Exemple #1
0
size_t
Net::Request::Read(void *_buffer, size_t buffer_size, unsigned long timeout)
{
    assert(handle != NULL);

    const int timeout_ms = timeout == INFINITE ? -1 : timeout;

    Buffer::Range range;
    CURLMcode mcode = CURLM_CALL_MULTI_PERFORM;
    while (true) {
        range = buffer.Read();
        if (!range.IsEmpty())
            break;

        CURLcode code = session.InfoRead(handle);
        if (code != CURLE_AGAIN)
            return 0;

        if (mcode != CURLM_CALL_MULTI_PERFORM &&
                !session.Select(timeout_ms))
            return 0;

        CURLMcode mcode = session.Perform();
        if (mcode != CURLM_OK && mcode != CURLM_CALL_MULTI_PERFORM)
            return 0;
    }

    --buffer_size;
    if (buffer_size > range.length)
        buffer_size = range.length;

    uint8_t *p = (uint8_t *)_buffer;
    std::copy(range.data, range.data + buffer_size, p);
    p[buffer_size] = 0;

    buffer.Consume(buffer_size);
    curl_easy_pause(handle, CURLPAUSE_CONT);

    return buffer_size;
}
Exemple #2
0
size_t
Net::Request::Read(void *_buffer, size_t buffer_size, unsigned _timeout_ms)
{
    const int timeout_ms = _timeout_ms == INFINITE ? -1 : _timeout_ms;

    Buffer::Range range;
    CURLMcode mcode = CURLM_CALL_MULTI_PERFORM;
    while (true) {
        range = buffer.Read();
        if (!range.IsEmpty())
            break;

        CURLcode code = session.InfoRead(handle.GetHandle());
        if (code != CURLE_AGAIN) {
            if (code != CURLE_OK)
                throw std::runtime_error(curl_easy_strerror(code));
            return 0;
        }

        if (mcode != CURLM_CALL_MULTI_PERFORM)
            session.Select(timeout_ms);

        mcode = session.Perform();
        if (mcode != CURLM_OK && mcode != CURLM_CALL_MULTI_PERFORM)
            throw std::runtime_error(curl_multi_strerror(mcode));
    }

    --buffer_size;
    if (buffer_size > range.size)
        buffer_size = range.size;

    uint8_t *p = (uint8_t *)_buffer;
    std::copy_n(range.data, buffer_size, p);
    p[buffer_size] = 0;

    buffer.Consume(buffer_size);
    handle.Unpause();

    return buffer_size;
}