Delete BurpExtender.java
This commit is contained in:
ᴋᴇʏ
@@ -1,292 +0,0 @@
|
|||||||
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