Check for IO stream failures in various format's truncate/seek operations
[asterisk/asterisk.git] / formats / format_h263.c
old mode 100755 (executable)
new mode 100644 (file)
index 17805b2..ec08867
 /*
- * Asterisk -- A telephony toolkit for Linux.
+ * Asterisk -- An open source telephony toolkit.
  *
- * Save to raw, headerless h263 data.
- * 
- * Copyright (C) 1999, Mark Spencer
+ * Copyright (C) 1999 - 2006, Digium, Inc.
  *
- * Mark Spencer <markster@linux-support.net>
+ * Mark Spencer <markster@digium.com>
+ *
+ * See http://www.asterisk.org for more information about
+ * the Asterisk project. Please do not directly contact
+ * any of the maintainers of this project for assistance;
+ * the project provides a web site, mailing lists and IRC
+ * channels for your use.
  *
  * This program is free software, distributed under the terms of
- * the GNU General Public License
+ * the GNU General Public License Version 2. See the LICENSE file
+ * at the top of the source tree.
+ */
+
+/*! \file
+ *
+ * \brief Save to raw, headerless h263 data.
+ * \arg File name extension: h263
+ * \ingroup formats
+ * \arg See \ref AstVideo
  */
+
+/*** MODULEINFO
+       <support_level>core</support_level>
+ ***/
  
-#include <asterisk/lock.h>
-#include <asterisk/channel.h>
-#include <asterisk/file.h>
-#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>
-#include <stdio.h>
-#include <unistd.h>
-#include <errno.h>
-#include <string.h>
-#ifdef __linux__
-#include <endian.h>
-#else
-#ifdef SOLARIS
-#include "solaris-compat/compat.h"
-#else
-#include <machine/endian.h>
-#endif
-#endif
+#include "asterisk.h"
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include "asterisk/mod_format.h"
+#include "asterisk/module.h"
+#include "asterisk/endian.h"
 
 /* Some Ideas for this code came from makeh263e.c by Jeffrey Chilton */
 
 /* Portions of the conversion code are by guido@sienanet.it */
 
