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