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