Refactor custom Editor Tabs

usdAG · Jul 16, 2024 · 9e2a885 · 9e2a885
1 parent 532d68d
commit 9e2a885
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 100 deletions.
diff --git a/src/main/java/burp/BurpExtender.java b/src/main/java/burp/BurpExtender.java
@@ -26,7 +26,6 @@ public void initialize(MontoyaApi api) {
         api.http().registerHttpHandler(new CstcHttpHandler(view));
         api.userInterface().registerSuiteTab(extensionName, view);
         api.userInterface().registerHttpRequestEditorProvider(new MyHttpRequestEditorProvider(view));
-        api.userInterface().registerHttpResponseEditorProvider(new MyHttpResponseEditorProvider(view));
 
         if (!api.burpSuite().version().edition().equals(BurpSuiteEdition.COMMUNITY_EDITION)) {
             PersistedObject persistence = api.persistence().extensionData();

diff --git a/src/main/java/burp/MyExtensionProvidedHttpRequestEditor.java b/src/main/java/burp/MyExtensionProvidedHttpRequestEditor.java
@@ -2,8 +2,10 @@
 
 import burp.api.montoya.MontoyaApi;
 import burp.api.montoya.core.ByteArray;
+import burp.api.montoya.core.ToolType;
 import burp.api.montoya.http.message.HttpRequestResponse;
 import burp.api.montoya.http.message.requests.HttpRequest;
+import burp.api.montoya.repeater.Repeater;
 import burp.api.montoya.ui.Selection;
 import burp.api.montoya.ui.editor.EditorOptions;
 import burp.api.montoya.ui.editor.RawEditor;
@@ -23,9 +25,16 @@ public class MyExtensionProvidedHttpRequestEditor implements ExtensionProvidedHt
 
     MyExtensionProvidedHttpRequestEditor(EditorCreationContext creationContext, View view)
     {
-        this.api = BurpUtils.getInstance().getApi();
-        this.view = view;
-        requestEditor = api.userInterface().createRawEditor(EditorOptions.READ_ONLY);
+        if(creationContext.toolSource().isFromTool(ToolType.REPEATER)) {
+            this.api = BurpUtils.getInstance().getApi();
+            this.view = view;
+            requestEditor = api.userInterface().createRawEditor(EditorOptions.READ_ONLY);
+        }
+        else {
+            this.api = null;
+            this.view = null;
+            this.requestEditor = null;
+        }
     }
 
     @Override

diff --git a/src/main/java/burp/MyExtensionProvidedHttpResponseEditor.java b/src/main/java/burp/MyExtensionProvidedHttpResponseEditor.java
diff --git a/src/main/java/burp/MyHttpResponseEditorProvider.java b/src/main/java/burp/MyHttpResponseEditorProvider.java