xref: /freebsd/contrib/unbound/daemon/remote.c (revision b7579f77d18196a58ff700756c84dc9a302a7f67)
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