2 * This file is part of DisOrder
3 * Copyright (C) 2005-2009 Richard Kettlewell
4 * Portions (C) 2007 Mark Wooding
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
19 /** @file server/speaker.c
20 * @brief Speaker process
22 * This program is responsible for transmitting a single coherent audio stream
23 * to its destination (over the network, to some sound API, to some
24 * subprocess). It receives connections from decoders (or rather from the
25 * process that is about to become disorder-normalize) and plays them in the
28 * @b Model. mainloop() implements a select loop awaiting commands from the
29 * main server, new connections to the speaker socket, and audio data on those
30 * connections. Each connection starts with a queue ID (with a 32-bit
31 * native-endian length word), allowing it to be referred to in commands from
34 * Data read on connections is buffered, up to a limit (currently 1Mbyte per
35 * track). No attempt is made here to limit the number of tracks, it is
36 * assumed that the main server won't start outrageously many decoders.
38 * Audio is supplied from this buffer to the uaudio play callback. Playback is
39 * enabled when a track is to be played and disabled when the its last bytes
40 * have been return by the callback; pause and resume is implemneted the
41 * obvious way. If the callback finds itself required to play when there is no
42 * playing track it returns dead air.
44 * @b Encodings. The encodings supported depend entirely on the uaudio backend
45 * chosen. See @ref uaudio.h, etc.
47 * Inbound data is expected to match @c config->sample_format. In normal use
48 * this is arranged by the @c disorder-normalize program (see @ref
49 * server/normalize.c).
51 * @b Garbage @b Collection. This program deliberately does not use the
52 * garbage collector even though it might be convenient to do so. This is for
53 * two reasons. Firstly some sound APIs use thread threads and we do not want
54 * to have to deal with potential interactions between threading and garbage
55 * collection. Secondly this process needs to be able to respond quickly and
56 * this is not compatible with the collector hanging the program even
59 * @b Units. This program thinks at various times in three different units.
60 * Bytes are obvious. A sample is a single sample on a single channel. A
61 * frame is several samples on different channels at the same point in time.
62 * So (for instance) a 16-bit stereo frame is 4 bytes and consists of a pair of
74 #include <sys/select.h>
82 #include <sys/resource.h>
85 #include "configuration.h"
90 #include "speaker-protocol.h"
96 /** @brief Maximum number of FDs to poll for */
99 /** @brief Track structure
101 * Known tracks are kept in a linked list. Usually there will be at most two
102 * of these but rearranging the queue can cause there to be more.
105 /** @brief Next track */
108 /** @brief Input file descriptor */
109 int fd
; /* input FD */
111 /** @brief Track ID */
114 /** @brief Start position of data in buffer */
117 /** @brief Number of bytes of data in buffer */
120 /** @brief Set @c fd is at EOF */
123 /** @brief Total number of samples played */
124 unsigned long long played
;
126 /** @brief Slot in @ref fds */
129 /** @brief Set when playable
131 * A track becomes playable whenever it fills its buffer or reaches EOF; it
132 * stops being playable when it entirely empties its buffer. Tracks start
133 * out life not playable.
137 /** @brief Input buffer
139 * 1Mbyte is enough for nearly 6s of 44100Hz 16-bit stereo
141 char buffer
[1048576];
144 /** @brief Lock protecting data structures
146 * This lock protects values shared between the main thread and the callback.
147 * It is needed e.g. if changing @ref playing or if modifying buffer pointers.
148 * It is not needed to add a new track, to read values only modified in the
151 static pthread_mutex_t lock
= PTHREAD_MUTEX_INITIALIZER
;
153 /** @brief Linked list of all prepared tracks */
154 static struct track
*tracks
;
156 /** @brief Playing track, or NULL
158 * This means the DESIRED playing track. It does not reflect any other state
159 * (e.g. activation of uaudio backend).
161 static struct track
*playing
;
163 /** @brief Array of file descriptors for poll() */
164 static struct pollfd fds
[NFDS
];
166 /** @brief Next free slot in @ref fds */
169 /** @brief Listen socket */
172 /** @brief Timestamp of last potential report to server */
173 static time_t last_report
;
175 /** @brief Set when paused */
178 /** @brief Set when back end activated */
179 static int activated
;
181 /** @brief Signal pipe back into the poll() loop */
182 static int sigpipe
[2];
184 /** @brief Selected backend */
185 static const struct uaudio
*backend
;
187 static const struct option options
[] = {
188 { "help", no_argument
, 0, 'h' },
189 { "version", no_argument
, 0, 'V' },
190 { "config", required_argument
, 0, 'c' },
191 { "debug", no_argument
, 0, 'd' },
192 { "no-debug", no_argument
, 0, 'D' },
193 { "syslog", no_argument
, 0, 's' },
194 { "no-syslog", no_argument
, 0, 'S' },
198 /* Display usage message and terminate. */
199 static void help(void) {
201 " disorder-speaker [OPTIONS]\n"
203 " --help, -h Display usage message\n"
204 " --version, -V Display version number\n"
205 " --config PATH, -c PATH Set configuration file\n"
206 " --debug, -d Turn on debugging\n"
207 " --[no-]syslog Force logging\n"
209 "Speaker process for DisOrder. Not intended to be run\n"
215 /** @brief Find track @p id, maybe creating it if not found
216 * @param id Track ID to find
217 * @param create If nonzero, create track structure of @p id not found
218 * @return Pointer to track structure or NULL
220 static struct track
*findtrack(const char *id
, int create
) {
223 D(("findtrack %s %d", id
, create
));
224 for(t
= tracks
; t
&& strcmp(id
, t
->id
); t
= t
->next
)
227 t
= xmalloc(sizeof *t
);
236 /** @brief Remove track @p id (but do not destroy it)
237 * @param id Track ID to remove
238 * @return Track structure or NULL if not found
240 static struct track
*removetrack(const char *id
) {
241 struct track
*t
, **tt
;
243 D(("removetrack %s", id
));
244 for(tt
= &tracks
; (t
= *tt
) && strcmp(id
, t
->id
); tt
= &t
->next
)
251 /** @brief Destroy a track
252 * @param t Track structure
254 static void destroy(struct track
*t
) {
255 D(("destroy %s", t
->id
));
261 /** @brief Read data into a sample buffer
262 * @param t Pointer to track
263 * @return 0 on success, -1 on EOF
265 * This is effectively the read callback on @c t->fd. It is called from the
266 * main loop whenever the track's file descriptor is readable, assuming the
267 * buffer has not reached the maximum allowed occupancy.
269 static int speaker_fill(struct track
*t
) {
273 D(("fill %s: eof=%d used=%zu",
274 t
->id
, t
->eof
, t
->used
));
277 pthread_mutex_lock(&lock
);
278 if(t
->used
< sizeof t
->buffer
) {
279 /* there is room left in the buffer */
280 where
= (t
->start
+ t
->used
) % sizeof t
->buffer
;
281 /* Get as much data as we can */
282 if(where
>= t
->start
)
283 left
= (sizeof t
->buffer
) - where
;
285 left
= t
->start
- where
;
286 pthread_mutex_unlock(&lock
);
288 n
= read(t
->fd
, t
->buffer
+ where
, left
);
289 } while(n
< 0 && errno
== EINTR
);
290 pthread_mutex_lock(&lock
);
293 fatal(errno
, "error reading sample stream");
296 D(("fill %s: eof detected", t
->id
));
298 /* A track always becomes playable at EOF; we're not going to see any
304 /* A track becomes playable when it (first) fills its buffer. For
305 * 44.1KHz 16-bit stereo this is ~6s of audio data. The latency will
306 * depend how long that takes to decode (hopefuly not very!) */
307 if(t
->used
== sizeof t
->buffer
)
312 pthread_mutex_unlock(&lock
);
316 /** @brief Return nonzero if we want to play some audio
318 * We want to play audio if there is a current track; and it is not paused; and
319 * it is playable according to the rules for @ref track::playable.
321 static int playable(void) {
324 && playing
->playable
;
327 /** @brief Notify the server what we're up to */
328 static void report(void) {
329 struct speaker_message sm
;
332 memset(&sm
, 0, sizeof sm
);
333 sm
.type
= paused ? SM_PAUSED
: SM_PLAYING
;
334 strcpy(sm
.id
, playing
->id
);
335 pthread_mutex_lock(&lock
);
336 sm
.data
= playing
->played
/ (uaudio_rate
* uaudio_channels
);
337 pthread_mutex_unlock(&lock
);
338 speaker_send(1, &sm
);
343 /** @brief Add a file descriptor to the set to poll() for
344 * @param fd File descriptor
345 * @param events Events to wait for e.g. @c POLLIN
346 * @return Slot number
348 static int addfd(int fd
, int events
) {
351 fds
[fdno
].events
= events
;
357 /** @brief Callback to return some sampled data
358 * @param buffer Where to put sample data
359 * @param max_samples How many samples to return
360 * @param userdata User data
361 * @return Number of samples written
363 * See uaudio_callback().
365 static size_t speaker_callback(void *buffer
,
367 void attribute((unused
)) *userdata
) {
368 const size_t max_bytes
= max_samples
* uaudio_sample_size
;
369 size_t provided_samples
= 0;
371 pthread_mutex_lock(&lock
);
372 /* TODO perhaps we should immediately go silent if we've been asked to pause
373 * or cancel the playing track (maybe block in the cancel case and see what
376 if(playing
->used
> 0) {
378 /* Compute size of largest contiguous chunk. We get called as often as
379 * necessary so there's no need for cleverness here. */
380 if(playing
->start
+ playing
->used
> sizeof playing
->buffer
)
381 bytes
= sizeof playing
->buffer
- playing
->start
;
383 bytes
= playing
->used
;
384 /* Limit to what we were asked for */
385 if(bytes
> max_bytes
)
388 memcpy(buffer
, playing
->buffer
+ playing
->start
, bytes
);
389 playing
->start
+= bytes
;
390 playing
->used
-= bytes
;
391 /* Wrap around to start of buffer */
392 if(playing
->start
== sizeof playing
->buffer
)
394 /* See if we've reached the end of the track */
395 if(playing
->used
== 0 && playing
->eof
)
396 write(sigpipe
[1], "", 1);
397 provided_samples
= bytes
/ uaudio_sample_size
;
398 playing
->played
+= provided_samples
;
401 /* If we couldn't provide anything at all, play dead air */
402 /* TODO maybe it would be better to block, in some cases? */
403 if(!provided_samples
) {
404 memset(buffer
, 0, max_bytes
);
405 provided_samples
= max_samples
;
407 pthread_mutex_unlock(&lock
);
408 return provided_samples
;
411 /** @brief Main event loop */
412 static void mainloop(void) {
414 struct speaker_message sm
;
415 int n
, fd
, stdin_slot
, timeout
, listen_slot
, sigpipe_slot
;
417 /* Keep going while our parent process is alive */
418 while(getppid() != 1) {
419 int force_report
= 0;
422 /* By default we will wait up to a second before thinking about current
425 /* Always ready for commands from the main server. */
426 stdin_slot
= addfd(0, POLLIN
);
427 /* Also always ready for inbound connections */
428 listen_slot
= addfd(listenfd
, POLLIN
);
429 /* Try to read sample data for the currently playing track if there is
434 && playing
->used
< (sizeof playing
->buffer
))
435 playing
->slot
= addfd(playing
->fd
, POLLIN
);
438 /* If any other tracks don't have a full buffer, try to read sample data
439 * from them. We do this last of all, so that if we run out of slots,
440 * nothing important can't be monitored. */
441 for(t
= tracks
; t
; t
= t
->next
)
445 && t
->used
< sizeof t
->buffer
) {
446 t
->slot
= addfd(t
->fd
, POLLIN
| POLLHUP
);
450 sigpipe_slot
= addfd(sigpipe
[1], POLLIN
);
451 /* Wait for something interesting to happen */
452 n
= poll(fds
, fdno
, timeout
);
454 if(errno
== EINTR
) continue;
455 fatal(errno
, "error calling poll");
457 /* Perhaps a connection has arrived */
458 if(fds
[listen_slot
].revents
& POLLIN
) {
459 struct sockaddr_un addr
;
460 socklen_t addrlen
= sizeof addr
;
464 if((fd
= accept(listenfd
, (struct sockaddr
*)&addr
, &addrlen
)) >= 0) {
466 if(read(fd
, &l
, sizeof l
) < 4) {
467 error(errno
, "reading length from inbound connection");
469 } else if(l
>= sizeof id
) {
470 error(0, "id length too long");
472 } else if(read(fd
, id
, l
) < (ssize_t
)l
) {
473 error(errno
, "reading id from inbound connection");
477 D(("id %s fd %d", id
, fd
));
478 t
= findtrack(id
, 1/*create*/);
479 if (write(fd
, "", 1) < 0) /* write an ack */
480 error(errno
, "writing ack to inbound connection");
482 error(0, "%s: already got a connection", id
);
486 t
->fd
= fd
; /* yay */
490 error(errno
, "accept");
492 /* Perhaps we have a command to process */
493 if(fds
[stdin_slot
].revents
& POLLIN
) {
494 /* There might (in theory) be several commands queued up, but in general
495 * this won't be the case, so we don't bother looping around to pick them
497 n
= speaker_recv(0, &sm
);
503 fatal(0, "got SM_PLAY but already playing something");
504 t
= findtrack(sm
.id
, 1);
505 D(("SM_PLAY %s fd %d", t
->id
, t
->fd
));
507 error(0, "cannot play track because no connection arrived");
522 D(("SM_CANCEL %s", sm
.id
));
523 t
= removetrack(sm
.id
);
525 pthread_mutex_lock(&lock
);
527 /* scratching the playing track */
528 sm
.type
= SM_FINISHED
;
531 /* Could be scratching the playing track before it's quite got
532 * going, or could be just removing a track from the queue. We
533 * log more because there's been a bug here recently than because
534 * it's particularly interesting; the log message will be removed
535 * if no further problems show up. */
536 info("SM_CANCEL for nonplaying track %s", sm
.id
);
537 sm
.type
= SM_STILLBORN
;
539 strcpy(sm
.id
, t
->id
);
541 pthread_mutex_unlock(&lock
);
543 /* Probably scratching the playing track well before it's got
544 * going, but could indicate a bug, so we log this as an error. */
545 sm
.type
= SM_UNKNOWN
;
546 error(0, "SM_CANCEL for unknown track %s", sm
.id
);
548 speaker_send(1, &sm
);
554 error(0, "cannot read configuration");
555 info("reloaded configuration");
558 error(0, "unknown message type %d", sm
.type
);
561 /* Read in any buffered data */
562 for(t
= tracks
; t
; t
= t
->next
)
565 && (fds
[t
->slot
].revents
& (POLLIN
| POLLHUP
)))
567 /* Drain the signal pipe. We don't care about its contents, merely that it
568 * interrupted poll(). */
569 if(fds
[sigpipe_slot
].revents
& POLLIN
) {
572 read(sigpipe
[0], buffer
, sizeof buffer
);
574 if(playing
&& playing
->used
== 0 && playing
->eof
) {
575 /* The playing track is done. Tell the server, and destroy it. */
576 memset(&sm
, 0, sizeof sm
);
577 sm
.type
= SM_FINISHED
;
578 strcpy(sm
.id
, playing
->id
);
579 speaker_send(1, &sm
);
580 removetrack(playing
->id
);
581 pthread_mutex_lock(&lock
);
584 pthread_mutex_unlock(&lock
);
585 /* The server will presumalby send as an SM_PLAY by return */
587 /* Impose any state change required by the above */
596 backend
->deactivate();
599 /* If we've not reported our state for a second do so now. */
600 if(force_report
|| time(0) > last_report
)
605 int main(int argc
, char **argv
) {
606 int n
, logsyslog
= !isatty(2);
607 struct sockaddr_un addr
;
608 static const int one
= 1;
609 struct speaker_message sm
;
615 if(!setlocale(LC_CTYPE
, "")) fatal(errno
, "error calling setlocale");
616 while((n
= getopt_long(argc
, argv
, "hVc:dDSs", options
, 0)) >= 0) {
619 case 'V': version("disorder-speaker");
620 case 'c': configfile
= optarg
; break;
621 case 'd': debugging
= 1; break;
622 case 'D': debugging
= 0; break;
623 case 'S': logsyslog
= 0; break;
624 case 's': logsyslog
= 1; break;
625 default: fatal(0, "invalid option");
628 if((d
= getenv("DISORDER_DEBUG_SPEAKER"))) debugging
= atoi(d
);
630 openlog(progname
, LOG_PID
, LOG_DAEMON
);
631 log_default
= &log_syslog
;
633 config_uaudio_apis
= uaudio_apis
;
634 if(config_read(1)) fatal(0, "cannot read configuration");
636 signal(SIGPIPE
, SIG_IGN
);
638 xnice(config
->nice_speaker
);
641 /* make sure we're not root, whatever the config says */
642 if(getuid() == 0 || geteuid() == 0)
643 fatal(0, "do not run as root");
644 /* Make sure we can't have more than NFDS files open (it would bust our
646 if(getrlimit(RLIMIT_NOFILE
, rl
) < 0)
647 fatal(errno
, "getrlimit RLIMIT_NOFILE");
648 if(rl
->rlim_cur
> NFDS
) {
650 if(setrlimit(RLIMIT_NOFILE
, rl
) < 0)
651 fatal(errno
, "setrlimit to reduce RLIMIT_NOFILE to %lu",
652 (unsigned long)rl
->rlim_cur
);
653 info("set RLIM_NOFILE to %lu", (unsigned long)rl
->rlim_cur
);
655 info("RLIM_NOFILE is %lu", (unsigned long)rl
->rlim_cur
);
656 /* gcrypt initialization */
657 if(!gcry_check_version(NULL
))
658 disorder_fatal(0, "gcry_check_version failed");
659 gcry_control(GCRYCTL_INIT_SECMEM
, 0);
660 gcry_control (GCRYCTL_INITIALIZATION_FINISHED
, 0);
661 /* create a pipe between the backend callback and the poll() loop */
663 nonblock(sigpipe
[0]);
664 /* set up audio backend */
665 uaudio_set_format(config
->sample_format
.rate
,
666 config
->sample_format
.channels
,
667 config
->sample_format
.bits
,
668 config
->sample_format
.bits
!= 8);
669 /* TODO other parameters! */
670 backend
= uaudio_find(config
->api
);
671 /* backend-specific initialization */
672 backend
->start(speaker_callback
, NULL
);
673 /* create the socket directory */
674 byte_xasprintf(&dir
, "%s/speaker", config
->home
);
675 unlink(dir
); /* might be a leftover socket */
676 if(mkdir(dir
, 0700) < 0 && errno
!= EEXIST
)
677 fatal(errno
, "error creating %s", dir
);
678 /* set up the listen socket */
679 listenfd
= xsocket(PF_UNIX
, SOCK_STREAM
, 0);
680 memset(&addr
, 0, sizeof addr
);
681 addr
.sun_family
= AF_UNIX
;
682 snprintf(addr
.sun_path
, sizeof addr
.sun_path
, "%s/speaker/socket",
684 if(unlink(addr
.sun_path
) < 0 && errno
!= ENOENT
)
685 error(errno
, "removing %s", addr
.sun_path
);
686 xsetsockopt(listenfd
, SOL_SOCKET
, SO_REUSEADDR
, &one
, sizeof one
);
687 if(bind(listenfd
, (const struct sockaddr
*)&addr
, sizeof addr
) < 0)
688 fatal(errno
, "error binding socket to %s", addr
.sun_path
);
689 xlisten(listenfd
, 128);
691 info("listening on %s", addr
.sun_path
);
692 memset(&sm
, 0, sizeof sm
);
694 speaker_send(1, &sm
);
696 info("stopped (parent terminated)");