path: root/usr.sbin/unwindctl/unwindctl.c

/*	$OpenBSD: unwindctl.c,v 1.24 2019/12/08 09:47:51 florian Exp $	*/

/*
 * Copyright (c) 2005 Claudio Jeker <claudio@openbsd.org>
 * Copyright (c) 2004, 2005 Esben Norby <norby@openbsd.org>
 * Copyright (c) 2003 Henning Brauer <henning@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>
#include <sys/queue.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <net/if_media.h>
#include <net/if_types.h>
#include <net/route.h>

#include <err.h>
#include <errno.h>
#include <event.h>
#include <imsg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "unwind.h"
#include "frontend.h"
#include "resolver.h"
#include "parser.h"

__dead void	 usage(void);
int		 show_status_msg(struct imsg *);
int		 show_autoconf_msg(struct imsg *);
void		 histogram_header(void);
void		 print_histogram(const char *name, int64_t[], size_t);

struct imsgbuf		*ibuf;
int		 	 histogram_cnt;
struct ctl_resolver_info info[UW_RES_NONE];

__dead void
usage(void)
{
	extern char *__progname;

	fprintf(stderr, "usage: %s [-s socket] command [argument ...]\n",
	    __progname);
	exit(1);
}

int
main(int argc, char *argv[])
{
	struct sockaddr_un	 sun;
	struct parse_result	*res;
	struct imsg		 imsg;
	int			 ctl_sock;
	int			 done = 0;
	int			 i, n, verbose = 0;
	int			 ch;
	char			*sockname, *oDoT;

	sockname = UNWIND_SOCKET;
	while ((ch = getopt(argc, argv, "s:")) != -1) {
		switch (ch) {
		case 's':
			sockname = optarg;
			break;
		default:
			usage();
		}
	}
	argc -= optind;
	argv += optind;

	/* Parse command line. */
	if ((res = parse(argc, argv)) == NULL)
		exit(1);

	/* Connect to control socket. */
	if ((ctl_sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
		err(1, "socket");

	memset(&sun, 0, sizeof(sun));
	sun.sun_family = AF_UNIX;
	strlcpy(sun.sun_path, sockname, sizeof(sun.sun_path));

	if (connect(ctl_sock, (struct sockaddr *)&sun, sizeof(sun)) == -1)
		err(1, "connect: %s", sockname);

	if (pledge("stdio", NULL) == -1)
		err(1, "pledge");

	if ((ibuf = malloc(sizeof(struct imsgbuf))) == NULL)
		err(1, NULL);
	imsg_init(ibuf, ctl_sock);
	done = 0;

	/* Check for root-only actions */
	switch (res->action) {
	case LOG_DEBUG:
	case LOG_VERBOSE:
	case LOG_BRIEF:
	case RELOAD:
		if (geteuid() != 0)
			errx(1, "need root privileges");
		break;
	default:
		break;
	}

	/* Process user request. */
	switch (res->action) {
	case LOG_DEBUG:
		verbose |= OPT_VERBOSE2;
		/* FALLTHROUGH */
	case LOG_VERBOSE:
		verbose |= OPT_VERBOSE;
		/* FALLTHROUGH */
	case LOG_BRIEF:
		imsg_compose(ibuf, IMSG_CTL_LOG_VERBOSE, 0, 0, -1,
		    &verbose, sizeof(verbose));
		printf("logging request sent.\n");
		done = 1;
		break;
	case RELOAD:
		imsg_compose(ibuf, IMSG_CTL_RELOAD, 0, 0, -1, NULL, 0);
		printf("reload request sent.\n");
		done = 1;
		break;
	case STATUS:
		imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1, NULL, 0);
		break;
	case AUTOCONF:
		imsg_compose(ibuf, IMSG_CTL_AUTOCONF, 0, 0, -1, NULL, 0);
		break;
	default:
		usage();
	}

	while (ibuf->w.queued)
		if (msgbuf_write(&ibuf->w) <= 0 && errno != EAGAIN)
			err(1, "write error");

	while (!done) {
		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
			errx(1, "imsg_read error");
		if (n == 0)
			errx(1, "pipe closed");

		while (!done) {
			if ((n = imsg_get(ibuf, &imsg)) == -1)
				errx(1, "imsg_get error");
			if (n == 0)
				break;

			switch (res->action) {
			case STATUS:
				done = show_status_msg(&imsg);
				break;
			case AUTOCONF:
				done = show_autoconf_msg(&imsg);
				break;
			default:
				break;
			}
			imsg_free(&imsg);
		}
	}
	close(ctl_sock);
	free(ibuf);

	if (histogram_cnt)
		histogram_header();
	for (i = 0; i < histogram_cnt; i++) {
		switch(info[i].type) {
		case UW_RES_ODOT_FORWARDER:
		case UW_RES_ODOT_DHCP:
			if (info[i].state == DEAD)
				continue;
			oDoT = "(oT)";
			break;
		default:
			oDoT = "";
			break;
		}
		print_histogram(uw_resolver_type_short[info[i].type],
		    info[i].histogram, nitems(info[i].histogram));
		print_histogram(oDoT, info[i].latest_histogram,
		    nitems(info[i].latest_histogram));
	}
	return (0);
}

