OpenGrok

/*
 * daemon/remote.h - 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 SSLv3/TLS capable web browser.
 * The channel is secured using SSLv3 or TLSv1, and certificates.
 * Both the server and the client(control tool) have their own keys.
 */

#ifndef DAEMON_REMOTE_H
#define DAEMON_REMOTE_H
#ifdef HAVE_OPENSSL_SSL_H
#include <openssl/ssl.h>
#endif
#include "util/locks.h"
#include "libunbound/remote.h"
struct config_file;
struct listen_list;
struct listen_port;
struct worker;
struct comm_reply;
struct comm_point;
struct daemon_remote;
struct config_strlist_head;

/** number of milliseconds timeout on incoming remote control handshake */
#define REMOTE_CONTROL_TCP_TIMEOUT 120000

/**
 * a busy control command connection, SSL state
 */
struct rc_state {
	/** the next item in list */
	struct rc_state* next;
	/** the commpoint */
	struct comm_point* c;
	/** in the handshake part */
	enum { rc_none, rc_hs_read, rc_hs_write } shake_state;
#ifdef HAVE_SSL
	/** the ssl state */
	SSL* ssl;
#endif
	/** file descriptor */
        int fd;
	/** the rc this is part of */
	struct daemon_remote* rc;
};

/**
 * The remote control tool state.
 * The state is only created for the first thread, other threads
 * are called from this thread.  Only the first threads listens to
 * the control port.  The other threads do not, but are called on the
 * command channel(pipe) from the first thread.
 */
struct daemon_remote {
	/** the worker for this remote control */
	struct worker* worker;
	/** commpoints for accepting remote control connections */
	struct listen_list* accept_list;
	/* if certificates are used */
	int use_cert;
	/** number of active commpoints that are handling remote control */
	int active;
	/** max active commpoints */
	int max_active;
	/** current commpoints busy; should be a short list, malloced */
	struct rc_state* busy_list;
#ifdef HAVE_SSL
	/** the SSL context for creating new SSL streams */
	SSL_CTX* ctx;
#endif
};

/**
 * Connection to print to, either SSL or plain over fd
 */
struct remote_stream {
#ifdef HAVE_SSL
	/** SSL structure, nonNULL if using SSL */
	SSL* ssl;
#endif
	/** file descriptor for plain transfer */
	int fd;
};
typedef struct remote_stream RES;

/**
 * Notification status. This is exchanged between the fast reload thread
 * and the server thread, over the commpair sockets.
 */
enum fast_reload_notification {
	/** nothing, not used */
	fast_reload_notification_none = 0,
	/** the fast reload thread is done */
	fast_reload_notification_done = 1,
	/** the fast reload thread is done but with an error, it failed */
	fast_reload_notification_done_error = 2,
	/** the fast reload thread is told to exit by the server thread.
	 * Sent on server quit while the reload is running. */
	fast_reload_notification_exit = 3,
	/** the fast reload thread has exited, after being told to exit */
	fast_reload_notification_exited = 4,
	/** the fast reload thread has information to print out */
	fast_reload_notification_printout = 5,
	/** stop as part of the reload the thread and other threads */
	fast_reload_notification_reload_stop = 6,
	/** ack the stop as part of the reload, and also ack start */
	fast_reload_notification_reload_ack = 7,
	/** resume from stop as part of the reload */
	fast_reload_notification_reload_start = 8,
	/** the fast reload thread wants the mainthread to poll workers,
	 * after the reload, sent when nopause is used */
	fast_reload_notification_reload_nopause_poll = 9
};

/**
 * Fast reload printout queue. Contains a list of strings, that need to be
 * printed over the file descriptor.
 */
struct fast_reload_printq {
	/** if this item is in a list, the previous and next */
	struct fast_reload_printq *prev, *next;
	/** if this item is in a list, it is true. */
	int in_list;
	/** list of strings to printout */
	struct config_strlist_head* to_print;
	/** the current item to print. It is malloced. NULL if none. */
	char* client_item;
	/** The length, strlen, of the client_item, that has to be sent. */
	int client_len;
	/** The number of bytes sent of client_item. */
	int client_byte_count;
	/** the comm point for the client connection, the remote control
	 * client. */
	struct comm_point* client_cp;
	/** the remote control connection to print output to. */
	struct remote_stream remote;
	/** the worker that the event is added in */
	struct worker* worker;
};

/**
 * Fast reload auth zone change. Keeps track if an auth zone was removed,
 * added or changed. This is needed because workers can have events for
 * dealing with auth zones, like transfers, and those have to be removed
 * too, not just the auth zone structure from the tree. */
struct fast_reload_auth_change {
	/** next in the list of auth zone changes. */
	struct fast_reload_auth_change* next;
	/** the zone in the old config */
	struct auth_zone* old_z;
	/** the zone in the new config */
	struct auth_zone* new_z;
	/** if the zone was deleted */
	int is_deleted;
	/** if the zone was added */
	int is_added;
	/** if the zone has been changed */
	int is_changed;
};

/**
 * Fast reload thread structure
 */
