res_rtp: Fix regression when IPv6 is not available.
[asterisk/asterisk.git] / formats / format_wav.c
index 956b13e..a415140 100644 (file)
 
 #include "asterisk.h"
 
-ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
-
 #include "asterisk/mod_format.h"
 #include "asterisk/module.h"
 #include "asterisk/endian.h"
+#include "asterisk/format_cache.h"
+#include "asterisk/format.h"
+#include "asterisk/codec.h"
 
 /* Some Ideas for this code came from makewave.c by Jeffrey Chilton */
 
@@ -42,6 +43,8 @@ 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;
@@ -59,14 +62,14 @@ struct wav_desc {   /* format-specific parameters */
 #define ltohs(b) (b)
 #else
 #if __BYTE_ORDER == __BIG_ENDIAN
-#define htoll(b)  \
-          (((((b)      ) & 0xFF) << 24) | \
-              ((((b) >>  8) & 0xFF) << 16) | \
-                  ((((b) >> 16) & 0xFF) <<  8) | \
-                  ((((b) >> 24) & 0xFF)      ))
+#define htoll(b) \
+       (((((b)      ) & 0xFF) << 24) | \
+       ((( (b) >>  8) & 0xFF) << 16) | \
+       ((( (b) >> 16) & 0xFF) <<  8) | \
+       ((( (b) >> 24) & 0xFF)      ))
 #define htols(b) \
-          (((((b)      ) & 0xFF) << 8) | \
-                  ((((b) >> 8) & 0xFF)      ))
+       (((((b)      ) & 0xFF) <<  8) | \
+       ((( (b) >>  8) & 0xFF)      ))
 #define ltohl(b) htoll(b)
 #define ltohs(b) htols(b)
 #else
@@ -81,7 +84,7 @@ static int check_header_fmt(FILE *f, int hsize, int hz)
        int bysec;
        int freq;
        if (hsize < 16) {
-               ast_log(LOG_WARNING, "Unexpected header size %d\n", ltohl(hsize));
+               ast_log(LOG_WARNING, "Unexpected header size %d\n", hsize);
                return -1;
        }
        if (fread(&format, 1, 2, f) != 2) {
@@ -89,7 +92,7 @@ static int check_header_fmt(FILE *f, int hsize, int hz)
                return -1;
        }
        if (ltohs(format) != 1) {
-               ast_log(LOG_WARNING, "Not a wav file %d\n", ltohs(format));
+               ast_log(LOG_WARNING, "Not a supported wav file format (%d). Only PCM encoded, 16 bit, mono, 8kHz files are supported with a lowercase '.wav' extension.\n", ltohs(format));
                return -1;
        }
        if (fread(&chans, 1, 2, f) != 2) {
@@ -105,8 +108,8 @@ static int check_header_fmt(FILE *f, int hsize, int hz)
                return -1;
        }
        if (((ltohl(freq) != 8000) && (ltohl(freq) != 16000)) ||
-           ((ltohl(freq) == 8000) && (hz != 8000)) ||
-           ((ltohl(freq) == 16000) && (hz != 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;
        }
@@ -129,8 +132,8 @@ static int check_header_fmt(FILE *f, int hsize, int hz)
                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;
@@ -148,7 +151,9 @@ static int check_header(FILE *f, int hz)
                ast_log(LOG_WARNING, "Read failed (size)\n");
                return -1;
        }
+#if __BYTE_ORDER == __BIG_ENDIAN
        size = ltohl(size);
+#endif
        if (fread(&formtype, 1, 4, f) != 4) {
                ast_log(LOG_WARNING, "Read failed (formtype)\n");
                return -1;
@@ -165,30 +170,32 @@ static int check_header(FILE *f, int hz)
        for(;;)
        { 
                char buf[4];
-           
-           /* Begin data chunk */
-           if (fread(&buf, 1, 4, f) != 4) {
+               
+               /* Begin data chunk */
+               if (fread(&buf, 1, 4, f) != 4) {
                        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) {
+               }
+               /* Data has the actual length of data in it */
+               if (fread(&data, 1, 4, f) != 4) {
                        ast_log(LOG_WARNING, "Read failed (block '%.4s' header length)\n", buf);
                        return -1;
-           }
-           data = ltohl(data);
+               }
+#if __BYTE_ORDER == __BIG_ENDIAN
+               data = ltohl(data);
+#endif
                if (memcmp(&buf, "fmt ", 4) == 0) {
                        if (check_header_fmt(f, data, hz))
                                return -1;
                        continue;
                }
-           if(memcmp(buf, "data", 4) == 0 ) 
+               if(memcmp(buf, "data", 4) == 0 ) 
                        break;
                ast_log(LOG_DEBUG, "Skipping unknown block '%.4s'\n", buf);
-           if (fseek(f,data,SEEK_CUR) == -1 ) {
+               if (fseek(f,data,SEEK_CUR) == -1 ) {
                        ast_log(LOG_WARNING, "Failed to skip '%.4s' block: %d\n", buf, data);
                        return -1;
-           }
+               }
        }
 #if 0
        curpos = lseek(fd, 0, SEEK_CUR);
@@ -317,7 +324,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, (s->fmt->format.id == AST_FORMAT_SLINEAR16 ? 16000 : 8000))) < 0)
+       if ((tmp->maxlen = check_header(s->f, ast_format_get_sample_rate(s->fmt->format))) < 0)
                return -1;
        return 0;
 }