int
show_status_msg(struct imsg *imsg)
{
	static char			 fwd_line[80];
	struct ctl_resolver_info	*cri;

	switch (imsg->hdr.type) {
	case IMSG_CTL_RESOLVER_INFO:
		cri = imsg->data;
		memcpy(&info[histogram_cnt++], cri, sizeof(info[0]));
		switch(cri->type) {
		case UW_RES_ODOT_FORWARDER:
		case UW_RES_ODOT_DHCP:
			if (cri->state == DEAD)
				return (0);
		default:
			break;
		}
		printf("%-15s %s, median RTT: ",
		    uw_resolver_type_str[cri->type],
		    uw_resolver_state_str[cri->state]);
		if (cri->median == 0)
			printf("N/A\n");
		else if (cri->median == INT64_MAX)
			printf("Inf\n");
		else
			printf("%lldms\n", cri->median);
		break;
	case IMSG_CTL_END:
		if (fwd_line[0] != '\0')
			printf("%s\n", fwd_line);
		return (1);
	default:
		break;
	}

	return (0);
}

int
show_autoconf_msg(struct imsg *imsg)
{
	static int			 autoconf_forwarders, last_src;
	static int			 label_len, line_len;
	static uint32_t			 last_if_index;
	static char			 fwd_line[80];
	struct ctl_forwarder_info	*cfi;
	char				 ifnamebuf[IFNAMSIZ];
	char				*if_name;

	switch (imsg->hdr.type) {
	case IMSG_CTL_AUTOCONF_RESOLVER_INFO:
		cfi = imsg->data;
		if (!autoconf_forwarders++)
			printf("autoconfiguration forwarders:\n");
		if (cfi->if_index != last_if_index || cfi->src != last_src) {
			if_name = if_indextoname(cfi->if_index, ifnamebuf);
			if (fwd_line[0] != '\0') {
				printf("%s\n", fwd_line);
				fwd_line[0] = '\0';
			}
			label_len = snprintf(fwd_line, sizeof(fwd_line),
			    "%s[%s]:", cfi->src == RTP_PROPOSAL_DHCLIENT ?
			    " DHCP" : "SLAAC", if_name ? if_name : "unknown");
			line_len = label_len;
			last_if_index = cfi->if_index;
			last_src = cfi->src;
		}

		if (line_len + 1 + strlen(cfi->ip) > sizeof(fwd_line)) {
			printf("%s\n", fwd_line);
			snprintf(fwd_line, sizeof(fwd_line), "%*s", label_len,
			    " ");
		}
		strlcat(fwd_line, " ", sizeof(fwd_line));
		line_len = strlcat(fwd_line, cfi->ip, sizeof(fwd_line));
		break;
	case IMSG_CTL_END:
		if (fwd_line[0] != '\0')
			printf("%s\n", fwd_line);
		return (1);
	default:
		break;
	}

	return (0);
}

void
histogram_header(void)
{
	const char	 head[] = "histograms: lifetime[ms], decaying[ms]";
	char	 	 buf[10];
	size_t	 	 i;

	printf("\n%*s%*s\n%*s", 5, "",
	    (int)(72/2 + (sizeof(head)-1)/2), head, 5, "");
	for(i = 0; i < nitems(histogram_limits) - 1; i++) {
		snprintf(buf, sizeof(buf), "<%lld", histogram_limits[i]);
		printf("%6s", buf);
	}
	printf("%6s\n", ">");
}

void
print_histogram(const char *name, int64_t histogram[], size_t n)
{
	size_t	 i;

	printf("%4s ", name);
	for(i = 0; i < n; i++)
		printf("%6lld", histogram[i]);
	printf("\n");
}