115 lines
4.2 KiB
C++
115 lines
4.2 KiB
C++
// Copyright (c) 2013 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#include "atom/browser/atom_browser_context.h"
|
|
|
|
#include "atom/browser/atom_browser_main_parts.h"
|
|
#include "atom/browser/net/atom_url_request_job_factory.h"
|
|
#include "atom/browser/net/asar/asar_protocol_handler.h"
|
|
#include "atom/browser/net/http_protocol_handler.h"
|
|
#include "atom/browser/web_view_manager.h"
|
|
#include "atom/common/options_switches.h"
|
|
#include "base/command_line.h"
|
|
#include "base/threading/sequenced_worker_pool.h"
|
|
#include "base/threading/worker_pool.h"
|
|
#include "chrome/browser/browser_process.h"
|
|
#include "content/public/browser/browser_thread.h"
|
|
#include "content/public/common/url_constants.h"
|
|
#include "net/ftp/ftp_network_layer.h"
|
|
#include "net/url_request/data_protocol_handler.h"
|
|
#include "net/url_request/ftp_protocol_handler.h"
|
|
#include "net/url_request/url_request_intercepting_job_factory.h"
|
|
#include "net/url_request/url_request_context.h"
|
|
#include "url/url_constants.h"
|
|
|
|
using content::BrowserThread;
|
|
|
|
namespace atom {
|
|
|
|
namespace {
|
|
|
|
class NoCacheBackend : public net::HttpCache::BackendFactory {
|
|
int CreateBackend(net::NetLog* net_log,
|
|
scoped_ptr<disk_cache::Backend>* backend,
|
|
const net::CompletionCallback& callback) override {
|
|
return net::ERR_FAILED;
|
|
}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
AtomBrowserContext::AtomBrowserContext()
|
|
: fake_browser_process_(new BrowserProcess),
|
|
job_factory_(new AtomURLRequestJobFactory) {
|
|
}
|
|
|
|
AtomBrowserContext::~AtomBrowserContext() {
|
|
}
|
|
|
|
net::URLRequestJobFactory* AtomBrowserContext::CreateURLRequestJobFactory(
|
|
content::ProtocolHandlerMap* handlers,
|
|
content::URLRequestInterceptorScopedVector* interceptors) {
|
|
scoped_ptr<AtomURLRequestJobFactory> job_factory(job_factory_);
|
|
|
|
for (content::ProtocolHandlerMap::iterator it = handlers->begin();
|
|
it != handlers->end(); ++it)
|
|
job_factory->SetProtocolHandler(it->first, it->second.release());
|
|
handlers->clear();
|
|
|
|
job_factory->SetProtocolHandler(
|
|
url::kDataScheme, new net::DataProtocolHandler);
|
|
job_factory->SetProtocolHandler(
|
|
url::kFileScheme, new asar::AsarProtocolHandler(
|
|
BrowserThread::GetBlockingPool()->GetTaskRunnerWithShutdownBehavior(
|
|
base::SequencedWorkerPool::SKIP_ON_SHUTDOWN)));
|
|
job_factory->SetProtocolHandler(
|
|
url::kHttpScheme, new HttpProtocolHandler(url::kHttpScheme));
|
|
job_factory->SetProtocolHandler(
|
|
url::kHttpsScheme, new HttpProtocolHandler(url::kHttpsScheme));
|
|
job_factory->SetProtocolHandler(
|
|
url::kWsScheme, new HttpProtocolHandler(url::kWsScheme));
|
|
job_factory->SetProtocolHandler(
|
|
url::kWssScheme, new HttpProtocolHandler(url::kWssScheme));
|
|
|
|
auto host_resolver = url_request_context_getter()
|
|
->GetURLRequestContext()
|
|
->host_resolver();
|
|
job_factory->SetProtocolHandler(
|
|
url::kFtpScheme, new net::FtpProtocolHandler(
|
|
new net::FtpNetworkLayer(host_resolver)));
|
|
|
|
// Set up interceptors in the reverse order.
|
|
scoped_ptr<net::URLRequestJobFactory> top_job_factory = job_factory.Pass();
|
|
content::URLRequestInterceptorScopedVector::reverse_iterator it;
|
|
for (it = interceptors->rbegin(); it != interceptors->rend(); ++it)
|
|
top_job_factory.reset(new net::URLRequestInterceptingJobFactory(
|
|
top_job_factory.Pass(), make_scoped_ptr(*it)));
|
|
interceptors->weak_clear();
|
|
|
|
return top_job_factory.release();
|
|
}
|
|
|
|
net::HttpCache::BackendFactory*
|
|
AtomBrowserContext::CreateHttpCacheBackendFactory(
|
|
const base::FilePath& base_path) {
|
|
base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
|
|
if (command_line->HasSwitch(switches::kDisableHttpCache))
|
|
return new NoCacheBackend;
|
|
else
|
|
return brightray::BrowserContext::CreateHttpCacheBackendFactory(base_path);
|
|
}
|
|
|
|
content::BrowserPluginGuestManager* AtomBrowserContext::GetGuestManager() {
|
|
if (!guest_manager_)
|
|
guest_manager_.reset(new WebViewManager(this));
|
|
return guest_manager_.get();
|
|
}
|
|
|
|
// static
|
|
AtomBrowserContext* AtomBrowserContext::Get() {
|
|
return static_cast<AtomBrowserContext*>(
|
|
AtomBrowserMainParts::Get()->browser_context());
|
|
}
|
|
|
|
} // namespace atom
|