diff --git a/src/main/java/burp/BurpExtender.java b/src/main/java/burp/BurpExtender.java index d8aed77..10b2d38 100644 --- a/src/main/java/burp/BurpExtender.java +++ b/src/main/java/burp/BurpExtender.java @@ -29,7 +29,7 @@ public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEdito this.callbacks = callbacks; BurpExtender.helpers = callbacks.getHelpers(); - String version = "2.1.4"; + String version = "2.1.5"; callbacks.setExtensionName(String.format("HaE (%s) - Highlighter and Extractor", version)); // 定义输出 stdout = new PrintWriter(callbacks.getStdout(), true); diff --git a/src/main/java/burp/Config.java b/src/main/java/burp/Config.java index 9f3f356..c05113a 100644 --- a/src/main/java/burp/Config.java +++ b/src/main/java/burp/Config.java @@ -39,5 +39,5 @@ public class Config { "gray" }; - public static Map ruleConfig = LoadConfig.getRules(); + public static Map ruleConfig = null; } \ No newline at end of file diff --git a/src/main/java/burp/yaml/LoadConfig.java b/src/main/java/burp/yaml/LoadConfig.java index 2af1344..fafa7cf 100644 --- a/src/main/java/burp/yaml/LoadConfig.java +++ b/src/main/java/burp/yaml/LoadConfig.java @@ -27,6 +27,7 @@ public class LoadConfig { initSetting(); initRules(); } + Config.ruleConfig = LoadConfig.getRules(); } // 初始化设置信息 @@ -91,14 +92,21 @@ public class LoadConfig { // 获取不包含的后缀名 public String getExcludeSuffix(){ - try { - InputStream inorder = new FileInputStream(SettingPath); - Map r = yaml.load(inorder); - return r.get("excludeSuffix").toString(); - } catch (FileNotFoundException e) { - e.printStackTrace(); - return Config.excludeSuffix; + String excludeSuffix = ""; + File yamlSetting = new File(SettingPath); + if (yamlSetting.exists() && yamlSetting.isFile()) { + try { + InputStream inorder = new FileInputStream(SettingPath); + Map r = yaml.load(inorder); + excludeSuffix = r.get("excludeSuffix").toString(); + } catch (Exception e) { + // e.printStackTrace(); + excludeSuffix = ""; + } + } else { + excludeSuffix = Config.excludeSuffix; } + return excludeSuffix; } // 获取规则配置