OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 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 "net/proxy/mojo_proxy_resolver_impl.h" | |
6 | |
7 #include "base/stl_util.h" | |
8 #include "mojo/common/common_type_converters.h" | |
9 #include "net/base/net_errors.h" | |
10 #include "net/base/net_log.h" | |
11 #include "net/proxy/proxy_info.h" | |
12 #include "net/proxy/proxy_resolver.h" | |
13 #include "net/proxy/proxy_resolver_script_data.h" | |
14 #include "net/proxy/type_converters.h" | |
15 | |
16 namespace net { | |
17 | |
18 class MojoProxyResolverImpl::Job : public mojo::ErrorHandler { | |
19 public: | |
20 Job(interfaces::ProxyResolverRequestClientPtr client, | |
21 MojoProxyResolverImpl* resolver); | |
22 ~Job() override; | |
23 | |
24 void Start(const GURL& url); | |
25 | |
26 private: | |
27 // mojo::ErrorHandler override. | |
28 void OnConnectionError() override; | |
29 | |
30 void GetProxyDone(int32_t error); | |
31 | |
32 MojoProxyResolverImpl* resolver_; | |
33 | |
34 interfaces::ProxyResolverRequestClientPtr client_; | |
35 ProxyInfo result_; | |
36 net::ProxyResolver::RequestHandle request_handle_; | |
37 | |
38 DISALLOW_COPY_AND_ASSIGN(Job); | |
39 }; | |
40 | |
41 MojoProxyResolverImpl::MojoProxyResolverImpl( | |
42 scoped_ptr<net::ProxyResolver> resolver) | |
43 : resolver_(resolver.Pass()) { | |
44 DCHECK(resolver_->expects_pac_bytes()); | |
45 } | |
46 | |
47 MojoProxyResolverImpl::~MojoProxyResolverImpl() { | |
48 resolver_->CancelSetPacScript(); | |
49 if (!set_pac_script_callback_.is_null()) | |
50 set_pac_script_callback_.Run(ERR_ABORTED); | |
51 STLDeleteElements(&resolve_jobs_); | |
52 } | |
53 | |
54 void MojoProxyResolverImpl::SetPacScript( | |
55 const mojo::String& data, | |
56 const mojo::Callback<void(int32_t)>& callback) { | |
57 set_pac_script_callback_ = callback; | |
Anand Mistry (off Chromium)
2015/02/10 03:54:08
DCHECK(set_pac_script_callback_.is_null());
Sam McNally
2015/02/10 06:27:42
Done.
| |
58 int result = resolver_->SetPacScript( | |
59 ProxyResolverScriptData::FromUTF8(data), | |
60 base::Bind(&MojoProxyResolverImpl::SetPacScriptDone, | |
61 base::Unretained(this))); | |
62 if (result != ERR_IO_PENDING) | |
63 SetPacScriptDone(result); | |
64 } | |
65 | |
66 void MojoProxyResolverImpl::GetProxyForUrl( | |
Anand Mistry (off Chromium)
2015/02/10 03:54:08
I think it would be nice to have a few DVLOGs scat
Sam McNally
2015/02/10 06:27:42
Done.
| |
67 const mojo::String& url, | |
68 interfaces::ProxyResolverRequestClientPtr client) { | |
69 Job* job = new Job(client.Pass(), this); | |
70 bool inserted = resolve_jobs_.insert(job).second; | |
71 DCHECK(inserted); | |
72 job->Start(url.To<GURL>()); | |
Anand Mistry (off Chromium)
2015/02/10 03:54:08
URL is a property of the job. Shouldn't it be in t
Sam McNally
2015/02/10 06:27:41
Done.
| |
73 } | |
74 | |
75 void MojoProxyResolverImpl::DeleteJob(Job* job) { | |
76 size_t num_erased = resolve_jobs_.erase(job); | |
77 DCHECK(num_erased); | |
78 delete job; | |
79 } | |
80 | |
81 void MojoProxyResolverImpl::SetPacScriptDone(int result) { | |
82 DCHECK(!set_pac_script_callback_.is_null()); | |
83 set_pac_script_callback_.Run(result); | |
84 set_pac_script_callback_ = mojo::Callback<void(int32_t)>(); | |
Anand Mistry (off Chromium)
2015/02/10 03:54:08
I've been using .reset() instead. Not sure which o
Sam McNally
2015/02/10 06:27:41
Changed to reset(). I got used to mojo::Callbacks
| |
85 } | |
86 | |
87 MojoProxyResolverImpl::Job::Job( | |
88 interfaces::ProxyResolverRequestClientPtr client, | |
89 MojoProxyResolverImpl* resolver) | |
90 : resolver_(resolver), client_(client.Pass()), request_handle_(nullptr) { | |
91 } | |
92 | |
93 MojoProxyResolverImpl::Job::~Job() { | |
94 if (request_handle_) | |
95 resolver_->resolver_->CancelRequest(request_handle_); | |
96 } | |
97 | |
98 void MojoProxyResolverImpl::Job::Start(const GURL& url) { | |
99 int result = resolver_->resolver_->GetProxyForURL( | |
100 url, &result_, base::Bind(&Job::GetProxyDone, base::Unretained(this)), | |
101 &request_handle_, BoundNetLog()); | |
102 if (result != ERR_IO_PENDING) { | |
103 GetProxyDone(result); | |
104 return; | |
105 } | |
106 client_.set_error_handler(this); | |
107 } | |
108 | |
109 void MojoProxyResolverImpl::Job::GetProxyDone(int32_t error) { | |
110 request_handle_ = nullptr; | |
111 if (error) { | |
112 client_->ReportResult(error, mojo::Array<interfaces::ProxyServerPtr>()); | |
113 } else { | |
114 client_->ReportResult(OK, mojo::Array<interfaces::ProxyServerPtr>::From( | |
115 result_.proxy_list().GetAll())); | |
116 } | |
117 resolver_->DeleteJob(this); | |
118 } | |
119 | |
120 void MojoProxyResolverImpl::Job::OnConnectionError() { | |
121 resolver_->DeleteJob(this); | |
122 } | |
123 | |
124 } // namespace net | |
OLD | NEW |