Removed accesses past array bounds.
[speexdsp.git] / src / wav_io.c
1 /* Copyright (C) 2002 Jean-Marc Valin 
2    File: wav_io.c
3    Routines to handle wav (RIFF) headers
4
5    Redistribution and use in source and binary forms, with or without
6    modification, are permitted provided that the following conditions
7    are met:
8    
9    - Redistributions of source code must retain the above copyright
10    notice, this list of conditions and the following disclaimer.
11    
12    - Redistributions in binary form must reproduce the above copyright
13    notice, this list of conditions and the following disclaimer in the
14    documentation and/or other materials provided with the distribution.
15    
16    - Neither the name of the Xiph.org Foundation nor the names of its
17    contributors may be used to endorse or promote products derived from
18    this software without specific prior written permission.
19    
20    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
24    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
25    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
26    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
27    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
28    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
29    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
30    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #ifdef HAVE_CONFIG_H
34 # include "config.h"
35 #endif
36
37 #include <stdio.h>
38 #include <string.h>
39 #include "speex/speex_types.h"
40
41 static spx_uint32_t le_int(spx_uint32_t i)
42 {
43    spx_uint32_t ret=i;
44 #ifdef WORDS_BIGENDIAN
45    ret =  i>>24;
46    ret += (i>>8)&0x0000ff00;
47    ret += (i<<8)&0x00ff0000;
48    ret += (i<<24);
49 #endif
50    return ret;
51 }
52
53 unsigned short be_short(unsigned short s)
54 {
55    unsigned short ret=s;
56 #ifndef WORDS_BIGENDIAN
57    ret =  s>>8;
58    ret += s<<8;
59 #endif
60    return ret;
61 }
62
63 unsigned short le_short(unsigned short s)
64 {
65    unsigned short ret=s;
66 #ifdef WORDS_BIGENDIAN
67    ret =  s>>8;
68    ret += s<<8;
69 #endif
70    return ret;
71 }
72
73
74 int read_wav_header(FILE *file, int *rate, int *channels, int *format, int *size)
75 {
76    char ch[5];
77    int itmp;
78    short stmp;
79    int bpersec;
80    short balign;
81    int skip_bytes;
82    int i;
83
84    ch[4]=0;
85 #if 0
86    fread(ch, 1, 4, file);
87    if (strcmp(ch, "RIFF")!=0)
88    {
89       fseek(file, 0, SEEK_SET);
90       return 0;
91    }
92
93    fread(&itmp, 4, 1, file);
94    *size = le_int(itmp-36);
95
96    fread(ch, 1, 4, file);
97    if (strcmp(ch, "WAVE")!=0)
98    {
99       fprintf (stderr, "RIFF file is not a WAVE file\n");
100       return -1;
101    }
102 #endif
103    fread(ch, 1, 4, file);
104    while (strcmp(ch, "fmt ")!=0)
105    {
106       fread(&itmp, 4, 1, file);
107       itmp = le_int(itmp);
108       /*fprintf (stderr, "skip=%d\n", itmp);*/
109       /*strange way of seeking, but it works even for pipes*/
110       for (i=0;i<itmp;i++)
111          fgetc(file);
112       /*fseek(file, itmp, SEEK_CUR);*/
113       fread(ch, 1, 4, file);
114       if (feof(file))
115       {
116          fprintf (stderr, "Corrupted WAVE file: no \"fmt \"\n");
117          return -1;
118       }
119    }
120    /*if (strcmp(ch, "fmt ")!=0)
121    {
122       fprintf (stderr, "Corrupted WAVE file: no \"fmt \"\n");
123       return -1;
124       }*/
125    
126    fread(&itmp, 4, 1, file);
127    itmp = le_int(itmp);
128    skip_bytes=itmp-16;
129    /*fprintf (stderr, "skip=%d\n", skip_bytes);*/
130    
131    fread(&stmp, 2, 1, file);
132    stmp = le_short(stmp);
133    if (stmp!=1)
134    {
135       fprintf (stderr, "Only PCM encoding is supported\n");
136       return -1;
137    }
138
139    fread(&stmp, 2, 1, file);
140    stmp = le_short(stmp);
141    *channels = stmp;
142    
143    if (stmp>2)
144    {
145       fprintf (stderr, "Only mono and (intensity) stereo supported\n");
146       return -1;
147    }
148
149    fread(&itmp, 4, 1, file);
150    itmp = le_int(itmp);
151    *rate = itmp;
152    if (*rate != 8000 && *rate != 16000 && *rate != 11025 && *rate != 22050 && *rate != 32000 && *rate != 44100 && *rate != 48000)
153    {
154       fprintf (stderr, "Only 8 kHz (narrowband) and 16 kHz (wideband) supported (plus 11.025 kHz and 22.05 kHz, but your mileage may vary)\n");
155       return -1;
156    }
157
158    fread(&itmp, 4, 1, file);
159    bpersec = le_int(itmp);
160
161    fread(&stmp, 2, 1, file);
162    balign = le_short(stmp);
163
164    fread(&stmp, 2, 1, file);
165    stmp = le_short(stmp);
166    if (stmp!=16 && stmp!=8)
167    {
168       fprintf (stderr, "Only 8/16-bit linear supported\n");
169       return -1;
170    }
171    *format=stmp;
172
173    if (bpersec!=*rate**channels*stmp/8)
174    {
175       fprintf (stderr, "Corrupted header: ByteRate mismatch\n");
176       return -1;
177    }
178
179    if (balign!=*channels*stmp/8)
180    {
181       fprintf (stderr, "Corrupted header: BlockAlign mismatch\n");
182       return -1;
183    }
184
185    
186    /*strange way of seeking, but it works even for pipes*/
187    if (skip_bytes>0)
188       for (i=0;i<skip_bytes;i++)
189          fgetc(file);
190
191    /*fseek(file, skip_bytes, SEEK_CUR);*/
192
193    fread(ch, 1, 4, file);
194    while (strcmp(ch, "data")!=0)
195    {
196       fread(&itmp, 4, 1, file);
197       itmp = le_int(itmp);
198       /*strange way of seeking, but it works even for pipes*/
199       for (i=0;i<itmp;i++)
200          fgetc(file);
201       /*fseek(file, itmp, SEEK_CUR);*/
202       fread(ch, 1, 4, file);
203       if (feof(file))
204       {
205          fprintf (stderr, "Corrupted WAVE file: no \"data\"\n");
206          return -1;
207       }
208    }
209
210    /*Ignore this for now*/
211    fread(&itmp, 4, 1, file);
212    itmp = le_int(itmp);
213
214    *size=itmp;
215
216    return 1;
217 }
218
219
220
221 void write_wav_header(FILE *file, int rate, int channels, int format, int size)
222 {
223    char ch[5];
224    int itmp;
225    short stmp;
226
227    ch[4]=0;
228
229    fprintf (file, "RIFF");
230
231    itmp = 0x7fffffff;
232    fwrite(&itmp, 4, 1, file);
233
234    fprintf (file, "WAVEfmt ");
235
236    itmp = le_int(16);
237    fwrite(&itmp, 4, 1, file);
238
239    stmp = le_short(1);
240    fwrite(&stmp, 2, 1, file);
241
242    stmp = le_short(channels);
243    fwrite(&stmp, 2, 1, file);
244
245    itmp = le_int(rate);
246    fwrite(&itmp, 4, 1, file);
247
248    itmp = le_int(rate*channels*2);
249    fwrite(&itmp, 4, 1, file);
250
251    stmp = le_short(2*channels);
252    fwrite(&stmp, 2, 1, file);
253
254    stmp = le_short(16);
255    fwrite(&stmp, 2, 1, file);
256
257    fprintf (file, "data");
258
259    itmp = le_int(0x7fffffff);
260    fwrite(&itmp, 4, 1, file);
261
262
263 }