channel locking: Add locking for channel snapshot creation
[asterisk/asterisk.git] / formats / format_h264.c
index 0b689dc..e090c2b 100644 (file)
  * \ingroup formats
  * \arg See \ref AstVideo
  */
+
+/*** MODULEINFO
+       <support_level>core</support_level>
+ ***/
  
 #include "asterisk.h"
 
 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
 
-#include <unistd.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
-#include <stdlib.h>
-#include <sys/time.h>
-#include <stdio.h>
-#include <errno.h>
-#include <string.h>
-
-#include "asterisk/lock.h"
-#include "asterisk/channel.h"
-#include "asterisk/file.h"
-#include "asterisk/logger.h"
-#include "asterisk/sched.h"
+#include "asterisk/mod_format.h"
 #include "asterisk/module.h"
 #include "asterisk/endian.h"
 
@@ -58,8 +49,7 @@ struct h264_desc {
 static int h264_open(struct ast_filestream *s)
 {
        unsigned int ts;
-       int res;
-       if ((res = fread(&ts, 1, sizeof(ts), s->f)) < sizeof(ts)) {
+       if (fread(&ts, 1, sizeof(ts), s->f) < sizeof(ts)) {
                ast_log(LOG_WARNING, "Empty file!\n");
                return -1;
        }
@@ -69,10 +59,10 @@ static int h264_open(struct ast_filestream *s)
 static struct ast_frame *h264_read(struct ast_filestream *s, int *whennext)
 {
        int res;
-       int mark=0;
+       int mark = 0;
        unsigned short len;
        unsigned int ts;
-       struct h264_desc *fs = (struct h264_desc *)s->private;
+       struct h264_desc *fs = (struct h264_desc *)s->_private;
 
        /* Send a frame from the file to the appropriate channel */
        if ((res = fread(&len, 1, sizeof(len), s->f)) < 1)
@@ -85,17 +75,19 @@ static struct ast_frame *h264_read(struct ast_filestream *s, int *whennext)
                len = BUF_SIZE; /* XXX truncate */
        }
        s->fr.frametype = AST_FRAME_VIDEO;
-       s->fr.subclass = AST_FORMAT_H264;
+       ast_format_set(&s->fr.subclass.format, AST_FORMAT_H264, 0);
        s->fr.mallocd = 0;
        AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, len);
-       if ((res = fread(s->fr.data, 1, s->fr.datalen, s->f)) != s->fr.datalen) {
+       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 of %d) (%s)!\n", res, len, strerror(errno));
                return NULL;
        }
        s->fr.samples = fs->lastts;
        s->fr.datalen = len;
-       s->fr.subclass |= mark;
+       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)) {
@@ -117,9 +109,9 @@ static int h264_write(struct ast_filestream *s, struct ast_frame *f)
                ast_log(LOG_WARNING, "Asked to write non-video frame!\n");
                return -1;
        }
-       mark = (f->subclass & 0x1) ? 0x8000 : 0;
-       if ((f->subclass & ~0x1) != AST_FORMAT_H264) {
-               ast_log(LOG_WARNING, "Asked to write non-h264 frame (%d)!\n", f->subclass);
+       mark = ast_format_get_video_mark(&f->subclass.format) ? 0x8000 : 0;
+       if (f->subclass.format.id != AST_FORMAT_H264) {
+               ast_log(LOG_WARNING, "Asked to write non-h264 frame (%s)!\n", ast_getformatname(&f->subclass.format));
                return -1;
        }
        ts = htonl(f->samples);
@@ -132,7 +124,7 @@ static int h264_write(struct ast_filestream *s, struct ast_frame *f)
                ast_log(LOG_WARNING, "Bad write (%d/2): %s\n", res, strerror(errno));
                return -1;
        }
-       if ((res = fwrite(f->data, 1, f->datalen, s->f)) != f->datalen) {
+       if ((res = fwrite(f->data.ptr, 1, f->datalen, s->f)) != f->datalen) {
                ast_log(LOG_WARNING, "Bad write (%d/%d): %s\n", res, f->datalen, strerror(errno));
                return -1;
        }
@@ -147,10 +139,19 @@ static int h264_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
 
 static int h264_trunc(struct ast_filestream *fs)
 {
-       /* Truncate file to current length */
-       if (ftruncate(fileno(fs->f), ftell(fs->f)) < 0)
+       int fd;
+       off_t cur;
+
+       if ((fd = fileno(fs->f)) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to determine file descriptor for h264 filestream %p: %s\n", fs, strerror(errno));
                return -1;
-       return 0;
+       }
+       if ((cur = ftello(fs->f)) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to determine current position in h264 filestream %p: %s\n", fs, strerror(errno));
+               return -1;
+       }
+       /* Truncate file to current length */
+       return ftruncate(fd, cur);
 }
 
 static off_t h264_tell(struct ast_filestream *fs)
@@ -159,10 +160,9 @@ static off_t h264_tell(struct ast_filestream *fs)
        return offset; /* XXX totally bogus, needs fixing */
 }
 
-static const struct ast_format h264_f = {
+static struct ast_format_def h264_f = {
        .name = "h264",
        .exts = "h264",
-       .format = AST_FORMAT_H264,
        .open = h264_open,
        .write = h264_write,
        .seek = h264_seek,
@@ -171,27 +171,23 @@ static const struct ast_format h264_f = {
        .read = h264_read,
        .buf_size = BUF_SIZE + AST_FRIENDLY_OFFSET,
        .desc_size = sizeof(struct h264_desc),
-       .module = &mod_data,    /* XXX */
 };
 
-static int load_module(void *mod)
-{
-       return ast_format_register(&h264_f);
-}
-
-static int unload_module(void *mod)
-{
-       return ast_format_unregister(h264_f.name);
-}      
-
-static const char *description(void)
+static int load_module(void)
 {
-       return "Raw h264 data";
+       ast_format_set(&h264_f.format, AST_FORMAT_H264, 0);
+       if (ast_format_def_register(&h264_f))
+               return AST_MODULE_LOAD_FAILURE;
+       return AST_MODULE_LOAD_SUCCESS;
 }
 
-static const char *key(void)
+static int unload_module(void)
 {
-       return ASTERISK_GPL_KEY;
+       return ast_format_def_unregister(h264_f.name);
 }
 
-STD_MOD1;
+AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Raw H.264 data",
+       .load = load_module,
+       .unload = unload_module,
+       .load_pri = AST_MODPRI_APP_DEPEND
+);