1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
3 /* GIO - GLib Input, Output and Streaming Library
5 * Copyright (C) 2008 Red Hat, Inc.
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General
18 * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
28 #include "gthreadedresolver.h"
29 #include "gnetworkingprivate.h"
31 #include "gcancellable.h"
32 #include "ginetaddress.h"
33 #include "ginetsocketaddress.h"
35 #include "gsocketaddress.h"
36 #include "gsrvtarget.h"
39 G_DEFINE_TYPE (GThreadedResolver, g_threaded_resolver, G_TYPE_RESOLVER)
42 g_threaded_resolver_init (GThreadedResolver *gtr)
47 g_resolver_error_from_addrinfo_error (gint err)
52 #if defined(EAI_NODATA) && (EAI_NODATA != EAI_NONAME)
56 return G_RESOLVER_ERROR_NOT_FOUND;
59 return G_RESOLVER_ERROR_TEMPORARY_FAILURE;
62 return G_RESOLVER_ERROR_INTERNAL;
66 static struct addrinfo addrinfo_hints;
69 do_lookup_by_name (GTask *task,
70 gpointer source_object,
72 GCancellable *cancellable)
74 const char *hostname = task_data;
75 struct addrinfo *res = NULL;
79 retval = getaddrinfo (hostname, NULL, &addrinfo_hints, &res);
84 GSocketAddress *sockaddr;
88 for (ai = res; ai; ai = ai->ai_next)
90 sockaddr = g_socket_address_new_from_native (ai->ai_addr, ai->ai_addrlen);
91 if (!sockaddr || !G_IS_INET_SOCKET_ADDRESS (sockaddr))
94 addr = g_object_ref (g_inet_socket_address_get_address ((GInetSocketAddress *)sockaddr));
95 addresses = g_list_prepend (addresses, addr);
96 g_object_unref (sockaddr);
99 if (addresses != NULL)
101 addresses = g_list_reverse (addresses);
102 g_task_return_pointer (task, addresses,
103 (GDestroyNotify)g_resolver_free_addresses);
107 /* All addresses failed to be converted to GSocketAddresses. */
108 g_task_return_new_error (task,
110 G_RESOLVER_ERROR_NOT_FOUND,
111 _("Error resolving '%s': %s"),
113 _("No valid addresses were found"));
118 g_task_return_new_error (task,
120 g_resolver_error_from_addrinfo_error (retval),
121 _("Error resolving '%s': %s"),
122 hostname, gai_strerror (retval));
130 lookup_by_name (GResolver *resolver,
131 const gchar *hostname,
132 GCancellable *cancellable,
138 task = g_task_new (resolver, cancellable, NULL, NULL);
139 g_task_set_task_data (task, g_strdup (hostname), g_free);
140 g_task_set_return_on_cancel (task, TRUE);
141 g_task_run_in_thread_sync (task, do_lookup_by_name);
142 addresses = g_task_propagate_pointer (task, error);
143 g_object_unref (task);
149 lookup_by_name_async (GResolver *resolver,
150 const gchar *hostname,
151 GCancellable *cancellable,
152 GAsyncReadyCallback callback,
157 task = g_task_new (resolver, cancellable, callback, user_data);
158 g_task_set_task_data (task, g_strdup (hostname), g_free);
159 g_task_set_return_on_cancel (task, TRUE);
160 g_task_run_in_thread (task, do_lookup_by_name);
161 g_object_unref (task);
165 lookup_by_name_finish (GResolver *resolver,
166 GAsyncResult *result,
169 g_return_val_if_fail (g_task_is_valid (result, resolver), NULL);
171 return g_task_propagate_pointer (G_TASK (result), error);
176 do_lookup_by_address (GTask *task,
177 gpointer source_object,
179 GCancellable *cancellable)
181 GInetAddress *address = task_data;
182 struct sockaddr_storage sockaddr;
184 GSocketAddress *gsockaddr;
185 gchar name[NI_MAXHOST];
188 gsockaddr = g_inet_socket_address_new (address, 0);
189 g_socket_address_to_native (gsockaddr, (struct sockaddr *)&sockaddr,
190 sizeof (sockaddr), NULL);
191 sockaddr_size = g_socket_address_get_native_size (gsockaddr);
192 g_object_unref (gsockaddr);
194 retval = getnameinfo ((struct sockaddr *)&sockaddr, sockaddr_size,
195 name, sizeof (name), NULL, 0, NI_NAMEREQD);
197 g_task_return_pointer (task, g_strdup (name), g_free);
202 phys = g_inet_address_to_string (address);
203 g_task_return_new_error (task,
205 g_resolver_error_from_addrinfo_error (retval),
206 _("Error reverse-resolving '%s': %s"),
207 phys ? phys : "(unknown)",
208 gai_strerror (retval));
214 lookup_by_address (GResolver *resolver,
215 GInetAddress *address,
216 GCancellable *cancellable,
222 task = g_task_new (resolver, cancellable, NULL, NULL);
223 g_task_set_task_data (task, g_object_ref (address), g_object_unref);
224 g_task_set_return_on_cancel (task, TRUE);
225 g_task_run_in_thread_sync (task, do_lookup_by_address);
226 name = g_task_propagate_pointer (task, error);
227 g_object_unref (task);
233 lookup_by_address_async (GResolver *resolver,
234 GInetAddress *address,
235 GCancellable *cancellable,
236 GAsyncReadyCallback callback,
241 task = g_task_new (resolver, cancellable, callback, user_data);
242 g_task_set_task_data (task, g_object_ref (address), g_object_unref);
243 g_task_set_return_on_cancel (task, TRUE);
244 g_task_run_in_thread (task, do_lookup_by_address);
245 g_object_unref (task);
249 lookup_by_address_finish (GResolver *resolver,
250 GAsyncResult *result,
253 g_return_val_if_fail (g_task_is_valid (result, resolver), NULL);
255 return g_task_propagate_pointer (G_TASK (result), error);
259 #if defined(G_OS_UNIX)
262 /* Copy from bionic/libc/private/arpa_nameser_compat.h
263 * and bionic/libc/private/arpa_nameser.h */
265 unsigned id :16; /* query identification number */
266 #if BYTE_ORDER == BIG_ENDIAN
267 /* fields in third byte */
268 unsigned qr: 1; /* response flag */
269 unsigned opcode: 4; /* purpose of message */
270 unsigned aa: 1; /* authoritive answer */
271 unsigned tc: 1; /* truncated message */
272 unsigned rd: 1; /* recursion desired */
273 /* fields in fourth byte */
274 unsigned ra: 1; /* recursion available */
275 unsigned unused :1; /* unused bits (MBZ as of 4.9.3a3) */
276 unsigned ad: 1; /* authentic data from named */
277 unsigned cd: 1; /* checking disabled by resolver */
278 unsigned rcode :4; /* response code */
280 #if BYTE_ORDER == LITTLE_ENDIAN || BYTE_ORDER == PDP_ENDIAN
281 /* fields in third byte */
282 unsigned rd :1; /* recursion desired */
283 unsigned tc :1; /* truncated message */
284 unsigned aa :1; /* authoritive answer */
285 unsigned opcode :4; /* purpose of message */
286 unsigned qr :1; /* response flag */
287 /* fields in fourth byte */
288 unsigned rcode :4; /* response code */
289 unsigned cd: 1; /* checking disabled by resolver */
290 unsigned ad: 1; /* authentic data from named */
291 unsigned unused :1; /* unused bits (MBZ as of 4.9.3a3) */
292 unsigned ra :1; /* recursion available */
294 /* remaining bytes */
295 unsigned qdcount :16; /* number of question entries */
296 unsigned ancount :16; /* number of answer entries */
297 unsigned nscount :16; /* number of authority entries */
298 unsigned arcount :16; /* number of resource entries */
301 #define NS_INT32SZ 4 /* #/bytes of data in a uint32_t */
302 #define NS_INT16SZ 2 /* #/bytes of data in a uint16_t */
304 #define NS_GET16(s, cp) do { \
305 const u_char *t_cp = (const u_char *)(cp); \
306 (s) = ((uint16_t)t_cp[0] << 8) \
307 | ((uint16_t)t_cp[1]) \
309 (cp) += NS_INT16SZ; \
310 } while (/*CONSTCOND*/0)
312 #define NS_GET32(l, cp) do { \
313 const u_char *t_cp = (const u_char *)(cp); \
314 (l) = ((uint32_t)t_cp[0] << 24) \
315 | ((uint32_t)t_cp[1] << 16) \
316 | ((uint32_t)t_cp[2] << 8) \
317 | ((uint32_t)t_cp[3]) \
319 (cp) += NS_INT32SZ; \
320 } while (/*CONSTCOND*/0)
322 #define GETSHORT NS_GET16
323 #define GETLONG NS_GET32
327 /* From bionic/libc/private/resolv_private.h */
328 int dn_expand(const u_char *, const u_char *, const u_char *, char *, int);
329 #define dn_skipname __dn_skipname
330 int dn_skipname(const u_char *, const u_char *);
332 /* From bionic/libc/private/arpa_nameser_compat.h */
334 #define T_TXT ns_t_txt
335 #define T_SOA ns_t_soa
338 /* From bionic/libc/private/arpa_nameser.h */
339 typedef enum __ns_type {
340 ns_t_invalid = 0, /* Cookie. */
341 ns_t_a = 1, /* Host address. */
342 ns_t_ns = 2, /* Authoritative server. */
343 ns_t_md = 3, /* Mail destination. */
344 ns_t_mf = 4, /* Mail forwarder. */
345 ns_t_cname = 5, /* Canonical name. */
346 ns_t_soa = 6, /* Start of authority zone. */
347 ns_t_mb = 7, /* Mailbox domain name. */
348 ns_t_mg = 8, /* Mail group member. */
349 ns_t_mr = 9, /* Mail rename name. */
350 ns_t_null = 10, /* Null resource record. */
351 ns_t_wks = 11, /* Well known service. */
352 ns_t_ptr = 12, /* Domain name pointer. */
353 ns_t_hinfo = 13, /* Host information. */
354 ns_t_minfo = 14, /* Mailbox information. */
355 ns_t_mx = 15, /* Mail routing information. */
356 ns_t_txt = 16, /* Text strings. */
357 ns_t_rp = 17, /* Responsible person. */
358 ns_t_afsdb = 18, /* AFS cell database. */
359 ns_t_x25 = 19, /* X_25 calling address. */
360 ns_t_isdn = 20, /* ISDN calling address. */
361 ns_t_rt = 21, /* Router. */
362 ns_t_nsap = 22, /* NSAP address. */
363 ns_t_nsap_ptr = 23, /* Reverse NSAP lookup (deprecated). */
364 ns_t_sig = 24, /* Security signature. */
365 ns_t_key = 25, /* Security key. */
366 ns_t_px = 26, /* X.400 mail mapping. */
367 ns_t_gpos = 27, /* Geographical position (withdrawn). */
368 ns_t_aaaa = 28, /* Ip6 Address. */
369 ns_t_loc = 29, /* Location Information. */
370 ns_t_nxt = 30, /* Next domain (security). */
371 ns_t_eid = 31, /* Endpoint identifier. */
372 ns_t_nimloc = 32, /* Nimrod Locator. */
373 ns_t_srv = 33, /* Server Selection. */
374 ns_t_atma = 34, /* ATM Address */
375 ns_t_naptr = 35, /* Naming Authority PoinTeR */
376 ns_t_kx = 36, /* Key Exchange */
377 ns_t_cert = 37, /* Certification record */
378 ns_t_a6 = 38, /* IPv6 address (deprecates AAAA) */
379 ns_t_dname = 39, /* Non-terminal DNAME (for IPv6) */
380 ns_t_sink = 40, /* Kitchen sink (experimentatl) */
381 ns_t_opt = 41, /* EDNS0 option (meta-RR) */
382 ns_t_apl = 42, /* Address prefix list (RFC 3123) */
383 ns_t_tkey = 249, /* Transaction key */
384 ns_t_tsig = 250, /* Transaction signature. */
385 ns_t_ixfr = 251, /* Incremental zone transfer. */
386 ns_t_axfr = 252, /* Transfer zone of authority. */
387 ns_t_mailb = 253, /* Transfer mailbox records. */
388 ns_t_maila = 254, /* Transfer mail agent records. */
389 ns_t_any = 255, /* Wildcard match. */
390 ns_t_zxfr = 256, /* BIND-specific, nonstandard. */
394 #endif /* __BIONIC__ */
397 parse_res_srv (guchar *answer,
402 guint16 priority, weight, port;
404 GETSHORT (priority, *p);
405 GETSHORT (weight, *p);
407 *p += dn_expand (answer, end, *p, namebuf, sizeof (namebuf));
409 return g_variant_new ("(qqqs)",
417 parse_res_soa (guchar *answer,
421 gchar mnamebuf[1024];
422 gchar rnamebuf[1024];
423 guint32 serial, refresh, retry, expire, ttl;
425 *p += dn_expand (answer, end, *p, mnamebuf, sizeof (mnamebuf));
426 *p += dn_expand (answer, end, *p, rnamebuf, sizeof (rnamebuf));
428 GETLONG (serial, *p);
429 GETLONG (refresh, *p);
431 GETLONG (expire, *p);
434 return g_variant_new ("(ssuuuuu)",
445 parse_res_ns (guchar *answer,
451 *p += dn_expand (answer, end, *p, namebuf, sizeof (namebuf));
453 return g_variant_new ("(s)", namebuf);
457 parse_res_mx (guchar *answer,
464 GETSHORT (preference, *p);
466 *p += dn_expand (answer, end, *p, namebuf, sizeof (namebuf));
468 return g_variant_new ("(qs)",
474 parse_res_txt (guchar *answer,
483 array = g_ptr_array_new_with_free_func (g_free);
489 g_ptr_array_add (array, g_strndup ((gchar *)at, len));
494 record = g_variant_new ("(@as)",
495 g_variant_new_strv ((const gchar **)array->pdata, array->len));
496 g_ptr_array_free (array, TRUE);
501 g_resolver_record_type_to_rrtype (GResolverRecordType type)
505 case G_RESOLVER_RECORD_SRV:
507 case G_RESOLVER_RECORD_TXT:
509 case G_RESOLVER_RECORD_SOA:
511 case G_RESOLVER_RECORD_NS:
513 case G_RESOLVER_RECORD_MX:
516 g_return_val_if_reached (-1);
520 g_resolver_records_from_res_query (const gchar *rrname,
530 guint16 type, qclass, rdlength;
538 if (len == 0 || herr == HOST_NOT_FOUND || herr == NO_DATA)
540 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_NOT_FOUND,
541 _("No DNS record of the requested type for '%s'"), rrname);
543 else if (herr == TRY_AGAIN)
545 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_TEMPORARY_FAILURE,
546 _("Temporarily unable to resolve '%s'"), rrname);
550 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_INTERNAL,
551 _("Error resolving '%s'"), rrname);
559 header = (HEADER *)answer;
560 p = answer + sizeof (HEADER);
564 count = ntohs (header->qdcount);
565 while (count-- && p < end)
567 p += dn_expand (answer, end, p, namebuf, sizeof (namebuf));
570 /* To silence gcc warnings */
571 namebuf[0] = namebuf[1];
575 count = ntohs (header->ancount);
576 while (count-- && p < end)
578 p += dn_expand (answer, end, p, namebuf, sizeof (namebuf));
580 GETSHORT (qclass, p);
582 ttl = ttl; /* To avoid -Wunused-but-set-variable */
583 GETSHORT (rdlength, p);
585 if (type != rrtype || qclass != C_IN)
594 record = parse_res_srv (answer, end, &p);
597 record = parse_res_mx (answer, end, &p);
600 record = parse_res_soa (answer, end, &p);
603 record = parse_res_ns (answer, end, &p);
606 record = parse_res_txt (answer, p + rdlength, &p);
609 g_warn_if_reached ();
615 records = g_list_prepend (records, record);
620 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_NOT_FOUND,
621 _("No DNS record of the requested type for '%s'"), rrname);
629 #elif defined(G_OS_WIN32)
632 parse_dns_srv (DNS_RECORD *rec)
634 return g_variant_new ("(qqqs)",
635 (guint16)rec->Data.SRV.wPriority,
636 (guint16)rec->Data.SRV.wWeight,
637 (guint16)rec->Data.SRV.wPort,
638 rec->Data.SRV.pNameTarget);
642 parse_dns_soa (DNS_RECORD *rec)
644 return g_variant_new ("(ssuuuuu)",
645 rec->Data.SOA.pNamePrimaryServer,
646 rec->Data.SOA.pNameAdministrator,
647 (guint32)rec->Data.SOA.dwSerialNo,
648 (guint32)rec->Data.SOA.dwRefresh,
649 (guint32)rec->Data.SOA.dwRetry,
650 (guint32)rec->Data.SOA.dwExpire,
651 (guint32)rec->Data.SOA.dwDefaultTtl);
655 parse_dns_ns (DNS_RECORD *rec)
657 return g_variant_new ("(s)", rec->Data.NS.pNameHost);
661 parse_dns_mx (DNS_RECORD *rec)
663 return g_variant_new ("(qs)",
664 (guint16)rec->Data.MX.wPreference,
665 rec->Data.MX.pNameExchange);
669 parse_dns_txt (DNS_RECORD *rec)
675 array = g_ptr_array_new ();
676 for (i = 0; i < rec->Data.TXT.dwStringCount; i++)
677 g_ptr_array_add (array, rec->Data.TXT.pStringArray[i]);
678 record = g_variant_new ("(@as)",
679 g_variant_new_strv ((const gchar **)array->pdata, array->len));
680 g_ptr_array_free (array, TRUE);
685 g_resolver_record_type_to_dnstype (GResolverRecordType type)
689 case G_RESOLVER_RECORD_SRV:
691 case G_RESOLVER_RECORD_TXT:
692 return DNS_TYPE_TEXT;
693 case G_RESOLVER_RECORD_SOA:
695 case G_RESOLVER_RECORD_NS:
697 case G_RESOLVER_RECORD_MX:
700 g_return_val_if_reached (-1);
704 g_resolver_records_from_DnsQuery (const gchar *rrname,
714 if (status != ERROR_SUCCESS)
716 if (status == DNS_ERROR_RCODE_NAME_ERROR)
718 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_NOT_FOUND,
719 _("No DNS record of the requested type for '%s'"), rrname);
721 else if (status == DNS_ERROR_RCODE_SERVER_FAILURE)
723 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_TEMPORARY_FAILURE,
724 _("Temporarily unable to resolve '%s'"), rrname);
728 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_INTERNAL,
729 _("Error resolving '%s'"), rrname);
736 for (rec = results; rec; rec = rec->pNext)
738 if (rec->wType != dnstype)
743 record = parse_dns_srv (rec);
746 record = parse_dns_soa (rec);
749 record = parse_dns_ns (rec);
752 record = parse_dns_mx (rec);
755 record = parse_dns_txt (rec);
758 g_warn_if_reached ();
763 records = g_list_prepend (records, g_variant_ref_sink (record));
768 g_set_error (error, G_RESOLVER_ERROR, G_RESOLVER_ERROR_NOT_FOUND,
769 _("No DNS record of the requested type for '%s'"), rrname);
781 GResolverRecordType record_type;
785 free_lookup_records_data (LookupRecordsData *lrd)
787 g_free (lrd->rrname);
788 g_slice_free (LookupRecordsData, lrd);
792 free_records (GList *records)
794 g_list_free_full (records, (GDestroyNotify) g_variant_unref);
797 #if defined(G_OS_UNIX)
800 int res_query(const char *, int, int, u_char *, int);
805 do_lookup_records (GTask *task,
806 gpointer source_object,
808 GCancellable *cancellable)
810 LookupRecordsData *lrd = task_data;
812 GError *error = NULL;
814 #if defined(G_OS_UNIX)
820 rrtype = g_resolver_record_type_to_rrtype (lrd->record_type);
821 answer = g_byte_array_new ();
824 g_byte_array_set_size (answer, len * 2);
825 len = res_query (lrd->rrname, C_IN, rrtype, answer->data, answer->len);
827 /* If answer fit in the buffer then we're done */
828 if (len < 0 || len < (gint)answer->len)
832 * On overflow some res_query's return the length needed, others
833 * return the full length entered. This code works in either case.
838 records = g_resolver_records_from_res_query (lrd->rrname, rrtype, answer->data, len, herr, &error);
839 g_byte_array_free (answer, TRUE);
844 DNS_RECORD *results = NULL;
847 dnstype = g_resolver_record_type_to_dnstype (lrd->record_type);
848 status = DnsQuery_A (lrd->rrname, dnstype, DNS_QUERY_STANDARD, NULL, &results, NULL);
849 records = g_resolver_records_from_DnsQuery (lrd->rrname, dnstype, status, results, &error);
851 DnsRecordListFree (results, DnsFreeRecordList);
856 g_task_return_pointer (task, records, (GDestroyNotify) free_records);
858 g_task_return_error (task, error);
862 lookup_records (GResolver *resolver,
864 GResolverRecordType record_type,
865 GCancellable *cancellable,
870 LookupRecordsData *lrd;
872 task = g_task_new (resolver, cancellable, NULL, NULL);
874 lrd = g_slice_new (LookupRecordsData);
875 lrd->rrname = g_strdup (rrname);
876 lrd->record_type = record_type;
877 g_task_set_task_data (task, lrd, (GDestroyNotify) free_lookup_records_data);
879 g_task_set_return_on_cancel (task, TRUE);
880 g_task_run_in_thread_sync (task, do_lookup_records);
881 records = g_task_propagate_pointer (task, error);
882 g_object_unref (task);
888 lookup_records_async (GResolver *resolver,
890 GResolverRecordType record_type,
891 GCancellable *cancellable,
892 GAsyncReadyCallback callback,
896 LookupRecordsData *lrd;
898 task = g_task_new (resolver, cancellable, callback, user_data);
900 lrd = g_slice_new (LookupRecordsData);
901 lrd->rrname = g_strdup (rrname);
902 lrd->record_type = record_type;
903 g_task_set_task_data (task, lrd, (GDestroyNotify) free_lookup_records_data);
905 g_task_set_return_on_cancel (task, TRUE);
906 g_task_run_in_thread (task, do_lookup_records);
907 g_object_unref (task);
911 lookup_records_finish (GResolver *resolver,
912 GAsyncResult *result,
915 g_return_val_if_fail (g_task_is_valid (result, resolver), NULL);
917 return g_task_propagate_pointer (G_TASK (result), error);
922 g_threaded_resolver_class_init (GThreadedResolverClass *threaded_class)
924 GResolverClass *resolver_class = G_RESOLVER_CLASS (threaded_class);
926 resolver_class->lookup_by_name = lookup_by_name;
927 resolver_class->lookup_by_name_async = lookup_by_name_async;
928 resolver_class->lookup_by_name_finish = lookup_by_name_finish;
929 resolver_class->lookup_by_address = lookup_by_address;
930 resolver_class->lookup_by_address_async = lookup_by_address_async;
931 resolver_class->lookup_by_address_finish = lookup_by_address_finish;
932 resolver_class->lookup_records = lookup_records;
933 resolver_class->lookup_records_async = lookup_records_async;
934 resolver_class->lookup_records_finish = lookup_records_finish;
936 /* Initialize addrinfo_hints */
938 addrinfo_hints.ai_flags |= AI_ADDRCONFIG;
940 /* These two don't actually matter, they just get copied into the
941 * returned addrinfo structures (and then we ignore them). But if
942 * we leave them unset, we'll get back duplicate answers.
944 addrinfo_hints.ai_socktype = SOCK_STREAM;
945 addrinfo_hints.ai_protocol = IPPROTO_TCP;