NetworkDataTaskCurl.h [plain text]
#pragma once
#include "CurlRequestClient.h"
#include "NetworkDataTask.h"
#include <WebCore/NetworkLoadMetrics.h>
#include <WebCore/ResourceResponse.h>
namespace WebCore {
class CurlRequest;
}
namespace WebKit {
class NetworkDataTaskCurl final : public NetworkDataTask, public WebCore::CurlRequestClient {
public:
static Ref<NetworkDataTask> create(NetworkSession& session, NetworkDataTaskClient& client, const WebCore::ResourceRequest& request, WebCore::StoredCredentialsPolicy storedCredentialsPolicy, WebCore::ContentSniffingPolicy shouldContentSniff, WebCore::ContentEncodingSniffingPolicy shouldContentEncodingSniff, bool shouldClearReferrerOnHTTPSToHTTPRedirect, bool dataTaskIsForMainFrameNavigation)
{
return adoptRef(*new NetworkDataTaskCurl(session, client, request, storedCredentialsPolicy, shouldContentSniff, shouldContentEncodingSniff, shouldClearReferrerOnHTTPSToHTTPRedirect, dataTaskIsForMainFrameNavigation));
}
~NetworkDataTaskCurl();
void ref() override { RefCounted<NetworkDataTask>::ref(); }
void deref() override { RefCounted<NetworkDataTask>::deref(); }
private:
enum class ShouldPreprocess : bool {
No = false,
Yes = true
};
NetworkDataTaskCurl(NetworkSession&, NetworkDataTaskClient&, const WebCore::ResourceRequest&, WebCore::StoredCredentialsPolicy, WebCore::ContentSniffingPolicy, WebCore::ContentEncodingSniffingPolicy, bool shouldClearReferrerOnHTTPSToHTTPRedirect, bool dataTaskIsForMainFrameNavigation);
void suspend() override;
void cancel() override;
void resume() override;
void invalidateAndCancel() override;
NetworkDataTask::State state() const override;
Ref<WebCore::CurlRequest> createCurlRequest(const WebCore::ResourceRequest&, ShouldPreprocess);
void curlDidSendData(WebCore::CurlRequest&, unsigned long long, unsigned long long) override;
void curlDidReceiveResponse(WebCore::CurlRequest&, const WebCore::CurlResponse&) override;
void curlDidReceiveBuffer(WebCore::CurlRequest&, Ref<WebCore::SharedBuffer>&&) override;
void curlDidComplete(WebCore::CurlRequest&) override;
void curlDidFailWithError(WebCore::CurlRequest&, const WebCore::ResourceError&) override;
bool shouldRedirectAsGET(const WebCore::ResourceRequest&, bool crossOrigin);
void willPerformHTTPRedirection();
void tryHttpAuthentication(WebCore::AuthenticationChallenge&&);
void restartWithCredential(const WebCore::Credential&);
void appendCookieHeader(WebCore::ResourceRequest&);
void handleCookieHeaders(const WebCore::CurlResponse&);
State m_state { State::Suspended };
WebCore::ResourceRequest m_currentRequest;
RefPtr<WebCore::CurlRequest> m_curlRequest;
WebCore::ResourceResponse m_response;
unsigned m_redirectCount { 0 };
unsigned m_authFailureCount { 0 };
};
}