Largely simplify format handlers (for file copy etc.)
[asterisk/asterisk.git] / formats / format_h263.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Mark Spencer <markster@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief Save to raw, headerless h263 data.
22  * \arg File name extension: h263
23  * \ingroup formats
24  */
25  
26 #include <unistd.h>
27 #include <netinet/in.h>
28 #include <arpa/inet.h>
29 #include <stdlib.h>
30 #include <sys/time.h>
31 #include <stdio.h>
32 #include <errno.h>
33 #include <string.h>
34
35 #include "asterisk.h"
36
37 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
38
39 #include "asterisk/lock.h"
40 #include "asterisk/channel.h"
41 #include "asterisk/file.h"
42 #include "asterisk/logger.h"
43 #include "asterisk/sched.h"
44 #include "asterisk/module.h"
45 #include "asterisk/endian.h"
46
47 /* Some Ideas for this code came from makeh263e.c by Jeffrey Chilton */
48
49 /* Portions of the conversion code are by guido@sienanet.it */
50
51 #define BUF_SIZE        4096    /* Two Real h263 Frames */
52
53 struct h263_desc {
54         unsigned int lastts;
55 };
56
57
58 static int h263_open(struct ast_filestream *s)
59 {
60         unsigned int ts;
61         int res;
62
63         if ((res = fread(&ts, 1, sizeof(ts), s->f)) < sizeof(ts)) {
64                 ast_log(LOG_WARNING, "Empty file!\n");
65                 return -1;
66         }
67         return 0;
68 }
69
70 static struct ast_frame *h263_read(struct ast_filestream *s, int *whennext)
71 {
72         int res;
73         int mark;
74         unsigned short len;
75         unsigned int ts;
76         struct h263_desc *fs = (struct h263_desc *)s->private;
77
78         /* Send a frame from the file to the appropriate channel */
79         if ((res = fread(&len, 1, sizeof(len), s->f)) < 1)
80                 return NULL;
81         len = ntohs(len);
82         mark = (len & 0x8000) ? 1 : 0;
83         len &= 0x7fff;
84         if (len > BUF_SIZE) {
85                 ast_log(LOG_WARNING, "Length %d is too long\n", len);
86                 len = BUF_SIZE; /* XXX truncate ? */
87         }
88         s->fr.frametype = AST_FRAME_VIDEO;
89         s->fr.subclass = AST_FORMAT_H263;
90         s->fr.mallocd = 0;
91         FR_SET_BUF(&s->fr, s->buf, AST_FRIENDLY_OFFSET, len);
92         if ((res = fread(s->fr.data, 1, s->fr.datalen, s->f)) != s->fr.datalen) {
93                 if (res)
94                         ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", res, strerror(errno));
95                 return NULL;
96         }
97         s->fr.samples = fs->lastts;     /* XXX what ? */
98         s->fr.datalen = len;
99         s->fr.subclass |= mark;
100         s->fr.delivery.tv_sec = 0;
101         s->fr.delivery.tv_usec = 0;
102         if ((res = fread(&ts, 1, sizeof(ts), s->f)) == sizeof(ts)) {
103                 fs->lastts = ntohl(ts);
104                 *whennext = fs->lastts * 4/45;
105         } else
106                 *whennext = 0;
107         return &s->fr;
108 }
109
110 static int h263_write(struct ast_filestream *fs, struct ast_frame *f)
111 {
112         int res;
113         unsigned int ts;
114         unsigned short len;
115         int subclass;
116         int mark=0;
117         if (f->frametype != AST_FRAME_VIDEO) {
118                 ast_log(LOG_WARNING, "Asked to write non-video frame!\n");
119                 return -1;
120         }
121         subclass = f->subclass;
122         if (subclass & 0x1)
123                 mark=0x8000;
124         subclass &= ~0x1;
125         if (subclass != AST_FORMAT_H263) {
126                 ast_log(LOG_WARNING, "Asked to write non-h263 frame (%d)!\n", f->subclass);
127                 return -1;
128         }
129         ts = htonl(f->samples);
130         if ((res = fwrite(&ts, 1, sizeof(ts), fs->f)) != sizeof(ts)) {
131                         ast_log(LOG_WARNING, "Bad write (%d/4): %s\n", res, strerror(errno));
132                         return -1;
133         }
134         len = htons(f->datalen | mark);
135         if ((res = fwrite(&len, 1, sizeof(len), fs->f)) != sizeof(len)) {
136                         ast_log(LOG_WARNING, "Bad write (%d/2): %s\n", res, strerror(errno));
137                         return -1;
138         }
139         if ((res = fwrite(f->data, 1, f->datalen, fs->f)) != f->datalen) {
140                         ast_log(LOG_WARNING, "Bad write (%d/%d): %s\n", res, f->datalen, strerror(errno));
141                         return -1;
142         }
143         return 0;
144 }
145
146 static int h263_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
147 {
148         /* No way Jose */
149         return -1;
150 }
151
152 static int h263_trunc(struct ast_filestream *fs)
153 {
154         /* Truncate file to current length */
155         if (ftruncate(fileno(fs->f), ftello(fs->f)) < 0)
156                 return -1;
157         return 0;
158 }
159
160 static off_t h263_tell(struct ast_filestream *fs)
161 {
162         off_t offset = ftello(fs->f);
163         return offset;  /* XXX totally bogus, needs fixing */
164 }
165
166 static struct ast_format_lock me = { .usecnt = -1 };
167
168 static const struct ast_format h263_f = {
169         .name = "h263",
170         .exts = "h264",
171         .format = AST_FORMAT_H263,
172         .open = h263_open,
173         .write = h263_write,
174         .seek = h263_seek,
175         .trunc = h263_trunc,
176         .tell = h263_tell,
177         .read = h263_read,
178         .buf_size = BUF_SIZE + AST_FRIENDLY_OFFSET,
179         .desc_size = sizeof(struct h263_desc),
180         .lockp = &me,
181 };
182
183 int load_module()
184 {
185         return ast_format_register(&h263_f);
186 }
187
188 int unload_module()
189 {
190         return ast_format_unregister(h263_f.name);
191 }       
192
193 int usecount()
194 {
195         return me.usecnt;
196 }
197
198 char *description()
199 {
200         return "Raw h263 data";
201 }
202
203 char *key()
204 {
205         return ASTERISK_GPL_KEY;
206 }