Index: net/dns/mdns_client_impl.cc |
diff --git a/net/dns/mdns_client_impl.cc b/net/dns/mdns_client_impl.cc |
deleted file mode 100644 |
index 66873abc36bca278a33e0194f0dd85ae52f91740..0000000000000000000000000000000000000000 |
--- a/net/dns/mdns_client_impl.cc |
+++ /dev/null |
@@ -1,735 +0,0 @@ |
-// Copyright 2013 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "net/dns/mdns_client_impl.h" |
- |
-#include <queue> |
- |
-#include "base/bind.h" |
-#include "base/message_loop/message_loop_proxy.h" |
-#include "base/stl_util.h" |
-#include "base/time/default_clock.h" |
-#include "base/time/time.h" |
-#include "net/base/dns_util.h" |
-#include "net/base/net_errors.h" |
-#include "net/base/net_log.h" |
-#include "net/base/rand_callback.h" |
-#include "net/dns/dns_protocol.h" |
-#include "net/dns/record_rdata.h" |
-#include "net/udp/datagram_socket.h" |
- |
-// TODO(gene): Remove this temporary method of disabling NSEC support once it |
-// becomes clear whether this feature should be |
-// supported. http://crbug.com/255232 |
-#define ENABLE_NSEC |
- |
-namespace net { |
- |
-namespace { |
- |
-const unsigned MDnsTransactionTimeoutSeconds = 3; |
-// The fractions of the record's original TTL after which an active listener |
-// (one that had |SetActiveRefresh(true)| called) will send a query to refresh |
-// its cache. This happens both at 85% of the original TTL and again at 95% of |
-// the original TTL. |
-const double kListenerRefreshRatio1 = 0.85; |
-const double kListenerRefreshRatio2 = 0.95; |
- |
-} // namespace |
- |
-void MDnsSocketFactoryImpl::CreateSockets( |
- ScopedVector<DatagramServerSocket>* sockets) { |
- InterfaceIndexFamilyList interfaces(GetMDnsInterfacesToBind()); |
- for (size_t i = 0; i < interfaces.size(); ++i) { |
- DCHECK(interfaces[i].second == net::ADDRESS_FAMILY_IPV4 || |
- interfaces[i].second == net::ADDRESS_FAMILY_IPV6); |
- scoped_ptr<DatagramServerSocket> socket( |
- CreateAndBindMDnsSocket(interfaces[i].second, interfaces[i].first)); |
- if (socket) |
- sockets->push_back(socket.release()); |
- } |
-} |
- |
-MDnsConnection::SocketHandler::SocketHandler( |
- scoped_ptr<DatagramServerSocket> socket, |
- MDnsConnection* connection) |
- : socket_(socket.Pass()), |
- connection_(connection), |
- response_(dns_protocol::kMaxMulticastSize), |
- send_in_progress_(false) { |
-} |
- |
-MDnsConnection::SocketHandler::~SocketHandler() { |
-} |
- |
-int MDnsConnection::SocketHandler::Start() { |
- IPEndPoint end_point; |
- int rv = socket_->GetLocalAddress(&end_point); |
- if (rv != OK) |
- return rv; |
- DCHECK(end_point.GetFamily() == ADDRESS_FAMILY_IPV4 || |
- end_point.GetFamily() == ADDRESS_FAMILY_IPV6); |
- multicast_addr_ = GetMDnsIPEndPoint(end_point.GetFamily()); |
- return DoLoop(0); |
-} |
- |
-int MDnsConnection::SocketHandler::DoLoop(int rv) { |
- do { |
- if (rv > 0) |
- connection_->OnDatagramReceived(&response_, recv_addr_, rv); |
- |
- rv = socket_->RecvFrom( |
- response_.io_buffer(), |
- response_.io_buffer()->size(), |
- &recv_addr_, |
- base::Bind(&MDnsConnection::SocketHandler::OnDatagramReceived, |
- base::Unretained(this))); |
- } while (rv > 0); |
- |
- if (rv != ERR_IO_PENDING) |
- return rv; |
- |
- return OK; |
-} |
- |
-void MDnsConnection::SocketHandler::OnDatagramReceived(int rv) { |
- if (rv >= OK) |
- rv = DoLoop(rv); |
- |
- if (rv != OK) |
- connection_->PostOnError(this, rv); |
-} |
- |
-void MDnsConnection::SocketHandler::Send(const scoped_refptr<IOBuffer>& buffer, |
- unsigned size) { |
- if (send_in_progress_) { |
- send_queue_.push(std::make_pair(buffer, size)); |
- return; |
- } |
- int rv = socket_->SendTo(buffer.get(), |
- size, |
- multicast_addr_, |
- base::Bind(&MDnsConnection::SocketHandler::SendDone, |
- base::Unretained(this))); |
- if (rv == ERR_IO_PENDING) { |
- send_in_progress_ = true; |
- } else if (rv < OK) { |
- connection_->PostOnError(this, rv); |
- } |
-} |
- |
-void MDnsConnection::SocketHandler::SendDone(int rv) { |
- DCHECK(send_in_progress_); |
- send_in_progress_ = false; |
- if (rv != OK) |
- connection_->PostOnError(this, rv); |
- while (!send_in_progress_ && !send_queue_.empty()) { |
- std::pair<scoped_refptr<IOBuffer>, unsigned> buffer = send_queue_.front(); |
- send_queue_.pop(); |
- Send(buffer.first, buffer.second); |
- } |
-} |
- |
-MDnsConnection::MDnsConnection(MDnsConnection::Delegate* delegate) |
- : delegate_(delegate), weak_ptr_factory_(this) { |
-} |
- |
-MDnsConnection::~MDnsConnection() { |
-} |
- |
-bool MDnsConnection::Init(MDnsSocketFactory* socket_factory) { |
- ScopedVector<DatagramServerSocket> sockets; |
- socket_factory->CreateSockets(&sockets); |
- |
- for (size_t i = 0; i < sockets.size(); ++i) { |
- socket_handlers_.push_back( |
- new MDnsConnection::SocketHandler(make_scoped_ptr(sockets[i]), this)); |
- } |
- sockets.weak_clear(); |
- |
- // All unbound sockets need to be bound before processing untrusted input. |
- // This is done for security reasons, so that an attacker can't get an unbound |
- // socket. |
- for (size_t i = 0; i < socket_handlers_.size();) { |
- int rv = socket_handlers_[i]->Start(); |
- if (rv != OK) { |
- socket_handlers_.erase(socket_handlers_.begin() + i); |
- VLOG(1) << "Start failed, socket=" << i << ", error=" << rv; |
- } else { |
- ++i; |
- } |
- } |
- VLOG(1) << "Sockets ready:" << socket_handlers_.size(); |
- return !socket_handlers_.empty(); |
-} |
- |
-void MDnsConnection::Send(const scoped_refptr<IOBuffer>& buffer, |
- unsigned size) { |
- for (size_t i = 0; i < socket_handlers_.size(); ++i) |
- socket_handlers_[i]->Send(buffer, size); |
-} |
- |
-void MDnsConnection::PostOnError(SocketHandler* loop, int rv) { |
- VLOG(1) << "Socket error. id=" |
- << std::find(socket_handlers_.begin(), socket_handlers_.end(), loop) - |
- socket_handlers_.begin() << ", error=" << rv; |
- // Post to allow deletion of this object by delegate. |
- base::MessageLoop::current()->PostTask( |
- FROM_HERE, |
- base::Bind(&MDnsConnection::OnError, weak_ptr_factory_.GetWeakPtr(), rv)); |
-} |
- |
-void MDnsConnection::OnError(int rv) { |
- // TODO(noamsml): Specific handling of intermittent errors that can be handled |
- // in the connection. |
- delegate_->OnConnectionError(rv); |
-} |
- |
-void MDnsConnection::OnDatagramReceived( |
- DnsResponse* response, |
- const IPEndPoint& recv_addr, |
- int bytes_read) { |
- // TODO(noamsml): More sophisticated error handling. |
- DCHECK_GT(bytes_read, 0); |
- delegate_->HandlePacket(response, bytes_read); |
-} |
- |
-MDnsClientImpl::Core::Core() : connection_(new MDnsConnection(this)) { |
-} |
- |
-MDnsClientImpl::Core::~Core() { |
- STLDeleteValues(&listeners_); |
-} |
- |
-bool MDnsClientImpl::Core::Init(MDnsSocketFactory* socket_factory) { |
- return connection_->Init(socket_factory); |
-} |
- |
-bool MDnsClientImpl::Core::SendQuery(uint16 rrtype, std::string name) { |
- std::string name_dns; |
- if (!DNSDomainFromDot(name, &name_dns)) |
- return false; |
- |
- DnsQuery query(0, name_dns, rrtype); |
- query.set_flags(0); // Remove the RD flag from the query. It is unneeded. |
- |
- connection_->Send(query.io_buffer(), query.io_buffer()->size()); |
- return true; |
-} |
- |
-void MDnsClientImpl::Core::HandlePacket(DnsResponse* response, |
- int bytes_read) { |
- unsigned offset; |
- // Note: We store cache keys rather than record pointers to avoid |
- // erroneous behavior in case a packet contains multiple exclusive |
- // records with the same type and name. |
- std::map<MDnsCache::Key, MDnsCache::UpdateType> update_keys; |
- |
- if (!response->InitParseWithoutQuery(bytes_read)) { |
- DVLOG(1) << "Could not understand an mDNS packet."; |
- return; // Message is unreadable. |
- } |
- |
- // TODO(noamsml): duplicate query suppression. |
- if (!(response->flags() & dns_protocol::kFlagResponse)) |
- return; // Message is a query. ignore it. |
- |
- DnsRecordParser parser = response->Parser(); |
- unsigned answer_count = response->answer_count() + |
- response->additional_answer_count(); |
- |
- for (unsigned i = 0; i < answer_count; i++) { |
- offset = parser.GetOffset(); |
- scoped_ptr<const RecordParsed> record = RecordParsed::CreateFrom( |
- &parser, base::Time::Now()); |
- |
- if (!record) { |
- DVLOG(1) << "Could not understand an mDNS record."; |
- |
- if (offset == parser.GetOffset()) { |
- DVLOG(1) << "Abandoned parsing the rest of the packet."; |
- return; // The parser did not advance, abort reading the packet. |
- } else { |
- continue; // We may be able to extract other records from the packet. |
- } |
- } |
- |
- if ((record->klass() & dns_protocol::kMDnsClassMask) != |
- dns_protocol::kClassIN) { |
- DVLOG(1) << "Received an mDNS record with non-IN class. Ignoring."; |
- continue; // Ignore all records not in the IN class. |
- } |
- |
- MDnsCache::Key update_key = MDnsCache::Key::CreateFor(record.get()); |
- MDnsCache::UpdateType update = cache_.UpdateDnsRecord(record.Pass()); |
- |
- // Cleanup time may have changed. |
- ScheduleCleanup(cache_.next_expiration()); |
- |
- update_keys.insert(std::make_pair(update_key, update)); |
- } |
- |
- for (std::map<MDnsCache::Key, MDnsCache::UpdateType>::iterator i = |
- update_keys.begin(); i != update_keys.end(); i++) { |
- const RecordParsed* record = cache_.LookupKey(i->first); |
- if (!record) |
- continue; |
- |
- if (record->type() == dns_protocol::kTypeNSEC) { |
-#if defined(ENABLE_NSEC) |
- NotifyNsecRecord(record); |
-#endif |
- } else { |
- AlertListeners(i->second, ListenerKey(record->name(), record->type()), |
- record); |
- } |
- } |
-} |
- |
-void MDnsClientImpl::Core::NotifyNsecRecord(const RecordParsed* record) { |
- DCHECK_EQ(dns_protocol::kTypeNSEC, record->type()); |
- const NsecRecordRdata* rdata = record->rdata<NsecRecordRdata>(); |
- DCHECK(rdata); |
- |
- // Remove all cached records matching the nonexistent RR types. |
- std::vector<const RecordParsed*> records_to_remove; |
- |
- cache_.FindDnsRecords(0, record->name(), &records_to_remove, |
- base::Time::Now()); |
- |
- for (std::vector<const RecordParsed*>::iterator i = records_to_remove.begin(); |
- i != records_to_remove.end(); i++) { |
- if ((*i)->type() == dns_protocol::kTypeNSEC) |
- continue; |
- if (!rdata->GetBit((*i)->type())) { |
- scoped_ptr<const RecordParsed> record_removed = cache_.RemoveRecord((*i)); |
- DCHECK(record_removed); |
- OnRecordRemoved(record_removed.get()); |
- } |
- } |
- |
- // Alert all listeners waiting for the nonexistent RR types. |
- ListenerMap::iterator i = |
- listeners_.upper_bound(ListenerKey(record->name(), 0)); |
- for (; i != listeners_.end() && i->first.first == record->name(); i++) { |
- if (!rdata->GetBit(i->first.second)) { |
- FOR_EACH_OBSERVER(MDnsListenerImpl, *i->second, AlertNsecRecord()); |
- } |
- } |
-} |
- |
-void MDnsClientImpl::Core::OnConnectionError(int error) { |
- // TODO(noamsml): On connection error, recreate connection and flush cache. |
-} |
- |
-void MDnsClientImpl::Core::AlertListeners( |
- MDnsCache::UpdateType update_type, |
- const ListenerKey& key, |
- const RecordParsed* record) { |
- ListenerMap::iterator listener_map_iterator = listeners_.find(key); |
- if (listener_map_iterator == listeners_.end()) return; |
- |
- FOR_EACH_OBSERVER(MDnsListenerImpl, *listener_map_iterator->second, |
- HandleRecordUpdate(update_type, record)); |
-} |
- |
-void MDnsClientImpl::Core::AddListener( |
- MDnsListenerImpl* listener) { |
- ListenerKey key(listener->GetName(), listener->GetType()); |
- std::pair<ListenerMap::iterator, bool> observer_insert_result = |
- listeners_.insert( |
- make_pair(key, static_cast<ObserverList<MDnsListenerImpl>*>(NULL))); |
- |
- // If an equivalent key does not exist, actually create the observer list. |
- if (observer_insert_result.second) |
- observer_insert_result.first->second = new ObserverList<MDnsListenerImpl>(); |
- |
- ObserverList<MDnsListenerImpl>* observer_list = |
- observer_insert_result.first->second; |
- |
- observer_list->AddObserver(listener); |
-} |
- |
-void MDnsClientImpl::Core::RemoveListener(MDnsListenerImpl* listener) { |
- ListenerKey key(listener->GetName(), listener->GetType()); |
- ListenerMap::iterator observer_list_iterator = listeners_.find(key); |
- |
- DCHECK(observer_list_iterator != listeners_.end()); |
- DCHECK(observer_list_iterator->second->HasObserver(listener)); |
- |
- observer_list_iterator->second->RemoveObserver(listener); |
- |
- // Remove the observer list from the map if it is empty |
- if (!observer_list_iterator->second->might_have_observers()) { |
- // Schedule the actual removal for later in case the listener removal |
- // happens while iterating over the observer list. |
- base::MessageLoop::current()->PostTask( |
- FROM_HERE, base::Bind( |
- &MDnsClientImpl::Core::CleanupObserverList, AsWeakPtr(), key)); |
- } |
-} |
- |
-void MDnsClientImpl::Core::CleanupObserverList(const ListenerKey& key) { |
- ListenerMap::iterator found = listeners_.find(key); |
- if (found != listeners_.end() && !found->second->might_have_observers()) { |
- delete found->second; |
- listeners_.erase(found); |
- } |
-} |
- |
-void MDnsClientImpl::Core::ScheduleCleanup(base::Time cleanup) { |
- // Cleanup is already scheduled, no need to do anything. |
- if (cleanup == scheduled_cleanup_) return; |
- scheduled_cleanup_ = cleanup; |
- |
- // This cancels the previously scheduled cleanup. |
- cleanup_callback_.Reset(base::Bind( |
- &MDnsClientImpl::Core::DoCleanup, base::Unretained(this))); |
- |
- // If |cleanup| is empty, then no cleanup necessary. |
- if (cleanup != base::Time()) { |
- base::MessageLoop::current()->PostDelayedTask( |
- FROM_HERE, |
- cleanup_callback_.callback(), |
- cleanup - base::Time::Now()); |
- } |
-} |
- |
-void MDnsClientImpl::Core::DoCleanup() { |
- cache_.CleanupRecords(base::Time::Now(), base::Bind( |
- &MDnsClientImpl::Core::OnRecordRemoved, base::Unretained(this))); |
- |
- ScheduleCleanup(cache_.next_expiration()); |
-} |
- |
-void MDnsClientImpl::Core::OnRecordRemoved( |
- const RecordParsed* record) { |
- AlertListeners(MDnsCache::RecordRemoved, |
- ListenerKey(record->name(), record->type()), record); |
-} |
- |
-void MDnsClientImpl::Core::QueryCache( |
- uint16 rrtype, const std::string& name, |
- std::vector<const RecordParsed*>* records) const { |
- cache_.FindDnsRecords(rrtype, name, records, base::Time::Now()); |
-} |
- |
-MDnsClientImpl::MDnsClientImpl() { |
-} |
- |
-MDnsClientImpl::~MDnsClientImpl() { |
-} |
- |
-bool MDnsClientImpl::StartListening(MDnsSocketFactory* socket_factory) { |
- DCHECK(!core_.get()); |
- core_.reset(new Core()); |
- if (!core_->Init(socket_factory)) { |
- core_.reset(); |
- return false; |
- } |
- return true; |
-} |
- |
-void MDnsClientImpl::StopListening() { |
- core_.reset(); |
-} |
- |
-bool MDnsClientImpl::IsListening() const { |
- return core_.get() != NULL; |
-} |
- |
-scoped_ptr<MDnsListener> MDnsClientImpl::CreateListener( |
- uint16 rrtype, |
- const std::string& name, |
- MDnsListener::Delegate* delegate) { |
- return scoped_ptr<net::MDnsListener>( |
- new MDnsListenerImpl(rrtype, name, delegate, this)); |
-} |
- |
-scoped_ptr<MDnsTransaction> MDnsClientImpl::CreateTransaction( |
- uint16 rrtype, |
- const std::string& name, |
- int flags, |
- const MDnsTransaction::ResultCallback& callback) { |
- return scoped_ptr<MDnsTransaction>( |
- new MDnsTransactionImpl(rrtype, name, flags, callback, this)); |
-} |
- |
-MDnsListenerImpl::MDnsListenerImpl( |
- uint16 rrtype, |
- const std::string& name, |
- MDnsListener::Delegate* delegate, |
- MDnsClientImpl* client) |
- : rrtype_(rrtype), name_(name), client_(client), delegate_(delegate), |
- started_(false), active_refresh_(false) { |
-} |
- |
-MDnsListenerImpl::~MDnsListenerImpl() { |
- if (started_) { |
- DCHECK(client_->core()); |
- client_->core()->RemoveListener(this); |
- } |
-} |
- |
-bool MDnsListenerImpl::Start() { |
- DCHECK(!started_); |
- |
- started_ = true; |
- |
- DCHECK(client_->core()); |
- client_->core()->AddListener(this); |
- |
- return true; |
-} |
- |
-void MDnsListenerImpl::SetActiveRefresh(bool active_refresh) { |
- active_refresh_ = active_refresh; |
- |
- if (started_) { |
- if (!active_refresh_) { |
- next_refresh_.Cancel(); |
- } else if (last_update_ != base::Time()) { |
- ScheduleNextRefresh(); |
- } |
- } |
-} |
- |
-const std::string& MDnsListenerImpl::GetName() const { |
- return name_; |
-} |
- |
-uint16 MDnsListenerImpl::GetType() const { |
- return rrtype_; |
-} |
- |
-void MDnsListenerImpl::HandleRecordUpdate(MDnsCache::UpdateType update_type, |
- const RecordParsed* record) { |
- DCHECK(started_); |
- |
- if (update_type != MDnsCache::RecordRemoved) { |
- ttl_ = record->ttl(); |
- last_update_ = record->time_created(); |
- |
- ScheduleNextRefresh(); |
- } |
- |
- if (update_type != MDnsCache::NoChange) { |
- MDnsListener::UpdateType update_external; |
- |
- switch (update_type) { |
- case MDnsCache::RecordAdded: |
- update_external = MDnsListener::RECORD_ADDED; |
- break; |
- case MDnsCache::RecordChanged: |
- update_external = MDnsListener::RECORD_CHANGED; |
- break; |
- case MDnsCache::RecordRemoved: |
- update_external = MDnsListener::RECORD_REMOVED; |
- break; |
- case MDnsCache::NoChange: |
- default: |
- NOTREACHED(); |
- // Dummy assignment to suppress compiler warning. |
- update_external = MDnsListener::RECORD_CHANGED; |
- break; |
- } |
- |
- delegate_->OnRecordUpdate(update_external, record); |
- } |
-} |
- |
-void MDnsListenerImpl::AlertNsecRecord() { |
- DCHECK(started_); |
- delegate_->OnNsecRecord(name_, rrtype_); |
-} |
- |
-void MDnsListenerImpl::ScheduleNextRefresh() { |
- DCHECK(last_update_ != base::Time()); |
- |
- if (!active_refresh_) |
- return; |
- |
- // A zero TTL is a goodbye packet and should not be refreshed. |
- if (ttl_ == 0) { |
- next_refresh_.Cancel(); |
- return; |
- } |
- |
- next_refresh_.Reset(base::Bind(&MDnsListenerImpl::DoRefresh, |
- AsWeakPtr())); |
- |
- // Schedule refreshes at both 85% and 95% of the original TTL. These will both |
- // be canceled and rescheduled if the record's TTL is updated due to a |
- // response being received. |
- base::Time next_refresh1 = last_update_ + base::TimeDelta::FromMilliseconds( |
- static_cast<int>(base::Time::kMillisecondsPerSecond * |
- kListenerRefreshRatio1 * ttl_)); |
- |
- base::Time next_refresh2 = last_update_ + base::TimeDelta::FromMilliseconds( |
- static_cast<int>(base::Time::kMillisecondsPerSecond * |
- kListenerRefreshRatio2 * ttl_)); |
- |
- base::MessageLoop::current()->PostDelayedTask( |
- FROM_HERE, |
- next_refresh_.callback(), |
- next_refresh1 - base::Time::Now()); |
- |
- base::MessageLoop::current()->PostDelayedTask( |
- FROM_HERE, |
- next_refresh_.callback(), |
- next_refresh2 - base::Time::Now()); |
-} |
- |
-void MDnsListenerImpl::DoRefresh() { |
- client_->core()->SendQuery(rrtype_, name_); |
-} |
- |
-MDnsTransactionImpl::MDnsTransactionImpl( |
- uint16 rrtype, |
- const std::string& name, |
- int flags, |
- const MDnsTransaction::ResultCallback& callback, |
- MDnsClientImpl* client) |
- : rrtype_(rrtype), name_(name), callback_(callback), client_(client), |
- started_(false), flags_(flags) { |
- DCHECK((flags_ & MDnsTransaction::FLAG_MASK) == flags_); |
- DCHECK(flags_ & MDnsTransaction::QUERY_CACHE || |
- flags_ & MDnsTransaction::QUERY_NETWORK); |
-} |
- |
-MDnsTransactionImpl::~MDnsTransactionImpl() { |
- timeout_.Cancel(); |
-} |
- |
-bool MDnsTransactionImpl::Start() { |
- DCHECK(!started_); |
- started_ = true; |
- |
- base::WeakPtr<MDnsTransactionImpl> weak_this = AsWeakPtr(); |
- if (flags_ & MDnsTransaction::QUERY_CACHE) { |
- ServeRecordsFromCache(); |
- |
- if (!weak_this || !is_active()) return true; |
- } |
- |
- if (flags_ & MDnsTransaction::QUERY_NETWORK) { |
- return QueryAndListen(); |
- } |
- |
- // If this is a cache only query, signal that the transaction is over |
- // immediately. |
- SignalTransactionOver(); |
- return true; |
-} |
- |
-const std::string& MDnsTransactionImpl::GetName() const { |
- return name_; |
-} |
- |
-uint16 MDnsTransactionImpl::GetType() const { |
- return rrtype_; |
-} |
- |
-void MDnsTransactionImpl::CacheRecordFound(const RecordParsed* record) { |
- DCHECK(started_); |
- OnRecordUpdate(MDnsListener::RECORD_ADDED, record); |
-} |
- |
-void MDnsTransactionImpl::TriggerCallback(MDnsTransaction::Result result, |
- const RecordParsed* record) { |
- DCHECK(started_); |
- if (!is_active()) return; |
- |
- // Ensure callback is run after touching all class state, so that |
- // the callback can delete the transaction. |
- MDnsTransaction::ResultCallback callback = callback_; |
- |
- // Reset the transaction if it expects a single result, or if the result |
- // is a final one (everything except for a record). |
- if (flags_ & MDnsTransaction::SINGLE_RESULT || |
- result != MDnsTransaction::RESULT_RECORD) { |
- Reset(); |
- } |
- |
- callback.Run(result, record); |
-} |
- |
-void MDnsTransactionImpl::Reset() { |
- callback_.Reset(); |
- listener_.reset(); |
- timeout_.Cancel(); |
-} |
- |
-void MDnsTransactionImpl::OnRecordUpdate(MDnsListener::UpdateType update, |
- const RecordParsed* record) { |
- DCHECK(started_); |
- if (update == MDnsListener::RECORD_ADDED || |
- update == MDnsListener::RECORD_CHANGED) |
- TriggerCallback(MDnsTransaction::RESULT_RECORD, record); |
-} |
- |
-void MDnsTransactionImpl::SignalTransactionOver() { |
- DCHECK(started_); |
- if (flags_ & MDnsTransaction::SINGLE_RESULT) { |
- TriggerCallback(MDnsTransaction::RESULT_NO_RESULTS, NULL); |
- } else { |
- TriggerCallback(MDnsTransaction::RESULT_DONE, NULL); |
- } |
-} |
- |
-void MDnsTransactionImpl::ServeRecordsFromCache() { |
- std::vector<const RecordParsed*> records; |
- base::WeakPtr<MDnsTransactionImpl> weak_this = AsWeakPtr(); |
- |
- if (client_->core()) { |
- client_->core()->QueryCache(rrtype_, name_, &records); |
- for (std::vector<const RecordParsed*>::iterator i = records.begin(); |
- i != records.end() && weak_this; ++i) { |
- weak_this->TriggerCallback(MDnsTransaction::RESULT_RECORD, *i); |
- } |
- |
-#if defined(ENABLE_NSEC) |
- if (records.empty()) { |
- DCHECK(weak_this); |
- client_->core()->QueryCache(dns_protocol::kTypeNSEC, name_, &records); |
- if (!records.empty()) { |
- const NsecRecordRdata* rdata = |
- records.front()->rdata<NsecRecordRdata>(); |
- DCHECK(rdata); |
- if (!rdata->GetBit(rrtype_)) |
- weak_this->TriggerCallback(MDnsTransaction::RESULT_NSEC, NULL); |
- } |
- } |
-#endif |
- } |
-} |
- |
-bool MDnsTransactionImpl::QueryAndListen() { |
- listener_ = client_->CreateListener(rrtype_, name_, this); |
- if (!listener_->Start()) |
- return false; |
- |
- DCHECK(client_->core()); |
- if (!client_->core()->SendQuery(rrtype_, name_)) |
- return false; |
- |
- timeout_.Reset(base::Bind(&MDnsTransactionImpl::SignalTransactionOver, |
- AsWeakPtr())); |
- base::MessageLoop::current()->PostDelayedTask( |
- FROM_HERE, |
- timeout_.callback(), |
- base::TimeDelta::FromSeconds(MDnsTransactionTimeoutSeconds)); |
- |
- return true; |
-} |
- |
-void MDnsTransactionImpl::OnNsecRecord(const std::string& name, unsigned type) { |
- TriggerCallback(RESULT_NSEC, NULL); |
-} |
- |
-void MDnsTransactionImpl::OnCachePurged() { |
- // TODO(noamsml): Cache purge situations not yet implemented |
-} |
- |
-} // namespace net |