Home | History | Annotate | Download | only in dns 
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      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 "net/dns/dns_client.h"
      6 
      7 #include "base/bind.h"
      8 #include "base/rand_util.h"
      9 #include "net/base/net_log.h"
     10 #include "net/dns/address_sorter.h"
     11 #include "net/dns/dns_config_service.h"
     12 #include "net/dns/dns_session.h"
     13 #include "net/dns/dns_socket_pool.h"
     14 #include "net/dns/dns_transaction.h"
     15 #include "net/socket/client_socket_factory.h"
     16 
     17 namespace net {
     18 
     19 namespace {
     20 
     21 class DnsClientImpl : public DnsClient {
     22  public:
     23   explicit DnsClientImpl(NetLog* net_log)
     24       : address_sorter_(AddressSorter::CreateAddressSorter()),
     25         net_log_(net_log) {}
     26 
     27   virtual void SetConfig(const DnsConfig& config) OVERRIDE {
     28     factory_.reset();
     29     session_ = NULL;
     30     if (config.IsValid() && !config.unhandled_options) {
     31       ClientSocketFactory* factory = ClientSocketFactory::GetDefaultFactory();
     32       scoped_ptr<DnsSocketPool> socket_pool(
     33           config.randomize_ports ? DnsSocketPool::CreateDefault(factory)
     34                                  : DnsSocketPool::CreateNull(factory));
     35       session_ = new DnsSession(config,
     36                                 socket_pool.Pass(),
     37                                 base::Bind(&base::RandInt),
     38                                 net_log_);
     39       factory_ = DnsTransactionFactory::CreateFactory(session_.get());
     40     }
     41   }
     42 
     43   virtual const DnsConfig* GetConfig() const OVERRIDE {
     44     return session_.get() ? &session_->config() : NULL;
     45   }
     46 
     47   virtual DnsTransactionFactory* GetTransactionFactory() OVERRIDE {
     48     return session_.get() ? factory_.get() : NULL;
     49   }
     50 
     51   virtual AddressSorter* GetAddressSorter() OVERRIDE {
     52     return address_sorter_.get();
     53   }
     54 
     55  private:
     56   scoped_refptr<DnsSession> session_;
     57   scoped_ptr<DnsTransactionFactory> factory_;
     58   scoped_ptr<AddressSorter> address_sorter_;
     59 
     60   NetLog* net_log_;
     61 };
     62 
     63 }  // namespace
     64 
     65 // static
     66 scoped_ptr<DnsClient> DnsClient::CreateClient(NetLog* net_log) {
     67   return scoped_ptr<DnsClient>(new DnsClientImpl(net_log));
     68 }
     69 
     70 }  // namespace net
     71 
     72