audiod: make signal handling functions static
[paraslash.git] / audiod.c
1 /*
2 * Copyright (C) 2005-2006 Andre Noll <noll@mathematik.tu-darmstadt.de>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
17 */
18
19 /** \file audiod.c the paraslash's audio daemon */
20
21 #include "para.h"
22
23 #include "audiod.cmdline.h"
24 #include "list.h"
25 #include "sched.h"
26 #include "recv.h"
27 #include "filter.h"
28 #include "grab_client.cmdline.h"
29 #include "grab_client.h"
30 #include "client.cmdline.h"
31 #include "client.h"
32 #include "audiod.h"
33 #include "net.h"
34 #include "daemon.h"
35 #include "string.h"
36 #include "fd.h"
37 #include "write.h"
38 #include "write_common.h"
39 #include "error.h"
40
41 /** define the array of error lists needed by para_audiod */
42 INIT_AUDIOD_ERRLISTS;
43 /** define the array containing all supported audio formats */
44 DEFINE_AUDIO_FORMAT_ARRAY;
45
46 /** defines how to handle one supported audio format */
47 struct audio_format_info {
48 /** pointer to the receiver for this audio format */
49 struct receiver *receiver;
50 /** the receiver configuration */
51 void *receiver_conf;
52 /** the number of filters that should be activated for this audio format */
53 unsigned int num_filters;
54 /** pointer to the array of filters to be activated */
55 struct filter **filters;
56 /** pointer to the array of filter configurations */
57 void **filter_conf;
58 /** the number of filters that should be activated for this audio format */
59 unsigned int num_writers;
60 /** pointer to the array of writers to be activated */
61 struct writer **writers;
62 /** pointer to the array of writer configurations */
63 void **writer_conf;
64 /** do not start receiver/filters/writer before this time */
65 struct timeval restart_barrier;
66 };
67
68 struct slot_info slot[MAX_STREAM_SLOTS];
69
70
71 int audiod_status = AUDIOD_ON;
72
73 struct audiod_args_info conf;
74 static char *socket_name;
75 static FILE *logfile;
76 static struct audio_format_info afi[NUM_AUDIO_FORMATS];
77
78 static struct signal_task signal_task_struct, *sig_task = &signal_task_struct;
79
80 static struct status_task status_task_struct;
81 struct status_task *stat_task = &status_task_struct;
82
83 /**
84 * the task for handling audiod commands
85 *
86 * \sa struct task, struct sched
87 */
88 struct command_task {
89 /** the local listening socket */
90 int fd;
91 /** the associated task structure */
92 struct task task;
93 };
94
95 struct signal_task {
96 int fd;
97 int signum;
98 struct task task;
99 };
100
101 /** iterate over all supported audio formats */
102 #define FOR_EACH_AUDIO_FORMAT(af) for (af = 0; af < NUM_AUDIO_FORMATS; af++)
103
104 /**
105 * get the audio format number
106 * \param name the name of the audio format
107 *
108 * \return The audio format number on success, -E_UNSUPPORTED_AUDIO_FORMAT if
109 * \a name is not a supported audio format.
110 */
111 int get_audio_format_num(char *name)
112 {
113 int i;
114 FOR_EACH_AUDIO_FORMAT(i)
115 if (!strcmp(name, audio_formats[i]))
116 return i;
117 return -E_UNSUPPORTED_AUDIO_FORMAT;
118 }
119
120 /*
121 * log function. first argument is loglevel.
122 */
123 void para_log(int ll, const char* fmt,...)
124 {
125 va_list argp;
126 FILE *outfd;
127 struct tm *tm;
128 time_t t1;
129 char str[MAXLINE] = "";
130 static char *hostname;
131
132 if (ll < conf.loglevel_arg)
133 return;
134 if (!hostname)
135 hostname = para_hostname();
136 outfd = logfile? logfile : stderr;
137 time(&t1);
138 tm = localtime(&t1);
139 strftime(str, MAXLINE, "%b %d %H:%M:%S", tm);
140 fprintf(outfd, "%s %s ", str, hostname);
141 if (conf.loglevel_arg <= INFO)
142 fprintf(outfd, "%i ", ll);
143 va_start(argp, fmt);
144 vfprintf(outfd, fmt, argp);
145 va_end(argp);
146 }
147
148 static char *configfile_exists(void)
149 {
150 static char *config_file;
151
152 if (!config_file) {
153 char *home = para_homedir();
154 config_file = make_message("%s/.paraslash/audiod.conf", home);
155 free(home);
156 }
157 return file_exists(config_file)? config_file : NULL;
158 }
159
160 static void setup_signal_handling(void)
161 {
162 sig_task->fd = para_signal_init();
163 PARA_INFO_LOG("signal pipe: fd %d\n", sig_task->fd);
164 para_install_sighandler(SIGINT);
165 para_install_sighandler(SIGTERM);
166 para_install_sighandler(SIGHUP);
167 signal(SIGPIPE, SIG_IGN);
168 }
169
170 static void clear_slot(int slot_num)
171 {
172 struct slot_info *s = &slot[slot_num];
173
174 PARA_INFO_LOG("clearing slot %d\n", slot_num);
175 memset(s, 0, sizeof(struct slot_info));
176 s->format = -1;
177 }
178
179 static void close_receiver(int slot_num)
180 {
181 struct slot_info *s = &slot[slot_num];
182 struct audio_format_info *a;
183 const struct timeval restart_delay = {0, 200 * 1000};
184
185 if (s->format < 0 || !s->receiver_node)
186 return;
187 a = &afi[s->format];
188 PARA_NOTICE_LOG("closing %s receiver in slot %d (eof = %d)\n",
189 audio_formats[s->format] , slot_num, s->receiver_node->eof);
190 // if (!s->receiver_node->eof)
191 // unregister_task(&s->receiver_node->task);
192 a->receiver->close(s->receiver_node);
193 free(s->receiver_node);
194 s->receiver_node = NULL;
195 /* set restart barrier */
196 tv_add(now, &restart_delay, &afi[s->format].restart_barrier);
197 }
198
199 static void kill_all_decoders(void)
200 {
201 int i;
202
203 FOR_EACH_SLOT(i) {
204 struct slot_info *s = &slot[i];
205 if (s->wng && !s->wng->eof) {
206 PARA_INFO_LOG("unregistering writer node group in slot %d\n",
207 i);
208 wng_unregister(s->wng);
209 s->wng->eof = 1;
210 }
211 if (s->fc && !s->fc->eof) {
212 PARA_INFO_LOG("unregistering filter chain in slot %d\n", i);
213 unregister_task(&s->fc->task);
214 s->fc->eof = 1;
215 }
216 if (s->receiver_node && !s->receiver_node->eof) {
217 PARA_INFO_LOG("unregistering receiver_node in slot %d\n", i);
218 unregister_task(&s->receiver_node->task);
219 s->receiver_node->eof = 1;
220 }
221 }
222 }
223
224 static int get_empty_slot(void)
225 {
226 int i;
227 struct slot_info *s;
228
229 FOR_EACH_SLOT(i) {
230 s = &slot[i];
231 if (s->format < 0) {
232 clear_slot(i);
233 return i;
234 }
235 if (s->wng || s->receiver_node || s->fc)
236 continue;
237 clear_slot(i);
238 return i;
239 }
240 return -E_NO_MORE_SLOTS;
241 }
242
243 static int decoder_running(int format)
244 {
245 int i, ret = 0;
246 struct slot_info *s;
247
248 FOR_EACH_SLOT(i) {
249 s = &slot[i];
250 if (s->format == format && s->receiver_node)
251 ret |= 1;
252 if (s->format == format && s->wng)
253 ret |= 2;
254 }
255 return ret;
256 }
257
258 static void close_stat_pipe(void)
259 {
260 int i;
261
262 if (!stat_task->pcd)
263 return;
264 client_close(stat_task->pcd);
265 stat_task->pcd = NULL;
266 for (i = 0; i < NUM_STAT_ITEMS; i++) {
267 free(stat_task->stat_item_values[i]);
268 stat_task->stat_item_values[i] = NULL;
269 }
270 dump_empty_status();
271 stat_task->length_seconds = 0;
272 stat_task->offset_seconds = 0;
273 audiod_status_dump();
274 stat_task->playing = 0;
275 stat_task->stat_item_values[SI_STATUS_BAR] = make_message(
276 "%s:no connection to para_server\n",
277 status_item_list[SI_STATUS_BAR]);
278 stat_client_write(stat_task->stat_item_values[SI_STATUS_BAR], SI_STATUS_BAR);
279 }
280
281 void __noreturn clean_exit(int status, const char *msg)
282 {
283 PARA_EMERG_LOG("%s\n", msg);
284 if (socket_name)
285 unlink(socket_name);
286 close_stat_pipe();
287 exit(status);
288 }
289
290 /**
291 * get the number of filters
292 *
293 * \param audio_format_num the number identifying the audio format
294 *
295 * \return the number of filters for the given audio format
296 *
297 * \sa struct filter;
298 */
299 int num_filters(int audio_format_num)
300 {
301 return afi[audio_format_num].num_filters;
302 }
303
304 static void filter_event_handler(struct task *t)
305 {
306 PARA_NOTICE_LOG("%s\n", PARA_STRERROR(-t->ret));
307 struct filter_chain *fc = t->private_data;
308 fc->eof = 1;
309 unregister_task(t);
310 }
311
312 static void open_filters(int slot_num)
313 {
314 struct slot_info *s = &slot[slot_num];
315 struct audio_format_info *a = &afi[s->format];
316 int nf = a->num_filters;
317 int i;
318
319 s->fc = NULL;
320 if (!nf)
321 return;
322 PARA_INFO_LOG("opening %s filters\n", audio_formats[s->format]);
323 s->fc = para_calloc(sizeof(struct filter_chain));
324 INIT_LIST_HEAD(&s->fc->filters);
325 s->fc->inbuf = s->receiver_node->buf;
326 s->fc->in_loaded = &s->receiver_node->loaded;
327 s->fc->input_eof = &s->receiver_node->eof;
328 s->fc->task.pre_select = filter_pre_select;
329 s->fc->task.event_handler = filter_event_handler;
330 s->fc->task.private_data = s->fc;
331 s->fc->eof = 0;
332
333 s->receiver_node->output_eof = &s->fc->eof;
334 sprintf(s->fc->task.status, "filter chain");
335 for (i = 0; i < nf; i++) {
336 struct filter_node *fn = para_calloc(sizeof(struct filter_node));
337 fn->conf = a->filter_conf[i];
338 fn->fc = s->fc;
339 fn->filter = a->filters[i];
340 INIT_LIST_HEAD(&fn->callbacks);
341 list_add_tail(&fn->node, &s->fc->filters);
342 fn->filter->open(fn);
343 PARA_NOTICE_LOG("%s filter %d/%d (%s) started in slot %d\n",
344 audio_formats[s->format], i + 1, nf,
345 fn->filter->name, slot_num);
346 s->fc->outbuf = fn->buf;
347 s->fc->out_loaded = &fn->loaded;
348 }
349 register_task(&s->fc->task);
350 }
351
352 static void wng_event_handler(struct task *t)
353 {
354 struct writer_node_group *wng = t->private_data;
355
356 PARA_INFO_LOG("%s\n", PARA_STRERROR(-t->ret));
357 wng->eof = 1;
358 wng_unregister(wng);
359 }
360
361 static void open_writers(int slot_num)
362 {
363 int ret, i;
364 struct slot_info *s = &slot[slot_num];
365 struct audio_format_info *a = &afi[s->format];
366
367 PARA_INFO_LOG("opening %s writers\n", audio_formats[s->format]);
368 if (!a->num_writers)
369 s->wng = setup_default_wng();
370 else
371 s->wng = wng_new(a->num_writers);
372 if (s->fc) {
373 s->wng->buf = s->fc->outbuf;
374 s->wng->loaded = s->fc->out_loaded;
375 s->wng->input_eof = &s->fc->eof;
376 s->wng->channels = &s->fc->channels;
377 s->wng->samplerate = &s->fc->samplerate;
378 s->fc->output_eof = &s->wng->eof;
379 PARA_INFO_LOG("samplerate: %d\n", *s->wng->samplerate);
380 } else {
381 s->wng->buf = s->receiver_node->buf;
382 s->wng->loaded = &s->receiver_node->loaded;
383 s->wng->input_eof = &s->receiver_node->eof;
384 }
385 s->wng->task.event_handler = wng_event_handler;
386 for (i = 0; i < a->num_writers; i++) {
387 s->wng->writer_nodes[i].conf = a->writer_conf[i];
388 s->wng->writer_nodes[i].writer = a->writers[i];
389 }
390 ret = wng_open(s->wng);
391 if (ret < 0) {
392 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
393 return;
394 }
395 s->wstime = *now;
396 activate_inactive_grab_clients(slot_num, s->format, &s->fc->filters);
397 }
398
399 static void rn_event_handler(struct task *t)
400 {
401 struct receiver_node *rn = t->private_data;
402
403 PARA_NOTICE_LOG("%s\n", PARA_STRERROR(-t->ret));
404 unregister_task(t);
405 rn->eof = 1;
406 }
407
408 static void open_receiver(int format)
409 {
410 struct audio_format_info *a = &afi[format];
411 struct slot_info *s;
412 int ret, slot_num;
413 struct receiver_node *rn;
414
415 slot_num = get_empty_slot();
416 if (slot_num < 0)
417 clean_exit(EXIT_FAILURE, PARA_STRERROR(-slot_num));
418 s = &slot[slot_num];
419 s->format = format;
420 s->receiver_node = para_calloc(sizeof(struct receiver_node));
421 rn = s->receiver_node;
422 rn->receiver = a->receiver;
423 rn->conf = a->receiver_conf;
424 ret = a->receiver->open(s->receiver_node);
425 if (ret < 0) {
426 PARA_ERROR_LOG("failed to open receiver (%s)\n",
427 PARA_STRERROR(-ret));
428 free(s->receiver_node);
429 s->receiver_node = NULL;
430 return;
431 }
432 PARA_NOTICE_LOG("started %s: %s receiver in slot %d\n",
433 audio_formats[s->format], a->receiver->name, slot_num);
434 rn->task.private_data = s->receiver_node;
435 rn->task.pre_select = a->receiver->pre_select;
436 rn->task.post_select = a->receiver->post_select;
437 rn->task.event_handler = rn_event_handler;
438 sprintf(rn->task.status, "%s receiver node", rn->receiver->name);
439 register_task(&rn->task);
440 }
441
442 static int open_current_receiver(struct sched *s)
443 {
444 int i;
445 struct timeval diff;
446
447 if (!stat_task->af_status || !stat_task->pcd)
448 return 0;
449 i = get_audio_format_num(stat_task->af_status);
450 if (i < 0)
451 return 0;
452 if (decoder_running(i))
453 return 0;
454 if (tv_diff(now, &afi[i].restart_barrier, &diff) < 0) {
455 s->timeout = diff;
456 return 0;
457 }
458 open_receiver(i);
459 return 1;
460 }
461
462 static void compute_time_diff(const struct timeval *status_time)
463 {
464 struct timeval tmp, diff;
465 static int count;
466 int sign, sa_time_diff_sign = stat_task->sa_time_diff_sign;
467 const struct timeval max_deviation = {0, 500 * 1000};
468 const int time_smooth = 5;
469
470 sign = tv_diff(status_time, now, &diff);
471 // PARA_NOTICE_LOG("%s: sign = %i, sa_time_diff_sign = %i\n", __func__,
472 // sign, sa_time_diff_sign);
473 if (!count) {
474 sa_time_diff_sign = sign;
475 stat_task->sa_time_diff = diff;
476 count++;
477 return;
478 }
479 if (count > 5) {
480 int s = tv_diff(&diff, &stat_task->sa_time_diff, &tmp);
481 if (tv_diff(&max_deviation, &tmp, NULL) < 0)
482 PARA_WARNING_LOG("time diff jump: %lims\n",
483 s * tv2ms(&tmp));
484 }
485 count++;
486 sa_time_diff_sign = tv_convex_combination(
487 sa_time_diff_sign * time_smooth, &stat_task->sa_time_diff,
488 count > 10? sign : sign * time_smooth, &diff,
489 &tmp);
490 stat_task->sa_time_diff = tmp;
491 PARA_DEBUG_LOG("time diff (cur/avg): %s%lums/%s%lums\n",
492 sign > 0? "+" : "-",
493 tv2ms(&diff),
494 sa_time_diff_sign ? "+" : "-",
495 tv2ms(&stat_task->sa_time_diff)
496 );
497 }
498
499 static void check_stat_line(char *line)
500 {
501 int itemnum;
502 size_t ilen = 0;
503 long unsigned sec, usec;
504 char *tmp;
505
506 // PARA_INFO_LOG("line: %s\n", line);
507 if (!line)
508 return;
509 itemnum = stat_line_valid(line);
510 if (itemnum < 0) {
511 PARA_WARNING_LOG("invalid status line: %s\n", line);
512 return;
513 }
514 tmp = make_message("%s\n", line);
515 stat_client_write(tmp, itemnum);
516 free(tmp);
517 free(stat_task->stat_item_values[itemnum]);
518 stat_task->stat_item_values[itemnum] = para_strdup(line);
519 ilen = strlen(status_item_list[itemnum]);
520 switch (itemnum) {
521 case SI_STATUS:
522 stat_task->playing = strstr(line, "playing")? 1 : 0;
523 break;
524 case SI_FORMAT:
525 free(stat_task->af_status);
526 stat_task->af_status = para_strdup(line + ilen + 1);
527 break;
528 case SI_OFFSET:
529 stat_task->offset_seconds = atoi(line + ilen + 1);
530 break;
531 case SI_LENGTH:
532 stat_task->length_seconds = atoi(line + ilen + 1);
533 break;
534 case SI_STREAM_START:
535 if (sscanf(line + ilen + 1, "%lu.%lu", &sec, &usec) == 2) {
536 stat_task->server_stream_start.tv_sec = sec;
537 stat_task->server_stream_start.tv_usec = usec;
538 }
539 break;
540 case SI_CURRENT_TIME:
541 if (sscanf(line + ilen + 1, "%lu.%lu", &sec, &usec) == 2) {
542 struct timeval tv = {sec, usec};
543 compute_time_diff(&tv);
544 }
545 break;
546 }
547 }
548
549 static void handle_signal(int sig)
550 {
551 switch (sig) {
552 case SIGINT:
553 case SIGTERM:
554 case SIGHUP:
555 PARA_EMERG_LOG("terminating on signal %d\n", sig);
556 clean_exit(EXIT_FAILURE, "caught deadly signal");
557 return;
558 }
559 }
560
561 static void try_to_close_slot(int slot_num)
562 {
563 struct slot_info *s = &slot[slot_num];
564
565 if (s->format < 0)
566 return;
567 if (s->receiver_node && !s->receiver_node->eof)
568 return;
569 if (s->fc && !s->fc->eof)
570 return;
571 if (s->wng && !s->wng->eof)
572 return;
573 PARA_INFO_LOG("closing slot %d \n", slot_num);
574 wng_close(s->wng);
575 close_filters(s->fc);
576 free(s->fc);
577 close_receiver(slot_num);
578 clear_slot(slot_num);
579 }
580
581 /*
582 * Check if any receivers/filters/writers need to be started and do so if
583 * neccessary. Since the pre_select function didn't have a chance yet to put
584 * file descriptors into the fd sets given by s, make the upcoming select()
585 * return immediately to avoid a long timeout in case we started something.
586 */
587 static void audiod_pre_select(struct sched *s, __a_unused struct task *t)
588 {
589 int i;
590 struct timeval min_delay = {0, 1};
591
592 t->ret = 1;
593 if (audiod_status != AUDIOD_ON || !stat_task->playing)
594 return kill_all_decoders();
595 if (open_current_receiver(s))
596 s->timeout = min_delay;
597 FOR_EACH_SLOT(i) {
598 struct slot_info *sl = &slot[i];
599 struct audio_format_info *a;
600
601 if (sl->format < 0)
602 continue;
603 a = &afi[sl->format];
604 if (!sl->receiver_node)
605 continue;
606 if (!a->num_filters) {
607 if (sl->receiver_node->loaded && !sl->wng) {
608 open_writers(i);
609 s->timeout = min_delay;
610 }
611 continue;
612 }
613 if (sl->receiver_node->loaded && !sl->fc) {
614 open_filters(i);
615 s->timeout = min_delay;
616 continue;
617 }
618 if (sl->fc && *sl->fc->out_loaded && !sl->wng) {
619 open_writers(i);
620 s->timeout = min_delay;
621 }
622 }
623 }
624
625 static void audiod_post_select(__a_unused struct sched *s,
626 __a_unused struct task *t)
627 {
628 int i;
629
630 /* save away the current time for other users */
631 t->ret = 1;
632 FOR_EACH_SLOT(i)
633 try_to_close_slot(i);
634 }
635
636 static void init_audiod_task(struct task *t)
637 {
638 t->pre_select = audiod_pre_select;
639 t->post_select = audiod_post_select;
640 t->event_handler = NULL;
641 t->private_data = t;
642 sprintf(t->status, "audiod task");
643 }
644
645 static int parse_stream_command(const char *txt, char **cmd)
646 {
647 char *p = strchr(txt, ':');
648 int i;
649
650 if (!p)
651 return -E_MISSING_COLON;
652 p++;
653 FOR_EACH_AUDIO_FORMAT(i) {
654 if (strncmp(txt, audio_formats[i], strlen(audio_formats[i])))
655 continue;
656 *cmd = p;
657 return i;
658 }
659 return -E_UNSUPPORTED_AUDIO_FORMAT;
660 }
661
662 static int add_filter(int format, char *cmdline)
663 {
664 struct audio_format_info *a = &afi[format];
665 int filter_num, nf = a->num_filters;
666
667 filter_num = check_filter_arg(cmdline, &a->filter_conf[nf]);
668 if (filter_num < 0)
669 return filter_num;
670 a->filters[nf] = &filters[filter_num];
671 a->num_filters++;
672 PARA_INFO_LOG("%s filter %d: %s\n", audio_formats[format], nf + 1,
673 a->filters[nf]->name);
674 return filter_num;
675 }
676
677 static int init_writers(void)
678 {
679 int i, ret, nw;
680 char *cmd;
681 struct audio_format_info *a;
682
683 init_supported_writers();
684 nw = PARA_MAX(1, conf.writer_given);
685 PARA_INFO_LOG("maximal number of writers: %d\n", nw);
686 FOR_EACH_AUDIO_FORMAT(i) {
687 a = &afi[i];
688 a->writer_conf = para_malloc(nw * sizeof(void *));
689 a->writers = para_malloc(nw * sizeof(struct writer *));
690 a->num_writers = 0;
691 }
692 for (i = 0; i < conf.writer_given; i++) {
693 void *wconf;
694 int writer_num;
695 ret = parse_stream_command(conf.writer_arg[i], &cmd);
696 if (ret < 0)
697 goto out;
698 a = &afi[ret];
699 nw = a->num_writers;
700 wconf = check_writer_arg(cmd, &writer_num);
701 if (!wconf) {
702 ret = writer_num;
703 goto out;
704 }
705 a->writers[nw] = &writers[writer_num];
706 a->writer_conf[nw] = wconf;
707 PARA_INFO_LOG("%s writer #%d: %s\n", audio_formats[ret],
708 nw, writer_names[writer_num]);
709 a->num_writers++;
710 }
711 ret = 1;
712 out:
713 return ret;
714 }
715
716 static int init_receivers(void)
717 {
718 int i, ret, receiver_num;
719 char *cmd = NULL;
720 struct audio_format_info *a;
721
722 for (i = 0; receivers[i].name; i++) {
723 PARA_INFO_LOG("initializing %s receiver\n", receivers[i].name);
724 receivers[i].init(&receivers[i]);
725 }
726 for (i = 0; i < conf.receiver_given; i++) {
727 char *arg = conf.receiver_arg[i];
728 char *recv = strchr(arg, ':');
729 ret = -E_MISSING_COLON;
730 if (!recv)
731 goto out;
732 *recv = '\0';
733 recv++;
734 ret = get_audio_format_num(arg);
735 if (ret < 0)
736 goto out;
737 afi[ret].receiver_conf = check_receiver_arg(recv, &receiver_num);
738 if (!afi[ret].receiver_conf) {
739 ret = -E_RECV_SYNTAX;
740 goto out;
741 }
742 afi[ret].receiver = &receivers[receiver_num];
743 }
744 /* use the first available receiver with no arguments
745 * for those audio formats for which no receiver
746 * was specified
747 */
748 cmd = para_strdup(receivers[0].name);
749 FOR_EACH_AUDIO_FORMAT(i) {
750 a = &afi[i];
751 if (a->receiver_conf)
752 continue;
753 a->receiver_conf = check_receiver_arg(cmd, &receiver_num);
754 if (!a->receiver_conf)
755 return -E_RECV_SYNTAX;
756 a->receiver = &receivers[receiver_num];
757 }
758 ret = 1;
759 out:
760 free(cmd);
761 return ret;
762 }
763
764 static int init_default_filters(void)
765 {
766 int i, ret = 1;
767
768 FOR_EACH_AUDIO_FORMAT(i) {
769 struct audio_format_info *a = &afi[i];
770 char *tmp;
771 int j;
772
773 if (a->num_filters)
774 continue; /* no default -- nothing to to */
775 /* add "dec" to audio format name */
776 tmp = make_message("%sdec", audio_formats[i]);
777 for (j = 0; filters[j].name; j++)
778 if (!strcmp(tmp, filters[j].name))
779 break;
780 free(tmp);
781 ret = -E_UNSUPPORTED_FILTER;
782 if (!filters[j].name)
783 goto out;
784 tmp = para_strdup(filters[j].name);
785 ret = add_filter(i, tmp);
786 free(tmp);
787 if (ret < 0)
788 goto out;
789 PARA_INFO_LOG("%s -> default filter: %s\n", audio_formats[i],
790 filters[j].name);
791 }
792 out:
793 return ret;
794 }
795
796 static int init_filters(void)
797 {
798 int i, ret, nf;
799
800 filter_init(filters);
801 nf = PARA_MAX(1, conf.filter_given);
802 PARA_INFO_LOG("maximal number of filters: %d\n", nf);
803 FOR_EACH_AUDIO_FORMAT(i) {
804 afi[i].filter_conf = para_malloc(nf * sizeof(void *));
805 afi[i].filters = para_malloc(nf * sizeof(struct filter *));
806 }
807 if (!conf.no_default_filters_given)
808 return init_default_filters();
809 for (i = 0; i < conf.filter_given; i++) {
810 char *arg = conf.filter_arg[i];
811 char *filter_name = strchr(arg, ':');
812 ret = -E_MISSING_COLON;
813 if (!filter_name)
814 goto out;
815 *filter_name = '\0';
816 filter_name++;
817 ret = get_audio_format_num(arg);
818 if (ret < 0)
819 goto out;
820 ret = add_filter(ret, filter_name);
821 if (ret < 0)
822 goto out;
823 }
824 ret = init_default_filters(); /* use default values for the rest */
825 out:
826 return ret;
827 }
828
829 static int init_stream_io(void)
830 {
831 int ret;
832
833 ret = init_writers();
834 if (ret < 0)
835 return ret;
836 ret = init_receivers();
837 if (ret < 0)
838 return ret;
839 ret = init_filters();
840 if (ret < 0)
841 return ret;
842 return 1;
843 }
844
845 static int audiod_get_socket(void)
846 {
847 struct sockaddr_un unix_addr;
848 int fd;
849
850 if (conf.socket_given)
851 socket_name = para_strdup(conf.socket_arg);
852 else {
853 char *hn = para_hostname();
854 socket_name = make_message("/var/paraslash/audiod_socket.%s",
855 hn);
856 free(hn);
857 }
858 PARA_NOTICE_LOG("local socket: %s\n", socket_name);
859 if (conf.force_given)
860 unlink(socket_name);
861 fd = create_pf_socket(socket_name, &unix_addr,
862 S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IWOTH);
863 if (fd < 0) {
864 PARA_EMERG_LOG("%s", "can not connect to socket\n");
865 exit(EXIT_FAILURE); /* do not unlink socket */
866 }
867 if (listen(fd , 5) < 0) {
868 PARA_EMERG_LOG("%s", "can not listen on socket\n");
869 exit(EXIT_FAILURE); /* do not unlink socket */
870 }
871 mark_fd_nonblock(fd);
872 return fd;
873 }
874
875 static void signal_event_handler(struct task *t)
876 {
877 struct signal_task *st = t->private_data;
878
879 if (t->ret != -E_SIGNAL_CAUGHT)
880 PARA_ERROR_LOG("%s (ignored)\n", PARA_STRERROR(-t->ret));
881 else
882 handle_signal(st->signum);
883 }
884
885 static void signal_pre_select(struct sched *s, struct task *t)
886 {
887 struct signal_task *st = t->private_data;
888 t->ret = 1;
889 para_fd_set(st->fd, &s->rfds, &s->max_fileno);
890 }
891
892 static void signal_post_select(struct sched *s, struct task *t)
893 {
894 struct signal_task *st = t->private_data;
895 t->ret = 1;
896 if (!FD_ISSET(st->fd, &s->rfds))
897 return;
898 t->ret = -E_SIGNAL_CAUGHT;
899 st->signum = para_next_signal();
900 }
901
902 static void signal_setup_default(struct signal_task *st)
903 {
904 st->task.pre_select = signal_pre_select;
905 st->task.post_select = signal_post_select;
906 st->task.private_data = st;
907 sprintf(st->task.status, "signal task");
908 }
909
910 static void command_pre_select(struct sched *s, struct task *t)
911 {
912 struct command_task *ct = t->private_data;
913 t->ret = 1;
914 para_fd_set(ct->fd, &s->rfds, &s->max_fileno);
915
916 }
917
918 static void command_post_select(struct sched *s, struct task *t)
919 {
920 int ret;
921 struct command_task *ct = t->private_data;
922
923 t->ret = 1; /* always successful */
924 if (audiod_status != AUDIOD_OFF)
925 audiod_status_dump();
926 if (!FD_ISSET(ct->fd, &s->rfds))
927 return;
928 ret = handle_connect(ct->fd);
929 if (ret < 0)
930 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
931 }
932
933 static void init_command_task(struct command_task *ct)
934 {
935 ct->task.pre_select = command_pre_select;
936 ct->task.post_select = command_post_select;
937 ct->task.event_handler = NULL;
938 ct->task.private_data = ct;
939 ct->fd = audiod_get_socket(); /* doesn't return on errors */
940 sprintf(ct->task.status, "command task");
941 }
942
943 static void client_task_event_handler(__a_unused struct task *t)
944 {
945 struct private_client_data *pcd = t->private_data;
946 if (t->ret == -E_HANDSHAKE_COMPLETE)
947 return;
948 unregister_task(t);
949 pcd->eof = 1;
950 }
951
952 static void status_event_handler(__a_unused struct task *t)
953 {
954 struct timeval delay = {1, 0};
955 int i;
956 struct status_task *st = t->private_data;
957
958 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
959 close_stat_pipe();
960 /* avoid busy loop if server is down */
961 tv_add(now, &delay, &st->restart_barrier);
962 FOR_EACH_AUDIO_FORMAT(i)
963 afi[i].restart_barrier = st->restart_barrier;
964 }
965
966 static void status_pre_select(struct sched *s, struct task *t)
967 {
968 struct status_task *st = t->private_data;
969 int argc = 2;
970 char *argv[] = {"audiod", "stat", NULL};
971 t->ret = 1;
972 if (st->pcd && (audiod_status == AUDIOD_OFF || st->pcd->eof))
973 close_stat_pipe();
974 if (!st->pcd && audiod_status != AUDIOD_OFF
975 && tv_diff(now, &st->restart_barrier, NULL) > 0) {
976 t->ret = client_parse_config(argc, argv, &st->pcd);
977 if (t->ret < 0)
978 return;
979 t->ret = client_open(st->pcd);
980 if (t->ret < 0)
981 return;
982 st->pcd->task.event_handler = client_task_event_handler;
983 s->timeout.tv_sec = 0;
984 s->timeout.tv_usec = 1;
985 }
986 }
987
988 static void status_post_select(__a_unused struct sched *s, struct task *t)
989 {
990 struct status_task *st = t->private_data;
991
992 t->ret = 1;
993 if (!st->pcd || !st->pcd->loaded
994 || st->pcd->status != CL_RECEIVING)
995 return;
996 st->pcd->loaded = for_each_line(st->pcd->buf, st->pcd->loaded,
997 &check_stat_line);
998 }
999
1000 static void init_status_task(struct status_task *st)
1001 {
1002 memset(st, 0, sizeof(struct status_task));
1003 st->task.pre_select = status_pre_select;
1004 st->task.post_select = status_post_select;
1005 st->task.event_handler = status_event_handler;
1006 st->task.private_data = st;
1007 st->sa_time_diff_sign = 1;
1008 sprintf(st->task.status, "status task");
1009 }
1010
1011 static void set_initial_status(void)
1012 {
1013 audiod_status = AUDIOD_ON;
1014 if (!conf.mode_given)
1015 return;
1016 if (!strcmp(conf.mode_arg, "sb")) {
1017 audiod_status = AUDIOD_STANDBY;
1018 return;
1019 }
1020 if (!strcmp(conf.mode_arg, "off")) {
1021 audiod_status = AUDIOD_OFF;
1022 return;
1023 }
1024 if (strcmp(conf.mode_arg, "on"))
1025 PARA_WARNING_LOG("%s", "invalid mode\n");
1026 }
1027
1028 int main(int argc, char *argv[])
1029 {
1030 char *cf;
1031 int ret, i;
1032 struct sched s;
1033 struct command_task command_task_struct, *cmd_task = &command_task_struct;
1034 struct task audiod_task_struct, *audiod_task = &audiod_task_struct;
1035
1036 valid_fd_012();
1037 audiod_cmdline_parser(argc, argv, &conf);
1038 para_drop_privileges(conf.user_arg, conf.group_arg);
1039 cf = configfile_exists();
1040 if (cf) {
1041 if (audiod_cmdline_parser_configfile(cf, &conf, 0, 0, 0)) {
1042 PARA_EMERG_LOG("%s", "parse error in config file\n");
1043 exit(EXIT_FAILURE);
1044 }
1045 }
1046 if (conf.logfile_given)
1047 logfile = open_log(conf.logfile_arg);
1048 log_welcome("para_audiod", conf.loglevel_arg);
1049 i = init_stream_io();
1050 if (i < 0) {
1051 PARA_EMERG_LOG("init stream io error: %s\n", PARA_STRERROR(-i));
1052 exit(EXIT_FAILURE);
1053 }
1054 server_uptime(UPTIME_SET);
1055 set_initial_status();
1056 FOR_EACH_SLOT(i)
1057 clear_slot(i);
1058 init_grabbing();
1059 setup_signal_handling();
1060 signal_setup_default(sig_task);
1061 sig_task->task.event_handler = signal_event_handler;
1062
1063 init_status_task(stat_task);
1064 init_command_task(cmd_task);
1065 init_audiod_task(audiod_task);
1066
1067 if (conf.daemon_given)
1068 daemon_init();
1069
1070 register_task(&sig_task->task);
1071 register_task(&cmd_task->task);
1072 register_task(&stat_task->task);
1073 register_task(audiod_task);
1074 s.default_timeout.tv_sec = 0;
1075 s.default_timeout.tv_usec = 99 * 1000;
1076 ret = sched(&s);
1077
1078 PARA_EMERG_LOG("%s\n", PARA_STRERROR(-ret));
1079 return EXIT_FAILURE;
1080 }