Paul Stewart | e692740 | 2012-01-23 16:11:30 -0800 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 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 "shill/dns_client.h" |
| 6 | |
| 7 | #include <arpa/inet.h> |
| 8 | #include <netdb.h> |
| 9 | #include <netinet/in.h> |
| 10 | #include <sys/socket.h> |
| 11 | |
| 12 | #include <map> |
Alex Vakulenko | a41ab51 | 2014-07-23 14:24:23 -0700 | [diff] [blame] | 13 | #include <memory> |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 14 | #include <set> |
| 15 | #include <string> |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 16 | #include <vector> |
| 17 | |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 18 | #include <base/bind.h> |
| 19 | #include <base/bind_helpers.h> |
| 20 | #include <base/stl_util.h> |
Ben Chan | a0ddf46 | 2014-02-06 11:32:42 -0800 | [diff] [blame] | 21 | #include <base/strings/string_number_conversions.h> |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 22 | |
Christopher Wiley | b691efd | 2012-08-09 13:51:51 -0700 | [diff] [blame] | 23 | #include "shill/logging.h" |
Peter Qiu | 8d6b597 | 2014-10-28 15:33:34 -0700 | [diff] [blame] | 24 | #include "shill/net/shill_time.h" |
Darin Petkov | 89593a9 | 2012-02-21 10:59:48 +0100 | [diff] [blame] | 25 | #include "shill/shill_ares.h" |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 26 | |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 27 | using base::Bind; |
| 28 | using base::Unretained; |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 29 | using std::map; |
| 30 | using std::set; |
| 31 | using std::string; |
| 32 | using std::vector; |
| 33 | |
| 34 | namespace shill { |
| 35 | |
Rebecca Silberstein | c9c31d8 | 2014-10-21 15:01:00 -0700 | [diff] [blame^] | 36 | namespace Logging { |
| 37 | static auto kModuleLogScope = ScopeLogger::kDNS; |
| 38 | static string ObjectID(DNSClient *d) { return d->interface_name(); } |
| 39 | } |
| 40 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 41 | const char DNSClient::kErrorNoData[] = "The query response contains no answers"; |
| 42 | const char DNSClient::kErrorFormErr[] = "The server says the query is bad"; |
| 43 | const char DNSClient::kErrorServerFail[] = "The server says it had a failure"; |
| 44 | const char DNSClient::kErrorNotFound[] = "The queried-for domain was not found"; |
| 45 | const char DNSClient::kErrorNotImp[] = "The server doesn't implement operation"; |
| 46 | const char DNSClient::kErrorRefused[] = "The server replied, refused the query"; |
| 47 | const char DNSClient::kErrorBadQuery[] = "Locally we could not format a query"; |
| 48 | const char DNSClient::kErrorNetRefused[] = "The network connection was refused"; |
| 49 | const char DNSClient::kErrorTimedOut[] = "The network connection was timed out"; |
| 50 | const char DNSClient::kErrorUnknown[] = "DNS Resolver unknown internal error"; |
| 51 | |
Peter Qiu | f3a8f90 | 2014-08-20 10:05:42 -0700 | [diff] [blame] | 52 | const int DNSClient::kDefaultDNSPort = 53; |
| 53 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 54 | // Private to the implementation of resolver so callers don't include ares.h |
| 55 | struct DNSClientState { |
Ben Chan | cc225ef | 2014-09-30 13:26:51 -0700 | [diff] [blame] | 56 | DNSClientState() : channel(nullptr), start_time{} {} |
Darin Petkov | e636c69 | 2012-05-31 10:22:17 +0200 | [diff] [blame] | 57 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 58 | ares_channel channel; |
Ben Chan | e2ee5e0 | 2014-09-19 19:29:42 -0700 | [diff] [blame] | 59 | map<ares_socket_t, std::shared_ptr<IOHandler>> read_handlers; |
| 60 | map<ares_socket_t, std::shared_ptr<IOHandler>> write_handlers; |
Darin Petkov | e636c69 | 2012-05-31 10:22:17 +0200 | [diff] [blame] | 61 | struct timeval start_time; |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 62 | }; |
| 63 | |
| 64 | DNSClient::DNSClient(IPAddress::Family family, |
| 65 | const string &interface_name, |
| 66 | const vector<string> &dns_servers, |
| 67 | int timeout_ms, |
| 68 | EventDispatcher *dispatcher, |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 69 | const ClientCallback &callback) |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 70 | : address_(IPAddress(family)), |
| 71 | interface_name_(interface_name), |
| 72 | dns_servers_(dns_servers), |
| 73 | dispatcher_(dispatcher), |
| 74 | callback_(callback), |
| 75 | timeout_ms_(timeout_ms), |
| 76 | running_(false), |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 77 | weak_ptr_factory_(this), |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 78 | ares_(Ares::GetInstance()), |
| 79 | time_(Time::GetInstance()) {} |
| 80 | |
| 81 | DNSClient::~DNSClient() { |
| 82 | Stop(); |
| 83 | } |
| 84 | |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 85 | bool DNSClient::Start(const string &hostname, Error *error) { |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 86 | if (running_) { |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 87 | Error::PopulateAndLog(error, Error::kInProgress, |
| 88 | "Only one DNS request is allowed at a time"); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 89 | return false; |
| 90 | } |
| 91 | |
| 92 | if (!resolver_state_.get()) { |
| 93 | struct ares_options options; |
| 94 | memset(&options, 0, sizeof(options)); |
Peter Qiu | f3a8f90 | 2014-08-20 10:05:42 -0700 | [diff] [blame] | 95 | options.timeout = timeout_ms_; |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 96 | |
Peter Qiu | f3a8f90 | 2014-08-20 10:05:42 -0700 | [diff] [blame] | 97 | if (dns_servers_.empty()) { |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 98 | Error::PopulateAndLog(error, Error::kInvalidArguments, |
| 99 | "No valid DNS server addresses"); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 100 | return false; |
| 101 | } |
| 102 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 103 | resolver_state_.reset(new DNSClientState); |
| 104 | int status = ares_->InitOptions(&resolver_state_->channel, |
| 105 | &options, |
Peter Qiu | f3a8f90 | 2014-08-20 10:05:42 -0700 | [diff] [blame] | 106 | ARES_OPT_TIMEOUTMS); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 107 | if (status != ARES_SUCCESS) { |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 108 | Error::PopulateAndLog(error, Error::kOperationFailed, |
| 109 | "ARES initialization returns error code: " + |
| 110 | base::IntToString(status)); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 111 | resolver_state_.reset(); |
| 112 | return false; |
| 113 | } |
| 114 | |
Peter Qiu | f3a8f90 | 2014-08-20 10:05:42 -0700 | [diff] [blame] | 115 | // Format DNS server addresses string as "host:port[,host:port...]" to be |
| 116 | // used in call to ares_set_servers_csv for setting DNS server addresses. |
| 117 | // There is a bug in ares library when parsing IPv6 addresses, where it |
| 118 | // always assumes the port number are specified when address contains ":". |
| 119 | // So when IPv6 address are given without port number as "xx:xx:xx::yy",the |
| 120 | // parser would parse the address as "xx:xx:xx:" and port number as "yy". |
| 121 | // To work around this bug, port number are added to each address. |
| 122 | // |
| 123 | // Alternatively, we can use ares_set_servers instead, where we would |
| 124 | // explicitly construct a link list of ares_addr_node. |
| 125 | string server_addresses; |
| 126 | bool first = true; |
| 127 | for (const auto &ip : dns_servers_) { |
| 128 | if (!first) { |
| 129 | server_addresses += ","; |
| 130 | } else { |
| 131 | first = false; |
| 132 | } |
| 133 | server_addresses += (ip + ":" + base::IntToString(kDefaultDNSPort)); |
| 134 | } |
| 135 | status = ares_->SetServersCsv(resolver_state_->channel, |
| 136 | server_addresses.c_str()); |
| 137 | if (status != ARES_SUCCESS) { |
| 138 | Error::PopulateAndLog(error, Error::kOperationFailed, |
| 139 | "ARES set DNS servers error code: " + |
| 140 | base::IntToString(status)); |
| 141 | resolver_state_.reset(); |
| 142 | return false; |
| 143 | } |
| 144 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 145 | ares_->SetLocalDev(resolver_state_->channel, interface_name_.c_str()); |
| 146 | } |
| 147 | |
| 148 | running_ = true; |
Darin Petkov | e636c69 | 2012-05-31 10:22:17 +0200 | [diff] [blame] | 149 | time_->GetTimeMonotonic(&resolver_state_->start_time); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 150 | ares_->GetHostByName(resolver_state_->channel, hostname.c_str(), |
| 151 | address_.family(), ReceiveDNSReplyCB, this); |
| 152 | |
| 153 | if (!RefreshHandles()) { |
| 154 | LOG(ERROR) << "Impossibly short timeout."; |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 155 | error->CopyFrom(error_); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 156 | Stop(); |
| 157 | return false; |
| 158 | } |
| 159 | |
| 160 | return true; |
| 161 | } |
| 162 | |
| 163 | void DNSClient::Stop() { |
Rebecca Silberstein | c9c31d8 | 2014-10-21 15:01:00 -0700 | [diff] [blame^] | 164 | SLOG(this, 3) << "In " << __func__; |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 165 | if (!resolver_state_.get()) { |
| 166 | return; |
| 167 | } |
| 168 | |
| 169 | running_ = false; |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 170 | weak_ptr_factory_.InvalidateWeakPtrs(); |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 171 | error_.Reset(); |
| 172 | address_.SetAddressToDefault(); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 173 | ares_->Destroy(resolver_state_->channel); |
| 174 | resolver_state_.reset(); |
| 175 | } |
| 176 | |
Prathmesh Prabhu | 3e452f8 | 2013-04-10 16:31:44 -0700 | [diff] [blame] | 177 | bool DNSClient::IsActive() const { |
| 178 | return running_; |
| 179 | } |
| 180 | |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 181 | // We delay our call to completion so that we exit all IOHandlers, and |
| 182 | // can clean up all of our local state before calling the callback, or |
| 183 | // during the process of the execution of the callee (which is free to |
| 184 | // call our destructor safely). |
| 185 | void DNSClient::HandleCompletion() { |
Rebecca Silberstein | c9c31d8 | 2014-10-21 15:01:00 -0700 | [diff] [blame^] | 186 | SLOG(this, 3) << "In " << __func__; |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 187 | Error error; |
| 188 | error.CopyFrom(error_); |
| 189 | IPAddress address(address_); |
| 190 | if (!error.IsSuccess()) { |
| 191 | // If the DNS request did not succeed, do not trust it for future |
| 192 | // attempts. |
| 193 | Stop(); |
| 194 | } else { |
| 195 | // Prepare our state for the next request without destroying the |
| 196 | // current ARES state. |
| 197 | error_.Reset(); |
| 198 | address_.SetAddressToDefault(); |
| 199 | } |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 200 | callback_.Run(error, address); |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 201 | } |
| 202 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 203 | void DNSClient::HandleDNSRead(int fd) { |
| 204 | ares_->ProcessFd(resolver_state_->channel, fd, ARES_SOCKET_BAD); |
| 205 | RefreshHandles(); |
| 206 | } |
| 207 | |
| 208 | void DNSClient::HandleDNSWrite(int fd) { |
| 209 | ares_->ProcessFd(resolver_state_->channel, ARES_SOCKET_BAD, fd); |
| 210 | RefreshHandles(); |
| 211 | } |
| 212 | |
| 213 | void DNSClient::HandleTimeout() { |
| 214 | ares_->ProcessFd(resolver_state_->channel, ARES_SOCKET_BAD, ARES_SOCKET_BAD); |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 215 | RefreshHandles(); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 216 | } |
| 217 | |
| 218 | void DNSClient::ReceiveDNSReply(int status, struct hostent *hostent) { |
| 219 | if (!running_) { |
| 220 | // We can be called during ARES shutdown -- ignore these events. |
| 221 | return; |
| 222 | } |
Rebecca Silberstein | c9c31d8 | 2014-10-21 15:01:00 -0700 | [diff] [blame^] | 223 | SLOG(this, 3) << "In " << __func__; |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 224 | running_ = false; |
Paul Stewart | f582b50 | 2012-04-04 21:39:22 -0700 | [diff] [blame] | 225 | timeout_closure_.Cancel(); |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 226 | dispatcher_->PostTask(Bind(&DNSClient::HandleCompletion, |
| 227 | weak_ptr_factory_.GetWeakPtr())); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 228 | |
| 229 | if (status == ARES_SUCCESS && |
Ben Chan | cc225ef | 2014-09-30 13:26:51 -0700 | [diff] [blame] | 230 | hostent != nullptr && |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 231 | hostent->h_addrtype == address_.family() && |
Eric Shienbrood | c74cf9c | 2012-03-02 15:00:35 -0500 | [diff] [blame] | 232 | static_cast<size_t>(hostent->h_length) == |
| 233 | IPAddress::GetAddressLength(address_.family()) && |
Ben Chan | cc225ef | 2014-09-30 13:26:51 -0700 | [diff] [blame] | 234 | hostent->h_addr_list != nullptr && |
| 235 | hostent->h_addr_list[0] != nullptr) { |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 236 | address_ = IPAddress(address_.family(), |
| 237 | ByteString(reinterpret_cast<unsigned char *>( |
| 238 | hostent->h_addr_list[0]), hostent->h_length)); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 239 | } else { |
| 240 | switch (status) { |
| 241 | case ARES_ENODATA: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 242 | error_.Populate(Error::kOperationFailed, kErrorNoData); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 243 | break; |
| 244 | case ARES_EFORMERR: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 245 | error_.Populate(Error::kOperationFailed, kErrorFormErr); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 246 | break; |
| 247 | case ARES_ESERVFAIL: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 248 | error_.Populate(Error::kOperationFailed, kErrorServerFail); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 249 | break; |
| 250 | case ARES_ENOTFOUND: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 251 | error_.Populate(Error::kOperationFailed, kErrorNotFound); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 252 | break; |
| 253 | case ARES_ENOTIMP: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 254 | error_.Populate(Error::kOperationFailed, kErrorNotImp); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 255 | break; |
| 256 | case ARES_EREFUSED: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 257 | error_.Populate(Error::kOperationFailed, kErrorRefused); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 258 | break; |
| 259 | case ARES_EBADQUERY: |
| 260 | case ARES_EBADNAME: |
| 261 | case ARES_EBADFAMILY: |
| 262 | case ARES_EBADRESP: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 263 | error_.Populate(Error::kOperationFailed, kErrorBadQuery); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 264 | break; |
| 265 | case ARES_ECONNREFUSED: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 266 | error_.Populate(Error::kOperationFailed, kErrorNetRefused); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 267 | break; |
| 268 | case ARES_ETIMEOUT: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 269 | error_.Populate(Error::kOperationTimeout, kErrorTimedOut); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 270 | break; |
| 271 | default: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 272 | error_.Populate(Error::kOperationFailed, kErrorUnknown); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 273 | if (status == ARES_SUCCESS) { |
| 274 | LOG(ERROR) << "ARES returned success but hostent was invalid!"; |
| 275 | } else { |
| 276 | LOG(ERROR) << "ARES returned unhandled error status " << status; |
| 277 | } |
| 278 | break; |
| 279 | } |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 280 | } |
| 281 | } |
| 282 | |
| 283 | void DNSClient::ReceiveDNSReplyCB(void *arg, int status, |
| 284 | int /*timeouts*/, |
| 285 | struct hostent *hostent) { |
| 286 | DNSClient *res = static_cast<DNSClient *>(arg); |
| 287 | res->ReceiveDNSReply(status, hostent); |
| 288 | } |
| 289 | |
| 290 | bool DNSClient::RefreshHandles() { |
Ben Chan | e2ee5e0 | 2014-09-19 19:29:42 -0700 | [diff] [blame] | 291 | map<ares_socket_t, std::shared_ptr<IOHandler>> old_read = |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 292 | resolver_state_->read_handlers; |
Ben Chan | e2ee5e0 | 2014-09-19 19:29:42 -0700 | [diff] [blame] | 293 | map<ares_socket_t, std::shared_ptr<IOHandler>> old_write = |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 294 | resolver_state_->write_handlers; |
| 295 | |
| 296 | resolver_state_->read_handlers.clear(); |
| 297 | resolver_state_->write_handlers.clear(); |
| 298 | |
| 299 | ares_socket_t sockets[ARES_GETSOCK_MAXNUM]; |
| 300 | int action_bits = ares_->GetSock(resolver_state_->channel, sockets, |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 301 | ARES_GETSOCK_MAXNUM); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 302 | |
Paul Stewart | f582b50 | 2012-04-04 21:39:22 -0700 | [diff] [blame] | 303 | base::Callback<void(int)> read_callback( |
| 304 | Bind(&DNSClient::HandleDNSRead, weak_ptr_factory_.GetWeakPtr())); |
| 305 | base::Callback<void(int)> write_callback( |
| 306 | Bind(&DNSClient::HandleDNSWrite, weak_ptr_factory_.GetWeakPtr())); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 307 | for (int i = 0; i < ARES_GETSOCK_MAXNUM; i++) { |
| 308 | if (ARES_GETSOCK_READABLE(action_bits, i)) { |
| 309 | if (ContainsKey(old_read, sockets[i])) { |
| 310 | resolver_state_->read_handlers[sockets[i]] = old_read[sockets[i]]; |
| 311 | } else { |
| 312 | resolver_state_->read_handlers[sockets[i]] = |
Alex Vakulenko | 8a53229 | 2014-06-16 17:18:44 -0700 | [diff] [blame] | 313 | std::shared_ptr<IOHandler> ( |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 314 | dispatcher_->CreateReadyHandler(sockets[i], |
| 315 | IOHandler::kModeInput, |
Paul Stewart | f582b50 | 2012-04-04 21:39:22 -0700 | [diff] [blame] | 316 | read_callback)); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 317 | } |
| 318 | } |
| 319 | if (ARES_GETSOCK_WRITABLE(action_bits, i)) { |
| 320 | if (ContainsKey(old_write, sockets[i])) { |
| 321 | resolver_state_->write_handlers[sockets[i]] = old_write[sockets[i]]; |
| 322 | } else { |
| 323 | resolver_state_->write_handlers[sockets[i]] = |
Alex Vakulenko | 8a53229 | 2014-06-16 17:18:44 -0700 | [diff] [blame] | 324 | std::shared_ptr<IOHandler> ( |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 325 | dispatcher_->CreateReadyHandler(sockets[i], |
| 326 | IOHandler::kModeOutput, |
Paul Stewart | f582b50 | 2012-04-04 21:39:22 -0700 | [diff] [blame] | 327 | write_callback)); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 328 | } |
| 329 | } |
| 330 | } |
| 331 | |
| 332 | if (!running_) { |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 333 | // We are here just to clean up socket handles, and the ARES state was |
| 334 | // cleaned up during the last call to ares_->ProcessFd(). |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 335 | return false; |
| 336 | } |
| 337 | |
| 338 | // Schedule timer event for the earlier of our timeout or one requested by |
| 339 | // the resolver library. |
| 340 | struct timeval now, elapsed_time, timeout_tv; |
Paul Stewart | e692740 | 2012-01-23 16:11:30 -0800 | [diff] [blame] | 341 | time_->GetTimeMonotonic(&now); |
Darin Petkov | e636c69 | 2012-05-31 10:22:17 +0200 | [diff] [blame] | 342 | timersub(&now, &resolver_state_->start_time, &elapsed_time); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 343 | timeout_tv.tv_sec = timeout_ms_ / 1000; |
| 344 | timeout_tv.tv_usec = (timeout_ms_ % 1000) * 1000; |
Paul Stewart | f582b50 | 2012-04-04 21:39:22 -0700 | [diff] [blame] | 345 | timeout_closure_.Cancel(); |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 346 | |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 347 | if (timercmp(&elapsed_time, &timeout_tv, >=)) { |
| 348 | // There are 3 cases of interest: |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 349 | // - If we got here from Start(), when we return, Stop() will be |
| 350 | // called, so our cleanup task will not run, so we will not have the |
| 351 | // side-effect of both invoking the callback and returning False |
| 352 | // in Start(). |
| 353 | // - If we got here from the tail of an IO event, we can't call |
| 354 | // Stop() since that will blow away the IOHandler we are running |
| 355 | // in. We will perform the cleanup in the posted task below. |
| 356 | // - If we got here from a timeout handler, we will perform cleanup |
| 357 | // in the posted task. |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 358 | running_ = false; |
Paul Stewart | bdb02e6 | 2012-02-22 16:24:33 -0800 | [diff] [blame] | 359 | error_.Populate(Error::kOperationTimeout, kErrorTimedOut); |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 360 | dispatcher_->PostTask(Bind(&DNSClient::HandleCompletion, |
| 361 | weak_ptr_factory_.GetWeakPtr())); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 362 | return false; |
| 363 | } else { |
| 364 | struct timeval max, ret_tv; |
| 365 | timersub(&timeout_tv, &elapsed_time, &max); |
| 366 | struct timeval *tv = ares_->Timeout(resolver_state_->channel, |
| 367 | &max, &ret_tv); |
Paul Stewart | f582b50 | 2012-04-04 21:39:22 -0700 | [diff] [blame] | 368 | timeout_closure_.Reset( |
| 369 | Bind(&DNSClient::HandleTimeout, weak_ptr_factory_.GetWeakPtr())); |
| 370 | dispatcher_->PostDelayedTask(timeout_closure_.callback(), |
| 371 | tv->tv_sec * 1000 + tv->tv_usec / 1000); |
Paul Stewart | c2350ee | 2011-10-19 12:28:40 -0700 | [diff] [blame] | 372 | } |
| 373 | |
| 374 | return true; |
| 375 | } |
| 376 | |
| 377 | } // namespace shill |