解码WebSocket掩码

Question

我试图编写一个函数来解码通过WebSocket连接发送到节点服务器的屏蔽缓冲区。结合使用RFC6455和一些旧的SO答案，这就是我迄今为止所做的：

function decodeMessage(buffer) {

    const opCode = buffer.readUInt8(0) & 0xF;

    if(opCode === 0x1) {
        const data_length = buffer.readUInt8(1) & 0x7F;
        let decoded = Buffer.alloc(data_length);

        const mask_key = buffer.readUInt32BE(2);

        for (let i = 0; i < data_length; i++) {
            decoded[i] = buffer.readUInt8(i) ^ mask_key[i % 4];
        }

        buffer.copy(decoded, 0, 2);

        return decoded.toString('utf8');

    } else {
        return null;
    }
}

此代码有两个问题：

>

它不能正确地揭开消息掩码，只是输出乱码Unicode

它每次输出不同的解码消息，即使输入是恒定的

（有效负载长度永远不会超过126，因此不需要担心处理额外的字节）

Answer 1

通过修改本文中的一个函数，我能够实现以下功能：

function decodeMessage(buffer) {

    if((buffer.readUInt8(0) & 0xF) === 0x1) {
        const length = (buffer.readUInt8(1) & 0x7F)+4;

        let currentOffset = 2;
        const mask_key = buffer.readUInt32BE(2);

        const data = Buffer.alloc(length);

        for (let i = 0, j = 0; i < length; ++i, j = i % 4) {
            const shift = j === 3 ? 0 : (3 - j) << 3;

            const mask = (shift === 0 ? mask_key : (mask_key >>> shift)) & 0xFF;
            const source = buffer.readUInt8(currentOffset++);
            data.writeUInt8(mask ^ source, i);
        }

        return data.toString('utf8');

    } else {
        return null;
    }
}