aboutsummaryrefslogtreecommitdiff
path: root/external/unbound/libunbound/libworker.c
diff options
context:
space:
mode:
authoranonimal <anonimal@i2pmail.org>2017-06-28 21:07:24 +0000
committeranonimal <anonimal@i2pmail.org>2018-03-18 15:52:19 +0000
commit84c5a9ba481d7a33cc0fd0ca43867b61d127d907 (patch)
treef05d3d3f107da02005b4a61f0e5074c113a7165c /external/unbound/libunbound/libworker.c
parentMerge pull request #3416 (diff)
downloadmonero-84c5a9ba481d7a33cc0fd0ca43867b61d127d907.tar.xz
Unbound: remove unbound from in-tree source
We'll instead use a git submodule to pull from our unbound repo.
Diffstat (limited to 'external/unbound/libunbound/libworker.c')
-rw-r--r--external/unbound/libunbound/libworker.c1031
1 files changed, 0 insertions, 1031 deletions
diff --git a/external/unbound/libunbound/libworker.c b/external/unbound/libunbound/libworker.c
deleted file mode 100644
index b42ba0bd8..000000000
--- a/external/unbound/libunbound/libworker.c
+++ /dev/null
@@ -1,1031 +0,0 @@
-/*
- * libunbound/worker.c - worker thread or process that resolves
- *
- * Copyright (c) 2007, NLnet Labs. All rights reserved.
- *
- * This software is open source.
- *
- * 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 name of the NLNET LABS 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 THE COPYRIGHT
- * HOLDER OR CONTRIBUTORS 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.
- */
-
-/**
- * \file
- *
- * This file contains the worker process or thread that performs
- * the DNS resolving and validation. The worker is called by a procedure
- * and if in the background continues until exit, if in the foreground
- * returns from the procedure when done.
- */
-#include "config.h"
-#ifdef HAVE_SSL
-#include <openssl/ssl.h>
-#endif
-#include "libunbound/libworker.h"
-#include "libunbound/context.h"
-#include "libunbound/unbound.h"
-#include "libunbound/worker.h"
-#include "libunbound/unbound-event.h"
-#include "services/outside_network.h"
-#include "services/mesh.h"
-#include "services/localzone.h"
-#include "services/cache/rrset.h"
-#include "services/outbound_list.h"
-#include "util/fptr_wlist.h"
-#include "util/module.h"
-#include "util/regional.h"
-#include "util/random.h"
-#include "util/config_file.h"
-#include "util/netevent.h"
-#include "util/storage/lookup3.h"
-#include "util/storage/slabhash.h"
-#include "util/net_help.h"
-#include "util/data/dname.h"
-#include "util/data/msgreply.h"
-#include "util/data/msgencode.h"
-#include "util/tube.h"
-#include "iterator/iter_fwd.h"
-#include "iterator/iter_hints.h"
-#include "sldns/sbuffer.h"
-#include "sldns/str2wire.h"
-
-/** handle new query command for bg worker */
-static void handle_newq(struct libworker* w, uint8_t* buf, uint32_t len);
-
-/** delete libworker env */
-static void
-libworker_delete_env(struct libworker* w)
-{
- if(w->env) {
- outside_network_quit_prepare(w->back);
- mesh_delete(w->env->mesh);
- context_release_alloc(w->ctx, w->env->alloc,
- !w->is_bg || w->is_bg_thread);
- sldns_buffer_free(w->env->scratch_buffer);
- regional_destroy(w->env->scratch);
- forwards_delete(w->env->fwds);
- hints_delete(w->env->hints);
- ub_randfree(w->env->rnd);
- free(w->env);
- }
-#ifdef HAVE_SSL
- SSL_CTX_free(w->sslctx);
-#endif
- outside_network_delete(w->back);
-}
-
-/** delete libworker struct */
-static void
-libworker_delete(struct libworker* w)
-{
- if(!w) return;
- libworker_delete_env(w);
- comm_base_delete(w->base);
- free(w);
-}
-
-void
-libworker_delete_event(struct libworker* w)
-{
- if(!w) return;
- libworker_delete_env(w);
- comm_base_delete_no_base(w->base);
- free(w);
-}
-
-/** setup fresh libworker struct */
-static struct libworker*
-libworker_setup(struct ub_ctx* ctx, int is_bg, struct ub_event_base* eb)
-{
- unsigned int seed;
- struct libworker* w = (struct libworker*)calloc(1, sizeof(*w));
- struct config_file* cfg = ctx->env->cfg;
- int* ports;
- int numports;
- if(!w) return NULL;
- w->is_bg = is_bg;
- w->ctx = ctx;
- w->env = (struct module_env*)malloc(sizeof(*w->env));
- if(!w->env) {
- free(w);
- return NULL;
- }
- *w->env = *ctx->env;
- w->env->alloc = context_obtain_alloc(ctx, !w->is_bg || w->is_bg_thread);
- if(!w->env->alloc) {
- libworker_delete(w);
- return NULL;
- }
- w->thread_num = w->env->alloc->thread_num;
- alloc_set_id_cleanup(w->env->alloc, &libworker_alloc_cleanup, w);
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_lock(&ctx->cfglock);
- }
- w->env->scratch = regional_create_custom(cfg->msg_buffer_size);
- w->env->scratch_buffer = sldns_buffer_new(cfg->msg_buffer_size);
- w->env->fwds = forwards_create();
- if(w->env->fwds && !forwards_apply_cfg(w->env->fwds, cfg)) {
- forwards_delete(w->env->fwds);
- w->env->fwds = NULL;
- }
- w->env->hints = hints_create();
- if(w->env->hints && !hints_apply_cfg(w->env->hints, cfg)) {
- hints_delete(w->env->hints);
- w->env->hints = NULL;
- }
- if(cfg->ssl_upstream) {
- w->sslctx = connect_sslctx_create(NULL, NULL, NULL);
- if(!w->sslctx) {
- /* to make the setup fail after unlock */
- hints_delete(w->env->hints);
- w->env->hints = NULL;
- }
- }
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_unlock(&ctx->cfglock);
- }
- if(!w->env->scratch || !w->env->scratch_buffer || !w->env->fwds ||
- !w->env->hints) {
- libworker_delete(w);
- return NULL;
- }
- w->env->worker = (struct worker*)w;
- w->env->probe_timer = NULL;
- seed = (unsigned int)time(NULL) ^ (unsigned int)getpid() ^
- (((unsigned int)w->thread_num)<<17);
- seed ^= (unsigned int)w->env->alloc->next_id;
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_lock(&ctx->cfglock);
- }
- if(!(w->env->rnd = ub_initstate(seed, ctx->seed_rnd))) {
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_unlock(&ctx->cfglock);
- }
- seed = 0;
- libworker_delete(w);
- return NULL;
- }
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_unlock(&ctx->cfglock);
- }
- if(1) {
- /* primitive lockout for threading: if it overwrites another
- * thread it is like wiping the cache (which is likely empty
- * at the start) */
- /* note we are holding the ctx lock in normal threaded
- * cases so that is solved properly, it is only for many ctx
- * in different threads that this may clash */
- static int done_raninit = 0;
- if(!done_raninit) {
- done_raninit = 1;
- hash_set_raninit((uint32_t)ub_random(w->env->rnd));
- }
- }
- seed = 0;
-
- if(eb)
- w->base = comm_base_create_event(eb);
- else w->base = comm_base_create(0);
- if(!w->base) {
- libworker_delete(w);
- return NULL;
- }
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_lock(&ctx->cfglock);
- }
- numports = cfg_condense_ports(cfg, &ports);
- if(numports == 0) {
- int locked = !w->is_bg || w->is_bg_thread;
- libworker_delete(w);
- if(locked) {
- lock_basic_unlock(&ctx->cfglock);
- }
- return NULL;
- }
- w->back = outside_network_create(w->base, cfg->msg_buffer_size,
- (size_t)cfg->outgoing_num_ports, cfg->out_ifs,
- cfg->num_out_ifs, cfg->do_ip4, cfg->do_ip6,
- cfg->do_tcp?cfg->outgoing_num_tcp:0,
- w->env->infra_cache, w->env->rnd, cfg->use_caps_bits_for_id,
- ports, numports, cfg->unwanted_threshold,
- cfg->outgoing_tcp_mss,
- &libworker_alloc_cleanup, w, cfg->do_udp, w->sslctx,
- cfg->delay_close, NULL);
- if(!w->is_bg || w->is_bg_thread) {
- lock_basic_unlock(&ctx->cfglock);
- }
- free(ports);
- if(!w->back) {
- libworker_delete(w);
- return NULL;
- }
- w->env->mesh = mesh_create(&ctx->mods, w->env);
- if(!w->env->mesh) {
- libworker_delete(w);
- return NULL;
- }
- w->env->send_query = &libworker_send_query;
- w->env->detach_subs = &mesh_detach_subs;
- w->env->attach_sub = &mesh_attach_sub;
- w->env->kill_sub = &mesh_state_delete;
- w->env->detect_cycle = &mesh_detect_cycle;
- comm_base_timept(w->base, &w->env->now, &w->env->now_tv);
- return w;
-}
-
-struct libworker* libworker_create_event(struct ub_ctx* ctx,
- struct ub_event_base* eb)
-{
- return libworker_setup(ctx, 0, eb);
-}
-
-/** handle cancel command for bg worker */
-static void
-handle_cancel(struct libworker* w, uint8_t* buf, uint32_t len)
-{
- struct ctx_query* q;
- if(w->is_bg_thread) {
- lock_basic_lock(&w->ctx->cfglock);
- q = context_deserialize_cancel(w->ctx, buf, len);
- lock_basic_unlock(&w->ctx->cfglock);
- } else {
- q = context_deserialize_cancel(w->ctx, buf, len);
- }
- if(!q) {
- /* probably simply lookup failed, i.e. the message had been
- * processed and answered before the cancel arrived */
- return;
- }
- q->cancelled = 1;
- free(buf);
-}
-
-/** do control command coming into bg server */
-static void
-libworker_do_cmd(struct libworker* w, uint8_t* msg, uint32_t len)
-{
- switch(context_serial_getcmd(msg, len)) {
- default:
- case UB_LIBCMD_ANSWER:
- log_err("unknown command for bg worker %d",
- (int)context_serial_getcmd(msg, len));
- /* and fall through to quit */
- case UB_LIBCMD_QUIT:
- free(msg);
- comm_base_exit(w->base);
- break;
- case UB_LIBCMD_NEWQUERY:
- handle_newq(w, msg, len);
- break;
- case UB_LIBCMD_CANCEL:
- handle_cancel(w, msg, len);
- break;
- }
-}
-
-/** handle control command coming into server */
-void
-libworker_handle_control_cmd(struct tube* ATTR_UNUSED(tube),
- uint8_t* msg, size_t len, int err, void* arg)
-{
- struct libworker* w = (struct libworker*)arg;
-
- if(err != 0) {
- free(msg);
- /* it is of no use to go on, exit */
- comm_base_exit(w->base);
- return;
- }
- libworker_do_cmd(w, msg, len); /* also frees the buf */
-}
-
-/** the background thread func */
-static void*
-libworker_dobg(void* arg)
-{
- /* setup */
- uint32_t m;
- struct libworker* w = (struct libworker*)arg;
- struct ub_ctx* ctx;
- if(!w) {
- log_err("libunbound bg worker init failed, nomem");
- return NULL;
- }
- ctx = w->ctx;
- log_thread_set(&w->thread_num);
-#ifdef THREADS_DISABLED
- /* we are forked */
- w->is_bg_thread = 0;
- /* close non-used parts of the pipes */
- tube_close_write(ctx->qq_pipe);
- tube_close_read(ctx->rr_pipe);
-#endif
- if(!tube_setup_bg_listen(ctx->qq_pipe, w->base,
- libworker_handle_control_cmd, w)) {
- log_err("libunbound bg worker init failed, no bglisten");
- return NULL;
- }
- if(!tube_setup_bg_write(ctx->rr_pipe, w->base)) {
- log_err("libunbound bg worker init failed, no bgwrite");
- return NULL;
- }
-
- /* do the work */
- comm_base_dispatch(w->base);
-
- /* cleanup */
- m = UB_LIBCMD_QUIT;
- tube_remove_bg_listen(w->ctx->qq_pipe);
- tube_remove_bg_write(w->ctx->rr_pipe);
- libworker_delete(w);
- (void)tube_write_msg(ctx->rr_pipe, (uint8_t*)&m,
- (uint32_t)sizeof(m), 0);
-#ifdef THREADS_DISABLED
- /* close pipes from forked process before exit */
- tube_close_read(ctx->qq_pipe);
- tube_close_write(ctx->rr_pipe);
-#endif
- return NULL;
-}
-
-int libworker_bg(struct ub_ctx* ctx)
-{
- struct libworker* w;
- /* fork or threadcreate */
- lock_basic_lock(&ctx->cfglock);
- if(ctx->dothread) {
- lock_basic_unlock(&ctx->cfglock);
- w = libworker_setup(ctx, 1, NULL);
- if(!w) return UB_NOMEM;
- w->is_bg_thread = 1;
-#ifdef ENABLE_LOCK_CHECKS
- w->thread_num = 1; /* for nicer DEBUG checklocks */
-#endif
- ub_thread_create(&ctx->bg_tid, libworker_dobg, w);
- } else {
- lock_basic_unlock(&ctx->cfglock);
-#ifndef HAVE_FORK
- /* no fork on windows */
- return UB_FORKFAIL;
-#else /* HAVE_FORK */
- switch((ctx->bg_pid=fork())) {
- case 0:
- w = libworker_setup(ctx, 1, NULL);
- if(!w) fatal_exit("out of memory");
- /* close non-used parts of the pipes */
- tube_close_write(ctx->qq_pipe);
- tube_close_read(ctx->rr_pipe);
- (void)libworker_dobg(w);
- exit(0);
- break;
- case -1:
- return UB_FORKFAIL;
- default:
- /* close non-used parts, so that the worker
- * bgprocess gets 'pipe closed' when the
- * main process exits */
- tube_close_read(ctx->qq_pipe);
- tube_close_write(ctx->rr_pipe);
- break;
- }
-#endif /* HAVE_FORK */
- }
- return UB_NOERROR;
-}
-
-/** get msg reply struct (in temp region) */
-static struct reply_info*
-parse_reply(sldns_buffer* pkt, struct regional* region, struct query_info* qi)
-{
- struct reply_info* rep;
- struct msg_parse* msg;
- if(!(msg = regional_alloc(region, sizeof(*msg)))) {
- return NULL;
- }
- memset(msg, 0, sizeof(*msg));
- sldns_buffer_set_position(pkt, 0);
- if(parse_packet(pkt, msg, region) != 0)
- return 0;
- if(!parse_create_msg(pkt, msg, NULL, qi, &rep, region)) {
- return 0;
- }
- return rep;
-}
-
-/** insert canonname */
-static int
-fill_canon(struct ub_result* res, uint8_t* s)
-{
- char buf[255+2];
- dname_str(s, buf);
- res->canonname = strdup(buf);
- return res->canonname != 0;
-}
-
-/** fill data into result */
-static int
-fill_res(struct ub_result* res, struct ub_packed_rrset_key* answer,
- uint8_t* finalcname, struct query_info* rq, struct reply_info* rep)
-{
- size_t i;
- struct packed_rrset_data* data;
- res->ttl = 0;
- if(!answer) {
- if(finalcname) {
- if(!fill_canon(res, finalcname))
- return 0; /* out of memory */
- }
- if(rep->rrset_count != 0)
- res->ttl = (int)rep->ttl;
- res->data = (char**)calloc(1, sizeof(char*));
- res->len = (int*)calloc(1, sizeof(int));
- return (res->data && res->len);
- }
- data = (struct packed_rrset_data*)answer->entry.data;
- if(query_dname_compare(rq->qname, answer->rk.dname) != 0) {
- if(!fill_canon(res, answer->rk.dname))
- return 0; /* out of memory */
- } else res->canonname = NULL;
- res->data = (char**)calloc(data->count+1, sizeof(char*));
- res->len = (int*)calloc(data->count+1, sizeof(int));
- if(!res->data || !res->len)
- return 0; /* out of memory */
- for(i=0; i<data->count; i++) {
- /* remove rdlength from rdata */
- res->len[i] = (int)(data->rr_len[i] - 2);
- res->data[i] = memdup(data->rr_data[i]+2, (size_t)res->len[i]);
- if(!res->data[i])
- return 0; /* out of memory */
- }
- /* ttl for positive answers, from CNAME and answer RRs */
- if(data->count != 0) {
- size_t j;
- res->ttl = (int)data->ttl;
- for(j=0; j<rep->an_numrrsets; j++) {
- struct packed_rrset_data* d =
- (struct packed_rrset_data*)rep->rrsets[j]->
- entry.data;
- if((int)d->ttl < res->ttl)
- res->ttl = (int)d->ttl;
- }
- }
- /* ttl for negative answers */
- if(data->count == 0 && rep->rrset_count != 0)
- res->ttl = (int)rep->ttl;
- res->data[data->count] = NULL;
- res->len[data->count] = 0;
- return 1;
-}
-
-/** fill result from parsed message, on error fills servfail */
-void
-libworker_enter_result(struct ub_result* res, sldns_buffer* buf,
- struct regional* temp, enum sec_status msg_security)
-{
- struct query_info rq;
- struct reply_info* rep;
- res->rcode = LDNS_RCODE_SERVFAIL;
- rep = parse_reply(buf, temp, &rq);
- if(!rep) {
- log_err("cannot parse buf");
- return; /* error parsing buf, or out of memory */
- }
- if(!fill_res(res, reply_find_answer_rrset(&rq, rep),
- reply_find_final_cname_target(&rq, rep), &rq, rep))
- return; /* out of memory */
- /* rcode, havedata, nxdomain, secure, bogus */
- res->rcode = (int)FLAGS_GET_RCODE(rep->flags);
- if(res->data && res->data[0])
- res->havedata = 1;
- if(res->rcode == LDNS_RCODE_NXDOMAIN)
- res->nxdomain = 1;
- if(msg_security == sec_status_secure)
- res->secure = 1;
- if(msg_security == sec_status_bogus)
- res->bogus = 1;
-}
-
-/** fillup fg results */
-static void
-libworker_fillup_fg(struct ctx_query* q, int rcode, sldns_buffer* buf,
- enum sec_status s, char* why_bogus)
-{
- if(why_bogus)
- q->res->why_bogus = strdup(why_bogus);
- if(rcode != 0) {
- q->res->rcode = rcode;
- q->msg_security = s;
- return;
- }
-
- q->res->rcode = LDNS_RCODE_SERVFAIL;
- q->msg_security = 0;
- q->msg = memdup(sldns_buffer_begin(buf), sldns_buffer_limit(buf));
- q->msg_len = sldns_buffer_limit(buf);
- if(!q->msg) {
- return; /* the error is in the rcode */
- }
-
- /* canonname and results */
- q->msg_security = s;
- libworker_enter_result(q->res, buf, q->w->env->scratch, s);
-}
-
-void
-libworker_fg_done_cb(void* arg, int rcode, sldns_buffer* buf, enum sec_status s,
- char* why_bogus)
-{
- struct ctx_query* q = (struct ctx_query*)arg;
- /* fg query is done; exit comm base */
- comm_base_exit(q->w->base);
-
- libworker_fillup_fg(q, rcode, buf, s, why_bogus);
-}
-
-/** setup qinfo and edns */
-static int
-setup_qinfo_edns(struct libworker* w, struct ctx_query* q,
- struct query_info* qinfo, struct edns_data* edns)
-{
- qinfo->qtype = (uint16_t)q->res->qtype;
- qinfo->qclass = (uint16_t)q->res->qclass;
- qinfo->local_alias = NULL;
- qinfo->qname = sldns_str2wire_dname(q->res->qname, &qinfo->qname_len);
- if(!qinfo->qname) {
- return 0;
- }
- qinfo->local_alias = NULL;
- edns->edns_present = 1;
- edns->ext_rcode = 0;
- edns->edns_version = 0;
- edns->bits = EDNS_DO;
- edns->opt_list = NULL;
- if(sldns_buffer_capacity(w->back->udp_buff) < 65535)
- edns->udp_size = (uint16_t)sldns_buffer_capacity(
- w->back->udp_buff);
- else edns->udp_size = 65535;
- return 1;
-}
-
-int libworker_fg(struct ub_ctx* ctx, struct ctx_query* q)
-{
- struct libworker* w = libworker_setup(ctx, 0, NULL);
- uint16_t qflags, qid;
- struct query_info qinfo;
- struct edns_data edns;
- if(!w)
- return UB_INITFAIL;
- if(!setup_qinfo_edns(w, q, &qinfo, &edns)) {
- libworker_delete(w);
- return UB_SYNTAX;
- }
- qid = 0;
- qflags = BIT_RD;
- q->w = w;
- /* see if there is a fixed answer */
- sldns_buffer_write_u16_at(w->back->udp_buff, 0, qid);
- sldns_buffer_write_u16_at(w->back->udp_buff, 2, qflags);
- if(local_zones_answer(ctx->local_zones, w->env, &qinfo, &edns,
- w->back->udp_buff, w->env->scratch, NULL, NULL, 0, NULL, 0,
- NULL, 0, NULL, 0, NULL)) {
- regional_free_all(w->env->scratch);
- libworker_fillup_fg(q, LDNS_RCODE_NOERROR,
- w->back->udp_buff, sec_status_insecure, NULL);
- libworker_delete(w);
- free(qinfo.qname);
- return UB_NOERROR;
- }
- /* process new query */
- if(!mesh_new_callback(w->env->mesh, &qinfo, qflags, &edns,
- w->back->udp_buff, qid, libworker_fg_done_cb, q)) {
- free(qinfo.qname);
- return UB_NOMEM;
- }
- free(qinfo.qname);
-
- /* wait for reply */
- comm_base_dispatch(w->base);
-
- libworker_delete(w);
- return UB_NOERROR;
-}
-
-void
-libworker_event_done_cb(void* arg, int rcode, sldns_buffer* buf,
- enum sec_status s, char* why_bogus)
-{
- struct ctx_query* q = (struct ctx_query*)arg;
- ub_event_callback_type cb = (ub_event_callback_type)q->cb;
- void* cb_arg = q->cb_arg;
- int cancelled = q->cancelled;
-
- /* delete it now */
- struct ub_ctx* ctx = q->w->ctx;
- lock_basic_lock(&ctx->cfglock);
- (void)rbtree_delete(&ctx->queries, q->node.key);
- ctx->num_async--;
- context_query_delete(q);
- lock_basic_unlock(&ctx->cfglock);
-
- if(!cancelled) {
- /* call callback */
- int sec = 0;
- if(s == sec_status_bogus)
- sec = 1;
- else if(s == sec_status_secure)
- sec = 2;
- (*cb)(cb_arg, rcode, (void*)sldns_buffer_begin(buf),
- (int)sldns_buffer_limit(buf), sec, why_bogus);
- }
-}
-
-int libworker_attach_mesh(struct ub_ctx* ctx, struct ctx_query* q,
- int* async_id)
-{
- struct libworker* w = ctx->event_worker;
- uint16_t qflags, qid;
- struct query_info qinfo;
- struct edns_data edns;
- if(!w)
- return UB_INITFAIL;
- if(!setup_qinfo_edns(w, q, &qinfo, &edns))
- return UB_SYNTAX;
- qid = 0;
- qflags = BIT_RD;
- q->w = w;
- /* see if there is a fixed answer */
- sldns_buffer_write_u16_at(w->back->udp_buff, 0, qid);
- sldns_buffer_write_u16_at(w->back->udp_buff, 2, qflags);
- if(local_zones_answer(ctx->local_zones, w->env, &qinfo, &edns,
- w->back->udp_buff, w->env->scratch, NULL, NULL, 0, NULL, 0,
- NULL, 0, NULL, 0, NULL)) {
- regional_free_all(w->env->scratch);
- free(qinfo.qname);
- libworker_event_done_cb(q, LDNS_RCODE_NOERROR,
- w->back->udp_buff, sec_status_insecure, NULL);
- return UB_NOERROR;
- }
- /* process new query */
- if(async_id)
- *async_id = q->querynum;
- if(!mesh_new_callback(w->env->mesh, &qinfo, qflags, &edns,
- w->back->udp_buff, qid, libworker_event_done_cb, q)) {
- free(qinfo.qname);
- return UB_NOMEM;
- }
- free(qinfo.qname);
- return UB_NOERROR;
-}
-
-/** add result to the bg worker result queue */
-static void
-add_bg_result(struct libworker* w, struct ctx_query* q, sldns_buffer* pkt,
- int err, char* reason)
-{
- uint8_t* msg = NULL;
- uint32_t len = 0;
-
- /* serialize and delete unneeded q */
- if(w->is_bg_thread) {
- lock_basic_lock(&w->ctx->cfglock);
- if(reason)
- q->res->why_bogus = strdup(reason);
- if(pkt) {
- q->msg_len = sldns_buffer_remaining(pkt);
- q->msg = memdup(sldns_buffer_begin(pkt), q->msg_len);
- if(!q->msg)
- msg = context_serialize_answer(q, UB_NOMEM,
- NULL, &len);
- else msg = context_serialize_answer(q, err,
- NULL, &len);
- } else msg = context_serialize_answer(q, err, NULL, &len);
- lock_basic_unlock(&w->ctx->cfglock);
- } else {
- if(reason)
- q->res->why_bogus = strdup(reason);
- msg = context_serialize_answer(q, err, pkt, &len);
- (void)rbtree_delete(&w->ctx->queries, q->node.key);
- w->ctx->num_async--;
- context_query_delete(q);
- }
-
- if(!msg) {
- log_err("out of memory for async answer");
- return;
- }
- if(!tube_queue_item(w->ctx->rr_pipe, msg, len)) {
- log_err("out of memory for async answer");
- return;
- }
-}
-
-void
-libworker_bg_done_cb(void* arg, int rcode, sldns_buffer* buf, enum sec_status s,
- char* why_bogus)
-{
- struct ctx_query* q = (struct ctx_query*)arg;
-
- if(q->cancelled) {
- if(q->w->is_bg_thread) {
- /* delete it now */
- struct ub_ctx* ctx = q->w->ctx;
- lock_basic_lock(&ctx->cfglock);
- (void)rbtree_delete(&ctx->queries, q->node.key);
- ctx->num_async--;
- context_query_delete(q);
- lock_basic_unlock(&ctx->cfglock);
- }
- /* cancelled, do not give answer */
- return;
- }
- q->msg_security = s;
- if(!buf)
- buf = q->w->env->scratch_buffer;
- if(rcode != 0) {
- error_encode(buf, rcode, NULL, 0, BIT_RD, NULL);
- }
- add_bg_result(q->w, q, buf, UB_NOERROR, why_bogus);
-}
-
-
-/** handle new query command for bg worker */
-static void
-handle_newq(struct libworker* w, uint8_t* buf, uint32_t len)
-{
- uint16_t qflags, qid;
- struct query_info qinfo;
- struct edns_data edns;
- struct ctx_query* q;
- if(w->is_bg_thread) {
- lock_basic_lock(&w->ctx->cfglock);
- q = context_lookup_new_query(w->ctx, buf, len);
- lock_basic_unlock(&w->ctx->cfglock);
- } else {
- q = context_deserialize_new_query(w->ctx, buf, len);
- }
- free(buf);
- if(!q) {
- log_err("failed to deserialize newq");
- return;
- }
- if(!setup_qinfo_edns(w, q, &qinfo, &edns)) {
- add_bg_result(w, q, NULL, UB_SYNTAX, NULL);
- return;
- }
- qid = 0;
- qflags = BIT_RD;
- /* see if there is a fixed answer */
- sldns_buffer_write_u16_at(w->back->udp_buff, 0, qid);
- sldns_buffer_write_u16_at(w->back->udp_buff, 2, qflags);
- if(local_zones_answer(w->ctx->local_zones, w->env, &qinfo, &edns,
- w->back->udp_buff, w->env->scratch, NULL, NULL, 0, NULL, 0,
- NULL, 0, NULL, 0, NULL)) {
- regional_free_all(w->env->scratch);
- q->msg_security = sec_status_insecure;
- add_bg_result(w, q, w->back->udp_buff, UB_NOERROR, NULL);
- free(qinfo.qname);
- return;
- }
- q->w = w;
- /* process new query */
- if(!mesh_new_callback(w->env->mesh, &qinfo, qflags, &edns,
- w->back->udp_buff, qid, libworker_bg_done_cb, q)) {
- add_bg_result(w, q, NULL, UB_NOMEM, NULL);
- }
- free(qinfo.qname);
-}
-
-void libworker_alloc_cleanup(void* arg)
-{
- struct libworker* w = (struct libworker*)arg;
- slabhash_clear(&w->env->rrset_cache->table);
- slabhash_clear(w->env->msg_cache);
-}
-
-struct outbound_entry* libworker_send_query(struct query_info* qinfo,
- uint16_t flags, int dnssec, int want_dnssec, int nocaps,
- struct sockaddr_storage* addr, socklen_t addrlen, uint8_t* zone,
- size_t zonelen, int ssl_upstream, struct module_qstate* q)
-{
- struct libworker* w = (struct libworker*)q->env->worker;
- struct outbound_entry* e = (struct outbound_entry*)regional_alloc(
- q->region, sizeof(*e));
- if(!e)
- return NULL;
- e->qstate = q;
- e->qsent = outnet_serviced_query(w->back, qinfo, flags, dnssec,
- want_dnssec, nocaps, q->env->cfg->tcp_upstream, ssl_upstream,
- addr, addrlen, zone, zonelen, q, libworker_handle_service_reply,
- e, w->back->udp_buff, q->env);
- if(!e->qsent) {
- return NULL;
- }
- return e;
-}
-
-int
-libworker_handle_reply(struct comm_point* c, void* arg, int error,
- struct comm_reply* reply_info)
-{
- struct module_qstate* q = (struct module_qstate*)arg;
- struct libworker* lw = (struct libworker*)q->env->worker;
- struct outbound_entry e;
- e.qstate = q;
- e.qsent = NULL;
-
- if(error != 0) {
- mesh_report_reply(lw->env->mesh, &e, reply_info, error);
- return 0;
- }
- /* sanity check. */
- if(!LDNS_QR_WIRE(sldns_buffer_begin(c->buffer))
- || LDNS_OPCODE_WIRE(sldns_buffer_begin(c->buffer)) !=
- LDNS_PACKET_QUERY
- || LDNS_QDCOUNT(sldns_buffer_begin(c->buffer)) > 1) {
- /* error becomes timeout for the module as if this reply
- * never arrived. */
- mesh_report_reply(lw->env->mesh, &e, reply_info,
- NETEVENT_TIMEOUT);
- return 0;
- }
- mesh_report_reply(lw->env->mesh, &e, reply_info, NETEVENT_NOERROR);
- return 0;
-}
-
-int
-libworker_handle_service_reply(struct comm_point* c, void* arg, int error,
- struct comm_reply* reply_info)
-{
- struct outbound_entry* e = (struct outbound_entry*)arg;
- struct libworker* lw = (struct libworker*)e->qstate->env->worker;
-
- if(error != 0) {
- mesh_report_reply(lw->env->mesh, e, reply_info, error);
- return 0;
- }
- /* sanity check. */
- if(!LDNS_QR_WIRE(sldns_buffer_begin(c->buffer))
- || LDNS_OPCODE_WIRE(sldns_buffer_begin(c->buffer)) !=
- LDNS_PACKET_QUERY
- || LDNS_QDCOUNT(sldns_buffer_begin(c->buffer)) > 1) {
- /* error becomes timeout for the module as if this reply
- * never arrived. */
- mesh_report_reply(lw->env->mesh, e, reply_info,
- NETEVENT_TIMEOUT);
- return 0;
- }
- mesh_report_reply(lw->env->mesh, e, reply_info, NETEVENT_NOERROR);
- return 0;
-}
-
-/* --- fake callbacks for fptr_wlist to work --- */
-void worker_handle_control_cmd(struct tube* ATTR_UNUSED(tube),
- uint8_t* ATTR_UNUSED(buffer), size_t ATTR_UNUSED(len),
- int ATTR_UNUSED(error), void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-int worker_handle_request(struct comm_point* ATTR_UNUSED(c),
- void* ATTR_UNUSED(arg), int ATTR_UNUSED(error),
- struct comm_reply* ATTR_UNUSED(repinfo))
-{
- log_assert(0);
- return 0;
-}
-
-int worker_handle_reply(struct comm_point* ATTR_UNUSED(c),
- void* ATTR_UNUSED(arg), int ATTR_UNUSED(error),
- struct comm_reply* ATTR_UNUSED(reply_info))
-{
- log_assert(0);
- return 0;
-}
-
-int worker_handle_service_reply(struct comm_point* ATTR_UNUSED(c),
- void* ATTR_UNUSED(arg), int ATTR_UNUSED(error),
- struct comm_reply* ATTR_UNUSED(reply_info))
-{
- log_assert(0);
- return 0;
-}
-
-int remote_accept_callback(struct comm_point* ATTR_UNUSED(c),
- void* ATTR_UNUSED(arg), int ATTR_UNUSED(error),
- struct comm_reply* ATTR_UNUSED(repinfo))
-{
- log_assert(0);
- return 0;
-}
-
-int remote_control_callback(struct comm_point* ATTR_UNUSED(c),
- void* ATTR_UNUSED(arg), int ATTR_UNUSED(error),
- struct comm_reply* ATTR_UNUSED(repinfo))
-{
- log_assert(0);
- return 0;
-}
-
-void worker_sighandler(int ATTR_UNUSED(sig), void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-struct outbound_entry* worker_send_query(struct query_info* ATTR_UNUSED(qinfo),
- uint16_t ATTR_UNUSED(flags), int ATTR_UNUSED(dnssec),
- int ATTR_UNUSED(want_dnssec), int ATTR_UNUSED(nocaps),
- struct sockaddr_storage* ATTR_UNUSED(addr), socklen_t ATTR_UNUSED(addrlen),
- uint8_t* ATTR_UNUSED(zone), size_t ATTR_UNUSED(zonelen),
- int ATTR_UNUSED(ssl_upstream), struct module_qstate* ATTR_UNUSED(q))
-{
- log_assert(0);
- return 0;
-}
-
-void
-worker_alloc_cleanup(void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-void worker_stat_timer_cb(void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-void worker_probe_timer_cb(void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-void worker_start_accept(void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-void worker_stop_accept(void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-int order_lock_cmp(const void* ATTR_UNUSED(e1), const void* ATTR_UNUSED(e2))
-{
- log_assert(0);
- return 0;
-}
-
-int
-codeline_cmp(const void* ATTR_UNUSED(a), const void* ATTR_UNUSED(b))
-{
- log_assert(0);
- return 0;
-}
-
-int replay_var_compare(const void* ATTR_UNUSED(a), const void* ATTR_UNUSED(b))
-{
- log_assert(0);
- return 0;
-}
-
-void remote_get_opt_ssl(char* ATTR_UNUSED(str), void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-
-#ifdef UB_ON_WINDOWS
-void
-worker_win_stop_cb(int ATTR_UNUSED(fd), short ATTR_UNUSED(ev), void*
- ATTR_UNUSED(arg)) {
- log_assert(0);
-}
-
-void
-wsvc_cron_cb(void* ATTR_UNUSED(arg))
-{
- log_assert(0);
-}
-#endif /* UB_ON_WINDOWS */