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 <poll.h>
27 #include <signal.h>
28 #include <sndio.h>
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <stdint.h>
32 #include <imsg.h>
33 #include <string.h>
34 #include <syslog.h>
35 #include <unistd.h>
36 
37 #include "amused.h"
38 #include "log.h"
39 #include "xmalloc.h"
40 
41 struct sio_hdl		*hdl;
42 struct sio_par		 par;
43 struct pollfd		*player_pfds;
44 static struct imsgbuf	*ibuf;
45 
46 static int stopped = 1;
47 static int nextfd = -1;
48 static int64_t samples;
49 
50 volatile sig_atomic_t halted;
51 
52 static void
53 player_signal_handler(int signo)
54 {
55 	halted = 1;
56 }
57 
58 int
59 player_setup(unsigned int bits, unsigned int rate, unsigned int channels)
60 {
61 	int nfds, fpct;
62 
63 	log_debug("%s: bits=%d, rate=%d, channels=%d", __func__,
64 	    bits, rate, channels);
65 
66 	fpct = (rate*5)/100;
67 
68 	/* don't stop if the parameters are the same */
69 	if (bits == par.bits && channels == par.pchan &&
70 	    par.rate - fpct <= rate && rate <= par.rate + fpct) {
71 		if (stopped)
72 			goto start;
73 		return 0;
74 	}
75 
76 again:
77 	if (!stopped) {
78 		sio_stop(hdl);
79 		stopped = 1;
80 	}
81 
82 	sio_initpar(&par);
83 	par.bits = bits;
84 	par.rate = rate;
85 	par.pchan = channels;
86 	if (!sio_setpar(hdl, &par)) {
87 		if (errno == EAGAIN) {
88 			nfds = sio_pollfd(hdl, player_pfds + 1, POLLOUT);
89 			if (poll(player_pfds + 1, nfds, INFTIM) == -1)
90 				fatal("poll");
91 			goto again;
92 		}
93 		log_warnx("invalid params (bits=%d, rate=%d, channels=%d",
94 		    bits, rate, channels);
95 		return -1;
96 	}
97 	if (!sio_getpar(hdl, &par)) {
98 		log_warnx("can't get params");
99 		return -1;
100 	}
101 
102 	if (par.bits != bits || par.pchan != channels) {
103 		log_warnx("failed to set params");
104 		return -1;
105 	}
106 
107 	/* TODO: check the sample rate? */
108 
109 start:
110 	if (!sio_start(hdl)) {
111 		log_warn("sio_start");
112 		return -1;
113 	}
114 	stopped = 0;
115 	return 0;
116 }
117 
118 void
119 player_setduration(int64_t duration)
120 {
121 	int64_t seconds;
122 
123 	seconds = duration / par.rate;
124 	imsg_compose(ibuf, IMSG_LEN, 0, 0, -1, &seconds, sizeof(seconds));
125 	imsg_flush(ibuf);
126 }
127 
128 static void
129 player_onmove(void *arg, int delta)
130 {
131 	static int64_t reported;
132 	int64_t sec;
133 
134 	samples += delta;
135 	if (llabs(samples - reported) >= par.rate) {
136 		reported = samples;
137 		sec = samples / par.rate;
138 
139 		imsg_compose(ibuf, IMSG_POS, 0, 0, -1, &sec, sizeof(sec));
140 		imsg_flush(ibuf);
141 	}
142 }
143 
144 void
145 player_setpos(int64_t pos)
146 {
147 	samples = pos;
148 	player_onmove(NULL, 0);
149 }
150 
151 /* process only one message */
152 static int
153 player_dispatch(int64_t *s)
154 {
155 	struct player_seek seek;
156 	struct pollfd	pfd;
157 	struct imsg	imsg;
158 	ssize_t		n;
159 	int		ret;
160 
161 	if (halted != 0)
162 		return IMSG_STOP;
163 
164 again:
165 	if ((n = imsg_get(ibuf, &imsg)) == -1)
166 		fatal("imsg_get");
167 	if (n == 0) {
168 		pfd.fd = ibuf->fd;
169 		pfd.events = POLLIN;
170 		if (poll(&pfd, 1, INFTIM) == -1)
171 			fatal("poll");
172 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
173 			fatal("imsg_read");
174 		if (n == 0)
175 			fatalx("pipe closed");
176 		goto again;
177 	}
178 
179 	ret = imsg.hdr.type;
180 	switch (imsg.hdr.type) {
181 	case IMSG_PLAY:
182 		if (nextfd != -1)
183 			fatalx("track already enqueued");
184 		if ((nextfd = imsg.fd) == -1)
185 			fatalx("%s: got invalid file descriptor", __func__);
186 		log_debug("song enqueued");
187 		ret = IMSG_STOP;
188 		break;
189 	case IMSG_RESUME:
190 	case IMSG_PAUSE:
191 	case IMSG_STOP:
192 		break;
193 	case IMSG_CTL_SEEK:
194 		if (s == NULL)
195 			break;
196 		if (IMSG_DATA_SIZE(imsg) != sizeof(seek))
197 			fatalx("wrong size for seek ctl");
198 		memcpy(&seek, imsg.data, sizeof(seek));
199 		log_debug("got to seek: {%lld, %d}", seek.offset,
200 		    seek.relative);
201 		*s = seek.offset * par.rate;
202 		if (seek.relative)
203 			*s += samples;
204 		if (*s < 0)
205 			*s = -1;
206 		break;
207 	default:
208 		fatalx("unknown imsg %d", imsg.hdr.type);
209 	}
210 
211 	imsg_free(&imsg);
212 	return ret;
213 }
214 
215 static void
216 player_senderr(const char *errstr)
217 {
218 	size_t len = 0;
219 
220 	if (errstr != NULL)
221 		len = strlen(errstr) + 1;
222 
223 	imsg_compose(ibuf, IMSG_ERR, 0, 0, -1, errstr, len);
224 	imsg_flush(ibuf);
225 }
226 
227 static void
228 player_sendeof(void)
229 {
230 	imsg_compose(ibuf, IMSG_EOF, 0, 0, -1, NULL, 0);
231 	imsg_flush(ibuf);
232 }
233 
234 static int
235 player_playnext(const char **errstr)
236 {
237 	static char buf[512];
238 	ssize_t r;
239 	int fd = nextfd;
240 
241 	assert(nextfd != -1);
242 	nextfd = -1;
243 
244 	/* reset samples and set position to zero */
245 	samples = 0;
246 	imsg_compose(ibuf, IMSG_POS, 0, 0, -1, &samples, sizeof(samples));
247 	imsg_flush(ibuf);
248 
249 	r = read(fd, buf, sizeof(buf));
250 
251 	/* 8 byte is the larger magic number */
252 	if (r < 8) {
253 		*errstr = "read failed";
254 		goto err;
255 	}
256 
257 	if (lseek(fd, 0, SEEK_SET) == -1) {
258 		*errstr = "lseek failed";
259 		goto err;
260 	}
261 
262 	if (memcmp(buf, "fLaC", 4) == 0)
263 		return play_flac(fd, errstr);
264 	if (memcmp(buf, "ID3", 3) == 0 ||
265 	    memcmp(buf, "\xFF\xFB", 2) == 0)
266 		return play_mp3(fd, errstr);
267 	if (memmem(buf, r, "OpusHead", 8) != NULL)
268 		return play_opus(fd, errstr);
269 	if (memmem(buf, r, "OggS", 4) != NULL)
270 		return play_oggvorbis(fd, errstr);
271 
272 	*errstr = "unknown file type";
273 err:
274 	close(fd);
275 	return -1;
276 }
277 
278 static int
279 player_pause(int64_t *s)
280 {
281 	int r;
282 
283 	r = player_dispatch(s);
284 	return r == IMSG_RESUME;
285 }
286 
287 static int
288 player_shouldstop(int64_t *s)
289 {
290 	switch (player_dispatch(s)) {
291 	case IMSG_PAUSE:
292 		if (player_pause(s))
293 			break;
294 		/* fallthrough */
295 	case IMSG_STOP:
296 		return 1;
297 	}
298 
299 	return 0;
300 }
301 
302 int
303 play(const void *buf, size_t len, int64_t *s)
304 {
305 	size_t w;
306 	int nfds, revents, r;
307 
308 	*s = -1;
309 	while (len != 0) {
310 		nfds = sio_pollfd(hdl, player_pfds + 1, POLLOUT);
311 		r = poll(player_pfds, nfds + 1, INFTIM);
312 		if (r == -1)
313 			fatal("poll");
314 
315 		if (player_pfds[0].revents & (POLLHUP|POLLIN)) {
316 			if (player_shouldstop(s)) {
317 				sio_flush(hdl);
318 				stopped = 1;
319 				return 0;
320 			}
321 		}
322 
323 		revents = sio_revents(hdl, player_pfds + 1);
324 		if (revents & POLLHUP)
325 			fatalx("sndio hang-up");
326 		if (revents & POLLOUT) {
327 			w = sio_write(hdl, buf, len);
328 			len -= w;
329 			buf += w;
330 		}
331 	}
332 
333 	return 1;
334 }
335 
336 int
337 player(int debug, int verbose)
338 {
339 	int r;
340 
341 	log_init(debug, LOG_DAEMON);
342 	log_setverbose(verbose);
343 
344 	setproctitle("player");
345 	log_procinit("player");
346 
347 #if 0
348 	{
349 		static int attached;
350 
351 		while (!attached)
352 			sleep(1);
353 	}
354 #endif
355 
356 	if ((hdl = sio_open(SIO_DEVANY, SIO_PLAY, 1)) == NULL)
357 		fatal("sio_open");
358 
359 	sio_onmove(hdl, player_onmove, NULL);
360 
361 	/* allocate one extra for imsg */
362 	player_pfds = calloc(sio_nfds(hdl) + 1, sizeof(*player_pfds));
363 	if (player_pfds == NULL)
364 		fatal("calloc");
365 
366 	player_pfds[0].events = POLLIN;
367 	player_pfds[0].fd = 3;
368 
369 	ibuf = xmalloc(sizeof(*ibuf));
370 	imsg_init(ibuf, 3);
371 
372 	signal(SIGINT, player_signal_handler);
373 	signal(SIGTERM, player_signal_handler);
374 
375 	signal(SIGHUP, SIG_IGN);
376 	signal(SIGPIPE, SIG_IGN);
377 
378 	if (pledge("stdio recvfd audio", NULL) == -1)
379 		fatal("pledge");
380 
381 	while (!halted) {
382 		const char *errstr = NULL;
383 
384 		while (nextfd == -1)
385 			player_dispatch(NULL);
386 
387 		r = player_playnext(&errstr);
388 		if (r == -1)
389 			player_senderr(errstr);
390 		if (r == 0)
391 			player_sendeof();
392 	}
393 
394 	return 0;
395 }