3 * Resolver library with GLib integration
5 * Copyright (C) 2009-2010 Intel Corporation. All rights reserved.
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License version 2 as
9 * published by the Free Software Foundation.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
31 #include <sys/types.h>
32 #include <sys/socket.h>
34 #include <arpa/inet.h>
35 #include <arpa/nameser.h>
47 struct sockaddr_in sin;
48 struct sockaddr_in6 sin6;
52 struct sockaddr_in sin;
53 struct sockaddr_in6 sin6;
59 struct resolv_lookup {
64 struct sort_result *results;
66 struct resolv_query *ipv4_query;
67 struct resolv_query *ipv6_query;
72 GResolvResultFunc result_func;
83 struct resolv_lookup *lookup;
86 struct resolv_nameserver {
93 GIOChannel *udp_channel;
100 guint next_lookup_id;
101 GQueue *lookup_queue;
105 GList *nameserver_list;
107 struct __res_state res;
109 GResolvDebugFunc debug_func;
113 static void sort_and_return_results(struct resolv_lookup *lookup);
114 static void rfc3484_sort_results(struct resolv_lookup *lookup);
116 static inline void debug(GResolv *resolv, const char *format, ...)
121 if (resolv->debug_func == NULL)
124 va_start(ap, format);
126 if (vsnprintf(str, sizeof(str), format, ap) > 0)
127 resolv->debug_func(str, resolv->debug_data);
132 static void destroy_query(struct resolv_query *query)
134 if (query->timeout > 0)
135 g_source_remove(query->timeout);
140 static void destroy_lookup(struct resolv_lookup *lookup)
142 if (lookup->ipv4_query) {
143 destroy_query(lookup->ipv4_query);
144 g_queue_remove(lookup->resolv->query_queue, lookup->ipv4_query);
146 if (lookup->ipv6_query) {
147 destroy_query(lookup->ipv6_query);
148 g_queue_remove(lookup->resolv->query_queue, lookup->ipv4_query);
150 g_free(lookup->results);
154 static gboolean query_timeout(gpointer user_data)
156 struct resolv_query *query = user_data;
157 struct resolv_lookup *lookup = query->lookup;
158 GResolv *resolv = query->resolv;
162 if (query == lookup->ipv4_query) {
163 lookup->ipv4_status = G_RESOLV_RESULT_STATUS_NO_RESPONSE;
164 lookup->ipv4_query = NULL;
165 } else if (query == lookup->ipv6_query) {
166 lookup->ipv6_status = G_RESOLV_RESULT_STATUS_NO_RESPONSE;
167 lookup->ipv6_query = NULL;
169 if (!lookup->ipv4_query && !lookup->ipv4_query)
170 sort_and_return_results(lookup);
172 destroy_query(query);
173 g_queue_remove(resolv->query_queue, query);
178 static void free_nameserver(struct resolv_nameserver *nameserver)
180 if (nameserver == NULL)
183 if (nameserver->udp_watch > 0)
184 g_source_remove(nameserver->udp_watch);
186 if (nameserver->udp_channel != NULL)
187 g_io_channel_unref(nameserver->udp_channel);
189 g_free(nameserver->address);
193 static void flush_nameservers(GResolv *resolv)
197 for (list = g_list_first(resolv->nameserver_list);
198 list; list = g_list_next(list))
199 free_nameserver(list->data);
201 g_list_free(resolv->nameserver_list);
202 resolv->nameserver_list = NULL;
205 static int send_query(GResolv *resolv, const unsigned char *buf, int len)
209 if (resolv->nameserver_list == NULL)
212 for (list = g_list_first(resolv->nameserver_list);
213 list; list = g_list_next(list)) {
214 struct resolv_nameserver *nameserver = list->data;
217 if (nameserver->udp_channel == NULL)
220 sk = g_io_channel_unix_get_fd(nameserver->udp_channel);
222 sent = send(sk, buf, len, 0);
228 static gint compare_lookup_id(gconstpointer a, gconstpointer b)
230 const struct resolv_lookup *lookup = a;
231 guint id = GPOINTER_TO_UINT(b);
242 static gint compare_query_msgid(gconstpointer a, gconstpointer b)
244 const struct resolv_query *query = a;
245 uint16_t msgid = GPOINTER_TO_UINT(b);
247 if (query->msgid < msgid)
250 if (query->msgid > msgid)
256 static void add_result(struct resolv_lookup *lookup, int family, const void *data)
258 int n = lookup->nr_results++;
259 lookup->results = g_realloc(lookup->results,
260 sizeof(struct sort_result) * (n+1));
262 memset(&lookup->results[n], 0, sizeof(struct sort_result));
264 lookup->results[n].dst.sa.sa_family = family;
265 if (family == AF_INET)
266 memcpy(&lookup->results[n].dst.sin.sin_addr, data, NS_INADDRSZ);
268 memcpy(&lookup->results[n].dst.sin6.sin6_addr, data, NS_IN6ADDRSZ);
271 static void parse_response(struct resolv_nameserver *nameserver,
272 const unsigned char *buf, int len)
274 GResolv *resolv = nameserver->resolv;
275 GResolvResultStatus status;
276 struct resolv_query *query;
277 struct resolv_lookup *lookup;
281 int i, n, rcode, count;
283 debug(resolv, "response from %s", nameserver->address);
285 ns_initparse(buf, len, &msg);
287 rcode = ns_msg_getflag(msg, ns_f_rcode);
288 count = ns_msg_count(msg, ns_s_an);
290 debug(resolv, "msg id: 0x%04x rcode: %d count: %d",
291 ns_msg_id(msg), rcode, count);
295 status = G_RESOLV_RESULT_STATUS_SUCCESS;
298 status = G_RESOLV_RESULT_STATUS_FORMAT_ERROR;
301 status = G_RESOLV_RESULT_STATUS_SERVER_FAILURE;
304 status = G_RESOLV_RESULT_STATUS_NAME_ERROR;
307 status = G_RESOLV_RESULT_STATUS_NOT_IMPLEMENTED;
310 status = G_RESOLV_RESULT_STATUS_REFUSED;
313 status = G_RESOLV_RESULT_STATUS_ERROR;
317 list = g_queue_find_custom(resolv->query_queue,
318 GUINT_TO_POINTER(ns_msg_id(msg)), compare_query_msgid);
323 lookup = query->lookup;
325 if (query == lookup->ipv6_query) {
326 lookup->ipv6_status = status;
327 lookup->ipv6_query = NULL;
328 } else if (query == lookup->ipv4_query) {
329 lookup->ipv4_status = status;
330 lookup->ipv4_query = NULL;
333 for (i = 0, n = 0; i < count; i++) {
334 ns_parserr(&msg, ns_s_an, i, &rr);
336 if (ns_rr_class(rr) != ns_c_in)
339 g_assert(offsetof(struct sockaddr_in, sin_addr) ==
340 offsetof(struct sockaddr_in6, sin6_flowinfo));
342 if (ns_rr_type(rr) == ns_t_a &&
343 ns_rr_rdlen(rr) == NS_INADDRSZ) {
344 add_result(lookup, AF_INET, ns_rr_rdata(rr));
345 } else if (ns_rr_type(rr) == ns_t_aaaa &&
346 ns_rr_rdlen(rr) == NS_IN6ADDRSZ) {
347 add_result(lookup, AF_INET6, ns_rr_rdata(rr));
351 if (!lookup->ipv4_query && !lookup->ipv6_query)
352 sort_and_return_results(lookup);
354 destroy_query(query);
355 g_queue_remove(resolv->query_queue, query);
358 static gboolean received_udp_data(GIOChannel *channel, GIOCondition cond,
361 struct resolv_nameserver *nameserver = user_data;
362 unsigned char buf[4096];
365 if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
366 nameserver->udp_watch = 0;
370 sk = g_io_channel_unix_get_fd(nameserver->udp_channel);
372 len = recv(sk, buf, sizeof(buf), 0);
376 parse_response(nameserver, buf, len);
381 static int connect_udp_channel(struct resolv_nameserver *nameserver)
383 struct addrinfo hints, *rp;
387 memset(&hints, 0, sizeof(hints));
388 hints.ai_family = AF_UNSPEC;
389 hints.ai_socktype = SOCK_DGRAM;
390 hints.ai_flags = AI_PASSIVE | AI_NUMERICSERV | AI_NUMERICHOST;
392 sprintf(portnr, "%d", nameserver->port);
393 err = getaddrinfo(nameserver->address, portnr, &hints, &rp);
397 /* Do not blindly copy this code elsewhere; it doesn't loop over the
398 results using ->ai_next as it should. That's OK in *this* case
399 because it was a numeric lookup; we *know* there's only one. */
403 sk = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
409 if (connect(sk, rp->ai_addr, rp->ai_addrlen) < 0) {
417 nameserver->udp_channel = g_io_channel_unix_new(sk);
418 if (nameserver->udp_channel == NULL) {
423 g_io_channel_set_close_on_unref(nameserver->udp_channel, TRUE);
425 nameserver->udp_watch = g_io_add_watch(nameserver->udp_channel,
426 G_IO_IN | G_IO_NVAL | G_IO_ERR | G_IO_HUP,
427 received_udp_data, nameserver);
432 GResolv *g_resolv_new(int index)
439 resolv = g_try_new0(GResolv, 1);
443 resolv->ref_count = 1;
445 resolv->next_lookup_id = 1;
447 resolv->query_queue = g_queue_new();
448 if (resolv->query_queue == NULL) {
453 resolv->lookup_queue = g_queue_new();
454 if (resolv->lookup_queue == NULL) {
455 g_queue_free(resolv->query_queue);
460 resolv->index = index;
461 resolv->nameserver_list = NULL;
463 res_ninit(&resolv->res);
468 GResolv *g_resolv_ref(GResolv *resolv)
473 g_atomic_int_inc(&resolv->ref_count);
478 void g_resolv_unref(GResolv *resolv)
480 struct resolv_query *query;
485 if (g_atomic_int_dec_and_test(&resolv->ref_count) == FALSE)
488 while ((query = g_queue_pop_head(resolv->query_queue)))
489 destroy_query(query);
491 g_queue_free(resolv->query_queue);
492 g_queue_free(resolv->lookup_queue);
494 flush_nameservers(resolv);
496 res_nclose(&resolv->res);
501 void g_resolv_set_debug(GResolv *resolv,
502 GResolvDebugFunc func, gpointer user_data)
507 resolv->debug_func = func;
508 resolv->debug_data = user_data;
511 gboolean g_resolv_add_nameserver(GResolv *resolv, const char *address,
512 uint16_t port, unsigned long flags)
514 struct resolv_nameserver *nameserver;
519 nameserver = g_try_new0(struct resolv_nameserver, 1);
520 if (nameserver == NULL)
523 nameserver->address = g_strdup(address);
524 nameserver->port = port;
525 nameserver->flags = flags;
527 if (connect_udp_channel(nameserver) < 0) {
528 free_nameserver(nameserver);
532 nameserver->resolv = resolv;
534 resolv->nameserver_list = g_list_append(resolv->nameserver_list,
537 debug(resolv, "setting nameserver %s", address);
542 void g_resolv_flush_nameservers(GResolv *resolv)
547 flush_nameservers(resolv);
550 static void sort_and_return_results(struct resolv_lookup *lookup)
553 GResolvResultStatus status;
554 char **results = g_try_new0(char *, lookup->nr_results + 1);
560 rfc3484_sort_results(lookup);
562 for (i = 0; i < lookup->nr_results; i++) {
563 if (lookup->results[i].dst.sa.sa_family == AF_INET) {
564 if (!inet_ntop(AF_INET, &lookup->results[i].dst.sin.sin_addr,
567 } else if (lookup->results[i].dst.sa.sa_family == AF_INET6) {
568 if (!inet_ntop(AF_INET6, &lookup->results[i].dst.sin6.sin6_addr,
574 results[n++] = strdup(buf);
578 status = lookup->ipv4_status;
579 if (status == G_RESOLV_RESULT_STATUS_SUCCESS)
580 status = lookup->ipv6_status;
582 lookup->result_func(status, results, lookup->result_data);
585 g_queue_remove(lookup->resolv->lookup_queue, lookup);
586 destroy_lookup(lookup);
589 static gint add_query(struct resolv_lookup *lookup, const char *hostname, int type)
591 struct resolv_query *query = g_try_new0(struct resolv_query, 1);
592 unsigned char buf[4096];
598 len = res_mkquery(ns_o_query, hostname, ns_c_in, type,
599 NULL, 0, NULL, buf, sizeof(buf));
601 query->msgid = buf[0] << 8 | buf[1];
603 if (send_query(lookup->resolv, buf, len) < 0)
606 query->resolv = lookup->resolv;
607 query->lookup = lookup;
609 g_queue_push_tail(lookup->resolv->query_queue, query);
611 query->timeout = g_timeout_add_seconds(5, query_timeout, query);
613 if (type == ns_t_aaaa)
614 lookup->ipv6_query = query;
616 lookup->ipv4_query = query;
621 guint g_resolv_lookup_hostname(GResolv *resolv, const char *hostname,
622 GResolvResultFunc func, gpointer user_data)
624 struct resolv_lookup *lookup;
626 debug(resolv, "lookup hostname %s", hostname);
631 if (resolv->nameserver_list == NULL) {
634 for (i = 0; i < resolv->res.nscount; i++) {
636 int family = resolv->res.nsaddr_list[i].sin_family;
637 void *sa_addr = &resolv->res.nsaddr_list[i].sin_addr;
639 if (family != AF_INET && resolv->res._u._ext.nsaddrs[i]) {
641 sa_addr = &resolv->res._u._ext.nsaddrs[i]->sin6_addr;
643 if (family != AF_INET && family != AF_INET6)
646 if (inet_ntop(family, sa_addr, buf, sizeof(buf)))
647 g_resolv_add_nameserver(resolv, buf, 53, 0);
650 if (resolv->nameserver_list == NULL)
651 g_resolv_add_nameserver(resolv, "127.0.0.1", 53, 0);
654 lookup = g_try_new0(struct resolv_lookup, 1);
658 lookup->resolv = resolv;
659 lookup->result_func = func;
660 lookup->result_data = user_data;
661 lookup->id = resolv->next_lookup_id++;
663 if (add_query(lookup, hostname, ns_t_a)) {
667 if (add_query(lookup, hostname, ns_t_aaaa)) {
668 destroy_query(lookup->ipv4_query);
669 g_queue_remove(resolv->query_queue, lookup->ipv4_query);
674 g_queue_push_tail(resolv->lookup_queue, lookup);
678 gboolean g_resolv_cancel_lookup(GResolv *resolv, guint id)
682 list = g_queue_find_custom(resolv->lookup_queue,
683 GUINT_TO_POINTER(id), compare_lookup_id);
688 destroy_lookup(list->data);
689 g_queue_remove(resolv->query_queue, list->data);
694 static void find_srcaddr(struct sort_result *res)
697 socklen_t sl = sizeof(res->src);
699 fd = socket(res->dst.sa.sa_family, SOCK_DGRAM, IPPROTO_IP);
703 if (connect(fd, &res->dst.sa, sizeof(res->dst))) {
707 if (getsockname(fd, &res->src.sa, &sl)) {
711 res->reachable = TRUE;
717 unsigned char addr[NS_IN6ADDRSZ];
722 static const struct gai_table gai_labels[] = {
724 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
725 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01 },
729 .addr = { 0x20, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
730 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
734 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
735 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
739 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
740 0x00, 0x00, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00 },
744 /* Variations from RFC 3484, matching glibc behaviour */
745 .addr = { 0xfe, 0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
746 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
750 .addr = { 0xfc, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
751 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
755 .addr = { 0x20, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
756 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
761 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
762 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
768 static const struct gai_table gai_precedences[] = {
770 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
771 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01 },
775 .addr = { 0x20, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
776 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
780 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
781 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
785 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
786 0x00, 0x00, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00 },
790 .addr = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
791 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
797 static unsigned char v4mapped[] = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
798 0x00, 0x00, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00 };
800 static gboolean mask_compare(const unsigned char *one, const unsigned char *two, int mask)
803 if (memcmp(one, two, mask / 8))
809 if (mask && ((*one ^ *two) >> (8-mask)))
815 static int match_gai_table(struct sockaddr *sa, const struct gai_table *tbl)
817 struct sockaddr_in *sin = (void *)sa;
818 struct sockaddr_in6 *sin6 = (void *)sa;
821 if (sa->sa_family == AF_INET) {
823 memcpy(v4mapped+12, &sin->sin_addr, NS_INADDRSZ);
825 addr = &sin6->sin6_addr;
828 if (mask_compare(addr, tbl->addr, tbl->mask))
834 static void rfc3484_sort_results(struct resolv_lookup *lookup)
838 for (i = 0; i < lookup->nr_results; i++) {
839 struct sort_result *res = &lookup->results[i];
841 res->precedence = match_gai_table(&res->dst.sa, gai_precedences);
842 res->dst_label = match_gai_table(&res->dst.sa, gai_labels);
843 res->src_label = match_gai_table(&res->src.sa, gai_labels);
845 /* FIXME: Actually *sort* them... */