diff --git a/src/main/java/burp/BurpExtender.java b/src/main/java/burp/BurpExtender.java index 708089e..596676c 100644 --- a/src/main/java/burp/BurpExtender.java +++ b/src/main/java/burp/BurpExtender.java @@ -120,34 +120,15 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito public void processHttpMessage(int toolFlag, boolean messageIsRequest, IHttpRequestResponse messageInfo) { // 判断是否是响应,且该代码作用域为:REPEATER、INTRUDER、PROXY(分别对应toolFlag 64、32、4) if (toolFlag == 64 || toolFlag == 32 || toolFlag == 4) { - byte[] content; - - if (messageIsRequest) { - content = messageInfo.getRequest(); - } else { - content = messageInfo.getResponse(); - } - - IHttpService iHttpService = null; - - String host = ""; - - try { - iHttpService = messageInfo.getHttpService(); - host = iHttpService.getHost(); - } catch (Exception ignored) { - } - - if (Objects.equals(host, "")) { - host = helpers.analyzeRequest(content).getUrl().getHost(); - } - - List> result = null; - - String originalColor = messageInfo.getHighlight(); - String originalComment = messageInfo.getComment(); - if (!messageIsRequest) { + IHttpService iHttpService = messageInfo.getHttpService(); + String host = iHttpService.getHost(); + + List> result = null; + + String originalColor = messageInfo.getHighlight(); + String originalComment = messageInfo.getComment(); + try { result = messageProcessor.processMessage(helpers, messageInfo, host, true); @@ -167,7 +148,7 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito String resComment = mergeComment(allComment); messageInfo.setComment(resComment); - messagePanel.add(messageInfo, resComment, String.valueOf(content.length), resColor); + messagePanel.add(messageInfo, resComment, resColor); } } catch (Exception e) { e.printStackTrace(); diff --git a/src/main/java/burp/ui/board/MessagePanel.java b/src/main/java/burp/ui/board/MessagePanel.java index 46fba32..728dc22 100644 --- a/src/main/java/burp/ui/board/MessagePanel.java +++ b/src/main/java/burp/ui/board/MessagePanel.java @@ -308,13 +308,13 @@ public class MessagePanel extends AbstractTableModel implements IMessageEditorCo return currentlyDisplayedItem.getHttpService(); } - public void add(IHttpRequestResponse messageInfo, String comment, String length, String color) { + public void add(IHttpRequestResponse messageInfo, String comment, String color) { synchronized(log) { IRequestInfo iRequestInfo = helpers.analyzeRequest(messageInfo); URL url = iRequestInfo.getUrl(); String method = iRequestInfo.getMethod(); String status = String.valueOf(helpers.analyzeResponse(messageInfo.getResponse()).getStatusCode()); - + String length = String.valueOf(messageInfo.getResponse().length); LogEntry logEntry = new LogEntry(callbacks.saveBuffersToTempFiles(messageInfo), method, url, comment, length, color, status); try {