Commit | Line | Data |
---|---|---|
36ad976b CH |
1 | /* |
2 | The software in this package is distributed under the GNU General | |
3 | Public License version 2 (with a special exception described below). | |
4 | ||
5 | A copy of GNU General Public License (GPL) is included in this distribution, | |
6 | in the file COPYING.GPL. | |
7 | ||
8 | As a special exception, if other files instantiate templates or use macros | |
9 | or inline functions from this file, or you compile this file and link it | |
10 | with other works to produce a work based on this file, this file | |
11 | does not by itself cause the resulting work to be covered | |
12 | by the GNU General Public License. | |
13 | ||
14 | However the source code for this file must still be made available | |
15 | in accordance with section (3) of the GNU General Public License. | |
16 | ||
17 | This exception does not invalidate any other reasons why a work based | |
18 | on this file might be covered by the GNU General Public License. | |
19 | ||
20 | Christian Herdtweck, Intra2net AG 2015 | |
21 | */ | |
22 | ||
23 | #ifndef DNS_RESOLVER_H | |
24 | #define DNS_RESOLVER_H | |
25 | ||
c5b4902d | 26 | #include "dns/dnsmaster.h" |
36ad976b | 27 | |
36ad976b | 28 | #include <boost/asio/deadline_timer.hpp> |
96779587 CH |
29 | #include <boost/asio/ip/udp.hpp> |
30 | #include <boost/asio/ip/address.hpp> | |
36ad976b | 31 | #include <boost/system/error_code.hpp> |
946356e1 | 32 | #include <boost/net/network_array.hpp> // dns_buffer_t |
ad83004d | 33 | #include <boost/uuid/uuid_generators.hpp> |
36ad976b | 34 | |
c5b4902d CH |
35 | #include "dns/resolverbase.h" |
36 | #include "dns/dnsmaster.h" | |
37 | #include "dns/dnscache.h" | |
36ad976b | 38 | |
36ad976b | 39 | |
ad83004d | 40 | typedef std::pair<std::string, std::string> string_pair; |
dbe986b9 | 41 | typedef std::pair<std::string, Cname> src_cname_pair; |
e18c1337 | 42 | typedef std::pair<std::string, HostAddress> host_addr_pair; |
ad83004d CH |
43 | |
44 | ||
36ad976b CH |
45 | class DnsResolver : public ResolverBase |
46 | { | |
36ad976b | 47 | public: |
96779587 | 48 | ~DnsResolver(); |
36ad976b | 49 | |
96779587 CH |
50 | // constructor accessible from friend DnsMaster |
51 | public: | |
4e7b6ff9 | 52 | friend ResolverItem& DnsMaster::get_resolver_for( |
923626c0 CH |
53 | const std::string &hostname, |
54 | const DnsIpProtocol &protocol); | |
e18c1337 CH |
55 | friend ResolverItem DnsMaster::get_recursor_for( |
56 | const std::string &hostname, | |
57 | const DnsIpProtocol &protocol, | |
58 | const boost::asio::ip::address &name_server); | |
36ad976b | 59 | private: |
4e7b6ff9 CH |
60 | DnsResolver(IoServiceItem &io_serv, |
61 | const std::string &hostname, | |
923626c0 | 62 | const DnsIpProtocol &protocol, |
96779587 | 63 | const DnsCacheItem cache, |
96779587 | 64 | const boost::asio::ip::address &name_server); |
36ad976b | 65 | |
4e7b6ff9 | 66 | // only real public function (called from pingers) |
36ad976b | 67 | public: |
4e7b6ff9 | 68 | HostAddress get_next_ip(); |
923626c0 CH |
69 | bool have_up_to_date_ip(); |
70 | int get_resolved_ip_count(); | |
e18c1337 CH |
71 | void cancel_resolve(); |
72 | bool is_resolving(); | |
4e7b6ff9 CH |
73 | |
74 | // implementation of ResolverBase::async_resolve | |
75 | protected: | |
76 | void do_resolve(); | |
36ad976b | 77 | |
c5b4902d | 78 | // internal helper functions |
36ad976b | 79 | private: |
36ad976b CH |
80 | void handle_resolve_timeout(const boost::system::error_code &error); |
81 | void handle_dns_result(const boost::system::error_code &error, | |
82 | const std::size_t bytes_transferred); | |
4e7b6ff9 | 83 | void handle_unavailable(); |
e18c1337 | 84 | void handle_ips(const std::vector<host_addr_pair> &result_ips); |
dbe986b9 | 85 | void handle_cname(const std::vector<src_cname_pair> &result_cnames); |
e18c1337 | 86 | void cname_resolve_callback(const bool was_success, |
23f51766 | 87 | const int cname_count); |
dbe986b9 | 88 | void schedule_retry(); |
23f51766 | 89 | void finalize_resolve(const bool success, const int cname_count=0); |
96779587 CH |
90 | void stop_trying(); |
91 | void wait_timer_timeout_handler(const boost::system::error_code &error); | |
23f51766 | 92 | void gather_results( const boost::net::dns::rr_list_t *rr_list, |
e18c1337 | 93 | std::vector<host_addr_pair> *result_ips, |
dbe986b9 | 94 | std::vector<src_cname_pair> *result_cnames, |
ad83004d | 95 | std::vector<string_pair> *result_nameservers ) const; |
36ad976b | 96 | |
c5b4902d | 97 | // variables |
36ad976b | 98 | private: |
96779587 | 99 | boost::asio::ip::udp::socket Socket; |
4e7b6ff9 | 100 | boost::net::dns_buffer_t ReceiveBuffer; |
ad83004d | 101 | boost::net::dns_buffer_t RequestBuffer; |
923626c0 | 102 | DnsIpProtocol Protocol; |
96779587 | 103 | boost::asio::ip::udp::endpoint NameServer; |
36ad976b CH |
104 | boost::asio::deadline_timer ResolveTimeoutTimer; |
105 | boost::asio::deadline_timer PauseBeforeRetryTimer; | |
106 | boost::asio::deadline_timer StaleDataLongtermTimer; | |
4e7b6ff9 | 107 | std::size_t NextIpIndex; |
36ad976b | 108 | int RetryCount; |
23f51766 CH |
109 | bool IsResolving; // true from async_resolve until finalize_resolve; |
110 | // false e.g. while waiting for StaleDataLongtermTimer | |
e91538f0 | 111 | std::string LogPrefix; |
ad83004d CH |
112 | boost::uuids::random_generator RandomIdGenerator; |
113 | uint16_t RequestId; | |
e18c1337 | 114 | bool OperationCancelled; |
96779587 | 115 | }; |
36ad976b | 116 | |
36ad976b CH |
117 | #endif |
118 | // (created using vim -- the world's best text editor) | |
119 |