| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/resolve_proxy_msg_helper.h" | 5 #include "content/browser/resolve_proxy_msg_helper.h" |
| 6 | 6 |
| 7 #include "base/compiler_specific.h" | 7 #include "base/compiler_specific.h" |
| 8 #include "content/browser/content_browser_client.h" | 8 #include "content/common/view_messages.h" |
| 9 #include "content/common/child_process_messages.h" | |
| 10 #include "content/common/content_client.h" | |
| 11 #include "net/base/net_errors.h" | 9 #include "net/base/net_errors.h" |
| 12 #include "net/url_request/url_request_context.h" | 10 #include "net/url_request/url_request_context.h" |
| 13 #include "net/url_request/url_request_context_getter.h" | 11 #include "net/url_request/url_request_context_getter.h" |
| 14 | 12 |
| 13 ResolveProxyMsgHelper::ResolveProxyMsgHelper( |
| 14 net::URLRequestContextGetter* getter) |
| 15 : ALLOW_THIS_IN_INITIALIZER_LIST(callback_( |
| 16 this, &ResolveProxyMsgHelper::OnResolveProxyCompleted)), |
| 17 context_getter_(getter), |
| 18 proxy_service_(NULL) { |
| 19 } |
| 20 |
| 15 ResolveProxyMsgHelper::ResolveProxyMsgHelper(net::ProxyService* proxy_service) | 21 ResolveProxyMsgHelper::ResolveProxyMsgHelper(net::ProxyService* proxy_service) |
| 16 : proxy_service_(NULL), | 22 : ALLOW_THIS_IN_INITIALIZER_LIST(callback_( |
| 17 ALLOW_THIS_IN_INITIALIZER_LIST(callback_( | |
| 18 this, &ResolveProxyMsgHelper::OnResolveProxyCompleted)), | 23 this, &ResolveProxyMsgHelper::OnResolveProxyCompleted)), |
| 19 proxy_service_override_(proxy_service) { | 24 proxy_service_(proxy_service) { |
| 20 } | 25 } |
| 21 | 26 |
| 22 bool ResolveProxyMsgHelper::OnMessageReceived(const IPC::Message& message, | 27 bool ResolveProxyMsgHelper::OnMessageReceived(const IPC::Message& message, |
| 23 bool* message_was_ok) { | 28 bool* message_was_ok) { |
| 24 bool handled = true; | 29 bool handled = true; |
| 25 IPC_BEGIN_MESSAGE_MAP_EX(ResolveProxyMsgHelper, message, *message_was_ok) | 30 IPC_BEGIN_MESSAGE_MAP_EX(ResolveProxyMsgHelper, message, *message_was_ok) |
| 26 IPC_MESSAGE_HANDLER_DELAY_REPLY(ChildProcessHostMsg_ResolveProxy, | 31 IPC_MESSAGE_HANDLER_DELAY_REPLY(ViewHostMsg_ResolveProxy, OnResolveProxy) |
| 27 OnResolveProxy) | |
| 28 IPC_MESSAGE_UNHANDLED(handled = false) | 32 IPC_MESSAGE_UNHANDLED(handled = false) |
| 29 IPC_END_MESSAGE_MAP() | 33 IPC_END_MESSAGE_MAP() |
| 30 return handled; | 34 return handled; |
| 31 } | 35 } |
| 32 | 36 |
| 33 void ResolveProxyMsgHelper::OnResolveProxy(const GURL& url, | 37 void ResolveProxyMsgHelper::OnResolveProxy(const GURL& url, |
| 34 IPC::Message* reply_msg) { | 38 IPC::Message* reply_msg) { |
| 35 // Enqueue the pending request. | 39 // Enqueue the pending request. |
| 36 pending_requests_.push_back(PendingRequest(url, reply_msg)); | 40 pending_requests_.push_back(PendingRequest(url, reply_msg)); |
| 37 | 41 |
| 38 // If nothing is in progress, start. | 42 // If nothing is in progress, start. |
| 39 if (pending_requests_.size() == 1) | 43 if (pending_requests_.size() == 1) |
| 40 StartPendingRequest(); | 44 StartPendingRequest(); |
| 41 } | 45 } |
| 42 | 46 |
| 43 void ResolveProxyMsgHelper::OnResolveProxyCompleted(int result) { | 47 void ResolveProxyMsgHelper::OnResolveProxyCompleted(int result) { |
| 44 CHECK(!pending_requests_.empty()); | 48 CHECK(!pending_requests_.empty()); |
| 45 | 49 |
| 46 const PendingRequest& completed_req = pending_requests_.front(); | 50 const PendingRequest& completed_req = pending_requests_.front(); |
| 47 ChildProcessHostMsg_ResolveProxy::WriteReplyParams( | 51 ViewHostMsg_ResolveProxy::WriteReplyParams( |
| 48 completed_req.reply_msg, result, proxy_info_.ToPacString()); | 52 completed_req.reply_msg, result == net::OK, proxy_info_.ToPacString()); |
| 49 Send(completed_req.reply_msg); | 53 Send(completed_req.reply_msg); |
| 50 | 54 |
| 51 // Clear the current (completed) request. | 55 // Clear the current (completed) request. |
| 52 pending_requests_.pop_front(); | 56 pending_requests_.pop_front(); |
| 53 proxy_service_ = NULL; | |
| 54 | 57 |
| 55 // Start the next request. | 58 // Start the next request. |
| 56 if (!pending_requests_.empty()) | 59 if (!pending_requests_.empty()) |
| 57 StartPendingRequest(); | 60 StartPendingRequest(); |
| 58 } | 61 } |
| 59 | 62 |
| 60 void ResolveProxyMsgHelper::StartPendingRequest() { | 63 void ResolveProxyMsgHelper::StartPendingRequest() { |
| 61 PendingRequest& req = pending_requests_.front(); | 64 PendingRequest& req = pending_requests_.front(); |
| 62 | 65 |
| 63 // Verify the request wasn't started yet. | 66 // Verify the request wasn't started yet. |
| 64 DCHECK(NULL == req.pac_req); | 67 DCHECK(NULL == req.pac_req); |
| 65 DCHECK(NULL == proxy_service_); | 68 |
| 69 if (context_getter_.get()) { |
| 70 proxy_service_ = context_getter_->GetURLRequestContext()->proxy_service(); |
| 71 context_getter_ = NULL; |
| 72 } |
| 66 | 73 |
| 67 // Start the request. | 74 // Start the request. |
| 68 bool ok = GetProxyService(&proxy_service_); | |
| 69 | |
| 70 if (!ok) { | |
| 71 // During shutdown, there may be no ProxyService to use, because the | |
| 72 // default ChromeURLRequestContext has already been NULL-ed out. | |
| 73 LOG(WARNING) << "Failed getting default URLRequestContext"; | |
| 74 OnResolveProxyCompleted(net::ERR_FAILED); | |
| 75 return; | |
| 76 } | |
| 77 | |
| 78 int result = proxy_service_->ResolveProxy( | 75 int result = proxy_service_->ResolveProxy( |
| 79 req.url, &proxy_info_, &callback_, &req.pac_req, net::BoundNetLog()); | 76 req.url, &proxy_info_, &callback_, &req.pac_req, net::BoundNetLog()); |
| 80 | 77 |
| 81 // Completed synchronously. | 78 // Completed synchronously. |
| 82 if (result != net::ERR_IO_PENDING) | 79 if (result != net::ERR_IO_PENDING) |
| 83 OnResolveProxyCompleted(result); | 80 OnResolveProxyCompleted(result); |
| 84 } | 81 } |
| 85 | 82 |
| 86 bool ResolveProxyMsgHelper::GetProxyService(net::ProxyService** out) const { | |
| 87 // Unit-tests specify their own proxy service to use. | |
| 88 if (proxy_service_override_) { | |
| 89 *out = proxy_service_override_; | |
| 90 return true; | |
| 91 } | |
| 92 | |
| 93 // If there is no default request context (say during shut down). | |
| 94 // Deprecated; see http://crbug.com/92361 | |
| 95 net::URLRequestContextGetter* context_getter = | |
| 96 content::GetContentClient()->browser()-> | |
| 97 GetDefaultRequestContextDeprecatedCrBug64339(); | |
| 98 if (!context_getter) | |
| 99 return false; | |
| 100 | |
| 101 // Otherwise use the browser's global proxy service. | |
| 102 *out = context_getter->GetURLRequestContext()->proxy_service(); | |
| 103 return true; | |
| 104 } | |
| 105 | |
| 106 ResolveProxyMsgHelper::~ResolveProxyMsgHelper() { | 83 ResolveProxyMsgHelper::~ResolveProxyMsgHelper() { |
| 107 // Clear all pending requests if the ProxyService is still alive (if we have a | 84 // Clear all pending requests if the ProxyService is still alive (if we have a |
| 108 // default request context or override). | 85 // default request context or override). |
| 109 // Deprecated; see http://crbug.com/92361 | 86 if (!pending_requests_.empty()) { |
| 110 net::URLRequestContextGetter* context_getter = | |
| 111 content::GetContentClient()->browser()-> | |
| 112 GetDefaultRequestContextDeprecatedCrBug64339(); | |
| 113 if (!pending_requests_.empty() && | |
| 114 (context_getter || proxy_service_override_)) { | |
| 115 PendingRequest req = pending_requests_.front(); | 87 PendingRequest req = pending_requests_.front(); |
| 116 proxy_service_->CancelPacRequest(req.pac_req); | 88 proxy_service_->CancelPacRequest(req.pac_req); |
| 117 } | 89 } |
| 118 | 90 |
| 119 for (PendingRequestList::iterator it = pending_requests_.begin(); | 91 for (PendingRequestList::iterator it = pending_requests_.begin(); |
| 120 it != pending_requests_.end(); | 92 it != pending_requests_.end(); |
| 121 ++it) { | 93 ++it) { |
| 122 delete it->reply_msg; | 94 delete it->reply_msg; |
| 123 } | 95 } |
| 124 | 96 |
| 125 proxy_service_ = NULL; | |
| 126 pending_requests_.clear(); | 97 pending_requests_.clear(); |
| 127 } | 98 } |
| OLD | NEW |