/** * * \file context.c * @brief getdns context management functions * * Declarations taken from the getdns API description pseudo implementation. * */ /* * Copyright (c) 2013, NLnet Labs, Verisign, Inc. * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * Neither the names of the copyright holders nor the * names of its contributors may be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL Verisign, Inc. BE LIABLE FOR ANY * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include #include #include #include #include #include #include #include #include #include #include "config.h" #include "context.h" #include "types-internal.h" #include "util-internal.h" #include "dnssec.h" #include "stub.h" void *plain_mem_funcs_user_arg = MF_PLAIN; typedef struct host_name_addrs { getdns_rbnode_t node; ldns_rdf *host_name; ldns_rr_list *ipv4addrs; ldns_rr_list *ipv6addrs; } host_name_addrs; /* Private functions */ getdns_return_t create_default_namespaces(struct getdns_context *context); static void create_local_hosts(struct getdns_context *context); getdns_return_t destroy_local_hosts(struct getdns_context *context); static struct getdns_list *create_default_root_servers(void); static getdns_return_t set_os_defaults(struct getdns_context *); static int transaction_id_cmp(const void *, const void *); static int local_host_cmp(const void *, const void *); static void dispatch_updated(struct getdns_context *, uint16_t); static void cancel_dns_req(getdns_dns_req *); static void cancel_outstanding_requests(struct getdns_context*, int); /* unbound helpers */ static getdns_return_t rebuild_ub_ctx(struct getdns_context* context); static void set_ub_string_opt(struct getdns_context *, char *, char *); static void set_ub_number_opt(struct getdns_context *, char *, uint16_t); static getdns_return_t set_ub_dns_transport(struct getdns_context*, getdns_transport_t); static void set_ub_limit_outstanding_queries(struct getdns_context*, uint16_t); static void set_ub_dnssec_allowed_skew(struct getdns_context*, uint32_t); static void set_ub_edns_maximum_udp_payload_size(struct getdns_context*, int); /* Stuff to make it compile pedantically */ #define RETURN_IF_NULL(ptr, code) if(ptr == NULL) return code; static void destroy_local_host(getdns_rbnode_t * node, void *arg) { getdns_context *context = (getdns_context *)arg; host_name_addrs *hnas = (host_name_addrs *)node; ldns_rdf_deep_free(hnas->host_name); ldns_rr_list_deep_free(hnas->ipv4addrs); ldns_rr_list_deep_free(hnas->ipv6addrs); GETDNS_FREE(context->my_mf, hnas); } /** * Helper to get default lookup namespaces. * TODO: Determine from OS */ getdns_return_t create_default_namespaces(struct getdns_context *context) { context->namespaces = GETDNS_XMALLOC(context->my_mf, getdns_namespace_t, 2); if(context->namespaces == NULL) return GETDNS_RETURN_GENERIC_ERROR; context->namespaces[0] = GETDNS_NAMESPACE_LOCALNAMES; context->namespaces[1] = GETDNS_NAMESPACE_DNS; context->namespace_count = 2; return GETDNS_RETURN_GOOD; } /** * Helper to get contents from hosts file */ static void create_local_hosts(struct getdns_context *context) { ldns_rr_list *host_names = ldns_get_rr_list_hosts_frm_file(NULL); size_t i; ldns_rr *rr; host_name_addrs *hnas; if (host_names == NULL) return; /* We have a 1:1 list of name -> ip address where there is an underlying many to many relationship. Need to create a lookup of (unique name + A/AAAA)-> list of IPV4/IPv6 ip addresses*/ for (i = 0; i < ldns_rr_list_rr_count(host_names); i++) { rr = ldns_rr_list_rr(host_names, i); if (ldns_rr_get_type(rr) != LDNS_RR_TYPE_A && ldns_rr_get_type(rr) != LDNS_RR_TYPE_AAAA) continue; /*Check to see if we already have an entry*/ hnas = (host_name_addrs *)getdns_rbtree_search( &context->local_hosts, ldns_rr_owner(rr)); if (!hnas) { if (!(hnas = GETDNS_MALLOC( context->my_mf, host_name_addrs))) break; if (!(hnas->host_name = ldns_rdf_clone(ldns_rr_owner(rr))) || !(hnas->ipv4addrs = ldns_rr_list_new()) || !(hnas->ipv6addrs = ldns_rr_list_new())) { ldns_rdf_free(hnas->host_name); ldns_rr_list_free(hnas->ipv4addrs); ldns_rr_list_free(hnas->ipv6addrs); GETDNS_FREE(context->my_mf, hnas); break; } hnas->node.key = hnas->host_name; (void) getdns_rbtree_insert( &context->local_hosts, &hnas->node); } if (!(rr = ldns_rr_clone(rr))) break; if (ldns_rr_get_type(rr) == LDNS_RR_TYPE_A) { if (!ldns_rr_list_push_rr(hnas->ipv4addrs, rr)) ldns_rr_free(rr); } else if (!ldns_rr_list_push_rr(hnas->ipv6addrs, rr)) ldns_rr_free(rr); } ldns_rr_list_deep_free(host_names); } /** * Helper to get the default root servers. * TODO: Implement */ static struct getdns_list * create_default_root_servers() { return NULL; } /** * check a file for changes since the last check * and refresh the current data if changes are detected * @param context pointer to a previously created context to be used for this call * @param fchg file to check * @returns changes as OR'd list of GETDNS_FCHG_* values * @returns GETDNS_FCHG_NONE if no changes * @returns GETDNS_FCHG_ERRORS if problems (see fchg->errors for details) */ int filechg_check(struct getdns_context *context, struct filechg *fchg) { struct stat *finfo; if(fchg == NULL) return 0; fchg->errors = GETDNS_FCHG_NOERROR; fchg->changes = GETDNS_FCHG_NOCHANGES; finfo = GETDNS_MALLOC(context->my_mf, struct stat); if(finfo == NULL) { fchg->errors = errno; return GETDNS_FCHG_ERRORS; } if(stat(fchg->fn, finfo) != 0) { GETDNS_FREE(context->my_mf, finfo); fchg->errors = errno; return GETDNS_FCHG_ERRORS; } /* we want to consider a file that previously returned error for stat() as a change */ if(fchg->prevstat == NULL) fchg->changes = GETDNS_FCHG_MTIME | GETDNS_FCHG_CTIME; else { if(fchg->prevstat->st_mtime != finfo->st_mtime) fchg->changes |= GETDNS_FCHG_MTIME; if(fchg->prevstat->st_ctime != finfo->st_ctime) fchg->changes |= GETDNS_FCHG_CTIME; GETDNS_FREE(context->my_mf, fchg->prevstat); } fchg->prevstat = finfo; return fchg->changes; } /* filechg */ static getdns_upstreams * upstreams_create(getdns_context *context, size_t size) { getdns_upstreams *r = (void *) GETDNS_XMALLOC(context->mf, char, sizeof(getdns_upstreams) + sizeof(getdns_upstream) * size); r->mf = context->mf; r->referenced = 1; r->count = 0; r->current = 0; return r; } static getdns_upstreams * upstreams_resize(getdns_upstreams *upstreams, size_t size) { getdns_upstreams *r = (void *) GETDNS_XREALLOC( upstreams->mf, upstreams, char, sizeof(getdns_upstreams) + sizeof(getdns_upstream) * size); return r; } static void upstreams_dereference(getdns_upstreams *upstreams) { if (upstreams && --upstreams->referenced == 0) GETDNS_FREE(upstreams->mf, upstreams); } static size_t upstream_addr_len(getdns_upstream *upstream) { return upstream->addr.ss_family == AF_INET ? 4 : 16; } static uint8_t* upstream_addr(getdns_upstream *upstream) { return upstream->addr.ss_family == AF_INET ? (void *)&((struct sockaddr_in*)&upstream->addr)->sin_addr : (void *)&((struct sockaddr_in6*)&upstream->addr)->sin6_addr; } static in_port_t upstream_port(getdns_upstream *upstream) { return ntohs(upstream->addr.ss_family == AF_INET ? ((struct sockaddr_in *)&upstream->addr)->sin_port : ((struct sockaddr_in6*)&upstream->addr)->sin6_port); } static uint32_t * upstream_scope_id(getdns_upstream *upstream) { return upstream->addr.ss_family == AF_INET ? NULL : (upstream_addr(upstream)[0] == 0xFE && (upstream_addr(upstream)[1] & 0xC0) == 0x80 ? &((struct sockaddr_in6*)&upstream->addr)->sin6_scope_id : NULL); } static void upstream_ntop_buf(getdns_upstream *upstream, char *buf, size_t len) { /* Also possible but prints scope_id by name (nor parsed by unbound) * * getnameinfo((struct sockaddr *)&upstream->addr, upstream->addr_len, * buf, len, NULL, 0, NI_NUMERICHOST) */ (void) inet_ntop(upstream->addr.ss_family, upstream_addr(upstream), buf, len); if (upstream_scope_id(upstream)) (void) snprintf(buf + strlen(buf), len - strlen(buf), "%%%d", (int)*upstream_scope_id(upstream)); if (upstream_port(upstream) != 53 && upstream_port(upstream) != 0) (void) snprintf(buf + strlen(buf), len - strlen(buf), "@%d", (int)upstream_port(upstream)); } static getdns_dict * upstream_dict(getdns_context *context, getdns_upstream *upstream) { getdns_dict *r = getdns_dict_create_with_context(context); char addrstr[1024], *b; getdns_bindata bindata; getdns_dict_util_set_string(r, "address_type", upstream->addr.ss_family == AF_INET ? "IPv4" : "IPv6"); bindata.size = upstream_addr_len(upstream); bindata.data = upstream_addr(upstream); getdns_dict_set_bindata(r, "address_data", &bindata); if (upstream_port(upstream) != 53) getdns_dict_set_int(r, "port", upstream_port(upstream)); (void) getnameinfo((struct sockaddr *)&upstream->addr, upstream->addr_len, addrstr, 1024, NULL, 0, NI_NUMERICHOST); if ((b = strchr(addrstr, '%'))) getdns_dict_util_set_string(r, "scope_id", b+1); return r; } static int net_req_query_id_cmp(const void *id1, const void *id2) { return (intptr_t)id1 - (intptr_t)id2; } static void upstream_init(getdns_upstream *upstream, getdns_upstreams *parent, struct addrinfo *ai) { upstream->upstreams = parent; upstream->addr_len = ai->ai_addrlen; (void) memcpy(&upstream->addr, ai->ai_addr, ai->ai_addrlen); /* How is this upstream doing? */ upstream->to_retry = 2; upstream->back_off = 1; /* For sharing a socket to this upstream with TCP */ upstream->fd = -1; upstream->loop = NULL; (void) getdns_eventloop_event_init( &upstream->event, upstream, NULL, NULL, NULL); (void) memset(&upstream->tcp, 0, sizeof(upstream->tcp)); upstream->write_queue = NULL; upstream->write_queue_last = NULL; /* Tracking of network requests on this socket */ getdns_rbtree_init(&upstream->netreq_by_query_id, net_req_query_id_cmp); } /*---------------------------------------- set_os_defaults we use ldns to read the resolv.conf file - the ldns resolver is destroyed once the file is read */ static getdns_return_t set_os_defaults(struct getdns_context *context) { FILE *in; char line[1024], domain[1024]; char *parse, *token, prev_ch; size_t upstreams_limit = 10, length; struct getdns_bindata bindata; struct addrinfo hints; struct addrinfo *result; getdns_upstream *upstream; int s; if(context->fchg_resolvconf == NULL) { context->fchg_resolvconf = GETDNS_MALLOC(context->my_mf, struct filechg); if(context->fchg_resolvconf == NULL) return GETDNS_RETURN_MEMORY_ERROR; context->fchg_resolvconf->fn = "/etc/resolv.conf"; context->fchg_resolvconf->prevstat = NULL; context->fchg_resolvconf->changes = GETDNS_FCHG_NOCHANGES; context->fchg_resolvconf->errors = GETDNS_FCHG_NOERROR; } filechg_check(context, context->fchg_resolvconf); context->suffix = getdns_list_create_with_context(context); context->upstreams = upstreams_create(context, upstreams_limit); in = fopen(context->fchg_resolvconf->fn, "r"); if (!in) return GETDNS_RETURN_GOOD; memset(&hints, 0, sizeof(struct addrinfo)); hints.ai_family = AF_UNSPEC; /* Allow IPv4 or IPv6 */ hints.ai_socktype = 0; /* Datagram socket */ hints.ai_flags = AI_NUMERICHOST; /* No reverse name lookups */ hints.ai_protocol = 0; /* Any protocol */ hints.ai_canonname = NULL; hints.ai_addr = NULL; hints.ai_next = NULL; *domain = 0; while (fgets(line, (int)sizeof(line), in)) { line[sizeof(line)-1] = 0; /* parse = line + strspn(line, " \t"); */ /* No leading whitespace */ parse = line; if (strncmp(parse, "domain", 6) == 0) { parse += 6; parse += strspn(parse, " \t"); if (*parse == 0 || *parse == '#') continue; token = parse + strcspn(parse, " \t\r\n"); *token = 0; (void) strcpy(domain, parse); } else if (strncmp(parse, "search", 6) == 0) { parse += 6; do { parse += strspn(parse, " \t"); if (*parse == '#' || *parse == '\n') break; token = parse + strcspn(parse, " \t\r\n"); prev_ch = *token; *token = 0; bindata.data = (uint8_t *)parse; bindata.size = strlen(parse) + 1; (void) getdns_list_get_length( context->suffix, &length); (void) getdns_list_set_bindata( context->suffix, length, &bindata); *token = prev_ch; parse = token; } while (*parse); } else if (strncmp(parse, "nameserver", 10) != 0) continue; parse += 10; parse += strspn(parse, " \t"); if (*parse == 0 || *parse == '#') continue; token = parse + strcspn(parse, " \t\r\n"); *token = 0; if ((s = getaddrinfo(parse, "53", &hints, &result))) continue; /* No lookups, so maximal 1 result */ if (! result) continue; /* Grow array when needed */ if (context->upstreams->count == upstreams_limit) context->upstreams = upstreams_resize( context->upstreams, (upstreams_limit *= 2)); upstream = &context->upstreams-> upstreams[context->upstreams->count++]; upstream_init(upstream, context->upstreams, result); freeaddrinfo(result); } fclose(in); (void) getdns_list_get_length(context->suffix, &length); if (length == 0 && *domain != 0) { bindata.data = (uint8_t *)domain; bindata.size = strlen(domain) + 1; (void) getdns_list_set_bindata(context->suffix, 0, &bindata); } return GETDNS_RETURN_GOOD; } /* set_os_defaults */ /* compare of transaction ids in DESCENDING order so that 0 comes last */ static int transaction_id_cmp(const void *id1, const void *id2) { if (id1 == NULL && id2 == NULL) { return 0; } else if (id1 == NULL && id2 != NULL) { return 1; } else if (id1 != NULL && id2 == NULL) { return -1; } else { getdns_transaction_t t1 = *((const getdns_transaction_t *) id1); getdns_transaction_t t2 = *((const getdns_transaction_t *) id2); if (t1 == t2) { return 0; } else if (t1 > t2) { return -1; } else { return 1; } } } static int local_host_cmp(const void *id1, const void *id2) { return (ldns_rdf_compare((const ldns_rdf *)id1, (const ldns_rdf *)id2)); } /* * getdns_context_create * * Call this to initialize the context that is used in other getdns calls. */ getdns_return_t getdns_context_create_with_extended_memory_functions( struct getdns_context ** context, int set_from_os, void *userarg, void *(*malloc)(void *userarg, size_t), void *(*realloc)(void *userarg, void *, size_t), void (*free)(void *userarg, void *) ) { getdns_return_t r; struct getdns_context *result = NULL; mf_union mf; if (!context || !malloc || !realloc || !free) return GETDNS_RETURN_INVALID_PARAMETER; /** default init **/ mf.ext.malloc = malloc; result = userarg == MF_PLAIN ? (*mf.pln.malloc)( sizeof(struct getdns_context)) : (*mf.ext.malloc)(userarg, sizeof(struct getdns_context)); if (!result) return GETDNS_RETURN_MEMORY_ERROR; result->processing = 0; result->destroying = 0; result->my_mf.mf_arg = userarg; result->my_mf.mf.ext.malloc = malloc; result->my_mf.mf.ext.realloc = realloc; result->my_mf.mf.ext.free = free; result->update_callback = NULL; result->mf.mf_arg = userarg; result->mf.mf.ext.malloc = malloc; result->mf.mf.ext.realloc = realloc; result->mf.mf.ext.free = free; result->resolution_type_set = 0; getdns_rbtree_init(&result->outbound_requests, transaction_id_cmp); getdns_rbtree_init(&result->local_hosts, local_host_cmp); result->resolution_type = GETDNS_RESOLUTION_RECURSING; if ((r = create_default_namespaces(result))) goto error; result->timeout = 5000; result->follow_redirects = GETDNS_REDIRECTS_FOLLOW; result->dns_root_servers = create_default_root_servers(); result->append_name = GETDNS_APPEND_NAME_ALWAYS; result->suffix = NULL; result->dnssec_trust_anchors = NULL; result->upstreams = NULL; result->edns_extended_rcode = 0; result->edns_version = 0; result->edns_do_bit = 0; result->extension = &result->mini_event.loop; if ((r = getdns_mini_event_init(result, &result->mini_event))) goto error; result->fchg_resolvconf = NULL; result->fchg_hosts = NULL; if (set_from_os && (r = set_os_defaults(result))) goto error; result->dnssec_allowed_skew = 0; result->edns_maximum_udp_payload_size = -1; result->dns_transport = GETDNS_TRANSPORT_UDP_FIRST_AND_FALL_BACK_TO_TCP; result->limit_outstanding_queries = 0; result->has_ta = priv_getdns_parse_ta_file(NULL, NULL); result->return_dnssec_status = GETDNS_EXTENSION_FALSE; /* unbound context is initialized here */ result->unbound_ctx = NULL; if ((r = rebuild_ub_ctx(result))) goto error; create_local_hosts(result); *context = result; return GETDNS_RETURN_GOOD; error: getdns_context_destroy(result); return r; } /* getdns_context_create_with_extended_memory_functions */ /* * getdns_context_create * * Call this to initialize the context that is used in other getdns calls. */ getdns_return_t getdns_context_create_with_memory_functions(struct getdns_context ** context, int set_from_os, void *(*malloc)(size_t), void *(*realloc)(void *, size_t), void (*free)(void *) ) { mf_union mf; mf.pln.malloc = malloc; mf.pln.realloc = realloc; mf.pln.free = free; return getdns_context_create_with_extended_memory_functions( context, set_from_os, MF_PLAIN, mf.ext.malloc, mf.ext.realloc, mf.ext.free); } /* getdns_context_create */ /* * getdns_context_create * * Call this to initialize the context that is used in other getdns calls. */ getdns_return_t getdns_context_create(struct getdns_context ** context, int set_from_os) { return getdns_context_create_with_memory_functions(context, set_from_os, malloc, realloc, free); } /* getdns_context_create */ /* * getdns_context_destroy * * Call this to dispose of resources associated with a context once you * are done with it. */ void getdns_context_destroy(struct getdns_context *context) { if (context == NULL) { return; } // If being destroyed during getdns callback, // fail via assert assert(context->processing == 0); if (context->destroying) { return ; } context->destroying = 1; context->processing = 1; /* cancel all outstanding requests */ cancel_outstanding_requests(context, 1); context->processing = 0; context->extension->vmt->cleanup(context->extension); if (context->namespaces) GETDNS_FREE(context->my_mf, context->namespaces); if(context->fchg_resolvconf) { if(context->fchg_resolvconf->prevstat) GETDNS_FREE(context->my_mf, context->fchg_resolvconf->prevstat); GETDNS_FREE(context->my_mf, context->fchg_resolvconf); } if(context->fchg_hosts) { if(context->fchg_hosts->prevstat) GETDNS_FREE(context->my_mf, context->fchg_hosts->prevstat); GETDNS_FREE(context->my_mf, context->fchg_hosts); } getdns_list_destroy(context->dns_root_servers); getdns_list_destroy(context->suffix); getdns_list_destroy(context->dnssec_trust_anchors); /* destroy the contexts */ if (context->unbound_ctx) ub_ctx_delete(context->unbound_ctx); traverse_postorder(&context->local_hosts, destroy_local_host, context); upstreams_dereference(context->upstreams); GETDNS_FREE(context->my_mf, context); } /* getdns_context_destroy */ /* * getdns_context_set_context_update_callback * */ getdns_return_t getdns_context_set_context_update_callback(struct getdns_context *context, void (*value) (struct getdns_context *context, getdns_context_code_t changed_item)) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); context->update_callback = value; return GETDNS_RETURN_GOOD; } /* getdns_context_set_context_update_callback */ /* * Helpers to set options on the unbound ctx */ static void set_ub_string_opt(struct getdns_context *ctx, char *opt, char *value) { if (ctx->unbound_ctx) ub_ctx_set_option(ctx->unbound_ctx, opt, value); } static void set_ub_number_opt(struct getdns_context *ctx, char *opt, uint16_t value) { char buffer[64]; snprintf(buffer, 64, "%hu", value); set_ub_string_opt(ctx, opt, buffer); } static void getdns_context_request_count_changed(getdns_context *context) { DEBUG_SCHED("getdns_context_request_count_changed(%d)\n", (int) context->outbound_requests.count); if (context->outbound_requests.count) { if (context->ub_event.ev) return; DEBUG_SCHED("gc_request_count_changed " "-> ub schedule(el_ev = %p, el_ev->ev = %p)\n", &context->ub_event, context->ub_event.ev); context->extension->vmt->schedule( context->extension, ub_fd(context->unbound_ctx), TIMEOUT_FOREVER, &context->ub_event); } else if (context->ub_event.ev) /* Only test if count == 0! */ { DEBUG_SCHED("gc_request_count_changed " "-> ub clear(el_ev = %p, el_ev->ev = %p)\n", &context->ub_event, context->ub_event.ev); context->extension->vmt->clear( context->extension, &context->ub_event); } } void priv_getdns_context_ub_read_cb(void *userarg) { getdns_context *context = (getdns_context *)userarg; /* getdns_context_process_async, but without reinvoking an eventloop * (with context->extension->vmt->run*), because we are already * called from a running eventloop. */ context->processing = 1; if (ub_poll(context->unbound_ctx)) (void) ub_process(context->unbound_ctx); context->processing = 0; /* No need to handle timeouts. They are handled by the extension. */ getdns_context_request_count_changed(context); } static getdns_return_t rebuild_ub_ctx(struct getdns_context* context) { if (context->unbound_ctx != NULL) { /* cancel all requests and delete */ cancel_outstanding_requests(context, 1); ub_ctx_delete(context->unbound_ctx); context->unbound_ctx = NULL; } /* setup */ context->unbound_ctx = ub_ctx_create(); if (!context->unbound_ctx) { return GETDNS_RETURN_MEMORY_ERROR; } set_ub_dnssec_allowed_skew(context, context->dnssec_allowed_skew); set_ub_edns_maximum_udp_payload_size(context, context->edns_maximum_udp_payload_size); set_ub_dns_transport(context, context->dns_transport); /* Set default trust anchor */ if (context->has_ta) { (void) ub_ctx_add_ta_file( context->unbound_ctx, TRUST_ANCHOR_FILE); } context->ub_event.userarg = context; context->ub_event.read_cb = priv_getdns_context_ub_read_cb; context->ub_event.write_cb = NULL; context->ub_event.timeout_cb = NULL; context->ub_event.ev = NULL; return GETDNS_RETURN_GOOD; } /** * Helper to dispatch the updated callback */ static void dispatch_updated(struct getdns_context *context, uint16_t item) { if (context->update_callback) { context->update_callback(context, item); } } /* * getdns_context_set_resolution_type * */ getdns_return_t getdns_context_set_resolution_type(struct getdns_context *context, getdns_resolution_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (value != GETDNS_RESOLUTION_STUB && value != GETDNS_RESOLUTION_RECURSING) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } if (context->resolution_type_set != 0) { /* already setup */ return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } context->resolution_type = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_RESOLUTION_TYPE); return GETDNS_RETURN_GOOD; } /* getdns_context_set_resolution_type */ /* * getdns_context_set_namespaces * */ getdns_return_t getdns_context_set_namespaces(struct getdns_context *context, size_t namespace_count, getdns_namespace_t *namespaces) { size_t i; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (namespace_count == 0 || namespaces == NULL) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } for(i=0; imy_mf, context->namespaces); /** duplicate **/ context->namespaces = GETDNS_XMALLOC(context->my_mf, getdns_namespace_t, namespace_count); memcpy(context->namespaces, namespaces, namespace_count * sizeof(getdns_namespace_t)); context->namespace_count = namespace_count; dispatch_updated(context, GETDNS_CONTEXT_CODE_NAMESPACES); return GETDNS_RETURN_GOOD; } /* getdns_context_set_namespaces */ static getdns_return_t set_ub_dns_transport(struct getdns_context* context, getdns_transport_t value) { switch (value) { case GETDNS_TRANSPORT_UDP_FIRST_AND_FALL_BACK_TO_TCP: set_ub_string_opt(context, "do-udp:", "yes"); set_ub_string_opt(context, "do-tcp:", "yes"); break; case GETDNS_TRANSPORT_UDP_ONLY: set_ub_string_opt(context, "do-udp:", "yes"); set_ub_string_opt(context, "do-tcp:", "no"); break; case GETDNS_TRANSPORT_TCP_ONLY: case GETDNS_TRANSPORT_TCP_ONLY_KEEP_CONNECTIONS_OPEN: set_ub_string_opt(context, "do-udp:", "no"); set_ub_string_opt(context, "do-tcp:", "yes"); break; default: /* TODO GETDNS_CONTEXT_TCP_ONLY_KEEP_CONNECTIONS_OPEN */ return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } return GETDNS_RETURN_GOOD; } /* * getdns_context_set_dns_transport * */ getdns_return_t getdns_context_set_dns_transport(struct getdns_context *context, getdns_transport_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (set_ub_dns_transport(context, value) != GETDNS_RETURN_GOOD) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } if (value != context->dns_transport) { context->dns_transport = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_DNS_TRANSPORT); } return GETDNS_RETURN_GOOD; } /* getdns_context_set_dns_transport */ static void set_ub_limit_outstanding_queries(struct getdns_context* context, uint16_t value) { /* num-queries-per-thread */ set_ub_number_opt(context, "num-queries-per-thread:", value); } /* * getdns_context_set_limit_outstanding_queries * */ getdns_return_t getdns_context_set_limit_outstanding_queries(struct getdns_context *context, uint16_t limit) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); set_ub_limit_outstanding_queries(context, limit); if (limit != context->limit_outstanding_queries) { context->limit_outstanding_queries = limit; dispatch_updated(context, GETDNS_CONTEXT_CODE_LIMIT_OUTSTANDING_QUERIES); } return GETDNS_RETURN_GOOD; } /* getdns_context_set_limit_outstanding_queries */ /* * getdns_context_set_timeout * */ getdns_return_t getdns_context_set_timeout(struct getdns_context *context, uint64_t timeout) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (timeout == 0) { return GETDNS_RETURN_INVALID_PARAMETER; } context->timeout = timeout; dispatch_updated(context, GETDNS_CONTEXT_CODE_TIMEOUT); return GETDNS_RETURN_GOOD; } /* getdns_context_set_timeout */ /* * getdns_context_set_follow_redirects * */ getdns_return_t getdns_context_set_follow_redirects(struct getdns_context *context, getdns_redirects_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (value != GETDNS_REDIRECTS_FOLLOW && value != GETDNS_REDIRECTS_DO_NOT_FOLLOW) return GETDNS_RETURN_INVALID_PARAMETER; context->follow_redirects = value; if (context->resolution_type_set != 0) { /* already setup */ return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } dispatch_updated(context, GETDNS_CONTEXT_CODE_FOLLOW_REDIRECTS); return GETDNS_RETURN_GOOD; } /* getdns_context_set_follow_redirects */ /* * getdns_context_set_dns_root_servers * */ getdns_return_t getdns_context_set_dns_root_servers(struct getdns_context *context, struct getdns_list * addresses) { struct getdns_list *copy = NULL; size_t count = 0; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (context->resolution_type_set != 0) { /* already setup */ return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } if (addresses != NULL) { if (getdns_list_copy(addresses, ©) != GETDNS_RETURN_GOOD) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } addresses = copy; getdns_list_get_length(addresses, &count); if (count == 0) { getdns_list_destroy(addresses); addresses = NULL; } else { size_t i = 0; getdns_return_t r = GETDNS_RETURN_GOOD; /* validate and add ip str */ for (i = 0; i < count; ++i) { struct getdns_dict *dict = NULL; getdns_list_get_dict(addresses, i, &dict); if (r != GETDNS_RETURN_GOOD) { break; } } if (r != GETDNS_RETURN_GOOD) { getdns_list_destroy(addresses); return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } } } getdns_list_destroy(context->dns_root_servers); context->dns_root_servers = addresses; dispatch_updated(context, GETDNS_CONTEXT_CODE_DNS_ROOT_SERVERS); return GETDNS_RETURN_GOOD; } /* getdns_context_set_dns_root_servers */ /* * getdns_context_set_append_name * */ getdns_return_t getdns_context_set_append_name(struct getdns_context *context, getdns_append_name_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (value != GETDNS_APPEND_NAME_ALWAYS && value != GETDNS_APPEND_NAME_ONLY_TO_SINGLE_LABEL_AFTER_FAILURE && value != GETDNS_APPEND_NAME_ONLY_TO_MULTIPLE_LABEL_NAME_AFTER_FAILURE && value != GETDNS_APPEND_NAME_NEVER) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } context->append_name = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_APPEND_NAME); return GETDNS_RETURN_GOOD; } /* getdns_context_set_append_name */ /* * getdns_context_set_suffix * */ getdns_return_t getdns_context_set_suffix(struct getdns_context *context, struct getdns_list * value) { struct getdns_list *copy = NULL; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (context->resolution_type_set != 0) { /* already setup */ return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } if (value != NULL) { if (getdns_list_copy(value, ©) != GETDNS_RETURN_GOOD) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } value = copy; } getdns_list_destroy(context->suffix); context->suffix = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_SUFFIX); return GETDNS_RETURN_GOOD; } /* getdns_context_set_suffix */ /* * getdns_context_set_dnssec_trust_anchors * */ getdns_return_t getdns_context_set_dnssec_trust_anchors(struct getdns_context *context, struct getdns_list * value) { struct getdns_list *copy = NULL; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (value != NULL) { if (getdns_list_copy(value, ©) != GETDNS_RETURN_GOOD) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } value = copy; } getdns_list_destroy(context->dnssec_trust_anchors); context->dnssec_trust_anchors = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_DNSSEC_TRUST_ANCHORS); return GETDNS_RETURN_GOOD; } /* getdns_context_set_dnssec_trust_anchors */ static void set_ub_dnssec_allowed_skew(struct getdns_context* context, uint32_t value) { set_ub_number_opt(context, "val-sig-skew-min:", value); set_ub_number_opt(context, "val-sig-skew-max:", value); } /* * getdns_context_set_dnssec_allowed_skew * */ getdns_return_t getdns_context_set_dnssec_allowed_skew(struct getdns_context *context, uint32_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); set_ub_dnssec_allowed_skew(context, value); if (value != context->dnssec_allowed_skew) { context->dnssec_allowed_skew = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_DNSSEC_ALLOWED_SKEW); } return GETDNS_RETURN_GOOD; } /* getdns_context_set_dnssec_allowed_skew */ /* * getdns_context_set_upstream_recursive_servers * */ getdns_return_t getdns_context_set_upstream_recursive_servers(struct getdns_context *context, struct getdns_list *upstream_list) { getdns_return_t r; size_t count = 0; size_t i; getdns_upstreams *upstreams; char addrstr[1024], portstr[1024], *eos; struct addrinfo hints; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); RETURN_IF_NULL(upstream_list, GETDNS_RETURN_INVALID_PARAMETER); r = getdns_list_get_length(upstream_list, &count); if (count == 0 || r != GETDNS_RETURN_GOOD) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } memset(&hints, 0, sizeof(struct addrinfo)); hints.ai_family = AF_UNSPEC; /* Allow IPv4 or IPv6 */ hints.ai_socktype = 0; /* Datagram socket */ hints.ai_flags = AI_NUMERICHOST; /* No reverse name lookups */ hints.ai_protocol = 0; /* Any protocol */ hints.ai_canonname = NULL; hints.ai_addr = NULL; hints.ai_next = NULL; upstreams = upstreams_create(context, count); for (i = 0; i < count; i++) { getdns_dict *dict; getdns_bindata *address_type; getdns_bindata *address_data; uint32_t port; getdns_bindata *scope_id; struct addrinfo *ai; getdns_upstream *upstream; upstream = &upstreams->upstreams[upstreams->count]; if ((r = getdns_list_get_dict(upstream_list, i, &dict))) goto error; if ((r = getdns_dict_get_bindata( dict, "address_type",&address_type))) goto error; if (address_type->size < 4) goto invalid_parameter; if (strncmp((char *)address_type->data, "IPv4", 4) == 0) upstream->addr.ss_family = AF_INET; else if (strncmp((char *)address_type->data, "IPv6", 4) == 0) upstream->addr.ss_family = AF_INET6; else goto invalid_parameter; if ((r = getdns_dict_get_bindata( dict, "address_data", &address_data))) goto error; if ((upstream->addr.ss_family == AF_INET && address_data->size != 4) || (upstream->addr.ss_family == AF_INET6 && address_data->size != 16)) goto invalid_parameter; if (inet_ntop(upstream->addr.ss_family, address_data->data, addrstr, 1024) == NULL) goto invalid_parameter; port = 53; (void) getdns_dict_get_int(dict, "port", &port); (void) snprintf(portstr, 1024, "%d", (int)port); if (getdns_dict_get_bindata(dict, "scope_id", &scope_id) == GETDNS_RETURN_GOOD) { if (strlen(addrstr) + scope_id->size > 1022) goto invalid_parameter; eos = &addrstr[strlen(addrstr)]; *eos++ = '%'; (void) memcpy(eos, scope_id->data, scope_id->size); eos[scope_id->size] = 0; } if (getaddrinfo(addrstr, portstr, &hints, &ai)) goto invalid_parameter; upstream_init(upstream, upstreams, ai); upstreams->count++; freeaddrinfo(ai); } upstreams_dereference(context->upstreams); context->upstreams = upstreams; dispatch_updated(context, GETDNS_CONTEXT_CODE_UPSTREAM_RECURSIVE_SERVERS); return GETDNS_RETURN_GOOD; invalid_parameter: r = GETDNS_RETURN_INVALID_PARAMETER; error: upstreams_dereference(upstreams); return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } /* getdns_context_set_upstream_recursive_servers */ static void set_ub_edns_maximum_udp_payload_size(struct getdns_context* context, int value) { /* edns-buffer-size */ if (value >= 512 && value <= 65535) set_ub_number_opt(context, "edns-buffer-size:", (uint16_t)value); } /* * getdns_context_set_edns_maximum_udp_payload_size * */ getdns_return_t getdns_context_set_edns_maximum_udp_payload_size(struct getdns_context *context, uint16_t value) { if (!context) return GETDNS_RETURN_INVALID_PARAMETER; /* check for < 512. uint16_t won't let it go above max) */ if (value < 512) value = 512; set_ub_edns_maximum_udp_payload_size(context, value); if (value != context->edns_maximum_udp_payload_size) { context->edns_maximum_udp_payload_size = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_EDNS_MAXIMUM_UDP_PAYLOAD_SIZE); } return GETDNS_RETURN_GOOD; } /* getdns_context_set_edns_maximum_udp_payload_size */ /* * getdns_context_set_edns_extended_rcode * */ getdns_return_t getdns_context_set_edns_extended_rcode(struct getdns_context *context, uint8_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); context->edns_extended_rcode = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_EDNS_EXTENDED_RCODE); return GETDNS_RETURN_GOOD; } /* getdns_context_set_edns_extended_rcode */ /* * getdns_context_set_edns_version * */ getdns_return_t getdns_context_set_edns_version(struct getdns_context *context, uint8_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); context->edns_version = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_EDNS_VERSION); return GETDNS_RETURN_GOOD; } /* getdns_context_set_edns_version */ /* * getdns_context_set_edns_do_bit * */ getdns_return_t getdns_context_set_edns_do_bit(struct getdns_context *context, uint8_t value) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); /* only allow 1 */ if (value != 0 && value != 1) { return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } context->edns_do_bit = value; dispatch_updated(context, GETDNS_CONTEXT_CODE_EDNS_DO_BIT); return GETDNS_RETURN_GOOD; } /* getdns_context_set_edns_do_bit */ /* * getdns_context_set_extended_memory_functions * */ getdns_return_t getdns_context_set_extended_memory_functions( struct getdns_context *context, void *userarg, void *(*malloc) (void *userarg, size_t), void *(*realloc) (void *userarg, void *, size_t), void (*free) (void *userarg, void *) ) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (!malloc || !realloc || !free) return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; context->mf.mf_arg = userarg; context->mf.mf.ext.malloc = malloc; context->mf.mf.ext.realloc = realloc; context->mf.mf.ext.free = free; dispatch_updated(context, GETDNS_CONTEXT_CODE_MEMORY_FUNCTIONS); return GETDNS_RETURN_GOOD; } /* getdns_context_set_extended_memory_functions*/ /* * getdns_context_set_memory_functions * */ getdns_return_t getdns_context_set_memory_functions(struct getdns_context *context, void *(*malloc) (size_t), void *(*realloc) (void *, size_t), void (*free) (void *) ) { mf_union mf; mf.pln.malloc = malloc; mf.pln.realloc = realloc; mf.pln.free = free; return getdns_context_set_extended_memory_functions( context, MF_PLAIN, mf.ext.malloc, mf.ext.realloc, mf.ext.free); } /* getdns_context_set_memory_functions*/ /* cancel the request */ static void cancel_dns_req(getdns_dns_req *req) { getdns_network_req *netreq; for (netreq = req->first_req; netreq; netreq = netreq->next) if (netreq->unbound_id != -1) { ub_cancel(req->context->unbound_ctx, netreq->unbound_id); netreq->unbound_id = -1; } else priv_getdns_cancel_stub_request(netreq); req->canceled = 1; } getdns_return_t getdns_context_cancel_request(getdns_context *context, getdns_transaction_t transaction_id, int fire_callback) { getdns_dns_req *dnsreq; if (!context) return GETDNS_RETURN_INVALID_PARAMETER; /* delete the node from the tree */ if (!(dnsreq = (getdns_dns_req *)getdns_rbtree_delete( &context->outbound_requests, &transaction_id))) return GETDNS_RETURN_UNKNOWN_TRANSACTION; /* do the cancel */ cancel_dns_req(dnsreq); if (fire_callback) dnsreq->user_callback(context, GETDNS_CALLBACK_CANCEL, NULL, dnsreq->user_pointer, transaction_id); /* clean up */ dns_req_free(dnsreq); return GETDNS_RETURN_GOOD; } /* * getdns_cancel_callback * */ getdns_return_t getdns_cancel_callback(getdns_context *context, getdns_transaction_t transaction_id) { if (!context) return GETDNS_RETURN_INVALID_PARAMETER; context->processing = 1; getdns_return_t r = getdns_context_cancel_request(context, transaction_id, 1); context->processing = 0; getdns_context_request_count_changed(context); return r; } /* getdns_cancel_callback */ static getdns_return_t ub_setup_stub(struct ub_ctx *ctx, getdns_upstreams *upstreams) { getdns_return_t r = GETDNS_RETURN_GOOD; size_t i; getdns_upstream *upstream; char addr[1024]; (void) ub_ctx_set_fwd(ctx, NULL); for (i = 0; i < upstreams->count; i++) { upstream = &upstreams->upstreams[i]; upstream_ntop_buf(upstream, addr, 1024); ub_ctx_set_fwd(ctx, addr); } /* Allow lookups of: */ /* - localhost */ (void)ub_ctx_zone_remove(ctx, "localhost."); /* - reverse IPv4 loopback */ (void)ub_ctx_zone_remove(ctx, "127.in-addr.arpa."); /* - reverse IPv6 loopback */ (void)ub_ctx_zone_remove(ctx, "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0." "0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa."); /* - reverse RFC1918 local use zones */ (void)ub_ctx_zone_remove(ctx, "10.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "16.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "17.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "18.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "19.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "20.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "21.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "22.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "23.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "24.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "25.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "26.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "27.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "28.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "29.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "30.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "31.172.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "168.192.in-addr.arpa."); /* - reverse RFC3330 IP4 this, link-local, testnet and broadcast */ (void)ub_ctx_zone_remove(ctx, "0.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "254.169.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "2.0.192.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "100.51.198.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "113.0.203.in-addr.arpa."); (void)ub_ctx_zone_remove(ctx, "255.255.255.255.in-addr.arpa."); /* - reverse RFC4291 IP6 unspecified */ (void)ub_ctx_zone_remove(ctx, "0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0." "0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa."); /* - reverse RFC4193 IPv6 Locally Assigned Local Addresses */ (void)ub_ctx_zone_remove(ctx, "D.F.ip6.arpa."); /* - reverse RFC4291 IPv6 Link Local Addresses */ (void)ub_ctx_zone_remove(ctx, "8.E.F.ip6.arpa."); (void)ub_ctx_zone_remove(ctx, "9.E.F.ip6.arpa."); (void)ub_ctx_zone_remove(ctx, "A.E.F.ip6.arpa."); (void)ub_ctx_zone_remove(ctx, "B.E.F.ip6.arpa."); /* - reverse IPv6 Example Prefix */ (void)ub_ctx_zone_remove(ctx, "8.B.D.0.1.0.0.2.ip6.arpa."); return r; } static getdns_return_t priv_getdns_ns_dns_setup(struct getdns_context *context) { assert(context); switch (context->resolution_type) { case GETDNS_RESOLUTION_STUB: /* Since we don't know if the resolution will be sync or async at this * point and we only support ldns in sync mode then we must set _both_ * contexts up */ /* We get away with just setting up ldns here here because sync mode * always hits this method because at the moment all sync calls use DNS * namespace */ if (!context->upstreams || !context->upstreams->count) return GETDNS_RETURN_GENERIC_ERROR; return ub_setup_stub(context->unbound_ctx, context->upstreams); case GETDNS_RESOLUTION_RECURSING: /* TODO: use the root servers via root hints file */ (void) ub_ctx_set_fwd(context->unbound_ctx, NULL); return GETDNS_RETURN_GOOD; } return GETDNS_RETURN_BAD_CONTEXT; } getdns_return_t getdns_context_prepare_for_resolution(struct getdns_context *context, int usenamespaces) { int i; getdns_return_t r; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (context->destroying) { return GETDNS_RETURN_BAD_CONTEXT; } if (context->resolution_type_set == context->resolution_type) /* already set and no config changes * have caused this to be bad. */ return GETDNS_RETURN_GOOD; /* TODO: respect namespace order (unbound always uses local first if cfg * the spec calls for us to treat the namespace list as ordered * so we need to respect that order */ if (! usenamespaces) { r = priv_getdns_ns_dns_setup(context); if (r == GETDNS_RETURN_GOOD) context->resolution_type_set = context->resolution_type; return r; } r = GETDNS_RETURN_GOOD; for (i = 0; i < context->namespace_count; i++) { switch (context->namespaces[i]) { case GETDNS_NAMESPACE_LOCALNAMES: /* TODO: Note to self! This must change once we have * proper namespace hanlding or asynch stub mode using ldns.*/ (void) ub_ctx_hosts(context->unbound_ctx, NULL); break; case GETDNS_NAMESPACE_DNS: r = priv_getdns_ns_dns_setup(context); break; default: r = GETDNS_RETURN_BAD_CONTEXT; break; } if (r != GETDNS_RETURN_GOOD) return r; /* try again later (resolution_type_set) */ } context->resolution_type_set = context->resolution_type; return r; } /* getdns_context_prepare_for_resolution */ getdns_return_t getdns_context_track_outbound_request(getdns_dns_req *dnsreq) { if (!dnsreq) return GETDNS_RETURN_INVALID_PARAMETER; dnsreq->node.key = &(dnsreq->trans_id); if (!getdns_rbtree_insert( &dnsreq->context->outbound_requests, &dnsreq->node)) return GETDNS_RETURN_GENERIC_ERROR; getdns_context_request_count_changed(dnsreq->context); return GETDNS_RETURN_GOOD; } getdns_return_t getdns_context_clear_outbound_request(getdns_dns_req *dnsreq) { if (!dnsreq) return GETDNS_RETURN_INVALID_PARAMETER; if (!getdns_rbtree_delete( &dnsreq->context->outbound_requests, &dnsreq->trans_id)) return GETDNS_RETURN_GENERIC_ERROR; getdns_context_request_count_changed(dnsreq->context); return GETDNS_RETURN_GOOD; } getdns_return_t getdns_context_request_timed_out(struct getdns_dns_req *req) { /* Don't use req after callback */ getdns_context* context = req->context; getdns_transaction_t trans_id = req->trans_id; getdns_callback_t cb = req->user_callback; void *user_arg = req->user_pointer; getdns_dict *response = create_getdns_response(req); /* cancel the req - also clears it from outbound and cleans up*/ getdns_context_cancel_request(context, trans_id, 0); context->processing = 1; cb(context, GETDNS_CALLBACK_TIMEOUT, response, user_arg, trans_id); context->processing = 0; getdns_context_request_count_changed(context); return GETDNS_RETURN_GOOD; } char * getdns_strdup(const struct mem_funcs *mfs, const char *s) { size_t sz = strlen(s) + 1; char *r = GETDNS_XMALLOC(*mfs, char, sz); if (r) return memcpy(r, s, sz); else return NULL; } struct getdns_bindata * getdns_bindata_copy(struct mem_funcs *mfs, const struct getdns_bindata *src) { struct getdns_bindata *dst; if (!src) return NULL; dst = GETDNS_MALLOC(*mfs, struct getdns_bindata); if (!dst) return NULL; dst->size = src->size; dst->data = GETDNS_XMALLOC(*mfs, uint8_t, src->size); if (!dst->data) { GETDNS_FREE(*mfs, dst); return NULL; } (void) memcpy(dst->data, src->data, src->size); return dst; } void getdns_bindata_destroy(struct mem_funcs *mfs, struct getdns_bindata *bindata) { if (!bindata) return; GETDNS_FREE(*mfs, bindata->data); GETDNS_FREE(*mfs, bindata); } /* TODO: Remove next_timeout argument from getdns_context_get_num_pending_requests */ void getdns_handle_timeouts(struct getdns_event_base* base, struct timeval* now, struct timeval* wait); uint32_t getdns_context_get_num_pending_requests(struct getdns_context* context, struct timeval* next_timeout) { struct timeval dispose; RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (context->outbound_requests.count) context->extension->vmt->run_once(context->extension, 0); /* TODO: Remove this when next_timeout is gone */ getdns_handle_timeouts(context->mini_event.base, &context->mini_event.time_tv, next_timeout ? next_timeout : &dispose); return context->outbound_requests.count; } /* process async reqs */ getdns_return_t getdns_context_process_async(struct getdns_context* context) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); context->processing = 1; if (ub_poll(context->unbound_ctx) && ub_process(context->unbound_ctx)){ /* need an async return code? */ context->processing = 0; return GETDNS_RETURN_GENERIC_ERROR; } context->processing = 0; context->extension->vmt->run_once(context->extension, 0); return GETDNS_RETURN_GOOD; } void getdns_context_run(getdns_context *context) { if (getdns_context_get_num_pending_requests(context, NULL) > 0 && !getdns_context_process_async(context)) context->extension->vmt->run(context->extension); } typedef struct timeout_accumulator { getdns_transaction_t* ids; int idx; } timeout_accumulator; static void accumulate_outstanding_transactions(getdns_rbnode_t* node, void* arg) { timeout_accumulator* acc = (timeout_accumulator*) arg; acc->ids[acc->idx] = *((getdns_transaction_t*) node->key); acc->idx++; } static void cancel_outstanding_requests(struct getdns_context* context, int fire_callback) { if (context->outbound_requests.count > 0) { timeout_accumulator acc; int i; acc.idx = 0; acc.ids = GETDNS_XMALLOC(context->my_mf, getdns_transaction_t, context->outbound_requests.count); traverse_postorder(&context->outbound_requests, accumulate_outstanding_transactions, &acc); for (i = 0; i < acc.idx; ++i) { getdns_context_cancel_request(context, acc.ids[i], fire_callback); } GETDNS_FREE(context->my_mf, acc.ids); } } getdns_return_t getdns_context_detach_eventloop(struct getdns_context* context) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); /* When called from within a callback, do not execute pending * context destroys. * The (other) callback handler will handle it. * * ( because callbacks occur in cancel_outstanding_requests, * and they may destroy the context ) */ context->processing = 1; /* cancel all outstanding requests */ cancel_outstanding_requests(context, 1); context->processing = 0; context->extension->vmt->cleanup(context->extension); context->extension = &context->mini_event.loop; return getdns_mini_event_init(context, &context->mini_event); } getdns_return_t getdns_context_set_eventloop(struct getdns_context* context, getdns_eventloop* loop) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); RETURN_IF_NULL(loop , GETDNS_RETURN_INVALID_PARAMETER); getdns_return_t r = getdns_context_detach_eventloop(context); if (r != GETDNS_RETURN_GOOD) return r; context->extension = loop; return GETDNS_RETURN_GOOD; } static getdns_dict* priv_get_context_settings(getdns_context* context) { getdns_return_t r = GETDNS_RETURN_GOOD; getdns_dict* result = getdns_dict_create_with_context(context); if (!result) { return NULL; } /* int fields */ r = getdns_dict_set_int(result, "dns_transport", context->dns_transport); r |= getdns_dict_set_int(result, "timeout", context->timeout); r |= getdns_dict_set_int(result, "limit_outstanding_queries", context->limit_outstanding_queries); r |= getdns_dict_set_int(result, "dnssec_allowed_skew", context->dnssec_allowed_skew); r |= getdns_dict_set_int(result, "follow_redirects", context->follow_redirects); if (context->edns_maximum_udp_payload_size != -1) r |= getdns_dict_set_int(result, "edns_maximum_udp_payload_size", context->edns_maximum_udp_payload_size); r |= getdns_dict_set_int(result, "edns_extended_rcode", context->edns_extended_rcode); r |= getdns_dict_set_int(result, "edns_version", context->edns_version); r |= getdns_dict_set_int(result, "edns_do_bit", context->edns_do_bit); r |= getdns_dict_set_int(result, "append_name", context->append_name); /* list fields */ if (context->suffix) r |= getdns_dict_set_list(result, "suffix", context->suffix); if (context->upstreams && context->upstreams->count > 0) { size_t i; getdns_upstream *upstream; getdns_list *upstreams = getdns_list_create_with_context(context); for (i = 0; i < context->upstreams->count; i++) { getdns_dict *d; upstream = &context->upstreams->upstreams[i]; d = upstream_dict(context, upstream); r |= getdns_list_set_dict(upstreams, i, d); getdns_dict_destroy(d); } r |= getdns_dict_set_list(result, "upstream_recursive_servers", upstreams); getdns_list_destroy(upstreams); } if (context->namespace_count > 0) { /* create a namespace list */ size_t i; getdns_list* namespaces = getdns_list_create_with_context(context); if (namespaces) { for (i = 0; i < context->namespace_count; ++i) { r |= getdns_list_set_int(namespaces, i, context->namespaces[i]); } r |= getdns_dict_set_list(result, "namespaces", namespaces); } } if (r != GETDNS_RETURN_GOOD) { getdns_dict_destroy(result); result = NULL; } return result; } getdns_dict* getdns_context_get_api_information(getdns_context* context) { getdns_return_t r = GETDNS_RETURN_GOOD; getdns_dict* result = getdns_dict_create_with_context(context); getdns_dict* settings; if (!result) { return NULL; } r = getdns_dict_util_set_string(result, "version_string", PACKAGE_VERSION); r |= getdns_dict_util_set_string(result, "implementation_string", PACKAGE_URL); r |= getdns_dict_set_int(result, "resolution_type", context->resolution_type); settings = priv_get_context_settings(context); r |= getdns_dict_set_dict(result, "all_context", settings); getdns_dict_destroy(settings); if (r != GETDNS_RETURN_GOOD) { getdns_dict_destroy(result); result = NULL; } return result; } getdns_return_t getdns_context_set_return_dnssec_status(getdns_context* context, int enabled) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (enabled != GETDNS_EXTENSION_TRUE && enabled != GETDNS_EXTENSION_FALSE) { return GETDNS_RETURN_INVALID_PARAMETER; } context->return_dnssec_status = enabled; return GETDNS_RETURN_GOOD; } getdns_return_t getdns_context_set_use_threads(getdns_context* context, int use_threads) { RETURN_IF_NULL(context, GETDNS_RETURN_INVALID_PARAMETER); if (context->resolution_type_set != 0) { /* already setup */ return GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } int r = 0; if (use_threads) r = ub_ctx_async(context->unbound_ctx, 1); else r = ub_ctx_async(context->unbound_ctx, 0); return r == 0 ? GETDNS_RETURN_GOOD : GETDNS_RETURN_CONTEXT_UPDATE_FAIL; } getdns_return_t getdns_context_local_namespace_resolve( getdns_dns_req *dnsreq, getdns_dict **response) { getdns_context *context = dnsreq->context; ldns_rr_list *result_list = NULL; host_name_addrs *hnas; ldns_rdf *query_name; int ipv4 = dnsreq->first_req->request_type == GETDNS_RRTYPE_A || (dnsreq->first_req->next && dnsreq->first_req->next->request_type == GETDNS_RRTYPE_A); int ipv6 = dnsreq->first_req->request_type == GETDNS_RRTYPE_AAAA || (dnsreq->first_req->next && dnsreq->first_req->next->request_type == GETDNS_RRTYPE_AAAA); if (!ipv4 && !ipv6) return GETDNS_RETURN_GENERIC_ERROR; /*Do the lookup*/ if (!(query_name = ldns_rdf_new_frm_str( LDNS_RDF_TYPE_DNAME, dnsreq->name))) return GETDNS_RETURN_MEMORY_ERROR; if ((hnas = (host_name_addrs *)getdns_rbtree_search( &context->local_hosts, query_name))) { result_list = ldns_rr_list_new(); if (ipv4) (void) ldns_rr_list_cat(result_list, hnas->ipv4addrs); if (ipv6) (void) ldns_rr_list_cat(result_list, hnas->ipv6addrs); } ldns_rdf_deep_free(query_name); if (!result_list) return GETDNS_RETURN_GENERIC_ERROR; *response = create_getdns_response_from_rr_list(dnsreq, result_list); /* Not deep_free because ldns_rr_list_cat doesn't clone the rr's */ ldns_rr_list_free(result_list); return *response ? GETDNS_RETURN_GOOD : GETDNS_RETURN_GENERIC_ERROR; } struct mem_funcs * priv_getdns_context_mf(getdns_context *context) { return &context->mf; } /* context.c */