X-Git-Url: https://git.saurik.com/apple/libresolv.git/blobdiff_plain/e9b5a3fca496bd4c85503d2ea34e927e631d5aaa..70a8bbea946a32c8c17d2ed726496d0814b6cbc2:/dns_plugin.c diff --git a/dns_plugin.c b/dns_plugin.c deleted file mode 100644 index 8b17275..0000000 --- a/dns_plugin.c +++ /dev/null @@ -1,790 +0,0 @@ -/* - * Copyright (c) 2008 Apple, Inc. All rights reserved. - * - * @APPLE_LICENSE_HEADER_START@ - * - * This file contains Original Code and/or Modifications of Original Code - * as defined in and that are subject to the Apple Public Source License - * Version 2.0 (the 'License'). You may not use this file except in - * compliance with the License. Please obtain a copy of the License at - * http://www.opensource.apple.com/apsl/ and read it before using this - * file. - * - * The Original Code and all software distributed under the License are - * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER - * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, - * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, - * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. - * Please see the License for the specific language governing rights and - * limitations under the License. - * - * @APPLE_LICENSE_HEADER_END@ - */ - -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include - -/* from dns_util.c */ -#define DNS_MAX_RECEIVE_SIZE 65536 - -#define MDNS_HANDLE_NAME "*MDNS*" - -#define DNS_HANDLE_BUSY 0x00000001 -#define MDNS_HANDLE_BUSY 0x00000002 - -#define MODULE_DNS 0 -#define MODULE_MDNS 1 - -static pthread_mutex_t dns_plugin_lock = PTHREAD_MUTEX_INITIALIZER; - -typedef struct -{ - struct hostent host; - int alias_count; - int addr_count; - uint64_t ttl; -} dns_build_hostent_t; - -typedef struct -{ - dns_handle_t dns; - dns_handle_t mdns; - uint32_t flags; -} dns_plugin_private_t; - -#define DNS_FLAGS_RCODE_MASK 0x000f -static const char hexchar[] = "0123456789abcdef"; - -static dns_handle_t -dns_checkout_handle(si_mod_t *si) -{ - dns_plugin_private_t *pp; - dns_handle_t dns; - - if (si == NULL) return NULL; - - pthread_mutex_lock(&dns_plugin_lock); - if (si->private == NULL) - { - pp = (dns_plugin_private_t *)calloc(1, sizeof(dns_plugin_private_t)); - si->private = pp; - } - - dns = NULL; - pp = (dns_plugin_private_t *)si->private; - - if (pp == NULL) - { - /* shouldn't happen */ - dns = dns_open(NULL); - pthread_mutex_unlock(&dns_plugin_lock); - return dns; - } - - if ((pp->flags & DNS_HANDLE_BUSY) == 0) - { - if (pp->dns == NULL) pp->dns = dns_open(NULL); - pp->flags |= DNS_HANDLE_BUSY; - pthread_mutex_unlock(&dns_plugin_lock); - return pp->dns; - } - - /* main dns handle is busy - create a temporary one */ - dns = dns_open(NULL); - pthread_mutex_unlock(&dns_plugin_lock); - return dns; -} - -static dns_handle_t -mdns_checkout_handle(si_mod_t *si) -{ - dns_plugin_private_t *pp; - dns_handle_t dns; - - if (si == NULL) return NULL; - - pthread_mutex_lock(&dns_plugin_lock); - if (si->private == NULL) - { - pp = (dns_plugin_private_t *)calloc(1, sizeof(dns_plugin_private_t)); - si->private = pp; - } - - dns = NULL; - pp = (dns_plugin_private_t *)si->private; - - if (pp == NULL) - { - /* shouldn't happen */ - dns = dns_open(MDNS_HANDLE_NAME); - pthread_mutex_unlock(&dns_plugin_lock); - return dns; - } - - if ((pp->flags & MDNS_HANDLE_BUSY) == 0) - { - if (pp->mdns == NULL) pp->mdns = dns_open(MDNS_HANDLE_NAME); - pp->flags |= MDNS_HANDLE_BUSY; - pthread_mutex_unlock(&dns_plugin_lock); - return pp->mdns; - } - - /* main mdns handle is busy - create a temporary one */ - dns = dns_open(MDNS_HANDLE_NAME); - pthread_mutex_unlock(&dns_plugin_lock); - return dns; -} - -static void -dns_checkin_handle(si_mod_t *si, dns_handle_t dns) -{ - dns_plugin_private_t *pp; - - if (si == NULL) return; - if (dns == NULL) return; - - pthread_mutex_lock(&dns_plugin_lock); - if (si->private == NULL) - { - /* shouldn't happen */ - pp = (dns_plugin_private_t *)calloc(1, sizeof(dns_plugin_private_t)); - si->private = pp; - } - - pp = (dns_plugin_private_t *)si->private; - - if (pp == NULL) - { - /* shouldn't happen */ - dns_free(dns); - pthread_mutex_unlock(&dns_plugin_lock); - return; - } - - if (pp->dns == dns) - { - pp->flags &= ~DNS_HANDLE_BUSY; - pthread_mutex_unlock(&dns_plugin_lock); - return; - } - else if (pp->mdns == dns) - { - pp->flags &= ~MDNS_HANDLE_BUSY; - pthread_mutex_unlock(&dns_plugin_lock); - return; - } - - dns_free(dns); - pthread_mutex_unlock(&dns_plugin_lock); -} - -static char * -dns_reverse_ipv4(const char *addr) -{ - union - { - uint32_t a; - unsigned char b[4]; - } ab; - char *p; - - if (addr == NULL) return NULL; - - memcpy(&(ab.a), addr, 4); - - asprintf(&p, "%u.%u.%u.%u.in-addr.arpa.", ab.b[3], ab.b[2], ab.b[1], ab.b[0]); - return p; -} - -static char * -dns_reverse_ipv6(const char *addr) -{ - char x[65], *p; - int i, j; - u_int8_t d, hi, lo; - - if (addr == NULL) return NULL; - - x[64] = '\0'; - j = 63; - for (i = 0; i < 16; i++) - { - d = addr[i]; - lo = d & 0x0f; - hi = d >> 4; - x[j--] = '.'; - x[j--] = hexchar[hi]; - x[j--] = '.'; - x[j--] = hexchar[lo]; - } - - p = calloc(1, 75); - if (p == NULL) return NULL; - - memmove(p, x, 64); - strcat(p, "ip6.arpa."); - - return p; -} - -static char * -dns_lower_case(const char *s) -{ - int i; - char *t; - - if (s == NULL) return NULL; - t = malloc(strlen(s) + 1); - - for (i = 0; s[i] != '\0'; i++) - { - if ((s[i] >= 'A') && (s[i] <= 'Z')) t[i] = s[i] + 32; - else t[i] = s[i]; - } - t[i] = '\0'; - return t; -} - -static int -dns_host_merge_alias(const char *name, dns_build_hostent_t *h) -{ - int i; - - if (name == NULL) return 0; - if (h == NULL) return 0; - - if ((h->host.h_name != NULL) && (!strcmp(name, h->host.h_name))) return 0; - for (i = 0; i < h->alias_count; i++) if (!strcmp(name, h->host.h_aliases[i])) return 0; - - h->host.h_aliases = (char **)reallocf(h->host.h_aliases, (h->alias_count + 2) * sizeof(char *)); - if (h->host.h_aliases == NULL) - { - h->alias_count = 0; - return -1; - } - - h->host.h_aliases[h->alias_count] = dns_lower_case(name); - h->alias_count++; - h->host.h_aliases[h->alias_count] = NULL; - - return 0; -} - -static int -dns_host_append_addr(const char *addr, uint32_t len, dns_build_hostent_t *h) -{ - if (addr == NULL) return 0; - if (h == NULL) return 0; - - if (h->addr_count == 0) h->host.h_addr_list = (char **)calloc(2, sizeof(char *)); - else h->host.h_addr_list = (char **)reallocf(h->host.h_addr_list, (h->addr_count + 2) * sizeof(char *)); - - if (h->host.h_addr_list == NULL) - { - h->addr_count = 0; - return -1; - } - - h->host.h_addr_list[h->addr_count] = malloc(len); - if (h->host.h_addr_list[h->addr_count] == NULL) return -1; - - memcpy(h->host.h_addr_list[h->addr_count], addr, len); - h->addr_count++; - h->host.h_addr_list[h->addr_count] = NULL; - - return 0; -} - -static void -dns_plugin_clear_host(dns_build_hostent_t *h) -{ - uint32_t i; - char **aliases; - - if (h == NULL) return; - - if (h->host.h_name != NULL) free(h->host.h_name); - h->host.h_name = NULL; - - aliases = h->host.h_aliases; - if (aliases != NULL) - { - while (*aliases != NULL) free(*aliases++); - free(h->host.h_aliases); - } - - h->host.h_aliases = NULL; - - if (h->host.h_addr_list != NULL) - { - for (i = 0; h->host.h_addr_list[i] != NULL; i++) free(h->host.h_addr_list[i]); - free(h->host.h_addr_list); - } - - h->host.h_addr_list = NULL; -} - -static int -dns_reply_to_hostent(dns_reply_t *r, int af, const char *addr, dns_build_hostent_t *h) -{ - int i, got_data, got_addr; - int ptrx, cnamex; - - if (r == NULL) return -1; - if (r->status != DNS_STATUS_OK) return -1; - if ((r->header->flags & DNS_FLAGS_RCODE_MASK) != ns_r_noerror) return -1; - - if (r == NULL) return -1; - if (r->header == NULL) return -1; - if (r->header->ancount == 0) return -1; - - got_data = 0; - got_addr = 0; - ptrx = -1; - cnamex = -1; - - for (i = 0; i < r->header->ancount; i++) - { - if ((af == AF_INET) && (r->answer[i]->dnstype == ns_t_a)) - { - got_data++; - got_addr++; - dns_host_append_addr((const char *)&(r->answer[i]->data.A->addr), 4, h); - if (h->ttl == 0) h->ttl = r->answer[i]->ttl; - else if (r->answer[i]->ttl < h->ttl) h->ttl = r->answer[i]->ttl; - } - - else if ((af == AF_INET6) && (r->answer[i]->dnstype == ns_t_aaaa)) - { - got_data++; - got_addr++; - dns_host_append_addr((const char *)&(r->answer[i]->data.AAAA->addr), 16, h); - if (h->ttl == 0) h->ttl = r->answer[i]->ttl; - else if (r->answer[i]->ttl < h->ttl) h->ttl = r->answer[i]->ttl; - } - - else if (r->answer[i]->dnstype == ns_t_cname) - { - got_data++; - if (cnamex == -1) cnamex = i; - if (h->ttl == 0) h->ttl = r->answer[i]->ttl; - else if (r->answer[i]->ttl < h->ttl) h->ttl = r->answer[i]->ttl; - } - - else if (r->answer[i]->dnstype == ns_t_ptr) - { - got_data++; - if (ptrx == -1) ptrx = i; - if (h->ttl == 0) h->ttl = r->answer[i]->ttl; - else if (r->answer[i]->ttl < h->ttl) h->ttl = r->answer[i]->ttl; - } - } - - if (addr != NULL) - { - if (af == AF_INET) - { - got_addr++; - dns_host_append_addr(addr, 4, h); - } - else if (af == AF_INET6) - { - got_addr++; - dns_host_append_addr(addr, 16, h); - } - } - - if (got_data == 0) return -1; - if (got_addr == 0) return -1; - - h->host.h_addrtype = af; - if (af == AF_INET) h->host.h_length = 4; - else h->host.h_length = 16; - - if (ptrx != -1) - { - /* use name from PTR record */ - h->host.h_name = dns_lower_case(r->answer[ptrx]->data.PTR->name); - } - else if (cnamex != -1) - { - /* use name from CNAME record */ - h->host.h_name = dns_lower_case(r->answer[cnamex]->data.CNAME->name); - } - else - { - /* use name in first answer */ - h->host.h_name = dns_lower_case(r->answer[0]->name); - } - - for (i = 0; i < r->header->ancount; i++) - { - if (r->answer[i]->dnstype == ns_t_cname) - { - dns_host_merge_alias(r->answer[cnamex]->data.CNAME->name, h); - dns_host_merge_alias(r->answer[cnamex]->name, h); - } - } - - if (h->alias_count == 0) h->host.h_aliases = (char **)calloc(1, sizeof(char *)); - - return 0; -} - -static si_item_t * -_internal_host_byname(si_mod_t *si, const char *name, int af, const char *ignored, uint32_t *err, int which) -{ - uint32_t type; - dns_reply_t *r; - dns_build_hostent_t h; - si_item_t *out; - dns_handle_t dns; - uint64_t bb; - int status; - - if (err != NULL) *err = SI_STATUS_NO_ERROR; - - if (name == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - if (af == AF_INET) type = ns_t_a; - else if (af == AF_INET6) type = ns_t_aaaa; - else - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - r = NULL; - dns = NULL; - - if (which == MODULE_DNS) dns = dns_checkout_handle(si); - else if (which == MODULE_MDNS) dns = mdns_checkout_handle(si); - - if (dns == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - r = dns_lookup(dns, name, ns_c_in, type); - dns_checkin_handle(si, dns); - - if (r == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND; - return NULL; - } - - memset(&h, 0, sizeof(dns_build_hostent_t)); - - status = dns_reply_to_hostent(r, af, NULL, &h); - dns_free_reply(r); - - if (status < 0) - { - dns_plugin_clear_host(&h); - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - bb = h.ttl + time(NULL); - - if (af == AF_INET) - { - out = (si_item_t *)LI_ils_create("L4488s*44a", (unsigned long)si, CATEGORY_HOST_IPV4, 1, bb, 0LL, h.host.h_name, h.host.h_aliases, af, h.host.h_length, h.host.h_addr_list); - } - else - { - out = (si_item_t *)LI_ils_create("L4488s*44c", (unsigned long)si, CATEGORY_HOST_IPV6, 1, bb, 0LL, h.host.h_name, h.host.h_aliases, af, h.host.h_length, h.host.h_addr_list); - } - - dns_plugin_clear_host(&h); - - if ((out == NULL) && (err != NULL)) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - - return out; -} - -si_item_t * -dns_host_byname(si_mod_t *si, const char *name, int af, const char *ignored, uint32_t *err) -{ - return _internal_host_byname(si, name, af, NULL, err, MODULE_DNS); -} - -si_item_t * -mdns_host_byname(si_mod_t *si, const char *name, int af, const char *ignored, uint32_t *err) -{ - return _internal_host_byname(si, name, af, NULL, err, MODULE_MDNS); -} - -static si_item_t * -_internal_host_byaddr(si_mod_t *si, const void *addr, int af, const char *ignored, uint32_t *err, int which) -{ - uint32_t type; - dns_reply_t *r; - dns_build_hostent_t h; - char *name; - si_item_t *out; - dns_handle_t dns; - socklen_t len; - uint64_t bb; - int cat; - struct sockaddr_storage from; - uint32_t fromlen; - - if (err != NULL) *err = SI_STATUS_NO_ERROR; - - if (addr == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - if (si == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - fromlen = sizeof(struct sockaddr_storage); - memset(&from, 0, fromlen); - - name = NULL; - type = ns_t_ptr; - - len = 0; - cat = -1; - - if (af == AF_INET) - { - len = 4; - name = dns_reverse_ipv4(addr); - cat = CATEGORY_HOST_IPV4; - } - else if (af == AF_INET6) - { - len = 16; - name = dns_reverse_ipv6(addr); - cat = CATEGORY_HOST_IPV6; - } - else - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - r = NULL; - dns = NULL; - - if (which == MODULE_DNS) dns = dns_checkout_handle(si); - else if (which == MODULE_MDNS) dns = mdns_checkout_handle(si); - - if (dns == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - r = dns_lookup(dns, name, ns_c_in, type); - dns_checkin_handle(si, dns); - - free(name); - if (r == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND; - return NULL; - } - - memset(&h, 0, sizeof(dns_build_hostent_t)); - - if (dns_reply_to_hostent(r, af, addr, &h) < 0) - { - dns_plugin_clear_host(&h); - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - bb = h.ttl + time(NULL); - out = (si_item_t *)LI_ils_create("L4488s*44a", (unsigned long)si, cat, 1, bb, 0LL, h.host.h_name, h.host.h_aliases, af, h.host.h_length, h.host.h_addr_list); - - dns_plugin_clear_host(&h); - - if ((out == NULL) && (err != NULL)) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - - return out; -} - -si_item_t * -dns_host_byaddr(si_mod_t *si, const void *addr, int af, const char *ignored, uint32_t *err) -{ - return _internal_host_byaddr(si, addr, af, NULL, err, MODULE_DNS); -} - -si_item_t * -mdns_host_byaddr(si_mod_t *si, const void *addr, int af, const char *ignored, uint32_t *err) -{ - return _internal_host_byaddr(si, addr, af, NULL, err, MODULE_MDNS); -} - -/* - * We support dns_async_start / cancel / handle_reply using dns_item_call - */ -static si_item_t * -_internal_item_call(si_mod_t *si, int call, const char *name, const char *ignored1, const char *ignored2, uint32_t class, uint32_t type, uint32_t *err, int which) -{ - dns_handle_t dns; - char buf[DNS_MAX_RECEIVE_SIZE]; - int len; - struct sockaddr_storage from; - uint32_t fromlen; - si_item_t *out; - - if (err != NULL) *err = SI_STATUS_NO_ERROR; - if ((call != SI_CALL_DNS_QUERY) && (call != SI_CALL_DNS_SEARCH)) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - if (name == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - dns = NULL; - - if (which == MODULE_DNS) dns = dns_checkout_handle(si); - else if (which == MODULE_MDNS) dns = mdns_checkout_handle(si); - - if (dns == NULL) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - return NULL; - } - - fromlen = sizeof(struct sockaddr_storage); - memset(&from, 0, fromlen); - - len = 0; - if (call == SI_CALL_DNS_QUERY) len = dns_query(dns, name, class, type, buf, sizeof(buf), (struct sockaddr *)&from, &fromlen); - else if (call == SI_CALL_DNS_SEARCH) len = dns_search(dns, name, class, type, buf, sizeof(buf), (struct sockaddr *)&from, &fromlen); - - dns_checkin_handle(si, dns); - - if ((len <= 0) || (len > DNS_MAX_RECEIVE_SIZE)) - { - if (err != NULL) *err = SI_STATUS_H_ERRNO_HOST_NOT_FOUND; - return NULL; - } - - out = (si_item_t *)LI_ils_create("L4488@@", (unsigned long)si, CATEGORY_DNSPACKET, 1, 0LL, 0LL, len, buf, fromlen, &from); - if ((out == NULL) && (err != NULL)) *err = SI_STATUS_H_ERRNO_NO_RECOVERY; - - return out; -} - -si_item_t * -dns_item_call(si_mod_t *si, int call, const char *name, const char *ignored1, const char *ignored2, uint32_t class, uint32_t type, uint32_t *err) -{ - return _internal_item_call(si, call, name, ignored1, ignored2, class, type, err, MODULE_DNS); -} - -si_item_t * -mdns_item_call(si_mod_t *si, int call, const char *name, const char *ignored1, const char *ignored2, uint32_t class, uint32_t type, uint32_t *err) -{ - return _internal_item_call(si, call, name, ignored1, ignored2, class, type, err, MODULE_MDNS); -} - -int -dns_is_valid(si_mod_t *si, si_item_t *item) -{ - uint64_t now; - si_mod_t *src; - - if (si == NULL) return 0; - if (item == NULL) return 0; - if (si->name == NULL) return 0; - if (item->src == NULL) return 0; - - src = (si_mod_t *)item->src; - - if (src->name == NULL) return 0; - if (string_not_equal(si->name, src->name)) return 0; - - now = time(NULL); - if (item->validation_a < now) return 0; - return 1; -} - -int -mdns_is_valid(si_mod_t *si, si_item_t *item) -{ - return 0; -} - -void -dns_close(si_mod_t *si) -{ - dns_plugin_private_t *pp; - - if (si == NULL) return; - - pp = (dns_plugin_private_t *)si->private; - if (pp == NULL) return; - - if (pp->dns != NULL) dns_free(pp->dns); - free(si->private); -} - -int -dns_init(si_mod_t *si) -{ - if (si == NULL) return 1; - - si->vers = 1; - - si->sim_close = dns_close; - si->sim_is_valid = dns_is_valid; - si->sim_host_byname = dns_host_byname; - si->sim_host_byaddr = dns_host_byaddr; - si->sim_item_call = dns_item_call; - - return 0; -} - -void -mdns_close(si_mod_t *si) -{ - dns_close(si); -} - -int -mdns_init(si_mod_t *si) -{ - if (si == NULL) return 1; - - si->vers = 1; - - si->sim_close = dns_close; - si->sim_is_valid = mdns_is_valid; - si->sim_host_byname = mdns_host_byname; - si->sim_host_byaddr = mdns_host_byaddr; - si->sim_item_call = mdns_item_call; - - return 0; -}