write_ok(): Always reset the select timeout
[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         t->ret = 1;
665         FOR_EACH_SLOT(i)
666                 try_to_close_slot(i);
667 }
668
669 static void init_audiod_task(struct task *t)
670 {
671         t->pre_select = audiod_pre_select;
672         t->post_select = audiod_post_select;
673         t->event_handler = NULL;
674         t->private_data = t;
675         sprintf(t->status, "audiod task");
676 }
677
678 static int parse_stream_command(const char *txt, char **cmd)
679 {
680         char *p = strchr(txt, ':');
681         int i;
682
683         if (!p)
684                 return -E_MISSING_COLON;
685         p++;
686         FOR_EACH_AUDIO_FORMAT(i) {
687                 if (strncmp(txt, audio_formats[i], strlen(audio_formats[i])))
688                         continue;
689                 *cmd = p;
690                 return i;
691         }
692         return -E_UNSUPPORTED_AUDIO_FORMAT;
693 }
694
695 static int add_filter(int format, char *cmdline)
696 {
697         struct audio_format_info *a = &afi[format];
698         int filter_num, nf = a->num_filters;
699
700         filter_num = check_filter_arg(cmdline, &a->filter_conf[nf]);
701         if (filter_num < 0)
702                 return filter_num;
703         a->filters[nf] = &filters[filter_num];
704         a->num_filters++;
705         PARA_INFO_LOG("%s filter %d: %s\n", audio_formats[format], nf + 1,
706                 a->filters[nf]->name);
707         return filter_num;
708 }
709
710 static int init_writers(void)
711 {
712         int i, ret, nw;
713         char *cmd;
714         struct audio_format_info *a;
715
716         init_supported_writers();
717         nw = PARA_MAX(1, conf.writer_given);
718         PARA_INFO_LOG("maximal number of writers: %d\n", nw);
719         FOR_EACH_AUDIO_FORMAT(i) {
720                 a = &afi[i];
721                 a->writer_conf = para_malloc(nw * sizeof(void *));
722                 a->writers = para_malloc(nw * sizeof(struct writer *));
723                 a->num_writers = 0;
724         }
725         for (i = 0; i < conf.writer_given; i++) {
726                 void *wconf;
727                 int writer_num;
728                 ret = parse_stream_command(conf.writer_arg[i], &cmd);
729                 if (ret < 0)
730                         goto out;
731                 a = &afi[ret];
732                 nw = a->num_writers;
733                 wconf = check_writer_arg(cmd, &writer_num);
734                 if (!wconf) {
735                         ret = writer_num;
736                         goto out;
737                 }
738                 a->writers[nw] = &writers[writer_num];
739                 a->writer_conf[nw] = wconf;
740                 PARA_INFO_LOG("%s writer #%d: %s\n", audio_formats[ret],
741                         nw, writer_names[writer_num]);
742                 a->num_writers++;
743         }
744         ret = 1;
745 out:
746         return ret;
747 }
748
749 static int init_receivers(void)
750 {
751         int i, ret, receiver_num;
752         char *cmd = NULL;
753         struct audio_format_info *a;
754
755         for (i = 0; receivers[i].name; i++) {
756                 PARA_INFO_LOG("initializing %s receiver\n", receivers[i].name);
757                 receivers[i].init(&receivers[i]);
758         }
759         for (i = conf.receiver_given - 1; i >= 0; i--) {
760                 char *arg = conf.receiver_arg[i];
761                 char *recv_arg = strchr(arg, ':');
762                 PARA_INFO_LOG("arg: %s\n", arg);
763                 ret = -E_MISSING_COLON;
764                 if (!recv_arg)
765                         goto out;
766                 *recv_arg = '\0';
767                 recv_arg++;
768                 ret = get_audio_format_num(arg);
769                 if (ret < 0)
770                         goto out;
771                 afi[ret].receiver_conf = check_receiver_arg(recv_arg, &receiver_num);
772                 if (!afi[ret].receiver_conf) {
773                         ret = -E_RECV_SYNTAX;
774                         goto out;
775                 }
776                 afi[ret].receiver = &receivers[receiver_num];
777         }
778         /* use the first available receiver with no arguments
779          * for those audio formats for which no receiver
780          * was specified
781          */
782         cmd = para_strdup(receivers[0].name);
783         FOR_EACH_AUDIO_FORMAT(i) {
784                 a = &afi[i];
785                 if (a->receiver_conf)
786                         continue;
787                 a->receiver_conf = check_receiver_arg(cmd, &receiver_num);
788                 if (!a->receiver_conf)
789                         return -E_RECV_SYNTAX;
790                 a->receiver = &receivers[receiver_num];
791         }
792         ret = 1;
793 out:
794         free(cmd);
795         return ret;
796 }
797
798 static int init_default_filters(void)
799 {
800         int i, ret = 1;
801
802         FOR_EACH_AUDIO_FORMAT(i) {
803                 struct audio_format_info *a = &afi[i];
804                 char *tmp;
805                 int j;
806
807                 if (a->num_filters)
808                         continue; /* no default -- nothing to to */
809                 /* add "dec" to audio format name */
810                 tmp = make_message("%sdec", audio_formats[i]);
811                 for (j = 0; filters[j].name; j++)
812                         if (!strcmp(tmp, filters[j].name))
813                                 break;
814                 free(tmp);
815                 ret = -E_UNSUPPORTED_FILTER;
816                 if (!filters[j].name)
817                         goto out;
818                 tmp = para_strdup(filters[j].name);
819                 ret = add_filter(i, tmp);
820                 free(tmp);
821                 if (ret < 0)
822                         goto out;
823                 PARA_INFO_LOG("%s -> default filter: %s\n", audio_formats[i],
824                         filters[j].name);
825         }
826 out:
827         return ret;
828 }
829
830 static int init_filters(void)
831 {
832         int i, ret, nf;
833
834         filter_init(filters);
835         nf = PARA_MAX(1,  conf.filter_given);
836         PARA_INFO_LOG("maximal number of filters: %d\n", nf);
837         FOR_EACH_AUDIO_FORMAT(i) {
838                 afi[i].filter_conf = para_malloc(nf * sizeof(void *));
839                 afi[i].filters = para_malloc(nf * sizeof(struct filter *));
840         }
841         if (!conf.no_default_filters_given)
842                 return init_default_filters();
843         for (i = 0; i < conf.filter_given; i++) {
844                 char *arg = conf.filter_arg[i];
845                 char *filter_name = strchr(arg, ':');
846                 ret = -E_MISSING_COLON;
847                 if (!filter_name)
848                         goto out;
849                 *filter_name = '\0';
850                 filter_name++;
851                 ret = get_audio_format_num(arg);
852                 if (ret < 0)
853                         goto out;
854                 ret = add_filter(ret, filter_name);
855                 if (ret < 0)
856                         goto out;
857         }
858         ret = init_default_filters(); /* use default values for the rest */
859 out:
860         return ret;
861 }
862
863 static int init_stream_io(void)
864 {
865         int ret;
866
867         ret = init_writers();
868         if (ret < 0)
869                 return ret;
870         ret = init_receivers();
871         if (ret < 0)
872                 return ret;
873         ret = init_filters();
874         if (ret < 0)
875                 return ret;
876         return 1;
877 }
878
879 static int audiod_get_socket(void)
880 {
881         struct sockaddr_un unix_addr;
882         int fd;
883
884         if (conf.socket_given)
885                 socket_name = para_strdup(conf.socket_arg);
886         else {
887                 char *hn = para_hostname();
888                 socket_name = make_message("/var/paraslash/audiod_socket.%s",
889                         hn);
890                 free(hn);
891         }
892         PARA_NOTICE_LOG("local socket: %s\n", socket_name);
893         if (conf.force_given)
894                 unlink(socket_name);
895         fd = create_pf_socket(socket_name, &unix_addr,
896                         S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IWOTH);
897         if (fd < 0) {
898                 PARA_EMERG_LOG("%s", "can not connect to socket\n");
899                 exit(EXIT_FAILURE); /* do not unlink socket */
900         }
901         if (listen(fd , 5) < 0) {
902                 PARA_EMERG_LOG("%s", "can not listen on socket\n");
903                 exit(EXIT_FAILURE); /* do not unlink socket */
904         }
905         mark_fd_nonblock(fd);
906         return fd;
907 }
908
909 static void signal_event_handler(struct task *t)
910 {
911         struct signal_task *st = t->private_data;
912
913         if (t->ret != -E_SIGNAL_CAUGHT)
914                 PARA_ERROR_LOG("%s (ignored)\n", PARA_STRERROR(-t->ret));
915         else
916                 handle_signal(st->signum);
917 }
918
919 static void signal_pre_select(struct sched *s, struct task *t)
920 {
921         struct signal_task *st = t->private_data;
922         t->ret = 1;
923         para_fd_set(st->fd, &s->rfds, &s->max_fileno);
924 }
925
926 static void signal_post_select(struct sched *s, struct task *t)
927 {
928         struct signal_task *st = t->private_data;
929         t->ret = 1;
930         if (!FD_ISSET(st->fd, &s->rfds))
931                 return;
932         t->ret = -E_SIGNAL_CAUGHT;
933         st->signum = para_next_signal();
934 }
935
936 static void signal_setup_default(struct signal_task *st)
937 {
938         st->task.pre_select = signal_pre_select;
939         st->task.post_select = signal_post_select;
940         st->task.private_data = st;
941         sprintf(st->task.status, "signal task");
942 }
943
944 static void command_pre_select(struct sched *s, struct task *t)
945 {
946         struct command_task *ct = t->private_data;
947         t->ret = 1;
948         para_fd_set(ct->fd, &s->rfds, &s->max_fileno);
949
950 }
951
952 static void command_post_select(struct sched *s, struct task *t)
953 {
954         int ret;
955         struct command_task *ct = t->private_data;
956
957         t->ret = 1; /* always successful */
958         audiod_status_dump();
959         if (!FD_ISSET(ct->fd, &s->rfds))
960                 return;
961         ret = handle_connect(ct->fd);
962         if (ret < 0)
963                 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
964 }
965
966 static void init_command_task(struct command_task *ct)
967 {
968         ct->task.pre_select = command_pre_select;
969         ct->task.post_select = command_post_select;
970         ct->task.event_handler = NULL;
971         ct->task.private_data = ct;
972         ct->fd = audiod_get_socket(); /* doesn't return on errors */
973         sprintf(ct->task.status, "command task");
974 }
975
976 static void client_task_event_handler(__a_unused struct task *t)
977 {
978         struct private_client_data *pcd = t->private_data;
979         if (t->ret == -E_HANDSHAKE_COMPLETE)
980                 return;
981         unregister_task(t);
982         pcd->eof = 1;
983 }
984
985 static void status_event_handler(__a_unused struct task *t)
986 {
987         struct timeval delay = {1, 0};
988         int i;
989         struct status_task *st = t->private_data;
990
991         PARA_ERROR_LOG("%s\n", PARA_STRERROR(-t->ret));
992         close_stat_pipe();
993         /* avoid busy loop if server is down */
994         tv_add(now, &delay, &st->restart_barrier);
995         FOR_EACH_AUDIO_FORMAT(i)
996                 afi[i].restart_barrier = st->restart_barrier;
997 }
998
999 static void status_pre_select(struct sched *s, struct task *t)
1000 {
1001         struct status_task *st = t->private_data;
1002         int argc = 2;
1003         char *argv[] = {"audiod", "stat", NULL};
1004         t->ret = 1;
1005         if (st->pcd && (audiod_status == AUDIOD_OFF || st->pcd->eof))
1006                 close_stat_pipe();
1007         if (!st->pcd && audiod_status != AUDIOD_OFF
1008                 && tv_diff(now, &st->restart_barrier, NULL) > 0) {
1009                 t->ret = client_parse_config(argc, argv, &st->pcd);
1010                 if (t->ret < 0)
1011                         return;
1012                 t->ret = client_open(st->pcd);
1013                 if (t->ret < 0)
1014                         return;
1015                 st->pcd->task.event_handler = client_task_event_handler;
1016                 s->timeout.tv_sec = 0;
1017                 s->timeout.tv_usec = 1;
1018         }
1019 }
1020
1021 static void status_post_select(__a_unused struct sched *s, struct task *t)
1022 {
1023         struct status_task *st = t->private_data;
1024
1025         t->ret = 1;
1026         if (!st->pcd || !st->pcd->loaded
1027                         || st->pcd->status != CL_RECEIVING)
1028                 return;
1029         st->pcd->loaded = for_each_line(st->pcd->buf, st->pcd->loaded,
1030                 &check_stat_line);
1031 }
1032
1033 static void init_status_task(struct status_task *st)
1034 {
1035         memset(st, 0, sizeof(struct status_task));
1036         st->task.pre_select = status_pre_select;
1037         st->task.post_select = status_post_select;
1038         st->task.event_handler = status_event_handler;
1039         st->task.private_data = st;
1040         st->sa_time_diff_sign = 1;
1041         sprintf(st->task.status, "status task");
1042 }
1043
1044 static void set_initial_status(void)
1045 {
1046         audiod_status = AUDIOD_ON;
1047         if (!conf.mode_given)
1048                 return;
1049         if (!strcmp(conf.mode_arg, "sb")) {
1050                 audiod_status = AUDIOD_STANDBY;
1051                 return;
1052         }
1053         if (!strcmp(conf.mode_arg, "off")) {
1054                 audiod_status = AUDIOD_OFF;
1055                 return;
1056         }
1057         if (strcmp(conf.mode_arg, "on"))
1058                 PARA_WARNING_LOG("%s", "invalid mode\n");
1059 }
1060
1061 int main(int argc, char *argv[])
1062 {
1063         char *cf;
1064         int ret, i;
1065         struct sched s;
1066         struct command_task command_task_struct, *cmd_task = &command_task_struct;
1067         struct task audiod_task_struct, *audiod_task = &audiod_task_struct;
1068
1069         valid_fd_012();
1070         audiod_cmdline_parser(argc, argv, &conf);
1071         para_drop_privileges(conf.user_arg, conf.group_arg);
1072         cf = configfile_exists();
1073         if (cf) {
1074                 if (audiod_cmdline_parser_configfile(cf, &conf, 0, 0, 0)) {
1075                         PARA_EMERG_LOG("%s", "parse error in config file\n");
1076                         exit(EXIT_FAILURE);
1077                 }
1078         }
1079         if (conf.logfile_given)
1080                 logfile = open_log(conf.logfile_arg);
1081         log_welcome("para_audiod", conf.loglevel_arg);
1082         i = init_stream_io();
1083         if (i < 0) {
1084                 PARA_EMERG_LOG("init stream io error: %s\n", PARA_STRERROR(-i));
1085                 exit(EXIT_FAILURE);
1086         }
1087         server_uptime(UPTIME_SET);
1088         set_initial_status();
1089         FOR_EACH_SLOT(i)
1090                 clear_slot(i);
1091         init_grabbing();
1092         setup_signal_handling();
1093         signal_setup_default(sig_task);
1094         sig_task->task.event_handler = signal_event_handler;
1095
1096         init_status_task(stat_task);
1097         init_command_task(cmd_task);
1098         init_audiod_task(audiod_task);
1099
1100         if (conf.daemon_given)
1101                 daemon_init();
1102
1103         register_task(&sig_task->task);
1104         register_task(&cmd_task->task);
1105         register_task(&stat_task->task);
1106         register_task(audiod_task);
1107         s.default_timeout.tv_sec = 0;
1108         s.default_timeout.tv_usec = 99 * 1000;
1109         ret = sched(&s);
1110
1111         PARA_EMERG_LOG("%s\n", PARA_STRERROR(-ret));
1112         return EXIT_FAILURE;
1113 }