解码 websocket 框架

Question

我正在尝试解码 websocket 帧，但在解码扩展有效负载时没有成功。到目前为止我所取得的成就是：

char *in = data;
char *buffer;
unsigned int i;
unsigned char mask[4];
unsigned int packet_length = 0;
int rc;

/* Expect a finished text frame. */
assert(in[0] == '\x81');
packet_length = ((unsigned char) in[1]) & 0x7f;

mask[0] = in[2];
mask[1] = in[3];
mask[2] = in[4];
mask[3] = in[5];

if (packet_length <= 125) {           **// This decoding works**
     /* Unmask the payload. */
    for (i = 0; i < packet_length; i++)
        in[6 + i] ^= mask[i % 4];
    rc = asprintf(&buffer, "%.*s", packet_length, in + 6);
} else
    if (packet_length == 126) {       **//This decosing does NOT work**
        /* Unmask the payload. */
        for (i = 0; i < packet_length; i++)
          in[8 + i] ^= mask[i % 4];
        rc = asprintf(&buffer, "%.*s", packet_length, in + 8);
}

我做错了什么？如何对扩展有效负载进行编码？

Answer 1

症结点在于>。 125 字节有效负载。

格式非常简单，假设您在 JavaScript 中发送十个 a：

ws.send("a".repeat(10))

然后服务器将收到：

bytes[16]=818a8258a610e339c771e339c771e339

• 字节 0：0x81 只是消息接收的指示符
• 字节 1：0x8a 是长度，从中减去 0x80，0x0A = = 10
• 字节 2, 3, 4, 5：用于解密有效负载的 4 字节异或密钥
• 其余：有效负载

但现在假设您发送 126 个 a在JavaScript中：

ws.send("a".repeat(126))

那么服务器将收到：

bytes[134]=81fe007ee415f1e5857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574908485749084857490848574

如果有效负载的长度> 125，字节 1 的值为 0xfe，格式随后更改为：

• 字节 0：0x81 只是一个指示符，表明收到的消息
• 字节 1：将是 0xfe
• 字节 2、3：作为 uint16 数字的有效负载的长度
• 字节 4, 5, 6, 7：用于解密有效负载的 4 字节异或密钥
• 其余部分：有效负载

C# 中的示例代码：

List<byte[]> decodeWebsocketFrame(Byte[] bytes)
{
    List<Byte[]> ret = new List<Byte[]>();
    int offset = 0;
    while (offset + 6 < bytes.Length)
    {
        // format: 0==ascii/binary 1=length-0x80, byte 2,3,4,5=key, 6+len=message, repeat with offset for next...
        int len = bytes[offset + 1] - 0x80;

        if (len <= 125)
        {

            //String data = Encoding.UTF8.GetString(bytes);
            //Debug.Log("len=" + len + "bytes[" + bytes.Length + "]=" + ByteArrayToString(bytes) + " data[" + data.Length + "]=" + data);
            Debug.Log("len=" + len + " offset=" + offset);
            Byte[] key = new Byte[] { bytes[offset + 2], bytes[offset + 3], bytes[offset + 4], bytes[offset + 5] };
            Byte[] decoded = new Byte[len];
            for (int i = 0; i < len; i++)
            {
                int realPos = offset + 6 + i;
                decoded[i] = (Byte)(bytes[realPos] ^ key[i % 4]);
            }
            offset += 6 + len;
            ret.Add(decoded);
        } else
        {
            int a = bytes[offset + 2];
            int b = bytes[offset + 3];
            len = (a << 8) + b;
            //Debug.Log("Length of ws: " + len);

            Byte[] key = new Byte[] { bytes[offset + 4], bytes[offset + 5], bytes[offset + 6], bytes[offset + 7] };
            Byte[] decoded = new Byte[len];
            for (int i = 0; i < len; i++)
            {
                int realPos = offset + 8 + i;
                decoded[i] = (Byte)(bytes[realPos] ^ key[i % 4]);
            }

            offset += 8 + len;
            ret.Add(decoded);
        }
    }
    return ret;
}

Answer 2

如果packet_length为126，则接下来的2个字节给出要读取的数据的长度。

如果packet_length为127，则接下来的8个字节给出了要读取的数据的长度。

掩码包含在以下 4 个字节中（长度之后）。

要解码的消息如下。

数据框架部分规范对此有一个有用的说明。

如果您将代码重新排序为

• “读取 packet_length”
• 之类的内容，请检查 packet_length 是否为 126 或 127。如果需要，请将 packet_length 重新分配为以下 2/4 字节的值。
• 读取掩码（packet_length 之后的 4 个字节，包括为上述步骤读取的任何附加 2 或 8 个字节）。
• 解码消息（掩码后的所有内容）。

那么事情就应该可以了。