修复已知bug

Author: Chave0v0

src/main/java/com/chave/editor/RequestEditor.java

@@ -12,6 +12,7 @@
 import com.chave.pojo.APIItem;
 import com.chave.service.APIMatchService;
 import com.chave.service.SensitiveInfoMatchService;
+import com.chave.utils.Util;
 
 import javax.swing.*;
 import javax.swing.table.DefaultTableCellRenderer;
@@ -33,7 +34,6 @@ public ExtensionProvidedHttpRequestEditor provideHttpRequestEditor(EditorCreatio
     private static class Editor implements ExtensionProvidedHttpRequestEditor {
         private HttpRequestResponse requestResponse;
         private final JTabbedPane jTabbedPane = new JTabbedPane();
-        private APIMatchService apiMatchService = new APIMatchService();
         private SensitiveInfoMatchService sensitiveInfoMatchService = new SensitiveInfoMatchService();
 
         public Editor() {
@@ -53,8 +53,7 @@ public void setRequestResponse(HttpRequestResponse requestResponse) {
         public boolean isEnabledFor(HttpRequestResponse requestResponse) {
             HttpRequest request = requestResponse.request();
             try {
-                Method matchMethod = APIMatchService.class.getMethod(UserConfig.MATCH_MOD.name(), HttpRequest.class);
-                HashMap apiMatchResult = (HashMap) matchMethod.invoke(apiMatchService, request);
+                HashMap apiMatchResult = Util.getAPIMatchResult(request);
                 boolean isMatched = (boolean) apiMatchResult.get("isMatched");
                 if (isMatched && SensitiveInfoConfig.IS_CHECK_SENSITIVE_INFO) {
                     HashMap result = sensitiveInfoMatchService.sensitiveInfoMatch(request);

src/main/java/com/chave/editor/ResponseEditor.java

@@ -7,17 +7,15 @@
 import burp.api.montoya.ui.editor.extension.*;
 import com.chave.Main;
 import com.chave.config.SensitiveInfoConfig;
-import com.chave.config.UserConfig;
-import com.chave.service.APIMatchService;
 import com.chave.service.SensitiveInfoMatchService;
+import com.chave.utils.Util;
 
 import javax.swing.*;
 import javax.swing.table.DefaultTableCellRenderer;
 import javax.swing.table.DefaultTableModel;
 import javax.swing.table.JTableHeader;
 import javax.swing.table.TableCellRenderer;
 import java.awt.*;
-import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.Set;
@@ -32,7 +30,6 @@ public ExtensionProvidedHttpResponseEditor provideHttpResponseEditor(EditorCreat
     private static class Editor implements ExtensionProvidedHttpResponseEditor {
         private HttpRequestResponse requestResponse;
         private final JTabbedPane jTabbedPane = new JTabbedPane();
-        private APIMatchService apiMatchService = new APIMatchService();
         private SensitiveInfoMatchService sensitiveInfoMatchService = new SensitiveInfoMatchService();
 
         public Editor() {
@@ -53,8 +50,7 @@ public boolean isEnabledFor(HttpRequestResponse requestResponse) {
             HttpResponse response = requestResponse.response();
             HttpRequest request = requestResponse.request();
             try {
-                Method matchMethod = APIMatchService.class.getMethod(UserConfig.MATCH_MOD.name(), HttpRequest.class);
-                HashMap apiMatchResult = (HashMap) matchMethod.invoke(apiMatchService, request);
+                HashMap apiMatchResult = Util.getAPIMatchResult(request);
                 boolean isMatched = (boolean) apiMatchResult.get("isMatched");
                 if (isMatched && SensitiveInfoConfig.IS_CHECK_SENSITIVE_INFO) {
                     HashMap result = sensitiveInfoMatchService.sensitiveInfoMatch(response);