Burp插件开发：实现数据包自定义修改

背景与需求

在安全测试工作中，经常遇到加密、加签或添加了其他安全措施的数据包。为了测试正常进行，需要解密后修改数据包再进行加密还原。手工操作繁琐，因此需要开发Burp插件实现自动化处理。

主要需求：

1. 在Burp收发数据包的生命周期各个环节都能进行修改
2. 使用Python脚本完成实际修改，实现高度可定制化
3. Python脚本修改效果实时反映在插件中，方便调试
4. 对不同测试项目的数据包实现自定义拦截修改
5. 支持不发送数据包情况下对选定文本的自定义修改

Burp数据包处理流程

Burp作为中间代理转发数据包的流程：

1. 客户端请求 → Burp Proxy模块（可拦截） → Burp请求引擎 → 发送到目标服务器
2. 服务器响应 → Burp响应引擎 → Burp Proxy模块（可拦截） → 返回客户端

技术实现方案

1. 使用Montoya API处理数据包生命周期

Burp的新API Montoya API提供了在数据包流转各环节插入处理逻辑的接口：

// 示例：处理Proxy接收的客户端请求
public class MyHttpHandler implements HttpHandler {
    @Override
    public RequestHandledAction handleHttpRequest(HttpRequestToBeSent request) {
        // 自定义处理逻辑
        return RequestHandledAction.continueWith(request);
    }
}

// 注册处理器
montoya.proxy().registerRequestHandler(new MyHttpHandler());

2. Java与Python通信实现

使用Pyrolite库实现Java与Python的RPC通信：

Java端实现：

1. 添加依赖：

<dependency>
    <groupId>com.thoughtworks.xstream</groupId>
    <artifactId>xstream</artifactId>
    <version>1.4.20</version>
</dependency>
<dependency>
    <groupId>net.sf.py4j</groupId>
    <artifactId>py4j</artifactId>
    <version>0.10.9.5</version>
</dependency>

2. 客户端连接服务端：

PyroProxy pyroProxy = new PyroProxy("localhost", 9999);

3. 调用Python方法：

String result = (String) pyroProxy.call("invoke_method", 
    "custom_method", 
    Base64.getEncoder().encodeToString(request.getBytes()));

Python服务端实现：

from burpy import Burpy

class BurpyHandler:
    def invoke_method(self, method_name, encoded_data):
        data = base64.b64decode(encoded_data).decode()
        # 调用对应方法处理数据
        result = getattr(Burpy(), method_name)(data)
        return base64.b64encode(result.encode()).decode()

3. Python脚本热加载

使用watchdog监控文件变化，实现不重启服务的脚本热加载：

from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler

class ScriptReloadHandler(FileSystemEventHandler):
    def on_modified(self, event):
        if event.src_path.endswith('.py'):
            # 重新加载脚本
            import importlib
            import sys
            module_name = os.path.basename(event.src_path)[:-3]
            if module_name in sys.modules:
                importlib.reload(sys.modules[module_name])

observer = Observer()
observer.schedule(ScriptReloadHandler(), path='scripts/', recursive=True)
observer.start()

4. 多项目配置管理

通过配置文件区分不同项目的处理逻辑：

配置文件示例 (config.json):

{
    "project1": {
        "script": "project1_script.burpy",
        "intercept_request": true,
        "intercept_response": false
    },
    "project2": {
        "script": "project2_script.burpy",
        "intercept_request": false,
        "intercept_response": true
    }
}

Java加载配置：

JSONObject config = new JSONObject(new JSONTokener(new FileReader("config.json")));
String scriptPath = config.getJSONObject("project1").getString("script");

5. 手动处理选定文本

在Repeater等模块中右键处理选定文本：

public class CustomContextMenu implements ContextMenuItemsProvider {
    @Override
    public List<Component> provideMenuItems(ContextMenuEvent event) {
        String selectedText = event.getSelectedText();
        return List.of(new JMenuItem(new AbstractAction("Process with Python") {
            @Override
            public void actionPerformed(ActionEvent e) {
                String processed = pyroProxy.call("process_text", selectedText);
                event.replaceSelectedText(processed);
            }
        }));
    }
}

Python脚本模板

class Burpy:
    def process_request(self, request):
        """处理请求数据包"""
        # 解密逻辑
        decrypted = self.decrypt(request)
        # 修改逻辑
        modified = decrypted.replace("old", "new")
        # 加密逻辑
        encrypted = self.encrypt(modified)
        return encrypted
        
    def process_response(self, response):
        """处理响应数据包"""
        # 处理逻辑
        return response
        
    def process_text(self, text):
        """处理选中的文本"""
        return text.upper()
        
    def decrypt(self, data):
        # 实现解密算法
        return data
        
    def encrypt(self, data):
        # 实现加密算法
        return data

下一步计划

1. 重构Java与Python的通信机制，替换Pyrolite
2. 开发Python库，预设常用加解密、签名算法模板
3. 优化高并发场景下的稳定性
4. 添加图形化配置界面

总结

本方案实现了：

• 基于Montoya API的Burp插件开发
• 通过Pyrolite实现Java与Python的RPC通信
• Python脚本热加载机制
• 多项目配置管理
• 手动处理选定文本功能

这种架构将固定的Java插件框架与灵活的Python处理逻辑分离，既保持了Burp插件的稳定性，又提供了高度可定制的数据处理能力。