Sword websocket分析二

//websocket发送数据
int send(uint8_t* message, uint64_t message_size)
{
    //掩码
    const uint8_t masking_key[] = { 0x12, 0x34, 0x56, 0x78 };
    //协议最基本的两个字节标志
    //如果data长度在126~65535之间，则此处playload的值为126
    //(message_size >= 126 ? 2 : 0) --如果playload == 126,增加额外两个字节表示data的长度
    //如果数据的长度大于65535，则此处playload的值为127
    //需要增加8个字节来表示数据的长度
    //MASK 掩码标识位，用来表明负载是否经过掩码处理，浏览器发送的数据都是经过掩码处理(浏览器自动处理，
    //无需开发者编码)，服务器发送的帧必须不经过掩码处理。所以此处浏览器发送的帧必为1，服务器发送的帧必为0，否则应断开WebSocket连接
    //payload length 后面4个字节可能是掩码的key(如果掩码位是1则有这4个字节的key，否则没有)
    const int headlen =  + (message_size >=  ?  : ) + (message_size >=  ?  : ) + (_useMask ?  : );

    uint8_t* header = new uint8_t[headlen];
    uint8_t* txbuf = NULL;
    uint64_t i = ;

    memset(header, , sizeof(uint8_t)*headlen);
    //0x80(1000 0000)  表示一帧之内将所有的数据全部发过去
    header[] = 0x80 | type;

    if (message_size < ) {
        header[] = (message_size & 0xff) | (_useMask ? 0x80 : );
        if (_useMask) {
            //payload length 后面4个字节是掩码的key
            header[] = masking_key[];
            header[] = masking_key[];
            header[] = masking_key[];
            header[] = masking_key[];
        }
    }
    else if (message_size < ) {
        header[] =  | (_useMask ? 0x80 : );
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        if (_useMask) {
            header[] = masking_key[];
            header[] = masking_key[];
            header[] = masking_key[];
            header[] = masking_key[];
        }
    }
    else { // TODO: run coverage testing here
        header[] =  | (_useMask ? 0x80 : );
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        header[] = (message_size >> ) & 0xff;
        if (_useMask) {
            header[] = masking_key[];
            header[] = masking_key[];
            header[] = masking_key[];
            header[] = masking_key[];
        }
    }

}

//服务器响应客户端websocket升级请求
int response()
{
    strcat(resData, "HTTP/1.1 101 Switching Protocols\r\n");
    strcat(resData, "Upgrade:websocket\r\n");
    strcat(resData, "Connection: upgrade\r\n");
    strcat(resData, "Sec-WebSocket-Version:13\r\n");

    snprintf(serverkey, , "%s258EAFA5-E914-47DA-95CA-C5AB0DC85B11", stHeader.websocketKey);

    /*
    把Sec-WebSocket-Key加上一个魔幻字符串258EAFA5-E914-47DA-95CA-C5AB0DC85B11。
    使用 SHA-1 加密，之后进行 BASE-64编码，将结果作为 Sec-WebSocket-Accept 头的值
    此处注意，网上的sha1加密都已经字符串化了，并非实际得到的sha1密文,sha1的密文不完全是由[0-9a-z]组成的
    */
    result = sha_encode(SHA_1, (unsigned char *)serverkey, strlen(serverkey), (unsigned char **)&shaTmp, &shalen);
    if (result)
    {
        return -;
    }
    result = base64_encode(shaTmp, shalen, &pcOut, &outlen);
    if (result)
    {
        return -;
    }
    strcat(resData, "Sec-WebSocket-Accept:");
    strcat(resData, pcOut);
    strcat(resData, "\r\n");
}