添加显示和隐藏Intruder模块识别结果,以便在极端情况下节省内存资源,防止burp卡住

Author: c0ny1

src/main/java/burp/BurpExtender.java

@@ -36,6 +36,7 @@ public class BurpExtender implements IBurpExtender,ITab,IIntruderPayloadGenerato
     public static IExtensionHelpers helpers;
     private String extensionName = "captcha-killer";
     private String version ="0.1.1";
+    public static boolean isShowIntruderResult = true; // 识别结果是否显示Intruder模块结果
     public static PrintWriter stdout;
     public static PrintWriter stderr;
     public static GUI gui;

src/main/java/burp/GeneratePayloadSwingWorker.java

@@ -27,11 +27,12 @@ protected Object doInBackground() throws Exception {
                     break;
                 }
             }
-
-            synchronized (BurpExtender.gui.captcha){
-                int row = BurpExtender.gui.captcha.size();
-                BurpExtender.gui.captcha.add(cap);
-                BurpExtender.gui.getModel().fireTableRowsInserted(row,row);
+            if(BurpExtender.isShowIntruderResult) {
+                synchronized (BurpExtender.gui.captcha) {
+                    int row = BurpExtender.gui.captcha.size();
+                    BurpExtender.gui.captcha.add(cap);
+                    BurpExtender.gui.getModel().fireTableRowsInserted(row, row);
+                }
             }
         } catch (Exception e) {
             cap.setResult(e.getMessage());

src/main/java/ui/GUI.java

@@ -69,6 +69,7 @@ public class GUI {
     private TableModel model;
     private JPopupMenu pppMenu = new JPopupMenu();
     private JMenuItem miClear = new JMenuItem("清空");
+    private JMenuItem miShowIntruderResult = new JMenuItem("关闭Intruder识别结果显示");
 
     //一些公共变量
     private byte[] byteImg;
@@ -254,6 +255,7 @@ public void initGUI(){
         //识别结果面板
         plResult = new JPanel();
         pppMenu.add(miClear);
+        pppMenu.add(miShowIntruderResult);
         table = new JTable();
         model = new TableModel(table);
         table.setModel(model);
@@ -719,6 +721,19 @@ public void actionPerformed(ActionEvent e) {
                 }
             }
         });
+
+        miShowIntruderResult.addActionListener(new ActionListener() {
+            @Override
+            public void actionPerformed(ActionEvent e) {
+                if(BurpExtender.isShowIntruderResult) {
+                    BurpExtender.isShowIntruderResult = false;
+                    miShowIntruderResult.setText("显示Intruder识别结果");
+                }else{
+                    BurpExtender.isShowIntruderResult = true;
+                    miShowIntruderResult.setText("隐藏Intruder识别结果");
+                }
+            }
+        });
     }