2 8d1b399b 2021-07-22 op * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
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.
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.
17 8d1b399b 2021-07-22 op #include <sys/socket.h>
18 8d1b399b 2021-07-22 op #include <sys/un.h>
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>
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"
31 06a84967 2021-07-22 op ATTR_DEAD void usage(void);
33 8d1b399b 2021-07-22 op struct imsgbuf *ibuf;
35 06a84967 2021-07-22 op ATTR_DEAD void
39 8d1b399b 2021-07-22 op * XXX: this will print `kamid' if compat/getprogname.c is
42 8d1b399b 2021-07-22 op fprintf(stderr, "usage: %s [-s socket] command [argument ...]\n",
43 8d1b399b 2021-07-22 op getprogname());
48 8d1b399b 2021-07-22 op main(int argc, char **argv)
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;
55 8d1b399b 2021-07-22 op int n, verbose = 0;
57 8d1b399b 2021-07-22 op const char *sockname;
59 8d1b399b 2021-07-22 op log_init(1, LOG_DAEMON); /* Log to stderr. */
61 8d1b399b 2021-07-22 op sockname = KD_SOCKET;
62 8d1b399b 2021-07-22 op while ((ch = getopt(argc, argv, "s:")) != -1) {
65 8d1b399b 2021-07-22 op sockname = optarg;
71 8d1b399b 2021-07-22 op argc -= optind;
72 8d1b399b 2021-07-22 op argv += optind;
74 8d1b399b 2021-07-22 op /* parse command line */
75 8d1b399b 2021-07-22 op if ((res = parse(argc, argv)) == NULL)
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");
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));
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);
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");
94 8d1b399b 2021-07-22 op if ((ibuf = calloc(1, sizeof(*ibuf))) == NULL)
96 8d1b399b 2021-07-22 op imsg_init(ibuf, ctl_sock);
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:
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.");
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.");
119 8d1b399b 2021-07-22 op imsg_flush(ibuf);
122 8d1b399b 2021-07-22 op * Later we may add commands which requires a response.
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");
130 8d1b399b 2021-07-22 op switch (res->action) {
134 8d1b399b 2021-07-22 op imsg_free(&imsg);
136 8d1b399b 2021-07-22 op close(ctl_sock);