OLD | NEW |
| (Empty) |
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 #ifndef NET_BASE_HOST_RESOLVER_PROC_H_ | |
6 #define NET_BASE_HOST_RESOLVER_PROC_H_ | |
7 | |
8 #include <string> | |
9 | |
10 #include "base/memory/ref_counted.h" | |
11 #include "net/base/address_family.h" | |
12 #include "net/base/net_export.h" | |
13 | |
14 namespace net { | |
15 | |
16 class AddressList; | |
17 | |
18 // Interface for a getaddrinfo()-like procedure. This is used by unit-tests | |
19 // to control the underlying resolutions in HostResolverImpl. HostResolverProcs | |
20 // can be chained together; they fallback to the next procedure in the chain | |
21 // by calling ResolveUsingPrevious(). | |
22 // | |
23 // Note that implementations of HostResolverProc *MUST BE THREADSAFE*, since | |
24 // the HostResolver implementation using them can be multi-threaded. | |
25 class NET_EXPORT HostResolverProc | |
26 : public base::RefCountedThreadSafe<HostResolverProc> { | |
27 public: | |
28 explicit HostResolverProc(HostResolverProc* previous); | |
29 | |
30 // Resolves |host| to an address list, restricting the results to addresses | |
31 // in |address_family|. If successful returns OK and fills |addrlist| with | |
32 // a list of socket addresses. Otherwise returns a network error code, and | |
33 // fills |os_error| with a more specific error if it was non-NULL. | |
34 virtual int Resolve(const std::string& host, | |
35 AddressFamily address_family, | |
36 HostResolverFlags host_resolver_flags, | |
37 AddressList* addrlist, | |
38 int* os_error) = 0; | |
39 | |
40 protected: | |
41 friend class base::RefCountedThreadSafe<HostResolverProc>; | |
42 | |
43 virtual ~HostResolverProc(); | |
44 | |
45 // Asks the fallback procedure (if set) to do the resolve. | |
46 int ResolveUsingPrevious(const std::string& host, | |
47 AddressFamily address_family, | |
48 HostResolverFlags host_resolver_flags, | |
49 AddressList* addrlist, | |
50 int* os_error); | |
51 | |
52 private: | |
53 friend class HostResolverImpl; | |
54 friend class MockHostResolverBase; | |
55 friend class ScopedDefaultHostResolverProc; | |
56 | |
57 // Sets the previous procedure in the chain. Aborts if this would result in a | |
58 // cycle. | |
59 void SetPreviousProc(HostResolverProc* proc); | |
60 | |
61 // Sets the last procedure in the chain, i.e. appends |proc| to the end of the | |
62 // current chain. Aborts if this would result in a cycle. | |
63 void SetLastProc(HostResolverProc* proc); | |
64 | |
65 // Returns the last procedure in the chain starting at |proc|. Will return | |
66 // NULL iff |proc| is NULL. | |
67 static HostResolverProc* GetLastProc(HostResolverProc* proc); | |
68 | |
69 // Sets the default host resolver procedure that is used by HostResolverImpl. | |
70 // This can be used through ScopedDefaultHostResolverProc to set a catch-all | |
71 // DNS block in unit-tests (individual tests should use MockHostResolver to | |
72 // prevent hitting the network). | |
73 static HostResolverProc* SetDefault(HostResolverProc* proc); | |
74 static HostResolverProc* GetDefault(); | |
75 | |
76 scoped_refptr<HostResolverProc> previous_proc_; | |
77 static HostResolverProc* default_proc_; | |
78 | |
79 DISALLOW_COPY_AND_ASSIGN(HostResolverProc); | |
80 }; | |
81 | |
82 // Resolves |host| to an address list, using the system's default host resolver. | |
83 // (i.e. this calls out to getaddrinfo()). If successful returns OK and fills | |
84 // |addrlist| with a list of socket addresses. Otherwise returns a | |
85 // network error code, and fills |os_error| with a more specific error if it | |
86 // was non-NULL. | |
87 NET_EXPORT_PRIVATE int SystemHostResolverProc( | |
88 const std::string& host, | |
89 AddressFamily address_family, | |
90 HostResolverFlags host_resolver_flags, | |
91 AddressList* addrlist, | |
92 int* os_error); | |
93 | |
94 } // namespace net | |
95 | |
96 #endif // NET_BASE_HOST_RESOLVER_PROC_H_ | |
OLD | NEW |