1 3baa2617 2022-02-16 op /* $OpenBSD: control.c,v 1.8 2021/03/02 04:10:07 jsg Exp $ */
4 578f8d0c 2023-05-02 op * Copyright (c) 2022, 2023 Omar Polo <op@omarpolo.com>
5 3baa2617 2022-02-16 op * Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
7 3baa2617 2022-02-16 op * Permission to use, copy, modify, and distribute this software for any
8 3baa2617 2022-02-16 op * purpose with or without fee is hereby granted, provided that the above
9 3baa2617 2022-02-16 op * copyright notice and this permission notice appear in all copies.
11 3baa2617 2022-02-16 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12 3baa2617 2022-02-16 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 3baa2617 2022-02-16 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14 3baa2617 2022-02-16 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 3baa2617 2022-02-16 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 3baa2617 2022-02-16 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17 3baa2617 2022-02-16 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
20 f36fd90a 2022-07-09 op #include "config.h"
22 3baa2617 2022-02-16 op #include <sys/stat.h>
23 3baa2617 2022-02-16 op #include <sys/socket.h>
24 3baa2617 2022-02-16 op #include <sys/un.h>
26 3baa2617 2022-02-16 op #include <netinet/in.h>
27 3baa2617 2022-02-16 op #include <net/if.h>
29 3baa2617 2022-02-16 op #include <errno.h>
30 1738062c 2023-10-07 op #include <fcntl.h>
31 bb3f279f 2022-02-16 op #include <limits.h>
32 0a53bd62 2023-08-16 op #include <poll.h>
33 14be015f 2022-02-17 op #include <stdio.h>
34 3baa2617 2022-02-16 op #include <stdlib.h>
35 3baa2617 2022-02-16 op #include <string.h>
36 3baa2617 2022-02-16 op #include <unistd.h>
38 3baa2617 2022-02-16 op #include "amused.h"
39 0a53bd62 2023-08-16 op #include "ev.h"
40 3baa2617 2022-02-16 op #include "log.h"
41 3baa2617 2022-02-16 op #include "control.h"
42 3baa2617 2022-02-16 op #include "playlist.h"
44 3baa2617 2022-02-16 op #define CONTROL_BACKLOG 5
48 19d6b480 2022-02-17 op struct playlist play;
50 19d6b480 2022-02-17 op } control_state = {.fd = -1, .tx = -1};
52 3baa2617 2022-02-16 op struct ctl_conn {
53 3baa2617 2022-02-16 op TAILQ_ENTRY(ctl_conn) entry;
54 87f575c3 2022-02-21 op int monitor; /* 1 if client is in monitor mode */
55 3baa2617 2022-02-16 op struct imsgev iev;
58 3baa2617 2022-02-16 op TAILQ_HEAD(ctl_conns, ctl_conn) ctl_conns = TAILQ_HEAD_INITIALIZER(ctl_conns);
60 3baa2617 2022-02-16 op struct ctl_conn *control_connbyfd(int);
61 3baa2617 2022-02-16 op struct ctl_conn *control_connbypid(pid_t);
62 3baa2617 2022-02-16 op void control_close(int);
65 3baa2617 2022-02-16 op control_init(char *path)
67 3baa2617 2022-02-16 op struct sockaddr_un sun;
68 1738062c 2023-10-07 op int fd, flags;
69 3baa2617 2022-02-16 op mode_t old_umask;
71 1738062c 2023-10-07 op if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
72 3baa2617 2022-02-16 op log_warn("%s: socket", __func__);
76 1738062c 2023-10-07 op if ((flags = fcntl(fd, F_GETFL)) == -1 ||
77 1738062c 2023-10-07 op fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
78 1738062c 2023-10-07 op fatal("fcntl(O_NONBLOCK)");
80 1738062c 2023-10-07 op if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1)
81 1738062c 2023-10-07 op fatal("fcntl(CLOEXEC)");
83 3baa2617 2022-02-16 op memset(&sun, 0, sizeof(sun));
84 3baa2617 2022-02-16 op sun.sun_family = AF_UNIX;
85 3baa2617 2022-02-16 op strlcpy(sun.sun_path, path, sizeof(sun.sun_path));
87 3baa2617 2022-02-16 op if (unlink(path) == -1)
88 3baa2617 2022-02-16 op if (errno != ENOENT) {
89 3baa2617 2022-02-16 op log_warn("%s: unlink %s", __func__, path);
94 3baa2617 2022-02-16 op old_umask = umask(S_IXUSR|S_IXGRP|S_IWOTH|S_IROTH|S_IXOTH);
95 3baa2617 2022-02-16 op if (bind(fd, (struct sockaddr *)&sun, sizeof(sun)) == -1) {
96 3baa2617 2022-02-16 op log_warn("%s: bind: %s", __func__, path);
98 3baa2617 2022-02-16 op umask(old_umask);
101 3baa2617 2022-02-16 op umask(old_umask);
103 3baa2617 2022-02-16 op if (chmod(path, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP) == -1) {
104 3baa2617 2022-02-16 op log_warn("%s: chmod", __func__);
106 3baa2617 2022-02-16 op (void)unlink(path);
114 3c6e1aae 2023-08-16 op enable_accept(int fd, int ev, void *bula)
116 3c6e1aae 2023-08-16 op ev_add(control_state.fd, POLLIN, control_accept, NULL);
120 3baa2617 2022-02-16 op control_listen(int fd)
122 3baa2617 2022-02-16 op if (control_state.fd != -1)
123 3baa2617 2022-02-16 op fatalx("%s: received unexpected controlsock", __func__);
125 3baa2617 2022-02-16 op control_state.fd = fd;
126 3baa2617 2022-02-16 op if (listen(control_state.fd, CONTROL_BACKLOG) == -1) {
127 3baa2617 2022-02-16 op log_warn("%s: listen", __func__);
131 3c6e1aae 2023-08-16 op enable_accept(-1, 0, NULL);
136 0a53bd62 2023-08-16 op control_accept(int listenfd, int event, void *bula)
138 23a99671 2023-10-08 op int connfd, flags;
139 3baa2617 2022-02-16 op socklen_t len;
140 3baa2617 2022-02-16 op struct sockaddr_un sun;
141 3baa2617 2022-02-16 op struct ctl_conn *c;
143 3baa2617 2022-02-16 op len = sizeof(sun);
144 23a99671 2023-10-08 op if ((connfd = accept(listenfd, (struct sockaddr *)&sun, &len)) == -1) {
146 3baa2617 2022-02-16 op * Pause accept if we are out of file descriptors, or
147 3baa2617 2022-02-16 op * libevent will haunt us here too.
149 3baa2617 2022-02-16 op if (errno == ENFILE || errno == EMFILE) {
150 3baa2617 2022-02-16 op struct timeval evtpause = { 1, 0 };
152 0a53bd62 2023-08-16 op ev_del(control_state.fd);
153 3c6e1aae 2023-08-16 op ev_timer(&evtpause, enable_accept, NULL);
154 3baa2617 2022-02-16 op } else if (errno != EWOULDBLOCK && errno != EINTR &&
155 3baa2617 2022-02-16 op errno != ECONNABORTED)
156 3baa2617 2022-02-16 op log_warn("%s: accept4", __func__);
160 23a99671 2023-10-08 op if ((flags = fcntl(connfd, F_GETFL)) == -1 ||
161 23a99671 2023-10-08 op fcntl(connfd, F_SETFL, flags | O_NONBLOCK) == -1)
162 23a99671 2023-10-08 op fatal("fcntl(O_NONBLOCK)");
163 23a99671 2023-10-08 op if (fcntl(connfd, F_SETFD, FD_CLOEXEC) == -1)
164 23a99671 2023-10-08 op fatal("fcntl(CLOEXEC)");
166 3baa2617 2022-02-16 op if ((c = calloc(1, sizeof(struct ctl_conn))) == NULL) {
167 3baa2617 2022-02-16 op log_warn("%s: calloc", __func__);
168 3baa2617 2022-02-16 op close(connfd);
172 3baa2617 2022-02-16 op imsg_init(&c->iev.ibuf, connfd);
173 3baa2617 2022-02-16 op c->iev.handler = control_dispatch_imsg;
174 0a53bd62 2023-08-16 op c->iev.events = POLLIN;
175 0a53bd62 2023-08-16 op ev_add(c->iev.ibuf.fd, c->iev.events, c->iev.handler, &c->iev);
177 3baa2617 2022-02-16 op TAILQ_INSERT_TAIL(&ctl_conns, c, entry);
180 3baa2617 2022-02-16 op struct ctl_conn *
181 3baa2617 2022-02-16 op control_connbyfd(int fd)
183 3baa2617 2022-02-16 op struct ctl_conn *c;
185 3baa2617 2022-02-16 op TAILQ_FOREACH(c, &ctl_conns, entry) {
186 3baa2617 2022-02-16 op if (c->iev.ibuf.fd == fd)
193 3baa2617 2022-02-16 op struct ctl_conn *
194 3baa2617 2022-02-16 op control_connbypid(pid_t pid)
196 3baa2617 2022-02-16 op struct ctl_conn *c;
198 3baa2617 2022-02-16 op TAILQ_FOREACH(c, &ctl_conns, entry) {
199 3baa2617 2022-02-16 op if (c->iev.ibuf.pid == pid)
207 3baa2617 2022-02-16 op control_close(int fd)
209 3baa2617 2022-02-16 op struct ctl_conn *c;
211 3baa2617 2022-02-16 op if ((c = control_connbyfd(fd)) == NULL) {
212 3baa2617 2022-02-16 op log_warnx("%s: fd %d: not found", __func__, fd);
216 19d6b480 2022-02-17 op /* abort the transaction if running by this user */
217 c6304ab9 2022-02-17 op if (control_state.tx != -1 && c->iev.ibuf.fd == control_state.tx) {
218 19d6b480 2022-02-17 op playlist_free(&control_state.play);
219 19d6b480 2022-02-17 op control_state.tx = -1;
222 3baa2617 2022-02-16 op msgbuf_clear(&c->iev.ibuf.w);
223 3baa2617 2022-02-16 op TAILQ_REMOVE(&ctl_conns, c, entry);
225 0a53bd62 2023-08-16 op ev_del(c->iev.ibuf.fd);
226 3baa2617 2022-02-16 op close(c->iev.ibuf.fd);
228 3baa2617 2022-02-16 op /* Some file descriptors are available again. */
229 0a53bd62 2023-08-16 op if (ev_timer_pending()) {
230 0a53bd62 2023-08-16 op ev_timer(NULL, NULL, NULL);
231 0a53bd62 2023-08-16 op ev_add(control_state.fd, POLLIN, control_accept, NULL);
238 949b5c2e 2022-07-12 op control_notify(int type)
240 87f575c3 2022-02-21 op struct ctl_conn *c;
241 3dd90731 2023-01-09 op struct player_event ev;
243 3dd90731 2023-01-09 op memset(&ev, 0, sizeof(ev));
244 3dd90731 2023-01-09 op ev.event = type;
245 3dd90731 2023-01-09 op ev.position = current_position;
246 3dd90731 2023-01-09 op ev.duration = current_duration;
247 3dd90731 2023-01-09 op ev.mode.repeat_one = repeat_one;
248 3dd90731 2023-01-09 op ev.mode.repeat_all = repeat_all;
249 3dd90731 2023-01-09 op ev.mode.consume = consume;
251 87f575c3 2022-02-21 op TAILQ_FOREACH(c, &ctl_conns, entry) {
252 949b5c2e 2022-07-12 op if (!c->monitor)
255 87f575c3 2022-02-21 op imsg_compose_event(&c->iev, IMSG_CTL_MONITOR, 0, 0,
256 3dd90731 2023-01-09 op -1, &ev, sizeof(ev));
261 9881c98c 2022-07-13 op new_mode(int val, int newval)
263 9881c98c 2022-07-13 op if (newval == MODE_UNDEF)
265 9881c98c 2022-07-13 op if (newval == MODE_TOGGLE)
267 9881c98c 2022-07-13 op return !!newval;
271 0a53bd62 2023-08-16 op control_dispatch_imsg(int fd, int event, void *bula)
273 310ef57c 2022-02-19 op struct ctl_conn *c;
274 310ef57c 2022-02-19 op struct imsg imsg;
275 9fb94242 2022-07-13 op struct player_mode mode;
276 15aecb89 2022-07-12 op struct player_seek seek;
277 3af93963 2022-03-02 op ssize_t n, off;
279 3baa2617 2022-02-16 op if ((c = control_connbyfd(fd)) == NULL) {
280 3baa2617 2022-02-16 op log_warnx("%s: fd %d: not found", __func__, fd);
284 0a53bd62 2023-08-16 op if (event & POLLIN) {
285 3baa2617 2022-02-16 op if (((n = imsg_read(&c->iev.ibuf)) == -1 && errno != EAGAIN) ||
287 3baa2617 2022-02-16 op control_close(fd);
291 0a53bd62 2023-08-16 op if (event & POLLOUT) {
292 3baa2617 2022-02-16 op if (msgbuf_write(&c->iev.ibuf.w) <= 0 && errno != EAGAIN) {
293 3baa2617 2022-02-16 op control_close(fd);
299 3baa2617 2022-02-16 op if ((n = imsg_get(&c->iev.ibuf, &imsg)) == -1) {
300 3baa2617 2022-02-16 op control_close(fd);
306 3baa2617 2022-02-16 op switch (imsg.hdr.type) {
307 3baa2617 2022-02-16 op case IMSG_CTL_PLAY:
308 3baa2617 2022-02-16 op switch (play_state) {
309 3baa2617 2022-02-16 op case STATE_STOPPED:
310 2f589330 2022-02-17 op main_playlist_resume();
312 3baa2617 2022-02-16 op case STATE_PLAYING:
313 3baa2617 2022-02-16 op /* do nothing */
315 3baa2617 2022-02-16 op case STATE_PAUSED:
316 950ad05c 2022-02-16 op play_state = STATE_PLAYING;
317 791d3db3 2022-07-09 op main_send_player(IMSG_RESUME, -1, NULL ,0);
320 949b5c2e 2022-07-12 op control_notify(imsg.hdr.type);
322 3baa2617 2022-02-16 op case IMSG_CTL_TOGGLE_PLAY:
323 3baa2617 2022-02-16 op switch (play_state) {
324 3baa2617 2022-02-16 op case STATE_STOPPED:
325 15aecb89 2022-07-12 op control_notify(IMSG_CTL_PLAY);
326 2f589330 2022-02-17 op main_playlist_resume();
328 3baa2617 2022-02-16 op case STATE_PLAYING:
329 15aecb89 2022-07-12 op control_notify(IMSG_CTL_PAUSE);
330 2ac7eafd 2022-02-16 op play_state = STATE_PAUSED;
331 791d3db3 2022-07-09 op main_send_player(IMSG_PAUSE, -1, NULL, 0);
333 3baa2617 2022-02-16 op case STATE_PAUSED:
334 15aecb89 2022-07-12 op control_notify(IMSG_CTL_PLAY);
335 2ac7eafd 2022-02-16 op play_state = STATE_PLAYING;
336 791d3db3 2022-07-09 op main_send_player(IMSG_RESUME, -1, NULL, 0);
340 3baa2617 2022-02-16 op case IMSG_CTL_PAUSE:
341 3baa2617 2022-02-16 op if (play_state != STATE_PLAYING)
343 950ad05c 2022-02-16 op play_state = STATE_PAUSED;
344 791d3db3 2022-07-09 op main_send_player(IMSG_PAUSE, -1, NULL, 0);
345 949b5c2e 2022-07-12 op control_notify(imsg.hdr.type);
347 3baa2617 2022-02-16 op case IMSG_CTL_STOP:
348 3baa2617 2022-02-16 op if (play_state == STATE_STOPPED)
350 bfaa7897 2022-02-17 op play_state = STATE_STOPPED;
351 791d3db3 2022-07-09 op main_send_player(IMSG_STOP, -1, NULL, 0);
352 949b5c2e 2022-07-12 op control_notify(imsg.hdr.type);
354 3baa2617 2022-02-16 op case IMSG_CTL_FLUSH:
355 3baa2617 2022-02-16 op playlist_truncate();
356 15aecb89 2022-07-12 op control_notify(IMSG_CTL_COMMIT);
358 3baa2617 2022-02-16 op case IMSG_CTL_SHOW:
359 d980494c 2022-02-16 op main_send_playlist(&c->iev);
361 bb3f279f 2022-02-16 op case IMSG_CTL_STATUS:
362 bb3f279f 2022-02-16 op main_send_status(&c->iev);
364 af27e631 2022-02-17 op case IMSG_CTL_NEXT:
365 15aecb89 2022-07-12 op control_notify(imsg.hdr.type);
366 791d3db3 2022-07-09 op main_send_player(IMSG_STOP, -1, NULL, 0);
367 af27e631 2022-02-17 op main_playlist_advance();
369 af27e631 2022-02-17 op case IMSG_CTL_PREV:
370 15aecb89 2022-07-12 op control_notify(imsg.hdr.type);
371 791d3db3 2022-07-09 op main_send_player(IMSG_STOP, -1, NULL, 0);
372 af27e631 2022-02-17 op main_playlist_previous();
374 a913de21 2022-02-17 op case IMSG_CTL_JUMP:
375 a913de21 2022-02-17 op main_playlist_jump(&c->iev, &imsg);
377 9fb94242 2022-07-13 op case IMSG_CTL_MODE:
378 9fb94242 2022-07-13 op if (IMSG_DATA_SIZE(imsg) != sizeof(mode)) {
379 310ef57c 2022-02-19 op log_warnx("%s: got wrong size", __func__);
382 9fb94242 2022-07-13 op memcpy(&mode, imsg.data, sizeof(mode));
383 9881c98c 2022-07-13 op consume = new_mode(consume, mode.consume);
384 9881c98c 2022-07-13 op repeat_all = new_mode(repeat_all, mode.repeat_all);
385 9881c98c 2022-07-13 op repeat_one = new_mode(repeat_one, mode.repeat_one);
386 949b5c2e 2022-07-12 op control_notify(imsg.hdr.type);
388 19d6b480 2022-02-17 op case IMSG_CTL_BEGIN:
389 19d6b480 2022-02-17 op if (control_state.tx != -1) {
390 19d6b480 2022-02-17 op main_senderr(&c->iev, "locked");
393 c6304ab9 2022-02-17 op control_state.tx = c->iev.ibuf.fd;
394 19d6b480 2022-02-17 op imsg_compose_event(&c->iev, IMSG_CTL_BEGIN, 0, 0, -1,
397 19d6b480 2022-02-17 op case IMSG_CTL_ADD:
398 19d6b480 2022-02-17 op if (control_state.tx != -1 &&
399 c6304ab9 2022-02-17 op control_state.tx != c->iev.ibuf.fd) {
400 19d6b480 2022-02-17 op main_senderr(&c->iev, "locked");
403 19d6b480 2022-02-17 op main_enqueue(control_state.tx != -1,
404 19d6b480 2022-02-17 op &control_state.play,&c->iev, &imsg);
405 87f575c3 2022-02-21 op if (control_state.tx == -1)
406 949b5c2e 2022-07-12 op control_notify(imsg.hdr.type);
408 19d6b480 2022-02-17 op case IMSG_CTL_COMMIT:
409 c6304ab9 2022-02-17 op if (control_state.tx != c->iev.ibuf.fd) {
410 19d6b480 2022-02-17 op main_senderr(&c->iev, "locked");
413 3af93963 2022-03-02 op if (IMSG_DATA_SIZE(imsg) != sizeof(off)) {
414 3af93963 2022-03-02 op main_senderr(&c->iev, "wrong size");
417 3af93963 2022-03-02 op memcpy(&off, imsg.data, sizeof(off));
418 3af93963 2022-03-02 op playlist_swap(&control_state.play, off);
419 19d6b480 2022-02-17 op memset(&control_state.play, 0,
420 19d6b480 2022-02-17 op sizeof(control_state.play));
421 19d6b480 2022-02-17 op control_state.tx = -1;
422 19d6b480 2022-02-17 op imsg_compose_event(&c->iev, IMSG_CTL_COMMIT, 0, 0, -1,
424 949b5c2e 2022-07-12 op control_notify(imsg.hdr.type);
426 87f575c3 2022-02-21 op case IMSG_CTL_MONITOR:
427 87f575c3 2022-02-21 op c->monitor = 1;
429 791d3db3 2022-07-09 op case IMSG_CTL_SEEK:
430 15aecb89 2022-07-12 op if (IMSG_DATA_SIZE(imsg) != sizeof(seek)) {
431 791d3db3 2022-07-09 op main_senderr(&c->iev, "wrong size");
434 15aecb89 2022-07-12 op memcpy(&seek, imsg.data, sizeof(seek));
435 15aecb89 2022-07-12 op main_seek(&seek);
438 3baa2617 2022-02-16 op log_debug("%s: error handling imsg %d", __func__,
439 3baa2617 2022-02-16 op imsg.hdr.type);
442 3baa2617 2022-02-16 op imsg_free(&imsg);
445 3baa2617 2022-02-16 op imsg_event_add(&c->iev);
449 3baa2617 2022-02-16 op control_imsg_relay(struct imsg *imsg)
451 3baa2617 2022-02-16 op struct ctl_conn *c;
453 3baa2617 2022-02-16 op if ((c = control_connbypid(imsg->hdr.pid)) == NULL)
456 3baa2617 2022-02-16 op return (imsg_compose_event(&c->iev, imsg->hdr.type, 0, imsg->hdr.pid,
457 3baa2617 2022-02-16 op -1, imsg->data, IMSG_DATA_SIZE(*imsg)));