addons/format_mp3.c

   1 /*
   2  * Asterisk -- An open source telephony toolkit.
   3  *
   4  * Anthony Minessale <anthmct@yahoo.com>
   5  *
   6  * Derived from other asterisk sound formats by
   7  * Mark Spencer <markster@linux-support.net>
   8  *
   9  * Thanks to mpglib from http://www.mpg123.org/
  10  * and Chris Stenton [jacs@gnome.co.uk]
  11  * for coding the ability to play stereo and non-8khz files
  12
  13  * See http://www.asterisk.org for more information about
  14  * the Asterisk project. Please do not directly contact
  15  * any of the maintainers of this project for assistance;
  16  * the project provides a web site, mailing lists and IRC
  17  * channels for your use.
  18  *
  19  * This program is free software, distributed under the terms of
  20  * the GNU General Public License Version 2. See the LICENSE file
  21  * at the top of the source tree.
  22  */
  23
  24 /*!
  25  * \file
  26  * \brief MP3 Format Handler
  27  * \ingroup formats
  28  */
  29
  30 /*** MODULEINFO
  31         <defaultenabled>no</defaultenabled>
  32         <support_level>extended</support_level>
  33  ***/
  34
  35 #include "asterisk.h"
  36
  37 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
  38
  39 #include "mp3/mpg123.h"
  40 #include "mp3/mpglib.h"
  41
  42 #include "asterisk/module.h"
  43 #include "asterisk/mod_format.h"
  44 #include "asterisk/logger.h"
  45
  46 #define MP3_BUFLEN 320
  47 #define MP3_SCACHE 16384
  48 #define MP3_DCACHE 8192
  49
  50 struct mp3_private {
  51         /*! state for the mp3 decoder */
  52         struct mpstr mp;
  53         /*! buffer to hold mp3 data after read from disk */
  54         char sbuf[MP3_SCACHE];
  55         /*! buffer for slinear audio after being decoded out of sbuf */
  56         char dbuf[MP3_DCACHE];
  57         /*! how much data has been written to the output buffer in the ast_filestream */
  58         int buflen;
  59         /*! how much data has been written to sbuf */
  60         int sbuflen;
  61         /*! how much data is left to be read out of dbuf, starting at dbufoffset */
  62         int dbuflen;
  63         /*! current offset for reading data out of dbuf */
  64         int dbufoffset;
  65         int offset;
  66         long seek;
  67 };
  68
  69 static const char name[] = "mp3";
  70
  71 #define BLOCKSIZE 160
  72 #define OUTSCALE 4096
  73
  74 #define GAIN -4         /* 2^GAIN is the multiple to increase the volume by */
  75
  76 #if __BYTE_ORDER == __LITTLE_ENDIAN
  77 #define htoll(b) (b)
  78 #define htols(b) (b)
  79 #define ltohl(b) (b)
  80 #define ltohs(b) (b)
  81 #else
  82 #if __BYTE_ORDER == __BIG_ENDIAN
  83 #define htoll(b)  \
  84           (((((b)      ) & 0xFF) << 24) | \
  85                ((((b) >>  8) & 0xFF) << 16) | \
  86                    ((((b) >> 16) & 0xFF) <<  8) | \
  87                    ((((b) >> 24) & 0xFF)      ))
  88 #define htols(b) \
  89           (((((b)      ) & 0xFF) << 8) | \
  90                    ((((b) >> 8) & 0xFF)      ))
  91 #define ltohl(b) htoll(b)
  92 #define ltohs(b) htols(b)
  93 #else
  94 #error "Endianess not defined"
  95 #endif
  96 #endif
  97
  98
  99 static int mp3_open(struct ast_filestream *s)
 100 {
 101         struct mp3_private *p = s->_private;
 102         InitMP3(&p->mp, OUTSCALE);
 103         return 0;
 104 }
 105
 106
 107 static void mp3_close(struct ast_filestream *s)
 108 {
 109         struct mp3_private *p = s->_private;
 110
 111         ExitMP3(&p->mp);
 112         return;
 113 }
 114
 115 static int mp3_squeue(struct ast_filestream *s)
 116 {
 117         struct mp3_private *p = s->_private;
 118         int res=0;
 119
 120         res = ftell(s->f);
 121         p->sbuflen = fread(p->sbuf, 1, MP3_SCACHE, s->f);
 122         if(p->sbuflen < 0) {
 123                 ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", p->sbuflen, strerror(errno));
 124                 return -1;
 125         }
 126         res = decodeMP3(&p->mp,p->sbuf,p->sbuflen,p->dbuf,MP3_DCACHE,&p->dbuflen);
 127         if(res != MP3_OK)
 128                 return -1;
 129         p->sbuflen -= p->dbuflen;
 130         p->dbufoffset = 0;
 131         return 0;
 132 }
 133
 134 static int mp3_dqueue(struct ast_filestream *s)
 135 {
 136         struct mp3_private *p = s->_private;
 137         int res=0;
 138
 139         if((res = decodeMP3(&p->mp,NULL,0,p->dbuf,MP3_DCACHE,&p->dbuflen)) == MP3_OK) {
 140                 p->sbuflen -= p->dbuflen;
 141                 p->dbufoffset = 0;
 142         }
 143         return res;
 144 }
 145
 146 static int mp3_queue(struct ast_filestream *s)
 147 {
 148         struct mp3_private *p = s->_private;
 149         int res = 0, bytes = 0;
 150
 151         if(p->seek) {
 152                 ExitMP3(&p->mp);
 153                 InitMP3(&p->mp, OUTSCALE);
 154                 fseek(s->f, 0, SEEK_SET);
 155                 p->sbuflen = p->dbuflen = p->offset = 0;
 156                 while(p->offset < p->seek) {
 157                         if(mp3_squeue(s))
 158                                 return -1;
 159                         while(p->offset < p->seek && ((res = mp3_dqueue(s))) == MP3_OK) {
 160                                 for(bytes = 0 ; bytes < p->dbuflen ; bytes++) {
 161                                         p->dbufoffset++;
 162                                         p->offset++;
 163                                         if(p->offset >= p->seek)
 164                                                 break;
 165                                 }
 166                         }
 167                         if(res == MP3_ERR)
 168                                 return -1;
 169                 }
 170
 171                 p->seek = 0;
 172                 return 0;
 173         }
 174         if(p->dbuflen == 0) {
 175                 if(p->sbuflen) {
 176                         res = mp3_dqueue(s);
 177                         if(res == MP3_ERR)
 178                                 return -1;
 179                 }
 180                 if(! p->sbuflen || res != MP3_OK) {
 181                         if(mp3_squeue(s))
 182                                 return -1;
 183                 }
 184
 185         }
 186
 187         return 0;
 188 }
 189
 190 static struct ast_frame *mp3_read(struct ast_filestream *s, int *whennext)
 191 {
 192
 193         struct mp3_private *p = s->_private;
 194         int delay =0;
 195         int save=0;
 196
 197         /* Pre-populate the buffer that holds audio to be returned (dbuf) */
 198         if (mp3_queue(s)) {
 199                 return NULL;
 200         }
 201
 202         if (p->dbuflen) {
 203                 /* Read out what's waiting in dbuf */
 204                 for (p->buflen = 0; p->buflen < MP3_BUFLEN && p->buflen < p->dbuflen; p->buflen++) {
 205                         s->buf[p->buflen + AST_FRIENDLY_OFFSET] = p->dbuf[p->buflen + p->dbufoffset];
 206                 }
 207                 p->dbufoffset += p->buflen;
 208                 p->dbuflen -= p->buflen;
 209         }
 210
 211         if (p->buflen < MP3_BUFLEN) {
 212                 /* dbuf didn't have enough, so reset dbuf, fill it back up and continue */
 213                 p->dbuflen = p->dbufoffset = 0;
 214
 215                 if (mp3_queue(s)) {
 216                         return NULL;
 217                 }
 218
 219                 /* Make sure dbuf has enough to complete this read attempt */
 220                 if (p->dbuflen >= (MP3_BUFLEN - p->buflen)) {
 221                         for (save = p->buflen; p->buflen < MP3_BUFLEN; p->buflen++) {
 222                                 s->buf[p->buflen + AST_FRIENDLY_OFFSET] = p->dbuf[(p->buflen - save) + p->dbufoffset];
 223                         }
 224                         p->dbufoffset += (MP3_BUFLEN - save);
 225                         p->dbuflen -= (MP3_BUFLEN - save);
 226                 }
 227
 228         }
 229
 230         p->offset += p->buflen;
 231         delay = p->buflen / 2;
 232         s->fr.frametype = AST_FRAME_VOICE;
 233         ast_format_set(&s->fr.subclass.format, AST_FORMAT_SLINEAR, 0);
 234         AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, p->buflen);
 235         s->fr.mallocd = 0;
 236         s->fr.samples = delay;
 237         *whennext = delay;
 238         return &s->fr;
 239 }
 240
 241
 242 static int mp3_write(struct ast_filestream *fs, struct ast_frame *f)
 243 {
 244         ast_log(LOG_ERROR,"I Can't write MP3 only read them.\n");
 245         return -1;
 246
 247 }
 248
 249
 250 static int mp3_seek(struct ast_filestream *s, off_t sample_offset, int whence)
 251 {
 252         struct mp3_private *p = s->_private;
 253         off_t min,max,cur;
 254         long offset=0,samples;
 255         samples = sample_offset * 2;
 256
 257         min = 0;
 258         fseek(s->f, 0, SEEK_END);
 259         max = ftell(s->f) * 100;
 260         cur = p->offset;
 261
 262         if (whence == SEEK_SET)
 263                 offset = samples + min;
 264         else if (whence == SEEK_CUR || whence == SEEK_FORCECUR)
 265                 offset = samples + cur;
 266         else if (whence == SEEK_END)
 267                 offset = max - samples;
 268         if (whence != SEEK_FORCECUR) {
 269                 offset = (offset > max)?max:offset;
 270         }
 271
 272         p->seek = offset;
 273         return fseek(s->f, offset, SEEK_SET);
 274
 275 }
 276
 277 static int mp3_rewrite(struct ast_filestream *s, const char *comment)
 278 {
 279         ast_log(LOG_ERROR,"I Can't write MP3 only read them.\n");
 280         return -1;
 281 }
 282
 283 static int mp3_trunc(struct ast_filestream *s)
 284 {
 285
 286         ast_log(LOG_ERROR,"I Can't write MP3 only read them.\n");
 287         return -1;
 288 }
 289
 290 static off_t mp3_tell(struct ast_filestream *s)
 291 {
 292         struct mp3_private *p = s->_private;
 293
 294         return p->offset/2;
 295 }
 296
 297 static char *mp3_getcomment(struct ast_filestream *s)
 298 {
 299         return NULL;
 300 }
 301
 302 static struct ast_format_def mp3_f = {
 303         .name = "mp3",
 304         .exts = "mp3",
 305         .open = mp3_open,
 306         .write = mp3_write,
 307         .rewrite = mp3_rewrite,
 308         .seek = mp3_seek,
 309         .trunc = mp3_trunc,
 310         .tell = mp3_tell,
 311         .read = mp3_read,
 312         .close = mp3_close,
 313         .getcomment = mp3_getcomment,
 314         .buf_size = MP3_BUFLEN + AST_FRIENDLY_OFFSET,
 315         .desc_size = sizeof(struct mp3_private),
 316 };
 317
 318
 319 static int load_module(void)
 320 {
 321         ast_format_set(&mp3_f.format, AST_FORMAT_SLINEAR, 0);
 322         InitMP3Constants();
 323         return ast_format_def_register(&mp3_f);
 324 }
 325
 326 static int unload_module(void)
 327 {
 328         return ast_format_def_unregister(name);
 329 }
 330
 331 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "MP3 format [Any rate but 8000hz mono is optimal]");