~mmach/netext73/webkit2gtk

/*

 * Copyright (C) 2019 Apple Inc. All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''

 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,

 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS

 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF

 * THE POSSIBILITY OF SUCH DAMAGE.

 */

 

#include "config.h"

#include "NetworkSendQueue.h"

 

#include "BlobLoader.h"

 

namespace WebCore {

 

NetworkSendQueue::NetworkSendQueue(Document& document, WriteString&& writeString, WriteRawData&& writeRawData, ProcessError&& processError)

    : m_document(makeWeakPtr(document))

    , m_writeString(WTFMove(writeString))

    , m_writeRawData(WTFMove(writeRawData))

    , m_processError(WTFMove(processError))

{

}

 

NetworkSendQueue::~NetworkSendQueue() = default;

 

void NetworkSendQueue::enqueue(const String& data)

{

    if (m_queue.isEmpty()) {

        m_writeString(data);

        return;

    }

    m_queue.append(data);

}

 

void NetworkSendQueue::enqueue(const JSC::ArrayBuffer& binaryData, unsigned byteOffset, unsigned byteLength)

{

    if (m_queue.isEmpty()) {

        auto* data = static_cast<const char*>(binaryData.data());

        m_writeRawData(data + byteOffset, byteLength);

        return;

    }

    m_queue.append(SharedBuffer::create(static_cast<const uint8_t*>(binaryData.data()) + byteOffset, byteLength));

}

 

void NetworkSendQueue::enqueue(WebCore::Blob& blob)

{

    auto byteLength = blob.size();

    if (!byteLength) {

        enqueue(JSC::ArrayBuffer::create(0U, 1), 0, 0);

        return;

    }

    m_queue.append(makeUniqueRef<BlobLoader>(m_document.get(), blob, [this] {

        processMessages();

    }));

}

 

void NetworkSendQueue::clear()

{

    m_queue.clear();

}

 

void NetworkSendQueue::processMessages()

{

    while (!m_queue.isEmpty()) {

        bool shouldStopProcessing = false;

        switchOn(m_queue.first(), [this](const String& message) {

            m_writeString(message);

        }, [this](Ref<SharedBuffer>& data) {

            m_writeRawData(data->data(), data->size());

        }, [this, &shouldStopProcessing](UniqueRef<BlobLoader>& loader) {

            if (loader->isLoading() || loader->errorCode() == FileError::ABORT_ERR) {

                shouldStopProcessing = true;

                return;

            }

 

            if (const auto& result = loader->result()) {

                m_writeRawData(static_cast<const char*>(result->data()), result->byteLength());

                return;

            }

            ASSERT(loader->errorCode());

            shouldStopProcessing = m_processError(loader->errorCode()) == Continue::No;

        });

        if (shouldStopProcessing)

            return;

        m_queue.removeFirst();

    }

 

}

 

} // namespace WebCore