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
17namespace net {
18
19namespace {
20
21class 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
66scoped_ptr<DnsClient> DnsClient::CreateClient(NetLog* net_log) {
67  return scoped_ptr<DnsClient>(new DnsClientImpl(net_log));
68}
69
70}  // namespace net
71
72