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