-struct ast_filestream {
-       void *reserved[AST_RESERVED_POINTERS];
-       /* Believe it or not, we must decode/recode to account for the
-          weird MS format */
-       /* This is what a filestream means to us */
-       int fd; /* Descriptor */
+/* According to:
+ * http://lists.mpegif.org/pipermail/mp4-tech/2005-July/005741.html
+ * the maximum actual frame size is not 2048, but 8192.  Since the maximum
+ * theoretical limit is not much larger (32k = 15bits), we'll go for that
+ * size to ensure we don't corrupt frames sent to us (unless they're
+ * ridiculously large). */
+#define        BUF_SIZE        32768   /* Four real h.263 Frames */
+
+struct h263_desc {
        unsigned int lastts;
-       struct ast_frame fr;                            /* Frame information */
-       char waste[AST_FRIENDLY_OFFSET];        /* Buffer for sending frames, etc */
-       char empty;                                                     /* Empty character */
-       unsigned char h263[4096];                               /* Two Real h263 Frames */
 };
 
 
-AST_MUTEX_DEFINE_STATIC(h263_lock);
-static int glistcnt = 0;
-
-static char *name = "h263";
-static char *desc = "Raw h263 data";
-static char *exts = "h263";
-
-static struct ast_filestream *h263_open(int fd)
+static int h263_open(struct ast_filestream *s)
 {
-       /* We don't have any header to read or anything really, but
-          if we did, it would go here.  We also might want to check
-          and be sure it's a valid file.  */
-       struct ast_filestream *tmp;
        unsigned int ts;
-       int res;
-       if ((res = read(fd, &ts, sizeof(ts))) < sizeof(ts)) {
-               ast_log(LOG_WARNING, "Empty file!\n");
-               return NULL;
-       }
-               
-       if ((tmp = malloc(sizeof(struct ast_filestream)))) {
-               memset(tmp, 0, sizeof(struct ast_filestream));
-               if (ast_mutex_lock(&h263_lock)) {
-                       ast_log(LOG_WARNING, "Unable to lock h263 list\n");
-                       free(tmp);
-                       return NULL;
-               }
-               tmp->fd = fd;
-               tmp->fr.data = tmp->h263;
-               tmp->fr.frametype = AST_FRAME_VIDEO;
-               tmp->fr.subclass = AST_FORMAT_H263;
-               /* datalen will vary for each frame */
-               tmp->fr.src = name;
-               tmp->fr.mallocd = 0;
-               glistcnt++;
-               ast_mutex_unlock(&h263_lock);
-               ast_update_use_count();
-       }
-       return tmp;
-}
 
-static struct ast_filestream *h263_rewrite(int fd, const char *comment)
-{
-       /* We don't have any header to read or anything really, but
-          if we did, it would go here.  We also might want to check
-          and be sure it's a valid file.  */
-       struct ast_filestream *tmp;
-       if ((tmp = malloc(sizeof(struct ast_filestream)))) {
-               memset(tmp, 0, sizeof(struct ast_filestream));
-               if (ast_mutex_lock(&h263_lock)) {
-                       ast_log(LOG_WARNING, "Unable to lock h263 list\n");
-                       free(tmp);
-                       return NULL;
-               }
-               tmp->fd = fd;
-               glistcnt++;
-               ast_mutex_unlock(&h263_lock);
-               ast_update_use_count();
-       } else
-               ast_log(LOG_WARNING, "Out of memory\n");
-       return tmp;
-}
-
-static void h263_close(struct ast_filestream *s)
-{
-       if (ast_mutex_lock(&h263_lock)) {
-               ast_log(LOG_WARNING, "Unable to lock h263 list\n");
-               return;
+       if (fread(&ts, 1, sizeof(ts), s->f) < sizeof(ts)) {
+               ast_log(LOG_WARNING, "Empty file!\n");
+               return -1;
        }
-       glistcnt--;
-       ast_mutex_unlock(&h263_lock);
-       ast_update_use_count();
-       close(s->fd);
-       free(s);
-       s = NULL;
+       return 0;
 }
 
 static struct ast_frame *h263_read(struct ast_filestream *s, int *whennext)
 {
        int res;
-       int mark=0;
+       uint32_t mark;
        unsigned short len;
        unsigned int ts;
+       struct h263_desc *fs = (struct h263_desc *)s->_private;
+
        /* Send a frame from the file to the appropriate channel */
-       s->fr.frametype = AST_FRAME_VIDEO;
-       s->fr.subclass = AST_FORMAT_H263;
-       s->fr.offset = AST_FRIENDLY_OFFSET;
-       s->fr.mallocd = 0;
-       s->fr.data = s->h263;
-       if ((res = read(s->fd, &len, sizeof(len))) < 1) {
+       if ((res = fread(&len, 1, sizeof(len), s->f)) < 1)
                return NULL;
-       }
        len = ntohs(len);
-       if (len & 0x8000) {
-               mark = 1;
-       }
+       mark = (len & 0x8000) ? 1 : 0;
        len &= 0x7fff;
-       if (len > sizeof(s->h263)) {
+       if (len > BUF_SIZE) {
                ast_log(LOG_WARNING, "Length %d is too long\n", len);
+               return NULL;
        }
-       if ((res = read(s->fd, s->h263, len)) != len) {
+       s->fr.frametype = AST_FRAME_VIDEO;
+       ast_format_set(&s->fr.subclass.format, AST_FORMAT_H263, 0);
+       s->fr.mallocd = 0;
+       AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, len);
+       if ((res = fread(s->fr.data.ptr, 1, s->fr.datalen, s->f)) != s->fr.datalen) {
                if (res)
                        ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", res, strerror(errno));
                return NULL;
        }
-       s->fr.samples = s->lastts;
+       s->fr.samples = fs->lastts;     /* XXX what ? */
        s->fr.datalen = len;
-       s->fr.subclass |= mark;
-       if ((res = read(s->fd, &ts, sizeof(ts))) == sizeof(ts)) {
-               s->lastts = *whennext = ntohl(ts) * 4/45;
+       if (mark) {
+               ast_format_set_video_mark(&s->fr.subclass.format);
+       }
+       s->fr.delivery.tv_sec = 0;
+       s->fr.delivery.tv_usec = 0;
+       if ((res = fread(&ts, 1, sizeof(ts), s->f)) == sizeof(ts)) {
+               fs->lastts = ntohl(ts);
+               *whennext = fs->lastts * 4/45;
        } else
                *whennext = 0;
        return &s->fr;
@@ -173,43 +111,34 @@ static int h263_write(struct ast_filestream *fs, struct ast_frame *f)
        int res;
        unsigned int ts;
        unsigned short len;
-       int subclass;
-       int mark=0;
+       uint32_t mark = 0;
        if (f->frametype != AST_FRAME_VIDEO) {
                ast_log(LOG_WARNING, "Asked to write non-video frame!\n");
                return -1;
        }
-       subclass = f->subclass;
-       if (subclass & 0x1)
-               mark=0x8000;
-       subclass &= ~0x1;
-       if (subclass != AST_FORMAT_H263) {
-               ast_log(LOG_WARNING, "Asked to write non-h263 frame (%d)!\n", f->subclass);
+       mark = ast_format_get_video_mark(&f->subclass.format) ? 0x8000 : 0;
+       if (f->subclass.format.id != AST_FORMAT_H263) {
+               ast_log(LOG_WARNING, "Asked to write non-h263 frame (%s)!\n", ast_getformatname(&f->subclass.format));
                return -1;
        }
        ts = htonl(f->samples);
-       if ((res = write(fs->fd, &ts, sizeof(ts))) != sizeof(ts)) {
+       if ((res = fwrite(&ts, 1, sizeof(ts), fs->f)) != sizeof(ts)) {
                        ast_log(LOG_WARNING, "Bad write (%d/4): %s\n", res, strerror(errno));
                        return -1;
        }
        len = htons(f->datalen | mark);
-       if ((res = write(fs->fd, &len, sizeof(len))) != sizeof(len)) {
+       if ((res = fwrite(&len, 1, sizeof(len), fs->f)) != sizeof(len)) {
                        ast_log(LOG_WARNING, "Bad write (%d/2): %s\n", res, strerror(errno));
                        return -1;
        }
-       if ((res = write(fs->fd, f->data, f->datalen)) != f->datalen) {
+       if ((res = fwrite(f->data.ptr, 1, f->datalen, fs->f)) != f->datalen) {
                        ast_log(LOG_WARNING, "Bad write (%d/%d): %s\n", res, f->datalen, strerror(errno));
                        return -1;
        }
        return 0;
 }
 
-static char *h263_getcomment(struct ast_filestream *s)
-{
-       return NULL;
-}
-
-static int h263_seek(struct ast_filestream *fs, long sample_offset, int whence)
+static int h263_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
 {
        /* No way Jose */
        return -1;
@@ -217,59 +146,55 @@ static int h263_seek(struct ast_filestream *fs, long sample_offset, int whence)
 
 static int h263_trunc(struct ast_filestream *fs)
 {
-       /* Truncate file to current length */
-       if (ftruncate(fs->fd, lseek(fs->fd, 0, SEEK_CUR)) < 0)
-               return -1;
-       return 0;
-}
+       int fd;
+       off_t cur;
 
-static long h263_tell(struct ast_filestream *fs)
-{
-       off_t offset;
-       offset = lseek(fs->fd, 0, SEEK_CUR);
-       return (offset/20)*160;
+       if ((fd = fileno(fs->f)) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to determine file descriptor for h263 filestream %p: %s\n", fs, strerror(errno));
+               return -1;
+       }
+       if ((cur = ftello(fs->f) < 0)) {
+               ast_log(AST_LOG_WARNING, "Unable to determine current position in h263 filestream %p: %s\n", fs, strerror(errno));
+               return -1;
+       }
+       /* Truncate file to current length */
+       return ftruncate(fd, cur);
 }
 
-int load_module()
+static off_t h263_tell(struct ast_filestream *fs)
 {
-       return ast_format_register(name, exts, AST_FORMAT_H263,
-                                                               h263_open,
-                                                               h263_rewrite,
-                                                               h263_write,
-                                                               h263_seek,
-                                                               h263_trunc,
-                                                               h263_tell,
-                                                               h263_read,
-                                                               h263_close,
-                                                               h263_getcomment);
-                                                               
-                                                               
+       off_t offset = ftello(fs->f);
+       return offset;  /* XXX totally bogus, needs fixing */
 }
 
-int unload_module()
-{
-       return ast_format_unregister(name);
-}      
+static struct ast_format_def h263_f = {
+       .name = "h263",
+       .exts = "h263",
+       .open = h263_open,
+       .write = h263_write,
+       .seek = h263_seek,
+       .trunc = h263_trunc,
+       .tell = h263_tell,
+       .read = h263_read,
+       .buf_size = BUF_SIZE + AST_FRIENDLY_OFFSET,
+       .desc_size = sizeof(struct h263_desc),
+};
 
-int usecount()
+static int load_module(void)
 {
-       int res;
-       if (ast_mutex_lock(&h263_lock)) {
-               ast_log(LOG_WARNING, "Unable to lock h263 list\n");
-               return -1;
-       }
-       res = glistcnt;
-       ast_mutex_unlock(&h263_lock);
-       return res;
+       ast_format_set(&h263_f.format, AST_FORMAT_H263, 0);
+       if (ast_format_def_register(&h263_f))
+               return AST_MODULE_LOAD_FAILURE;
+       return AST_MODULE_LOAD_SUCCESS;
 }
 
-char *description()
+static int unload_module(void)
 {
-       return desc;
+       return ast_format_def_unregister(h263_f.name);
 }
 
-
-char *key()
-{
-       return ASTERISK_GPL_KEY;
-}
+AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Raw H.263 data",
+       .load = load_module,
+       .unload = unload_module,
+       .load_pri = AST_MODPRI_APP_DEPEND
+);