Websocket Poco无法接收大数据

Websocket Poco can't receive big data

本文关键字:数据 Poco Websocket      更新时间:2023-10-16

我从poco下载并构建了poco项目,并在poco-1.7.6 net net samples websocketserver中下载并运行WebSocketerver项目。

当我使用较小的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;

缩写代码:

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

可以正常工作

设置setchunkedtransferencoding应该使其起作用。

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