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

Version: 4.0.5 Update

Browse Source
This commit is contained in:
chen
2025-02-11 18:29:55 +08:00
parent 5977e82ca6
commit 0c795af101
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/hae/HaE.java
View File

@@ -20,7 +20,7 @@ public class HaE implements BurpExtension {
public void initialize(MontoyaApi api) { public void initialize(MontoyaApi api) {
// 设置扩展名称 // 设置扩展名称
api.extension().setName("HaE - Highlighter and Extractor"); api.extension().setName("HaE - Highlighter and Extractor");
String version = "4.0.4"; String version = "4.0.5";
// 加载扩展后输出的项目信息 // 加载扩展后输出的项目信息
Logging logging = api.logging(); Logging logging = api.logging();

12
src/main/java/hae/utils/DataManager.java
View File

@@ -24,10 +24,11 @@ public class DataManager {
persistence.extensionData().deleteChildObject(dataName); persistence.extensionData().deleteChildObject(dataName);
} }
persistence.extensionData().setChildObject(dataName, persistedObject); persistence.extensionData().setChildObject(dataName, persistedObject);
saveIndex(dataType, dataName); saveIndex(dataType, dataName);
} }
public void loadData(MessageTableModel messageTableModel) { public synchronized void loadData(MessageTableModel messageTableModel) {
// 1. 获取索引 // 1. 获取索引
PersistedList<String> dataIndex = persistence.extensionData().getStringList("data"); // 数据索引 PersistedList<String> dataIndex = persistence.extensionData().getStringList("data"); // 数据索引
PersistedList<String> messageIndex = persistence.extensionData().getStringList("message"); // 消息索引 PersistedList<String> messageIndex = persistence.extensionData().getStringList("message"); // 消息索引
@@ -42,7 +43,7 @@ public class DataManager {
if (indexList != null && !indexList.isEmpty()) { if (indexList != null && !indexList.isEmpty()) {
persistence.extensionData().deleteStringList(indexName); persistence.extensionData().deleteStringList(indexName);
} else { } else if (indexList == null) {
indexList = PersistedList.persistedStringList(); indexList = PersistedList.persistedStringList();
} }
@@ -55,14 +56,13 @@ public class DataManager {
private void loadHaEData(PersistedList<String> dataIndex) { private void loadHaEData(PersistedList<String> dataIndex) {
if (dataIndex != null && !dataIndex.isEmpty()) { if (dataIndex != null && !dataIndex.isEmpty()) {
dataIndex.parallelStream().forEach(index -> { dataIndex.forEach(index -> {
PersistedObject dataObj = persistence.extensionData().getChildObject(index); PersistedObject dataObj = persistence.extensionData().getChildObject(index);
try { try {
dataObj.stringListKeys().forEach(dataKey -> { dataObj.stringListKeys().forEach(dataKey -> {
RegularMatcher.putDataToGlobalMap(api, index, dataKey, dataObj.getStringList(dataKey).stream().toList(), false); RegularMatcher.putDataToGlobalMap(api, index, dataKey, dataObj.getStringList(dataKey).stream().toList(), false);
}); });
} catch (Exception e) { } catch (Exception ignored) {
// api.logging().logToOutput("loadHaEData:" + e.getMessage());
} }
}); });
} }
@@ -70,7 +70,7 @@ public class DataManager {
private void loadMessageData(PersistedList<String> messageIndex, MessageTableModel messageTableModel) { private void loadMessageData(PersistedList<String> messageIndex, MessageTableModel messageTableModel) {
if (messageIndex != null && !messageIndex.isEmpty()) { if (messageIndex != null && !messageIndex.isEmpty()) {
messageIndex.parallelStream().forEach(index -> { messageIndex.forEach(index -> {
PersistedObject dataObj = persistence.extensionData().getChildObject(index); PersistedObject dataObj = persistence.extensionData().getChildObject(index);
if (dataObj != null) { if (dataObj != null) {
HttpRequestResponse messageInfo = dataObj.getHttpRequestResponse("messageInfo"); HttpRequestResponse messageInfo = dataObj.getHttpRequestResponse("messageInfo");