SocketStreamHandleImplCurl.cpp [plain text]
#include "config.h"
#include "SocketStreamHandleImpl.h"
#if USE(CURL)
#include "CurlStreamScheduler.h"
#include "DeprecatedGlobalSettings.h"
#include "SocketStreamError.h"
#include "SocketStreamHandleClient.h"
#include "StorageSessionProvider.h"
namespace WebCore {
SocketStreamHandleImpl::SocketStreamHandleImpl(const URL& url, SocketStreamHandleClient& client, const StorageSessionProvider* provider)
: SocketStreamHandle(url, client)
, m_storageSessionProvider(provider)
, m_scheduler(CurlContext::singleton().streamScheduler())
{
if (m_url.protocolIs("wss") && DeprecatedGlobalSettings::allowsAnySSLCertificate())
CurlContext::singleton().sslHandle().setIgnoreSSLErrors(true);
m_streamID = m_scheduler.createStream(m_url, *this);
}
SocketStreamHandleImpl::~SocketStreamHandleImpl()
{
destructStream();
}
Optional<size_t> SocketStreamHandleImpl::platformSendInternal(const uint8_t* data, size_t length)
{
if (isStreamInvalidated())
return WTF::nullopt;
if (m_totalSendDataSize + length > maxBufferSize)
return 0;
m_totalSendDataSize += length;
auto buffer = makeUniqueArray<uint8_t>(length);
memcpy(buffer.get(), data, length);
m_scheduler.send(m_streamID, WTFMove(buffer), length);
return length;
}
void SocketStreamHandleImpl::platformClose()
{
destructStream();
if (m_state == Closed)
return;
m_state = Closed;
m_client.didCloseSocketStream(*this);
}
void SocketStreamHandleImpl::didOpen(CurlStreamID)
{
if (m_state != Connecting)
return;
m_state = Open;
m_client.didOpenSocketStream(*this);
}
void SocketStreamHandleImpl::didSendData(CurlStreamID, size_t length)
{
ASSERT(m_totalSendDataSize - length >= 0);
m_totalSendDataSize -= length;
sendPendingData();
}
void SocketStreamHandleImpl::didReceiveData(CurlStreamID, const char* data, size_t length)
{
if (m_state != Open)
return;
m_client.didReceiveSocketStreamData(*this, data, length);
}
void SocketStreamHandleImpl::didFail(CurlStreamID, CURLcode errorCode)
{
destructStream();
if (m_state == Closed)
return;
if (errorCode == CURLE_RECV_ERROR)
m_client.didFailToReceiveSocketStreamData(*this);
else
m_client.didFailSocketStream(*this, SocketStreamError(errorCode, m_url.string(), CurlHandle::errorDescription(errorCode)));
}
void SocketStreamHandleImpl::destructStream()
{
if (isStreamInvalidated())
return;
m_scheduler.destroyStream(m_streamID);
m_streamID = invalidCurlStreamID;
}
}
#endif