Index: chrome/browser/net/chrome_extensions_network_delegate.cc |
diff --git a/chrome/browser/net/chrome_extensions_network_delegate.cc b/chrome/browser/net/chrome_extensions_network_delegate.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..cbc9654ddea608e3895ee59068ab370df5e0b89e |
--- /dev/null |
+++ b/chrome/browser/net/chrome_extensions_network_delegate.cc |
@@ -0,0 +1,351 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/browser/net/chrome_extensions_network_delegate.h" |
+ |
+#include "net/base/net_errors.h" |
+ |
+#if defined(ENABLE_EXTENSIONS) |
+#include "chrome/browser/browser_process.h" |
+#include "chrome/browser/extensions/api/proxy/proxy_api.h" |
+#include "chrome/browser/extensions/api/web_request/web_request_api.h" |
+#include "chrome/browser/extensions/event_router_forwarder.h" |
+#include "chrome/browser/profiles/profile_manager.h" |
+#include "content/public/browser/browser_thread.h" |
+#include "content/public/browser/render_frame_host.h" |
+#include "content/public/browser/resource_request_info.h" |
+#include "extensions/browser/extension_system.h" |
+#include "extensions/browser/info_map.h" |
+#include "extensions/browser/process_manager.h" |
+#include "net/url_request/url_request.h" |
+ |
+using content::BrowserThread; |
+using content::ResourceRequestInfo; |
+ |
+namespace { |
+ |
+enum RequestStatus { REQUEST_STARTED, REQUEST_DONE }; |
+ |
+// Notifies the extensions::ProcessManager that a request has started or stopped |
+// for a particular RenderFrame. |
+void NotifyEPMRequestStatus(RequestStatus status, |
+ void* profile_id, |
+ int process_id, |
+ int render_frame_id) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ Profile* profile = reinterpret_cast<Profile*>(profile_id); |
+ if (!g_browser_process->profile_manager()->IsValidProfile(profile)) |
+ return; |
+ |
+ extensions::ProcessManager* process_manager = |
+ extensions::ExtensionSystem::Get(profile)->process_manager(); |
+ // This may be NULL in unit tests. |
+ if (!process_manager) |
+ return; |
+ |
+ // Will be NULL if the request was not issued on behalf of a renderer (e.g. a |
+ // system-level request). |
+ content::RenderFrameHost* render_frame_host = |
+ content::RenderFrameHost::FromID(process_id, render_frame_id); |
+ if (render_frame_host) { |
+ if (status == REQUEST_STARTED) { |
+ process_manager->OnNetworkRequestStarted(render_frame_host); |
+ } else if (status == REQUEST_DONE) { |
+ process_manager->OnNetworkRequestDone(render_frame_host); |
+ } else { |
+ NOTREACHED(); |
+ } |
+ } |
+} |
+ |
+void ForwardRequestStatus( |
+ RequestStatus status, net::URLRequest* request, void* profile_id) { |
+ const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request); |
+ if (!info) |
+ return; |
+ |
+ int process_id, render_frame_id; |
+ if (info->GetAssociatedRenderFrame(&process_id, &render_frame_id)) { |
+ BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
+ base::Bind(&NotifyEPMRequestStatus, |
+ status, profile_id, process_id, render_frame_id)); |
+ } |
+} |
+ |
+class ChromeExtensionsNetworkDelegateImpl |
+ : public ChromeExtensionsNetworkDelegate { |
+ public: |
+ explicit ChromeExtensionsNetworkDelegateImpl( |
+ extensions::EventRouterForwarder* event_router); |
+ virtual ~ChromeExtensionsNetworkDelegateImpl(); |
+ |
+ private: |
+ // ChromeExtensionsNetworkDelegate implementation. |
+ virtual void ForwardProxyErrors(net::URLRequest* request) OVERRIDE; |
+ virtual void ForwardStartRequestStatus(net::URLRequest* request) OVERRIDE; |
+ virtual void ForwardDoneRequestStatus(net::URLRequest* request) OVERRIDE; |
+ virtual int OnBeforeURLRequest(net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ GURL* new_url) OVERRIDE; |
+ virtual int OnBeforeSendHeaders(net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ net::HttpRequestHeaders* headers) OVERRIDE; |
+ virtual void OnSendHeaders(net::URLRequest* request, |
+ const net::HttpRequestHeaders& headers) OVERRIDE; |
+ virtual int OnHeadersReceived( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ const net::HttpResponseHeaders* original_response_headers, |
+ scoped_refptr<net::HttpResponseHeaders>* override_response_headers, |
+ GURL* allowed_unsafe_redirect_url) OVERRIDE; |
+ virtual void OnBeforeRedirect(net::URLRequest* request, |
+ const GURL& new_location) OVERRIDE; |
+ virtual void OnResponseStarted(net::URLRequest* request) OVERRIDE; |
+ virtual void OnCompleted(net::URLRequest* request, bool started) OVERRIDE; |
+ virtual void OnURLRequestDestroyed(net::URLRequest* request) OVERRIDE; |
+ virtual void OnPACScriptError(int line_number, |
+ const base::string16& error) OVERRIDE; |
+ virtual net::NetworkDelegate::AuthRequiredResponse OnAuthRequired( |
+ net::URLRequest* request, |
+ const net::AuthChallengeInfo& auth_info, |
+ const AuthCallback& callback, |
+ net::AuthCredentials* credentials) OVERRIDE; |
+ |
+ scoped_refptr<extensions::EventRouterForwarder> event_router_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(ChromeExtensionsNetworkDelegateImpl); |
+}; |
+ |
+ChromeExtensionsNetworkDelegateImpl::ChromeExtensionsNetworkDelegateImpl( |
+ extensions::EventRouterForwarder* event_router) { |
+ DCHECK(event_router); |
+ event_router_ = event_router; |
+} |
+ |
+ChromeExtensionsNetworkDelegateImpl::~ChromeExtensionsNetworkDelegateImpl() {} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::ForwardProxyErrors( |
+ net::URLRequest* request) { |
+ if (request->status().status() == net::URLRequestStatus::FAILED) { |
+ switch (request->status().error()) { |
+ case net::ERR_PROXY_AUTH_UNSUPPORTED: |
+ case net::ERR_PROXY_CONNECTION_FAILED: |
+ case net::ERR_TUNNEL_CONNECTION_FAILED: |
+ extensions::ProxyEventRouter::GetInstance()->OnProxyError( |
+ event_router_.get(), profile_, request->status().error()); |
+ } |
+ } |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::ForwardStartRequestStatus( |
+ net::URLRequest* request) { |
+ ForwardRequestStatus(REQUEST_STARTED, request, profile_); |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::ForwardDoneRequestStatus( |
+ net::URLRequest* request) { |
+ ForwardRequestStatus(REQUEST_DONE, request, profile_); |
+} |
+ |
+int ChromeExtensionsNetworkDelegateImpl::OnBeforeURLRequest( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ GURL* new_url) { |
+ return ExtensionWebRequestEventRouter::GetInstance()->OnBeforeRequest( |
+ profile_, extension_info_map_.get(), request, callback, new_url); |
+} |
+ |
+int ChromeExtensionsNetworkDelegateImpl::OnBeforeSendHeaders( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ net::HttpRequestHeaders* headers) { |
+ return ExtensionWebRequestEventRouter::GetInstance()->OnBeforeSendHeaders( |
+ profile_, extension_info_map_.get(), request, callback, headers); |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::OnSendHeaders( |
+ net::URLRequest* request, |
+ const net::HttpRequestHeaders& headers) { |
+ ExtensionWebRequestEventRouter::GetInstance()->OnSendHeaders( |
+ profile_, extension_info_map_.get(), request, headers); |
+} |
+ |
+int ChromeExtensionsNetworkDelegateImpl::OnHeadersReceived( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ const net::HttpResponseHeaders* original_response_headers, |
+ scoped_refptr<net::HttpResponseHeaders>* override_response_headers, |
+ GURL* allowed_unsafe_redirect_url) { |
+ return ExtensionWebRequestEventRouter::GetInstance()->OnHeadersReceived( |
+ profile_, |
+ extension_info_map_.get(), |
+ request, |
+ callback, |
+ original_response_headers, |
+ override_response_headers, |
+ allowed_unsafe_redirect_url); |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::OnBeforeRedirect( |
+ net::URLRequest* request, |
+ const GURL& new_location) { |
+ ExtensionWebRequestEventRouter::GetInstance()->OnBeforeRedirect( |
+ profile_, extension_info_map_.get(), request, new_location); |
+} |
+ |
+ |
+void ChromeExtensionsNetworkDelegateImpl::OnResponseStarted( |
+ net::URLRequest* request) { |
+ ExtensionWebRequestEventRouter::GetInstance()->OnResponseStarted( |
+ profile_, extension_info_map_.get(), request); |
+ ForwardProxyErrors(request); |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::OnCompleted( |
+ net::URLRequest* request, |
+ bool started) { |
+ if (request->status().status() == net::URLRequestStatus::SUCCESS) { |
+ bool is_redirect = request->response_headers() && |
+ net::HttpResponseHeaders::IsRedirectResponseCode( |
+ request->response_headers()->response_code()); |
+ if (!is_redirect) { |
+ ExtensionWebRequestEventRouter::GetInstance()->OnCompleted( |
+ profile_, extension_info_map_.get(), request); |
+ } |
+ return; |
+ } |
+ |
+ if (request->status().status() == net::URLRequestStatus::FAILED || |
+ request->status().status() == net::URLRequestStatus::CANCELED) { |
+ ExtensionWebRequestEventRouter::GetInstance()->OnErrorOccurred( |
+ profile_, extension_info_map_.get(), request, started); |
+ return; |
+ } |
+ |
+ NOTREACHED(); |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::OnURLRequestDestroyed( |
+ net::URLRequest* request) { |
+ ExtensionWebRequestEventRouter::GetInstance()->OnURLRequestDestroyed( |
+ profile_, request); |
+} |
+ |
+void ChromeExtensionsNetworkDelegateImpl::OnPACScriptError( |
+ int line_number, |
+ const base::string16& error) { |
+ extensions::ProxyEventRouter::GetInstance()->OnPACScriptError( |
+ event_router_.get(), profile_, line_number, error); |
+} |
+ |
+net::NetworkDelegate::AuthRequiredResponse |
+ChromeExtensionsNetworkDelegateImpl::OnAuthRequired( |
+ net::URLRequest* request, |
+ const net::AuthChallengeInfo& auth_info, |
+ const AuthCallback& callback, |
+ net::AuthCredentials* credentials) { |
+ return ExtensionWebRequestEventRouter::GetInstance()->OnAuthRequired( |
+ profile_, extension_info_map_.get(), request, auth_info, callback, |
+ credentials); |
+} |
+ |
+} // namespace |
+ |
+#endif // defined(ENABLE_EXTENSIONS) |
+ |
+// static |
+ChromeExtensionsNetworkDelegate* ChromeExtensionsNetworkDelegate::Create( |
+ extensions::EventRouterForwarder* event_router) { |
+#if defined(ENABLE_EXTENSIONS) |
+ return new ChromeExtensionsNetworkDelegateImpl(event_router); |
+#else |
+ return new ChromeExtensionsNetworkDelegate(); |
+#endif |
+} |
+ |
+ChromeExtensionsNetworkDelegate::ChromeExtensionsNetworkDelegate() |
+ : profile_(NULL) { |
+} |
+ |
+ChromeExtensionsNetworkDelegate::~ChromeExtensionsNetworkDelegate() {} |
+ |
+void ChromeExtensionsNetworkDelegate::set_extension_info_map( |
+ extensions::InfoMap* extension_info_map) { |
+#if defined(ENABLE_EXTENSIONS) |
+ extension_info_map_ = extension_info_map; |
+#endif |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::ForwardProxyErrors( |
Yoyo Zhou
2014/06/18 23:38:54
Is it necessary to override these?
Lei Zhang
2014/06/18 23:44:27
I could add #ifdefs for these three. Did you have
Yoyo Zhou
2014/06/18 23:55:25
The NetworkDelegate implementation does nothing he
|
+ net::URLRequest* request) { |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::ForwardStartRequestStatus( |
+ net::URLRequest* request) { |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::ForwardDoneRequestStatus( |
+ net::URLRequest* request) { |
+} |
+ |
+int ChromeExtensionsNetworkDelegate::OnBeforeURLRequest( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ GURL* new_url) { |
+ return net::OK; |
+} |
+ |
+int ChromeExtensionsNetworkDelegate::OnBeforeSendHeaders( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ net::HttpRequestHeaders* headers) { |
+ return net::OK; |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::OnSendHeaders( |
+ net::URLRequest* request, |
+ const net::HttpRequestHeaders& headers) { |
+} |
+ |
+int ChromeExtensionsNetworkDelegate::OnHeadersReceived( |
+ net::URLRequest* request, |
+ const net::CompletionCallback& callback, |
+ const net::HttpResponseHeaders* original_response_headers, |
+ scoped_refptr<net::HttpResponseHeaders>* override_response_headers, |
+ GURL* allowed_unsafe_redirect_url) { |
+ return net::OK; |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::OnBeforeRedirect( |
+ net::URLRequest* request, |
+ const GURL& new_location) { |
+} |
+ |
+ |
+void ChromeExtensionsNetworkDelegate::OnResponseStarted( |
+ net::URLRequest* request) { |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::OnCompleted( |
+ net::URLRequest* request, |
+ bool started) { |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::OnURLRequestDestroyed( |
+ net::URLRequest* request) { |
+} |
+ |
+void ChromeExtensionsNetworkDelegate::OnPACScriptError( |
+ int line_number, |
+ const base::string16& error) { |
+} |
+ |
+net::NetworkDelegate::AuthRequiredResponse |
+ChromeExtensionsNetworkDelegate::OnAuthRequired( |
+ net::URLRequest* request, |
+ const net::AuthChallengeInfo& auth_info, |
+ const AuthCallback& callback, |
+ net::AuthCredentials* credentials) { |
+ return net::NetworkDelegate::AUTH_REQUIRED_RESPONSE_NO_ACTION; |
+} |