1 /*
2  * Copyright (c) 2022 Omar Polo <op@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/types.h>
18 #include <sys/queue.h>
19 #include <sys/uio.h>
20 
21 #include <limits.h>
22 
23 #include <assert.h>
24 #include <errno.h>
25 #include <event.h>
26 #include <fcntl.h>
27 #include <poll.h>
28 #include <signal.h>
29 #include <sndio.h>
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <stdint.h>
33 #include <imsg.h>
34 #include <string.h>
35 #include <syslog.h>
36 #include <unistd.h>
37 
38 #include "amused.h"
39 #include "log.h"
40 #include "xmalloc.h"
41 
42 static struct imsgbuf *ibuf;
43 
44 static int got_stop;
45 static int nextfd = -1;
46 static char nextpath[PATH_MAX];
47 
48 volatile sig_atomic_t halted;
49 
50 static void
51 player_signal_handler(int signo)
52 {
53 	halted = 1;
54 }
55 
56 static void
57 audio_init(void)
58 {
59 	struct sio_par par;
60 
61 	if ((hdl = sio_open(SIO_DEVANY, SIO_PLAY, 0)) == NULL)
62 		fatal("sio_open");
63 
64 	sio_initpar(&par);
65 	par.bits = 16;
66 	par.appbufsz = 50 * par.rate / 1000;
67 	par.pchan = 2;
68 	if (!sio_setpar(hdl, &par) || !sio_getpar(hdl, &par))
69 		fatal("couldn't set audio params");
70 	if (par.bits != 16 || par.le != SIO_LE_NATIVE)
71 		fatalx("unsupported audio params");
72 	if (!sio_start(hdl))
73 		fatal("sio_start");
74 }
75 
76 int
77 player_setup(int rate, int channels)
78 {
79 	struct sio_par par;
80 
81 	log_debug("switching to sample rate %d", rate);
82 
83 	sio_stop(hdl);
84 
85 	sio_initpar(&par);
86 	par.rate = rate;
87 	par.pchan = channels;
88 	if (!sio_setpar(hdl, &par) || !sio_getpar(hdl, &par)) {
89 		log_warnx("invalid params");
90 		return -1;
91 	}
92 
93 	if (par.pchan != channels) {
94 		log_warnx("failed to set params");
95 		return -1;
96 	}
97 
98 	/* TODO: check the sample rate? */
99 
100 	if (!sio_start(hdl)) {
101 		log_warn("sio_start");
102 		return -1;
103 	}
104 	return 0;
105 }
106 
107 int
108 player_pendingimsg(void)
109 {
110 	struct pollfd pfd;
111 	int r;
112 
113 	if (halted != 0)
114 		return 1;
115 
116 	pfd.fd = ibuf->fd;
117 	pfd.events = POLLIN;
118 
119 	r = poll(&pfd, 1, 0);
120 	if (r == -1)
121 		fatal("poll");
122 	return r;
123 }
124 
125 void
126 player_enqueue(struct imsg *imsg)
127 {
128 	size_t datalen;
129 
130 	if (nextfd != -1)
131 		fatalx("track already enqueued");
132 
133 	datalen = IMSG_DATA_SIZE(*imsg);
134 	if (datalen != sizeof(nextpath))
135 		fatalx("%s: size mismatch", __func__);
136 	memcpy(nextpath, imsg->data, sizeof(nextpath));
137 	if (nextpath[datalen-1] != '\0')
138 		fatalx("%s: corrupted path", __func__);
139 	if ((nextfd = imsg->fd) == -1)
140 		fatalx("%s: got invalid file descriptor", __func__);
141 	log_debug("enqueued %s", nextpath);
142 }
143 
144 /* process only one message */
145 int
146 player_dispatch(void)
147 {
148 	struct imsg	imsg;
149 	ssize_t		n;
150 	int		ret;
151 
152 	if (halted != 0)
153 		return IMSG_STOP;
154 
155 	if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
156 		fatalx("imsg_read");
157 	if (n == 0)
158 		fatalx("pipe closed");
159 
160 	if ((n = imsg_get(ibuf, &imsg)) == -1)
161 		fatal("imsg_get");
162 	if (n == 0) /* no more messages */
163 		fatalx("expected at least a message");
164 
165 	ret = imsg.hdr.type;
166 	if (ret == IMSG_STOP)
167 		got_stop = 1;
168 	switch (imsg.hdr.type) {
169 	case IMSG_PLAY:
170 		player_enqueue(&imsg);
171 		ret = IMSG_STOP;
172 		break;
173 	case IMSG_RESUME:
174 	case IMSG_PAUSE:
175 	case IMSG_STOP:
176 		break;
177 	default:
178 		fatalx("unknown imsg %d", imsg.hdr.type);
179 	}
180 
181 	return ret;
182 }
183 
184 void
185 player_senderr(void)
186 {
187 	imsg_compose(ibuf, IMSG_ERR, 0, 0, -1, NULL, 0);
188 	imsg_flush(ibuf);
189 }
190 
191 void
192 player_sendeof(void)
193 {
194 	imsg_compose(ibuf, IMSG_EOF, 0, 0, -1, NULL, 0);
195 	imsg_flush(ibuf);
196 }
197 
198 void
199 player_playnext(void)
200 {
201 	int fd = nextfd;
202 
203 	assert(nextfd != -1);
204 	nextfd = -1;
205 
206 	/* XXX: use magic(5) for this, not file extensions */
207 	if (strstr(nextpath, ".ogg") != NULL)
208 		play_oggvorbis(fd);
209 	else if (strstr(nextpath, ".mp3") != NULL)
210 		play_mp3(fd);
211 	else if (strstr(nextpath, ".flac") != NULL)
212 		play_flac(fd);
213 	else if (strstr(nextpath, ".opus") != NULL)
214 		play_opus(fd);
215 	else {
216 		log_warnx("unknown file type for %s", nextpath);
217 		player_senderr();
218 		return;
219 	}
220 }
221 
222 int
223 player_pause(void)
224 {
225 	int r;
226 
227 	r = player_dispatch();
228 	return r == IMSG_RESUME;
229 }
230 
231 int
232 player_shouldstop(void)
233 {
234 	if (!player_pendingimsg())
235 		return 0;
236 
237 	switch (player_dispatch()) {
238 	case IMSG_PAUSE:
239 		if (player_pause())
240 			break;
241 		/* fallthrough */
242 	case IMSG_STOP:
243 		return 1;
244 	}
245 
246 	return 0;
247 }
248 
249 int
250 player(int debug, int verbose)
251 {
252 	int flags;
253 	log_init(debug, LOG_DAEMON);
254 	log_setverbose(verbose);
255 
256 	setproctitle("player");
257 	log_procinit("player");
258 
259 #if 0
260 	{
261 		static int attached;
262 
263 		while (!attached)
264 			sleep(1);
265 	}
266 #endif
267 
268 	audio_init();
269 
270 	/* mark fd as blocking i/o mode */
271 	if ((flags = fcntl(3, F_GETFL)) == -1)
272 		fatal("fcntl(F_GETFL)");
273 	if (fcntl(3, F_SETFL, flags & ~O_NONBLOCK) == -1)
274 		fatal("fcntl F_SETFL O_NONBLOCK");
275 
276 	ibuf = xmalloc(sizeof(*ibuf));
277 	imsg_init(ibuf, 3);
278 
279 	signal(SIGINT, player_signal_handler);
280 	signal(SIGTERM, player_signal_handler);
281 
282 	signal(SIGHUP, SIG_IGN);
283 	signal(SIGPIPE, SIG_IGN);
284 
285 	if (pledge("stdio recvfd", NULL) == -1)
286 		fatal("pledge");
287 
288 	while (!halted) {
289 		while (nextfd == -1)
290 			player_dispatch();
291 
292 		player_playnext();
293 
294 		if (!got_stop)
295 			player_sendeof();
296 		else
297 			got_stop = 0;
298 	}
299 
300 	return 0;
301 }