6afe539da41a3a0a3bddfb0928c55faecc5f3fda
[asterisk/asterisk.git] / formats / format_g729.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 G729 data.
22  * \note This is not an encoder/decoder. The codec fo g729 is only
23  * available with a commercial license from Digium, due to patent
24  * restrictions. Check http://www.digium.com for information.
25  * \arg Extensions: g729 
26  * \ingroup formats
27  */
28  
29 #include <unistd.h>
30 #include <netinet/in.h>
31 #include <arpa/inet.h>
32 #include <stdlib.h>
33 #include <sys/time.h>
34 #include <stdio.h>
35 #include <errno.h>
36 #include <string.h>
37
38 #include "asterisk.h"
39
40 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
41
42 #include "asterisk/lock.h"
43 #include "asterisk/channel.h"
44 #include "asterisk/file.h"
45 #include "asterisk/logger.h"
46 #include "asterisk/sched.h"
47 #include "asterisk/module.h"
48 #include "asterisk/endian.h"
49
50 /* Some Ideas for this code came from makeg729e.c by Jeffrey Chilton */
51
52 /* Portions of the conversion code are by guido@sienanet.it */
53
54 struct ast_filestream {
55         void *reserved[AST_RESERVED_POINTERS];
56         /* Believe it or not, we must decode/recode to account for the
57            weird MS format */
58         /* This is what a filestream means to us */
59         FILE *f; /* Descriptor */
60         struct ast_frame fr;                            /* Frame information */
61         char waste[AST_FRIENDLY_OFFSET];        /* Buffer for sending frames, etc */
62         char empty;                                                     /* Empty character */
63         unsigned char g729[20];                         /* Two Real G729 Frames */
64 };
65
66
67 AST_MUTEX_DEFINE_STATIC(g729_lock);
68 static int glistcnt = 0;
69
70 static char *name = "g729";
71 static char *desc = "Raw G729 data";
72 static char *exts = "g729";
73
74 static struct ast_filestream *g729_open(FILE *f)
75 {
76         /* We don't have any header to read or anything really, but
77            if we did, it would go here.  We also might want to check
78            and be sure it's a valid file.  */
79         struct ast_filestream *tmp;
80         if ((tmp = malloc(sizeof(struct ast_filestream)))) {
81                 memset(tmp, 0, sizeof(struct ast_filestream));
82                 if (ast_mutex_lock(&g729_lock)) {
83                         ast_log(LOG_WARNING, "Unable to lock g729 list\n");
84                         free(tmp);
85                         return NULL;
86                 }
87                 tmp->f = f;
88                 tmp->fr.data = tmp->g729;
89                 tmp->fr.frametype = AST_FRAME_VOICE;
90                 tmp->fr.subclass = AST_FORMAT_G729A;
91                 /* datalen will vary for each frame */
92                 tmp->fr.src = name;
93                 tmp->fr.mallocd = 0;
94                 glistcnt++;
95                 ast_mutex_unlock(&g729_lock);
96                 ast_update_use_count();
97         }
98         return tmp;
99 }
100
101 static struct ast_filestream *g729_rewrite(FILE *f, const char *comment)
102 {
103         /* We don't have any header to read or anything really, but
104            if we did, it would go here.  We also might want to check
105            and be sure it's a valid file.  */
106         struct ast_filestream *tmp;
107         if ((tmp = malloc(sizeof(struct ast_filestream)))) {
108                 memset(tmp, 0, sizeof(struct ast_filestream));
109                 if (ast_mutex_lock(&g729_lock)) {
110                         ast_log(LOG_WARNING, "Unable to lock g729 list\n");
111                         free(tmp);
112                         return NULL;
113                 }
114                 tmp->f = f;
115                 glistcnt++;
116                 ast_mutex_unlock(&g729_lock);
117                 ast_update_use_count();
118         } else
119                 ast_log(LOG_WARNING, "Out of memory\n");
120         return tmp;
121 }
122
123 static void g729_close(struct ast_filestream *s)
124 {
125         if (ast_mutex_lock(&g729_lock)) {
126                 ast_log(LOG_WARNING, "Unable to lock g729 list\n");
127                 return;
128         }
129         glistcnt--;
130         ast_mutex_unlock(&g729_lock);
131         ast_update_use_count();
132         fclose(s->f);
133         free(s);
134         s = NULL;
135 }
136
137 static struct ast_frame *g729_read(struct ast_filestream *s, int *whennext)
138 {
139         int res;
140         /* Send a frame from the file to the appropriate channel */
141         s->fr.frametype = AST_FRAME_VOICE;
142         s->fr.subclass = AST_FORMAT_G729A;
143         s->fr.offset = AST_FRIENDLY_OFFSET;
144         s->fr.samples = 160;
145         s->fr.datalen = 20;
146         s->fr.mallocd = 0;
147         s->fr.data = s->g729;
148         if ((res = fread(s->g729, 1, 20, s->f)) != 20) {
149                 if (res && (res != 10))
150                         ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", res, strerror(errno));
151                 return NULL;
152         }
153         *whennext = s->fr.samples;
154         return &s->fr;
155 }
156
157 static int g729_write(struct ast_filestream *fs, struct ast_frame *f)
158 {
159         int res;
160         if (f->frametype != AST_FRAME_VOICE) {
161                 ast_log(LOG_WARNING, "Asked to write non-voice frame!\n");
162                 return -1;
163         }
164         if (f->subclass != AST_FORMAT_G729A) {
165                 ast_log(LOG_WARNING, "Asked to write non-G729 frame (%d)!\n", f->subclass);
166                 return -1;
167         }
168         if (f->datalen % 10) {
169                 ast_log(LOG_WARNING, "Invalid data length, %d, should be multiple of 10\n", f->datalen);
170                 return -1;
171         }
172         if ((res = fwrite(f->data, 1, f->datalen, fs->f)) != f->datalen) {
173                         ast_log(LOG_WARNING, "Bad write (%d/10): %s\n", res, strerror(errno));
174                         return -1;
175         }
176         return 0;
177 }
178
179 static char *g729_getcomment(struct ast_filestream *s)
180 {
181         return NULL;
182 }
183
184 static int g729_seek(struct ast_filestream *fs, long sample_offset, int whence)
185 {
186         long bytes;
187         off_t min,cur,max,offset=0;
188         min = 0;
189         cur = ftell(fs->f);
190         fseek(fs->f, 0, SEEK_END);
191         max = ftell(fs->f);
192         
193         bytes = 20 * (sample_offset / 160);
194         if (whence == SEEK_SET)
195                 offset = bytes;
196         else if (whence == SEEK_CUR || whence == SEEK_FORCECUR)
197                 offset = cur + bytes;
198         else if (whence == SEEK_END)
199                 offset = max - bytes;
200         if (whence != SEEK_FORCECUR) {
201                 offset = (offset > max)?max:offset;
202         }
203         /* protect against seeking beyond begining. */
204         offset = (offset < min)?min:offset;
205         if (fseek(fs->f, offset, SEEK_SET) < 0)
206                 return -1;
207         return 0;
208 }
209
210 static int g729_trunc(struct ast_filestream *fs)
211 {
212         /* Truncate file to current length */
213         if (ftruncate(fileno(fs->f), ftell(fs->f)) < 0)
214                 return -1;
215         return 0;
216 }
217
218 static long g729_tell(struct ast_filestream *fs)
219 {
220         off_t offset;
221         offset = ftell(fs->f);
222         return (offset/20)*160;
223 }
224
225 int load_module()
226 {
227         return ast_format_register(name, exts, AST_FORMAT_G729A,
228                                                                 g729_open,
229                                                                 g729_rewrite,
230                                                                 g729_write,
231                                                                 g729_seek,
232                                                                 g729_trunc,
233                                                                 g729_tell,
234                                                                 g729_read,
235                                                                 g729_close,
236                                                                 g729_getcomment);
237                                                                 
238                                                                 
239 }
240
241 int unload_module()
242 {
243         return ast_format_unregister(name);
244 }       
245
246 int usecount()
247 {
248         return glistcnt;
249 }
250
251 char *description()
252 {
253         return desc;
254 }
255
256
257 char *key()
258 {
259         return ASTERISK_GPL_KEY;
260 }