admin/HaE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

支持完整响应报文匹配

Browse Source
This commit is contained in:
EvilOS
2020-03-25 17:50:09 +08:00
parent 7863d9640d
commit d568a7d653
2 changed files with 6 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
.DS_Store vendored Normal file
View File

Binary file not shown.

18
HaE/HaE.py
View File

@@ -58,11 +58,9 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory):
if messageIsRequest: if messageIsRequest:
return return
content = messageInfo.getResponse() content = messageInfo.getResponse()
r = self._helpers.analyzeResponse(content) # content 为响应正文信息
msg = content[r.getBodyOffset():].tostring()
# msg 为响应正文信息
info = getConfig() info = getConfig()
results = findContent(info, msg) results = findContent(info, content)
colorList = [] colorList = []
if results != {}: if results != {}:
for i in results: for i in results:
@@ -205,13 +203,11 @@ class MarkINFOTab(IMessageEditorTab):
# 非响应 没有匹配到不返回Tab标签页 # 非响应 没有匹配到不返回Tab标签页
def isEnabled(self, content, isRequest): def isEnabled(self, content, isRequest):
r = self._helpers.analyzeResponse(content)
msg = content[r.getBodyOffset():].tostring()
info = getConfig() info = getConfig()
if not isRequest: if not isRequest:
content = findContent(info, msg) contents = findContent(info, content)
if content != {}: if contents != {}:
for i in content: for i in contents:
if info[i]['extract'] == 1 : if info[i]['extract'] == 1 :
return True return True
@@ -220,10 +216,8 @@ class MarkINFOTab(IMessageEditorTab):
# 判断是否有内容 # 判断是否有内容
if content: if content:
if not isRequest: if not isRequest:
r = self._helpers.analyzeResponse(content)
msg = content[r.getBodyOffset():].tostring()
info = getConfig() info = getConfig()
contents = findContent(info, msg) contents = findContent(info, content)
result = "" result = ""
for i in contents: for i in contents:
if info[i]['extract'] == 1 : if info[i]['extract'] == 1 :