Function Receive

engine/dlib/src/dlib/http_server.cpp:462–503

#undef HTTP_SERVER_SENDALL_AND_BAIL

    Result Receive(const Request* request, void* buffer, uint32_t buffer_size, uint32_t* received_bytes)
    {
        InternalRequest* internal_req = (InternalRequest*) request->m_Internal;
        if (internal_req->m_Result != RESULT_OK)
            return internal_req->m_Result;

        uint32_t total = 0;
        assert(internal_req->m_TotalReceived >= internal_req->m_ContentOffset);
        uint32_t bytes_in_buffer = internal_req->m_TotalReceived - internal_req->m_ContentOffset;
        if (bytes_in_buffer > 0)
        {
            uint32_t to_copy = dmMath::Min(buffer_size, bytes_in_buffer);
            memcpy(buffer, internal_req->m_Server->m_Buffer + internal_req->m_ContentOffset, to_copy);
            internal_req->m_ContentOffset += to_copy;
            total += to_copy;
        }

        while (total < buffer_size)
        {
            void* p = (void*) (((uintptr_t) buffer) + total);
            int to_recv = buffer_size - total;
            int recv_bytes = 0;
            dmSocket::Result r = dmSocket::Receive(internal_req->m_Socket, p, to_recv, &recv_bytes);
            if (r == dmSocket::RESULT_TRY_AGAIN)
            {
                // Ok
            }
            else if (r == dmSocket::RESULT_OK)
            {
                total += recv_bytes;
            }
            else
            {
                internal_req->m_Result = RESULT_SOCKET_ERROR;
                break;
            }
        }
        internal_req->m_TotalContentReceived += total;
        *received_bytes = total;

        return internal_req->m_Result;
    }

    /*
     * Handle an http-connection