データベースのリセット方法を追記

ankokuty · Oct 7, 2017 · f76f909 · f76f909
1 parent 0d66145
commit f76f909
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/src/main/java/badstore/StartBadStore.java b/src/main/java/badstore/StartBadStore.java
@@ -2,9 +2,25 @@
 
 public class StartBadStore {
 	public static void main(String[] args) throws Exception {
+		String host = "127.0.0.1";
+		int port = 8528;
+
 		Class.forName("org.sqlite.JDBC");
-		BadStore badstore = new BadStore(8528, "127.0.0.1", null);
 
+		try {
+			if (args[0] != null) {
+				host = args[0];
+			}
+		} catch (ArrayIndexOutOfBoundsException ignore) {
+		}
+
+		try {
+			port = Integer.valueOf(args[1]);
+		} catch (ArrayIndexOutOfBoundsException ignore) {
+		} catch (NumberFormatException ignore) {
+		}
+
+		BadStore badstore = new BadStore(port, host, null);
 		badstore.startServer(null);
 	}
 }
diff --git a/src/main/java/burp/BurpExtender.java b/src/main/java/burp/BurpExtender.java
@@ -22,8 +22,8 @@ public void registerExtenderCallbacks(IBurpExtenderCallbacks callbacks) {
 			callbacks.addSuiteTab(BurpExtender.this);
 		});
 
-		callbacks.printOutput("Hakoniwa BadStore is using " + BadStore.storeDir.getAbsolutePath() + " as a temporary directory.");
-		callbacks.printOutput("If something goes wrong, delete this directory and restart the BadStore server.");
+		callbacks.printOutput("When something goes wrong, access /cgi-bin/initdbs.cgi to reset the database.");
+		callbacks.printOutput("Or delete temporary directory (" + BadStore.storeDir.getAbsolutePath() + ") and restart the Hakoniwa BadStore server.");
 	}
 
 	@Override