1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "content/browser/resource_context_impl.h" 6 7 #include "base/logging.h" 8 #include "content/browser/fileapi/chrome_blob_storage_context.h" 9 #include "content/browser/host_zoom_map_impl.h" 10 #include "content/browser/loader/resource_dispatcher_host_impl.h" 11 #include "content/browser/loader/resource_request_info_impl.h" 12 #include "content/browser/streams/stream_context.h" 13 #include "content/browser/webui/url_data_manager_backend.h" 14 #include "content/public/browser/browser_context.h" 15 #include "content/public/browser/browser_thread.h" 16 17 using base::UserDataAdapter; 18 19 namespace content { 20 21 namespace { 22 23 // Key names on ResourceContext. 24 const char kBlobStorageContextKeyName[] = "content_blob_storage_context"; 25 const char kHostZoomMapKeyName[] = "content_host_zoom_map"; 26 const char kStreamContextKeyName[] = "content_stream_context"; 27 const char kURLDataManagerBackendKeyName[] = "url_data_manager_backend"; 28 29 class NonOwningZoomData : public base::SupportsUserData::Data { 30 public: 31 explicit NonOwningZoomData(HostZoomMap* hzm) : host_zoom_map_(hzm) {} 32 HostZoomMap* host_zoom_map() { return host_zoom_map_; } 33 34 private: 35 HostZoomMap* host_zoom_map_; 36 }; 37 38 } // namespace 39 40 41 ResourceContext::ResourceContext() { 42 if (ResourceDispatcherHostImpl::Get()) 43 ResourceDispatcherHostImpl::Get()->AddResourceContext(this); 44 } 45 46 ResourceContext::~ResourceContext() { 47 ResourceDispatcherHostImpl* rdhi = ResourceDispatcherHostImpl::Get(); 48 if (rdhi) { 49 rdhi->CancelRequestsForContext(this); 50 rdhi->RemoveResourceContext(this); 51 } 52 53 // In some tests this object is destructed on UI thread. 54 DetachUserDataThread(); 55 } 56 57 ChromeBlobStorageContext* GetChromeBlobStorageContextForResourceContext( 58 ResourceContext* resource_context) { 59 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 60 return UserDataAdapter<ChromeBlobStorageContext>::Get( 61 resource_context, kBlobStorageContextKeyName); 62 } 63 64 StreamContext* GetStreamContextForResourceContext( 65 ResourceContext* resource_context) { 66 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 67 return UserDataAdapter<StreamContext>::Get( 68 resource_context, kStreamContextKeyName); 69 } 70 71 HostZoomMap* GetHostZoomMapForResourceContext(ResourceContext* context) { 72 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 73 return static_cast<NonOwningZoomData*>( 74 context->GetUserData(kHostZoomMapKeyName))->host_zoom_map(); 75 } 76 77 URLDataManagerBackend* GetURLDataManagerForResourceContext( 78 ResourceContext* context) { 79 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 80 if (!context->GetUserData(kURLDataManagerBackendKeyName)) { 81 context->SetUserData(kURLDataManagerBackendKeyName, 82 new URLDataManagerBackend()); 83 } 84 return static_cast<URLDataManagerBackend*>( 85 context->GetUserData(kURLDataManagerBackendKeyName)); 86 } 87 88 void InitializeResourceContext(BrowserContext* browser_context) { 89 ResourceContext* resource_context = browser_context->GetResourceContext(); 90 DCHECK(!resource_context->GetUserData(kHostZoomMapKeyName)); 91 92 resource_context->SetUserData( 93 kBlobStorageContextKeyName, 94 new UserDataAdapter<ChromeBlobStorageContext>( 95 ChromeBlobStorageContext::GetFor(browser_context))); 96 97 resource_context->SetUserData( 98 kStreamContextKeyName, 99 new UserDataAdapter<StreamContext>( 100 StreamContext::GetFor(browser_context))); 101 102 // This object is owned by the BrowserContext and not ResourceContext, so 103 // store a non-owning pointer here. 104 resource_context->SetUserData( 105 kHostZoomMapKeyName, 106 new NonOwningZoomData( 107 HostZoomMap::GetForBrowserContext(browser_context))); 108 109 resource_context->DetachUserDataThread(); 110 } 111 112 } // namespace content 113