format_pcm: Track actual header size of .au files
[asterisk/asterisk.git] / formats / format_pcm.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2006, 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 Flat, binary, ulaw PCM file format.
22  * \arg File name extension: alaw, al, alw, pcm, ulaw, ul, mu, ulw, g722, au
23  * 
24  * \ingroup formats
25  */
26
27 /*** MODULEINFO
28         <support_level>core</support_level>
29  ***/
30  
31 #include "asterisk.h"
32
33 #include "asterisk/mod_format.h"
34 #include "asterisk/module.h"
35 #include "asterisk/endian.h"
36 #include "asterisk/ulaw.h"
37 #include "asterisk/alaw.h"
38 #include "asterisk/format_cache.h"
39
40 #define BUF_SIZE 160            /* 160 bytes, and same number of samples */
41
42 static char ulaw_silence[BUF_SIZE];
43 static char alaw_silence[BUF_SIZE];
44
45 /* #define REALTIME_WRITE */    /* XXX does it work at all ? */
46
47 #ifdef REALTIME_WRITE
48 struct pcm_desc {
49         unsigned long start_time;
50 };
51
52 /* Returns time in msec since system boot. */
53 static unsigned long get_time(void)
54 {
55         struct tms buf;
56         clock_t cur;
57
58         cur = times( &buf );
59         if( cur < 0 ) {
60                 ast_log( LOG_WARNING, "Cannot get current time\n" );
61                 return 0;
62         }
63         return cur * 1000 / sysconf( _SC_CLK_TCK );
64 }
65
66 static int pcma_open(struct ast_filestream *s)
67 {
68         if (ast_format_cmp(s->fmt->format, ast_format_alaw) == AST_FORMAT_CMP_EQUAL)
69                 pd->starttime = get_time();
70         return 0;
71 }
72
73 static int pcma_rewrite(struct ast_filestream *s, const char *comment)
74 {
75         return pcma_open(s);
76 }
77 #endif
78
79 static struct ast_frame *pcm_read(struct ast_filestream *s, int *whennext)
80 {
81         int res;
82         
83         /* Send a frame from the file to the appropriate channel */
84
85         AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, BUF_SIZE);
86         if ((res = fread(s->fr.data.ptr, 1, s->fr.datalen, s->f)) < 1) {
87                 if (res)
88                         ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", res, strerror(errno));
89                 return NULL;
90         }
91         s->fr.datalen = res;
92         if (ast_format_cmp(s->fmt->format, ast_format_g722) == AST_FORMAT_CMP_EQUAL)
93                 *whennext = s->fr.samples = res * 2;
94         else
95                 *whennext = s->fr.samples = res;
96         return &s->fr;
97 }
98
99 static int pcm_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
100 {
101         off_t cur, max, offset = 0;
102         int ret = -1;   /* assume error */
103
104         if ((cur = ftello(fs->f)) < 0) {
105                 ast_log(AST_LOG_WARNING, "Unable to determine current position in pcm filestream %p: %s\n", fs, strerror(errno));
106                 return -1;
107         }
108
109         if (fseeko(fs->f, 0, SEEK_END) < 0) {
110                 ast_log(AST_LOG_WARNING, "Unable to seek to end of pcm filestream %p: %s\n", fs, strerror(errno));
111                 return -1;
112         }
113
114         if ((max = ftello(fs->f)) < 0) {
115                 ast_log(AST_LOG_WARNING, "Unable to determine max position in pcm filestream %p: %s\n", fs, strerror(errno));
116                 return -1;
117         }
118
119         switch (whence) {
120         case SEEK_SET:
121                 offset = sample_offset;
122                 break;
123         case SEEK_END:
124                 offset = max - sample_offset;
125                 break;
126         case SEEK_CUR:
127         case SEEK_FORCECUR:
128                 offset = cur + sample_offset;
129                 break;
130         default:
131                 ast_log(LOG_WARNING, "invalid whence %d, assuming SEEK_SET\n", whence);
132                 offset = sample_offset;
133         }
134         if (offset < 0) {
135                 ast_log(LOG_WARNING, "negative offset %ld, resetting to 0\n", (long) offset);
136                 offset = 0;
137         }
138         if (whence == SEEK_FORCECUR && offset > max) { /* extend the file */
139                 size_t left = offset - max;
140                 const char *src = (ast_format_cmp(fs->fmt->format, ast_format_alaw) == AST_FORMAT_CMP_EQUAL) ? alaw_silence : ulaw_silence;
141
142                 while (left) {
143                         size_t written = fwrite(src, 1, (left > BUF_SIZE) ? BUF_SIZE : left, fs->f);
144                         if (written == -1)
145                                 break;  /* error */
146                         left -= written;
147                 }
148                 ret = 0; /* successful */
149         } else {
150                 if (offset > max) {
151                         ast_log(LOG_WARNING, "offset too large %ld, truncating to %ld\n", (long) offset, (long) max);
152                         offset = max;
153                 }
154                 ret = fseeko(fs->f, offset, SEEK_SET);
155         }
156         return ret;
157 }
158
159 static int pcm_trunc(struct ast_filestream *fs)
160 {
161         int cur, fd;
162
163         if ((fd = fileno(fs->f)) < 0) {
164                 ast_log(AST_LOG_WARNING, "Unable to determine file descriptor for pcm filestream %p: %s\n", fs, strerror(errno));
165                 return -1;
166         }
167         if ((cur = ftello(fs->f)) < 0) {
168                 ast_log(AST_LOG_WARNING, "Unable to determine current position in pcm filestream %p: %s\n", fs, strerror(errno));
169                 return -1;
170         }
171         /* Truncate file to current length */
172         return ftruncate(fd, cur);
173 }
174
175 static off_t pcm_tell(struct ast_filestream *fs)
176 {
177         return ftello(fs->f);
178 }
179
180 static int pcm_write(struct ast_filestream *fs, struct ast_frame *f)
181 {
182         int res;
183
184 #ifdef REALTIME_WRITE
185         if (ast_format_cmp(s->fmt->format, ast_format_alaw) == AST_FORMAT_CMP_EQUAL) {
186                 struct pcm_desc *pd = (struct pcm_desc *)fs->_private;
187                 struct stat stat_buf;
188                 unsigned long cur_time = get_time();
189                 unsigned long fpos = ( cur_time - pd->start_time ) * 8; /* 8 bytes per msec */
190                 /* Check if we have written to this position yet. If we have, then increment pos by one frame
191                 *  for some degree of protection against receiving packets in the same clock tick.
192                 */
193                 
194                 fstat(fileno(fs->f), &stat_buf );
195                 if (stat_buf.st_size > fpos )
196                         fpos += f->datalen;     /* Incrementing with the size of this current frame */
197
198                 if (stat_buf.st_size < fpos) {
199                         /* fill the gap with 0x55 rather than 0. */
200                         char buf[1024];
201                         unsigned long cur, to_write;
202
203                         cur = stat_buf.st_size;
204                         if (fseek(fs->f, cur, SEEK_SET) < 0) {
205                                 ast_log( LOG_WARNING, "Cannot seek in file: %s\n", strerror(errno) );
206                                 return -1;
207                         }
208                         memset(buf, 0x55, 512);
209                         while (cur < fpos) {
210                                 to_write = fpos - cur;
211                                 if (to_write > sizeof(buf))
212                                         to_write = sizeof(buf);
213                                 fwrite(buf, 1, to_write, fs->f);
214                                 cur += to_write;
215                         }
216                 }
217
218                 if (fseek(s->f, fpos, SEEK_SET) < 0) {
219                         ast_log( LOG_WARNING, "Cannot seek in file: %s\n", strerror(errno) );
220                         return -1;
221                 }
222         }
223 #endif  /* REALTIME_WRITE */
224         
225         if ((res = fwrite(f->data.ptr, 1, f->datalen, fs->f)) != f->datalen) {
226                 ast_log(LOG_WARNING, "Bad write (%d/%d): %s\n", res, f->datalen, strerror(errno));
227                 return -1;
228         }
229         return 0;
230 }
231
232 /* SUN .au support routines */
233
234 #define MIN_AU_HEADER_SIZE      24
235 #define AU_HEADER(var)          uint32_t var[6]
236
237 #define AU_HDR_MAGIC_OFF        0
238 #define AU_HDR_HDR_SIZE_OFF     1
239 #define AU_HDR_DATA_SIZE_OFF    2
240 #define AU_HDR_ENCODING_OFF     3
241 #define AU_HDR_SAMPLE_RATE_OFF  4
242 #define AU_HDR_CHANNELS_OFF     5
243
244 #define AU_ENC_8BIT_ULAW        1
245
246 #define AU_MAGIC 0x2e736e64
247 #if __BYTE_ORDER == __BIG_ENDIAN
248 #define htoll(b) (b)
249 #define htols(b) (b)
250 #define ltohl(b) (b)
251 #define ltohs(b) (b)
252 #else
253 #if __BYTE_ORDER == __LITTLE_ENDIAN
254 #define htoll(b)  \
255           (((((b)      ) & 0xFF) << 24) | \
256                ((((b) >>  8) & 0xFF) << 16) | \
257                    ((((b) >> 16) & 0xFF) <<  8) | \
258                    ((((b) >> 24) & 0xFF)      ))
259 #define htols(b) \
260           (((((b)      ) & 0xFF) << 8) | \
261                    ((((b) >> 8) & 0xFF)      ))
262 #define ltohl(b) htoll(b)
263 #define ltohs(b) htols(b)
264 #else
265 #error "Endianess not defined"
266 #endif
267 #endif
268
269 struct au_desc {
270         uint32_t hdr_size;
271 };
272
273 static int check_header(struct ast_filestream *fs)
274 {
275         AU_HEADER(header);
276         uint32_t magic;
277         uint32_t hdr_size;
278         uint32_t data_size;
279         uint32_t encoding;
280         uint32_t sample_rate;
281         uint32_t channels;
282
283         struct au_desc *desc = fs->_private;
284         FILE *f = fs->f;
285
286         if (fread(header, 1, MIN_AU_HEADER_SIZE, f) != MIN_AU_HEADER_SIZE) {
287                 ast_log(LOG_WARNING, "Read failed (header)\n");
288                 return -1;
289         }
290         magic = ltohl(header[AU_HDR_MAGIC_OFF]);
291         if (magic != (uint32_t) AU_MAGIC) {
292                 ast_log(LOG_WARNING, "Bad magic: 0x%x\n", magic);
293         }
294         hdr_size = ltohl(header[AU_HDR_HDR_SIZE_OFF]);
295         if (hdr_size < MIN_AU_HEADER_SIZE) {
296                 hdr_size = MIN_AU_HEADER_SIZE;
297         }
298 /*      data_size = ltohl(header[AU_HDR_DATA_SIZE_OFF]); */
299         encoding = ltohl(header[AU_HDR_ENCODING_OFF]);
300         if (encoding != AU_ENC_8BIT_ULAW) {
301                 ast_log(LOG_WARNING, "Unexpected format: %u. Only 8bit ULAW allowed (%d)\n", encoding, AU_ENC_8BIT_ULAW);
302                 return -1;
303         }
304         sample_rate = ltohl(header[AU_HDR_SAMPLE_RATE_OFF]);
305         if (sample_rate != DEFAULT_SAMPLE_RATE) {
306                 ast_log(LOG_WARNING, "Sample rate can only be 8000 not %u\n", sample_rate);
307                 return -1;
308         }
309         channels = ltohl(header[AU_HDR_CHANNELS_OFF]);
310         if (channels != 1) {
311                 ast_log(LOG_WARNING, "Not in mono: channels=%u\n", channels);
312                 return -1;
313         }
314         /* Skip to data */
315         fseek(f, 0, SEEK_END);
316         data_size = ftell(f) - hdr_size;
317         if (fseek(f, hdr_size, SEEK_SET) == -1 ) {
318                 ast_log(LOG_WARNING, "Failed to skip to data: %u\n", hdr_size);
319                 return -1;
320         }
321
322         /* We'll need this later */
323         desc->hdr_size = hdr_size;
324
325         return data_size;
326 }
327
328 static int update_header(struct ast_filestream *fs)
329 {
330         off_t cur, end;
331         uint32_t datalen;
332         int bytes;
333         struct au_desc *desc = fs->_private;
334         FILE *f = fs->f;
335
336         cur = ftell(f);
337         fseek(f, 0, SEEK_END);
338         end = ftell(f);
339         /* data starts 24 bytes in */
340         bytes = end - desc->hdr_size;
341         datalen = htoll(bytes);
342
343         if (cur < 0) {
344                 ast_log(LOG_WARNING, "Unable to find our position\n");
345                 return -1;
346         }
347         if (fseek(f, AU_HDR_DATA_SIZE_OFF * sizeof(uint32_t), SEEK_SET)) {
348                 ast_log(LOG_WARNING, "Unable to set our position\n");
349                 return -1;
350         }
351         if (fwrite(&datalen, 1, sizeof(datalen), f) != sizeof(datalen)) {
352                 ast_log(LOG_WARNING, "Unable to set write file size\n");
353                 return -1;
354         }
355         if (fseek(f, cur, SEEK_SET)) {
356                 ast_log(LOG_WARNING, "Unable to return to position\n");
357                 return -1;
358         }
359         return 0;
360 }
361
362 static int write_header(struct ast_filestream *fs)
363 {
364         struct au_desc *desc = fs->_private;
365         FILE *f = fs->f;
366
367         AU_HEADER(header);
368
369         header[AU_HDR_MAGIC_OFF] = htoll((uint32_t) AU_MAGIC);
370         header[AU_HDR_HDR_SIZE_OFF] = htoll(desc->hdr_size);
371         header[AU_HDR_DATA_SIZE_OFF] = 0;
372         header[AU_HDR_ENCODING_OFF] = htoll(AU_ENC_8BIT_ULAW);
373         header[AU_HDR_SAMPLE_RATE_OFF] = htoll(DEFAULT_SAMPLE_RATE);
374         header[AU_HDR_CHANNELS_OFF] = htoll(1);
375
376         /* Write an au header, ignoring sizes which will be filled in later */
377         fseek(f, 0, SEEK_SET);
378         if (fwrite(header, 1, MIN_AU_HEADER_SIZE, f) != MIN_AU_HEADER_SIZE) {
379                 ast_log(LOG_WARNING, "Unable to write header\n");
380                 return -1;
381         }
382         return 0;
383 }
384
385 static int au_open(struct ast_filestream *s)
386 {
387         if (check_header(s) < 0)
388                 return -1;
389         return 0;
390 }
391
392 static int au_rewrite(struct ast_filestream *s, const char *comment)
393 {
394         struct au_desc *desc = s->_private;
395
396         desc->hdr_size = MIN_AU_HEADER_SIZE;
397
398         if (write_header(s))
399                 return -1;
400         return 0;
401 }
402
403 /* XXX check this, probably incorrect */
404 static int au_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
405 {
406         off_t min, max, cur;
407         long offset = 0, bytes;
408         struct au_desc *desc = fs->_private;
409
410         min = desc->hdr_size;
411
412         if (ast_format_cmp(fs->fmt->format, ast_format_g722) == AST_FORMAT_CMP_EQUAL)
413                 bytes = sample_offset / 2;
414         else
415                 bytes = sample_offset;
416
417         if ((cur = ftello(fs->f)) < 0) {
418                 ast_log(AST_LOG_WARNING, "Unable to determine current position in au filestream %p: %s\n", fs, strerror(errno));
419                 return -1;
420         }
421
422         if (fseeko(fs->f, 0, SEEK_END) < 0) {
423                 ast_log(AST_LOG_WARNING, "Unable to seek to end of au filestream %p: %s\n", fs, strerror(errno));
424                 return -1;
425         }
426
427         if ((max = ftello(fs->f)) < 0) {
428                 ast_log(AST_LOG_WARNING, "Unable to determine max position in au filestream %p: %s\n", fs, strerror(errno));
429                 return -1;
430         }
431
432         if (whence == SEEK_SET)
433                 offset = bytes + min;
434         else if (whence == SEEK_CUR || whence == SEEK_FORCECUR)
435                 offset = bytes + cur;
436         else if (whence == SEEK_END)
437                 offset = max - bytes;
438
439         if (whence != SEEK_FORCECUR) {
440                 offset = (offset > max) ? max : offset;
441         }
442
443         /* always protect the header space. */
444         offset = (offset < min) ? min : offset;
445
446         return fseeko(fs->f, offset, SEEK_SET);
447 }
448
449 static int au_trunc(struct ast_filestream *fs)
450 {
451         int fd;
452         off_t cur;
453
454         if ((fd = fileno(fs->f)) < 0) {
455                 ast_log(AST_LOG_WARNING, "Unable to determine file descriptor for au filestream %p: %s\n", fs, strerror(errno));
456                 return -1;
457         }
458         if ((cur = ftello(fs->f)) < 0) {
459                 ast_log(AST_LOG_WARNING, "Unable to determine current position in au filestream %p: %s\n", fs, strerror(errno));
460                 return -1;
461         }
462         /* Truncate file to current length */
463         if (ftruncate(fd, cur)) {
464                 return -1;
465         }
466         return update_header(fs);
467 }
468
469 static off_t au_tell(struct ast_filestream *fs)
470 {
471         struct au_desc *desc = fs->_private;
472         off_t offset = ftello(fs->f);
473         return offset - desc->hdr_size;
474 }
475
476 static struct ast_format_def alaw_f = {
477         .name = "alaw",
478         .exts = "alaw|al|alw",
479         .write = pcm_write,
480         .seek = pcm_seek,
481         .trunc = pcm_trunc,
482         .tell = pcm_tell,
483         .read = pcm_read,
484         .buf_size = BUF_SIZE + AST_FRIENDLY_OFFSET,
485 #ifdef REALTIME_WRITE
486         .open = pcma_open,
487         .rewrite = pcma_rewrite,
488         .desc_size = sizeof(struct pcm_desc),
489 #endif
490 };
491
492 static struct ast_format_def pcm_f = {
493         .name = "pcm",
494         .exts = "pcm|ulaw|ul|mu|ulw",
495         .write = pcm_write,
496         .seek = pcm_seek,
497         .trunc = pcm_trunc,
498         .tell = pcm_tell,
499         .read = pcm_read,
500         .buf_size = BUF_SIZE + AST_FRIENDLY_OFFSET,
501 };
502
503 static struct ast_format_def g722_f = {
504         .name = "g722",
505         .exts = "g722",
506         .write = pcm_write,
507         .seek = pcm_seek,
508         .trunc = pcm_trunc,
509         .tell = pcm_tell,
510         .read = pcm_read,
511         .buf_size = (BUF_SIZE * 2) + AST_FRIENDLY_OFFSET,
512 };
513
514 static struct ast_format_def au_f = {
515         .name = "au",
516         .exts = "au",
517         .open = au_open,
518         .rewrite = au_rewrite,
519         .write = pcm_write,
520         .seek = au_seek,
521         .trunc = au_trunc,
522         .tell = au_tell,
523         .read = pcm_read,
524         .buf_size = BUF_SIZE + AST_FRIENDLY_OFFSET,     /* this many shorts */
525         .desc_size = sizeof(struct au_desc),
526 };
527
528 static int unload_module(void)
529 {
530         return ast_format_def_unregister(pcm_f.name)
531                 || ast_format_def_unregister(alaw_f.name)
532                 || ast_format_def_unregister(au_f.name)
533                 || ast_format_def_unregister(g722_f.name);
534 }
535
536 static int load_module(void)
537 {
538         int i;
539
540         /* XXX better init ? */
541         for (i = 0; i < ARRAY_LEN(ulaw_silence); i++)
542                 ulaw_silence[i] = AST_LIN2MU(0);
543         for (i = 0; i < ARRAY_LEN(alaw_silence); i++)
544                 alaw_silence[i] = AST_LIN2A(0);
545
546         pcm_f.format = ast_format_ulaw;
547         alaw_f.format = ast_format_alaw;
548         au_f.format = ast_format_ulaw;
549         g722_f.format = ast_format_g722;
550         if ( ast_format_def_register(&pcm_f)
551                 || ast_format_def_register(&alaw_f)
552                 || ast_format_def_register(&au_f)
553                 || ast_format_def_register(&g722_f) ) {
554                 unload_module();
555                 return AST_MODULE_LOAD_DECLINE;
556         }
557         return AST_MODULE_LOAD_SUCCESS;
558 }
559
560 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Raw/Sun uLaw/ALaw 8KHz (PCM,PCMA,AU), G.722 16Khz",
561         .support_level = AST_MODULE_SUPPORT_CORE,
562         .load = load_module,
563         .unload = unload_module,
564         .load_pri = AST_MODPRI_APP_DEPEND
565 );