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