| /* |
| * Copyright (C) 2010 Google Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: |
| * |
| * * Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * * Redistributions in binary form must reproduce the above |
| * copyright notice, this list of conditions and the following disclaimer |
| * in the documentation and/or other materials provided with the |
| * distribution. |
| * * Neither the name of Google Inc. nor the names of its |
| * contributors may be used to endorse or promote products derived from |
| * this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "config.h" |
| #include "core/fileapi/BlobRegistry.h" |
| |
| #include "core/fileapi/BlobURL.h" |
| #include "core/platform/network/BlobData.h" |
| #include "public/platform/Platform.h" |
| #include "public/platform/WebBlobData.h" |
| #include "public/platform/WebBlobRegistry.h" |
| #include "public/platform/WebString.h" |
| #include "public/platform/WebThreadSafeData.h" |
| #include "weborigin/SecurityOrigin.h" |
| #include "weborigin/SecurityOriginCache.h" |
| #include "wtf/Assertions.h" |
| #include "wtf/HashMap.h" |
| #include "wtf/MainThread.h" |
| #include "wtf/RefPtr.h" |
| #include "wtf/ThreadSpecific.h" |
| #include "wtf/Threading.h" |
| #include "wtf/text/StringHash.h" |
| #include "wtf/text/WTFString.h" |
| |
| using WebKit::WebBlobData; |
| using WebKit::WebBlobRegistry; |
| using WebKit::WebThreadSafeData; |
| using WTF::ThreadSpecific; |
| |
| namespace WebCore { |
| |
| class BlobOriginCache : public SecurityOriginCache { |
| public: |
| BlobOriginCache(); |
| virtual SecurityOrigin* cachedOrigin(const KURL&) OVERRIDE; |
| }; |
| |
| struct BlobRegistryContext { |
| WTF_MAKE_FAST_ALLOCATED; |
| public: |
| BlobRegistryContext(const KURL& url, PassOwnPtr<BlobData> blobData) |
| : url(url.copy()) |
| , blobData(blobData) |
| { |
| this->blobData->detachFromCurrentThread(); |
| } |
| |
| BlobRegistryContext(const KURL& url, const String& type) |
| : url(url.copy()) |
| , type(type.isolatedCopy()) |
| { |
| } |
| |
| BlobRegistryContext(const KURL& url, const KURL& srcURL) |
| : url(url.copy()) |
| , srcURL(srcURL.copy()) |
| { |
| } |
| |
| BlobRegistryContext(const KURL& url, PassRefPtr<RawData> streamData) |
| : url(url.copy()) |
| , streamData(streamData) |
| { |
| } |
| |
| BlobRegistryContext(const KURL& url) |
| : url(url.copy()) |
| { |
| } |
| |
| KURL url; |
| KURL srcURL; |
| OwnPtr<BlobData> blobData; |
| PassRefPtr<RawData> streamData; |
| String type; |
| }; |
| |
| static WebBlobRegistry* blobRegistry() |
| { |
| ASSERT(isMainThread()); |
| return WebKit::Platform::current()->blobRegistry(); |
| } |
| |
| typedef HashMap<String, RefPtr<SecurityOrigin> > BlobURLOriginMap; |
| static ThreadSpecific<BlobURLOriginMap>& originMap() |
| { |
| // We want to create the BlobOriginCache exactly once because it is shared by all the threads. |
| AtomicallyInitializedStatic(BlobOriginCache*, cache = new BlobOriginCache); |
| |
| AtomicallyInitializedStatic(ThreadSpecific<BlobURLOriginMap>*, map = new ThreadSpecific<BlobURLOriginMap>); |
| return *map; |
| } |
| |
| static void saveToOriginMap(SecurityOrigin* origin, const KURL& url) |
| { |
| // If the blob URL contains null origin, as in the context with unique |
| // security origin or file URL, save the mapping between url and origin so |
| // that the origin can be retrived when doing security origin check. |
| if (origin && BlobURL::getOrigin(url) == "null") |
| originMap()->add(url.string(), origin); |
| } |
| |
| static void removeFromOriginMap(const KURL& url) |
| { |
| if (BlobURL::getOrigin(url) == "null") |
| originMap()->remove(url.string()); |
| } |
| |
| static void registerBlobURLTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) { |
| WebBlobData webBlobData(blobRegistryContext->blobData.release()); |
| registry->registerBlobURL(blobRegistryContext->url, webBlobData); |
| } |
| } |
| |
| void BlobRegistry::registerBlobURL(const KURL& url, PassOwnPtr<BlobData> blobData) |
| { |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) { |
| WebBlobData webBlobData(blobData); |
| registry->registerBlobURL(url, webBlobData); |
| } |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url, blobData)); |
| callOnMainThread(®isterBlobURLTask, context.leakPtr()); |
| } |
| } |
| |
| static void registerBlobURLFromTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->registerBlobURL(blobRegistryContext->url, blobRegistryContext->srcURL); |
| } |
| |
| void BlobRegistry::registerBlobURL(SecurityOrigin* origin, const KURL& url, const KURL& srcURL) |
| { |
| saveToOriginMap(origin, url); |
| |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->registerBlobURL(url, srcURL); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url, srcURL)); |
| callOnMainThread(®isterBlobURLFromTask, context.leakPtr()); |
| } |
| } |
| |
| static void unregisterBlobURLTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->unregisterBlobURL(blobRegistryContext->url); |
| } |
| |
| void BlobRegistry::unregisterBlobURL(const KURL& url) |
| { |
| removeFromOriginMap(url); |
| |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->unregisterBlobURL(url); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url)); |
| callOnMainThread(&unregisterBlobURLTask, context.leakPtr()); |
| } |
| } |
| |
| static void registerStreamURLTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->registerStreamURL(blobRegistryContext->url, blobRegistryContext->type); |
| } |
| |
| void BlobRegistry::registerStreamURL(const KURL& url, const String& type) |
| { |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->registerStreamURL(url, type); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url, type)); |
| callOnMainThread(®isterStreamURLTask, context.leakPtr()); |
| } |
| } |
| |
| static void registerStreamURLFromTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->registerStreamURL(blobRegistryContext->url, blobRegistryContext->srcURL); |
| } |
| |
| void BlobRegistry::registerStreamURL(SecurityOrigin* origin, const KURL& url, const KURL& srcURL) |
| { |
| saveToOriginMap(origin, url); |
| |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->registerStreamURL(url, srcURL); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url, srcURL)); |
| callOnMainThread(®isterStreamURLFromTask, context.leakPtr()); |
| } |
| } |
| |
| static void addDataToStreamTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) { |
| WebThreadSafeData webThreadSafeData(blobRegistryContext->streamData); |
| registry->addDataToStream(blobRegistryContext->url, webThreadSafeData); |
| } |
| } |
| |
| void BlobRegistry::addDataToStream(const KURL& url, PassRefPtr<RawData> streamData) |
| { |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) { |
| WebThreadSafeData webThreadSafeData(streamData); |
| registry->addDataToStream(url, webThreadSafeData); |
| } |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url, streamData)); |
| callOnMainThread(&addDataToStreamTask, context.leakPtr()); |
| } |
| } |
| |
| static void finalizeStreamTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->finalizeStream(blobRegistryContext->url); |
| } |
| |
| void BlobRegistry::finalizeStream(const KURL& url) |
| { |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->finalizeStream(url); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url)); |
| callOnMainThread(&finalizeStreamTask, context.leakPtr()); |
| } |
| } |
| |
| static void abortStreamTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->abortStream(blobRegistryContext->url); |
| } |
| |
| void BlobRegistry::abortStream(const KURL& url) |
| { |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->abortStream(url); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url)); |
| callOnMainThread(&abortStreamTask, context.leakPtr()); |
| } |
| } |
| |
| static void unregisterStreamURLTask(void* context) |
| { |
| OwnPtr<BlobRegistryContext> blobRegistryContext = adoptPtr(static_cast<BlobRegistryContext*>(context)); |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->unregisterStreamURL(blobRegistryContext->url); |
| } |
| |
| void BlobRegistry::unregisterStreamURL(const KURL& url) |
| { |
| removeFromOriginMap(url); |
| |
| if (isMainThread()) { |
| if (WebBlobRegistry* registry = blobRegistry()) |
| registry->unregisterStreamURL(url); |
| } else { |
| OwnPtr<BlobRegistryContext> context = adoptPtr(new BlobRegistryContext(url)); |
| callOnMainThread(&unregisterStreamURLTask, context.leakPtr()); |
| } |
| } |
| |
| BlobOriginCache::BlobOriginCache() |
| { |
| SecurityOrigin::setCache(this); |
| } |
| |
| SecurityOrigin* BlobOriginCache::cachedOrigin(const KURL& url) |
| { |
| if (url.protocolIs("blob")) |
| return originMap()->get(url.string()); |
| return 0; |
| } |
| |
| } // namespace WebCore |