1 // Copyright (c) 2011 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 "chrome/browser/net/chrome_network_delegate.h"
6
7 #include "base/logging.h"
8 #include "chrome/browser/custom_handlers/protocol_handler_registry.h"
9 #include "chrome/browser/extensions/extension_event_router_forwarder.h"
10 #include "chrome/browser/extensions/extension_proxy_api.h"
11 #include "chrome/browser/extensions/extension_webrequest_api.h"
12 #include "chrome/browser/prefs/pref_member.h"
13 #include "chrome/common/pref_names.h"
14 #include "content/browser/browser_thread.h"
15 #include "net/base/net_errors.h"
16 #include "net/http/http_request_headers.h"
17 #include "net/url_request/url_request.h"
18
19 namespace {
20
21 // If the |request| failed due to problems with a proxy, forward the error to
22 // the proxy extension API.
ForwardProxyErrors(net::URLRequest * request,ExtensionEventRouterForwarder * event_router,ProfileId profile_id)23 void ForwardProxyErrors(net::URLRequest* request,
24 ExtensionEventRouterForwarder* event_router,
25 ProfileId profile_id) {
26 if (request->status().status() == net::URLRequestStatus::FAILED) {
27 switch (request->status().os_error()) {
28 case net::ERR_PROXY_AUTH_UNSUPPORTED:
29 case net::ERR_PROXY_CONNECTION_FAILED:
30 case net::ERR_TUNNEL_CONNECTION_FAILED:
31 ExtensionProxyEventRouter::GetInstance()->OnProxyError(
32 event_router, profile_id, request->status().os_error());
33 }
34 }
35 }
36
37 } // namespace
38
ChromeNetworkDelegate(ExtensionEventRouterForwarder * event_router,ProfileId profile_id,BooleanPrefMember * enable_referrers,ProtocolHandlerRegistry * protocol_handler_registry)39 ChromeNetworkDelegate::ChromeNetworkDelegate(
40 ExtensionEventRouterForwarder* event_router,
41 ProfileId profile_id,
42 BooleanPrefMember* enable_referrers,
43 ProtocolHandlerRegistry* protocol_handler_registry)
44 : event_router_(event_router),
45 profile_id_(profile_id),
46 enable_referrers_(enable_referrers),
47 protocol_handler_registry_(protocol_handler_registry) {
48 DCHECK(event_router);
49 DCHECK(enable_referrers);
50 }
51
~ChromeNetworkDelegate()52 ChromeNetworkDelegate::~ChromeNetworkDelegate() {}
53
54 // static
InitializeReferrersEnabled(BooleanPrefMember * enable_referrers,PrefService * pref_service)55 void ChromeNetworkDelegate::InitializeReferrersEnabled(
56 BooleanPrefMember* enable_referrers,
57 PrefService* pref_service) {
58 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
59 enable_referrers->Init(prefs::kEnableReferrers, pref_service, NULL);
60 enable_referrers->MoveToThread(BrowserThread::IO);
61 }
62
OnBeforeURLRequest(net::URLRequest * request,net::CompletionCallback * callback,GURL * new_url)63 int ChromeNetworkDelegate::OnBeforeURLRequest(
64 net::URLRequest* request,
65 net::CompletionCallback* callback,
66 GURL* new_url) {
67 if (!enable_referrers_->GetValue())
68 request->set_referrer(std::string());
69 return ExtensionWebRequestEventRouter::GetInstance()->OnBeforeRequest(
70 profile_id_, event_router_.get(), request, callback, new_url);
71 }
72
OnBeforeSendHeaders(uint64 request_id,net::CompletionCallback * callback,net::HttpRequestHeaders * headers)73 int ChromeNetworkDelegate::OnBeforeSendHeaders(
74 uint64 request_id,
75 net::CompletionCallback* callback,
76 net::HttpRequestHeaders* headers) {
77 return ExtensionWebRequestEventRouter::GetInstance()->OnBeforeSendHeaders(
78 profile_id_, event_router_.get(), request_id, callback, headers);
79 }
80
OnResponseStarted(net::URLRequest * request)81 void ChromeNetworkDelegate::OnResponseStarted(net::URLRequest* request) {
82 ForwardProxyErrors(request, event_router_.get(), profile_id_);
83 }
84
OnReadCompleted(net::URLRequest * request,int bytes_read)85 void ChromeNetworkDelegate::OnReadCompleted(net::URLRequest* request,
86 int bytes_read) {
87 ForwardProxyErrors(request, event_router_.get(), profile_id_);
88 }
89
OnURLRequestDestroyed(net::URLRequest * request)90 void ChromeNetworkDelegate::OnURLRequestDestroyed(net::URLRequest* request) {
91 ExtensionWebRequestEventRouter::GetInstance()->OnURLRequestDestroyed(
92 profile_id_, request);
93 }
94
OnMaybeCreateURLRequestJob(net::URLRequest * request)95 net::URLRequestJob* ChromeNetworkDelegate::OnMaybeCreateURLRequestJob(
96 net::URLRequest* request) {
97 if (!protocol_handler_registry_)
98 return NULL;
99 return protocol_handler_registry_->MaybeCreateJob(request);
100 }
101