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

Version: 2.5.1 Update

Browse Source
This commit is contained in:
ᴋᴇʏ
2023-10-18 15:14:33 +08:00
committed by GitHub
parent 6546446e4f
commit cf90a9366a
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/burp/BurpExtender.java
View File

@@ -127,7 +127,6 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
String allComment = !Objects.equals(originalComment, "") ? String.format("%s, %s", originalComment, addComment) : addComment; String allComment = !Objects.equals(originalComment, "") ? String.format("%s, %s", originalComment, addComment) : addComment;
resComment = mergeComment(allComment); resComment = mergeComment(allComment);
messageInfo.setComment(resComment);
} }
String endComment = resComment.isEmpty() ? originalComment : resComment; String endComment = resComment.isEmpty() ? originalComment : resComment;
@@ -140,12 +139,19 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
} }
private String mergeComment(String comment) { private String mergeComment(String comment) {
if (!comment.contains("(") || !comment.contains(")")) {
// 没有括号的情况，直接返回原始Comment
return comment;
}
Map<String, Integer> itemCounts = new HashMap<>(); Map<String, Integer> itemCounts = new HashMap<>();
String[] items = comment.split(", "); String[] items = comment.split(", ");
for (String item : items) { for (String item : items) {
String itemName = item.substring(0, item.lastIndexOf("(")).trim(); int openParenIndex = item.lastIndexOf("(");
int count = Integer.parseInt(item.substring(item.lastIndexOf("(") + 1, item.lastIndexOf(")")).trim()); int closeParenIndex = item.lastIndexOf(")");
String itemName = item.substring(0, openParenIndex).trim();
int count = Integer.parseInt(item.substring(openParenIndex + 1, closeParenIndex).trim());
itemCounts.put(itemName, itemCounts.getOrDefault(itemName, 0) + count); itemCounts.put(itemName, itemCounts.getOrDefault(itemName, 0) + count);
} }
@@ -289,4 +295,4 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
public IMessageEditorTab createNewInstance(IMessageEditorController controller, boolean editable) { public IMessageEditorTab createNewInstance(IMessageEditorController controller, boolean editable) {
return new MarkInfoTab(controller, editable); return new MarkInfoTab(controller, editable);
} }
} }