163
* This wad of functions supports filename- and callback-based chain reading/writing.
164
* Everything up to set_file_stats_() is copied from libFLAC/metadata_iterators.c
166
FLAC__bool open_tempfile_(const char *filename, FILE **tempfile, char **tempfilename)
168
static const char *tempfile_suffix = ".metadata_edit";
170
if(0 == (*tempfilename = (char*)malloc(strlen(filename) + strlen(tempfile_suffix) + 1)))
172
strcpy(*tempfilename, filename);
173
strcat(*tempfilename, tempfile_suffix);
175
if(0 == (*tempfile = fopen(*tempfilename, "wb")))
181
void cleanup_tempfile_(FILE **tempfile, char **tempfilename)
184
(void)fclose(*tempfile);
188
if(0 != *tempfilename) {
189
(void)unlink(*tempfilename);
195
FLAC__bool transport_tempfile_(const char *filename, FILE **tempfile, char **tempfilename)
197
FLAC__ASSERT(0 != filename);
198
FLAC__ASSERT(0 != tempfile);
199
FLAC__ASSERT(0 != tempfilename);
200
FLAC__ASSERT(0 != *tempfilename);
203
(void)fclose(*tempfile);
207
#if defined _MSC_VER || defined __MINGW32__
208
if(unlink(filename) < 0) {
209
cleanup_tempfile_(tempfile, tempfilename);
214
if(0 != rename(*tempfilename, filename)) {
215
cleanup_tempfile_(tempfile, tempfilename);
219
cleanup_tempfile_(tempfile, tempfilename);
224
FLAC__bool get_file_stats_(const char *filename, struct stat *stats)
226
FLAC__ASSERT(0 != filename);
227
FLAC__ASSERT(0 != stats);
228
return (0 == stat(filename, stats));
231
void set_file_stats_(const char *filename, struct stat *stats)
233
struct utimbuf srctime;
235
FLAC__ASSERT(0 != filename);
236
FLAC__ASSERT(0 != stats);
238
srctime.actime = stats->st_atime;
239
srctime.modtime = stats->st_mtime;
240
(void)chmod(filename, stats->st_mode);
241
(void)utime(filename, &srctime);
242
#if !defined _MSC_VER && !defined __MINGW32__
243
(void)chown(filename, stats->st_uid, -1);
244
(void)chown(filename, -1, stats->st_gid);
248
#ifdef FLAC__VALGRIND_TESTING
249
static size_t chain_write_cb_(const void *ptr, size_t size, size_t nmemb, FLAC__IOHandle handle)
251
FILE *stream = (FILE*)handle;
252
size_t ret = fwrite(ptr, size, nmemb, stream);
259
static int chain_seek_cb_(FLAC__IOHandle handle, FLAC__int64 offset, int whence)
261
long o = (long)offset;
262
FLAC__ASSERT(offset == o);
263
return fseek((FILE*)handle, o, whence);
266
static FLAC__int64 chain_tell_cb_(FLAC__IOHandle handle)
268
return ftell((FILE*)handle);
271
static int chain_eof_cb_(FLAC__IOHandle handle)
273
return feof((FILE*)handle);
276
static FLAC__bool write_chain_(FLAC__Metadata_Chain *chain, FLAC__bool use_padding, FLAC__bool preserve_file_stats, FLAC__bool filename_based, const char *filename)
279
return FLAC__metadata_chain_write(chain, use_padding, preserve_file_stats);
281
FLAC__IOCallbacks callbacks;
283
memset(&callbacks, 0, sizeof(callbacks));
284
callbacks.read = (FLAC__IOCallback_Read)fread;
285
#ifdef FLAC__VALGRIND_TESTING
286
callbacks.write = chain_write_cb_;
288
callbacks.write = (FLAC__IOCallback_Write)fwrite;
290
callbacks.seek = chain_seek_cb_;
291
callbacks.eof = chain_eof_cb_;
293
if(FLAC__metadata_chain_check_if_tempfile_needed(chain, use_padding)) {
295
FILE *file, *tempfile;
297
if(preserve_file_stats) {
298
if(!get_file_stats_(filename, &stats))
301
if(0 == (file = fopen(filename, "rb")))
302
return false; /*@@@ chain status still says OK though */
303
if(!open_tempfile_(filename, &tempfile, &tempfilename)) {
305
cleanup_tempfile_(&tempfile, &tempfilename);
306
return false; /*@@@ chain status still says OK though */
308
if(!FLAC__metadata_chain_write_with_callbacks_and_tempfile(chain, use_padding, (FLAC__IOHandle)file, callbacks, (FLAC__IOHandle)tempfile, callbacks)) {
316
if(!transport_tempfile_(filename, &tempfile, &tempfilename))
318
if(preserve_file_stats)
319
set_file_stats_(filename, &stats);
322
FILE *file = fopen(filename, "r+b");
324
return false; /*@@@ chain status still says OK though */
325
if(!FLAC__metadata_chain_write_with_callbacks(chain, use_padding, (FLAC__IOHandle)file, callbacks))
334
static FLAC__bool read_chain_(FLAC__Metadata_Chain *chain, const char *filename, FLAC__bool filename_based)
337
return FLAC__metadata_chain_read(chain, flacfile_);
339
FLAC__IOCallbacks callbacks;
341
memset(&callbacks, 0, sizeof(callbacks));
342
callbacks.read = (FLAC__IOCallback_Read)fread;
343
callbacks.seek = chain_seek_cb_;
344
callbacks.tell = chain_tell_cb_;
348
FILE *file = fopen(filename, "rb");
350
return false; /*@@@ chain status still says OK though */
351
ret = FLAC__metadata_chain_read_with_callbacks(chain, (FLAC__IOHandle)file, callbacks);
151
358
/* function for comparing our metadata to a FLAC__Metadata_Chain */
153
360
static FLAC__bool compare_chain_(FLAC__Metadata_Chain *chain, unsigned current_position, FLAC__StreamMetadata *current_block)
1822
static FLAC__bool test_level_2_misc_()
1824
FLAC__Metadata_Iterator *iterator;
1825
FLAC__Metadata_Chain *chain;
1826
FLAC__IOCallbacks callbacks;
1828
memset(&callbacks, 0, sizeof(callbacks));
1829
callbacks.read = (FLAC__IOCallback_Read)fread;
1830
#ifdef FLAC__VALGRIND_TESTING
1831
callbacks.write = chain_write_cb_;
1833
callbacks.write = (FLAC__IOCallback_Write)fwrite;
1835
callbacks.seek = chain_seek_cb_;
1836
callbacks.tell = chain_tell_cb_;
1837
callbacks.eof = chain_eof_cb_;
1839
printf("\n\n++++++ testing level 2 interface (mismatched read/write protections)\n");
1841
printf("generate file\n");
1843
if(!generate_file_())
1846
printf("create chain\n");
1848
if(0 == (chain = FLAC__metadata_chain_new()))
1849
return die_("allocating chain");
1851
printf("read chain (filename-based)\n");
1853
if(!FLAC__metadata_chain_read(chain, flacfile_))
1854
return die_c_("reading chain", FLAC__metadata_chain_status(chain));
1856
printf("write chain with wrong method FLAC__metadata_chain_write_with_callbacks()\n");
1858
if(FLAC__metadata_chain_write_with_callbacks(chain, /*use_padding=*/false, 0, callbacks))
1859
return die_c_("mismatched write should have failed", FLAC__metadata_chain_status(chain));
1860
if(FLAC__metadata_chain_status(chain) != FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH)
1861
return die_c_("expected FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH", FLAC__metadata_chain_status(chain));
1862
printf(" OK: FLAC__metadata_chain_write_with_callbacks() returned false,FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH like it should\n");
1865
printf("read chain (filename-based)\n");
1867
if(!FLAC__metadata_chain_read(chain, flacfile_))
1868
return die_c_("reading chain", FLAC__metadata_chain_status(chain));
1870
printf("write chain with wrong method FLAC__metadata_chain_write_with_callbacks_and_tempfile()\n");
1872
if(FLAC__metadata_chain_write_with_callbacks_and_tempfile(chain, /*use_padding=*/false, 0, callbacks, 0, callbacks))
1873
return die_c_("mismatched write should have failed", FLAC__metadata_chain_status(chain));
1874
if(FLAC__metadata_chain_status(chain) != FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH)
1875
return die_c_("expected FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH", FLAC__metadata_chain_status(chain));
1876
printf(" OK: FLAC__metadata_chain_write_with_callbacks_and_tempfile() returned false,FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH like it should\n");
1879
printf("read chain (callback-based)\n");
1881
FILE *file = fopen(flacfile_, "rb");
1883
return die_("opening file");
1884
if(!FLAC__metadata_chain_read_with_callbacks(chain, (FLAC__IOHandle)file, callbacks)) {
1886
return die_c_("reading chain", FLAC__metadata_chain_status(chain));
1891
printf("write chain with wrong method FLAC__metadata_chain_write()\n");
1893
if(FLAC__metadata_chain_write(chain, /*use_padding=*/false, /*preserve_file_stats=*/false))
1894
return die_c_("mismatched write should have failed", FLAC__metadata_chain_status(chain));
1895
if(FLAC__metadata_chain_status(chain) != FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH)
1896
return die_c_("expected FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH", FLAC__metadata_chain_status(chain));
1897
printf(" OK: FLAC__metadata_chain_write() returned false,FLAC__METADATA_CHAIN_STATUS_READ_WRITE_MISMATCH like it should\n");
1900
printf("read chain (callback-based)\n");
1902
FILE *file = fopen(flacfile_, "rb");
1904
return die_("opening file");
1905
if(!FLAC__metadata_chain_read_with_callbacks(chain, (FLAC__IOHandle)file, callbacks)) {
1907
return die_c_("reading chain", FLAC__metadata_chain_status(chain));
1912
printf("testing FLAC__metadata_chain_check_if_tempfile_needed()... ");
1914
if(!FLAC__metadata_chain_check_if_tempfile_needed(chain, /*use_padding=*/false))
1915
printf("OK: FLAC__metadata_chain_check_if_tempfile_needed() returned false like it should\n");
1917
return die_("FLAC__metadata_chain_check_if_tempfile_needed() returned true but shouldn't have");
1919
printf("write chain with wrong method FLAC__metadata_chain_write_with_callbacks_and_tempfile()\n");
1921
if(FLAC__metadata_chain_write_with_callbacks_and_tempfile(chain, /*use_padding=*/false, 0, callbacks, 0, callbacks))
1922
return die_c_("mismatched write should have failed", FLAC__metadata_chain_status(chain));
1923
if(FLAC__metadata_chain_status(chain) != FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL)
1924
return die_c_("expected FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL", FLAC__metadata_chain_status(chain));
1925
printf(" OK: FLAC__metadata_chain_write_with_callbacks_and_tempfile() returned false,FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL like it should\n");
1928
printf("read chain (callback-based)\n");
1930
FILE *file = fopen(flacfile_, "rb");
1932
return die_("opening file");
1933
if(!FLAC__metadata_chain_read_with_callbacks(chain, (FLAC__IOHandle)file, callbacks)) {
1935
return die_c_("reading chain", FLAC__metadata_chain_status(chain));
1940
printf("create iterator\n");
1941
if(0 == (iterator = FLAC__metadata_iterator_new()))
1942
return die_("allocating memory for iterator");
1944
FLAC__metadata_iterator_init(iterator, chain);
1946
printf("[S]VP\tnext\n");
1947
if(!FLAC__metadata_iterator_next(iterator))
1948
return die_("iterator ended early\n");
1950
printf("S[V]P\tdelete VORBIS_COMMENT, write\n");
1951
if(!FLAC__metadata_iterator_delete_block(iterator, /*replace_with_padding=*/false))
1952
return die_c_("block delete failed\n", FLAC__metadata_chain_status(chain));
1954
printf("testing FLAC__metadata_chain_check_if_tempfile_needed()... ");
1956
if(FLAC__metadata_chain_check_if_tempfile_needed(chain, /*use_padding=*/false))
1957
printf("OK: FLAC__metadata_chain_check_if_tempfile_needed() returned true like it should\n");
1959
return die_("FLAC__metadata_chain_check_if_tempfile_needed() returned false but shouldn't have");
1961
printf("write chain with wrong method FLAC__metadata_chain_write_with_callbacks()\n");
1963
if(FLAC__metadata_chain_write_with_callbacks(chain, /*use_padding=*/false, 0, callbacks))
1964
return die_c_("mismatched write should have failed", FLAC__metadata_chain_status(chain));
1965
if(FLAC__metadata_chain_status(chain) != FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL)
1966
return die_c_("expected FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL", FLAC__metadata_chain_status(chain));
1967
printf(" OK: FLAC__metadata_chain_write_with_callbacks() returned false,FLAC__METADATA_CHAIN_STATUS_WRONG_WRITE_CALL like it should\n");
1970
printf("delete iterator\n");
1972
FLAC__metadata_iterator_delete(iterator);
1974
printf("delete chain\n");
1976
FLAC__metadata_chain_delete(chain);
1978
if(!remove_file_(flacfile_))
1597
1984
FLAC__bool test_metadata_file_manipulation()
1599
1986
printf("\n+++ libFLAC unit test: metadata manipulation\n\n");