1*b7579f77SDag-Erling Smørgrav /* 2*b7579f77SDag-Erling Smørgrav * daemon/remote.c - remote control for the unbound daemon. 3*b7579f77SDag-Erling Smørgrav * 4*b7579f77SDag-Erling Smørgrav * Copyright (c) 2008, NLnet Labs. All rights reserved. 5*b7579f77SDag-Erling Smørgrav * 6*b7579f77SDag-Erling Smørgrav * This software is open source. 7*b7579f77SDag-Erling Smørgrav * 8*b7579f77SDag-Erling Smørgrav * Redistribution and use in source and binary forms, with or without 9*b7579f77SDag-Erling Smørgrav * modification, are permitted provided that the following conditions 10*b7579f77SDag-Erling Smørgrav * are met: 11*b7579f77SDag-Erling Smørgrav * 12*b7579f77SDag-Erling Smørgrav * Redistributions of source code must retain the above copyright notice, 13*b7579f77SDag-Erling Smørgrav * this list of conditions and the following disclaimer. 14*b7579f77SDag-Erling Smørgrav * 15*b7579f77SDag-Erling Smørgrav * Redistributions in binary form must reproduce the above copyright notice, 16*b7579f77SDag-Erling Smørgrav * this list of conditions and the following disclaimer in the documentation 17*b7579f77SDag-Erling Smørgrav * and/or other materials provided with the distribution. 18*b7579f77SDag-Erling Smørgrav * 19*b7579f77SDag-Erling Smørgrav * Neither the name of the NLNET LABS nor the names of its contributors may 20*b7579f77SDag-Erling Smørgrav * be used to endorse or promote products derived from this software without 21*b7579f77SDag-Erling Smørgrav * specific prior written permission. 22*b7579f77SDag-Erling Smørgrav * 23*b7579f77SDag-Erling Smørgrav * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 24*b7579f77SDag-Erling Smørgrav * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 25*b7579f77SDag-Erling Smørgrav * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 26*b7579f77SDag-Erling Smørgrav * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE 27*b7579f77SDag-Erling Smørgrav * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 28*b7579f77SDag-Erling Smørgrav * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 29*b7579f77SDag-Erling Smørgrav * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 30*b7579f77SDag-Erling Smørgrav * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 31*b7579f77SDag-Erling Smørgrav * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 32*b7579f77SDag-Erling Smørgrav * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 33*b7579f77SDag-Erling Smørgrav * POSSIBILITY OF SUCH DAMAGE. 34*b7579f77SDag-Erling Smørgrav */ 35*b7579f77SDag-Erling Smørgrav 36*b7579f77SDag-Erling Smørgrav /** 37*b7579f77SDag-Erling Smørgrav * \file 38*b7579f77SDag-Erling Smørgrav * 39*b7579f77SDag-Erling Smørgrav * This file contains the remote control functionality for the daemon. 40*b7579f77SDag-Erling Smørgrav * The remote control can be performed using either the commandline 41*b7579f77SDag-Erling Smørgrav * unbound-control tool, or a SSLv3/TLS capable web browser. 42*b7579f77SDag-Erling Smørgrav * The channel is secured using SSLv3 or TLSv1, and certificates. 43*b7579f77SDag-Erling Smørgrav * Both the server and the client(control tool) have their own keys. 44*b7579f77SDag-Erling Smørgrav */ 45*b7579f77SDag-Erling Smørgrav #include "config.h" 46*b7579f77SDag-Erling Smørgrav #ifdef HAVE_OPENSSL_ERR_H 47*b7579f77SDag-Erling Smørgrav #include <openssl/err.h> 48*b7579f77SDag-Erling Smørgrav #endif 49*b7579f77SDag-Erling Smørgrav #include <ctype.h> 50*b7579f77SDag-Erling Smørgrav #include <ldns/ldns.h> 51*b7579f77SDag-Erling Smørgrav #include "daemon/remote.h" 52*b7579f77SDag-Erling Smørgrav #include "daemon/worker.h" 53*b7579f77SDag-Erling Smørgrav #include "daemon/daemon.h" 54*b7579f77SDag-Erling Smørgrav #include "daemon/stats.h" 55*b7579f77SDag-Erling Smørgrav #include "daemon/cachedump.h" 56*b7579f77SDag-Erling Smørgrav #include "util/log.h" 57*b7579f77SDag-Erling Smørgrav #include "util/config_file.h" 58*b7579f77SDag-Erling Smørgrav #include "util/net_help.h" 59*b7579f77SDag-Erling Smørgrav #include "util/module.h" 60*b7579f77SDag-Erling Smørgrav #include "services/listen_dnsport.h" 61*b7579f77SDag-Erling Smørgrav #include "services/cache/rrset.h" 62*b7579f77SDag-Erling Smørgrav #include "services/cache/infra.h" 63*b7579f77SDag-Erling Smørgrav #include "services/mesh.h" 64*b7579f77SDag-Erling Smørgrav #include "services/localzone.h" 65*b7579f77SDag-Erling Smørgrav #include "util/storage/slabhash.h" 66*b7579f77SDag-Erling Smørgrav #include "util/fptr_wlist.h" 67*b7579f77SDag-Erling Smørgrav #include "util/data/dname.h" 68*b7579f77SDag-Erling Smørgrav #include "validator/validator.h" 69*b7579f77SDag-Erling Smørgrav #include "validator/val_kcache.h" 70*b7579f77SDag-Erling Smørgrav #include "validator/val_kentry.h" 71*b7579f77SDag-Erling Smørgrav #include "validator/val_anchor.h" 72*b7579f77SDag-Erling Smørgrav #include "iterator/iterator.h" 73*b7579f77SDag-Erling Smørgrav #include "iterator/iter_fwd.h" 74*b7579f77SDag-Erling Smørgrav #include "iterator/iter_hints.h" 75*b7579f77SDag-Erling Smørgrav #include "iterator/iter_delegpt.h" 76*b7579f77SDag-Erling Smørgrav #include "services/outbound_list.h" 77*b7579f77SDag-Erling Smørgrav #include "services/outside_network.h" 78*b7579f77SDag-Erling Smørgrav 79*b7579f77SDag-Erling Smørgrav #ifdef HAVE_SYS_TYPES_H 80*b7579f77SDag-Erling Smørgrav # include <sys/types.h> 81*b7579f77SDag-Erling Smørgrav #endif 82*b7579f77SDag-Erling Smørgrav #ifdef HAVE_NETDB_H 83*b7579f77SDag-Erling Smørgrav #include <netdb.h> 84*b7579f77SDag-Erling Smørgrav #endif 85*b7579f77SDag-Erling Smørgrav 86*b7579f77SDag-Erling Smørgrav /* just for portability */ 87*b7579f77SDag-Erling Smørgrav #ifdef SQ 88*b7579f77SDag-Erling Smørgrav #undef SQ 89*b7579f77SDag-Erling Smørgrav #endif 90*b7579f77SDag-Erling Smørgrav 91*b7579f77SDag-Erling Smørgrav /** what to put on statistics lines between var and value, ": " or "=" */ 92*b7579f77SDag-Erling Smørgrav #define SQ "=" 93*b7579f77SDag-Erling Smørgrav /** if true, inhibits a lot of =0 lines from the stats output */ 94*b7579f77SDag-Erling Smørgrav static const int inhibit_zero = 1; 95*b7579f77SDag-Erling Smørgrav 96*b7579f77SDag-Erling Smørgrav /** subtract timers and the values do not overflow or become negative */ 97*b7579f77SDag-Erling Smørgrav static void 98*b7579f77SDag-Erling Smørgrav timeval_subtract(struct timeval* d, const struct timeval* end, 99*b7579f77SDag-Erling Smørgrav const struct timeval* start) 100*b7579f77SDag-Erling Smørgrav { 101*b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S 102*b7579f77SDag-Erling Smørgrav time_t end_usec = end->tv_usec; 103*b7579f77SDag-Erling Smørgrav d->tv_sec = end->tv_sec - start->tv_sec; 104*b7579f77SDag-Erling Smørgrav if(end_usec < start->tv_usec) { 105*b7579f77SDag-Erling Smørgrav end_usec += 1000000; 106*b7579f77SDag-Erling Smørgrav d->tv_sec--; 107*b7579f77SDag-Erling Smørgrav } 108*b7579f77SDag-Erling Smørgrav d->tv_usec = end_usec - start->tv_usec; 109*b7579f77SDag-Erling Smørgrav #endif 110*b7579f77SDag-Erling Smørgrav } 111*b7579f77SDag-Erling Smørgrav 112*b7579f77SDag-Erling Smørgrav /** divide sum of timers to get average */ 113*b7579f77SDag-Erling Smørgrav static void 114*b7579f77SDag-Erling Smørgrav timeval_divide(struct timeval* avg, const struct timeval* sum, size_t d) 115*b7579f77SDag-Erling Smørgrav { 116*b7579f77SDag-Erling Smørgrav #ifndef S_SPLINT_S 117*b7579f77SDag-Erling Smørgrav size_t leftover; 118*b7579f77SDag-Erling Smørgrav if(d == 0) { 119*b7579f77SDag-Erling Smørgrav avg->tv_sec = 0; 120*b7579f77SDag-Erling Smørgrav avg->tv_usec = 0; 121*b7579f77SDag-Erling Smørgrav return; 122*b7579f77SDag-Erling Smørgrav } 123*b7579f77SDag-Erling Smørgrav avg->tv_sec = sum->tv_sec / d; 124*b7579f77SDag-Erling Smørgrav avg->tv_usec = sum->tv_usec / d; 125*b7579f77SDag-Erling Smørgrav /* handle fraction from seconds divide */ 126*b7579f77SDag-Erling Smørgrav leftover = sum->tv_sec - avg->tv_sec*d; 127*b7579f77SDag-Erling Smørgrav avg->tv_usec += (leftover*1000000)/d; 128*b7579f77SDag-Erling Smørgrav #endif 129*b7579f77SDag-Erling Smørgrav } 130*b7579f77SDag-Erling Smørgrav 131*b7579f77SDag-Erling Smørgrav struct daemon_remote* 132*b7579f77SDag-Erling Smørgrav daemon_remote_create(struct config_file* cfg) 133*b7579f77SDag-Erling Smørgrav { 134*b7579f77SDag-Erling Smørgrav char* s_cert; 135*b7579f77SDag-Erling Smørgrav char* s_key; 136*b7579f77SDag-Erling Smørgrav struct daemon_remote* rc = (struct daemon_remote*)calloc(1, 137*b7579f77SDag-Erling Smørgrav sizeof(*rc)); 138*b7579f77SDag-Erling Smørgrav if(!rc) { 139*b7579f77SDag-Erling Smørgrav log_err("out of memory in daemon_remote_create"); 140*b7579f77SDag-Erling Smørgrav return NULL; 141*b7579f77SDag-Erling Smørgrav } 142*b7579f77SDag-Erling Smørgrav rc->max_active = 10; 143*b7579f77SDag-Erling Smørgrav 144*b7579f77SDag-Erling Smørgrav if(!cfg->remote_control_enable) { 145*b7579f77SDag-Erling Smørgrav rc->ctx = NULL; 146*b7579f77SDag-Erling Smørgrav return rc; 147*b7579f77SDag-Erling Smørgrav } 148*b7579f77SDag-Erling Smørgrav rc->ctx = SSL_CTX_new(SSLv23_server_method()); 149*b7579f77SDag-Erling Smørgrav if(!rc->ctx) { 150*b7579f77SDag-Erling Smørgrav log_crypto_err("could not SSL_CTX_new"); 151*b7579f77SDag-Erling Smørgrav free(rc); 152*b7579f77SDag-Erling Smørgrav return NULL; 153*b7579f77SDag-Erling Smørgrav } 154*b7579f77SDag-Erling Smørgrav /* no SSLv2 because has defects */ 155*b7579f77SDag-Erling Smørgrav if(!(SSL_CTX_set_options(rc->ctx, SSL_OP_NO_SSLv2) & SSL_OP_NO_SSLv2)){ 156*b7579f77SDag-Erling Smørgrav log_crypto_err("could not set SSL_OP_NO_SSLv2"); 157*b7579f77SDag-Erling Smørgrav daemon_remote_delete(rc); 158*b7579f77SDag-Erling Smørgrav return NULL; 159*b7579f77SDag-Erling Smørgrav } 160*b7579f77SDag-Erling Smørgrav s_cert = fname_after_chroot(cfg->server_cert_file, cfg, 1); 161*b7579f77SDag-Erling Smørgrav s_key = fname_after_chroot(cfg->server_key_file, cfg, 1); 162*b7579f77SDag-Erling Smørgrav if(!s_cert || !s_key) { 163*b7579f77SDag-Erling Smørgrav log_err("out of memory in remote control fname"); 164*b7579f77SDag-Erling Smørgrav goto setup_error; 165*b7579f77SDag-Erling Smørgrav } 166*b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "setup SSL certificates"); 167*b7579f77SDag-Erling Smørgrav if (!SSL_CTX_use_certificate_file(rc->ctx,s_cert,SSL_FILETYPE_PEM)) { 168*b7579f77SDag-Erling Smørgrav log_err("Error for server-cert-file: %s", s_cert); 169*b7579f77SDag-Erling Smørgrav log_crypto_err("Error in SSL_CTX use_certificate_file"); 170*b7579f77SDag-Erling Smørgrav goto setup_error; 171*b7579f77SDag-Erling Smørgrav } 172*b7579f77SDag-Erling Smørgrav if(!SSL_CTX_use_PrivateKey_file(rc->ctx,s_key,SSL_FILETYPE_PEM)) { 173*b7579f77SDag-Erling Smørgrav log_err("Error for server-key-file: %s", s_key); 174*b7579f77SDag-Erling Smørgrav log_crypto_err("Error in SSL_CTX use_PrivateKey_file"); 175*b7579f77SDag-Erling Smørgrav goto setup_error; 176*b7579f77SDag-Erling Smørgrav } 177*b7579f77SDag-Erling Smørgrav if(!SSL_CTX_check_private_key(rc->ctx)) { 178*b7579f77SDag-Erling Smørgrav log_err("Error for server-key-file: %s", s_key); 179*b7579f77SDag-Erling Smørgrav log_crypto_err("Error in SSL_CTX check_private_key"); 180*b7579f77SDag-Erling Smørgrav goto setup_error; 181*b7579f77SDag-Erling Smørgrav } 182*b7579f77SDag-Erling Smørgrav if(!SSL_CTX_load_verify_locations(rc->ctx, s_cert, NULL)) { 183*b7579f77SDag-Erling Smørgrav log_crypto_err("Error setting up SSL_CTX verify locations"); 184*b7579f77SDag-Erling Smørgrav setup_error: 185*b7579f77SDag-Erling Smørgrav free(s_cert); 186*b7579f77SDag-Erling Smørgrav free(s_key); 187*b7579f77SDag-Erling Smørgrav daemon_remote_delete(rc); 188*b7579f77SDag-Erling Smørgrav return NULL; 189*b7579f77SDag-Erling Smørgrav } 190*b7579f77SDag-Erling Smørgrav SSL_CTX_set_client_CA_list(rc->ctx, SSL_load_client_CA_file(s_cert)); 191*b7579f77SDag-Erling Smørgrav SSL_CTX_set_verify(rc->ctx, SSL_VERIFY_PEER, NULL); 192*b7579f77SDag-Erling Smørgrav free(s_cert); 193*b7579f77SDag-Erling Smørgrav free(s_key); 194*b7579f77SDag-Erling Smørgrav 195*b7579f77SDag-Erling Smørgrav return rc; 196*b7579f77SDag-Erling Smørgrav } 197*b7579f77SDag-Erling Smørgrav 198*b7579f77SDag-Erling Smørgrav void daemon_remote_clear(struct daemon_remote* rc) 199*b7579f77SDag-Erling Smørgrav { 200*b7579f77SDag-Erling Smørgrav struct rc_state* p, *np; 201*b7579f77SDag-Erling Smørgrav if(!rc) return; 202*b7579f77SDag-Erling Smørgrav /* but do not close the ports */ 203*b7579f77SDag-Erling Smørgrav listen_list_delete(rc->accept_list); 204*b7579f77SDag-Erling Smørgrav rc->accept_list = NULL; 205*b7579f77SDag-Erling Smørgrav /* do close these sockets */ 206*b7579f77SDag-Erling Smørgrav p = rc->busy_list; 207*b7579f77SDag-Erling Smørgrav while(p) { 208*b7579f77SDag-Erling Smørgrav np = p->next; 209*b7579f77SDag-Erling Smørgrav if(p->ssl) 210*b7579f77SDag-Erling Smørgrav SSL_free(p->ssl); 211*b7579f77SDag-Erling Smørgrav comm_point_delete(p->c); 212*b7579f77SDag-Erling Smørgrav free(p); 213*b7579f77SDag-Erling Smørgrav p = np; 214*b7579f77SDag-Erling Smørgrav } 215*b7579f77SDag-Erling Smørgrav rc->busy_list = NULL; 216*b7579f77SDag-Erling Smørgrav rc->active = 0; 217*b7579f77SDag-Erling Smørgrav rc->worker = NULL; 218*b7579f77SDag-Erling Smørgrav } 219*b7579f77SDag-Erling Smørgrav 220*b7579f77SDag-Erling Smørgrav void daemon_remote_delete(struct daemon_remote* rc) 221*b7579f77SDag-Erling Smørgrav { 222*b7579f77SDag-Erling Smørgrav if(!rc) return; 223*b7579f77SDag-Erling Smørgrav daemon_remote_clear(rc); 224*b7579f77SDag-Erling Smørgrav if(rc->ctx) { 225*b7579f77SDag-Erling Smørgrav SSL_CTX_free(rc->ctx); 226*b7579f77SDag-Erling Smørgrav } 227*b7579f77SDag-Erling Smørgrav free(rc); 228*b7579f77SDag-Erling Smørgrav } 229*b7579f77SDag-Erling Smørgrav 230*b7579f77SDag-Erling Smørgrav /** 231*b7579f77SDag-Erling Smørgrav * Add and open a new control port 232*b7579f77SDag-Erling Smørgrav * @param ip: ip str 233*b7579f77SDag-Erling Smørgrav * @param nr: port nr 234*b7579f77SDag-Erling Smørgrav * @param list: list head 235*b7579f77SDag-Erling Smørgrav * @param noproto_is_err: if lack of protocol support is an error. 236*b7579f77SDag-Erling Smørgrav * @return false on failure. 237*b7579f77SDag-Erling Smørgrav */ 238*b7579f77SDag-Erling Smørgrav static int 239*b7579f77SDag-Erling Smørgrav add_open(const char* ip, int nr, struct listen_port** list, int noproto_is_err) 240*b7579f77SDag-Erling Smørgrav { 241*b7579f77SDag-Erling Smørgrav struct addrinfo hints; 242*b7579f77SDag-Erling Smørgrav struct addrinfo* res; 243*b7579f77SDag-Erling Smørgrav struct listen_port* n; 244*b7579f77SDag-Erling Smørgrav int noproto; 245*b7579f77SDag-Erling Smørgrav int fd, r; 246*b7579f77SDag-Erling Smørgrav char port[15]; 247*b7579f77SDag-Erling Smørgrav snprintf(port, sizeof(port), "%d", nr); 248*b7579f77SDag-Erling Smørgrav port[sizeof(port)-1]=0; 249*b7579f77SDag-Erling Smørgrav memset(&hints, 0, sizeof(hints)); 250*b7579f77SDag-Erling Smørgrav hints.ai_socktype = SOCK_STREAM; 251*b7579f77SDag-Erling Smørgrav hints.ai_flags = AI_PASSIVE | AI_NUMERICHOST; 252*b7579f77SDag-Erling Smørgrav if((r = getaddrinfo(ip, port, &hints, &res)) != 0 || !res) { 253*b7579f77SDag-Erling Smørgrav #ifdef USE_WINSOCK 254*b7579f77SDag-Erling Smørgrav if(!noproto_is_err && r == EAI_NONAME) { 255*b7579f77SDag-Erling Smørgrav /* tried to lookup the address as name */ 256*b7579f77SDag-Erling Smørgrav return 1; /* return success, but do nothing */ 257*b7579f77SDag-Erling Smørgrav } 258*b7579f77SDag-Erling Smørgrav #endif /* USE_WINSOCK */ 259*b7579f77SDag-Erling Smørgrav log_err("control interface %s:%s getaddrinfo: %s %s", 260*b7579f77SDag-Erling Smørgrav ip?ip:"default", port, gai_strerror(r), 261*b7579f77SDag-Erling Smørgrav #ifdef EAI_SYSTEM 262*b7579f77SDag-Erling Smørgrav r==EAI_SYSTEM?(char*)strerror(errno):"" 263*b7579f77SDag-Erling Smørgrav #else 264*b7579f77SDag-Erling Smørgrav "" 265*b7579f77SDag-Erling Smørgrav #endif 266*b7579f77SDag-Erling Smørgrav ); 267*b7579f77SDag-Erling Smørgrav return 0; 268*b7579f77SDag-Erling Smørgrav } 269*b7579f77SDag-Erling Smørgrav 270*b7579f77SDag-Erling Smørgrav /* open fd */ 271*b7579f77SDag-Erling Smørgrav fd = create_tcp_accept_sock(res, 1, &noproto); 272*b7579f77SDag-Erling Smørgrav freeaddrinfo(res); 273*b7579f77SDag-Erling Smørgrav if(fd == -1 && noproto) { 274*b7579f77SDag-Erling Smørgrav if(!noproto_is_err) 275*b7579f77SDag-Erling Smørgrav return 1; /* return success, but do nothing */ 276*b7579f77SDag-Erling Smørgrav log_err("cannot open control interface %s %d : " 277*b7579f77SDag-Erling Smørgrav "protocol not supported", ip, nr); 278*b7579f77SDag-Erling Smørgrav return 0; 279*b7579f77SDag-Erling Smørgrav } 280*b7579f77SDag-Erling Smørgrav if(fd == -1) { 281*b7579f77SDag-Erling Smørgrav log_err("cannot open control interface %s %d", ip, nr); 282*b7579f77SDag-Erling Smørgrav return 0; 283*b7579f77SDag-Erling Smørgrav } 284*b7579f77SDag-Erling Smørgrav 285*b7579f77SDag-Erling Smørgrav /* alloc */ 286*b7579f77SDag-Erling Smørgrav n = (struct listen_port*)calloc(1, sizeof(*n)); 287*b7579f77SDag-Erling Smørgrav if(!n) { 288*b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK 289*b7579f77SDag-Erling Smørgrav close(fd); 290*b7579f77SDag-Erling Smørgrav #else 291*b7579f77SDag-Erling Smørgrav closesocket(fd); 292*b7579f77SDag-Erling Smørgrav #endif 293*b7579f77SDag-Erling Smørgrav log_err("out of memory"); 294*b7579f77SDag-Erling Smørgrav return 0; 295*b7579f77SDag-Erling Smørgrav } 296*b7579f77SDag-Erling Smørgrav n->next = *list; 297*b7579f77SDag-Erling Smørgrav *list = n; 298*b7579f77SDag-Erling Smørgrav n->fd = fd; 299*b7579f77SDag-Erling Smørgrav return 1; 300*b7579f77SDag-Erling Smørgrav } 301*b7579f77SDag-Erling Smørgrav 302*b7579f77SDag-Erling Smørgrav struct listen_port* daemon_remote_open_ports(struct config_file* cfg) 303*b7579f77SDag-Erling Smørgrav { 304*b7579f77SDag-Erling Smørgrav struct listen_port* l = NULL; 305*b7579f77SDag-Erling Smørgrav log_assert(cfg->remote_control_enable && cfg->control_port); 306*b7579f77SDag-Erling Smørgrav if(cfg->control_ifs) { 307*b7579f77SDag-Erling Smørgrav struct config_strlist* p; 308*b7579f77SDag-Erling Smørgrav for(p = cfg->control_ifs; p; p = p->next) { 309*b7579f77SDag-Erling Smørgrav if(!add_open(p->str, cfg->control_port, &l, 1)) { 310*b7579f77SDag-Erling Smørgrav listening_ports_free(l); 311*b7579f77SDag-Erling Smørgrav return NULL; 312*b7579f77SDag-Erling Smørgrav } 313*b7579f77SDag-Erling Smørgrav } 314*b7579f77SDag-Erling Smørgrav } else { 315*b7579f77SDag-Erling Smørgrav /* defaults */ 316*b7579f77SDag-Erling Smørgrav if(cfg->do_ip6 && 317*b7579f77SDag-Erling Smørgrav !add_open("::1", cfg->control_port, &l, 0)) { 318*b7579f77SDag-Erling Smørgrav listening_ports_free(l); 319*b7579f77SDag-Erling Smørgrav return NULL; 320*b7579f77SDag-Erling Smørgrav } 321*b7579f77SDag-Erling Smørgrav if(cfg->do_ip4 && 322*b7579f77SDag-Erling Smørgrav !add_open("127.0.0.1", cfg->control_port, &l, 1)) { 323*b7579f77SDag-Erling Smørgrav listening_ports_free(l); 324*b7579f77SDag-Erling Smørgrav return NULL; 325*b7579f77SDag-Erling Smørgrav } 326*b7579f77SDag-Erling Smørgrav } 327*b7579f77SDag-Erling Smørgrav return l; 328*b7579f77SDag-Erling Smørgrav } 329*b7579f77SDag-Erling Smørgrav 330*b7579f77SDag-Erling Smørgrav /** open accept commpoint */ 331*b7579f77SDag-Erling Smørgrav static int 332*b7579f77SDag-Erling Smørgrav accept_open(struct daemon_remote* rc, int fd) 333*b7579f77SDag-Erling Smørgrav { 334*b7579f77SDag-Erling Smørgrav struct listen_list* n = (struct listen_list*)malloc(sizeof(*n)); 335*b7579f77SDag-Erling Smørgrav if(!n) { 336*b7579f77SDag-Erling Smørgrav log_err("out of memory"); 337*b7579f77SDag-Erling Smørgrav return 0; 338*b7579f77SDag-Erling Smørgrav } 339*b7579f77SDag-Erling Smørgrav n->next = rc->accept_list; 340*b7579f77SDag-Erling Smørgrav rc->accept_list = n; 341*b7579f77SDag-Erling Smørgrav /* open commpt */ 342*b7579f77SDag-Erling Smørgrav n->com = comm_point_create_raw(rc->worker->base, fd, 0, 343*b7579f77SDag-Erling Smørgrav &remote_accept_callback, rc); 344*b7579f77SDag-Erling Smørgrav if(!n->com) 345*b7579f77SDag-Erling Smørgrav return 0; 346*b7579f77SDag-Erling Smørgrav /* keep this port open, its fd is kept in the rc portlist */ 347*b7579f77SDag-Erling Smørgrav n->com->do_not_close = 1; 348*b7579f77SDag-Erling Smørgrav return 1; 349*b7579f77SDag-Erling Smørgrav } 350*b7579f77SDag-Erling Smørgrav 351*b7579f77SDag-Erling Smørgrav int daemon_remote_open_accept(struct daemon_remote* rc, 352*b7579f77SDag-Erling Smørgrav struct listen_port* ports, struct worker* worker) 353*b7579f77SDag-Erling Smørgrav { 354*b7579f77SDag-Erling Smørgrav struct listen_port* p; 355*b7579f77SDag-Erling Smørgrav rc->worker = worker; 356*b7579f77SDag-Erling Smørgrav for(p = ports; p; p = p->next) { 357*b7579f77SDag-Erling Smørgrav if(!accept_open(rc, p->fd)) { 358*b7579f77SDag-Erling Smørgrav log_err("could not create accept comm point"); 359*b7579f77SDag-Erling Smørgrav return 0; 360*b7579f77SDag-Erling Smørgrav } 361*b7579f77SDag-Erling Smørgrav } 362*b7579f77SDag-Erling Smørgrav return 1; 363*b7579f77SDag-Erling Smørgrav } 364*b7579f77SDag-Erling Smørgrav 365*b7579f77SDag-Erling Smørgrav void daemon_remote_stop_accept(struct daemon_remote* rc) 366*b7579f77SDag-Erling Smørgrav { 367*b7579f77SDag-Erling Smørgrav struct listen_list* p; 368*b7579f77SDag-Erling Smørgrav for(p=rc->accept_list; p; p=p->next) { 369*b7579f77SDag-Erling Smørgrav comm_point_stop_listening(p->com); 370*b7579f77SDag-Erling Smørgrav } 371*b7579f77SDag-Erling Smørgrav } 372*b7579f77SDag-Erling Smørgrav 373*b7579f77SDag-Erling Smørgrav void daemon_remote_start_accept(struct daemon_remote* rc) 374*b7579f77SDag-Erling Smørgrav { 375*b7579f77SDag-Erling Smørgrav struct listen_list* p; 376*b7579f77SDag-Erling Smørgrav for(p=rc->accept_list; p; p=p->next) { 377*b7579f77SDag-Erling Smørgrav comm_point_start_listening(p->com, -1, -1); 378*b7579f77SDag-Erling Smørgrav } 379*b7579f77SDag-Erling Smørgrav } 380*b7579f77SDag-Erling Smørgrav 381*b7579f77SDag-Erling Smørgrav int remote_accept_callback(struct comm_point* c, void* arg, int err, 382*b7579f77SDag-Erling Smørgrav struct comm_reply* ATTR_UNUSED(rep)) 383*b7579f77SDag-Erling Smørgrav { 384*b7579f77SDag-Erling Smørgrav struct daemon_remote* rc = (struct daemon_remote*)arg; 385*b7579f77SDag-Erling Smørgrav struct sockaddr_storage addr; 386*b7579f77SDag-Erling Smørgrav socklen_t addrlen; 387*b7579f77SDag-Erling Smørgrav int newfd; 388*b7579f77SDag-Erling Smørgrav struct rc_state* n; 389*b7579f77SDag-Erling Smørgrav if(err != NETEVENT_NOERROR) { 390*b7579f77SDag-Erling Smørgrav log_err("error %d on remote_accept_callback", err); 391*b7579f77SDag-Erling Smørgrav return 0; 392*b7579f77SDag-Erling Smørgrav } 393*b7579f77SDag-Erling Smørgrav /* perform the accept */ 394*b7579f77SDag-Erling Smørgrav newfd = comm_point_perform_accept(c, &addr, &addrlen); 395*b7579f77SDag-Erling Smørgrav if(newfd == -1) 396*b7579f77SDag-Erling Smørgrav return 0; 397*b7579f77SDag-Erling Smørgrav /* create new commpoint unless we are servicing already */ 398*b7579f77SDag-Erling Smørgrav if(rc->active >= rc->max_active) { 399*b7579f77SDag-Erling Smørgrav log_warn("drop incoming remote control: too many connections"); 400*b7579f77SDag-Erling Smørgrav close_exit: 401*b7579f77SDag-Erling Smørgrav #ifndef USE_WINSOCK 402*b7579f77SDag-Erling Smørgrav close(newfd); 403*b7579f77SDag-Erling Smørgrav #else 404*b7579f77SDag-Erling Smørgrav closesocket(newfd); 405*b7579f77SDag-Erling Smørgrav #endif 406*b7579f77SDag-Erling Smørgrav return 0; 407*b7579f77SDag-Erling Smørgrav } 408*b7579f77SDag-Erling Smørgrav 409*b7579f77SDag-Erling Smørgrav /* setup commpoint to service the remote control command */ 410*b7579f77SDag-Erling Smørgrav n = (struct rc_state*)calloc(1, sizeof(*n)); 411*b7579f77SDag-Erling Smørgrav if(!n) { 412*b7579f77SDag-Erling Smørgrav log_err("out of memory"); 413*b7579f77SDag-Erling Smørgrav goto close_exit; 414*b7579f77SDag-Erling Smørgrav } 415*b7579f77SDag-Erling Smørgrav /* start in reading state */ 416*b7579f77SDag-Erling Smørgrav n->c = comm_point_create_raw(rc->worker->base, newfd, 0, 417*b7579f77SDag-Erling Smørgrav &remote_control_callback, n); 418*b7579f77SDag-Erling Smørgrav if(!n->c) { 419*b7579f77SDag-Erling Smørgrav log_err("out of memory"); 420*b7579f77SDag-Erling Smørgrav free(n); 421*b7579f77SDag-Erling Smørgrav goto close_exit; 422*b7579f77SDag-Erling Smørgrav } 423*b7579f77SDag-Erling Smørgrav log_addr(VERB_QUERY, "new control connection from", &addr, addrlen); 424*b7579f77SDag-Erling Smørgrav n->c->do_not_close = 0; 425*b7579f77SDag-Erling Smørgrav comm_point_stop_listening(n->c); 426*b7579f77SDag-Erling Smørgrav comm_point_start_listening(n->c, -1, REMOTE_CONTROL_TCP_TIMEOUT); 427*b7579f77SDag-Erling Smørgrav memcpy(&n->c->repinfo.addr, &addr, addrlen); 428*b7579f77SDag-Erling Smørgrav n->c->repinfo.addrlen = addrlen; 429*b7579f77SDag-Erling Smørgrav n->shake_state = rc_hs_read; 430*b7579f77SDag-Erling Smørgrav n->ssl = SSL_new(rc->ctx); 431*b7579f77SDag-Erling Smørgrav if(!n->ssl) { 432*b7579f77SDag-Erling Smørgrav log_crypto_err("could not SSL_new"); 433*b7579f77SDag-Erling Smørgrav comm_point_delete(n->c); 434*b7579f77SDag-Erling Smørgrav free(n); 435*b7579f77SDag-Erling Smørgrav goto close_exit; 436*b7579f77SDag-Erling Smørgrav } 437*b7579f77SDag-Erling Smørgrav SSL_set_accept_state(n->ssl); 438*b7579f77SDag-Erling Smørgrav (void)SSL_set_mode(n->ssl, SSL_MODE_AUTO_RETRY); 439*b7579f77SDag-Erling Smørgrav if(!SSL_set_fd(n->ssl, newfd)) { 440*b7579f77SDag-Erling Smørgrav log_crypto_err("could not SSL_set_fd"); 441*b7579f77SDag-Erling Smørgrav SSL_free(n->ssl); 442*b7579f77SDag-Erling Smørgrav comm_point_delete(n->c); 443*b7579f77SDag-Erling Smørgrav free(n); 444*b7579f77SDag-Erling Smørgrav goto close_exit; 445*b7579f77SDag-Erling Smørgrav } 446*b7579f77SDag-Erling Smørgrav 447*b7579f77SDag-Erling Smørgrav n->rc = rc; 448*b7579f77SDag-Erling Smørgrav n->next = rc->busy_list; 449*b7579f77SDag-Erling Smørgrav rc->busy_list = n; 450*b7579f77SDag-Erling Smørgrav rc->active ++; 451*b7579f77SDag-Erling Smørgrav 452*b7579f77SDag-Erling Smørgrav /* perform the first nonblocking read already, for windows, 453*b7579f77SDag-Erling Smørgrav * so it can return wouldblock. could be faster too. */ 454*b7579f77SDag-Erling Smørgrav (void)remote_control_callback(n->c, n, NETEVENT_NOERROR, NULL); 455*b7579f77SDag-Erling Smørgrav return 0; 456*b7579f77SDag-Erling Smørgrav } 457*b7579f77SDag-Erling Smørgrav 458*b7579f77SDag-Erling Smørgrav /** delete from list */ 459*b7579f77SDag-Erling Smørgrav static void 460*b7579f77SDag-Erling Smørgrav state_list_remove_elem(struct rc_state** list, struct comm_point* c) 461*b7579f77SDag-Erling Smørgrav { 462*b7579f77SDag-Erling Smørgrav while(*list) { 463*b7579f77SDag-Erling Smørgrav if( (*list)->c == c) { 464*b7579f77SDag-Erling Smørgrav *list = (*list)->next; 465*b7579f77SDag-Erling Smørgrav return; 466*b7579f77SDag-Erling Smørgrav } 467*b7579f77SDag-Erling Smørgrav list = &(*list)->next; 468*b7579f77SDag-Erling Smørgrav } 469*b7579f77SDag-Erling Smørgrav } 470*b7579f77SDag-Erling Smørgrav 471*b7579f77SDag-Erling Smørgrav /** decrease active count and remove commpoint from busy list */ 472*b7579f77SDag-Erling Smørgrav static void 473*b7579f77SDag-Erling Smørgrav clean_point(struct daemon_remote* rc, struct rc_state* s) 474*b7579f77SDag-Erling Smørgrav { 475*b7579f77SDag-Erling Smørgrav state_list_remove_elem(&rc->busy_list, s->c); 476*b7579f77SDag-Erling Smørgrav rc->active --; 477*b7579f77SDag-Erling Smørgrav if(s->ssl) { 478*b7579f77SDag-Erling Smørgrav SSL_shutdown(s->ssl); 479*b7579f77SDag-Erling Smørgrav SSL_free(s->ssl); 480*b7579f77SDag-Erling Smørgrav } 481*b7579f77SDag-Erling Smørgrav comm_point_delete(s->c); 482*b7579f77SDag-Erling Smørgrav free(s); 483*b7579f77SDag-Erling Smørgrav } 484*b7579f77SDag-Erling Smørgrav 485*b7579f77SDag-Erling Smørgrav int 486*b7579f77SDag-Erling Smørgrav ssl_print_text(SSL* ssl, const char* text) 487*b7579f77SDag-Erling Smørgrav { 488*b7579f77SDag-Erling Smørgrav int r; 489*b7579f77SDag-Erling Smørgrav if(!ssl) 490*b7579f77SDag-Erling Smørgrav return 0; 491*b7579f77SDag-Erling Smørgrav ERR_clear_error(); 492*b7579f77SDag-Erling Smørgrav if((r=SSL_write(ssl, text, (int)strlen(text))) <= 0) { 493*b7579f77SDag-Erling Smørgrav if(SSL_get_error(ssl, r) == SSL_ERROR_ZERO_RETURN) { 494*b7579f77SDag-Erling Smørgrav verbose(VERB_QUERY, "warning, in SSL_write, peer " 495*b7579f77SDag-Erling Smørgrav "closed connection"); 496*b7579f77SDag-Erling Smørgrav return 0; 497*b7579f77SDag-Erling Smørgrav } 498*b7579f77SDag-Erling Smørgrav log_crypto_err("could not SSL_write"); 499*b7579f77SDag-Erling Smørgrav return 0; 500*b7579f77SDag-Erling Smørgrav } 501*b7579f77SDag-Erling Smørgrav return 1; 502*b7579f77SDag-Erling Smørgrav } 503*b7579f77SDag-Erling Smørgrav 504*b7579f77SDag-Erling Smørgrav /** print text over the ssl connection */ 505*b7579f77SDag-Erling Smørgrav static int 506*b7579f77SDag-Erling Smørgrav ssl_print_vmsg(SSL* ssl, const char* format, va_list args) 507*b7579f77SDag-Erling Smørgrav { 508*b7579f77SDag-Erling Smørgrav char msg[1024]; 509*b7579f77SDag-Erling Smørgrav vsnprintf(msg, sizeof(msg), format, args); 510*b7579f77SDag-Erling Smørgrav return ssl_print_text(ssl, msg); 511*b7579f77SDag-Erling Smørgrav } 512*b7579f77SDag-Erling Smørgrav 513*b7579f77SDag-Erling Smørgrav /** printf style printing to the ssl connection */ 514*b7579f77SDag-Erling Smørgrav int ssl_printf(SSL* ssl, const char* format, ...) 515*b7579f77SDag-Erling Smørgrav { 516*b7579f77SDag-Erling Smørgrav va_list args; 517*b7579f77SDag-Erling Smørgrav int ret; 518*b7579f77SDag-Erling Smørgrav va_start(args, format); 519*b7579f77SDag-Erling Smørgrav ret = ssl_print_vmsg(ssl, format, args); 520*b7579f77SDag-Erling Smørgrav va_end(args); 521*b7579f77SDag-Erling Smørgrav return ret; 522*b7579f77SDag-Erling Smørgrav } 523*b7579f77SDag-Erling Smørgrav 524*b7579f77SDag-Erling Smørgrav int 525*b7579f77SDag-Erling Smørgrav ssl_read_line(SSL* ssl, char* buf, size_t max) 526*b7579f77SDag-Erling Smørgrav { 527*b7579f77SDag-Erling Smørgrav int r; 528*b7579f77SDag-Erling Smørgrav size_t len = 0; 529*b7579f77SDag-Erling Smørgrav if(!ssl) 530*b7579f77SDag-Erling Smørgrav return 0; 531*b7579f77SDag-Erling Smørgrav while(len < max) { 532*b7579f77SDag-Erling Smørgrav ERR_clear_error(); 533*b7579f77SDag-Erling Smørgrav if((r=SSL_read(ssl, buf+len, 1)) <= 0) { 534*b7579f77SDag-Erling Smørgrav if(SSL_get_error(ssl, r) == SSL_ERROR_ZERO_RETURN) { 535*b7579f77SDag-Erling Smørgrav buf[len] = 0; 536*b7579f77SDag-Erling Smørgrav return 1; 537*b7579f77SDag-Erling Smørgrav } 538*b7579f77SDag-Erling Smørgrav log_crypto_err("could not SSL_read"); 539*b7579f77SDag-Erling Smørgrav return 0; 540*b7579f77SDag-Erling Smørgrav } 541*b7579f77SDag-Erling Smørgrav if(buf[len] == '\n') { 542*b7579f77SDag-Erling Smørgrav /* return string without \n */ 543*b7579f77SDag-Erling Smørgrav buf[len] = 0; 544*b7579f77SDag-Erling Smørgrav return 1; 545*b7579f77SDag-Erling Smørgrav } 546*b7579f77SDag-Erling Smørgrav len++; 547*b7579f77SDag-Erling Smørgrav } 548*b7579f77SDag-Erling Smørgrav buf[max-1] = 0; 549*b7579f77SDag-Erling Smørgrav log_err("control line too long (%d): %s", (int)max, buf); 550*b7579f77SDag-Erling Smørgrav return 0; 551*b7579f77SDag-Erling Smørgrav } 552*b7579f77SDag-Erling Smørgrav 553*b7579f77SDag-Erling Smørgrav /** skip whitespace, return new pointer into string */ 554*b7579f77SDag-Erling Smørgrav static char* 555*b7579f77SDag-Erling Smørgrav skipwhite(char* str) 556*b7579f77SDag-Erling Smørgrav { 557*b7579f77SDag-Erling Smørgrav /* EOS \0 is not a space */ 558*b7579f77SDag-Erling Smørgrav while( isspace(*str) ) 559*b7579f77SDag-Erling Smørgrav str++; 560*b7579f77SDag-Erling Smørgrav return str; 561*b7579f77SDag-Erling Smørgrav } 562*b7579f77SDag-Erling Smørgrav 563*b7579f77SDag-Erling Smørgrav /** send the OK to the control client */ 564*b7579f77SDag-Erling Smørgrav static void send_ok(SSL* ssl) 565*b7579f77SDag-Erling Smørgrav { 566*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "ok\n"); 567*b7579f77SDag-Erling Smørgrav } 568*b7579f77SDag-Erling Smørgrav 569*b7579f77SDag-Erling Smørgrav /** do the stop command */ 570*b7579f77SDag-Erling Smørgrav static void 571*b7579f77SDag-Erling Smørgrav do_stop(SSL* ssl, struct daemon_remote* rc) 572*b7579f77SDag-Erling Smørgrav { 573*b7579f77SDag-Erling Smørgrav rc->worker->need_to_exit = 1; 574*b7579f77SDag-Erling Smørgrav comm_base_exit(rc->worker->base); 575*b7579f77SDag-Erling Smørgrav send_ok(ssl); 576*b7579f77SDag-Erling Smørgrav } 577*b7579f77SDag-Erling Smørgrav 578*b7579f77SDag-Erling Smørgrav /** do the reload command */ 579*b7579f77SDag-Erling Smørgrav static void 580*b7579f77SDag-Erling Smørgrav do_reload(SSL* ssl, struct daemon_remote* rc) 581*b7579f77SDag-Erling Smørgrav { 582*b7579f77SDag-Erling Smørgrav rc->worker->need_to_exit = 0; 583*b7579f77SDag-Erling Smørgrav comm_base_exit(rc->worker->base); 584*b7579f77SDag-Erling Smørgrav send_ok(ssl); 585*b7579f77SDag-Erling Smørgrav } 586*b7579f77SDag-Erling Smørgrav 587*b7579f77SDag-Erling Smørgrav /** do the verbosity command */ 588*b7579f77SDag-Erling Smørgrav static void 589*b7579f77SDag-Erling Smørgrav do_verbosity(SSL* ssl, char* str) 590*b7579f77SDag-Erling Smørgrav { 591*b7579f77SDag-Erling Smørgrav int val = atoi(str); 592*b7579f77SDag-Erling Smørgrav if(val == 0 && strcmp(str, "0") != 0) { 593*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error in verbosity number syntax: %s\n", str); 594*b7579f77SDag-Erling Smørgrav return; 595*b7579f77SDag-Erling Smørgrav } 596*b7579f77SDag-Erling Smørgrav verbosity = val; 597*b7579f77SDag-Erling Smørgrav send_ok(ssl); 598*b7579f77SDag-Erling Smørgrav } 599*b7579f77SDag-Erling Smørgrav 600*b7579f77SDag-Erling Smørgrav /** print stats from statinfo */ 601*b7579f77SDag-Erling Smørgrav static int 602*b7579f77SDag-Erling Smørgrav print_stats(SSL* ssl, const char* nm, struct stats_info* s) 603*b7579f77SDag-Erling Smørgrav { 604*b7579f77SDag-Erling Smørgrav struct timeval avg; 605*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.num.queries"SQ"%u\n", nm, 606*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.num_queries)) return 0; 607*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.num.cachehits"SQ"%u\n", nm, 608*b7579f77SDag-Erling Smørgrav (unsigned)(s->svr.num_queries 609*b7579f77SDag-Erling Smørgrav - s->svr.num_queries_missed_cache))) return 0; 610*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.num.cachemiss"SQ"%u\n", nm, 611*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.num_queries_missed_cache)) return 0; 612*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.num.prefetch"SQ"%u\n", nm, 613*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.num_queries_prefetch)) return 0; 614*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.num.recursivereplies"SQ"%u\n", nm, 615*b7579f77SDag-Erling Smørgrav (unsigned)s->mesh_replies_sent)) return 0; 616*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.requestlist.avg"SQ"%g\n", nm, 617*b7579f77SDag-Erling Smørgrav (s->svr.num_queries_missed_cache+s->svr.num_queries_prefetch)? 618*b7579f77SDag-Erling Smørgrav (double)s->svr.sum_query_list_size/ 619*b7579f77SDag-Erling Smørgrav (s->svr.num_queries_missed_cache+ 620*b7579f77SDag-Erling Smørgrav s->svr.num_queries_prefetch) : 0.0)) return 0; 621*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.requestlist.max"SQ"%u\n", nm, 622*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.max_query_list_size)) return 0; 623*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.requestlist.overwritten"SQ"%u\n", nm, 624*b7579f77SDag-Erling Smørgrav (unsigned)s->mesh_jostled)) return 0; 625*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.requestlist.exceeded"SQ"%u\n", nm, 626*b7579f77SDag-Erling Smørgrav (unsigned)s->mesh_dropped)) return 0; 627*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.requestlist.current.all"SQ"%u\n", nm, 628*b7579f77SDag-Erling Smørgrav (unsigned)s->mesh_num_states)) return 0; 629*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.requestlist.current.user"SQ"%u\n", nm, 630*b7579f77SDag-Erling Smørgrav (unsigned)s->mesh_num_reply_states)) return 0; 631*b7579f77SDag-Erling Smørgrav timeval_divide(&avg, &s->mesh_replies_sum_wait, s->mesh_replies_sent); 632*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.recursion.time.avg"SQ"%d.%6.6d\n", nm, 633*b7579f77SDag-Erling Smørgrav (int)avg.tv_sec, (int)avg.tv_usec)) return 0; 634*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s.recursion.time.median"SQ"%g\n", nm, 635*b7579f77SDag-Erling Smørgrav s->mesh_time_median)) return 0; 636*b7579f77SDag-Erling Smørgrav return 1; 637*b7579f77SDag-Erling Smørgrav } 638*b7579f77SDag-Erling Smørgrav 639*b7579f77SDag-Erling Smørgrav /** print stats for one thread */ 640*b7579f77SDag-Erling Smørgrav static int 641*b7579f77SDag-Erling Smørgrav print_thread_stats(SSL* ssl, int i, struct stats_info* s) 642*b7579f77SDag-Erling Smørgrav { 643*b7579f77SDag-Erling Smørgrav char nm[16]; 644*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "thread%d", i); 645*b7579f77SDag-Erling Smørgrav nm[sizeof(nm)-1]=0; 646*b7579f77SDag-Erling Smørgrav return print_stats(ssl, nm, s); 647*b7579f77SDag-Erling Smørgrav } 648*b7579f77SDag-Erling Smørgrav 649*b7579f77SDag-Erling Smørgrav /** print long number */ 650*b7579f77SDag-Erling Smørgrav static int 651*b7579f77SDag-Erling Smørgrav print_longnum(SSL* ssl, char* desc, size_t x) 652*b7579f77SDag-Erling Smørgrav { 653*b7579f77SDag-Erling Smørgrav if(x > 1024*1024*1024) { 654*b7579f77SDag-Erling Smørgrav /* more than a Gb */ 655*b7579f77SDag-Erling Smørgrav size_t front = x / (size_t)1000000; 656*b7579f77SDag-Erling Smørgrav size_t back = x % (size_t)1000000; 657*b7579f77SDag-Erling Smørgrav return ssl_printf(ssl, "%s%u%6.6u\n", desc, 658*b7579f77SDag-Erling Smørgrav (unsigned)front, (unsigned)back); 659*b7579f77SDag-Erling Smørgrav } else { 660*b7579f77SDag-Erling Smørgrav return ssl_printf(ssl, "%s%u\n", desc, (unsigned)x); 661*b7579f77SDag-Erling Smørgrav } 662*b7579f77SDag-Erling Smørgrav } 663*b7579f77SDag-Erling Smørgrav 664*b7579f77SDag-Erling Smørgrav /** print mem stats */ 665*b7579f77SDag-Erling Smørgrav static int 666*b7579f77SDag-Erling Smørgrav print_mem(SSL* ssl, struct worker* worker, struct daemon* daemon) 667*b7579f77SDag-Erling Smørgrav { 668*b7579f77SDag-Erling Smørgrav int m; 669*b7579f77SDag-Erling Smørgrav size_t msg, rrset, val, iter; 670*b7579f77SDag-Erling Smørgrav #ifdef HAVE_SBRK 671*b7579f77SDag-Erling Smørgrav extern void* unbound_start_brk; 672*b7579f77SDag-Erling Smørgrav void* cur = sbrk(0); 673*b7579f77SDag-Erling Smørgrav if(!print_longnum(ssl, "mem.total.sbrk"SQ, 674*b7579f77SDag-Erling Smørgrav (size_t)((char*)cur - (char*)unbound_start_brk))) return 0; 675*b7579f77SDag-Erling Smørgrav #endif /* HAVE_SBRK */ 676*b7579f77SDag-Erling Smørgrav msg = slabhash_get_mem(daemon->env->msg_cache); 677*b7579f77SDag-Erling Smørgrav rrset = slabhash_get_mem(&daemon->env->rrset_cache->table); 678*b7579f77SDag-Erling Smørgrav val=0; 679*b7579f77SDag-Erling Smørgrav iter=0; 680*b7579f77SDag-Erling Smørgrav m = modstack_find(&worker->env.mesh->mods, "validator"); 681*b7579f77SDag-Erling Smørgrav if(m != -1) { 682*b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mod_get_mem(worker->env.mesh-> 683*b7579f77SDag-Erling Smørgrav mods.mod[m]->get_mem)); 684*b7579f77SDag-Erling Smørgrav val = (*worker->env.mesh->mods.mod[m]->get_mem) 685*b7579f77SDag-Erling Smørgrav (&worker->env, m); 686*b7579f77SDag-Erling Smørgrav } 687*b7579f77SDag-Erling Smørgrav m = modstack_find(&worker->env.mesh->mods, "iterator"); 688*b7579f77SDag-Erling Smørgrav if(m != -1) { 689*b7579f77SDag-Erling Smørgrav fptr_ok(fptr_whitelist_mod_get_mem(worker->env.mesh-> 690*b7579f77SDag-Erling Smørgrav mods.mod[m]->get_mem)); 691*b7579f77SDag-Erling Smørgrav iter = (*worker->env.mesh->mods.mod[m]->get_mem) 692*b7579f77SDag-Erling Smørgrav (&worker->env, m); 693*b7579f77SDag-Erling Smørgrav } 694*b7579f77SDag-Erling Smørgrav 695*b7579f77SDag-Erling Smørgrav if(!print_longnum(ssl, "mem.cache.rrset"SQ, rrset)) 696*b7579f77SDag-Erling Smørgrav return 0; 697*b7579f77SDag-Erling Smørgrav if(!print_longnum(ssl, "mem.cache.message"SQ, msg)) 698*b7579f77SDag-Erling Smørgrav return 0; 699*b7579f77SDag-Erling Smørgrav if(!print_longnum(ssl, "mem.mod.iterator"SQ, iter)) 700*b7579f77SDag-Erling Smørgrav return 0; 701*b7579f77SDag-Erling Smørgrav if(!print_longnum(ssl, "mem.mod.validator"SQ, val)) 702*b7579f77SDag-Erling Smørgrav return 0; 703*b7579f77SDag-Erling Smørgrav return 1; 704*b7579f77SDag-Erling Smørgrav } 705*b7579f77SDag-Erling Smørgrav 706*b7579f77SDag-Erling Smørgrav /** print uptime stats */ 707*b7579f77SDag-Erling Smørgrav static int 708*b7579f77SDag-Erling Smørgrav print_uptime(SSL* ssl, struct worker* worker, int reset) 709*b7579f77SDag-Erling Smørgrav { 710*b7579f77SDag-Erling Smørgrav struct timeval now = *worker->env.now_tv; 711*b7579f77SDag-Erling Smørgrav struct timeval up, dt; 712*b7579f77SDag-Erling Smørgrav timeval_subtract(&up, &now, &worker->daemon->time_boot); 713*b7579f77SDag-Erling Smørgrav timeval_subtract(&dt, &now, &worker->daemon->time_last_stat); 714*b7579f77SDag-Erling Smørgrav if(reset) 715*b7579f77SDag-Erling Smørgrav worker->daemon->time_last_stat = now; 716*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "time.now"SQ"%d.%6.6d\n", 717*b7579f77SDag-Erling Smørgrav (unsigned)now.tv_sec, (unsigned)now.tv_usec)) return 0; 718*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "time.up"SQ"%d.%6.6d\n", 719*b7579f77SDag-Erling Smørgrav (unsigned)up.tv_sec, (unsigned)up.tv_usec)) return 0; 720*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "time.elapsed"SQ"%d.%6.6d\n", 721*b7579f77SDag-Erling Smørgrav (unsigned)dt.tv_sec, (unsigned)dt.tv_usec)) return 0; 722*b7579f77SDag-Erling Smørgrav return 1; 723*b7579f77SDag-Erling Smørgrav } 724*b7579f77SDag-Erling Smørgrav 725*b7579f77SDag-Erling Smørgrav /** print extended histogram */ 726*b7579f77SDag-Erling Smørgrav static int 727*b7579f77SDag-Erling Smørgrav print_hist(SSL* ssl, struct stats_info* s) 728*b7579f77SDag-Erling Smørgrav { 729*b7579f77SDag-Erling Smørgrav struct timehist* hist; 730*b7579f77SDag-Erling Smørgrav size_t i; 731*b7579f77SDag-Erling Smørgrav hist = timehist_setup(); 732*b7579f77SDag-Erling Smørgrav if(!hist) { 733*b7579f77SDag-Erling Smørgrav log_err("out of memory"); 734*b7579f77SDag-Erling Smørgrav return 0; 735*b7579f77SDag-Erling Smørgrav } 736*b7579f77SDag-Erling Smørgrav timehist_import(hist, s->svr.hist, NUM_BUCKETS_HIST); 737*b7579f77SDag-Erling Smørgrav for(i=0; i<hist->num; i++) { 738*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, 739*b7579f77SDag-Erling Smørgrav "histogram.%6.6d.%6.6d.to.%6.6d.%6.6d=%u\n", 740*b7579f77SDag-Erling Smørgrav (int)hist->buckets[i].lower.tv_sec, 741*b7579f77SDag-Erling Smørgrav (int)hist->buckets[i].lower.tv_usec, 742*b7579f77SDag-Erling Smørgrav (int)hist->buckets[i].upper.tv_sec, 743*b7579f77SDag-Erling Smørgrav (int)hist->buckets[i].upper.tv_usec, 744*b7579f77SDag-Erling Smørgrav (unsigned)hist->buckets[i].count)) { 745*b7579f77SDag-Erling Smørgrav timehist_delete(hist); 746*b7579f77SDag-Erling Smørgrav return 0; 747*b7579f77SDag-Erling Smørgrav } 748*b7579f77SDag-Erling Smørgrav } 749*b7579f77SDag-Erling Smørgrav timehist_delete(hist); 750*b7579f77SDag-Erling Smørgrav return 1; 751*b7579f77SDag-Erling Smørgrav } 752*b7579f77SDag-Erling Smørgrav 753*b7579f77SDag-Erling Smørgrav /** print extended stats */ 754*b7579f77SDag-Erling Smørgrav static int 755*b7579f77SDag-Erling Smørgrav print_ext(SSL* ssl, struct stats_info* s) 756*b7579f77SDag-Erling Smørgrav { 757*b7579f77SDag-Erling Smørgrav int i; 758*b7579f77SDag-Erling Smørgrav char nm[16]; 759*b7579f77SDag-Erling Smørgrav const ldns_rr_descriptor* desc; 760*b7579f77SDag-Erling Smørgrav const ldns_lookup_table* lt; 761*b7579f77SDag-Erling Smørgrav /* TYPE */ 762*b7579f77SDag-Erling Smørgrav for(i=0; i<STATS_QTYPE_NUM; i++) { 763*b7579f77SDag-Erling Smørgrav if(inhibit_zero && s->svr.qtype[i] == 0) 764*b7579f77SDag-Erling Smørgrav continue; 765*b7579f77SDag-Erling Smørgrav desc = ldns_rr_descript((uint16_t)i); 766*b7579f77SDag-Erling Smørgrav if(desc && desc->_name) { 767*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "%s", desc->_name); 768*b7579f77SDag-Erling Smørgrav } else if (i == LDNS_RR_TYPE_IXFR) { 769*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "IXFR"); 770*b7579f77SDag-Erling Smørgrav } else if (i == LDNS_RR_TYPE_AXFR) { 771*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "AXFR"); 772*b7579f77SDag-Erling Smørgrav } else if (i == LDNS_RR_TYPE_MAILA) { 773*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "MAILA"); 774*b7579f77SDag-Erling Smørgrav } else if (i == LDNS_RR_TYPE_MAILB) { 775*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "MAILB"); 776*b7579f77SDag-Erling Smørgrav } else if (i == LDNS_RR_TYPE_ANY) { 777*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "ANY"); 778*b7579f77SDag-Erling Smørgrav } else { 779*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "TYPE%d", i); 780*b7579f77SDag-Erling Smørgrav } 781*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.type.%s"SQ"%u\n", 782*b7579f77SDag-Erling Smørgrav nm, (unsigned)s->svr.qtype[i])) return 0; 783*b7579f77SDag-Erling Smørgrav } 784*b7579f77SDag-Erling Smørgrav if(!inhibit_zero || s->svr.qtype_big) { 785*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.type.other"SQ"%u\n", 786*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qtype_big)) return 0; 787*b7579f77SDag-Erling Smørgrav } 788*b7579f77SDag-Erling Smørgrav /* CLASS */ 789*b7579f77SDag-Erling Smørgrav for(i=0; i<STATS_QCLASS_NUM; i++) { 790*b7579f77SDag-Erling Smørgrav if(inhibit_zero && s->svr.qclass[i] == 0) 791*b7579f77SDag-Erling Smørgrav continue; 792*b7579f77SDag-Erling Smørgrav lt = ldns_lookup_by_id(ldns_rr_classes, i); 793*b7579f77SDag-Erling Smørgrav if(lt && lt->name) { 794*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "%s", lt->name); 795*b7579f77SDag-Erling Smørgrav } else { 796*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "CLASS%d", i); 797*b7579f77SDag-Erling Smørgrav } 798*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.class.%s"SQ"%u\n", 799*b7579f77SDag-Erling Smørgrav nm, (unsigned)s->svr.qclass[i])) return 0; 800*b7579f77SDag-Erling Smørgrav } 801*b7579f77SDag-Erling Smørgrav if(!inhibit_zero || s->svr.qclass_big) { 802*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.class.other"SQ"%u\n", 803*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qclass_big)) return 0; 804*b7579f77SDag-Erling Smørgrav } 805*b7579f77SDag-Erling Smørgrav /* OPCODE */ 806*b7579f77SDag-Erling Smørgrav for(i=0; i<STATS_OPCODE_NUM; i++) { 807*b7579f77SDag-Erling Smørgrav if(inhibit_zero && s->svr.qopcode[i] == 0) 808*b7579f77SDag-Erling Smørgrav continue; 809*b7579f77SDag-Erling Smørgrav lt = ldns_lookup_by_id(ldns_opcodes, i); 810*b7579f77SDag-Erling Smørgrav if(lt && lt->name) { 811*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "%s", lt->name); 812*b7579f77SDag-Erling Smørgrav } else { 813*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "OPCODE%d", i); 814*b7579f77SDag-Erling Smørgrav } 815*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.opcode.%s"SQ"%u\n", 816*b7579f77SDag-Erling Smørgrav nm, (unsigned)s->svr.qopcode[i])) return 0; 817*b7579f77SDag-Erling Smørgrav } 818*b7579f77SDag-Erling Smørgrav /* transport */ 819*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.tcp"SQ"%u\n", 820*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qtcp)) return 0; 821*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.ipv6"SQ"%u\n", 822*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qipv6)) return 0; 823*b7579f77SDag-Erling Smørgrav /* flags */ 824*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.QR"SQ"%u\n", 825*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_QR)) return 0; 826*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.AA"SQ"%u\n", 827*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_AA)) return 0; 828*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.TC"SQ"%u\n", 829*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_TC)) return 0; 830*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.RD"SQ"%u\n", 831*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_RD)) return 0; 832*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.RA"SQ"%u\n", 833*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_RA)) return 0; 834*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.Z"SQ"%u\n", 835*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_Z)) return 0; 836*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.AD"SQ"%u\n", 837*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_AD)) return 0; 838*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.flags.CD"SQ"%u\n", 839*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qbit_CD)) return 0; 840*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.edns.present"SQ"%u\n", 841*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qEDNS)) return 0; 842*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.query.edns.DO"SQ"%u\n", 843*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.qEDNS_DO)) return 0; 844*b7579f77SDag-Erling Smørgrav 845*b7579f77SDag-Erling Smørgrav /* RCODE */ 846*b7579f77SDag-Erling Smørgrav for(i=0; i<STATS_RCODE_NUM; i++) { 847*b7579f77SDag-Erling Smørgrav if(inhibit_zero && s->svr.ans_rcode[i] == 0) 848*b7579f77SDag-Erling Smørgrav continue; 849*b7579f77SDag-Erling Smørgrav lt = ldns_lookup_by_id(ldns_rcodes, i); 850*b7579f77SDag-Erling Smørgrav if(lt && lt->name) { 851*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "%s", lt->name); 852*b7579f77SDag-Erling Smørgrav } else { 853*b7579f77SDag-Erling Smørgrav snprintf(nm, sizeof(nm), "RCODE%d", i); 854*b7579f77SDag-Erling Smørgrav } 855*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.answer.rcode.%s"SQ"%u\n", 856*b7579f77SDag-Erling Smørgrav nm, (unsigned)s->svr.ans_rcode[i])) return 0; 857*b7579f77SDag-Erling Smørgrav } 858*b7579f77SDag-Erling Smørgrav if(!inhibit_zero || s->svr.ans_rcode_nodata) { 859*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.answer.rcode.nodata"SQ"%u\n", 860*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.ans_rcode_nodata)) return 0; 861*b7579f77SDag-Erling Smørgrav } 862*b7579f77SDag-Erling Smørgrav /* validation */ 863*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.answer.secure"SQ"%u\n", 864*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.ans_secure)) return 0; 865*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.answer.bogus"SQ"%u\n", 866*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.ans_bogus)) return 0; 867*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "num.rrset.bogus"SQ"%u\n", 868*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.rrset_bogus)) return 0; 869*b7579f77SDag-Erling Smørgrav /* threat detection */ 870*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "unwanted.queries"SQ"%u\n", 871*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.unwanted_queries)) return 0; 872*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "unwanted.replies"SQ"%u\n", 873*b7579f77SDag-Erling Smørgrav (unsigned)s->svr.unwanted_replies)) return 0; 874*b7579f77SDag-Erling Smørgrav return 1; 875*b7579f77SDag-Erling Smørgrav } 876*b7579f77SDag-Erling Smørgrav 877*b7579f77SDag-Erling Smørgrav /** do the stats command */ 878*b7579f77SDag-Erling Smørgrav static void 879*b7579f77SDag-Erling Smørgrav do_stats(SSL* ssl, struct daemon_remote* rc, int reset) 880*b7579f77SDag-Erling Smørgrav { 881*b7579f77SDag-Erling Smørgrav struct daemon* daemon = rc->worker->daemon; 882*b7579f77SDag-Erling Smørgrav struct stats_info total; 883*b7579f77SDag-Erling Smørgrav struct stats_info s; 884*b7579f77SDag-Erling Smørgrav int i; 885*b7579f77SDag-Erling Smørgrav log_assert(daemon->num > 0); 886*b7579f77SDag-Erling Smørgrav /* gather all thread statistics in one place */ 887*b7579f77SDag-Erling Smørgrav for(i=0; i<daemon->num; i++) { 888*b7579f77SDag-Erling Smørgrav server_stats_obtain(rc->worker, daemon->workers[i], &s, reset); 889*b7579f77SDag-Erling Smørgrav if(!print_thread_stats(ssl, i, &s)) 890*b7579f77SDag-Erling Smørgrav return; 891*b7579f77SDag-Erling Smørgrav if(i == 0) 892*b7579f77SDag-Erling Smørgrav total = s; 893*b7579f77SDag-Erling Smørgrav else server_stats_add(&total, &s); 894*b7579f77SDag-Erling Smørgrav } 895*b7579f77SDag-Erling Smørgrav /* print the thread statistics */ 896*b7579f77SDag-Erling Smørgrav total.mesh_time_median /= (double)daemon->num; 897*b7579f77SDag-Erling Smørgrav if(!print_stats(ssl, "total", &total)) 898*b7579f77SDag-Erling Smørgrav return; 899*b7579f77SDag-Erling Smørgrav if(!print_uptime(ssl, rc->worker, reset)) 900*b7579f77SDag-Erling Smørgrav return; 901*b7579f77SDag-Erling Smørgrav if(daemon->cfg->stat_extended) { 902*b7579f77SDag-Erling Smørgrav if(!print_mem(ssl, rc->worker, daemon)) 903*b7579f77SDag-Erling Smørgrav return; 904*b7579f77SDag-Erling Smørgrav if(!print_hist(ssl, &total)) 905*b7579f77SDag-Erling Smørgrav return; 906*b7579f77SDag-Erling Smørgrav if(!print_ext(ssl, &total)) 907*b7579f77SDag-Erling Smørgrav return; 908*b7579f77SDag-Erling Smørgrav } 909*b7579f77SDag-Erling Smørgrav } 910*b7579f77SDag-Erling Smørgrav 911*b7579f77SDag-Erling Smørgrav /** parse commandline argument domain name */ 912*b7579f77SDag-Erling Smørgrav static int 913*b7579f77SDag-Erling Smørgrav parse_arg_name(SSL* ssl, char* str, uint8_t** res, size_t* len, int* labs) 914*b7579f77SDag-Erling Smørgrav { 915*b7579f77SDag-Erling Smørgrav ldns_rdf* rdf; 916*b7579f77SDag-Erling Smørgrav *res = NULL; 917*b7579f77SDag-Erling Smørgrav *len = 0; 918*b7579f77SDag-Erling Smørgrav *labs = 0; 919*b7579f77SDag-Erling Smørgrav rdf = ldns_dname_new_frm_str(str); 920*b7579f77SDag-Erling Smørgrav if(!rdf) { 921*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error cannot parse name %s\n", str); 922*b7579f77SDag-Erling Smørgrav return 0; 923*b7579f77SDag-Erling Smørgrav } 924*b7579f77SDag-Erling Smørgrav *res = memdup(ldns_rdf_data(rdf), ldns_rdf_size(rdf)); 925*b7579f77SDag-Erling Smørgrav ldns_rdf_deep_free(rdf); 926*b7579f77SDag-Erling Smørgrav if(!*res) { 927*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error out of memory\n"); 928*b7579f77SDag-Erling Smørgrav return 0; 929*b7579f77SDag-Erling Smørgrav } 930*b7579f77SDag-Erling Smørgrav *labs = dname_count_size_labels(*res, len); 931*b7579f77SDag-Erling Smørgrav return 1; 932*b7579f77SDag-Erling Smørgrav } 933*b7579f77SDag-Erling Smørgrav 934*b7579f77SDag-Erling Smørgrav /** find second argument, modifies string */ 935*b7579f77SDag-Erling Smørgrav static int 936*b7579f77SDag-Erling Smørgrav find_arg2(SSL* ssl, char* arg, char** arg2) 937*b7579f77SDag-Erling Smørgrav { 938*b7579f77SDag-Erling Smørgrav char* as = strchr(arg, ' '); 939*b7579f77SDag-Erling Smørgrav char* at = strchr(arg, '\t'); 940*b7579f77SDag-Erling Smørgrav if(as && at) { 941*b7579f77SDag-Erling Smørgrav if(at < as) 942*b7579f77SDag-Erling Smørgrav as = at; 943*b7579f77SDag-Erling Smørgrav as[0]=0; 944*b7579f77SDag-Erling Smørgrav *arg2 = skipwhite(as+1); 945*b7579f77SDag-Erling Smørgrav } else if(as) { 946*b7579f77SDag-Erling Smørgrav as[0]=0; 947*b7579f77SDag-Erling Smørgrav *arg2 = skipwhite(as+1); 948*b7579f77SDag-Erling Smørgrav } else if(at) { 949*b7579f77SDag-Erling Smørgrav at[0]=0; 950*b7579f77SDag-Erling Smørgrav *arg2 = skipwhite(at+1); 951*b7579f77SDag-Erling Smørgrav } else { 952*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error could not find next argument " 953*b7579f77SDag-Erling Smørgrav "after %s\n", arg); 954*b7579f77SDag-Erling Smørgrav return 0; 955*b7579f77SDag-Erling Smørgrav } 956*b7579f77SDag-Erling Smørgrav return 1; 957*b7579f77SDag-Erling Smørgrav } 958*b7579f77SDag-Erling Smørgrav 959*b7579f77SDag-Erling Smørgrav /** Add a new zone */ 960*b7579f77SDag-Erling Smørgrav static void 961*b7579f77SDag-Erling Smørgrav do_zone_add(SSL* ssl, struct worker* worker, char* arg) 962*b7579f77SDag-Erling Smørgrav { 963*b7579f77SDag-Erling Smørgrav uint8_t* nm; 964*b7579f77SDag-Erling Smørgrav int nmlabs; 965*b7579f77SDag-Erling Smørgrav size_t nmlen; 966*b7579f77SDag-Erling Smørgrav char* arg2; 967*b7579f77SDag-Erling Smørgrav enum localzone_type t; 968*b7579f77SDag-Erling Smørgrav struct local_zone* z; 969*b7579f77SDag-Erling Smørgrav if(!find_arg2(ssl, arg, &arg2)) 970*b7579f77SDag-Erling Smørgrav return; 971*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 972*b7579f77SDag-Erling Smørgrav return; 973*b7579f77SDag-Erling Smørgrav if(!local_zone_str2type(arg2, &t)) { 974*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error not a zone type. %s\n", arg2); 975*b7579f77SDag-Erling Smørgrav free(nm); 976*b7579f77SDag-Erling Smørgrav return; 977*b7579f77SDag-Erling Smørgrav } 978*b7579f77SDag-Erling Smørgrav lock_quick_lock(&worker->daemon->local_zones->lock); 979*b7579f77SDag-Erling Smørgrav if((z=local_zones_find(worker->daemon->local_zones, nm, nmlen, 980*b7579f77SDag-Erling Smørgrav nmlabs, LDNS_RR_CLASS_IN))) { 981*b7579f77SDag-Erling Smørgrav /* already present in tree */ 982*b7579f77SDag-Erling Smørgrav lock_rw_wrlock(&z->lock); 983*b7579f77SDag-Erling Smørgrav z->type = t; /* update type anyway */ 984*b7579f77SDag-Erling Smørgrav lock_rw_unlock(&z->lock); 985*b7579f77SDag-Erling Smørgrav free(nm); 986*b7579f77SDag-Erling Smørgrav lock_quick_unlock(&worker->daemon->local_zones->lock); 987*b7579f77SDag-Erling Smørgrav send_ok(ssl); 988*b7579f77SDag-Erling Smørgrav return; 989*b7579f77SDag-Erling Smørgrav } 990*b7579f77SDag-Erling Smørgrav if(!local_zones_add_zone(worker->daemon->local_zones, nm, nmlen, 991*b7579f77SDag-Erling Smørgrav nmlabs, LDNS_RR_CLASS_IN, t)) { 992*b7579f77SDag-Erling Smørgrav lock_quick_unlock(&worker->daemon->local_zones->lock); 993*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error out of memory\n"); 994*b7579f77SDag-Erling Smørgrav return; 995*b7579f77SDag-Erling Smørgrav } 996*b7579f77SDag-Erling Smørgrav lock_quick_unlock(&worker->daemon->local_zones->lock); 997*b7579f77SDag-Erling Smørgrav send_ok(ssl); 998*b7579f77SDag-Erling Smørgrav } 999*b7579f77SDag-Erling Smørgrav 1000*b7579f77SDag-Erling Smørgrav /** Remove a zone */ 1001*b7579f77SDag-Erling Smørgrav static void 1002*b7579f77SDag-Erling Smørgrav do_zone_remove(SSL* ssl, struct worker* worker, char* arg) 1003*b7579f77SDag-Erling Smørgrav { 1004*b7579f77SDag-Erling Smørgrav uint8_t* nm; 1005*b7579f77SDag-Erling Smørgrav int nmlabs; 1006*b7579f77SDag-Erling Smørgrav size_t nmlen; 1007*b7579f77SDag-Erling Smørgrav struct local_zone* z; 1008*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 1009*b7579f77SDag-Erling Smørgrav return; 1010*b7579f77SDag-Erling Smørgrav lock_quick_lock(&worker->daemon->local_zones->lock); 1011*b7579f77SDag-Erling Smørgrav if((z=local_zones_find(worker->daemon->local_zones, nm, nmlen, 1012*b7579f77SDag-Erling Smørgrav nmlabs, LDNS_RR_CLASS_IN))) { 1013*b7579f77SDag-Erling Smørgrav /* present in tree */ 1014*b7579f77SDag-Erling Smørgrav local_zones_del_zone(worker->daemon->local_zones, z); 1015*b7579f77SDag-Erling Smørgrav } 1016*b7579f77SDag-Erling Smørgrav lock_quick_unlock(&worker->daemon->local_zones->lock); 1017*b7579f77SDag-Erling Smørgrav free(nm); 1018*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1019*b7579f77SDag-Erling Smørgrav } 1020*b7579f77SDag-Erling Smørgrav 1021*b7579f77SDag-Erling Smørgrav /** Add new RR data */ 1022*b7579f77SDag-Erling Smørgrav static void 1023*b7579f77SDag-Erling Smørgrav do_data_add(SSL* ssl, struct worker* worker, char* arg) 1024*b7579f77SDag-Erling Smørgrav { 1025*b7579f77SDag-Erling Smørgrav if(!local_zones_add_RR(worker->daemon->local_zones, arg, 1026*b7579f77SDag-Erling Smørgrav worker->env.scratch_buffer)) { 1027*b7579f77SDag-Erling Smørgrav ssl_printf(ssl,"error in syntax or out of memory, %s\n", arg); 1028*b7579f77SDag-Erling Smørgrav return; 1029*b7579f77SDag-Erling Smørgrav } 1030*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1031*b7579f77SDag-Erling Smørgrav } 1032*b7579f77SDag-Erling Smørgrav 1033*b7579f77SDag-Erling Smørgrav /** Remove RR data */ 1034*b7579f77SDag-Erling Smørgrav static void 1035*b7579f77SDag-Erling Smørgrav do_data_remove(SSL* ssl, struct worker* worker, char* arg) 1036*b7579f77SDag-Erling Smørgrav { 1037*b7579f77SDag-Erling Smørgrav uint8_t* nm; 1038*b7579f77SDag-Erling Smørgrav int nmlabs; 1039*b7579f77SDag-Erling Smørgrav size_t nmlen; 1040*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 1041*b7579f77SDag-Erling Smørgrav return; 1042*b7579f77SDag-Erling Smørgrav local_zones_del_data(worker->daemon->local_zones, nm, 1043*b7579f77SDag-Erling Smørgrav nmlen, nmlabs, LDNS_RR_CLASS_IN); 1044*b7579f77SDag-Erling Smørgrav free(nm); 1045*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1046*b7579f77SDag-Erling Smørgrav } 1047*b7579f77SDag-Erling Smørgrav 1048*b7579f77SDag-Erling Smørgrav /** cache lookup of nameservers */ 1049*b7579f77SDag-Erling Smørgrav static void 1050*b7579f77SDag-Erling Smørgrav do_lookup(SSL* ssl, struct worker* worker, char* arg) 1051*b7579f77SDag-Erling Smørgrav { 1052*b7579f77SDag-Erling Smørgrav uint8_t* nm; 1053*b7579f77SDag-Erling Smørgrav int nmlabs; 1054*b7579f77SDag-Erling Smørgrav size_t nmlen; 1055*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 1056*b7579f77SDag-Erling Smørgrav return; 1057*b7579f77SDag-Erling Smørgrav (void)print_deleg_lookup(ssl, worker, nm, nmlen, nmlabs); 1058*b7579f77SDag-Erling Smørgrav free(nm); 1059*b7579f77SDag-Erling Smørgrav } 1060*b7579f77SDag-Erling Smørgrav 1061*b7579f77SDag-Erling Smørgrav /** flush something from rrset and msg caches */ 1062*b7579f77SDag-Erling Smørgrav static void 1063*b7579f77SDag-Erling Smørgrav do_cache_remove(struct worker* worker, uint8_t* nm, size_t nmlen, 1064*b7579f77SDag-Erling Smørgrav uint16_t t, uint16_t c) 1065*b7579f77SDag-Erling Smørgrav { 1066*b7579f77SDag-Erling Smørgrav hashvalue_t h; 1067*b7579f77SDag-Erling Smørgrav struct query_info k; 1068*b7579f77SDag-Erling Smørgrav rrset_cache_remove(worker->env.rrset_cache, nm, nmlen, t, c, 0); 1069*b7579f77SDag-Erling Smørgrav if(t == LDNS_RR_TYPE_SOA) 1070*b7579f77SDag-Erling Smørgrav rrset_cache_remove(worker->env.rrset_cache, nm, nmlen, t, c, 1071*b7579f77SDag-Erling Smørgrav PACKED_RRSET_SOA_NEG); 1072*b7579f77SDag-Erling Smørgrav k.qname = nm; 1073*b7579f77SDag-Erling Smørgrav k.qname_len = nmlen; 1074*b7579f77SDag-Erling Smørgrav k.qtype = t; 1075*b7579f77SDag-Erling Smørgrav k.qclass = c; 1076*b7579f77SDag-Erling Smørgrav h = query_info_hash(&k); 1077*b7579f77SDag-Erling Smørgrav slabhash_remove(worker->env.msg_cache, h, &k); 1078*b7579f77SDag-Erling Smørgrav } 1079*b7579f77SDag-Erling Smørgrav 1080*b7579f77SDag-Erling Smørgrav /** flush a type */ 1081*b7579f77SDag-Erling Smørgrav static void 1082*b7579f77SDag-Erling Smørgrav do_flush_type(SSL* ssl, struct worker* worker, char* arg) 1083*b7579f77SDag-Erling Smørgrav { 1084*b7579f77SDag-Erling Smørgrav uint8_t* nm; 1085*b7579f77SDag-Erling Smørgrav int nmlabs; 1086*b7579f77SDag-Erling Smørgrav size_t nmlen; 1087*b7579f77SDag-Erling Smørgrav char* arg2; 1088*b7579f77SDag-Erling Smørgrav uint16_t t; 1089*b7579f77SDag-Erling Smørgrav if(!find_arg2(ssl, arg, &arg2)) 1090*b7579f77SDag-Erling Smørgrav return; 1091*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 1092*b7579f77SDag-Erling Smørgrav return; 1093*b7579f77SDag-Erling Smørgrav t = ldns_get_rr_type_by_name(arg2); 1094*b7579f77SDag-Erling Smørgrav do_cache_remove(worker, nm, nmlen, t, LDNS_RR_CLASS_IN); 1095*b7579f77SDag-Erling Smørgrav 1096*b7579f77SDag-Erling Smørgrav free(nm); 1097*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1098*b7579f77SDag-Erling Smørgrav } 1099*b7579f77SDag-Erling Smørgrav 1100*b7579f77SDag-Erling Smørgrav /** flush statistics */ 1101*b7579f77SDag-Erling Smørgrav static void 1102*b7579f77SDag-Erling Smørgrav do_flush_stats(SSL* ssl, struct worker* worker) 1103*b7579f77SDag-Erling Smørgrav { 1104*b7579f77SDag-Erling Smørgrav worker_stats_clear(worker); 1105*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1106*b7579f77SDag-Erling Smørgrav } 1107*b7579f77SDag-Erling Smørgrav 1108*b7579f77SDag-Erling Smørgrav /** 1109*b7579f77SDag-Erling Smørgrav * Local info for deletion functions 1110*b7579f77SDag-Erling Smørgrav */ 1111*b7579f77SDag-Erling Smørgrav struct del_info { 1112*b7579f77SDag-Erling Smørgrav /** worker */ 1113*b7579f77SDag-Erling Smørgrav struct worker* worker; 1114*b7579f77SDag-Erling Smørgrav /** name to delete */ 1115*b7579f77SDag-Erling Smørgrav uint8_t* name; 1116*b7579f77SDag-Erling Smørgrav /** length */ 1117*b7579f77SDag-Erling Smørgrav size_t len; 1118*b7579f77SDag-Erling Smørgrav /** labels */ 1119*b7579f77SDag-Erling Smørgrav int labs; 1120*b7579f77SDag-Erling Smørgrav /** now */ 1121*b7579f77SDag-Erling Smørgrav uint32_t now; 1122*b7579f77SDag-Erling Smørgrav /** time to invalidate to */ 1123*b7579f77SDag-Erling Smørgrav uint32_t expired; 1124*b7579f77SDag-Erling Smørgrav /** number of rrsets removed */ 1125*b7579f77SDag-Erling Smørgrav size_t num_rrsets; 1126*b7579f77SDag-Erling Smørgrav /** number of msgs removed */ 1127*b7579f77SDag-Erling Smørgrav size_t num_msgs; 1128*b7579f77SDag-Erling Smørgrav /** number of key entries removed */ 1129*b7579f77SDag-Erling Smørgrav size_t num_keys; 1130*b7579f77SDag-Erling Smørgrav /** length of addr */ 1131*b7579f77SDag-Erling Smørgrav socklen_t addrlen; 1132*b7579f77SDag-Erling Smørgrav /** socket address for host deletion */ 1133*b7579f77SDag-Erling Smørgrav struct sockaddr_storage addr; 1134*b7579f77SDag-Erling Smørgrav }; 1135*b7579f77SDag-Erling Smørgrav 1136*b7579f77SDag-Erling Smørgrav /** callback to delete hosts in infra cache */ 1137*b7579f77SDag-Erling Smørgrav static void 1138*b7579f77SDag-Erling Smørgrav infra_del_host(struct lruhash_entry* e, void* arg) 1139*b7579f77SDag-Erling Smørgrav { 1140*b7579f77SDag-Erling Smørgrav /* entry is locked */ 1141*b7579f77SDag-Erling Smørgrav struct del_info* inf = (struct del_info*)arg; 1142*b7579f77SDag-Erling Smørgrav struct infra_key* k = (struct infra_key*)e->key; 1143*b7579f77SDag-Erling Smørgrav if(sockaddr_cmp(&inf->addr, inf->addrlen, &k->addr, k->addrlen) == 0) { 1144*b7579f77SDag-Erling Smørgrav struct infra_data* d = (struct infra_data*)e->data; 1145*b7579f77SDag-Erling Smørgrav d->probedelay = 0; 1146*b7579f77SDag-Erling Smørgrav d->timeout_A = 0; 1147*b7579f77SDag-Erling Smørgrav d->timeout_AAAA = 0; 1148*b7579f77SDag-Erling Smørgrav d->timeout_other = 0; 1149*b7579f77SDag-Erling Smørgrav rtt_init(&d->rtt); 1150*b7579f77SDag-Erling Smørgrav if(d->ttl >= inf->now) { 1151*b7579f77SDag-Erling Smørgrav d->ttl = inf->expired; 1152*b7579f77SDag-Erling Smørgrav inf->num_keys++; 1153*b7579f77SDag-Erling Smørgrav } 1154*b7579f77SDag-Erling Smørgrav } 1155*b7579f77SDag-Erling Smørgrav } 1156*b7579f77SDag-Erling Smørgrav 1157*b7579f77SDag-Erling Smørgrav /** flush infra cache */ 1158*b7579f77SDag-Erling Smørgrav static void 1159*b7579f77SDag-Erling Smørgrav do_flush_infra(SSL* ssl, struct worker* worker, char* arg) 1160*b7579f77SDag-Erling Smørgrav { 1161*b7579f77SDag-Erling Smørgrav struct sockaddr_storage addr; 1162*b7579f77SDag-Erling Smørgrav socklen_t len; 1163*b7579f77SDag-Erling Smørgrav struct del_info inf; 1164*b7579f77SDag-Erling Smørgrav if(strcmp(arg, "all") == 0) { 1165*b7579f77SDag-Erling Smørgrav slabhash_clear(worker->env.infra_cache->hosts); 1166*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1167*b7579f77SDag-Erling Smørgrav return; 1168*b7579f77SDag-Erling Smørgrav } 1169*b7579f77SDag-Erling Smørgrav if(!ipstrtoaddr(arg, UNBOUND_DNS_PORT, &addr, &len)) { 1170*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error parsing ip addr: '%s'\n", arg); 1171*b7579f77SDag-Erling Smørgrav return; 1172*b7579f77SDag-Erling Smørgrav } 1173*b7579f77SDag-Erling Smørgrav /* delete all entries from cache */ 1174*b7579f77SDag-Erling Smørgrav /* what we do is to set them all expired */ 1175*b7579f77SDag-Erling Smørgrav inf.worker = worker; 1176*b7579f77SDag-Erling Smørgrav inf.name = 0; 1177*b7579f77SDag-Erling Smørgrav inf.len = 0; 1178*b7579f77SDag-Erling Smørgrav inf.labs = 0; 1179*b7579f77SDag-Erling Smørgrav inf.now = *worker->env.now; 1180*b7579f77SDag-Erling Smørgrav inf.expired = *worker->env.now; 1181*b7579f77SDag-Erling Smørgrav inf.expired -= 3; /* handle 3 seconds skew between threads */ 1182*b7579f77SDag-Erling Smørgrav inf.num_rrsets = 0; 1183*b7579f77SDag-Erling Smørgrav inf.num_msgs = 0; 1184*b7579f77SDag-Erling Smørgrav inf.num_keys = 0; 1185*b7579f77SDag-Erling Smørgrav inf.addrlen = len; 1186*b7579f77SDag-Erling Smørgrav memmove(&inf.addr, &addr, len); 1187*b7579f77SDag-Erling Smørgrav slabhash_traverse(worker->env.infra_cache->hosts, 1, &infra_del_host, 1188*b7579f77SDag-Erling Smørgrav &inf); 1189*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1190*b7579f77SDag-Erling Smørgrav } 1191*b7579f77SDag-Erling Smørgrav 1192*b7579f77SDag-Erling Smørgrav /** flush requestlist */ 1193*b7579f77SDag-Erling Smørgrav static void 1194*b7579f77SDag-Erling Smørgrav do_flush_requestlist(SSL* ssl, struct worker* worker) 1195*b7579f77SDag-Erling Smørgrav { 1196*b7579f77SDag-Erling Smørgrav mesh_delete_all(worker->env.mesh); 1197*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1198*b7579f77SDag-Erling Smørgrav } 1199*b7579f77SDag-Erling Smørgrav 1200*b7579f77SDag-Erling Smørgrav /** callback to delete rrsets in a zone */ 1201*b7579f77SDag-Erling Smørgrav static void 1202*b7579f77SDag-Erling Smørgrav zone_del_rrset(struct lruhash_entry* e, void* arg) 1203*b7579f77SDag-Erling Smørgrav { 1204*b7579f77SDag-Erling Smørgrav /* entry is locked */ 1205*b7579f77SDag-Erling Smørgrav struct del_info* inf = (struct del_info*)arg; 1206*b7579f77SDag-Erling Smørgrav struct ub_packed_rrset_key* k = (struct ub_packed_rrset_key*)e->key; 1207*b7579f77SDag-Erling Smørgrav if(dname_subdomain_c(k->rk.dname, inf->name)) { 1208*b7579f77SDag-Erling Smørgrav struct packed_rrset_data* d = 1209*b7579f77SDag-Erling Smørgrav (struct packed_rrset_data*)e->data; 1210*b7579f77SDag-Erling Smørgrav if(d->ttl >= inf->now) { 1211*b7579f77SDag-Erling Smørgrav d->ttl = inf->expired; 1212*b7579f77SDag-Erling Smørgrav inf->num_rrsets++; 1213*b7579f77SDag-Erling Smørgrav } 1214*b7579f77SDag-Erling Smørgrav } 1215*b7579f77SDag-Erling Smørgrav } 1216*b7579f77SDag-Erling Smørgrav 1217*b7579f77SDag-Erling Smørgrav /** callback to delete messages in a zone */ 1218*b7579f77SDag-Erling Smørgrav static void 1219*b7579f77SDag-Erling Smørgrav zone_del_msg(struct lruhash_entry* e, void* arg) 1220*b7579f77SDag-Erling Smørgrav { 1221*b7579f77SDag-Erling Smørgrav /* entry is locked */ 1222*b7579f77SDag-Erling Smørgrav struct del_info* inf = (struct del_info*)arg; 1223*b7579f77SDag-Erling Smørgrav struct msgreply_entry* k = (struct msgreply_entry*)e->key; 1224*b7579f77SDag-Erling Smørgrav if(dname_subdomain_c(k->key.qname, inf->name)) { 1225*b7579f77SDag-Erling Smørgrav struct reply_info* d = (struct reply_info*)e->data; 1226*b7579f77SDag-Erling Smørgrav if(d->ttl >= inf->now) { 1227*b7579f77SDag-Erling Smørgrav d->ttl = inf->expired; 1228*b7579f77SDag-Erling Smørgrav inf->num_msgs++; 1229*b7579f77SDag-Erling Smørgrav } 1230*b7579f77SDag-Erling Smørgrav } 1231*b7579f77SDag-Erling Smørgrav } 1232*b7579f77SDag-Erling Smørgrav 1233*b7579f77SDag-Erling Smørgrav /** callback to delete keys in zone */ 1234*b7579f77SDag-Erling Smørgrav static void 1235*b7579f77SDag-Erling Smørgrav zone_del_kcache(struct lruhash_entry* e, void* arg) 1236*b7579f77SDag-Erling Smørgrav { 1237*b7579f77SDag-Erling Smørgrav /* entry is locked */ 1238*b7579f77SDag-Erling Smørgrav struct del_info* inf = (struct del_info*)arg; 1239*b7579f77SDag-Erling Smørgrav struct key_entry_key* k = (struct key_entry_key*)e->key; 1240*b7579f77SDag-Erling Smørgrav if(dname_subdomain_c(k->name, inf->name)) { 1241*b7579f77SDag-Erling Smørgrav struct key_entry_data* d = (struct key_entry_data*)e->data; 1242*b7579f77SDag-Erling Smørgrav if(d->ttl >= inf->now) { 1243*b7579f77SDag-Erling Smørgrav d->ttl = inf->expired; 1244*b7579f77SDag-Erling Smørgrav inf->num_keys++; 1245*b7579f77SDag-Erling Smørgrav } 1246*b7579f77SDag-Erling Smørgrav } 1247*b7579f77SDag-Erling Smørgrav } 1248*b7579f77SDag-Erling Smørgrav 1249*b7579f77SDag-Erling Smørgrav /** remove all rrsets and keys from zone from cache */ 1250*b7579f77SDag-Erling Smørgrav static void 1251*b7579f77SDag-Erling Smørgrav do_flush_zone(SSL* ssl, struct worker* worker, char* arg) 1252*b7579f77SDag-Erling Smørgrav { 1253*b7579f77SDag-Erling Smørgrav uint8_t* nm; 1254*b7579f77SDag-Erling Smørgrav int nmlabs; 1255*b7579f77SDag-Erling Smørgrav size_t nmlen; 1256*b7579f77SDag-Erling Smørgrav struct del_info inf; 1257*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 1258*b7579f77SDag-Erling Smørgrav return; 1259*b7579f77SDag-Erling Smørgrav /* delete all RRs and key entries from zone */ 1260*b7579f77SDag-Erling Smørgrav /* what we do is to set them all expired */ 1261*b7579f77SDag-Erling Smørgrav inf.worker = worker; 1262*b7579f77SDag-Erling Smørgrav inf.name = nm; 1263*b7579f77SDag-Erling Smørgrav inf.len = nmlen; 1264*b7579f77SDag-Erling Smørgrav inf.labs = nmlabs; 1265*b7579f77SDag-Erling Smørgrav inf.now = *worker->env.now; 1266*b7579f77SDag-Erling Smørgrav inf.expired = *worker->env.now; 1267*b7579f77SDag-Erling Smørgrav inf.expired -= 3; /* handle 3 seconds skew between threads */ 1268*b7579f77SDag-Erling Smørgrav inf.num_rrsets = 0; 1269*b7579f77SDag-Erling Smørgrav inf.num_msgs = 0; 1270*b7579f77SDag-Erling Smørgrav inf.num_keys = 0; 1271*b7579f77SDag-Erling Smørgrav slabhash_traverse(&worker->env.rrset_cache->table, 1, 1272*b7579f77SDag-Erling Smørgrav &zone_del_rrset, &inf); 1273*b7579f77SDag-Erling Smørgrav 1274*b7579f77SDag-Erling Smørgrav slabhash_traverse(worker->env.msg_cache, 1, &zone_del_msg, &inf); 1275*b7579f77SDag-Erling Smørgrav 1276*b7579f77SDag-Erling Smørgrav /* and validator cache */ 1277*b7579f77SDag-Erling Smørgrav if(worker->env.key_cache) { 1278*b7579f77SDag-Erling Smørgrav slabhash_traverse(worker->env.key_cache->slab, 1, 1279*b7579f77SDag-Erling Smørgrav &zone_del_kcache, &inf); 1280*b7579f77SDag-Erling Smørgrav } 1281*b7579f77SDag-Erling Smørgrav 1282*b7579f77SDag-Erling Smørgrav free(nm); 1283*b7579f77SDag-Erling Smørgrav 1284*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "ok removed %u rrsets, %u messages " 1285*b7579f77SDag-Erling Smørgrav "and %u key entries\n", (unsigned)inf.num_rrsets, 1286*b7579f77SDag-Erling Smørgrav (unsigned)inf.num_msgs, (unsigned)inf.num_keys); 1287*b7579f77SDag-Erling Smørgrav } 1288*b7579f77SDag-Erling Smørgrav 1289*b7579f77SDag-Erling Smørgrav /** remove name rrset from cache */ 1290*b7579f77SDag-Erling Smørgrav static void 1291*b7579f77SDag-Erling Smørgrav do_flush_name(SSL* ssl, struct worker* w, char* arg) 1292*b7579f77SDag-Erling Smørgrav { 1293*b7579f77SDag-Erling Smørgrav uint8_t* nm; 1294*b7579f77SDag-Erling Smørgrav int nmlabs; 1295*b7579f77SDag-Erling Smørgrav size_t nmlen; 1296*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs)) 1297*b7579f77SDag-Erling Smørgrav return; 1298*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_A, LDNS_RR_CLASS_IN); 1299*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_AAAA, LDNS_RR_CLASS_IN); 1300*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_NS, LDNS_RR_CLASS_IN); 1301*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_SOA, LDNS_RR_CLASS_IN); 1302*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_CNAME, LDNS_RR_CLASS_IN); 1303*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_DNAME, LDNS_RR_CLASS_IN); 1304*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_MX, LDNS_RR_CLASS_IN); 1305*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_PTR, LDNS_RR_CLASS_IN); 1306*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_SRV, LDNS_RR_CLASS_IN); 1307*b7579f77SDag-Erling Smørgrav do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_NAPTR, LDNS_RR_CLASS_IN); 1308*b7579f77SDag-Erling Smørgrav 1309*b7579f77SDag-Erling Smørgrav free(nm); 1310*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1311*b7579f77SDag-Erling Smørgrav } 1312*b7579f77SDag-Erling Smørgrav 1313*b7579f77SDag-Erling Smørgrav /** printout a delegation point info */ 1314*b7579f77SDag-Erling Smørgrav static int 1315*b7579f77SDag-Erling Smørgrav ssl_print_name_dp(SSL* ssl, char* str, uint8_t* nm, uint16_t dclass, 1316*b7579f77SDag-Erling Smørgrav struct delegpt* dp) 1317*b7579f77SDag-Erling Smørgrav { 1318*b7579f77SDag-Erling Smørgrav char buf[257]; 1319*b7579f77SDag-Erling Smørgrav struct delegpt_ns* ns; 1320*b7579f77SDag-Erling Smørgrav struct delegpt_addr* a; 1321*b7579f77SDag-Erling Smørgrav int f = 0; 1322*b7579f77SDag-Erling Smørgrav if(str) { /* print header for forward, stub */ 1323*b7579f77SDag-Erling Smørgrav char* c = ldns_rr_class2str(dclass); 1324*b7579f77SDag-Erling Smørgrav dname_str(nm, buf); 1325*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s %s %s: ", buf, c, str)) { 1326*b7579f77SDag-Erling Smørgrav free(c); 1327*b7579f77SDag-Erling Smørgrav return 0; 1328*b7579f77SDag-Erling Smørgrav } 1329*b7579f77SDag-Erling Smørgrav free(c); 1330*b7579f77SDag-Erling Smørgrav } 1331*b7579f77SDag-Erling Smørgrav for(ns = dp->nslist; ns; ns = ns->next) { 1332*b7579f77SDag-Erling Smørgrav dname_str(ns->name, buf); 1333*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s%s", (f?" ":""), buf)) 1334*b7579f77SDag-Erling Smørgrav return 0; 1335*b7579f77SDag-Erling Smørgrav f = 1; 1336*b7579f77SDag-Erling Smørgrav } 1337*b7579f77SDag-Erling Smørgrav for(a = dp->target_list; a; a = a->next_target) { 1338*b7579f77SDag-Erling Smørgrav addr_to_str(&a->addr, a->addrlen, buf, sizeof(buf)); 1339*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%s%s", (f?" ":""), buf)) 1340*b7579f77SDag-Erling Smørgrav return 0; 1341*b7579f77SDag-Erling Smørgrav f = 1; 1342*b7579f77SDag-Erling Smørgrav } 1343*b7579f77SDag-Erling Smørgrav return ssl_printf(ssl, "\n"); 1344*b7579f77SDag-Erling Smørgrav } 1345*b7579f77SDag-Erling Smørgrav 1346*b7579f77SDag-Erling Smørgrav 1347*b7579f77SDag-Erling Smørgrav /** print root forwards */ 1348*b7579f77SDag-Erling Smørgrav static int 1349*b7579f77SDag-Erling Smørgrav print_root_fwds(SSL* ssl, struct iter_forwards* fwds, uint8_t* root) 1350*b7579f77SDag-Erling Smørgrav { 1351*b7579f77SDag-Erling Smørgrav struct delegpt* dp; 1352*b7579f77SDag-Erling Smørgrav dp = forwards_lookup(fwds, root, LDNS_RR_CLASS_IN); 1353*b7579f77SDag-Erling Smørgrav if(!dp) 1354*b7579f77SDag-Erling Smørgrav return ssl_printf(ssl, "off (using root hints)\n"); 1355*b7579f77SDag-Erling Smørgrav /* if dp is returned it must be the root */ 1356*b7579f77SDag-Erling Smørgrav log_assert(query_dname_compare(dp->name, root)==0); 1357*b7579f77SDag-Erling Smørgrav return ssl_print_name_dp(ssl, NULL, root, LDNS_RR_CLASS_IN, dp); 1358*b7579f77SDag-Erling Smørgrav } 1359*b7579f77SDag-Erling Smørgrav 1360*b7579f77SDag-Erling Smørgrav /** parse args into delegpt */ 1361*b7579f77SDag-Erling Smørgrav static struct delegpt* 1362*b7579f77SDag-Erling Smørgrav parse_delegpt(SSL* ssl, char* args, uint8_t* nm, int allow_names) 1363*b7579f77SDag-Erling Smørgrav { 1364*b7579f77SDag-Erling Smørgrav /* parse args and add in */ 1365*b7579f77SDag-Erling Smørgrav char* p = args; 1366*b7579f77SDag-Erling Smørgrav char* todo; 1367*b7579f77SDag-Erling Smørgrav struct delegpt* dp = delegpt_create_mlc(nm); 1368*b7579f77SDag-Erling Smørgrav struct sockaddr_storage addr; 1369*b7579f77SDag-Erling Smørgrav socklen_t addrlen; 1370*b7579f77SDag-Erling Smørgrav if(!dp) { 1371*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1372*b7579f77SDag-Erling Smørgrav return NULL; 1373*b7579f77SDag-Erling Smørgrav } 1374*b7579f77SDag-Erling Smørgrav while(p) { 1375*b7579f77SDag-Erling Smørgrav todo = p; 1376*b7579f77SDag-Erling Smørgrav p = strchr(p, ' '); /* find next spot, if any */ 1377*b7579f77SDag-Erling Smørgrav if(p) { 1378*b7579f77SDag-Erling Smørgrav *p++ = 0; /* end this spot */ 1379*b7579f77SDag-Erling Smørgrav p = skipwhite(p); /* position at next spot */ 1380*b7579f77SDag-Erling Smørgrav } 1381*b7579f77SDag-Erling Smørgrav /* parse address */ 1382*b7579f77SDag-Erling Smørgrav if(!extstrtoaddr(todo, &addr, &addrlen)) { 1383*b7579f77SDag-Erling Smørgrav if(allow_names) { 1384*b7579f77SDag-Erling Smørgrav uint8_t* n = NULL; 1385*b7579f77SDag-Erling Smørgrav size_t ln; 1386*b7579f77SDag-Erling Smørgrav int lb; 1387*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, todo, &n, &ln, &lb)) { 1388*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error cannot " 1389*b7579f77SDag-Erling Smørgrav "parse IP address or name " 1390*b7579f77SDag-Erling Smørgrav "'%s'\n", todo); 1391*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1392*b7579f77SDag-Erling Smørgrav return NULL; 1393*b7579f77SDag-Erling Smørgrav } 1394*b7579f77SDag-Erling Smørgrav if(!delegpt_add_ns_mlc(dp, n, 0)) { 1395*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1396*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1397*b7579f77SDag-Erling Smørgrav return NULL; 1398*b7579f77SDag-Erling Smørgrav } 1399*b7579f77SDag-Erling Smørgrav free(n); 1400*b7579f77SDag-Erling Smørgrav 1401*b7579f77SDag-Erling Smørgrav } else { 1402*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error cannot parse" 1403*b7579f77SDag-Erling Smørgrav " IP address '%s'\n", todo); 1404*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1405*b7579f77SDag-Erling Smørgrav return NULL; 1406*b7579f77SDag-Erling Smørgrav } 1407*b7579f77SDag-Erling Smørgrav } else { 1408*b7579f77SDag-Erling Smørgrav /* add address */ 1409*b7579f77SDag-Erling Smørgrav if(!delegpt_add_addr_mlc(dp, &addr, addrlen, 0, 0)) { 1410*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1411*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1412*b7579f77SDag-Erling Smørgrav return NULL; 1413*b7579f77SDag-Erling Smørgrav } 1414*b7579f77SDag-Erling Smørgrav } 1415*b7579f77SDag-Erling Smørgrav } 1416*b7579f77SDag-Erling Smørgrav return dp; 1417*b7579f77SDag-Erling Smørgrav } 1418*b7579f77SDag-Erling Smørgrav 1419*b7579f77SDag-Erling Smørgrav /** do the status command */ 1420*b7579f77SDag-Erling Smørgrav static void 1421*b7579f77SDag-Erling Smørgrav do_forward(SSL* ssl, struct worker* worker, char* args) 1422*b7579f77SDag-Erling Smørgrav { 1423*b7579f77SDag-Erling Smørgrav struct iter_forwards* fwd = worker->env.fwds; 1424*b7579f77SDag-Erling Smørgrav uint8_t* root = (uint8_t*)"\000"; 1425*b7579f77SDag-Erling Smørgrav if(!fwd) { 1426*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error: structure not allocated\n"); 1427*b7579f77SDag-Erling Smørgrav return; 1428*b7579f77SDag-Erling Smørgrav } 1429*b7579f77SDag-Erling Smørgrav if(args == NULL || args[0] == 0) { 1430*b7579f77SDag-Erling Smørgrav (void)print_root_fwds(ssl, fwd, root); 1431*b7579f77SDag-Erling Smørgrav return; 1432*b7579f77SDag-Erling Smørgrav } 1433*b7579f77SDag-Erling Smørgrav /* set root forwards for this thread. since we are in remote control 1434*b7579f77SDag-Erling Smørgrav * the actual mesh is not running, so we can freely edit it. */ 1435*b7579f77SDag-Erling Smørgrav /* delete all the existing queries first */ 1436*b7579f77SDag-Erling Smørgrav mesh_delete_all(worker->env.mesh); 1437*b7579f77SDag-Erling Smørgrav if(strcmp(args, "off") == 0) { 1438*b7579f77SDag-Erling Smørgrav forwards_delete_zone(fwd, LDNS_RR_CLASS_IN, root); 1439*b7579f77SDag-Erling Smørgrav } else { 1440*b7579f77SDag-Erling Smørgrav struct delegpt* dp; 1441*b7579f77SDag-Erling Smørgrav if(!(dp = parse_delegpt(ssl, args, root, 0))) 1442*b7579f77SDag-Erling Smørgrav return; 1443*b7579f77SDag-Erling Smørgrav if(!forwards_add_zone(fwd, LDNS_RR_CLASS_IN, dp)) { 1444*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1445*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1446*b7579f77SDag-Erling Smørgrav return; 1447*b7579f77SDag-Erling Smørgrav } 1448*b7579f77SDag-Erling Smørgrav } 1449*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1450*b7579f77SDag-Erling Smørgrav } 1451*b7579f77SDag-Erling Smørgrav 1452*b7579f77SDag-Erling Smørgrav static int 1453*b7579f77SDag-Erling Smørgrav parse_fs_args(SSL* ssl, char* args, uint8_t** nm, struct delegpt** dp, 1454*b7579f77SDag-Erling Smørgrav int* insecure, int* prime) 1455*b7579f77SDag-Erling Smørgrav { 1456*b7579f77SDag-Erling Smørgrav char* zonename; 1457*b7579f77SDag-Erling Smørgrav char* rest; 1458*b7579f77SDag-Erling Smørgrav size_t nmlen; 1459*b7579f77SDag-Erling Smørgrav int nmlabs; 1460*b7579f77SDag-Erling Smørgrav /* parse all -x args */ 1461*b7579f77SDag-Erling Smørgrav while(args[0] == '+') { 1462*b7579f77SDag-Erling Smørgrav if(!find_arg2(ssl, args, &rest)) 1463*b7579f77SDag-Erling Smørgrav return 0; 1464*b7579f77SDag-Erling Smørgrav while(*(++args) != 0) { 1465*b7579f77SDag-Erling Smørgrav if(*args == 'i' && insecure) 1466*b7579f77SDag-Erling Smørgrav *insecure = 1; 1467*b7579f77SDag-Erling Smørgrav else if(*args == 'p' && prime) 1468*b7579f77SDag-Erling Smørgrav *prime = 1; 1469*b7579f77SDag-Erling Smørgrav else { 1470*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error: unknown option %s\n", args); 1471*b7579f77SDag-Erling Smørgrav return 0; 1472*b7579f77SDag-Erling Smørgrav } 1473*b7579f77SDag-Erling Smørgrav } 1474*b7579f77SDag-Erling Smørgrav args = rest; 1475*b7579f77SDag-Erling Smørgrav } 1476*b7579f77SDag-Erling Smørgrav /* parse name */ 1477*b7579f77SDag-Erling Smørgrav if(dp) { 1478*b7579f77SDag-Erling Smørgrav if(!find_arg2(ssl, args, &rest)) 1479*b7579f77SDag-Erling Smørgrav return 0; 1480*b7579f77SDag-Erling Smørgrav zonename = args; 1481*b7579f77SDag-Erling Smørgrav args = rest; 1482*b7579f77SDag-Erling Smørgrav } else zonename = args; 1483*b7579f77SDag-Erling Smørgrav if(!parse_arg_name(ssl, zonename, nm, &nmlen, &nmlabs)) 1484*b7579f77SDag-Erling Smørgrav return 0; 1485*b7579f77SDag-Erling Smørgrav 1486*b7579f77SDag-Erling Smørgrav /* parse dp */ 1487*b7579f77SDag-Erling Smørgrav if(dp) { 1488*b7579f77SDag-Erling Smørgrav if(!(*dp = parse_delegpt(ssl, args, *nm, 1))) { 1489*b7579f77SDag-Erling Smørgrav free(*nm); 1490*b7579f77SDag-Erling Smørgrav return 0; 1491*b7579f77SDag-Erling Smørgrav } 1492*b7579f77SDag-Erling Smørgrav } 1493*b7579f77SDag-Erling Smørgrav return 1; 1494*b7579f77SDag-Erling Smørgrav } 1495*b7579f77SDag-Erling Smørgrav 1496*b7579f77SDag-Erling Smørgrav /** do the forward_add command */ 1497*b7579f77SDag-Erling Smørgrav static void 1498*b7579f77SDag-Erling Smørgrav do_forward_add(SSL* ssl, struct worker* worker, char* args) 1499*b7579f77SDag-Erling Smørgrav { 1500*b7579f77SDag-Erling Smørgrav struct iter_forwards* fwd = worker->env.fwds; 1501*b7579f77SDag-Erling Smørgrav int insecure = 0; 1502*b7579f77SDag-Erling Smørgrav uint8_t* nm = NULL; 1503*b7579f77SDag-Erling Smørgrav struct delegpt* dp = NULL; 1504*b7579f77SDag-Erling Smørgrav if(!parse_fs_args(ssl, args, &nm, &dp, &insecure, NULL)) 1505*b7579f77SDag-Erling Smørgrav return; 1506*b7579f77SDag-Erling Smørgrav if(insecure) { 1507*b7579f77SDag-Erling Smørgrav if(!anchors_add_insecure(worker->env.anchors, LDNS_RR_CLASS_IN, 1508*b7579f77SDag-Erling Smørgrav nm)) { 1509*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1510*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1511*b7579f77SDag-Erling Smørgrav free(nm); 1512*b7579f77SDag-Erling Smørgrav return; 1513*b7579f77SDag-Erling Smørgrav } 1514*b7579f77SDag-Erling Smørgrav } 1515*b7579f77SDag-Erling Smørgrav if(!forwards_add_zone(fwd, LDNS_RR_CLASS_IN, dp)) { 1516*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1517*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1518*b7579f77SDag-Erling Smørgrav free(nm); 1519*b7579f77SDag-Erling Smørgrav return; 1520*b7579f77SDag-Erling Smørgrav } 1521*b7579f77SDag-Erling Smørgrav free(nm); 1522*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1523*b7579f77SDag-Erling Smørgrav } 1524*b7579f77SDag-Erling Smørgrav 1525*b7579f77SDag-Erling Smørgrav /** do the forward_remove command */ 1526*b7579f77SDag-Erling Smørgrav static void 1527*b7579f77SDag-Erling Smørgrav do_forward_remove(SSL* ssl, struct worker* worker, char* args) 1528*b7579f77SDag-Erling Smørgrav { 1529*b7579f77SDag-Erling Smørgrav struct iter_forwards* fwd = worker->env.fwds; 1530*b7579f77SDag-Erling Smørgrav int insecure = 0; 1531*b7579f77SDag-Erling Smørgrav uint8_t* nm = NULL; 1532*b7579f77SDag-Erling Smørgrav if(!parse_fs_args(ssl, args, &nm, NULL, &insecure, NULL)) 1533*b7579f77SDag-Erling Smørgrav return; 1534*b7579f77SDag-Erling Smørgrav if(insecure) 1535*b7579f77SDag-Erling Smørgrav anchors_delete_insecure(worker->env.anchors, LDNS_RR_CLASS_IN, 1536*b7579f77SDag-Erling Smørgrav nm); 1537*b7579f77SDag-Erling Smørgrav forwards_delete_zone(fwd, LDNS_RR_CLASS_IN, nm); 1538*b7579f77SDag-Erling Smørgrav free(nm); 1539*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1540*b7579f77SDag-Erling Smørgrav } 1541*b7579f77SDag-Erling Smørgrav 1542*b7579f77SDag-Erling Smørgrav /** do the stub_add command */ 1543*b7579f77SDag-Erling Smørgrav static void 1544*b7579f77SDag-Erling Smørgrav do_stub_add(SSL* ssl, struct worker* worker, char* args) 1545*b7579f77SDag-Erling Smørgrav { 1546*b7579f77SDag-Erling Smørgrav struct iter_forwards* fwd = worker->env.fwds; 1547*b7579f77SDag-Erling Smørgrav int insecure = 0, prime = 0; 1548*b7579f77SDag-Erling Smørgrav uint8_t* nm = NULL; 1549*b7579f77SDag-Erling Smørgrav struct delegpt* dp = NULL; 1550*b7579f77SDag-Erling Smørgrav if(!parse_fs_args(ssl, args, &nm, &dp, &insecure, &prime)) 1551*b7579f77SDag-Erling Smørgrav return; 1552*b7579f77SDag-Erling Smørgrav if(insecure) { 1553*b7579f77SDag-Erling Smørgrav if(!anchors_add_insecure(worker->env.anchors, LDNS_RR_CLASS_IN, 1554*b7579f77SDag-Erling Smørgrav nm)) { 1555*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1556*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1557*b7579f77SDag-Erling Smørgrav free(nm); 1558*b7579f77SDag-Erling Smørgrav return; 1559*b7579f77SDag-Erling Smørgrav } 1560*b7579f77SDag-Erling Smørgrav } 1561*b7579f77SDag-Erling Smørgrav if(!forwards_add_stub_hole(fwd, LDNS_RR_CLASS_IN, nm)) { 1562*b7579f77SDag-Erling Smørgrav if(insecure) anchors_delete_insecure(worker->env.anchors, 1563*b7579f77SDag-Erling Smørgrav LDNS_RR_CLASS_IN, nm); 1564*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1565*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1566*b7579f77SDag-Erling Smørgrav free(nm); 1567*b7579f77SDag-Erling Smørgrav return; 1568*b7579f77SDag-Erling Smørgrav } 1569*b7579f77SDag-Erling Smørgrav if(!hints_add_stub(worker->env.hints, LDNS_RR_CLASS_IN, dp, !prime)) { 1570*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error out of memory\n"); 1571*b7579f77SDag-Erling Smørgrav forwards_delete_stub_hole(fwd, LDNS_RR_CLASS_IN, nm); 1572*b7579f77SDag-Erling Smørgrav if(insecure) anchors_delete_insecure(worker->env.anchors, 1573*b7579f77SDag-Erling Smørgrav LDNS_RR_CLASS_IN, nm); 1574*b7579f77SDag-Erling Smørgrav delegpt_free_mlc(dp); 1575*b7579f77SDag-Erling Smørgrav free(nm); 1576*b7579f77SDag-Erling Smørgrav return; 1577*b7579f77SDag-Erling Smørgrav } 1578*b7579f77SDag-Erling Smørgrav free(nm); 1579*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1580*b7579f77SDag-Erling Smørgrav } 1581*b7579f77SDag-Erling Smørgrav 1582*b7579f77SDag-Erling Smørgrav /** do the stub_remove command */ 1583*b7579f77SDag-Erling Smørgrav static void 1584*b7579f77SDag-Erling Smørgrav do_stub_remove(SSL* ssl, struct worker* worker, char* args) 1585*b7579f77SDag-Erling Smørgrav { 1586*b7579f77SDag-Erling Smørgrav struct iter_forwards* fwd = worker->env.fwds; 1587*b7579f77SDag-Erling Smørgrav int insecure = 0; 1588*b7579f77SDag-Erling Smørgrav uint8_t* nm = NULL; 1589*b7579f77SDag-Erling Smørgrav if(!parse_fs_args(ssl, args, &nm, NULL, &insecure, NULL)) 1590*b7579f77SDag-Erling Smørgrav return; 1591*b7579f77SDag-Erling Smørgrav if(insecure) 1592*b7579f77SDag-Erling Smørgrav anchors_delete_insecure(worker->env.anchors, LDNS_RR_CLASS_IN, 1593*b7579f77SDag-Erling Smørgrav nm); 1594*b7579f77SDag-Erling Smørgrav forwards_delete_stub_hole(fwd, LDNS_RR_CLASS_IN, nm); 1595*b7579f77SDag-Erling Smørgrav hints_delete_stub(worker->env.hints, LDNS_RR_CLASS_IN, nm); 1596*b7579f77SDag-Erling Smørgrav free(nm); 1597*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1598*b7579f77SDag-Erling Smørgrav } 1599*b7579f77SDag-Erling Smørgrav 1600*b7579f77SDag-Erling Smørgrav /** do the status command */ 1601*b7579f77SDag-Erling Smørgrav static void 1602*b7579f77SDag-Erling Smørgrav do_status(SSL* ssl, struct worker* worker) 1603*b7579f77SDag-Erling Smørgrav { 1604*b7579f77SDag-Erling Smørgrav int i; 1605*b7579f77SDag-Erling Smørgrav time_t uptime; 1606*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "version: %s\n", PACKAGE_VERSION)) 1607*b7579f77SDag-Erling Smørgrav return; 1608*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "verbosity: %d\n", verbosity)) 1609*b7579f77SDag-Erling Smørgrav return; 1610*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "threads: %d\n", worker->daemon->num)) 1611*b7579f77SDag-Erling Smørgrav return; 1612*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "modules: %d [", worker->daemon->mods.num)) 1613*b7579f77SDag-Erling Smørgrav return; 1614*b7579f77SDag-Erling Smørgrav for(i=0; i<worker->daemon->mods.num; i++) { 1615*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, " %s", worker->daemon->mods.mod[i]->name)) 1616*b7579f77SDag-Erling Smørgrav return; 1617*b7579f77SDag-Erling Smørgrav } 1618*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, " ]\n")) 1619*b7579f77SDag-Erling Smørgrav return; 1620*b7579f77SDag-Erling Smørgrav uptime = (time_t)time(NULL) - (time_t)worker->daemon->time_boot.tv_sec; 1621*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "uptime: %u seconds\n", (unsigned)uptime)) 1622*b7579f77SDag-Erling Smørgrav return; 1623*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "unbound (pid %d) is running...\n", 1624*b7579f77SDag-Erling Smørgrav (int)getpid())) 1625*b7579f77SDag-Erling Smørgrav return; 1626*b7579f77SDag-Erling Smørgrav } 1627*b7579f77SDag-Erling Smørgrav 1628*b7579f77SDag-Erling Smørgrav /** get age for the mesh state */ 1629*b7579f77SDag-Erling Smørgrav static void 1630*b7579f77SDag-Erling Smørgrav get_mesh_age(struct mesh_state* m, char* buf, size_t len, 1631*b7579f77SDag-Erling Smørgrav struct module_env* env) 1632*b7579f77SDag-Erling Smørgrav { 1633*b7579f77SDag-Erling Smørgrav if(m->reply_list) { 1634*b7579f77SDag-Erling Smørgrav struct timeval d; 1635*b7579f77SDag-Erling Smørgrav struct mesh_reply* r = m->reply_list; 1636*b7579f77SDag-Erling Smørgrav /* last reply is the oldest */ 1637*b7579f77SDag-Erling Smørgrav while(r && r->next) 1638*b7579f77SDag-Erling Smørgrav r = r->next; 1639*b7579f77SDag-Erling Smørgrav timeval_subtract(&d, env->now_tv, &r->start_time); 1640*b7579f77SDag-Erling Smørgrav snprintf(buf, len, "%d.%6.6d", (int)d.tv_sec, (int)d.tv_usec); 1641*b7579f77SDag-Erling Smørgrav } else { 1642*b7579f77SDag-Erling Smørgrav snprintf(buf, len, "-"); 1643*b7579f77SDag-Erling Smørgrav } 1644*b7579f77SDag-Erling Smørgrav } 1645*b7579f77SDag-Erling Smørgrav 1646*b7579f77SDag-Erling Smørgrav /** get status of a mesh state */ 1647*b7579f77SDag-Erling Smørgrav static void 1648*b7579f77SDag-Erling Smørgrav get_mesh_status(struct mesh_area* mesh, struct mesh_state* m, 1649*b7579f77SDag-Erling Smørgrav char* buf, size_t len) 1650*b7579f77SDag-Erling Smørgrav { 1651*b7579f77SDag-Erling Smørgrav enum module_ext_state s = m->s.ext_state[m->s.curmod]; 1652*b7579f77SDag-Erling Smørgrav const char *modname = mesh->mods.mod[m->s.curmod]->name; 1653*b7579f77SDag-Erling Smørgrav size_t l; 1654*b7579f77SDag-Erling Smørgrav if(strcmp(modname, "iterator") == 0 && s == module_wait_reply && 1655*b7579f77SDag-Erling Smørgrav m->s.minfo[m->s.curmod]) { 1656*b7579f77SDag-Erling Smørgrav /* break into iterator to find out who its waiting for */ 1657*b7579f77SDag-Erling Smørgrav struct iter_qstate* qstate = (struct iter_qstate*) 1658*b7579f77SDag-Erling Smørgrav m->s.minfo[m->s.curmod]; 1659*b7579f77SDag-Erling Smørgrav struct outbound_list* ol = &qstate->outlist; 1660*b7579f77SDag-Erling Smørgrav struct outbound_entry* e; 1661*b7579f77SDag-Erling Smørgrav snprintf(buf, len, "%s wait for", modname); 1662*b7579f77SDag-Erling Smørgrav l = strlen(buf); 1663*b7579f77SDag-Erling Smørgrav buf += l; len -= l; 1664*b7579f77SDag-Erling Smørgrav if(ol->first == NULL) 1665*b7579f77SDag-Erling Smørgrav snprintf(buf, len, " (empty_list)"); 1666*b7579f77SDag-Erling Smørgrav for(e = ol->first; e; e = e->next) { 1667*b7579f77SDag-Erling Smørgrav snprintf(buf, len, " "); 1668*b7579f77SDag-Erling Smørgrav l = strlen(buf); 1669*b7579f77SDag-Erling Smørgrav buf += l; len -= l; 1670*b7579f77SDag-Erling Smørgrav addr_to_str(&e->qsent->addr, e->qsent->addrlen, 1671*b7579f77SDag-Erling Smørgrav buf, len); 1672*b7579f77SDag-Erling Smørgrav l = strlen(buf); 1673*b7579f77SDag-Erling Smørgrav buf += l; len -= l; 1674*b7579f77SDag-Erling Smørgrav } 1675*b7579f77SDag-Erling Smørgrav } else if(s == module_wait_subquery) { 1676*b7579f77SDag-Erling Smørgrav /* look in subs from mesh state to see what */ 1677*b7579f77SDag-Erling Smørgrav char nm[257]; 1678*b7579f77SDag-Erling Smørgrav struct mesh_state_ref* sub; 1679*b7579f77SDag-Erling Smørgrav snprintf(buf, len, "%s wants", modname); 1680*b7579f77SDag-Erling Smørgrav l = strlen(buf); 1681*b7579f77SDag-Erling Smørgrav buf += l; len -= l; 1682*b7579f77SDag-Erling Smørgrav if(m->sub_set.count == 0) 1683*b7579f77SDag-Erling Smørgrav snprintf(buf, len, " (empty_list)"); 1684*b7579f77SDag-Erling Smørgrav RBTREE_FOR(sub, struct mesh_state_ref*, &m->sub_set) { 1685*b7579f77SDag-Erling Smørgrav char* t = ldns_rr_type2str(sub->s->s.qinfo.qtype); 1686*b7579f77SDag-Erling Smørgrav char* c = ldns_rr_class2str(sub->s->s.qinfo.qclass); 1687*b7579f77SDag-Erling Smørgrav dname_str(sub->s->s.qinfo.qname, nm); 1688*b7579f77SDag-Erling Smørgrav snprintf(buf, len, " %s %s %s", t, c, nm); 1689*b7579f77SDag-Erling Smørgrav l = strlen(buf); 1690*b7579f77SDag-Erling Smørgrav buf += l; len -= l; 1691*b7579f77SDag-Erling Smørgrav free(t); 1692*b7579f77SDag-Erling Smørgrav free(c); 1693*b7579f77SDag-Erling Smørgrav } 1694*b7579f77SDag-Erling Smørgrav } else { 1695*b7579f77SDag-Erling Smørgrav snprintf(buf, len, "%s is %s", modname, strextstate(s)); 1696*b7579f77SDag-Erling Smørgrav } 1697*b7579f77SDag-Erling Smørgrav } 1698*b7579f77SDag-Erling Smørgrav 1699*b7579f77SDag-Erling Smørgrav /** do the dump_requestlist command */ 1700*b7579f77SDag-Erling Smørgrav static void 1701*b7579f77SDag-Erling Smørgrav do_dump_requestlist(SSL* ssl, struct worker* worker) 1702*b7579f77SDag-Erling Smørgrav { 1703*b7579f77SDag-Erling Smørgrav struct mesh_area* mesh; 1704*b7579f77SDag-Erling Smørgrav struct mesh_state* m; 1705*b7579f77SDag-Erling Smørgrav int num = 0; 1706*b7579f77SDag-Erling Smørgrav char buf[257]; 1707*b7579f77SDag-Erling Smørgrav char timebuf[32]; 1708*b7579f77SDag-Erling Smørgrav char statbuf[10240]; 1709*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "thread #%d\n", worker->thread_num)) 1710*b7579f77SDag-Erling Smørgrav return; 1711*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "# type cl name seconds module status\n")) 1712*b7579f77SDag-Erling Smørgrav return; 1713*b7579f77SDag-Erling Smørgrav /* show worker mesh contents */ 1714*b7579f77SDag-Erling Smørgrav mesh = worker->env.mesh; 1715*b7579f77SDag-Erling Smørgrav if(!mesh) return; 1716*b7579f77SDag-Erling Smørgrav RBTREE_FOR(m, struct mesh_state*, &mesh->all) { 1717*b7579f77SDag-Erling Smørgrav char* t = ldns_rr_type2str(m->s.qinfo.qtype); 1718*b7579f77SDag-Erling Smørgrav char* c = ldns_rr_class2str(m->s.qinfo.qclass); 1719*b7579f77SDag-Erling Smørgrav dname_str(m->s.qinfo.qname, buf); 1720*b7579f77SDag-Erling Smørgrav get_mesh_age(m, timebuf, sizeof(timebuf), &worker->env); 1721*b7579f77SDag-Erling Smørgrav get_mesh_status(mesh, m, statbuf, sizeof(statbuf)); 1722*b7579f77SDag-Erling Smørgrav if(!ssl_printf(ssl, "%3d %4s %2s %s %s %s\n", 1723*b7579f77SDag-Erling Smørgrav num, t, c, buf, timebuf, statbuf)) { 1724*b7579f77SDag-Erling Smørgrav free(t); 1725*b7579f77SDag-Erling Smørgrav free(c); 1726*b7579f77SDag-Erling Smørgrav return; 1727*b7579f77SDag-Erling Smørgrav } 1728*b7579f77SDag-Erling Smørgrav num++; 1729*b7579f77SDag-Erling Smørgrav free(t); 1730*b7579f77SDag-Erling Smørgrav free(c); 1731*b7579f77SDag-Erling Smørgrav } 1732*b7579f77SDag-Erling Smørgrav } 1733*b7579f77SDag-Erling Smørgrav 1734*b7579f77SDag-Erling Smørgrav /** structure for argument data for dump infra host */ 1735*b7579f77SDag-Erling Smørgrav struct infra_arg { 1736*b7579f77SDag-Erling Smørgrav /** the infra cache */ 1737*b7579f77SDag-Erling Smørgrav struct infra_cache* infra; 1738*b7579f77SDag-Erling Smørgrav /** the SSL connection */ 1739*b7579f77SDag-Erling Smørgrav SSL* ssl; 1740*b7579f77SDag-Erling Smørgrav /** the time now */ 1741*b7579f77SDag-Erling Smørgrav uint32_t now; 1742*b7579f77SDag-Erling Smørgrav }; 1743*b7579f77SDag-Erling Smørgrav 1744*b7579f77SDag-Erling Smørgrav /** callback for every host element in the infra cache */ 1745*b7579f77SDag-Erling Smørgrav static void 1746*b7579f77SDag-Erling Smørgrav dump_infra_host(struct lruhash_entry* e, void* arg) 1747*b7579f77SDag-Erling Smørgrav { 1748*b7579f77SDag-Erling Smørgrav struct infra_arg* a = (struct infra_arg*)arg; 1749*b7579f77SDag-Erling Smørgrav struct infra_key* k = (struct infra_key*)e->key; 1750*b7579f77SDag-Erling Smørgrav struct infra_data* d = (struct infra_data*)e->data; 1751*b7579f77SDag-Erling Smørgrav char ip_str[1024]; 1752*b7579f77SDag-Erling Smørgrav char name[257]; 1753*b7579f77SDag-Erling Smørgrav addr_to_str(&k->addr, k->addrlen, ip_str, sizeof(ip_str)); 1754*b7579f77SDag-Erling Smørgrav dname_str(k->zonename, name); 1755*b7579f77SDag-Erling Smørgrav /* skip expired stuff (only backed off) */ 1756*b7579f77SDag-Erling Smørgrav if(d->ttl < a->now) { 1757*b7579f77SDag-Erling Smørgrav if(d->rtt.rto >= USEFUL_SERVER_TOP_TIMEOUT) { 1758*b7579f77SDag-Erling Smørgrav if(!ssl_printf(a->ssl, "%s %s expired rto %d\n", ip_str, 1759*b7579f77SDag-Erling Smørgrav name, d->rtt.rto)) return; 1760*b7579f77SDag-Erling Smørgrav } 1761*b7579f77SDag-Erling Smørgrav return; 1762*b7579f77SDag-Erling Smørgrav } 1763*b7579f77SDag-Erling Smørgrav if(!ssl_printf(a->ssl, "%s %s ttl %d ping %d var %d rtt %d rto %d " 1764*b7579f77SDag-Erling Smørgrav "tA %d tAAAA %d tother %d " 1765*b7579f77SDag-Erling Smørgrav "ednsknown %d edns %d delay %d lame dnssec %d rec %d A %d " 1766*b7579f77SDag-Erling Smørgrav "other %d\n", ip_str, name, (int)(d->ttl - a->now), 1767*b7579f77SDag-Erling Smørgrav d->rtt.srtt, d->rtt.rttvar, rtt_notimeout(&d->rtt), d->rtt.rto, 1768*b7579f77SDag-Erling Smørgrav d->timeout_A, d->timeout_AAAA, d->timeout_other, 1769*b7579f77SDag-Erling Smørgrav (int)d->edns_lame_known, (int)d->edns_version, 1770*b7579f77SDag-Erling Smørgrav (int)(a->now<d->probedelay?d->probedelay-a->now:0), 1771*b7579f77SDag-Erling Smørgrav (int)d->isdnsseclame, (int)d->rec_lame, (int)d->lame_type_A, 1772*b7579f77SDag-Erling Smørgrav (int)d->lame_other)) 1773*b7579f77SDag-Erling Smørgrav return; 1774*b7579f77SDag-Erling Smørgrav } 1775*b7579f77SDag-Erling Smørgrav 1776*b7579f77SDag-Erling Smørgrav /** do the dump_infra command */ 1777*b7579f77SDag-Erling Smørgrav static void 1778*b7579f77SDag-Erling Smørgrav do_dump_infra(SSL* ssl, struct worker* worker) 1779*b7579f77SDag-Erling Smørgrav { 1780*b7579f77SDag-Erling Smørgrav struct infra_arg arg; 1781*b7579f77SDag-Erling Smørgrav arg.infra = worker->env.infra_cache; 1782*b7579f77SDag-Erling Smørgrav arg.ssl = ssl; 1783*b7579f77SDag-Erling Smørgrav arg.now = *worker->env.now; 1784*b7579f77SDag-Erling Smørgrav slabhash_traverse(arg.infra->hosts, 0, &dump_infra_host, (void*)&arg); 1785*b7579f77SDag-Erling Smørgrav } 1786*b7579f77SDag-Erling Smørgrav 1787*b7579f77SDag-Erling Smørgrav /** do the log_reopen command */ 1788*b7579f77SDag-Erling Smørgrav static void 1789*b7579f77SDag-Erling Smørgrav do_log_reopen(SSL* ssl, struct worker* worker) 1790*b7579f77SDag-Erling Smørgrav { 1791*b7579f77SDag-Erling Smørgrav struct config_file* cfg = worker->env.cfg; 1792*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1793*b7579f77SDag-Erling Smørgrav log_init(cfg->logfile, cfg->use_syslog, cfg->chrootdir); 1794*b7579f77SDag-Erling Smørgrav } 1795*b7579f77SDag-Erling Smørgrav 1796*b7579f77SDag-Erling Smørgrav /** do the set_option command */ 1797*b7579f77SDag-Erling Smørgrav static void 1798*b7579f77SDag-Erling Smørgrav do_set_option(SSL* ssl, struct worker* worker, char* arg) 1799*b7579f77SDag-Erling Smørgrav { 1800*b7579f77SDag-Erling Smørgrav char* arg2; 1801*b7579f77SDag-Erling Smørgrav if(!find_arg2(ssl, arg, &arg2)) 1802*b7579f77SDag-Erling Smørgrav return; 1803*b7579f77SDag-Erling Smørgrav if(!config_set_option(worker->env.cfg, arg, arg2)) { 1804*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error setting option\n"); 1805*b7579f77SDag-Erling Smørgrav return; 1806*b7579f77SDag-Erling Smørgrav } 1807*b7579f77SDag-Erling Smørgrav send_ok(ssl); 1808*b7579f77SDag-Erling Smørgrav } 1809*b7579f77SDag-Erling Smørgrav 1810*b7579f77SDag-Erling Smørgrav /* routine to printout option values over SSL */ 1811*b7579f77SDag-Erling Smørgrav void remote_get_opt_ssl(char* line, void* arg) 1812*b7579f77SDag-Erling Smørgrav { 1813*b7579f77SDag-Erling Smørgrav SSL* ssl = (SSL*)arg; 1814*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "%s\n", line); 1815*b7579f77SDag-Erling Smørgrav } 1816*b7579f77SDag-Erling Smørgrav 1817*b7579f77SDag-Erling Smørgrav /** do the get_option command */ 1818*b7579f77SDag-Erling Smørgrav static void 1819*b7579f77SDag-Erling Smørgrav do_get_option(SSL* ssl, struct worker* worker, char* arg) 1820*b7579f77SDag-Erling Smørgrav { 1821*b7579f77SDag-Erling Smørgrav int r; 1822*b7579f77SDag-Erling Smørgrav r = config_get_option(worker->env.cfg, arg, remote_get_opt_ssl, ssl); 1823*b7579f77SDag-Erling Smørgrav if(!r) { 1824*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error unknown option\n"); 1825*b7579f77SDag-Erling Smørgrav return; 1826*b7579f77SDag-Erling Smørgrav } 1827*b7579f77SDag-Erling Smørgrav } 1828*b7579f77SDag-Erling Smørgrav 1829*b7579f77SDag-Erling Smørgrav /** do the list_forwards command */ 1830*b7579f77SDag-Erling Smørgrav static void 1831*b7579f77SDag-Erling Smørgrav do_list_forwards(SSL* ssl, struct worker* worker) 1832*b7579f77SDag-Erling Smørgrav { 1833*b7579f77SDag-Erling Smørgrav /* since its a per-worker structure no locks needed */ 1834*b7579f77SDag-Erling Smørgrav struct iter_forwards* fwds = worker->env.fwds; 1835*b7579f77SDag-Erling Smørgrav struct iter_forward_zone* z; 1836*b7579f77SDag-Erling Smørgrav RBTREE_FOR(z, struct iter_forward_zone*, fwds->tree) { 1837*b7579f77SDag-Erling Smørgrav if(!z->dp) continue; /* skip empty marker for stub */ 1838*b7579f77SDag-Erling Smørgrav if(!ssl_print_name_dp(ssl, "forward", z->name, z->dclass, 1839*b7579f77SDag-Erling Smørgrav z->dp)) 1840*b7579f77SDag-Erling Smørgrav return; 1841*b7579f77SDag-Erling Smørgrav } 1842*b7579f77SDag-Erling Smørgrav } 1843*b7579f77SDag-Erling Smørgrav 1844*b7579f77SDag-Erling Smørgrav /** do the list_stubs command */ 1845*b7579f77SDag-Erling Smørgrav static void 1846*b7579f77SDag-Erling Smørgrav do_list_stubs(SSL* ssl, struct worker* worker) 1847*b7579f77SDag-Erling Smørgrav { 1848*b7579f77SDag-Erling Smørgrav struct iter_hints_stub* z; 1849*b7579f77SDag-Erling Smørgrav RBTREE_FOR(z, struct iter_hints_stub*, &worker->env.hints->tree) { 1850*b7579f77SDag-Erling Smørgrav if(!ssl_print_name_dp(ssl, 1851*b7579f77SDag-Erling Smørgrav z->noprime?"stub noprime":"stub prime", z->node.name, 1852*b7579f77SDag-Erling Smørgrav z->node.dclass, z->dp)) 1853*b7579f77SDag-Erling Smørgrav return; 1854*b7579f77SDag-Erling Smørgrav } 1855*b7579f77SDag-Erling Smørgrav } 1856*b7579f77SDag-Erling Smørgrav 1857*b7579f77SDag-Erling Smørgrav /** do the list_local_zones command */ 1858*b7579f77SDag-Erling Smørgrav static void 1859*b7579f77SDag-Erling Smørgrav do_list_local_zones(SSL* ssl, struct worker* worker) 1860*b7579f77SDag-Erling Smørgrav { 1861*b7579f77SDag-Erling Smørgrav struct local_zones* zones = worker->daemon->local_zones; 1862*b7579f77SDag-Erling Smørgrav struct local_zone* z; 1863*b7579f77SDag-Erling Smørgrav char buf[257]; 1864*b7579f77SDag-Erling Smørgrav lock_quick_lock(&zones->lock); 1865*b7579f77SDag-Erling Smørgrav RBTREE_FOR(z, struct local_zone*, &zones->ztree) { 1866*b7579f77SDag-Erling Smørgrav lock_rw_rdlock(&z->lock); 1867*b7579f77SDag-Erling Smørgrav dname_str(z->name, buf); 1868*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "%s %s\n", buf, 1869*b7579f77SDag-Erling Smørgrav local_zone_type2str(z->type)); 1870*b7579f77SDag-Erling Smørgrav lock_rw_unlock(&z->lock); 1871*b7579f77SDag-Erling Smørgrav } 1872*b7579f77SDag-Erling Smørgrav lock_quick_unlock(&zones->lock); 1873*b7579f77SDag-Erling Smørgrav } 1874*b7579f77SDag-Erling Smørgrav 1875*b7579f77SDag-Erling Smørgrav /** do the list_local_data command */ 1876*b7579f77SDag-Erling Smørgrav static void 1877*b7579f77SDag-Erling Smørgrav do_list_local_data(SSL* ssl, struct worker* worker) 1878*b7579f77SDag-Erling Smørgrav { 1879*b7579f77SDag-Erling Smørgrav struct local_zones* zones = worker->daemon->local_zones; 1880*b7579f77SDag-Erling Smørgrav struct local_zone* z; 1881*b7579f77SDag-Erling Smørgrav struct local_data* d; 1882*b7579f77SDag-Erling Smørgrav struct local_rrset* p; 1883*b7579f77SDag-Erling Smørgrav lock_quick_lock(&zones->lock); 1884*b7579f77SDag-Erling Smørgrav RBTREE_FOR(z, struct local_zone*, &zones->ztree) { 1885*b7579f77SDag-Erling Smørgrav lock_rw_rdlock(&z->lock); 1886*b7579f77SDag-Erling Smørgrav RBTREE_FOR(d, struct local_data*, &z->data) { 1887*b7579f77SDag-Erling Smørgrav for(p = d->rrsets; p; p = p->next) { 1888*b7579f77SDag-Erling Smørgrav ldns_rr_list* rr = packed_rrset_to_rr_list( 1889*b7579f77SDag-Erling Smørgrav p->rrset, worker->env.scratch_buffer); 1890*b7579f77SDag-Erling Smørgrav char* str = ldns_rr_list2str(rr); 1891*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "%s", str); 1892*b7579f77SDag-Erling Smørgrav free(str); 1893*b7579f77SDag-Erling Smørgrav ldns_rr_list_free(rr); 1894*b7579f77SDag-Erling Smørgrav } 1895*b7579f77SDag-Erling Smørgrav } 1896*b7579f77SDag-Erling Smørgrav lock_rw_unlock(&z->lock); 1897*b7579f77SDag-Erling Smørgrav } 1898*b7579f77SDag-Erling Smørgrav lock_quick_unlock(&zones->lock); 1899*b7579f77SDag-Erling Smørgrav } 1900*b7579f77SDag-Erling Smørgrav 1901*b7579f77SDag-Erling Smørgrav /** tell other processes to execute the command */ 1902*b7579f77SDag-Erling Smørgrav static void 1903*b7579f77SDag-Erling Smørgrav distribute_cmd(struct daemon_remote* rc, SSL* ssl, char* cmd) 1904*b7579f77SDag-Erling Smørgrav { 1905*b7579f77SDag-Erling Smørgrav int i; 1906*b7579f77SDag-Erling Smørgrav if(!cmd || !ssl) 1907*b7579f77SDag-Erling Smørgrav return; 1908*b7579f77SDag-Erling Smørgrav /* skip i=0 which is me */ 1909*b7579f77SDag-Erling Smørgrav for(i=1; i<rc->worker->daemon->num; i++) { 1910*b7579f77SDag-Erling Smørgrav worker_send_cmd(rc->worker->daemon->workers[i], 1911*b7579f77SDag-Erling Smørgrav worker_cmd_remote); 1912*b7579f77SDag-Erling Smørgrav if(!tube_write_msg(rc->worker->daemon->workers[i]->cmd, 1913*b7579f77SDag-Erling Smørgrav (uint8_t*)cmd, strlen(cmd)+1, 0)) { 1914*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error could not distribute cmd\n"); 1915*b7579f77SDag-Erling Smørgrav return; 1916*b7579f77SDag-Erling Smørgrav } 1917*b7579f77SDag-Erling Smørgrav } 1918*b7579f77SDag-Erling Smørgrav } 1919*b7579f77SDag-Erling Smørgrav 1920*b7579f77SDag-Erling Smørgrav /** check for name with end-of-string, space or tab after it */ 1921*b7579f77SDag-Erling Smørgrav static int 1922*b7579f77SDag-Erling Smørgrav cmdcmp(char* p, const char* cmd, size_t len) 1923*b7579f77SDag-Erling Smørgrav { 1924*b7579f77SDag-Erling Smørgrav return strncmp(p,cmd,len)==0 && (p[len]==0||p[len]==' '||p[len]=='\t'); 1925*b7579f77SDag-Erling Smørgrav } 1926*b7579f77SDag-Erling Smørgrav 1927*b7579f77SDag-Erling Smørgrav /** execute a remote control command */ 1928*b7579f77SDag-Erling Smørgrav static void 1929*b7579f77SDag-Erling Smørgrav execute_cmd(struct daemon_remote* rc, SSL* ssl, char* cmd, 1930*b7579f77SDag-Erling Smørgrav struct worker* worker) 1931*b7579f77SDag-Erling Smørgrav { 1932*b7579f77SDag-Erling Smørgrav char* p = skipwhite(cmd); 1933*b7579f77SDag-Erling Smørgrav /* compare command */ 1934*b7579f77SDag-Erling Smørgrav if(cmdcmp(p, "stop", 4)) { 1935*b7579f77SDag-Erling Smørgrav do_stop(ssl, rc); 1936*b7579f77SDag-Erling Smørgrav return; 1937*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "reload", 6)) { 1938*b7579f77SDag-Erling Smørgrav do_reload(ssl, rc); 1939*b7579f77SDag-Erling Smørgrav return; 1940*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "stats_noreset", 13)) { 1941*b7579f77SDag-Erling Smørgrav do_stats(ssl, rc, 0); 1942*b7579f77SDag-Erling Smørgrav return; 1943*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "stats", 5)) { 1944*b7579f77SDag-Erling Smørgrav do_stats(ssl, rc, 1); 1945*b7579f77SDag-Erling Smørgrav return; 1946*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "status", 6)) { 1947*b7579f77SDag-Erling Smørgrav do_status(ssl, worker); 1948*b7579f77SDag-Erling Smørgrav return; 1949*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "dump_cache", 10)) { 1950*b7579f77SDag-Erling Smørgrav (void)dump_cache(ssl, worker); 1951*b7579f77SDag-Erling Smørgrav return; 1952*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "load_cache", 10)) { 1953*b7579f77SDag-Erling Smørgrav if(load_cache(ssl, worker)) send_ok(ssl); 1954*b7579f77SDag-Erling Smørgrav return; 1955*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "list_forwards", 13)) { 1956*b7579f77SDag-Erling Smørgrav do_list_forwards(ssl, worker); 1957*b7579f77SDag-Erling Smørgrav return; 1958*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "list_stubs", 10)) { 1959*b7579f77SDag-Erling Smørgrav do_list_stubs(ssl, worker); 1960*b7579f77SDag-Erling Smørgrav return; 1961*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "list_local_zones", 16)) { 1962*b7579f77SDag-Erling Smørgrav do_list_local_zones(ssl, worker); 1963*b7579f77SDag-Erling Smørgrav return; 1964*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "list_local_data", 15)) { 1965*b7579f77SDag-Erling Smørgrav do_list_local_data(ssl, worker); 1966*b7579f77SDag-Erling Smørgrav return; 1967*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "stub_add", 8)) { 1968*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1969*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 1970*b7579f77SDag-Erling Smørgrav do_stub_add(ssl, worker, skipwhite(p+8)); 1971*b7579f77SDag-Erling Smørgrav return; 1972*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "stub_remove", 11)) { 1973*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1974*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 1975*b7579f77SDag-Erling Smørgrav do_stub_remove(ssl, worker, skipwhite(p+11)); 1976*b7579f77SDag-Erling Smørgrav return; 1977*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "forward_add", 11)) { 1978*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1979*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 1980*b7579f77SDag-Erling Smørgrav do_forward_add(ssl, worker, skipwhite(p+11)); 1981*b7579f77SDag-Erling Smørgrav return; 1982*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "forward_remove", 14)) { 1983*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1984*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 1985*b7579f77SDag-Erling Smørgrav do_forward_remove(ssl, worker, skipwhite(p+14)); 1986*b7579f77SDag-Erling Smørgrav return; 1987*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "forward", 7)) { 1988*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1989*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 1990*b7579f77SDag-Erling Smørgrav do_forward(ssl, worker, skipwhite(p+7)); 1991*b7579f77SDag-Erling Smørgrav return; 1992*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "flush_stats", 11)) { 1993*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1994*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 1995*b7579f77SDag-Erling Smørgrav do_flush_stats(ssl, worker); 1996*b7579f77SDag-Erling Smørgrav return; 1997*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "flush_requestlist", 17)) { 1998*b7579f77SDag-Erling Smørgrav /* must always distribute this cmd */ 1999*b7579f77SDag-Erling Smørgrav if(rc) distribute_cmd(rc, ssl, cmd); 2000*b7579f77SDag-Erling Smørgrav do_flush_requestlist(ssl, worker); 2001*b7579f77SDag-Erling Smørgrav return; 2002*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "lookup", 6)) { 2003*b7579f77SDag-Erling Smørgrav do_lookup(ssl, worker, skipwhite(p+6)); 2004*b7579f77SDag-Erling Smørgrav return; 2005*b7579f77SDag-Erling Smørgrav } 2006*b7579f77SDag-Erling Smørgrav 2007*b7579f77SDag-Erling Smørgrav #ifdef THREADS_DISABLED 2008*b7579f77SDag-Erling Smørgrav /* other processes must execute the command as well */ 2009*b7579f77SDag-Erling Smørgrav /* commands that should not be distributed, returned above. */ 2010*b7579f77SDag-Erling Smørgrav if(rc) { /* only if this thread is the master (rc) thread */ 2011*b7579f77SDag-Erling Smørgrav /* done before the code below, which may split the string */ 2012*b7579f77SDag-Erling Smørgrav distribute_cmd(rc, ssl, cmd); 2013*b7579f77SDag-Erling Smørgrav } 2014*b7579f77SDag-Erling Smørgrav #endif 2015*b7579f77SDag-Erling Smørgrav if(cmdcmp(p, "verbosity", 9)) { 2016*b7579f77SDag-Erling Smørgrav do_verbosity(ssl, skipwhite(p+9)); 2017*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "local_zone_remove", 17)) { 2018*b7579f77SDag-Erling Smørgrav do_zone_remove(ssl, worker, skipwhite(p+17)); 2019*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "local_zone", 10)) { 2020*b7579f77SDag-Erling Smørgrav do_zone_add(ssl, worker, skipwhite(p+10)); 2021*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "local_data_remove", 17)) { 2022*b7579f77SDag-Erling Smørgrav do_data_remove(ssl, worker, skipwhite(p+17)); 2023*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "local_data", 10)) { 2024*b7579f77SDag-Erling Smørgrav do_data_add(ssl, worker, skipwhite(p+10)); 2025*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "flush_zone", 10)) { 2026*b7579f77SDag-Erling Smørgrav do_flush_zone(ssl, worker, skipwhite(p+10)); 2027*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "flush_type", 10)) { 2028*b7579f77SDag-Erling Smørgrav do_flush_type(ssl, worker, skipwhite(p+10)); 2029*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "flush_infra", 11)) { 2030*b7579f77SDag-Erling Smørgrav do_flush_infra(ssl, worker, skipwhite(p+11)); 2031*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "flush", 5)) { 2032*b7579f77SDag-Erling Smørgrav do_flush_name(ssl, worker, skipwhite(p+5)); 2033*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "dump_requestlist", 16)) { 2034*b7579f77SDag-Erling Smørgrav do_dump_requestlist(ssl, worker); 2035*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "dump_infra", 10)) { 2036*b7579f77SDag-Erling Smørgrav do_dump_infra(ssl, worker); 2037*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "log_reopen", 10)) { 2038*b7579f77SDag-Erling Smørgrav do_log_reopen(ssl, worker); 2039*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "set_option", 10)) { 2040*b7579f77SDag-Erling Smørgrav do_set_option(ssl, worker, skipwhite(p+10)); 2041*b7579f77SDag-Erling Smørgrav } else if(cmdcmp(p, "get_option", 10)) { 2042*b7579f77SDag-Erling Smørgrav do_get_option(ssl, worker, skipwhite(p+10)); 2043*b7579f77SDag-Erling Smørgrav } else { 2044*b7579f77SDag-Erling Smørgrav (void)ssl_printf(ssl, "error unknown command '%s'\n", p); 2045*b7579f77SDag-Erling Smørgrav } 2046*b7579f77SDag-Erling Smørgrav } 2047*b7579f77SDag-Erling Smørgrav 2048*b7579f77SDag-Erling Smørgrav void 2049*b7579f77SDag-Erling Smørgrav daemon_remote_exec(struct worker* worker) 2050*b7579f77SDag-Erling Smørgrav { 2051*b7579f77SDag-Erling Smørgrav /* read the cmd string */ 2052*b7579f77SDag-Erling Smørgrav uint8_t* msg = NULL; 2053*b7579f77SDag-Erling Smørgrav uint32_t len = 0; 2054*b7579f77SDag-Erling Smørgrav if(!tube_read_msg(worker->cmd, &msg, &len, 0)) { 2055*b7579f77SDag-Erling Smørgrav log_err("daemon_remote_exec: tube_read_msg failed"); 2056*b7579f77SDag-Erling Smørgrav return; 2057*b7579f77SDag-Erling Smørgrav } 2058*b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "remote exec distributed: %s", (char*)msg); 2059*b7579f77SDag-Erling Smørgrav execute_cmd(NULL, NULL, (char*)msg, worker); 2060*b7579f77SDag-Erling Smørgrav free(msg); 2061*b7579f77SDag-Erling Smørgrav } 2062*b7579f77SDag-Erling Smørgrav 2063*b7579f77SDag-Erling Smørgrav /** handle remote control request */ 2064*b7579f77SDag-Erling Smørgrav static void 2065*b7579f77SDag-Erling Smørgrav handle_req(struct daemon_remote* rc, struct rc_state* s, SSL* ssl) 2066*b7579f77SDag-Erling Smørgrav { 2067*b7579f77SDag-Erling Smørgrav int r; 2068*b7579f77SDag-Erling Smørgrav char pre[10]; 2069*b7579f77SDag-Erling Smørgrav char magic[7]; 2070*b7579f77SDag-Erling Smørgrav char buf[1024]; 2071*b7579f77SDag-Erling Smørgrav #ifdef USE_WINSOCK 2072*b7579f77SDag-Erling Smørgrav /* makes it possible to set the socket blocking again. */ 2073*b7579f77SDag-Erling Smørgrav /* basically removes it from winsock_event ... */ 2074*b7579f77SDag-Erling Smørgrav WSAEventSelect(s->c->fd, NULL, 0); 2075*b7579f77SDag-Erling Smørgrav #endif 2076*b7579f77SDag-Erling Smørgrav fd_set_block(s->c->fd); 2077*b7579f77SDag-Erling Smørgrav 2078*b7579f77SDag-Erling Smørgrav /* try to read magic UBCT[version]_space_ string */ 2079*b7579f77SDag-Erling Smørgrav ERR_clear_error(); 2080*b7579f77SDag-Erling Smørgrav if((r=SSL_read(ssl, magic, (int)sizeof(magic)-1)) <= 0) { 2081*b7579f77SDag-Erling Smørgrav if(SSL_get_error(ssl, r) == SSL_ERROR_ZERO_RETURN) 2082*b7579f77SDag-Erling Smørgrav return; 2083*b7579f77SDag-Erling Smørgrav log_crypto_err("could not SSL_read"); 2084*b7579f77SDag-Erling Smørgrav return; 2085*b7579f77SDag-Erling Smørgrav } 2086*b7579f77SDag-Erling Smørgrav magic[6] = 0; 2087*b7579f77SDag-Erling Smørgrav if( r != 6 || strncmp(magic, "UBCT", 4) != 0) { 2088*b7579f77SDag-Erling Smørgrav verbose(VERB_QUERY, "control connection has bad magic string"); 2089*b7579f77SDag-Erling Smørgrav /* probably wrong tool connected, ignore it completely */ 2090*b7579f77SDag-Erling Smørgrav return; 2091*b7579f77SDag-Erling Smørgrav } 2092*b7579f77SDag-Erling Smørgrav 2093*b7579f77SDag-Erling Smørgrav /* read the command line */ 2094*b7579f77SDag-Erling Smørgrav if(!ssl_read_line(ssl, buf, sizeof(buf))) { 2095*b7579f77SDag-Erling Smørgrav return; 2096*b7579f77SDag-Erling Smørgrav } 2097*b7579f77SDag-Erling Smørgrav snprintf(pre, sizeof(pre), "UBCT%d ", UNBOUND_CONTROL_VERSION); 2098*b7579f77SDag-Erling Smørgrav if(strcmp(magic, pre) != 0) { 2099*b7579f77SDag-Erling Smørgrav verbose(VERB_QUERY, "control connection had bad " 2100*b7579f77SDag-Erling Smørgrav "version %s, cmd: %s", magic, buf); 2101*b7579f77SDag-Erling Smørgrav ssl_printf(ssl, "error version mismatch\n"); 2102*b7579f77SDag-Erling Smørgrav return; 2103*b7579f77SDag-Erling Smørgrav } 2104*b7579f77SDag-Erling Smørgrav verbose(VERB_DETAIL, "control cmd: %s", buf); 2105*b7579f77SDag-Erling Smørgrav 2106*b7579f77SDag-Erling Smørgrav /* figure out what to do */ 2107*b7579f77SDag-Erling Smørgrav execute_cmd(rc, ssl, buf, rc->worker); 2108*b7579f77SDag-Erling Smørgrav } 2109*b7579f77SDag-Erling Smørgrav 2110*b7579f77SDag-Erling Smørgrav int remote_control_callback(struct comm_point* c, void* arg, int err, 2111*b7579f77SDag-Erling Smørgrav struct comm_reply* ATTR_UNUSED(rep)) 2112*b7579f77SDag-Erling Smørgrav { 2113*b7579f77SDag-Erling Smørgrav struct rc_state* s = (struct rc_state*)arg; 2114*b7579f77SDag-Erling Smørgrav struct daemon_remote* rc = s->rc; 2115*b7579f77SDag-Erling Smørgrav int r; 2116*b7579f77SDag-Erling Smørgrav if(err != NETEVENT_NOERROR) { 2117*b7579f77SDag-Erling Smørgrav if(err==NETEVENT_TIMEOUT) 2118*b7579f77SDag-Erling Smørgrav log_err("remote control timed out"); 2119*b7579f77SDag-Erling Smørgrav clean_point(rc, s); 2120*b7579f77SDag-Erling Smørgrav return 0; 2121*b7579f77SDag-Erling Smørgrav } 2122*b7579f77SDag-Erling Smørgrav /* (continue to) setup the SSL connection */ 2123*b7579f77SDag-Erling Smørgrav ERR_clear_error(); 2124*b7579f77SDag-Erling Smørgrav r = SSL_do_handshake(s->ssl); 2125*b7579f77SDag-Erling Smørgrav if(r != 1) { 2126*b7579f77SDag-Erling Smørgrav int r2 = SSL_get_error(s->ssl, r); 2127*b7579f77SDag-Erling Smørgrav if(r2 == SSL_ERROR_WANT_READ) { 2128*b7579f77SDag-Erling Smørgrav if(s->shake_state == rc_hs_read) { 2129*b7579f77SDag-Erling Smørgrav /* try again later */ 2130*b7579f77SDag-Erling Smørgrav return 0; 2131*b7579f77SDag-Erling Smørgrav } 2132*b7579f77SDag-Erling Smørgrav s->shake_state = rc_hs_read; 2133*b7579f77SDag-Erling Smørgrav comm_point_listen_for_rw(c, 1, 0); 2134*b7579f77SDag-Erling Smørgrav return 0; 2135*b7579f77SDag-Erling Smørgrav } else if(r2 == SSL_ERROR_WANT_WRITE) { 2136*b7579f77SDag-Erling Smørgrav if(s->shake_state == rc_hs_write) { 2137*b7579f77SDag-Erling Smørgrav /* try again later */ 2138*b7579f77SDag-Erling Smørgrav return 0; 2139*b7579f77SDag-Erling Smørgrav } 2140*b7579f77SDag-Erling Smørgrav s->shake_state = rc_hs_write; 2141*b7579f77SDag-Erling Smørgrav comm_point_listen_for_rw(c, 0, 1); 2142*b7579f77SDag-Erling Smørgrav return 0; 2143*b7579f77SDag-Erling Smørgrav } else { 2144*b7579f77SDag-Erling Smørgrav if(r == 0) 2145*b7579f77SDag-Erling Smørgrav log_err("remote control connection closed prematurely"); 2146*b7579f77SDag-Erling Smørgrav log_addr(1, "failed connection from", 2147*b7579f77SDag-Erling Smørgrav &s->c->repinfo.addr, s->c->repinfo.addrlen); 2148*b7579f77SDag-Erling Smørgrav log_crypto_err("remote control failed ssl"); 2149*b7579f77SDag-Erling Smørgrav clean_point(rc, s); 2150*b7579f77SDag-Erling Smørgrav return 0; 2151*b7579f77SDag-Erling Smørgrav } 2152*b7579f77SDag-Erling Smørgrav } 2153*b7579f77SDag-Erling Smørgrav s->shake_state = rc_none; 2154*b7579f77SDag-Erling Smørgrav 2155*b7579f77SDag-Erling Smørgrav /* once handshake has completed, check authentication */ 2156*b7579f77SDag-Erling Smørgrav if(SSL_get_verify_result(s->ssl) == X509_V_OK) { 2157*b7579f77SDag-Erling Smørgrav X509* x = SSL_get_peer_certificate(s->ssl); 2158*b7579f77SDag-Erling Smørgrav if(!x) { 2159*b7579f77SDag-Erling Smørgrav verbose(VERB_DETAIL, "remote control connection " 2160*b7579f77SDag-Erling Smørgrav "provided no client certificate"); 2161*b7579f77SDag-Erling Smørgrav clean_point(rc, s); 2162*b7579f77SDag-Erling Smørgrav return 0; 2163*b7579f77SDag-Erling Smørgrav } 2164*b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "remote control connection authenticated"); 2165*b7579f77SDag-Erling Smørgrav X509_free(x); 2166*b7579f77SDag-Erling Smørgrav } else { 2167*b7579f77SDag-Erling Smørgrav verbose(VERB_DETAIL, "remote control connection failed to " 2168*b7579f77SDag-Erling Smørgrav "authenticate with client certificate"); 2169*b7579f77SDag-Erling Smørgrav clean_point(rc, s); 2170*b7579f77SDag-Erling Smørgrav return 0; 2171*b7579f77SDag-Erling Smørgrav } 2172*b7579f77SDag-Erling Smørgrav 2173*b7579f77SDag-Erling Smørgrav /* if OK start to actually handle the request */ 2174*b7579f77SDag-Erling Smørgrav handle_req(rc, s, s->ssl); 2175*b7579f77SDag-Erling Smørgrav 2176*b7579f77SDag-Erling Smørgrav verbose(VERB_ALGO, "remote control operation completed"); 2177*b7579f77SDag-Erling Smørgrav clean_point(rc, s); 2178*b7579f77SDag-Erling Smørgrav return 0; 2179*b7579f77SDag-Erling Smørgrav } 2180