diff --git a/.DS_Store b/.DS_Store new file mode 100644 index 0000000..0ba057a Binary files /dev/null and b/.DS_Store differ diff --git a/HaE/HaE.py b/HaE/HaE.py index 32322cb..f276723 100644 --- a/HaE/HaE.py +++ b/HaE/HaE.py @@ -58,11 +58,9 @@ class BurpExtender(IBurpExtender, ITab,IHttpListener, IMessageEditorTabFactory): if messageIsRequest: return content = messageInfo.getResponse() - r = self._helpers.analyzeResponse(content) - msg = content[r.getBodyOffset():].tostring() - # msg 为响应正文信息 + # content 为响应正文信息 info = getConfig() - results = findContent(info, msg) + results = findContent(info, content) colorList = [] if results != {}: for i in results: @@ -205,13 +203,11 @@ class MarkINFOTab(IMessageEditorTab): # 非响应 没有匹配到不返回Tab标签页 def isEnabled(self, content, isRequest): - r = self._helpers.analyzeResponse(content) - msg = content[r.getBodyOffset():].tostring() info = getConfig() if not isRequest: - content = findContent(info, msg) - if content != {}: - for i in content: + contents = findContent(info, content) + if contents != {}: + for i in contents: if info[i]['extract'] == 1 : return True @@ -220,10 +216,8 @@ class MarkINFOTab(IMessageEditorTab): # 判断是否有内容 if content: if not isRequest: - r = self._helpers.analyzeResponse(content) - msg = content[r.getBodyOffset():].tostring() info = getConfig() - contents = findContent(info, msg) + contents = findContent(info, content) result = "" for i in contents: if info[i]['extract'] == 1 :