#pragma once
#include "CurlRequestSchedulerClient.h"
#include "CurlResponse.h"
#include "CurlSSLVerifier.h"
#include "FileSystem.h"
#include "FormDataStreamCurl.h"
#include "NetworkLoadMetrics.h"
#include "ResourceRequest.h"
#include <wtf/Noncopyable.h>
namespace WebCore {
class CurlRequestClient;
class ResourceError;
class SharedBuffer;
class CurlRequest : public ThreadSafeRefCounted<CurlRequest>, public CurlRequestSchedulerClient {
WTF_MAKE_NONCOPYABLE(CurlRequest);
public:
static Ref<CurlRequest> create(const ResourceRequest& request, CurlRequestClient* client, bool shouldSuspend = false)
{
return adoptRef(*new CurlRequest(request, client, shouldSuspend));
}
virtual ~CurlRequest() = default;
void setClient(CurlRequestClient* client) { m_client = client; }
void setUserPass(const String&, const String&);
void start(bool isSyncRequest = false);
void cancel();
void suspend();
void resume();
bool isSyncRequest() const { return m_isSyncRequest; }
bool isCompleted() const { return !m_curlHandle; }
bool isCancelled() const { return m_cancelled; }
bool isCompletedOrCancelled() const { return isCompleted() || isCancelled(); }
void completeDidReceiveResponse();
void enableDownloadToFile();
const String& getDownloadedFilePath();
NetworkLoadMetrics getNetworkLoadMetrics() { return m_networkLoadMetrics.isolatedCopy(); }
private:
enum class Action {
None,
ReceiveData,
StartTransfer,
FinishTransfer
};
CurlRequest(const ResourceRequest&, CurlRequestClient*, bool shouldSuspend);
void retain() override { ref(); }
void release() override { deref(); }
CURL* handle() override { return m_curlHandle ? m_curlHandle->handle() : nullptr; }
void startWithJobManager();
void callClient(WTF::Function<void(CurlRequestClient*)>);
CURL* setupTransfer() override;
CURLcode willSetupSslCtx(void*);
size_t willSendData(char*, size_t, size_t);
size_t didReceiveHeader(String&&);
size_t didReceiveData(Ref<SharedBuffer>&&);
void didCompleteTransfer(CURLcode) override;
void didCancelTransfer() override;
void finalizeTransfer();
void resolveBlobReferences(ResourceRequest&);
void setupPOST(ResourceRequest&);
void setupPUT(ResourceRequest&);
void setupFormData(ResourceRequest&, bool);
bool needToInvokeDidReceiveResponse() const { return !m_didNotifyResponse || !m_didReturnFromNotify; }
bool needToInvokeDidCancelTransfer() const { return m_didNotifyResponse && !m_didReturnFromNotify && m_actionAfterInvoke == Action::FinishTransfer; }
void invokeDidReceiveResponseForFile(URL&);
void invokeDidReceiveResponse(Action);
void setRequestPaused(bool);
void setCallbackPaused(bool);
void pausedStatusChanged();
bool isPaused() const { return m_isPausedOfRequest || m_isPausedOfCallback; };
void writeDataToDownloadFileIfEnabled(const SharedBuffer&);
void closeDownloadFile();
void cleanupDownloadFile();
static CURLcode willSetupSslCtxCallback(CURL*, void*, void*);
static size_t willSendDataCallback(char*, size_t, size_t, void*);
static size_t didReceiveHeaderCallback(char*, size_t, size_t, void*);
static size_t didReceiveDataCallback(char*, size_t, size_t, void*);
std::atomic<CurlRequestClient*> m_client { };
bool m_isSyncRequest { false };
bool m_cancelled { false };
ResourceRequest m_request;
String m_user;
String m_password;
bool m_shouldSuspend { false };
std::unique_ptr<CurlHandle> m_curlHandle;
std::unique_ptr<FormDataStream> m_formDataStream;
Vector<char> m_postBuffer;
CurlSSLVerifier m_sslVerifier;
CurlResponse m_response;
bool m_didReceiveResponse { false };
bool m_didNotifyResponse { false };
bool m_didReturnFromNotify { false };
Action m_actionAfterInvoke { Action::None };
CURLcode m_finishedResultCode { CURLE_OK };
bool m_isPausedOfRequest { false };
bool m_isPausedOfCallback { false };
Lock m_downloadMutex;
bool m_isEnabledDownloadToFile { false };
String m_downloadFilePath;
FileSystem::PlatformFileHandle m_downloadFileHandle { FileSystem::invalidPlatformFileHandle };
NetworkLoadMetrics m_networkLoadMetrics;
};
}