a39a57a56e1692689d02b049da97f5fef375374c
[Faustine.git] / interpretor / libsndfile-1.0.25 / programs / sndfile-salvage.c
1 /*
2 ** Copyright (C) 2010-2011 Erik de Castro Lopo <erikd@mega-nerd.com>
3 **
4 ** All rights reserved.
5 **
6 ** Redistribution and use in source and binary forms, with or without
7 ** modification, are permitted provided that the following conditions are
8 ** met:
9 **
10 ** * Redistributions of source code must retain the above copyright
11 ** notice, this list of conditions and the following disclaimer.
12 ** * Redistributions in binary form must reproduce the above copyright
13 ** notice, this list of conditions and the following disclaimer in
14 ** the documentation and/or other materials provided with the
15 ** distribution.
16 ** * Neither the author nor the names of any contributors may be used
17 ** to endorse or promote products derived from this software without
18 ** 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 LIMITED
22 ** TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23 ** PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 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 PROFITS;
27 ** OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
28 ** WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
29 ** OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
30 ** ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <inttypes.h>
37 #include <ctype.h>
38 #include <math.h>
39 #include <errno.h>
40 #include <unistd.h>
41 #include <fcntl.h>
42 #include <sys/stat.h>
43 #include <sys/types.h>
44
45 #include <sndfile.h>
46
47 #include "common.h"
48
49 #define BUFFER_LEN (1 << 16)
50
51 #define NOT(x) (! (x))
52
53
54 static void usage_exit (const char *progname) ;
55 static void salvage_file (const char * broken_wav, const char * fixed_w64) ;
56
57 int
58 main (int argc, char *argv [])
59 {
60 if (argc != 3)
61 usage_exit (program_name (argv [0])) ;
62
63 salvage_file (argv [1], argv [2]) ;
64
65 return 0 ;
66 } /* main */
67
68 /*==============================================================================
69 */
70
71 static void lseek_or_die (int fd, off_t offset, int whence) ;
72 static sf_count_t get_file_length (int fd, const char * name) ;
73 static sf_count_t find_data_offset (int fd, int format) ;
74 static void copy_data (int fd, SNDFILE * sndfile, int readsize) ;
75
76
77 static void
78 usage_exit (const char *progname)
79 { printf ("Usage :\n\n %s <broken wav file> <fixed w64 file>\n\n", progname) ;
80 puts ("Salvages the audio data from WAV files which are more than 4G in length.\n") ;
81 printf ("Using %s.\n\n", sf_version_string ()) ;
82 exit (0) ;
83 } /* usage_exit */
84
85 static void
86 salvage_file (const char * broken_wav, const char * fixed_w64)
87 { SNDFILE * sndfile ;
88 SF_INFO sfinfo ;
89 sf_count_t broken_len, data_offset ;
90 int fd, read_size ;
91
92 if (strcmp (broken_wav, fixed_w64) == 0)
93 { printf ("Error : Input and output files must be different.\n\n") ;
94 exit (1) ;
95 } ;
96
97 if ((fd = open (broken_wav, O_RDONLY)) < 0)
98 { printf ("Error : Not able to open file '%s' : %s\n", broken_wav, strerror (errno)) ;
99 exit (1) ;
100 } ;
101
102 broken_len = get_file_length (fd, broken_wav) ;
103 if (broken_len <= 0xffffffff)
104 printf ("File is not greater than 4Gig but salvaging anyway.\n") ;
105
106 /* Grab the format info from the broken file. */
107 memset (&sfinfo, 0, sizeof (sfinfo)) ;
108 if ((sndfile = sf_open (broken_wav, SFM_READ, &sfinfo)) == NULL)
109 { printf ("sf_open ('%s') failed : %s\n", broken_wav, sf_strerror (NULL)) ;
110 exit (1) ;
111 } ;
112 sf_close (sndfile) ;
113
114 data_offset = find_data_offset (fd, sfinfo.format & SF_FORMAT_TYPEMASK) ;
115
116 printf ("Offset to audio data : %" PRId64 "\n", data_offset) ;
117
118 switch (sfinfo.format & SF_FORMAT_TYPEMASK)
119 { case SF_FORMAT_WAV :
120 case SF_FORMAT_WAVEX :
121 sfinfo.format = SF_FORMAT_W64 | (sfinfo.format & SF_FORMAT_SUBMASK) ;
122 break ;
123
124 default :
125 printf ("Don't currently support this file type.\n") ;
126 exit (1) ;
127 } ;
128
129 switch (sfinfo.format & SF_FORMAT_SUBMASK)
130 { case SF_FORMAT_PCM_U8 :
131 case SF_FORMAT_PCM_S8 :
132 read_size = 1 ;
133 break ;
134
135 case SF_FORMAT_PCM_16 :
136 read_size = 2 ;
137 break ;
138
139 case SF_FORMAT_PCM_24 :
140 read_size = 3 ;
141 break ;
142
143 case SF_FORMAT_PCM_32 :
144 case SF_FORMAT_FLOAT :
145 read_size = 4 ;
146 break ;
147
148 case SF_FORMAT_DOUBLE :
149 read_size = 8 ;
150 break ;
151
152 default :
153 printf ("Sorry, don't currently support this file encoding type.\n") ;
154 exit (1) ;
155 } ;
156
157 read_size *= sfinfo.channels ;
158
159 if ((sndfile = sf_open (fixed_w64, SFM_WRITE, &sfinfo)) == NULL)
160 { printf ("sf_open ('%s') failed : %s\n", broken_wav, sf_strerror (NULL)) ;
161 exit (1) ;
162 } ;
163
164 lseek_or_die (fd, data_offset, SEEK_SET) ;
165
166 copy_data (fd, sndfile, read_size) ;
167
168 sf_close (sndfile) ;
169
170 puts ("Done!") ;
171 } /* salvage_file */
172
173 /*------------------------------------------------------------------------------
174 */
175
176 static void
177 lseek_or_die (int fd, off_t offset, int whence)
178 {
179 if (lseek (fd, offset, whence) < 0)
180 { printf ("lseek failed : %s\n", strerror (errno)) ;
181 exit (1) ;
182 } ;
183
184 return ;
185 } /* lseek_or_die */
186
187
188 static sf_count_t
189 get_file_length (int fd, const char * name)
190 { struct stat sbuf ;
191
192 if (sizeof (sbuf.st_size) != 8)
193 { puts ("Error : sizeof (sbuf.st_size) != 8. Was program compiled with\n"
194 " 64 bit file offsets?\n") ;
195 exit (1) ;
196 } ;
197
198 if (fstat (fd, &sbuf) != 0)
199 { printf ("Error : fstat ('%s') failed : %s\n", name, strerror (errno)) ;
200 exit (1) ;
201 } ;
202
203 return sbuf.st_size ;
204 } /* get_file_length */
205
206 static sf_count_t
207 find_data_offset (int fd, int format)
208 { char buffer [8192], *cptr ;
209 const char * target = "XXXX" ;
210 sf_count_t offset = -1, extra ;
211 int rlen, slen ;
212
213 switch (format)
214 { case SF_FORMAT_WAV :
215 case SF_FORMAT_WAVEX :
216 target = "data" ;
217 extra = 8 ;
218 break ;
219
220 case SF_FORMAT_AIFF :
221 target = "SSND" ;
222 extra = 16 ;
223 break ;
224
225 default :
226 puts ("Error : Sorry, don't handle this input file format.\n") ;
227 exit (1) ;
228 } ;
229
230 slen = strlen (target) ;
231
232 lseek_or_die (fd, 0, SEEK_SET) ;
233
234 printf ("Searching for '%s' maker.\n", target) ;
235
236 if ((rlen = read (fd, buffer, sizeof (buffer))) < 0)
237 { printf ("Error : failed read : %s\n", strerror (errno)) ;
238 exit (1) ;
239 } ;
240
241 cptr = memchr (buffer, target [0], rlen - slen) ;
242 if (cptr && memcmp (cptr, target, slen) == 0)
243 offset = cptr - buffer ;
244 else
245 { printf ("Error : Could not find data offset.\n") ;
246 exit (1) ;
247 } ;
248
249 return offset + extra ;
250 } /* find_data_offset */
251
252 static void
253 copy_data (int fd, SNDFILE * sndfile, int readsize)
254 { static char * buffer ;
255 sf_count_t readlen, count ;
256 int bufferlen, done = 0 ;
257
258 bufferlen = readsize * 1024 ;
259 buffer = malloc (bufferlen) ;
260
261 while (NOT (done) && (readlen = read (fd, buffer, bufferlen)) >= 0)
262 { if (readlen < bufferlen)
263 { readlen -= readlen % readsize ;
264 done = 1 ;
265 } ;
266
267 if ((count = sf_write_raw (sndfile, buffer, readlen)) != readlen)
268 { printf ("Error : sf_write_raw returned %" PRId64 " : %s\n", count, sf_strerror (sndfile)) ;
269 return ;
270 } ;
271 } ;
272
273 free (buffer) ;
274
275 return ;
276 } /* copy_data */
277