Burp Suite API插件开发指南：篡改WebSocket报文

1. 背景与概述

WebSocket协议相比HTTP具有更低的传输开销，广泛应用于即时聊天、在线游戏、实时位置等低延迟、高并发的服务场景。随着Burp Suite新版本对WebSocket的支持，开发者现在可以通过API插件来拦截和篡改WebSocket报文。

2. 模拟场景

本教程基于一个GoLang编写的简易聊天室项目，该场景具有以下特点：

• 前端使用JavaScript计算消息的SHA-256哈希值
• 哈希值附加到消息中一起发送至服务器
• 后端校验哈希值，校验失败则返回错误信息

3. 插件开发步骤

3.1 初始化插件

package arr;

import burp.api.montoya.BurpExtension;
import burp.api.montoya.MontoyaApi;
import burp.api.montoya.logging.Logging;

public class WebsocketExample implements BurpExtension {
    MontoyaApi api;
    Logging logging;
    
    @Override
    public void initialize(MontoyaApi api) {
        // 保存对 MontoyaApi 对象的引用
        this.api = api;
        // 获取日志对象
        this.logging = api.logging();
        // 设置扩展名称
        api.extension().setName("WebsocketExample");
        // 打印加载消息
        this.logging.logToOutput("*** Burp Suite API插件开发指北 - WebsocketExample 已加载");
        // 注册WebSocket处理器
        api.websockets().registerWebSocketCreatedHandler(new CustomWebsocketCreatedHandler(api));
    }
}

3.2 创建WebSocket监听器

创建一个实现WebSocketCreatedHandler接口的类来处理WebSocket连接创建事件：

package arr;

import burp.api.montoya.MontoyaApi;
import burp.api.montoya.logging.Logging;
import burp.api.montoya.websocket.WebSocketCreated;

public class CustomWebsocketCreatedHandler implements WebSocketCreatedHandler {
    private final MontoyaApi api;
    private final Logging logging;
    
    public CustomWebsocketCreatedHandler(MontoyaApi api) {
        this.api = api;
        this.logging = api.logging();
    }
    
    @Override
    public void handleWebSocketCreated(WebSocketCreated webSocketCreated) {
        // 在这里处理WebSocket连接创建事件
        logging.logToOutput("WebSocket连接已创建: " + webSocketCreated);
    }
}

3.3 处理WebSocket消息

为了处理WebSocket消息，我们需要注册消息处理器：

@Override
public void handleWebSocketCreated(WebSocketCreated webSocketCreated) {
    // 注册二进制消息处理器
    webSocketCreated.webSocket().registerBinaryMessageHandler(new BinaryMessageHandler() {
        @Override
        public void handleBinaryMessage(BinaryMessage binaryMessage) {
            // 处理二进制消息
            byte[] payload = binaryMessage.payload();
            // 修改消息内容
            byte[] modifiedPayload = modifyPayload(payload);
            // 发送修改后的消息
            binaryMessage.webSocket().sendBinaryMessage(modifiedPayload);
        }
    });
    
    // 注册文本消息处理器
    webSocketCreated.webSocket().registerTextMessageHandler(new TextMessageHandler() {
        @Override
        public void handleTextMessage(TextMessage textMessage) {
            // 处理文本消息
            String payload = textMessage.payload();
            // 修改消息内容
            String modifiedPayload = modifyPayload(payload);
            // 发送修改后的消息
            textMessage.webSocket().sendTextMessage(modifiedPayload);
        }
    });
}

3.4 实现消息篡改逻辑

private String modifyPayload(String originalPayload) {
    // 示例：修改聊天消息的哈希值
    try {
        JSONObject json = new JSONObject(originalPayload);
        String content = json.getString("content");
        // 计算新的哈希值
        String newHash = calculateSHA256(content);
        // 更新哈希值
        json.put("hash", newHash);
        return json.toString();
    } catch (JSONException e) {
        logging.logToError("解析JSON失败: " + e.getMessage());
        return originalPayload;
    }
}

private byte[] modifyPayload(byte[] originalPayload) {
    // 处理二进制消息的修改逻辑
    // 这里可以根据实际需求实现
    return originalPayload;
}

private String calculateSHA256(String input) {
    try {
        MessageDigest digest = MessageDigest.getInstance("SHA-256");
        byte[] hashBytes = digest.digest(input.getBytes(StandardCharsets.UTF_8));
        return bytesToHex(hashBytes);
    } catch (NoSuchAlgorithmException e) {
        logging.logToError("计算哈希失败: " + e.getMessage());
        return "";
    }
}

private static String bytesToHex(byte[] bytes) {
    StringBuilder hexString = new StringBuilder();
    for (byte b : bytes) {
        String hex = Integer.toHexString(0xff & b);
        if (hex.length() == 1) hexString.append('0');
        hexString.append(hex);
    }
    return hexString.toString();
}

4. 关键API说明

4.1 WebSocket相关API

1. registerWebSocketCreatedHandler() - 注册WebSocket连接创建时的处理器
2. WebSocket接口提供的方法：
   • registerBinaryMessageHandler() - 注册二进制消息处理器
   • registerTextMessageHandler() - 注册文本消息处理器
   • sendBinaryMessage() - 发送二进制消息
   • sendTextMessage() - 发送文本消息

4.2 消息处理接口

1. WebSocketCreatedHandler - WebSocket连接创建时的处理接口
2. BinaryMessageHandler - 二进制消息处理接口
3. TextMessageHandler - 文本消息处理接口

5. 实际应用场景

本插件可用于以下场景：

1. 绕过WebSocket消息的签名校验
2. 修改实时聊天应用的消息内容
3. 篡改游戏中的实时数据
4. 测试WebSocket协议的安全性

6. 注意事项

1. 确保正确处理二进制和文本两种消息类型
2. 修改消息时要保持协议格式的一致性
3. 注意异常处理，避免插件崩溃
4. 考虑性能影响，避免在消息处理中执行耗时操作

7. 总结

通过本教程，我们学习了如何开发Burp Suite插件来拦截和篡改WebSocket报文。关键点包括：

1. 注册WebSocket连接创建处理器
2. 实现二进制和文本消息处理器
3. 修改消息内容并重新计算签名
4. 发送修改后的消息

这种技术可以用于安全测试，帮助发现WebSocket协议实现中的安全漏洞。