op public repos

Blame

Date:: Thu Dec 23 16:44:12 2021 UTC
Message:: cosmetic s/ATTR_DEAD/__dead
Actions:: History | Blob | Raw File
  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 87b7e08f 2021-11-30 op #include "compat.h"
 18 87b7e08f 2021-11-30 op 
 19 8d1b399b 2021-07-22 op #include <sys/socket.h>
 20 8d1b399b 2021-07-22 op #include <sys/un.h>
 21 8d1b399b 2021-07-22 op 
 22 8d1b399b 2021-07-22 op #include <err.h>
 23 8d1b399b 2021-07-22 op #include <stdio.h>
 24 8d1b399b 2021-07-22 op #include <stdlib.h>
 25 8d1b399b 2021-07-22 op #include <string.h>
 26 8d1b399b 2021-07-22 op #include <syslog.h>
 27 8d1b399b 2021-07-22 op #include <unistd.h>
 28 8d1b399b 2021-07-22 op 
 29 8d1b399b 2021-07-22 op #include "ctl_parser.h"
 30 8d1b399b 2021-07-22 op #include "kamid.h"
 31 8d1b399b 2021-07-22 op #include "log.h"
 32 8d1b399b 2021-07-22 op 
 33 a97ec9eb 2021-12-23 op __dead void	 usage(void);
 34 8d1b399b 2021-07-22 op 
 35 8d1b399b 2021-07-22 op struct imsgbuf	*ibuf;
 36 8d1b399b 2021-07-22 op 
 37 a97ec9eb 2021-12-23 op __dead void
 38 8d1b399b 2021-07-22 op usage(void)
 39 8d1b399b 2021-07-22 op {
 40 8d1b399b 2021-07-22 op 	/*
 41 8d1b399b 2021-07-22 op 	 * XXX: this will print `kamid' if compat/getprogname.c is
 42 8d1b399b 2021-07-22 op 	 * used.
 43 8d1b399b 2021-07-22 op 	 */
 44 8d1b399b 2021-07-22 op 	fprintf(stderr, "usage: %s [-s socket] command [argument ...]\n",
 45 8d1b399b 2021-07-22 op 	    getprogname());
 46 8d1b399b 2021-07-22 op 	exit(1);
 47 8d1b399b 2021-07-22 op }
 48 8d1b399b 2021-07-22 op 
 49 8d1b399b 2021-07-22 op int
 50 8d1b399b 2021-07-22 op main(int argc, char **argv)
 51 8d1b399b 2021-07-22 op {
 52 8d1b399b 2021-07-22 op 	struct sockaddr_un	 sun;
 53 8d1b399b 2021-07-22 op 	struct parse_result	*res;
 54 8d1b399b 2021-07-22 op 	struct imsg		 imsg;
 55 8d1b399b 2021-07-22 op 	int			 ctl_sock;
 56 8d1b399b 2021-07-22 op 	int			 done = 0;
 57 8d1b399b 2021-07-22 op 	int			 n, verbose = 0;
 58 8d1b399b 2021-07-22 op 	int			 ch;
 59 8d1b399b 2021-07-22 op 	const char		*sockname;
 60 8d1b399b 2021-07-22 op 
 61 8d1b399b 2021-07-22 op 	log_init(1, LOG_DAEMON); /* Log to stderr. */
 62 8d1b399b 2021-07-22 op 
 63 8d1b399b 2021-07-22 op 	sockname = KD_SOCKET;
 64 8d1b399b 2021-07-22 op 	while ((ch = getopt(argc, argv, "s:")) != -1) {
 65 8d1b399b 2021-07-22 op 		switch (ch) {
 66 8d1b399b 2021-07-22 op 		case 's':
 67 8d1b399b 2021-07-22 op 			sockname = optarg;
 68 8d1b399b 2021-07-22 op 			break;
 69 8d1b399b 2021-07-22 op 		default:
 70 8d1b399b 2021-07-22 op 			usage();
 71 8d1b399b 2021-07-22 op 		}
 72 8d1b399b 2021-07-22 op 	}
 73 8d1b399b 2021-07-22 op 	argc -= optind;
 74 8d1b399b 2021-07-22 op 	argv += optind;
 75 8d1b399b 2021-07-22 op 
 76 8d1b399b 2021-07-22 op 	/* parse command line */
 77 8d1b399b 2021-07-22 op 	if ((res = parse(argc, argv)) == NULL)
 78 8d1b399b 2021-07-22 op 		exit(1);
 79 8d1b399b 2021-07-22 op 
 80 8d1b399b 2021-07-22 op 	/* connect to control socket */
 81 8d1b399b 2021-07-22 op 	if ((ctl_sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
 82 8d1b399b 2021-07-22 op 		err(1, "socket");
 83 8d1b399b 2021-07-22 op 
 84 8d1b399b 2021-07-22 op 	memset(&sun, 0, sizeof(sun));
 85 8d1b399b 2021-07-22 op 	sun.sun_family = AF_UNIX;
 86 8d1b399b 2021-07-22 op 	strlcpy(sun.sun_path, sockname, sizeof(sun.sun_path));
 87 8d1b399b 2021-07-22 op 
 88 8d1b399b 2021-07-22 op 	if (connect(ctl_sock, (struct sockaddr*)&sun, sizeof(sun)) == -1)
 89 8d1b399b 2021-07-22 op 		err(1, "connect: %s", sockname);
 90 8d1b399b 2021-07-22 op 
 91 8d1b399b 2021-07-22 op #ifdef __OpenBSD__
 92 8d1b399b 2021-07-22 op 	if (pledge("stdio", NULL) == -1)
 93 8d1b399b 2021-07-22 op 		err(1, "pledge");
 94 8d1b399b 2021-07-22 op #endif
 95 8d1b399b 2021-07-22 op 
 96 8d1b399b 2021-07-22 op 	if ((ibuf = calloc(1, sizeof(*ibuf))) == NULL)
 97 8d1b399b 2021-07-22 op 		err(1, NULL);
 98 8d1b399b 2021-07-22 op 	imsg_init(ibuf, ctl_sock);
 99 8d1b399b 2021-07-22 op 	done = 0;
100 8d1b399b 2021-07-22 op 
101 8d1b399b 2021-07-22 op 	/* process user request */
102 8d1b399b 2021-07-22 op 	switch (res->action) {
103 8d1b399b 2021-07-22 op 	case LOG_VERBOSE:
104 8d1b399b 2021-07-22 op 		verbose = 1;
105 8d1b399b 2021-07-22 op 		/* fallthrough */
106 8d1b399b 2021-07-22 op 	case LOG_BRIEF:
107 8d1b399b 2021-07-22 op 		imsg_compose(ibuf, IMSG_CTL_LOG_VERBOSE, 0, 0, -1,
108 8d1b399b 2021-07-22 op 		    &verbose, sizeof(verbose));
109 8d1b399b 2021-07-22 op 		puts("logging request sent.");
110 8d1b399b 2021-07-22 op 		done = 1;
111 8d1b399b 2021-07-22 op 		break;
112 8d1b399b 2021-07-22 op 	case RELOAD:
113 8d1b399b 2021-07-22 op 		imsg_compose(ibuf, IMSG_CTL_RELOAD, 0, 0, -1, NULL, 0);
114 8d1b399b 2021-07-22 op 		puts("reload request sent.");
115 8d1b399b 2021-07-22 op 		done = 1;
116 8d1b399b 2021-07-22 op 		break;
117 8d1b399b 2021-07-22 op 	default:
118 8d1b399b 2021-07-22 op 		usage();
119 8d1b399b 2021-07-22 op 	}
120 8d1b399b 2021-07-22 op 
121 8d1b399b 2021-07-22 op 	imsg_flush(ibuf);
122 8d1b399b 2021-07-22 op 
123 8d1b399b 2021-07-22 op 	/*
124 8d1b399b 2021-07-22 op 	 * Later we may add commands which requires a response.
125 8d1b399b 2021-07-22 op 	 */
126 8d1b399b 2021-07-22 op 	while (!done) {
127 8d1b399b 2021-07-22 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
128 8d1b399b 2021-07-22 op 			errx(1, "imsg_get error");
129 8d1b399b 2021-07-22 op 		if (n == 0)
130 8d1b399b 2021-07-22 op 			break;
131 8d1b399b 2021-07-22 op 
132 8d1b399b 2021-07-22 op 		switch (res->action) {
133 8d1b399b 2021-07-22 op 		default:
134 8d1b399b 2021-07-22 op 			break;
135 8d1b399b 2021-07-22 op 		}
136 8d1b399b 2021-07-22 op 		imsg_free(&imsg);
137 8d1b399b 2021-07-22 op 	}
138 8d1b399b 2021-07-22 op 	close(ctl_sock);
139 8d1b399b 2021-07-22 op 	free(ibuf);
140 8d1b399b 2021-07-22 op 
141 8d1b399b 2021-07-22 op 	return 0;
142 8d1b399b 2021-07-22 op }