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