Fixes #4

righettod · Oct 7, 2018 · cc15f40 · cc15f40
1 parent b750c67
commit cc15f40
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/build.gradle b/build.gradle
@@ -1,7 +1,7 @@
 apply plugin: 'java'
 
 group 'eu.righettod'
-version '1.0.2'
+version '1.0.3'
 def burpExtensionHomepage = 'https://github.com/righettod/log-requests-to-sqlite'
 def burpExtensionJarName = 'LogRequestsToSQLite.jar'
 

diff --git a/src/burp/BurpExtender.java b/src/burp/BurpExtender.java
@@ -12,11 +12,12 @@ public class BurpExtender implements IBurpExtender {
      * {@inheritDoc}
      */
     public void registerExtenderCallbacks(IBurpExtenderCallbacks callbacks) {
+        ConfigMenu configMenu = null;
         try {
             String extensionName = "LogRequestsToSQLite";
             callbacks.setExtensionName(extensionName);
             Trace trace = new Trace(callbacks);
-            ConfigMenu configMenu = new ConfigMenu(callbacks, trace);
+            configMenu = new ConfigMenu(callbacks, trace);
             SwingUtilities.invokeLater(configMenu);
             String storeFileName = new File(System.getProperty("user.home"), extensionName + ".db").getAbsolutePath().replaceAll("\\\\", "/");
             ActivityLogger activityLogger = new ActivityLogger(storeFileName, callbacks, trace);
@@ -26,6 +27,10 @@ public void registerExtenderCallbacks(IBurpExtenderCallbacks callbacks) {
             callbacks.registerExtensionStateListener(configMenu);
         } catch (Exception e) {
             callbacks.issueAlert("Cannot start the extension: " + e.getMessage());
+            //Unload the menu if the extension cannot be loaded
+            if (configMenu != null) {
+                configMenu.extensionUnloaded();
+            }
         }
     }
 }