blob: 13f1c3f7c38f242fafa9143c6f17fbbde03f627c [file] [log] [blame]
Josh Coalsoncae58732002-05-04 17:30:32 +00001/* libFLAC - Free Lossless Audio Codec library
Josh Coalson95643902004-01-17 04:14:43 +00002 * Copyright (C) 2001,2002,2003,2004 Josh Coalson
Josh Coalsoncae58732002-05-04 17:30:32 +00003 *
Josh Coalsonafd81072003-01-31 23:34:56 +00004 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
Josh Coalsoncae58732002-05-04 17:30:32 +00007 *
Josh Coalsonafd81072003-01-31 23:34:56 +00008 * - Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
Josh Coalsoncae58732002-05-04 17:30:32 +000010 *
Josh Coalsonafd81072003-01-31 23:34:56 +000011 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * - Neither the name of the Xiph.org Foundation nor the names of its
16 * contributors may be used to endorse or promote products derived from
17 * this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Josh Coalsoncae58732002-05-04 17:30:32 +000030 */
31
32#include <errno.h>
33#include <stdio.h>
34#include <stdlib.h>
35#include <string.h>
36
37#if defined _MSC_VER || defined __MINGW32__
38#include <sys/utime.h> /* for utime() */
39#include <io.h> /* for chmod() */
40#else
41#include <sys/types.h> /* some flavors of BSD (like OS X) require this to get time_t */
42#include <utime.h> /* for utime() */
43#include <unistd.h> /* for chown(), unlink() */
44#endif
45#include <sys/stat.h> /* for stat(), maybe chmod() */
46
Josh Coalson90ced912002-05-30 05:23:38 +000047#include "private/metadata.h"
48
Josh Coalsoncae58732002-05-04 17:30:32 +000049#include "FLAC/assert.h"
Josh Coalsoncae58732002-05-04 17:30:32 +000050#include "FLAC/file_decoder.h"
51
52#ifdef max
53#undef max
54#endif
55#define max(a,b) ((a)>(b)?(a):(b))
56#ifdef min
57#undef min
58#endif
59#define min(a,b) ((a)<(b)?(a):(b))
60
61
62/****************************************************************************
63 *
64 * Local function declarations
65 *
66 ***************************************************************************/
67
Josh Coalsoncae58732002-05-04 17:30:32 +000068static void pack_uint32_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes);
69static void pack_uint32_little_endian_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes);
70static void pack_uint64_(FLAC__uint64 val, FLAC__byte *b, unsigned bytes);
71static FLAC__uint32 unpack_uint32_(FLAC__byte *b, unsigned bytes);
72static FLAC__uint32 unpack_uint32_little_endian_(FLAC__byte *b, unsigned bytes);
73static FLAC__uint64 unpack_uint64_(FLAC__byte *b, unsigned bytes);
74
Josh Coalsoncc682512002-06-08 04:53:42 +000075static FLAC__bool read_metadata_block_header_(FLAC__Metadata_SimpleIterator *iterator);
76static FLAC__bool read_metadata_block_data_(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block);
Josh Coalsonea25cb32004-07-10 00:39:01 +000077static FLAC__bool read_metadata_block_header_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__bool *is_last, FLAC__MetadataType *type, unsigned *length);
78static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Seek seek_cb, FLAC__StreamMetadata *block);
79static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_streaminfo_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_StreamInfo *block);
80static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_padding_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Seek seek_cb, FLAC__StreamMetadata_Padding *block, unsigned block_length);
81static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_application_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_Application *block, unsigned block_length);
82static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_seektable_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_SeekTable *block, unsigned block_length);
83static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_entry_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_VorbisComment_Entry *entry);
84static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_VorbisComment *block);
85static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_cuesheet_track_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_CueSheet_Track *track);
86static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_cuesheet_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_CueSheet *block);
87static FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_unknown_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_Unknown *block, unsigned block_length);
Josh Coalsoncae58732002-05-04 17:30:32 +000088
Josh Coalsoncc682512002-06-08 04:53:42 +000089static FLAC__bool write_metadata_block_header_(FILE *file, FLAC__Metadata_SimpleIteratorStatus *status, const FLAC__StreamMetadata *block);
90static FLAC__bool write_metadata_block_data_(FILE *file, FLAC__Metadata_SimpleIteratorStatus *status, const FLAC__StreamMetadata *block);
Josh Coalson463a22c2004-07-11 06:41:00 +000091static FLAC__bool write_metadata_block_header_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata *block);
92static FLAC__bool write_metadata_block_data_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata *block);
93static FLAC__bool write_metadata_block_data_streaminfo_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_StreamInfo *block);
94static FLAC__bool write_metadata_block_data_padding_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_Padding *block, unsigned block_length);
95static FLAC__bool write_metadata_block_data_application_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_Application *block, unsigned block_length);
96static FLAC__bool write_metadata_block_data_seektable_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_SeekTable *block);
97static FLAC__bool write_metadata_block_data_vorbis_comment_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_VorbisComment *block);
98static FLAC__bool write_metadata_block_data_cuesheet_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_CueSheet *block);
99static FLAC__bool write_metadata_block_data_unknown_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_Unknown *block, unsigned block_length);
100
Josh Coalsoncc682512002-06-08 04:53:42 +0000101static FLAC__bool write_metadata_block_stationary_(FLAC__Metadata_SimpleIterator *iterator, const FLAC__StreamMetadata *block);
102static FLAC__bool write_metadata_block_stationary_with_padding_(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block, unsigned padding_length, FLAC__bool padding_is_last);
103static FLAC__bool rewrite_whole_file_(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block, FLAC__bool append);
Josh Coalsoncae58732002-05-04 17:30:32 +0000104
Josh Coalsoncc682512002-06-08 04:53:42 +0000105static void simple_iterator_push_(FLAC__Metadata_SimpleIterator *iterator);
106static FLAC__bool simple_iterator_pop_(FLAC__Metadata_SimpleIterator *iterator);
Josh Coalsoncae58732002-05-04 17:30:32 +0000107
Josh Coalsonea25cb32004-07-10 00:39:01 +0000108static unsigned seek_to_first_metadata_block_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Seek seek_cb);
Josh Coalsoncae58732002-05-04 17:30:32 +0000109static unsigned seek_to_first_metadata_block_(FILE *f);
110
Josh Coalsoncc682512002-06-08 04:53:42 +0000111static FLAC__bool simple_iterator_copy_file_prefix_(FLAC__Metadata_SimpleIterator *iterator, FILE **tempfile, char **tempfilename, FLAC__bool append);
112static FLAC__bool simple_iterator_copy_file_postfix_(FLAC__Metadata_SimpleIterator *iterator, FILE **tempfile, char **tempfilename, int fixup_is_last_code, long fixup_is_last_flag_offset, FLAC__bool backup);
Josh Coalsoncae58732002-05-04 17:30:32 +0000113
Josh Coalsoncc682512002-06-08 04:53:42 +0000114static FLAC__bool copy_n_bytes_from_file_(FILE *file, FILE *tempfile, unsigned bytes/*@@@ 4G limit*/, FLAC__Metadata_SimpleIteratorStatus *status);
Josh Coalson463a22c2004-07-11 06:41:00 +0000115static FLAC__bool copy_n_bytes_from_file_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOHandle temp_handle, FLAC__IOCallback_Write temp_write_cb, unsigned bytes/*@@@ 4G limit*/, FLAC__Metadata_SimpleIteratorStatus *status);
Josh Coalsoncc682512002-06-08 04:53:42 +0000116static FLAC__bool copy_remaining_bytes_from_file_(FILE *file, FILE *tempfile, FLAC__Metadata_SimpleIteratorStatus *status);
Josh Coalson463a22c2004-07-11 06:41:00 +0000117static FLAC__bool copy_remaining_bytes_from_file_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Eof eof_cb, FLAC__IOHandle temp_handle, FLAC__IOCallback_Write temp_write_cb, FLAC__Metadata_SimpleIteratorStatus *status);
Josh Coalsoncae58732002-05-04 17:30:32 +0000118
Josh Coalsoncc682512002-06-08 04:53:42 +0000119static FLAC__bool open_tempfile_(const char *filename, const char *tempfile_path_prefix, FILE **tempfile, char **tempfilename, FLAC__Metadata_SimpleIteratorStatus *status);
120static FLAC__bool transport_tempfile_(const char *filename, FILE **tempfile, char **tempfilename, FLAC__Metadata_SimpleIteratorStatus *status);
Josh Coalsoncae58732002-05-04 17:30:32 +0000121static void cleanup_tempfile_(FILE **tempfile, char **tempfilename);
122
123static FLAC__bool get_file_stats_(const char *filename, struct stat *stats);
124static void set_file_stats_(const char *filename, struct stat *stats);
125
Josh Coalsonea25cb32004-07-10 00:39:01 +0000126static int fseek_wrapper_(FLAC__IOHandle handle, FLAC__int64 offset, int whence);
127static FLAC__int64 ftell_wrapper_(FLAC__IOHandle handle);
128
Josh Coalsoncc682512002-06-08 04:53:42 +0000129static FLAC__Metadata_ChainStatus get_equivalent_status_(FLAC__Metadata_SimpleIteratorStatus status);
Josh Coalsoncae58732002-05-04 17:30:32 +0000130
131
Josh Coalsonf91251d2002-12-28 07:04:49 +0000132#ifdef FLAC__VALGRIND_TESTING
133static size_t local__fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream)
134{
135 size_t ret = fwrite(ptr, size, nmemb, stream);
136 if(!ferror(stream))
137 fflush(stream);
138 return ret;
139}
140#else
141#define local__fwrite fwrite
142#endif
143
Josh Coalsoncae58732002-05-04 17:30:32 +0000144/****************************************************************************
145 *
146 * Level 0 implementation
147 *
148 ***************************************************************************/
149
Josh Coalson57ba6f42002-06-07 05:27:37 +0000150static FLAC__StreamDecoderWriteStatus write_callback_(const FLAC__FileDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data);
Josh Coalsoncc682512002-06-08 04:53:42 +0000151static void metadata_callback_(const FLAC__FileDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000152static void error_callback_(const FLAC__FileDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data);
153
Josh Coalson57ba6f42002-06-07 05:27:37 +0000154typedef struct {
155 FLAC__bool got_error;
Josh Coalson1aca6b12004-07-30 01:54:29 +0000156 FLAC__bool got_object;
157 FLAC__StreamMetadata *object;
Josh Coalson57ba6f42002-06-07 05:27:37 +0000158} level0_client_data;
159
Josh Coalson6afed9f2002-10-16 22:29:47 +0000160FLAC_API FLAC__bool FLAC__metadata_get_streaminfo(const char *filename, FLAC__StreamMetadata *streaminfo)
Josh Coalsoncae58732002-05-04 17:30:32 +0000161{
Josh Coalson57ba6f42002-06-07 05:27:37 +0000162 level0_client_data cd;
Josh Coalson9b86c962002-07-10 07:11:15 +0000163 FLAC__FileDecoder *decoder;
164
165 FLAC__ASSERT(0 != filename);
166 FLAC__ASSERT(0 != streaminfo);
167
168 decoder = FLAC__file_decoder_new();
Josh Coalsoncae58732002-05-04 17:30:32 +0000169
170 if(0 == decoder)
171 return false;
172
Josh Coalson57ba6f42002-06-07 05:27:37 +0000173 cd.got_error = false;
Josh Coalson1aca6b12004-07-30 01:54:29 +0000174 cd.got_object = false;
175 cd.object = 0;
Josh Coalson57ba6f42002-06-07 05:27:37 +0000176
Josh Coalsoncae58732002-05-04 17:30:32 +0000177 FLAC__file_decoder_set_md5_checking(decoder, false);
178 FLAC__file_decoder_set_filename(decoder, filename);
Josh Coalsonb4e186a2002-08-12 21:35:43 +0000179 FLAC__file_decoder_set_metadata_ignore_all(decoder);
180 FLAC__file_decoder_set_metadata_respond(decoder, FLAC__METADATA_TYPE_STREAMINFO);
Josh Coalsoncae58732002-05-04 17:30:32 +0000181 FLAC__file_decoder_set_write_callback(decoder, write_callback_);
182 FLAC__file_decoder_set_metadata_callback(decoder, metadata_callback_);
183 FLAC__file_decoder_set_error_callback(decoder, error_callback_);
Josh Coalson57ba6f42002-06-07 05:27:37 +0000184 FLAC__file_decoder_set_client_data(decoder, &cd);
Josh Coalsoncae58732002-05-04 17:30:32 +0000185
Josh Coalson57ba6f42002-06-07 05:27:37 +0000186 if(FLAC__file_decoder_init(decoder) != FLAC__FILE_DECODER_OK || cd.got_error) {
187 FLAC__file_decoder_finish(decoder);
188 FLAC__file_decoder_delete(decoder);
Josh Coalsoncae58732002-05-04 17:30:32 +0000189 return false;
Josh Coalson57ba6f42002-06-07 05:27:37 +0000190 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000191
Josh Coalsonb4e186a2002-08-12 21:35:43 +0000192 if(!FLAC__file_decoder_process_until_end_of_metadata(decoder) || cd.got_error) {
Josh Coalson57ba6f42002-06-07 05:27:37 +0000193 FLAC__file_decoder_finish(decoder);
194 FLAC__file_decoder_delete(decoder);
Josh Coalson1aca6b12004-07-30 01:54:29 +0000195 if(0 != cd.object)
196 FLAC__metadata_object_delete(cd.object);
Josh Coalsoncae58732002-05-04 17:30:32 +0000197 return false;
Josh Coalson57ba6f42002-06-07 05:27:37 +0000198 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000199
Josh Coalson51ee4652002-05-09 05:46:23 +0000200 FLAC__file_decoder_finish(decoder);
Josh Coalsoncae58732002-05-04 17:30:32 +0000201 FLAC__file_decoder_delete(decoder);
202
Josh Coalson1aca6b12004-07-30 01:54:29 +0000203 if(cd.got_object) {
204 /* can just copy the contents since STREAMINFO has no internal structure */
205 *streaminfo = *(cd.object);
206 }
207
208 if(0 != cd.object)
209 FLAC__metadata_object_delete(cd.object);
210
211 return cd.got_object;
212}
213
214FLAC_API FLAC__bool FLAC__metadata_get_tags(const char *filename, FLAC__StreamMetadata **tags)
215{
216 level0_client_data cd;
217 FLAC__FileDecoder *decoder;
218
219 FLAC__ASSERT(0 != filename);
220 FLAC__ASSERT(0 != tags);
221
222 decoder = FLAC__file_decoder_new();
223
224 if(0 == decoder)
225 return false;
226
227 *tags = 0;
228
229 cd.got_error = false;
230 cd.got_object = false;
231 cd.object = 0;
232
233 FLAC__file_decoder_set_md5_checking(decoder, false);
234 FLAC__file_decoder_set_filename(decoder, filename);
235 FLAC__file_decoder_set_metadata_ignore_all(decoder);
236 FLAC__file_decoder_set_metadata_respond(decoder, FLAC__METADATA_TYPE_VORBIS_COMMENT);
237 FLAC__file_decoder_set_write_callback(decoder, write_callback_);
238 FLAC__file_decoder_set_metadata_callback(decoder, metadata_callback_);
239 FLAC__file_decoder_set_error_callback(decoder, error_callback_);
240 FLAC__file_decoder_set_client_data(decoder, &cd);
241
242 if(FLAC__file_decoder_init(decoder) != FLAC__FILE_DECODER_OK || cd.got_error) {
243 FLAC__file_decoder_finish(decoder);
244 FLAC__file_decoder_delete(decoder);
245 return false;
246 }
247
248 if(!FLAC__file_decoder_process_until_end_of_metadata(decoder) || cd.got_error) {
249 FLAC__file_decoder_finish(decoder);
250 FLAC__file_decoder_delete(decoder);
251 if(0 != cd.object)
252 FLAC__metadata_object_delete(cd.object);
253 return false;
254 }
255
256 FLAC__file_decoder_finish(decoder);
257 FLAC__file_decoder_delete(decoder);
258
259 if(cd.got_object)
260 *tags = cd.object;
261
262 return cd.got_object;
Josh Coalsoncae58732002-05-04 17:30:32 +0000263}
264
Josh Coalson57ba6f42002-06-07 05:27:37 +0000265FLAC__StreamDecoderWriteStatus write_callback_(const FLAC__FileDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data)
Josh Coalsoncae58732002-05-04 17:30:32 +0000266{
267 (void)decoder, (void)frame, (void)buffer, (void)client_data;
268
Josh Coalson2835f432002-06-04 05:54:44 +0000269 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
Josh Coalsoncae58732002-05-04 17:30:32 +0000270}
271
Josh Coalsoncc682512002-06-08 04:53:42 +0000272void metadata_callback_(const FLAC__FileDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data)
Josh Coalsoncae58732002-05-04 17:30:32 +0000273{
Josh Coalson57ba6f42002-06-07 05:27:37 +0000274 level0_client_data *cd = (level0_client_data *)client_data;
Josh Coalsoncae58732002-05-04 17:30:32 +0000275 (void)decoder;
276
Josh Coalson1aca6b12004-07-30 01:54:29 +0000277 /*
278 * we assume we only get here when the one metadata block we were
279 * looking for was passed to us
280 */
281 if(!cd->got_object) {
282 if(0 == (cd->object = FLAC__metadata_object_clone(metadata)))
283 cd->got_error = true;
284 else
285 cd->got_object = true;
Josh Coalson57ba6f42002-06-07 05:27:37 +0000286 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000287}
288
289void error_callback_(const FLAC__FileDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
290{
Josh Coalson57ba6f42002-06-07 05:27:37 +0000291 level0_client_data *cd = (level0_client_data *)client_data;
Josh Coalsoncae58732002-05-04 17:30:32 +0000292 (void)decoder;
293
Josh Coalson2835f432002-06-04 05:54:44 +0000294 if(status != FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC)
Josh Coalson57ba6f42002-06-07 05:27:37 +0000295 cd->got_error = true;
Josh Coalsoncae58732002-05-04 17:30:32 +0000296}
297
298
299/****************************************************************************
300 *
301 * Level 1 implementation
302 *
303 ***************************************************************************/
304
305#define SIMPLE_ITERATOR_MAX_PUSH_DEPTH (1+4)
306/* 1 for initial offset, +4 for our own personal use */
307
Josh Coalsoncc682512002-06-08 04:53:42 +0000308struct FLAC__Metadata_SimpleIterator {
Josh Coalsoncae58732002-05-04 17:30:32 +0000309 FILE *file;
310 char *filename, *tempfile_path_prefix;
311 struct stat stats;
312 FLAC__bool has_stats;
313 FLAC__bool is_writable;
Josh Coalsoncc682512002-06-08 04:53:42 +0000314 FLAC__Metadata_SimpleIteratorStatus status;
Josh Coalsoncae58732002-05-04 17:30:32 +0000315 /*@@@ 2G limits here because of the offset type: */
316 long offset[SIMPLE_ITERATOR_MAX_PUSH_DEPTH];
317 long first_offset; /* this is the offset to the STREAMINFO block */
318 unsigned depth;
319 /* this is the metadata block header of the current block we are pointing to: */
320 FLAC__bool is_last;
Josh Coalsoncc682512002-06-08 04:53:42 +0000321 FLAC__MetadataType type;
Josh Coalsoncae58732002-05-04 17:30:32 +0000322 unsigned length;
323};
324
Josh Coalson6afed9f2002-10-16 22:29:47 +0000325FLAC_API const char * const FLAC__Metadata_SimpleIteratorStatusString[] = {
Josh Coalsoncae58732002-05-04 17:30:32 +0000326 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK",
327 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT",
328 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE",
329 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_A_FLAC_FILE",
330 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_WRITABLE",
Josh Coalsona7b673c2002-05-25 02:13:38 +0000331 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_BAD_METADATA",
Josh Coalsoncae58732002-05-04 17:30:32 +0000332 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR",
333 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR",
334 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR",
335 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_RENAME_ERROR",
336 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_UNLINK_ERROR",
337 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR",
338 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_INTERNAL_ERROR"
339};
340
341
Josh Coalson6afed9f2002-10-16 22:29:47 +0000342FLAC_API FLAC__Metadata_SimpleIterator *FLAC__metadata_simple_iterator_new()
Josh Coalsoncae58732002-05-04 17:30:32 +0000343{
Josh Coalson5e3fcb22002-11-06 07:11:26 +0000344 FLAC__Metadata_SimpleIterator *iterator = (FLAC__Metadata_SimpleIterator*)calloc(1, sizeof(FLAC__Metadata_SimpleIterator));
Josh Coalsoncae58732002-05-04 17:30:32 +0000345
346 if(0 != iterator) {
347 iterator->file = 0;
348 iterator->filename = 0;
349 iterator->tempfile_path_prefix = 0;
350 iterator->has_stats = false;
351 iterator->is_writable = false;
352 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
353 iterator->first_offset = iterator->offset[0] = -1;
354 iterator->depth = 0;
355 }
356
357 return iterator;
358}
359
Josh Coalsoncc682512002-06-08 04:53:42 +0000360static void simple_iterator_free_guts_(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000361{
362 FLAC__ASSERT(0 != iterator);
363
364 if(0 != iterator->file) {
365 fclose(iterator->file);
366 iterator->file = 0;
367 if(iterator->has_stats)
368 set_file_stats_(iterator->filename, &iterator->stats);
369 }
370 if(0 != iterator->filename) {
371 free(iterator->filename);
372 iterator->filename = 0;
373 }
374 if(0 != iterator->tempfile_path_prefix) {
375 free(iterator->tempfile_path_prefix);
376 iterator->tempfile_path_prefix = 0;
377 }
378}
379
Josh Coalson6afed9f2002-10-16 22:29:47 +0000380FLAC_API void FLAC__metadata_simple_iterator_delete(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000381{
382 FLAC__ASSERT(0 != iterator);
383
384 simple_iterator_free_guts_(iterator);
385 free(iterator);
386}
387
Josh Coalson6afed9f2002-10-16 22:29:47 +0000388FLAC_API FLAC__Metadata_SimpleIteratorStatus FLAC__metadata_simple_iterator_status(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000389{
Josh Coalsoncc682512002-06-08 04:53:42 +0000390 FLAC__Metadata_SimpleIteratorStatus status;
Josh Coalsoncae58732002-05-04 17:30:32 +0000391
392 FLAC__ASSERT(0 != iterator);
393
394 status = iterator->status;
395 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
396 return status;
397}
398
Josh Coalson3ac66932002-08-30 05:41:31 +0000399static FLAC__bool simple_iterator_prime_input_(FLAC__Metadata_SimpleIterator *iterator, FLAC__bool read_only)
Josh Coalsoncae58732002-05-04 17:30:32 +0000400{
401 unsigned ret;
402
403 FLAC__ASSERT(0 != iterator);
404
Josh Coalson3ac66932002-08-30 05:41:31 +0000405 if(read_only || 0 == (iterator->file = fopen(iterator->filename, "r+b"))) {
406 iterator->is_writable = false;
Josh Coalson3ac66932002-08-30 05:41:31 +0000407 if(read_only || errno == EACCES) {
Josh Coalsoncae58732002-05-04 17:30:32 +0000408 if(0 == (iterator->file = fopen(iterator->filename, "rb"))) {
409 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE;
410 return false;
411 }
412 }
413 else {
414 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE;
415 return false;
416 }
417 }
418 else {
419 iterator->is_writable = true;
420 }
421
422 ret = seek_to_first_metadata_block_(iterator->file);
423 switch(ret) {
424 case 0:
425 iterator->depth = 0;
426 iterator->first_offset = iterator->offset[iterator->depth] = ftell(iterator->file);
427 return read_metadata_block_header_(iterator);
428 case 1:
429 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
430 return false;
431 case 2:
Josh Coalsonea25cb32004-07-10 00:39:01 +0000432 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
433 return false;
434 case 3:
Josh Coalsoncae58732002-05-04 17:30:32 +0000435 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_A_FLAC_FILE;
436 return false;
437 default:
438 FLAC__ASSERT(0);
439 return false;
440 }
441}
442
443#if 0
444@@@ If we decide to finish implementing this, put this comment back in metadata.h
445/*
446 * The 'tempfile_path_prefix' allows you to specify a directory where
447 * tempfiles should go. Remember that if your metadata edits cause the
448 * FLAC file to grow, the entire file will have to be rewritten. If
449 * 'tempfile_path_prefix' is NULL, the temp file will be written in the
450 * same directory as the original FLAC file. This makes replacing the
451 * original with the tempfile fast but requires extra space in the same
452 * partition for the tempfile. If space is a problem, you can pass a
453 * directory name belonging to a different partition in
454 * 'tempfile_path_prefix'. Note that you should use the forward slash
455 * '/' as the directory separator. A trailing slash is not needed; it
456 * will be added automatically.
457 */
Josh Coalsoncc682512002-06-08 04:53:42 +0000458FLAC__bool FLAC__metadata_simple_iterator_init(FLAC__Metadata_SimpleIterator *iterator, const char *filename, FLAC__bool preserve_file_stats, const char *tempfile_path_prefix);
Josh Coalsoncae58732002-05-04 17:30:32 +0000459#endif
460
Josh Coalson6afed9f2002-10-16 22:29:47 +0000461FLAC_API FLAC__bool FLAC__metadata_simple_iterator_init(FLAC__Metadata_SimpleIterator *iterator, const char *filename, FLAC__bool read_only, FLAC__bool preserve_file_stats)
Josh Coalsoncae58732002-05-04 17:30:32 +0000462{
Josh Coalson72943092002-05-22 05:35:43 +0000463 const char *tempfile_path_prefix = 0; /*@@@ search for comments near 'rename(...)' for what it will take to finish implementing this */
Josh Coalsoncae58732002-05-04 17:30:32 +0000464
465 FLAC__ASSERT(0 != iterator);
466 FLAC__ASSERT(0 != filename);
467
468 simple_iterator_free_guts_(iterator);
469
Josh Coalson3ac66932002-08-30 05:41:31 +0000470 if(!read_only && preserve_file_stats)
Josh Coalsoncae58732002-05-04 17:30:32 +0000471 iterator->has_stats = get_file_stats_(filename, &iterator->stats);
472
473 if(0 == (iterator->filename = strdup(filename))) {
474 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
475 return false;
476 }
477 if(0 != tempfile_path_prefix && 0 == (iterator->tempfile_path_prefix = strdup(tempfile_path_prefix))) {
478 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
479 return false;
480 }
481
Josh Coalson3ac66932002-08-30 05:41:31 +0000482 return simple_iterator_prime_input_(iterator, read_only);
Josh Coalsoncae58732002-05-04 17:30:32 +0000483}
484
Josh Coalson6afed9f2002-10-16 22:29:47 +0000485FLAC_API FLAC__bool FLAC__metadata_simple_iterator_is_writable(const FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000486{
487 FLAC__ASSERT(0 != iterator);
488 FLAC__ASSERT(0 != iterator->file);
489
490 return iterator->is_writable;
491}
492
Josh Coalson6afed9f2002-10-16 22:29:47 +0000493FLAC_API FLAC__bool FLAC__metadata_simple_iterator_next(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000494{
495 FLAC__ASSERT(0 != iterator);
496 FLAC__ASSERT(0 != iterator->file);
497
498 if(iterator->is_last)
499 return false;
500
501 if(0 != fseek(iterator->file, iterator->length, SEEK_CUR)) {
502 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
503 return false;
504 }
505
506 iterator->offset[iterator->depth] = ftell(iterator->file);
507
508 return read_metadata_block_header_(iterator);
509}
510
Josh Coalson6afed9f2002-10-16 22:29:47 +0000511FLAC_API FLAC__bool FLAC__metadata_simple_iterator_prev(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000512{
513 long this_offset;
514
515 FLAC__ASSERT(0 != iterator);
516 FLAC__ASSERT(0 != iterator->file);
517
518 if(iterator->offset[iterator->depth] == iterator->first_offset)
519 return false;
520
521 if(0 != fseek(iterator->file, iterator->first_offset, SEEK_SET)) {
522 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
523 return false;
524 }
525 this_offset = iterator->first_offset;
526 if(!read_metadata_block_header_(iterator))
527 return false;
528
529 /* we ignore any error from ftell() and catch it in fseek() */
530 while(ftell(iterator->file) + (long)iterator->length < iterator->offset[iterator->depth]) {
531 if(0 != fseek(iterator->file, iterator->length, SEEK_CUR)) {
532 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
533 return false;
534 }
535 this_offset = ftell(iterator->file);
536 if(!read_metadata_block_header_(iterator))
537 return false;
538 }
539
540 iterator->offset[iterator->depth] = this_offset;
541
542 return true;
543}
544
Josh Coalson6afed9f2002-10-16 22:29:47 +0000545FLAC_API FLAC__MetadataType FLAC__metadata_simple_iterator_get_block_type(const FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000546{
547 FLAC__ASSERT(0 != iterator);
548 FLAC__ASSERT(0 != iterator->file);
549
550 return iterator->type;
551}
552
Josh Coalson6afed9f2002-10-16 22:29:47 +0000553FLAC_API FLAC__StreamMetadata *FLAC__metadata_simple_iterator_get_block(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +0000554{
Josh Coalsoncc682512002-06-08 04:53:42 +0000555 FLAC__StreamMetadata *block = FLAC__metadata_object_new(iterator->type);
Josh Coalsoncae58732002-05-04 17:30:32 +0000556
557 FLAC__ASSERT(0 != iterator);
558 FLAC__ASSERT(0 != iterator->file);
559
560 if(0 != block) {
561 block->is_last = iterator->is_last;
562 block->length = iterator->length;
563
564 if(!read_metadata_block_data_(iterator, block)) {
565 FLAC__metadata_object_delete(block);
566 return 0;
567 }
568
569 /* back up to the beginning of the block data to stay consistent */
Josh Coalsoneb643a22002-05-20 05:58:50 +0000570 if(0 != fseek(iterator->file, iterator->offset[iterator->depth] + FLAC__STREAM_METADATA_HEADER_LENGTH, SEEK_SET)) {
Josh Coalsoncae58732002-05-04 17:30:32 +0000571 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
572 FLAC__metadata_object_delete(block);
573 return 0;
574 }
575 }
576 else
577 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
578
579 return block;
580}
581
Josh Coalson6afed9f2002-10-16 22:29:47 +0000582FLAC_API FLAC__bool FLAC__metadata_simple_iterator_set_block(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block, FLAC__bool use_padding)
Josh Coalsoncae58732002-05-04 17:30:32 +0000583{
Josh Coalson51ee4652002-05-09 05:46:23 +0000584 FLAC__ASSERT_DECLARATION(long debug_target_offset = iterator->offset[iterator->depth];)
585 FLAC__bool ret;
586
Josh Coalsoncae58732002-05-04 17:30:32 +0000587 FLAC__ASSERT(0 != iterator);
588 FLAC__ASSERT(0 != iterator->file);
Josh Coalsond8ab3462002-07-11 05:54:05 +0000589 FLAC__ASSERT(0 != block);
Josh Coalsoncae58732002-05-04 17:30:32 +0000590
591 if(!iterator->is_writable) {
592 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_WRITABLE;
593 return false;
594 }
595
596 if(iterator->type == FLAC__METADATA_TYPE_STREAMINFO || block->type == FLAC__METADATA_TYPE_STREAMINFO) {
597 if(iterator->type != block->type) {
598 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT;
599 return false;
600 }
601 }
602
603 block->is_last = iterator->is_last;
Josh Coalsoncae58732002-05-04 17:30:32 +0000604
605 if(iterator->length == block->length)
606 return write_metadata_block_stationary_(iterator, block);
607 else if(iterator->length > block->length) {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000608 if(use_padding && iterator->length >= FLAC__STREAM_METADATA_HEADER_LENGTH + block->length) {
Josh Coalsonccce9ed2003-01-13 02:06:40 +0000609 ret = write_metadata_block_stationary_with_padding_(iterator, block, iterator->length - FLAC__STREAM_METADATA_HEADER_LENGTH - block->length, block->is_last);
610 FLAC__ASSERT(!ret || iterator->offset[iterator->depth] == debug_target_offset);
611 FLAC__ASSERT(!ret || ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000612 return ret;
613 }
614 else {
615 ret = rewrite_whole_file_(iterator, block, /*append=*/false);
Josh Coalsonccce9ed2003-01-13 02:06:40 +0000616 FLAC__ASSERT(!ret || iterator->offset[iterator->depth] == debug_target_offset);
617 FLAC__ASSERT(!ret || ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000618 return ret;
619 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000620 }
621 else /* iterator->length < block->length */ {
622 unsigned padding_leftover = 0;
623 FLAC__bool padding_is_last = false;
624 if(use_padding) {
625 /* first see if we can even use padding */
626 if(iterator->is_last) {
627 use_padding = false;
628 }
629 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000630 const unsigned extra_padding_bytes_required = block->length - iterator->length;
Josh Coalsoncae58732002-05-04 17:30:32 +0000631 simple_iterator_push_(iterator);
632 if(!FLAC__metadata_simple_iterator_next(iterator)) {
633 (void)simple_iterator_pop_(iterator);
634 return false;
635 }
636 if(iterator->type != FLAC__METADATA_TYPE_PADDING) {
637 use_padding = false;
638 }
639 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000640 if(FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length == extra_padding_bytes_required) {
Josh Coalsoncae58732002-05-04 17:30:32 +0000641 padding_leftover = 0;
642 block->is_last = iterator->is_last;
643 }
Josh Coalson51ee4652002-05-09 05:46:23 +0000644 else if(iterator->length < extra_padding_bytes_required)
Josh Coalsoncae58732002-05-04 17:30:32 +0000645 use_padding = false;
646 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000647 padding_leftover = FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length - extra_padding_bytes_required;
Josh Coalsoncae58732002-05-04 17:30:32 +0000648 padding_is_last = iterator->is_last;
649 block->is_last = false;
650 }
651 }
652 if(!simple_iterator_pop_(iterator))
653 return false;
654 }
655 }
656 if(use_padding) {
Josh Coalson51ee4652002-05-09 05:46:23 +0000657 if(padding_leftover == 0) {
658 ret = write_metadata_block_stationary_(iterator, block);
Josh Coalsonccce9ed2003-01-13 02:06:40 +0000659 FLAC__ASSERT(!ret || iterator->offset[iterator->depth] == debug_target_offset);
660 FLAC__ASSERT(!ret || ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000661 return ret;
662 }
663 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000664 FLAC__ASSERT(padding_leftover >= FLAC__STREAM_METADATA_HEADER_LENGTH);
665 ret = write_metadata_block_stationary_with_padding_(iterator, block, padding_leftover - FLAC__STREAM_METADATA_HEADER_LENGTH, padding_is_last);
Josh Coalsonccce9ed2003-01-13 02:06:40 +0000666 FLAC__ASSERT(!ret || iterator->offset[iterator->depth] == debug_target_offset);
667 FLAC__ASSERT(!ret || ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000668 return ret;
669 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000670 }
671 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000672 ret = rewrite_whole_file_(iterator, block, /*append=*/false);
Josh Coalsonccce9ed2003-01-13 02:06:40 +0000673 FLAC__ASSERT(!ret || iterator->offset[iterator->depth] == debug_target_offset);
674 FLAC__ASSERT(!ret || ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000675 return ret;
Josh Coalsoncae58732002-05-04 17:30:32 +0000676 }
677 }
678}
679
Josh Coalson6afed9f2002-10-16 22:29:47 +0000680FLAC_API FLAC__bool FLAC__metadata_simple_iterator_insert_block_after(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block, FLAC__bool use_padding)
Josh Coalsoncae58732002-05-04 17:30:32 +0000681{
682 unsigned padding_leftover = 0;
683 FLAC__bool padding_is_last = false;
684
Josh Coalsoneb643a22002-05-20 05:58:50 +0000685 FLAC__ASSERT_DECLARATION(long debug_target_offset = iterator->offset[iterator->depth] + FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length;)
Josh Coalson51ee4652002-05-09 05:46:23 +0000686 FLAC__bool ret;
687
Josh Coalsoncae58732002-05-04 17:30:32 +0000688 FLAC__ASSERT(0 != iterator);
689 FLAC__ASSERT(0 != iterator->file);
Josh Coalsond8ab3462002-07-11 05:54:05 +0000690 FLAC__ASSERT(0 != block);
Josh Coalsoncae58732002-05-04 17:30:32 +0000691
692 if(!iterator->is_writable)
693 return false;
694
695 if(block->type == FLAC__METADATA_TYPE_STREAMINFO) {
696 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT;
697 return false;
698 }
699
Josh Coalson6b02a752002-05-10 06:42:02 +0000700 block->is_last = iterator->is_last;
Josh Coalson51ee4652002-05-09 05:46:23 +0000701
Josh Coalsoncae58732002-05-04 17:30:32 +0000702 if(use_padding) {
703 /* first see if we can even use padding */
704 if(iterator->is_last) {
705 use_padding = false;
706 }
707 else {
708 simple_iterator_push_(iterator);
709 if(!FLAC__metadata_simple_iterator_next(iterator)) {
710 (void)simple_iterator_pop_(iterator);
711 return false;
712 }
713 if(iterator->type != FLAC__METADATA_TYPE_PADDING) {
714 use_padding = false;
715 }
716 else {
Josh Coalsoncae58732002-05-04 17:30:32 +0000717 if(iterator->length == block->length) {
718 padding_leftover = 0;
719 block->is_last = iterator->is_last;
720 }
Josh Coalsoneb643a22002-05-20 05:58:50 +0000721 else if(iterator->length < FLAC__STREAM_METADATA_HEADER_LENGTH + block->length)
Josh Coalsoncae58732002-05-04 17:30:32 +0000722 use_padding = false;
723 else {
Josh Coalson6b02a752002-05-10 06:42:02 +0000724 padding_leftover = iterator->length - block->length;
Josh Coalsoncae58732002-05-04 17:30:32 +0000725 padding_is_last = iterator->is_last;
726 block->is_last = false;
727 }
728 }
729 if(!simple_iterator_pop_(iterator))
730 return false;
731 }
732 }
733 if(use_padding) {
734 /* move to the next block, which is suitable padding */
735 if(!FLAC__metadata_simple_iterator_next(iterator))
736 return false;
Josh Coalson51ee4652002-05-09 05:46:23 +0000737 if(padding_leftover == 0) {
738 ret = write_metadata_block_stationary_(iterator, block);
Josh Coalson51ee4652002-05-09 05:46:23 +0000739 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalson72943092002-05-22 05:35:43 +0000740 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000741 return ret;
742 }
743 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000744 FLAC__ASSERT(padding_leftover >= FLAC__STREAM_METADATA_HEADER_LENGTH);
745 ret = write_metadata_block_stationary_with_padding_(iterator, block, padding_leftover - FLAC__STREAM_METADATA_HEADER_LENGTH, padding_is_last);
Josh Coalson51ee4652002-05-09 05:46:23 +0000746 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalson72943092002-05-22 05:35:43 +0000747 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000748 return ret;
749 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000750 }
751 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000752 ret = rewrite_whole_file_(iterator, block, /*append=*/true);
Josh Coalson51ee4652002-05-09 05:46:23 +0000753 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalson72943092002-05-22 05:35:43 +0000754 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + (long)FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000755 return ret;
Josh Coalsoncae58732002-05-04 17:30:32 +0000756 }
757}
758
Josh Coalson6afed9f2002-10-16 22:29:47 +0000759FLAC_API FLAC__bool FLAC__metadata_simple_iterator_delete_block(FLAC__Metadata_SimpleIterator *iterator, FLAC__bool use_padding)
Josh Coalsoncae58732002-05-04 17:30:32 +0000760{
Josh Coalson51ee4652002-05-09 05:46:23 +0000761 FLAC__ASSERT_DECLARATION(long debug_target_offset = iterator->offset[iterator->depth];)
762 FLAC__bool ret;
763
Josh Coalsoncae58732002-05-04 17:30:32 +0000764 if(iterator->type == FLAC__METADATA_TYPE_STREAMINFO) {
765 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT;
766 return false;
767 }
768
769 if(use_padding) {
Josh Coalsoncc682512002-06-08 04:53:42 +0000770 FLAC__StreamMetadata *padding = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING);
Josh Coalsoncae58732002-05-04 17:30:32 +0000771 if(0 == padding) {
772 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
773 return false;
774 }
775 padding->length = iterator->length;
776 if(!FLAC__metadata_simple_iterator_set_block(iterator, padding, false)) {
777 FLAC__metadata_object_delete(padding);
778 return false;
779 }
780 FLAC__metadata_object_delete(padding);
Josh Coalson6b02a752002-05-10 06:42:02 +0000781 if(!FLAC__metadata_simple_iterator_prev(iterator))
782 return false;
Josh Coalson72943092002-05-22 05:35:43 +0000783 FLAC__ASSERT(iterator->offset[iterator->depth] + (long)FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length == debug_target_offset);
Josh Coalson51ee4652002-05-09 05:46:23 +0000784 FLAC__ASSERT(ftell(iterator->file) + (long)iterator->length == debug_target_offset);
Josh Coalsoncae58732002-05-04 17:30:32 +0000785 return true;
786 }
787 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000788 ret = rewrite_whole_file_(iterator, 0, /*append=*/false);
Josh Coalson72943092002-05-22 05:35:43 +0000789 FLAC__ASSERT(iterator->offset[iterator->depth] + (long)FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length == debug_target_offset);
Josh Coalson51ee4652002-05-09 05:46:23 +0000790 FLAC__ASSERT(ftell(iterator->file) + (long)iterator->length == debug_target_offset);
791 return ret;
Josh Coalsoncae58732002-05-04 17:30:32 +0000792 }
793}
794
795
796
797/****************************************************************************
798 *
799 * Level 2 implementation
800 *
801 ***************************************************************************/
802
803
Josh Coalsoncc682512002-06-08 04:53:42 +0000804typedef struct FLAC__Metadata_Node {
805 FLAC__StreamMetadata *data;
806 struct FLAC__Metadata_Node *prev, *next;
807} FLAC__Metadata_Node;
Josh Coalsoncae58732002-05-04 17:30:32 +0000808
Josh Coalsoncc682512002-06-08 04:53:42 +0000809struct FLAC__Metadata_Chain {
Josh Coalson91748be2004-07-10 22:28:03 +0000810 char *filename; /* will be NULL if using callbacks */
Josh Coalsoncc682512002-06-08 04:53:42 +0000811 FLAC__Metadata_Node *head;
812 FLAC__Metadata_Node *tail;
Josh Coalsoncae58732002-05-04 17:30:32 +0000813 unsigned nodes;
Josh Coalsoncc682512002-06-08 04:53:42 +0000814 FLAC__Metadata_ChainStatus status;
Josh Coalsoncae58732002-05-04 17:30:32 +0000815 long first_offset, last_offset; /*@@@ 2G limit */
816 /*
817 * This is the length of the chain initially read from the FLAC file.
Josh Coalson3ecb1172002-06-15 05:09:47 +0000818 * it is used to compare against the current length to decide whether
Josh Coalsoncae58732002-05-04 17:30:32 +0000819 * or not the whole file has to be rewritten.
820 */
821 unsigned initial_length; /*@@@ 4G limit */
Josh Coalsoncae58732002-05-04 17:30:32 +0000822};
823
Josh Coalsoncc682512002-06-08 04:53:42 +0000824struct FLAC__Metadata_Iterator {
825 FLAC__Metadata_Chain *chain;
826 FLAC__Metadata_Node *current;
Josh Coalsoncae58732002-05-04 17:30:32 +0000827};
828
Josh Coalson6afed9f2002-10-16 22:29:47 +0000829FLAC_API const char * const FLAC__Metadata_ChainStatusString[] = {
Josh Coalsoncae58732002-05-04 17:30:32 +0000830 "FLAC__METADATA_CHAIN_STATUS_OK",
831 "FLAC__METADATA_CHAIN_STATUS_ILLEGAL_INPUT",
832 "FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE",
833 "FLAC__METADATA_CHAIN_STATUS_NOT_A_FLAC_FILE",
834 "FLAC__METADATA_CHAIN_STATUS_NOT_WRITABLE",
Josh Coalsona7b673c2002-05-25 02:13:38 +0000835 "FLAC__METADATA_CHAIN_STATUS_BAD_METADATA",
Josh Coalsoncae58732002-05-04 17:30:32 +0000836 "FLAC__METADATA_CHAIN_STATUS_READ_ERROR",
837 "FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR",
838 "FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR",
839 "FLAC__METADATA_CHAIN_STATUS_RENAME_ERROR",
840 "FLAC__METADATA_CHAIN_STATUS_UNLINK_ERROR",
841 "FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR",
Josh Coalson91748be2004-07-10 22:28:03 +0000842 "FLAC__METADATA_CHAIN_STATUS_INTERNAL_ERROR",
Josh Coalson463a22c2004-07-11 06:41:00 +0000843 "FLAC__METADATA_CHAIN_STATUS_INVALID_CALLBACKS",
844 "FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH",
845 "FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL"
Josh Coalsoncae58732002-05-04 17:30:32 +0000846};
847
848
Josh Coalsoncc682512002-06-08 04:53:42 +0000849static FLAC__Metadata_Node *node_new_()
Josh Coalsoncae58732002-05-04 17:30:32 +0000850{
Josh Coalsonea7155f2002-10-18 05:49:19 +0000851 return (FLAC__Metadata_Node*)calloc(1, sizeof(FLAC__Metadata_Node));
Josh Coalsoncae58732002-05-04 17:30:32 +0000852}
853
Josh Coalsoncc682512002-06-08 04:53:42 +0000854static void node_delete_(FLAC__Metadata_Node *node)
Josh Coalsoncae58732002-05-04 17:30:32 +0000855{
856 FLAC__ASSERT(0 != node);
857 if(0 != node->data)
858 FLAC__metadata_object_delete(node->data);
859 free(node);
860}
861
Josh Coalson91748be2004-07-10 22:28:03 +0000862static void chain_init_(FLAC__Metadata_Chain *chain)
863{
864 FLAC__ASSERT(0 != chain);
865
866 chain->filename = 0;
867 chain->head = chain->tail = 0;
868 chain->nodes = 0;
869 chain->status = FLAC__METADATA_CHAIN_STATUS_OK;
870 chain->initial_length = 0;
871}
872
873static void chain_clear_(FLAC__Metadata_Chain *chain)
874{
875 FLAC__Metadata_Node *node, *next;
876
877 FLAC__ASSERT(0 != chain);
878
879 for(node = chain->head; node; ) {
880 next = node->next;
881 node_delete_(node);
882 node = next;
883 }
884
885 if(0 != chain->filename)
886 free(chain->filename);
887
888 chain_init_(chain);
889}
890
Josh Coalsoncc682512002-06-08 04:53:42 +0000891static void chain_append_node_(FLAC__Metadata_Chain *chain, FLAC__Metadata_Node *node)
Josh Coalsoncae58732002-05-04 17:30:32 +0000892{
893 FLAC__ASSERT(0 != chain);
894 FLAC__ASSERT(0 != node);
Josh Coalson6b02a752002-05-10 06:42:02 +0000895 FLAC__ASSERT(0 != node->data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000896
Josh Coalson6b02a752002-05-10 06:42:02 +0000897 node->next = node->prev = 0;
Josh Coalsoncae58732002-05-04 17:30:32 +0000898 node->data->is_last = true;
899 if(0 != chain->tail)
900 chain->tail->data->is_last = false;
901
902 if(0 == chain->head)
903 chain->head = node;
904 else {
Josh Coalson6b02a752002-05-10 06:42:02 +0000905 FLAC__ASSERT(0 != chain->tail);
Josh Coalsoncae58732002-05-04 17:30:32 +0000906 chain->tail->next = node;
907 node->prev = chain->tail;
908 }
909 chain->tail = node;
910 chain->nodes++;
Josh Coalsoncae58732002-05-04 17:30:32 +0000911}
912
Josh Coalsoncc682512002-06-08 04:53:42 +0000913static void chain_remove_node_(FLAC__Metadata_Chain *chain, FLAC__Metadata_Node *node)
Josh Coalsoncae58732002-05-04 17:30:32 +0000914{
915 FLAC__ASSERT(0 != chain);
916 FLAC__ASSERT(0 != node);
917
918 if(node == chain->head)
919 chain->head = node->next;
920 else
921 node->prev->next = node->next;
922
923 if(node == chain->tail)
924 chain->tail = node->prev;
925 else
926 node->next->prev = node->prev;
927
Josh Coalson6b02a752002-05-10 06:42:02 +0000928 if(0 != chain->tail)
929 chain->tail->data->is_last = true;
930
Josh Coalsoncae58732002-05-04 17:30:32 +0000931 chain->nodes--;
Josh Coalsoncae58732002-05-04 17:30:32 +0000932}
933
Josh Coalsoncc682512002-06-08 04:53:42 +0000934static void chain_delete_node_(FLAC__Metadata_Chain *chain, FLAC__Metadata_Node *node)
Josh Coalsoncae58732002-05-04 17:30:32 +0000935{
936 chain_remove_node_(chain, node);
937 node_delete_(node);
938}
939
Josh Coalson3ecb1172002-06-15 05:09:47 +0000940static unsigned chain_calculate_length_(FLAC__Metadata_Chain *chain)
941{
942 const FLAC__Metadata_Node *node;
943 unsigned length = 0;
944 for(node = chain->head; node; node = node->next)
945 length += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
946 return length;
947}
948
Josh Coalsoncc682512002-06-08 04:53:42 +0000949static void iterator_insert_node_(FLAC__Metadata_Iterator *iterator, FLAC__Metadata_Node *node)
Josh Coalsoncae58732002-05-04 17:30:32 +0000950{
951 FLAC__ASSERT(0 != node);
Josh Coalson6b02a752002-05-10 06:42:02 +0000952 FLAC__ASSERT(0 != node->data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000953 FLAC__ASSERT(0 != iterator);
954 FLAC__ASSERT(0 != iterator->current);
955 FLAC__ASSERT(0 != iterator->chain);
956 FLAC__ASSERT(0 != iterator->chain->head);
957 FLAC__ASSERT(0 != iterator->chain->tail);
958
Josh Coalson6b02a752002-05-10 06:42:02 +0000959 node->data->is_last = false;
Josh Coalson2745fe72002-05-11 05:31:56 +0000960
Josh Coalsoncae58732002-05-04 17:30:32 +0000961 node->prev = iterator->current->prev;
Josh Coalson2745fe72002-05-11 05:31:56 +0000962 node->next = iterator->current;
963
Josh Coalsoncae58732002-05-04 17:30:32 +0000964 if(0 == node->prev)
965 iterator->chain->head = node;
966 else
967 node->prev->next = node;
968
Josh Coalsoncae58732002-05-04 17:30:32 +0000969 iterator->current->prev = node;
970
971 iterator->chain->nodes++;
Josh Coalsoncae58732002-05-04 17:30:32 +0000972}
973
Josh Coalsoncc682512002-06-08 04:53:42 +0000974static void iterator_insert_node_after_(FLAC__Metadata_Iterator *iterator, FLAC__Metadata_Node *node)
Josh Coalsoncae58732002-05-04 17:30:32 +0000975{
976 FLAC__ASSERT(0 != node);
Josh Coalson6b02a752002-05-10 06:42:02 +0000977 FLAC__ASSERT(0 != node->data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000978 FLAC__ASSERT(0 != iterator);
979 FLAC__ASSERT(0 != iterator->current);
980 FLAC__ASSERT(0 != iterator->chain);
981 FLAC__ASSERT(0 != iterator->chain->head);
982 FLAC__ASSERT(0 != iterator->chain->tail);
983
984 iterator->current->data->is_last = false;
985
986 node->prev = iterator->current;
Josh Coalsoncae58732002-05-04 17:30:32 +0000987 node->next = iterator->current->next;
Josh Coalson2745fe72002-05-11 05:31:56 +0000988
Josh Coalsoncae58732002-05-04 17:30:32 +0000989 if(0 == node->next)
990 iterator->chain->tail = node;
991 else
992 node->next->prev = node;
993
Josh Coalson2745fe72002-05-11 05:31:56 +0000994 node->prev->next = node;
995
Josh Coalsoncae58732002-05-04 17:30:32 +0000996 iterator->chain->tail->data->is_last = true;
997
998 iterator->chain->nodes++;
Josh Coalsoncae58732002-05-04 17:30:32 +0000999}
1000
Josh Coalson2745fe72002-05-11 05:31:56 +00001001/* return true iff node and node->next are both padding */
Josh Coalsoncc682512002-06-08 04:53:42 +00001002static FLAC__bool chain_merge_adjacent_padding_(FLAC__Metadata_Chain *chain, FLAC__Metadata_Node *node)
Josh Coalson2745fe72002-05-11 05:31:56 +00001003{
1004 if(node->data->type == FLAC__METADATA_TYPE_PADDING && 0 != node->next && node->next->data->type == FLAC__METADATA_TYPE_PADDING) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00001005 const unsigned growth = FLAC__STREAM_METADATA_HEADER_LENGTH + node->next->data->length;
Josh Coalson2745fe72002-05-11 05:31:56 +00001006 node->data->length += growth;
Josh Coalson2745fe72002-05-11 05:31:56 +00001007
1008 chain_delete_node_(chain, node->next);
1009 return true;
1010 }
1011 else
1012 return false;
1013}
1014
Josh Coalson463a22c2004-07-11 06:41:00 +00001015/* Returns the new length of the chain, or 0 if there was an error. */
1016/* WATCHOUT: This can get called multiple times before a write, so
1017 * it should still work when this happens.
1018 */
1019/* WATCHOUT: Make sure to also update the logic in
1020 * FLAC__metadata_chain_check_if_tempfile_needed() if the logic here changes.
1021 */
1022static unsigned chain_prepare_for_write_(FLAC__Metadata_Chain *chain, FLAC__bool use_padding)
1023{
1024 unsigned current_length = chain_calculate_length_(chain);
1025
1026 if(use_padding) {
1027 /* if the metadata shrank and the last block is padding, we just extend the last padding block */
1028 if(current_length < chain->initial_length && chain->tail->data->type == FLAC__METADATA_TYPE_PADDING) {
1029 const unsigned delta = chain->initial_length - current_length;
1030 chain->tail->data->length += delta;
1031 current_length += delta;
1032 FLAC__ASSERT(current_length == chain->initial_length);
1033 }
1034 /* if the metadata shrank more than 4 bytes then there's room to add another padding block */
1035 else if(current_length + FLAC__STREAM_METADATA_HEADER_LENGTH <= chain->initial_length) {
1036 FLAC__StreamMetadata *padding;
1037 FLAC__Metadata_Node *node;
1038 if(0 == (padding = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING))) {
1039 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
1040 return 0;
1041 }
1042 padding->length = chain->initial_length - (FLAC__STREAM_METADATA_HEADER_LENGTH + current_length);
1043 if(0 == (node = node_new_())) {
1044 FLAC__metadata_object_delete(padding);
1045 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
1046 return 0;
1047 }
1048 node->data = padding;
1049 chain_append_node_(chain, node);
1050 current_length = chain_calculate_length_(chain);
1051 FLAC__ASSERT(current_length == chain->initial_length);
1052 }
1053 /* if the metadata grew but the last block is padding, try cutting the padding to restore the original length so we don't have to rewrite the whole file */
1054 else if(current_length > chain->initial_length) {
1055 const unsigned delta = current_length - chain->initial_length;
1056 if(chain->tail->data->type == FLAC__METADATA_TYPE_PADDING) {
1057 /* if the delta is exactly the size of the last padding block, remove the padding block */
1058 if(chain->tail->data->length + FLAC__STREAM_METADATA_HEADER_LENGTH == delta) {
1059 chain_delete_node_(chain, chain->tail);
1060 current_length = chain_calculate_length_(chain);
1061 FLAC__ASSERT(current_length == chain->initial_length);
1062 }
1063 /* if there is at least 'delta' bytes of padding, trim the padding down */
1064 else if(chain->tail->data->length >= delta) {
1065 chain->tail->data->length -= delta;
1066 current_length -= delta;
1067 FLAC__ASSERT(current_length == chain->initial_length);
1068 }
1069 }
1070 }
1071 }
1072
1073 return current_length;
1074}
1075
Josh Coalson0e46bf52004-07-15 00:04:46 +00001076static FLAC__bool chain_read_cb_(FLAC__Metadata_Chain *chain, FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Seek seek_cb, FLAC__IOCallback_Tell tell_cb)
Josh Coalsonea25cb32004-07-10 00:39:01 +00001077{
1078 FLAC__Metadata_Node *node;
1079
1080 FLAC__ASSERT(0 != chain);
1081
1082 /* we assume we're already at the beginning of the file */
1083
1084 switch(seek_to_first_metadata_block_cb_(handle, read_cb, seek_cb)) {
1085 case 0:
1086 break;
1087 case 1:
1088 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_ERROR;
1089 return false;
1090 case 2:
1091 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
1092 return false;
1093 case 3:
1094 chain->status = FLAC__METADATA_CHAIN_STATUS_NOT_A_FLAC_FILE;
1095 return false;
1096 default:
1097 FLAC__ASSERT(0);
1098 return false;
1099 }
1100
1101 {
1102 FLAC__int64 pos = tell_cb(handle);
1103 if(pos < 0) {
1104 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_ERROR;
1105 return false;
1106 }
1107 chain->first_offset = (long)pos;
1108 }
1109
1110 {
1111 FLAC__bool is_last;
1112 FLAC__MetadataType type;
1113 unsigned length;
1114
1115 do {
1116 node = node_new_();
1117 if(0 == node) {
1118 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
1119 return false;
1120 }
1121
1122 if(!read_metadata_block_header_cb_(handle, read_cb, &is_last, &type, &length)) {
1123 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_ERROR;
1124 return false;
1125 }
1126
1127 node->data = FLAC__metadata_object_new(type);
1128 if(0 == node->data) {
1129 node_delete_(node);
1130 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
1131 return false;
1132 }
1133
1134 node->data->is_last = is_last;
1135 node->data->length = length;
1136
1137 chain->status = get_equivalent_status_(read_metadata_block_data_cb_(handle, read_cb, seek_cb, node->data));
1138 if(chain->status != FLAC__METADATA_CHAIN_STATUS_OK) {
1139 node_delete_(node);
1140 return false;
1141 }
1142 chain_append_node_(chain, node);
1143 } while(!is_last);
1144 }
1145
1146 {
1147 FLAC__int64 pos = tell_cb(handle);
1148 if(pos < 0) {
1149 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_ERROR;
1150 return false;
1151 }
1152 chain->last_offset = (long)pos;
1153 }
1154
Josh Coalsonea25cb32004-07-10 00:39:01 +00001155 chain->initial_length = chain_calculate_length_(chain);
1156
1157 return true;
1158}
1159
Josh Coalson0e46bf52004-07-15 00:04:46 +00001160static FLAC__bool chain_rewrite_metadata_in_place_cb_(FLAC__Metadata_Chain *chain, FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, FLAC__IOCallback_Seek seek_cb)
Josh Coalson463a22c2004-07-11 06:41:00 +00001161{
1162 FLAC__Metadata_Node *node;
1163
1164 FLAC__ASSERT(0 != chain);
1165 FLAC__ASSERT(0 != chain->head);
1166
1167 if(0 != seek_cb(handle, chain->first_offset, SEEK_SET)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001168 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
1169 return false;
1170 }
1171
1172 for(node = chain->head; node; node = node->next) {
1173 if(!write_metadata_block_header_cb_(handle, write_cb, node->data)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001174 chain->status = FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR;
1175 return false;
1176 }
1177 if(!write_metadata_block_data_cb_(handle, write_cb, node->data)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001178 chain->status = FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR;
1179 return false;
1180 }
1181 }
1182
1183 /*FLAC__ASSERT(fflush(), ftell() == chain->last_offset);*/
1184
Josh Coalson463a22c2004-07-11 06:41:00 +00001185 chain->status = FLAC__METADATA_CHAIN_STATUS_OK;
1186 return true;
1187}
1188
1189static FLAC__bool chain_rewrite_metadata_in_place_(FLAC__Metadata_Chain *chain)
1190{
1191 FILE *file;
Josh Coalson0e46bf52004-07-15 00:04:46 +00001192 FLAC__bool ret;
Josh Coalson463a22c2004-07-11 06:41:00 +00001193
1194 FLAC__ASSERT(0 != chain->filename);
1195
1196 if(0 == (file = fopen(chain->filename, "r+b"))) {
1197 chain->status = FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
1198 return false;
1199 }
1200
Josh Coalson0e46bf52004-07-15 00:04:46 +00001201 /* chain_rewrite_metadata_in_place_cb_() sets chain->status for us */
1202 ret = chain_rewrite_metadata_in_place_cb_(chain, (FLAC__IOHandle)file, (FLAC__IOCallback_Write)fwrite, fseek_wrapper_);
1203
1204 fclose(file);
1205
1206 return ret;
Josh Coalson463a22c2004-07-11 06:41:00 +00001207}
1208
1209static FLAC__bool chain_rewrite_file_(FLAC__Metadata_Chain *chain, const char *tempfile_path_prefix)
1210{
1211 FILE *f, *tempfile;
1212 char *tempfilename;
1213 FLAC__Metadata_SimpleIteratorStatus status;
1214 const FLAC__Metadata_Node *node;
1215
1216 FLAC__ASSERT(0 != chain);
1217 FLAC__ASSERT(0 != chain->filename);
1218 FLAC__ASSERT(0 != chain->head);
1219
1220 /* copy the file prefix (data up to first metadata block */
1221 if(0 == (f = fopen(chain->filename, "rb"))) {
1222 chain->status = FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
1223 return false;
1224 }
1225 if(!open_tempfile_(chain->filename, tempfile_path_prefix, &tempfile, &tempfilename, &status)) {
1226 chain->status = get_equivalent_status_(status);
1227 cleanup_tempfile_(&tempfile, &tempfilename);
1228 return false;
1229 }
1230 if(!copy_n_bytes_from_file_(f, tempfile, chain->first_offset, &status)) {
1231 chain->status = get_equivalent_status_(status);
1232 cleanup_tempfile_(&tempfile, &tempfilename);
1233 return false;
1234 }
1235
1236 /* write the metadata */
1237 for(node = chain->head; node; node = node->next) {
1238 if(!write_metadata_block_header_(tempfile, &status, node->data)) {
1239 chain->status = get_equivalent_status_(status);
1240 return false;
1241 }
1242 if(!write_metadata_block_data_(tempfile, &status, node->data)) {
1243 chain->status = get_equivalent_status_(status);
1244 return false;
1245 }
1246 }
1247 /*FLAC__ASSERT(fflush(), ftell() == chain->last_offset);*/
1248
1249 /* copy the file postfix (everything after the metadata) */
1250 if(0 != fseek(f, chain->last_offset, SEEK_SET)) {
1251 cleanup_tempfile_(&tempfile, &tempfilename);
1252 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
1253 return false;
1254 }
1255 if(!copy_remaining_bytes_from_file_(f, tempfile, &status)) {
1256 cleanup_tempfile_(&tempfile, &tempfilename);
1257 chain->status = get_equivalent_status_(status);
1258 return false;
1259 }
1260
1261 /* move the tempfile on top of the original */
1262 (void)fclose(f);
1263 if(!transport_tempfile_(chain->filename, &tempfile, &tempfilename, &status))
1264 return false;
1265
1266 return true;
1267}
1268
Josh Coalson9aac6702004-09-08 00:50:21 +00001269/* assumes 'handle' is already at beginning of file */
Josh Coalson0e46bf52004-07-15 00:04:46 +00001270static FLAC__bool chain_rewrite_file_cb_(FLAC__Metadata_Chain *chain, FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Seek seek_cb, FLAC__IOCallback_Eof eof_cb, FLAC__IOHandle temp_handle, FLAC__IOCallback_Write temp_write_cb)
Josh Coalson463a22c2004-07-11 06:41:00 +00001271{
1272 FLAC__Metadata_SimpleIteratorStatus status;
1273 const FLAC__Metadata_Node *node;
1274
1275 FLAC__ASSERT(0 != chain);
1276 FLAC__ASSERT(0 == chain->filename);
1277 FLAC__ASSERT(0 != chain->head);
1278
1279 /* copy the file prefix (data up to first metadata block */
1280 if(!copy_n_bytes_from_file_cb_(handle, read_cb, temp_handle, temp_write_cb, chain->first_offset, &status)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001281 chain->status = get_equivalent_status_(status);
1282 return false;
1283 }
1284
1285 /* write the metadata */
1286 for(node = chain->head; node; node = node->next) {
1287 if(!write_metadata_block_header_cb_(temp_handle, temp_write_cb, node->data)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001288 chain->status = FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR;
1289 return false;
1290 }
1291 if(!write_metadata_block_data_cb_(temp_handle, temp_write_cb, node->data)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001292 chain->status = FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR;
1293 return false;
1294 }
1295 }
1296 /*FLAC__ASSERT(fflush(), ftell() == chain->last_offset);*/
1297
1298 /* copy the file postfix (everything after the metadata) */
1299 if(0 != seek_cb(handle, chain->last_offset, SEEK_SET)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001300 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
1301 return false;
1302 }
1303 if(!copy_remaining_bytes_from_file_cb_(handle, read_cb, eof_cb, temp_handle, temp_write_cb, &status)) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001304 chain->status = get_equivalent_status_(status);
1305 return false;
1306 }
1307
Josh Coalson463a22c2004-07-11 06:41:00 +00001308 return true;
1309}
1310
Josh Coalson6afed9f2002-10-16 22:29:47 +00001311FLAC_API FLAC__Metadata_Chain *FLAC__metadata_chain_new()
Josh Coalsoncae58732002-05-04 17:30:32 +00001312{
Josh Coalson5e3fcb22002-11-06 07:11:26 +00001313 FLAC__Metadata_Chain *chain = (FLAC__Metadata_Chain*)calloc(1, sizeof(FLAC__Metadata_Chain));
Josh Coalsoncae58732002-05-04 17:30:32 +00001314
Josh Coalson91748be2004-07-10 22:28:03 +00001315 if(0 != chain)
1316 chain_init_(chain);
Josh Coalsoncae58732002-05-04 17:30:32 +00001317
1318 return chain;
1319}
1320
Josh Coalson6afed9f2002-10-16 22:29:47 +00001321FLAC_API void FLAC__metadata_chain_delete(FLAC__Metadata_Chain *chain)
Josh Coalsoncae58732002-05-04 17:30:32 +00001322{
Josh Coalsoncae58732002-05-04 17:30:32 +00001323 FLAC__ASSERT(0 != chain);
1324
Josh Coalson91748be2004-07-10 22:28:03 +00001325 chain_clear_(chain);
Josh Coalsoncae58732002-05-04 17:30:32 +00001326
1327 free(chain);
1328}
1329
Josh Coalson6afed9f2002-10-16 22:29:47 +00001330FLAC_API FLAC__Metadata_ChainStatus FLAC__metadata_chain_status(FLAC__Metadata_Chain *chain)
Josh Coalsoncae58732002-05-04 17:30:32 +00001331{
Josh Coalsoncc682512002-06-08 04:53:42 +00001332 FLAC__Metadata_ChainStatus status;
Josh Coalsoncae58732002-05-04 17:30:32 +00001333
1334 FLAC__ASSERT(0 != chain);
1335
1336 status = chain->status;
1337 chain->status = FLAC__METADATA_CHAIN_STATUS_OK;
1338 return status;
1339}
1340
Josh Coalson6afed9f2002-10-16 22:29:47 +00001341FLAC_API FLAC__bool FLAC__metadata_chain_read(FLAC__Metadata_Chain *chain, const char *filename)
Josh Coalsoncae58732002-05-04 17:30:32 +00001342{
Josh Coalsonea25cb32004-07-10 00:39:01 +00001343 FILE *file;
Josh Coalson0e46bf52004-07-15 00:04:46 +00001344 FLAC__bool ret;
Josh Coalsoncae58732002-05-04 17:30:32 +00001345
1346 FLAC__ASSERT(0 != chain);
1347 FLAC__ASSERT(0 != filename);
1348
Josh Coalson91748be2004-07-10 22:28:03 +00001349 chain_clear_(chain);
1350
Josh Coalsoncae58732002-05-04 17:30:32 +00001351 if(0 == (chain->filename = strdup(filename))) {
1352 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
1353 return false;
1354 }
1355
Josh Coalsonea25cb32004-07-10 00:39:01 +00001356 if(0 == (file = fopen(filename, "rb"))) {
1357 chain->status = FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
Josh Coalsoncae58732002-05-04 17:30:32 +00001358 return false;
1359 }
1360
Josh Coalson0e46bf52004-07-15 00:04:46 +00001361 /* chain_read_cb_() sets chain->status for us */
1362 ret = chain_read_cb_(chain, file, (FLAC__IOCallback_Read)fread, fseek_wrapper_, ftell_wrapper_);
Josh Coalsoncae58732002-05-04 17:30:32 +00001363
Josh Coalson0e46bf52004-07-15 00:04:46 +00001364 fclose(file);
Josh Coalson91748be2004-07-10 22:28:03 +00001365
Josh Coalson0e46bf52004-07-15 00:04:46 +00001366 return ret;
Josh Coalson91748be2004-07-10 22:28:03 +00001367}
1368
1369FLAC_API FLAC__bool FLAC__metadata_chain_read_with_callbacks(FLAC__Metadata_Chain *chain, FLAC__IOHandle handle, FLAC__IOCallbacks callbacks)
1370{
Josh Coalson91748be2004-07-10 22:28:03 +00001371 FLAC__ASSERT(0 != chain);
1372
1373 chain_clear_(chain);
1374
Josh Coalson0e46bf52004-07-15 00:04:46 +00001375 if (0 == callbacks.read || 0 == callbacks.seek || 0 == callbacks.tell) {
Josh Coalson91748be2004-07-10 22:28:03 +00001376 chain->status = FLAC__METADATA_CHAIN_STATUS_INVALID_CALLBACKS;
1377 return false;
1378 }
1379
Josh Coalson9aac6702004-09-08 00:50:21 +00001380 /* rewind */
1381 if(0 != callbacks.seek(handle, 0, SEEK_SET)) {
1382 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
1383 return false;
1384 }
1385
Josh Coalson0e46bf52004-07-15 00:04:46 +00001386 if(!chain_read_cb_(chain, handle, callbacks.read, callbacks.seek, callbacks.tell))
Josh Coalson91748be2004-07-10 22:28:03 +00001387 return false; /* chain->status is already set by chain_read_cb_ */
1388
Josh Coalsoncae58732002-05-04 17:30:32 +00001389 return true;
1390}
1391
Josh Coalson463a22c2004-07-11 06:41:00 +00001392FLAC_API FLAC__bool FLAC__metadata_chain_check_if_tempfile_needed(FLAC__Metadata_Chain *chain, FLAC__bool use_padding)
1393{
1394 /* This does all the same checks that are in chain_prepare_for_write_()
1395 * but doesn't actually alter the chain. Make sure to update the logic
1396 * here if chain_prepare_for_write_() changes.
1397 */
1398 const unsigned current_length = chain_calculate_length_(chain);
1399
1400 FLAC__ASSERT(0 != chain);
1401
1402 if(use_padding) {
1403 /* if the metadata shrank and the last block is padding, we just extend the last padding block */
1404 if(current_length < chain->initial_length && chain->tail->data->type == FLAC__METADATA_TYPE_PADDING)
1405 return false;
1406 /* if the metadata shrank more than 4 bytes then there's room to add another padding block */
1407 else if(current_length + FLAC__STREAM_METADATA_HEADER_LENGTH <= chain->initial_length)
1408 return false;
1409 /* if the metadata grew but the last block is padding, try cutting the padding to restore the original length so we don't have to rewrite the whole file */
1410 else if(current_length > chain->initial_length) {
1411 const unsigned delta = current_length - chain->initial_length;
1412 if(chain->tail->data->type == FLAC__METADATA_TYPE_PADDING) {
1413 /* if the delta is exactly the size of the last padding block, remove the padding block */
1414 if(chain->tail->data->length + FLAC__STREAM_METADATA_HEADER_LENGTH == delta)
1415 return false;
1416 /* if there is at least 'delta' bytes of padding, trim the padding down */
1417 else if(chain->tail->data->length >= delta)
1418 return false;
1419 }
1420 }
1421 }
1422
Josh Coalson028d97e2004-07-14 00:49:56 +00001423 return (current_length != chain->initial_length);
Josh Coalson463a22c2004-07-11 06:41:00 +00001424}
1425
Josh Coalson6afed9f2002-10-16 22:29:47 +00001426FLAC_API FLAC__bool FLAC__metadata_chain_write(FLAC__Metadata_Chain *chain, FLAC__bool use_padding, FLAC__bool preserve_file_stats)
Josh Coalsoncae58732002-05-04 17:30:32 +00001427{
1428 struct stat stats;
1429 const char *tempfile_path_prefix = 0;
Josh Coalson3ecb1172002-06-15 05:09:47 +00001430 unsigned current_length;
1431
1432 FLAC__ASSERT(0 != chain);
1433
Josh Coalson463a22c2004-07-11 06:41:00 +00001434 if (0 == chain->filename) {
1435 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH;
1436 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00001437 }
1438
Josh Coalson463a22c2004-07-11 06:41:00 +00001439 current_length = chain_prepare_for_write_(chain, use_padding);
1440
1441 /* a return value of 0 means there was an error; chain->status is already set */
1442 if (0 == current_length)
1443 return false;
1444
Josh Coalsoncae58732002-05-04 17:30:32 +00001445 if(preserve_file_stats)
1446 get_file_stats_(chain->filename, &stats);
1447
Josh Coalson3ecb1172002-06-15 05:09:47 +00001448 if(current_length == chain->initial_length) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001449 if(!chain_rewrite_metadata_in_place_(chain))
Josh Coalsoncae58732002-05-04 17:30:32 +00001450 return false;
1451 }
1452 else {
1453 if(!chain_rewrite_file_(chain, tempfile_path_prefix))
1454 return false;
Josh Coalson463a22c2004-07-11 06:41:00 +00001455
1456 /* recompute lengths and offsets */
1457 {
1458 const FLAC__Metadata_Node *node;
1459 chain->initial_length = current_length;
1460 chain->last_offset = chain->first_offset;
1461 for(node = chain->head; node; node = node->next)
1462 chain->last_offset += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
1463 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001464 }
1465
1466 if(preserve_file_stats)
1467 set_file_stats_(chain->filename, &stats);
1468
Josh Coalson463a22c2004-07-11 06:41:00 +00001469 return true;
1470}
1471
1472FLAC_API FLAC__bool FLAC__metadata_chain_write_with_callbacks(FLAC__Metadata_Chain *chain, FLAC__bool use_padding, FLAC__IOHandle handle, FLAC__IOCallbacks callbacks)
1473{
1474 unsigned current_length;
1475
1476 FLAC__ASSERT(0 != chain);
1477
1478 if (0 != chain->filename) {
1479 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH;
1480 return false;
1481 }
1482
Josh Coalson0e46bf52004-07-15 00:04:46 +00001483 if (0 == callbacks.write || 0 == callbacks.seek) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001484 chain->status = FLAC__METADATA_CHAIN_STATUS_INVALID_CALLBACKS;
1485 return false;
1486 }
1487
1488 if (FLAC__metadata_chain_check_if_tempfile_needed(chain, use_padding)) {
1489 chain->status = FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL;
1490 return false;
1491 }
1492
1493 current_length = chain_prepare_for_write_(chain, use_padding);
1494
1495 /* a return value of 0 means there was an error; chain->status is already set */
1496 if (0 == current_length)
1497 return false;
1498
1499 FLAC__ASSERT(current_length == chain->initial_length);
1500
Josh Coalson0e46bf52004-07-15 00:04:46 +00001501 return chain_rewrite_metadata_in_place_cb_(chain, handle, callbacks.write, callbacks.seek);
Josh Coalson463a22c2004-07-11 06:41:00 +00001502}
1503
1504FLAC_API FLAC__bool FLAC__metadata_chain_write_with_callbacks_and_tempfile(FLAC__Metadata_Chain *chain, FLAC__bool use_padding, FLAC__IOHandle handle, FLAC__IOCallbacks callbacks, FLAC__IOHandle temp_handle, FLAC__IOCallbacks temp_callbacks)
1505{
1506 unsigned current_length;
1507
1508 FLAC__ASSERT(0 != chain);
1509
1510 if (0 != chain->filename) {
1511 chain->status = FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH;
1512 return false;
1513 }
1514
Josh Coalson0e46bf52004-07-15 00:04:46 +00001515 if (0 == callbacks.read || 0 == callbacks.seek || 0 == callbacks.eof) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001516 chain->status = FLAC__METADATA_CHAIN_STATUS_INVALID_CALLBACKS;
1517 return false;
1518 }
Josh Coalson0e46bf52004-07-15 00:04:46 +00001519 if (0 == temp_callbacks.write) {
Josh Coalson463a22c2004-07-11 06:41:00 +00001520 chain->status = FLAC__METADATA_CHAIN_STATUS_INVALID_CALLBACKS;
1521 return false;
1522 }
1523
1524 if (!FLAC__metadata_chain_check_if_tempfile_needed(chain, use_padding)) {
1525 chain->status = FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL;
1526 return false;
1527 }
1528
1529 current_length = chain_prepare_for_write_(chain, use_padding);
1530
1531 /* a return value of 0 means there was an error; chain->status is already set */
1532 if (0 == current_length)
1533 return false;
1534
1535 FLAC__ASSERT(current_length != chain->initial_length);
1536
Josh Coalson9aac6702004-09-08 00:50:21 +00001537 /* rewind */
1538 if(0 != callbacks.seek(handle, 0, SEEK_SET)) {
1539 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
1540 return false;
1541 }
1542
Josh Coalson0e46bf52004-07-15 00:04:46 +00001543 if(!chain_rewrite_file_cb_(chain, handle, callbacks.read, callbacks.seek, callbacks.eof, temp_handle, temp_callbacks.write))
Josh Coalson463a22c2004-07-11 06:41:00 +00001544 return false;
1545
1546 /* recompute lengths and offsets */
1547 {
Josh Coalsoncc682512002-06-08 04:53:42 +00001548 const FLAC__Metadata_Node *node;
Josh Coalson3ecb1172002-06-15 05:09:47 +00001549 chain->initial_length = current_length;
Josh Coalsone2ba10f2002-05-14 06:03:17 +00001550 chain->last_offset = chain->first_offset;
1551 for(node = chain->head; node; node = node->next)
Josh Coalsoneb643a22002-05-20 05:58:50 +00001552 chain->last_offset += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
Josh Coalsone2ba10f2002-05-14 06:03:17 +00001553 }
1554
Josh Coalsoncae58732002-05-04 17:30:32 +00001555 return true;
1556}
1557
Josh Coalson6afed9f2002-10-16 22:29:47 +00001558FLAC_API void FLAC__metadata_chain_merge_padding(FLAC__Metadata_Chain *chain)
Josh Coalsoncae58732002-05-04 17:30:32 +00001559{
Josh Coalsoncc682512002-06-08 04:53:42 +00001560 FLAC__Metadata_Node *node;
Josh Coalsoncae58732002-05-04 17:30:32 +00001561
1562 FLAC__ASSERT(0 != chain);
1563
1564 for(node = chain->head; node; ) {
Josh Coalson2745fe72002-05-11 05:31:56 +00001565 if(!chain_merge_adjacent_padding_(chain, node))
Josh Coalsoncae58732002-05-04 17:30:32 +00001566 node = node->next;
Josh Coalsoncae58732002-05-04 17:30:32 +00001567 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001568}
1569
Josh Coalson6afed9f2002-10-16 22:29:47 +00001570FLAC_API void FLAC__metadata_chain_sort_padding(FLAC__Metadata_Chain *chain)
Josh Coalsoncae58732002-05-04 17:30:32 +00001571{
Josh Coalsoncc682512002-06-08 04:53:42 +00001572 FLAC__Metadata_Node *node, *save;
Josh Coalsoncae58732002-05-04 17:30:32 +00001573 unsigned i;
1574
1575 FLAC__ASSERT(0 != chain);
1576
1577 /*
1578 * Don't try and be too smart... this simple algo is good enough for
1579 * the small number of nodes that we deal with.
1580 */
1581 for(i = 0, node = chain->head; i < chain->nodes; i++) {
Josh Coalsoncae58732002-05-04 17:30:32 +00001582 if(node->data->type == FLAC__METADATA_TYPE_PADDING) {
1583 save = node->next;
1584 chain_remove_node_(chain, node);
1585 chain_append_node_(chain, node);
1586 node = save;
1587 }
1588 else {
1589 node = node->next;
1590 }
1591 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001592
1593 FLAC__metadata_chain_merge_padding(chain);
1594}
1595
1596
Josh Coalson6afed9f2002-10-16 22:29:47 +00001597FLAC_API FLAC__Metadata_Iterator *FLAC__metadata_iterator_new()
Josh Coalsoncae58732002-05-04 17:30:32 +00001598{
Josh Coalson5e3fcb22002-11-06 07:11:26 +00001599 FLAC__Metadata_Iterator *iterator = (FLAC__Metadata_Iterator*)calloc(1, sizeof(FLAC__Metadata_Iterator));
Josh Coalsoncae58732002-05-04 17:30:32 +00001600
Josh Coalsonea7155f2002-10-18 05:49:19 +00001601 /* calloc() implies:
Josh Coalsoncae58732002-05-04 17:30:32 +00001602 iterator->current = 0;
1603 iterator->chain = 0;
Josh Coalsonea7155f2002-10-18 05:49:19 +00001604 */
Josh Coalsoncae58732002-05-04 17:30:32 +00001605
1606 return iterator;
1607}
1608
Josh Coalson6afed9f2002-10-16 22:29:47 +00001609FLAC_API void FLAC__metadata_iterator_delete(FLAC__Metadata_Iterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00001610{
1611 FLAC__ASSERT(0 != iterator);
1612
1613 free(iterator);
1614}
1615
Josh Coalson6afed9f2002-10-16 22:29:47 +00001616FLAC_API void FLAC__metadata_iterator_init(FLAC__Metadata_Iterator *iterator, FLAC__Metadata_Chain *chain)
Josh Coalsoncae58732002-05-04 17:30:32 +00001617{
1618 FLAC__ASSERT(0 != iterator);
1619 FLAC__ASSERT(0 != chain);
1620 FLAC__ASSERT(0 != chain->head);
1621
1622 iterator->chain = chain;
1623 iterator->current = chain->head;
1624}
1625
Josh Coalson6afed9f2002-10-16 22:29:47 +00001626FLAC_API FLAC__bool FLAC__metadata_iterator_next(FLAC__Metadata_Iterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00001627{
1628 FLAC__ASSERT(0 != iterator);
1629
1630 if(0 == iterator->current || 0 == iterator->current->next)
1631 return false;
1632
1633 iterator->current = iterator->current->next;
1634 return true;
1635}
1636
Josh Coalson6afed9f2002-10-16 22:29:47 +00001637FLAC_API FLAC__bool FLAC__metadata_iterator_prev(FLAC__Metadata_Iterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00001638{
1639 FLAC__ASSERT(0 != iterator);
1640
1641 if(0 == iterator->current || 0 == iterator->current->prev)
1642 return false;
1643
1644 iterator->current = iterator->current->prev;
1645 return true;
1646}
1647
Josh Coalson6afed9f2002-10-16 22:29:47 +00001648FLAC_API FLAC__MetadataType FLAC__metadata_iterator_get_block_type(const FLAC__Metadata_Iterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00001649{
1650 FLAC__ASSERT(0 != iterator);
1651 FLAC__ASSERT(0 != iterator->current);
Josh Coalson2745fe72002-05-11 05:31:56 +00001652 FLAC__ASSERT(0 != iterator->current->data);
Josh Coalsoncae58732002-05-04 17:30:32 +00001653
1654 return iterator->current->data->type;
1655}
1656
Josh Coalson6afed9f2002-10-16 22:29:47 +00001657FLAC_API FLAC__StreamMetadata *FLAC__metadata_iterator_get_block(FLAC__Metadata_Iterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00001658{
1659 FLAC__ASSERT(0 != iterator);
1660 FLAC__ASSERT(0 != iterator->current);
1661
1662 return iterator->current->data;
1663}
1664
Josh Coalson6afed9f2002-10-16 22:29:47 +00001665FLAC_API FLAC__bool FLAC__metadata_iterator_set_block(FLAC__Metadata_Iterator *iterator, FLAC__StreamMetadata *block)
Josh Coalson2745fe72002-05-11 05:31:56 +00001666{
Josh Coalson9b86c962002-07-10 07:11:15 +00001667 FLAC__ASSERT(0 != iterator);
1668 FLAC__ASSERT(0 != block);
Josh Coalson2745fe72002-05-11 05:31:56 +00001669 return FLAC__metadata_iterator_delete_block(iterator, false) && FLAC__metadata_iterator_insert_block_after(iterator, block);
1670}
1671
Josh Coalson6afed9f2002-10-16 22:29:47 +00001672FLAC_API FLAC__bool FLAC__metadata_iterator_delete_block(FLAC__Metadata_Iterator *iterator, FLAC__bool replace_with_padding)
Josh Coalsoncae58732002-05-04 17:30:32 +00001673{
Josh Coalsoncc682512002-06-08 04:53:42 +00001674 FLAC__Metadata_Node *save;
Josh Coalsoncae58732002-05-04 17:30:32 +00001675
1676 FLAC__ASSERT(0 != iterator);
1677 FLAC__ASSERT(0 != iterator->current);
1678
1679 if(0 == iterator->current->prev) {
1680 FLAC__ASSERT(iterator->current->data->type == FLAC__METADATA_TYPE_STREAMINFO);
1681 return false;
1682 }
1683
1684 save = iterator->current->prev;
1685
1686 if(replace_with_padding) {
Josh Coalson90ced912002-05-30 05:23:38 +00001687 FLAC__metadata_object_delete_data(iterator->current->data);
Josh Coalsoncae58732002-05-04 17:30:32 +00001688 iterator->current->data->type = FLAC__METADATA_TYPE_PADDING;
1689 }
1690 else {
1691 chain_delete_node_(iterator->chain, iterator->current);
1692 }
1693
1694 iterator->current = save;
1695 return true;
1696}
1697
Josh Coalson6afed9f2002-10-16 22:29:47 +00001698FLAC_API FLAC__bool FLAC__metadata_iterator_insert_block_before(FLAC__Metadata_Iterator *iterator, FLAC__StreamMetadata *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00001699{
Josh Coalsoncc682512002-06-08 04:53:42 +00001700 FLAC__Metadata_Node *node;
Josh Coalsoncae58732002-05-04 17:30:32 +00001701
1702 FLAC__ASSERT(0 != iterator);
1703 FLAC__ASSERT(0 != iterator->current);
Josh Coalson9b86c962002-07-10 07:11:15 +00001704 FLAC__ASSERT(0 != block);
Josh Coalsoncae58732002-05-04 17:30:32 +00001705
1706 if(block->type == FLAC__METADATA_TYPE_STREAMINFO)
1707 return false;
1708
1709 if(0 == iterator->current->prev) {
1710 FLAC__ASSERT(iterator->current->data->type == FLAC__METADATA_TYPE_STREAMINFO);
1711 return false;
1712 }
1713
1714 if(0 == (node = node_new_()))
1715 return false;
1716
1717 node->data = block;
1718 iterator_insert_node_(iterator, node);
1719 iterator->current = node;
1720 return true;
1721}
1722
Josh Coalson6afed9f2002-10-16 22:29:47 +00001723FLAC_API FLAC__bool FLAC__metadata_iterator_insert_block_after(FLAC__Metadata_Iterator *iterator, FLAC__StreamMetadata *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00001724{
Josh Coalsoncc682512002-06-08 04:53:42 +00001725 FLAC__Metadata_Node *node;
Josh Coalsoncae58732002-05-04 17:30:32 +00001726
1727 FLAC__ASSERT(0 != iterator);
1728 FLAC__ASSERT(0 != iterator->current);
Josh Coalson9b86c962002-07-10 07:11:15 +00001729 FLAC__ASSERT(0 != block);
Josh Coalsoncae58732002-05-04 17:30:32 +00001730
1731 if(block->type == FLAC__METADATA_TYPE_STREAMINFO)
1732 return false;
1733
1734 if(0 == (node = node_new_()))
1735 return false;
1736
1737 node->data = block;
1738 iterator_insert_node_after_(iterator, node);
1739 iterator->current = node;
1740 return true;
1741}
1742
1743
1744/****************************************************************************
1745 *
Josh Coalsoncae58732002-05-04 17:30:32 +00001746 * Local function definitions
1747 *
1748 ***************************************************************************/
1749
1750void pack_uint32_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes)
1751{
1752 unsigned i;
1753
1754 b += bytes;
1755
1756 for(i = 0; i < bytes; i++) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00001757 *(--b) = (FLAC__byte)(val & 0xff);
Josh Coalsoncae58732002-05-04 17:30:32 +00001758 val >>= 8;
1759 }
1760}
1761
1762void pack_uint32_little_endian_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes)
1763{
1764 unsigned i;
1765
1766 for(i = 0; i < bytes; i++) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00001767 *(b++) = (FLAC__byte)(val & 0xff);
Josh Coalsoncae58732002-05-04 17:30:32 +00001768 val >>= 8;
1769 }
1770}
1771
1772void pack_uint64_(FLAC__uint64 val, FLAC__byte *b, unsigned bytes)
1773{
1774 unsigned i;
1775
1776 b += bytes;
1777
1778 for(i = 0; i < bytes; i++) {
Josh Coalson72943092002-05-22 05:35:43 +00001779 *(--b) = (FLAC__byte)(val & 0xff);
Josh Coalsoncae58732002-05-04 17:30:32 +00001780 val >>= 8;
1781 }
1782}
1783
1784FLAC__uint32 unpack_uint32_(FLAC__byte *b, unsigned bytes)
1785{
1786 FLAC__uint32 ret = 0;
1787 unsigned i;
1788
1789 for(i = 0; i < bytes; i++)
1790 ret = (ret << 8) | (FLAC__uint32)(*b++);
1791
1792 return ret;
1793}
1794
1795FLAC__uint32 unpack_uint32_little_endian_(FLAC__byte *b, unsigned bytes)
1796{
1797 FLAC__uint32 ret = 0;
1798 unsigned i;
1799
1800 b += bytes;
1801
1802 for(i = 0; i < bytes; i++)
1803 ret = (ret << 8) | (FLAC__uint32)(*--b);
1804
1805 return ret;
1806}
1807
1808FLAC__uint64 unpack_uint64_(FLAC__byte *b, unsigned bytes)
1809{
1810 FLAC__uint64 ret = 0;
1811 unsigned i;
1812
1813 for(i = 0; i < bytes; i++)
1814 ret = (ret << 8) | (FLAC__uint64)(*b++);
1815
1816 return ret;
1817}
1818
Josh Coalsoncc682512002-06-08 04:53:42 +00001819FLAC__bool read_metadata_block_header_(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00001820{
Josh Coalsoncae58732002-05-04 17:30:32 +00001821 FLAC__ASSERT(0 != iterator);
1822 FLAC__ASSERT(0 != iterator->file);
1823
Josh Coalsonea25cb32004-07-10 00:39:01 +00001824 if(!read_metadata_block_header_cb_((FLAC__IOHandle)iterator->file, (FLAC__IOCallback_Read)fread, &iterator->is_last, &iterator->type, &iterator->length)) {
Josh Coalsoncae58732002-05-04 17:30:32 +00001825 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1826 return false;
1827 }
1828
Josh Coalsoncae58732002-05-04 17:30:32 +00001829 return true;
1830}
1831
Josh Coalsoncc682512002-06-08 04:53:42 +00001832FLAC__bool read_metadata_block_data_(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00001833{
1834 FLAC__ASSERT(0 != iterator);
1835 FLAC__ASSERT(0 != iterator->file);
1836
Josh Coalsonea25cb32004-07-10 00:39:01 +00001837 iterator->status = read_metadata_block_data_cb_((FLAC__IOHandle)iterator->file, (FLAC__IOCallback_Read)fread, fseek_wrapper_, block);
Josh Coalsoncae58732002-05-04 17:30:32 +00001838
1839 return (iterator->status == FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK);
1840}
1841
Josh Coalsonea25cb32004-07-10 00:39:01 +00001842FLAC__bool read_metadata_block_header_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__bool *is_last, FLAC__MetadataType *type, unsigned *length)
1843{
1844 FLAC__byte raw_header[FLAC__STREAM_METADATA_HEADER_LENGTH];
1845
1846 if(read_cb(raw_header, 1, FLAC__STREAM_METADATA_HEADER_LENGTH, handle) != FLAC__STREAM_METADATA_HEADER_LENGTH)
1847 return false;
1848
1849 *is_last = raw_header[0] & 0x80? true : false;
1850 *type = (FLAC__MetadataType)(raw_header[0] & 0x7f);
1851 *length = unpack_uint32_(raw_header + 1, 3);
1852
1853 /* Note that we don't check:
1854 * if(iterator->type >= FLAC__METADATA_TYPE_UNDEFINED)
1855 * we just will read in an opaque block
1856 */
1857
1858 return true;
1859}
1860
1861FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Seek seek_cb, FLAC__StreamMetadata *block)
1862{
1863 switch(block->type) {
1864 case FLAC__METADATA_TYPE_STREAMINFO:
1865 return read_metadata_block_data_streaminfo_cb_(handle, read_cb, &block->data.stream_info);
1866 case FLAC__METADATA_TYPE_PADDING:
1867 return read_metadata_block_data_padding_cb_(handle, seek_cb, &block->data.padding, block->length);
1868 case FLAC__METADATA_TYPE_APPLICATION:
1869 return read_metadata_block_data_application_cb_(handle, read_cb, &block->data.application, block->length);
1870 case FLAC__METADATA_TYPE_SEEKTABLE:
1871 return read_metadata_block_data_seektable_cb_(handle, read_cb, &block->data.seek_table, block->length);
1872 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1873 return read_metadata_block_data_vorbis_comment_cb_(handle, read_cb, &block->data.vorbis_comment);
1874 case FLAC__METADATA_TYPE_CUESHEET:
1875 return read_metadata_block_data_cuesheet_cb_(handle, read_cb, &block->data.cue_sheet);
1876 default:
1877 return read_metadata_block_data_unknown_cb_(handle, read_cb, &block->data.unknown, block->length);
1878 }
1879}
1880
1881FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_streaminfo_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_StreamInfo *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00001882{
1883 FLAC__byte buffer[FLAC__STREAM_METADATA_STREAMINFO_LENGTH], *b;
1884
Josh Coalsonea25cb32004-07-10 00:39:01 +00001885 if(read_cb(buffer, 1, FLAC__STREAM_METADATA_STREAMINFO_LENGTH, handle) != FLAC__STREAM_METADATA_STREAMINFO_LENGTH)
Josh Coalsoncae58732002-05-04 17:30:32 +00001886 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1887
1888 b = buffer;
1889
Josh Coalsonf0a8c4f2003-01-15 03:17:51 +00001890 /* we are using hardcoded numbers for simplicity but we should
Josh Coalsoncae58732002-05-04 17:30:32 +00001891 * probably eventually write a bit-level unpacker and use the
1892 * _STREAMINFO_ constants.
1893 */
1894 block->min_blocksize = unpack_uint32_(b, 2); b += 2;
1895 block->max_blocksize = unpack_uint32_(b, 2); b += 2;
1896 block->min_framesize = unpack_uint32_(b, 3); b += 3;
1897 block->max_framesize = unpack_uint32_(b, 3); b += 3;
1898 block->sample_rate = (unpack_uint32_(b, 2) << 4) | ((unsigned)(b[2] & 0xf0) >> 4);
1899 block->channels = (unsigned)((b[2] & 0x0e) >> 1) + 1;
Josh Coalsonb4e186a2002-08-12 21:35:43 +00001900 block->bits_per_sample = ((((unsigned)(b[2] & 0x01)) << 4) | (((unsigned)(b[3] & 0xf0)) >> 4)) + 1;
Josh Coalsoncae58732002-05-04 17:30:32 +00001901 block->total_samples = (((FLAC__uint64)(b[3] & 0x0f)) << 32) | unpack_uint64_(b+4, 4);
1902 memcpy(block->md5sum, b+8, 16);
1903
1904 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1905}
1906
1907
Josh Coalsonea25cb32004-07-10 00:39:01 +00001908FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_padding_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Seek seek_cb, FLAC__StreamMetadata_Padding *block, unsigned block_length)
Josh Coalsoncae58732002-05-04 17:30:32 +00001909{
Josh Coalsoncae58732002-05-04 17:30:32 +00001910 (void)block; /* nothing to do; we don't care about reading the padding bytes */
1911
Josh Coalsonea25cb32004-07-10 00:39:01 +00001912 if(0 != seek_cb(handle, block_length, SEEK_CUR))
Josh Coalsoncae58732002-05-04 17:30:32 +00001913 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
1914
1915 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1916}
1917
Josh Coalsonea25cb32004-07-10 00:39:01 +00001918FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_application_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_Application *block, unsigned block_length)
Josh Coalsoncae58732002-05-04 17:30:32 +00001919{
1920 const unsigned id_bytes = FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8;
1921
Josh Coalsonea25cb32004-07-10 00:39:01 +00001922 if(read_cb(block->id, 1, id_bytes, handle) != id_bytes)
Josh Coalsoncae58732002-05-04 17:30:32 +00001923 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1924
1925 block_length -= id_bytes;
1926
Josh Coalson72943092002-05-22 05:35:43 +00001927 if(block_length == 0) {
1928 block->data = 0;
1929 }
1930 else {
Josh Coalson5e3fcb22002-11-06 07:11:26 +00001931 if(0 == (block->data = (FLAC__byte*)malloc(block_length)))
Josh Coalson72943092002-05-22 05:35:43 +00001932 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
Josh Coalsoncae58732002-05-04 17:30:32 +00001933
Josh Coalsonea25cb32004-07-10 00:39:01 +00001934 if(read_cb(block->data, 1, block_length, handle) != block_length)
Josh Coalson72943092002-05-22 05:35:43 +00001935 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1936 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001937
1938 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1939}
1940
Josh Coalsonea25cb32004-07-10 00:39:01 +00001941FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_seektable_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_SeekTable *block, unsigned block_length)
Josh Coalsoncae58732002-05-04 17:30:32 +00001942{
1943 unsigned i;
1944 FLAC__byte buffer[FLAC__STREAM_METADATA_SEEKPOINT_LENGTH];
1945
Josh Coalsoncae58732002-05-04 17:30:32 +00001946 FLAC__ASSERT(block_length % FLAC__STREAM_METADATA_SEEKPOINT_LENGTH == 0);
1947
1948 block->num_points = block_length / FLAC__STREAM_METADATA_SEEKPOINT_LENGTH;
1949
Josh Coalson72943092002-05-22 05:35:43 +00001950 if(block->num_points == 0)
1951 block->points = 0;
Josh Coalson5e3fcb22002-11-06 07:11:26 +00001952 else if(0 == (block->points = (FLAC__StreamMetadata_SeekPoint*)malloc(block->num_points * sizeof(FLAC__StreamMetadata_SeekPoint))))
Josh Coalsoncae58732002-05-04 17:30:32 +00001953 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
1954
1955 for(i = 0; i < block->num_points; i++) {
Josh Coalsonea25cb32004-07-10 00:39:01 +00001956 if(read_cb(buffer, 1, FLAC__STREAM_METADATA_SEEKPOINT_LENGTH, handle) != FLAC__STREAM_METADATA_SEEKPOINT_LENGTH)
Josh Coalsoncae58732002-05-04 17:30:32 +00001957 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
Josh Coalsonf0a8c4f2003-01-15 03:17:51 +00001958 /* some MAGIC NUMBERs here */
Josh Coalsoncae58732002-05-04 17:30:32 +00001959 block->points[i].sample_number = unpack_uint64_(buffer, 8);
1960 block->points[i].stream_offset = unpack_uint64_(buffer+8, 8);
1961 block->points[i].frame_samples = unpack_uint32_(buffer+16, 2);
1962 }
1963
1964 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1965}
1966
Josh Coalsonea25cb32004-07-10 00:39:01 +00001967FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_entry_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_VorbisComment_Entry *entry)
Josh Coalsoncae58732002-05-04 17:30:32 +00001968{
1969 const unsigned entry_length_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8;
Josh Coalson7ee4d172002-06-01 06:31:27 +00001970 FLAC__byte buffer[4]; /* magic number is asserted below */
Josh Coalsoncae58732002-05-04 17:30:32 +00001971
Josh Coalson7ee4d172002-06-01 06:31:27 +00001972 FLAC__ASSERT(FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8 == 4);
Josh Coalsoncae58732002-05-04 17:30:32 +00001973
Josh Coalsonea25cb32004-07-10 00:39:01 +00001974 if(read_cb(buffer, 1, entry_length_len, handle) != entry_length_len)
Josh Coalsoncae58732002-05-04 17:30:32 +00001975 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
Josh Coalsoneb643a22002-05-20 05:58:50 +00001976 entry->length = unpack_uint32_little_endian_(buffer, entry_length_len);
Josh Coalsoncae58732002-05-04 17:30:32 +00001977
Josh Coalson6ffe8da2002-10-26 04:35:00 +00001978 if(0 != entry->entry)
1979 free(entry->entry);
1980
Josh Coalson72943092002-05-22 05:35:43 +00001981 if(entry->length == 0) {
1982 entry->entry = 0;
1983 }
1984 else {
Josh Coalsondef597e2004-12-30 00:59:30 +00001985 if(0 == (entry->entry = (FLAC__byte*)malloc(entry->length+1)))
Josh Coalson72943092002-05-22 05:35:43 +00001986 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
Josh Coalsoncae58732002-05-04 17:30:32 +00001987
Josh Coalsonea25cb32004-07-10 00:39:01 +00001988 if(read_cb(entry->entry, 1, entry->length, handle) != entry->length)
Josh Coalson72943092002-05-22 05:35:43 +00001989 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
Josh Coalsondef597e2004-12-30 00:59:30 +00001990
1991 entry->entry[entry->length] = '\0';
Josh Coalson72943092002-05-22 05:35:43 +00001992 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001993
1994 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1995}
1996
Josh Coalsonea25cb32004-07-10 00:39:01 +00001997FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_VorbisComment *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00001998{
1999 unsigned i;
Josh Coalsoncc682512002-06-08 04:53:42 +00002000 FLAC__Metadata_SimpleIteratorStatus status;
Josh Coalsoncae58732002-05-04 17:30:32 +00002001 const unsigned num_comments_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN / 8;
Josh Coalson7ee4d172002-06-01 06:31:27 +00002002 FLAC__byte buffer[4]; /* magic number is asserted below */
Josh Coalsoncae58732002-05-04 17:30:32 +00002003
Josh Coalson7ee4d172002-06-01 06:31:27 +00002004 FLAC__ASSERT(FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN / 8 == 4);
Josh Coalsoncae58732002-05-04 17:30:32 +00002005
Josh Coalsonea25cb32004-07-10 00:39:01 +00002006 if(FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK != (status = read_metadata_block_data_vorbis_comment_entry_cb_(handle, read_cb, &(block->vendor_string))))
Josh Coalsoncae58732002-05-04 17:30:32 +00002007 return status;
2008
Josh Coalsonea25cb32004-07-10 00:39:01 +00002009 if(read_cb(buffer, 1, num_comments_len, handle) != num_comments_len)
Josh Coalsoncae58732002-05-04 17:30:32 +00002010 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2011 block->num_comments = unpack_uint32_little_endian_(buffer, num_comments_len);
2012
Josh Coalson72943092002-05-22 05:35:43 +00002013 if(block->num_comments == 0) {
2014 block->comments = 0;
2015 }
Josh Coalson38162d52002-11-16 06:31:54 +00002016 else if(0 == (block->comments = (FLAC__StreamMetadata_VorbisComment_Entry*)calloc(block->num_comments, sizeof(FLAC__StreamMetadata_VorbisComment_Entry))))
Josh Coalsoncae58732002-05-04 17:30:32 +00002017 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2018
2019 for(i = 0; i < block->num_comments; i++) {
Josh Coalsonea25cb32004-07-10 00:39:01 +00002020 if(FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK != (status = read_metadata_block_data_vorbis_comment_entry_cb_(handle, read_cb, block->comments + i)))
Josh Coalsoncae58732002-05-04 17:30:32 +00002021 return status;
2022 }
2023
2024 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
2025}
2026
Josh Coalsonea25cb32004-07-10 00:39:01 +00002027FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_cuesheet_track_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_CueSheet_Track *track)
Josh Coalsone4869382002-11-15 05:41:48 +00002028{
2029 unsigned i, len;
2030 FLAC__byte buffer[32]; /* asserted below that this is big enough */
2031
2032 FLAC__ASSERT(sizeof(buffer) >= sizeof(FLAC__uint64));
2033 FLAC__ASSERT(sizeof(buffer) >= FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN/8);
2034 FLAC__ASSERT(sizeof(buffer) >= (FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN) / 8);
2035
Josh Coalsone4869382002-11-15 05:41:48 +00002036 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_OFFSET_LEN % 8 == 0);
2037 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_OFFSET_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002038 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002039 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2040 track->offset = unpack_uint64_(buffer, len);
2041
2042 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_NUMBER_LEN % 8 == 0);
2043 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_NUMBER_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002044 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002045 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2046 track->number = (FLAC__byte)unpack_uint32_(buffer, len);
2047
2048 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN % 8 == 0);
2049 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002050 if(read_cb(track->isrc, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002051 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2052
2053 FLAC__ASSERT((FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN) % 8 == 0);
2054 len = (FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN) / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002055 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002056 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2057 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN == 1);
2058 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN == 1);
2059 track->type = buffer[0] >> 7;
2060 track->pre_emphasis = (buffer[0] >> 6) & 1;
2061
2062 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_NUM_INDICES_LEN % 8 == 0);
2063 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_NUM_INDICES_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002064 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002065 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2066 track->num_indices = (FLAC__byte)unpack_uint32_(buffer, len);
2067
2068 if(track->num_indices == 0) {
2069 track->indices = 0;
2070 }
2071 else if(0 == (track->indices = (FLAC__StreamMetadata_CueSheet_Index*)calloc(track->num_indices, sizeof(FLAC__StreamMetadata_CueSheet_Index))))
2072 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2073
2074 for(i = 0; i < track->num_indices; i++) {
2075 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_INDEX_OFFSET_LEN % 8 == 0);
2076 len = FLAC__STREAM_METADATA_CUESHEET_INDEX_OFFSET_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002077 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002078 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2079 track->indices[i].offset = unpack_uint64_(buffer, len);
2080
2081 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_INDEX_NUMBER_LEN % 8 == 0);
2082 len = FLAC__STREAM_METADATA_CUESHEET_INDEX_NUMBER_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002083 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002084 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2085 track->indices[i].number = (FLAC__byte)unpack_uint32_(buffer, len);
2086
2087 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN % 8 == 0);
2088 len = FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002089 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002090 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2091 }
2092
2093 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
2094}
2095
Josh Coalsonea25cb32004-07-10 00:39:01 +00002096FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_cuesheet_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_CueSheet *block)
Josh Coalsone4869382002-11-15 05:41:48 +00002097{
2098 unsigned i, len;
2099 FLAC__Metadata_SimpleIteratorStatus status;
Josh Coalsonbfc8e312002-11-21 09:00:25 +00002100 FLAC__byte buffer[1024]; /* MSVC needs a constant expression so we put a magic number and assert */
Josh Coalsone4869382002-11-15 05:41:48 +00002101
Josh Coalson8f0c71b2002-12-05 06:37:46 +00002102 FLAC__ASSERT((FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN + FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN)/8 <= sizeof(buffer));
Josh Coalsonbfc8e312002-11-21 09:00:25 +00002103 FLAC__ASSERT(sizeof(FLAC__uint64) <= sizeof(buffer));
Josh Coalsone4869382002-11-15 05:41:48 +00002104
2105 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN % 8 == 0);
2106 len = FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002107 if(read_cb(block->media_catalog_number, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002108 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2109
2110 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_LEAD_IN_LEN % 8 == 0);
2111 len = FLAC__STREAM_METADATA_CUESHEET_LEAD_IN_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002112 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002113 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2114 block->lead_in = unpack_uint64_(buffer, len);
2115
Josh Coalson8f0c71b2002-12-05 06:37:46 +00002116 FLAC__ASSERT((FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN + FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN) % 8 == 0);
2117 len = (FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN + FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN) / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002118 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalson38162d52002-11-16 06:31:54 +00002119 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
Josh Coalson8f0c71b2002-12-05 06:37:46 +00002120 block->is_cd = buffer[0]&0x80? true : false;
Josh Coalson38162d52002-11-16 06:31:54 +00002121
Josh Coalsone4869382002-11-15 05:41:48 +00002122 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_NUM_TRACKS_LEN % 8 == 0);
2123 len = FLAC__STREAM_METADATA_CUESHEET_NUM_TRACKS_LEN / 8;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002124 if(read_cb(buffer, 1, len, handle) != len)
Josh Coalsone4869382002-11-15 05:41:48 +00002125 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2126 block->num_tracks = unpack_uint32_(buffer, len);
2127
2128 if(block->num_tracks == 0) {
2129 block->tracks = 0;
2130 }
2131 else if(0 == (block->tracks = (FLAC__StreamMetadata_CueSheet_Track*)calloc(block->num_tracks, sizeof(FLAC__StreamMetadata_CueSheet_Track))))
2132 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2133
2134 for(i = 0; i < block->num_tracks; i++) {
Josh Coalsonea25cb32004-07-10 00:39:01 +00002135 if(FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK != (status = read_metadata_block_data_cuesheet_track_cb_(handle, read_cb, block->tracks + i)))
Josh Coalsone4869382002-11-15 05:41:48 +00002136 return status;
2137 }
2138
2139 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
2140}
2141
Josh Coalsonea25cb32004-07-10 00:39:01 +00002142FLAC__Metadata_SimpleIteratorStatus read_metadata_block_data_unknown_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__StreamMetadata_Unknown *block, unsigned block_length)
Josh Coalsond0609472003-01-10 05:37:13 +00002143{
Josh Coalsond0609472003-01-10 05:37:13 +00002144 if(block_length == 0) {
2145 block->data = 0;
2146 }
2147 else {
2148 if(0 == (block->data = (FLAC__byte*)malloc(block_length)))
2149 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2150
Josh Coalsonea25cb32004-07-10 00:39:01 +00002151 if(read_cb(block->data, 1, block_length, handle) != block_length)
Josh Coalsond0609472003-01-10 05:37:13 +00002152 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2153 }
2154
2155 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
2156}
2157
Josh Coalsoncc682512002-06-08 04:53:42 +00002158FLAC__bool write_metadata_block_header_(FILE *file, FLAC__Metadata_SimpleIteratorStatus *status, const FLAC__StreamMetadata *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00002159{
Josh Coalsoncae58732002-05-04 17:30:32 +00002160 FLAC__ASSERT(0 != file);
2161 FLAC__ASSERT(0 != status);
Josh Coalsoncae58732002-05-04 17:30:32 +00002162
Josh Coalson463a22c2004-07-11 06:41:00 +00002163 if(!write_metadata_block_header_cb_((FLAC__IOHandle)file, (FLAC__IOCallback_Write)fwrite, block)) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002164 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2165 return false;
2166 }
2167
2168 return true;
2169}
2170
Josh Coalsoncc682512002-06-08 04:53:42 +00002171FLAC__bool write_metadata_block_data_(FILE *file, FLAC__Metadata_SimpleIteratorStatus *status, const FLAC__StreamMetadata *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00002172{
2173 FLAC__ASSERT(0 != file);
2174 FLAC__ASSERT(0 != status);
2175
Josh Coalson463a22c2004-07-11 06:41:00 +00002176 if (write_metadata_block_data_cb_((FLAC__IOHandle)file, (FLAC__IOCallback_Write)fwrite, block)) {
2177 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
2178 return true;
Josh Coalsoncae58732002-05-04 17:30:32 +00002179 }
Josh Coalson463a22c2004-07-11 06:41:00 +00002180 else {
2181 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2182 return false;
2183 }
Josh Coalsoncae58732002-05-04 17:30:32 +00002184}
2185
Josh Coalson463a22c2004-07-11 06:41:00 +00002186FLAC__bool write_metadata_block_header_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata *block)
2187{
2188 FLAC__byte buffer[FLAC__STREAM_METADATA_HEADER_LENGTH];
2189
2190 FLAC__ASSERT(block->length < (1u << FLAC__STREAM_METADATA_LENGTH_LEN));
2191
2192 buffer[0] = (block->is_last? 0x80 : 0) | (FLAC__byte)block->type;
2193 pack_uint32_(block->length, buffer + 1, 3);
2194
2195 if(write_cb(buffer, 1, FLAC__STREAM_METADATA_HEADER_LENGTH, handle) != FLAC__STREAM_METADATA_HEADER_LENGTH)
2196 return false;
2197
2198 return true;
2199}
2200
2201FLAC__bool write_metadata_block_data_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata *block)
2202{
2203 FLAC__ASSERT(0 != block);
2204
2205 switch(block->type) {
2206 case FLAC__METADATA_TYPE_STREAMINFO:
2207 return write_metadata_block_data_streaminfo_cb_(handle, write_cb, &block->data.stream_info);
2208 case FLAC__METADATA_TYPE_PADDING:
2209 return write_metadata_block_data_padding_cb_(handle, write_cb, &block->data.padding, block->length);
2210 case FLAC__METADATA_TYPE_APPLICATION:
2211 return write_metadata_block_data_application_cb_(handle, write_cb, &block->data.application, block->length);
2212 case FLAC__METADATA_TYPE_SEEKTABLE:
2213 return write_metadata_block_data_seektable_cb_(handle, write_cb, &block->data.seek_table);
2214 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
2215 return write_metadata_block_data_vorbis_comment_cb_(handle, write_cb, &block->data.vorbis_comment);
2216 case FLAC__METADATA_TYPE_CUESHEET:
2217 return write_metadata_block_data_cuesheet_cb_(handle, write_cb, &block->data.cue_sheet);
2218 default:
2219 return write_metadata_block_data_unknown_cb_(handle, write_cb, &block->data.unknown, block->length);
2220 }
2221}
2222
2223FLAC__bool write_metadata_block_data_streaminfo_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_StreamInfo *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00002224{
2225 FLAC__byte buffer[FLAC__STREAM_METADATA_STREAMINFO_LENGTH];
2226 const unsigned channels1 = block->channels - 1;
2227 const unsigned bps1 = block->bits_per_sample - 1;
2228
Josh Coalsonf0a8c4f2003-01-15 03:17:51 +00002229 /* we are using hardcoded numbers for simplicity but we should
Josh Coalsoncae58732002-05-04 17:30:32 +00002230 * probably eventually write a bit-level packer and use the
2231 * _STREAMINFO_ constants.
2232 */
2233 pack_uint32_(block->min_blocksize, buffer, 2);
2234 pack_uint32_(block->max_blocksize, buffer+2, 2);
2235 pack_uint32_(block->min_framesize, buffer+4, 3);
2236 pack_uint32_(block->max_framesize, buffer+7, 3);
2237 buffer[10] = (block->sample_rate >> 12) & 0xff;
2238 buffer[11] = (block->sample_rate >> 4) & 0xff;
2239 buffer[12] = ((block->sample_rate & 0x0f) << 4) | (channels1 << 1) | (bps1 >> 4);
Josh Coalson7ee4d172002-06-01 06:31:27 +00002240 buffer[13] = (FLAC__byte)(((bps1 & 0x0f) << 4) | ((block->total_samples >> 32) & 0x0f));
Josh Coalsoncae58732002-05-04 17:30:32 +00002241 pack_uint32_((FLAC__uint32)block->total_samples, buffer+14, 4);
2242 memcpy(buffer+18, block->md5sum, 16);
2243
Josh Coalson463a22c2004-07-11 06:41:00 +00002244 if(write_cb(buffer, 1, FLAC__STREAM_METADATA_STREAMINFO_LENGTH, handle) != FLAC__STREAM_METADATA_STREAMINFO_LENGTH)
2245 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002246
Josh Coalson463a22c2004-07-11 06:41:00 +00002247 return true;
Josh Coalsoncae58732002-05-04 17:30:32 +00002248}
2249
Josh Coalson463a22c2004-07-11 06:41:00 +00002250FLAC__bool write_metadata_block_data_padding_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_Padding *block, unsigned block_length)
Josh Coalsoncae58732002-05-04 17:30:32 +00002251{
2252 unsigned i, n = block_length;
2253 FLAC__byte buffer[1024];
2254
Josh Coalsoncae58732002-05-04 17:30:32 +00002255 (void)block;
2256
2257 memset(buffer, 0, 1024);
2258
2259 for(i = 0; i < n/1024; i++)
Josh Coalson463a22c2004-07-11 06:41:00 +00002260 if(write_cb(buffer, 1, 1024, handle) != 1024)
2261 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002262
2263 n %= 1024;
Josh Coalson765ff502002-08-27 05:46:11 +00002264
Josh Coalson463a22c2004-07-11 06:41:00 +00002265 if(write_cb(buffer, 1, n, handle) != n)
2266 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002267
Josh Coalson463a22c2004-07-11 06:41:00 +00002268 return true;
Josh Coalsoncae58732002-05-04 17:30:32 +00002269}
2270
Josh Coalson463a22c2004-07-11 06:41:00 +00002271FLAC__bool write_metadata_block_data_application_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_Application *block, unsigned block_length)
Josh Coalsoncae58732002-05-04 17:30:32 +00002272{
2273 const unsigned id_bytes = FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8;
2274
Josh Coalson463a22c2004-07-11 06:41:00 +00002275 if(write_cb(block->id, 1, id_bytes, handle) != id_bytes)
2276 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002277
2278 block_length -= id_bytes;
2279
Josh Coalson463a22c2004-07-11 06:41:00 +00002280 if(write_cb(block->data, 1, block_length, handle) != block_length)
2281 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002282
Josh Coalson463a22c2004-07-11 06:41:00 +00002283 return true;
Josh Coalsoncae58732002-05-04 17:30:32 +00002284}
2285
Josh Coalson463a22c2004-07-11 06:41:00 +00002286FLAC__bool write_metadata_block_data_seektable_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_SeekTable *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00002287{
2288 unsigned i;
2289 FLAC__byte buffer[FLAC__STREAM_METADATA_SEEKPOINT_LENGTH];
2290
Josh Coalsoncae58732002-05-04 17:30:32 +00002291 for(i = 0; i < block->num_points; i++) {
Josh Coalsonf0a8c4f2003-01-15 03:17:51 +00002292 /* some MAGIC NUMBERs here */
Josh Coalsoncae58732002-05-04 17:30:32 +00002293 pack_uint64_(block->points[i].sample_number, buffer, 8);
2294 pack_uint64_(block->points[i].stream_offset, buffer+8, 8);
2295 pack_uint32_(block->points[i].frame_samples, buffer+16, 2);
Josh Coalson463a22c2004-07-11 06:41:00 +00002296 if(write_cb(buffer, 1, FLAC__STREAM_METADATA_SEEKPOINT_LENGTH, handle) != FLAC__STREAM_METADATA_SEEKPOINT_LENGTH)
2297 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002298 }
2299
Josh Coalson463a22c2004-07-11 06:41:00 +00002300 return true;
Josh Coalsoncae58732002-05-04 17:30:32 +00002301}
2302
Josh Coalson463a22c2004-07-11 06:41:00 +00002303FLAC__bool write_metadata_block_data_vorbis_comment_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_VorbisComment *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00002304{
2305 unsigned i;
2306 const unsigned entry_length_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8;
2307 const unsigned num_comments_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN / 8;
Josh Coalson7ee4d172002-06-01 06:31:27 +00002308 FLAC__byte buffer[4]; /* magic number is asserted below */
Josh Coalsoncae58732002-05-04 17:30:32 +00002309
Josh Coalson7ee4d172002-06-01 06:31:27 +00002310 FLAC__ASSERT(max(FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN, FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN) / 8 == 4);
Josh Coalsoncae58732002-05-04 17:30:32 +00002311
2312 pack_uint32_little_endian_(block->vendor_string.length, buffer, entry_length_len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002313 if(write_cb(buffer, 1, entry_length_len, handle) != entry_length_len)
2314 return false;
2315 if(write_cb(block->vendor_string.entry, 1, block->vendor_string.length, handle) != block->vendor_string.length)
2316 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002317
2318 pack_uint32_little_endian_(block->num_comments, buffer, num_comments_len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002319 if(write_cb(buffer, 1, num_comments_len, handle) != num_comments_len)
2320 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002321
2322 for(i = 0; i < block->num_comments; i++) {
2323 pack_uint32_little_endian_(block->comments[i].length, buffer, entry_length_len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002324 if(write_cb(buffer, 1, entry_length_len, handle) != entry_length_len)
2325 return false;
2326 if(write_cb(block->comments[i].entry, 1, block->comments[i].length, handle) != block->comments[i].length)
2327 return false;
Josh Coalsoncae58732002-05-04 17:30:32 +00002328 }
2329
Josh Coalson463a22c2004-07-11 06:41:00 +00002330 return true;
Josh Coalsoncae58732002-05-04 17:30:32 +00002331}
2332
Josh Coalson463a22c2004-07-11 06:41:00 +00002333FLAC__bool write_metadata_block_data_cuesheet_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_CueSheet *block)
Josh Coalsone4869382002-11-15 05:41:48 +00002334{
2335 unsigned i, j, len;
Josh Coalson38162d52002-11-16 06:31:54 +00002336 FLAC__byte buffer[1024]; /* asserted below that this is big enough */
Josh Coalsone4869382002-11-15 05:41:48 +00002337
2338 FLAC__ASSERT(sizeof(buffer) >= sizeof(FLAC__uint64));
Josh Coalson38162d52002-11-16 06:31:54 +00002339 FLAC__ASSERT(sizeof(buffer) >= FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN/8);
Josh Coalson8f0c71b2002-12-05 06:37:46 +00002340 FLAC__ASSERT(sizeof(buffer) >= (FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN + FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN)/8);
Josh Coalsone4869382002-11-15 05:41:48 +00002341 FLAC__ASSERT(sizeof(buffer) >= FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN/8);
2342
Josh Coalsone4869382002-11-15 05:41:48 +00002343 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN % 8 == 0);
2344 len = FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN / 8;
Josh Coalson463a22c2004-07-11 06:41:00 +00002345 if(write_cb(block->media_catalog_number, 1, len, handle) != len)
2346 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002347
2348 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_LEAD_IN_LEN % 8 == 0);
2349 len = FLAC__STREAM_METADATA_CUESHEET_LEAD_IN_LEN / 8;
2350 pack_uint64_(block->lead_in, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002351 if(write_cb(buffer, 1, len, handle) != len)
2352 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002353
Josh Coalson8f0c71b2002-12-05 06:37:46 +00002354 FLAC__ASSERT((FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN + FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN) % 8 == 0);
2355 len = (FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN + FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN) / 8;
Josh Coalson38162d52002-11-16 06:31:54 +00002356 memset(buffer, 0, len);
Josh Coalson8f0c71b2002-12-05 06:37:46 +00002357 if(block->is_cd)
2358 buffer[0] |= 0x80;
Josh Coalson463a22c2004-07-11 06:41:00 +00002359 if(write_cb(buffer, 1, len, handle) != len)
2360 return false;
Josh Coalson38162d52002-11-16 06:31:54 +00002361
Josh Coalsone4869382002-11-15 05:41:48 +00002362 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_NUM_TRACKS_LEN % 8 == 0);
2363 len = FLAC__STREAM_METADATA_CUESHEET_NUM_TRACKS_LEN / 8;
2364 pack_uint32_(block->num_tracks, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002365 if(write_cb(buffer, 1, len, handle) != len)
2366 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002367
2368 for(i = 0; i < block->num_tracks; i++) {
2369 FLAC__StreamMetadata_CueSheet_Track *track = block->tracks + i;
2370
2371 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_OFFSET_LEN % 8 == 0);
2372 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_OFFSET_LEN / 8;
2373 pack_uint64_(track->offset, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002374 if(write_cb(buffer, 1, len, handle) != len)
2375 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002376
2377 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_NUMBER_LEN % 8 == 0);
2378 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_NUMBER_LEN / 8;
2379 pack_uint32_(track->number, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002380 if(write_cb(buffer, 1, len, handle) != len)
2381 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002382
2383 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN % 8 == 0);
2384 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN / 8;
Josh Coalson463a22c2004-07-11 06:41:00 +00002385 if(write_cb(track->isrc, 1, len, handle) != len)
2386 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002387
2388 FLAC__ASSERT((FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN) % 8 == 0);
2389 len = (FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN + FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN) / 8;
2390 memset(buffer, 0, len);
2391 buffer[0] = (track->type << 7) | (track->pre_emphasis << 6);
Josh Coalson463a22c2004-07-11 06:41:00 +00002392 if(write_cb(buffer, 1, len, handle) != len)
2393 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002394
2395 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_NUM_INDICES_LEN % 8 == 0);
2396 len = FLAC__STREAM_METADATA_CUESHEET_TRACK_NUM_INDICES_LEN / 8;
2397 pack_uint32_(track->num_indices, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002398 if(write_cb(buffer, 1, len, handle) != len)
2399 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002400
2401 for(j = 0; j < track->num_indices; j++) {
2402 FLAC__StreamMetadata_CueSheet_Index *index = track->indices + j;
2403
2404 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_INDEX_OFFSET_LEN % 8 == 0);
2405 len = FLAC__STREAM_METADATA_CUESHEET_INDEX_OFFSET_LEN / 8;
2406 pack_uint64_(index->offset, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002407 if(write_cb(buffer, 1, len, handle) != len)
2408 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002409
2410 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_INDEX_NUMBER_LEN % 8 == 0);
2411 len = FLAC__STREAM_METADATA_CUESHEET_INDEX_NUMBER_LEN / 8;
2412 pack_uint32_(index->number, buffer, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002413 if(write_cb(buffer, 1, len, handle) != len)
2414 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002415
2416 FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN % 8 == 0);
2417 len = FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN / 8;
2418 memset(buffer, 0, len);
Josh Coalson463a22c2004-07-11 06:41:00 +00002419 if(write_cb(buffer, 1, len, handle) != len)
2420 return false;
Josh Coalsone4869382002-11-15 05:41:48 +00002421 }
2422 }
2423
Josh Coalson463a22c2004-07-11 06:41:00 +00002424 return true;
Josh Coalsone4869382002-11-15 05:41:48 +00002425}
2426
Josh Coalson463a22c2004-07-11 06:41:00 +00002427FLAC__bool write_metadata_block_data_unknown_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Write write_cb, const FLAC__StreamMetadata_Unknown *block, unsigned block_length)
Josh Coalsond0609472003-01-10 05:37:13 +00002428{
Josh Coalson463a22c2004-07-11 06:41:00 +00002429 if(write_cb(block->data, 1, block_length, handle) != block_length)
2430 return false;
Josh Coalsond0609472003-01-10 05:37:13 +00002431
Josh Coalson463a22c2004-07-11 06:41:00 +00002432 return true;
Josh Coalsond0609472003-01-10 05:37:13 +00002433}
2434
Josh Coalsoncc682512002-06-08 04:53:42 +00002435FLAC__bool write_metadata_block_stationary_(FLAC__Metadata_SimpleIterator *iterator, const FLAC__StreamMetadata *block)
Josh Coalsoncae58732002-05-04 17:30:32 +00002436{
2437 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2438 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2439 return false;
2440 }
2441
2442 if(!write_metadata_block_header_(iterator->file, &iterator->status, block))
2443 return false;
2444
2445 if(!write_metadata_block_data_(iterator->file, &iterator->status, block))
2446 return false;
2447
2448 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2449 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2450 return false;
2451 }
2452
2453 return read_metadata_block_header_(iterator);
2454}
2455
Josh Coalsoncc682512002-06-08 04:53:42 +00002456FLAC__bool write_metadata_block_stationary_with_padding_(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block, unsigned padding_length, FLAC__bool padding_is_last)
Josh Coalsoncae58732002-05-04 17:30:32 +00002457{
Josh Coalsoncc682512002-06-08 04:53:42 +00002458 FLAC__StreamMetadata *padding;
Josh Coalsoncae58732002-05-04 17:30:32 +00002459
2460 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2461 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2462 return false;
2463 }
2464
2465 block->is_last = false;
2466
2467 if(!write_metadata_block_header_(iterator->file, &iterator->status, block))
2468 return false;
2469
2470 if(!write_metadata_block_data_(iterator->file, &iterator->status, block))
2471 return false;
2472
2473 if(0 == (padding = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING)))
2474 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2475
2476 padding->is_last = padding_is_last;
2477 padding->length = padding_length;
2478
2479 if(!write_metadata_block_header_(iterator->file, &iterator->status, padding)) {
2480 FLAC__metadata_object_delete(padding);
2481 return false;
2482 }
2483
2484 if(!write_metadata_block_data_(iterator->file, &iterator->status, padding)) {
2485 FLAC__metadata_object_delete(padding);
2486 return false;
2487 }
2488
2489 FLAC__metadata_object_delete(padding);
2490
2491 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2492 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2493 return false;
2494 }
2495
2496 return read_metadata_block_header_(iterator);
2497}
2498
Josh Coalsoncc682512002-06-08 04:53:42 +00002499FLAC__bool rewrite_whole_file_(FLAC__Metadata_SimpleIterator *iterator, FLAC__StreamMetadata *block, FLAC__bool append)
Josh Coalsoncae58732002-05-04 17:30:32 +00002500{
2501 FILE *tempfile;
2502 char *tempfilename;
Josh Coalson6b02a752002-05-10 06:42:02 +00002503 int fixup_is_last_code = 0; /* 0 => no need to change any is_last flags */
2504 long fixup_is_last_flag_offset = -1;
Josh Coalsoncae58732002-05-04 17:30:32 +00002505
Josh Coalson51ee4652002-05-09 05:46:23 +00002506 FLAC__ASSERT(0 != block || append == false);
2507
Josh Coalson6b02a752002-05-10 06:42:02 +00002508 if(iterator->is_last) {
2509 if(append) {
2510 fixup_is_last_code = 1; /* 1 => clear the is_last flag at the following offset */
2511 fixup_is_last_flag_offset = iterator->offset[iterator->depth];
2512 }
2513 else if(0 == block) {
2514 simple_iterator_push_(iterator);
2515 if(!FLAC__metadata_simple_iterator_prev(iterator)) {
2516 (void)simple_iterator_pop_(iterator);
2517 return false;
2518 }
2519 fixup_is_last_code = -1; /* -1 => set the is_last the flag at the following offset */
2520 fixup_is_last_flag_offset = iterator->offset[iterator->depth];
2521 if(!simple_iterator_pop_(iterator))
2522 return false;
2523 }
2524 }
2525
Josh Coalsoncae58732002-05-04 17:30:32 +00002526 if(!simple_iterator_copy_file_prefix_(iterator, &tempfile, &tempfilename, append))
2527 return false;
2528
2529 if(0 != block) {
2530 if(!write_metadata_block_header_(tempfile, &iterator->status, block)) {
2531 cleanup_tempfile_(&tempfile, &tempfilename);
2532 return false;
2533 }
2534
2535 if(!write_metadata_block_data_(tempfile, &iterator->status, block)) {
2536 cleanup_tempfile_(&tempfile, &tempfilename);
2537 return false;
2538 }
2539 }
2540
Josh Coalson6b02a752002-05-10 06:42:02 +00002541 if(!simple_iterator_copy_file_postfix_(iterator, &tempfile, &tempfilename, fixup_is_last_code, fixup_is_last_flag_offset, block==0))
Josh Coalsoncae58732002-05-04 17:30:32 +00002542 return false;
2543
Josh Coalson51ee4652002-05-09 05:46:23 +00002544 if(append)
2545 return FLAC__metadata_simple_iterator_next(iterator);
2546
Josh Coalsoncae58732002-05-04 17:30:32 +00002547 return true;
2548}
2549
Josh Coalsoncc682512002-06-08 04:53:42 +00002550void simple_iterator_push_(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00002551{
2552 FLAC__ASSERT(iterator->depth+1 < SIMPLE_ITERATOR_MAX_PUSH_DEPTH);
2553 iterator->offset[iterator->depth+1] = iterator->offset[iterator->depth];
2554 iterator->depth++;
2555}
2556
Josh Coalsoncc682512002-06-08 04:53:42 +00002557FLAC__bool simple_iterator_pop_(FLAC__Metadata_SimpleIterator *iterator)
Josh Coalsoncae58732002-05-04 17:30:32 +00002558{
2559 FLAC__ASSERT(iterator->depth > 0);
2560 iterator->depth--;
2561 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2562 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2563 return false;
2564 }
2565
2566 return read_metadata_block_header_(iterator);
2567}
2568
Josh Coalsonea25cb32004-07-10 00:39:01 +00002569/* return meanings:
2570 * 0: ok
2571 * 1: read error
2572 * 2: seek error
2573 * 3: not a FLAC file
2574 */
2575unsigned seek_to_first_metadata_block_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Seek seek_cb)
Josh Coalsoncae58732002-05-04 17:30:32 +00002576{
Josh Coalsona7b673c2002-05-25 02:13:38 +00002577 FLAC__byte buffer[4];
2578 size_t n;
2579 unsigned i;
Josh Coalsoncae58732002-05-04 17:30:32 +00002580
Josh Coalsona7b673c2002-05-25 02:13:38 +00002581 FLAC__ASSERT(FLAC__STREAM_SYNC_LENGTH == 4);
Josh Coalsoncae58732002-05-04 17:30:32 +00002582
Josh Coalsona7b673c2002-05-25 02:13:38 +00002583 /* skip any id3v2 tag */
2584 errno = 0;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002585 n = read_cb(buffer, 1, 4, handle);
Josh Coalsona7b673c2002-05-25 02:13:38 +00002586 if(errno)
Josh Coalsoncae58732002-05-04 17:30:32 +00002587 return 1;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002588 else if(n != 4)
2589 return 3;
Josh Coalsona7b673c2002-05-25 02:13:38 +00002590 else if(0 == memcmp(buffer, "ID3", 3)) {
2591 unsigned tag_length = 0;
2592
2593 /* skip to the tag length */
Josh Coalsonea25cb32004-07-10 00:39:01 +00002594 if(seek_cb(handle, 2, SEEK_CUR) < 0)
2595 return 2;
Josh Coalsona7b673c2002-05-25 02:13:38 +00002596
2597 /* read the length */
2598 for(i = 0; i < 4; i++) {
Josh Coalsonea25cb32004-07-10 00:39:01 +00002599 if(read_cb(buffer, 1, 1, handle) < 1 || buffer[0] & 0x80)
Josh Coalsona7b673c2002-05-25 02:13:38 +00002600 return 1;
2601 tag_length <<= 7;
2602 tag_length |= (buffer[0] & 0x7f);
2603 }
2604
2605 /* skip the rest of the tag */
Josh Coalsonea25cb32004-07-10 00:39:01 +00002606 if(seek_cb(handle, tag_length, SEEK_CUR) < 0)
2607 return 2;
Josh Coalsona7b673c2002-05-25 02:13:38 +00002608
2609 /* read the stream sync code */
2610 errno = 0;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002611 n = read_cb(buffer, 1, 4, handle);
Josh Coalsona7b673c2002-05-25 02:13:38 +00002612 if(errno)
2613 return 1;
Josh Coalsonea25cb32004-07-10 00:39:01 +00002614 else if(n != 4)
2615 return 3;
Josh Coalsoncae58732002-05-04 17:30:32 +00002616 }
Josh Coalsona7b673c2002-05-25 02:13:38 +00002617
2618 /* check for the fLaC signature */
2619 if(0 == memcmp(FLAC__STREAM_SYNC_STRING, buffer, FLAC__STREAM_SYNC_LENGTH))
2620 return 0;
2621 else
Josh Coalsonea25cb32004-07-10 00:39:01 +00002622 return 3;
2623}
2624
2625unsigned seek_to_first_metadata_block_(FILE *f)
2626{
2627 return seek_to_first_metadata_block_cb_((FLAC__IOHandle)f, (FLAC__IOCallback_Read)fread, fseek_wrapper_);
Josh Coalsoncae58732002-05-04 17:30:32 +00002628}
2629
Josh Coalsoncc682512002-06-08 04:53:42 +00002630FLAC__bool simple_iterator_copy_file_prefix_(FLAC__Metadata_SimpleIterator *iterator, FILE **tempfile, char **tempfilename, FLAC__bool append)
Josh Coalsoncae58732002-05-04 17:30:32 +00002631{
Josh Coalson72943092002-05-22 05:35:43 +00002632 const long offset_end = append? iterator->offset[iterator->depth] + (long)FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length : iterator->offset[iterator->depth];
Josh Coalsoncae58732002-05-04 17:30:32 +00002633
2634 if(0 != fseek(iterator->file, 0, SEEK_SET)) {
2635 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2636 return false;
2637 }
2638 if(!open_tempfile_(iterator->filename, iterator->tempfile_path_prefix, tempfile, tempfilename, &iterator->status)) {
2639 cleanup_tempfile_(tempfile, tempfilename);
2640 return false;
2641 }
2642 if(!copy_n_bytes_from_file_(iterator->file, *tempfile, offset_end, &iterator->status)) {
2643 cleanup_tempfile_(tempfile, tempfilename);
2644 return false;
2645 }
2646
2647 return true;
2648}
2649
Josh Coalsoncc682512002-06-08 04:53:42 +00002650FLAC__bool simple_iterator_copy_file_postfix_(FLAC__Metadata_SimpleIterator *iterator, FILE **tempfile, char **tempfilename, int fixup_is_last_code, long fixup_is_last_flag_offset, FLAC__bool backup)
Josh Coalsoncae58732002-05-04 17:30:32 +00002651{
2652 long save_offset = iterator->offset[iterator->depth]; /*@@@ 2G limit */
2653 FLAC__ASSERT(0 != *tempfile);
2654
Josh Coalsoneb643a22002-05-20 05:58:50 +00002655 if(0 != fseek(iterator->file, save_offset + FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length, SEEK_SET)) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002656 cleanup_tempfile_(tempfile, tempfilename);
2657 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2658 return false;
2659 }
2660 if(!copy_remaining_bytes_from_file_(iterator->file, *tempfile, &iterator->status)) {
2661 cleanup_tempfile_(tempfile, tempfilename);
2662 return false;
2663 }
2664
Josh Coalson6b02a752002-05-10 06:42:02 +00002665 if(fixup_is_last_code != 0) {
2666 /*
2667 * if code == 1, it means a block was appended to the end so
2668 * we have to clear the is_last flag of the previous block
2669 * if code == -1, it means the last block was deleted so
2670 * we have to set the is_last flag of the previous block
2671 */
Josh Coalsonf0a8c4f2003-01-15 03:17:51 +00002672 /* MAGIC NUMBERs here; we know the is_last flag is the high bit of the byte at this location */
Josh Coalsoncae58732002-05-04 17:30:32 +00002673 FLAC__byte x;
2674 if(0 != fseek(*tempfile, fixup_is_last_flag_offset, SEEK_SET)) {
2675 cleanup_tempfile_(tempfile, tempfilename);
2676 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2677 return false;
2678 }
2679 if(fread(&x, 1, 1, *tempfile) != 1) {
2680 cleanup_tempfile_(tempfile, tempfilename);
2681 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2682 return false;
2683 }
Josh Coalson6b02a752002-05-10 06:42:02 +00002684 if(fixup_is_last_code > 0) {
2685 FLAC__ASSERT(x & 0x80);
2686 x &= 0x7f;
2687 }
2688 else {
2689 FLAC__ASSERT(!(x & 0x80));
2690 x |= 0x80;
2691 }
Josh Coalsoncae58732002-05-04 17:30:32 +00002692 if(0 != fseek(*tempfile, fixup_is_last_flag_offset, SEEK_SET)) {
2693 cleanup_tempfile_(tempfile, tempfilename);
2694 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2695 return false;
2696 }
Josh Coalsonf91251d2002-12-28 07:04:49 +00002697 if(local__fwrite(&x, 1, 1, *tempfile) != 1) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002698 cleanup_tempfile_(tempfile, tempfilename);
2699 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2700 return false;
2701 }
2702 }
2703
2704 (void)fclose(iterator->file);
2705
2706 if(!transport_tempfile_(iterator->filename, tempfile, tempfilename, &iterator->status))
2707 return false;
2708
2709 if(iterator->has_stats)
2710 set_file_stats_(iterator->filename, &iterator->stats);
2711
Josh Coalson3ac66932002-08-30 05:41:31 +00002712 if(!simple_iterator_prime_input_(iterator, !iterator->is_writable))
Josh Coalsoncae58732002-05-04 17:30:32 +00002713 return false;
2714 if(backup) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00002715 while(iterator->offset[iterator->depth] + (long)FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length < save_offset)
Josh Coalsoncae58732002-05-04 17:30:32 +00002716 if(!FLAC__metadata_simple_iterator_next(iterator))
2717 return false;
2718 return true;
2719 }
2720 else {
2721 /* move the iterator to it's original block faster by faking a push, then doing a pop_ */
2722 FLAC__ASSERT(iterator->depth == 0);
2723 iterator->offset[0] = save_offset;
2724 iterator->depth++;
2725 return simple_iterator_pop_(iterator);
2726 }
2727}
2728
Josh Coalsoncc682512002-06-08 04:53:42 +00002729FLAC__bool copy_n_bytes_from_file_(FILE *file, FILE *tempfile, unsigned bytes/*@@@ 4G limit*/, FLAC__Metadata_SimpleIteratorStatus *status)
Josh Coalsoncae58732002-05-04 17:30:32 +00002730{
2731 FLAC__byte buffer[8192];
2732 unsigned n;
2733
2734 while(bytes > 0) {
2735 n = min(sizeof(buffer), bytes);
2736 if(fread(buffer, 1, n, file) != n) {
2737 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2738 return false;
2739 }
Josh Coalsonf91251d2002-12-28 07:04:49 +00002740 if(local__fwrite(buffer, 1, n, tempfile) != n) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002741 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2742 return false;
2743 }
2744 bytes -= n;
2745 }
2746
2747 return true;
2748}
2749
Josh Coalson463a22c2004-07-11 06:41:00 +00002750FLAC__bool copy_n_bytes_from_file_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOHandle temp_handle, FLAC__IOCallback_Write temp_write_cb, unsigned bytes/*@@@ 4G limit*/, FLAC__Metadata_SimpleIteratorStatus *status)
2751{
2752 FLAC__byte buffer[8192];
2753 unsigned n;
2754
2755 while(bytes > 0) {
2756 n = min(sizeof(buffer), bytes);
2757 if(read_cb(buffer, 1, n, handle) != n) {
2758 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2759 return false;
2760 }
2761 if(temp_write_cb(buffer, 1, n, temp_handle) != n) {
2762 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2763 return false;
2764 }
2765 bytes -= n;
2766 }
2767
2768 return true;
2769}
2770
Josh Coalsoncc682512002-06-08 04:53:42 +00002771FLAC__bool copy_remaining_bytes_from_file_(FILE *file, FILE *tempfile, FLAC__Metadata_SimpleIteratorStatus *status)
Josh Coalsoncae58732002-05-04 17:30:32 +00002772{
2773 FLAC__byte buffer[8192];
2774 size_t n;
2775
2776 while(!feof(file)) {
2777 n = fread(buffer, 1, sizeof(buffer), file);
2778 if(n == 0 && !feof(file)) {
2779 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2780 return false;
2781 }
Josh Coalsonf91251d2002-12-28 07:04:49 +00002782 if(n > 0 && local__fwrite(buffer, 1, n, tempfile) != n) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002783 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2784 return false;
2785 }
2786 }
2787
2788 return true;
2789}
2790
Josh Coalson463a22c2004-07-11 06:41:00 +00002791FLAC__bool copy_remaining_bytes_from_file_cb_(FLAC__IOHandle handle, FLAC__IOCallback_Read read_cb, FLAC__IOCallback_Eof eof_cb, FLAC__IOHandle temp_handle, FLAC__IOCallback_Write temp_write_cb, FLAC__Metadata_SimpleIteratorStatus *status)
2792{
2793 FLAC__byte buffer[8192];
2794 size_t n;
2795
2796 while(!eof_cb(handle)) {
2797 n = read_cb(buffer, 1, sizeof(buffer), handle);
2798 if(n == 0 && !eof_cb(handle)) {
2799 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2800 return false;
2801 }
2802 if(n > 0 && temp_write_cb(buffer, 1, n, temp_handle) != n) {
2803 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2804 return false;
2805 }
2806 }
2807
2808 return true;
2809}
2810
Josh Coalsoncc682512002-06-08 04:53:42 +00002811FLAC__bool open_tempfile_(const char *filename, const char *tempfile_path_prefix, FILE **tempfile, char **tempfilename, FLAC__Metadata_SimpleIteratorStatus *status)
Josh Coalsoncae58732002-05-04 17:30:32 +00002812{
2813 static const char *tempfile_suffix = ".metadata_edit";
2814 if(0 == tempfile_path_prefix) {
Josh Coalson5e3fcb22002-11-06 07:11:26 +00002815 if(0 == (*tempfilename = (char*)malloc(strlen(filename) + strlen(tempfile_suffix) + 1))) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002816 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2817 return false;
2818 }
2819 strcpy(*tempfilename, filename);
2820 strcat(*tempfilename, tempfile_suffix);
2821 }
2822 else {
2823 const char *p = strrchr(filename, '/');
2824 if(0 == p)
2825 p = filename;
2826 else
2827 p++;
2828
Josh Coalson5e3fcb22002-11-06 07:11:26 +00002829 if(0 == (*tempfilename = (char*)malloc(strlen(tempfile_path_prefix) + 1 + strlen(p) + strlen(tempfile_suffix) + 1))) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002830 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2831 return false;
2832 }
2833 strcpy(*tempfilename, tempfile_path_prefix);
2834 strcat(*tempfilename, "/");
2835 strcat(*tempfilename, p);
2836 strcat(*tempfilename, tempfile_suffix);
2837 }
2838
Josh Coalson51ee4652002-05-09 05:46:23 +00002839 if(0 == (*tempfile = fopen(*tempfilename, "w+b"))) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002840 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE;
2841 return false;
2842 }
2843
2844 return true;
2845}
2846
Josh Coalsoncc682512002-06-08 04:53:42 +00002847FLAC__bool transport_tempfile_(const char *filename, FILE **tempfile, char **tempfilename, FLAC__Metadata_SimpleIteratorStatus *status)
Josh Coalsoncae58732002-05-04 17:30:32 +00002848{
2849 FLAC__ASSERT(0 != filename);
2850 FLAC__ASSERT(0 != tempfile);
2851 FLAC__ASSERT(0 != *tempfile);
2852 FLAC__ASSERT(0 != tempfilename);
2853 FLAC__ASSERT(0 != *tempfilename);
2854 FLAC__ASSERT(0 != status);
2855
2856 (void)fclose(*tempfile);
2857 *tempfile = 0;
Josh Coalsond57c8d32002-06-11 06:15:28 +00002858
2859#if defined _MSC_VER || defined __MINGW32__
2860 if(unlink(filename) < 0) {
2861 cleanup_tempfile_(tempfile, tempfilename);
2862 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_UNLINK_ERROR;
2863 return false;
2864 }
2865#endif
2866
Josh Coalsoncae58732002-05-04 17:30:32 +00002867 /*@@@ to fully support the tempfile_path_prefix we need to update this piece to actually copy across filesystems instead of just rename(): */
2868 if(0 != rename(*tempfilename, filename)) {
2869 cleanup_tempfile_(tempfile, tempfilename);
2870 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_RENAME_ERROR;
2871 return false;
2872 }
2873
2874 cleanup_tempfile_(tempfile, tempfilename);
2875
2876 return true;
2877}
2878
2879void cleanup_tempfile_(FILE **tempfile, char **tempfilename)
2880{
2881 if(0 != *tempfile) {
2882 (void)fclose(*tempfile);
2883 *tempfile = 0;
2884 }
2885
2886 if(0 != *tempfilename) {
2887 (void)unlink(*tempfilename);
2888 free(*tempfilename);
2889 *tempfilename = 0;
2890 }
2891}
2892
2893FLAC__bool get_file_stats_(const char *filename, struct stat *stats)
2894{
2895 FLAC__ASSERT(0 != filename);
2896 FLAC__ASSERT(0 != stats);
2897 return (0 == stat(filename, stats));
2898}
2899
2900void set_file_stats_(const char *filename, struct stat *stats)
2901{
2902 struct utimbuf srctime;
2903
2904 FLAC__ASSERT(0 != filename);
2905 FLAC__ASSERT(0 != stats);
2906
2907 srctime.actime = stats->st_atime;
2908 srctime.modtime = stats->st_mtime;
2909 (void)chmod(filename, stats->st_mode);
2910 (void)utime(filename, &srctime);
2911#if !defined _MSC_VER && !defined __MINGW32__
2912 (void)chown(filename, stats->st_uid, -1);
2913 (void)chown(filename, -1, stats->st_gid);
2914#endif
2915}
2916
Josh Coalsonea25cb32004-07-10 00:39:01 +00002917/* @@@ WATCHOUT @@@
2918 * We cast FLAC__int64 to long and use fseek()/ftell() because
2919 * none of our operations on metadata is ever likely to go past
2920 * 2 gigabytes.
2921 */
2922int fseek_wrapper_(FLAC__IOHandle handle, FLAC__int64 offset, int whence)
2923{
2924 FLAC__ASSERT(offset <= 0x7fffffff);
2925 return fseek((FILE*)handle, (long)offset, whence);
2926}
2927
2928FLAC__int64 ftell_wrapper_(FLAC__IOHandle handle)
2929{
2930 return (long)ftell((FILE*)handle);
2931}
2932
Josh Coalsoncc682512002-06-08 04:53:42 +00002933FLAC__Metadata_ChainStatus get_equivalent_status_(FLAC__Metadata_SimpleIteratorStatus status)
Josh Coalsoncae58732002-05-04 17:30:32 +00002934{
2935 switch(status) {
2936 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK:
2937 return FLAC__METADATA_CHAIN_STATUS_OK;
2938 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT:
2939 return FLAC__METADATA_CHAIN_STATUS_ILLEGAL_INPUT;
2940 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE:
2941 return FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
2942 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_A_FLAC_FILE:
2943 return FLAC__METADATA_CHAIN_STATUS_NOT_A_FLAC_FILE;
2944 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_WRITABLE:
2945 return FLAC__METADATA_CHAIN_STATUS_NOT_WRITABLE;
Josh Coalsona7b673c2002-05-25 02:13:38 +00002946 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_BAD_METADATA:
2947 return FLAC__METADATA_CHAIN_STATUS_BAD_METADATA;
Josh Coalsoncae58732002-05-04 17:30:32 +00002948 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR:
2949 return FLAC__METADATA_CHAIN_STATUS_READ_ERROR;
2950 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR:
2951 return FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
2952 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR:
2953 return FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR;
2954 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_RENAME_ERROR:
2955 return FLAC__METADATA_CHAIN_STATUS_RENAME_ERROR;
2956 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_UNLINK_ERROR:
2957 return FLAC__METADATA_CHAIN_STATUS_UNLINK_ERROR;
2958 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR:
2959 return FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
2960 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_INTERNAL_ERROR:
2961 default:
2962 return FLAC__METADATA_CHAIN_STATUS_INTERNAL_ERROR;
2963 }
2964}