This source file includes following definitions.
- CreateSockets
- response_
- Start
- DoLoop
- OnDatagramReceived
- Send
- SendDone
- Init
- Send
- OnError
- OnDatagramReceived
- connection_
- Init
- SendQuery
- HandlePacket
- NotifyNsecRecord
- OnConnectionError
- AlertListeners
- AddListener
- RemoveListener
- CleanupObserverList
- ScheduleCleanup
- DoCleanup
- OnRecordRemoved
- QueryCache
- StartListening
- StopListening
- IsListening
- CreateListener
- CreateTransaction
- active_refresh_
- Start
- SetActiveRefresh
- GetName
- GetType
- HandleRecordUpdate
- AlertNsecRecord
- ScheduleNextRefresh
- DoRefresh
- flags_
- Start
- GetName
- GetType
- CacheRecordFound
- TriggerCallback
- Reset
- OnRecordUpdate
- SignalTransactionOver
- ServeRecordsFromCache
- QueryAndListen
- OnNsecRecord
- OnCachePurged
#include "net/dns/mdns_client_impl.h"
#include "base/bind.h"
#include "base/message_loop/message_loop_proxy.h"
#include "base/stl_util.h"
#include "base/time/default_clock.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"
#define ENABLE_NSEC
namespace net {
namespace {
const unsigned MDnsTransactionTimeoutSeconds = 3;
const double kListenerRefreshRatio1 = 0.85;
const double kListenerRefreshRatio2 = 0.95;
const unsigned kMillisecondsPerSecond = 1000;
}
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) {
}
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_->OnError(this, rv);
}
int MDnsConnection::SocketHandler::Send(IOBuffer* buffer, unsigned size) {
return socket_->SendTo(buffer, size, multicast_addr_,
base::Bind(&MDnsConnection::SocketHandler::SendDone,
base::Unretained(this) ));
}
void MDnsConnection::SocketHandler::SendDone(int rv) {
}
MDnsConnection::MDnsConnection(MDnsConnection::Delegate* delegate) :
delegate_(delegate) {
}
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();
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();
}
bool MDnsConnection::Send(IOBuffer* buffer, unsigned size) {
bool success = false;
for (size_t i = 0; i < socket_handlers_.size(); ++i) {
int rv = socket_handlers_[i]->Send(buffer, size);
if (rv >= OK || rv == ERR_IO_PENDING) {
success = true;
} else {
VLOG(1) << "Send failed, socket=" << i << ", error=" << rv;
}
}
return success;
}
void MDnsConnection::OnError(SocketHandler* loop,
int error) {
delegate_->OnConnectionError(error);
}
void MDnsConnection::OnDatagramReceived(
DnsResponse* response,
const IPEndPoint& recv_addr,
int bytes_read) {
DCHECK_GT(bytes_read, 0);
delegate_->HandlePacket(response, bytes_read);
}
MDnsClientImpl::Core::Core(MDnsClientImpl* client)
: client_(client), 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);
return connection_->Send(query.io_buffer(), query.io_buffer()->size());
}
void MDnsClientImpl::Core::HandlePacket(DnsResponse* response,
int bytes_read) {
unsigned offset;
std::map<MDnsCache::Key, MDnsCache::UpdateType> update_keys;
if (!response->InitParseWithoutQuery(bytes_read)) {
DVLOG(1) << "Could not understand an mDNS packet.";
return;
}
if (!(response->flags() & dns_protocol::kFlagResponse))
return;
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;
} else {
continue;
}
}
if ((record->klass() & dns_protocol::kMDnsClassMask) !=
dns_protocol::kClassIN) {
DVLOG(1) << "Received an mDNS record with non-IN class. Ignoring.";
continue;
}
MDnsCache::Key update_key = MDnsCache::Key::CreateFor(record.get());
MDnsCache::UpdateType update = cache_.UpdateDnsRecord(record.Pass());
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);
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());
}
}
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) {
}
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 (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);
if (!observer_list_iterator->second->might_have_observers()) {
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) {
if (cleanup == scheduled_cleanup_) return;
scheduled_cleanup_ = cleanup;
cleanup_callback_.Reset(base::Bind(
&MDnsClientImpl::Core::DoCleanup, base::Unretained(this)));
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(this));
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();
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;
if (ttl_ == 0) {
next_refresh_.Cancel();
return;
}
next_refresh_.Reset(base::Bind(&MDnsListenerImpl::DoRefresh,
AsWeakPtr()));
base::Time next_refresh1 = last_update_ + base::TimeDelta::FromMilliseconds(
static_cast<int>(kMillisecondsPerSecond *
kListenerRefreshRatio1 * ttl_));
base::Time next_refresh2 = last_update_ + base::TimeDelta::FromMilliseconds(
static_cast<int>(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();
}
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;
MDnsTransaction::ResultCallback callback = callback_;
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() {
}
}