Fix stream grabbing.
[paraslash.git] / audiod.c
1 /*
2 * Copyright (C) 2005-2009 Andre Noll <maan@systemlinux.org>
3 *
4 * Licensed under the GPL v2. For licencing details see COPYING.
5 */
6
7 /** \file audiod.c the paraslash's audio daemon */
8 #include <regex.h>
9 #include <sys/types.h>
10 #include <dirent.h>
11 #include <signal.h>
12 #include <openssl/rc4.h>
13 #include <stdbool.h>
14
15 #include "para.h"
16 #include "error.h"
17 #include "crypt.h"
18 #include "audiod.cmdline.h"
19 #include "list.h"
20 #include "sched.h"
21 #include "ggo.h"
22 #include "recv.h"
23 #include "buffer_tree.h"
24 #include "filter.h"
25 #include "grab_client.h"
26 #include "client.cmdline.h"
27 #include "client.h"
28 #include "audiod.h"
29 #include "net.h"
30 #include "daemon.h"
31 #include "string.h"
32 #include "fd.h"
33 #include "write.h"
34 #include "write_common.h"
35 #include "signal.h"
36
37 /** define the array of error lists needed by para_audiod */
38 INIT_AUDIOD_ERRLISTS;
39 /** define the array containing all supported audio formats */
40 const char *audio_formats[] = {AUDIOD_AUDIO_FORMAT_ARRAY NULL};
41
42 /** Defines how audiod handles one supported audio format. */
43 struct audio_format_info {
44 /** pointer to the receiver for this audio format */
45 struct receiver *receiver;
46 /** the receiver configuration */
47 void *receiver_conf;
48 /** the number of filters that should be activated for this audio format */
49 unsigned int num_filters;
50 /** Array of filter numbers to be activated. */
51 unsigned *filter_nums;
52 /** Pointer to the array of filter configurations. */
53 void **filter_conf;
54 /** the number of filters that should be activated for this audio format */
55 unsigned int num_writers;
56 /** Array of writer numbers to be activated. */
57 int *writer_nums;
58 /** pointer to the array of writer configurations */
59 void **writer_conf;
60 /** do not start receiver/filters/writer before this time */
61 struct timeval restart_barrier;
62 };
63
64 /**
65 * para_audiod uses \p MAX_STREAM_SLOTS different slots, each of which may
66 * be associated with a receiver/filter/writer triple. This array holds all
67 * information on the status of these slots.
68 *
69 * \sa struct slot_info
70 * */
71 struct slot_info slot[MAX_STREAM_SLOTS];
72
73 /** The vss status flags audiod is interested in. */
74 enum vss_status_flags {
75 /** Whether the 'N' flag is set. */
76 VSS_STATUS_FLAG_NEXT = 1,
77 /** The 'P' flag is set. */
78 VSS_STATUS_FLAG_PLAYING = 2,
79 };
80
81 /**
82 * The task for obtaining para_server's status (para_client stat).
83 *
84 * \sa struct task, struct sched.
85 */
86 struct status_task {
87 /** The associated task structure of audiod. */
88 struct task task;
89 /** Client data associated with the stat task. */
90 struct client_task *ct;
91 /** Do not restart client command until this time. */
92 struct timeval restart_barrier;
93 /** Last time we received status data from para_server. */
94 struct timeval last_status_read;
95 /** The offset value announced by para_server. */
96 int offset_seconds;
97 /** The length of the current audio file as announced by para_server. */
98 int length_seconds;
99 /** The start of the current stream from the view of para_server. */
100 struct timeval server_stream_start;
101 /** The average time deviation between para_server and para_audiod. */
102 struct timeval sa_time_diff;
103 /** Whether client time is ahead of server time. */
104 int sa_time_diff_sign;
105 /** The 'P' and the 'N' flags as announced by para_server. */
106 enum vss_status_flags vss_status;
107 /** Number of times the clock difference is to be checked. */
108 unsigned clock_diff_count;
109 /** When to start the next check for clock difference. */
110 struct timeval clock_diff_barrier;
111 /** Number of the audio format as announced by para_server. */
112 int current_audio_format_num;
113 };
114
115 /** The array of status items sent by para_server. */
116 char *stat_item_values[NUM_STAT_ITEMS] = {NULL};
117
118 /**
119 * the current mode of operation of which can be changed by the on/off/cycle
120 * commands. It is either, AUDIOD_OFF, AUDIOD_ON or AUDIOD_STANDBY.
121 */
122 int audiod_status = AUDIOD_ON;
123
124 /**
125 * the gengetopt args_info struct that holds information on all command line
126 * arguments
127 */
128 struct audiod_args_info conf;
129
130 static char *socket_name;
131 static struct audio_format_info afi[NUM_AUDIO_FORMATS];
132
133 static struct signal_task signal_task_struct, *sig_task = &signal_task_struct;
134
135 static struct status_task status_task_struct;
136
137 /**
138 * the task that calls the status command of para_server
139 *
140 * \sa struct status_task
141 */
142 static struct status_task *stat_task = &status_task_struct;
143 static struct timeval initial_delay_barrier;
144
145 /**
146 * the task for handling audiod commands
147 *
148 * \sa struct task, struct sched
149 */
150 struct command_task {
151 /** the local listening socket */
152 int fd;
153 /** the associated task structure */
154 struct task task;
155 };
156
157 /** iterate over all supported audio formats */
158 #define FOR_EACH_AUDIO_FORMAT(af) for (af = 0; af < NUM_AUDIO_FORMATS; af++)
159
160 /**
161 * get the audio format number
162 * \param name the name of the audio format
163 *
164 * \return The audio format number on success, -E_UNSUPPORTED_AUDIO_FORMAT if
165 * \a name is not a supported audio format.
166 */
167 int get_audio_format_num(const char *name)
168 {
169 int i;
170
171 while (para_isspace(*name))
172 name++;
173 FOR_EACH_AUDIO_FORMAT(i)
174 if (!strcmp(name, audio_formats[i]))
175 return i;
176 return -E_UNSUPPORTED_AUDIO_FORMAT;
177 }
178
179 char *get_time_string(int slot_num)
180 {
181 int ret, seconds = 0, length;
182 struct timeval *tmp, sum, sss, /* server stream start */
183 rstime, /* receiver start time */
184 wstime, /* writer start time */
185 wtime, /* now - writer start */
186 rskip; /* receiver start - sss */
187 struct slot_info *s = slot_num < 0? NULL : &slot[slot_num];
188 char *msg;
189
190 if (audiod_status == AUDIOD_OFF)
191 goto empty;
192 if (!(stat_task->vss_status & VSS_STATUS_FLAG_PLAYING)) {
193 if (stat_task->length_seconds) /* paused */
194 return NULL;
195 goto empty; /* stopped */
196 }
197 if (audiod_status == AUDIOD_ON && !s)
198 goto empty;
199 /* valid status items and playing */
200 if (s && s->wns) { /* writer active in this slot */
201 length = s->seconds_total;
202 tmp = &s->server_stream_start;
203 } else { /* standby mode, rely on status items */
204 length = stat_task->length_seconds;
205 tmp = &stat_task->server_stream_start;
206 }
207 if (stat_task->sa_time_diff_sign > 0)
208 tv_diff(tmp, &stat_task->sa_time_diff, &sss);
209 else
210 tv_add(tmp, &stat_task->sa_time_diff, &sss);
211 if (!s || !s->wns) {
212 struct timeval diff;
213 tv_diff(now, &sss, &diff);
214 seconds = diff.tv_sec + stat_task->offset_seconds;
215 goto out;
216 }
217 btr_get_node_start(s->wns[0].btrn, &wstime);
218 tv_diff(now, &wstime, &wtime);
219 //PARA_CRIT_LOG("offset %d\n", s->offset_seconds);
220 seconds = s->offset_seconds;
221 btr_get_node_start(s->receiver_node->btrn, &rstime);
222 ret = tv_diff(&rstime, &sss, &rskip);
223 if (ret > 0) { /* audiod was started in the middle of the stream */
224 tv_add(&wtime, &rskip, &sum);
225 seconds += sum.tv_sec;
226 } else
227 seconds += wtime.tv_sec;
228 out:
229 seconds = PARA_MIN(seconds, length);
230 seconds = PARA_MAX(seconds, 0);
231 msg = make_message(
232 "%s%d:%02d [%d:%02d] (%d%%/%d:%02d)",
233 s? "" : "~",
234 seconds / 60,
235 seconds % 60,
236 (length - seconds) / 60,
237 (length - seconds) % 60,
238 length? (seconds * 100 + length / 2) / length : 0,
239 length / 60,
240 length % 60
241 );
242 PARA_DEBUG_LOG("slot %d: %s\n", slot_num, msg);
243 return msg;
244 empty:
245 return para_strdup(NULL);
246 }
247
248 static int want_colors(void)
249 {
250 if (conf.color_arg == color_arg_no)
251 return 0;
252 if (conf.color_arg == color_arg_yes)
253 return 1;
254 if (conf.logfile_given)
255 return 0;
256 return isatty(STDERR_FILENO);
257 }
258
259 static void parse_config_or_die(void)
260 {
261 int ret;
262 char *config_file;
263 struct audiod_cmdline_parser_params params = {
264 .override = 0,
265 .initialize = 0,
266 .check_required = 1,
267 .check_ambiguity = 0,
268 .print_errors = 1
269 };
270
271 if (conf.config_file_given)
272 config_file = para_strdup(conf.config_file_arg);
273 else {
274 char *home = para_homedir();
275 config_file = make_message("%s/.paraslash/audiod.conf", home);
276 free(home);
277 }
278 ret = file_exists(config_file);
279 if (conf.config_file_given && !ret) {
280 PARA_EMERG_LOG("can not read config file %s\n", config_file);
281 goto err;
282 }
283 if (ret)
284 audiod_cmdline_parser_config_file(config_file, &conf, &params);
285 free(config_file);
286 daemon_set_loglevel(conf.loglevel_arg);
287 return;
288 err:
289 free(config_file);
290 exit(EXIT_FAILURE);
291 }
292
293 static void setup_signal_handling(void)
294 {
295 sig_task->fd = para_signal_init();
296 PARA_INFO_LOG("signal pipe: fd %d\n", sig_task->fd);
297 para_install_sighandler(SIGINT);
298 para_install_sighandler(SIGTERM);
299 para_install_sighandler(SIGHUP);
300 para_sigaction(SIGPIPE, SIG_IGN);
301 }
302
303 static void clear_slot(int slot_num)
304 {
305 struct slot_info *s = &slot[slot_num];
306
307 PARA_INFO_LOG("clearing slot %d\n", slot_num);
308 memset(s, 0, sizeof(struct slot_info));
309 s->format = -1;
310 }
311
312 static void close_receiver(int slot_num)
313 {
314 struct slot_info *s = &slot[slot_num];
315 struct audio_format_info *a;
316
317 if (s->format < 0 || !s->receiver_node)
318 return;
319 a = &afi[s->format];
320 PARA_NOTICE_LOG("closing %s receiver in slot %d\n",
321 audio_formats[s->format], slot_num);
322 a->receiver->close(s->receiver_node);
323 btr_free_node(s->receiver_node->btrn);
324 free(s->receiver_node);
325 s->receiver_node = NULL;
326 stat_task->current_audio_format_num = -1;
327 }
328
329 static void writer_cleanup(struct writer_node *wn)
330 {
331 struct writer *w = writers + wn->writer_num;
332
333 w->close(wn);
334 btr_free_node(wn->btrn);
335 }
336
337 static void close_writers(struct slot_info *s)
338 {
339 struct audio_format_info *a;
340 int i;
341
342 if (s->format < 0)
343 return;
344 a = afi + s->format;
345 if (a->num_writers == 0)
346 writer_cleanup(s->wns);
347 else {
348 for (i = 0; i < a->num_writers; i++)
349 writer_cleanup(s->wns + i);
350 }
351 free(s->wns);
352 s->wns = NULL;
353 }
354
355 static void _close_filters(struct slot_info *s)
356 {
357 int i;
358 struct audio_format_info *a = afi + s->format;
359 if (a->num_filters == 0)
360 return;
361 for (i = 0; i < a->num_filters; i++) {
362 struct filter_node *fn = s->fns + i;
363 struct filter *f = filters + fn->filter_num;
364
365 f->close(fn);
366 btr_free_node(fn->btrn);
367 }
368 free(s->fns);
369 s->fns = NULL;
370 }
371
372 static void kill_all_decoders(int error)
373 {
374 int i, j;
375
376 FOR_EACH_SLOT(i) {
377 struct slot_info *s = &slot[i];
378 struct audio_format_info *a;
379 if (s->format < 0)
380 continue;
381 a = afi + s->format;
382 if (s->wns)
383 for (j = 0; j < a->num_writers; j++)
384 s->wns[j].task.error = error;
385 if (s->fns)
386 for (j = 0; j < a->num_writers; j++)
387 s->fns[j].task.error = error;
388 if (s->receiver_node)
389 s->receiver_node->task.error = error;
390 }
391 }
392
393 static int get_empty_slot(void)
394 {
395 int i;
396 struct slot_info *s;
397
398 FOR_EACH_SLOT(i) {
399 s = &slot[i];
400 if (s->format < 0) {
401 clear_slot(i);
402 return i;
403 }
404 if (s->wns || s->receiver_node || s->fns)
405 continue;
406 clear_slot(i);
407 return i;
408 }
409 return -E_NO_MORE_SLOTS;
410 }
411
412 /**
413 * get the number of filters
414 *
415 * \param audio_format_num the number identifying the audio format
416 *
417 * \return the number of filters for the given audio format
418 *
419 * \sa struct filter;
420 */
421 int num_filters(int audio_format_num)
422 {
423 return afi[audio_format_num].num_filters;
424 }
425
426 static void open_filters(struct slot_info *s)
427 {
428 struct audio_format_info *a = afi + s->format;
429 struct filter_node *fn;
430 int nf = a->num_filters;
431 struct btr_node *parent;
432 int i;
433
434 if (nf == 0)
435 return;
436 PARA_INFO_LOG("opening %s filters\n", audio_formats[s->format]);
437 assert(s->fns == NULL);
438 s->fns = para_calloc(nf * sizeof(struct filter_node));
439 parent = s->receiver_node->btrn;
440 for (i = 0; i < nf; i++) {
441 struct filter *f = filters + a->filter_nums[i];
442 fn = s->fns + i;
443 fn->filter_num = a->filter_nums[i];
444 fn->conf = a->filter_conf[i];
445 fn->task.pre_select = f->pre_select;
446 fn->task.post_select = f->post_select;
447 fn->btrn = btr_new_node(f->name, parent, f->execute, fn);
448 f->open(fn);
449 register_task(&fn->task);
450 parent = fn->btrn;
451 PARA_NOTICE_LOG("%s filter %d/%d (%s) started in slot %d\n",
452 audio_formats[s->format], i, nf, f->name, s - slot);
453 sprintf(fn->task.status, "%s (slot %d)", f->name, s - slot);
454 }
455 }
456
457 static void open_writers(struct slot_info *s)
458 {
459 int i;
460 struct audio_format_info *a = afi + s->format;
461 struct writer_node *wn;
462 struct btr_node *parent = s->fns[a->num_filters - 1].btrn;
463
464 assert(s->wns == NULL);
465 s->wns = para_calloc(PARA_MAX(1U, a->num_writers)
466 * sizeof(struct writer_node));
467 if (a->num_writers == 0)
468 setup_writer_node(NULL, parent, s->wns);
469 else {
470 PARA_INFO_LOG("opening %s writers\n", audio_formats[s->format]);
471 for (i = 0; i < a->num_writers; i++) {
472 wn = s->wns + i;
473 wn->conf = a->writer_conf[i];
474 wn->writer_num = a->writer_nums[i];
475 register_writer_node(wn, parent);
476 }
477 }
478 s->server_stream_start = stat_task->server_stream_start.tv_sec?
479 stat_task->server_stream_start : *now;
480 s->offset_seconds = stat_task->offset_seconds;
481 s->seconds_total = stat_task->length_seconds;
482 }
483
484 /* returns slot num on success */
485 static int open_receiver(int format)
486 {
487 struct audio_format_info *a = &afi[format];
488 struct slot_info *s;
489 int ret, slot_num;
490 struct receiver *r = a->receiver;
491 struct receiver_node *rn;
492 const struct timeval restart_delay = {2, 0};
493
494 ret = get_empty_slot();
495 if (ret < 0)
496 goto err;
497 slot_num = ret;
498 s = &slot[slot_num];
499 s->format = format;
500 s->receiver_node = para_calloc(sizeof(struct receiver_node));
501 rn = s->receiver_node;
502 rn->receiver = r;
503 rn->conf = a->receiver_conf;
504 rn->btrn = btr_new_node(r->name, NULL, NULL, rn);
505 ret = r->open(rn);
506 if (ret < 0) {
507 btr_free_node(rn->btrn);
508 free(rn);
509 s->receiver_node = NULL;
510 goto err;
511 }
512 PARA_NOTICE_LOG("started %s: %s receiver in slot %d\n",
513 audio_formats[s->format], r->name, slot_num);
514 rn->task.pre_select = r->pre_select;
515 rn->task.post_select = r->post_select;
516 sprintf(rn->task.status, "%s receiver node", r->name);
517 register_task(&rn->task);
518 ret = slot_num;
519 err:
520 if (ret < 0)
521 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
522 tv_add(now, &restart_delay, &afi[format].restart_barrier);
523 return ret;
524 }
525
526 /* return: 0: Not running, 1: Running, -1: Running but eof (or error) */
527 static int receiver_running(int format)
528 {
529 int i, ret = 0;
530
531 FOR_EACH_SLOT(i) {
532 struct slot_info *s = &slot[i];
533 if (s->format != format)
534 continue;
535 if (!s->receiver_node)
536 continue;
537 if (s->receiver_node->task.error >= 0)
538 return 1;
539 ret = -1;
540 }
541 return ret;
542 }
543
544 struct btr_node *audiod_get_btr_root(void)
545 {
546 int i, newest_slot = -1;
547 struct timeval newest_rstime = {0, 0};
548
549 FOR_EACH_SLOT(i) {
550 struct slot_info *s = &slot[i];
551 struct timeval rstime;
552 if (!s->receiver_node)
553 continue;
554 if (s->receiver_node->task.error < 0)
555 continue;
556 btr_get_node_start(s->receiver_node->btrn, &rstime);
557 if (newest_slot >= 0 && tv_diff(&rstime, &newest_rstime, NULL) < 0)
558 continue;
559 newest_rstime = rstime;
560 newest_slot = i;
561 }
562 if (newest_slot == -1)
563 return NULL;
564 return slot[newest_slot].receiver_node->btrn;
565 }
566
567 /* returns slot num on success. */
568 static int open_current_receiver(struct sched *s)
569 {
570 struct timeval diff;
571 int ret, cafn = stat_task->current_audio_format_num;
572
573 if (cafn < 0 || !stat_task->ct)
574 return -1;
575 /* Do nothing if the 'N' flag is set or the 'P' flag is unset */
576 if (stat_task->vss_status != VSS_STATUS_FLAG_PLAYING)
577 return -1;
578 ret = receiver_running(cafn);
579 if (ret > 0) /* already running and not eof */
580 return -1;
581 if (ret < 0) { /* eof */
582 /*
583 * para_server uses a zero start time during the announcement
584 * period, i.e. before it sends the first chunk. Wait until
585 * this period begins to avoid restarting the receiver that
586 * belongs to the file just completed.
587 */
588 if (stat_task->server_stream_start.tv_sec != 0)
589 return -1;
590 }
591 if (tv_diff(now, &afi[cafn].restart_barrier, &diff) < 0) {
592 /* avoid busy loop */
593 s->timeout = diff;
594 return -1;
595 }
596 /* start a new receiver */
597 return open_receiver(cafn);
598 }
599
600 static unsigned compute_time_diff(const struct timeval *status_time)
601 {
602 struct timeval tmp, diff;
603 static unsigned count;
604 int sign, sa_time_diff_sign = stat_task->sa_time_diff_sign;
605 const struct timeval max_deviation = {0, 500 * 1000};
606 const int time_smooth = 5;
607
608 if (!status_time)
609 return count;
610 sign = tv_diff(status_time, now, &diff);
611 // PARA_NOTICE_LOG("%s: sign = %i, sa_time_diff_sign = %i\n", __func__,
612 // sign, sa_time_diff_sign);
613 if (!count) {
614 sa_time_diff_sign = sign;
615 stat_task->sa_time_diff = diff;
616 count++;
617 goto out;
618 }
619 if (count > 5) {
620 int s = tv_diff(&diff, &stat_task->sa_time_diff, &tmp);
621 if (tv_diff(&max_deviation, &tmp, NULL) < 0)
622 PARA_WARNING_LOG("time diff jump: %lims\n",
623 s * tv2ms(&tmp));
624 }
625 count++;
626 sa_time_diff_sign = tv_convex_combination(
627 sa_time_diff_sign * time_smooth, &stat_task->sa_time_diff,
628 count > 10? sign : sign * time_smooth, &diff,
629 &tmp);
630 stat_task->sa_time_diff = tmp;
631 PARA_INFO_LOG("time diff (cur/avg): %s%lums/%s%lums\n",
632 sign > 0? "+" : "-",
633 tv2ms(&diff),
634 sa_time_diff_sign ? "+" : "-",
635 tv2ms(&stat_task->sa_time_diff)
636 );
637 out:
638 stat_task->sa_time_diff_sign = sa_time_diff_sign;
639 return count;
640 }
641
642 static int update_item(int itemnum, char *buf)
643 {
644 long unsigned sec, usec;
645
646 if (stat_task->clock_diff_count && itemnum != SI_CURRENT_TIME)
647 return 1;
648 free(stat_item_values[itemnum]);
649 stat_item_values[itemnum] = para_strdup(buf);
650 stat_client_write_item(itemnum);
651 switch (itemnum) {
652 case SI_STATUS_FLAGS:
653 stat_task->vss_status = 0;
654 if (strchr(buf, 'N'))
655 stat_task->vss_status |= VSS_STATUS_FLAG_NEXT;
656 if (strchr(buf, 'P'))
657 stat_task->vss_status |= VSS_STATUS_FLAG_PLAYING;
658 break;
659 case SI_OFFSET:
660 stat_task->offset_seconds = atoi(buf);
661 break;
662 case SI_SECONDS_TOTAL:
663 stat_task->length_seconds = atoi(buf);
664 break;
665 case SI_STREAM_START:
666 if (sscanf(buf, "%lu.%lu", &sec, &usec) == 2) {
667 struct timeval a_start, delay;
668 delay.tv_sec = conf.stream_delay_arg / 1000;
669 delay.tv_usec = (conf.stream_delay_arg % 1000) * 1000;
670 stat_task->server_stream_start.tv_sec = sec;
671 stat_task->server_stream_start.tv_usec = usec;
672 if (compute_time_diff(NULL) > 2) {
673 if (stat_task->sa_time_diff_sign < 0)
674 tv_add(&stat_task->server_stream_start,
675 &stat_task->sa_time_diff, &a_start);
676 else
677 tv_diff(&stat_task->server_stream_start,
678 &stat_task->sa_time_diff, &a_start);
679 tv_add(&a_start, &delay, &initial_delay_barrier);
680 }
681 }
682 break;
683 case SI_CURRENT_TIME:
684 if (sscanf(buf, "%lu.%lu", &sec, &usec) == 2) {
685 struct timeval tv = {sec, usec};
686 compute_time_diff(&tv);
687 }
688 break;
689 case SI_FORMAT:
690 stat_task->current_audio_format_num
691 = get_audio_format_num(buf);
692 }
693 return 1;
694 }
695
696 static int parse_stream_command(const char *txt, char **cmd)
697 {
698 char *p = strchr(txt, ':');
699 int i;
700
701 if (!p)
702 return -E_MISSING_COLON;
703 p++;
704 FOR_EACH_AUDIO_FORMAT(i) {
705 if (strncmp(txt, audio_formats[i], strlen(audio_formats[i])))
706 continue;
707 *cmd = p;
708 return i;
709 }
710 return -E_UNSUPPORTED_AUDIO_FORMAT;
711 }
712
713 static int add_filter(int format, char *cmdline)
714 {
715 struct audio_format_info *a = &afi[format];
716 int filter_num, nf = a->num_filters;
717
718 filter_num = check_filter_arg(cmdline, &a->filter_conf[nf]);
719 if (filter_num < 0)
720 return filter_num;
721 a->filter_nums[nf] = filter_num;
722 a->num_filters++;
723 PARA_INFO_LOG("%s filter %d: %s\n", audio_formats[format], nf,
724 filters[filter_num].name);
725 return filter_num;
726 }
727
728 static int parse_writer_args(void)
729 {
730 int i, ret, nw;
731 char *cmd;
732 struct audio_format_info *a;
733
734 nw = PARA_MAX(1U, conf.writer_given);
735 PARA_INFO_LOG("maximal number of writers: %d\n", nw);
736 FOR_EACH_AUDIO_FORMAT(i) {
737 a = &afi[i];
738 a->writer_conf = para_malloc(nw * sizeof(void *));
739 a->writer_nums = para_malloc(nw * sizeof(int));
740 a->num_writers = 0;
741 }
742 for (i = 0; i < conf.writer_given; i++) {
743 void *wconf;
744 int writer_num;
745 ret = parse_stream_command(conf.writer_arg[i], &cmd);
746 if (ret < 0)
747 goto out;
748 a = &afi[ret];
749 nw = a->num_writers;
750 wconf = check_writer_arg(cmd, &writer_num);
751 if (!wconf) {
752 ret = writer_num;
753 goto out;
754 }
755 a->writer_nums[nw] = writer_num;
756 a->writer_conf[nw] = wconf;
757 PARA_INFO_LOG("%s writer #%d: %s\n", audio_formats[ret],
758 nw, writer_names[writer_num]);
759 a->num_writers++;
760 }
761 ret = 1;
762 out:
763 return ret;
764 }
765
766 static int parse_receiver_args(void)
767 {
768 int i, ret, receiver_num;
769 char *cmd = NULL;
770 struct audio_format_info *a;
771
772 for (i = conf.receiver_given - 1; i >= 0; i--) {
773 char *arg = conf.receiver_arg[i];
774 char *recv_arg = strchr(arg, ':');
775 ret = -E_MISSING_COLON;
776 if (!recv_arg)
777 goto out;
778 *recv_arg = '\0';
779 recv_arg++;
780 ret = get_audio_format_num(arg);
781 if (ret < 0)
782 goto out;
783 afi[ret].receiver_conf = check_receiver_arg(recv_arg, &receiver_num);
784 if (!afi[ret].receiver_conf) {
785 ret = -E_RECV_SYNTAX;
786 goto out;
787 }
788 afi[ret].receiver = &receivers[receiver_num];
789 }
790 /* use the first available receiver with no arguments
791 * for those audio formats for which no receiver
792 * was specified
793 */
794 cmd = para_strdup(receivers[0].name);
795 FOR_EACH_AUDIO_FORMAT(i) {
796 a = &afi[i];
797 if (a->receiver_conf)
798 continue;
799 a->receiver_conf = check_receiver_arg(cmd, &receiver_num);
800 if (!a->receiver_conf)
801 return -E_RECV_SYNTAX;
802 a->receiver = &receivers[receiver_num];
803 }
804 ret = 1;
805 out:
806 free(cmd);
807 return ret;
808 }
809
810 static int init_default_filters(void)
811 {
812 int i, ret = 1;
813
814 FOR_EACH_AUDIO_FORMAT(i) {
815 struct audio_format_info *a = &afi[i];
816 char *tmp;
817 int j;
818
819 if (a->num_filters)
820 continue; /* no default -- nothing to to */
821 /* add "dec" to audio format name */
822 tmp = make_message("%sdec", audio_formats[i]);
823 for (j = 0; filters[j].name; j++)
824 if (!strcmp(tmp, filters[j].name))
825 break;
826 free(tmp);
827 ret = -E_UNSUPPORTED_FILTER;
828 if (!filters[j].name)
829 goto out;
830 tmp = para_strdup(filters[j].name);
831 ret = add_filter(i, tmp);
832 free(tmp);
833 if (ret < 0)
834 goto out;
835 PARA_INFO_LOG("%s -> default filter: %s\n", audio_formats[i],
836 filters[j].name);
837 }
838 out:
839 return ret;
840 }
841
842 static int parse_filter_args(void)
843 {
844 int i, ret, nf;
845
846 nf = PARA_MAX(1U, conf.filter_given);
847 PARA_INFO_LOG("maximal number of filters: %d\n", nf);
848 FOR_EACH_AUDIO_FORMAT(i) {
849 afi[i].filter_conf = para_malloc(nf * sizeof(void *));
850 afi[i].filter_nums = para_malloc(nf * sizeof(unsigned));
851 }
852 if (!conf.no_default_filters_given)
853 return init_default_filters();
854 for (i = 0; i < conf.filter_given; i++) {
855 char *arg = conf.filter_arg[i];
856 char *filter_name = strchr(arg, ':');
857 ret = -E_MISSING_COLON;
858 if (!filter_name)
859 goto out;
860 *filter_name = '\0';
861 filter_name++;
862 ret = get_audio_format_num(arg);
863 if (ret < 0)
864 goto out;
865 ret = add_filter(ret, filter_name);
866 if (ret < 0)
867 goto out;
868 }
869 ret = init_default_filters(); /* use default values for the rest */
870 out:
871 return ret;
872 }
873
874 static int parse_stream_args(void)
875 {
876 int ret;
877
878 ret = parse_receiver_args();
879 if (ret < 0)
880 return ret;
881 ret = parse_filter_args();
882 if (ret < 0)
883 return ret;
884 ret = parse_writer_args();
885 if (ret < 0)
886 return ret;
887 return 1;
888 }
889
890 /* does not unlink socket on errors */
891 static int audiod_get_socket(void)
892 {
893 struct sockaddr_un unix_addr;
894 int ret, fd;
895
896 if (conf.socket_given)
897 socket_name = para_strdup(conf.socket_arg);
898 else {
899 char *hn = para_hostname();
900 socket_name = make_message("/var/paraslash/audiod_socket.%s",
901 hn);
902 free(hn);
903 }
904 PARA_NOTICE_LOG("local socket: %s\n", socket_name);
905 if (conf.force_given)
906 unlink(socket_name);
907 ret = create_local_socket(socket_name, &unix_addr,
908 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IWOTH);
909 if (ret < 0)
910 goto err;
911 fd = ret;
912 if (listen(fd , 5) < 0) {
913 ret = -ERRNO_TO_PARA_ERROR(errno);
914 goto err;
915 }
916 ret = mark_fd_nonblocking(fd);
917 if (ret < 0)
918 goto err;
919 return fd;
920 err:
921 PARA_EMERG_LOG("%s\n", para_strerror(-ret));
922 exit(EXIT_FAILURE);
923 }
924
925 static void signal_pre_select(struct sched *s, struct task *t)
926 {
927 struct signal_task *st = container_of(t, struct signal_task, task);
928 para_fd_set(st->fd, &s->rfds, &s->max_fileno);
929 }
930
931 static void signal_post_select(struct sched *s, struct task *t)
932 {
933 struct signal_task *st = container_of(t, struct signal_task, task);
934
935 if (!FD_ISSET(st->fd, &s->rfds))
936 return;
937
938 st->signum = para_next_signal();
939 switch (st->signum) {
940 case SIGINT:
941 case SIGTERM:
942 case SIGHUP:
943 PARA_EMERG_LOG("terminating on signal %d\n", st->signum);
944 clean_exit(EXIT_FAILURE, "caught deadly signal");
945 }
946 }
947
948 static void signal_setup_default(struct signal_task *st)
949 {
950 st->task.pre_select = signal_pre_select;
951 st->task.post_select = signal_post_select;
952 sprintf(st->task.status, "signal task");
953 }
954
955 static void command_pre_select(struct sched *s, struct task *t)
956 {
957 struct command_task *ct = container_of(t, struct command_task, task);
958 para_fd_set(ct->fd, &s->rfds, &s->max_fileno);
959 }
960
961 static void command_post_select(struct sched *s, struct task *t)
962 {
963 int ret;
964 struct command_task *ct = container_of(t, struct command_task, task);
965 static struct timeval last_status_dump;
966 struct timeval tmp, delay = {0, 500 * 1000};
967
968 tv_add(&last_status_dump, &delay, &tmp);
969 if (tv_diff(&tmp, now, NULL) < 0) {
970 audiod_status_dump();
971 last_status_dump = *now;
972 }
973
974 if (!FD_ISSET(ct->fd, &s->rfds))
975 return;
976 ret = handle_connect(ct->fd);
977 if (ret < 0)
978 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
979 }
980
981 static void init_command_task(struct command_task *ct)
982 {
983 ct->task.pre_select = command_pre_select;
984 ct->task.post_select = command_post_select;
985 ct->task.error = 0;
986 ct->fd = audiod_get_socket(); /* doesn't return on errors */
987 sprintf(ct->task.status, "command task");
988 }
989
990 static void close_stat_pipe(void)
991 {
992 if (!stat_task->ct)
993 return;
994 client_close(stat_task->ct);
995 stat_task->ct = NULL;
996 clear_and_dump_items();
997 stat_task->length_seconds = 0;
998 stat_task->offset_seconds = 0;
999 stat_task->vss_status = 0;
1000 stat_task->current_audio_format_num = -1;
1001 audiod_status_dump();
1002 }
1003
1004 /**
1005 * close the connection to para_server and exit
1006 *
1007 * \param status the exit status which is passed to exit(3)
1008 * \param msg the log message
1009 *
1010 * Log \a msg with loglevel \p EMERG, close the connection to para_server if
1011 * open, and call \p exit(status). \a status should be either EXIT_SUCCESS or
1012 * EXIT_FAILURE.
1013 *
1014 * \sa exit(3)
1015 */
1016 void __noreturn clean_exit(int status, const char *msg)
1017 {
1018 PARA_EMERG_LOG("%s\n", msg);
1019 if (socket_name)
1020 unlink(socket_name);
1021 close_stat_pipe();
1022 exit(status);
1023 }
1024
1025 /* avoid busy loop if server is down */
1026 static void set_stat_task_restart_barrier(unsigned seconds)
1027 {
1028 struct timeval delay = {seconds, 0};
1029 tv_add(now, &delay, &stat_task->restart_barrier);
1030 }
1031
1032 static void try_to_close_slot(int slot_num)
1033 {
1034 struct slot_info *s = &slot[slot_num];
1035 struct audio_format_info *a = afi + s->format;
1036 int i;
1037
1038 if (s->format < 0)
1039 return;
1040 if (s->receiver_node && s->receiver_node->task.error != -E_TASK_UNREGISTERED)
1041 return;
1042 for (i = 0; i < a->num_filters; i++)
1043 if (s->fns && s->fns[i].task.error != -E_TASK_UNREGISTERED)
1044 return;
1045 for (i = 0; i < a->num_writers; i++)
1046 if (s->wns && s->wns[i].task.error != -E_TASK_UNREGISTERED)
1047 return;
1048 PARA_INFO_LOG("closing slot %d\n", slot_num);
1049 close_writers(s);
1050 _close_filters(s);
1051 close_receiver(slot_num);
1052 clear_slot(slot_num);
1053 }
1054
1055 /*
1056 * Check if any receivers/filters/writers need to be started and do so if
1057 * necessary.
1058 */
1059 static void start_stop_decoders(struct sched *s)
1060 {
1061 int i, ret;
1062 struct slot_info *sl;
1063 struct audio_format_info *a;
1064
1065 FOR_EACH_SLOT(i)
1066 try_to_close_slot(i);
1067 // if (audiod_status != AUDIOD_ON ||
1068 // !(stat_task->vss_status & VSS_STATUS_FLAG_PLAYING))
1069 // return kill_all_decoders(-E_NOT_PLAYING);
1070 ret = open_current_receiver(s);
1071 if (ret < 0)
1072 return;
1073 sl = slot + ret;
1074 a = afi + sl->format;
1075 if (a->num_filters)
1076 open_filters(sl);
1077 open_writers(sl);
1078 activate_grab_clients();
1079 btr_log_tree(sl->receiver_node->btrn, LL_NOTICE);
1080 s->timeout.tv_sec = 0;
1081 s->timeout.tv_usec = 1;
1082 }
1083
1084 /* restart the client task if necessary */
1085 static void status_pre_select(struct sched *s, struct task *t)
1086 {
1087 struct status_task *st = container_of(t, struct status_task, task);
1088
1089 if (audiod_status == AUDIOD_OFF) {
1090 if (!st->ct)
1091 goto out;
1092 if (st->ct->task.error >= 0) {
1093 st->ct->task.error = -E_AUDIOD_OFF;
1094 goto out;
1095 }
1096 if (st->ct->task.error != -E_TASK_UNREGISTERED)
1097 goto out;
1098 close_stat_pipe();
1099 st->clock_diff_count = conf.clock_diff_count_arg;
1100 goto out;
1101 }
1102 if (st->ct) {
1103 int ret;
1104 if (st->ct->task.error < 0) {
1105 if (st->ct->task.error != -E_TASK_UNREGISTERED)
1106 goto out;
1107 close_stat_pipe();
1108 goto out;
1109 }
1110 if (st->ct->status != CL_RECEIVING)
1111 goto out;
1112 ret = for_each_stat_item(st->ct->buf, st->ct->loaded,
1113 update_item);
1114 if (ret < 0) {
1115 st->ct->task.error = ret;
1116 goto out;
1117 }
1118 if (st->ct->loaded != ret) {
1119 st->last_status_read = *now;
1120 st->ct->loaded = ret;
1121 } else {
1122 struct timeval diff;
1123 tv_diff(now, &st->last_status_read, &diff);
1124 if (diff.tv_sec > 61)
1125 st->ct->task.error = -E_STATUS_TIMEOUT;
1126 }
1127 goto out;
1128 }
1129 if (tv_diff(now, &st->restart_barrier, NULL) < 0)
1130 goto out;
1131 if (st->clock_diff_count) { /* get status only one time */
1132 char *argv[] = {"audiod", "--", "stat", "-p", "1", NULL};
1133 int argc = 5;
1134 PARA_INFO_LOG("clock diff count: %d\n", st->clock_diff_count);
1135 st->clock_diff_count--;
1136 client_open(argc, argv, &st->ct, NULL);
1137 set_stat_task_restart_barrier(2);
1138
1139 } else {
1140 char *argv[] = {"audiod", "--", "stat", "-p", NULL};
1141 int argc = 4;
1142 client_open(argc, argv, &st->ct, NULL);
1143 set_stat_task_restart_barrier(5);
1144 }
1145 free(stat_item_values[SI_BASENAME]);
1146 stat_item_values[SI_BASENAME] = para_strdup(
1147 "no connection to para_server");
1148 stat_client_write_item(SI_BASENAME);
1149 st->last_status_read = *now;
1150 out:
1151 start_stop_decoders(s);
1152 }
1153
1154 static void init_status_task(struct status_task *st)
1155 {
1156 memset(st, 0, sizeof(struct status_task));
1157 st->task.pre_select = status_pre_select;
1158 st->sa_time_diff_sign = 1;
1159 st->clock_diff_count = conf.clock_diff_count_arg;
1160 st->current_audio_format_num = -1;
1161 sprintf(st->task.status, "status task");
1162 }
1163
1164 static void set_initial_status(void)
1165 {
1166 audiod_status = AUDIOD_ON;
1167 if (!conf.mode_given)
1168 return;
1169 if (!strcmp(conf.mode_arg, "sb")) {
1170 audiod_status = AUDIOD_STANDBY;
1171 return;
1172 }
1173 if (!strcmp(conf.mode_arg, "off")) {
1174 audiod_status = AUDIOD_OFF;
1175 return;
1176 }
1177 if (strcmp(conf.mode_arg, "on"))
1178 PARA_WARNING_LOG("invalid mode\n");
1179 }
1180
1181 __noreturn static void print_help_and_die(void)
1182 {
1183 int d = conf.detailed_help_given;
1184 const char **p = d? audiod_args_info_detailed_help
1185 : audiod_args_info_help;
1186
1187 printf_or_die("%s\n\n", AUDIOD_CMDLINE_PARSER_PACKAGE "-"
1188 AUDIOD_CMDLINE_PARSER_VERSION);
1189 printf_or_die("%s\n\n", audiod_args_info_usage);
1190 for (; *p; p++)
1191 printf_or_die("%s\n", *p);
1192 print_receiver_helps(d);
1193 print_filter_helps(d);
1194 print_writer_helps(d);
1195 exit(0);
1196 }
1197
1198 static void init_colors_or_die(void)
1199 {
1200 int ret, i;
1201
1202 if (!want_colors())
1203 return;
1204 daemon_set_default_log_colors();
1205 daemon_set_flag(DF_COLOR_LOG);
1206 for (i = 0; i < conf.log_color_given; i++) {
1207 ret = daemon_set_log_color(conf.log_color_arg[i]);
1208 if (ret < 0)
1209 exit(EXIT_FAILURE);
1210 }
1211 }
1212
1213 /**
1214 * the main function of para_audiod
1215 *
1216 * \param argc usual argument count
1217 * \param argv usual argument vector
1218 *
1219 * \return EXIT_SUCCESS or EXIT_FAILURE
1220 *
1221 * \sa para_audiod(1)
1222 * */
1223 int main(int argc, char *argv[])
1224 {
1225 int ret, i;
1226 static struct sched s;
1227 struct command_task command_task_struct, *cmd_task = &command_task_struct;
1228 struct audiod_cmdline_parser_params params = {
1229 .override = 0,
1230 .initialize = 1,
1231 .check_required = 0,
1232 .check_ambiguity = 0,
1233 .print_errors = 1
1234 };
1235
1236 valid_fd_012();
1237 if (audiod_cmdline_parser_ext(argc, argv, &conf, &params))
1238 exit(EXIT_FAILURE);
1239 HANDLE_VERSION_FLAG("audiod", conf);
1240 /* init receivers/filters/writers early to make help work */
1241 recv_init();
1242 filter_init();
1243 writer_init();
1244 if (conf.help_given || conf.detailed_help_given)
1245 print_help_and_die();
1246 drop_privileges_or_die(conf.user_arg, conf.group_arg);
1247 parse_config_or_die();
1248 init_colors_or_die();
1249 init_random_seed_or_die();
1250 daemon_set_flag(DF_LOG_TIME);
1251 daemon_set_flag(DF_LOG_HOSTNAME);
1252 daemon_set_flag(DF_LOG_LL);
1253 if (conf.log_timing_given)
1254 daemon_set_flag(DF_LOG_TIMING);
1255 if (conf.logfile_given) {
1256 daemon_set_logfile(conf.logfile_arg);
1257 daemon_open_log_or_die();
1258 }
1259 ret = parse_stream_args();
1260 if (ret < 0) {
1261 PARA_EMERG_LOG("%s\n", para_strerror(-ret));
1262 exit(EXIT_FAILURE);
1263 }
1264 log_welcome("para_audiod");
1265 server_uptime(UPTIME_SET);
1266 set_initial_status();
1267 FOR_EACH_SLOT(i)
1268 clear_slot(i);
1269 init_grabbing();
1270 setup_signal_handling();
1271 signal_setup_default(sig_task);
1272
1273 init_status_task(stat_task);
1274 init_command_task(cmd_task);
1275
1276 if (conf.daemon_given)
1277 daemonize();
1278
1279 register_task(&sig_task->task);
1280 register_task(&cmd_task->task);
1281 register_task(&stat_task->task);
1282 s.default_timeout.tv_sec = 2;
1283 s.default_timeout.tv_usec = 999 * 1000;
1284 ret = schedule(&s);
1285
1286 PARA_EMERG_LOG("%s\n", para_strerror(-ret));
1287 return EXIT_FAILURE;
1288 }