1/*
2 * Copyright (C) 2010 Google Inc. All rights reserved.
3 * Copyright (C) 2013, 2014, 2016 Apple Inc. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are
7 * met:
8 *
9 * * Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * * Redistributions in binary form must reproduce the above
12 * copyright notice, this list of conditions and the following disclaimer
13 * in the documentation and/or other materials provided with the
14 * distribution.
15 * * Neither the name of Google Inc. nor the names of its
16 * contributors may be used to endorse or promote products derived from
17 * this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 */
31
32#pragma once
33
34#include "BlobData.h"
35#include "BlobRegistry.h"
36#include <wtf/HashMap.h>
37#include <wtf/URLHash.h>
38#include <wtf/text/StringHash.h>
39#include <wtf/text/WTFString.h>
40
41namespace WebCore {
42
43class ResourceHandle;
44class ResourceHandleClient;
45class ResourceRequest;
46class ThreadSafeDataBuffer;
47
48// BlobRegistryImpl is not thread-safe. It should only be called from main thread.
49class WEBCORE_EXPORT BlobRegistryImpl final : public BlobRegistry {
50 WTF_MAKE_FAST_ALLOCATED;
51public:
52 virtual ~BlobRegistryImpl();
53
54 BlobData* getBlobDataFromURL(const URL&) const;
55
56 Ref<ResourceHandle> createResourceHandle(const ResourceRequest&, ResourceHandleClient*);
57 void writeBlobToFilePath(const URL& blobURL, const String& path, Function<void(bool success)>&& completionHandler);
58
59 void appendStorageItems(BlobData*, const BlobDataItemList&, long long offset, long long length);
60
61 void registerFileBlobURL(const URL&, Ref<BlobDataFileReference>&&, const String& contentType) override;
62 void registerBlobURL(const URL&, Vector<BlobPart>&&, const String& contentType) override;
63 void registerBlobURL(const URL&, const URL& srcURL) override;
64 void registerBlobURLOptionallyFileBacked(const URL&, const URL& srcURL, RefPtr<BlobDataFileReference>&&, const String& contentType) override;
65 void registerBlobURLForSlice(const URL&, const URL& srcURL, long long start, long long end) override;
66 void unregisterBlobURL(const URL&) override;
67 bool isBlobRegistryImpl() const override { return true; }
68
69 unsigned long long blobSize(const URL&) override;
70
71 void writeBlobsToTemporaryFiles(const Vector<String>& blobURLs, CompletionHandler<void(Vector<String>&& filePaths)>&&) override;
72
73 struct BlobForFileWriting {
74 String blobURL;
75 Vector<std::pair<String, ThreadSafeDataBuffer>> filePathsOrDataBuffers;
76 };
77
78 bool populateBlobsForFileWriting(const Vector<String>& blobURLs, Vector<BlobForFileWriting>&);
79
80private:
81 HashMap<String, RefPtr<BlobData>> m_blobs;
82};
83
84} // namespace WebCore
85