Question

我从Poco下载并构建了Poco项目，并在poco-1.7.6 \ Net \ samples \ WebSocketServer中运行WebsocketServer项目。

当我使用较小的128 KB（131072字节）的小数据时，它可以正常工作。但是，如果我使用更大的数据（我需要发送20 MB），我的数据将会被削减，因此服务器无法获得足够的数据。

以下是我在项目中复制的代码：

        WebSocket ws(request, response);
        char *buffer = new char[1000000]; // It just receive 131072 bytes
        int flags;
        int n;
        do
        {
            n = ws.receiveFrame(buffer, sizeof(buffer), flags);
            ws.sendFrame(buffer, tmp.length(), flags);
        } while (n > 0 || (flags & WebSocket::FRAME_OP_BITMASK) != WebSocket::FRAME_OP_CLOSE);
        delete[] buffer;

Deframe代码：

int WebSocketImpl::receiveBytes(void* buffer, int length, int)
{
    char header[MAX_HEADER_LENGTH];
    int n = receiveNBytes(header, 2);
    if (n <= 0)
    {
        _frameFlags = 0;
        return n;
    }
    poco_assert (n == 2);
    Poco::UInt8 lengthByte = static_cast<Poco::UInt8>(header[1]);
    int maskOffset = 0;
    if (lengthByte & FRAME_FLAG_MASK) maskOffset += 4;
    lengthByte &= 0x7f;
    if (lengthByte > 0 || maskOffset > 0)
    {
        if (lengthByte + 2 + maskOffset < MAX_HEADER_LENGTH)
        {
            n = receiveNBytes(header + 2, lengthByte + maskOffset);
        }
        else
        {
            n = receiveNBytes(header + 2, MAX_HEADER_LENGTH - 2);
        }
        if (n <= 0) throw WebSocketException("Incomplete header received", WebSocket::WS_ERR_INCOMPLETE_FRAME);
        n += 2;
    }
    Poco::MemoryInputStream istr(header, n);
    Poco::BinaryReader reader(istr, Poco::BinaryReader::NETWORK_BYTE_ORDER);
    Poco::UInt8 flags;
    char mask[4];
    reader >> flags >> lengthByte;
    _frameFlags = flags;
    int payloadLength = 0;
    int payloadOffset = 2;
    if ((lengthByte & 0x7f) == 127)
    {
        Poco::UInt64 l;
        reader >> l;
        if (l > length) throw WebSocketException(Poco::format("Insufficient buffer for payload size %Lu", l), WebSocket::WS_ERR_PAYLOAD_TOO_BIG);
        payloadLength = static_cast<int>(l);
        payloadOffset += 8;
    }
    else if ((lengthByte & 0x7f) == 126)
    {
        Poco::UInt16 l;
        //lenBuffer = l;

        if (l > length) throw WebSocketException(Poco::format("Insufficient buffer for payload size %hu", l), WebSocket::WS_ERR_PAYLOAD_TOO_BIG);
        payloadLength = static_cast<int>(l);
        payloadOffset += 2;
    }
    else
    {
        Poco::UInt8 l = lengthByte & 0x7f;

        if (l > length) throw WebSocketException(Poco::format("Insufficient buffer for payload size %u", unsigned(l)), WebSocket::WS_ERR_PAYLOAD_TOO_BIG);
        payloadLength = static_cast<int>(l);
    }

    if (lengthByte & FRAME_FLAG_MASK)
    {
        reader.readRaw(mask, 4);
        payloadOffset += 4;
    }

    int received = 0;

    if (payloadOffset < n)
    {
        std::memcpy(buffer, header + payloadOffset, n - payloadOffset);
        received = n - payloadOffset;
    }

    if (received < payloadLength)
    {
        n = receiveNBytes(reinterpret_cast<char*>(buffer) + received, payloadLength - received);
        if (n <= 0) throw WebSocketException("Incomplete frame received", WebSocket::WS_ERR_INCOMPLETE_FRAME);
        received += n;
    }

    if (lengthByte & FRAME_FLAG_MASK)
    {
        char* p = reinterpret_cast<char*>(buffer);
        for (int i = 0; i < received; i++)
        {
            p[i] ^= mask[i % 4];
        }
    }
    return received;
}

请有人帮助我！

P / s：抱歉我的英文

更新：我刚刚在Chrome中遇到此问题。它适用于Firefox和Edge

Answer 1

设置setChunkedTransferEncoding应该会有效。

在响应发送之前的代码中设置： response.setChunkedTransferEncoding(true);

Websocket Poco无法接收大数据

1 个答案: