Version: 2.5.1 Update
This commit is contained in:
ᴋᴇʏ
292
BurpExtender.java
Normal file
292
BurpExtender.java
Normal file
@@ -0,0 +1,292 @@
|
|||||||
|
package burp;
|
||||||
|
|
||||||
|
import burp.core.processor.ColorProcessor;
|
||||||
|
import burp.core.processor.MessageProcessor;
|
||||||
|
import burp.ui.MainUI;
|
||||||
|
import burp.ui.board.MessagePanel;
|
||||||
|
import java.security.NoSuchAlgorithmException;
|
||||||
|
import java.util.*;
|
||||||
|
import javax.swing.*;
|
||||||
|
import java.awt.*;
|
||||||
|
import java.io.PrintWriter;
|
||||||
|
import java.util.List;
|
||||||
|
import javax.swing.event.ChangeEvent;
|
||||||
|
import javax.swing.event.ChangeListener;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author EvilChen & 0chencc
|
||||||
|
*/
|
||||||
|
|
||||||
|
public class BurpExtender implements IBurpExtender, IHttpListener, IMessageEditorTabFactory, ITab {
|
||||||
|
private MainUI main;
|
||||||
|
// stdout变成公开属性，便于其他类调用输出调试信息
|
||||||
|
public static PrintWriter stdout;
|
||||||
|
private IBurpExtenderCallbacks callbacks;
|
||||||
|
private static IExtensionHelpers helpers;
|
||||||
|
ColorProcessor colorProcessor = new ColorProcessor();
|
||||||
|
MessageProcessor messageProcessor = new MessageProcessor();
|
||||||
|
private MessagePanel messagePanel;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void registerExtenderCallbacks(final IBurpExtenderCallbacks callbacks)
|
||||||
|
{
|
||||||
|
this.callbacks = callbacks;
|
||||||
|
BurpExtender.helpers = callbacks.getHelpers();
|
||||||
|
|
||||||
|
String version = "2.5.1";
|
||||||
|
callbacks.setExtensionName(String.format("HaE (%s) - Highlighter and Extractor", version));
|
||||||
|
|
||||||
|
// 定义输出
|
||||||
|
stdout = new PrintWriter(callbacks.getStdout(), true);
|
||||||
|
stdout.println("[ HACK THE WORLD - TO DO IT ]");
|
||||||
|
stdout.println("[#] Author: EvilChen & 0chencc");
|
||||||
|
stdout.println("[#] Github: https://github.com/gh0stkey/HaE");
|
||||||
|
|
||||||
|
// UI
|
||||||
|
SwingUtilities.invokeLater(new Runnable() {
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
initialize();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
callbacks.registerHttpListener(BurpExtender.this);
|
||||||
|
callbacks.registerMessageEditorTabFactory(BurpExtender.this);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private void initialize(){
|
||||||
|
messagePanel = new MessagePanel(callbacks, helpers);
|
||||||
|
main = new MainUI(messagePanel);
|
||||||
|
callbacks.customizeUiComponent(main);
|
||||||
|
callbacks.addSuiteTab(BurpExtender.this);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getTabCaption(){
|
||||||
|
return "HaE";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Component getUiComponent() {
|
||||||
|
return main;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 使用processHttpMessage用来做Highlighter
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void processHttpMessage(int toolFlag, boolean messageIsRequest, IHttpRequestResponse messageInfo) {
|
||||||
|
// 判断是否是响应，且该代码作用域为：REPEATER、INTRUDER、PROXY（分别对应toolFlag 64、32、4）
|
||||||
|
if (toolFlag == 64 || toolFlag == 32 || toolFlag == 4) {
|
||||||
|
byte[] content;
|
||||||
|
|
||||||
|
if (messageIsRequest) {
|
||||||
|
content = messageInfo.getRequest();
|
||||||
|
} else {
|
||||||
|
content = messageInfo.getResponse();
|
||||||
|
}
|
||||||
|
|
||||||
|
IHttpService iHttpService = null;
|
||||||
|
|
||||||
|
String host = "";
|
||||||
|
|
||||||
|
try {
|
||||||
|
iHttpService = messageInfo.getHttpService();
|
||||||
|
host = iHttpService.getHost();
|
||||||
|
} catch (Exception ignored) {
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Objects.equals(host, "")) {
|
||||||
|
List<String> requestTmpHeaders = helpers.analyzeRequest(content).getHeaders();
|
||||||
|
host = requestTmpHeaders.get(1).split(":")[1].trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
List<Map<String, String>> result = null;
|
||||||
|
try {
|
||||||
|
result = messageProcessor.processMessage(helpers, content, messageIsRequest, true, host);
|
||||||
|
} catch (NoSuchAlgorithmException e) {
|
||||||
|
throw new RuntimeException(e);
|
||||||
|
}
|
||||||
|
String resComment = "";
|
||||||
|
String resColor = "";
|
||||||
|
String originalColor = messageInfo.getHighlight();
|
||||||
|
String originalComment = messageInfo.getComment();
|
||||||
|
if (result != null && !result.isEmpty() && result.size() > 0) {
|
||||||
|
List<String> colorList = new ArrayList<>();
|
||||||
|
|
||||||
|
if (originalColor != null) {
|
||||||
|
colorList.add(originalColor);
|
||||||
|
}
|
||||||
|
|
||||||
|
colorList.add(result.get(0).get("color"));
|
||||||
|
resColor = colorProcessor.retrieveFinalColor(colorProcessor.retrieveColorIndices(colorList));
|
||||||
|
messageInfo.setHighlight(resColor);
|
||||||
|
|
||||||
|
String addComment = String.join(", ", result.get(1).get("comment"));
|
||||||
|
String allComment = !Objects.equals(originalComment, "") ? String.format("%s, %s", originalComment, addComment) : addComment;
|
||||||
|
resComment = mergeComment(allComment);
|
||||||
|
|
||||||
|
messageInfo.setComment(resComment);
|
||||||
|
}
|
||||||
|
|
||||||
|
String endComment = resComment.isEmpty() ? originalComment : resComment;
|
||||||
|
String endColor = resColor.isEmpty() ? originalColor : resColor;
|
||||||
|
|
||||||
|
if (!messageIsRequest && !Objects.equals(endComment, "") && !Objects.equals(endColor, "")) {
|
||||||
|
messagePanel.add(messageInfo, endComment, String.valueOf(content.length), endColor);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private String mergeComment(String comment) {
|
||||||
|
Map<String, Integer> itemCounts = new HashMap<>();
|
||||||
|
String[] items = comment.split(", ");
|
||||||
|
|
||||||
|
for (String item : items) {
|
||||||
|
String itemName = item.substring(0, item.lastIndexOf("(")).trim();
|
||||||
|
int count = Integer.parseInt(item.substring(item.lastIndexOf("(") + 1, item.lastIndexOf(")")).trim());
|
||||||
|
itemCounts.put(itemName, itemCounts.getOrDefault(itemName, 0) + count);
|
||||||
|
}
|
||||||
|
|
||||||
|
StringBuilder mergedItems = new StringBuilder();
|
||||||
|
|
||||||
|
for (Map.Entry<String, Integer> entry : itemCounts.entrySet()) {
|
||||||
|
String itemName = entry.getKey();
|
||||||
|
int count = entry.getValue();
|
||||||
|
|
||||||
|
mergedItems.append(itemName).append(" (").append(count).append("), ");
|
||||||
|
}
|
||||||
|
|
||||||
|
return mergedItems.substring(0, mergedItems.length() - 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
class MarkInfoTab implements IMessageEditorTab {
|
||||||
|
private final JTabbedPane jTabbedPane = new JTabbedPane();
|
||||||
|
private JTable jTable = new JTable();
|
||||||
|
private final IMessageEditorController controller;
|
||||||
|
private Map<String, String> extractRequestMap;
|
||||||
|
private Map<String, String> extractResponseMap;
|
||||||
|
private ArrayList<String> titleList = new ArrayList<>();
|
||||||
|
|
||||||
|
public MarkInfoTab(IMessageEditorController controller, boolean editable) {
|
||||||
|
this.controller = controller;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getTabCaption() {
|
||||||
|
return "MarkInfo";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Component getUiComponent() {
|
||||||
|
jTabbedPane.addChangeListener(new ChangeListener() {
|
||||||
|
@Override
|
||||||
|
public void stateChanged(ChangeEvent arg0) {
|
||||||
|
jTable = (JTable) ((JScrollPane)jTabbedPane.getSelectedComponent()).getViewport().getView();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return this.jTabbedPane;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isEnabled(byte[] content, boolean isRequest) {
|
||||||
|
List<Map<String, String>> result = null;
|
||||||
|
|
||||||
|
try {
|
||||||
|
result = messageProcessor.processMessage(helpers, content, isRequest, false, "");
|
||||||
|
} catch (NoSuchAlgorithmException e) {
|
||||||
|
throw new RuntimeException(e);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (result != null && !result.isEmpty()) {
|
||||||
|
Map<String, String> dataMap = result.get(0);
|
||||||
|
if (isRequest) {
|
||||||
|
extractRequestMap = dataMap;
|
||||||
|
} else {
|
||||||
|
extractResponseMap = dataMap;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public byte[] getMessage() {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isModified() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 快捷键复制功能
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public byte[] getSelectedData() {
|
||||||
|
int[] selectRows = jTable.getSelectedRows();
|
||||||
|
StringBuilder selectData = new StringBuilder();
|
||||||
|
for (int row : selectRows) {
|
||||||
|
selectData.append(jTable.getValueAt(row, 0).toString()).append("\n");
|
||||||
|
}
|
||||||
|
// 便于单行复制，去除最后一个换行符
|
||||||
|
String revData = selectData.reverse().toString().replaceFirst("\n", "");
|
||||||
|
StringBuilder retData = new StringBuilder(revData).reverse();
|
||||||
|
return helpers.stringToBytes(retData.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 使用setMessage用来做Extractor
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void setMessage(byte[] content, boolean isRequest) {
|
||||||
|
if (content.length > 0) {
|
||||||
|
if (isRequest) {
|
||||||
|
makeTable(extractRequestMap);
|
||||||
|
} else {
|
||||||
|
makeTable(extractResponseMap);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建MarkInfo表单
|
||||||
|
*/
|
||||||
|
public void makeTable(Map<String, String> dataMap) {
|
||||||
|
ArrayList<String> lTitleList = new ArrayList<>();
|
||||||
|
dataMap.keySet().forEach(i->{
|
||||||
|
String[] extractData = dataMap.get(i).split("\n");
|
||||||
|
Object[][] data = new Object[extractData.length][1];
|
||||||
|
for (int x = 0; x < extractData.length; x++) {
|
||||||
|
data[x][0] = extractData[x];
|
||||||
|
}
|
||||||
|
JTable infoTable = new JTable(data, new Object[]{"Information"});
|
||||||
|
infoTable.setAutoCreateRowSorter(true);
|
||||||
|
JScrollPane jScrollPane = new JScrollPane(infoTable);
|
||||||
|
|
||||||
|
lTitleList.add(i);
|
||||||
|
this.jTabbedPane.addTab(i, jScrollPane);
|
||||||
|
});
|
||||||
|
|
||||||
|
/*
|
||||||
|
* 使用removeAll会导致MarkInfo UI出现空白的情况，为了改善用户侧体验，采用remove的方式进行删除；
|
||||||
|
* 采用全局ArrayList的方式遍历删除Tab，以此应对BurpSuite缓存机制导致的MarkInfo UI错误展示。
|
||||||
|
*/
|
||||||
|
titleList.forEach(t->{
|
||||||
|
int indexOfTab = this.jTabbedPane.indexOfTab(t);
|
||||||
|
if (indexOfTab != -1) {
|
||||||
|
this.jTabbedPane.removeTabAt(indexOfTab);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
titleList = lTitleList;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public IMessageEditorTab createNewInstance(IMessageEditorController controller, boolean editable) {
|
||||||
|
return new MarkInfoTab(controller, editable);
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user