@@ -329,7 +336,7 @@ static int wav_rewrite(struct ast_filestream *s, const char *comment)
           and be sure it's a valid file.  */
 
        struct wav_desc *tmp = (struct wav_desc *)s->_private;
-       tmp->hz = (s->fmt->format.id == AST_FORMAT_SLINEAR16 ? 16000 : 8000);
+       tmp->hz = ast_format_get_sample_rate(s->fmt->format);
        if (write_header(s->f,tmp->hz))
                return -1;
        return 0;
@@ -377,9 +384,6 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
        if (bytes < 0)
                bytes = 0;
 /*     ast_debug(1, "here: %d, maxlen: %d, bytes: %d\n", here, s->maxlen, bytes); */
-       s->fr.frametype = AST_FRAME_VOICE;
-       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);
        
        if ( (res = fread(s->fr.data.ptr, 1, s->fr.datalen, s->f)) <= 0 ) {
@@ -410,18 +414,6 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
        struct wav_desc *s = (struct wav_desc *)fs->_private;
        int res;
 
-       if (f->frametype != AST_FRAME_VOICE) {
-               ast_log(LOG_WARNING, "Asked to write non-voice frame!\n");
-               return -1;
-       }
-       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)
                return -1;
 
@@ -452,20 +444,33 @@ 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 (whence == SEEK_SET)
+
+       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)
+       } else if (whence == SEEK_CUR || whence == SEEK_FORCECUR) {
                offset = samples + cur;
-       else if (whence == SEEK_END)
+       } else if (whence == SEEK_END) {
                offset = max - samples;
-        if (whence != SEEK_FORCECUR) {
+       }
+       if (whence != SEEK_FORCECUR) {
                offset = (offset > max)?max:offset;
        }
        /* always protect the header space. */
@@ -475,8 +480,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);
 }
 
@@ -520,8 +538,8 @@ static struct ast_format_def wav_f = {
 
 static int load_module(void)
 {
-       ast_format_set(&wav_f.format, AST_FORMAT_SLINEAR, 0);
-       ast_format_set(&wav16_f.format, AST_FORMAT_SLINEAR16, 0);
+       wav_f.format = ast_format_slin;
+       wav16_f.format = ast_format_slin16;
        if (ast_format_def_register(&wav_f)
                || ast_format_def_register(&wav16_f))
                return AST_MODULE_LOAD_FAILURE;
@@ -535,6 +553,7 @@ static int unload_module(void)
 }
 
 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Microsoft WAV/WAV16 format (8kHz/16kHz Signed Linear)",
+       .support_level = AST_MODULE_SUPPORT_CORE,
        .load = load_module,
        .unload = unload_module,
        .load_pri = AST_MODPRI_APP_DEPEND