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
ResolveProxyMsgHelper(net::ProxyService * proxy_service)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
OnMessageReceived(const IPC::Message & message,bool * message_was_ok)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
OnResolveProxy(const GURL & url,IPC::Message * reply_msg)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
OnResolveProxyCompleted(int result)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
StartPendingRequest()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
GetProxyService(scoped_refptr<net::ProxyService> * out) const85 bool ResolveProxyMsgHelper::GetProxyService(
86 scoped_refptr<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 net::URLRequestContextGetter* context_getter =
95 Profile::GetDefaultRequestContext();
96 if (!context_getter)
97 return false;
98
99 // Otherwise use the browser's global proxy service.
100 *out = context_getter->GetURLRequestContext()->proxy_service();
101 return true;
102 }
103
~ResolveProxyMsgHelper()104 ResolveProxyMsgHelper::~ResolveProxyMsgHelper() {
105 // Clear all pending requests.
106 if (!pending_requests_.empty()) {
107 PendingRequest req = pending_requests_.front();
108 proxy_service_->CancelPacRequest(req.pac_req);
109 }
110
111 for (PendingRequestList::iterator it = pending_requests_.begin();
112 it != pending_requests_.end();
113 ++it) {
114 delete it->reply_msg;
115 }
116
117 proxy_service_ = NULL;
118 pending_requests_.clear();
119 }
120