Do proper bounds checking in formats (bug #1356)
[asterisk/asterisk.git] / formats / format_g729.c
index a2336c9..68dde21 100755 (executable)
@@ -17,6 +17,7 @@
 #include <asterisk/logger.h>
 #include <asterisk/sched.h>
 #include <asterisk/module.h>
+#include <netinet/in.h>
 #include <arpa/inet.h>
 #include <stdlib.h>
 #include <sys/time.h>
@@ -31,7 +32,7 @@
 #include <machine/endian.h>
 #endif
 
-/* Some Ideas for this code came from makeg729e.c by Jeffery Chilton */
+/* Some Ideas for this code came from makeg729e.c by Jeffrey Chilton */
 
 /* Portions of the conversion code are by guido@sienanet.it */
 
@@ -48,7 +49,7 @@ struct ast_filestream {
 };
 
 
-static pthread_mutex_t g729_lock = AST_MUTEX_INITIALIZER;
+static ast_mutex_t g729_lock = AST_MUTEX_INITIALIZER;
 static int glistcnt = 0;
 
 static char *name = "g729";
@@ -63,7 +64,7 @@ static struct ast_filestream *g729_open(int fd)
        struct ast_filestream *tmp;
        if ((tmp = malloc(sizeof(struct ast_filestream)))) {
                memset(tmp, 0, sizeof(struct ast_filestream));
-               if (ast_pthread_mutex_lock(&g729_lock)) {
+               if (ast_mutex_lock(&g729_lock)) {
                        ast_log(LOG_WARNING, "Unable to lock g729 list\n");
                        free(tmp);
                        return NULL;
@@ -76,7 +77,7 @@ static struct ast_filestream *g729_open(int fd)
                tmp->fr.src = name;
                tmp->fr.mallocd = 0;
                glistcnt++;
-               ast_pthread_mutex_unlock(&g729_lock);
+               ast_mutex_unlock(&g729_lock);
                ast_update_use_count();
        }
        return tmp;
@@ -90,14 +91,14 @@ static struct ast_filestream *g729_rewrite(int fd, char *comment)
        struct ast_filestream *tmp;
        if ((tmp = malloc(sizeof(struct ast_filestream)))) {
                memset(tmp, 0, sizeof(struct ast_filestream));
-               if (ast_pthread_mutex_lock(&g729_lock)) {
+               if (ast_mutex_lock(&g729_lock)) {
                        ast_log(LOG_WARNING, "Unable to lock g729 list\n");
                        free(tmp);
                        return NULL;
                }
                tmp->fd = fd;
                glistcnt++;
-               ast_pthread_mutex_unlock(&g729_lock);
+               ast_mutex_unlock(&g729_lock);
                ast_update_use_count();
        } else
                ast_log(LOG_WARNING, "Out of memory\n");
@@ -106,12 +107,12 @@ static struct ast_filestream *g729_rewrite(int fd, char *comment)
 
 static void g729_close(struct ast_filestream *s)
 {
-       if (ast_pthread_mutex_lock(&g729_lock)) {
+       if (ast_mutex_lock(&g729_lock)) {
                ast_log(LOG_WARNING, "Unable to lock g729 list\n");
                return;
        }
        glistcnt--;
-       ast_pthread_mutex_unlock(&g729_lock);
+       ast_mutex_unlock(&g729_lock);
        ast_update_use_count();
        close(s->fd);
        free(s);
@@ -149,12 +150,12 @@ static int g729_write(struct ast_filestream *fs, struct ast_frame *f)
                ast_log(LOG_WARNING, "Asked to write non-G729 frame (%d)!\n", f->subclass);
                return -1;
        }
-       if (f->datalen % 20) {
-               ast_log(LOG_WARNING, "Invalid data length, %d, should be multiple of 20\n", f->datalen);
+       if (f->datalen % 10) {
+               ast_log(LOG_WARNING, "Invalid data length, %d, should be multiple of 10\n", f->datalen);
                return -1;
        }
        if ((res = write(fs->fd, f->data, f->datalen)) != f->datalen) {
-                       ast_log(LOG_WARNING, "Bad write (%d/20): %s\n", res, strerror(errno));
+                       ast_log(LOG_WARNING, "Bad write (%d/10): %s\n", res, strerror(errno));
                        return -1;
        }
        return 0;
@@ -168,19 +169,22 @@ static char *g729_getcomment(struct ast_filestream *s)
 static int g729_seek(struct ast_filestream *fs, long sample_offset, int whence)
 {
        long bytes;
-       off_t min,cur,max,offset;
+       off_t min,cur,max,offset=0;
        min = 0;
        cur = lseek(fs->fd, 0, SEEK_CUR);
        max = lseek(fs->fd, 0, SEEK_END);
        
        bytes = 20 * (sample_offset / 160);
-       if(whence == SEEK_SET)
+       if (whence == SEEK_SET)
                offset = bytes;
-       if(whence == SEEK_CUR)
+       else if (whence == SEEK_CUR || whence == SEEK_FORCECUR)
                offset = cur + bytes;
-       if(whence == SEEK_END)
+       else if (whence == SEEK_END)
                offset = max - bytes;
-       offset = (offset > max)?max:offset;
+       if (whence != SEEK_FORCECUR) {
+               offset = (offset > max)?max:offset;
+       }
+       // protect against seeking beyond begining.
        offset = (offset < min)?min:offset;
        if (lseek(fs->fd, offset, SEEK_SET) < 0)
                return -1;
@@ -226,12 +230,12 @@ int unload_module()
 int usecount()
 {
        int res;
-       if (ast_pthread_mutex_lock(&g729_lock)) {
+       if (ast_mutex_lock(&g729_lock)) {
                ast_log(LOG_WARNING, "Unable to lock g729 list\n");
                return -1;
        }
        res = glistcnt;
-       ast_pthread_mutex_unlock(&g729_lock);
+       ast_mutex_unlock(&g729_lock);
        return res;
 }