struct fast_reload_thread {
	/** the thread number for the dtio thread,
	 * must be first to cast thread arg to int* in checklock code. */
	int threadnum;
	/** communication socket pair, that sends commands */
	int commpair[2];
	/** thread id, of the io thread */
	ub_thread_type tid;
#ifdef HAVE_GETTID
	/** thread tid, the LWP id */
	pid_t thread_tid;
	/** if logging should include the LWP id */
	int thread_tid_log;
#endif
	/** if the io processing has started */
	int started;
	/** if the thread has to quit */
	int need_to_quit;
	/** verbosity of the fast_reload command, the number of +v options */
	int fr_verb;
	/** option to not pause threads during reload */
	int fr_nopause;
	/** option to drop mesh queries */
	int fr_drop_mesh;

	/** the event that listens on the remote service worker to the
	 * commpair, it receives content from the fast reload thread. */
	void* service_event;
	/** if the event that listens on the remote service worker has
	 * been added to the comm base. */
	int service_event_is_added;
	/** the service event can read a cmd, nonblocking, so it can
	 * save the partial read cmd here */
	uint32_t service_read_cmd;
	/** the number of bytes in service_read_cmd */
	int service_read_cmd_count;
	/** the worker that the service_event is added in */
	struct worker* worker;

	/** the printout of output to the remote client. */
	struct fast_reload_printq *printq;

	/** lock on fr_output, to stop race when both remote control thread
	 * and fast reload thread use fr_output list. */
	lock_basic_type fr_output_lock;
	/** list of strings, that the fast reload thread produces that have
	 * to be printed. The remote control thread can pick them up with
	 * the lock. */
	struct config_strlist_head* fr_output;

	/** communication socket pair, to respond to the reload request */
	int commreload[2];

	/** the list of auth zone changes. */
	struct fast_reload_auth_change* auth_zone_change_list;
	/** the old tree of auth zones, to lookup. */
	struct auth_zones* old_auth_zones;
	/** If the ssl ctxs have changed. */
	int sslctxs_changed;
};

/**
 * Create new remote control state for the daemon.
 * @param cfg: config file with key file settings.
 * @return new state, or NULL on failure.
 */
struct daemon_remote* daemon_remote_create(struct config_file* cfg);

/**
 * remote control state to delete.
 * @param rc: state to delete.
 */
void daemon_remote_delete(struct daemon_remote* rc);

/**
 * remote control state to clear up. Busy and accept points are closed.
 * Does not delete the rc itself, or the ssl context (with its keys).
 * @param rc: state to clear.
 */
void daemon_remote_clear(struct daemon_remote* rc);

/**
 * Open and create listening ports for remote control.
 * @param cfg: config options.
 * @return list of ports or NULL on failure.
 *	can be freed with listening_ports_free().
 */
struct listen_port* daemon_remote_open_ports(struct config_file* cfg);

/**
 * Setup comm points for accepting remote control connections.
 * @param rc: state
 * @param ports: already opened ports.
 * @param worker: worker with communication base. and links to command channels.
 * @return false on error.
 */
int daemon_remote_open_accept(struct daemon_remote* rc,
	struct listen_port* ports, struct worker* worker);

/**
 * Stop accept handlers for TCP (until enabled again)
 * @param rc: state
 */
void daemon_remote_stop_accept(struct daemon_remote* rc);

/**
 * Stop accept handlers for TCP (until enabled again)
 * @param rc: state
 */
void daemon_remote_start_accept(struct daemon_remote* rc);

/**
 * Handle nonthreaded remote cmd execution.
 * @param worker: this worker (the remote worker).
 */
void daemon_remote_exec(struct worker* worker);

#ifdef HAVE_SSL
/**
 * Print fixed line of text over ssl connection in blocking mode
 * @param ssl: print to
 * @param text: the text.
 * @return false on connection failure.
 */
int ssl_print_text(RES* ssl, const char* text);

/**
 * printf style printing to the ssl connection
 * @param ssl: the RES connection to print to. Blocking.
 * @param format: printf style format string.
 * @return success or false on a network failure.
 */
int ssl_printf(RES* ssl, const char* format, ...)
        ATTR_FORMAT(printf, 2, 3);

/**
 * Read until \n is encountered
 * If stream signals EOF, the string up to then is returned (without \n).
 * @param ssl: the RES connection to read from. blocking.
 * @param buf: buffer to read to.
 * @param max: size of buffer.
 * @return false on connection failure.
 */
int ssl_read_line(RES* ssl, char* buf, size_t max);
#endif /* HAVE_SSL */

/**
 * Start fast reload thread
 * @param ssl: the RES connection to print to.
 * @param worker: the remote servicing worker.
 * @param s: the rc_state that is servicing the remote control connection to
 *	the remote control client. It needs to be moved away to stay connected
 *	while the fast reload is running.
 * @param fr_verb: verbosity to print output at. 0 is nothing, 1 is some
 *	and 2 is more detail.
 * @param fr_nopause: option to not pause threads during reload.
 * @param fr_drop_mesh: option to drop mesh queries.
 */
void fast_reload_thread_start(RES* ssl, struct worker* worker,
	struct rc_state* s, int fr_verb, int fr_nopause, int fr_drop_mesh);

/**
 * Stop fast reload thread
 * @param fast_reload_thread: the thread struct.
 */
void fast_reload_thread_stop(struct fast_reload_thread* fast_reload_thread);

/** fast reload printq delete list */
void fast_reload_printq_list_delete(struct fast_reload_printq* list);

/** Pick up per worker changes after a fast reload. */
void fast_reload_worker_pickup_changes(struct worker* worker);

#endif /* DAEMON_REMOTE_H */