Version 0.3.0 from FTP
[asterisk/asterisk.git] / formats / format_mp3.c
index 811acbe..e702e55 100755 (executable)
@@ -3,7 +3,7 @@
  *
  * Everybody's favorite format: MP3 Files!  Yay!
  * 
- * Copyright (C) 1999, Adtran Inc. and Linux Support Services, LLC
+ * Copyright (C) 1999, Mark Spencer
  *
  * Mark Spencer <markster@linux-support.net>
  *
@@ -11,6 +11,7 @@
  * the GNU General Public License
  */
  
+#include <asterisk/lock.h>
 #include <asterisk/channel.h>
 #include <asterisk/file.h>
 #include <asterisk/logger.h>
@@ -26,8 +27,7 @@
 #include <sys/time.h>
 #include "../channels/adtranvofr.h"
 
-
-#define MP3_MAX_SIZE 1400
+#define MAX_FRAME_SIZE 1441
 
 struct ast_filestream {
        /* First entry MUST be reserved for the channel type */
@@ -36,27 +36,25 @@ struct ast_filestream {
        int fd; /* Descriptor */
        struct ast_channel *owner;
        struct ast_filestream *next;
-       struct ast_frame *fr;   /* Frame representation of buf */
-       char buf[sizeof(struct ast_frame) + MP3_MAX_SIZE + AST_FRIENDLY_OFFSET];        /* Buffer for sending frames, etc */
+       struct ast_frame fr;    /* Frame representation of buf */
+       char offset[AST_FRIENDLY_OFFSET];
+       unsigned char buf[MAX_FRAME_SIZE * 2];
+       int lasttimeout;
        int pos;
+       int adj;
+       struct timeval last;
 };
 
 
 static struct ast_filestream *glist = NULL;
-static pthread_mutex_t mp3_lock = PTHREAD_MUTEX_INITIALIZER;
+static pthread_mutex_t mp3_lock = AST_MUTEX_INITIALIZER;
 static int glistcnt = 0;
 
 static char *name = "mp3";
-static char *desc = "MPEG-2 Layer 3 File Format Support";
+static char *desc = "MPEG-1,2 Layer 3 File Format Support";
 static char *exts = "mp3|mpeg3";
 
-#if 0
-#define MP3_FRAMELEN 417
-#else
-#define MP3_FRAMELEN 400
-#endif
-#define MP3_OUTPUTLEN 2304     /* Bytes */
-#define MP3_TIMELEN ((MP3_OUTPUTLEN * 1000 / 16000) )
+#include "../codecs/mp3anal.h"
 
 static struct ast_filestream *mp3_open(int fd)
 {
@@ -65,7 +63,7 @@ static struct ast_filestream *mp3_open(int fd)
           and be sure it's a valid file.  */
        struct ast_filestream *tmp;
        if ((tmp = malloc(sizeof(struct ast_filestream)))) {
-               if (pthread_mutex_lock(&mp3_lock)) {
+               if (ast_pthread_mutex_lock(&mp3_lock)) {
                        ast_log(LOG_WARNING, "Unable to lock mp3 list\n");
                        free(tmp);
                        return NULL;
@@ -74,15 +72,12 @@ static struct ast_filestream *mp3_open(int fd)
                glist = tmp;
                tmp->fd = fd;
                tmp->owner = NULL;
-               tmp->fr = (struct ast_frame *)tmp->buf;
-               tmp->fr->data = tmp->buf + sizeof(struct ast_frame);
-               tmp->fr->frametype = AST_FRAME_VOICE;
-               tmp->fr->subclass = AST_FORMAT_MP3;
-               /* datalen will vary for each frame */
-               tmp->fr->src = name;
-               tmp->fr->mallocd = 0;
+               tmp->lasttimeout = -1;
+               tmp->last.tv_usec = 0;
+               tmp->last.tv_sec = 0;
+               tmp->adj = 0;
                glistcnt++;
-               pthread_mutex_unlock(&mp3_lock);
+               ast_pthread_mutex_unlock(&mp3_lock);
                ast_update_use_count();
        }
        return tmp;
@@ -95,7 +90,7 @@ static struct ast_filestream *mp3_rewrite(int fd, char *comment)
           and be sure it's a valid file.  */
        struct ast_filestream *tmp;
        if ((tmp = malloc(sizeof(struct ast_filestream)))) {
-               if (pthread_mutex_lock(&mp3_lock)) {
+               if (ast_pthread_mutex_lock(&mp3_lock)) {
                        ast_log(LOG_WARNING, "Unable to lock mp3 list\n");
                        free(tmp);
                        return NULL;
@@ -104,9 +99,8 @@ static struct ast_filestream *mp3_rewrite(int fd, char *comment)
                glist = tmp;
                tmp->fd = fd;
                tmp->owner = NULL;
-               tmp->fr = NULL;
                glistcnt++;
-               pthread_mutex_unlock(&mp3_lock);
+               ast_pthread_mutex_unlock(&mp3_lock);
                ast_update_use_count();
        } else
                ast_log(LOG_WARNING, "Out of memory\n");
@@ -121,7 +115,7 @@ static struct ast_frame *mp3_read(struct ast_filestream *s)
 static void mp3_close(struct ast_filestream *s)
 {
        struct ast_filestream *tmp, *tmpl = NULL;
-       if (pthread_mutex_lock(&mp3_lock)) {
+       if (ast_pthread_mutex_lock(&mp3_lock)) {
                ast_log(LOG_WARNING, "Unable to lock mp3 list\n");
                return;
        }
@@ -144,7 +138,7 @@ static void mp3_close(struct ast_filestream *s)
                        ast_sched_del(s->owner->sched, s->owner->streamid);
                s->owner->streamid = -1;
        }
-       pthread_mutex_unlock(&mp3_lock);
+       ast_pthread_mutex_unlock(&mp3_lock);
        ast_update_use_count();
        if (!tmp) 
                ast_log(LOG_WARNING, "Freeing a filestream we don't seem to own\n");
@@ -155,28 +149,68 @@ static void mp3_close(struct ast_filestream *s)
 static int ast_read_callback(void *data)
 {
        /* XXX Don't assume frames are this size XXX */
-       u_int16_t size=MP3_FRAMELEN;
        u_int32_t delay = -1;
        int res;
        struct ast_filestream *s = data;
-       /* Send a frame from the file to the appropriate channel */
-       /* Read the data into the buffer */
-       s->fr->offset = AST_FRIENDLY_OFFSET;
-       s->fr->datalen = size;
-       s->fr->data = s->buf + sizeof(struct ast_frame) + AST_FRIENDLY_OFFSET;
-       if ((res = read(s->fd, s->fr->data , size)) != size) {
-               ast_log(LOG_WARNING, "Short read (%d of %d bytes) (%s)!\n", res, size, strerror(errno));
+       int size;
+       int ms=0;
+       struct timeval tv;
+       if ((res = read(s->fd, s->buf , 4)) != 4) {
+               ast_log(LOG_WARNING, "Short read (%d of 4 bytes) (%s)!\n", res, strerror(errno));
+               s->owner->streamid = -1;
+               return 0;
+       }
+       if (mp3_badheader(s->buf)) {
+               ast_log(LOG_WARNING, "Bad mp3 header\n");
+               return 0;
+       }
+       if ((size = mp3_framelen(s->buf)) < 0) {
+               ast_log(LOG_WARNING, "Unable to calculate frame size\n");
+               return 0;
+       }
+       if ((res = read(s->fd, s->buf + 4 , size - 4)) != size - 4) {
+               ast_log(LOG_WARNING, "Short read (%d of %d bytes) (%s)!\n", res, size - 4, strerror(errno));
                s->owner->streamid = -1;
                return 0;
        }
-       delay = MP3_TIMELEN;
-       s->fr->timelen = delay;
+       /* Send a frame from the file to the appropriate channel */
+       /* Read the data into the buffer */
+       s->fr.offset = AST_FRIENDLY_OFFSET;
+       s->fr.frametype = AST_FRAME_VOICE;
+       s->fr.subclass = AST_FORMAT_MP3;
+       s->fr.mallocd = 0;
+       s->fr.src = name;
+       s->fr.datalen = size;
+       s->fr.data = s->buf;
+       delay = mp3_samples(s->buf) * 1000 / mp3_samplerate(s->buf);
+       if (s->last.tv_sec || s->last.tv_usec) {
+               /* To keep things running smoothly, we watch how close we're coming */
+               gettimeofday(&tv, NULL);
+               ms = ((tv.tv_usec - s->last.tv_usec) / 1000 + (tv.tv_sec - s->last.tv_sec) * 1000);
+               /* If we're within 2 milliseconds, that's close enough */
+               if ((ms - delay) > 0 )
+                       s->adj -= (ms - delay);
+               s->adj -= 2;
+       }
+       s->fr.samples = delay * 8;
+#if 0
+       ast_log(LOG_DEBUG, "delay is %d, adjusting by %d, as last was %d\n", delay, s->adj, ms);
+#endif
+       delay += s->adj;
+       if (delay < 1)
+               delay = 1;
        /* Lastly, process the frame */
-       if (ast_write(s->owner, s->fr)) {
+       if (ast_write(s->owner, &s->fr)) {
                ast_log(LOG_WARNING, "Failed to write frame\n");
                s->owner->streamid = -1;
                return 0;
        }
+       gettimeofday(&s->last, NULL);
+       if (s->lasttimeout != delay) {
+               s->owner->streamid = ast_sched_add(s->owner->sched, delay, ast_read_callback, s);
+               s->lasttimeout = delay;
+               return 0;
+       }
        return -1;
 }
 
@@ -184,7 +218,11 @@ static int mp3_apply(struct ast_channel *c, struct ast_filestream *s)
 {
        /* Select our owner for this stream, and get the ball rolling. */
        s->owner = c;
-       s->owner->streamid = ast_sched_add(s->owner->sched, MP3_TIMELEN, ast_read_callback, s);
+       return 0;
+}
+
+static int mp3_play(struct ast_filestream *s)
+{
        ast_read_callback(s);
        return 0;
 }
@@ -192,10 +230,6 @@ static int mp3_apply(struct ast_channel *c, struct ast_filestream *s)
 static int mp3_write(struct ast_filestream *fs, struct ast_frame *f)
 {
        int res;
-       if (fs->fr) {
-               ast_log(LOG_WARNING, "Asked to write on a read stream??\n");
-               return -1;
-       }
        if (f->frametype != AST_FRAME_VOICE) {
                ast_log(LOG_WARNING, "Asked to write non-voice frame!\n");
                return -1;
@@ -211,7 +245,22 @@ static int mp3_write(struct ast_filestream *fs, struct ast_frame *f)
        return 0;
 }
 
-char *mp3_getcomment(struct ast_filestream *s)
+static int mp3_seek(struct ast_filestream *fs, long sample_offset, int whence)
+{
+       return -1;
+}
+
+static int mp3_trunc(struct ast_filestream *fs)
+{
+       return -1;
+}
+
+static long mp3_tell(struct ast_filestream *fs)
+{
+       return -1;
+}
+
+static char *mp3_getcomment(struct ast_filestream *s)
 {
        return NULL;
 }
@@ -222,7 +271,11 @@ int load_module()
                                                                mp3_open,
                                                                mp3_rewrite,
                                                                mp3_apply,
+                                                               mp3_play,
                                                                mp3_write,
+                                                               mp3_seek,
+                                                               mp3_trunc,
+                                                               mp3_tell,
                                                                mp3_read,
                                                                mp3_close,
                                                                mp3_getcomment);
@@ -233,31 +286,31 @@ int load_module()
 int unload_module()
 {
        struct ast_filestream *tmp, *tmpl;
-       if (pthread_mutex_lock(&mp3_lock)) {
+       if (ast_pthread_mutex_lock(&mp3_lock)) {
                ast_log(LOG_WARNING, "Unable to lock mp3 list\n");
                return -1;
        }
        tmp = glist;
        while(tmp) {
                if (tmp->owner)
-                       ast_softhangup(tmp->owner);
+                       ast_softhangup(tmp->owner, AST_SOFTHANGUP_APPUNLOAD);
                tmpl = tmp;
                tmp = tmp->next;
                free(tmpl);
        }
-       pthread_mutex_unlock(&mp3_lock);
+       ast_pthread_mutex_unlock(&mp3_lock);
        return ast_format_unregister(name);
 }      
 
 int usecount()
 {
        int res;
-       if (pthread_mutex_lock(&mp3_lock)) {
+       if (ast_pthread_mutex_lock(&mp3_lock)) {
                ast_log(LOG_WARNING, "Unable to lock mp3 list\n");
                return -1;
        }
        res = glistcnt;
-       pthread_mutex_unlock(&mp3_lock);
+       ast_pthread_mutex_unlock(&mp3_lock);
        return res;
 }
 
@@ -266,3 +319,8 @@ char *description()
        return desc;
 }
 
+
+char *key()
+{
+       return ASTERISK_GPL_KEY;
+}