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