From cf90a9366a4cd84a15f5976c96a6f44f7c99b533 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E1=B4=8B=E1=B4=87=CA=8F?=
 <24655118+gh0stkey@users.noreply.github.com>
Date: Wed, 18 Oct 2023 15:14:33 +0800
Subject: [PATCH] Version: 2.5.1 Update

---
 src/main/java/burp/BurpExtender.java | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/main/java/burp/BurpExtender.java b/src/main/java/burp/BurpExtender.java
index 2054e95..e33acea 100644
--- a/src/main/java/burp/BurpExtender.java
+++ b/src/main/java/burp/BurpExtender.java
@@ -127,7 +127,6 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
                 String allComment = !Objects.equals(originalComment, "") ? String.format("%s, %s", originalComment, addComment) : addComment;
                 resComment = mergeComment(allComment);
 
-                messageInfo.setComment(resComment);
             }
 
             String endComment = resComment.isEmpty() ? originalComment : resComment;
@@ -140,12 +139,19 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
     }
 
     private String mergeComment(String comment) {
+        if (!comment.contains("(") || !comment.contains(")")) {
+            // 没有括号的情况，直接返回原始Comment
+            return comment;
+        }
+
         Map<String, Integer> itemCounts = new HashMap<>();
         String[] items = comment.split(", ");
 
         for (String item : items) {
-            String itemName = item.substring(0, item.lastIndexOf("(")).trim();
-            int count = Integer.parseInt(item.substring(item.lastIndexOf("(") + 1, item.lastIndexOf(")")).trim());
+            int openParenIndex = item.lastIndexOf("(");
+            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);
         }
 
@@ -289,4 +295,4 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito
     public IMessageEditorTab createNewInstance(IMessageEditorController controller, boolean editable) {
         return new MarkInfoTab(controller, editable);
     }
-}
\ No newline at end of file
+}