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
28 #include <catta/domain.h>
29 #include <catta/timeval.h>
30 #include <catta/malloc.h>
31 #include <catta/error.h>
34 #include <catta/log.h>
36 #define TIMEOUT_MSEC 5000
38 struct CattaSServiceResolver {
43 CattaProtocol address_protocol;
46 CattaProtocol protocol;
48 CattaSRecordBrowser *record_browser_srv;
49 CattaSRecordBrowser *record_browser_txt;
50 CattaSRecordBrowser *record_browser_a;
51 CattaSRecordBrowser *record_browser_aaaa;
53 CattaRecord *srv_record, *txt_record, *address_record;
54 CattaLookupResultFlags srv_flags, txt_flags, address_flags;
56 CattaSServiceResolverCallback callback;
58 CattaLookupFlags user_flags;
60 CattaTimeEvent *time_event;
62 CATTA_LLIST_FIELDS(CattaSServiceResolver, resolver);
65 static void finish(CattaSServiceResolver *r, CattaResolverEvent event) {
66 CattaLookupResultFlags flags;
71 catta_time_event_free(r->time_event);
81 case CATTA_RESOLVER_FAILURE:
100 case CATTA_RESOLVER_FOUND: {
103 assert(event == CATTA_RESOLVER_FOUND);
105 assert(r->srv_record);
107 if (r->address_record) {
108 switch (r->address_record->key->type) {
109 case CATTA_DNS_TYPE_A:
110 a.proto = CATTA_PROTO_INET;
111 a.data.ipv4 = r->address_record->data.a.address;
114 case CATTA_DNS_TYPE_AAAA:
115 a.proto = CATTA_PROTO_INET6;
116 a.data.ipv6 = r->address_record->data.aaaa.address;
132 r->srv_record->data.srv.name,
133 r->address_record ? &a : NULL,
134 r->srv_record->data.srv.port,
135 r->txt_record ? r->txt_record->data.txt.string_list : NULL,
144 static void time_event_callback(CattaTimeEvent *e, void *userdata) {
145 CattaSServiceResolver *r = userdata;
150 catta_server_set_errno(r->server, CATTA_ERR_TIMEOUT);
151 finish(r, CATTA_RESOLVER_FAILURE);
154 static void start_timeout(CattaSServiceResolver *r) {
161 catta_elapse_time(&tv, TIMEOUT_MSEC, 0);
163 r->time_event = catta_time_event_new(r->server->time_event_queue, &tv, time_event_callback, r);
166 static void record_browser_callback(
167 CattaSRecordBrowser*rr,
169 CattaProtocol protocol,
170 CattaBrowserEvent event,
172 CattaLookupResultFlags flags,
175 CattaSServiceResolver *r = userdata;
180 if (rr == r->record_browser_aaaa || rr == r->record_browser_a)
181 r->address_flags = flags;
182 else if (rr == r->record_browser_srv)
183 r->srv_flags = flags;
184 else if (rr == r->record_browser_txt)
185 r->txt_flags = flags;
189 case CATTA_BROWSER_NEW: {
193 if (r->iface > 0 && iface > 0 && iface != r->iface)
196 if (r->protocol != CATTA_PROTO_UNSPEC && protocol != CATTA_PROTO_UNSPEC && protocol != r->protocol)
202 if (r->protocol == CATTA_PROTO_UNSPEC)
203 r->protocol = protocol;
205 switch (record->key->type) {
206 case CATTA_DNS_TYPE_SRV:
207 if (!r->srv_record) {
208 r->srv_record = catta_record_ref(record);
211 if (r->record_browser_a) {
212 catta_s_record_browser_free(r->record_browser_a);
213 r->record_browser_a = NULL;
216 if (r->record_browser_aaaa) {
217 catta_s_record_browser_free(r->record_browser_aaaa);
218 r->record_browser_aaaa = NULL;
221 if (!(r->user_flags & CATTA_LOOKUP_NO_ADDRESS)) {
223 if (r->address_protocol == CATTA_PROTO_INET || r->address_protocol == CATTA_PROTO_UNSPEC) {
224 CattaKey *k = catta_key_new(r->srv_record->data.srv.name, CATTA_DNS_CLASS_IN, CATTA_DNS_TYPE_A);
225 r->record_browser_a = catta_s_record_browser_new(r->server, r->iface, r->protocol, k, r->user_flags & ~(CATTA_LOOKUP_NO_TXT|CATTA_LOOKUP_NO_ADDRESS), record_browser_callback, r);
229 if (r->address_protocol == CATTA_PROTO_INET6 || r->address_protocol == CATTA_PROTO_UNSPEC) {
230 CattaKey *k = catta_key_new(r->srv_record->data.srv.name, CATTA_DNS_CLASS_IN, CATTA_DNS_TYPE_AAAA);
231 r->record_browser_aaaa = catta_s_record_browser_new(r->server, r->iface, r->protocol, k, r->user_flags & ~(CATTA_LOOKUP_NO_TXT|CATTA_LOOKUP_NO_ADDRESS), record_browser_callback, r);
238 case CATTA_DNS_TYPE_TXT:
240 assert(!(r->user_flags & CATTA_LOOKUP_NO_TXT));
242 if (!r->txt_record) {
243 r->txt_record = catta_record_ref(record);
248 case CATTA_DNS_TYPE_A:
249 case CATTA_DNS_TYPE_AAAA:
251 assert(!(r->user_flags & CATTA_LOOKUP_NO_ADDRESS));
253 if (!r->address_record) {
254 r->address_record = catta_record_ref(record);
266 (r->txt_record || (r->user_flags & CATTA_LOOKUP_NO_TXT)) &&
267 (r->address_record || (r->user_flags & CATTA_LOOKUP_NO_ADDRESS)))
268 finish(r, CATTA_RESOLVER_FOUND);
274 case CATTA_BROWSER_REMOVE:
278 switch (record->key->type) {
279 case CATTA_DNS_TYPE_SRV:
281 if (r->srv_record && catta_record_equal_no_ttl(record, r->srv_record)) {
282 catta_record_unref(r->srv_record);
283 r->srv_record = NULL;
285 if (r->record_browser_a) {
286 catta_s_record_browser_free(r->record_browser_a);
287 r->record_browser_a = NULL;
290 if (r->record_browser_aaaa) {
291 catta_s_record_browser_free(r->record_browser_aaaa);
292 r->record_browser_aaaa = NULL;
295 /** Look for a replacement */
296 catta_s_record_browser_restart(r->record_browser_srv);
302 case CATTA_DNS_TYPE_TXT:
304 assert(!(r->user_flags & CATTA_LOOKUP_NO_TXT));
306 if (r->txt_record && catta_record_equal_no_ttl(record, r->txt_record)) {
307 catta_record_unref(r->txt_record);
308 r->txt_record = NULL;
310 /** Look for a replacement */
311 catta_s_record_browser_restart(r->record_browser_txt);
316 case CATTA_DNS_TYPE_A:
317 case CATTA_DNS_TYPE_AAAA:
319 assert(!(r->user_flags & CATTA_LOOKUP_NO_ADDRESS));
321 if (r->address_record && catta_record_equal_no_ttl(record, r->address_record)) {
322 catta_record_unref(r->address_record);
323 r->address_record = NULL;
325 /** Look for a replacement */
326 if (r->record_browser_aaaa)
327 catta_s_record_browser_restart(r->record_browser_aaaa);
328 if (r->record_browser_a)
329 catta_s_record_browser_restart(r->record_browser_a);
340 case CATTA_BROWSER_CACHE_EXHAUSTED:
341 case CATTA_BROWSER_ALL_FOR_NOW:
344 case CATTA_BROWSER_FAILURE:
346 if (rr == r->record_browser_a && r->record_browser_aaaa) {
347 /* We were looking for both AAAA and A, and the other query is still living, so we'll not die */
348 catta_s_record_browser_free(r->record_browser_a);
349 r->record_browser_a = NULL;
353 if (rr == r->record_browser_aaaa && r->record_browser_a) {
354 /* We were looking for both AAAA and A, and the other query is still living, so we'll not die */
355 catta_s_record_browser_free(r->record_browser_aaaa);
356 r->record_browser_aaaa = NULL;
360 /* Hmm, everything's lost, tell the user */
362 if (r->record_browser_srv)
363 catta_s_record_browser_free(r->record_browser_srv);
364 if (r->record_browser_txt)
365 catta_s_record_browser_free(r->record_browser_txt);
366 if (r->record_browser_a)
367 catta_s_record_browser_free(r->record_browser_a);
368 if (r->record_browser_aaaa)
369 catta_s_record_browser_free(r->record_browser_aaaa);
371 r->record_browser_srv = r->record_browser_txt = r->record_browser_a = r->record_browser_aaaa = NULL;
373 finish(r, CATTA_RESOLVER_FAILURE);
378 CattaSServiceResolver *catta_s_service_resolver_new(
381 CattaProtocol protocol,
385 CattaProtocol aprotocol,
386 CattaLookupFlags flags,
387 CattaSServiceResolverCallback callback,
390 CattaSServiceResolver *r;
392 char n[CATTA_DOMAIN_NAME_MAX];
399 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_IF_VALID(iface), CATTA_ERR_INVALID_INTERFACE);
400 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_PROTO_VALID(protocol), CATTA_ERR_INVALID_PROTOCOL);
401 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_PROTO_VALID(aprotocol), CATTA_ERR_INVALID_PROTOCOL);
402 CATTA_CHECK_VALIDITY_RETURN_NULL(server, !domain || catta_is_valid_domain_name(domain), CATTA_ERR_INVALID_DOMAIN_NAME);
403 CATTA_CHECK_VALIDITY_RETURN_NULL(server, !name || catta_is_valid_service_name(name), CATTA_ERR_INVALID_SERVICE_NAME);
404 CATTA_CHECK_VALIDITY_RETURN_NULL(server, catta_is_valid_service_type_strict(type), CATTA_ERR_INVALID_SERVICE_TYPE);
405 CATTA_CHECK_VALIDITY_RETURN_NULL(server, CATTA_FLAGS_VALID(flags, CATTA_LOOKUP_USE_WIDE_AREA|CATTA_LOOKUP_USE_MULTICAST|CATTA_LOOKUP_NO_TXT|CATTA_LOOKUP_NO_ADDRESS), CATTA_ERR_INVALID_FLAGS);
408 domain = server->domain_name;
410 if ((ret = catta_service_name_join(n, sizeof(n), name, type, domain)) < 0) {
411 catta_server_set_errno(server, ret);
415 if (!(r = catta_new(CattaSServiceResolver, 1))) {
416 catta_server_set_errno(server, CATTA_ERR_NO_MEMORY);
421 r->service_name = catta_strdup(name);
422 r->service_type = catta_normalize_name_strdup(type);
423 r->domain_name = catta_normalize_name_strdup(domain);
424 r->callback = callback;
425 r->userdata = userdata;
426 r->address_protocol = aprotocol;
427 r->srv_record = r->txt_record = r->address_record = NULL;
428 r->srv_flags = r->txt_flags = r->address_flags = 0;
430 r->protocol = protocol;
431 r->user_flags = flags;
432 r->record_browser_a = r->record_browser_aaaa = r->record_browser_srv = r->record_browser_txt = NULL;
433 r->time_event = NULL;
434 CATTA_LLIST_PREPEND(CattaSServiceResolver, resolver, server->service_resolvers, r);
436 k = catta_key_new(n, CATTA_DNS_CLASS_IN, CATTA_DNS_TYPE_SRV);
437 r->record_browser_srv = catta_s_record_browser_new(server, iface, protocol, k, flags & ~(CATTA_LOOKUP_NO_TXT|CATTA_LOOKUP_NO_ADDRESS), record_browser_callback, r);
440 if (!r->record_browser_srv) {
441 catta_s_service_resolver_free(r);
445 if (!(flags & CATTA_LOOKUP_NO_TXT)) {
446 k = catta_key_new(n, CATTA_DNS_CLASS_IN, CATTA_DNS_TYPE_TXT);
447 r->record_browser_txt = catta_s_record_browser_new(server, iface, protocol, k, flags & ~(CATTA_LOOKUP_NO_TXT|CATTA_LOOKUP_NO_ADDRESS), record_browser_callback, r);
450 if (!r->record_browser_txt) {
451 catta_s_service_resolver_free(r);
461 void catta_s_service_resolver_free(CattaSServiceResolver *r) {
464 CATTA_LLIST_REMOVE(CattaSServiceResolver, resolver, r->server->service_resolvers, r);
467 catta_time_event_free(r->time_event);
469 if (r->record_browser_srv)
470 catta_s_record_browser_free(r->record_browser_srv);
471 if (r->record_browser_txt)
472 catta_s_record_browser_free(r->record_browser_txt);
473 if (r->record_browser_a)
474 catta_s_record_browser_free(r->record_browser_a);
475 if (r->record_browser_aaaa)
476 catta_s_record_browser_free(r->record_browser_aaaa);
479 catta_record_unref(r->srv_record);
481 catta_record_unref(r->txt_record);
482 if (r->address_record)
483 catta_record_unref(r->address_record);
485 catta_free(r->service_name);
486 catta_free(r->service_type);
487 catta_free(r->domain_name);