1 3baa2617 2022-02-16 op /*
  2 3baa2617 2022-02-16 op  * Copyright (c) 2022 Omar Polo <op@openbsd.org>
  3 96233476 2022-06-09 op  * Copyright (c) 2015 Theo de Raadt <deraadt@openbsd.org>
  4 3baa2617 2022-02-16 op  *
  5 3baa2617 2022-02-16 op  * Permission to use, copy, modify, and distribute this software for any
  6 3baa2617 2022-02-16 op  * purpose with or without fee is hereby granted, provided that the above
  7 3baa2617 2022-02-16 op  * copyright notice and this permission notice appear in all copies.
  8 3baa2617 2022-02-16 op  *
  9 3baa2617 2022-02-16 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 10 3baa2617 2022-02-16 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 11 3baa2617 2022-02-16 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 12 3baa2617 2022-02-16 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 13 3baa2617 2022-02-16 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 14 3baa2617 2022-02-16 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 15 3baa2617 2022-02-16 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 16 3baa2617 2022-02-16 op  */
 17 3baa2617 2022-02-16 op 
 18 3baa2617 2022-02-16 op #include <sys/types.h>
 19 3baa2617 2022-02-16 op #include <sys/socket.h>
 20 3baa2617 2022-02-16 op #include <sys/queue.h>
 21 3baa2617 2022-02-16 op #include <sys/uio.h>
 22 3baa2617 2022-02-16 op #include <sys/un.h>
 23 3baa2617 2022-02-16 op 
 24 3baa2617 2022-02-16 op #include <errno.h>
 25 3baa2617 2022-02-16 op #include <event.h>
 26 1d673950 2022-03-03 op #include <fcntl.h>
 27 3baa2617 2022-02-16 op #include <limits.h>
 28 3baa2617 2022-02-16 op #include <stdio.h>
 29 3baa2617 2022-02-16 op #include <stdlib.h>
 30 3baa2617 2022-02-16 op #include <stdint.h>
 31 3baa2617 2022-02-16 op #include <string.h>
 32 3baa2617 2022-02-16 op #include <syslog.h>
 33 3baa2617 2022-02-16 op #include <unistd.h>
 34 3baa2617 2022-02-16 op #include <imsg.h>
 35 3baa2617 2022-02-16 op 
 36 3baa2617 2022-02-16 op #include "amused.h"
 37 3baa2617 2022-02-16 op #include "log.h"
 38 bb3f279f 2022-02-16 op #include "playlist.h"
 39 3baa2617 2022-02-16 op #include "xmalloc.h"
 40 3baa2617 2022-02-16 op 
 41 96233476 2022-06-09 op static struct imsgbuf	*ibuf;
 42 96233476 2022-06-09 op char			 cwd[PATH_MAX];
 43 3baa2617 2022-02-16 op 
 44 3baa2617 2022-02-16 op int	ctl_noarg(struct parse_result *, int, char **);
 45 3baa2617 2022-02-16 op int	ctl_add(struct parse_result *, int, char **);
 46 8e916714 2022-02-17 op int	ctl_show(struct parse_result *, int, char **);
 47 14be015f 2022-02-17 op int	ctl_load(struct parse_result *, int, char **);
 48 a913de21 2022-02-17 op int	ctl_jump(struct parse_result *, int, char **);
 49 310ef57c 2022-02-19 op int	ctl_repeat(struct parse_result *, int, char **);
 50 90122a37 2022-05-14 op int	ctl_monitor(struct parse_result *, int, char **);
 51 3baa2617 2022-02-16 op 
 52 3baa2617 2022-02-16 op struct ctl_command ctl_commands[] = {
 53 3baa2617 2022-02-16 op 	{ "play",	PLAY,		ctl_noarg,	"" },
 54 3baa2617 2022-02-16 op 	{ "pause",	PAUSE,		ctl_noarg,	"" },
 55 3baa2617 2022-02-16 op 	{ "toggle",	TOGGLE,		ctl_noarg,	"" },
 56 3baa2617 2022-02-16 op 	{ "stop",	STOP,		ctl_noarg,	"" },
 57 3baa2617 2022-02-16 op 	{ "restart",	RESTART,	ctl_noarg,	"" },
 58 a830518b 2022-06-09 op 	{ "add",	ADD,		ctl_add,	"files..." },
 59 3baa2617 2022-02-16 op 	{ "flush",	FLUSH,		ctl_noarg,	"" },
 60 714abf92 2022-02-19 op 	{ "show",	SHOW,		ctl_show,	"[-p]" },
 61 bb3f279f 2022-02-16 op 	{ "status",	STATUS,		ctl_noarg,	"" },
 62 af27e631 2022-02-17 op 	{ "next",	NEXT,		ctl_noarg,	"" },
 63 af27e631 2022-02-17 op 	{ "prev",	PREV,		ctl_noarg,	"" },
 64 14be015f 2022-02-17 op 	{ "load",	LOAD,		ctl_load,	"[file]", 1 },
 65 a913de21 2022-02-17 op 	{ "jump",	JUMP,		ctl_jump,	"pattern" },
 66 310ef57c 2022-02-19 op 	{ "repeat",	REPEAT,		ctl_repeat,	"one|all on|off" },
 67 90122a37 2022-05-14 op 	{ "monitor",	MONITOR,	ctl_monitor,	"[events]" },
 68 3baa2617 2022-02-16 op 	{ NULL },
 69 3baa2617 2022-02-16 op };
 70 3baa2617 2022-02-16 op 
 71 3baa2617 2022-02-16 op __dead void
 72 3baa2617 2022-02-16 op usage(void)
 73 3baa2617 2022-02-16 op {
 74 3baa2617 2022-02-16 op 	fprintf(stderr, "usage: %s [-dv] [-s socket]\n", getprogname());
 75 3baa2617 2022-02-16 op 	exit(1);
 76 3baa2617 2022-02-16 op }
 77 3baa2617 2022-02-16 op 
 78 3baa2617 2022-02-16 op static __dead void
 79 3baa2617 2022-02-16 op ctl_usage(struct ctl_command *ctl)
 80 3baa2617 2022-02-16 op {
 81 3baa2617 2022-02-16 op 	fprintf(stderr, "usage: %s [-v] [-s socket] %s %s\n", getprogname(),
 82 3baa2617 2022-02-16 op 	    ctl->name, ctl->usage);
 83 3baa2617 2022-02-16 op 	exit(1);
 84 3baa2617 2022-02-16 op }
 85 3baa2617 2022-02-16 op 
 86 96233476 2022-06-09 op /* based on canonpath from kern_pledge.c */
 87 3baa2617 2022-02-16 op static int
 88 96233476 2022-06-09 op canonpath(const char *input, char *buf, size_t bufsize)
 89 96233476 2022-06-09 op {
 90 96233476 2022-06-09 op 	const char *p;
 91 96233476 2022-06-09 op 	char *q, path[PATH_MAX];
 92 96233476 2022-06-09 op 
 93 96233476 2022-06-09 op 	if (input[0] != '/') {
 94 96233476 2022-06-09 op 		if (snprintf(path, sizeof(path), "%s/%s", cwd, input)
 95 96233476 2022-06-09 op 		    >= sizeof(path)) {
 96 96233476 2022-06-09 op 			errno = ENAMETOOLONG;
 97 96233476 2022-06-09 op 			return -1;
 98 96233476 2022-06-09 op 		}
 99 96233476 2022-06-09 op 		input = path;
100 96233476 2022-06-09 op 	}
101 96233476 2022-06-09 op 
102 96233476 2022-06-09 op 	p = input;
103 96233476 2022-06-09 op 	q = buf;
104 96233476 2022-06-09 op 	while (*p && (q - buf < bufsize)) {
105 96233476 2022-06-09 op 		if (p[0] == '/' && (p[1] == '/' || p[1] == '\0')) {
106 96233476 2022-06-09 op 			p += 1;
107 96233476 2022-06-09 op 
108 96233476 2022-06-09 op 		} else if (p[0] == '/' && p[1] == '.' &&
109 96233476 2022-06-09 op 		    (p[2] == '/' || p[2] == '\0')) {
110 96233476 2022-06-09 op 			p += 2;
111 96233476 2022-06-09 op 
112 96233476 2022-06-09 op 		} else if (p[0] == '/' && p[1] == '.' && p[2] == '.' &&
113 96233476 2022-06-09 op 		    (p[3] == '/' || p[3] == '\0')) {
114 96233476 2022-06-09 op 			p += 3;
115 96233476 2022-06-09 op 			if (q != buf)	/* "/../" at start of buf */
116 96233476 2022-06-09 op 				while (*--q != '/')
117 96233476 2022-06-09 op 					continue;
118 96233476 2022-06-09 op 
119 96233476 2022-06-09 op 		} else {
120 96233476 2022-06-09 op 			*q++ = *p++;
121 96233476 2022-06-09 op 		}
122 96233476 2022-06-09 op 	}
123 96233476 2022-06-09 op 	if ((*p == '\0') && (q - buf < bufsize)) {
124 96233476 2022-06-09 op 		*q = 0;
125 96233476 2022-06-09 op 		return 0;
126 96233476 2022-06-09 op 	} else {
127 96233476 2022-06-09 op 		errno = ENAMETOOLONG;
128 96233476 2022-06-09 op 		return -1;
129 96233476 2022-06-09 op 	}
130 96233476 2022-06-09 op }
131 96233476 2022-06-09 op 
132 96233476 2022-06-09 op static int
133 3baa2617 2022-02-16 op parse(int argc, char **argv)
134 3baa2617 2022-02-16 op {
135 3baa2617 2022-02-16 op 	struct ctl_command	*ctl = NULL;
136 3baa2617 2022-02-16 op 	struct parse_result	 res;
137 1d673950 2022-03-03 op 	const char		*argv0;
138 3baa2617 2022-02-16 op 	int			 i, status;
139 3baa2617 2022-02-16 op 
140 3baa2617 2022-02-16 op 	memset(&res, 0, sizeof(res));
141 3baa2617 2022-02-16 op 
142 1d673950 2022-03-03 op 	if ((argv0 = argv[0]) == NULL)
143 1d673950 2022-03-03 op 		argv0 = "status";
144 1d673950 2022-03-03 op 
145 3baa2617 2022-02-16 op 	for (i = 0; ctl_commands[i].name != NULL; ++i) {
146 1d673950 2022-03-03 op 		if (strncmp(ctl_commands[i].name, argv0, strlen(argv0))
147 3baa2617 2022-02-16 op 		    == 0) {
148 3baa2617 2022-02-16 op 			if (ctl != NULL) {
149 3baa2617 2022-02-16 op 				fprintf(stderr,
150 1d673950 2022-03-03 op 				    "ambiguous argument: %s\n", argv0);
151 3baa2617 2022-02-16 op 				usage();
152 3baa2617 2022-02-16 op 			}
153 3baa2617 2022-02-16 op 			ctl = &ctl_commands[i];
154 3baa2617 2022-02-16 op 		}
155 3baa2617 2022-02-16 op 	}
156 3baa2617 2022-02-16 op 
157 3baa2617 2022-02-16 op 	if (ctl == NULL) {
158 3baa2617 2022-02-16 op 		fprintf(stderr, "unknown argument: %s\n", argv[0]);
159 3baa2617 2022-02-16 op 		usage();
160 3baa2617 2022-02-16 op 	}
161 3baa2617 2022-02-16 op 
162 3baa2617 2022-02-16 op 	res.action = ctl->action;
163 3baa2617 2022-02-16 op 	res.ctl = ctl;
164 3baa2617 2022-02-16 op 
165 3baa2617 2022-02-16 op 	if (!ctl->has_pledge) {
166 3baa2617 2022-02-16 op 		/* pledge(2) default if command doesn't have its own */
167 3baa2617 2022-02-16 op 		if (pledge("stdio", NULL) == -1)
168 3baa2617 2022-02-16 op 			fatal("pledge");
169 3baa2617 2022-02-16 op 	}
170 3baa2617 2022-02-16 op 
171 3baa2617 2022-02-16 op 	status = ctl->main(&res, argc, argv);
172 3baa2617 2022-02-16 op 	close(ibuf->fd);
173 3baa2617 2022-02-16 op 	free(ibuf);
174 3baa2617 2022-02-16 op 	return status;
175 3baa2617 2022-02-16 op }
176 3baa2617 2022-02-16 op 
177 3baa2617 2022-02-16 op static int
178 fad0fb69 2022-05-28 op load_files(struct parse_result *res, int *ret)
179 7a427ecd 2022-02-17 op {
180 ec1fb0c7 2022-02-17 op 	FILE		*f;
181 7a427ecd 2022-02-17 op 	const char	*file;
182 7a427ecd 2022-02-17 op 	char		*line = NULL;
183 96233476 2022-06-09 op 	char		 path[PATH_MAX];
184 96233476 2022-06-09 op 	size_t		 linesize = 0, i = 0;
185 3af93963 2022-03-02 op 	ssize_t		 linelen, curr = -1;
186 7a427ecd 2022-02-17 op 
187 ec1fb0c7 2022-02-17 op 	if (res->file == NULL)
188 ec1fb0c7 2022-02-17 op 		f = stdin;
189 ec1fb0c7 2022-02-17 op 	else if ((f = fopen(res->file, "r")) == NULL) {
190 ec1fb0c7 2022-02-17 op 		log_warn("can't open %s", res->file);
191 ec1fb0c7 2022-02-17 op 		*ret = 1;
192 ec1fb0c7 2022-02-17 op 		return 1;
193 ec1fb0c7 2022-02-17 op 	}
194 c00c1428 2022-03-25 op 
195 ec1fb0c7 2022-02-17 op 	while ((linelen = getline(&line, &linesize, f)) != -1) {
196 7a427ecd 2022-02-17 op 		if (linelen == 0)
197 7a427ecd 2022-02-17 op 			continue;
198 7a427ecd 2022-02-17 op 		line[linelen-1] = '\0';
199 7a427ecd 2022-02-17 op 		file = line;
200 1dd3b054 2022-03-25 op 		if (!strncmp(file, "> ", 2)) {
201 7a427ecd 2022-02-17 op 			file += 2;
202 3af93963 2022-03-02 op 			curr = i;
203 1dd3b054 2022-03-25 op 		} else if (!strncmp(file, "  ", 2))
204 7a427ecd 2022-02-17 op 			file += 2;
205 1dd3b054 2022-03-25 op 
206 18223c28 2022-06-13 op 		memset(path, 0, sizeof(path));
207 96233476 2022-06-09 op 		if (canonpath(file, path, sizeof(path)) == -1) {
208 f1beb69a 2022-06-13 op 			log_warn("canonpath %s", file);
209 7a427ecd 2022-02-17 op 			continue;
210 7a427ecd 2022-02-17 op 		}
211 7a427ecd 2022-02-17 op 
212 3af93963 2022-03-02 op 		i++;
213 7a427ecd 2022-02-17 op 		imsg_compose(ibuf, IMSG_CTL_ADD, 0, 0, -1,
214 7a427ecd 2022-02-17 op 		    path, sizeof(path));
215 7a427ecd 2022-02-17 op 	}
216 7a427ecd 2022-02-17 op 
217 7a427ecd 2022-02-17 op 	free(line);
218 ec1fb0c7 2022-02-17 op 	if (ferror(f))
219 7a427ecd 2022-02-17 op 		fatal("getline");
220 ec1fb0c7 2022-02-17 op 	fclose(f);
221 7a427ecd 2022-02-17 op 
222 3af93963 2022-03-02 op 	if (i == 0) {
223 7a427ecd 2022-02-17 op 		*ret = 1;
224 7a427ecd 2022-02-17 op 		return 1;
225 7a427ecd 2022-02-17 op 	}
226 7a427ecd 2022-02-17 op 
227 7a427ecd 2022-02-17 op 	imsg_compose(ibuf, IMSG_CTL_COMMIT, 0, 0, -1,
228 3af93963 2022-03-02 op 	    &curr, sizeof(curr));
229 7a427ecd 2022-02-17 op 	imsg_flush(ibuf);
230 7a427ecd 2022-02-17 op 	return 0;
231 3baa2617 2022-02-16 op }
232 3baa2617 2022-02-16 op 
233 fad0fb69 2022-05-28 op static const char *
234 fad0fb69 2022-05-28 op imsg_strerror(struct imsg *imsg)
235 6b47a39f 2022-02-21 op {
236 fad0fb69 2022-05-28 op 	size_t datalen;
237 fad0fb69 2022-05-28 op 	const char *msg;
238 6b47a39f 2022-02-21 op 
239 fad0fb69 2022-05-28 op 	datalen = IMSG_DATA_SIZE(*imsg);
240 fad0fb69 2022-05-28 op 	msg = imsg->data;
241 fad0fb69 2022-05-28 op 	if (datalen == 0 || msg[datalen-1] != '\0')
242 fad0fb69 2022-05-28 op 		fatalx("malformed error message");
243 6b47a39f 2022-02-21 op 
244 fad0fb69 2022-05-28 op 	return msg;
245 fad0fb69 2022-05-28 op }
246 90122a37 2022-05-14 op 
247 fad0fb69 2022-05-28 op static const char *
248 fad0fb69 2022-05-28 op imsg_name(int type)
249 fad0fb69 2022-05-28 op {
250 6b47a39f 2022-02-21 op 	switch (type) {
251 6b47a39f 2022-02-21 op 	case IMSG_CTL_PLAY:
252 fad0fb69 2022-05-28 op 		return "play";
253 6b47a39f 2022-02-21 op 	case IMSG_CTL_TOGGLE_PLAY:
254 fad0fb69 2022-05-28 op 		return "toggle";
255 6b47a39f 2022-02-21 op 	case IMSG_CTL_PAUSE:
256 fad0fb69 2022-05-28 op 		return "pause";
257 6b47a39f 2022-02-21 op 	case IMSG_CTL_STOP:
258 fad0fb69 2022-05-28 op 		return "stop";
259 6b47a39f 2022-02-21 op 	case IMSG_CTL_RESTART:
260 fad0fb69 2022-05-28 op 		return "restart";
261 6b47a39f 2022-02-21 op 	case IMSG_CTL_FLUSH:
262 fad0fb69 2022-05-28 op 		return "flush";
263 6b47a39f 2022-02-21 op 	case IMSG_CTL_NEXT:
264 fad0fb69 2022-05-28 op 		return "next";
265 6b47a39f 2022-02-21 op 	case IMSG_CTL_PREV:
266 fad0fb69 2022-05-28 op 		return "prev";
267 6b47a39f 2022-02-21 op 	case IMSG_CTL_JUMP:
268 fad0fb69 2022-05-28 op 		return "jump";
269 6b47a39f 2022-02-21 op 	case IMSG_CTL_REPEAT:
270 fad0fb69 2022-05-28 op 		return "repeat";
271 6b47a39f 2022-02-21 op 	case IMSG_CTL_ADD:
272 fad0fb69 2022-05-28 op 		return "add";
273 6b47a39f 2022-02-21 op 	case IMSG_CTL_COMMIT:
274 fad0fb69 2022-05-28 op 		return "load";
275 6b47a39f 2022-02-21 op 	default:
276 fad0fb69 2022-05-28 op 		return "unknown";
277 6b47a39f 2022-02-21 op 	}
278 6b47a39f 2022-02-21 op }
279 6b47a39f 2022-02-21 op 
280 6b47a39f 2022-02-21 op static int
281 3baa2617 2022-02-16 op ctlaction(struct parse_result *res)
282 3baa2617 2022-02-16 op {
283 fad0fb69 2022-05-28 op 	char path[PATH_MAX];
284 3baa2617 2022-02-16 op 	struct imsg imsg;
285 fad0fb69 2022-05-28 op 	struct player_status ps;
286 fad0fb69 2022-05-28 op 	size_t datalen;
287 3baa2617 2022-02-16 op 	ssize_t n;
288 fad0fb69 2022-05-28 op 	int i, type, ret = 0, done = 1;
289 3baa2617 2022-02-16 op 
290 3baa2617 2022-02-16 op 	switch (res->action) {
291 3baa2617 2022-02-16 op 	case PLAY:
292 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_PLAY, 0, 0, -1, NULL, 0);
293 00f500ff 2022-02-19 op 		if (verbose) {
294 00f500ff 2022-02-19 op 			imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1,
295 00f500ff 2022-02-19 op 			    NULL, 0);
296 00f500ff 2022-02-19 op 			done = 0;
297 00f500ff 2022-02-19 op 		}
298 3baa2617 2022-02-16 op 		break;
299 3baa2617 2022-02-16 op 	case PAUSE:
300 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_PAUSE, 0, 0, -1, NULL, 0);
301 3baa2617 2022-02-16 op 		break;
302 3baa2617 2022-02-16 op 	case TOGGLE:
303 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_TOGGLE_PLAY, 0, 0, -1, NULL, 0);
304 00f500ff 2022-02-19 op 		if (verbose) {
305 00f500ff 2022-02-19 op 			imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1,
306 00f500ff 2022-02-19 op 			    NULL, 0);
307 00f500ff 2022-02-19 op 			done = 0;
308 00f500ff 2022-02-19 op 		}
309 3baa2617 2022-02-16 op 		break;
310 3baa2617 2022-02-16 op 	case STOP:
311 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_STOP, 0, 0, -1, NULL, 0);
312 3baa2617 2022-02-16 op 		break;
313 3baa2617 2022-02-16 op 	case RESTART:
314 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_RESTART, 0, 0, -1, NULL, 0);
315 00f500ff 2022-02-19 op 		if (verbose) {
316 00f500ff 2022-02-19 op 			imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1,
317 00f500ff 2022-02-19 op 			    NULL, 0);
318 00f500ff 2022-02-19 op 			done = 0;
319 00f500ff 2022-02-19 op 		}
320 3baa2617 2022-02-16 op 		break;
321 3baa2617 2022-02-16 op 	case ADD:
322 8ff9231f 2022-02-16 op 		done = 0;
323 d903ec9a 2022-05-29 op 		for (i = 0; res->files[i] != NULL; ++i) {
324 18223c28 2022-06-13 op 			memset(path, 0, sizeof(path));
325 ef593b43 2022-06-09 op 			if (canonpath(res->files[i], path, sizeof(path))
326 ef593b43 2022-06-09 op 			    == -1) {
327 ef593b43 2022-06-09 op 				log_warn("canonpath %s", res->files[i]);
328 fad0fb69 2022-05-28 op 				continue;
329 fad0fb69 2022-05-28 op 			}
330 fad0fb69 2022-05-28 op 
331 fad0fb69 2022-05-28 op 			imsg_compose(ibuf, IMSG_CTL_ADD, 0, 0, -1,
332 fad0fb69 2022-05-28 op 			    path, sizeof(path));
333 fad0fb69 2022-05-28 op 		}
334 fad0fb69 2022-05-28 op 		ret = i == 0;
335 3baa2617 2022-02-16 op 		break;
336 3baa2617 2022-02-16 op 	case FLUSH:
337 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_FLUSH, 0, 0, -1, NULL, 0);
338 3baa2617 2022-02-16 op 		break;
339 3baa2617 2022-02-16 op 	case SHOW:
340 3baa2617 2022-02-16 op 		done = 0;
341 3baa2617 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_SHOW, 0, 0, -1, NULL, 0);
342 3baa2617 2022-02-16 op 		break;
343 bb3f279f 2022-02-16 op 	case STATUS:
344 bb3f279f 2022-02-16 op 		done = 0;
345 bb3f279f 2022-02-16 op 		imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1, NULL, 0);
346 af27e631 2022-02-17 op 		break;
347 af27e631 2022-02-17 op 	case NEXT:
348 af27e631 2022-02-17 op 		imsg_compose(ibuf, IMSG_CTL_NEXT, 0, 0, -1, NULL, 0);
349 00f500ff 2022-02-19 op 		if (verbose) {
350 00f500ff 2022-02-19 op 			imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1,
351 00f500ff 2022-02-19 op 			    NULL, 0);
352 00f500ff 2022-02-19 op 			done = 0;
353 00f500ff 2022-02-19 op 		}
354 bb3f279f 2022-02-16 op 		break;
355 af27e631 2022-02-17 op 	case PREV:
356 af27e631 2022-02-17 op 		imsg_compose(ibuf, IMSG_CTL_PREV, 0, 0, -1, NULL, 0);
357 00f500ff 2022-02-19 op 		if (verbose) {
358 00f500ff 2022-02-19 op 			imsg_compose(ibuf, IMSG_CTL_STATUS, 0, 0, -1,
359 00f500ff 2022-02-19 op 			    NULL, 0);
360 00f500ff 2022-02-19 op 			done = 0;
361 00f500ff 2022-02-19 op 		}
362 af27e631 2022-02-17 op 		break;
363 14be015f 2022-02-17 op 	case LOAD:
364 7a427ecd 2022-02-17 op 		done = 0;
365 7a427ecd 2022-02-17 op 		imsg_compose(ibuf, IMSG_CTL_BEGIN, 0, 0, -1, NULL, 0);
366 a913de21 2022-02-17 op 		break;
367 a913de21 2022-02-17 op 	case JUMP:
368 a913de21 2022-02-17 op 		done = 0;
369 fad0fb69 2022-05-28 op 		memset(path, 0, sizeof(path));
370 fad0fb69 2022-05-28 op 		strlcpy(path, res->file, sizeof(path));
371 fad0fb69 2022-05-28 op 		imsg_compose(ibuf, IMSG_CTL_JUMP, 0, 0, -1,
372 fad0fb69 2022-05-28 op 		    path, sizeof(path));
373 7a427ecd 2022-02-17 op 		break;
374 310ef57c 2022-02-19 op 	case REPEAT:
375 310ef57c 2022-02-19 op 		imsg_compose(ibuf, IMSG_CTL_REPEAT, 0, 0, -1,
376 310ef57c 2022-02-19 op 		    &res->rep, sizeof(res->rep));
377 310ef57c 2022-02-19 op 		break;
378 6b47a39f 2022-02-21 op 	case MONITOR:
379 6b47a39f 2022-02-21 op 		done = 0;
380 6b47a39f 2022-02-21 op 		imsg_compose(ibuf, IMSG_CTL_MONITOR, 0, 0, -1,
381 6b47a39f 2022-02-21 op 		    NULL, 0);
382 6b47a39f 2022-02-21 op 		break;
383 3baa2617 2022-02-16 op 	case NONE:
384 3baa2617 2022-02-16 op 		/* action not expected */
385 3baa2617 2022-02-16 op 		fatalx("invalid action %u", res->action);
386 3baa2617 2022-02-16 op 		break;
387 3baa2617 2022-02-16 op 	}
388 3baa2617 2022-02-16 op 
389 3baa2617 2022-02-16 op 	if (ret != 0)
390 3baa2617 2022-02-16 op 		goto end;
391 3baa2617 2022-02-16 op 
392 3baa2617 2022-02-16 op 	imsg_flush(ibuf);
393 3baa2617 2022-02-16 op 
394 fad0fb69 2022-05-28 op 	i = 0;
395 3baa2617 2022-02-16 op 	while (!done) {
396 3baa2617 2022-02-16 op 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
397 3baa2617 2022-02-16 op 			fatalx("imsg_read error");
398 3baa2617 2022-02-16 op 		if (n == 0)
399 3baa2617 2022-02-16 op 			fatalx("pipe closed");
400 3baa2617 2022-02-16 op 
401 3baa2617 2022-02-16 op 		while (!done) {
402 3baa2617 2022-02-16 op 			if ((n = imsg_get(ibuf, &imsg)) == -1)
403 3baa2617 2022-02-16 op 				fatalx("imsg_get error");
404 3baa2617 2022-02-16 op 			if (n == 0)
405 3baa2617 2022-02-16 op 				break;
406 3baa2617 2022-02-16 op 
407 fad0fb69 2022-05-28 op 			if (imsg.hdr.type == IMSG_CTL_ERR) {
408 fad0fb69 2022-05-28 op 				log_warnx("%s: %s", res->ctl->name,
409 fad0fb69 2022-05-28 op 				    imsg_strerror(&imsg));
410 fad0fb69 2022-05-28 op 				ret = 1;
411 fad0fb69 2022-05-28 op 				done = 1;
412 fad0fb69 2022-05-28 op 				break;
413 fad0fb69 2022-05-28 op 			}
414 fad0fb69 2022-05-28 op 
415 fad0fb69 2022-05-28 op 			datalen = IMSG_DATA_SIZE(imsg);
416 fad0fb69 2022-05-28 op 
417 3baa2617 2022-02-16 op 			switch (res->action) {
418 8ff9231f 2022-02-16 op 			case ADD:
419 d903ec9a 2022-05-29 op 				if (res->files[i] == NULL)
420 fad0fb69 2022-05-28 op 					fatalx("received more replies than "
421 fad0fb69 2022-05-28 op 					    "files enqueued.");
422 fad0fb69 2022-05-28 op 
423 fad0fb69 2022-05-28 op 				if (imsg.hdr.type == IMSG_CTL_ADD)
424 d903ec9a 2022-05-29 op 					log_debug("enqueued %s", res->files[i]);
425 fad0fb69 2022-05-28 op 				else
426 fad0fb69 2022-05-28 op 					fatalx("invalid message %d",
427 fad0fb69 2022-05-28 op 					    imsg.hdr.type);
428 fad0fb69 2022-05-28 op 				i++;
429 d903ec9a 2022-05-29 op 				done = res->files[i] == NULL;
430 8ff9231f 2022-02-16 op 				break;
431 3baa2617 2022-02-16 op 			case SHOW:
432 fad0fb69 2022-05-28 op 				if (datalen == 0) {
433 fad0fb69 2022-05-28 op 					done = 1;
434 fad0fb69 2022-05-28 op 					break;
435 fad0fb69 2022-05-28 op 				}
436 fad0fb69 2022-05-28 op 				if (datalen != sizeof(ps))
437 fad0fb69 2022-05-28 op 					fatalx("data size mismatch");
438 fad0fb69 2022-05-28 op 				memcpy(&ps, imsg.data, sizeof(ps));
439 fad0fb69 2022-05-28 op 				if (ps.path[sizeof(ps.path) - 1] != '\0')
440 fad0fb69 2022-05-28 op 					fatalx("received corrupted data");
441 fad0fb69 2022-05-28 op 				if (res->pretty) {
442 fad0fb69 2022-05-28 op 					char c = ' ';
443 fad0fb69 2022-05-28 op 					if (ps.status == STATE_PLAYING)
444 fad0fb69 2022-05-28 op 						c = '>';
445 fad0fb69 2022-05-28 op 					printf("%c ", c);
446 fad0fb69 2022-05-28 op 				}
447 fad0fb69 2022-05-28 op 				puts(ps.path);
448 3baa2617 2022-02-16 op 				break;
449 146307bd 2022-02-17 op 			case PLAY:
450 146307bd 2022-02-17 op 			case TOGGLE:
451 146307bd 2022-02-17 op 			case RESTART:
452 bb3f279f 2022-02-16 op 			case STATUS:
453 146307bd 2022-02-17 op 			case NEXT:
454 146307bd 2022-02-17 op 			case PREV:
455 a913de21 2022-02-17 op 			case JUMP:
456 fad0fb69 2022-05-28 op 				if (imsg.hdr.type != IMSG_CTL_STATUS)
457 fad0fb69 2022-05-28 op 					fatalx("invalid message %d",
458 fad0fb69 2022-05-28 op 					    imsg.hdr.type);
459 fad0fb69 2022-05-28 op 
460 fad0fb69 2022-05-28 op 				if (datalen != sizeof(ps))
461 fad0fb69 2022-05-28 op 					fatalx("data size mismatch");
462 fad0fb69 2022-05-28 op 				memcpy(&ps, imsg.data, sizeof(ps));
463 fad0fb69 2022-05-28 op 				if (ps.path[sizeof(ps.path) - 1] != '\0')
464 fad0fb69 2022-05-28 op 					fatalx("received corrupted data");
465 fad0fb69 2022-05-28 op 
466 fad0fb69 2022-05-28 op 				if (ps.status == STATE_STOPPED)
467 fad0fb69 2022-05-28 op 					printf("stopped ");
468 fad0fb69 2022-05-28 op 				else if (ps.status == STATE_PLAYING)
469 fad0fb69 2022-05-28 op 					printf("playing ");
470 fad0fb69 2022-05-28 op 				else if (ps.status == STATE_PAUSED)
471 fad0fb69 2022-05-28 op 					printf("paused ");
472 fad0fb69 2022-05-28 op 				else
473 fad0fb69 2022-05-28 op 					printf("unknown ");
474 fad0fb69 2022-05-28 op 
475 fad0fb69 2022-05-28 op 				puts(ps.path);
476 fad0fb69 2022-05-28 op 				printf("repat one %s\nrepeat all %s\n",
477 fad0fb69 2022-05-28 op 				    ps.rp.repeat_one ? "on" : "off",
478 fad0fb69 2022-05-28 op 				    ps.rp.repeat_all ? "on" : "off");
479 fad0fb69 2022-05-28 op 
480 fad0fb69 2022-05-28 op 				done = 1;
481 bb3f279f 2022-02-16 op 				break;
482 7a427ecd 2022-02-17 op 			case LOAD:
483 fad0fb69 2022-05-28 op 				if (imsg.hdr.type == IMSG_CTL_ADD)
484 fad0fb69 2022-05-28 op 					break;
485 fad0fb69 2022-05-28 op 				if (imsg.hdr.type == IMSG_CTL_COMMIT) {
486 fad0fb69 2022-05-28 op 					done = 1;
487 fad0fb69 2022-05-28 op 					break;
488 fad0fb69 2022-05-28 op 				}
489 fad0fb69 2022-05-28 op 
490 fad0fb69 2022-05-28 op 				if (imsg.hdr.type != IMSG_CTL_BEGIN)
491 fad0fb69 2022-05-28 op 					fatalx("invalid message %d",
492 fad0fb69 2022-05-28 op 					    imsg.hdr.type);
493 fad0fb69 2022-05-28 op 
494 fad0fb69 2022-05-28 op 				load_files(res, &ret);
495 7a427ecd 2022-02-17 op 				break;
496 6b47a39f 2022-02-21 op 			case MONITOR:
497 fad0fb69 2022-05-28 op 				if (imsg.hdr.type != IMSG_CTL_MONITOR)
498 fad0fb69 2022-05-28 op 					fatalx("invalid message %d",
499 fad0fb69 2022-05-28 op 					    imsg.hdr.type);
500 fad0fb69 2022-05-28 op 
501 fad0fb69 2022-05-28 op 				if (datalen != sizeof(type))
502 fad0fb69 2022-05-28 op 					fatalx("data size mismatch");
503 fad0fb69 2022-05-28 op 
504 fad0fb69 2022-05-28 op 				memcpy(&type, imsg.data, sizeof(type));
505 fad0fb69 2022-05-28 op 				if (type < 0 || type > IMSG__LAST)
506 fad0fb69 2022-05-28 op 					fatalx("received corrupted data");
507 fad0fb69 2022-05-28 op 
508 fad0fb69 2022-05-28 op 				if (!res->monitor[type])
509 fad0fb69 2022-05-28 op 					break;
510 fad0fb69 2022-05-28 op 
511 fad0fb69 2022-05-28 op 				puts(imsg_name(type));
512 fad0fb69 2022-05-28 op 				fflush(stdout);
513 6b47a39f 2022-02-21 op 				break;
514 3baa2617 2022-02-16 op 			default:
515 3baa2617 2022-02-16 op 				done = 1;
516 3baa2617 2022-02-16 op 				break;
517 3baa2617 2022-02-16 op 			}
518 3baa2617 2022-02-16 op 
519 3baa2617 2022-02-16 op 			imsg_free(&imsg);
520 3baa2617 2022-02-16 op 		}
521 3baa2617 2022-02-16 op 	}
522 3baa2617 2022-02-16 op 
523 3baa2617 2022-02-16 op end:
524 3baa2617 2022-02-16 op 	return ret;
525 3baa2617 2022-02-16 op }
526 3baa2617 2022-02-16 op 
527 3baa2617 2022-02-16 op int
528 3baa2617 2022-02-16 op ctl_noarg(struct parse_result *res, int argc, char **argv)
529 3baa2617 2022-02-16 op {
530 7e29fc4a 2022-03-03 op 	int ch;
531 7e29fc4a 2022-03-03 op 
532 7e29fc4a 2022-03-03 op 	while ((ch = getopt(argc, argv, "")) != -1)
533 7e29fc4a 2022-03-03 op 		ctl_usage(res->ctl);
534 7e29fc4a 2022-03-03 op 	argc -= optind;
535 7e29fc4a 2022-03-03 op 	argv += optind;
536 7e29fc4a 2022-03-03 op 
537 acaf7eb2 2022-03-05 op 	if (argc > 0)
538 3baa2617 2022-02-16 op 		ctl_usage(res->ctl);
539 7e29fc4a 2022-03-03 op 
540 3baa2617 2022-02-16 op 	return ctlaction(res);
541 3baa2617 2022-02-16 op }
542 3baa2617 2022-02-16 op 
543 3baa2617 2022-02-16 op int
544 3baa2617 2022-02-16 op ctl_add(struct parse_result *res, int argc, char **argv)
545 3baa2617 2022-02-16 op {
546 7e29fc4a 2022-03-03 op 	int ch;
547 7e29fc4a 2022-03-03 op 
548 7e29fc4a 2022-03-03 op 	while ((ch = getopt(argc, argv, "")) != -1)
549 3baa2617 2022-02-16 op 		ctl_usage(res->ctl);
550 7e29fc4a 2022-03-03 op 	argc -= optind;
551 7e29fc4a 2022-03-03 op 	argv += optind;
552 7e29fc4a 2022-03-03 op 
553 7e29fc4a 2022-03-03 op 	if (argc == 0)
554 7e29fc4a 2022-03-03 op 		ctl_usage(res->ctl);
555 7e29fc4a 2022-03-03 op 	res->files = argv;
556 14be015f 2022-02-17 op 
557 3baa2617 2022-02-16 op 	return ctlaction(res);
558 3baa2617 2022-02-16 op }
559 14be015f 2022-02-17 op 
560 14be015f 2022-02-17 op int
561 8e916714 2022-02-17 op ctl_show(struct parse_result *res, int argc, char **argv)
562 8e916714 2022-02-17 op {
563 8e916714 2022-02-17 op 	int ch;
564 8e916714 2022-02-17 op 
565 8e916714 2022-02-17 op 	while ((ch = getopt(argc, argv, "p")) != -1) {
566 8e916714 2022-02-17 op 		switch (ch) {
567 8e916714 2022-02-17 op 		case 'p':
568 8e916714 2022-02-17 op 			res->pretty = 1;
569 8e916714 2022-02-17 op 			break;
570 8e916714 2022-02-17 op 		default:
571 8e916714 2022-02-17 op 			ctl_usage(res->ctl);
572 8e916714 2022-02-17 op 		}
573 8e916714 2022-02-17 op 	}
574 8e916714 2022-02-17 op 
575 8e916714 2022-02-17 op 	return ctlaction(res);
576 8e916714 2022-02-17 op }
577 8e916714 2022-02-17 op 
578 8e916714 2022-02-17 op int
579 14be015f 2022-02-17 op ctl_load(struct parse_result *res, int argc, char **argv)
580 14be015f 2022-02-17 op {
581 7e29fc4a 2022-03-03 op 	int ch;
582 7e29fc4a 2022-03-03 op 
583 7e29fc4a 2022-03-03 op 	while ((ch = getopt(argc, argv, "")) != -1)
584 7e29fc4a 2022-03-03 op 		ctl_usage(res->ctl);
585 7e29fc4a 2022-03-03 op 	argc -= optind;
586 7e29fc4a 2022-03-03 op 	argv += optind;
587 7e29fc4a 2022-03-03 op 
588 7e29fc4a 2022-03-03 op 	if (argc == 0)
589 ec1fb0c7 2022-02-17 op 		res->file = NULL;
590 7e29fc4a 2022-03-03 op 	else if (argc == 1)
591 7e29fc4a 2022-03-03 op 		res->file = argv[0];
592 ec1fb0c7 2022-02-17 op 	else
593 14be015f 2022-02-17 op 		ctl_usage(res->ctl);
594 14be015f 2022-02-17 op 
595 14be015f 2022-02-17 op 	if (pledge("stdio rpath", NULL) == -1)
596 14be015f 2022-02-17 op 		fatal("pledge");
597 14be015f 2022-02-17 op 
598 7a427ecd 2022-02-17 op 	return ctlaction(res);
599 14be015f 2022-02-17 op }
600 14be015f 2022-02-17 op 
601 a913de21 2022-02-17 op int
602 a913de21 2022-02-17 op ctl_jump(struct parse_result *res, int argc, char **argv)
603 a913de21 2022-02-17 op {
604 a913de21 2022-02-17 op 	int ch;
605 a913de21 2022-02-17 op 
606 a913de21 2022-02-17 op 	while ((ch = getopt(argc, argv, "")) != -1)
607 a913de21 2022-02-17 op 		ctl_usage(res->ctl);
608 a913de21 2022-02-17 op 	argc -= optind;
609 a913de21 2022-02-17 op 	argv += optind;
610 a913de21 2022-02-17 op 
611 a913de21 2022-02-17 op 	if (argc != 1)
612 a913de21 2022-02-17 op 		ctl_usage(res->ctl);
613 a913de21 2022-02-17 op 
614 a913de21 2022-02-17 op 	res->file = argv[0];
615 310ef57c 2022-02-19 op 	return ctlaction(res);
616 310ef57c 2022-02-19 op }
617 310ef57c 2022-02-19 op 
618 310ef57c 2022-02-19 op int
619 310ef57c 2022-02-19 op ctl_repeat(struct parse_result *res, int argc, char **argv)
620 310ef57c 2022-02-19 op {
621 310ef57c 2022-02-19 op 	int ch, b;
622 310ef57c 2022-02-19 op 
623 310ef57c 2022-02-19 op 	while ((ch = getopt(argc, argv, "")) != -1)
624 310ef57c 2022-02-19 op 		ctl_usage(res->ctl);
625 310ef57c 2022-02-19 op 	argc -= optind;
626 310ef57c 2022-02-19 op 	argv += optind;
627 310ef57c 2022-02-19 op 
628 310ef57c 2022-02-19 op 	if (argc != 2)
629 310ef57c 2022-02-19 op 		ctl_usage(res->ctl);
630 310ef57c 2022-02-19 op 
631 310ef57c 2022-02-19 op 	if (!strcmp(argv[1], "on"))
632 310ef57c 2022-02-19 op 		b = 1;
633 310ef57c 2022-02-19 op 	else if (!strcmp(argv[1], "off"))
634 310ef57c 2022-02-19 op 		b = 0;
635 310ef57c 2022-02-19 op 	else
636 310ef57c 2022-02-19 op 		ctl_usage(res->ctl);
637 310ef57c 2022-02-19 op 
638 310ef57c 2022-02-19 op 	res->rep.repeat_one = -1;
639 310ef57c 2022-02-19 op 	res->rep.repeat_all = -1;
640 310ef57c 2022-02-19 op 	if (!strcmp(argv[0], "one"))
641 310ef57c 2022-02-19 op 		res->rep.repeat_one = b;
642 310ef57c 2022-02-19 op 	else if (!strcmp(argv[0], "all"))
643 310ef57c 2022-02-19 op 		res->rep.repeat_all = b;
644 310ef57c 2022-02-19 op 	else
645 310ef57c 2022-02-19 op 		ctl_usage(res->ctl);
646 90122a37 2022-05-14 op 
647 90122a37 2022-05-14 op 	return ctlaction(res);
648 90122a37 2022-05-14 op }
649 90122a37 2022-05-14 op 
650 90122a37 2022-05-14 op int
651 90122a37 2022-05-14 op ctl_monitor(struct parse_result *res, int argc, char **argv)
652 90122a37 2022-05-14 op {
653 90122a37 2022-05-14 op 	int ch;
654 90122a37 2022-05-14 op 	const char *events;
655 90122a37 2022-05-14 op 	char *dup, *tmp, *tok;
656 90122a37 2022-05-14 op 
657 90122a37 2022-05-14 op 	while ((ch = getopt(argc, argv, "")) != -1)
658 90122a37 2022-05-14 op 		ctl_usage(res->ctl);
659 90122a37 2022-05-14 op 	argc -= optind;
660 90122a37 2022-05-14 op 	argv += optind;
661 90122a37 2022-05-14 op 
662 90122a37 2022-05-14 op 	if (argc > 1)
663 90122a37 2022-05-14 op 		ctl_usage(res->ctl);
664 90122a37 2022-05-14 op 
665 90122a37 2022-05-14 op 	if (argc == 1)
666 90122a37 2022-05-14 op 		events = *argv;
667 90122a37 2022-05-14 op 	else
668 90122a37 2022-05-14 op 		events = "play,toggle,pause,stop,restart,flush,next,prev,"
669 90122a37 2022-05-14 op 			"jump,repeat,add,load";
670 310ef57c 2022-02-19 op 
671 90122a37 2022-05-14 op 	tmp = dup = xstrdup(events);
672 90122a37 2022-05-14 op 	while ((tok = strsep(&tmp, ",")) != NULL) {
673 90122a37 2022-05-14 op 		if (*tok == '\0')
674 90122a37 2022-05-14 op 			continue;
675 90122a37 2022-05-14 op 
676 90122a37 2022-05-14 op 		if (!strcmp(tok, "play"))
677 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_PLAY] = 1;
678 90122a37 2022-05-14 op 		else if (!strcmp(tok, "toggle"))
679 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_TOGGLE_PLAY] = 1;
680 90122a37 2022-05-14 op 		else if (!strcmp(tok, "pause"))
681 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_PAUSE] = 1;
682 90122a37 2022-05-14 op 		else if (!strcmp(tok, "stop"))
683 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_STOP] = 1;
684 90122a37 2022-05-14 op 		else if (!strcmp(tok, "restart"))
685 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_RESTART] = 1;
686 90122a37 2022-05-14 op 		else if (!strcmp(tok, "flush"))
687 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_FLUSH] = 1;
688 90122a37 2022-05-14 op 		else if (!strcmp(tok, "next"))
689 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_NEXT] = 1;
690 90122a37 2022-05-14 op 		else if (!strcmp(tok, "prev"))
691 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_PREV] = 1;
692 90122a37 2022-05-14 op 		else if (!strcmp(tok, "jump"))
693 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_JUMP] = 1;
694 90122a37 2022-05-14 op 		else if (!strcmp(tok, "repeat"))
695 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_REPEAT] = 1;
696 90122a37 2022-05-14 op 		else if (!strcmp(tok, "add"))
697 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_ADD] = 1;
698 90122a37 2022-05-14 op 		else if (!strcmp(tok, "load"))
699 90122a37 2022-05-14 op 			res->monitor[IMSG_CTL_COMMIT] = 1;
700 90122a37 2022-05-14 op 		else
701 90122a37 2022-05-14 op 			fatalx("unknown event \"%s\"", tok);
702 90122a37 2022-05-14 op 	}
703 90122a37 2022-05-14 op 
704 90122a37 2022-05-14 op 	free(dup);
705 a913de21 2022-02-17 op 	return ctlaction(res);
706 a913de21 2022-02-17 op }
707 a913de21 2022-02-17 op 
708 fea541a8 2022-02-16 op static int
709 1d673950 2022-03-03 op ctl_get_lock(const char *lockfile)
710 3baa2617 2022-02-16 op {
711 1d673950 2022-03-03 op 	int lockfd;
712 3baa2617 2022-02-16 op 
713 1d673950 2022-03-03 op 	if ((lockfd = open(lockfile, O_WRONLY|O_CREAT, 0600)) == -1) {
714 1d673950 2022-03-03 op 		log_debug("open failed: %s", strerror(errno));
715 1d673950 2022-03-03 op 		return -1;
716 1d673950 2022-03-03 op 	}
717 3baa2617 2022-02-16 op 
718 1d673950 2022-03-03 op 	if (flock(lockfd, LOCK_EX|LOCK_NB) == -1) {
719 1d673950 2022-03-03 op 		log_debug("flock failed: %s", strerror(errno));
720 1d673950 2022-03-03 op 		if (errno != EAGAIN)
721 1d673950 2022-03-03 op 			return -1;
722 1d673950 2022-03-03 op 		while (flock(lockfd, LOCK_EX) == -1 && errno == EINTR)
723 1d673950 2022-03-03 op 			/* nop */;
724 1d673950 2022-03-03 op 		close(lockfd);
725 1d673950 2022-03-03 op 		return -2;
726 1d673950 2022-03-03 op 	}
727 1d673950 2022-03-03 op 	log_debug("flock succeeded");
728 1d673950 2022-03-03 op 
729 1d673950 2022-03-03 op 	return lockfd;
730 1d673950 2022-03-03 op }
731 1d673950 2022-03-03 op 
732 1d673950 2022-03-03 op static int
733 1d673950 2022-03-03 op ctl_connect(void)
734 1d673950 2022-03-03 op {
735 1d673950 2022-03-03 op 	struct timespec		 ts = { 0, 50000000 }; /* 0.05 seconds */
736 1d673950 2022-03-03 op 	struct sockaddr_un	 sa;
737 1d673950 2022-03-03 op 	size_t			 size;
738 1d673950 2022-03-03 op 	int			 fd, lockfd = -1, locked = 0, spawned = 0;
739 1d673950 2022-03-03 op 	char			*lockfile = NULL;
740 1d673950 2022-03-03 op 
741 1d673950 2022-03-03 op 	memset(&sa, 0, sizeof(sa));
742 1d673950 2022-03-03 op 	sa.sun_family = AF_UNIX;
743 1d673950 2022-03-03 op 	size = strlcpy(sa.sun_path, csock, sizeof(sa.sun_path));
744 1d673950 2022-03-03 op 	if (size >= sizeof(sa.sun_path)) {
745 1d673950 2022-03-03 op 		errno = ENAMETOOLONG;
746 fea541a8 2022-02-16 op 		return -1;
747 fea541a8 2022-02-16 op 	}
748 3baa2617 2022-02-16 op 
749 1d673950 2022-03-03 op retry:
750 1d673950 2022-03-03 op 	if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
751 1d673950 2022-03-03 op 		return -1;
752 1d673950 2022-03-03 op 
753 1d673950 2022-03-03 op 	if (connect(fd, (struct sockaddr *)&sa, sizeof(sa)) == -1) {
754 1d673950 2022-03-03 op 		log_debug("connection failed: %s", strerror(errno));
755 1d673950 2022-03-03 op 		if (errno != ECONNREFUSED && errno != ENOENT)
756 1d673950 2022-03-03 op 			goto failed;
757 1d673950 2022-03-03 op 		close(fd);
758 1d673950 2022-03-03 op 
759 1d673950 2022-03-03 op 		if (!locked) {
760 1d673950 2022-03-03 op 			xasprintf(&lockfile, "%s.lock", csock);
761 1d673950 2022-03-03 op 			if ((lockfd = ctl_get_lock(lockfile)) < 0) {
762 1d673950 2022-03-03 op 				log_debug("didn't get the lock (%d)", lockfd);
763 1d673950 2022-03-03 op 
764 1d673950 2022-03-03 op 				free(lockfile);
765 1d673950 2022-03-03 op 				lockfile = NULL;
766 1d673950 2022-03-03 op 
767 1d673950 2022-03-03 op 				if (lockfd == -1)
768 1d673950 2022-03-03 op 					goto retry;
769 1d673950 2022-03-03 op 			}
770 1d673950 2022-03-03 op 
771 1d673950 2022-03-03 op 			/*
772 1d673950 2022-03-03 op 			 * Always retry at least once, even if we got
773 1d673950 2022-03-03 op 			 * the lock, because another client could have
774 1d673950 2022-03-03 op 			 * taken the lock, started the server and released
775 1d673950 2022-03-03 op 			 * the lock between our connect() and flock()
776 1d673950 2022-03-03 op 			 */
777 1d673950 2022-03-03 op 			locked = 1;
778 1d673950 2022-03-03 op 			goto retry;
779 1d673950 2022-03-03 op 		}
780 1d673950 2022-03-03 op 
781 1d673950 2022-03-03 op 		if (!spawned) {
782 1d673950 2022-03-03 op 			log_debug("spawning the daemon");
783 1d673950 2022-03-03 op 			spawn_daemon();
784 1d673950 2022-03-03 op 			spawned = 1;
785 1d673950 2022-03-03 op 		}
786 1d673950 2022-03-03 op 
787 1d673950 2022-03-03 op 		nanosleep(&ts, NULL);
788 1d673950 2022-03-03 op 		goto retry;
789 1d673950 2022-03-03 op 	}
790 1d673950 2022-03-03 op 
791 1d673950 2022-03-03 op 	if (locked && lockfd >= 0) {
792 1d673950 2022-03-03 op 		unlink(lockfile);
793 1d673950 2022-03-03 op 		free(lockfile);
794 1d673950 2022-03-03 op 		close(lockfd);
795 1d673950 2022-03-03 op 	}
796 1d673950 2022-03-03 op 	return fd;
797 1d673950 2022-03-03 op 
798 1d673950 2022-03-03 op failed:
799 1d673950 2022-03-03 op 	if (locked) {
800 1d673950 2022-03-03 op 		free(lockfile);
801 1d673950 2022-03-03 op 		close(lockfd);
802 1d673950 2022-03-03 op 	}
803 1d673950 2022-03-03 op 	close(fd);
804 1d673950 2022-03-03 op 	return -1;
805 fea541a8 2022-02-16 op }
806 fea541a8 2022-02-16 op 
807 fea541a8 2022-02-16 op __dead void
808 fea541a8 2022-02-16 op ctl(int argc, char **argv)
809 fea541a8 2022-02-16 op {
810 1d673950 2022-03-03 op 	int ctl_sock;
811 3baa2617 2022-02-16 op 
812 fea541a8 2022-02-16 op 	log_init(1, LOG_DAEMON);
813 fea541a8 2022-02-16 op 	log_setverbose(verbose);
814 fea541a8 2022-02-16 op 
815 96233476 2022-06-09 op 	if (getcwd(cwd, sizeof(cwd)) == NULL)
816 96233476 2022-06-09 op 		fatal("getcwd");
817 96233476 2022-06-09 op 
818 1d673950 2022-03-03 op 	if ((ctl_sock = ctl_connect()) == -1)
819 1d673950 2022-03-03 op 		fatal("can't connect");
820 fea541a8 2022-02-16 op 
821 fea541a8 2022-02-16 op 	if (ctl_sock == -1)
822 fea541a8 2022-02-16 op 		fatalx("failed to connect to the daemon");
823 fea541a8 2022-02-16 op 
824 3baa2617 2022-02-16 op 	ibuf = xmalloc(sizeof(*ibuf));
825 3baa2617 2022-02-16 op 	imsg_init(ibuf, ctl_sock);
826 3baa2617 2022-02-16 op 
827 3baa2617 2022-02-16 op 	optreset = 1;
828 3baa2617 2022-02-16 op 	optind = 1;
829 3baa2617 2022-02-16 op 
830 3baa2617 2022-02-16 op 	exit(parse(argc, argv));
831 3baa2617 2022-02-16 op }