Home | History | Annotate | Download | only in browser 
      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