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