| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 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 "headless/lib/browser/headless_browser_context.h" | |
| 6 | |
| 7 #include <memory> | |
| 8 | |
| 9 #include "base/path_service.h" | |
| 10 #include "content/public/browser/browser_thread.h" | |
| 11 #include "content/public/browser/resource_context.h" | |
| 12 #include "content/public/browser/storage_partition.h" | |
| 13 #include "headless/lib/browser/headless_url_request_context_getter.h" | |
| 14 #include "net/url_request/url_request_context.h" | |
| 15 | |
| 16 namespace headless { | |
| 17 | |
| 18 // Contains net::URLRequestContextGetter required for resource loading. | |
| 19 // Must be destructed on the IO thread as per content::ResourceContext | |
| 20 // requirements. | |
| 21 class HeadlessResourceContext : public content::ResourceContext { | |
| 22 public: | |
| 23 HeadlessResourceContext(); | |
| 24 ~HeadlessResourceContext() override; | |
| 25 | |
| 26 // ResourceContext implementation: | |
| 27 net::HostResolver* GetHostResolver() override; | |
| 28 net::URLRequestContext* GetRequestContext() override; | |
| 29 | |
| 30 // Configure the URL request context getter to be used for resource fetching. | |
| 31 // Must be called before any of the other methods of this class are used. Must | |
| 32 // be called on the browser UI thread. | |
| 33 void set_url_request_context_getter( | |
| 34 scoped_refptr<net::URLRequestContextGetter> url_request_context_getter) { | |
| 35 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
| 36 url_request_context_getter_ = std::move(url_request_context_getter); | |
| 37 } | |
| 38 | |
| 39 net::URLRequestContextGetter* url_request_context_getter() { | |
| 40 return url_request_context_getter_.get(); | |
| 41 } | |
| 42 | |
| 43 private: | |
| 44 scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_; | |
| 45 | |
| 46 DISALLOW_COPY_AND_ASSIGN(HeadlessResourceContext); | |
| 47 }; | |
| 48 | |
| 49 HeadlessResourceContext::HeadlessResourceContext() { | |
| 50 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | |
| 51 } | |
| 52 | |
| 53 HeadlessResourceContext::~HeadlessResourceContext() { | |
| 54 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); | |
| 55 } | |
| 56 | |
| 57 net::HostResolver* HeadlessResourceContext::GetHostResolver() { | |
| 58 CHECK(url_request_context_getter_); | |
| 59 return url_request_context_getter_->GetURLRequestContext()->host_resolver(); | |
| 60 } | |
| 61 | |
| 62 net::URLRequestContext* HeadlessResourceContext::GetRequestContext() { | |
| 63 CHECK(url_request_context_getter_); | |
| 64 return url_request_context_getter_->GetURLRequestContext(); | |
| 65 } | |
| 66 | |
| 67 HeadlessBrowserContext::HeadlessBrowserContext( | |
| 68 HeadlessBrowser::Options* options) | |
| 69 : resource_context_(new HeadlessResourceContext), options_(options) { | |
| 70 InitWhileIOAllowed(); | |
| 71 } | |
| 72 | |
| 73 HeadlessBrowserContext::~HeadlessBrowserContext() { | |
| 74 if (resource_context_) { | |
| 75 content::BrowserThread::DeleteSoon(content::BrowserThread::IO, FROM_HERE, | |
| 76 resource_context_.release()); | |
| 77 } | |
| 78 } | |
| 79 | |
| 80 void HeadlessBrowserContext::InitWhileIOAllowed() { | |
| 81 // TODO(skyostil): Allow the embedder to override this. | |
| 82 PathService::Get(base::DIR_EXE, &path_); | |
| 83 BrowserContext::Initialize(this, path_); | |
| 84 } | |
| 85 | |
| 86 std::unique_ptr<content::ZoomLevelDelegate> | |
| 87 HeadlessBrowserContext::CreateZoomLevelDelegate( | |
| 88 const base::FilePath& partition_path) { | |
| 89 return std::unique_ptr<content::ZoomLevelDelegate>(); | |
| 90 } | |
| 91 | |
| 92 base::FilePath HeadlessBrowserContext::GetPath() const { | |
| 93 return path_; | |
| 94 } | |
| 95 | |
| 96 bool HeadlessBrowserContext::IsOffTheRecord() const { | |
| 97 return false; | |
| 98 } | |
| 99 | |
| 100 content::ResourceContext* HeadlessBrowserContext::GetResourceContext() { | |
| 101 return resource_context_.get(); | |
| 102 } | |
| 103 | |
| 104 content::DownloadManagerDelegate* | |
| 105 HeadlessBrowserContext::GetDownloadManagerDelegate() { | |
| 106 return nullptr; | |
| 107 } | |
| 108 | |
| 109 content::BrowserPluginGuestManager* HeadlessBrowserContext::GetGuestManager() { | |
| 110 // TODO(altimin): Should be non-null? (is null in content/shell). | |
| 111 return nullptr; | |
| 112 } | |
| 113 | |
| 114 storage::SpecialStoragePolicy* | |
| 115 HeadlessBrowserContext::GetSpecialStoragePolicy() { | |
| 116 return nullptr; | |
| 117 } | |
| 118 | |
| 119 content::PushMessagingService* | |
| 120 HeadlessBrowserContext::GetPushMessagingService() { | |
| 121 return nullptr; | |
| 122 } | |
| 123 | |
| 124 content::SSLHostStateDelegate* | |
| 125 HeadlessBrowserContext::GetSSLHostStateDelegate() { | |
| 126 return nullptr; | |
| 127 } | |
| 128 | |
| 129 content::PermissionManager* HeadlessBrowserContext::GetPermissionManager() { | |
| 130 return nullptr; | |
| 131 } | |
| 132 | |
| 133 content::BackgroundSyncController* | |
| 134 HeadlessBrowserContext::GetBackgroundSyncController() { | |
| 135 return nullptr; | |
| 136 } | |
| 137 | |
| 138 net::URLRequestContextGetter* HeadlessBrowserContext::CreateRequestContext( | |
| 139 content::ProtocolHandlerMap* protocol_handlers, | |
| 140 content::URLRequestInterceptorScopedVector request_interceptors) { | |
| 141 scoped_refptr<HeadlessURLRequestContextGetter> url_request_context_getter( | |
| 142 new HeadlessURLRequestContextGetter( | |
| 143 content::BrowserThread::GetMessageLoopProxyForThread( | |
| 144 content::BrowserThread::IO), | |
| 145 content::BrowserThread::GetMessageLoopProxyForThread( | |
| 146 content::BrowserThread::FILE), | |
| 147 protocol_handlers, std::move(request_interceptors), options())); | |
| 148 resource_context_->set_url_request_context_getter(url_request_context_getter); | |
| 149 return url_request_context_getter.get(); | |
| 150 } | |
| 151 | |
| 152 net::URLRequestContextGetter* | |
| 153 HeadlessBrowserContext::CreateRequestContextForStoragePartition( | |
| 154 const base::FilePath& partition_path, | |
| 155 bool in_memory, | |
| 156 content::ProtocolHandlerMap* protocol_handlers, | |
| 157 content::URLRequestInterceptorScopedVector request_interceptors) { | |
| 158 return nullptr; | |
| 159 } | |
| 160 | |
| 161 net::URLRequestContextGetter* | |
| 162 HeadlessBrowserContext::CreateMediaRequestContext() { | |
| 163 return resource_context_->url_request_context_getter(); | |
| 164 } | |
| 165 | |
| 166 net::URLRequestContextGetter* | |
| 167 HeadlessBrowserContext::CreateMediaRequestContextForStoragePartition( | |
| 168 const base::FilePath& partition_path, | |
| 169 bool in_memory) { | |
| 170 return nullptr; | |
| 171 } | |
| 172 | |
| 173 void HeadlessBrowserContext::SetOptionsForTesting( | |
| 174 HeadlessBrowser::Options* options) { | |
| 175 options_ = options; | |
| 176 } | |
| 177 | |
| 178 } // namespace headless | |
| OLD | NEW |