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