1 8d1b399b 2021-07-22 op /*
  2 8d1b399b 2021-07-22 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 8d1b399b 2021-07-22 op  *
  4 8d1b399b 2021-07-22 op  * Permission to use, copy, modify, and distribute this software for any
  5 8d1b399b 2021-07-22 op  * purpose with or without fee is hereby granted, provided that the above
  6 8d1b399b 2021-07-22 op  * copyright notice and this permission notice appear in all copies.
  7 8d1b399b 2021-07-22 op  *
  8 8d1b399b 2021-07-22 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 8d1b399b 2021-07-22 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 8d1b399b 2021-07-22 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 8d1b399b 2021-07-22 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 8d1b399b 2021-07-22 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 8d1b399b 2021-07-22 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 8d1b399b 2021-07-22 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 8d1b399b 2021-07-22 op  */
 16 8d1b399b 2021-07-22 op 
 17 8d1b399b 2021-07-22 op #include <sys/socket.h>
 18 8d1b399b 2021-07-22 op #include <sys/un.h>
 19 8d1b399b 2021-07-22 op 
 20 8d1b399b 2021-07-22 op #include <err.h>
 21 8d1b399b 2021-07-22 op #include <stdio.h>
 22 8d1b399b 2021-07-22 op #include <stdlib.h>
 23 8d1b399b 2021-07-22 op #include <string.h>
 24 8d1b399b 2021-07-22 op #include <syslog.h>
 25 8d1b399b 2021-07-22 op #include <unistd.h>
 26 8d1b399b 2021-07-22 op 
 27 8d1b399b 2021-07-22 op #include "ctl_parser.h"
 28 8d1b399b 2021-07-22 op #include "kamid.h"
 29 8d1b399b 2021-07-22 op #include "log.h"
 30 8d1b399b 2021-07-22 op 
 31 8d1b399b 2021-07-22 op __dead void	 usage(void);
 32 8d1b399b 2021-07-22 op 
 33 8d1b399b 2021-07-22 op struct imsgbuf	*ibuf;
 34 8d1b399b 2021-07-22 op 
 35 8d1b399b 2021-07-22 op __dead void
 36 8d1b399b 2021-07-22 op usage(void)
 37 8d1b399b 2021-07-22 op {
 38 8d1b399b 2021-07-22 op 	/*
 39 8d1b399b 2021-07-22 op 	 * XXX: this will print `kamid' if compat/getprogname.c is
 40 8d1b399b 2021-07-22 op 	 * used.
 41 8d1b399b 2021-07-22 op 	 */
 42 8d1b399b 2021-07-22 op 	fprintf(stderr, "usage: %s [-s socket] command [argument ...]\n",
 43 8d1b399b 2021-07-22 op 	    getprogname());
 44 8d1b399b 2021-07-22 op 	exit(1);
 45 8d1b399b 2021-07-22 op }
 46 8d1b399b 2021-07-22 op 
 47 8d1b399b 2021-07-22 op int
 48 8d1b399b 2021-07-22 op main(int argc, char **argv)
 49 8d1b399b 2021-07-22 op {
 50 8d1b399b 2021-07-22 op 	struct sockaddr_un	 sun;
 51 8d1b399b 2021-07-22 op 	struct parse_result	*res;
 52 8d1b399b 2021-07-22 op 	struct imsg		 imsg;
 53 8d1b399b 2021-07-22 op 	int			 ctl_sock;
 54 8d1b399b 2021-07-22 op 	int			 done = 0;
 55 8d1b399b 2021-07-22 op 	int			 n, verbose = 0;
 56 8d1b399b 2021-07-22 op 	int			 ch;
 57 8d1b399b 2021-07-22 op 	const char		*sockname;
 58 8d1b399b 2021-07-22 op 
 59 8d1b399b 2021-07-22 op 	log_init(1, LOG_DAEMON); /* Log to stderr. */
 60 8d1b399b 2021-07-22 op 
 61 8d1b399b 2021-07-22 op 	sockname = KD_SOCKET;
 62 8d1b399b 2021-07-22 op 	while ((ch = getopt(argc, argv, "s:")) != -1) {
 63 8d1b399b 2021-07-22 op 		switch (ch) {
 64 8d1b399b 2021-07-22 op 		case 's':
 65 8d1b399b 2021-07-22 op 			sockname = optarg;
 66 8d1b399b 2021-07-22 op 			break;
 67 8d1b399b 2021-07-22 op 		default:
 68 8d1b399b 2021-07-22 op 			usage();
 69 8d1b399b 2021-07-22 op 		}
 70 8d1b399b 2021-07-22 op 	}
 71 8d1b399b 2021-07-22 op 	argc -= optind;
 72 8d1b399b 2021-07-22 op 	argv += optind;
 73 8d1b399b 2021-07-22 op 
 74 8d1b399b 2021-07-22 op 	/* parse command line */
 75 8d1b399b 2021-07-22 op 	if ((res = parse(argc, argv)) == NULL)
 76 8d1b399b 2021-07-22 op 		exit(1);
 77 8d1b399b 2021-07-22 op 
 78 8d1b399b 2021-07-22 op 	/* connect to control socket */
 79 8d1b399b 2021-07-22 op 	if ((ctl_sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
 80 8d1b399b 2021-07-22 op 		err(1, "socket");
 81 8d1b399b 2021-07-22 op 
 82 8d1b399b 2021-07-22 op 	memset(&sun, 0, sizeof(sun));
 83 8d1b399b 2021-07-22 op 	sun.sun_family = AF_UNIX;
 84 8d1b399b 2021-07-22 op 	strlcpy(sun.sun_path, sockname, sizeof(sun.sun_path));
 85 8d1b399b 2021-07-22 op 
 86 8d1b399b 2021-07-22 op 	if (connect(ctl_sock, (struct sockaddr*)&sun, sizeof(sun)) == -1)
 87 8d1b399b 2021-07-22 op 		err(1, "connect: %s", sockname);
 88 8d1b399b 2021-07-22 op 
 89 8d1b399b 2021-07-22 op #ifdef __OpenBSD__
 90 8d1b399b 2021-07-22 op 	if (pledge("stdio", NULL) == -1)
 91 8d1b399b 2021-07-22 op 		err(1, "pledge");
 92 8d1b399b 2021-07-22 op #endif
 93 8d1b399b 2021-07-22 op 
 94 8d1b399b 2021-07-22 op 	if ((ibuf = calloc(1, sizeof(*ibuf))) == NULL)
 95 8d1b399b 2021-07-22 op 		err(1, NULL);
 96 8d1b399b 2021-07-22 op 	imsg_init(ibuf, ctl_sock);
 97 8d1b399b 2021-07-22 op 	done = 0;
 98 8d1b399b 2021-07-22 op 
 99 8d1b399b 2021-07-22 op 	/* process user request */
100 8d1b399b 2021-07-22 op 	switch (res->action) {
101 8d1b399b 2021-07-22 op 	case LOG_VERBOSE:
102 8d1b399b 2021-07-22 op 		verbose = 1;
103 8d1b399b 2021-07-22 op 		/* fallthrough */
104 8d1b399b 2021-07-22 op 	case LOG_BRIEF:
105 8d1b399b 2021-07-22 op 		imsg_compose(ibuf, IMSG_CTL_LOG_VERBOSE, 0, 0, -1,
106 8d1b399b 2021-07-22 op 		    &verbose, sizeof(verbose));
107 8d1b399b 2021-07-22 op 		puts("logging request sent.");
108 8d1b399b 2021-07-22 op 		done = 1;
109 8d1b399b 2021-07-22 op 		break;
110 8d1b399b 2021-07-22 op 	case RELOAD:
111 8d1b399b 2021-07-22 op 		imsg_compose(ibuf, IMSG_CTL_RELOAD, 0, 0, -1, NULL, 0);
112 8d1b399b 2021-07-22 op 		puts("reload request sent.");
113 8d1b399b 2021-07-22 op 		done = 1;
114 8d1b399b 2021-07-22 op 		break;
115 8d1b399b 2021-07-22 op 	default:
116 8d1b399b 2021-07-22 op 		usage();
117 8d1b399b 2021-07-22 op 	}
118 8d1b399b 2021-07-22 op 
119 8d1b399b 2021-07-22 op 	imsg_flush(ibuf);
120 8d1b399b 2021-07-22 op 
121 8d1b399b 2021-07-22 op 	/*
122 8d1b399b 2021-07-22 op 	 * Later we may add commands which requires a response.
123 8d1b399b 2021-07-22 op 	 */
124 8d1b399b 2021-07-22 op 	while (!done) {
125 8d1b399b 2021-07-22 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
126 8d1b399b 2021-07-22 op 			errx(1, "imsg_get error");
127 8d1b399b 2021-07-22 op 		if (n == 0)
128 8d1b399b 2021-07-22 op 			break;
129 8d1b399b 2021-07-22 op 
130 8d1b399b 2021-07-22 op 		switch (res->action) {
131 8d1b399b 2021-07-22 op 		default:
132 8d1b399b 2021-07-22 op 			break;
133 8d1b399b 2021-07-22 op 		}
134 8d1b399b 2021-07-22 op 		imsg_free(&imsg);
135 8d1b399b 2021-07-22 op 	}
136 8d1b399b 2021-07-22 op 	close(ctl_sock);
137 8d1b399b 2021-07-22 op 	free(ibuf);
138 8d1b399b 2021-07-22 op 
139 8d1b399b 2021-07-22 op 	return 0;
140 8d1b399b 2021-07-22 op }