dispatch.c revision 4423c99613db1399dbb5c51e86ef0d351a1418c2
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson/*
2f99b54e8ec8e908ea894bc808d1c18a5f51a850Automatic Updater * Copyright (C) 2004 Internet Systems Consortium, Inc. ("ISC")
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * Copyright (C) 1999-2003 Internet Software Consortium.
40f53fa8d9c6a4fc38c0014495e7a42b08f52481David Lawrence *
ec5347e2c775f027573ce5648b910361aa926c01Automatic Updater * Permission to use, copy, modify, and distribute this software for any
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson * purpose with or without fee is hereby granted, provided that the above
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson * copyright notice and this permission notice appear in all copies.
40f53fa8d9c6a4fc38c0014495e7a42b08f52481David Lawrence *
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
dafcb997e390efa4423883dafd100c975c4095d6Mark Andrews * PERFORMANCE OF THIS SOFTWARE.
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson */
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
28ad0be64ee756013c0f6a474fc447ee613ee0d1Evan Hunt/* $Id: dispatch.c,v 1.121 2005/02/23 01:06:37 marka Exp $ */
ab023a65562e62b85a824509d829b6fad87e00b1Rob Austein
ab023a65562e62b85a824509d829b6fad87e00b1Rob Austein#include <config.h>
9c3531d72aeaad6c5f01efe6a1c82023e1379e4dDavid Lawrence
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson#include <stdlib.h>
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson#include <isc/entropy.h>
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews#include <isc/lfsr.h>
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence#include <isc/mem.h>
364a82f7c25b62967678027043425201a5e5171aBob Halley#include <isc/mutex.h>
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson#include <isc/print.h>
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson#include <isc/string.h>
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#include <isc/task.h>
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson#include <isc/util.h>
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#include <dns/acl.h>
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#include <dns/dispatch.h>
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#include <dns/events.h>
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#include <dns/log.h>
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson#include <dns/message.h>
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence#include <dns/portlist.h>
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson#include <dns/tcpmsg.h>
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson#include <dns/types.h>
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrencetypedef ISC_LIST(dns_dispentry_t) dns_displist_t;
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrencetypedef struct dns_qid {
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson unsigned int magic;
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson unsigned int qid_nbuckets; /* hash table size */
40f53fa8d9c6a4fc38c0014495e7a42b08f52481David Lawrence unsigned int qid_increment; /* id increment on collision */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_mutex_t lock;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_lfsr_t qid_lfsr1; /* state generator info */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_lfsr_t qid_lfsr2; /* state generator info */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson dns_displist_t *qid_table; /* the table itself */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson} dns_qid_t;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
18d0b5e54be891a1aa938c165b6d439859121ec8Mark Andrewsstruct dns_dispatchmgr {
18d0b5e54be891a1aa938c165b6d439859121ec8Mark Andrews /* Unlocked. */
18d0b5e54be891a1aa938c165b6d439859121ec8Mark Andrews unsigned int magic;
18d0b5e54be891a1aa938c165b6d439859121ec8Mark Andrews isc_mem_t *mctx;
18d0b5e54be891a1aa938c165b6d439859121ec8Mark Andrews dns_acl_t *blackhole;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_portlist_t *portlist;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt /* Locked by "lock". */
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_mutex_t lock;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt unsigned int state;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt ISC_LIST(dns_dispatch_t) list;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson /* locked by buffer lock */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson dns_qid_t *qid;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_mutex_t buffer_lock;
114c14f8adfc249cf2e5cdcb9007af46fed257e3Mark Andrews unsigned int buffers; /* allocated buffers */
40f53fa8d9c6a4fc38c0014495e7a42b08f52481David Lawrence unsigned int buffersize; /* size of each buffer */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int maxbuffers; /* max buffers */
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence /* Locked internally. */
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence isc_mutex_t pool_lock;
40f53fa8d9c6a4fc38c0014495e7a42b08f52481David Lawrence isc_mempool_t *epool; /* memory pool for events */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_mempool_t *rpool; /* memory pool for replies */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_mempool_t *dpool; /* dispatch allocations */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_mempool_t *bpool; /* memory pool for buffers */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_entropy_t *entropy; /* entropy source */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson};
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson#define MGR_SHUTTINGDOWN 0x00000001U
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson#define MGR_IS_SHUTTINGDOWN(l) (((l)->state & MGR_SHUTTINGDOWN) != 0)
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson#define IS_PRIVATE(d) (((d)->attributes & DNS_DISPATCHATTR_PRIVATE) != 0)
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafssonstruct dns_dispentry {
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int magic;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson dns_dispatch_t *disp;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson dns_messageid_t id;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt unsigned int bucket;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_sockaddr_t host;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_task_t *task;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_taskaction_t action;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt void *arg;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_boolean_t item_out;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson ISC_LIST(dns_dispatchevent_t) items;
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence ISC_LINK(dns_dispentry_t) link;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson};
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#define INVALID_BUCKET (0xffffdead)
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafssonstruct dns_dispatch {
1aba9fe67899522364a9dbc3ee5a14da081f0314Evan Hunt /* Unlocked. */
1aba9fe67899522364a9dbc3ee5a14da081f0314Evan Hunt unsigned int magic; /* magic */
1aba9fe67899522364a9dbc3ee5a14da081f0314Evan Hunt dns_dispatchmgr_t *mgr; /* dispatch manager */
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater isc_task_t *task; /* internal task */
1aba9fe67899522364a9dbc3ee5a14da081f0314Evan Hunt isc_socket_t *socket; /* isc socket attached to */
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater isc_sockaddr_t local; /* local address */
1aba9fe67899522364a9dbc3ee5a14da081f0314Evan Hunt unsigned int maxrequests; /* max requests */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_event_t *ctlevent;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson /* Locked by mgr->lock. */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson ISC_LINK(dns_dispatch_t) link;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt /* Locked by "lock". */
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_mutex_t lock; /* locks all below */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_sockettype_t socktype;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int attributes;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int refcount; /* number of users */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson dns_dispatchevent_t *failsafe_ev; /* failsafe cancel event */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int shutting_down : 1,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt shutdown_out : 1,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt connected : 1,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt tcpmsg_valid : 1,
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson recv_pending : 1; /* is a recv() pending? */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson isc_result_t shutdown_why;
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int requests; /* how many requests we have */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson unsigned int tcpbuffers; /* allocated buffers */
7693d4de8fca501dfe6989a7f30d8d3c86fe096aAndreas Gustafsson dns_tcpmsg_t tcpmsg; /* for tcp streams */
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_qid_t *qid;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt};
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#define QID_MAGIC ISC_MAGIC('Q', 'i', 'd', ' ')
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#define VALID_QID(e) ISC_MAGIC_VALID((e), QID_MAGIC)
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#define RESPONSE_MAGIC ISC_MAGIC('D', 'r', 's', 'p')
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews#define VALID_RESPONSE(e) ISC_MAGIC_VALID((e), RESPONSE_MAGIC)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews#define DISPATCH_MAGIC ISC_MAGIC('D', 'i', 's', 'p')
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews#define VALID_DISPATCH(e) ISC_MAGIC_VALID((e), DISPATCH_MAGIC)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews#define DNS_DISPATCHMGR_MAGIC ISC_MAGIC('D', 'M', 'g', 'r')
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews#define VALID_DISPATCHMGR(e) ISC_MAGIC_VALID((e), DNS_DISPATCHMGR_MAGIC)
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt#define DNS_QID(disp) ((disp)->socktype == isc_sockettype_tcp) ? \
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt (disp)->qid : (disp)->mgr->qid
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt/*
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt * Statics.
28ad0be64ee756013c0f6a474fc447ee613ee0d1Evan Hunt */
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic dns_dispentry_t *bucket_search(dns_qid_t *, isc_sockaddr_t *,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_messageid_t, unsigned int);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic isc_boolean_t destroy_disp_ok(dns_dispatch_t *);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void destroy_disp(isc_task_t *task, isc_event_t *event);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void udp_recv(isc_task_t *, isc_event_t *);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void tcp_recv(isc_task_t *, isc_event_t *);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void startrecv(dns_dispatch_t *);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic dns_messageid_t dns_randomid(dns_qid_t *);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic isc_uint32_t dns_hash(dns_qid_t *, isc_sockaddr_t *, dns_messageid_t);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void free_buffer(dns_dispatch_t *disp, void *buf, unsigned int len);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void *allocate_udp_buffer(dns_dispatch_t *disp);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic inline void free_event(dns_dispatch_t *disp, dns_dispatchevent_t *ev);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic inline dns_dispatchevent_t *allocate_event(dns_dispatch_t *disp);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void do_cancel(dns_dispatch_t *disp);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic dns_dispentry_t *linear_first(dns_qid_t *disp);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic dns_dispentry_t *linear_next(dns_qid_t *disp,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_dispentry_t *resp);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void dispatch_free(dns_dispatch_t **dispp);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic isc_result_t dispatch_createudp(dns_dispatchmgr_t *mgr,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_socketmgr_t *sockmgr,
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_taskmgr_t *taskmgr,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_sockaddr_t *localaddr,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt unsigned int maxrequests,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt unsigned int attributes,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_dispatch_t **dispp);
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updaterstatic isc_boolean_t destroy_mgr_ok(dns_dispatchmgr_t *mgr);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void destroy_mgr(dns_dispatchmgr_t **mgrp);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic isc_result_t qid_allocate(dns_dispatchmgr_t *mgr, unsigned int buckets,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt unsigned int increment, dns_qid_t **qidp);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafssonstatic void qid_destroy(isc_mem_t *mctx, dns_qid_t **qidp);
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews#define LVL(x) ISC_LOG_DEBUG(x)
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrewsstatic void
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafssonmgr_log(dns_dispatchmgr_t *mgr, int level, const char *fmt, ...)
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt ISC_FORMAT_PRINTF(3, 4);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
262c39b2366bf79062f7f86b218947523dd1cbacEvan Huntstatic void
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntmgr_log(dns_dispatchmgr_t *mgr, int level, const char *fmt, ...) {
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt char msgbuf[2048];
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews va_list ap;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt if (! isc_log_wouldlog(dns_lctx, level))
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return;
8bb77cd31b7518fb5d2a6a9d75e16e4abd59df61Andreas Gustafsson
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson va_start(ap, fmt);
d8d0c5b1bc97ac0f07e35a31b58ced80ce613c55David Lawrence vsnprintf(msgbuf, sizeof(msgbuf), fmt, ap);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson va_end(ap);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_log_write(dns_lctx,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt DNS_LOGCATEGORY_DISPATCH, DNS_LOGMODULE_DISPATCH,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt level, "dispatchmgr %p: %s", mgr, msgbuf);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews}
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntdispatch_log(dns_dispatch_t *disp, int level, const char *fmt, ...)
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt ISC_FORMAT_PRINTF(3, 4);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void
fc7043d7d1294478c9988c10af9a7fb8fd810338Evan Huntdispatch_log(dns_dispatch_t *disp, int level, const char *fmt, ...) {
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt char msgbuf[2048];
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt va_list ap;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt if (! isc_log_wouldlog(dns_lctx, level))
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt return;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt va_start(ap, fmt);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt vsnprintf(msgbuf, sizeof(msgbuf), fmt, ap);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt va_end(ap);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_log_write(dns_lctx,
1fa2ce7eaef0c17d554495220565b681639b2ce5Mark Andrews DNS_LOGCATEGORY_DISPATCH, DNS_LOGMODULE_DISPATCH,
1fa2ce7eaef0c17d554495220565b681639b2ce5Mark Andrews level, "dispatch %p: %s", disp, msgbuf);
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt}
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic void
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsrequest_log(dns_dispatch_t *disp, dns_dispentry_t *resp,
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews int level, const char *fmt, ...)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ISC_FORMAT_PRINTF(4, 5);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic void
8bb77cd31b7518fb5d2a6a9d75e16e4abd59df61Andreas Gustafssonrequest_log(dns_dispatch_t *disp, dns_dispentry_t *resp,
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson int level, const char *fmt, ...)
40f53fa8d9c6a4fc38c0014495e7a42b08f52481David Lawrence{
114c14f8adfc249cf2e5cdcb9007af46fed257e3Mark Andrews char msgbuf[2048];
114c14f8adfc249cf2e5cdcb9007af46fed257e3Mark Andrews char peerbuf[256];
114c14f8adfc249cf2e5cdcb9007af46fed257e3Mark Andrews va_list ap;
114c14f8adfc249cf2e5cdcb9007af46fed257e3Mark Andrews
ba7ea2326d98edb4296098749fc9cf44b5157643David Lawrence if (! isc_log_wouldlog(dns_lctx, level))
ba7ea2326d98edb4296098749fc9cf44b5157643David Lawrence return;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews va_start(ap, fmt);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews vsnprintf(msgbuf, sizeof(msgbuf), fmt, ap);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews va_end(ap);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (VALID_RESPONSE(resp)) {
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson isc_sockaddr_format(&resp->host, peerbuf, sizeof(peerbuf));
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson isc_log_write(dns_lctx, DNS_LOGCATEGORY_DISPATCH,
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson DNS_LOGMODULE_DISPATCH, level,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt "dispatch %p response %p %s: %s", disp, resp,
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson peerbuf, msgbuf);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson } else {
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson isc_log_write(dns_lctx, DNS_LOGCATEGORY_DISPATCH,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt DNS_LOGMODULE_DISPATCH, level,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt "dispatch %p req/resp %p: %s", disp, resp,
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt msgbuf);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt }
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt}
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic void
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntreseed_lfsr(isc_lfsr_t *lfsr, void *arg)
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt{
aa39170da817cae7b4c6c735cc832e05ec3d2351Mark Andrews dns_dispatchmgr_t *mgr = arg;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_result_t result;
aa39170da817cae7b4c6c735cc832e05ec3d2351Mark Andrews isc_uint32_t val;
12e0477d4e132c9122312246ed60aaa646f819b2Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews REQUIRE(VALID_DISPATCHMGR(mgr));
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (mgr->entropy != NULL) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews result = isc_entropy_getdata(mgr->entropy, &val, sizeof(val),
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews NULL, 0);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews INSIST(result == ISC_R_SUCCESS);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews lfsr->count = (val & 0x1f) + 32;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews lfsr->state = val;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews }
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews lfsr->count = (random() & 0x1f) + 32; /* From 32 to 63 states */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews lfsr->state = random();
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews}
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews/*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Return an unpredictable message ID.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic dns_messageid_t
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsdns_randomid(dns_qid_t *qid) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_uint32_t id;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews id = isc_lfsr_generate32(&qid->qid_lfsr1, &qid->qid_lfsr2);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (dns_messageid_t)(id & 0xFFFF);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews}
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews/*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Return a hash of the destination and message id.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic isc_uint32_t
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsdns_hash(dns_qid_t *qid, isc_sockaddr_t *dest, dns_messageid_t id) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews unsigned int ret;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ret = isc_sockaddr_hash(dest, ISC_TRUE);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ret ^= id;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ret %= qid->qid_nbuckets;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews INSIST(ret < qid->qid_nbuckets);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (ret);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews}
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater/*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Find the first entry in 'qid'. Returns NULL if there are no entries.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic dns_dispentry_t *
262c39b2366bf79062f7f86b218947523dd1cbacEvan Huntlinear_first(dns_qid_t *qid) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dns_dispentry_t *ret;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews unsigned int bucket;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews bucket = 0;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews while (bucket < qid->qid_nbuckets) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ret = ISC_LIST_HEAD(qid->qid_table[bucket]);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (ret != NULL)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (ret);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews bucket++;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews }
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (NULL);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews}
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews/*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Find the next entry after 'resp' in 'qid'. Return NULL if there are
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * no more entries.
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic dns_dispentry_t *
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewslinear_next(dns_qid_t *qid, dns_dispentry_t *resp) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dns_dispentry_t *ret;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews unsigned int bucket;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ret = ISC_LIST_NEXT(resp, link);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (ret != NULL)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (ret);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews bucket = resp->bucket;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews bucket++;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews while (bucket < qid->qid_nbuckets) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ret = ISC_LIST_HEAD(qid->qid_table[bucket]);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (ret != NULL)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (ret);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews bucket++;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews }
aa39170da817cae7b4c6c735cc832e05ec3d2351Mark Andrews
aa39170da817cae7b4c6c735cc832e05ec3d2351Mark Andrews return (NULL);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt}
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt/*
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt * The dispatch must be locked.
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater */
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic isc_boolean_t
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntdestroy_disp_ok(dns_dispatch_t *disp)
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt{
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt if (disp->refcount != 0)
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson return (ISC_FALSE);
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews if (disp->recv_pending != 0)
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews return (ISC_FALSE);
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews if (disp->shutting_down == 0)
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson return (ISC_FALSE);
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson return (ISC_TRUE);
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson}
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews/*
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson * Called when refcount reaches 0 (and safe to destroy).
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt *
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * The dispatcher must not be locked.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * The manager must be locked.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic void
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsdestroy_disp(isc_task_t *task, isc_event_t *event) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dns_dispatch_t *disp;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dns_dispatchmgr_t *mgr;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_boolean_t killmgr;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews INSIST(event->ev_type == DNS_EVENT_DISPATCHCONTROL);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews UNUSED(task);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews disp = event->ev_arg;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews mgr = disp->mgr;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews LOCK(&mgr->lock);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ISC_LIST_UNLINK(mgr->list, disp, link);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dispatch_log(disp, LVL(90),
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews "shutting down; detaching from sock %p, task %p",
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews disp->socket, disp->task);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_socket_detach(&disp->socket);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_task_detach(&disp->task);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_event_free(&event);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater dispatch_free(&disp);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews killmgr = destroy_mgr_ok(mgr);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews UNLOCK(&mgr->lock);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (killmgr)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews destroy_mgr(&mgr);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews}
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews/*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Find an entry for query ID 'id' and socket address 'dest' in 'qid'.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Return NULL if no such entry exists.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic dns_dispentry_t *
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsbucket_search(dns_qid_t *qid, isc_sockaddr_t *dest, dns_messageid_t id,
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews unsigned int bucket)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews{
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater dns_dispentry_t *res;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews REQUIRE(bucket < qid->qid_nbuckets);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews res = ISC_LIST_HEAD(qid->qid_table[bucket]);
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews while (res != NULL) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if ((res->id == id) && isc_sockaddr_equal(dest, &res->host))
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater return (res);
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson res = ISC_LIST_NEXT(res, link);
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater }
febaa091847ab004f40500cc475a819f2c73fcddAndreas Gustafsson
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson return (NULL);
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence}
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
aa23a35d81a9618a40c4a9b44be48009553e4777Andreas Gustafssonstatic void
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafssonfree_buffer(dns_dispatch_t *disp, void *buf, unsigned int len) {
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson INSIST(buf != NULL && len != 0);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence switch (disp->socktype) {
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson case isc_sockettype_tcp:
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson INSIST(disp->tcpbuffers > 0);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson disp->tcpbuffers--;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_mem_put(disp->mgr->mctx, buf, len);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt break;
12e0477d4e132c9122312246ed60aaa646f819b2Mark Andrews case isc_sockettype_udp:
12e0477d4e132c9122312246ed60aaa646f819b2Mark Andrews LOCK(&disp->mgr->buffer_lock);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson INSIST(disp->mgr->buffers > 0);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson INSIST(len == disp->mgr->buffersize);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson disp->mgr->buffers--;
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson isc_mempool_put(disp->mgr->bpool, buf);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson UNLOCK(&disp->mgr->buffer_lock);
ea419adc4eca4c3e44f2c282035b5dce6b795fe2Andreas Gustafsson break;
ea419adc4eca4c3e44f2c282035b5dce6b795fe2Andreas Gustafsson default:
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews INSIST(0);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews break;
aa23a35d81a9618a40c4a9b44be48009553e4777Andreas Gustafsson }
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson}
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafssonstatic void *
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafssonallocate_udp_buffer(dns_dispatch_t *disp) {
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson void *temp;
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson LOCK(&disp->mgr->buffer_lock);
aa23a35d81a9618a40c4a9b44be48009553e4777Andreas Gustafsson temp = isc_mempool_get(disp->mgr->bpool);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
aa23a35d81a9618a40c4a9b44be48009553e4777Andreas Gustafsson if (temp != NULL)
aa23a35d81a9618a40c4a9b44be48009553e4777Andreas Gustafsson disp->mgr->buffers++;
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson UNLOCK(&disp->mgr->buffer_lock);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson return (temp);
6017f424ee3c02d7f22132c77576ea38542fa949Andreas Gustafsson}
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrewsstatic inline void
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrewsfree_event(dns_dispatch_t *disp, dns_dispatchevent_t *ev) {
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews if (disp->failsafe_ev == ev) {
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews INSIST(disp->shutdown_out == 1);
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews disp->shutdown_out = 0;
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return;
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews }
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_mempool_put(disp->mgr->epool, ev);
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt}
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntstatic inline dns_dispatchevent_t *
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Huntallocate_event(dns_dispatch_t *disp) {
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt dns_dispatchevent_t *ev;
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt ev = isc_mempool_get(disp->mgr->epool);
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt if (ev == NULL)
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt return (NULL);
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt ISC_EVENT_INIT(ev, sizeof(*ev), 0, NULL, 0,
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt NULL, NULL, NULL, NULL, NULL);
140a27777d6fba397720770b101967d5cf73f42bAutomatic Updater
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return (ev);
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt}
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
140a27777d6fba397720770b101967d5cf73f42bAutomatic Updater/*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * General flow:
140a27777d6fba397720770b101967d5cf73f42bAutomatic Updater *
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * If I/O result == CANCELED or error, free the buffer.
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt *
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * If query, free the buffer, restart.
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt *
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt * If response:
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Allocate event, fill in details.
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson * If cannot allocate, free buffer, restart.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * find target. If not found, free buffer, restart.
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt * if event queue is not empty, queue. else, send.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * restart.
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrewsstatic void
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafssonudp_recv(isc_task_t *task, isc_event_t *ev_in) {
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson isc_socketevent_t *ev = (isc_socketevent_t *)ev_in;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_dispatch_t *disp = ev_in->ev_arg;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dns_messageid_t id;
262c39b2366bf79062f7f86b218947523dd1cbacEvan Hunt isc_result_t dres;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_buffer_t source;
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson unsigned int flags;
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson dns_dispentry_t *resp;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_dispatchevent_t *rev;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt unsigned int bucket;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_boolean_t killit;
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater isc_boolean_t queue_response;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_dispatchmgr_t *mgr;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt dns_qid_t *qid;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt isc_netaddr_t netaddr;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt int match;
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson
45e1bd63587102c3bb361eaca42ee7b714fb3542Mark Andrews UNUSED(task);
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews LOCK(&disp->lock);
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews mgr = disp->mgr;
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews qid = mgr->qid;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews dispatch_log(disp, LVL(90),
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews "got packet: requests %d, buffers %d, recvs %d",
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews disp->requests, disp->mgr->buffers, disp->recv_pending);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews if (ev->ev_type == ISC_SOCKEVENT_RECVDONE) {
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews /*
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews * Unless the receive event was imported from a listening
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews * interface, in which case the event type is
69f3cb5abcb38f105c653c7b3df7cec33b87b292Mark Andrews * DNS_EVENT_IMPORTRECVDONE, receive operation must be pending.
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews */
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews INSIST(disp->recv_pending != 0);
2f012d936b5ccdf6520c96a4de23721dc58a2221Automatic Updater disp->recv_pending = 0;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews }
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson if (disp->shutting_down) {
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson /*
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson * This dispatcher is shutting down.
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson */
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson free_buffer(disp, ev->region.base, ev->region.length);
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson isc_event_free(&ev_in);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews ev = NULL;
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson killit = destroy_disp_ok(disp);
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson UNLOCK(&disp->lock);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews if (killit)
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews isc_task_send(disp->task, &disp->ctlevent);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews return;
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt }
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson if (ev->result != ISC_R_SUCCESS) {
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson free_buffer(disp, ev->region.base, ev->region.length);
604419a812b491cd35fb6fad129c3c39da7200a1Mark Andrews
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson if (ev->result != ISC_R_CANCELED)
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson dispatch_log(disp, ISC_LOG_ERROR,
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson "odd socket result in udp_recv(): %s",
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson isc_result_totext(ev->result));
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson UNLOCK(&disp->lock);
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson isc_event_free(&ev_in);
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson return;
9c566a852f31c3a5d0b9d6eaf11463114339c01dAndreas Gustafsson }
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt /*
c7e266b7e5675e12d1ca3cc929f24b3e86d41f8eEvan Hunt * If this is from a blackholed address, drop it.
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson */
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence isc_netaddr_fromsockaddr(&netaddr, &ev->address);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson if (disp->mgr->blackhole != NULL &&
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson dns_acl_match(&netaddr, NULL, disp->mgr->blackhole,
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson NULL, &match, NULL) == ISC_R_SUCCESS &&
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson match > 0)
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson {
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson if (isc_log_wouldlog(dns_lctx, LVL(10))) {
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson char netaddrstr[ISC_NETADDR_FORMATSIZE];
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson isc_netaddr_format(&netaddr, netaddrstr,
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson sizeof(netaddrstr));
6eccf5bd07eb9abf65cc08fec4a8fc97b62c0e1bBrian Wellington dispatch_log(disp, LVL(10),
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson "blackholed packet from %s",
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson netaddrstr);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson }
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson free_buffer(disp, ev->region.base, ev->region.length);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson goto restart;
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson }
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson /*
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence * Peek into the buffer to see what we can see.
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence */
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson isc_buffer_init(&source, ev->region.base, ev->region.length);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson isc_buffer_add(&source, ev->n);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson dres = dns_message_peekheader(&source, &id, &flags);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson if (dres != ISC_R_SUCCESS) {
6eccf5bd07eb9abf65cc08fec4a8fc97b62c0e1bBrian Wellington free_buffer(disp, ev->region.base, ev->region.length);
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson dispatch_log(disp, LVL(10), "got garbage packet");
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson goto restart;
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence }
1a69a1a78cfaa86f3b68bbc965232b7876d4da2aDavid Lawrence
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson dispatch_log(disp, LVL(92),
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson "got valid DNS message header, /QR %c, id %u",
a1747570262ed336c213aaf6bd31bc91993a46deAndreas Gustafsson ((flags & DNS_MESSAGEFLAG_QR) ? '1' : '0'), id);
/*
* Look at flags. If query, drop it. If response,
* look to see where it goes.
*/
queue_response = ISC_FALSE;
if ((flags & DNS_MESSAGEFLAG_QR) == 0) {
/* query */
free_buffer(disp, ev->region.base, ev->region.length);
goto restart;
}
/* response */
bucket = dns_hash(qid, &ev->address, id);
LOCK(&qid->lock);
resp = bucket_search(qid, &ev->address, id, bucket);
dispatch_log(disp, LVL(90),
"search for response in bucket %d: %s",
bucket, (resp == NULL ? "not found" : "found"));
if (resp == NULL) {
free_buffer(disp, ev->region.base, ev->region.length);
goto unlock;
}
queue_response = resp->item_out;
rev = allocate_event(resp->disp);
if (rev == NULL) {
free_buffer(disp, ev->region.base, ev->region.length);
goto unlock;
}
/*
* At this point, rev contains the event we want to fill in, and
* resp contains the information on the place to send it to.
* Send the event off.
*/
isc_buffer_init(&rev->buffer, ev->region.base, ev->region.length);
isc_buffer_add(&rev->buffer, ev->n);
rev->result = ISC_R_SUCCESS;
rev->id = id;
rev->addr = ev->address;
rev->pktinfo = ev->pktinfo;
rev->attributes = ev->attributes;
if (queue_response) {
ISC_LIST_APPEND(resp->items, rev, ev_link);
} else {
ISC_EVENT_INIT(rev, sizeof(*rev), 0, NULL,
DNS_EVENT_DISPATCH,
resp->action, resp->arg, resp, NULL, NULL);
request_log(disp, resp, LVL(90),
"[a] Sent event %p buffer %p len %d to task %p",
rev, rev->buffer.base, rev->buffer.length,
resp->task);
resp->item_out = ISC_TRUE;
isc_task_send(resp->task, ISC_EVENT_PTR(&rev));
}
unlock:
UNLOCK(&qid->lock);
/*
* Restart recv() to get the next packet.
*/
restart:
startrecv(disp);
UNLOCK(&disp->lock);
isc_event_free(&ev_in);
}
/*
* General flow:
*
* If I/O result == CANCELED, EOF, or error, notify everyone as the
* various queues drain.
*
* If query, restart.
*
* If response:
* Allocate event, fill in details.
* If cannot allocate, restart.
* find target. If not found, restart.
* if event queue is not empty, queue. else, send.
* restart.
*/
static void
tcp_recv(isc_task_t *task, isc_event_t *ev_in) {
dns_dispatch_t *disp = ev_in->ev_arg;
dns_tcpmsg_t *tcpmsg = &disp->tcpmsg;
dns_messageid_t id;
isc_result_t dres;
unsigned int flags;
dns_dispentry_t *resp;
dns_dispatchevent_t *rev;
unsigned int bucket;
isc_boolean_t killit;
isc_boolean_t queue_response;
dns_qid_t *qid;
int level;
char buf[ISC_SOCKADDR_FORMATSIZE];
UNUSED(task);
REQUIRE(VALID_DISPATCH(disp));
qid = disp->qid;
dispatch_log(disp, LVL(90),
"got TCP packet: requests %d, buffers %d, recvs %d",
disp->requests, disp->tcpbuffers, disp->recv_pending);
LOCK(&disp->lock);
INSIST(disp->recv_pending != 0);
disp->recv_pending = 0;
if (disp->refcount == 0) {
/*
* This dispatcher is shutting down. Force cancelation.
*/
tcpmsg->result = ISC_R_CANCELED;
}
if (tcpmsg->result != ISC_R_SUCCESS) {
switch (tcpmsg->result) {
case ISC_R_CANCELED:
break;
case ISC_R_EOF:
dispatch_log(disp, LVL(90), "shutting down on EOF");
do_cancel(disp);
break;
case ISC_R_CONNECTIONRESET:
level = ISC_LOG_INFO;
goto logit;
default:
level = ISC_LOG_ERROR;
logit:
isc_sockaddr_format(&tcpmsg->address, buf, sizeof(buf));
dispatch_log(disp, level, "shutting down due to TCP "
"receive error: %s: %s", buf,
isc_result_totext(tcpmsg->result));
do_cancel(disp);
break;
}
/*
* The event is statically allocated in the tcpmsg
* structure, and destroy_disp() frees the tcpmsg, so we must
* free the event *before* calling destroy_disp().
*/
isc_event_free(&ev_in);
disp->shutting_down = 1;
disp->shutdown_why = tcpmsg->result;
/*
* If the recv() was canceled pass the word on.
*/
killit = destroy_disp_ok(disp);
UNLOCK(&disp->lock);
if (killit)
isc_task_send(disp->task, &disp->ctlevent);
return;
}
dispatch_log(disp, LVL(90), "result %d, length == %d, addr = %p",
tcpmsg->result,
tcpmsg->buffer.length, tcpmsg->buffer.base);
/*
* Peek into the buffer to see what we can see.
*/
dres = dns_message_peekheader(&tcpmsg->buffer, &id, &flags);
if (dres != ISC_R_SUCCESS) {
dispatch_log(disp, LVL(10), "got garbage packet");
goto restart;
}
dispatch_log(disp, LVL(92),
"got valid DNS message header, /QR %c, id %u",
((flags & DNS_MESSAGEFLAG_QR) ? '1' : '0'), id);
/*
* Allocate an event to send to the query or response client, and
* allocate a new buffer for our use.
*/
/*
* Look at flags. If query, drop it. If response,
* look to see where it goes.
*/
queue_response = ISC_FALSE;
if ((flags & DNS_MESSAGEFLAG_QR) == 0) {
/*
* Query.
*/
goto restart;
}
/*
* Response.
*/
bucket = dns_hash(qid, &tcpmsg->address, id);
LOCK(&qid->lock);
resp = bucket_search(qid, &tcpmsg->address, id, bucket);
dispatch_log(disp, LVL(90),
"search for response in bucket %d: %s",
bucket, (resp == NULL ? "not found" : "found"));
if (resp == NULL)
goto unlock;
queue_response = resp->item_out;
rev = allocate_event(disp);
if (rev == NULL)
goto unlock;
/*
* At this point, rev contains the event we want to fill in, and
* resp contains the information on the place to send it to.
* Send the event off.
*/
dns_tcpmsg_keepbuffer(tcpmsg, &rev->buffer);
disp->tcpbuffers++;
rev->result = ISC_R_SUCCESS;
rev->id = id;
rev->addr = tcpmsg->address;
if (queue_response) {
ISC_LIST_APPEND(resp->items, rev, ev_link);
} else {
ISC_EVENT_INIT(rev, sizeof(*rev), 0, NULL, DNS_EVENT_DISPATCH,
resp->action, resp->arg, resp, NULL, NULL);
request_log(disp, resp, LVL(90),
"[b] Sent event %p buffer %p len %d to task %p",
rev, rev->buffer.base, rev->buffer.length,
resp->task);
resp->item_out = ISC_TRUE;
isc_task_send(resp->task, ISC_EVENT_PTR(&rev));
}
unlock:
UNLOCK(&qid->lock);
/*
* Restart recv() to get the next packet.
*/
restart:
startrecv(disp);
UNLOCK(&disp->lock);
isc_event_free(&ev_in);
}
/*
* disp must be locked.
*/
static void
startrecv(dns_dispatch_t *disp) {
isc_result_t res;
isc_region_t region;
if (disp->shutting_down == 1)
return;
if ((disp->attributes & DNS_DISPATCHATTR_NOLISTEN) != 0)
return;
if (disp->recv_pending != 0)
return;
if (disp->mgr->buffers >= disp->mgr->maxbuffers)
return;
switch (disp->socktype) {
/*
* UDP reads are always maximal.
*/
case isc_sockettype_udp:
region.length = disp->mgr->buffersize;
region.base = allocate_udp_buffer(disp);
if (region.base == NULL)
return;
res = isc_socket_recv(disp->socket, &region, 1,
disp->task, udp_recv, disp);
if (res != ISC_R_SUCCESS) {
free_buffer(disp, region.base, region.length);
disp->shutdown_why = res;
disp->shutting_down = 1;
do_cancel(disp);
return;
}
INSIST(disp->recv_pending == 0);
disp->recv_pending = 1;
break;
case isc_sockettype_tcp:
res = dns_tcpmsg_readmessage(&disp->tcpmsg, disp->task,
tcp_recv, disp);
if (res != ISC_R_SUCCESS) {
disp->shutdown_why = res;
disp->shutting_down = 1;
do_cancel(disp);
return;
}
INSIST(disp->recv_pending == 0);
disp->recv_pending = 1;
break;
default:
INSIST(0);
break;
}
}
/*
* Mgr must be locked when calling this function.
*/
static isc_boolean_t
destroy_mgr_ok(dns_dispatchmgr_t *mgr) {
mgr_log(mgr, LVL(90),
"destroy_mgr_ok: shuttingdown=%d, listnonempty=%d, "
"epool=%d, rpool=%d, dpool=%d",
MGR_IS_SHUTTINGDOWN(mgr), !ISC_LIST_EMPTY(mgr->list),
isc_mempool_getallocated(mgr->epool),
isc_mempool_getallocated(mgr->rpool),
isc_mempool_getallocated(mgr->dpool));
if (!MGR_IS_SHUTTINGDOWN(mgr))
return (ISC_FALSE);
if (!ISC_LIST_EMPTY(mgr->list))
return (ISC_FALSE);
if (isc_mempool_getallocated(mgr->epool) != 0)
return (ISC_FALSE);
if (isc_mempool_getallocated(mgr->rpool) != 0)
return (ISC_FALSE);
if (isc_mempool_getallocated(mgr->dpool) != 0)
return (ISC_FALSE);
return (ISC_TRUE);
}
/*
* Mgr must be unlocked when calling this function.
*/
static void
destroy_mgr(dns_dispatchmgr_t **mgrp) {
isc_mem_t *mctx;
dns_dispatchmgr_t *mgr;
mgr = *mgrp;
*mgrp = NULL;
mctx = mgr->mctx;
mgr->magic = 0;
mgr->mctx = NULL;
DESTROYLOCK(&mgr->lock);
mgr->state = 0;
isc_mempool_destroy(&mgr->epool);
isc_mempool_destroy(&mgr->rpool);
isc_mempool_destroy(&mgr->dpool);
isc_mempool_destroy(&mgr->bpool);
DESTROYLOCK(&mgr->pool_lock);
if (mgr->entropy != NULL)
isc_entropy_detach(&mgr->entropy);
if (mgr->qid != NULL)
qid_destroy(mctx, &mgr->qid);
DESTROYLOCK(&mgr->buffer_lock);
if (mgr->blackhole != NULL)
dns_acl_detach(&mgr->blackhole);
if (mgr->portlist != NULL)
dns_portlist_detach(&mgr->portlist);
isc_mem_put(mctx, mgr, sizeof(dns_dispatchmgr_t));
isc_mem_detach(&mctx);
}
static isc_result_t
create_socket(isc_socketmgr_t *mgr, isc_sockaddr_t *local,
isc_socket_t **sockp)
{
isc_socket_t *sock;
isc_result_t result;
sock = NULL;
result = isc_socket_create(mgr, isc_sockaddr_pf(local),
isc_sockettype_udp, &sock);
if (result != ISC_R_SUCCESS)
return (result);
#ifndef ISC_ALLOW_MAPPED
isc_socket_ipv6only(sock, ISC_TRUE);
#endif
result = isc_socket_bind(sock, local);
if (result != ISC_R_SUCCESS) {
isc_socket_detach(&sock);
return (result);
}
*sockp = sock;
return (ISC_R_SUCCESS);
}
/*
* Publics.
*/
isc_result_t
dns_dispatchmgr_create(isc_mem_t *mctx, isc_entropy_t *entropy,
dns_dispatchmgr_t **mgrp)
{
dns_dispatchmgr_t *mgr;
isc_result_t result;
REQUIRE(mctx != NULL);
REQUIRE(mgrp != NULL && *mgrp == NULL);
mgr = isc_mem_get(mctx, sizeof(dns_dispatchmgr_t));
if (mgr == NULL)
return (ISC_R_NOMEMORY);
mgr->mctx = NULL;
isc_mem_attach(mctx, &mgr->mctx);
mgr->blackhole = NULL;
mgr->portlist = NULL;
result = isc_mutex_init(&mgr->lock);
if (result != ISC_R_SUCCESS)
goto deallocate;
result = isc_mutex_init(&mgr->buffer_lock);
if (result != ISC_R_SUCCESS)
goto kill_lock;
result = isc_mutex_init(&mgr->pool_lock);
if (result != ISC_R_SUCCESS)
goto kill_buffer_lock;
mgr->epool = NULL;
if (isc_mempool_create(mgr->mctx, sizeof(dns_dispatchevent_t),
&mgr->epool) != ISC_R_SUCCESS) {
result = ISC_R_NOMEMORY;
goto kill_pool_lock;
}
mgr->rpool = NULL;
if (isc_mempool_create(mgr->mctx, sizeof(dns_dispentry_t),
&mgr->rpool) != ISC_R_SUCCESS) {
result = ISC_R_NOMEMORY;
goto kill_epool;
}
mgr->dpool = NULL;
if (isc_mempool_create(mgr->mctx, sizeof(dns_dispatch_t),
&mgr->dpool) != ISC_R_SUCCESS) {
result = ISC_R_NOMEMORY;
goto kill_rpool;
}
isc_mempool_setname(mgr->epool, "dispmgr_epool");
isc_mempool_setfreemax(mgr->epool, 1024);
isc_mempool_associatelock(mgr->epool, &mgr->pool_lock);
isc_mempool_setname(mgr->rpool, "dispmgr_rpool");
isc_mempool_setfreemax(mgr->rpool, 1024);
isc_mempool_associatelock(mgr->rpool, &mgr->pool_lock);
isc_mempool_setname(mgr->dpool, "dispmgr_dpool");
isc_mempool_setfreemax(mgr->dpool, 1024);
isc_mempool_associatelock(mgr->dpool, &mgr->pool_lock);
mgr->buffers = 0;
mgr->buffersize = 0;
mgr->maxbuffers = 0;
mgr->bpool = NULL;
mgr->entropy = NULL;
mgr->qid = NULL;
mgr->state = 0;
ISC_LIST_INIT(mgr->list);
mgr->magic = DNS_DISPATCHMGR_MAGIC;
if (entropy != NULL)
isc_entropy_attach(entropy, &mgr->entropy);
*mgrp = mgr;
return (ISC_R_SUCCESS);
kill_rpool:
isc_mempool_destroy(&mgr->rpool);
kill_epool:
isc_mempool_destroy(&mgr->epool);
kill_pool_lock:
DESTROYLOCK(&mgr->pool_lock);
kill_buffer_lock:
DESTROYLOCK(&mgr->buffer_lock);
kill_lock:
DESTROYLOCK(&mgr->lock);
deallocate:
isc_mem_put(mctx, mgr, sizeof(dns_dispatchmgr_t));
isc_mem_detach(&mctx);
return (result);
}
void
dns_dispatchmgr_setblackhole(dns_dispatchmgr_t *mgr, dns_acl_t *blackhole) {
REQUIRE(VALID_DISPATCHMGR(mgr));
if (mgr->blackhole != NULL)
dns_acl_detach(&mgr->blackhole);
dns_acl_attach(blackhole, &mgr->blackhole);
}
dns_acl_t *
dns_dispatchmgr_getblackhole(dns_dispatchmgr_t *mgr) {
REQUIRE(VALID_DISPATCHMGR(mgr));
return (mgr->blackhole);
}
void
dns_dispatchmgr_setblackportlist(dns_dispatchmgr_t *mgr,
dns_portlist_t *portlist)
{
REQUIRE(VALID_DISPATCHMGR(mgr));
if (mgr->portlist != NULL)
dns_portlist_detach(&mgr->portlist);
if (portlist != NULL)
dns_portlist_attach(portlist, &mgr->portlist);
}
dns_portlist_t *
dns_dispatchmgr_getblackportlist(dns_dispatchmgr_t *mgr) {
REQUIRE(VALID_DISPATCHMGR(mgr));
return (mgr->portlist);
}
static isc_result_t
dns_dispatchmgr_setudp(dns_dispatchmgr_t *mgr,
unsigned int buffersize, unsigned int maxbuffers,
unsigned int buckets, unsigned int increment)
{
isc_result_t result;
REQUIRE(VALID_DISPATCHMGR(mgr));
REQUIRE(buffersize >= 512 && buffersize < (64 * 1024));
REQUIRE(maxbuffers > 0);
REQUIRE(buckets < 2097169); /* next prime > 65536 * 32 */
REQUIRE(increment > buckets);
/*
* Keep some number of items around. This should be a config
* option. For now, keep 8, but later keep at least two even
* if the caller wants less. This allows us to ensure certain
* things, like an event can be "freed" and the next allocation
* will always succeed.
*
* Note that if limits are placed on anything here, we use one
* event internally, so the actual limit should be "wanted + 1."
*
* XXXMLG
*/
if (maxbuffers < 8)
maxbuffers = 8;
LOCK(&mgr->buffer_lock);
if (mgr->bpool != NULL) {
isc_mempool_setmaxalloc(mgr->bpool, maxbuffers);
mgr->maxbuffers = maxbuffers;
UNLOCK(&mgr->buffer_lock);
return (ISC_R_SUCCESS);
}
if (isc_mempool_create(mgr->mctx, buffersize,
&mgr->bpool) != ISC_R_SUCCESS) {
return (ISC_R_NOMEMORY);
}
isc_mempool_setname(mgr->bpool, "dispmgr_bpool");
isc_mempool_setmaxalloc(mgr->bpool, maxbuffers);
isc_mempool_associatelock(mgr->bpool, &mgr->pool_lock);
result = qid_allocate(mgr, buckets, increment, &mgr->qid);
if (result != ISC_R_SUCCESS)
goto cleanup;
mgr->buffersize = buffersize;
mgr->maxbuffers = maxbuffers;
UNLOCK(&mgr->buffer_lock);
return (ISC_R_SUCCESS);
cleanup:
isc_mempool_destroy(&mgr->bpool);
UNLOCK(&mgr->buffer_lock);
return (ISC_R_NOMEMORY);
}
void
dns_dispatchmgr_destroy(dns_dispatchmgr_t **mgrp) {
dns_dispatchmgr_t *mgr;
isc_boolean_t killit;
REQUIRE(mgrp != NULL);
REQUIRE(VALID_DISPATCHMGR(*mgrp));
mgr = *mgrp;
*mgrp = NULL;
LOCK(&mgr->lock);
mgr->state |= MGR_SHUTTINGDOWN;
killit = destroy_mgr_ok(mgr);
UNLOCK(&mgr->lock);
mgr_log(mgr, LVL(90), "destroy: killit=%d", killit);
if (killit)
destroy_mgr(&mgr);
}
static isc_boolean_t
blacklisted(dns_dispatchmgr_t *mgr, isc_socket_t *sock) {
isc_sockaddr_t sockaddr;
isc_result_t result;
if (mgr->portlist == NULL)
return (ISC_FALSE);
result = isc_socket_getsockname(sock, &sockaddr);
if (result != ISC_R_SUCCESS)
return (ISC_FALSE);
if (mgr->portlist != NULL &&
dns_portlist_match(mgr->portlist, isc_sockaddr_pf(&sockaddr),
isc_sockaddr_getport(&sockaddr)))
return (ISC_TRUE);
return (ISC_FALSE);
}
#define ATTRMATCH(_a1, _a2, _mask) (((_a1) & (_mask)) == ((_a2) & (_mask)))
static isc_boolean_t
local_addr_match(dns_dispatch_t *disp, isc_sockaddr_t *addr) {
isc_sockaddr_t sockaddr;
isc_result_t result;
if (addr == NULL)
return (ISC_TRUE);
/*
* Don't match wildcard ports against newly blacklisted ports.
*/
if (disp->mgr->portlist != NULL &&
isc_sockaddr_getport(addr) == 0 &&
isc_sockaddr_getport(&disp->local) == 0 &&
blacklisted(disp->mgr, disp->socket))
return (ISC_FALSE);
/*
* Check if we match the binding <address,port>.
* Wildcard ports match/fail here.
*/
if (isc_sockaddr_equal(&disp->local, addr))
return (ISC_TRUE);
if (isc_sockaddr_getport(addr) == 0)
return (ISC_FALSE);
/*
* Check if we match a bound wildcard port <address,port>.
*/
if (!isc_sockaddr_eqaddr(&disp->local, addr))
return (ISC_FALSE);
result = isc_socket_getsockname(disp->socket, &sockaddr);
if (result != ISC_R_SUCCESS)
return (ISC_FALSE);
return (isc_sockaddr_equal(&sockaddr, addr));
}
/*
* Requires mgr be locked.
*
* No dispatcher can be locked by this thread when calling this function.
*
*
* NOTE:
* If a matching dispatcher is found, it is locked after this function
* returns, and must be unlocked by the caller.
*/
static isc_result_t
dispatch_find(dns_dispatchmgr_t *mgr, isc_sockaddr_t *local,
unsigned int attributes, unsigned int mask,
dns_dispatch_t **dispp)
{
dns_dispatch_t *disp;
isc_result_t result;
/*
* Make certain that we will not match a private dispatch.
*/
attributes &= ~DNS_DISPATCHATTR_PRIVATE;
mask |= DNS_DISPATCHATTR_PRIVATE;
disp = ISC_LIST_HEAD(mgr->list);
while (disp != NULL) {
LOCK(&disp->lock);
if ((disp->shutting_down == 0)
&& ATTRMATCH(disp->attributes, attributes, mask)
&& local_addr_match(disp, local))
break;
UNLOCK(&disp->lock);
disp = ISC_LIST_NEXT(disp, link);
}
if (disp == NULL) {
result = ISC_R_NOTFOUND;
goto out;
}
*dispp = disp;
result = ISC_R_SUCCESS;
out:
return (result);
}
static isc_result_t
qid_allocate(dns_dispatchmgr_t *mgr, unsigned int buckets,
unsigned int increment, dns_qid_t **qidp)
{
dns_qid_t *qid;
unsigned int i;
REQUIRE(VALID_DISPATCHMGR(mgr));
REQUIRE(buckets < 2097169); /* next prime > 65536 * 32 */
REQUIRE(increment > buckets);
REQUIRE(qidp != NULL && *qidp == NULL);
qid = isc_mem_get(mgr->mctx, sizeof(*qid));
if (qid == NULL)
return (ISC_R_NOMEMORY);
qid->qid_table = isc_mem_get(mgr->mctx,
buckets * sizeof(dns_displist_t));
if (qid->qid_table == NULL) {
isc_mem_put(mgr->mctx, qid, sizeof(*qid));
return (ISC_R_NOMEMORY);
}
if (isc_mutex_init(&qid->lock) != ISC_R_SUCCESS) {
UNEXPECTED_ERROR(__FILE__, __LINE__, "isc_mutex_init failed");
isc_mem_put(mgr->mctx, qid->qid_table,
buckets * sizeof(dns_displist_t));
isc_mem_put(mgr->mctx, qid, sizeof(*qid));
return (ISC_R_UNEXPECTED);
}
for (i = 0; i < buckets; i++)
ISC_LIST_INIT(qid->qid_table[i]);
qid->qid_nbuckets = buckets;
qid->qid_increment = increment;
qid->magic = QID_MAGIC;
/*
* Initialize to a 32-bit LFSR. Both of these are from Applied
* Cryptography.
*
* lfsr1:
* x^32 + x^7 + x^5 + x^3 + x^2 + x + 1
*
* lfsr2:
* x^32 + x^7 + x^6 + x^2 + 1
*/
isc_lfsr_init(&qid->qid_lfsr1, 0, 32, 0x80000057U,
0, reseed_lfsr, mgr);
isc_lfsr_init(&qid->qid_lfsr2, 0, 32, 0x80000062U,
0, reseed_lfsr, mgr);
*qidp = qid;
return (ISC_R_SUCCESS);
}
static void
qid_destroy(isc_mem_t *mctx, dns_qid_t **qidp) {
dns_qid_t *qid;
REQUIRE(qidp != NULL);
qid = *qidp;
REQUIRE(VALID_QID(qid));
*qidp = NULL;
qid->magic = 0;
isc_mem_put(mctx, qid->qid_table,
qid->qid_nbuckets * sizeof(dns_displist_t));
DESTROYLOCK(&qid->lock);
isc_mem_put(mctx, qid, sizeof(*qid));
}
/*
* Allocate and set important limits.
*/
static isc_result_t
dispatch_allocate(dns_dispatchmgr_t *mgr, unsigned int maxrequests,
dns_dispatch_t **dispp)
{
dns_dispatch_t *disp;
isc_result_t res;
REQUIRE(VALID_DISPATCHMGR(mgr));
REQUIRE(dispp != NULL && *dispp == NULL);
/*
* Set up the dispatcher, mostly. Don't bother setting some of
* the options that are controlled by tcp vs. udp, etc.
*/
disp = isc_mempool_get(mgr->dpool);
if (disp == NULL)
return (ISC_R_NOMEMORY);
disp->magic = 0;
disp->mgr = mgr;
disp->maxrequests = maxrequests;
disp->attributes = 0;
ISC_LINK_INIT(disp, link);
disp->refcount = 1;
disp->recv_pending = 0;
memset(&disp->local, 0, sizeof(disp->local));
disp->shutting_down = 0;
disp->shutdown_out = 0;
disp->connected = 0;
disp->tcpmsg_valid = 0;
disp->shutdown_why = ISC_R_UNEXPECTED;
disp->requests = 0;
disp->tcpbuffers = 0;
disp->qid = NULL;
if (isc_mutex_init(&disp->lock) != ISC_R_SUCCESS) {
res = ISC_R_UNEXPECTED;
UNEXPECTED_ERROR(__FILE__, __LINE__, "isc_mutex_init failed");
goto deallocate;
}
disp->failsafe_ev = allocate_event(disp);
if (disp->failsafe_ev == NULL) {
res = ISC_R_NOMEMORY;
goto kill_lock;
}
disp->magic = DISPATCH_MAGIC;
*dispp = disp;
return (ISC_R_SUCCESS);
/*
* error returns
*/
kill_lock:
DESTROYLOCK(&disp->lock);
deallocate:
isc_mempool_put(mgr->dpool, disp);
return (res);
}
/*
* MUST be unlocked, and not used by anthing.
*/
static void
dispatch_free(dns_dispatch_t **dispp)
{
dns_dispatch_t *disp;
dns_dispatchmgr_t *mgr;
REQUIRE(VALID_DISPATCH(*dispp));
disp = *dispp;
*dispp = NULL;
mgr = disp->mgr;
REQUIRE(VALID_DISPATCHMGR(mgr));
if (disp->tcpmsg_valid) {
dns_tcpmsg_invalidate(&disp->tcpmsg);
disp->tcpmsg_valid = 0;
}
INSIST(disp->tcpbuffers == 0);
INSIST(disp->requests == 0);
INSIST(disp->recv_pending == 0);
isc_mempool_put(mgr->epool, disp->failsafe_ev);
disp->failsafe_ev = NULL;
if (disp->qid != NULL)
qid_destroy(mgr->mctx, &disp->qid);
disp->mgr = NULL;
DESTROYLOCK(&disp->lock);
disp->magic = 0;
isc_mempool_put(mgr->dpool, disp);
}
isc_result_t
dns_dispatch_createtcp(dns_dispatchmgr_t *mgr, isc_socket_t *sock,
isc_taskmgr_t *taskmgr, unsigned int buffersize,
unsigned int maxbuffers, unsigned int maxrequests,
unsigned int buckets, unsigned int increment,
unsigned int attributes, dns_dispatch_t **dispp)
{
isc_result_t result;
dns_dispatch_t *disp;
UNUSED(maxbuffers);
UNUSED(buffersize);
REQUIRE(VALID_DISPATCHMGR(mgr));
REQUIRE(isc_socket_gettype(sock) == isc_sockettype_tcp);
REQUIRE((attributes & DNS_DISPATCHATTR_TCP) != 0);
REQUIRE((attributes & DNS_DISPATCHATTR_UDP) == 0);
attributes |= DNS_DISPATCHATTR_PRIVATE; /* XXXMLG */
LOCK(&mgr->lock);
/*
* dispatch_allocate() checks mgr for us.
* qid_allocate() checks buckets and increment for us.
*/
disp = NULL;
result = dispatch_allocate(mgr, maxrequests, &disp);
if (result != ISC_R_SUCCESS) {
UNLOCK(&mgr->lock);
return (result);
}
result = qid_allocate(mgr, buckets, increment, &disp->qid);
if (result != ISC_R_SUCCESS)
goto deallocate_dispatch;
disp->socktype = isc_sockettype_tcp;
disp->socket = NULL;
isc_socket_attach(sock, &disp->socket);
disp->task = NULL;
result = isc_task_create(taskmgr, 0, &disp->task);
if (result != ISC_R_SUCCESS)
goto kill_socket;
disp->ctlevent = isc_event_allocate(mgr->mctx, disp,
DNS_EVENT_DISPATCHCONTROL,
destroy_disp, disp,
sizeof(isc_event_t));
if (disp->ctlevent == NULL)
goto kill_task;
isc_task_setname(disp->task, "tcpdispatch", disp);
dns_tcpmsg_init(mgr->mctx, disp->socket, &disp->tcpmsg);
disp->tcpmsg_valid = 1;
disp->attributes = attributes;
/*
* Append it to the dispatcher list.
*/
ISC_LIST_APPEND(mgr->list, disp, link);
UNLOCK(&mgr->lock);
mgr_log(mgr, LVL(90), "created TCP dispatcher %p", disp);
dispatch_log(disp, LVL(90), "created task %p", disp->task);
*dispp = disp;
return (ISC_R_SUCCESS);
/*
* Error returns.
*/
kill_task:
isc_task_detach(&disp->task);
kill_socket:
isc_socket_detach(&disp->socket);
deallocate_dispatch:
dispatch_free(&disp);
UNLOCK(&mgr->lock);
return (result);
}
isc_result_t
dns_dispatch_getudp(dns_dispatchmgr_t *mgr, isc_socketmgr_t *sockmgr,
isc_taskmgr_t *taskmgr, isc_sockaddr_t *localaddr,
unsigned int buffersize,
unsigned int maxbuffers, unsigned int maxrequests,
unsigned int buckets, unsigned int increment,
unsigned int attributes, unsigned int mask,
dns_dispatch_t **dispp)
{
isc_result_t result;
dns_dispatch_t *disp;
REQUIRE(VALID_DISPATCHMGR(mgr));
REQUIRE(sockmgr != NULL);
REQUIRE(localaddr != NULL);
REQUIRE(taskmgr != NULL);
REQUIRE(buffersize >= 512 && buffersize < (64 * 1024));
REQUIRE(maxbuffers > 0);
REQUIRE(buckets < 2097169); /* next prime > 65536 * 32 */
REQUIRE(increment > buckets);
REQUIRE(dispp != NULL && *dispp == NULL);
REQUIRE((attributes & DNS_DISPATCHATTR_TCP) == 0);
result = dns_dispatchmgr_setudp(mgr, buffersize, maxbuffers,
buckets, increment);
if (result != ISC_R_SUCCESS)
return (result);
LOCK(&mgr->lock);
/*
* First, see if we have a dispatcher that matches.
*/
disp = NULL;
result = dispatch_find(mgr, localaddr, attributes, mask, &disp);
if (result == ISC_R_SUCCESS) {
disp->refcount++;
if (disp->maxrequests < maxrequests)
disp->maxrequests = maxrequests;
if ((disp->attributes & DNS_DISPATCHATTR_NOLISTEN) == 0 &&
(attributes & DNS_DISPATCHATTR_NOLISTEN) != 0)
{
disp->attributes |= DNS_DISPATCHATTR_NOLISTEN;
if (disp->recv_pending != 0)
isc_socket_cancel(disp->socket, disp->task,
ISC_SOCKCANCEL_RECV);
}
UNLOCK(&disp->lock);
UNLOCK(&mgr->lock);
*dispp = disp;
return (ISC_R_SUCCESS);
}
/*
* Nope, create one.
*/
result = dispatch_createudp(mgr, sockmgr, taskmgr, localaddr,
maxrequests, attributes, &disp);
if (result != ISC_R_SUCCESS) {
UNLOCK(&mgr->lock);
return (result);
}
UNLOCK(&mgr->lock);
*dispp = disp;
return (ISC_R_SUCCESS);
}
/*
* mgr should be locked.
*/
static isc_result_t
dispatch_createudp(dns_dispatchmgr_t *mgr, isc_socketmgr_t *sockmgr,
isc_taskmgr_t *taskmgr,
isc_sockaddr_t *localaddr,
unsigned int maxrequests,
unsigned int attributes,
dns_dispatch_t **dispp)
{
isc_result_t result;
dns_dispatch_t *disp;
isc_socket_t *sock;
/*
* dispatch_allocate() checks mgr for us.
*/
disp = NULL;
result = dispatch_allocate(mgr, maxrequests, &disp);
if (result != ISC_R_SUCCESS)
return (result);
/*
* This assumes that the IP stack will *not* quickly reallocate
* the same port. If it does continually reallocate the same port
* then we need a mechanism to hold all the blacklisted sockets
* until we find a usable socket.
*/
getsocket:
result = create_socket(sockmgr, localaddr, &sock);
if (result != ISC_R_SUCCESS)
goto deallocate_dispatch;
if (isc_sockaddr_getport(localaddr) == 0 && blacklisted(mgr, sock)) {
isc_socket_detach(&sock);
goto getsocket;
}
disp->socktype = isc_sockettype_udp;
disp->socket = sock;
disp->local = *localaddr;
disp->task = NULL;
result = isc_task_create(taskmgr, 0, &disp->task);
if (result != ISC_R_SUCCESS)
goto kill_socket;
disp->ctlevent = isc_event_allocate(mgr->mctx, disp,
DNS_EVENT_DISPATCHCONTROL,
destroy_disp, disp,
sizeof(isc_event_t));
if (disp->ctlevent == NULL)
goto kill_task;
isc_task_setname(disp->task, "udpdispatch", disp);
attributes &= ~DNS_DISPATCHATTR_TCP;
attributes |= DNS_DISPATCHATTR_UDP;
disp->attributes = attributes;
/*
* Append it to the dispatcher list.
*/
ISC_LIST_APPEND(mgr->list, disp, link);
mgr_log(mgr, LVL(90), "created UDP dispatcher %p", disp);
dispatch_log(disp, LVL(90), "created task %p", disp->task);
dispatch_log(disp, LVL(90), "created socket %p", disp->socket);
*dispp = disp;
return (ISC_R_SUCCESS);
/*
* Error returns.
*/
kill_task:
isc_task_detach(&disp->task);
kill_socket:
isc_socket_detach(&disp->socket);
deallocate_dispatch:
dispatch_free(&disp);
return (result);
}
void
dns_dispatch_attach(dns_dispatch_t *disp, dns_dispatch_t **dispp) {
REQUIRE(VALID_DISPATCH(disp));
REQUIRE(dispp != NULL && *dispp == NULL);
LOCK(&disp->lock);
disp->refcount++;
UNLOCK(&disp->lock);
*dispp = disp;
}
/*
* It is important to lock the manager while we are deleting the dispatch,
* since dns_dispatch_getudp will call dispatch_find, which returns to
* the caller a dispatch but does not attach to it until later. _getudp
* locks the manager, however, so locking it here will keep us from attaching
* to a dispatcher that is in the process of going away.
*/
void
dns_dispatch_detach(dns_dispatch_t **dispp) {
dns_dispatch_t *disp;
isc_boolean_t killit;
REQUIRE(dispp != NULL && VALID_DISPATCH(*dispp));
disp = *dispp;
*dispp = NULL;
LOCK(&disp->lock);
INSIST(disp->refcount > 0);
disp->refcount--;
killit = ISC_FALSE;
if (disp->refcount == 0) {
if (disp->recv_pending > 0)
isc_socket_cancel(disp->socket, disp->task,
ISC_SOCKCANCEL_RECV);
disp->shutting_down = 1;
}
dispatch_log(disp, LVL(90), "detach: refcount %d", disp->refcount);
killit = destroy_disp_ok(disp);
UNLOCK(&disp->lock);
if (killit)
isc_task_send(disp->task, &disp->ctlevent);
}
isc_result_t
dns_dispatch_addresponse(dns_dispatch_t *disp, isc_sockaddr_t *dest,
isc_task_t *task, isc_taskaction_t action, void *arg,
dns_messageid_t *idp, dns_dispentry_t **resp)
{
dns_dispentry_t *res;
unsigned int bucket;
dns_messageid_t id;
int i;
isc_boolean_t ok;
dns_qid_t *qid;
REQUIRE(VALID_DISPATCH(disp));
REQUIRE(task != NULL);
REQUIRE(dest != NULL);
REQUIRE(resp != NULL && *resp == NULL);
REQUIRE(idp != NULL);
LOCK(&disp->lock);
if (disp->shutting_down == 1) {
UNLOCK(&disp->lock);
return (ISC_R_SHUTTINGDOWN);
}
if (disp->requests >= disp->maxrequests) {
UNLOCK(&disp->lock);
return (ISC_R_QUOTA);
}
/*
* Try somewhat hard to find an unique ID.
*/
qid = DNS_QID(disp);
LOCK(&qid->lock);
id = dns_randomid(qid);
bucket = dns_hash(qid, dest, id);
ok = ISC_FALSE;
for (i = 0; i < 64; i++) {
if (bucket_search(qid, dest, id, bucket) == NULL) {
ok = ISC_TRUE;
break;
}
id += qid->qid_increment;
id &= 0x0000ffff;
bucket = dns_hash(qid, dest, id);
}
if (!ok) {
UNLOCK(&qid->lock);
UNLOCK(&disp->lock);
return (ISC_R_NOMORE);
}
res = isc_mempool_get(disp->mgr->rpool);
if (res == NULL) {
UNLOCK(&qid->lock);
UNLOCK(&disp->lock);
return (ISC_R_NOMEMORY);
}
disp->refcount++;
disp->requests++;
res->task = NULL;
isc_task_attach(task, &res->task);
res->disp = disp;
res->id = id;
res->bucket = bucket;
res->host = *dest;
res->action = action;
res->arg = arg;
res->item_out = ISC_FALSE;
ISC_LIST_INIT(res->items);
ISC_LINK_INIT(res, link);
res->magic = RESPONSE_MAGIC;
ISC_LIST_APPEND(qid->qid_table[bucket], res, link);
UNLOCK(&qid->lock);
request_log(disp, res, LVL(90),
"attached to task %p", res->task);
if (((disp->attributes & DNS_DISPATCHATTR_UDP) != 0) ||
((disp->attributes & DNS_DISPATCHATTR_CONNECTED) != 0))
startrecv(disp);
UNLOCK(&disp->lock);
*idp = id;
*resp = res;
return (ISC_R_SUCCESS);
}
void
dns_dispatch_starttcp(dns_dispatch_t *disp) {
REQUIRE(VALID_DISPATCH(disp));
dispatch_log(disp, LVL(90), "starttcp %p", disp->task);
LOCK(&disp->lock);
disp->attributes |= DNS_DISPATCHATTR_CONNECTED;
startrecv(disp);
UNLOCK(&disp->lock);
}
void
dns_dispatch_removeresponse(dns_dispentry_t **resp,
dns_dispatchevent_t **sockevent)
{
dns_dispatchmgr_t *mgr;
dns_dispatch_t *disp;
dns_dispentry_t *res;
dns_dispatchevent_t *ev;
unsigned int bucket;
isc_boolean_t killit;
unsigned int n;
isc_eventlist_t events;
dns_qid_t *qid;
REQUIRE(resp != NULL);
REQUIRE(VALID_RESPONSE(*resp));
res = *resp;
*resp = NULL;
disp = res->disp;
REQUIRE(VALID_DISPATCH(disp));
mgr = disp->mgr;
REQUIRE(VALID_DISPATCHMGR(mgr));
qid = DNS_QID(disp);
if (sockevent != NULL) {
REQUIRE(*sockevent != NULL);
ev = *sockevent;
*sockevent = NULL;
} else {
ev = NULL;
}
LOCK(&disp->lock);
INSIST(disp->requests > 0);
disp->requests--;
INSIST(disp->refcount > 0);
disp->refcount--;
killit = ISC_FALSE;
if (disp->refcount == 0) {
if (disp->recv_pending > 0)
isc_socket_cancel(disp->socket, disp->task,
ISC_SOCKCANCEL_RECV);
disp->shutting_down = 1;
}
bucket = res->bucket;
LOCK(&qid->lock);
ISC_LIST_UNLINK(qid->qid_table[bucket], res, link);
UNLOCK(&qid->lock);
if (ev == NULL && res->item_out) {
/*
* We've posted our event, but the caller hasn't gotten it
* yet. Take it back.
*/
ISC_LIST_INIT(events);
n = isc_task_unsend(res->task, res, DNS_EVENT_DISPATCH,
NULL, &events);
/*
* We had better have gotten it back.
*/
INSIST(n == 1);
ev = (dns_dispatchevent_t *)ISC_LIST_HEAD(events);
}
if (ev != NULL) {
REQUIRE(res->item_out == ISC_TRUE);
res->item_out = ISC_FALSE;
if (ev->buffer.base != NULL)
free_buffer(disp, ev->buffer.base, ev->buffer.length);
free_event(disp, ev);
}
request_log(disp, res, LVL(90), "detaching from task %p", res->task);
isc_task_detach(&res->task);
/*
* Free any buffered requests as well
*/
ev = ISC_LIST_HEAD(res->items);
while (ev != NULL) {
ISC_LIST_UNLINK(res->items, ev, ev_link);
if (ev->buffer.base != NULL)
free_buffer(disp, ev->buffer.base, ev->buffer.length);
free_event(disp, ev);
ev = ISC_LIST_HEAD(res->items);
}
res->magic = 0;
isc_mempool_put(disp->mgr->rpool, res);
if (disp->shutting_down == 1)
do_cancel(disp);
else
startrecv(disp);
killit = destroy_disp_ok(disp);
UNLOCK(&disp->lock);
if (killit)
isc_task_send(disp->task, &disp->ctlevent);
}
static void
do_cancel(dns_dispatch_t *disp) {
dns_dispatchevent_t *ev;
dns_dispentry_t *resp;
dns_qid_t *qid;
if (disp->shutdown_out == 1)
return;
qid = DNS_QID(disp);
/*
* Search for the first response handler without packets outstanding.
*/
LOCK(&qid->lock);
for (resp = linear_first(qid);
resp != NULL && resp->item_out != ISC_FALSE;
/* Empty. */)
resp = linear_next(qid, resp);
/*
* No one to send the cancel event to, so nothing to do.
*/
if (resp == NULL)
goto unlock;
/*
* Send the shutdown failsafe event to this resp.
*/
ev = disp->failsafe_ev;
ISC_EVENT_INIT(ev, sizeof(*ev), 0, NULL, DNS_EVENT_DISPATCH,
resp->action, resp->arg, resp, NULL, NULL);
ev->result = disp->shutdown_why;
ev->buffer.base = NULL;
ev->buffer.length = 0;
disp->shutdown_out = 1;
request_log(disp, resp, LVL(10),
"cancel: failsafe event %p -> task %p",
ev, resp->task);
resp->item_out = ISC_TRUE;
isc_task_send(resp->task, ISC_EVENT_PTR(&ev));
unlock:
UNLOCK(&qid->lock);
}
isc_socket_t *
dns_dispatch_getsocket(dns_dispatch_t *disp) {
REQUIRE(VALID_DISPATCH(disp));
return (disp->socket);
}
isc_result_t
dns_dispatch_getlocaladdress(dns_dispatch_t *disp, isc_sockaddr_t *addrp) {
REQUIRE(VALID_DISPATCH(disp));
REQUIRE(addrp != NULL);
if (disp->socktype == isc_sockettype_udp) {
*addrp = disp->local;
return (ISC_R_SUCCESS);
}
return (ISC_R_NOTIMPLEMENTED);
}
void
dns_dispatch_cancel(dns_dispatch_t *disp) {
REQUIRE(VALID_DISPATCH(disp));
LOCK(&disp->lock);
if (disp->shutting_down == 1) {
UNLOCK(&disp->lock);
return;
}
disp->shutdown_why = ISC_R_CANCELED;
disp->shutting_down = 1;
do_cancel(disp);
UNLOCK(&disp->lock);
return;
}
void
dns_dispatch_changeattributes(dns_dispatch_t *disp,
unsigned int attributes, unsigned int mask)
{
REQUIRE(VALID_DISPATCH(disp));
/* XXXMLG
* Should check for valid attributes here!
*/
LOCK(&disp->lock);
if ((mask & DNS_DISPATCHATTR_NOLISTEN) != 0) {
if ((disp->attributes & DNS_DISPATCHATTR_NOLISTEN) != 0 &&
(attributes & DNS_DISPATCHATTR_NOLISTEN) == 0) {
disp->attributes &= ~DNS_DISPATCHATTR_NOLISTEN;
startrecv(disp);
} else if ((disp->attributes & DNS_DISPATCHATTR_NOLISTEN)
== 0 &&
(attributes & DNS_DISPATCHATTR_NOLISTEN) != 0) {
disp->attributes |= DNS_DISPATCHATTR_NOLISTEN;
if (disp->recv_pending != 0)
isc_socket_cancel(disp->socket, disp->task,
ISC_SOCKCANCEL_RECV);
}
}
disp->attributes &= ~mask;
disp->attributes |= (attributes & mask);
UNLOCK(&disp->lock);
}
void
dns_dispatch_importrecv(dns_dispatch_t *disp, isc_event_t *event) {
void *buf;
isc_socketevent_t *sevent, *newsevent;
REQUIRE(VALID_DISPATCH(disp));
REQUIRE((disp->attributes & DNS_DISPATCHATTR_NOLISTEN) != 0);
REQUIRE(event != NULL);
sevent = (isc_socketevent_t *)event;
INSIST(sevent->n <= disp->mgr->buffersize);
newsevent = (isc_socketevent_t *)
isc_event_allocate(disp->mgr->mctx, NULL,
DNS_EVENT_IMPORTRECVDONE, udp_recv,
disp, sizeof(isc_socketevent_t));
if (newsevent == NULL)
return;
buf = allocate_udp_buffer(disp);
if (buf == NULL) {
isc_event_free(ISC_EVENT_PTR(&newsevent));
return;
}
memcpy(buf, sevent->region.base, sevent->n);
newsevent->region.base = buf;
newsevent->region.length = disp->mgr->buffersize;
newsevent->n = sevent->n;
newsevent->result = sevent->result;
newsevent->address = sevent->address;
newsevent->timestamp = sevent->timestamp;
newsevent->pktinfo = sevent->pktinfo;
newsevent->attributes = sevent->attributes;
isc_task_send(disp->task, ISC_EVENT_PTR(&newsevent));
}
#if 0
void
dns_dispatchmgr_dump(dns_dispatchmgr_t *mgr) {
dns_dispatch_t *disp;
char foo[1024];
disp = ISC_LIST_HEAD(mgr->list);
while (disp != NULL) {
isc_sockaddr_format(&disp->local, foo, sizeof(foo));
printf("\tdispatch %p, addr %s\n", disp, foo);
disp = ISC_LIST_NEXT(disp, link);
}
}
#endif