audiod.c

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