contrib/unbound/daemon/worker.h

   1 /*
   2  * daemon/worker.h - worker that handles a pending list of requests.
   3  *
   4  * Copyright (c) 2007, NLnet Labs. All rights reserved.
   5  *
   6  * This software is open source.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * Redistributions of source code must retain the above copyright notice,
  13  * this list of conditions and the following disclaimer.
  14  *
  15  * Redistributions in binary form must reproduce the above copyright notice,
  16  * this list of conditions and the following disclaimer in the documentation
  17  * and/or other materials provided with the distribution.
  18  *
  19  * Neither the name of the NLNET LABS nor the names of its contributors may
  20  * be used to endorse or promote products derived from this software without
  21  * specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  24  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  25  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  26  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  27  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  28  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
  29  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  30  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  31  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  32  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  33  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  34  */
  35
  36 /**
  37  * \file
  38  *
  39  * This file describes the worker structure that holds a list of
  40  * pending requests and handles them.
  41  */
  42
  43 #ifndef DAEMON_WORKER_H
  44 #define DAEMON_WORKER_H
  45
  46 #include "libunbound/worker.h"
  47 #include "util/netevent.h"
  48 #include "util/locks.h"
  49 #include "util/alloc.h"
  50 #include "util/data/msgreply.h"
  51 #include "util/data/msgparse.h"
  52 #include "daemon/stats.h"
  53 #include "util/module.h"
  54 #include "dnstap/dnstap.h"
  55 struct listen_dnsport;
  56 struct outside_network;
  57 struct config_file;
  58 struct daemon;
  59 struct listen_port;
  60 struct ub_randstate;
  61 struct regional;
  62 struct tube;
  63 struct daemon_remote;
  64 struct query_info;
  65
  66 /** worker commands */
  67 enum worker_commands {
  68         /** make the worker quit */
  69         worker_cmd_quit,
  70         /** obtain statistics */
  71         worker_cmd_stats,
  72         /** obtain statistics without statsclear */
  73         worker_cmd_stats_noreset,
  74         /** execute remote control command */
  75         worker_cmd_remote
  76 };
  77
  78 /**
  79  * Structure holding working information for unbound.
  80  * Holds globally visible information.
  81  */
  82 struct worker {
  83         /** the thread number (in daemon array). First in struct for debug. */
  84         int thread_num;
  85         /** global shared daemon structure */
  86         struct daemon* daemon;
  87         /** thread id */
  88         ub_thread_type thr_id;
  89         /** pipe, for commands for this worker */
  90         struct tube* cmd;
  91         /** the event base this worker works with */
  92         struct comm_base* base;
  93         /** the frontside listening interface where request events come in */
  94         struct listen_dnsport* front;
  95         /** the backside outside network interface to the auth servers */
  96         struct outside_network* back;
  97         /** ports to be used by this worker. */
  98         int* ports;
  99         /** number of ports for this worker */
 100         int numports;
 101         /** the signal handler */
 102         struct comm_signal* comsig;
 103         /** commpoint to listen to commands. */
 104         struct comm_point* cmd_com;
 105         /** timer for statistics */
 106         struct comm_timer* stat_timer;
 107         /** ratelimit for errors, time value */
 108         time_t err_limit_time;
 109         /** ratelimit for errors, packet count */
 110         unsigned int err_limit_count;
 111
 112         /** random() table for this worker. */
 113         struct ub_randstate* rndstate;
 114         /** do we need to restart or quit (on signal) */
 115         int need_to_exit;
 116         /** allocation cache for this thread */
 117         struct alloc_cache alloc;
 118         /** per thread statistics */
 119         struct ub_server_stats stats;
 120         /** thread scratch regional */
 121         struct regional* scratchpad;
 122
 123         /** module environment passed to modules, changed for this thread */
 124         struct module_env env;
 125
 126 #ifdef USE_DNSTAP
 127         /** dnstap environment, changed for this thread */
 128         struct dt_env dtenv;
 129 #endif
 130 };
 131
 132 /**
 133  * Create the worker structure. Bare bones version, zeroed struct,
 134  * with backpointers only. Use worker_init on it later.
 135  * @param daemon: the daemon that this worker thread is part of.
 136  * @param id: the thread number from 0.. numthreads-1.
 137  * @param ports: the ports it is allowed to use, array.
 138  * @param n: the number of ports.
 139  * @return: the new worker or NULL on alloc failure.
 140  */
 141 struct worker* worker_create(struct daemon* daemon, int id, int* ports, int n);
 142
 143 /**
 144  * Initialize worker.
 145  * Allocates event base, listens to ports
 146  * @param worker: worker to initialize, created with worker_create.
 147  * @param cfg: configuration settings.
 148  * @param ports: list of shared query ports.
 149  * @param do_sigs: if true, worker installs signal handlers.
 150  * @return: false on error.
 151  */
 152 int worker_init(struct worker* worker, struct config_file *cfg,
 153         struct listen_port* ports, int do_sigs);
 154
 155 /**
 156  * Make worker work.
 157  */
 158 void worker_work(struct worker* worker);
 159
 160 /**
 161  * Delete worker.
 162  */
 163 void worker_delete(struct worker* worker);
 164
 165 /**
 166  * Send a command to a worker. Uses blocking writes.
 167  * @param worker: worker to send command to.
 168  * @param cmd: command to send.
 169  */
 170 void worker_send_cmd(struct worker* worker, enum worker_commands cmd);
 171
 172 /**
 173  * Init worker stats - includes server_stats_init, outside network and mesh.
 174  * @param worker: the worker to init
 175  */
 176 void worker_stats_clear(struct worker* worker);
 177
 178 #endif /* DAEMON_WORKER_H */