afs.c

   1 /*
   2  * Copyright (C) 1997-2006 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 afs.c audio file sending functions
  20  *
  21  * This contains the audio sending part of para_server which is independent of
  22  * the current audio format, audio file selector and of the activated senders.
  23  */
  24
  25 #include "server.h"
  26 #include <sys/time.h> /* gettimeofday */
  27 #include "server.cmdline.h"
  28 #include "db.h"
  29 #include "afh.h"
  30 #include "afs.h"
  31 #include "send.h"
  32 #include "error.h"
  33 #include "string.h"
  34
  35 extern const char *status_item_list[];
  36
  37 static struct timeval announce_tv;
  38 static struct timeval data_send_barrier;
  39 static struct timeval eof_barrier;
  40
  41 extern struct misc_meta_data *mmd;
  42 extern struct audio_file_selector selectors[];
  43 extern struct sender senders[];
  44
  45 static FILE *audio_file = NULL;
  46
  47 #if 1
  48         void mp3_init(struct audio_format_handler *);
  49 #endif
  50
  51 #ifdef HAVE_OGGVORBIS
  52         void ogg_init(struct audio_format_handler *);
  53 #endif
  54 #ifdef HAVE_FAAD
  55         void aac_afh_init(struct audio_format_handler *);
  56 #endif
  57
  58 /**
  59  * the list of supported  audio formats
  60  */
  61 static struct audio_format_handler afl[] = {
  62 #if 1
  63         {
  64                 .name = "mp3",
  65                 .init = mp3_init,
  66         },
  67 #endif
  68 #ifdef HAVE_OGGVORBIS
  69         {
  70                 .name = "ogg",
  71                 .init = ogg_init,
  72         },
  73 #endif
  74 #ifdef HAVE_FAAD
  75         {
  76                 .name = "aac",
  77                 .init = aac_afh_init,
  78         },
  79 #endif
  80         {
  81                 .name = NULL,
  82         }
  83 };
  84 #define FOR_EACH_AUDIO_FORMAT(i) for (i = 0; afl[i].name; i++)
  85
  86 /**
  87  * check if audio file sender is playing
  88  *
  89  * \return greater than zero if playing, zero otherwise.
  90  *
  91  */
  92 unsigned int afs_playing(void)
  93 {
  94         return mmd->new_afs_status_flags & AFS_PLAYING;
  95 }
  96
  97 /**
  98  * check if 'next' flag is set afs_status_flags
  99  *
 100  * \return greater than zero if set, zero if not.
 101  *
 102  */
 103 unsigned int afs_next(void)
 104 {
 105         return mmd->new_afs_status_flags & AFS_NEXT;
 106 }
 107
 108 /**
 109  * check if a reposition request is pending
 110  *
 111  * \return greater than zero if true, zero otherwise.
 112  *
 113  */
 114 unsigned int afs_repos(void)
 115 {
 116         return mmd->new_afs_status_flags & AFS_REPOS;
 117 }
 118
 119 /**
 120  * check if audio file sender is paused
 121  *
 122  * \return greater than zero if paused, zero otherwise.
 123  *
 124  */
 125 unsigned int afs_paused(void)
 126 {
 127         return !(mmd->new_afs_status_flags & AFS_NEXT)
 128                 && !(mmd->new_afs_status_flags & AFS_PLAYING);
 129 }
 130
 131 /**
 132  * get the name of the given audio format
 133  * \param i the audio format number
 134  *
 135  * This returns a pointer to statically allocated memory so it
 136  * must not be freed by the caller.
 137  */
 138 const char *audio_format_name(int i)
 139 {
 140         return i >= 0?  afl[i].name : "(none)";
 141 }
 142
 143 /**
 144  * initialize the audio file sender
 145  *
 146  * Call the init functions of all supported audio format handlers and
 147  * initialize all supported senders.
 148  */
 149 void afs_init(void)
 150 {
 151         int i;
 152         char *hn = para_hostname(), *home = para_homedir();
 153
 154         PARA_DEBUG_LOG("supported audio formats: %s\n",
 155                 SUPPORTED_AUDIO_FORMATS);
 156         for (i = 0; afl[i].name; i++) {
 157                 PARA_NOTICE_LOG("initializing %s handler\n",
 158                         afl[i].name);
 159                 afl[i].init(&afl[i]);
 160         }
 161         ms2tv(conf.announce_time_arg, &announce_tv);
 162         PARA_INFO_LOG("announce timeval: %lums\n", tv2ms(&announce_tv));
 163         for (i = 0; senders[i].name; i++) {
 164                 PARA_NOTICE_LOG("initializing %s sender\n", senders[i].name);
 165                 senders[i].init(&senders[i]);
 166         }
 167         free(hn);
 168         free(home);
 169 }
 170
 171 static int get_file_info(int i)
 172 {
 173         return  afl[i].get_file_info(audio_file, mmd->audio_file_info,
 174                 &mmd->chunks_total, &mmd->seconds_total);
 175 }
 176
 177 /**
 178  * guess the audio format judging from filename
 179  *
 180  * \param name the filename
 181  *
 182  * \return This function returns -1 if it has no idea what kind of audio
 183  * file this might be. Otherwise the (non-negative) number of the audio format
 184  * is returned.
 185  */
 186 int guess_audio_format(const char *name)
 187 {
 188         int i,j, len = strlen(name);
 189
 190         FOR_EACH_AUDIO_FORMAT(i) {
 191                 for (j = 0; afl[i].suffixes[j]; j++) {
 192                         const char *p = afl[i].suffixes[j];
 193                         int plen = strlen(p);
 194                         if (len < plen + 1)
 195                                 continue;
 196                         if (name[len - plen - 1] != '.')
 197                                 continue;
 198                         if (strcasecmp(name + len - plen, p))
 199                                 continue;
 200 //                      PARA_DEBUG_LOG("might be %s\n", audio_format_name(i));
 201                         return i;
 202                 }
 203         }
 204         return -1;
 205 }
 206
 207 static int get_audio_format(int omit)
 208 {
 209         int i;
 210
 211         FOR_EACH_AUDIO_FORMAT(i) {
 212                 if (i == omit || !afl[i].get_file_info)
 213                         continue;
 214                 rewind(audio_file);
 215                 if (get_file_info(i) > 0)
 216                         return i;
 217                 rewind(audio_file);
 218         }
 219         return -E_AUDIO_FORMAT;
 220 }
 221
 222 /*
 223  * upddate shared mem
 224  */
 225 static int update_mmd(void)
 226 {
 227         int i;
 228         struct stat file_status;
 229
 230         i = guess_audio_format(mmd->filename);
 231         if (i < 0 || get_file_info(i) < 0)
 232                 i = get_audio_format(i);
 233         if (i < 0)
 234                 return i;
 235         mmd->audio_format = i;
 236         mmd->chunks_sent = 0;
 237         mmd->current_chunk = 0;
 238         mmd->offset = 0;
 239         if (fstat(fileno(audio_file), &file_status) == -1)
 240                 return -E_FSTAT;
 241         mmd->size = file_status.st_size;
 242         mmd->mtime = file_status.st_mtime;
 243         mmd->events++;
 244         PARA_NOTICE_LOG("next audio file: %s\n", mmd->filename);
 245         return 1;
 246 }
 247
 248 static void get_song(void)
 249 {
 250         char **sl = selectors[mmd->selector_num].get_audio_file_list(10);
 251         int i;
 252
 253         if (!sl)
 254                 goto err_out;
 255         for (i = 0; sl[i]; i++) {
 256                 struct timeval now;
 257                 PARA_INFO_LOG("trying %s\n", sl[i]);
 258                 if (strlen(sl[i]) >= _POSIX_PATH_MAX)
 259                         continue;
 260                 audio_file = fopen(sl[i], "r");
 261                 if (!audio_file)
 262                         continue;
 263                 strcpy(mmd->filename, sl[i]);
 264                 if (update_mmd() < 0) {
 265                         fclose(audio_file);
 266                         audio_file = NULL;
 267                         continue;
 268                 }
 269                 mmd->num_played++;
 270                 if (selectors[mmd->selector_num].update_audio_file)
 271                         selectors[mmd->selector_num].update_audio_file(sl[i]);
 272                 PARA_DEBUG_LOG("%s", "success\n");
 273                 mmd->new_afs_status_flags &= (~AFS_NEXT);
 274                 gettimeofday(&now, NULL);
 275                 tv_add(&now, &announce_tv, &data_send_barrier);
 276
 277                 goto free;
 278         }
 279         PARA_ERROR_LOG("%s", "no valid files found\n");
 280 err_out:
 281         mmd->new_afs_status_flags = AFS_NEXT;
 282 free:
 283         if (sl) {
 284                 for (i = 0; sl[i]; i++)
 285                         free(sl[i]);
 286                 free(sl);
 287         }
 288 }
 289
 290 static int chk_barrier(const char *bname, const struct timeval *now,
 291                 const struct timeval *barrier, struct timeval *diff, int log)
 292 {
 293         long ms;
 294
 295         if (tv_diff(now, barrier, diff) > 0)
 296                 return 1;
 297         ms = tv2ms(diff);
 298         if (log && ms)
 299                 PARA_DEBUG_LOG("%s barrier: %lims left\n", bname, ms);
 300         return -1;
 301 }
 302
 303 static void afs_next_chunk_time(struct timeval *due)
 304 {
 305         struct timeval tmp;
 306
 307         tv_scale(mmd->chunks_sent, &afl[mmd->audio_format].chunk_tv, &tmp);
 308         tv_add(&tmp, &mmd->stream_start, due);
 309 }
 310
 311 /*
 312  * != NULL: timeout for next chunk
 313  * NULL: nothing to do
 314  */
 315 static struct timeval *afs_compute_timeout(void)
 316 {
 317         static struct timeval the_timeout;
 318         struct timeval now, next_chunk;
 319
 320         if (afs_next() && mmd->audio_format >= 0) {
 321                 /* only sleep a bit, nec*/
 322                 the_timeout.tv_sec = 0;
 323                 the_timeout.tv_usec = 100;
 324                 return &the_timeout;
 325         }
 326         gettimeofday(&now, NULL);
 327         if (chk_barrier("eof", &now, &eof_barrier, &the_timeout, 1) < 0)
 328                 return &the_timeout;
 329         if (chk_barrier("data send", &now, &data_send_barrier,
 330                         &the_timeout, 1) < 0)
 331                 return &the_timeout;
 332         if (mmd->audio_format < 0 || !afs_playing() || !audio_file)
 333                 return NULL;
 334         afs_next_chunk_time(&next_chunk);
 335         if (chk_barrier(afl[mmd->audio_format].name, &now, &next_chunk,
 336                         &the_timeout, 0) < 0)
 337                 return &the_timeout;
 338         /* chunk is due or bof */
 339         the_timeout.tv_sec = 0;
 340         the_timeout.tv_usec = 0;
 341         return &the_timeout;
 342 }
 343
 344 static void afs_eof(struct audio_format_handler *af)
 345 {
 346         struct timeval now;
 347         int i;
 348         char *tmp;
 349
 350         if (!af || !audio_file) {
 351                 for (i = 0; senders[i].name; i++)
 352                         senders[i].shutdown_clients();
 353                 return;
 354         }
 355         gettimeofday(&now, NULL);
 356         tv_add(&af->eof_tv, &now, &eof_barrier);
 357         af->close_audio_file();
 358         audio_file = NULL;
 359         mmd->audio_format = -1;
 360         af = NULL;
 361         mmd->chunks_sent = 0;
 362         mmd->offset = 0;
 363         mmd->seconds_total = 0;
 364         tmp  = make_message("%s:\n%s:\n%s:\n", status_item_list[SI_AUDIO_INFO1],
 365                 status_item_list[SI_AUDIO_INFO2], status_item_list[SI_AUDIO_INFO3]);
 366         strcpy(mmd->audio_file_info, tmp);
 367         free(tmp);
 368         tmp  = make_message("%s:\n%s:\n%s:\n", status_item_list[SI_DBINFO1],
 369                 status_item_list[SI_DBINFO2], status_item_list[SI_DBINFO3]);
 370         strcpy(mmd->selector_info, tmp);
 371         free(tmp);
 372         mmd->filename[0] = '\0';
 373         mmd->size = 0;
 374         mmd->events++;
 375 }
 376
 377 /**
 378  * get the header and of the current audio file
 379  *
 380  * \param header_len the length of the header is stored here
 381  *
 382  * \return a pointer to a buffer containing the header, or NULL, if no audio
 383  * file is selected or if the current audio format does not need special header
 384  * treamtment.
 385  *
 386  */
 387 char *afs_get_header(int *header_len)
 388 {
 389         *header_len = 0;
 390         if (mmd->audio_format < 0)
 391                 return NULL;
 392         if (!afl[mmd->audio_format].get_header_info)
 393                 return NULL;
 394         return afl[mmd->audio_format].get_header_info(header_len);
 395 }
 396 const char *supported_audio_formats(void)
 397 {
 398         return SUPPORTED_AUDIO_FORMATS;
 399 }
 400
 401 /**
 402  * get the chunk time of the current audio file
 403  *
 404  * \return a pointer to a struct containing the chunk time, or NULL,
 405  * if currently no audio file is selected.
 406  */
 407 struct timeval *afs_chunk_time(void)
 408 {
 409         if (mmd->audio_format < 0)
 410                 return NULL;
 411         return &afl[mmd->audio_format].chunk_tv;
 412 }
 413
 414 /**
 415  * compute the timeout for para_server's main select-loop
 416  *
 417  * This function gets called from para_server to determine the timeout value
 418  * for its main select loop.
 419  *
 420  * Before the timeout is computed, the current afs status flags are evaluated
 421  * and acted upon by calling appropriate functions from the lower layers.
 422  * Possible actions include
 423  *
 424  *      - request a new file list from the current audio file selector
 425  *      - shutdown of all senders (stop/pause command)
 426  *      - reposition the stream (ff/jmp command)
 427  *
 428  * \return A pointer to a struct timeval containing the timeout for the next
 429  * chunk of data to be sent, or NULL if we're not sending right now.
 430  */
 431 struct timeval *afs_preselect(void)
 432 {
 433         struct audio_format_handler *af = NULL;
 434         int i, format;
 435         struct timeval *ret;
 436 again:
 437         format = mmd->audio_format;
 438         if (format >= 0)
 439                 af = afl + format;
 440         else
 441                 for (i = 0; senders[i].name; i++)
 442                         senders[i].shutdown_clients();
 443         if (afs_next() && af) {
 444                 afs_eof(af);
 445                 return afs_compute_timeout();
 446         }
 447         if (afs_paused() || afs_repos()) {
 448                 for (i = 0; senders[i].name; i++)
 449                         senders[i].shutdown_clients();
 450                 if (af) {
 451                         struct timeval now;
 452                         gettimeofday(&now, NULL);
 453                         if (!afs_paused() || mmd->chunks_sent)
 454                                 tv_add(&af->eof_tv, &now, &eof_barrier);
 455                         if (afs_repos())
 456                                 tv_add(&now, &announce_tv, &data_send_barrier);
 457                         if (mmd->new_afs_status_flags & AFS_NOMORE)
 458                                 mmd->new_afs_status_flags = AFS_NEXT;
 459                 }
 460                 mmd->chunks_sent = 0;
 461         }
 462         if (af && afs_repos() && mmd->current_chunk != mmd->repos_request)
 463                 af->reposition_stream(mmd->repos_request);
 464         if (afs_repos()) {
 465                 mmd->new_afs_status_flags &= ~(AFS_REPOS);
 466                 mmd->current_chunk = mmd->repos_request;
 467         }
 468         ret = afs_compute_timeout();
 469         if (!ret && !audio_file && afs_playing() &&
 470                         !(mmd->new_afs_status_flags & AFS_NOMORE)) {
 471                 PARA_DEBUG_LOG("%s", "ready and playing, but no audio file\n");
 472                 get_song();
 473                 goto again;
 474         }
 475         return ret;
 476 }
 477
 478 /**
 479  * afs_send_chunk - paraslash's main sending function
 480  *
 481  * This function gets called from para_server as soon as the next chunk of
 482  * data should be pushed out. It first calls the read_chunk() function of
 483  * the current audio format handler to obtain a pointer to the data to be
 484  * sent out as well as its length. This information is then passed to each
 485  * supported sender's send() function which does the actual sending.
 486  *
 487  * Return value: Positive return value on success, zero on eof and negative
 488  * on errors.
 489  */
 490
 491 void afs_send_chunk(void)
 492 {
 493         int i;
 494         struct audio_format_handler *af;
 495         char *buf;
 496         ssize_t ret;
 497         struct timeval now, due;
 498
 499         if (mmd->audio_format < 0 || !audio_file || !afs_playing())
 500                 return;
 501         af = &afl[mmd->audio_format];
 502         gettimeofday(&now, NULL);
 503         afs_next_chunk_time(&due);
 504         if (tv_diff(&due, &now, NULL) > 0)
 505                 return;
 506         if (chk_barrier("eof", &now, &eof_barrier, &due, 1) < 0)
 507                 return;
 508         if (chk_barrier("data send", &now, &data_send_barrier,
 509                         &due, 1) < 0)
 510                 return;
 511         buf = af->read_chunk(mmd->current_chunk, &ret);
 512         mmd->new_afs_status_flags &= ~AFS_REPOS;
 513         if (!buf) {
 514                 if (ret < 0)
 515                         mmd->new_afs_status_flags = AFS_NEXT;
 516                 else
 517                         mmd->new_afs_status_flags |= AFS_NEXT;
 518                 afs_eof(af);
 519                 return;
 520         }
 521         if (!mmd->chunks_sent) {
 522                 struct timeval tmp;
 523                 gettimeofday(&mmd->stream_start, NULL);
 524                 tv_scale(mmd->current_chunk, &af->chunk_tv, &tmp);
 525                 mmd->offset = tv2ms(&tmp);
 526                 mmd->events++;
 527         }
 528         for (i = 0; senders[i].name; i++)
 529                 senders[i].send(mmd->current_chunk, mmd->chunks_sent, buf, ret);
 530         mmd->new_afs_status_flags |= AFS_PLAYING;
 531         mmd->chunks_sent++;
 532         mmd->current_chunk++;
 533 }