1/* 2 * Copyright (C) 2013 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' 14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS 17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 23 * THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26#include "config.h" 27#include "NetworkBlobRegistry.h" 28 29#if ENABLE(NETWORK_PROCESS) 30 31#include "BlobDataFileReferenceWithSandboxExtension.h" 32#include "SandboxExtension.h" 33#include <WebCore/BlobPart.h> 34#include <WebCore/BlobRegistryImpl.h> 35#include <wtf/NeverDestroyed.h> 36#include <wtf/RunLoop.h> 37 38using namespace WebCore; 39 40namespace WebKit { 41 42NetworkBlobRegistry& NetworkBlobRegistry::shared() 43{ 44 ASSERT(RunLoop::isMain()); 45 static NeverDestroyed<NetworkBlobRegistry> registry; 46 return registry; 47} 48 49NetworkBlobRegistry::NetworkBlobRegistry() 50{ 51} 52 53void NetworkBlobRegistry::registerFileBlobURL(NetworkConnectionToWebProcess* connection, const URL& url, const String& path, PassRefPtr<SandboxExtension> sandboxExtension, const String& contentType) 54{ 55 blobRegistry().registerFileBlobURL(url, BlobDataFileReferenceWithSandboxExtension::create(path, sandboxExtension), contentType); 56 57 ASSERT(!m_blobsForConnection.get(connection).contains(url)); 58 BlobForConnectionMap::iterator mapIterator = m_blobsForConnection.find(connection); 59 if (mapIterator == m_blobsForConnection.end()) 60 mapIterator = m_blobsForConnection.add(connection, HashSet<URL>()).iterator; 61 mapIterator->value.add(url); 62} 63 64void NetworkBlobRegistry::registerBlobURL(NetworkConnectionToWebProcess* connection, const URL& url, Vector<WebCore::BlobPart> blobParts, const String& contentType) 65{ 66 blobRegistry().registerBlobURL(url, WTF::move(blobParts), contentType); 67 68 ASSERT(!m_blobsForConnection.get(connection).contains(url)); 69 BlobForConnectionMap::iterator mapIterator = m_blobsForConnection.find(connection); 70 if (mapIterator == m_blobsForConnection.end()) 71 mapIterator = m_blobsForConnection.add(connection, HashSet<URL>()).iterator; 72 mapIterator->value.add(url); 73} 74 75void NetworkBlobRegistry::registerBlobURL(NetworkConnectionToWebProcess* connection, const WebCore::URL& url, const WebCore::URL& srcURL) 76{ 77 // The connection may not be registered if NetworkProcess prevously crashed for any reason. 78 BlobForConnectionMap::iterator mapIterator = m_blobsForConnection.find(connection); 79 if (mapIterator == m_blobsForConnection.end()) 80 return; 81 82 blobRegistry().registerBlobURL(url, srcURL); 83 84 ASSERT(mapIterator->value.contains(srcURL)); 85 mapIterator->value.add(url); 86} 87 88void NetworkBlobRegistry::registerBlobURLForSlice(NetworkConnectionToWebProcess* connection, const WebCore::URL& url, const WebCore::URL& srcURL, int64_t start, int64_t end) 89{ 90 // The connection may not be registered if NetworkProcess prevously crashed for any reason. 91 BlobForConnectionMap::iterator mapIterator = m_blobsForConnection.find(connection); 92 if (mapIterator == m_blobsForConnection.end()) 93 return; 94 95 blobRegistry().registerBlobURLForSlice(url, srcURL, start, end); 96 97 ASSERT(mapIterator->value.contains(srcURL)); 98 mapIterator->value.add(url); 99} 100 101void NetworkBlobRegistry::unregisterBlobURL(NetworkConnectionToWebProcess* connection, const WebCore::URL& url) 102{ 103 // The connection may not be registered if NetworkProcess prevously crashed for any reason. 104 BlobForConnectionMap::iterator mapIterator = m_blobsForConnection.find(connection); 105 if (mapIterator == m_blobsForConnection.end()) 106 return; 107 108 blobRegistry().unregisterBlobURL(url); 109 110 ASSERT(mapIterator->value.contains(url)); 111 mapIterator->value.remove(url); 112} 113 114uint64_t NetworkBlobRegistry::blobSize(NetworkConnectionToWebProcess* connection, const WebCore::URL& url) 115{ 116 if (!m_blobsForConnection.contains(connection) || !m_blobsForConnection.find(connection)->value.contains(url)) 117 return 0; 118 119 return blobRegistry().blobSize(url); 120} 121 122void NetworkBlobRegistry::connectionToWebProcessDidClose(NetworkConnectionToWebProcess* connection) 123{ 124 if (!m_blobsForConnection.contains(connection)) 125 return; 126 127 HashSet<URL>& blobsForConnection = m_blobsForConnection.find(connection)->value; 128 for (HashSet<URL>::iterator iter = blobsForConnection.begin(), end = blobsForConnection.end(); iter != end; ++iter) 129 blobRegistry().unregisterBlobURL(*iter); 130 131 m_blobsForConnection.remove(connection); 132} 133 134Vector<RefPtr<BlobDataFileReference>> NetworkBlobRegistry::filesInBlob(NetworkConnectionToWebProcess* connection, const WebCore::URL& url) 135{ 136 if (!m_blobsForConnection.contains(connection) || !m_blobsForConnection.find(connection)->value.contains(url)) 137 return Vector<RefPtr<BlobDataFileReference>>(); 138 139 ASSERT(blobRegistry().isBlobRegistryImpl()); 140 BlobData* blobData = static_cast<BlobRegistryImpl&>(blobRegistry()).getBlobDataFromURL(url); 141 if (!blobData) 142 return Vector<RefPtr<BlobDataFileReference>>(); 143 144 Vector<RefPtr<BlobDataFileReference>> result; 145 for (const BlobDataItem& item : blobData->items()) { 146 if (item.type == BlobDataItem::File) 147 result.append(item.file); 148 } 149 150 return result; 151} 152 153} 154 155#endif 156