HTTP 之 Web Sockets处理恶意的Payload的策略（十一）

处理恶意的 Payload 主要涉及到输入验证、清理和在某些情况下对数据进行适当的转义。

1. 输入验证（Validation）

验证所有通过 WebSockets 接收的数据以确保它们符合预期格式。例如，如果你期望一个数字，验证接收到的数据是否为数字。

socket.on('message', function(message) {
    const data = JSON.parse(message); // 假设消息是JSON格式
    if (typeof data.id !== 'number') {
        console.error('接收到无效的Payload:', message);
        return; // 或者采取其他错误处理措施
    }
    // 继续处理有效数据
});

2. 数据清理（Sanitization）

在将数据展示给用户或存储到数据库之前，清理数据以防止 XSS 攻击或SQL注入。

防止 XSS 攻击：
对所有接收到的数据进行HTML转义，避免HTML标签的执行。

function escapeHTML(unsafe) {
    return unsafe
        .replace(/&/g, "&")
        .replace(/</g, "<")
        .replace(/>/g, ">")
        .replace(/"/g, """)
        .replace(/'/g, "'");
}

socket.on('message', function(message) {
    const data = JSON.parse(message);
    const safeData = escapeHTML(data.content);
    // 使用safeData进行DOM操作或存储
});

防止 SQL 注入：
使用参数化查询，避免直接将用户输入拼接到SQL语句中。

// 假设使用Node.js的某个数据库库
function queryDatabase(id) {
    // 使用参数化查询防止SQL注入
    db.query('SELECT * FROM records WHERE id = ?', [id], function(err, results) {
        // 处理查询结果
    });
}

3. 使用白名单（Whitelisting）

定义允许的字符或格式，并拒绝所有不符合这些规则的输入。

function isValidInput(input) {
    // 定义允许的字符或模式
    const allowedPattern = /^[a-zA-Z0-9]+$/;
    return allowedPattern.test(input);
}

socket.on('message', function(message) {
    if (!isValidInput(message)) {
        console.error('接收到不允许的Payload:', message);
        return;
    }
    // 继续处理允许的数据
});

4. 限制消息大小

限制通过 WebSockets 接收的消息大小，避免大量数据的传输导致服务拒绝（DoS）。

socket.on('message', function(message) {
    if (message.length > MAX_MESSAGE_SIZE) {
        console.error('接收到的消息过大:', message);
        return;
    }
    // 处理有效大小的消息
});

5. 错误处理和断开连接

在检测到恶意数据时，除了记录错误，还应该考虑断开连接以防止进一步的攻击。

socket.on('message', function(message) {
    if (isMalicious(message)) {
        console.error('检测到恶意Payload:', message);
        socket.close(1008, 'Malicious payload detected');
        return;
    }
    // 正常处理消息
});

6. 使用安全框架和库

利用现有的安全框架和库来帮助处理输入验证和清理，例如使用OWASP的ESAPI等。

7. 监控和日志记录

记录所有WebSockets通信，并监控异常模式或潜在的攻击行为。

socket.on('message', function(message) {
    console.log('接收到消息:', message); // 记录消息
    // 消息处理逻辑...
});