NetworkDataTaskBlob.h [plain text]
#pragma once
#include "NetworkDataTask.h"
#include <WebCore/FileStreamClient.h>
#include <wtf/FileSystem.h>
namespace WebCore {
class AsyncFileStream;
class BlobDataFileReference;
class BlobData;
class BlobDataItem;
class BlobRegistryImpl;
}
namespace WebKit {
class NetworkProcess;
class NetworkDataTaskBlob final : public NetworkDataTask, public WebCore::FileStreamClient {
public:
static Ref<NetworkDataTask> create(NetworkSession& session, WebCore::BlobRegistryImpl& blobRegistry, NetworkDataTaskClient& client, const WebCore::ResourceRequest& request, WebCore::ContentSniffingPolicy shouldContentSniff, const Vector<RefPtr<WebCore::BlobDataFileReference>>& fileReferences)
{
return adoptRef(*new NetworkDataTaskBlob(session, blobRegistry, client, request, shouldContentSniff, fileReferences));
}
~NetworkDataTaskBlob();
private:
NetworkDataTaskBlob(NetworkSession&, WebCore::BlobRegistryImpl&, NetworkDataTaskClient&, const WebCore::ResourceRequest&, WebCore::ContentSniffingPolicy, const Vector<RefPtr<WebCore::BlobDataFileReference>>&);
void cancel() override;
void resume() override;
void invalidateAndCancel() override;
NetworkDataTask::State state() const override { return m_state; }
void setPendingDownloadLocation(const String&, SandboxExtension::Handle&&, bool ) override;
String suggestedFilename() const override;
void didGetSize(long long) override;
void didOpen(bool) override;
void didRead(int) override;
enum class Error {
NoError = 0,
NotFoundError = 1,
SecurityError = 2,
RangeError = 3,
NotReadableError = 4,
MethodNotAllowed = 5
};
void clearStream();
void getSizeForNext();
void dispatchDidReceiveResponse(Error = Error::NoError);
void seek();
void consumeData(const char* data, int bytesRead);
void read();
void readData(const WebCore::BlobDataItem&);
void readFile(const WebCore::BlobDataItem&);
void download();
bool writeDownload(const char* data, int bytesRead);
void cleanDownloadFiles();
void didFailDownload(const WebCore::ResourceError&);
void didFinishDownload();
void didFail(Error);
void didFinish();
enum { kPositionNotSpecified = -1 };
RefPtr<WebCore::BlobData> m_blobData;
std::unique_ptr<WebCore::AsyncFileStream> m_stream; Vector<char> m_buffer;
Vector<long long> m_itemLengthList;
State m_state { State::Suspended };
long long m_rangeOffset { kPositionNotSpecified };
long long m_rangeEnd { kPositionNotSpecified };
long long m_rangeSuffixLength { kPositionNotSpecified };
long long m_totalSize { 0 };
long long m_downloadBytesWritten { 0 };
long long m_totalRemainingSize { 0 };
long long m_currentItemReadSize { 0 };
unsigned m_sizeItemCount { 0 };
unsigned m_readItemCount { 0 };
bool m_fileOpened { false };
FileSystem::PlatformFileHandle m_downloadFile { FileSystem::invalidPlatformFileHandle };
Vector<RefPtr<WebCore::BlobDataFileReference>> m_fileReferences;
RefPtr<SandboxExtension> m_sandboxExtension;
Ref<NetworkProcess> m_networkProcess;
};
}