blob: 64b1d3caa6344372741b3ba2ee3fe7ed65cbf192 [file] [log] [blame]
Josh Coalsoncae58732002-05-04 17:30:32 +00001/* libFLAC - Free Lossless Audio Codec library
2 * Copyright (C) 2001,2002 Josh Coalson
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Library General Public
6 * License as published by the Free Software Foundation; either
7 * version 2 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Library General Public License for more details.
13 *
14 * You should have received a copy of the GNU Library General Public
15 * License along with this library; if not, write to the
16 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17 * Boston, MA 02111-1307, USA.
18 */
19
20#include <errno.h>
21#include <stdio.h>
22#include <stdlib.h>
23#include <string.h>
24
25#if defined _MSC_VER || defined __MINGW32__
26#include <sys/utime.h> /* for utime() */
27#include <io.h> /* for chmod() */
28#else
29#include <sys/types.h> /* some flavors of BSD (like OS X) require this to get time_t */
30#include <utime.h> /* for utime() */
31#include <unistd.h> /* for chown(), unlink() */
32#endif
33#include <sys/stat.h> /* for stat(), maybe chmod() */
34
35#include "FLAC/assert.h"
36#include "FLAC/metadata.h"
37#include "FLAC/file_decoder.h"
38
39#ifdef max
40#undef max
41#endif
42#define max(a,b) ((a)>(b)?(a):(b))
43#ifdef min
44#undef min
45#endif
46#define min(a,b) ((a)<(b)?(a):(b))
47
48
49/****************************************************************************
50 *
51 * Local function declarations
52 *
53 ***************************************************************************/
54
55/* this one should probably not go in the public interface: */
56static void FLAC__metadata_object_delete_data_(FLAC__StreamMetaData *object);
57
58static void pack_uint32_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes);
59static void pack_uint32_little_endian_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes);
60static void pack_uint64_(FLAC__uint64 val, FLAC__byte *b, unsigned bytes);
61static FLAC__uint32 unpack_uint32_(FLAC__byte *b, unsigned bytes);
62static FLAC__uint32 unpack_uint32_little_endian_(FLAC__byte *b, unsigned bytes);
63static FLAC__uint64 unpack_uint64_(FLAC__byte *b, unsigned bytes);
64
65static FLAC__bool read_metadata_block_header_(FLAC__MetaData_SimpleIterator *iterator);
66static FLAC__bool read_metadata_block_data_(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block);
67static FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_streaminfo_(FILE *file, FLAC__StreamMetaData_StreamInfo *block);
68static FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_padding_(FILE *file, FLAC__StreamMetaData_Padding *block, unsigned block_length);
69static FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_application_(FILE *file, FLAC__StreamMetaData_Application *block, unsigned block_length);
70static FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_seektable_(FILE *file, FLAC__StreamMetaData_SeekTable *block, unsigned block_length);
71static FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_entry_(FILE *file, FLAC__StreamMetaData_VorbisComment_Entry *entry);
72static FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_(FILE *file, FLAC__StreamMetaData_VorbisComment *block);
73
74static FLAC__bool write_metadata_block_header_(FILE *file, FLAC__MetaData_SimpleIteratorStatus *status, const FLAC__StreamMetaData *block);
75static FLAC__bool write_metadata_block_data_(FILE *file, FLAC__MetaData_SimpleIteratorStatus *status, const FLAC__StreamMetaData *block);
76static FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_streaminfo_(FILE *file, const FLAC__StreamMetaData_StreamInfo *block);
77static FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_padding_(FILE *file, const FLAC__StreamMetaData_Padding *block, unsigned block_length);
78static FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_application_(FILE *file, const FLAC__StreamMetaData_Application *block, unsigned block_length);
79static FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_seektable_(FILE *file, const FLAC__StreamMetaData_SeekTable *block);
80static FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_vorbis_comment_(FILE *file, const FLAC__StreamMetaData_VorbisComment *block);
81static FLAC__bool write_metadata_block_stationary_(FLAC__MetaData_SimpleIterator *iterator, const FLAC__StreamMetaData *block);
82static FLAC__bool write_metadata_block_stationary_with_padding_(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block, unsigned padding_length, FLAC__bool padding_is_last);
83static FLAC__bool rewrite_whole_file_(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block, FLAC__bool append);
84
85static FLAC__bool chain_rewrite_chain_(FLAC__MetaData_Chain *chain);
86static FLAC__bool chain_rewrite_file_(FLAC__MetaData_Chain *chain, const char *tempfile_path_prefix);
87
88static void simple_iterator_push_(FLAC__MetaData_SimpleIterator *iterator);
89static FLAC__bool simple_iterator_pop_(FLAC__MetaData_SimpleIterator *iterator);
90
91/* return 0 if OK, 1 if read error, 2 if not a FLAC file */
92static unsigned seek_to_first_metadata_block_(FILE *f);
93
94static FLAC__bool simple_iterator_copy_file_prefix_(FLAC__MetaData_SimpleIterator *iterator, FILE **tempfile, char **tempfilename, FLAC__bool append);
Josh Coalson6b02a752002-05-10 06:42:02 +000095static 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 +000096
97static FLAC__bool copy_n_bytes_from_file_(FILE *file, FILE *tempfile, unsigned bytes/*@@@ 4G limit*/, FLAC__MetaData_SimpleIteratorStatus *status);
98static FLAC__bool copy_remaining_bytes_from_file_(FILE *file, FILE *tempfile, FLAC__MetaData_SimpleIteratorStatus *status);
99
100static FLAC__bool open_tempfile_(const char *filename, const char *tempfile_path_prefix, FILE **tempfile, char **tempfilename, FLAC__MetaData_SimpleIteratorStatus *status);
101static FLAC__bool transport_tempfile_(const char *filename, FILE **tempfile, char **tempfilename, FLAC__MetaData_SimpleIteratorStatus *status);
102static void cleanup_tempfile_(FILE **tempfile, char **tempfilename);
103
104static FLAC__bool get_file_stats_(const char *filename, struct stat *stats);
105static void set_file_stats_(const char *filename, struct stat *stats);
106
107static FLAC__MetaData_ChainStatus get_equivalent_status_(FLAC__MetaData_SimpleIteratorStatus status);
108
109
110/****************************************************************************
111 *
112 * Level 0 implementation
113 *
114 ***************************************************************************/
115
116static FLAC__StreamDecoderWriteStatus write_callback_(const FLAC__FileDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 *buffer[], void *client_data);
117static void metadata_callback_(const FLAC__FileDecoder *decoder, const FLAC__StreamMetaData *metadata, void *client_data);
118static void error_callback_(const FLAC__FileDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data);
119
120FLAC__bool FLAC__metadata_get_streaminfo(const char *filename, FLAC__StreamMetaData_StreamInfo *streaminfo)
121{
122 FLAC__FileDecoder *decoder = FLAC__file_decoder_new();
123
124 if(0 == decoder)
125 return false;
126
127 FLAC__file_decoder_set_md5_checking(decoder, false);
128 FLAC__file_decoder_set_filename(decoder, filename);
129 FLAC__file_decoder_set_write_callback(decoder, write_callback_);
130 FLAC__file_decoder_set_metadata_callback(decoder, metadata_callback_);
131 FLAC__file_decoder_set_error_callback(decoder, error_callback_);
132 FLAC__file_decoder_set_client_data(decoder, &streaminfo);
133
134 if(FLAC__file_decoder_init(decoder) != FLAC__FILE_DECODER_OK)
135 return false;
136
137 if(!FLAC__file_decoder_process_metadata(decoder))
138 return false;
139
Josh Coalson51ee4652002-05-09 05:46:23 +0000140 FLAC__file_decoder_finish(decoder);
Josh Coalsoncae58732002-05-04 17:30:32 +0000141 FLAC__file_decoder_delete(decoder);
142
143 return 0 != streaminfo; /* the metadata_callback_() will set streaminfo to 0 on an error */
144}
145
146FLAC__StreamDecoderWriteStatus write_callback_(const FLAC__FileDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 *buffer[], void *client_data)
147{
148 (void)decoder, (void)frame, (void)buffer, (void)client_data;
149
150 return FLAC__STREAM_DECODER_WRITE_CONTINUE;
151}
152
153void metadata_callback_(const FLAC__FileDecoder *decoder, const FLAC__StreamMetaData *metadata, void *client_data)
154{
155 FLAC__StreamMetaData_StreamInfo **streaminfo = (FLAC__StreamMetaData_StreamInfo **)client_data;
156 (void)decoder;
157
158 if(metadata->type == FLAC__METADATA_TYPE_STREAMINFO && 0 != *streaminfo)
159 **streaminfo = metadata->data.stream_info;
160}
161
162void error_callback_(const FLAC__FileDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
163{
164 FLAC__StreamMetaData_StreamInfo **streaminfo = (FLAC__StreamMetaData_StreamInfo **)client_data;
165 (void)decoder;
166
167 if(status != FLAC__STREAM_DECODER_ERROR_LOST_SYNC)
168 *streaminfo = 0;
169}
170
171
172/****************************************************************************
173 *
174 * Level 1 implementation
175 *
176 ***************************************************************************/
177
178#define SIMPLE_ITERATOR_MAX_PUSH_DEPTH (1+4)
179/* 1 for initial offset, +4 for our own personal use */
180
181struct FLAC__MetaData_SimpleIterator {
182 FILE *file;
183 char *filename, *tempfile_path_prefix;
184 struct stat stats;
185 FLAC__bool has_stats;
186 FLAC__bool is_writable;
187 FLAC__MetaData_SimpleIteratorStatus status;
188 /*@@@ 2G limits here because of the offset type: */
189 long offset[SIMPLE_ITERATOR_MAX_PUSH_DEPTH];
190 long first_offset; /* this is the offset to the STREAMINFO block */
191 unsigned depth;
192 /* this is the metadata block header of the current block we are pointing to: */
193 FLAC__bool is_last;
194 FLAC__MetaDataType type;
195 unsigned length;
196};
197
198const char *FLAC__MetaData_SimpleIteratorStatusString[] = {
199 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK",
200 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT",
201 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE",
202 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_A_FLAC_FILE",
203 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_WRITABLE",
204 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR",
205 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR",
206 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR",
207 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_RENAME_ERROR",
208 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_UNLINK_ERROR",
209 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR",
210 "FLAC__METADATA_SIMPLE_ITERATOR_STATUS_INTERNAL_ERROR"
211};
212
213
214FLAC__MetaData_SimpleIterator *FLAC__metadata_simple_iterator_new()
215{
216 FLAC__MetaData_SimpleIterator *iterator = malloc(sizeof(FLAC__MetaData_SimpleIterator));
217
218 if(0 != iterator) {
219 iterator->file = 0;
220 iterator->filename = 0;
221 iterator->tempfile_path_prefix = 0;
222 iterator->has_stats = false;
223 iterator->is_writable = false;
224 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
225 iterator->first_offset = iterator->offset[0] = -1;
226 iterator->depth = 0;
227 }
228
229 return iterator;
230}
231
232static void simple_iterator_free_guts_(FLAC__MetaData_SimpleIterator *iterator)
233{
234 FLAC__ASSERT(0 != iterator);
235
236 if(0 != iterator->file) {
237 fclose(iterator->file);
238 iterator->file = 0;
239 if(iterator->has_stats)
240 set_file_stats_(iterator->filename, &iterator->stats);
241 }
242 if(0 != iterator->filename) {
243 free(iterator->filename);
244 iterator->filename = 0;
245 }
246 if(0 != iterator->tempfile_path_prefix) {
247 free(iterator->tempfile_path_prefix);
248 iterator->tempfile_path_prefix = 0;
249 }
250}
251
252void FLAC__metadata_simple_iterator_delete(FLAC__MetaData_SimpleIterator *iterator)
253{
254 FLAC__ASSERT(0 != iterator);
255
256 simple_iterator_free_guts_(iterator);
257 free(iterator);
258}
259
260FLAC__MetaData_SimpleIteratorStatus FLAC__metadata_simple_iterator_status(FLAC__MetaData_SimpleIterator *iterator)
261{
262 FLAC__MetaData_SimpleIteratorStatus status;
263
264 FLAC__ASSERT(0 != iterator);
265
266 status = iterator->status;
267 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
268 return status;
269}
270
271static FLAC__bool simple_iterator_prime_input_(FLAC__MetaData_SimpleIterator *iterator)
272{
273 unsigned ret;
274
275 FLAC__ASSERT(0 != iterator);
276
277 iterator->is_writable = false;
278
279 if(0 == (iterator->file = fopen(iterator->filename, "r+b"))) {
280 if(errno == EACCES) {
281 if(0 == (iterator->file = fopen(iterator->filename, "rb"))) {
282 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE;
283 return false;
284 }
285 }
286 else {
287 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE;
288 return false;
289 }
290 }
291 else {
292 iterator->is_writable = true;
293 }
294
295 ret = seek_to_first_metadata_block_(iterator->file);
296 switch(ret) {
297 case 0:
298 iterator->depth = 0;
299 iterator->first_offset = iterator->offset[iterator->depth] = ftell(iterator->file);
300 return read_metadata_block_header_(iterator);
301 case 1:
302 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
303 return false;
304 case 2:
305 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_A_FLAC_FILE;
306 return false;
307 default:
308 FLAC__ASSERT(0);
309 return false;
310 }
311}
312
313#if 0
314@@@ If we decide to finish implementing this, put this comment back in metadata.h
315/*
316 * The 'tempfile_path_prefix' allows you to specify a directory where
317 * tempfiles should go. Remember that if your metadata edits cause the
318 * FLAC file to grow, the entire file will have to be rewritten. If
319 * 'tempfile_path_prefix' is NULL, the temp file will be written in the
320 * same directory as the original FLAC file. This makes replacing the
321 * original with the tempfile fast but requires extra space in the same
322 * partition for the tempfile. If space is a problem, you can pass a
323 * directory name belonging to a different partition in
324 * 'tempfile_path_prefix'. Note that you should use the forward slash
325 * '/' as the directory separator. A trailing slash is not needed; it
326 * will be added automatically.
327 */
328FLAC__bool FLAC__metadata_simple_iterator_init(FLAC__MetaData_SimpleIterator *iterator, const char *filename, FLAC__bool preserve_file_stats, const char *tempfile_path_prefix);
329#endif
330
331FLAC__bool FLAC__metadata_simple_iterator_init(FLAC__MetaData_SimpleIterator *iterator, const char *filename, FLAC__bool preserve_file_stats)
332{
333 const char *tempfile_path_prefix = 0; /*@@@ search for comments near 'rename(' for what it will take to finish implementing this */
334
335 FLAC__ASSERT(0 != iterator);
336 FLAC__ASSERT(0 != filename);
337
338 simple_iterator_free_guts_(iterator);
339
340 if(preserve_file_stats)
341 iterator->has_stats = get_file_stats_(filename, &iterator->stats);
342
343 if(0 == (iterator->filename = strdup(filename))) {
344 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
345 return false;
346 }
347 if(0 != tempfile_path_prefix && 0 == (iterator->tempfile_path_prefix = strdup(tempfile_path_prefix))) {
348 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
349 return false;
350 }
351
352 return simple_iterator_prime_input_(iterator);
353}
354
355FLAC__bool FLAC__metadata_simple_iterator_is_writable(const FLAC__MetaData_SimpleIterator *iterator)
356{
357 FLAC__ASSERT(0 != iterator);
358 FLAC__ASSERT(0 != iterator->file);
359
360 return iterator->is_writable;
361}
362
363FLAC__bool FLAC__metadata_simple_iterator_next(FLAC__MetaData_SimpleIterator *iterator)
364{
365 FLAC__ASSERT(0 != iterator);
366 FLAC__ASSERT(0 != iterator->file);
367
368 if(iterator->is_last)
369 return false;
370
371 if(0 != fseek(iterator->file, iterator->length, SEEK_CUR)) {
372 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
373 return false;
374 }
375
376 iterator->offset[iterator->depth] = ftell(iterator->file);
377
378 return read_metadata_block_header_(iterator);
379}
380
381FLAC__bool FLAC__metadata_simple_iterator_prev(FLAC__MetaData_SimpleIterator *iterator)
382{
383 long this_offset;
384
385 FLAC__ASSERT(0 != iterator);
386 FLAC__ASSERT(0 != iterator->file);
387
388 if(iterator->offset[iterator->depth] == iterator->first_offset)
389 return false;
390
391 if(0 != fseek(iterator->file, iterator->first_offset, SEEK_SET)) {
392 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
393 return false;
394 }
395 this_offset = iterator->first_offset;
396 if(!read_metadata_block_header_(iterator))
397 return false;
398
399 /* we ignore any error from ftell() and catch it in fseek() */
400 while(ftell(iterator->file) + (long)iterator->length < iterator->offset[iterator->depth]) {
401 if(0 != fseek(iterator->file, iterator->length, SEEK_CUR)) {
402 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
403 return false;
404 }
405 this_offset = ftell(iterator->file);
406 if(!read_metadata_block_header_(iterator))
407 return false;
408 }
409
410 iterator->offset[iterator->depth] = this_offset;
411
412 return true;
413}
414
415FLAC__MetaDataType FLAC__metadata_simple_iterator_get_block_type(const FLAC__MetaData_SimpleIterator *iterator)
416{
417 FLAC__ASSERT(0 != iterator);
418 FLAC__ASSERT(0 != iterator->file);
419
420 return iterator->type;
421}
422
423FLAC__StreamMetaData *FLAC__metadata_simple_iterator_get_block(FLAC__MetaData_SimpleIterator *iterator)
424{
425 FLAC__StreamMetaData *block = FLAC__metadata_object_new(iterator->type);
426
427 FLAC__ASSERT(0 != iterator);
428 FLAC__ASSERT(0 != iterator->file);
429
430 if(0 != block) {
431 block->is_last = iterator->is_last;
432 block->length = iterator->length;
433
434 if(!read_metadata_block_data_(iterator, block)) {
435 FLAC__metadata_object_delete(block);
436 return 0;
437 }
438
439 /* back up to the beginning of the block data to stay consistent */
Josh Coalsoneb643a22002-05-20 05:58:50 +0000440 if(0 != fseek(iterator->file, iterator->offset[iterator->depth] + FLAC__STREAM_METADATA_HEADER_LENGTH, SEEK_SET)) {
Josh Coalsoncae58732002-05-04 17:30:32 +0000441 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
442 FLAC__metadata_object_delete(block);
443 return 0;
444 }
445 }
446 else
447 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
448
449 return block;
450}
451
452FLAC__bool FLAC__metadata_simple_iterator_set_block(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block, FLAC__bool use_padding)
453{
Josh Coalson51ee4652002-05-09 05:46:23 +0000454 FLAC__ASSERT_DECLARATION(long debug_target_offset = iterator->offset[iterator->depth];)
455 FLAC__bool ret;
456
Josh Coalsoncae58732002-05-04 17:30:32 +0000457 FLAC__ASSERT(0 != iterator);
458 FLAC__ASSERT(0 != iterator->file);
459
460 if(!iterator->is_writable) {
461 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_WRITABLE;
462 return false;
463 }
464
465 if(iterator->type == FLAC__METADATA_TYPE_STREAMINFO || block->type == FLAC__METADATA_TYPE_STREAMINFO) {
466 if(iterator->type != block->type) {
467 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT;
468 return false;
469 }
470 }
471
472 block->is_last = iterator->is_last;
Josh Coalsoncae58732002-05-04 17:30:32 +0000473
474 if(iterator->length == block->length)
475 return write_metadata_block_stationary_(iterator, block);
476 else if(iterator->length > block->length) {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000477 if(use_padding && iterator->length >= FLAC__STREAM_METADATA_HEADER_LENGTH + block->length) {
478 ret = write_metadata_block_stationary_with_padding_(iterator, block, iterator->length - FLAC__STREAM_METADATA_HEADER_LENGTH - block->length, block->is_last);
Josh Coalson51ee4652002-05-09 05:46:23 +0000479 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000480 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000481 return ret;
482 }
483 else {
484 ret = rewrite_whole_file_(iterator, block, /*append=*/false);
Josh Coalson51ee4652002-05-09 05:46:23 +0000485 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000486 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000487 return ret;
488 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000489 }
490 else /* iterator->length < block->length */ {
491 unsigned padding_leftover = 0;
492 FLAC__bool padding_is_last = false;
493 if(use_padding) {
494 /* first see if we can even use padding */
495 if(iterator->is_last) {
496 use_padding = false;
497 }
498 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000499 const unsigned extra_padding_bytes_required = block->length - iterator->length;
Josh Coalsoncae58732002-05-04 17:30:32 +0000500 simple_iterator_push_(iterator);
501 if(!FLAC__metadata_simple_iterator_next(iterator)) {
502 (void)simple_iterator_pop_(iterator);
503 return false;
504 }
505 if(iterator->type != FLAC__METADATA_TYPE_PADDING) {
506 use_padding = false;
507 }
508 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000509 if(FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length == extra_padding_bytes_required) {
Josh Coalsoncae58732002-05-04 17:30:32 +0000510 padding_leftover = 0;
511 block->is_last = iterator->is_last;
512 }
Josh Coalson51ee4652002-05-09 05:46:23 +0000513 else if(iterator->length < extra_padding_bytes_required)
Josh Coalsoncae58732002-05-04 17:30:32 +0000514 use_padding = false;
515 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000516 padding_leftover = FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length - extra_padding_bytes_required;
Josh Coalsoncae58732002-05-04 17:30:32 +0000517 padding_is_last = iterator->is_last;
518 block->is_last = false;
519 }
520 }
521 if(!simple_iterator_pop_(iterator))
522 return false;
523 }
524 }
525 if(use_padding) {
Josh Coalson51ee4652002-05-09 05:46:23 +0000526 if(padding_leftover == 0) {
527 ret = write_metadata_block_stationary_(iterator, block);
Josh Coalson51ee4652002-05-09 05:46:23 +0000528 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000529 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000530 return ret;
531 }
532 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000533 FLAC__ASSERT(padding_leftover >= FLAC__STREAM_METADATA_HEADER_LENGTH);
534 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 +0000535 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000536 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000537 return ret;
538 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000539 }
540 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000541 ret = rewrite_whole_file_(iterator, block, /*append=*/false);
Josh Coalson51ee4652002-05-09 05:46:23 +0000542 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000543 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000544 return ret;
Josh Coalsoncae58732002-05-04 17:30:32 +0000545 }
546 }
547}
548
549FLAC__bool FLAC__metadata_simple_iterator_insert_block_after(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block, FLAC__bool use_padding)
550{
551 unsigned padding_leftover = 0;
552 FLAC__bool padding_is_last = false;
553
Josh Coalsoneb643a22002-05-20 05:58:50 +0000554 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 +0000555 FLAC__bool ret;
556
Josh Coalsoncae58732002-05-04 17:30:32 +0000557 FLAC__ASSERT(0 != iterator);
558 FLAC__ASSERT(0 != iterator->file);
559
560 if(!iterator->is_writable)
561 return false;
562
563 if(block->type == FLAC__METADATA_TYPE_STREAMINFO) {
564 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT;
565 return false;
566 }
567
Josh Coalson6b02a752002-05-10 06:42:02 +0000568 block->is_last = iterator->is_last;
Josh Coalson51ee4652002-05-09 05:46:23 +0000569
Josh Coalsoncae58732002-05-04 17:30:32 +0000570 if(use_padding) {
571 /* first see if we can even use padding */
572 if(iterator->is_last) {
573 use_padding = false;
574 }
575 else {
576 simple_iterator_push_(iterator);
577 if(!FLAC__metadata_simple_iterator_next(iterator)) {
578 (void)simple_iterator_pop_(iterator);
579 return false;
580 }
581 if(iterator->type != FLAC__METADATA_TYPE_PADDING) {
582 use_padding = false;
583 }
584 else {
Josh Coalsoncae58732002-05-04 17:30:32 +0000585 if(iterator->length == block->length) {
586 padding_leftover = 0;
587 block->is_last = iterator->is_last;
588 }
Josh Coalsoneb643a22002-05-20 05:58:50 +0000589 else if(iterator->length < FLAC__STREAM_METADATA_HEADER_LENGTH + block->length)
Josh Coalsoncae58732002-05-04 17:30:32 +0000590 use_padding = false;
591 else {
Josh Coalson6b02a752002-05-10 06:42:02 +0000592 padding_leftover = iterator->length - block->length;
Josh Coalsoncae58732002-05-04 17:30:32 +0000593 padding_is_last = iterator->is_last;
594 block->is_last = false;
595 }
596 }
597 if(!simple_iterator_pop_(iterator))
598 return false;
599 }
600 }
601 if(use_padding) {
602 /* move to the next block, which is suitable padding */
603 if(!FLAC__metadata_simple_iterator_next(iterator))
604 return false;
Josh Coalson51ee4652002-05-09 05:46:23 +0000605 if(padding_leftover == 0) {
606 ret = write_metadata_block_stationary_(iterator, block);
Josh Coalson51ee4652002-05-09 05:46:23 +0000607 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000608 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000609 return ret;
610 }
611 else {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000612 FLAC__ASSERT(padding_leftover >= FLAC__STREAM_METADATA_HEADER_LENGTH);
613 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 +0000614 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000615 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000616 return ret;
617 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000618 }
619 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000620 ret = rewrite_whole_file_(iterator, block, /*append=*/true);
Josh Coalson51ee4652002-05-09 05:46:23 +0000621 FLAC__ASSERT(iterator->offset[iterator->depth] == debug_target_offset);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000622 FLAC__ASSERT(ftell(iterator->file) == debug_target_offset + FLAC__STREAM_METADATA_HEADER_LENGTH);
Josh Coalson51ee4652002-05-09 05:46:23 +0000623 return ret;
Josh Coalsoncae58732002-05-04 17:30:32 +0000624 }
625}
626
Josh Coalsoncae58732002-05-04 17:30:32 +0000627FLAC__bool FLAC__metadata_simple_iterator_delete_block(FLAC__MetaData_SimpleIterator *iterator, FLAC__bool use_padding)
628{
Josh Coalson51ee4652002-05-09 05:46:23 +0000629 FLAC__ASSERT_DECLARATION(long debug_target_offset = iterator->offset[iterator->depth];)
630 FLAC__bool ret;
631
Josh Coalsoncae58732002-05-04 17:30:32 +0000632 if(iterator->type == FLAC__METADATA_TYPE_STREAMINFO) {
633 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT;
634 return false;
635 }
636
637 if(use_padding) {
638 FLAC__StreamMetaData *padding = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING);
639 if(0 == padding) {
640 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
641 return false;
642 }
643 padding->length = iterator->length;
644 if(!FLAC__metadata_simple_iterator_set_block(iterator, padding, false)) {
645 FLAC__metadata_object_delete(padding);
646 return false;
647 }
648 FLAC__metadata_object_delete(padding);
Josh Coalson6b02a752002-05-10 06:42:02 +0000649 if(!FLAC__metadata_simple_iterator_prev(iterator))
650 return false;
Josh Coalsoneb643a22002-05-20 05:58:50 +0000651 FLAC__ASSERT(iterator->offset[iterator->depth] + FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length == debug_target_offset);
Josh Coalson51ee4652002-05-09 05:46:23 +0000652 FLAC__ASSERT(ftell(iterator->file) + (long)iterator->length == debug_target_offset);
Josh Coalsoncae58732002-05-04 17:30:32 +0000653 return true;
654 }
655 else {
Josh Coalson51ee4652002-05-09 05:46:23 +0000656 ret = rewrite_whole_file_(iterator, 0, /*append=*/false);
Josh Coalsoneb643a22002-05-20 05:58:50 +0000657 FLAC__ASSERT(iterator->offset[iterator->depth] + FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length == debug_target_offset);
Josh Coalson51ee4652002-05-09 05:46:23 +0000658 FLAC__ASSERT(ftell(iterator->file) + (long)iterator->length == debug_target_offset);
659 return ret;
Josh Coalsoncae58732002-05-04 17:30:32 +0000660 }
661}
662
663
664
665/****************************************************************************
666 *
667 * Level 2 implementation
668 *
669 ***************************************************************************/
670
671
672typedef struct FLAC__MetaData_Node {
673 FLAC__StreamMetaData *data;
674 struct FLAC__MetaData_Node *prev, *next;
675} FLAC__MetaData_Node;
676
677struct FLAC__MetaData_Chain {
678 char *filename;
679 FLAC__MetaData_Node *head;
680 FLAC__MetaData_Node *tail;
681 unsigned nodes;
682 FLAC__MetaData_ChainStatus status;
683 long first_offset, last_offset; /*@@@ 2G limit */
684 /*
685 * This is the length of the chain initially read from the FLAC file.
686 * it is used to compare against the current_length to decide whether
687 * or not the whole file has to be rewritten.
688 */
689 unsigned initial_length; /*@@@ 4G limit */
690 unsigned current_length; /*@@@ 4G limit */
691};
692
693struct FLAC__MetaData_Iterator {
694 FLAC__MetaData_Chain *chain;
695 FLAC__MetaData_Node *current;
696};
697
698const char *FLAC__MetaData_ChainStatusString[] = {
699 "FLAC__METADATA_CHAIN_STATUS_OK",
700 "FLAC__METADATA_CHAIN_STATUS_ILLEGAL_INPUT",
701 "FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE",
702 "FLAC__METADATA_CHAIN_STATUS_NOT_A_FLAC_FILE",
703 "FLAC__METADATA_CHAIN_STATUS_NOT_WRITABLE",
704 "FLAC__METADATA_CHAIN_STATUS_READ_ERROR",
705 "FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR",
706 "FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR",
707 "FLAC__METADATA_CHAIN_STATUS_RENAME_ERROR",
708 "FLAC__METADATA_CHAIN_STATUS_UNLINK_ERROR",
709 "FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR",
710 "FLAC__METADATA_CHAIN_STATUS_INTERNAL_ERROR"
711};
712
713
714static FLAC__MetaData_Node *node_new_()
715{
716 FLAC__MetaData_Node *node = (FLAC__MetaData_Node*)malloc(sizeof(FLAC__MetaData_Node));
717 if(0 != node)
718 memset(node, 0, sizeof(FLAC__MetaData_Node));
719 return node;
720}
721
722static void node_delete_(FLAC__MetaData_Node *node)
723{
724 FLAC__ASSERT(0 != node);
725 if(0 != node->data)
726 FLAC__metadata_object_delete(node->data);
727 free(node);
728}
729
730static void chain_append_node_(FLAC__MetaData_Chain *chain, FLAC__MetaData_Node *node)
731{
732 FLAC__ASSERT(0 != chain);
733 FLAC__ASSERT(0 != node);
Josh Coalson6b02a752002-05-10 06:42:02 +0000734 FLAC__ASSERT(0 != node->data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000735
Josh Coalson6b02a752002-05-10 06:42:02 +0000736 node->next = node->prev = 0;
Josh Coalsoncae58732002-05-04 17:30:32 +0000737 node->data->is_last = true;
738 if(0 != chain->tail)
739 chain->tail->data->is_last = false;
740
741 if(0 == chain->head)
742 chain->head = node;
743 else {
Josh Coalson6b02a752002-05-10 06:42:02 +0000744 FLAC__ASSERT(0 != chain->tail);
Josh Coalsoncae58732002-05-04 17:30:32 +0000745 chain->tail->next = node;
746 node->prev = chain->tail;
747 }
748 chain->tail = node;
749 chain->nodes++;
Josh Coalsoneb643a22002-05-20 05:58:50 +0000750 chain->current_length += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
Josh Coalsoncae58732002-05-04 17:30:32 +0000751}
752
753static void chain_remove_node_(FLAC__MetaData_Chain *chain, FLAC__MetaData_Node *node)
754{
755 FLAC__ASSERT(0 != chain);
756 FLAC__ASSERT(0 != node);
757
758 if(node == chain->head)
759 chain->head = node->next;
760 else
761 node->prev->next = node->next;
762
763 if(node == chain->tail)
764 chain->tail = node->prev;
765 else
766 node->next->prev = node->prev;
767
Josh Coalson6b02a752002-05-10 06:42:02 +0000768 if(0 != chain->tail)
769 chain->tail->data->is_last = true;
770
Josh Coalsoncae58732002-05-04 17:30:32 +0000771 chain->nodes--;
Josh Coalsoneb643a22002-05-20 05:58:50 +0000772 chain->current_length -= (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
Josh Coalsoncae58732002-05-04 17:30:32 +0000773}
774
775static void chain_delete_node_(FLAC__MetaData_Chain *chain, FLAC__MetaData_Node *node)
776{
777 chain_remove_node_(chain, node);
778 node_delete_(node);
779}
780
781static void iterator_insert_node_(FLAC__MetaData_Iterator *iterator, FLAC__MetaData_Node *node)
782{
783 FLAC__ASSERT(0 != node);
Josh Coalson6b02a752002-05-10 06:42:02 +0000784 FLAC__ASSERT(0 != node->data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000785 FLAC__ASSERT(0 != iterator);
786 FLAC__ASSERT(0 != iterator->current);
787 FLAC__ASSERT(0 != iterator->chain);
788 FLAC__ASSERT(0 != iterator->chain->head);
789 FLAC__ASSERT(0 != iterator->chain->tail);
790
Josh Coalson6b02a752002-05-10 06:42:02 +0000791 node->data->is_last = false;
Josh Coalson2745fe72002-05-11 05:31:56 +0000792
Josh Coalsoncae58732002-05-04 17:30:32 +0000793 node->prev = iterator->current->prev;
Josh Coalson2745fe72002-05-11 05:31:56 +0000794 node->next = iterator->current;
795
Josh Coalsoncae58732002-05-04 17:30:32 +0000796 if(0 == node->prev)
797 iterator->chain->head = node;
798 else
799 node->prev->next = node;
800
Josh Coalsoncae58732002-05-04 17:30:32 +0000801 iterator->current->prev = node;
802
803 iterator->chain->nodes++;
Josh Coalsoneb643a22002-05-20 05:58:50 +0000804 iterator->chain->current_length += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
Josh Coalsoncae58732002-05-04 17:30:32 +0000805}
806
807static void iterator_insert_node_after_(FLAC__MetaData_Iterator *iterator, FLAC__MetaData_Node *node)
808{
809 FLAC__ASSERT(0 != node);
Josh Coalson6b02a752002-05-10 06:42:02 +0000810 FLAC__ASSERT(0 != node->data);
Josh Coalsoncae58732002-05-04 17:30:32 +0000811 FLAC__ASSERT(0 != iterator);
812 FLAC__ASSERT(0 != iterator->current);
813 FLAC__ASSERT(0 != iterator->chain);
814 FLAC__ASSERT(0 != iterator->chain->head);
815 FLAC__ASSERT(0 != iterator->chain->tail);
816
817 iterator->current->data->is_last = false;
818
819 node->prev = iterator->current;
Josh Coalsoncae58732002-05-04 17:30:32 +0000820 node->next = iterator->current->next;
Josh Coalson2745fe72002-05-11 05:31:56 +0000821
Josh Coalsoncae58732002-05-04 17:30:32 +0000822 if(0 == node->next)
823 iterator->chain->tail = node;
824 else
825 node->next->prev = node;
826
Josh Coalson2745fe72002-05-11 05:31:56 +0000827 node->prev->next = node;
828
Josh Coalsoncae58732002-05-04 17:30:32 +0000829 iterator->chain->tail->data->is_last = true;
830
831 iterator->chain->nodes++;
Josh Coalsoneb643a22002-05-20 05:58:50 +0000832 iterator->chain->current_length += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
Josh Coalsoncae58732002-05-04 17:30:32 +0000833}
834
Josh Coalson2745fe72002-05-11 05:31:56 +0000835/* return true iff node and node->next are both padding */
836static FLAC__bool chain_merge_adjacent_padding_(FLAC__MetaData_Chain *chain, FLAC__MetaData_Node *node)
837{
838 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 +0000839 const unsigned growth = FLAC__STREAM_METADATA_HEADER_LENGTH + node->next->data->length;
Josh Coalson2745fe72002-05-11 05:31:56 +0000840 node->data->length += growth;
841 chain->current_length += growth;
842
843 chain_delete_node_(chain, node->next);
844 return true;
845 }
846 else
847 return false;
848}
849
Josh Coalsoncae58732002-05-04 17:30:32 +0000850FLAC__MetaData_Chain *FLAC__metadata_chain_new()
851{
852 FLAC__MetaData_Chain *chain = malloc(sizeof(FLAC__MetaData_Chain));
853
854 if(0 != chain) {
855 chain->filename = 0;
856 chain->head = chain->tail = 0;
857 chain->nodes = 0;
858 chain->status = FLAC__METADATA_CHAIN_STATUS_OK;
859 chain->initial_length = chain->current_length = 0;
860 }
861
862 return chain;
863}
864
865void FLAC__metadata_chain_delete(FLAC__MetaData_Chain *chain)
866{
867 FLAC__MetaData_Node *node, *next;
868
869 FLAC__ASSERT(0 != chain);
870
871 for(node = chain->head; node; ) {
872 next = node->next;
873 node_delete_(node);
874 node = next;
875 }
876
877 if(0 != chain->filename)
878 free(chain->filename);
879
880 free(chain);
881}
882
883FLAC__MetaData_ChainStatus FLAC__metadata_chain_status(FLAC__MetaData_Chain *chain)
884{
885 FLAC__MetaData_ChainStatus status;
886
887 FLAC__ASSERT(0 != chain);
888
889 status = chain->status;
890 chain->status = FLAC__METADATA_CHAIN_STATUS_OK;
891 return status;
892}
893
894FLAC__bool FLAC__metadata_chain_read(FLAC__MetaData_Chain *chain, const char *filename)
895{
896 FLAC__MetaData_SimpleIterator *iterator;
897 FLAC__MetaData_Node *node;
898
899 FLAC__ASSERT(0 != chain);
900 FLAC__ASSERT(0 != filename);
901
902 if(0 == (chain->filename = strdup(filename))) {
903 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
904 return false;
905 }
906
907 if(0 == (iterator = FLAC__metadata_simple_iterator_new())) {
908 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
909 return false;
910 }
911
912 if(!FLAC__metadata_simple_iterator_init(iterator, filename, /*preserve_file_stats=*/false)) {
913 chain->status = get_equivalent_status_(iterator->status);
914 return false;
915 }
916
Josh Coalson2745fe72002-05-11 05:31:56 +0000917 chain->first_offset = iterator->offset[iterator->depth];
918
Josh Coalsoncae58732002-05-04 17:30:32 +0000919 do {
920 node = node_new_();
921 if(0 == node) {
922 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
923 return false;
924 }
925 node->data = FLAC__metadata_simple_iterator_get_block(iterator);
926 if(0 == node->data) {
927 node_delete_(node);
928 chain->status = get_equivalent_status_(iterator->status);
929 return false;
930 }
931 chain_append_node_(chain, node);
932 } while(FLAC__metadata_simple_iterator_next(iterator));
933
934 if(!iterator->is_last || iterator->status != FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK) {
935 chain->status = get_equivalent_status_(iterator->status);
936 return false;
937 }
938
Josh Coalson2745fe72002-05-11 05:31:56 +0000939 chain->last_offset = ftell(iterator->file) + iterator->length;
Josh Coalsoncae58732002-05-04 17:30:32 +0000940 FLAC__metadata_simple_iterator_delete(iterator);
941
942 chain->initial_length = chain->current_length;
943 return true;
944}
945
Josh Coalsoncae58732002-05-04 17:30:32 +0000946FLAC__bool FLAC__metadata_chain_write(FLAC__MetaData_Chain *chain, FLAC__bool use_padding, FLAC__bool preserve_file_stats)
947{
948 struct stat stats;
949 const char *tempfile_path_prefix = 0;
950
Josh Coalson2745fe72002-05-11 05:31:56 +0000951 if(use_padding) {
Josh Coalson2745fe72002-05-11 05:31:56 +0000952 if(chain->current_length < chain->initial_length && chain->tail->data->type == FLAC__METADATA_TYPE_PADDING) {
953 const unsigned delta = chain->initial_length - chain->current_length;
954 chain->tail->data->length += delta;
955 chain->current_length += delta;
956 FLAC__ASSERT(chain->current_length == chain->initial_length);
Josh Coalsoncae58732002-05-04 17:30:32 +0000957 }
Josh Coalsoneb643a22002-05-20 05:58:50 +0000958 else if(chain->current_length + FLAC__STREAM_METADATA_HEADER_LENGTH <= chain->initial_length) {
Josh Coalson2745fe72002-05-11 05:31:56 +0000959 FLAC__StreamMetaData *padding;
960 FLAC__MetaData_Node *node;
961 if(0 == (padding = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING))) {
962 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
963 return false;
964 }
Josh Coalsoneb643a22002-05-20 05:58:50 +0000965 padding->length = chain->initial_length - (FLAC__STREAM_METADATA_HEADER_LENGTH + chain->current_length);
Josh Coalson2745fe72002-05-11 05:31:56 +0000966 if(0 == (node = node_new_())) {
967 FLAC__metadata_object_delete(padding);
968 chain->status = FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
969 return false;
970 }
971 node->data = padding;
972 chain_append_node_(chain, node);
973 FLAC__ASSERT(chain->current_length == chain->initial_length);
Josh Coalsoncae58732002-05-04 17:30:32 +0000974 }
Josh Coalson2745fe72002-05-11 05:31:56 +0000975 else if(chain->current_length > chain->initial_length) {
976 const unsigned delta = chain->current_length - chain->initial_length;
977 if(chain->tail->data->type == FLAC__METADATA_TYPE_PADDING) {
Josh Coalsoneb643a22002-05-20 05:58:50 +0000978 if(chain->tail->data->length + FLAC__STREAM_METADATA_HEADER_LENGTH == delta) {
Josh Coalson2745fe72002-05-11 05:31:56 +0000979 chain_delete_node_(chain, chain->tail);
980 FLAC__ASSERT(chain->current_length == chain->initial_length);
981 }
Josh Coalsone2ba10f2002-05-14 06:03:17 +0000982 else if(chain->tail->data->length >= delta) {
Josh Coalson2745fe72002-05-11 05:31:56 +0000983 chain->tail->data->length -= delta;
984 chain->current_length -= delta;
985 FLAC__ASSERT(chain->current_length == chain->initial_length);
986 }
987 }
988 }
Josh Coalsoncae58732002-05-04 17:30:32 +0000989 }
990
991 if(preserve_file_stats)
992 get_file_stats_(chain->filename, &stats);
993
994 if(chain->current_length == chain->initial_length) {
995 if(!chain_rewrite_chain_(chain))
996 return false;
997 }
998 else {
999 if(!chain_rewrite_file_(chain, tempfile_path_prefix))
1000 return false;
1001 }
1002
1003 if(preserve_file_stats)
1004 set_file_stats_(chain->filename, &stats);
1005
Josh Coalsone2ba10f2002-05-14 06:03:17 +00001006 /* recompute lengths and offsets if necessary */
1007 if(chain->initial_length != chain->current_length) {
1008 const FLAC__MetaData_Node *node;
1009 chain->initial_length = chain->current_length;
1010 chain->last_offset = chain->first_offset;
1011 for(node = chain->head; node; node = node->next)
Josh Coalsoneb643a22002-05-20 05:58:50 +00001012 chain->last_offset += (FLAC__STREAM_METADATA_HEADER_LENGTH + node->data->length);
Josh Coalsone2ba10f2002-05-14 06:03:17 +00001013 }
1014
Josh Coalsoncae58732002-05-04 17:30:32 +00001015 return true;
1016}
1017
1018void FLAC__metadata_chain_merge_padding(FLAC__MetaData_Chain *chain)
1019{
1020 FLAC__MetaData_Node *node;
1021
1022 FLAC__ASSERT(0 != chain);
1023
1024 for(node = chain->head; node; ) {
Josh Coalson2745fe72002-05-11 05:31:56 +00001025 if(!chain_merge_adjacent_padding_(chain, node))
Josh Coalsoncae58732002-05-04 17:30:32 +00001026 node = node->next;
Josh Coalsoncae58732002-05-04 17:30:32 +00001027 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001028}
1029
1030void FLAC__metadata_chain_sort_padding(FLAC__MetaData_Chain *chain)
1031{
1032 FLAC__MetaData_Node *node, *save;
1033 unsigned i;
1034
1035 FLAC__ASSERT(0 != chain);
1036
1037 /*
1038 * Don't try and be too smart... this simple algo is good enough for
1039 * the small number of nodes that we deal with.
1040 */
1041 for(i = 0, node = chain->head; i < chain->nodes; i++) {
1042 node->data->is_last = false; /* we'll fix at the end */
1043 if(node->data->type == FLAC__METADATA_TYPE_PADDING) {
1044 save = node->next;
1045 chain_remove_node_(chain, node);
1046 chain_append_node_(chain, node);
1047 node = save;
1048 }
1049 else {
1050 node = node->next;
1051 }
1052 }
Josh Coalsoncae58732002-05-04 17:30:32 +00001053
1054 FLAC__metadata_chain_merge_padding(chain);
1055}
1056
1057
1058FLAC__MetaData_Iterator *FLAC__metadata_iterator_new()
1059{
1060 FLAC__MetaData_Iterator *iterator = malloc(sizeof(FLAC__MetaData_Iterator));
1061
1062 if(0 != iterator) {
1063 iterator->current = 0;
1064 iterator->chain = 0;
1065 }
1066
1067 return iterator;
1068}
1069
1070void FLAC__metadata_iterator_delete(FLAC__MetaData_Iterator *iterator)
1071{
1072 FLAC__ASSERT(0 != iterator);
1073
1074 free(iterator);
1075}
1076
1077/*
1078 * Initialize the iterator to point to the first metadata block in the
1079 * given chain.
1080 */
1081void FLAC__metadata_iterator_init(FLAC__MetaData_Iterator *iterator, FLAC__MetaData_Chain *chain)
1082{
1083 FLAC__ASSERT(0 != iterator);
1084 FLAC__ASSERT(0 != chain);
1085 FLAC__ASSERT(0 != chain->head);
1086
1087 iterator->chain = chain;
1088 iterator->current = chain->head;
1089}
1090
1091FLAC__bool FLAC__metadata_iterator_next(FLAC__MetaData_Iterator *iterator)
1092{
1093 FLAC__ASSERT(0 != iterator);
1094
1095 if(0 == iterator->current || 0 == iterator->current->next)
1096 return false;
1097
1098 iterator->current = iterator->current->next;
1099 return true;
1100}
1101
1102FLAC__bool FLAC__metadata_iterator_prev(FLAC__MetaData_Iterator *iterator)
1103{
1104 FLAC__ASSERT(0 != iterator);
1105
1106 if(0 == iterator->current || 0 == iterator->current->prev)
1107 return false;
1108
1109 iterator->current = iterator->current->prev;
1110 return true;
1111}
1112
1113FLAC__MetaDataType FLAC__metadata_iterator_get_block_type(const FLAC__MetaData_Iterator *iterator)
1114{
1115 FLAC__ASSERT(0 != iterator);
1116 FLAC__ASSERT(0 != iterator->current);
Josh Coalson2745fe72002-05-11 05:31:56 +00001117 FLAC__ASSERT(0 != iterator->current->data);
Josh Coalsoncae58732002-05-04 17:30:32 +00001118
1119 return iterator->current->data->type;
1120}
1121
1122FLAC__StreamMetaData *FLAC__metadata_iterator_get_block(FLAC__MetaData_Iterator *iterator)
1123{
1124 FLAC__ASSERT(0 != iterator);
1125 FLAC__ASSERT(0 != iterator->current);
1126
1127 return iterator->current->data;
1128}
1129
Josh Coalson2745fe72002-05-11 05:31:56 +00001130FLAC__bool FLAC__metadata_iterator_set_block(FLAC__MetaData_Iterator *iterator, FLAC__StreamMetaData *block)
1131{
1132 return FLAC__metadata_iterator_delete_block(iterator, false) && FLAC__metadata_iterator_insert_block_after(iterator, block);
1133}
1134
Josh Coalsoncae58732002-05-04 17:30:32 +00001135FLAC__bool FLAC__metadata_iterator_delete_block(FLAC__MetaData_Iterator *iterator, FLAC__bool replace_with_padding)
1136{
1137 FLAC__MetaData_Node *save;
1138
1139 FLAC__ASSERT(0 != iterator);
1140 FLAC__ASSERT(0 != iterator->current);
1141
1142 if(0 == iterator->current->prev) {
1143 FLAC__ASSERT(iterator->current->data->type == FLAC__METADATA_TYPE_STREAMINFO);
1144 return false;
1145 }
1146
1147 save = iterator->current->prev;
1148
1149 if(replace_with_padding) {
1150 FLAC__metadata_object_delete_data_(iterator->current->data);
1151 iterator->current->data->type = FLAC__METADATA_TYPE_PADDING;
1152 }
1153 else {
1154 chain_delete_node_(iterator->chain, iterator->current);
1155 }
1156
1157 iterator->current = save;
1158 return true;
1159}
1160
1161FLAC__bool FLAC__metadata_iterator_insert_block_before(FLAC__MetaData_Iterator *iterator, FLAC__StreamMetaData *block)
1162{
1163 FLAC__MetaData_Node *node;
1164
1165 FLAC__ASSERT(0 != iterator);
1166 FLAC__ASSERT(0 != iterator->current);
1167
1168 if(block->type == FLAC__METADATA_TYPE_STREAMINFO)
1169 return false;
1170
1171 if(0 == iterator->current->prev) {
1172 FLAC__ASSERT(iterator->current->data->type == FLAC__METADATA_TYPE_STREAMINFO);
1173 return false;
1174 }
1175
1176 if(0 == (node = node_new_()))
1177 return false;
1178
1179 node->data = block;
1180 iterator_insert_node_(iterator, node);
1181 iterator->current = node;
1182 return true;
1183}
1184
1185FLAC__bool FLAC__metadata_iterator_insert_block_after(FLAC__MetaData_Iterator *iterator, FLAC__StreamMetaData *block)
1186{
1187 FLAC__MetaData_Node *node;
1188
1189 FLAC__ASSERT(0 != iterator);
1190 FLAC__ASSERT(0 != iterator->current);
1191
1192 if(block->type == FLAC__METADATA_TYPE_STREAMINFO)
1193 return false;
1194
1195 if(0 == (node = node_new_()))
1196 return false;
1197
1198 node->data = block;
1199 iterator_insert_node_after_(iterator, node);
1200 iterator->current = node;
1201 return true;
1202}
1203
1204
1205/****************************************************************************
1206 *
1207 * Metadata object routines
1208 *
1209 ***************************************************************************/
1210
1211FLAC__StreamMetaData *FLAC__metadata_object_new(FLAC__MetaDataType type)
1212{
1213 FLAC__StreamMetaData *object = malloc(sizeof(FLAC__StreamMetaData));
1214 if(0 != object) {
1215 memset(object, 0, sizeof(FLAC__StreamMetaData));
1216 object->is_last = false;
1217 object->type = type;
1218 switch(type) {
1219 case FLAC__METADATA_TYPE_STREAMINFO:
1220 object->length = FLAC__STREAM_METADATA_STREAMINFO_LENGTH;
1221 break;
1222 case FLAC__METADATA_TYPE_PADDING:
1223 break;
1224 case FLAC__METADATA_TYPE_APPLICATION:
1225 object->length = FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8;
1226 break;
1227 case FLAC__METADATA_TYPE_SEEKTABLE:
1228 break;
1229 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1230 object->length = (FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN + FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN) / 8;
1231 break;
1232 default:
1233 FLAC__ASSERT(0);
1234 }
1235 }
1236
1237 return object;
1238}
1239
1240static FLAC__bool copy_bytes_(FLAC__byte **to, const FLAC__byte *from, unsigned bytes)
1241{
1242 if(0 == (*to = malloc(bytes)))
1243 return false;
1244 memcpy(*to, from, bytes);
1245 return true;
1246}
1247
1248static FLAC__bool copy_vcentry_(FLAC__StreamMetaData_VorbisComment_Entry *to, const FLAC__StreamMetaData_VorbisComment_Entry *from)
1249{
1250 to->length = from->length;
1251 if(0 == from->entry) {
1252 FLAC__ASSERT(0 == from->length);
1253 to->entry = 0;
1254 }
1255 else {
1256 if(0 == (to->entry = malloc(from->length)))
1257 return false;
1258 memcpy(to->entry, from->entry, from->length);
1259 }
1260 return true;
1261}
1262
1263FLAC__StreamMetaData *FLAC__metadata_object_copy(const FLAC__StreamMetaData *object)
1264{
1265 FLAC__StreamMetaData *to;
1266 unsigned i;
1267
1268 FLAC__ASSERT(0 != object);
1269
1270 if(0 != (to = FLAC__metadata_object_new(object->type))) {
1271 to->is_last = object->is_last;
1272 to->type = object->type;
1273 to->length = object->length;
1274 switch(to->type) {
1275 case FLAC__METADATA_TYPE_STREAMINFO:
1276 memcpy(&to->data.stream_info, &object->data.stream_info, sizeof(FLAC__StreamMetaData_StreamInfo));
1277 break;
1278 case FLAC__METADATA_TYPE_PADDING:
1279 break;
1280 case FLAC__METADATA_TYPE_APPLICATION:
1281 memcpy(&to->data.application.id, &object->data.application.id, FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8);
1282 if(0 == object->data.application.data)
1283 to->data.application.data = 0;
1284 else {
1285 if(!copy_bytes_(&to->data.application.data, object->data.application.data, object->length - FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8)) {
1286 FLAC__metadata_object_delete(to);
1287 return 0;
1288 }
1289 }
1290 break;
1291 case FLAC__METADATA_TYPE_SEEKTABLE:
1292 to->data.seek_table.num_points = object->data.seek_table.num_points;
1293 if(0 == object->data.seek_table.points)
1294 to->data.seek_table.points = 0;
1295 else {
1296 if(!copy_bytes_((FLAC__byte**)&to->data.seek_table.points, (FLAC__byte*)object->data.seek_table.points, object->data.seek_table.num_points * sizeof(FLAC__StreamMetaData_SeekPoint))) {
1297 FLAC__metadata_object_delete(to);
1298 return 0;
1299 }
1300 }
1301 break;
1302 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1303 if(!copy_vcentry_(&to->data.vorbis_comment.vendor_string, &object->data.vorbis_comment.vendor_string)) {
1304 FLAC__metadata_object_delete(to);
1305 return 0;
1306 }
1307 to->data.vorbis_comment.num_comments = object->data.vorbis_comment.num_comments;
1308 if(0 == (to->data.vorbis_comment.comments = malloc(object->data.vorbis_comment.num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry*)))) {
1309 FLAC__metadata_object_delete(to);
1310 return 0;
1311 }
1312 /* Need to do this to set the pointers inside the comments to 0.
1313 * In case of an error in the following loop, the object will be
1314 * deleted and we don't want the destructor freeing uninitialized
1315 * pointers.
1316 */
1317 memset(to->data.vorbis_comment.comments, 0, object->data.vorbis_comment.num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry*));
1318 for(i = 0; i < object->data.vorbis_comment.num_comments; i++) {
1319 if(!copy_vcentry_(to->data.vorbis_comment.comments+i, object->data.vorbis_comment.comments+i)) {
1320 FLAC__metadata_object_delete(to);
1321 return 0;
1322 }
1323 }
1324 break;
1325 default:
1326 FLAC__ASSERT(0);
1327 }
1328 }
1329
1330 return to;
1331}
1332
1333void FLAC__metadata_object_delete_data_(FLAC__StreamMetaData *object)
1334{
1335 unsigned i;
1336
1337 FLAC__ASSERT(0 != object);
1338
1339 switch(object->type) {
1340 case FLAC__METADATA_TYPE_STREAMINFO:
1341 case FLAC__METADATA_TYPE_PADDING:
1342 break;
1343 case FLAC__METADATA_TYPE_APPLICATION:
1344 if(0 != object->data.application.data)
1345 free(object->data.application.data);
1346 break;
1347 case FLAC__METADATA_TYPE_SEEKTABLE:
1348 if(0 != object->data.seek_table.points)
1349 free(object->data.seek_table.points);
1350 break;
1351 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1352 if(0 != object->data.vorbis_comment.vendor_string.entry)
1353 free(object->data.vorbis_comment.vendor_string.entry);
1354 for(i = 0; i < object->data.vorbis_comment.num_comments; i++) {
1355 if(0 != object->data.vorbis_comment.comments[i].entry)
1356 free(object->data.vorbis_comment.comments[i].entry);
1357 }
Josh Coalsonfad2e612002-05-07 05:33:10 +00001358 if(0 != object->data.vorbis_comment.comments)
1359 free(object->data.vorbis_comment.comments);
Josh Coalsoncae58732002-05-04 17:30:32 +00001360 break;
1361 default:
1362 FLAC__ASSERT(0);
1363 }
1364}
1365
1366void FLAC__metadata_object_delete(FLAC__StreamMetaData *object)
1367{
1368 FLAC__metadata_object_delete_data_(object);
1369 free(object);
1370}
1371
Josh Coalsoncae58732002-05-04 17:30:32 +00001372FLAC__bool FLAC__metadata_object_application_set_data(FLAC__StreamMetaData *object, FLAC__byte *data, unsigned length, FLAC__bool copy)
1373{
1374 FLAC__ASSERT(object->type == FLAC__METADATA_TYPE_APPLICATION);
Josh Coalson31cf6a42002-05-17 06:25:14 +00001375 FLAC__ASSERT(0 != data || (length == 0 && copy == false));
Josh Coalsoncae58732002-05-04 17:30:32 +00001376
1377 if(0 != object->data.application.data)
1378 free(object->data.application.data);
1379
1380 if(copy) {
1381 if(!copy_bytes_(&object->data.application.data, data, length))
1382 return false;
1383 }
1384 else {
1385 object->data.application.data = data;
1386 }
1387 object->length = FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8 + length;
1388 return true;
1389}
1390
1391FLAC__StreamMetaData_SeekPoint *FLAC__metadata_object_seekpoint_array_new(unsigned num_points)
1392{
1393 FLAC__StreamMetaData_SeekPoint *object_array = malloc(num_points * sizeof(FLAC__StreamMetaData_SeekPoint));
1394
1395 if(0 != object_array)
1396 memset(object_array, 0, num_points * sizeof(FLAC__StreamMetaData_SeekPoint));
1397
1398 return object_array;
1399}
1400
1401FLAC__StreamMetaData_SeekPoint *FLAC__metadata_object_seekpoint_array_copy(const FLAC__StreamMetaData_SeekPoint *object_array, unsigned num_points)
1402{
1403 FLAC__StreamMetaData_SeekPoint *return_array;
1404
1405 FLAC__ASSERT(0 != object_array);
1406
1407 return_array = FLAC__metadata_object_seekpoint_array_new(num_points);
1408
1409 if(0 != return_array)
1410 memcpy(return_array, object_array, num_points * sizeof(FLAC__StreamMetaData_SeekPoint));
1411
1412 return return_array;
1413}
1414
1415void FLAC__metadata_object_seekpoint_array_delete(FLAC__StreamMetaData_SeekPoint *object_array)
1416{
1417 FLAC__ASSERT(0 != object_array);
1418
1419 free(object_array);
1420}
1421
1422FLAC__bool FLAC__metadata_object_seekpoint_array_resize(FLAC__StreamMetaData_SeekPoint **object_array, unsigned old_num_points, unsigned new_num_points)
1423{
1424 FLAC__ASSERT(0 != object_array);
1425
1426 if(0 == *object_array) {
1427 FLAC__ASSERT(old_num_points == 0);
1428 return 0 != (*object_array = FLAC__metadata_object_seekpoint_array_new(new_num_points));
1429 }
1430 else {
1431 const unsigned old_size = old_num_points * sizeof(FLAC__StreamMetaData_SeekPoint);
1432 const unsigned new_size = new_num_points * sizeof(FLAC__StreamMetaData_SeekPoint);
1433
1434 if(0 == (*object_array = realloc(*object_array, new_size)))
1435 return false;
1436
1437 if(new_size > old_size)
1438 memset(*object_array + old_num_points, 0, new_size - old_size);
1439
1440 return true;
1441 }
1442}
1443
1444FLAC__bool FLAC__metadata_object_seektable_set_points(FLAC__StreamMetaData *object, FLAC__StreamMetaData_SeekPoint *points, unsigned num_points, FLAC__bool copy)
1445{
1446 FLAC__ASSERT(object->type == FLAC__METADATA_TYPE_SEEKTABLE);
Josh Coalson31cf6a42002-05-17 06:25:14 +00001447 FLAC__ASSERT(0 != points || (num_points == 0 && copy == false));
Josh Coalsoncae58732002-05-04 17:30:32 +00001448
1449 object->data.seek_table.num_points = num_points;
1450
1451 if(0 != object->data.seek_table.points)
1452 FLAC__metadata_object_seekpoint_array_delete(object->data.seek_table.points);
1453
1454 if(copy) {
1455 if(0 == (object->data.seek_table.points = FLAC__metadata_object_seekpoint_array_copy(points, num_points)))
1456 return false;
1457 }
1458 else {
1459 object->data.seek_table.points = points;
1460 }
1461 object->length = num_points * FLAC__STREAM_METADATA_SEEKPOINT_LENGTH;
1462 return true;
1463}
1464
1465FLAC__StreamMetaData_VorbisComment_Entry *FLAC__metadata_object_vorbiscomment_entry_array_new(unsigned num_comments)
1466{
1467 FLAC__StreamMetaData_VorbisComment_Entry *object_array = malloc(num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry));
1468
1469 if(0 != object_array)
1470 memset(object_array, 0, num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry));
1471
1472 return object_array;
1473}
1474
1475FLAC__StreamMetaData_VorbisComment_Entry *FLAC__metadata_object_vorbiscomment_entry_array_copy(const FLAC__StreamMetaData_VorbisComment_Entry *object_array, unsigned num_comments)
1476{
1477 FLAC__StreamMetaData_VorbisComment_Entry *return_array;
1478
1479 FLAC__ASSERT(0 != object_array);
1480
1481 return_array = FLAC__metadata_object_vorbiscomment_entry_array_new(num_comments);
1482
1483 if(0 != return_array) {
1484 unsigned i;
1485 for(i = 0; i < num_comments; i++) {
1486 return_array[i].length = object_array[i].length;
1487 if(!copy_bytes_(&(return_array[i].entry), object_array[i].entry, object_array[i].length)) {
1488 FLAC__metadata_object_vorbiscomment_entry_array_delete(return_array, num_comments);
1489 return 0;
1490 }
1491 }
1492 }
1493
1494 return return_array;
1495}
1496
1497void FLAC__metadata_object_vorbiscomment_entry_array_delete(FLAC__StreamMetaData_VorbisComment_Entry *object_array, unsigned num_comments)
1498{
1499 unsigned i;
1500
1501 FLAC__ASSERT(0 != object_array);
1502
1503 for(i = 0; i < num_comments; i++)
1504 if(0 != object_array[i].entry)
1505 free(object_array[i].entry);
1506
1507 free(object_array);
1508}
1509
1510FLAC__bool FLAC__metadata_object_vorbiscomment_entry_array_resize(FLAC__StreamMetaData_VorbisComment_Entry **object_array, unsigned old_num_comments, unsigned new_num_comments)
1511{
1512 FLAC__ASSERT(0 != object_array);
1513
1514 if(0 == *object_array) {
1515 FLAC__ASSERT(old_num_comments == 0);
1516 return 0 != (*object_array = FLAC__metadata_object_vorbiscomment_entry_array_new(new_num_comments));
1517 }
1518 else {
1519 const unsigned old_size = old_num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry);
1520 const unsigned new_size = new_num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry);
1521
1522 /* if shrinking, free the truncated entries */
1523 if(new_num_comments < old_num_comments) {
1524 unsigned i;
1525 for(i = new_num_comments; i < old_num_comments; i++)
1526 free((*object_array)[i].entry);
1527 }
1528
1529 if(0 == (*object_array = realloc(*object_array, new_size)))
1530 return false;
1531
1532 /* if growing, zero all the length/pointers of new elements */
1533 if(new_size > old_size)
1534 memset(*object_array + old_num_comments, 0, new_size - old_size);
1535
1536 return true;
1537 }
1538}
1539
1540FLAC__bool FLAC__metadata_object_vorbiscomment_set_vendor_string(FLAC__StreamMetaData *object, FLAC__byte *entry, unsigned length, FLAC__bool copy)
1541{
1542 FLAC__ASSERT(object->type == FLAC__METADATA_TYPE_VORBIS_COMMENT);
Josh Coalson31cf6a42002-05-17 06:25:14 +00001543 FLAC__ASSERT(0 != entry || (length == 0 && copy == false));
Josh Coalsoncae58732002-05-04 17:30:32 +00001544
1545 if(0 != object->data.vorbis_comment.vendor_string.entry)
1546 free(object->data.vorbis_comment.vendor_string.entry);
1547
1548 object->length -= object->data.vorbis_comment.vendor_string.length;
1549
1550 if(copy) {
1551 if(!copy_bytes_(&object->data.vorbis_comment.vendor_string.entry, entry, length)) {
1552 object->data.vorbis_comment.vendor_string.length = 0;
1553 return false;
1554 }
1555 }
1556 else {
1557 object->data.vorbis_comment.vendor_string.entry = entry;
1558 }
1559
1560 object->data.vorbis_comment.vendor_string.length = length;
1561 object->length += object->data.vorbis_comment.vendor_string.length;
1562
1563 return true;
1564}
1565
1566FLAC__bool FLAC__metadata_object_vorbiscomment_set_comments(FLAC__StreamMetaData *object, FLAC__StreamMetaData_VorbisComment_Entry *comments, unsigned num_comments, FLAC__bool copy)
1567{
1568 unsigned i;
1569
1570 FLAC__ASSERT(object->type == FLAC__METADATA_TYPE_VORBIS_COMMENT);
Josh Coalson31cf6a42002-05-17 06:25:14 +00001571 FLAC__ASSERT(0 != comments || (num_comments == 0 && copy == false));
Josh Coalsoncae58732002-05-04 17:30:32 +00001572
1573 object->data.vorbis_comment.num_comments = num_comments;
1574
1575 if(0 != object->data.vorbis_comment.comments)
1576 FLAC__metadata_object_vorbiscomment_entry_array_delete(object->data.vorbis_comment.comments, num_comments);
1577
1578 if(copy) {
1579 if(0 == (object->data.vorbis_comment.comments = FLAC__metadata_object_vorbiscomment_entry_array_copy(comments, num_comments)))
1580 return false;
1581 }
1582 else {
1583 object->data.vorbis_comment.comments = comments;
1584 }
1585 object->length = num_comments * FLAC__STREAM_METADATA_SEEKPOINT_LENGTH;
1586
1587 /* calculate the new length */
1588 object->length = (FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN) / 8;
1589 object->length += object->data.vorbis_comment.vendor_string.length;
1590 object->length += (FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN) / 8;
1591 for(i = 0; i < object->data.vorbis_comment.num_comments; i++) {
1592 object->length += (FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8);
1593 object->length += object->data.vorbis_comment.comments[i].length;
1594 }
1595
1596 return true;
1597}
1598
1599
1600/****************************************************************************
1601 *
1602 * Local function definitions
1603 *
1604 ***************************************************************************/
1605
1606void pack_uint32_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes)
1607{
1608 unsigned i;
1609
1610 b += bytes;
1611
1612 for(i = 0; i < bytes; i++) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00001613 *(--b) = (FLAC__byte)(val & 0xff);
Josh Coalsoncae58732002-05-04 17:30:32 +00001614 val >>= 8;
1615 }
1616}
1617
1618void pack_uint32_little_endian_(FLAC__uint32 val, FLAC__byte *b, unsigned bytes)
1619{
1620 unsigned i;
1621
1622 for(i = 0; i < bytes; i++) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00001623 *(b++) = (FLAC__byte)(val & 0xff);
Josh Coalsoncae58732002-05-04 17:30:32 +00001624 val >>= 8;
1625 }
1626}
1627
1628void pack_uint64_(FLAC__uint64 val, FLAC__byte *b, unsigned bytes)
1629{
1630 unsigned i;
1631
1632 b += bytes;
1633
1634 for(i = 0; i < bytes; i++) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00001635 *(--b) (FLAC__byte)(= val & 0xff);
Josh Coalsoncae58732002-05-04 17:30:32 +00001636 val >>= 8;
1637 }
1638}
1639
1640FLAC__uint32 unpack_uint32_(FLAC__byte *b, unsigned bytes)
1641{
1642 FLAC__uint32 ret = 0;
1643 unsigned i;
1644
1645 for(i = 0; i < bytes; i++)
1646 ret = (ret << 8) | (FLAC__uint32)(*b++);
1647
1648 return ret;
1649}
1650
1651FLAC__uint32 unpack_uint32_little_endian_(FLAC__byte *b, unsigned bytes)
1652{
1653 FLAC__uint32 ret = 0;
1654 unsigned i;
1655
1656 b += bytes;
1657
1658 for(i = 0; i < bytes; i++)
1659 ret = (ret << 8) | (FLAC__uint32)(*--b);
1660
1661 return ret;
1662}
1663
1664FLAC__uint64 unpack_uint64_(FLAC__byte *b, unsigned bytes)
1665{
1666 FLAC__uint64 ret = 0;
1667 unsigned i;
1668
1669 for(i = 0; i < bytes; i++)
1670 ret = (ret << 8) | (FLAC__uint64)(*b++);
1671
1672 return ret;
1673}
1674
1675FLAC__bool read_metadata_block_header_(FLAC__MetaData_SimpleIterator *iterator)
1676{
Josh Coalsoneb643a22002-05-20 05:58:50 +00001677 FLAC__byte raw_header[FLAC__STREAM_METADATA_HEADER_LENGTH];
Josh Coalsoncae58732002-05-04 17:30:32 +00001678
1679 FLAC__ASSERT(0 != iterator);
1680 FLAC__ASSERT(0 != iterator->file);
1681
Josh Coalsoneb643a22002-05-20 05:58:50 +00001682 if(fread(raw_header, 1, FLAC__STREAM_METADATA_HEADER_LENGTH, iterator->file) != FLAC__STREAM_METADATA_HEADER_LENGTH) {
Josh Coalsoncae58732002-05-04 17:30:32 +00001683 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1684 return false;
1685 }
1686
1687 iterator->is_last = raw_header[0] & 0x80? true : false;
1688 iterator->type = (FLAC__MetaDataType)(raw_header[0] & 0x7f);
1689 iterator->length = unpack_uint32_(raw_header + 1, 3);
1690
1691 return true;
1692}
1693
1694FLAC__bool read_metadata_block_data_(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block)
1695{
1696 FLAC__ASSERT(0 != iterator);
1697 FLAC__ASSERT(0 != iterator->file);
1698
1699 switch(block->type) {
1700 case FLAC__METADATA_TYPE_STREAMINFO:
1701 iterator->status = read_metadata_block_data_streaminfo_(iterator->file, &block->data.stream_info);
1702 break;
1703 case FLAC__METADATA_TYPE_PADDING:
1704 iterator->status = read_metadata_block_data_padding_(iterator->file, &block->data.padding, block->length);
1705 break;
1706 case FLAC__METADATA_TYPE_APPLICATION:
1707 iterator->status = read_metadata_block_data_application_(iterator->file, &block->data.application, block->length);
1708 break;
1709 case FLAC__METADATA_TYPE_SEEKTABLE:
1710 iterator->status = read_metadata_block_data_seektable_(iterator->file, &block->data.seek_table, block->length);
1711 break;
1712 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1713 iterator->status = read_metadata_block_data_vorbis_comment_(iterator->file, &block->data.vorbis_comment);
1714 break;
1715 default:
1716 FLAC__ASSERT(0);
1717 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_INTERNAL_ERROR;
1718 }
1719
1720 return (iterator->status == FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK);
1721}
1722
1723FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_streaminfo_(FILE *file, FLAC__StreamMetaData_StreamInfo *block)
1724{
1725 FLAC__byte buffer[FLAC__STREAM_METADATA_STREAMINFO_LENGTH], *b;
1726
1727 FLAC__ASSERT(0 != file);
1728
1729 if(fread(buffer, 1, FLAC__STREAM_METADATA_STREAMINFO_LENGTH, file) != FLAC__STREAM_METADATA_STREAMINFO_LENGTH)
1730 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1731
1732 b = buffer;
1733
1734 /* @@@ we are using hardcoded numbers for simplicity but we should
1735 * probably eventually write a bit-level unpacker and use the
1736 * _STREAMINFO_ constants.
1737 */
1738 block->min_blocksize = unpack_uint32_(b, 2); b += 2;
1739 block->max_blocksize = unpack_uint32_(b, 2); b += 2;
1740 block->min_framesize = unpack_uint32_(b, 3); b += 3;
1741 block->max_framesize = unpack_uint32_(b, 3); b += 3;
1742 block->sample_rate = (unpack_uint32_(b, 2) << 4) | ((unsigned)(b[2] & 0xf0) >> 4);
1743 block->channels = (unsigned)((b[2] & 0x0e) >> 1) + 1;
1744 block->bits_per_sample = ((((unsigned)(b[2] & 0x01)) << 1) | (((unsigned)(b[3] & 0xf0)) >> 4)) + 1;
1745 block->total_samples = (((FLAC__uint64)(b[3] & 0x0f)) << 32) | unpack_uint64_(b+4, 4);
1746 memcpy(block->md5sum, b+8, 16);
1747
1748 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1749}
1750
1751
1752FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_padding_(FILE *file, FLAC__StreamMetaData_Padding *block, unsigned block_length)
1753{
1754 FLAC__ASSERT(0 != file);
1755
1756 (void)block; /* nothing to do; we don't care about reading the padding bytes */
1757
1758 if(0 != fseek(file, block_length, SEEK_CUR))
1759 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
1760
1761 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1762}
1763
1764FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_application_(FILE *file, FLAC__StreamMetaData_Application *block, unsigned block_length)
1765{
1766 const unsigned id_bytes = FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8;
1767
1768 FLAC__ASSERT(0 != file);
1769
1770 if(fread(block->id, 1, id_bytes, file) != id_bytes)
1771 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1772
1773 block_length -= id_bytes;
1774
1775 if(0 == (block->data = malloc(block_length)))
1776 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
1777
1778 if(fread(block->data, 1, block_length, file) != block_length)
1779 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1780
1781 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1782}
1783
1784FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_seektable_(FILE *file, FLAC__StreamMetaData_SeekTable *block, unsigned block_length)
1785{
1786 unsigned i;
1787 FLAC__byte buffer[FLAC__STREAM_METADATA_SEEKPOINT_LENGTH];
1788
1789 FLAC__ASSERT(0 != file);
1790 FLAC__ASSERT(block_length % FLAC__STREAM_METADATA_SEEKPOINT_LENGTH == 0);
1791
1792 block->num_points = block_length / FLAC__STREAM_METADATA_SEEKPOINT_LENGTH;
1793
1794 if(0 == (block->points = malloc(block->num_points * sizeof(FLAC__StreamMetaData_SeekPoint))))
1795 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
1796
1797 for(i = 0; i < block->num_points; i++) {
1798 if(fread(buffer, 1, FLAC__STREAM_METADATA_SEEKPOINT_LENGTH, file) != FLAC__STREAM_METADATA_SEEKPOINT_LENGTH)
1799 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1800 /*@@@ some MAGIC NUMBERs here */
1801 block->points[i].sample_number = unpack_uint64_(buffer, 8);
1802 block->points[i].stream_offset = unpack_uint64_(buffer+8, 8);
1803 block->points[i].frame_samples = unpack_uint32_(buffer+16, 2);
1804 }
1805
1806 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1807}
1808
1809FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_entry_(FILE *file, FLAC__StreamMetaData_VorbisComment_Entry *entry)
1810{
1811 const unsigned entry_length_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8;
1812 FLAC__byte buffer[FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8];
1813
1814 FLAC__ASSERT(0 != file);
1815
1816 if(fread(buffer, 1, entry_length_len, file) != entry_length_len)
1817 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
Josh Coalsoneb643a22002-05-20 05:58:50 +00001818 entry->length = unpack_uint32_little_endian_(buffer, entry_length_len);
Josh Coalsoncae58732002-05-04 17:30:32 +00001819
1820 if(0 == (entry->entry = malloc(entry->length)))
1821 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
1822
1823 if(fread(entry->entry, 1, entry->length, file) != entry->length)
1824 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1825
1826 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1827}
1828
1829FLAC__MetaData_SimpleIteratorStatus read_metadata_block_data_vorbis_comment_(FILE *file, FLAC__StreamMetaData_VorbisComment *block)
1830{
1831 unsigned i;
1832 FLAC__MetaData_SimpleIteratorStatus status;
1833 const unsigned num_comments_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN / 8;
1834 FLAC__byte buffer[FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN / 8];
1835
1836 FLAC__ASSERT(0 != file);
1837
1838 if(FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK != (status = read_metadata_block_data_vorbis_comment_entry_(file, &(block->vendor_string))))
1839 return status;
1840
1841 if(fread(buffer, 1, num_comments_len, file) != num_comments_len)
1842 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
1843 block->num_comments = unpack_uint32_little_endian_(buffer, num_comments_len);
1844
1845 if(0 == (block->comments = malloc(block->num_comments * sizeof(FLAC__StreamMetaData_VorbisComment_Entry))))
1846 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
1847
1848 for(i = 0; i < block->num_comments; i++) {
1849 if(FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK != (status = read_metadata_block_data_vorbis_comment_entry_(file, block->comments + i)))
1850 return status;
1851 }
1852
1853 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1854}
1855
1856FLAC__bool write_metadata_block_header_(FILE *file, FLAC__MetaData_SimpleIteratorStatus *status, const FLAC__StreamMetaData *block)
1857{
Josh Coalsoneb643a22002-05-20 05:58:50 +00001858 FLAC__byte buffer[FLAC__STREAM_METADATA_HEADER_LENGTH];
Josh Coalsoncae58732002-05-04 17:30:32 +00001859
1860 FLAC__ASSERT(0 != file);
1861 FLAC__ASSERT(0 != status);
1862 FLAC__ASSERT(block->length < (1u << FLAC__STREAM_METADATA_LENGTH_LEN));
1863
1864 buffer[0] = (block->is_last? 0x80 : 0) | (FLAC__byte)block->type;
1865 pack_uint32_(block->length, buffer + 1, 3);
1866
Josh Coalsoneb643a22002-05-20 05:58:50 +00001867 if(fwrite(buffer, 1, FLAC__STREAM_METADATA_HEADER_LENGTH, file) != FLAC__STREAM_METADATA_HEADER_LENGTH) {
Josh Coalsoncae58732002-05-04 17:30:32 +00001868 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1869 return false;
1870 }
1871
1872 return true;
1873}
1874
1875FLAC__bool write_metadata_block_data_(FILE *file, FLAC__MetaData_SimpleIteratorStatus *status, const FLAC__StreamMetaData *block)
1876{
1877 FLAC__ASSERT(0 != file);
1878 FLAC__ASSERT(0 != status);
1879
1880 switch(block->type) {
1881 case FLAC__METADATA_TYPE_STREAMINFO:
1882 *status = write_metadata_block_data_streaminfo_(file, &block->data.stream_info);
1883 break;
1884 case FLAC__METADATA_TYPE_PADDING:
1885 *status = write_metadata_block_data_padding_(file, &block->data.padding, block->length);
1886 break;
1887 case FLAC__METADATA_TYPE_APPLICATION:
1888 *status = write_metadata_block_data_application_(file, &block->data.application, block->length);
1889 break;
1890 case FLAC__METADATA_TYPE_SEEKTABLE:
1891 *status = write_metadata_block_data_seektable_(file, &block->data.seek_table);
1892 break;
1893 case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1894 *status = write_metadata_block_data_vorbis_comment_(file, &block->data.vorbis_comment);
1895 break;
1896 default:
1897 FLAC__ASSERT(0);
1898 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_INTERNAL_ERROR;
1899 }
1900 return (*status == FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK);
1901}
1902
1903FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_streaminfo_(FILE *file, const FLAC__StreamMetaData_StreamInfo *block)
1904{
1905 FLAC__byte buffer[FLAC__STREAM_METADATA_STREAMINFO_LENGTH];
1906 const unsigned channels1 = block->channels - 1;
1907 const unsigned bps1 = block->bits_per_sample - 1;
1908
1909 FLAC__ASSERT(0 != file);
1910
1911 /* @@@ we are using hardcoded numbers for simplicity but we should
1912 * probably eventually write a bit-level packer and use the
1913 * _STREAMINFO_ constants.
1914 */
1915 pack_uint32_(block->min_blocksize, buffer, 2);
1916 pack_uint32_(block->max_blocksize, buffer+2, 2);
1917 pack_uint32_(block->min_framesize, buffer+4, 3);
1918 pack_uint32_(block->max_framesize, buffer+7, 3);
1919 buffer[10] = (block->sample_rate >> 12) & 0xff;
1920 buffer[11] = (block->sample_rate >> 4) & 0xff;
1921 buffer[12] = ((block->sample_rate & 0x0f) << 4) | (channels1 << 1) | (bps1 >> 4);
1922 buffer[13] = ((bps1 & 0x0f) << 4) | ((block->total_samples >> 32) & 0x0f);
1923 pack_uint32_((FLAC__uint32)block->total_samples, buffer+14, 4);
1924 memcpy(buffer+18, block->md5sum, 16);
1925
1926 if(fwrite(buffer, 1, FLAC__STREAM_METADATA_STREAMINFO_LENGTH, file) != FLAC__STREAM_METADATA_STREAMINFO_LENGTH)
1927 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1928
1929 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1930}
1931
1932FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_padding_(FILE *file, const FLAC__StreamMetaData_Padding *block, unsigned block_length)
1933{
1934 unsigned i, n = block_length;
1935 FLAC__byte buffer[1024];
1936
1937 FLAC__ASSERT(0 != file);
1938
1939 (void)block;
1940
1941 memset(buffer, 0, 1024);
1942
1943 for(i = 0; i < n/1024; i++)
1944 if(fwrite(buffer, 1, 1024, file) != 1024)
1945 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1946
1947 n %= 1024;
1948
1949 if(fwrite(buffer, 1, n, file) != n)
1950 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1951
1952 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1953}
1954
1955FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_application_(FILE *file, const FLAC__StreamMetaData_Application *block, unsigned block_length)
1956{
1957 const unsigned id_bytes = FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8;
1958
1959 FLAC__ASSERT(0 != file);
1960
1961 if(fwrite(block->id, 1, id_bytes, file) != id_bytes)
1962 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1963
1964 block_length -= id_bytes;
1965
1966 if(fwrite(block->data, 1, block_length, file) != block_length)
1967 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1968
1969 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1970}
1971
1972FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_seektable_(FILE *file, const FLAC__StreamMetaData_SeekTable *block)
1973{
1974 unsigned i;
1975 FLAC__byte buffer[FLAC__STREAM_METADATA_SEEKPOINT_LENGTH];
1976
1977 FLAC__ASSERT(0 != file);
1978
1979 for(i = 0; i < block->num_points; i++) {
1980 /*@@@ some MAGIC NUMBERs here */
1981 pack_uint64_(block->points[i].sample_number, buffer, 8);
1982 pack_uint64_(block->points[i].stream_offset, buffer+8, 8);
1983 pack_uint32_(block->points[i].frame_samples, buffer+16, 2);
1984 if(fwrite(buffer, 1, FLAC__STREAM_METADATA_SEEKPOINT_LENGTH, file) != FLAC__STREAM_METADATA_SEEKPOINT_LENGTH)
1985 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
1986 }
1987
1988 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
1989}
1990
1991FLAC__MetaData_SimpleIteratorStatus write_metadata_block_data_vorbis_comment_(FILE *file, const FLAC__StreamMetaData_VorbisComment *block)
1992{
1993 unsigned i;
1994 const unsigned entry_length_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN / 8;
1995 const unsigned num_comments_len = FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN / 8;
1996 FLAC__byte buffer[max(FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN, FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN) / 8];
1997
1998 FLAC__ASSERT(0 != file);
1999
2000 pack_uint32_little_endian_(block->vendor_string.length, buffer, entry_length_len);
2001 if(fwrite(buffer, 1, entry_length_len, file) != entry_length_len)
2002 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2003 if(fwrite(block->vendor_string.entry, 1, block->vendor_string.length, file) != block->vendor_string.length)
2004 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2005
2006 pack_uint32_little_endian_(block->num_comments, buffer, num_comments_len);
2007 if(fwrite(buffer, 1, num_comments_len, file) != num_comments_len)
2008 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2009
2010 for(i = 0; i < block->num_comments; i++) {
2011 pack_uint32_little_endian_(block->comments[i].length, buffer, entry_length_len);
2012 if(fwrite(buffer, 1, entry_length_len, file) != entry_length_len)
2013 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2014 if(fwrite(block->comments[i].entry, 1, block->comments[i].length, file) != block->comments[i].length)
2015 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2016 }
2017
2018 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK;
2019}
2020
2021FLAC__bool write_metadata_block_stationary_(FLAC__MetaData_SimpleIterator *iterator, const FLAC__StreamMetaData *block)
2022{
2023 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2024 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2025 return false;
2026 }
2027
2028 if(!write_metadata_block_header_(iterator->file, &iterator->status, block))
2029 return false;
2030
2031 if(!write_metadata_block_data_(iterator->file, &iterator->status, block))
2032 return false;
2033
2034 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2035 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2036 return false;
2037 }
2038
2039 return read_metadata_block_header_(iterator);
2040}
2041
2042FLAC__bool write_metadata_block_stationary_with_padding_(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block, unsigned padding_length, FLAC__bool padding_is_last)
2043{
2044 FLAC__StreamMetaData *padding;
2045
2046 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2047 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2048 return false;
2049 }
2050
2051 block->is_last = false;
2052
2053 if(!write_metadata_block_header_(iterator->file, &iterator->status, block))
2054 return false;
2055
2056 if(!write_metadata_block_data_(iterator->file, &iterator->status, block))
2057 return false;
2058
2059 if(0 == (padding = FLAC__metadata_object_new(FLAC__METADATA_TYPE_PADDING)))
2060 return FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2061
2062 padding->is_last = padding_is_last;
2063 padding->length = padding_length;
2064
2065 if(!write_metadata_block_header_(iterator->file, &iterator->status, padding)) {
2066 FLAC__metadata_object_delete(padding);
2067 return false;
2068 }
2069
2070 if(!write_metadata_block_data_(iterator->file, &iterator->status, padding)) {
2071 FLAC__metadata_object_delete(padding);
2072 return false;
2073 }
2074
2075 FLAC__metadata_object_delete(padding);
2076
2077 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2078 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2079 return false;
2080 }
2081
2082 return read_metadata_block_header_(iterator);
2083}
2084
2085FLAC__bool rewrite_whole_file_(FLAC__MetaData_SimpleIterator *iterator, FLAC__StreamMetaData *block, FLAC__bool append)
2086{
2087 FILE *tempfile;
2088 char *tempfilename;
Josh Coalson6b02a752002-05-10 06:42:02 +00002089 int fixup_is_last_code = 0; /* 0 => no need to change any is_last flags */
2090 long fixup_is_last_flag_offset = -1;
Josh Coalsoncae58732002-05-04 17:30:32 +00002091
Josh Coalson51ee4652002-05-09 05:46:23 +00002092 FLAC__ASSERT(0 != block || append == false);
2093
Josh Coalson6b02a752002-05-10 06:42:02 +00002094 if(iterator->is_last) {
2095 if(append) {
2096 fixup_is_last_code = 1; /* 1 => clear the is_last flag at the following offset */
2097 fixup_is_last_flag_offset = iterator->offset[iterator->depth];
2098 }
2099 else if(0 == block) {
2100 simple_iterator_push_(iterator);
2101 if(!FLAC__metadata_simple_iterator_prev(iterator)) {
2102 (void)simple_iterator_pop_(iterator);
2103 return false;
2104 }
2105 fixup_is_last_code = -1; /* -1 => set the is_last the flag at the following offset */
2106 fixup_is_last_flag_offset = iterator->offset[iterator->depth];
2107 if(!simple_iterator_pop_(iterator))
2108 return false;
2109 }
2110 }
2111
Josh Coalsoncae58732002-05-04 17:30:32 +00002112 if(!simple_iterator_copy_file_prefix_(iterator, &tempfile, &tempfilename, append))
2113 return false;
2114
2115 if(0 != block) {
2116 if(!write_metadata_block_header_(tempfile, &iterator->status, block)) {
2117 cleanup_tempfile_(&tempfile, &tempfilename);
2118 return false;
2119 }
2120
2121 if(!write_metadata_block_data_(tempfile, &iterator->status, block)) {
2122 cleanup_tempfile_(&tempfile, &tempfilename);
2123 return false;
2124 }
2125 }
2126
Josh Coalson6b02a752002-05-10 06:42:02 +00002127 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 +00002128 return false;
2129
Josh Coalson51ee4652002-05-09 05:46:23 +00002130 if(append)
2131 return FLAC__metadata_simple_iterator_next(iterator);
2132
Josh Coalsoncae58732002-05-04 17:30:32 +00002133 return true;
2134}
2135
2136FLAC__bool chain_rewrite_chain_(FLAC__MetaData_Chain *chain)
2137{
2138 FILE *f;
2139 FLAC__MetaData_Node *node;
2140 FLAC__MetaData_SimpleIteratorStatus status;
2141
2142 FLAC__ASSERT(0 != chain);
2143 FLAC__ASSERT(0 != chain->filename);
2144 FLAC__ASSERT(0 != chain->head);
2145
2146 if(0 == (f = fopen(chain->filename, "r+b"))) {
2147 chain->status = FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
2148 return false;
2149 }
2150 if(0 != fseek(f, chain->first_offset, SEEK_SET)) {
2151 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
2152 return false;
2153 }
2154
2155 for(node = chain->head; node; node = node->next) {
2156 if(!write_metadata_block_header_(f, &status, node->data)) {
2157 chain->status = get_equivalent_status_(status);
2158 return false;
2159 }
2160 if(!write_metadata_block_data_(f, &status, node->data)) {
2161 chain->status = get_equivalent_status_(status);
2162 return false;
2163 }
2164 }
2165
2166 /*FLAC__ASSERT(fflush(), ftell() == chain->last_offset);*/
2167
2168 (void)fclose(f);
2169
2170 return true;
2171}
2172
2173FLAC__bool chain_rewrite_file_(FLAC__MetaData_Chain *chain, const char *tempfile_path_prefix)
2174{
2175 FILE *f, *tempfile;
2176 char *tempfilename;
2177 FLAC__MetaData_SimpleIteratorStatus status;
2178 const FLAC__MetaData_Node *node;
2179
2180 FLAC__ASSERT(0 != chain);
2181 FLAC__ASSERT(0 != chain->filename);
2182 FLAC__ASSERT(0 != chain->head);
2183
2184 /* copy the file prefix (data up to first metadata block */
2185 if(0 == (f = fopen(chain->filename, "rb"))) {
2186 chain->status = FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
2187 return false;
2188 }
2189 if(!open_tempfile_(chain->filename, tempfile_path_prefix, &tempfile, &tempfilename, &status)) {
2190 chain->status = get_equivalent_status_(status);
2191 cleanup_tempfile_(&tempfile, &tempfilename);
2192 return false;
2193 }
2194 if(!copy_n_bytes_from_file_(f, tempfile, chain->first_offset, &status)) {
2195 chain->status = get_equivalent_status_(status);
2196 cleanup_tempfile_(&tempfile, &tempfilename);
2197 return false;
2198 }
2199
2200 /* write the metadata */
2201 for(node = chain->head; node; node = node->next) {
2202 if(!write_metadata_block_header_(tempfile, &status, node->data)) {
2203 chain->status = get_equivalent_status_(status);
2204 return false;
2205 }
2206 if(!write_metadata_block_data_(tempfile, &status, node->data)) {
2207 chain->status = get_equivalent_status_(status);
2208 return false;
2209 }
2210 }
2211 /*FLAC__ASSERT(fflush(), ftell() == chain->last_offset);*/
2212
2213 /* copy the file postfix (everything after the metadata) */
2214 if(0 != fseek(f, chain->last_offset, SEEK_SET)) {
2215 cleanup_tempfile_(&tempfile, &tempfilename);
2216 chain->status = FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
2217 return false;
2218 }
2219 if(!copy_remaining_bytes_from_file_(f, tempfile, &status)) {
2220 cleanup_tempfile_(&tempfile, &tempfilename);
2221 chain->status = get_equivalent_status_(status);
2222 return false;
2223 }
2224
2225 /* move the tempfile on top of the original */
2226 (void)fclose(f);
2227 if(!transport_tempfile_(chain->filename, &tempfile, &tempfilename, &status))
2228 return false;
2229
2230 return true;
2231}
2232
2233void simple_iterator_push_(FLAC__MetaData_SimpleIterator *iterator)
2234{
2235 FLAC__ASSERT(iterator->depth+1 < SIMPLE_ITERATOR_MAX_PUSH_DEPTH);
2236 iterator->offset[iterator->depth+1] = iterator->offset[iterator->depth];
2237 iterator->depth++;
2238}
2239
2240FLAC__bool simple_iterator_pop_(FLAC__MetaData_SimpleIterator *iterator)
2241{
2242 FLAC__ASSERT(iterator->depth > 0);
2243 iterator->depth--;
2244 if(0 != fseek(iterator->file, iterator->offset[iterator->depth], SEEK_SET)) {
2245 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2246 return false;
2247 }
2248
2249 return read_metadata_block_header_(iterator);
2250}
2251
2252unsigned seek_to_first_metadata_block_(FILE *f)
2253{
Josh Coalsoneb643a22002-05-20 05:58:50 +00002254 FLAC__byte signature[FLAC__STREAM_METADATA_HEADER_LENGTH];
Josh Coalsoncae58732002-05-04 17:30:32 +00002255
2256 FLAC__ASSERT(0 != f);
2257
2258 /*@@@@ skip id3v2, change comment about id3v2 in metadata.h*/
2259
2260 /* search for the fLaC signature */
Josh Coalsoneb643a22002-05-20 05:58:50 +00002261 if(fread(signature, 1, FLAC__STREAM_METADATA_HEADER_LENGTH, f) == FLAC__STREAM_METADATA_HEADER_LENGTH) {
2262 if(0 == memcmp(FLAC__STREAM_SYNC_STRING, signature, FLAC__STREAM_SYNC_LENGTH)) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002263 return 0;
2264 }
2265 else {
2266 return 2;
2267 }
2268 }
2269 else {
2270 return 1;
2271 }
2272}
2273
2274FLAC__bool simple_iterator_copy_file_prefix_(FLAC__MetaData_SimpleIterator *iterator, FILE **tempfile, char **tempfilename, FLAC__bool append)
2275{
Josh Coalsoneb643a22002-05-20 05:58:50 +00002276 const long offset_end = append? iterator->offset[iterator->depth] + FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length : iterator->offset[iterator->depth];
Josh Coalsoncae58732002-05-04 17:30:32 +00002277
2278 if(0 != fseek(iterator->file, 0, SEEK_SET)) {
2279 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2280 return false;
2281 }
2282 if(!open_tempfile_(iterator->filename, iterator->tempfile_path_prefix, tempfile, tempfilename, &iterator->status)) {
2283 cleanup_tempfile_(tempfile, tempfilename);
2284 return false;
2285 }
2286 if(!copy_n_bytes_from_file_(iterator->file, *tempfile, offset_end, &iterator->status)) {
2287 cleanup_tempfile_(tempfile, tempfilename);
2288 return false;
2289 }
2290
2291 return true;
2292}
2293
Josh Coalson6b02a752002-05-10 06:42:02 +00002294FLAC__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 +00002295{
2296 long save_offset = iterator->offset[iterator->depth]; /*@@@ 2G limit */
2297 FLAC__ASSERT(0 != *tempfile);
2298
Josh Coalsoneb643a22002-05-20 05:58:50 +00002299 if(0 != fseek(iterator->file, save_offset + FLAC__STREAM_METADATA_HEADER_LENGTH + iterator->length, SEEK_SET)) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002300 cleanup_tempfile_(tempfile, tempfilename);
2301 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2302 return false;
2303 }
2304 if(!copy_remaining_bytes_from_file_(iterator->file, *tempfile, &iterator->status)) {
2305 cleanup_tempfile_(tempfile, tempfilename);
2306 return false;
2307 }
2308
Josh Coalson6b02a752002-05-10 06:42:02 +00002309 if(fixup_is_last_code != 0) {
2310 /*
2311 * if code == 1, it means a block was appended to the end so
2312 * we have to clear the is_last flag of the previous block
2313 * if code == -1, it means the last block was deleted so
2314 * we have to set the is_last flag of the previous block
2315 */
Josh Coalson51ee4652002-05-09 05:46:23 +00002316 /*@@@ 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 +00002317 FLAC__byte x;
2318 if(0 != fseek(*tempfile, fixup_is_last_flag_offset, SEEK_SET)) {
2319 cleanup_tempfile_(tempfile, tempfilename);
2320 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2321 return false;
2322 }
2323 if(fread(&x, 1, 1, *tempfile) != 1) {
2324 cleanup_tempfile_(tempfile, tempfilename);
2325 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2326 return false;
2327 }
Josh Coalson6b02a752002-05-10 06:42:02 +00002328 if(fixup_is_last_code > 0) {
2329 FLAC__ASSERT(x & 0x80);
2330 x &= 0x7f;
2331 }
2332 else {
2333 FLAC__ASSERT(!(x & 0x80));
2334 x |= 0x80;
2335 }
Josh Coalsoncae58732002-05-04 17:30:32 +00002336 if(0 != fseek(*tempfile, fixup_is_last_flag_offset, SEEK_SET)) {
2337 cleanup_tempfile_(tempfile, tempfilename);
2338 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR;
2339 return false;
2340 }
2341 if(fwrite(&x, 1, 1, *tempfile) != 1) {
2342 cleanup_tempfile_(tempfile, tempfilename);
2343 iterator->status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2344 return false;
2345 }
2346 }
2347
2348 (void)fclose(iterator->file);
2349
2350 if(!transport_tempfile_(iterator->filename, tempfile, tempfilename, &iterator->status))
2351 return false;
2352
2353 if(iterator->has_stats)
2354 set_file_stats_(iterator->filename, &iterator->stats);
2355
2356 if(!simple_iterator_prime_input_(iterator))
2357 return false;
2358 if(backup) {
Josh Coalsoneb643a22002-05-20 05:58:50 +00002359 while(iterator->offset[iterator->depth] + (long)FLAC__STREAM_METADATA_HEADER_LENGTH + (long)iterator->length < save_offset)
Josh Coalsoncae58732002-05-04 17:30:32 +00002360 if(!FLAC__metadata_simple_iterator_next(iterator))
2361 return false;
2362 return true;
2363 }
2364 else {
2365 /* move the iterator to it's original block faster by faking a push, then doing a pop_ */
2366 FLAC__ASSERT(iterator->depth == 0);
2367 iterator->offset[0] = save_offset;
2368 iterator->depth++;
2369 return simple_iterator_pop_(iterator);
2370 }
2371}
2372
2373FLAC__bool copy_n_bytes_from_file_(FILE *file, FILE *tempfile, unsigned bytes/*@@@ 4G limit*/, FLAC__MetaData_SimpleIteratorStatus *status)
2374{
2375 FLAC__byte buffer[8192];
2376 unsigned n;
2377
2378 while(bytes > 0) {
2379 n = min(sizeof(buffer), bytes);
2380 if(fread(buffer, 1, n, file) != n) {
2381 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2382 return false;
2383 }
2384 if(fwrite(buffer, 1, n, tempfile) != n) {
2385 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2386 return false;
2387 }
2388 bytes -= n;
2389 }
2390
2391 return true;
2392}
2393
2394FLAC__bool copy_remaining_bytes_from_file_(FILE *file, FILE *tempfile, FLAC__MetaData_SimpleIteratorStatus *status)
2395{
2396 FLAC__byte buffer[8192];
2397 size_t n;
2398
2399 while(!feof(file)) {
2400 n = fread(buffer, 1, sizeof(buffer), file);
2401 if(n == 0 && !feof(file)) {
2402 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR;
2403 return false;
2404 }
2405 if(n > 0 && fwrite(buffer, 1, n, tempfile) != n) {
2406 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR;
2407 return false;
2408 }
2409 }
2410
2411 return true;
2412}
2413
2414FLAC__bool open_tempfile_(const char *filename, const char *tempfile_path_prefix, FILE **tempfile, char **tempfilename, FLAC__MetaData_SimpleIteratorStatus *status)
2415{
2416 static const char *tempfile_suffix = ".metadata_edit";
2417 if(0 == tempfile_path_prefix) {
2418 if(0 == (*tempfilename = malloc(strlen(filename) + strlen(tempfile_suffix) + 1))) {
2419 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2420 return false;
2421 }
2422 strcpy(*tempfilename, filename);
2423 strcat(*tempfilename, tempfile_suffix);
2424 }
2425 else {
2426 const char *p = strrchr(filename, '/');
2427 if(0 == p)
2428 p = filename;
2429 else
2430 p++;
2431
2432 if(0 == (*tempfilename = malloc(strlen(tempfile_path_prefix) + 1 + strlen(p) + strlen(tempfile_suffix) + 1))) {
2433 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR;
2434 return false;
2435 }
2436 strcpy(*tempfilename, tempfile_path_prefix);
2437 strcat(*tempfilename, "/");
2438 strcat(*tempfilename, p);
2439 strcat(*tempfilename, tempfile_suffix);
2440 }
2441
Josh Coalson51ee4652002-05-09 05:46:23 +00002442 if(0 == (*tempfile = fopen(*tempfilename, "w+b"))) {
Josh Coalsoncae58732002-05-04 17:30:32 +00002443 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE;
2444 return false;
2445 }
2446
2447 return true;
2448}
2449
2450FLAC__bool transport_tempfile_(const char *filename, FILE **tempfile, char **tempfilename, FLAC__MetaData_SimpleIteratorStatus *status)
2451{
2452 FLAC__ASSERT(0 != filename);
2453 FLAC__ASSERT(0 != tempfile);
2454 FLAC__ASSERT(0 != *tempfile);
2455 FLAC__ASSERT(0 != tempfilename);
2456 FLAC__ASSERT(0 != *tempfilename);
2457 FLAC__ASSERT(0 != status);
2458
2459 (void)fclose(*tempfile);
2460 *tempfile = 0;
2461 /*@@@ to fully support the tempfile_path_prefix we need to update this piece to actually copy across filesystems instead of just rename(): */
2462 if(0 != rename(*tempfilename, filename)) {
2463 cleanup_tempfile_(tempfile, tempfilename);
2464 *status = FLAC__METADATA_SIMPLE_ITERATOR_STATUS_RENAME_ERROR;
2465 return false;
2466 }
2467
2468 cleanup_tempfile_(tempfile, tempfilename);
2469
2470 return true;
2471}
2472
2473void cleanup_tempfile_(FILE **tempfile, char **tempfilename)
2474{
2475 if(0 != *tempfile) {
2476 (void)fclose(*tempfile);
2477 *tempfile = 0;
2478 }
2479
2480 if(0 != *tempfilename) {
2481 (void)unlink(*tempfilename);
2482 free(*tempfilename);
2483 *tempfilename = 0;
2484 }
2485}
2486
2487FLAC__bool get_file_stats_(const char *filename, struct stat *stats)
2488{
2489 FLAC__ASSERT(0 != filename);
2490 FLAC__ASSERT(0 != stats);
2491 return (0 == stat(filename, stats));
2492}
2493
2494void set_file_stats_(const char *filename, struct stat *stats)
2495{
2496 struct utimbuf srctime;
2497
2498 FLAC__ASSERT(0 != filename);
2499 FLAC__ASSERT(0 != stats);
2500
2501 srctime.actime = stats->st_atime;
2502 srctime.modtime = stats->st_mtime;
2503 (void)chmod(filename, stats->st_mode);
2504 (void)utime(filename, &srctime);
2505#if !defined _MSC_VER && !defined __MINGW32__
2506 (void)chown(filename, stats->st_uid, -1);
2507 (void)chown(filename, -1, stats->st_gid);
2508#endif
2509}
2510
2511FLAC__MetaData_ChainStatus get_equivalent_status_(FLAC__MetaData_SimpleIteratorStatus status)
2512{
2513 switch(status) {
2514 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_OK:
2515 return FLAC__METADATA_CHAIN_STATUS_OK;
2516 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ILLEGAL_INPUT:
2517 return FLAC__METADATA_CHAIN_STATUS_ILLEGAL_INPUT;
2518 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_ERROR_OPENING_FILE:
2519 return FLAC__METADATA_CHAIN_STATUS_ERROR_OPENING_FILE;
2520 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_A_FLAC_FILE:
2521 return FLAC__METADATA_CHAIN_STATUS_NOT_A_FLAC_FILE;
2522 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_NOT_WRITABLE:
2523 return FLAC__METADATA_CHAIN_STATUS_NOT_WRITABLE;
2524 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_READ_ERROR:
2525 return FLAC__METADATA_CHAIN_STATUS_READ_ERROR;
2526 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_SEEK_ERROR:
2527 return FLAC__METADATA_CHAIN_STATUS_SEEK_ERROR;
2528 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_WRITE_ERROR:
2529 return FLAC__METADATA_CHAIN_STATUS_WRITE_ERROR;
2530 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_RENAME_ERROR:
2531 return FLAC__METADATA_CHAIN_STATUS_RENAME_ERROR;
2532 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_UNLINK_ERROR:
2533 return FLAC__METADATA_CHAIN_STATUS_UNLINK_ERROR;
2534 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_MEMORY_ALLOCATION_ERROR:
2535 return FLAC__METADATA_CHAIN_STATUS_MEMORY_ALLOCATION_ERROR;
2536 case FLAC__METADATA_SIMPLE_ITERATOR_STATUS_INTERNAL_ERROR:
2537 default:
2538 return FLAC__METADATA_CHAIN_STATUS_INTERNAL_ERROR;
2539 }
2540}