2 This file is part of catta.
4 catta is free software; you can redistribute it and/or modify it
5 under the terms of the GNU Lesser General Public License as
6 published by the Free Software Foundation; either version 2.1 of the
7 License, or (at your option) any later version.
9 catta is distributed in the hope that it will be useful, but WITHOUT
10 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
12 Public License for more details.
14 You should have received a copy of the GNU Lesser General Public
15 License along with catta; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
26 #include <catta/domain.h>
27 #include <catta/timeval.h>
28 #include <catta/malloc.h>
29 #include <catta/error.h>
32 #include <catta/log.h>
34 #define TIMEOUT_MSEC 5000
36 struct CattaSHostNameResolver {
40 CattaSRecordBrowser *record_browser_a;
41 CattaSRecordBrowser *record_browser_aaaa;
43 CattaSHostNameResolverCallback callback;
46 CattaRecord *address_record;
48 CattaProtocol protocol;
49 CattaLookupResultFlags flags;
51 CattaTimeEvent *time_event;
53 CATTA_LLIST_FIELDS(CattaSHostNameResolver, resolver);
56 static void finish(CattaSHostNameResolver *r, CattaResolverEvent event) {
60 catta_time_event_free(r->time_event);
65 case CATTA_RESOLVER_FOUND: {
68 assert(r->address_record);
70 switch (r->address_record->key->type) {
71 case CATTA_DNS_TYPE_A:
72 a.proto = CATTA_PROTO_INET;
73 a.data.ipv4 = r->address_record->data.a.address;
76 case CATTA_DNS_TYPE_AAAA:
77 a.proto = CATTA_PROTO_INET6;
78 a.data.ipv6 = r->address_record->data.aaaa.address;
85 r->callback(r, r->iface, r->protocol, CATTA_RESOLVER_FOUND, r->address_record->key->name, &a, r->flags, r->userdata);
90 case CATTA_RESOLVER_FAILURE:
92 r->callback(r, r->iface, r->protocol, event, r->host_name, NULL, r->flags, r->userdata);
97 static void time_event_callback(CattaTimeEvent *e, void *userdata) {
98 CattaSHostNameResolver *r = userdata;
103 catta_server_set_errno(r->server, CATTA_ERR_TIMEOUT);
104 finish(r, CATTA_RESOLVER_FAILURE);
107 static void start_timeout(CattaSHostNameResolver *r) {
114 catta_elapse_time(&tv, TIMEOUT_MSEC, 0);
116 r->time_event = catta_time_event_new(r->server->time_event_queue, &tv, time_event_callback, r);
119 static void record_browser_callback(
120 CattaSRecordBrowser*rr,
122 CattaProtocol protocol,
123 CattaBrowserEvent event,
125 CattaLookupResultFlags flags,
128 CattaSHostNameResolver *r = userdata;
135 case CATTA_BROWSER_NEW:
137 assert(record->key->type == CATTA_DNS_TYPE_A || record->key->type == CATTA_DNS_TYPE_AAAA);
139 if (r->iface > 0 && iface != r->iface)
142 if (r->protocol != CATTA_PROTO_UNSPEC && protocol != r->protocol)
148 if (r->protocol == CATTA_PROTO_UNSPEC)
149 r->protocol = protocol;
151 if (!r->address_record) {
152 r->address_record = catta_record_ref(record);
155 finish(r, CATTA_RESOLVER_FOUND);
160 case CATTA_BROWSER_REMOVE:
162 assert(record->key->type == CATTA_DNS_TYPE_A || record->key->type == CATTA_DNS_TYPE_AAAA);
164 if (r->address_record && catta_record_equal_no_ttl(record, r->address_record)) {
165 catta_record_unref(r->address_record);
166 r->address_record = NULL;
171 /** Look for a replacement */
172 if (r->record_browser_aaaa)
173 catta_s_record_browser_restart(r->record_browser_aaaa);
174 if (r->record_browser_a)
175 catta_s_record_browser_restart(r->record_browser_a);
182 case CATTA_BROWSER_CACHE_EXHAUSTED:
183 case CATTA_BROWSER_ALL_FOR_NOW:
187 case CATTA_BROWSER_FAILURE:
191 if (r->record_browser_aaaa)
192 catta_s_record_browser_free(r->record_browser_aaaa);
193 if (r->record_browser_a)
194 catta_s_record_browser_free(r->record_browser_a);
196 r->record_browser_a = r->record_browser_aaaa = NULL;
199 finish(r, CATTA_RESOLVER_FAILURE);
204 CattaSHostNameResolver *catta_s_host_name_resolver_new(
207 CattaProtocol protocol,
208 const char *host_name,
209 CattaProtocol aprotocol,
210 CattaLookupFlags flags,
211 CattaSHostNameResolverCallback callback,
214 CattaSHostNameResolver *r;
221 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_IF_VALID(iface), CATTA_ERR_INVALID_INTERFACE);
222 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_PROTO_VALID(protocol), CATTA_ERR_INVALID_PROTOCOL);
223 CATTA_CHECK_VALIDITY_RETURN_NULL(server, catta_is_valid_fqdn(host_name), CATTA_ERR_INVALID_HOST_NAME);
224 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_PROTO_VALID(aprotocol), CATTA_ERR_INVALID_PROTOCOL);
225 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_FLAGS_VALID(flags, CATTA_LOOKUP_USE_WIDE_AREA|CATTA_LOOKUP_USE_MULTICAST), CATTA_ERR_INVALID_FLAGS);
227 if (!(r = catta_new(CattaSHostNameResolver, 1))) {
228 catta_server_set_errno(server, CATTA_ERR_NO_MEMORY);
233 r->host_name = catta_normalize_name_strdup(host_name);
234 r->callback = callback;
235 r->userdata = userdata;
236 r->address_record = NULL;
238 r->protocol = protocol;
241 r->record_browser_a = r->record_browser_aaaa = NULL;
243 r->time_event = NULL;
245 CATTA_LLIST_PREPEND(CattaSHostNameResolver, resolver, server->host_name_resolvers, r);
247 r->record_browser_aaaa = r->record_browser_a = NULL;
249 if (aprotocol == CATTA_PROTO_INET || aprotocol == CATTA_PROTO_UNSPEC) {
250 k = catta_key_new(host_name, CATTA_DNS_CLASS_IN, CATTA_DNS_TYPE_A);
251 r->record_browser_a = catta_s_record_browser_new(server, iface, protocol, k, flags, record_browser_callback, r);
254 if (!r->record_browser_a)
258 if (aprotocol == CATTA_PROTO_INET6 || aprotocol == CATTA_PROTO_UNSPEC) {
259 k = catta_key_new(host_name, CATTA_DNS_CLASS_IN, CATTA_DNS_TYPE_AAAA);
260 r->record_browser_aaaa = catta_s_record_browser_new(server, iface, protocol, k, flags, record_browser_callback, r);
263 if (!r->record_browser_aaaa)
267 assert(r->record_browser_aaaa || r->record_browser_a);
274 catta_s_host_name_resolver_free(r);
278 void catta_s_host_name_resolver_free(CattaSHostNameResolver *r) {
281 CATTA_LLIST_REMOVE(CattaSHostNameResolver, resolver, r->server->host_name_resolvers, r);
283 if (r->record_browser_a)
284 catta_s_record_browser_free(r->record_browser_a);
286 if (r->record_browser_aaaa)
287 catta_s_record_browser_free(r->record_browser_aaaa);
290 catta_time_event_free(r->time_event);
292 if (r->address_record)
293 catta_record_unref(r->address_record);
295 catta_free(r->host_name);