mirror of
https://github.com/NLnetLabs/unbound.git
synced 2025-12-18 23:06:06 -05:00
* - dnsoverquic, configure --with-libngtcp2 option. * - dnsoverquic, create comm_point for doq and receive cmsg local address. * - dnsoverquic, less obtrusive debug. * - dnsoverquic, log and fix local port number. Neater subroutines and ifdefs. * - dnsoverquic, add testcode/doqclient. * - dnsoverquic, review fixes on doqclient. * - dnsoverquic, fix unit test testbound link. * - dnsoverquic, parse query in doqclient. * - dnsoverquic, link with libngtcp2_crypto_openssl and code for doqclient. * - dnsoverquic, random routine for doqclient and fix ngaddr allocation, and check ub_initstate return. * - dnsoverquic, fix doqclient free of allocated ngaddr addresses. * - dnsoverquic, enable debug output with -v for doqclient. * - dnsoverquic, create and set TLS object and TLS context in doqclient. * - dnsoverquic, work on quic tls context in doqclient. * - dnsoverquic, set default dnsoverquic port to the standardized 853 port. * - dnsoverquic, remove debug comment. * - dnsoverquic, dns-over-quic quic-port: 853 config option. * - dnsoverquic, log type of interface created at start of unbound. * - dnsoverquic, log type of no tls https as https when interface is created. * - dnsoverquic, setup client quic tls methods. * - dnsoverquic, event work in doqclient. * - dnsoverquic, explain in documentation that QUIC uses UDP. * - dnsoverquic, make doqclient exit. * - dnsoverquic, doqclient cleanup run routine. * - dnsoverquic, doqclient code nicer. * - dnsoverquic, doqclient read and timer. * - dnsoverquic, doqclient write work. * - dnsoverquic, review fixes. * - dnsoverquic, detect openssl quic support at configure time. * - dnsoverquic, do not allow QUIC on port 53 to stop confusion of DoQ and DNS. * - dnsoverquic, in doqclient, when idle close is returned, drop the connection without calling ngtcp2_conn_write_connection_close. * - dnsoverquic, in doqclient, log callbacks. * - dnsoverquic, in doqclient add extend_max_local_streams_bidi callback. * - dnsoverquic, in doqclient add client query lists. * - dnsoverquic, in doqclient, code cleaner, log text nicer. * - dnsoverquic, in doqclient, work on write_streams. * - dnsoverquic, in doqclient, use signed int for stream_id, work on the ngtcp2_recv_stream_data callback. * - dnsoverquic, in doqclient, print result and fixes for recv data. * - dnsoverquic, in doqclient, add the event callbacks to fptr wlist. * - dnsoverquic, in doqclient, when already expired, use zero timeout timer. * - dnsoverquic, in doqclient, ignore unused return codes from ngtcp2_conn_writev_stream. * - dnsoverquic, add doqclient event functions to the unbound-dnstap-socket test tool for linking. * - dnsoverquic, in doqclient, fix multiple operands for the commandline. neater dns message output. * - dnsoverquic, in doqclient, store packet when write blocks and try later. * - dnsoverquic, in doqclient, limit number of packets and number of bytes sent. * - dnsoverquic, in doqclient, better size estimate for outgoing packet. * - dnsoverquic, in doqclient, fix that already written next packet is not counted for data length to send. * - dnsoverquic, in doqclient, early data transmission and session resumption. * - dnsoverquic, send version negotiation packet. * - dnsoverquic, send retry and accept the connection. * - dnsoverquic, storage structures. * - dnsoverquic, doq connection setup. * - dnsoverquic, neater code layout for new conn. Fix verbosity of log print. * - dnsoverquic, doq conn callback functions. * - dnsoverquic, doq_fill_rand routine in header file. * - dnsoverquic, keep track of connection ids. * - dnsoverquic, get_new_connection_id callback. * - dnsoverquic, create doq_conid tree. * - dnsoverquic, settings for server connection. * - dnsoverquic, tls context. * - dnsoverquic, sendmsg error handling. * - dnsoverquic, neat code. * - dnsoverquic, track doq connection last error. * - dnsoverquic, neater packet address parameters. * - dnsoverquic, fix uninitialized bytes in msg control in doq sendmsg, and fix tree cleanup of conid tree. * - dnsoverquic, better usage text for doqclient. * - dnsoverquic, neat code. * - dnsoverquic, connection receive packet handling. * - dnsoverquic, debug output. * - dnsoverquic, debug switched meaning of scid and dcid gives ERR_TRANSPORT_PARAM. * - dnsoverquic, remove debug output. * - dnsoverquic, connection delete routine and error from connection read in more detail with less clutter. * - dnsoverquic, write to stream, and receive stream data, log packet. * - dnsoverquic, alpn set up. * - dnsoverquic, connection close. * - dnsoverquic, doq_table and locks. * - dnsoverquic, fix tests. * - dnsoverquic, better locking. * - dnsoverquic, doq_stream. * - dnsoverquic, remove compile warning. * - dnsoverquic, doq_stream receive data. * - dnsoverquic, fixes for locks and keep length bytes allocated. * - dnsoverquic, lock connection on initial insertion. * - dnsoverquic, reply information, and reply buffer. * - dnsoverquic, reply info from cache, local-zone and recursion lookups. * - dnsoverquic, spelling in comment about buffer storage. * - dnsoverquic, stream write list and doqclient fixes to exit and printout. * - dnsoverquic, doqclient -q option for short printout. * - dnsoverquic, unit test with local data reply. * - dnsoverquic, write connection and write event is set. * - dnsoverquic, neater logging for write event connection stream writes. * - dnsoverquic, log remote connection when the streams are written for it. * - dnsoverquic, better threaded use, threads can write to doq connections at the same time. * - dnsoverquic, unit test for the calculation of connection size with a query. * - dnsoverquic, use less memory per connection. * - dnsoverquic, remove unit test output. * - dnsoverquic, add MSG_DONTWAIT so that there is no mistakenly blocking socket operations. * - dnsoverquic, doqclient logs address on connection failures. * - dnsoverquic, compat code for clock get time routine. * - dnsoverquic, use skip_test for doq unit test. * - dnsoverquic, fixes for proxyprotocol, use remote_addr and set proxyprotocol disabled on the doq connection. * - dnsoverquic, doqclient sets log identity to its name, instead of "unbound". * - dnsoverquic, handle blocked udp packet writes. * - dnsoverquic, fix function documentation for verbose_print_addr from services/listen_dnsport.c. * - dnsoverquic, fix doq_conn lock protection. The checklock allows to set the output file name, and doqclient uses that. Print place of lock_protect. * - dnsoverquic, neater buffer clear when write of blocked packet fails, make sure that memory area does not overlap for blocked packet addresses when write of blocked packet fails, and size blocked packet buffer to the pkt buf. * - dnsoverquic, move lock check after the test to test script in doq test. * - dnsoverquic, the doq test uses valgrind when enabled. * - dnsoverquic, git ignore the doqclient test. * - dnsoverquic, limit the buffer for packets to max packet size with some more. * - dnsoverquic, spelling fix. * - dnsoverquic, timer work, structure and adds and deletes. * - dnsoverquic, timer_tree uses table.lock. * - dnsoverquic, fix timer tree remove and spelling in header file comment. * - dnsoverquic, fix testbound for timer compare function linkage. * - dnsoverquic, timer set add debug output. * - dnsoverquic, doq_conn_check_timer function. * - dnsoverquic, doq_done_setup_timer_and_write function. * - dnsoverquic, fix that doq conn is not deleted whilst editing write and timer. * - dnsoverquic, Fix #861 make ERROR netevent.h:1073:32: error: field 'blocked_pkt_pi' has incomplete type * - dnsoverquic, timer element has timeout setup when socket callback complete. * - dnsoverquic, fix unit test compile. * - dnsoverquic, timer callback routine, handle timeout and close and delete the connection if necessary. * - dnsoverquic, timer pickup stops at current time. * - dnsoverquic, timer comparable with the event base time. * - dnsoverquic, erase marked time when timer disabled. * - dnsoverquic, fix timer to set correctly and lock popped write connection early, before it is modified. * - dnsoverquic, fix to unlock connection lock when it is unlinked and deleted. * - dnsoverquic, fix to unlock connection lock when it is deleted because it is a duplicate connection. * - dnsoverquic, fix that doq timer is not disabled when not set. * - dnsoverquic, quic-size: 8m maximum number of bytes for QUIC buffers. * - dnsoverquic, flex and bison. * - dnsoverquic, quic-size turn away new connections when full. * - dnsoverquic, doqclient outputs stream reset information. * - dnsoverquic, detect stream close and reset. * - dnsoverquic, free stream buffers when data is acked and stream is closed. * - dnsoverquic, delete stream when closed. Unlink it. Allow stream_id 4 as first. * - dnsoverquic, stats output for mem.quic and num.query.quic. * - dnsoverquic, review fix. * - dnsoverquic, fix when compiled without ngtcp2. * - dnsoverquic, fix to detect ngtcp2_crypto_quictls for openssl crypto, after change in libngtcp2. * - dnsoverquic, fix for newer ngtcp2 versions. detect ngtcp2_ccerr_default, ngtcp2/ngtcp2_crypto_quictls.h, struct ngtcp2_pkt_hd.tokenlen, struct ngtcp2_settings.tokenlen and struct ngtcp2_version_cid. * - dnsoverquic, fix for newer ngtcp2 version, detect number of arguments for ngtcp2_conn_shutdown_stream. * - dnsoverquic, fix for newer ngtcp2. * - dnsoverquic, use the functions from util/timeval_func.h. * - dnsoverquic, fix in doqclient only write transport parameters once. * - dnsoverquic, debug log output removed. * - dnsoverquic, fix in doqclient to work with renamed NGTCP2_CC_ALGO_BBR_V2 from ngtcp2. * - dnsoverquic, fix to check in doq_server_socket_create that tls-service-key and tls-service-pem have a value. * - dnsoverquic, fix to error when doq_server_socket_create fails. * - dnsoverquic, improve linebreaks in configparser additions. * - dnsoverquic, fix port from interface pickup after main branch change. * Fix getting user data from SSL, fix calloc warning. * Fix fwrite return value check in doqclient * - timeval_substruct from timeval_func.h - lock_protect also for HAVE_NGTCP2_CCERR_DEFAULT - fix doq logging for inet_ntop failures * - memset for consistency - no value returned from msghdr_get_ecn when S_SPLINT_S is defined * - dnsoverquic, rerun autoconf. --------- Co-authored-by: Yorgos Thessalonikefs <yorgos@nlnetlabs.nl>
3791 lines
104 KiB
C
3791 lines
104 KiB
C
/*
|
|
* daemon/remote.c - remote control for the unbound daemon.
|
|
*
|
|
* Copyright (c) 2008, NLnet Labs. All rights reserved.
|
|
*
|
|
* This software is open source.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
*
|
|
* Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* Neither the name of the NLNET LABS nor the names of its contributors may
|
|
* be used to endorse or promote products derived from this software without
|
|
* specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
* HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
|
|
* TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
/**
|
|
* \file
|
|
*
|
|
* This file contains the remote control functionality for the daemon.
|
|
* The remote control can be performed using either the commandline
|
|
* unbound-control tool, or a TLS capable web browser.
|
|
* The channel is secured using TLSv1, and certificates.
|
|
* Both the server and the client(control tool) have their own keys.
|
|
*/
|
|
#include "config.h"
|
|
#ifdef HAVE_OPENSSL_ERR_H
|
|
#include <openssl/err.h>
|
|
#endif
|
|
#ifdef HAVE_OPENSSL_DH_H
|
|
#include <openssl/dh.h>
|
|
#endif
|
|
#ifdef HAVE_OPENSSL_BN_H
|
|
#include <openssl/bn.h>
|
|
#endif
|
|
|
|
#include <ctype.h>
|
|
#include "daemon/remote.h"
|
|
#include "daemon/worker.h"
|
|
#include "daemon/daemon.h"
|
|
#include "daemon/stats.h"
|
|
#include "daemon/cachedump.h"
|
|
#include "util/log.h"
|
|
#include "util/config_file.h"
|
|
#include "util/net_help.h"
|
|
#include "util/module.h"
|
|
#include "services/listen_dnsport.h"
|
|
#include "services/cache/rrset.h"
|
|
#include "services/cache/infra.h"
|
|
#include "services/mesh.h"
|
|
#include "services/localzone.h"
|
|
#include "services/authzone.h"
|
|
#include "services/rpz.h"
|
|
#include "util/storage/slabhash.h"
|
|
#include "util/fptr_wlist.h"
|
|
#include "util/data/dname.h"
|
|
#include "validator/validator.h"
|
|
#include "validator/val_kcache.h"
|
|
#include "validator/val_kentry.h"
|
|
#include "validator/val_anchor.h"
|
|
#include "iterator/iterator.h"
|
|
#include "iterator/iter_fwd.h"
|
|
#include "iterator/iter_hints.h"
|
|
#include "iterator/iter_delegpt.h"
|
|
#include "services/outbound_list.h"
|
|
#include "services/outside_network.h"
|
|
#include "sldns/str2wire.h"
|
|
#include "sldns/parseutil.h"
|
|
#include "sldns/wire2str.h"
|
|
#include "sldns/sbuffer.h"
|
|
#include "util/timeval_func.h"
|
|
#include "util/edns.h"
|
|
#ifdef USE_CACHEDB
|
|
#include "cachedb/cachedb.h"
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
# include <sys/types.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_STAT_H
|
|
#include <sys/stat.h>
|
|
#endif
|
|
#ifdef HAVE_NETDB_H
|
|
#include <netdb.h>
|
|
#endif
|
|
|
|
/* just for portability */
|
|
#ifdef SQ
|
|
#undef SQ
|
|
#endif
|
|
|
|
/** what to put on statistics lines between var and value, ": " or "=" */
|
|
#define SQ "="
|
|
|
|
/** Acceptable lengths of str lines */
|
|
#define MAX_CMD_STRLINE 1024
|
|
#define MAX_STDIN_STRLINE 2048
|
|
|
|
static int
|
|
remote_setup_ctx(struct daemon_remote* rc, struct config_file* cfg)
|
|
{
|
|
char* s_cert;
|
|
char* s_key;
|
|
rc->ctx = SSL_CTX_new(SSLv23_server_method());
|
|
if(!rc->ctx) {
|
|
log_crypto_err("could not SSL_CTX_new");
|
|
return 0;
|
|
}
|
|
if(!listen_sslctx_setup(rc->ctx)) {
|
|
return 0;
|
|
}
|
|
|
|
s_cert = fname_after_chroot(cfg->server_cert_file, cfg, 1);
|
|
s_key = fname_after_chroot(cfg->server_key_file, cfg, 1);
|
|
if(!s_cert || !s_key) {
|
|
log_err("out of memory in remote control fname");
|
|
goto setup_error;
|
|
}
|
|
verbose(VERB_ALGO, "setup SSL certificates");
|
|
if (!SSL_CTX_use_certificate_chain_file(rc->ctx,s_cert)) {
|
|
log_err("Error for server-cert-file: %s", s_cert);
|
|
log_crypto_err("Error in SSL_CTX use_certificate_chain_file");
|
|
goto setup_error;
|
|
}
|
|
if(!SSL_CTX_use_PrivateKey_file(rc->ctx,s_key,SSL_FILETYPE_PEM)) {
|
|
log_err("Error for server-key-file: %s", s_key);
|
|
log_crypto_err("Error in SSL_CTX use_PrivateKey_file");
|
|
goto setup_error;
|
|
}
|
|
if(!SSL_CTX_check_private_key(rc->ctx)) {
|
|
log_err("Error for server-key-file: %s", s_key);
|
|
log_crypto_err("Error in SSL_CTX check_private_key");
|
|
goto setup_error;
|
|
}
|
|
listen_sslctx_setup_2(rc->ctx);
|
|
if(!SSL_CTX_load_verify_locations(rc->ctx, s_cert, NULL)) {
|
|
log_crypto_err("Error setting up SSL_CTX verify locations");
|
|
setup_error:
|
|
free(s_cert);
|
|
free(s_key);
|
|
return 0;
|
|
}
|
|
SSL_CTX_set_client_CA_list(rc->ctx, SSL_load_client_CA_file(s_cert));
|
|
SSL_CTX_set_verify(rc->ctx, SSL_VERIFY_PEER, NULL);
|
|
free(s_cert);
|
|
free(s_key);
|
|
return 1;
|
|
}
|
|
|
|
struct daemon_remote*
|
|
daemon_remote_create(struct config_file* cfg)
|
|
{
|
|
struct daemon_remote* rc = (struct daemon_remote*)calloc(1,
|
|
sizeof(*rc));
|
|
if(!rc) {
|
|
log_err("out of memory in daemon_remote_create");
|
|
return NULL;
|
|
}
|
|
rc->max_active = 10;
|
|
|
|
if(!cfg->remote_control_enable) {
|
|
rc->ctx = NULL;
|
|
return rc;
|
|
}
|
|
if(options_remote_is_address(cfg) && cfg->control_use_cert) {
|
|
if(!remote_setup_ctx(rc, cfg)) {
|
|
daemon_remote_delete(rc);
|
|
return NULL;
|
|
}
|
|
rc->use_cert = 1;
|
|
} else {
|
|
struct config_strlist* p;
|
|
rc->ctx = NULL;
|
|
rc->use_cert = 0;
|
|
if(!options_remote_is_address(cfg))
|
|
for(p = cfg->control_ifs.first; p; p = p->next) {
|
|
if(p->str && p->str[0] != '/')
|
|
log_warn("control-interface %s is not using TLS, but plain transfer, because first control-interface in config file is a local socket (starts with a /).", p->str);
|
|
}
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
void daemon_remote_clear(struct daemon_remote* rc)
|
|
{
|
|
struct rc_state* p, *np;
|
|
if(!rc) return;
|
|
/* but do not close the ports */
|
|
listen_list_delete(rc->accept_list);
|
|
rc->accept_list = NULL;
|
|
/* do close these sockets */
|
|
p = rc->busy_list;
|
|
while(p) {
|
|
np = p->next;
|
|
if(p->ssl)
|
|
SSL_free(p->ssl);
|
|
comm_point_delete(p->c);
|
|
free(p);
|
|
p = np;
|
|
}
|
|
rc->busy_list = NULL;
|
|
rc->active = 0;
|
|
rc->worker = NULL;
|
|
}
|
|
|
|
void daemon_remote_delete(struct daemon_remote* rc)
|
|
{
|
|
if(!rc) return;
|
|
daemon_remote_clear(rc);
|
|
if(rc->ctx) {
|
|
SSL_CTX_free(rc->ctx);
|
|
}
|
|
free(rc);
|
|
}
|
|
|
|
/**
|
|
* Add and open a new control port
|
|
* @param ip: ip str
|
|
* @param nr: port nr
|
|
* @param list: list head
|
|
* @param noproto_is_err: if lack of protocol support is an error.
|
|
* @param cfg: config with username for chown of unix-sockets.
|
|
* @return false on failure.
|
|
*/
|
|
static int
|
|
add_open(const char* ip, int nr, struct listen_port** list, int noproto_is_err,
|
|
struct config_file* cfg)
|
|
{
|
|
struct addrinfo hints;
|
|
struct addrinfo* res;
|
|
struct listen_port* n;
|
|
int noproto = 0;
|
|
int fd, r;
|
|
char port[15];
|
|
snprintf(port, sizeof(port), "%d", nr);
|
|
port[sizeof(port)-1]=0;
|
|
memset(&hints, 0, sizeof(hints));
|
|
log_assert(ip);
|
|
|
|
if(ip[0] == '/') {
|
|
/* This looks like a local socket */
|
|
fd = create_local_accept_sock(ip, &noproto, cfg->use_systemd);
|
|
/*
|
|
* Change socket ownership and permissions so users other
|
|
* than root can access it provided they are in the same
|
|
* group as the user we run as.
|
|
*/
|
|
if(fd != -1) {
|
|
#ifdef HAVE_CHOWN
|
|
chmod(ip, (mode_t)(S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP));
|
|
if (cfg->username && cfg->username[0] &&
|
|
cfg_uid != (uid_t)-1) {
|
|
if(chown(ip, cfg_uid, cfg_gid) == -1)
|
|
verbose(VERB_QUERY, "cannot chown %u.%u %s: %s",
|
|
(unsigned)cfg_uid, (unsigned)cfg_gid,
|
|
ip, strerror(errno));
|
|
}
|
|
#else
|
|
(void)cfg;
|
|
#endif
|
|
}
|
|
} else {
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
hints.ai_flags = AI_PASSIVE | AI_NUMERICHOST;
|
|
if((r = getaddrinfo(ip, port, &hints, &res)) != 0 || !res) {
|
|
#ifdef USE_WINSOCK
|
|
if(!noproto_is_err && r == EAI_NONAME) {
|
|
/* tried to lookup the address as name */
|
|
return 1; /* return success, but do nothing */
|
|
}
|
|
#endif /* USE_WINSOCK */
|
|
log_err("control interface %s:%s getaddrinfo: %s %s",
|
|
ip?ip:"default", port, gai_strerror(r),
|
|
#ifdef EAI_SYSTEM
|
|
r==EAI_SYSTEM?(char*)strerror(errno):""
|
|
#else
|
|
""
|
|
#endif
|
|
);
|
|
return 0;
|
|
}
|
|
|
|
/* open fd */
|
|
fd = create_tcp_accept_sock(res, 1, &noproto, 0,
|
|
cfg->ip_transparent, 0, 0, cfg->ip_freebind,
|
|
cfg->use_systemd, cfg->ip_dscp, "unbound-control");
|
|
freeaddrinfo(res);
|
|
}
|
|
|
|
if(fd == -1 && noproto) {
|
|
if(!noproto_is_err)
|
|
return 1; /* return success, but do nothing */
|
|
log_err("cannot open control interface %s %d : "
|
|
"protocol not supported", ip, nr);
|
|
return 0;
|
|
}
|
|
if(fd == -1) {
|
|
log_err("cannot open control interface %s %d", ip, nr);
|
|
return 0;
|
|
}
|
|
|
|
/* alloc */
|
|
n = (struct listen_port*)calloc(1, sizeof(*n));
|
|
if(!n) {
|
|
sock_close(fd);
|
|
log_err("out of memory");
|
|
return 0;
|
|
}
|
|
n->next = *list;
|
|
*list = n;
|
|
n->fd = fd;
|
|
return 1;
|
|
}
|
|
|
|
struct listen_port* daemon_remote_open_ports(struct config_file* cfg)
|
|
{
|
|
struct listen_port* l = NULL;
|
|
log_assert(cfg->remote_control_enable && cfg->control_port);
|
|
if(cfg->control_ifs.first) {
|
|
char** rcif = NULL;
|
|
int i, num_rcif = 0;
|
|
if(!resolve_interface_names(NULL, 0, cfg->control_ifs.first,
|
|
&rcif, &num_rcif)) {
|
|
return NULL;
|
|
}
|
|
for(i=0; i<num_rcif; i++) {
|
|
if(!add_open(rcif[i], cfg->control_port, &l, 1, cfg)) {
|
|
listening_ports_free(l);
|
|
config_del_strarray(rcif, num_rcif);
|
|
return NULL;
|
|
}
|
|
}
|
|
config_del_strarray(rcif, num_rcif);
|
|
} else {
|
|
/* defaults */
|
|
if(cfg->do_ip6 &&
|
|
!add_open("::1", cfg->control_port, &l, 0, cfg)) {
|
|
listening_ports_free(l);
|
|
return NULL;
|
|
}
|
|
if(cfg->do_ip4 &&
|
|
!add_open("127.0.0.1", cfg->control_port, &l, 1, cfg)) {
|
|
listening_ports_free(l);
|
|
return NULL;
|
|
}
|
|
}
|
|
return l;
|
|
}
|
|
|
|
/** open accept commpoint */
|
|
static int
|
|
accept_open(struct daemon_remote* rc, int fd)
|
|
{
|
|
struct listen_list* n = (struct listen_list*)malloc(sizeof(*n));
|
|
if(!n) {
|
|
log_err("out of memory");
|
|
return 0;
|
|
}
|
|
n->next = rc->accept_list;
|
|
rc->accept_list = n;
|
|
/* open commpt */
|
|
n->com = comm_point_create_raw(rc->worker->base, fd, 0,
|
|
&remote_accept_callback, rc);
|
|
if(!n->com)
|
|
return 0;
|
|
/* keep this port open, its fd is kept in the rc portlist */
|
|
n->com->do_not_close = 1;
|
|
return 1;
|
|
}
|
|
|
|
int daemon_remote_open_accept(struct daemon_remote* rc,
|
|
struct listen_port* ports, struct worker* worker)
|
|
{
|
|
struct listen_port* p;
|
|
rc->worker = worker;
|
|
for(p = ports; p; p = p->next) {
|
|
if(!accept_open(rc, p->fd)) {
|
|
log_err("could not create accept comm point");
|
|
return 0;
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
void daemon_remote_stop_accept(struct daemon_remote* rc)
|
|
{
|
|
struct listen_list* p;
|
|
for(p=rc->accept_list; p; p=p->next) {
|
|
comm_point_stop_listening(p->com);
|
|
}
|
|
}
|
|
|
|
void daemon_remote_start_accept(struct daemon_remote* rc)
|
|
{
|
|
struct listen_list* p;
|
|
for(p=rc->accept_list; p; p=p->next) {
|
|
comm_point_start_listening(p->com, -1, -1);
|
|
}
|
|
}
|
|
|
|
int remote_accept_callback(struct comm_point* c, void* arg, int err,
|
|
struct comm_reply* ATTR_UNUSED(rep))
|
|
{
|
|
struct daemon_remote* rc = (struct daemon_remote*)arg;
|
|
struct sockaddr_storage addr;
|
|
socklen_t addrlen;
|
|
int newfd;
|
|
struct rc_state* n;
|
|
if(err != NETEVENT_NOERROR) {
|
|
log_err("error %d on remote_accept_callback", err);
|
|
return 0;
|
|
}
|
|
/* perform the accept */
|
|
newfd = comm_point_perform_accept(c, &addr, &addrlen);
|
|
if(newfd == -1)
|
|
return 0;
|
|
/* create new commpoint unless we are servicing already */
|
|
if(rc->active >= rc->max_active) {
|
|
log_warn("drop incoming remote control: too many connections");
|
|
close_exit:
|
|
sock_close(newfd);
|
|
return 0;
|
|
}
|
|
|
|
/* setup commpoint to service the remote control command */
|
|
n = (struct rc_state*)calloc(1, sizeof(*n));
|
|
if(!n) {
|
|
log_err("out of memory");
|
|
goto close_exit;
|
|
}
|
|
n->fd = newfd;
|
|
/* start in reading state */
|
|
n->c = comm_point_create_raw(rc->worker->base, newfd, 0,
|
|
&remote_control_callback, n);
|
|
if(!n->c) {
|
|
log_err("out of memory");
|
|
free(n);
|
|
goto close_exit;
|
|
}
|
|
log_addr(VERB_QUERY, "new control connection from", &addr, addrlen);
|
|
n->c->do_not_close = 0;
|
|
comm_point_stop_listening(n->c);
|
|
comm_point_start_listening(n->c, -1, REMOTE_CONTROL_TCP_TIMEOUT);
|
|
memcpy(&n->c->repinfo.remote_addr, &addr, addrlen);
|
|
n->c->repinfo.remote_addrlen = addrlen;
|
|
if(rc->use_cert) {
|
|
n->shake_state = rc_hs_read;
|
|
n->ssl = SSL_new(rc->ctx);
|
|
if(!n->ssl) {
|
|
log_crypto_err("could not SSL_new");
|
|
comm_point_delete(n->c);
|
|
free(n);
|
|
goto close_exit;
|
|
}
|
|
SSL_set_accept_state(n->ssl);
|
|
(void)SSL_set_mode(n->ssl, (long)SSL_MODE_AUTO_RETRY);
|
|
if(!SSL_set_fd(n->ssl, newfd)) {
|
|
log_crypto_err("could not SSL_set_fd");
|
|
SSL_free(n->ssl);
|
|
comm_point_delete(n->c);
|
|
free(n);
|
|
goto close_exit;
|
|
}
|
|
} else {
|
|
n->ssl = NULL;
|
|
}
|
|
|
|
n->rc = rc;
|
|
n->next = rc->busy_list;
|
|
rc->busy_list = n;
|
|
rc->active ++;
|
|
|
|
/* perform the first nonblocking read already, for windows,
|
|
* so it can return wouldblock. could be faster too. */
|
|
(void)remote_control_callback(n->c, n, NETEVENT_NOERROR, NULL);
|
|
return 0;
|
|
}
|
|
|
|
/** delete from list */
|
|
static void
|
|
state_list_remove_elem(struct rc_state** list, struct comm_point* c)
|
|
{
|
|
while(*list) {
|
|
if( (*list)->c == c) {
|
|
*list = (*list)->next;
|
|
return;
|
|
}
|
|
list = &(*list)->next;
|
|
}
|
|
}
|
|
|
|
/** decrease active count and remove commpoint from busy list */
|
|
static void
|
|
clean_point(struct daemon_remote* rc, struct rc_state* s)
|
|
{
|
|
state_list_remove_elem(&rc->busy_list, s->c);
|
|
rc->active --;
|
|
if(s->ssl) {
|
|
SSL_shutdown(s->ssl);
|
|
SSL_free(s->ssl);
|
|
}
|
|
comm_point_delete(s->c);
|
|
free(s);
|
|
}
|
|
|
|
int
|
|
ssl_print_text(RES* res, const char* text)
|
|
{
|
|
int r;
|
|
if(!res)
|
|
return 0;
|
|
if(res->ssl) {
|
|
ERR_clear_error();
|
|
if((r=SSL_write(res->ssl, text, (int)strlen(text))) <= 0) {
|
|
int r2;
|
|
if((r2=SSL_get_error(res->ssl, r)) == SSL_ERROR_ZERO_RETURN) {
|
|
verbose(VERB_QUERY, "warning, in SSL_write, peer "
|
|
"closed connection");
|
|
return 0;
|
|
}
|
|
log_crypto_err_io("could not SSL_write", r2);
|
|
return 0;
|
|
}
|
|
} else {
|
|
size_t at = 0;
|
|
while(at < strlen(text)) {
|
|
ssize_t r = send(res->fd, text+at, strlen(text)-at, 0);
|
|
if(r == -1) {
|
|
if(errno == EAGAIN || errno == EINTR)
|
|
continue;
|
|
log_err("could not send: %s",
|
|
sock_strerror(errno));
|
|
return 0;
|
|
}
|
|
at += r;
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/** print text over the ssl connection */
|
|
static int
|
|
ssl_print_vmsg(RES* ssl, const char* format, va_list args)
|
|
{
|
|
char msg[65535];
|
|
vsnprintf(msg, sizeof(msg), format, args);
|
|
return ssl_print_text(ssl, msg);
|
|
}
|
|
|
|
/** printf style printing to the ssl connection */
|
|
int ssl_printf(RES* ssl, const char* format, ...)
|
|
{
|
|
va_list args;
|
|
int ret;
|
|
va_start(args, format);
|
|
ret = ssl_print_vmsg(ssl, format, args);
|
|
va_end(args);
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
ssl_read_line(RES* res, char* buf, size_t max)
|
|
{
|
|
int r;
|
|
size_t len = 0;
|
|
if(!res)
|
|
return 0;
|
|
while(len < max) {
|
|
if(res->ssl) {
|
|
ERR_clear_error();
|
|
if((r=SSL_read(res->ssl, buf+len, 1)) <= 0) {
|
|
int r2;
|
|
if((r2=SSL_get_error(res->ssl, r)) == SSL_ERROR_ZERO_RETURN) {
|
|
buf[len] = 0;
|
|
return 1;
|
|
}
|
|
log_crypto_err_io("could not SSL_read", r2);
|
|
return 0;
|
|
}
|
|
} else {
|
|
while(1) {
|
|
ssize_t rr = recv(res->fd, buf+len, 1, 0);
|
|
if(rr <= 0) {
|
|
if(rr == 0) {
|
|
buf[len] = 0;
|
|
return 1;
|
|
}
|
|
if(errno == EINTR || errno == EAGAIN)
|
|
continue;
|
|
if(rr < 0) log_err("could not recv: %s",
|
|
sock_strerror(errno));
|
|
return 0;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if(buf[len] == '\n') {
|
|
/* return string without \n */
|
|
buf[len] = 0;
|
|
return 1;
|
|
}
|
|
len++;
|
|
}
|
|
buf[max-1] = 0;
|
|
log_err("control line too long (%d): %s", (int)max, buf);
|
|
return 0;
|
|
}
|
|
|
|
/** skip whitespace, return new pointer into string */
|
|
static char*
|
|
skipwhite(char* str)
|
|
{
|
|
/* EOS \0 is not a space */
|
|
while( isspace((unsigned char)*str) )
|
|
str++;
|
|
return str;
|
|
}
|
|
|
|
/** send the OK to the control client */
|
|
static void send_ok(RES* ssl)
|
|
{
|
|
(void)ssl_printf(ssl, "ok\n");
|
|
}
|
|
|
|
/** tell other processes to execute the command */
|
|
static void
|
|
distribute_cmd(struct daemon_remote* rc, RES* ssl, char* cmd)
|
|
{
|
|
int i;
|
|
if(!cmd || !ssl)
|
|
return;
|
|
/* skip i=0 which is me */
|
|
for(i=1; i<rc->worker->daemon->num; i++) {
|
|
worker_send_cmd(rc->worker->daemon->workers[i],
|
|
worker_cmd_remote);
|
|
if(!tube_write_msg(rc->worker->daemon->workers[i]->cmd,
|
|
(uint8_t*)cmd, strlen(cmd)+1, 0)) {
|
|
(void)ssl_printf(ssl, "error could not distribute cmd\n");
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
/** do the stop command */
|
|
static void
|
|
do_stop(RES* ssl, struct worker* worker)
|
|
{
|
|
worker->need_to_exit = 1;
|
|
comm_base_exit(worker->base);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the reload command */
|
|
static void
|
|
do_reload(RES* ssl, struct worker* worker, int reuse_cache)
|
|
{
|
|
worker->reuse_cache = reuse_cache;
|
|
worker->need_to_exit = 0;
|
|
comm_base_exit(worker->base);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the verbosity command */
|
|
static void
|
|
do_verbosity(RES* ssl, char* str)
|
|
{
|
|
int val = atoi(str);
|
|
if(val == 0 && strcmp(str, "0") != 0) {
|
|
ssl_printf(ssl, "error in verbosity number syntax: %s\n", str);
|
|
return;
|
|
}
|
|
verbosity = val;
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** print stats from statinfo */
|
|
static int
|
|
print_stats(RES* ssl, const char* nm, struct ub_stats_info* s)
|
|
{
|
|
struct timeval sumwait, avg;
|
|
if(!ssl_printf(ssl, "%s.num.queries"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.queries_ip_ratelimited"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_ip_ratelimited)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.queries_cookie_valid"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_cookie_valid)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.queries_cookie_client"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_cookie_client)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.queries_cookie_invalid"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_cookie_invalid)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.cachehits"SQ"%lu\n", nm,
|
|
(unsigned long)(s->svr.num_queries
|
|
- s->svr.num_queries_missed_cache))) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.cachemiss"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_missed_cache)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.prefetch"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_prefetch)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.queries_timed_out"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_queries_timed_out)) return 0;
|
|
if(!ssl_printf(ssl, "%s.query.queue_time_us.max"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.max_query_time_us)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.expired"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.ans_expired)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.recursivereplies"SQ"%lu\n", nm,
|
|
(unsigned long)s->mesh_replies_sent)) return 0;
|
|
#ifdef USE_DNSCRYPT
|
|
if(!ssl_printf(ssl, "%s.num.dnscrypt.crypted"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_query_dnscrypt_crypted)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.dnscrypt.cert"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_query_dnscrypt_cert)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.dnscrypt.cleartext"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_query_dnscrypt_cleartext)) return 0;
|
|
if(!ssl_printf(ssl, "%s.num.dnscrypt.malformed"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.num_query_dnscrypt_crypted_malformed)) return 0;
|
|
#endif
|
|
if(!ssl_printf(ssl, "%s.requestlist.avg"SQ"%g\n", nm,
|
|
(s->svr.num_queries_missed_cache+s->svr.num_queries_prefetch)?
|
|
(double)s->svr.sum_query_list_size/
|
|
(double)(s->svr.num_queries_missed_cache+
|
|
s->svr.num_queries_prefetch) : 0.0)) return 0;
|
|
if(!ssl_printf(ssl, "%s.requestlist.max"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.max_query_list_size)) return 0;
|
|
if(!ssl_printf(ssl, "%s.requestlist.overwritten"SQ"%lu\n", nm,
|
|
(unsigned long)s->mesh_jostled)) return 0;
|
|
if(!ssl_printf(ssl, "%s.requestlist.exceeded"SQ"%lu\n", nm,
|
|
(unsigned long)s->mesh_dropped)) return 0;
|
|
if(!ssl_printf(ssl, "%s.requestlist.current.all"SQ"%lu\n", nm,
|
|
(unsigned long)s->mesh_num_states)) return 0;
|
|
if(!ssl_printf(ssl, "%s.requestlist.current.user"SQ"%lu\n", nm,
|
|
(unsigned long)s->mesh_num_reply_states)) return 0;
|
|
#ifndef S_SPLINT_S
|
|
sumwait.tv_sec = s->mesh_replies_sum_wait_sec;
|
|
sumwait.tv_usec = s->mesh_replies_sum_wait_usec;
|
|
#endif
|
|
timeval_divide(&avg, &sumwait, s->mesh_replies_sent);
|
|
if(!ssl_printf(ssl, "%s.recursion.time.avg"SQ ARG_LL "d.%6.6d\n", nm,
|
|
(long long)avg.tv_sec, (int)avg.tv_usec)) return 0;
|
|
if(!ssl_printf(ssl, "%s.recursion.time.median"SQ"%g\n", nm,
|
|
s->mesh_time_median)) return 0;
|
|
if(!ssl_printf(ssl, "%s.tcpusage"SQ"%lu\n", nm,
|
|
(unsigned long)s->svr.tcp_accept_usage)) return 0;
|
|
return 1;
|
|
}
|
|
|
|
/** print stats for one thread */
|
|
static int
|
|
print_thread_stats(RES* ssl, int i, struct ub_stats_info* s)
|
|
{
|
|
char nm[32];
|
|
snprintf(nm, sizeof(nm), "thread%d", i);
|
|
nm[sizeof(nm)-1]=0;
|
|
return print_stats(ssl, nm, s);
|
|
}
|
|
|
|
/** print long number */
|
|
static int
|
|
print_longnum(RES* ssl, const char* desc, size_t x)
|
|
{
|
|
if(x > 1024*1024*1024) {
|
|
/* more than a Gb */
|
|
size_t front = x / (size_t)1000000;
|
|
size_t back = x % (size_t)1000000;
|
|
return ssl_printf(ssl, "%s%u%6.6u\n", desc,
|
|
(unsigned)front, (unsigned)back);
|
|
} else {
|
|
return ssl_printf(ssl, "%s%lu\n", desc, (unsigned long)x);
|
|
}
|
|
}
|
|
|
|
/** print mem stats */
|
|
static int
|
|
print_mem(RES* ssl, struct worker* worker, struct daemon* daemon,
|
|
struct ub_stats_info* s)
|
|
{
|
|
size_t msg, rrset, val, iter, respip;
|
|
#ifdef CLIENT_SUBNET
|
|
size_t subnet = 0;
|
|
#endif /* CLIENT_SUBNET */
|
|
#ifdef USE_IPSECMOD
|
|
size_t ipsecmod = 0;
|
|
#endif /* USE_IPSECMOD */
|
|
#ifdef USE_DNSCRYPT
|
|
size_t dnscrypt_shared_secret = 0;
|
|
size_t dnscrypt_nonce = 0;
|
|
#endif /* USE_DNSCRYPT */
|
|
#ifdef WITH_DYNLIBMODULE
|
|
size_t dynlib = 0;
|
|
#endif /* WITH_DYNLIBMODULE */
|
|
msg = slabhash_get_mem(daemon->env->msg_cache);
|
|
rrset = slabhash_get_mem(&daemon->env->rrset_cache->table);
|
|
val = mod_get_mem(&worker->env, "validator");
|
|
iter = mod_get_mem(&worker->env, "iterator");
|
|
respip = mod_get_mem(&worker->env, "respip");
|
|
#ifdef CLIENT_SUBNET
|
|
subnet = mod_get_mem(&worker->env, "subnetcache");
|
|
#endif /* CLIENT_SUBNET */
|
|
#ifdef USE_IPSECMOD
|
|
ipsecmod = mod_get_mem(&worker->env, "ipsecmod");
|
|
#endif /* USE_IPSECMOD */
|
|
#ifdef USE_DNSCRYPT
|
|
if(daemon->dnscenv) {
|
|
dnscrypt_shared_secret = slabhash_get_mem(
|
|
daemon->dnscenv->shared_secrets_cache);
|
|
dnscrypt_nonce = slabhash_get_mem(daemon->dnscenv->nonces_cache);
|
|
}
|
|
#endif /* USE_DNSCRYPT */
|
|
#ifdef WITH_DYNLIBMODULE
|
|
dynlib = mod_get_mem(&worker->env, "dynlib");
|
|
#endif /* WITH_DYNLIBMODULE */
|
|
|
|
if(!print_longnum(ssl, "mem.cache.rrset"SQ, rrset))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.cache.message"SQ, msg))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.mod.iterator"SQ, iter))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.mod.validator"SQ, val))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.mod.respip"SQ, respip))
|
|
return 0;
|
|
#ifdef CLIENT_SUBNET
|
|
if(!print_longnum(ssl, "mem.mod.subnet"SQ, subnet))
|
|
return 0;
|
|
#endif /* CLIENT_SUBNET */
|
|
#ifdef USE_IPSECMOD
|
|
if(!print_longnum(ssl, "mem.mod.ipsecmod"SQ, ipsecmod))
|
|
return 0;
|
|
#endif /* USE_IPSECMOD */
|
|
#ifdef USE_DNSCRYPT
|
|
if(!print_longnum(ssl, "mem.cache.dnscrypt_shared_secret"SQ,
|
|
dnscrypt_shared_secret))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.cache.dnscrypt_nonce"SQ,
|
|
dnscrypt_nonce))
|
|
return 0;
|
|
#endif /* USE_DNSCRYPT */
|
|
#ifdef WITH_DYNLIBMODULE
|
|
if(!print_longnum(ssl, "mem.mod.dynlibmod"SQ, dynlib))
|
|
return 0;
|
|
#endif /* WITH_DYNLIBMODULE */
|
|
if(!print_longnum(ssl, "mem.streamwait"SQ,
|
|
(size_t)s->svr.mem_stream_wait))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.http.query_buffer"SQ,
|
|
(size_t)s->svr.mem_http2_query_buffer))
|
|
return 0;
|
|
if(!print_longnum(ssl, "mem.http.response_buffer"SQ,
|
|
(size_t)s->svr.mem_http2_response_buffer))
|
|
return 0;
|
|
#ifdef HAVE_NGTCP2
|
|
if(!print_longnum(ssl, "mem.quic"SQ, (size_t)s->svr.mem_quic))
|
|
return 0;
|
|
#endif /* HAVE_NGTCP2 */
|
|
return 1;
|
|
}
|
|
|
|
/** print uptime stats */
|
|
static int
|
|
print_uptime(RES* ssl, struct worker* worker, int reset)
|
|
{
|
|
struct timeval now = *worker->env.now_tv;
|
|
struct timeval up, dt;
|
|
timeval_subtract(&up, &now, &worker->daemon->time_boot);
|
|
timeval_subtract(&dt, &now, &worker->daemon->time_last_stat);
|
|
if(reset)
|
|
worker->daemon->time_last_stat = now;
|
|
if(!ssl_printf(ssl, "time.now"SQ ARG_LL "d.%6.6d\n",
|
|
(long long)now.tv_sec, (unsigned)now.tv_usec)) return 0;
|
|
if(!ssl_printf(ssl, "time.up"SQ ARG_LL "d.%6.6d\n",
|
|
(long long)up.tv_sec, (unsigned)up.tv_usec)) return 0;
|
|
if(!ssl_printf(ssl, "time.elapsed"SQ ARG_LL "d.%6.6d\n",
|
|
(long long)dt.tv_sec, (unsigned)dt.tv_usec)) return 0;
|
|
return 1;
|
|
}
|
|
|
|
/** print extended histogram */
|
|
static int
|
|
print_hist(RES* ssl, struct ub_stats_info* s)
|
|
{
|
|
struct timehist* hist;
|
|
size_t i;
|
|
hist = timehist_setup();
|
|
if(!hist) {
|
|
log_err("out of memory");
|
|
return 0;
|
|
}
|
|
timehist_import(hist, s->svr.hist, NUM_BUCKETS_HIST);
|
|
for(i=0; i<hist->num; i++) {
|
|
if(!ssl_printf(ssl,
|
|
"histogram.%6.6d.%6.6d.to.%6.6d.%6.6d=%lu\n",
|
|
(int)hist->buckets[i].lower.tv_sec,
|
|
(int)hist->buckets[i].lower.tv_usec,
|
|
(int)hist->buckets[i].upper.tv_sec,
|
|
(int)hist->buckets[i].upper.tv_usec,
|
|
(unsigned long)hist->buckets[i].count)) {
|
|
timehist_delete(hist);
|
|
return 0;
|
|
}
|
|
}
|
|
timehist_delete(hist);
|
|
return 1;
|
|
}
|
|
|
|
/** print extended stats */
|
|
static int
|
|
print_ext(RES* ssl, struct ub_stats_info* s, int inhibit_zero)
|
|
{
|
|
int i;
|
|
char nm[32];
|
|
const sldns_rr_descriptor* desc;
|
|
const sldns_lookup_table* lt;
|
|
/* TYPE */
|
|
for(i=0; i<UB_STATS_QTYPE_NUM; i++) {
|
|
if(inhibit_zero && s->svr.qtype[i] == 0)
|
|
continue;
|
|
desc = sldns_rr_descript((uint16_t)i);
|
|
if(desc && desc->_name) {
|
|
snprintf(nm, sizeof(nm), "%s", desc->_name);
|
|
} else if (i == LDNS_RR_TYPE_IXFR) {
|
|
snprintf(nm, sizeof(nm), "IXFR");
|
|
} else if (i == LDNS_RR_TYPE_AXFR) {
|
|
snprintf(nm, sizeof(nm), "AXFR");
|
|
} else if (i == LDNS_RR_TYPE_MAILA) {
|
|
snprintf(nm, sizeof(nm), "MAILA");
|
|
} else if (i == LDNS_RR_TYPE_MAILB) {
|
|
snprintf(nm, sizeof(nm), "MAILB");
|
|
} else if (i == LDNS_RR_TYPE_ANY) {
|
|
snprintf(nm, sizeof(nm), "ANY");
|
|
} else {
|
|
snprintf(nm, sizeof(nm), "TYPE%d", i);
|
|
}
|
|
if(!ssl_printf(ssl, "num.query.type.%s"SQ"%lu\n",
|
|
nm, (unsigned long)s->svr.qtype[i])) return 0;
|
|
}
|
|
if(!inhibit_zero || s->svr.qtype_big) {
|
|
if(!ssl_printf(ssl, "num.query.type.other"SQ"%lu\n",
|
|
(unsigned long)s->svr.qtype_big)) return 0;
|
|
}
|
|
/* CLASS */
|
|
for(i=0; i<UB_STATS_QCLASS_NUM; i++) {
|
|
if(inhibit_zero && s->svr.qclass[i] == 0)
|
|
continue;
|
|
lt = sldns_lookup_by_id(sldns_rr_classes, i);
|
|
if(lt && lt->name) {
|
|
snprintf(nm, sizeof(nm), "%s", lt->name);
|
|
} else {
|
|
snprintf(nm, sizeof(nm), "CLASS%d", i);
|
|
}
|
|
if(!ssl_printf(ssl, "num.query.class.%s"SQ"%lu\n",
|
|
nm, (unsigned long)s->svr.qclass[i])) return 0;
|
|
}
|
|
if(!inhibit_zero || s->svr.qclass_big) {
|
|
if(!ssl_printf(ssl, "num.query.class.other"SQ"%lu\n",
|
|
(unsigned long)s->svr.qclass_big)) return 0;
|
|
}
|
|
/* OPCODE */
|
|
for(i=0; i<UB_STATS_OPCODE_NUM; i++) {
|
|
if(inhibit_zero && s->svr.qopcode[i] == 0)
|
|
continue;
|
|
lt = sldns_lookup_by_id(sldns_opcodes, i);
|
|
if(lt && lt->name) {
|
|
snprintf(nm, sizeof(nm), "%s", lt->name);
|
|
} else {
|
|
snprintf(nm, sizeof(nm), "OPCODE%d", i);
|
|
}
|
|
if(!ssl_printf(ssl, "num.query.opcode.%s"SQ"%lu\n",
|
|
nm, (unsigned long)s->svr.qopcode[i])) return 0;
|
|
}
|
|
/* transport */
|
|
if(!ssl_printf(ssl, "num.query.tcp"SQ"%lu\n",
|
|
(unsigned long)s->svr.qtcp)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.tcpout"SQ"%lu\n",
|
|
(unsigned long)s->svr.qtcp_outgoing)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.udpout"SQ"%lu\n",
|
|
(unsigned long)s->svr.qudp_outgoing)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.tls"SQ"%lu\n",
|
|
(unsigned long)s->svr.qtls)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.tls.resume"SQ"%lu\n",
|
|
(unsigned long)s->svr.qtls_resume)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.ipv6"SQ"%lu\n",
|
|
(unsigned long)s->svr.qipv6)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.https"SQ"%lu\n",
|
|
(unsigned long)s->svr.qhttps)) return 0;
|
|
#ifdef HAVE_NGTCP2
|
|
if(!ssl_printf(ssl, "num.query.quic"SQ"%lu\n",
|
|
(unsigned long)s->svr.qquic)) return 0;
|
|
#endif /* HAVE_NGTCP2 */
|
|
/* flags */
|
|
if(!ssl_printf(ssl, "num.query.flags.QR"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_QR)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.AA"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_AA)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.TC"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_TC)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.RD"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_RD)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.RA"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_RA)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.Z"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_Z)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.AD"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_AD)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.flags.CD"SQ"%lu\n",
|
|
(unsigned long)s->svr.qbit_CD)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.edns.present"SQ"%lu\n",
|
|
(unsigned long)s->svr.qEDNS)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.edns.DO"SQ"%lu\n",
|
|
(unsigned long)s->svr.qEDNS_DO)) return 0;
|
|
|
|
/* RCODE */
|
|
for(i=0; i<UB_STATS_RCODE_NUM; i++) {
|
|
/* Always include RCODEs 0-5 */
|
|
if(inhibit_zero && i > LDNS_RCODE_REFUSED && s->svr.ans_rcode[i] == 0)
|
|
continue;
|
|
lt = sldns_lookup_by_id(sldns_rcodes, i);
|
|
if(lt && lt->name) {
|
|
snprintf(nm, sizeof(nm), "%s", lt->name);
|
|
} else {
|
|
snprintf(nm, sizeof(nm), "RCODE%d", i);
|
|
}
|
|
if(!ssl_printf(ssl, "num.answer.rcode.%s"SQ"%lu\n",
|
|
nm, (unsigned long)s->svr.ans_rcode[i])) return 0;
|
|
}
|
|
if(!inhibit_zero || s->svr.ans_rcode_nodata) {
|
|
if(!ssl_printf(ssl, "num.answer.rcode.nodata"SQ"%lu\n",
|
|
(unsigned long)s->svr.ans_rcode_nodata)) return 0;
|
|
}
|
|
/* iteration */
|
|
if(!ssl_printf(ssl, "num.query.ratelimited"SQ"%lu\n",
|
|
(unsigned long)s->svr.queries_ratelimited)) return 0;
|
|
/* validation */
|
|
if(!ssl_printf(ssl, "num.answer.secure"SQ"%lu\n",
|
|
(unsigned long)s->svr.ans_secure)) return 0;
|
|
if(!ssl_printf(ssl, "num.answer.bogus"SQ"%lu\n",
|
|
(unsigned long)s->svr.ans_bogus)) return 0;
|
|
if(!ssl_printf(ssl, "num.rrset.bogus"SQ"%lu\n",
|
|
(unsigned long)s->svr.rrset_bogus)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.aggressive.NOERROR"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_neg_cache_noerror)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.aggressive.NXDOMAIN"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_neg_cache_nxdomain)) return 0;
|
|
/* threat detection */
|
|
if(!ssl_printf(ssl, "unwanted.queries"SQ"%lu\n",
|
|
(unsigned long)s->svr.unwanted_queries)) return 0;
|
|
if(!ssl_printf(ssl, "unwanted.replies"SQ"%lu\n",
|
|
(unsigned long)s->svr.unwanted_replies)) return 0;
|
|
/* cache counts */
|
|
if(!ssl_printf(ssl, "msg.cache.count"SQ"%u\n",
|
|
(unsigned)s->svr.msg_cache_count)) return 0;
|
|
if(!ssl_printf(ssl, "rrset.cache.count"SQ"%u\n",
|
|
(unsigned)s->svr.rrset_cache_count)) return 0;
|
|
if(!ssl_printf(ssl, "infra.cache.count"SQ"%u\n",
|
|
(unsigned)s->svr.infra_cache_count)) return 0;
|
|
if(!ssl_printf(ssl, "key.cache.count"SQ"%u\n",
|
|
(unsigned)s->svr.key_cache_count)) return 0;
|
|
/* max collisions */
|
|
if(!ssl_printf(ssl, "msg.cache.max_collisions"SQ"%u\n",
|
|
(unsigned)s->svr.msg_cache_max_collisions)) return 0;
|
|
if(!ssl_printf(ssl, "rrset.cache.max_collisions"SQ"%u\n",
|
|
(unsigned)s->svr.rrset_cache_max_collisions)) return 0;
|
|
/* applied RPZ actions */
|
|
for(i=0; i<UB_STATS_RPZ_ACTION_NUM; i++) {
|
|
if(i == RPZ_NO_OVERRIDE_ACTION)
|
|
continue;
|
|
if(inhibit_zero && s->svr.rpz_action[i] == 0)
|
|
continue;
|
|
if(!ssl_printf(ssl, "num.rpz.action.%s"SQ"%lu\n",
|
|
rpz_action_to_string(i),
|
|
(unsigned long)s->svr.rpz_action[i])) return 0;
|
|
}
|
|
#ifdef USE_DNSCRYPT
|
|
if(!ssl_printf(ssl, "dnscrypt_shared_secret.cache.count"SQ"%u\n",
|
|
(unsigned)s->svr.shared_secret_cache_count)) return 0;
|
|
if(!ssl_printf(ssl, "dnscrypt_nonce.cache.count"SQ"%u\n",
|
|
(unsigned)s->svr.nonce_cache_count)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.dnscrypt.shared_secret.cachemiss"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_dnscrypt_secret_missed_cache)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.dnscrypt.replay"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_dnscrypt_replay)) return 0;
|
|
#endif /* USE_DNSCRYPT */
|
|
if(!ssl_printf(ssl, "num.query.authzone.up"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_authzone_up)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.authzone.down"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_authzone_down)) return 0;
|
|
#ifdef CLIENT_SUBNET
|
|
if(!ssl_printf(ssl, "num.query.subnet"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_subnet)) return 0;
|
|
if(!ssl_printf(ssl, "num.query.subnet_cache"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_subnet_cache)) return 0;
|
|
#endif /* CLIENT_SUBNET */
|
|
#ifdef USE_CACHEDB
|
|
if(!ssl_printf(ssl, "num.query.cachedb"SQ"%lu\n",
|
|
(unsigned long)s->svr.num_query_cachedb)) return 0;
|
|
#endif /* USE_CACHEDB */
|
|
return 1;
|
|
}
|
|
|
|
/** do the stats command */
|
|
static void
|
|
do_stats(RES* ssl, struct worker* worker, int reset)
|
|
{
|
|
struct daemon* daemon = worker->daemon;
|
|
struct ub_stats_info total;
|
|
struct ub_stats_info s;
|
|
int i;
|
|
memset(&total, 0, sizeof(total));
|
|
log_assert(daemon->num > 0);
|
|
/* gather all thread statistics in one place */
|
|
for(i=0; i<daemon->num; i++) {
|
|
server_stats_obtain(worker, daemon->workers[i], &s, reset);
|
|
if(!print_thread_stats(ssl, i, &s))
|
|
return;
|
|
if(i == 0)
|
|
total = s;
|
|
else server_stats_add(&total, &s);
|
|
}
|
|
/* print the thread statistics */
|
|
total.mesh_time_median /= (double)daemon->num;
|
|
if(!print_stats(ssl, "total", &total))
|
|
return;
|
|
if(!print_uptime(ssl, worker, reset))
|
|
return;
|
|
if(daemon->cfg->stat_extended) {
|
|
if(!print_mem(ssl, worker, daemon, &total))
|
|
return;
|
|
if(!print_hist(ssl, &total))
|
|
return;
|
|
if(!print_ext(ssl, &total, daemon->cfg->stat_inhibit_zero))
|
|
return;
|
|
}
|
|
}
|
|
|
|
/** parse commandline argument domain name */
|
|
static int
|
|
parse_arg_name(RES* ssl, char* str, uint8_t** res, size_t* len, int* labs)
|
|
{
|
|
uint8_t nm[LDNS_MAX_DOMAINLEN+1];
|
|
size_t nmlen = sizeof(nm);
|
|
int status;
|
|
*res = NULL;
|
|
*len = 0;
|
|
*labs = 0;
|
|
if(str[0] == '\0') {
|
|
ssl_printf(ssl, "error: this option requires a domain name\n");
|
|
return 0;
|
|
}
|
|
status = sldns_str2wire_dname_buf(str, nm, &nmlen);
|
|
if(status != 0) {
|
|
ssl_printf(ssl, "error cannot parse name %s at %d: %s\n", str,
|
|
LDNS_WIREPARSE_OFFSET(status),
|
|
sldns_get_errorstr_parse(status));
|
|
return 0;
|
|
}
|
|
*res = memdup(nm, nmlen);
|
|
if(!*res) {
|
|
ssl_printf(ssl, "error out of memory\n");
|
|
return 0;
|
|
}
|
|
*labs = dname_count_size_labels(*res, len);
|
|
return 1;
|
|
}
|
|
|
|
/** find second argument, modifies string */
|
|
static int
|
|
find_arg2(RES* ssl, char* arg, char** arg2)
|
|
{
|
|
char* as = strchr(arg, ' ');
|
|
char* at = strchr(arg, '\t');
|
|
if(as && at) {
|
|
if(at < as)
|
|
as = at;
|
|
as[0]=0;
|
|
*arg2 = skipwhite(as+1);
|
|
} else if(as) {
|
|
as[0]=0;
|
|
*arg2 = skipwhite(as+1);
|
|
} else if(at) {
|
|
at[0]=0;
|
|
*arg2 = skipwhite(at+1);
|
|
} else {
|
|
ssl_printf(ssl, "error could not find next argument "
|
|
"after %s\n", arg);
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/** Add a new zone */
|
|
static int
|
|
perform_zone_add(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
char* arg2;
|
|
enum localzone_type t;
|
|
struct local_zone* z;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return 0;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return 0;
|
|
if(!local_zone_str2type(arg2, &t)) {
|
|
ssl_printf(ssl, "error not a zone type. %s\n", arg2);
|
|
free(nm);
|
|
return 0;
|
|
}
|
|
lock_rw_wrlock(&zones->lock);
|
|
if((z=local_zones_find(zones, nm, nmlen,
|
|
nmlabs, LDNS_RR_CLASS_IN))) {
|
|
/* already present in tree */
|
|
lock_rw_wrlock(&z->lock);
|
|
z->type = t; /* update type anyway */
|
|
lock_rw_unlock(&z->lock);
|
|
free(nm);
|
|
lock_rw_unlock(&zones->lock);
|
|
return 1;
|
|
}
|
|
if(!local_zones_add_zone(zones, nm, nmlen,
|
|
nmlabs, LDNS_RR_CLASS_IN, t)) {
|
|
lock_rw_unlock(&zones->lock);
|
|
ssl_printf(ssl, "error out of memory\n");
|
|
return 0;
|
|
}
|
|
lock_rw_unlock(&zones->lock);
|
|
return 1;
|
|
}
|
|
|
|
/** Do the local_zone command */
|
|
static void
|
|
do_zone_add(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
if(!perform_zone_add(ssl, zones, arg))
|
|
return;
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Do the local_zones command */
|
|
static void
|
|
do_zones_add(struct daemon_remote* rc, RES* ssl, struct worker* worker)
|
|
{
|
|
char buf[MAX_CMD_STRLINE + MAX_STDIN_STRLINE] = "local_zone ";
|
|
int num = 0;
|
|
size_t cmd_len = strlen(buf);
|
|
while(ssl_read_line(ssl, buf+cmd_len, MAX_STDIN_STRLINE)) {
|
|
if(buf[0+cmd_len] == 0 ||
|
|
(buf[0+cmd_len] == 0x04 && buf[1+cmd_len] == 0))
|
|
break; /* zero byte line or end of transmission */
|
|
#ifdef THREADS_DISABLED
|
|
/* distribute single item command */
|
|
if(rc) distribute_cmd(rc, ssl, buf);
|
|
#else
|
|
(void)rc; /* unused */
|
|
#endif
|
|
if(!perform_zone_add(ssl, worker->daemon->local_zones,
|
|
buf+cmd_len)) {
|
|
if(!ssl_printf(ssl, "error for input line: %s\n",
|
|
buf+cmd_len))
|
|
return;
|
|
}
|
|
else num++;
|
|
}
|
|
(void)ssl_printf(ssl, "added %d zones\n", num);
|
|
}
|
|
|
|
/** Remove a zone */
|
|
static int
|
|
perform_zone_remove(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
struct local_zone* z;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return 0;
|
|
lock_rw_wrlock(&zones->lock);
|
|
if((z=local_zones_find(zones, nm, nmlen,
|
|
nmlabs, LDNS_RR_CLASS_IN))) {
|
|
/* present in tree */
|
|
local_zones_del_zone(zones, z);
|
|
}
|
|
lock_rw_unlock(&zones->lock);
|
|
free(nm);
|
|
return 1;
|
|
}
|
|
|
|
/** Do the local_zone_remove command */
|
|
static void
|
|
do_zone_remove(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
if(!perform_zone_remove(ssl, zones, arg))
|
|
return;
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Do the local_zones_remove command */
|
|
static void
|
|
do_zones_remove(struct daemon_remote* rc, RES* ssl, struct worker* worker)
|
|
{
|
|
char buf[MAX_CMD_STRLINE + MAX_STDIN_STRLINE] = "local_zone_remove ";
|
|
int num = 0;
|
|
size_t cmd_len = strlen(buf);
|
|
while(ssl_read_line(ssl, buf+cmd_len, MAX_STDIN_STRLINE)) {
|
|
if(buf[0+cmd_len] == 0 ||
|
|
(buf[0+cmd_len] == 0x04 && buf[1+cmd_len] == 0))
|
|
break; /* zero byte line or end of transmission */
|
|
#ifdef THREADS_DISABLED
|
|
/* distribute single item command */
|
|
if(rc) distribute_cmd(rc, ssl, buf);
|
|
#else
|
|
(void)rc; /* unused */
|
|
#endif
|
|
if(!perform_zone_remove(ssl, worker->daemon->local_zones,
|
|
buf+cmd_len)) {
|
|
if(!ssl_printf(ssl, "error for input line: %s\n",
|
|
buf+cmd_len))
|
|
return;
|
|
}
|
|
else num++;
|
|
}
|
|
(void)ssl_printf(ssl, "removed %d zones\n", num);
|
|
}
|
|
|
|
/** check syntax of newly added RR */
|
|
static int
|
|
check_RR_syntax(RES* ssl, char* str, int line)
|
|
{
|
|
uint8_t rr[LDNS_RR_BUF_SIZE];
|
|
size_t len = sizeof(rr), dname_len = 0;
|
|
int s = sldns_str2wire_rr_buf(str, rr, &len, &dname_len, 3600,
|
|
NULL, 0, NULL, 0);
|
|
if(s != 0) {
|
|
char linestr[32];
|
|
if(line == 0)
|
|
linestr[0]=0;
|
|
else snprintf(linestr, sizeof(linestr), "line %d ", line);
|
|
if(!ssl_printf(ssl, "error parsing local-data at %sposition %d '%s': %s\n",
|
|
linestr, LDNS_WIREPARSE_OFFSET(s), str,
|
|
sldns_get_errorstr_parse(s)))
|
|
return 0;
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/** Add new RR data */
|
|
static int
|
|
perform_data_add(RES* ssl, struct local_zones* zones, char* arg, int line)
|
|
{
|
|
if(!check_RR_syntax(ssl, arg, line)) {
|
|
return 0;
|
|
}
|
|
if(!local_zones_add_RR(zones, arg)) {
|
|
ssl_printf(ssl,"error in syntax or out of memory, %s\n", arg);
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/** Do the local_data command */
|
|
static void
|
|
do_data_add(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
if(!perform_data_add(ssl, zones, arg, 0))
|
|
return;
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Do the local_datas command */
|
|
static void
|
|
do_datas_add(struct daemon_remote* rc, RES* ssl, struct worker* worker)
|
|
{
|
|
char buf[MAX_CMD_STRLINE + MAX_STDIN_STRLINE] = "local_data ";
|
|
int num = 0, line = 0;
|
|
size_t cmd_len = strlen(buf);
|
|
while(ssl_read_line(ssl, buf+cmd_len, MAX_STDIN_STRLINE)) {
|
|
if(buf[0+cmd_len] == 0 ||
|
|
(buf[0+cmd_len] == 0x04 && buf[1+cmd_len] == 0))
|
|
break; /* zero byte line or end of transmission */
|
|
#ifdef THREADS_DISABLED
|
|
/* distribute single item command */
|
|
if(rc) distribute_cmd(rc, ssl, buf);
|
|
#else
|
|
(void)rc; /* unused */
|
|
#endif
|
|
line++;
|
|
if(perform_data_add(ssl, worker->daemon->local_zones,
|
|
buf+cmd_len, line))
|
|
num++;
|
|
}
|
|
(void)ssl_printf(ssl, "added %d datas\n", num);
|
|
}
|
|
|
|
/** Remove RR data */
|
|
static int
|
|
perform_data_remove(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return 0;
|
|
local_zones_del_data(zones, nm,
|
|
nmlen, nmlabs, LDNS_RR_CLASS_IN);
|
|
free(nm);
|
|
return 1;
|
|
}
|
|
|
|
/** Do the local_data_remove command */
|
|
static void
|
|
do_data_remove(RES* ssl, struct local_zones* zones, char* arg)
|
|
{
|
|
if(!perform_data_remove(ssl, zones, arg))
|
|
return;
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Do the local_datas_remove command */
|
|
static void
|
|
do_datas_remove(struct daemon_remote* rc, RES* ssl, struct worker* worker)
|
|
{
|
|
char buf[MAX_CMD_STRLINE + MAX_STDIN_STRLINE] = "local_data_remove ";
|
|
int num = 0;
|
|
size_t cmd_len = strlen(buf);
|
|
while(ssl_read_line(ssl, buf+cmd_len, MAX_STDIN_STRLINE)) {
|
|
if(buf[0+cmd_len] == 0 ||
|
|
(buf[0+cmd_len] == 0x04 && buf[1+cmd_len] == 0))
|
|
break; /* zero byte line or end of transmission */
|
|
#ifdef THREADS_DISABLED
|
|
/* distribute single item command */
|
|
if(rc) distribute_cmd(rc, ssl, buf);
|
|
#else
|
|
(void)rc; /* unused */
|
|
#endif
|
|
if(!perform_data_remove(ssl, worker->daemon->local_zones,
|
|
buf+cmd_len)) {
|
|
if(!ssl_printf(ssl, "error for input line: %s\n",
|
|
buf+cmd_len))
|
|
return;
|
|
}
|
|
else num++;
|
|
}
|
|
(void)ssl_printf(ssl, "removed %d datas\n", num);
|
|
}
|
|
|
|
/** Add a new zone to view */
|
|
static void
|
|
do_view_zone_add(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
char* arg2;
|
|
struct view* v;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return;
|
|
v = views_find_view(worker->daemon->views,
|
|
arg, 1 /* get write lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(!v->local_zones) {
|
|
if(!(v->local_zones = local_zones_create())){
|
|
lock_rw_unlock(&v->lock);
|
|
ssl_printf(ssl,"error out of memory\n");
|
|
return;
|
|
}
|
|
if(!v->isfirst) {
|
|
/* Global local-zone is not used for this view,
|
|
* therefore add defaults to this view-specic
|
|
* local-zone. */
|
|
struct config_file lz_cfg;
|
|
memset(&lz_cfg, 0, sizeof(lz_cfg));
|
|
local_zone_enter_defaults(v->local_zones, &lz_cfg);
|
|
}
|
|
}
|
|
do_zone_add(ssl, v->local_zones, arg2);
|
|
lock_rw_unlock(&v->lock);
|
|
}
|
|
|
|
/** Remove a zone from view */
|
|
static void
|
|
do_view_zone_remove(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
char* arg2;
|
|
struct view* v;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return;
|
|
v = views_find_view(worker->daemon->views,
|
|
arg, 1 /* get write lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(!v->local_zones) {
|
|
lock_rw_unlock(&v->lock);
|
|
send_ok(ssl);
|
|
return;
|
|
}
|
|
do_zone_remove(ssl, v->local_zones, arg2);
|
|
lock_rw_unlock(&v->lock);
|
|
}
|
|
|
|
/** Add new RR data to view */
|
|
static void
|
|
do_view_data_add(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
char* arg2;
|
|
struct view* v;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return;
|
|
v = views_find_view(worker->daemon->views,
|
|
arg, 1 /* get write lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(!v->local_zones) {
|
|
if(!(v->local_zones = local_zones_create())){
|
|
lock_rw_unlock(&v->lock);
|
|
ssl_printf(ssl,"error out of memory\n");
|
|
return;
|
|
}
|
|
}
|
|
do_data_add(ssl, v->local_zones, arg2);
|
|
lock_rw_unlock(&v->lock);
|
|
}
|
|
|
|
/** Add new RR data from stdin to view */
|
|
static void
|
|
do_view_datas_add(struct daemon_remote* rc, RES* ssl, struct worker* worker,
|
|
char* arg)
|
|
{
|
|
struct view* v;
|
|
char buf[MAX_CMD_STRLINE + MAX_STDIN_STRLINE] = "view_local_data ";
|
|
size_t cmd_len;
|
|
int num = 0, line = 0;
|
|
v = views_find_view(worker->daemon->views,
|
|
arg, 1 /* get write lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(!v->local_zones) {
|
|
if(!(v->local_zones = local_zones_create())){
|
|
lock_rw_unlock(&v->lock);
|
|
ssl_printf(ssl,"error out of memory\n");
|
|
return;
|
|
}
|
|
}
|
|
/* put the view name in the command buf */
|
|
(void)snprintf(buf+strlen(buf), sizeof(buf)-strlen(buf), "%s ", arg);
|
|
cmd_len = strlen(buf);
|
|
while(ssl_read_line(ssl, buf+cmd_len, MAX_STDIN_STRLINE)) {
|
|
if(buf[0+cmd_len] == 0 ||
|
|
(buf[0+cmd_len] == 0x04 && buf[1+cmd_len] == 0))
|
|
break; /* zero byte line or end of transmission */
|
|
#ifdef THREADS_DISABLED
|
|
/* distribute single item command */
|
|
if(rc) distribute_cmd(rc, ssl, buf);
|
|
#else
|
|
(void)rc; /* unused */
|
|
#endif
|
|
line++;
|
|
if(perform_data_add(ssl, v->local_zones, buf+cmd_len, line))
|
|
num++;
|
|
}
|
|
lock_rw_unlock(&v->lock);
|
|
(void)ssl_printf(ssl, "added %d datas\n", num);
|
|
}
|
|
|
|
/** Remove RR data from view */
|
|
static void
|
|
do_view_data_remove(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
char* arg2;
|
|
struct view* v;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return;
|
|
v = views_find_view(worker->daemon->views,
|
|
arg, 1 /* get write lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(!v->local_zones) {
|
|
lock_rw_unlock(&v->lock);
|
|
send_ok(ssl);
|
|
return;
|
|
}
|
|
do_data_remove(ssl, v->local_zones, arg2);
|
|
lock_rw_unlock(&v->lock);
|
|
}
|
|
|
|
/** Remove RR data from stdin from view */
|
|
static void
|
|
do_view_datas_remove(struct daemon_remote* rc, RES* ssl, struct worker* worker,
|
|
char* arg)
|
|
{
|
|
struct view* v;
|
|
char buf[MAX_CMD_STRLINE + MAX_STDIN_STRLINE] = "view_local_data_remove ";
|
|
int num = 0;
|
|
size_t cmd_len;
|
|
v = views_find_view(worker->daemon->views,
|
|
arg, 1 /* get write lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(!v->local_zones){
|
|
lock_rw_unlock(&v->lock);
|
|
ssl_printf(ssl, "removed 0 datas\n");
|
|
return;
|
|
}
|
|
/* put the view name in the command buf */
|
|
(void)snprintf(buf+strlen(buf), sizeof(buf)-strlen(buf), "%s ", arg);
|
|
cmd_len = strlen(buf);
|
|
while(ssl_read_line(ssl, buf+cmd_len, MAX_STDIN_STRLINE)) {
|
|
if(buf[0+cmd_len] == 0 ||
|
|
(buf[0+cmd_len] == 0x04 && buf[1+cmd_len] == 0))
|
|
break; /* zero byte line or end of transmission */
|
|
#ifdef THREADS_DISABLED
|
|
/* distribute single item command */
|
|
if(rc) distribute_cmd(rc, ssl, buf);
|
|
#else
|
|
(void)rc; /* unused */
|
|
#endif
|
|
if(!perform_data_remove(ssl, v->local_zones, buf+cmd_len)) {
|
|
if(!ssl_printf(ssl, "error for input line: %s\n",
|
|
buf+cmd_len))
|
|
return;
|
|
}
|
|
else num++;
|
|
}
|
|
lock_rw_unlock(&v->lock);
|
|
(void)ssl_printf(ssl, "removed %d datas\n", num);
|
|
}
|
|
|
|
/** cache lookup of nameservers */
|
|
static void
|
|
do_lookup(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
(void)print_deleg_lookup(ssl, worker, nm, nmlen, nmlabs);
|
|
free(nm);
|
|
}
|
|
|
|
/** flush something from rrset and msg caches */
|
|
static void
|
|
do_cache_remove(struct worker* worker, uint8_t* nm, size_t nmlen,
|
|
uint16_t t, uint16_t c, int remcachedb)
|
|
{
|
|
hashvalue_type h;
|
|
struct query_info k;
|
|
rrset_cache_remove(worker->env.rrset_cache, nm, nmlen, t, c, 0);
|
|
if(t == LDNS_RR_TYPE_SOA)
|
|
rrset_cache_remove(worker->env.rrset_cache, nm, nmlen, t, c,
|
|
PACKED_RRSET_SOA_NEG);
|
|
k.qname = nm;
|
|
k.qname_len = nmlen;
|
|
k.qtype = t;
|
|
k.qclass = c;
|
|
k.local_alias = NULL;
|
|
h = query_info_hash(&k, 0);
|
|
slabhash_remove(worker->env.msg_cache, h, &k);
|
|
if(t == LDNS_RR_TYPE_AAAA) {
|
|
/* for AAAA also flush dns64 bit_cd packet */
|
|
h = query_info_hash(&k, BIT_CD);
|
|
slabhash_remove(worker->env.msg_cache, h, &k);
|
|
}
|
|
#ifdef USE_CACHEDB
|
|
if(remcachedb && worker->env.cachedb_enabled)
|
|
cachedb_msg_remove_qinfo(&worker->env, &k);
|
|
#else
|
|
(void)remcachedb;
|
|
#endif
|
|
}
|
|
|
|
/** parse '+c' option, modifies string to return remainder. */
|
|
static int
|
|
parse_remcachedb(RES* ssl, char** arg, int* pc)
|
|
{
|
|
*arg = skipwhite(*arg);
|
|
if((*arg)[0] == '+' && (*arg)[1] == 'c') {
|
|
char* arg2;
|
|
*pc = 1;
|
|
if(!find_arg2(ssl, *arg, &arg2))
|
|
return 0;
|
|
*arg = arg2;
|
|
return 1;
|
|
}
|
|
/* The option was not found, no problem */
|
|
return 1;
|
|
}
|
|
|
|
/** flush a type */
|
|
static void
|
|
do_flush_type(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
char* arg2;
|
|
uint16_t t;
|
|
int pc = 0; /* '+c' option */
|
|
if(!parse_remcachedb(ssl, &arg, &pc))
|
|
return;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
t = sldns_get_rr_type_by_name(arg2);
|
|
if(t == 0 && strcmp(arg2, "TYPE0") != 0) {
|
|
(void)ssl_printf(ssl, "error parsing RRset type: '%s'\n", arg2);
|
|
free(nm);
|
|
return;
|
|
}
|
|
do_cache_remove(worker, nm, nmlen, t, LDNS_RR_CLASS_IN, pc);
|
|
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** flush statistics */
|
|
static void
|
|
do_flush_stats(RES* ssl, struct worker* worker)
|
|
{
|
|
worker_stats_clear(worker);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/**
|
|
* Local info for deletion functions
|
|
*/
|
|
struct del_info {
|
|
/** worker */
|
|
struct worker* worker;
|
|
/** name to delete */
|
|
uint8_t* name;
|
|
/** length */
|
|
size_t len;
|
|
/** labels */
|
|
int labs;
|
|
/** time to invalidate to */
|
|
time_t expired;
|
|
/** number of rrsets removed */
|
|
size_t num_rrsets;
|
|
/** number of msgs removed */
|
|
size_t num_msgs;
|
|
/** number of key entries removed */
|
|
size_t num_keys;
|
|
/** length of addr */
|
|
socklen_t addrlen;
|
|
/** socket address for host deletion */
|
|
struct sockaddr_storage addr;
|
|
/** if cachedb information should be flushed too */
|
|
int remcachedb;
|
|
};
|
|
|
|
/** callback to delete hosts in infra cache */
|
|
static void
|
|
infra_del_host(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct infra_key* k = (struct infra_key*)e->key;
|
|
if(sockaddr_cmp(&inf->addr, inf->addrlen, &k->addr, k->addrlen) == 0) {
|
|
struct infra_data* d = (struct infra_data*)e->data;
|
|
d->probedelay = 0;
|
|
d->timeout_A = 0;
|
|
d->timeout_AAAA = 0;
|
|
d->timeout_other = 0;
|
|
rtt_init(&d->rtt);
|
|
if(d->ttl > inf->expired) {
|
|
d->ttl = inf->expired;
|
|
inf->num_keys++;
|
|
}
|
|
}
|
|
}
|
|
|
|
/** flush infra cache */
|
|
static void
|
|
do_flush_infra(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct sockaddr_storage addr;
|
|
socklen_t len;
|
|
struct del_info inf;
|
|
if(strcmp(arg, "all") == 0) {
|
|
slabhash_clear(worker->env.infra_cache->hosts);
|
|
send_ok(ssl);
|
|
return;
|
|
}
|
|
if(!ipstrtoaddr(arg, UNBOUND_DNS_PORT, &addr, &len)) {
|
|
(void)ssl_printf(ssl, "error parsing ip addr: '%s'\n", arg);
|
|
return;
|
|
}
|
|
/* delete all entries from cache */
|
|
/* what we do is to set them all expired */
|
|
inf.worker = worker;
|
|
inf.name = 0;
|
|
inf.len = 0;
|
|
inf.labs = 0;
|
|
inf.expired = *worker->env.now;
|
|
inf.expired -= 3; /* handle 3 seconds skew between threads */
|
|
inf.num_rrsets = 0;
|
|
inf.num_msgs = 0;
|
|
inf.num_keys = 0;
|
|
inf.addrlen = len;
|
|
inf.remcachedb = 0;
|
|
memmove(&inf.addr, &addr, len);
|
|
slabhash_traverse(worker->env.infra_cache->hosts, 1, &infra_del_host,
|
|
&inf);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** flush requestlist */
|
|
static void
|
|
do_flush_requestlist(RES* ssl, struct worker* worker)
|
|
{
|
|
mesh_delete_all(worker->env.mesh);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** callback to delete rrsets in a zone */
|
|
static void
|
|
zone_del_rrset(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct ub_packed_rrset_key* k = (struct ub_packed_rrset_key*)e->key;
|
|
if(dname_subdomain_c(k->rk.dname, inf->name)) {
|
|
struct packed_rrset_data* d =
|
|
(struct packed_rrset_data*)e->data;
|
|
if(d->ttl > inf->expired) {
|
|
d->ttl = inf->expired;
|
|
inf->num_rrsets++;
|
|
}
|
|
}
|
|
}
|
|
|
|
/** callback to delete messages in a zone */
|
|
static void
|
|
zone_del_msg(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct msgreply_entry* k = (struct msgreply_entry*)e->key;
|
|
if(dname_subdomain_c(k->key.qname, inf->name)) {
|
|
struct reply_info* d = (struct reply_info*)e->data;
|
|
if(d->ttl > inf->expired) {
|
|
d->ttl = inf->expired;
|
|
d->prefetch_ttl = inf->expired;
|
|
d->serve_expired_ttl = inf->expired;
|
|
inf->num_msgs++;
|
|
}
|
|
#ifdef USE_CACHEDB
|
|
if(inf->remcachedb && inf->worker->env.cachedb_enabled)
|
|
cachedb_msg_remove_qinfo(&inf->worker->env, &k->key);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
/** callback to delete keys in zone */
|
|
static void
|
|
zone_del_kcache(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct key_entry_key* k = (struct key_entry_key*)e->key;
|
|
if(dname_subdomain_c(k->name, inf->name)) {
|
|
struct key_entry_data* d = (struct key_entry_data*)e->data;
|
|
if(d->ttl > inf->expired) {
|
|
d->ttl = inf->expired;
|
|
inf->num_keys++;
|
|
}
|
|
}
|
|
}
|
|
|
|
/** remove all rrsets and keys from zone from cache */
|
|
static void
|
|
do_flush_zone(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
struct del_info inf;
|
|
int pc = 0; /* '+c' option */
|
|
if(!parse_remcachedb(ssl, &arg, &pc))
|
|
return;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
/* delete all RRs and key entries from zone */
|
|
/* what we do is to set them all expired */
|
|
inf.worker = worker;
|
|
inf.name = nm;
|
|
inf.len = nmlen;
|
|
inf.labs = nmlabs;
|
|
inf.expired = *worker->env.now;
|
|
inf.expired -= 3; /* handle 3 seconds skew between threads */
|
|
inf.num_rrsets = 0;
|
|
inf.num_msgs = 0;
|
|
inf.num_keys = 0;
|
|
inf.remcachedb = pc;
|
|
slabhash_traverse(&worker->env.rrset_cache->table, 1,
|
|
&zone_del_rrset, &inf);
|
|
|
|
slabhash_traverse(worker->env.msg_cache, 1, &zone_del_msg, &inf);
|
|
|
|
/* and validator cache */
|
|
if(worker->env.key_cache) {
|
|
slabhash_traverse(worker->env.key_cache->slab, 1,
|
|
&zone_del_kcache, &inf);
|
|
}
|
|
|
|
free(nm);
|
|
|
|
(void)ssl_printf(ssl, "ok removed %lu rrsets, %lu messages "
|
|
"and %lu key entries\n", (unsigned long)inf.num_rrsets,
|
|
(unsigned long)inf.num_msgs, (unsigned long)inf.num_keys);
|
|
}
|
|
|
|
/** callback to delete bogus rrsets */
|
|
static void
|
|
bogus_del_rrset(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct packed_rrset_data* d = (struct packed_rrset_data*)e->data;
|
|
if(d->security == sec_status_bogus) {
|
|
d->ttl = inf->expired;
|
|
inf->num_rrsets++;
|
|
}
|
|
}
|
|
|
|
/** callback to delete bogus messages */
|
|
static void
|
|
bogus_del_msg(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct reply_info* d = (struct reply_info*)e->data;
|
|
if(d->security == sec_status_bogus) {
|
|
d->ttl = inf->expired;
|
|
d->prefetch_ttl = inf->expired;
|
|
d->serve_expired_ttl = inf->expired;
|
|
inf->num_msgs++;
|
|
#ifdef USE_CACHEDB
|
|
if(inf->remcachedb && inf->worker->env.cachedb_enabled)
|
|
cachedb_msg_remove_qinfo(&inf->worker->env,
|
|
&((struct msgreply_entry*)e->key)->key);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
/** callback to delete bogus keys */
|
|
static void
|
|
bogus_del_kcache(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct key_entry_data* d = (struct key_entry_data*)e->data;
|
|
if(d->isbad) {
|
|
d->ttl = inf->expired;
|
|
inf->num_keys++;
|
|
}
|
|
}
|
|
|
|
/** remove all bogus rrsets, msgs and keys from cache */
|
|
static void
|
|
do_flush_bogus(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct del_info inf;
|
|
int pc = 0; /* '+c' option */
|
|
if(!parse_remcachedb(ssl, &arg, &pc))
|
|
return;
|
|
/* what we do is to set them all expired */
|
|
inf.worker = worker;
|
|
inf.expired = *worker->env.now;
|
|
inf.expired -= 3; /* handle 3 seconds skew between threads */
|
|
inf.num_rrsets = 0;
|
|
inf.num_msgs = 0;
|
|
inf.num_keys = 0;
|
|
inf.remcachedb = pc;
|
|
slabhash_traverse(&worker->env.rrset_cache->table, 1,
|
|
&bogus_del_rrset, &inf);
|
|
|
|
slabhash_traverse(worker->env.msg_cache, 1, &bogus_del_msg, &inf);
|
|
|
|
/* and validator cache */
|
|
if(worker->env.key_cache) {
|
|
slabhash_traverse(worker->env.key_cache->slab, 1,
|
|
&bogus_del_kcache, &inf);
|
|
}
|
|
|
|
(void)ssl_printf(ssl, "ok removed %lu rrsets, %lu messages "
|
|
"and %lu key entries\n", (unsigned long)inf.num_rrsets,
|
|
(unsigned long)inf.num_msgs, (unsigned long)inf.num_keys);
|
|
}
|
|
|
|
/** callback to delete negative and servfail rrsets */
|
|
static void
|
|
negative_del_rrset(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct ub_packed_rrset_key* k = (struct ub_packed_rrset_key*)e->key;
|
|
struct packed_rrset_data* d = (struct packed_rrset_data*)e->data;
|
|
/* delete the parentside negative cache rrsets,
|
|
* these are nameserver rrsets that failed lookup, rdata empty */
|
|
if((k->rk.flags & PACKED_RRSET_PARENT_SIDE) && d->count == 1 &&
|
|
d->rrsig_count == 0 && d->rr_len[0] == 0) {
|
|
d->ttl = inf->expired;
|
|
inf->num_rrsets++;
|
|
}
|
|
}
|
|
|
|
/** callback to delete negative and servfail messages */
|
|
static void
|
|
negative_del_msg(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct reply_info* d = (struct reply_info*)e->data;
|
|
/* rcode not NOERROR: NXDOMAIN, SERVFAIL, ..: an nxdomain or error
|
|
* or NOERROR rcode with ANCOUNT==0: a NODATA answer */
|
|
if(FLAGS_GET_RCODE(d->flags) != 0 || d->an_numrrsets == 0) {
|
|
d->ttl = inf->expired;
|
|
d->prefetch_ttl = inf->expired;
|
|
d->serve_expired_ttl = inf->expired;
|
|
inf->num_msgs++;
|
|
#ifdef USE_CACHEDB
|
|
if(inf->remcachedb && inf->worker->env.cachedb_enabled)
|
|
cachedb_msg_remove_qinfo(&inf->worker->env,
|
|
&((struct msgreply_entry*)e->key)->key);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
/** callback to delete negative key entries */
|
|
static void
|
|
negative_del_kcache(struct lruhash_entry* e, void* arg)
|
|
{
|
|
/* entry is locked */
|
|
struct del_info* inf = (struct del_info*)arg;
|
|
struct key_entry_data* d = (struct key_entry_data*)e->data;
|
|
/* could be bad because of lookup failure on the DS, DNSKEY, which
|
|
* was nxdomain or servfail, and thus a result of negative lookups */
|
|
if(d->isbad) {
|
|
d->ttl = inf->expired;
|
|
inf->num_keys++;
|
|
}
|
|
}
|
|
|
|
/** remove all negative(NODATA,NXDOMAIN), and servfail messages from cache */
|
|
static void
|
|
do_flush_negative(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct del_info inf;
|
|
int pc = 0; /* '+c' option */
|
|
if(!parse_remcachedb(ssl, &arg, &pc))
|
|
return;
|
|
/* what we do is to set them all expired */
|
|
inf.worker = worker;
|
|
inf.expired = *worker->env.now;
|
|
inf.expired -= 3; /* handle 3 seconds skew between threads */
|
|
inf.num_rrsets = 0;
|
|
inf.num_msgs = 0;
|
|
inf.num_keys = 0;
|
|
inf.remcachedb = pc;
|
|
slabhash_traverse(&worker->env.rrset_cache->table, 1,
|
|
&negative_del_rrset, &inf);
|
|
|
|
slabhash_traverse(worker->env.msg_cache, 1, &negative_del_msg, &inf);
|
|
|
|
/* and validator cache */
|
|
if(worker->env.key_cache) {
|
|
slabhash_traverse(worker->env.key_cache->slab, 1,
|
|
&negative_del_kcache, &inf);
|
|
}
|
|
|
|
(void)ssl_printf(ssl, "ok removed %lu rrsets, %lu messages "
|
|
"and %lu key entries\n", (unsigned long)inf.num_rrsets,
|
|
(unsigned long)inf.num_msgs, (unsigned long)inf.num_keys);
|
|
}
|
|
|
|
/** remove name rrset from cache */
|
|
static void
|
|
do_flush_name(RES* ssl, struct worker* w, char* arg)
|
|
{
|
|
uint8_t* nm;
|
|
int nmlabs;
|
|
size_t nmlen;
|
|
int pc = 0; /* '+c' option */
|
|
if(!parse_remcachedb(ssl, &arg, &pc))
|
|
return;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_A, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_AAAA, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_NS, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_SOA, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_CNAME, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_DNAME, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_MX, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_PTR, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_SRV, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_NAPTR, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_SVCB, LDNS_RR_CLASS_IN, pc);
|
|
do_cache_remove(w, nm, nmlen, LDNS_RR_TYPE_HTTPS, LDNS_RR_CLASS_IN, pc);
|
|
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** printout a delegation point info */
|
|
static int
|
|
ssl_print_name_dp(RES* ssl, const char* str, uint8_t* nm, uint16_t dclass,
|
|
struct delegpt* dp)
|
|
{
|
|
char buf[257];
|
|
struct delegpt_ns* ns;
|
|
struct delegpt_addr* a;
|
|
int f = 0;
|
|
if(str) { /* print header for forward, stub */
|
|
char* c = sldns_wire2str_class(dclass);
|
|
dname_str(nm, buf);
|
|
if(!ssl_printf(ssl, "%s %s %s ", buf, (c?c:"CLASS??"), str)) {
|
|
free(c);
|
|
return 0;
|
|
}
|
|
free(c);
|
|
}
|
|
for(ns = dp->nslist; ns; ns = ns->next) {
|
|
dname_str(ns->name, buf);
|
|
if(!ssl_printf(ssl, "%s%s", (f?" ":""), buf))
|
|
return 0;
|
|
f = 1;
|
|
}
|
|
for(a = dp->target_list; a; a = a->next_target) {
|
|
addr_to_str(&a->addr, a->addrlen, buf, sizeof(buf));
|
|
if(!ssl_printf(ssl, "%s%s", (f?" ":""), buf))
|
|
return 0;
|
|
f = 1;
|
|
}
|
|
return ssl_printf(ssl, "\n");
|
|
}
|
|
|
|
|
|
/** print root forwards */
|
|
static int
|
|
print_root_fwds(RES* ssl, struct iter_forwards* fwds, uint8_t* root)
|
|
{
|
|
struct delegpt* dp;
|
|
int nolock = 0;
|
|
dp = forwards_lookup(fwds, root, LDNS_RR_CLASS_IN, nolock);
|
|
if(!dp) {
|
|
return ssl_printf(ssl, "off (using root hints)\n");
|
|
}
|
|
/* if dp is returned it must be the root */
|
|
log_assert(query_dname_compare(dp->name, root)==0);
|
|
if(!ssl_print_name_dp(ssl, NULL, root, LDNS_RR_CLASS_IN, dp)) {
|
|
lock_rw_unlock(&fwds->lock);
|
|
return 0;
|
|
}
|
|
lock_rw_unlock(&fwds->lock);
|
|
return 1;
|
|
}
|
|
|
|
/** parse args into delegpt */
|
|
static struct delegpt*
|
|
parse_delegpt(RES* ssl, char* args, uint8_t* nm)
|
|
{
|
|
/* parse args and add in */
|
|
char* p = args;
|
|
char* todo;
|
|
struct delegpt* dp = delegpt_create_mlc(nm);
|
|
struct sockaddr_storage addr;
|
|
socklen_t addrlen;
|
|
char* auth_name;
|
|
if(!dp) {
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
return NULL;
|
|
}
|
|
while(p) {
|
|
todo = p;
|
|
p = strchr(p, ' '); /* find next spot, if any */
|
|
if(p) {
|
|
*p++ = 0; /* end this spot */
|
|
p = skipwhite(p); /* position at next spot */
|
|
}
|
|
/* parse address */
|
|
if(!authextstrtoaddr(todo, &addr, &addrlen, &auth_name)) {
|
|
uint8_t* dname= NULL;
|
|
int port;
|
|
dname = authextstrtodname(todo, &port, &auth_name);
|
|
if(!dname) {
|
|
(void)ssl_printf(ssl, "error cannot parse"
|
|
" '%s'\n", todo);
|
|
delegpt_free_mlc(dp);
|
|
return NULL;
|
|
}
|
|
#if ! defined(HAVE_SSL_SET1_HOST) && ! defined(HAVE_X509_VERIFY_PARAM_SET1_HOST)
|
|
if(auth_name)
|
|
log_err("no name verification functionality in "
|
|
"ssl library, ignored name for %s", todo);
|
|
#endif
|
|
if(!delegpt_add_ns_mlc(dp, dname, 0, auth_name, port)) {
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
free(dname);
|
|
delegpt_free_mlc(dp);
|
|
return NULL;
|
|
}
|
|
} else {
|
|
#if ! defined(HAVE_SSL_SET1_HOST) && ! defined(HAVE_X509_VERIFY_PARAM_SET1_HOST)
|
|
if(auth_name)
|
|
log_err("no name verification functionality in "
|
|
"ssl library, ignored name for %s", todo);
|
|
#endif
|
|
/* add address */
|
|
if(!delegpt_add_addr_mlc(dp, &addr, addrlen, 0, 0,
|
|
auth_name, -1)) {
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
delegpt_free_mlc(dp);
|
|
return NULL;
|
|
}
|
|
}
|
|
}
|
|
dp->has_parent_side_NS = 1;
|
|
return dp;
|
|
}
|
|
|
|
/** do the forward command */
|
|
static void
|
|
do_forward(RES* ssl, struct worker* worker, char* args)
|
|
{
|
|
struct iter_forwards* fwd = worker->env.fwds;
|
|
uint8_t* root = (uint8_t*)"\000";
|
|
int nolock = 0;
|
|
if(!fwd) {
|
|
(void)ssl_printf(ssl, "error: structure not allocated\n");
|
|
return;
|
|
}
|
|
if(args == NULL || args[0] == 0) {
|
|
(void)print_root_fwds(ssl, fwd, root);
|
|
return;
|
|
}
|
|
/* set root forwards for this thread. since we are in remote control
|
|
* the actual mesh is not running, so we can freely edit it. */
|
|
/* delete all the existing queries first */
|
|
mesh_delete_all(worker->env.mesh);
|
|
if(strcmp(args, "off") == 0) {
|
|
forwards_delete_zone(fwd, LDNS_RR_CLASS_IN, root, nolock);
|
|
} else {
|
|
struct delegpt* dp;
|
|
if(!(dp = parse_delegpt(ssl, args, root)))
|
|
return;
|
|
if(!forwards_add_zone(fwd, LDNS_RR_CLASS_IN, dp, nolock)) {
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
return;
|
|
}
|
|
}
|
|
send_ok(ssl);
|
|
}
|
|
|
|
static int
|
|
parse_fs_args(RES* ssl, char* args, uint8_t** nm, struct delegpt** dp,
|
|
int* insecure, int* prime, int* tls)
|
|
{
|
|
char* zonename;
|
|
char* rest;
|
|
size_t nmlen;
|
|
int nmlabs;
|
|
/* parse all -x args */
|
|
while(args[0] == '+') {
|
|
if(!find_arg2(ssl, args, &rest))
|
|
return 0;
|
|
while(*(++args) != 0) {
|
|
if(*args == 'i' && insecure)
|
|
*insecure = 1;
|
|
else if(*args == 'p' && prime)
|
|
*prime = 1;
|
|
else if(*args == 't' && tls)
|
|
*tls = 1;
|
|
else {
|
|
(void)ssl_printf(ssl, "error: unknown option %s\n", args);
|
|
return 0;
|
|
}
|
|
}
|
|
args = rest;
|
|
}
|
|
/* parse name */
|
|
if(dp) {
|
|
if(!find_arg2(ssl, args, &rest))
|
|
return 0;
|
|
zonename = args;
|
|
args = rest;
|
|
} else zonename = args;
|
|
if(!parse_arg_name(ssl, zonename, nm, &nmlen, &nmlabs))
|
|
return 0;
|
|
|
|
/* parse dp */
|
|
if(dp) {
|
|
if(!(*dp = parse_delegpt(ssl, args, *nm))) {
|
|
free(*nm);
|
|
return 0;
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/** do the forward_add command */
|
|
static void
|
|
do_forward_add(RES* ssl, struct worker* worker, char* args)
|
|
{
|
|
struct iter_forwards* fwd = worker->env.fwds;
|
|
int insecure = 0, tls = 0;
|
|
uint8_t* nm = NULL;
|
|
struct delegpt* dp = NULL;
|
|
int nolock = 1;
|
|
if(!parse_fs_args(ssl, args, &nm, &dp, &insecure, NULL, &tls))
|
|
return;
|
|
if(tls)
|
|
dp->ssl_upstream = 1;
|
|
/* prelock forwarders for atomic operation with anchors */
|
|
lock_rw_wrlock(&fwd->lock);
|
|
if(insecure && worker->env.anchors) {
|
|
if(!anchors_add_insecure(worker->env.anchors, LDNS_RR_CLASS_IN,
|
|
nm)) {
|
|
lock_rw_unlock(&fwd->lock);
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
delegpt_free_mlc(dp);
|
|
free(nm);
|
|
return;
|
|
}
|
|
}
|
|
if(!forwards_add_zone(fwd, LDNS_RR_CLASS_IN, dp, nolock)) {
|
|
lock_rw_unlock(&fwd->lock);
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
free(nm);
|
|
return;
|
|
}
|
|
lock_rw_unlock(&fwd->lock);
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the forward_remove command */
|
|
static void
|
|
do_forward_remove(RES* ssl, struct worker* worker, char* args)
|
|
{
|
|
struct iter_forwards* fwd = worker->env.fwds;
|
|
int insecure = 0;
|
|
uint8_t* nm = NULL;
|
|
int nolock = 1;
|
|
if(!parse_fs_args(ssl, args, &nm, NULL, &insecure, NULL, NULL))
|
|
return;
|
|
/* prelock forwarders for atomic operation with anchors */
|
|
lock_rw_wrlock(&fwd->lock);
|
|
if(insecure && worker->env.anchors)
|
|
anchors_delete_insecure(worker->env.anchors, LDNS_RR_CLASS_IN,
|
|
nm);
|
|
forwards_delete_zone(fwd, LDNS_RR_CLASS_IN, nm, nolock);
|
|
lock_rw_unlock(&fwd->lock);
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the stub_add command */
|
|
static void
|
|
do_stub_add(RES* ssl, struct worker* worker, char* args)
|
|
{
|
|
struct iter_forwards* fwd = worker->env.fwds;
|
|
int insecure = 0, prime = 0, tls = 0;
|
|
uint8_t* nm = NULL;
|
|
struct delegpt* dp = NULL;
|
|
int nolock = 1;
|
|
if(!parse_fs_args(ssl, args, &nm, &dp, &insecure, &prime, &tls))
|
|
return;
|
|
if(tls)
|
|
dp->ssl_upstream = 1;
|
|
/* prelock forwarders and hints for atomic operation with anchors */
|
|
lock_rw_wrlock(&fwd->lock);
|
|
lock_rw_wrlock(&worker->env.hints->lock);
|
|
if(insecure && worker->env.anchors) {
|
|
if(!anchors_add_insecure(worker->env.anchors, LDNS_RR_CLASS_IN,
|
|
nm)) {
|
|
lock_rw_unlock(&fwd->lock);
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
delegpt_free_mlc(dp);
|
|
free(nm);
|
|
return;
|
|
}
|
|
}
|
|
if(!forwards_add_stub_hole(fwd, LDNS_RR_CLASS_IN, nm, nolock)) {
|
|
if(insecure && worker->env.anchors)
|
|
anchors_delete_insecure(worker->env.anchors,
|
|
LDNS_RR_CLASS_IN, nm);
|
|
lock_rw_unlock(&fwd->lock);
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
delegpt_free_mlc(dp);
|
|
free(nm);
|
|
return;
|
|
}
|
|
if(!hints_add_stub(worker->env.hints, LDNS_RR_CLASS_IN, dp, !prime,
|
|
nolock)) {
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
forwards_delete_stub_hole(fwd, LDNS_RR_CLASS_IN, nm, nolock);
|
|
if(insecure && worker->env.anchors)
|
|
anchors_delete_insecure(worker->env.anchors,
|
|
LDNS_RR_CLASS_IN, nm);
|
|
lock_rw_unlock(&fwd->lock);
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
free(nm);
|
|
return;
|
|
}
|
|
lock_rw_unlock(&fwd->lock);
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the stub_remove command */
|
|
static void
|
|
do_stub_remove(RES* ssl, struct worker* worker, char* args)
|
|
{
|
|
struct iter_forwards* fwd = worker->env.fwds;
|
|
int insecure = 0;
|
|
uint8_t* nm = NULL;
|
|
int nolock = 1;
|
|
if(!parse_fs_args(ssl, args, &nm, NULL, &insecure, NULL, NULL))
|
|
return;
|
|
/* prelock forwarders and hints for atomic operation with anchors */
|
|
lock_rw_wrlock(&fwd->lock);
|
|
lock_rw_wrlock(&worker->env.hints->lock);
|
|
if(insecure && worker->env.anchors)
|
|
anchors_delete_insecure(worker->env.anchors, LDNS_RR_CLASS_IN,
|
|
nm);
|
|
forwards_delete_stub_hole(fwd, LDNS_RR_CLASS_IN, nm, nolock);
|
|
hints_delete_stub(worker->env.hints, LDNS_RR_CLASS_IN, nm, nolock);
|
|
lock_rw_unlock(&fwd->lock);
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the insecure_add command */
|
|
static void
|
|
do_insecure_add(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
size_t nmlen;
|
|
int nmlabs;
|
|
uint8_t* nm = NULL;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
if(worker->env.anchors) {
|
|
if(!anchors_add_insecure(worker->env.anchors,
|
|
LDNS_RR_CLASS_IN, nm)) {
|
|
(void)ssl_printf(ssl, "error out of memory\n");
|
|
free(nm);
|
|
return;
|
|
}
|
|
}
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the insecure_remove command */
|
|
static void
|
|
do_insecure_remove(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
size_t nmlen;
|
|
int nmlabs;
|
|
uint8_t* nm = NULL;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
if(worker->env.anchors)
|
|
anchors_delete_insecure(worker->env.anchors,
|
|
LDNS_RR_CLASS_IN, nm);
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
static void
|
|
do_insecure_list(RES* ssl, struct worker* worker)
|
|
{
|
|
char buf[257];
|
|
struct trust_anchor* a;
|
|
if(worker->env.anchors) {
|
|
RBTREE_FOR(a, struct trust_anchor*, worker->env.anchors->tree) {
|
|
if(a->numDS == 0 && a->numDNSKEY == 0) {
|
|
dname_str(a->name, buf);
|
|
ssl_printf(ssl, "%s\n", buf);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/** do the status command */
|
|
static void
|
|
do_status(RES* ssl, struct worker* worker)
|
|
{
|
|
int i;
|
|
time_t uptime;
|
|
if(!ssl_printf(ssl, "version: %s\n", PACKAGE_VERSION))
|
|
return;
|
|
if(!ssl_printf(ssl, "verbosity: %d\n", verbosity))
|
|
return;
|
|
if(!ssl_printf(ssl, "threads: %d\n", worker->daemon->num))
|
|
return;
|
|
if(!ssl_printf(ssl, "modules: %d [", worker->daemon->mods.num))
|
|
return;
|
|
for(i=0; i<worker->daemon->mods.num; i++) {
|
|
if(!ssl_printf(ssl, " %s", worker->daemon->mods.mod[i]->name))
|
|
return;
|
|
}
|
|
if(!ssl_printf(ssl, " ]\n"))
|
|
return;
|
|
uptime = (time_t)time(NULL) - (time_t)worker->daemon->time_boot.tv_sec;
|
|
if(!ssl_printf(ssl, "uptime: " ARG_LL "d seconds\n", (long long)uptime))
|
|
return;
|
|
if(!ssl_printf(ssl, "options:%s%s%s%s\n" ,
|
|
(worker->daemon->reuseport?" reuseport":""),
|
|
(worker->daemon->rc->accept_list?" control":""),
|
|
(worker->daemon->rc->accept_list && worker->daemon->rc->use_cert?"(ssl)":""),
|
|
(worker->daemon->rc->accept_list && worker->daemon->cfg->control_ifs.first && worker->daemon->cfg->control_ifs.first->str && worker->daemon->cfg->control_ifs.first->str[0] == '/'?"(namedpipe)":"")
|
|
))
|
|
return;
|
|
if(!ssl_printf(ssl, "unbound (pid %d) is running...\n",
|
|
(int)getpid()))
|
|
return;
|
|
}
|
|
|
|
/** get age for the mesh state */
|
|
static void
|
|
get_mesh_age(struct mesh_state* m, char* buf, size_t len,
|
|
struct module_env* env)
|
|
{
|
|
if(m->reply_list) {
|
|
struct timeval d;
|
|
struct mesh_reply* r = m->reply_list;
|
|
/* last reply is the oldest */
|
|
while(r && r->next)
|
|
r = r->next;
|
|
timeval_subtract(&d, env->now_tv, &r->start_time);
|
|
snprintf(buf, len, ARG_LL "d.%6.6d",
|
|
(long long)d.tv_sec, (int)d.tv_usec);
|
|
} else {
|
|
snprintf(buf, len, "-");
|
|
}
|
|
}
|
|
|
|
/** get status of a mesh state */
|
|
static void
|
|
get_mesh_status(struct mesh_area* mesh, struct mesh_state* m,
|
|
char* buf, size_t len)
|
|
{
|
|
enum module_ext_state s = m->s.ext_state[m->s.curmod];
|
|
const char *modname = mesh->mods.mod[m->s.curmod]->name;
|
|
size_t l;
|
|
if(strcmp(modname, "iterator") == 0 && s == module_wait_reply &&
|
|
m->s.minfo[m->s.curmod]) {
|
|
/* break into iterator to find out who its waiting for */
|
|
struct iter_qstate* qstate = (struct iter_qstate*)
|
|
m->s.minfo[m->s.curmod];
|
|
struct outbound_list* ol = &qstate->outlist;
|
|
struct outbound_entry* e;
|
|
snprintf(buf, len, "%s wait for", modname);
|
|
l = strlen(buf);
|
|
buf += l; len -= l;
|
|
if(ol->first == NULL)
|
|
snprintf(buf, len, " (empty_list)");
|
|
for(e = ol->first; e; e = e->next) {
|
|
snprintf(buf, len, " ");
|
|
l = strlen(buf);
|
|
buf += l; len -= l;
|
|
addr_to_str(&e->qsent->addr, e->qsent->addrlen,
|
|
buf, len);
|
|
l = strlen(buf);
|
|
buf += l; len -= l;
|
|
}
|
|
} else if(s == module_wait_subquery) {
|
|
/* look in subs from mesh state to see what */
|
|
char nm[257];
|
|
struct mesh_state_ref* sub;
|
|
snprintf(buf, len, "%s wants", modname);
|
|
l = strlen(buf);
|
|
buf += l; len -= l;
|
|
if(m->sub_set.count == 0)
|
|
snprintf(buf, len, " (empty_list)");
|
|
RBTREE_FOR(sub, struct mesh_state_ref*, &m->sub_set) {
|
|
char* t = sldns_wire2str_type(sub->s->s.qinfo.qtype);
|
|
char* c = sldns_wire2str_class(sub->s->s.qinfo.qclass);
|
|
dname_str(sub->s->s.qinfo.qname, nm);
|
|
snprintf(buf, len, " %s %s %s", (t?t:"TYPE??"),
|
|
(c?c:"CLASS??"), nm);
|
|
l = strlen(buf);
|
|
buf += l; len -= l;
|
|
free(t);
|
|
free(c);
|
|
}
|
|
} else {
|
|
snprintf(buf, len, "%s is %s", modname, strextstate(s));
|
|
}
|
|
}
|
|
|
|
/** do the dump_requestlist command */
|
|
static void
|
|
do_dump_requestlist(RES* ssl, struct worker* worker)
|
|
{
|
|
struct mesh_area* mesh;
|
|
struct mesh_state* m;
|
|
int num = 0;
|
|
char buf[257];
|
|
char timebuf[32];
|
|
char statbuf[10240];
|
|
if(!ssl_printf(ssl, "thread #%d\n", worker->thread_num))
|
|
return;
|
|
if(!ssl_printf(ssl, "# type cl name seconds module status\n"))
|
|
return;
|
|
/* show worker mesh contents */
|
|
mesh = worker->env.mesh;
|
|
if(!mesh) return;
|
|
RBTREE_FOR(m, struct mesh_state*, &mesh->all) {
|
|
char* t = sldns_wire2str_type(m->s.qinfo.qtype);
|
|
char* c = sldns_wire2str_class(m->s.qinfo.qclass);
|
|
dname_str(m->s.qinfo.qname, buf);
|
|
get_mesh_age(m, timebuf, sizeof(timebuf), &worker->env);
|
|
get_mesh_status(mesh, m, statbuf, sizeof(statbuf));
|
|
if(!ssl_printf(ssl, "%3d %4s %2s %s %s %s\n",
|
|
num, (t?t:"TYPE??"), (c?c:"CLASS??"), buf, timebuf,
|
|
statbuf)) {
|
|
free(t);
|
|
free(c);
|
|
return;
|
|
}
|
|
num++;
|
|
free(t);
|
|
free(c);
|
|
}
|
|
}
|
|
|
|
/** structure for argument data for dump infra host */
|
|
struct infra_arg {
|
|
/** the infra cache */
|
|
struct infra_cache* infra;
|
|
/** the SSL connection */
|
|
RES* ssl;
|
|
/** the time now */
|
|
time_t now;
|
|
/** ssl failure? stop writing and skip the rest. If the tcp
|
|
* connection is broken, and writes fail, we then stop writing. */
|
|
int ssl_failed;
|
|
};
|
|
|
|
/** callback for every host element in the infra cache */
|
|
static void
|
|
dump_infra_host(struct lruhash_entry* e, void* arg)
|
|
{
|
|
struct infra_arg* a = (struct infra_arg*)arg;
|
|
struct infra_key* k = (struct infra_key*)e->key;
|
|
struct infra_data* d = (struct infra_data*)e->data;
|
|
char ip_str[1024];
|
|
char name[257];
|
|
int port;
|
|
if(a->ssl_failed)
|
|
return;
|
|
addr_to_str(&k->addr, k->addrlen, ip_str, sizeof(ip_str));
|
|
dname_str(k->zonename, name);
|
|
port = (int)ntohs(((struct sockaddr_in*)&k->addr)->sin_port);
|
|
if(port != UNBOUND_DNS_PORT) {
|
|
snprintf(ip_str+strlen(ip_str), sizeof(ip_str)-strlen(ip_str),
|
|
"@%d", port);
|
|
}
|
|
/* skip expired stuff (only backed off) */
|
|
if(d->ttl < a->now) {
|
|
if(d->rtt.rto >= USEFUL_SERVER_TOP_TIMEOUT) {
|
|
if(!ssl_printf(a->ssl, "%s %s expired rto %d\n", ip_str,
|
|
name, d->rtt.rto)) {
|
|
a->ssl_failed = 1;
|
|
return;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
if(!ssl_printf(a->ssl, "%s %s ttl %lu ping %d var %d rtt %d rto %d "
|
|
"tA %d tAAAA %d tother %d "
|
|
"ednsknown %d edns %d delay %d lame dnssec %d rec %d A %d "
|
|
"other %d\n", ip_str, name, (unsigned long)(d->ttl - a->now),
|
|
d->rtt.srtt, d->rtt.rttvar, rtt_notimeout(&d->rtt), d->rtt.rto,
|
|
d->timeout_A, d->timeout_AAAA, d->timeout_other,
|
|
(int)d->edns_lame_known, (int)d->edns_version,
|
|
(int)(a->now<d->probedelay?(d->probedelay - a->now):0),
|
|
(int)d->isdnsseclame, (int)d->rec_lame, (int)d->lame_type_A,
|
|
(int)d->lame_other)) {
|
|
a->ssl_failed = 1;
|
|
return;
|
|
}
|
|
}
|
|
|
|
/** do the dump_infra command */
|
|
static void
|
|
do_dump_infra(RES* ssl, struct worker* worker)
|
|
{
|
|
struct infra_arg arg;
|
|
arg.infra = worker->env.infra_cache;
|
|
arg.ssl = ssl;
|
|
arg.now = *worker->env.now;
|
|
arg.ssl_failed = 0;
|
|
slabhash_traverse(arg.infra->hosts, 0, &dump_infra_host, (void*)&arg);
|
|
}
|
|
|
|
/** do the log_reopen command */
|
|
static void
|
|
do_log_reopen(RES* ssl, struct worker* worker)
|
|
{
|
|
struct config_file* cfg = worker->env.cfg;
|
|
send_ok(ssl);
|
|
log_init(cfg->logfile, cfg->use_syslog, cfg->chrootdir);
|
|
}
|
|
|
|
/** do the auth_zone_reload command */
|
|
static void
|
|
do_auth_zone_reload(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
size_t nmlen;
|
|
int nmlabs;
|
|
uint8_t* nm = NULL;
|
|
struct auth_zones* az = worker->env.auth_zones;
|
|
struct auth_zone* z = NULL;
|
|
struct auth_xfer* xfr = NULL;
|
|
char* reason = NULL;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
if(az) {
|
|
lock_rw_rdlock(&az->lock);
|
|
z = auth_zone_find(az, nm, nmlen, LDNS_RR_CLASS_IN);
|
|
if(z) {
|
|
lock_rw_wrlock(&z->lock);
|
|
}
|
|
xfr = auth_xfer_find(az, nm, nmlen, LDNS_RR_CLASS_IN);
|
|
if(xfr) {
|
|
lock_basic_lock(&xfr->lock);
|
|
}
|
|
lock_rw_unlock(&az->lock);
|
|
}
|
|
free(nm);
|
|
if(!z) {
|
|
if(xfr) {
|
|
lock_basic_unlock(&xfr->lock);
|
|
}
|
|
(void)ssl_printf(ssl, "error no auth-zone %s\n", arg);
|
|
return;
|
|
}
|
|
if(!auth_zone_read_zonefile(z, worker->env.cfg)) {
|
|
lock_rw_unlock(&z->lock);
|
|
if(xfr) {
|
|
lock_basic_unlock(&xfr->lock);
|
|
}
|
|
(void)ssl_printf(ssl, "error failed to read %s\n", arg);
|
|
return;
|
|
}
|
|
|
|
z->zone_expired = 0;
|
|
if(xfr) {
|
|
xfr->zone_expired = 0;
|
|
if(!xfr_find_soa(z, xfr)) {
|
|
if(z->data.count == 0) {
|
|
lock_rw_unlock(&z->lock);
|
|
lock_basic_unlock(&xfr->lock);
|
|
(void)ssl_printf(ssl, "zone %s has no contents\n", arg);
|
|
return;
|
|
}
|
|
lock_rw_unlock(&z->lock);
|
|
lock_basic_unlock(&xfr->lock);
|
|
(void)ssl_printf(ssl, "error: no SOA in zone after read %s\n", arg);
|
|
return;
|
|
}
|
|
if(xfr->have_zone)
|
|
xfr->lease_time = *worker->env.now;
|
|
lock_basic_unlock(&xfr->lock);
|
|
}
|
|
|
|
auth_zone_verify_zonemd(z, &worker->env, &worker->env.mesh->mods,
|
|
&reason, 0, 0);
|
|
if(reason && z->zone_expired) {
|
|
lock_rw_unlock(&z->lock);
|
|
(void)ssl_printf(ssl, "error zonemd for %s failed: %s\n",
|
|
arg, reason);
|
|
free(reason);
|
|
return;
|
|
} else if(reason && strcmp(reason, "ZONEMD verification successful")
|
|
==0) {
|
|
(void)ssl_printf(ssl, "%s: %s\n", arg, reason);
|
|
}
|
|
lock_rw_unlock(&z->lock);
|
|
free(reason);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the auth_zone_transfer command */
|
|
static void
|
|
do_auth_zone_transfer(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
size_t nmlen;
|
|
int nmlabs;
|
|
uint8_t* nm = NULL;
|
|
struct auth_zones* az = worker->env.auth_zones;
|
|
if(!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
if(!az || !auth_zones_startprobesequence(az, &worker->env, nm, nmlen,
|
|
LDNS_RR_CLASS_IN)) {
|
|
(void)ssl_printf(ssl, "error zone xfr task not found %s\n", arg);
|
|
free(nm);
|
|
return;
|
|
}
|
|
free(nm);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the set_option command */
|
|
static void
|
|
do_set_option(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
char* arg2;
|
|
if(!find_arg2(ssl, arg, &arg2))
|
|
return;
|
|
if(!config_set_option(worker->env.cfg, arg, arg2)) {
|
|
(void)ssl_printf(ssl, "error setting option\n");
|
|
return;
|
|
}
|
|
/* effectuate some arguments */
|
|
if(strcmp(arg, "val-override-date:") == 0) {
|
|
int m = modstack_find(&worker->env.mesh->mods, "validator");
|
|
struct val_env* val_env = NULL;
|
|
if(m != -1) val_env = (struct val_env*)worker->env.modinfo[m];
|
|
if(val_env)
|
|
val_env->date_override = worker->env.cfg->val_date_override;
|
|
}
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/* routine to printout option values over SSL */
|
|
void remote_get_opt_ssl(char* line, void* arg)
|
|
{
|
|
RES* ssl = (RES*)arg;
|
|
(void)ssl_printf(ssl, "%s\n", line);
|
|
}
|
|
|
|
/** do the get_option command */
|
|
static void
|
|
do_get_option(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
int r;
|
|
r = config_get_option(worker->env.cfg, arg, remote_get_opt_ssl, ssl);
|
|
if(!r) {
|
|
(void)ssl_printf(ssl, "error unknown option\n");
|
|
return;
|
|
}
|
|
}
|
|
|
|
/** do the list_forwards command */
|
|
static void
|
|
do_list_forwards(RES* ssl, struct worker* worker)
|
|
{
|
|
/* since its a per-worker structure no locks needed */
|
|
struct iter_forwards* fwds = worker->env.fwds;
|
|
struct iter_forward_zone* z;
|
|
struct trust_anchor* a;
|
|
int insecure;
|
|
lock_rw_rdlock(&fwds->lock);
|
|
RBTREE_FOR(z, struct iter_forward_zone*, fwds->tree) {
|
|
if(!z->dp) continue; /* skip empty marker for stub */
|
|
|
|
/* see if it is insecure */
|
|
insecure = 0;
|
|
if(worker->env.anchors &&
|
|
(a=anchor_find(worker->env.anchors, z->name,
|
|
z->namelabs, z->namelen, z->dclass))) {
|
|
if(!a->keylist && !a->numDS && !a->numDNSKEY)
|
|
insecure = 1;
|
|
lock_basic_unlock(&a->lock);
|
|
}
|
|
|
|
if(!ssl_print_name_dp(ssl, (insecure?"forward +i":"forward"),
|
|
z->name, z->dclass, z->dp)) {
|
|
lock_rw_unlock(&fwds->lock);
|
|
return;
|
|
}
|
|
}
|
|
lock_rw_unlock(&fwds->lock);
|
|
}
|
|
|
|
/** do the list_stubs command */
|
|
static void
|
|
do_list_stubs(RES* ssl, struct worker* worker)
|
|
{
|
|
struct iter_hints_stub* z;
|
|
struct trust_anchor* a;
|
|
int insecure;
|
|
char str[32];
|
|
lock_rw_rdlock(&worker->env.hints->lock);
|
|
RBTREE_FOR(z, struct iter_hints_stub*, &worker->env.hints->tree) {
|
|
|
|
/* see if it is insecure */
|
|
insecure = 0;
|
|
if(worker->env.anchors &&
|
|
(a=anchor_find(worker->env.anchors, z->node.name,
|
|
z->node.labs, z->node.len, z->node.dclass))) {
|
|
if(!a->keylist && !a->numDS && !a->numDNSKEY)
|
|
insecure = 1;
|
|
lock_basic_unlock(&a->lock);
|
|
}
|
|
|
|
snprintf(str, sizeof(str), "stub %sprime%s",
|
|
(z->noprime?"no":""), (insecure?" +i":""));
|
|
if(!ssl_print_name_dp(ssl, str, z->node.name,
|
|
z->node.dclass, z->dp)) {
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
return;
|
|
}
|
|
}
|
|
lock_rw_unlock(&worker->env.hints->lock);
|
|
}
|
|
|
|
/** do the list_auth_zones command */
|
|
static void
|
|
do_list_auth_zones(RES* ssl, struct auth_zones* az)
|
|
{
|
|
struct auth_zone* z;
|
|
char buf[257], buf2[256];
|
|
lock_rw_rdlock(&az->lock);
|
|
RBTREE_FOR(z, struct auth_zone*, &az->ztree) {
|
|
lock_rw_rdlock(&z->lock);
|
|
dname_str(z->name, buf);
|
|
if(z->zone_expired)
|
|
snprintf(buf2, sizeof(buf2), "expired");
|
|
else {
|
|
uint32_t serial = 0;
|
|
if(auth_zone_get_serial(z, &serial))
|
|
snprintf(buf2, sizeof(buf2), "serial %u",
|
|
(unsigned)serial);
|
|
else snprintf(buf2, sizeof(buf2), "no serial");
|
|
}
|
|
if(!ssl_printf(ssl, "%s\t%s\n", buf, buf2)) {
|
|
/* failure to print */
|
|
lock_rw_unlock(&z->lock);
|
|
lock_rw_unlock(&az->lock);
|
|
return;
|
|
}
|
|
lock_rw_unlock(&z->lock);
|
|
}
|
|
lock_rw_unlock(&az->lock);
|
|
}
|
|
|
|
/** do the list_local_zones command */
|
|
static void
|
|
do_list_local_zones(RES* ssl, struct local_zones* zones)
|
|
{
|
|
struct local_zone* z;
|
|
char buf[257];
|
|
lock_rw_rdlock(&zones->lock);
|
|
RBTREE_FOR(z, struct local_zone*, &zones->ztree) {
|
|
lock_rw_rdlock(&z->lock);
|
|
dname_str(z->name, buf);
|
|
if(!ssl_printf(ssl, "%s %s\n", buf,
|
|
local_zone_type2str(z->type))) {
|
|
/* failure to print */
|
|
lock_rw_unlock(&z->lock);
|
|
lock_rw_unlock(&zones->lock);
|
|
return;
|
|
}
|
|
lock_rw_unlock(&z->lock);
|
|
}
|
|
lock_rw_unlock(&zones->lock);
|
|
}
|
|
|
|
/** do the list_local_data command */
|
|
static void
|
|
do_list_local_data(RES* ssl, struct worker* worker, struct local_zones* zones)
|
|
{
|
|
struct local_zone* z;
|
|
struct local_data* d;
|
|
struct local_rrset* p;
|
|
char* s = (char*)sldns_buffer_begin(worker->env.scratch_buffer);
|
|
size_t slen = sldns_buffer_capacity(worker->env.scratch_buffer);
|
|
lock_rw_rdlock(&zones->lock);
|
|
RBTREE_FOR(z, struct local_zone*, &zones->ztree) {
|
|
lock_rw_rdlock(&z->lock);
|
|
RBTREE_FOR(d, struct local_data*, &z->data) {
|
|
for(p = d->rrsets; p; p = p->next) {
|
|
struct packed_rrset_data* d =
|
|
(struct packed_rrset_data*)p->rrset->entry.data;
|
|
size_t i;
|
|
for(i=0; i<d->count + d->rrsig_count; i++) {
|
|
if(!packed_rr_to_string(p->rrset, i,
|
|
0, s, slen)) {
|
|
if(!ssl_printf(ssl, "BADRR\n")) {
|
|
lock_rw_unlock(&z->lock);
|
|
lock_rw_unlock(&zones->lock);
|
|
return;
|
|
}
|
|
}
|
|
if(!ssl_printf(ssl, "%s\n", s)) {
|
|
lock_rw_unlock(&z->lock);
|
|
lock_rw_unlock(&zones->lock);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
lock_rw_unlock(&z->lock);
|
|
}
|
|
lock_rw_unlock(&zones->lock);
|
|
}
|
|
|
|
/** do the view_list_local_zones command */
|
|
static void
|
|
do_view_list_local_zones(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct view* v = views_find_view(worker->daemon->views,
|
|
arg, 0 /* get read lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(v->local_zones) {
|
|
do_list_local_zones(ssl, v->local_zones);
|
|
}
|
|
lock_rw_unlock(&v->lock);
|
|
}
|
|
|
|
/** do the view_list_local_data command */
|
|
static void
|
|
do_view_list_local_data(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct view* v = views_find_view(worker->daemon->views,
|
|
arg, 0 /* get read lock*/);
|
|
if(!v) {
|
|
ssl_printf(ssl,"no view with name: %s\n", arg);
|
|
return;
|
|
}
|
|
if(v->local_zones) {
|
|
do_list_local_data(ssl, worker, v->local_zones);
|
|
}
|
|
lock_rw_unlock(&v->lock);
|
|
}
|
|
|
|
/** struct for user arg ratelimit list */
|
|
struct ratelimit_list_arg {
|
|
/** the infra cache */
|
|
struct infra_cache* infra;
|
|
/** the SSL to print to */
|
|
RES* ssl;
|
|
/** all or only ratelimited */
|
|
int all;
|
|
/** current time */
|
|
time_t now;
|
|
/** if backoff is enabled */
|
|
int backoff;
|
|
};
|
|
|
|
#define ip_ratelimit_list_arg ratelimit_list_arg
|
|
|
|
/** list items in the ratelimit table */
|
|
static void
|
|
rate_list(struct lruhash_entry* e, void* arg)
|
|
{
|
|
struct ratelimit_list_arg* a = (struct ratelimit_list_arg*)arg;
|
|
struct rate_key* k = (struct rate_key*)e->key;
|
|
struct rate_data* d = (struct rate_data*)e->data;
|
|
char buf[257];
|
|
int lim = infra_find_ratelimit(a->infra, k->name, k->namelen);
|
|
int max = infra_rate_max(d, a->now, a->backoff);
|
|
if(a->all == 0) {
|
|
if(max < lim)
|
|
return;
|
|
}
|
|
dname_str(k->name, buf);
|
|
ssl_printf(a->ssl, "%s %d limit %d\n", buf, max, lim);
|
|
}
|
|
|
|
/** list items in the ip_ratelimit table */
|
|
static void
|
|
ip_rate_list(struct lruhash_entry* e, void* arg)
|
|
{
|
|
char ip[128];
|
|
struct ip_ratelimit_list_arg* a = (struct ip_ratelimit_list_arg*)arg;
|
|
struct ip_rate_key* k = (struct ip_rate_key*)e->key;
|
|
struct ip_rate_data* d = (struct ip_rate_data*)e->data;
|
|
int lim = infra_ip_ratelimit;
|
|
int max = infra_rate_max(d, a->now, a->backoff);
|
|
if(a->all == 0) {
|
|
if(max < lim)
|
|
return;
|
|
}
|
|
addr_to_str(&k->addr, k->addrlen, ip, sizeof(ip));
|
|
ssl_printf(a->ssl, "%s %d limit %d\n", ip, max, lim);
|
|
}
|
|
|
|
/** do the ratelimit_list command */
|
|
static void
|
|
do_ratelimit_list(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct ratelimit_list_arg a;
|
|
a.all = 0;
|
|
a.infra = worker->env.infra_cache;
|
|
a.now = *worker->env.now;
|
|
a.ssl = ssl;
|
|
a.backoff = worker->env.cfg->ratelimit_backoff;
|
|
arg = skipwhite(arg);
|
|
if(strcmp(arg, "+a") == 0)
|
|
a.all = 1;
|
|
if(a.infra->domain_rates==NULL ||
|
|
(a.all == 0 && infra_dp_ratelimit == 0))
|
|
return;
|
|
slabhash_traverse(a.infra->domain_rates, 0, rate_list, &a);
|
|
}
|
|
|
|
/** do the ip_ratelimit_list command */
|
|
static void
|
|
do_ip_ratelimit_list(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
struct ip_ratelimit_list_arg a;
|
|
a.all = 0;
|
|
a.infra = worker->env.infra_cache;
|
|
a.now = *worker->env.now;
|
|
a.ssl = ssl;
|
|
a.backoff = worker->env.cfg->ip_ratelimit_backoff;
|
|
arg = skipwhite(arg);
|
|
if(strcmp(arg, "+a") == 0)
|
|
a.all = 1;
|
|
if(a.infra->client_ip_rates==NULL ||
|
|
(a.all == 0 && infra_ip_ratelimit == 0))
|
|
return;
|
|
slabhash_traverse(a.infra->client_ip_rates, 0, ip_rate_list, &a);
|
|
}
|
|
|
|
/** do the rpz_enable/disable command */
|
|
static void
|
|
do_rpz_enable_disable(RES* ssl, struct worker* worker, char* arg, int enable) {
|
|
size_t nmlen;
|
|
int nmlabs;
|
|
uint8_t *nm = NULL;
|
|
struct auth_zones *az = worker->env.auth_zones;
|
|
struct auth_zone *z = NULL;
|
|
if (!parse_arg_name(ssl, arg, &nm, &nmlen, &nmlabs))
|
|
return;
|
|
if (az) {
|
|
lock_rw_rdlock(&az->lock);
|
|
z = auth_zone_find(az, nm, nmlen, LDNS_RR_CLASS_IN);
|
|
if (z) {
|
|
lock_rw_wrlock(&z->lock);
|
|
}
|
|
lock_rw_unlock(&az->lock);
|
|
}
|
|
free(nm);
|
|
if (!z) {
|
|
(void) ssl_printf(ssl, "error no auth-zone %s\n", arg);
|
|
return;
|
|
}
|
|
if (!z->rpz) {
|
|
(void) ssl_printf(ssl, "error auth-zone %s not RPZ\n", arg);
|
|
lock_rw_unlock(&z->lock);
|
|
return;
|
|
}
|
|
if (enable) {
|
|
rpz_enable(z->rpz);
|
|
} else {
|
|
rpz_disable(z->rpz);
|
|
}
|
|
lock_rw_unlock(&z->lock);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** do the rpz_enable command */
|
|
static void
|
|
do_rpz_enable(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
do_rpz_enable_disable(ssl, worker, arg, 1);
|
|
}
|
|
|
|
/** do the rpz_disable command */
|
|
static void
|
|
do_rpz_disable(RES* ssl, struct worker* worker, char* arg)
|
|
{
|
|
do_rpz_enable_disable(ssl, worker, arg, 0);
|
|
}
|
|
|
|
/** Write the cookie secrets to file, returns `0` on failure.
|
|
* Caller has to hold the lock. */
|
|
static int
|
|
cookie_secret_file_dump(RES* ssl, struct worker* worker) {
|
|
char const* secret_file = worker->env.cfg->cookie_secret_file;
|
|
struct cookie_secrets* cookie_secrets = worker->daemon->cookie_secrets;
|
|
char secret_hex[UNBOUND_COOKIE_SECRET_SIZE * 2 + 1];
|
|
FILE* f;
|
|
size_t i;
|
|
if(secret_file == NULL || secret_file[0]==0) {
|
|
(void)ssl_printf(ssl, "error: no cookie secret file configured\n");
|
|
return 0;
|
|
}
|
|
log_assert( secret_file != NULL );
|
|
|
|
/* open write only and truncate */
|
|
if((f = fopen(secret_file, "w")) == NULL ) {
|
|
(void)ssl_printf(ssl, "unable to open cookie secret file %s: %s",
|
|
secret_file, strerror(errno));
|
|
return 0;
|
|
}
|
|
if(cookie_secrets == NULL) {
|
|
/* nothing to write */
|
|
fclose(f);
|
|
return 1;
|
|
}
|
|
|
|
for(i = 0; i < cookie_secrets->cookie_count; i++) {
|
|
struct cookie_secret const* cs = &cookie_secrets->
|
|
cookie_secrets[i];
|
|
ssize_t const len = hex_ntop(cs->cookie_secret,
|
|
UNBOUND_COOKIE_SECRET_SIZE, secret_hex,
|
|
sizeof(secret_hex));
|
|
(void)len; /* silence unused variable warning with -DNDEBUG */
|
|
log_assert( len == UNBOUND_COOKIE_SECRET_SIZE * 2 );
|
|
secret_hex[UNBOUND_COOKIE_SECRET_SIZE * 2] = '\0';
|
|
fprintf(f, "%s\n", secret_hex);
|
|
}
|
|
explicit_bzero(secret_hex, sizeof(secret_hex));
|
|
fclose(f);
|
|
return 1;
|
|
}
|
|
|
|
/** Activate cookie secret */
|
|
static void
|
|
do_activate_cookie_secret(RES* ssl, struct worker* worker) {
|
|
char const* secret_file = worker->env.cfg->cookie_secret_file;
|
|
struct cookie_secrets* cookie_secrets = worker->daemon->cookie_secrets;
|
|
|
|
if(secret_file == NULL || secret_file[0] == 0) {
|
|
(void)ssl_printf(ssl, "error: no cookie secret file configured\n");
|
|
return;
|
|
}
|
|
if(cookie_secrets == NULL) {
|
|
(void)ssl_printf(ssl, "error: there are no cookie_secrets.");
|
|
return;
|
|
}
|
|
lock_basic_lock(&cookie_secrets->lock);
|
|
|
|
if(cookie_secrets->cookie_count <= 1 ) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
(void)ssl_printf(ssl, "error: no staging cookie secret to activate\n");
|
|
return;
|
|
}
|
|
/* Only the worker 0 writes to file, the others update state. */
|
|
if(worker->thread_num == 0 && !cookie_secret_file_dump(ssl, worker)) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
(void)ssl_printf(ssl, "error: writing to cookie secret file: \"%s\"\n",
|
|
secret_file);
|
|
return;
|
|
}
|
|
activate_cookie_secret(cookie_secrets);
|
|
if(worker->thread_num == 0)
|
|
(void)cookie_secret_file_dump(ssl, worker);
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Drop cookie secret */
|
|
static void
|
|
do_drop_cookie_secret(RES* ssl, struct worker* worker) {
|
|
char const* secret_file = worker->env.cfg->cookie_secret_file;
|
|
struct cookie_secrets* cookie_secrets = worker->daemon->cookie_secrets;
|
|
|
|
if(secret_file == NULL || secret_file[0] == 0) {
|
|
(void)ssl_printf(ssl, "error: no cookie secret file configured\n");
|
|
return;
|
|
}
|
|
if(cookie_secrets == NULL) {
|
|
(void)ssl_printf(ssl, "error: there are no cookie_secrets.");
|
|
return;
|
|
}
|
|
lock_basic_lock(&cookie_secrets->lock);
|
|
|
|
if(cookie_secrets->cookie_count <= 1 ) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
(void)ssl_printf(ssl, "error: can not drop the currently active cookie secret\n");
|
|
return;
|
|
}
|
|
/* Only the worker 0 writes to file, the others update state. */
|
|
if(worker->thread_num == 0 && !cookie_secret_file_dump(ssl, worker)) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
(void)ssl_printf(ssl, "error: writing to cookie secret file: \"%s\"\n",
|
|
secret_file);
|
|
return;
|
|
}
|
|
drop_cookie_secret(cookie_secrets);
|
|
if(worker->thread_num == 0)
|
|
(void)cookie_secret_file_dump(ssl, worker);
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Add cookie secret */
|
|
static void
|
|
do_add_cookie_secret(RES* ssl, struct worker* worker, char* arg) {
|
|
uint8_t secret[UNBOUND_COOKIE_SECRET_SIZE];
|
|
char const* secret_file = worker->env.cfg->cookie_secret_file;
|
|
struct cookie_secrets* cookie_secrets = worker->daemon->cookie_secrets;
|
|
|
|
if(secret_file == NULL || secret_file[0] == 0) {
|
|
(void)ssl_printf(ssl, "error: no cookie secret file configured\n");
|
|
return;
|
|
}
|
|
if(cookie_secrets == NULL) {
|
|
worker->daemon->cookie_secrets = cookie_secrets_create();
|
|
if(!worker->daemon->cookie_secrets) {
|
|
(void)ssl_printf(ssl, "error: out of memory");
|
|
return;
|
|
}
|
|
cookie_secrets = worker->daemon->cookie_secrets;
|
|
}
|
|
lock_basic_lock(&cookie_secrets->lock);
|
|
|
|
if(*arg == '\0') {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
(void)ssl_printf(ssl, "error: missing argument (cookie_secret)\n");
|
|
return;
|
|
}
|
|
if(strlen(arg) != 32) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
explicit_bzero(arg, strlen(arg));
|
|
(void)ssl_printf(ssl, "invalid cookie secret: invalid argument length\n");
|
|
(void)ssl_printf(ssl, "please provide a 128bit hex encoded secret\n");
|
|
return;
|
|
}
|
|
if(hex_pton(arg, secret, UNBOUND_COOKIE_SECRET_SIZE) !=
|
|
UNBOUND_COOKIE_SECRET_SIZE ) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
explicit_bzero(secret, UNBOUND_COOKIE_SECRET_SIZE);
|
|
explicit_bzero(arg, strlen(arg));
|
|
(void)ssl_printf(ssl, "invalid cookie secret: parse error\n");
|
|
(void)ssl_printf(ssl, "please provide a 128bit hex encoded secret\n");
|
|
return;
|
|
}
|
|
/* Only the worker 0 writes to file, the others update state. */
|
|
if(worker->thread_num == 0 && !cookie_secret_file_dump(ssl, worker)) {
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
explicit_bzero(secret, UNBOUND_COOKIE_SECRET_SIZE);
|
|
explicit_bzero(arg, strlen(arg));
|
|
(void)ssl_printf(ssl, "error: writing to cookie secret file: \"%s\"\n",
|
|
secret_file);
|
|
return;
|
|
}
|
|
add_cookie_secret(cookie_secrets, secret, UNBOUND_COOKIE_SECRET_SIZE);
|
|
explicit_bzero(secret, UNBOUND_COOKIE_SECRET_SIZE);
|
|
if(worker->thread_num == 0)
|
|
(void)cookie_secret_file_dump(ssl, worker);
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
explicit_bzero(arg, strlen(arg));
|
|
send_ok(ssl);
|
|
}
|
|
|
|
/** Print cookie secrets */
|
|
static void
|
|
do_print_cookie_secrets(RES* ssl, struct worker* worker) {
|
|
struct cookie_secrets* cookie_secrets = worker->daemon->cookie_secrets;
|
|
char secret_hex[UNBOUND_COOKIE_SECRET_SIZE * 2 + 1];
|
|
int i;
|
|
|
|
if(!cookie_secrets)
|
|
return; /* Output is empty. */
|
|
lock_basic_lock(&cookie_secrets->lock);
|
|
for(i = 0; (size_t)i < cookie_secrets->cookie_count; i++) {
|
|
struct cookie_secret const* cs = &cookie_secrets->
|
|
cookie_secrets[i];
|
|
ssize_t const len = hex_ntop(cs->cookie_secret,
|
|
UNBOUND_COOKIE_SECRET_SIZE, secret_hex,
|
|
sizeof(secret_hex));
|
|
(void)len; /* silence unused variable warning with -DNDEBUG */
|
|
log_assert( len == UNBOUND_COOKIE_SECRET_SIZE * 2 );
|
|
secret_hex[UNBOUND_COOKIE_SECRET_SIZE * 2] = '\0';
|
|
if (i == 0)
|
|
(void)ssl_printf(ssl, "active : %s\n", secret_hex);
|
|
else if (cookie_secrets->cookie_count == 2)
|
|
(void)ssl_printf(ssl, "staging: %s\n", secret_hex);
|
|
else
|
|
(void)ssl_printf(ssl, "staging[%d]: %s\n", i,
|
|
secret_hex);
|
|
}
|
|
lock_basic_unlock(&cookie_secrets->lock);
|
|
explicit_bzero(secret_hex, sizeof(secret_hex));
|
|
}
|
|
|
|
/** check for name with end-of-string, space or tab after it */
|
|
static int
|
|
cmdcmp(char* p, const char* cmd, size_t len)
|
|
{
|
|
return strncmp(p,cmd,len)==0 && (p[len]==0||p[len]==' '||p[len]=='\t');
|
|
}
|
|
|
|
/** execute a remote control command */
|
|
static void
|
|
execute_cmd(struct daemon_remote* rc, RES* ssl, char* cmd,
|
|
struct worker* worker)
|
|
{
|
|
char* p = skipwhite(cmd);
|
|
/* compare command */
|
|
if(cmdcmp(p, "stop", 4)) {
|
|
do_stop(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "reload_keep_cache", 17)) {
|
|
do_reload(ssl, worker, 1);
|
|
return;
|
|
} else if(cmdcmp(p, "reload", 6)) {
|
|
do_reload(ssl, worker, 0);
|
|
return;
|
|
} else if(cmdcmp(p, "stats_noreset", 13)) {
|
|
do_stats(ssl, worker, 0);
|
|
return;
|
|
} else if(cmdcmp(p, "stats", 5)) {
|
|
do_stats(ssl, worker, 1);
|
|
return;
|
|
} else if(cmdcmp(p, "status", 6)) {
|
|
do_status(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "dump_cache", 10)) {
|
|
#ifdef THREADS_DISABLED
|
|
if(worker->daemon->num > 1) {
|
|
(void)ssl_printf(ssl, "dump_cache/load_cache is not "
|
|
"supported in multi-process operation\n");
|
|
return;
|
|
}
|
|
#endif
|
|
(void)dump_cache(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "load_cache", 10)) {
|
|
#ifdef THREADS_DISABLED
|
|
if(worker->daemon->num > 1) {
|
|
/* The warning can't be printed when stdin is sending
|
|
* data; just return */
|
|
return;
|
|
}
|
|
#endif
|
|
if(load_cache(ssl, worker)) send_ok(ssl);
|
|
return;
|
|
} else if(cmdcmp(p, "list_forwards", 13)) {
|
|
do_list_forwards(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "list_stubs", 10)) {
|
|
do_list_stubs(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "list_insecure", 13)) {
|
|
do_insecure_list(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "list_local_zones", 16)) {
|
|
do_list_local_zones(ssl, worker->daemon->local_zones);
|
|
return;
|
|
} else if(cmdcmp(p, "list_local_data", 15)) {
|
|
do_list_local_data(ssl, worker, worker->daemon->local_zones);
|
|
return;
|
|
} else if(cmdcmp(p, "view_list_local_zones", 21)) {
|
|
do_view_list_local_zones(ssl, worker, skipwhite(p+21));
|
|
return;
|
|
} else if(cmdcmp(p, "view_list_local_data", 20)) {
|
|
do_view_list_local_data(ssl, worker, skipwhite(p+20));
|
|
return;
|
|
} else if(cmdcmp(p, "ratelimit_list", 14)) {
|
|
do_ratelimit_list(ssl, worker, p+14);
|
|
return;
|
|
} else if(cmdcmp(p, "ip_ratelimit_list", 17)) {
|
|
do_ip_ratelimit_list(ssl, worker, p+17);
|
|
return;
|
|
} else if(cmdcmp(p, "list_auth_zones", 15)) {
|
|
do_list_auth_zones(ssl, worker->env.auth_zones);
|
|
return;
|
|
} else if(cmdcmp(p, "auth_zone_reload", 16)) {
|
|
do_auth_zone_reload(ssl, worker, skipwhite(p+16));
|
|
return;
|
|
} else if(cmdcmp(p, "auth_zone_transfer", 18)) {
|
|
do_auth_zone_transfer(ssl, worker, skipwhite(p+18));
|
|
return;
|
|
} else if(cmdcmp(p, "insecure_add", 12)) {
|
|
/* must always distribute this cmd */
|
|
if(rc) distribute_cmd(rc, ssl, cmd);
|
|
do_insecure_add(ssl, worker, skipwhite(p+12));
|
|
return;
|
|
} else if(cmdcmp(p, "insecure_remove", 15)) {
|
|
/* must always distribute this cmd */
|
|
if(rc) distribute_cmd(rc, ssl, cmd);
|
|
do_insecure_remove(ssl, worker, skipwhite(p+15));
|
|
return;
|
|
} else if(cmdcmp(p, "flush_stats", 11)) {
|
|
/* must always distribute this cmd */
|
|
if(rc) distribute_cmd(rc, ssl, cmd);
|
|
do_flush_stats(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "flush_requestlist", 17)) {
|
|
/* must always distribute this cmd */
|
|
if(rc) distribute_cmd(rc, ssl, cmd);
|
|
do_flush_requestlist(ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "lookup", 6)) {
|
|
do_lookup(ssl, worker, skipwhite(p+6));
|
|
return;
|
|
/* The following are commands that read stdin.
|
|
* Each line needs to be distributed if THREADS_DISABLED.
|
|
*/
|
|
} else if(cmdcmp(p, "local_zones_remove", 18)) {
|
|
do_zones_remove(rc, ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "local_zones", 11)) {
|
|
do_zones_add(rc, ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "local_datas_remove", 18)) {
|
|
do_datas_remove(rc, ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "local_datas", 11)) {
|
|
do_datas_add(rc, ssl, worker);
|
|
return;
|
|
} else if(cmdcmp(p, "view_local_datas_remove", 23)){
|
|
do_view_datas_remove(rc, ssl, worker, skipwhite(p+23));
|
|
return;
|
|
} else if(cmdcmp(p, "view_local_datas", 16)) {
|
|
do_view_datas_add(rc, ssl, worker, skipwhite(p+16));
|
|
return;
|
|
} else if(cmdcmp(p, "print_cookie_secrets", 20)) {
|
|
do_print_cookie_secrets(ssl, worker);
|
|
return;
|
|
}
|
|
|
|
#ifdef THREADS_DISABLED
|
|
/* other processes must execute the command as well */
|
|
/* commands that should not be distributed, returned above. */
|
|
if(rc) { /* only if this thread is the master (rc) thread */
|
|
/* done before the code below, which may split the string */
|
|
distribute_cmd(rc, ssl, cmd);
|
|
}
|
|
#endif
|
|
if(cmdcmp(p, "verbosity", 9)) {
|
|
do_verbosity(ssl, skipwhite(p+9));
|
|
} else if(cmdcmp(p, "local_zone_remove", 17)) {
|
|
do_zone_remove(ssl, worker->daemon->local_zones, skipwhite(p+17));
|
|
} else if(cmdcmp(p, "local_zone", 10)) {
|
|
do_zone_add(ssl, worker->daemon->local_zones, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "local_data_remove", 17)) {
|
|
do_data_remove(ssl, worker->daemon->local_zones, skipwhite(p+17));
|
|
} else if(cmdcmp(p, "local_data", 10)) {
|
|
do_data_add(ssl, worker->daemon->local_zones, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "forward_add", 11)) {
|
|
do_forward_add(ssl, worker, skipwhite(p+11));
|
|
} else if(cmdcmp(p, "forward_remove", 14)) {
|
|
do_forward_remove(ssl, worker, skipwhite(p+14));
|
|
} else if(cmdcmp(p, "forward", 7)) {
|
|
do_forward(ssl, worker, skipwhite(p+7));
|
|
} else if(cmdcmp(p, "stub_add", 8)) {
|
|
do_stub_add(ssl, worker, skipwhite(p+8));
|
|
} else if(cmdcmp(p, "stub_remove", 11)) {
|
|
do_stub_remove(ssl, worker, skipwhite(p+11));
|
|
} else if(cmdcmp(p, "view_local_zone_remove", 22)) {
|
|
do_view_zone_remove(ssl, worker, skipwhite(p+22));
|
|
} else if(cmdcmp(p, "view_local_zone", 15)) {
|
|
do_view_zone_add(ssl, worker, skipwhite(p+15));
|
|
} else if(cmdcmp(p, "view_local_data_remove", 22)) {
|
|
do_view_data_remove(ssl, worker, skipwhite(p+22));
|
|
} else if(cmdcmp(p, "view_local_data", 15)) {
|
|
do_view_data_add(ssl, worker, skipwhite(p+15));
|
|
} else if(cmdcmp(p, "flush_zone", 10)) {
|
|
do_flush_zone(ssl, worker, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "flush_type", 10)) {
|
|
do_flush_type(ssl, worker, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "flush_infra", 11)) {
|
|
do_flush_infra(ssl, worker, skipwhite(p+11));
|
|
} else if(cmdcmp(p, "flush", 5)) {
|
|
do_flush_name(ssl, worker, skipwhite(p+5));
|
|
} else if(cmdcmp(p, "dump_requestlist", 16)) {
|
|
do_dump_requestlist(ssl, worker);
|
|
} else if(cmdcmp(p, "dump_infra", 10)) {
|
|
do_dump_infra(ssl, worker);
|
|
} else if(cmdcmp(p, "log_reopen", 10)) {
|
|
do_log_reopen(ssl, worker);
|
|
} else if(cmdcmp(p, "set_option", 10)) {
|
|
do_set_option(ssl, worker, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "get_option", 10)) {
|
|
do_get_option(ssl, worker, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "flush_bogus", 11)) {
|
|
do_flush_bogus(ssl, worker, skipwhite(p+11));
|
|
} else if(cmdcmp(p, "flush_negative", 14)) {
|
|
do_flush_negative(ssl, worker, skipwhite(p+14));
|
|
} else if(cmdcmp(p, "rpz_enable", 10)) {
|
|
do_rpz_enable(ssl, worker, skipwhite(p+10));
|
|
} else if(cmdcmp(p, "rpz_disable", 11)) {
|
|
do_rpz_disable(ssl, worker, skipwhite(p+11));
|
|
} else if(cmdcmp(p, "add_cookie_secret", 17)) {
|
|
do_add_cookie_secret(ssl, worker, skipwhite(p+17));
|
|
} else if(cmdcmp(p, "drop_cookie_secret", 18)) {
|
|
do_drop_cookie_secret(ssl, worker);
|
|
} else if(cmdcmp(p, "activate_cookie_secret", 22)) {
|
|
do_activate_cookie_secret(ssl, worker);
|
|
} else {
|
|
(void)ssl_printf(ssl, "error unknown command '%s'\n", p);
|
|
}
|
|
}
|
|
|
|
void
|
|
daemon_remote_exec(struct worker* worker)
|
|
{
|
|
/* read the cmd string */
|
|
uint8_t* msg = NULL;
|
|
uint32_t len = 0;
|
|
if(!tube_read_msg(worker->cmd, &msg, &len, 0)) {
|
|
log_err("daemon_remote_exec: tube_read_msg failed");
|
|
return;
|
|
}
|
|
verbose(VERB_ALGO, "remote exec distributed: %s", (char*)msg);
|
|
execute_cmd(NULL, NULL, (char*)msg, worker);
|
|
free(msg);
|
|
}
|
|
|
|
/** handle remote control request */
|
|
static void
|
|
handle_req(struct daemon_remote* rc, struct rc_state* s, RES* res)
|
|
{
|
|
int r;
|
|
char pre[10];
|
|
char magic[7];
|
|
char buf[MAX_CMD_STRLINE];
|
|
#ifdef USE_WINSOCK
|
|
/* makes it possible to set the socket blocking again. */
|
|
/* basically removes it from winsock_event ... */
|
|
WSAEventSelect(s->c->fd, NULL, 0);
|
|
#endif
|
|
fd_set_block(s->c->fd);
|
|
|
|
/* try to read magic UBCT[version]_space_ string */
|
|
if(res->ssl) {
|
|
ERR_clear_error();
|
|
if((r=SSL_read(res->ssl, magic, (int)sizeof(magic)-1)) <= 0) {
|
|
int r2;
|
|
if((r2=SSL_get_error(res->ssl, r)) == SSL_ERROR_ZERO_RETURN)
|
|
return;
|
|
log_crypto_err_io("could not SSL_read", r2);
|
|
return;
|
|
}
|
|
} else {
|
|
while(1) {
|
|
ssize_t rr = recv(res->fd, magic, sizeof(magic)-1, 0);
|
|
if(rr <= 0) {
|
|
if(rr == 0) return;
|
|
if(errno == EINTR || errno == EAGAIN)
|
|
continue;
|
|
log_err("could not recv: %s", sock_strerror(errno));
|
|
return;
|
|
}
|
|
r = (int)rr;
|
|
break;
|
|
}
|
|
}
|
|
magic[6] = 0;
|
|
if( r != 6 || strncmp(magic, "UBCT", 4) != 0) {
|
|
verbose(VERB_QUERY, "control connection has bad magic string");
|
|
/* probably wrong tool connected, ignore it completely */
|
|
return;
|
|
}
|
|
|
|
/* read the command line */
|
|
if(!ssl_read_line(res, buf, sizeof(buf))) {
|
|
return;
|
|
}
|
|
snprintf(pre, sizeof(pre), "UBCT%d ", UNBOUND_CONTROL_VERSION);
|
|
if(strcmp(magic, pre) != 0) {
|
|
verbose(VERB_QUERY, "control connection had bad "
|
|
"version %s, cmd: %s", magic, buf);
|
|
ssl_printf(res, "error version mismatch\n");
|
|
return;
|
|
}
|
|
verbose(VERB_DETAIL, "control cmd: %s", buf);
|
|
|
|
/* figure out what to do */
|
|
execute_cmd(rc, res, buf, rc->worker);
|
|
}
|
|
|
|
/** handle SSL_do_handshake changes to the file descriptor to wait for later */
|
|
static int
|
|
remote_handshake_later(struct daemon_remote* rc, struct rc_state* s,
|
|
struct comm_point* c, int r, int r2)
|
|
{
|
|
if(r2 == SSL_ERROR_WANT_READ) {
|
|
if(s->shake_state == rc_hs_read) {
|
|
/* try again later */
|
|
return 0;
|
|
}
|
|
s->shake_state = rc_hs_read;
|
|
comm_point_listen_for_rw(c, 1, 0);
|
|
return 0;
|
|
} else if(r2 == SSL_ERROR_WANT_WRITE) {
|
|
if(s->shake_state == rc_hs_write) {
|
|
/* try again later */
|
|
return 0;
|
|
}
|
|
s->shake_state = rc_hs_write;
|
|
comm_point_listen_for_rw(c, 0, 1);
|
|
return 0;
|
|
} else {
|
|
if(r == 0)
|
|
log_err("remote control connection closed prematurely");
|
|
log_addr(VERB_OPS, "failed connection from",
|
|
&s->c->repinfo.remote_addr, s->c->repinfo.remote_addrlen);
|
|
log_crypto_err_io("remote control failed ssl", r2);
|
|
clean_point(rc, s);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int remote_control_callback(struct comm_point* c, void* arg, int err,
|
|
struct comm_reply* ATTR_UNUSED(rep))
|
|
{
|
|
RES res;
|
|
struct rc_state* s = (struct rc_state*)arg;
|
|
struct daemon_remote* rc = s->rc;
|
|
int r;
|
|
if(err != NETEVENT_NOERROR) {
|
|
if(err==NETEVENT_TIMEOUT)
|
|
log_err("remote control timed out");
|
|
clean_point(rc, s);
|
|
return 0;
|
|
}
|
|
if(s->ssl) {
|
|
/* (continue to) setup the SSL connection */
|
|
ERR_clear_error();
|
|
r = SSL_do_handshake(s->ssl);
|
|
if(r != 1) {
|
|
int r2 = SSL_get_error(s->ssl, r);
|
|
return remote_handshake_later(rc, s, c, r, r2);
|
|
}
|
|
s->shake_state = rc_none;
|
|
}
|
|
|
|
/* once handshake has completed, check authentication */
|
|
if (!rc->use_cert) {
|
|
verbose(VERB_ALGO, "unauthenticated remote control connection");
|
|
} else if(SSL_get_verify_result(s->ssl) == X509_V_OK) {
|
|
#ifdef HAVE_SSL_GET1_PEER_CERTIFICATE
|
|
X509* x = SSL_get1_peer_certificate(s->ssl);
|
|
#else
|
|
X509* x = SSL_get_peer_certificate(s->ssl);
|
|
#endif
|
|
if(!x) {
|
|
verbose(VERB_DETAIL, "remote control connection "
|
|
"provided no client certificate");
|
|
clean_point(rc, s);
|
|
return 0;
|
|
}
|
|
verbose(VERB_ALGO, "remote control connection authenticated");
|
|
X509_free(x);
|
|
} else {
|
|
verbose(VERB_DETAIL, "remote control connection failed to "
|
|
"authenticate with client certificate");
|
|
clean_point(rc, s);
|
|
return 0;
|
|
}
|
|
|
|
/* if OK start to actually handle the request */
|
|
res.ssl = s->ssl;
|
|
res.fd = c->fd;
|
|
handle_req(rc, s, &res);
|
|
|
|
verbose(VERB_ALGO, "remote control operation completed");
|
|
clean_point(rc, s);
|
|
return 0;
|
|
}
|