Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 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 "net/dns/mojo_host_resolver_impl.h" | 5 #include "net/dns/mojo_host_resolver_impl.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/stl_util.h" | |
| 10 #include "net/base/address_list.h" | 9 #include "net/base/address_list.h" |
| 11 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
| 12 #include "net/base/network_interfaces.h" | 11 #include "net/base/network_interfaces.h" |
| 13 #include "net/dns/host_resolver.h" | 12 #include "net/dns/host_resolver.h" |
| 14 | 13 |
| 15 namespace net { | 14 namespace net { |
| 16 | 15 |
| 17 // Handles host resolution for a single request and sends a response when done. | 16 // Handles host resolution for a single request and sends a response when done. |
| 18 // Also detects connection errors for HostResolverRequestClient and cancels the | 17 // Also detects connection errors for HostResolverRequestClient and cancels the |
| 19 // outstanding resolve request. Owned by MojoHostResolverImpl. | 18 // outstanding resolve request. Owned by MojoHostResolverImpl. |
| 20 class MojoHostResolverImpl::Job { | 19 class MojoHostResolverImpl::Job { |
| 21 public: | 20 public: |
| 22 Job(MojoHostResolverImpl* resolver_service, | 21 Job(MojoHostResolverImpl* resolver_service, |
| 23 net::HostResolver* resolver, | 22 net::HostResolver* resolver, |
| 24 const net::HostResolver::RequestInfo& request_info, | 23 const net::HostResolver::RequestInfo& request_info, |
| 25 const NetLogWithSource& net_log, | 24 const NetLogWithSource& net_log, |
| 26 interfaces::HostResolverRequestClientPtr client); | 25 interfaces::HostResolverRequestClientPtr client); |
| 27 ~Job(); | 26 ~Job(); |
| 28 | 27 |
| 28 void set_iter(std::list<Job>::iterator iter) { iter_ = iter; } | |
| 29 | |
| 29 void Start(); | 30 void Start(); |
| 30 | 31 |
| 31 private: | 32 private: |
| 32 // Completion callback for the HostResolver::Resolve request. | 33 // Completion callback for the HostResolver::Resolve request. |
| 33 void OnResolveDone(int result); | 34 void OnResolveDone(int result); |
| 34 | 35 |
| 35 // Mojo error handler. | 36 // Mojo error handler. |
| 36 void OnConnectionError(); | 37 void OnConnectionError(); |
| 37 | 38 |
| 38 MojoHostResolverImpl* resolver_service_; | 39 MojoHostResolverImpl* resolver_service_; |
| 40 // This Job's iterator in |resolver_service_|, so the Job may be removed on | |
| 41 // completion. | |
| 42 std::list<Job>::iterator iter_; | |
| 39 net::HostResolver* resolver_; | 43 net::HostResolver* resolver_; |
| 40 net::HostResolver::RequestInfo request_info_; | 44 net::HostResolver::RequestInfo request_info_; |
| 41 const NetLogWithSource net_log_; | 45 const NetLogWithSource net_log_; |
| 42 interfaces::HostResolverRequestClientPtr client_; | 46 interfaces::HostResolverRequestClientPtr client_; |
| 43 std::unique_ptr<net::HostResolver::Request> request_; | 47 std::unique_ptr<net::HostResolver::Request> request_; |
| 44 AddressList result_; | 48 AddressList result_; |
| 45 base::ThreadChecker thread_checker_; | 49 base::ThreadChecker thread_checker_; |
| 46 }; | 50 }; |
| 47 | 51 |
| 48 MojoHostResolverImpl::MojoHostResolverImpl(net::HostResolver* resolver, | 52 MojoHostResolverImpl::MojoHostResolverImpl(net::HostResolver* resolver, |
| 49 const NetLogWithSource& net_log) | 53 const NetLogWithSource& net_log) |
| 50 : resolver_(resolver), net_log_(net_log) {} | 54 : resolver_(resolver), net_log_(net_log) {} |
| 51 | 55 |
| 52 MojoHostResolverImpl::~MojoHostResolverImpl() { | 56 MojoHostResolverImpl::~MojoHostResolverImpl() { |
| 53 DCHECK(thread_checker_.CalledOnValidThread()); | 57 DCHECK(thread_checker_.CalledOnValidThread()); |
| 54 base::STLDeleteElements(&pending_jobs_); | |
| 55 } | 58 } |
| 56 | 59 |
| 57 void MojoHostResolverImpl::Resolve( | 60 void MojoHostResolverImpl::Resolve( |
| 58 std::unique_ptr<HostResolver::RequestInfo> request_info, | 61 std::unique_ptr<HostResolver::RequestInfo> request_info, |
| 59 interfaces::HostResolverRequestClientPtr client) { | 62 interfaces::HostResolverRequestClientPtr client) { |
| 60 DCHECK(thread_checker_.CalledOnValidThread()); | 63 DCHECK(thread_checker_.CalledOnValidThread()); |
| 61 if (request_info->is_my_ip_address()) { | 64 if (request_info->is_my_ip_address()) { |
| 62 // The proxy resolver running inside a sandbox may not be able to get the | 65 // The proxy resolver running inside a sandbox may not be able to get the |
| 63 // correct host name. Instead, fill it ourself if the request is for our own | 66 // correct host name. Instead, fill it ourself if the request is for our own |
| 64 // IP address. | 67 // IP address. |
| 65 request_info->set_host_port_pair(HostPortPair(GetHostName(), 80)); | 68 request_info->set_host_port_pair(HostPortPair(GetHostName(), 80)); |
| 66 } | 69 } |
| 67 Job* job = | 70 |
| 68 new Job(this, resolver_, *request_info, net_log_, std::move(client)); | 71 pending_jobs_.emplace_front(this, resolver_, *request_info, net_log_, |
| 69 pending_jobs_.insert(job); | 72 std::move(client)); |
| 73 auto job = pending_jobs_.begin(); | |
| 74 job->set_iter(job); | |
| 70 job->Start(); | 75 job->Start(); |
| 71 } | 76 } |
| 72 | 77 |
| 73 void MojoHostResolverImpl::DeleteJob(Job* job) { | 78 void MojoHostResolverImpl::DeleteJob(std::list<Job>::iterator job) { |
| 74 DCHECK(thread_checker_.CalledOnValidThread()); | 79 DCHECK(thread_checker_.CalledOnValidThread()); |
| 75 size_t num_erased = pending_jobs_.erase(job); | 80 pending_jobs_.erase(job); |
|
mmenke
2016/11/11 16:15:25
Worth checking if &*job is in the list? I'm not r
| |
| 76 DCHECK(num_erased); | |
| 77 delete job; | |
| 78 } | 81 } |
| 79 | 82 |
| 80 MojoHostResolverImpl::Job::Job( | 83 MojoHostResolverImpl::Job::Job( |
| 81 MojoHostResolverImpl* resolver_service, | 84 MojoHostResolverImpl* resolver_service, |
| 82 net::HostResolver* resolver, | 85 net::HostResolver* resolver, |
| 83 const net::HostResolver::RequestInfo& request_info, | 86 const net::HostResolver::RequestInfo& request_info, |
| 84 const NetLogWithSource& net_log, | 87 const NetLogWithSource& net_log, |
| 85 interfaces::HostResolverRequestClientPtr client) | 88 interfaces::HostResolverRequestClientPtr client) |
| 86 : resolver_service_(resolver_service), | 89 : resolver_service_(resolver_service), |
| 87 resolver_(resolver), | 90 resolver_(resolver), |
| 88 request_info_(request_info), | 91 request_info_(request_info), |
| 89 net_log_(net_log), | 92 net_log_(net_log), |
| 90 client_(std::move(client)) { | 93 client_(std::move(client)) { |
| 91 client_.set_connection_error_handler(base::Bind( | 94 client_.set_connection_error_handler(base::Bind( |
| 92 &MojoHostResolverImpl::Job::OnConnectionError, base::Unretained(this))); | 95 &MojoHostResolverImpl::Job::OnConnectionError, base::Unretained(this))); |
| 93 } | 96 } |
| 94 | 97 |
| 95 void MojoHostResolverImpl::Job::Start() { | 98 void MojoHostResolverImpl::Job::Start() { |
| 99 // The caller is responsible for setting up |iter_|. | |
| 100 DCHECK_EQ(this, &*iter_); | |
|
mmenke
2016/11/11 16:15:25
optional: Maybe also do this in Job::OnConnection
| |
| 101 | |
| 96 DVLOG(1) << "Resolve " << request_info_.host_port_pair().ToString(); | 102 DVLOG(1) << "Resolve " << request_info_.host_port_pair().ToString(); |
| 97 int result = | 103 int result = |
| 98 resolver_->Resolve(request_info_, DEFAULT_PRIORITY, &result_, | 104 resolver_->Resolve(request_info_, DEFAULT_PRIORITY, &result_, |
| 99 base::Bind(&MojoHostResolverImpl::Job::OnResolveDone, | 105 base::Bind(&MojoHostResolverImpl::Job::OnResolveDone, |
| 100 base::Unretained(this)), | 106 base::Unretained(this)), |
| 101 &request_, net_log_); | 107 &request_, net_log_); |
| 102 | 108 |
| 103 if (result != ERR_IO_PENDING) | 109 if (result != ERR_IO_PENDING) |
| 104 OnResolveDone(result); | 110 OnResolveDone(result); |
| 105 } | 111 } |
| 106 | 112 |
| 107 MojoHostResolverImpl::Job::~Job() { | 113 MojoHostResolverImpl::Job::~Job() { |
| 108 } | 114 } |
| 109 | 115 |
| 110 void MojoHostResolverImpl::Job::OnResolveDone(int result) { | 116 void MojoHostResolverImpl::Job::OnResolveDone(int result) { |
| 111 DCHECK(thread_checker_.CalledOnValidThread()); | 117 DCHECK(thread_checker_.CalledOnValidThread()); |
| 112 request_.reset(); | 118 request_.reset(); |
| 113 DVLOG(1) << "Resolved " << request_info_.host_port_pair().ToString() | 119 DVLOG(1) << "Resolved " << request_info_.host_port_pair().ToString() |
| 114 << " with error " << result << " and " << result_.size() | 120 << " with error " << result << " and " << result_.size() |
| 115 << " results!"; | 121 << " results!"; |
| 116 for (const auto& address : result_) { | 122 for (const auto& address : result_) { |
| 117 DVLOG(1) << address.ToString(); | 123 DVLOG(1) << address.ToString(); |
| 118 } | 124 } |
| 119 client_->ReportResult(result, result_); | 125 client_->ReportResult(result, result_); |
| 120 resolver_service_->DeleteJob(this); | 126 resolver_service_->DeleteJob(iter_); |
| 121 } | 127 } |
| 122 | 128 |
| 123 void MojoHostResolverImpl::Job::OnConnectionError() { | 129 void MojoHostResolverImpl::Job::OnConnectionError() { |
| 124 DCHECK(thread_checker_.CalledOnValidThread()); | 130 DCHECK(thread_checker_.CalledOnValidThread()); |
| 125 // |resolver_service_| should always outlive us. | 131 // |resolver_service_| should always outlive us. |
| 126 DCHECK(resolver_service_); | 132 DCHECK(resolver_service_); |
| 127 DVLOG(1) << "Connection error on request for " | 133 DVLOG(1) << "Connection error on request for " |
| 128 << request_info_.host_port_pair().ToString(); | 134 << request_info_.host_port_pair().ToString(); |
| 129 resolver_service_->DeleteJob(this); | 135 resolver_service_->DeleteJob(iter_); |
| 130 } | 136 } |
| 131 | 137 |
| 132 } // namespace net | 138 } // namespace net |
| OLD | NEW |