往期文章:
0x01、前言 |
|
--->
|
--->
这里burp插件的功能很简单,它只负责将HTTP history的数据包保存到指定的文件夹。
# -*- coding: utf-8 -*-
from burp import IBurpExtender
from burp import IContextMenuFactory
from burp import IHttpListener
from java.io import File
from java.text import SimpleDateFormat
from java.util import Date
import time
import hashlib
import urlparse
class BurpExtender(IBurpExtender, IHttpListener, IContextMenuFactory):
def registerExtenderCallbacks(self, callbacks):
self.callbacks = callbacks
callbacks.setExtensionName("Auto Save HTTP History with In-Scope Filter")
callbacks.registerHttpListener(self)
self.save_directory = File("Z:/burp_history_xmls/")
if not self.save_directory.exists():
self.save_directory.mkdirs()
self.saved_requests = set()
def processHttpMessage(self, toolFlag, messageIsRequest, messageInfo):
if not self.isInScope(messageInfo):
return
request = messageInfo.getRequest()
response = messageInfo.getResponse()
status_code = self.getStatusCode(response)
request_hash = self.getRequestHash(request)
if request_hash in self.saved_requests:
return
else:
self.saved_requests.add(request_hash)
timestamp = self.getTimestamp()
file_name = "response_" + timestamp + "_status_" + str(status_code) + ".txt"
file_path = File(self.save_directory, file_name)
with open(file_path.getAbsolutePath(), "wb") as file:
file.write(request)
file.write(b"rnrn")
file.write(response)
def getTimestamp(self):
current_time = SimpleDateFormat("yyyyMMdd_HHmmss").format(Date())
return current_time
def isInScope(self, messageInfo):
return self.callbacks.isInScope(messageInfo.getUrl())
def getStatusCode(self, response):
if response is None:
return None
response_headers = self.callbacks.getHelpers().analyzeResponse(response).getHeaders()
if response_headers:
status_line = response_headers[0]
status_code = status_line.split(" ")[1]
return status_code
return None
def getRequestHash(self, request):
return hashlib.sha256(request).hexdigest()
|
--->
当burp有新数据包更新后,插件都会实时保存到指定目录,当AI大模型检测到有数据包更新时,会自动进行分析。
原文始发于微信公众号(偏远酒馆):如何使用AI进行漏洞挖掘?完结!
免责声明:文章中涉及的程序(方法)可能带有攻击性,仅供安全研究与教学之用,读者将其信息做其他用途,由读者承担全部法律及连带责任,本站不承担任何法律及连带责任;如有问题可邮件联系(建议使用企业邮箱或有效邮箱,避免邮件被拦截,联系方式见首页),望知悉。
- 左青龙
- 微信扫一扫
-
- 右白虎
- 微信扫一扫
-
评论