SocketStreamHandleImpl.h [plain text]
#pragma once
#include "SocketStreamHandle.h"
#if PLATFORM(WIN)
#include <winsock2.h>
#endif
#include <curl/curl.h>
#include <pal/SessionID.h>
#include <wtf/Deque.h>
#include <wtf/Lock.h>
#include <wtf/RefCounted.h>
#include <wtf/Seconds.h>
#include <wtf/StreamBuffer.h>
#include <wtf/Threading.h>
#include <wtf/UniqueArray.h>
namespace WebCore {
class SocketStreamHandleClient;
class SocketStreamHandleImpl : public SocketStreamHandle {
public:
static Ref<SocketStreamHandleImpl> create(const URL& url, SocketStreamHandleClient& client, PAL::SessionID, const String&, SourceApplicationAuditToken&&) { return adoptRef(*new SocketStreamHandleImpl(url, client)); }
virtual ~SocketStreamHandleImpl();
WEBCORE_EXPORT void platformSend(const uint8_t* data, size_t length, Function<void(bool)>&&) final;
WEBCORE_EXPORT void platformSendHandshake(const uint8_t* data, size_t length, const std::optional<CookieRequestHeaderFieldProxy>&, Function<void(bool, bool)>&&) final;
WEBCORE_EXPORT void platformClose() final;
private:
WEBCORE_EXPORT SocketStreamHandleImpl(const URL&, SocketStreamHandleClient&);
size_t bufferedAmount() final;
std::optional<size_t> platformSendInternal(const uint8_t*, size_t);
bool sendPendingData();
bool readData(CURL*);
bool sendData(CURL*);
bool waitForAvailableData(CURL*, Seconds selectTimeout);
void startThread();
void stopThread();
void didReceiveData();
void didOpenSocket();
struct SocketData {
SocketData(UniqueArray<char>&& source, size_t length)
{
data = WTFMove(source);
size = length;
}
SocketData(SocketData&& other)
{
data = WTFMove(other.data);
size = other.size;
other.size = 0;
}
UniqueArray<char> data;
size_t size { 0 };
};
RefPtr<Thread> m_workerThread;
std::atomic<bool> m_stopThread { false };
Lock m_mutexSend;
Lock m_mutexReceive;
Deque<SocketData> m_sendData;
Deque<SocketData> m_receiveData;
bool m_closed { false };
StreamBuffer<uint8_t, 1024 * 1024> m_buffer;
static const unsigned maxBufferSize = 100 * 1024 * 1024;
};
}