Title update
[asterisk/asterisk.git] / formats / format_wav.c
index 5cf39ce..4e24186 100644 (file)
  * \ingroup formats
  */
  
+/*** MODULEINFO
+       <support_level>core</support_level>
+ ***/
+
 #include "asterisk.h"
 
 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
@@ -38,7 +42,10 @@ ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
 
 #define        WAV_BUF_SIZE    320
 
+#define WAV_HEADER_SIZE 44
+
 struct wav_desc {      /* format-specific parameters */
+       int hz;
        int bytes;
        int lasttimeout;
        int maxlen;
@@ -70,49 +77,13 @@ struct wav_desc {   /* format-specific parameters */
 #endif
 
 
-static int check_header(FILE *f)
+static int check_header_fmt(FILE *f, int hsize, int hz)
 {
-       int type, size, formtype;
-       int fmt, hsize;
        short format, chans, bysam, bisam;
        int bysec;
        int freq;
-       int data;
-       if (fread(&type, 1, 4, f) != 4) {
-               ast_log(LOG_WARNING, "Read failed (type)\n");
-               return -1;
-       }
-       if (fread(&size, 1, 4, f) != 4) {
-               ast_log(LOG_WARNING, "Read failed (size)\n");
-               return -1;
-       }
-       size = ltohl(size);
-       if (fread(&formtype, 1, 4, f) != 4) {
-               ast_log(LOG_WARNING, "Read failed (formtype)\n");
-               return -1;
-       }
-       if (memcmp(&type, "RIFF", 4)) {
-               ast_log(LOG_WARNING, "Does not begin with RIFF\n");
-               return -1;
-       }
-       if (memcmp(&formtype, "WAVE", 4)) {
-               ast_log(LOG_WARNING, "Does not contain WAVE\n");
-               return -1;
-       }
-       if (fread(&fmt, 1, 4, f) != 4) {
-               ast_log(LOG_WARNING, "Read failed (fmt)\n");
-               return -1;
-       }
-       if (memcmp(&fmt, "fmt ", 4)) {
-               ast_log(LOG_WARNING, "Does not say fmt\n");
-               return -1;
-       }
-       if (fread(&hsize, 1, 4, f) != 4) {
-               ast_log(LOG_WARNING, "Read failed (formtype)\n");
-               return -1;
-       }
-       if (ltohl(hsize) < 16) {
-               ast_log(LOG_WARNING, "Unexpected header size %d\n", ltohl(hsize));
+       if (hsize < 16) {
+               ast_log(LOG_WARNING, "Unexpected header size %d\n", hsize);
                return -1;
        }
        if (fread(&format, 1, 2, f) != 2) {
@@ -135,8 +106,10 @@ static int check_header(FILE *f)
                ast_log(LOG_WARNING, "Read failed (freq)\n");
                return -1;
        }
-       if (ltohl(freq) != DEFAULT_SAMPLE_RATE) {
-               ast_log(LOG_WARNING, "Unexpected freqency %d\n", ltohl(freq));
+       if (((ltohl(freq) != 8000) && (ltohl(freq) != 16000)) ||
+           ((ltohl(freq) == 8000) && (hz != 8000)) ||
+           ((ltohl(freq) == 16000) && (hz != 16000))) {
+               ast_log(LOG_WARNING, "Unexpected frequency mismatch %d (expecting %d)\n", ltohl(freq),hz);
                return -1;
        }
        /* Ignore the byte frequency */
@@ -158,8 +131,36 @@ static int check_header(FILE *f)
                return -1;
        }
        /* Skip any additional header */
-       if (fseek(f,ltohl(hsize)-16,SEEK_CUR) == -1 ) {
-               ast_log(LOG_WARNING, "Failed to skip remaining header bytes: %d\n", ltohl(hsize)-16 );
+       if (fseek(f,hsize-16,SEEK_CUR) == -1 ) {
+               ast_log(LOG_WARNING, "Failed to skip remaining header bytes: %d\n", hsize-16 );
+               return -1;
+       }
+       return 0;
+}
+
+static int check_header(FILE *f, int hz)
+{
+       int type, size, formtype;
+       int data;
+       if (fread(&type, 1, 4, f) != 4) {
+               ast_log(LOG_WARNING, "Read failed (type)\n");
+               return -1;
+       }
+       if (fread(&size, 1, 4, f) != 4) {
+               ast_log(LOG_WARNING, "Read failed (size)\n");
+               return -1;
+       }
+       size = ltohl(size);
+       if (fread(&formtype, 1, 4, f) != 4) {
+               ast_log(LOG_WARNING, "Read failed (formtype)\n");
+               return -1;
+       }
+       if (memcmp(&type, "RIFF", 4)) {
+               ast_log(LOG_WARNING, "Does not begin with RIFF\n");
+               return -1;
+       }
+       if (memcmp(&formtype, "WAVE", 4)) {
+               ast_log(LOG_WARNING, "Does not contain WAVE\n");
                return -1;
        }
        /* Skip any facts and get the first data block */
@@ -169,23 +170,25 @@ static int check_header(FILE *f)
            
            /* Begin data chunk */
            if (fread(&buf, 1, 4, f) != 4) {
-                       ast_log(LOG_WARNING, "Read failed (data)\n");
+                       ast_log(LOG_WARNING, "Read failed (block header format)\n");
                        return -1;
            }
            /* Data has the actual length of data in it */
            if (fread(&data, 1, 4, f) != 4) {
-                       ast_log(LOG_WARNING, "Read failed (data)\n");
+                       ast_log(LOG_WARNING, "Read failed (block '%.4s' header length)\n", buf);
                        return -1;
            }
            data = ltohl(data);
+               if (memcmp(&buf, "fmt ", 4) == 0) {
+                       if (check_header_fmt(f, data, hz))
+                               return -1;
+                       continue;
+               }
            if(memcmp(buf, "data", 4) == 0 ) 
                        break;
-           if(memcmp(buf, "fact", 4) != 0 ) {
-                       ast_log(LOG_WARNING, "Unknown block - not fact or data\n");
-                       return -1;
-           }
+               ast_log(LOG_DEBUG, "Skipping unknown block '%.4s'\n", buf);
            if (fseek(f,data,SEEK_CUR) == -1 ) {
-                       ast_log(LOG_WARNING, "Failed to skip fact block: %d\n", data );
+                       ast_log(LOG_WARNING, "Failed to skip '%.4s' block: %d\n", buf, data);
                        return -1;
            }
        }
@@ -239,16 +242,24 @@ static int update_header(FILE *f)
        return 0;
 }
 
-static int write_header(FILE *f)
+static int write_header(FILE *f, int writehz)
 {
-       unsigned int hz=htoll(8000);
-       unsigned int bhz = htoll(16000);
+       unsigned int hz;
+       unsigned int bhz;
        unsigned int hs = htoll(16);
        unsigned short fmt = htols(1);
        unsigned short chans = htols(1);
        unsigned short bysam = htols(2);
        unsigned short bisam = htols(16);
        unsigned int size = htoll(0);
+
+       if (writehz == 16000) {
+               hz = htoll(16000);
+               bhz = htoll(32000);
+       } else {
+               hz = htoll(8000);
+               bhz = htoll(16000);
+       }
        /* Write a wav header, ignoring sizes which will be filled in later */
        fseek(f,0,SEEK_SET);
        if (fwrite("RIFF", 1, 4, f) != 4) {
@@ -308,7 +319,7 @@ static int wav_open(struct ast_filestream *s)
           if we did, it would go here.  We also might want to check
           and be sure it's a valid file.  */
        struct wav_desc *tmp = (struct wav_desc *)s->_private;
-       if ((tmp->maxlen = check_header(s->f)) < 0)
+       if ((tmp->maxlen = check_header(s->f, (s->fmt->format.id == AST_FORMAT_SLINEAR16 ? 16000 : 8000))) < 0)
                return -1;
        return 0;
 }
@@ -319,7 +330,9 @@ static int wav_rewrite(struct ast_filestream *s, const char *comment)
           if we did, it would go here.  We also might want to check
           and be sure it's a valid file.  */
 
-       if (write_header(s->f))
+       struct wav_desc *tmp = (struct wav_desc *)s->_private;
+       tmp->hz = (s->fmt->format.id == AST_FORMAT_SLINEAR16 ? 16000 : 8000);
+       if (write_header(s->f,tmp->hz))
                return -1;
        return 0;
 }
@@ -329,13 +342,20 @@ static void wav_close(struct ast_filestream *s)
        char zero = 0;
        struct wav_desc *fs = (struct wav_desc *)s->_private;
 
+       if (s->mode == O_RDONLY) {
+               return;
+       }
+
        if (s->filename) {
                update_header(s->f);
        }
 
        /* Pad to even length */
-       if (fs->bytes & 0x1)
-               fwrite(&zero, 1, 1, s->f);
+       if (fs->bytes & 0x1) {
+               if (!fwrite(&zero, 1, 1, s->f)) {
+                       ast_log(LOG_WARNING, "fwrite() failed: %s\n", strerror(errno));
+               }
+       }
 }
 
 static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
@@ -344,13 +364,15 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
        int samples;    /* actual samples read */
 #if __BYTE_ORDER == __BIG_ENDIAN
        int x;
-#endif
        short *tmp;
-       int bytes = WAV_BUF_SIZE;       /* in bytes */
+#endif
+       int bytes;
        off_t here;
        /* Send a frame from the file to the appropriate channel */
        struct wav_desc *fs = (struct wav_desc *)s->_private;
 
+       bytes = (fs->hz == 16000 ? (WAV_BUF_SIZE * 2) : WAV_BUF_SIZE);
+
        here = ftello(s->f);
        if (fs->maxlen - here < bytes)          /* truncate if necessary */
                bytes = fs->maxlen - here;
@@ -358,7 +380,7 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
                bytes = 0;
 /*     ast_debug(1, "here: %d, maxlen: %d, bytes: %d\n", here, s->maxlen, bytes); */
        s->fr.frametype = AST_FRAME_VOICE;
-       s->fr.subclass = AST_FORMAT_SLINEAR;
+       ast_format_set(&s->fr.subclass.format, (fs->hz == 16000 ? AST_FORMAT_SLINEAR16 : AST_FORMAT_SLINEAR), 0);
        s->fr.mallocd = 0;
        AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, bytes);
        
@@ -370,8 +392,8 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
        s->fr.datalen = res;
        s->fr.samples = samples = res / 2;
 
-       tmp = (short *)(s->fr.data.ptr);
 #if __BYTE_ORDER == __BIG_ENDIAN
+       tmp = (short *)(s->fr.data.ptr);
        /* file format is little endian so we need to swap */
        for( x = 0; x < samples; x++)
                tmp[x] = (tmp[x] << 8) | ((tmp[x] & 0xff00) >> 8);
@@ -385,7 +407,7 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
 {
 #if __BYTE_ORDER == __BIG_ENDIAN
        int x;
-       short tmp[8000], *tmpi;
+       short tmp[16000], *tmpi;
 #endif
        struct wav_desc *s = (struct wav_desc *)fs->_private;
        int res;
@@ -394,8 +416,12 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
                ast_log(LOG_WARNING, "Asked to write non-voice frame!\n");
                return -1;
        }
-       if (f->subclass != AST_FORMAT_SLINEAR) {
-               ast_log(LOG_WARNING, "Asked to write non-SLINEAR frame (%d)!\n", f->subclass);
+       if ((f->subclass.format.id != AST_FORMAT_SLINEAR) && (f->subclass.format.id != AST_FORMAT_SLINEAR16)) {
+               ast_log(LOG_WARNING, "Asked to write non-SLINEAR%s frame (%s)!\n", s->hz == 16000 ? "16" : "", ast_getformatname(&f->subclass.format));
+               return -1;
+       }
+       if (ast_format_cmp(&f->subclass.format, &fs->fmt->format) == AST_FORMAT_CMP_NOT_EQUAL) {
+               ast_log(LOG_WARNING, "Can't change SLINEAR frequency during write\n");
                return -1;
        }
        if (!f->datalen)
@@ -428,13 +454,25 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
 
 static int wav_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
 {
-       off_t min, max, cur, offset = 0, samples;
+       off_t min = WAV_HEADER_SIZE, max, cur, offset = 0, samples;
 
        samples = sample_offset * 2; /* SLINEAR is 16 bits mono, so sample_offset * 2 = bytes */
-       min = 44; /* wav header is 44 bytes */
-       cur = ftello(fs->f);
-       fseeko(fs->f, 0, SEEK_END);
-       max = ftello(fs->f);
+
+       if ((cur = ftello(fs->f)) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to determine current position in wav filestream %p: %s\n", fs, strerror(errno));
+               return -1;
+       }
+
+       if (fseeko(fs->f, 0, SEEK_END) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to seek to end of wav filestream %p: %s\n", fs, strerror(errno));
+               return -1;
+       }
+
+       if ((max = ftello(fs->f)) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to determine max position in wav filestream %p: %s\n", fs, strerror(errno));
+               return -1;
+       }
+
        if (whence == SEEK_SET)
                offset = samples + min;
        else if (whence == SEEK_CUR || whence == SEEK_FORCECUR)
@@ -451,8 +489,21 @@ static int wav_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
 
 static int wav_trunc(struct ast_filestream *fs)
 {
-       if (ftruncate(fileno(fs->f), ftello(fs->f)))
+       int fd;
+       off_t cur;
+
+       if ((fd = fileno(fs->f)) < 0) {
+               ast_log(AST_LOG_WARNING, "Unable to determine file descriptor for wav 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 wav filestream %p: %s\n", fs, strerror(errno));
                return -1;
+       }
+       /* Truncate file to current length */
+       if (ftruncate(fd, cur)) {
+               return -1;
+       }
        return update_header(fs->f);
 }
 
@@ -464,10 +515,24 @@ static off_t wav_tell(struct ast_filestream *fs)
        return (offset - 44)/2;
 }
 
-static const struct ast_format wav_f = {
+static struct ast_format_def wav16_f = {
+       .name = "wav16",
+       .exts = "wav16",
+       .open = wav_open,
+       .rewrite = wav_rewrite,
+       .write = wav_write,
+       .seek = wav_seek,
+       .trunc = wav_trunc,
+       .tell = wav_tell,
+       .read = wav_read,
+       .close = wav_close,
+       .buf_size = (WAV_BUF_SIZE * 2) + AST_FRIENDLY_OFFSET,
+       .desc_size = sizeof(struct wav_desc),
+};
+
+static struct ast_format_def wav_f = {
        .name = "wav",
        .exts = "wav",
-       .format = AST_FORMAT_SLINEAR,
        .open = wav_open,
        .rewrite = wav_rewrite,
        .write = wav_write,
@@ -482,14 +547,22 @@ static const struct ast_format wav_f = {
 
 static int load_module(void)
 {
-       if (ast_format_register(&wav_f))
+       ast_format_set(&wav_f.format, AST_FORMAT_SLINEAR, 0);
+       ast_format_set(&wav16_f.format, AST_FORMAT_SLINEAR16, 0);
+       if (ast_format_def_register(&wav_f)
+               || ast_format_def_register(&wav16_f))
                return AST_MODULE_LOAD_FAILURE;
        return AST_MODULE_LOAD_SUCCESS;
 }
 
 static int unload_module(void)
 {
-       return ast_format_unregister(wav_f.name);
-}      
+       return ast_format_def_unregister(wav_f.name)
+               || ast_format_def_unregister(wav16_f.name);
+}
 
-AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Microsoft WAV format (8000Hz Signed Linear)");
+AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Microsoft WAV/WAV16 format (8kHz/16kHz Signed Linear)",
+       .load = load_module,
+       .unload = unload_module,
+       .load_pri = AST_MODPRI_APP_DEPEND
+);