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 87b7e08f 2021-11-30 op #include "compat.h"
19 8d1b399b 2021-07-22 op #include <sys/socket.h>
20 8d1b399b 2021-07-22 op #include <sys/un.h>
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>
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"
33 06a84967 2021-07-22 op ATTR_DEAD void usage(void);
35 8d1b399b 2021-07-22 op struct imsgbuf *ibuf;
37 06a84967 2021-07-22 op ATTR_DEAD void
41 8d1b399b 2021-07-22 op * XXX: this will print `kamid' if compat/getprogname.c is
44 8d1b399b 2021-07-22 op fprintf(stderr, "usage: %s [-s socket] command [argument ...]\n",
45 8d1b399b 2021-07-22 op getprogname());
50 8d1b399b 2021-07-22 op main(int argc, char **argv)
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;
57 8d1b399b 2021-07-22 op int n, verbose = 0;
59 8d1b399b 2021-07-22 op const char *sockname;
61 8d1b399b 2021-07-22 op log_init(1, LOG_DAEMON); /* Log to stderr. */
63 8d1b399b 2021-07-22 op sockname = KD_SOCKET;
64 8d1b399b 2021-07-22 op while ((ch = getopt(argc, argv, "s:")) != -1) {
67 8d1b399b 2021-07-22 op sockname = optarg;
73 8d1b399b 2021-07-22 op argc -= optind;
74 8d1b399b 2021-07-22 op argv += optind;
76 8d1b399b 2021-07-22 op /* parse command line */
77 8d1b399b 2021-07-22 op if ((res = parse(argc, argv)) == NULL)
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");
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));
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);
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");
96 8d1b399b 2021-07-22 op if ((ibuf = calloc(1, sizeof(*ibuf))) == NULL)
98 8d1b399b 2021-07-22 op imsg_init(ibuf, ctl_sock);
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:
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.");
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.");
121 8d1b399b 2021-07-22 op imsg_flush(ibuf);
124 8d1b399b 2021-07-22 op * Later we may add commands which requires a response.
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");
132 8d1b399b 2021-07-22 op switch (res->action) {
136 8d1b399b 2021-07-22 op imsg_free(&imsg);
138 8d1b399b 2021-07-22 op close(ctl_sock);