summaryrefslogtreecommitdiffstats
path: root/chrome/browser/net/dns_slave.h
blob: 820ce43a9afe6fd02cbd9c04cdea36fe0a250593 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
// Copyright (c) 2006-2008 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.

// A DnsSlave object processes hostname lookups
// via DNS on a single thread, waiting for that blocking
// call to complete, and then getting its next hostname
// from its associated DnsMaster object.
// Since this class only is concerned with prefetching
// to warm the underlying DNS cache, the actual IP
// does not need to be recorded.  It is necessary to record
// when the lookup finished, so that the associated DnsMaster
// won't (wastefully) ask for the same name in "too short a
// period of time."
// This class does no "de-duping," and merely slavishly services
// items supplied by its DnsMaster.

#ifndef CHROME_BROWSER_NET_DNS_SLAVE_H__
#define CHROME_BROWSER_NET_DNS_SLAVE_H__

#include <windows.h>
#include <string>

#include "base/basictypes.h"

namespace chrome_browser_net {

class DnsMaster;
class DnsSlave;

// Support testing infrastructure, and allow off-line testing.
typedef void (__stdcall *FreeAddrInfoFunction)(struct addrinfo* ai);
typedef int (__stdcall *GetAddrInfoFunction)(const char* nodename,
                                             const char* servname,
                                             const struct addrinfo* hints,
                                             struct addrinfo** result);
void SetAddrinfoCallbacks(GetAddrInfoFunction getaddrinfo,
                          FreeAddrInfoFunction freeaddrinfo);

GetAddrInfoFunction get_getaddrinfo();
FreeAddrInfoFunction get_freeaddrinfo();

class DnsSlave {
 public:
  DnsSlave(DnsMaster* master, int slave_index)
      : slave_index_(slave_index),
        master_(master) {
  }

  ~DnsSlave() {
    master_ = NULL;
  }

  static DWORD WINAPI ThreadStart(void* pThis);

  unsigned Run();

 private:
  std::string hostname_;  // Name being looked up.

  DnsMaster* master_;  // Master that started us.
  int slave_index_;  // Our index into DnsMaster's array.

  void BlockingDnsLookup();

  DISALLOW_EVIL_CONSTRUCTORS(DnsSlave);
};

}  // namespace chrome_browser_net

#endif  // CHROME_BROWSER_NET_DNS_SLAVE_H__