admin/HaE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #235 from 0Chencc/master

Browse Source 
修复了社区版会无法正常使用HaE的问题
This commit is contained in:
EvilChen
2025-01-08 13:47:38 +08:00
committed by GitHub
parent e83a6a1478 b686b5e75e
commit 7e53e250af
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/main/java/hae/component/Config.java
View File

@@ -29,6 +29,7 @@ public class Config extends JPanel {
private final ConfigLoader configLoader;
private final MessageTableModel messageTableModel;
private final Rules rules;
private final boolean isProfessionalVersion;
private Registration activeHandler;
private Registration passiveHandler;
@@ -38,9 +39,15 @@ public class Config extends JPanel {
this.configLoader = configLoader;
this.messageTableModel = messageTableModel;
this.rules = rules;
// 检查版本并记录日志
this.isProfessionalVersion = api.burpSuite().version().name().contains("Professional");
api.logging().logToOutput("Current Burp Suite Version: " + api.burpSuite().version().name());
this.activeHandler = api.http().registerHttpHandler(new HttpMessageActiveHandler(api, configLoader, messageTableModel));
this.passiveHandler = api.scanner().registerScanCheck(new HttpMessagePassiveHandler(api, configLoader, messageTableModel));
if (isProfessionalVersion) {
this.passiveHandler = api.scanner().registerScanCheck(new HttpMessagePassiveHandler(api, configLoader, messageTableModel));
}
initComponents();
}
@@ -379,7 +386,7 @@ public class Config extends JPanel {
configLoader.setMode(selected ? "true" : "false");
if (checkBox.isSelected()) {
if (passiveHandler.isRegistered()) {
if (isProfessionalVersion && passiveHandler.isRegistered()) {
passiveHandler.deregister();
}
@@ -387,7 +394,7 @@ public class Config extends JPanel {
activeHandler = api.http().registerHttpHandler(new HttpMessageActiveHandler(api, configLoader, messageTableModel));
}
} else {
if (!passiveHandler.isRegistered()) {
if (isProfessionalVersion && !passiveHandler.isRegistered()) {
passiveHandler = api.scanner().registerScanCheck(new HttpMessagePassiveHandler(api, configLoader, messageTableModel));
}