/* test_libFLAC - Unit tester for libFLAC
- * Copyright (C) 2002 Josh Coalson
+ * Copyright (C) 2002,2003,2004,2005,2006,2007,2008,2009 Josh Coalson
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
-#include "encoders.h"
-#include "file_utils.h"
-#include "FLAC/assert.h"
-#include "FLAC/stream_encoder.h"
+#if HAVE_CONFIG_H
+# include <config.h>
+#endif
+
+#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
+#include "encoders.h"
+#include "FLAC/assert.h"
+#include "FLAC/stream_encoder.h"
+#include "share/grabbag.h"
+#include "share/compat.h"
+#include "test_libs_common/file_utils_flac.h"
+#include "test_libs_common/metadata_utils.h"
+
+typedef enum {
+ LAYER_STREAM = 0, /* FLAC__stream_encoder_init_[ogg_]stream() without seeking */
+ LAYER_SEEKABLE_STREAM, /* FLAC__stream_encoder_init_[ogg_]stream() with seeking */
+ LAYER_FILE, /* FLAC__stream_encoder_init_[ogg_]FILE() */
+ LAYER_FILENAME /* FLAC__stream_encoder_init_[ogg_]file() */
+} Layer;
+
+static const char * const LayerString[] = {
+ "Stream",
+ "Seekable Stream",
+ "FILE*",
+ "Filename"
+};
+
+static FLAC__StreamMetadata streaminfo_, padding_, seektable_, application1_, application2_, vorbiscomment_, cuesheet_, picture_, unknown_;
+static FLAC__StreamMetadata *metadata_sequence_[] = { &vorbiscomment_, &padding_, &seektable_, &application1_, &application2_, &cuesheet_, &picture_, &unknown_ };
+static const unsigned num_metadata_ = sizeof(metadata_sequence_) / sizeof(metadata_sequence_[0]);
+
+static const char *flacfilename(FLAC__bool is_ogg)
+{
+ return is_ogg? "metadata.oga" : "metadata.flac";
+}
-static FLAC__StreamMetadata streaminfo_, padding_, seektable_, application1_, application2_, vorbiscomment_;
-static FLAC__StreamMetadata *metadata_sequence_[] = { &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_ };
-static const unsigned num_metadata_ = 5;
-static const char *flacfilename_ = "metadata.flac";
+static FLAC__bool die_(const char *msg)
+{
+ printf("ERROR: %s\n", msg);
+ return false;
+}
static FLAC__bool die_s_(const char *msg, const FLAC__StreamEncoder *encoder)
{
FLAC__StreamEncoderState state = FLAC__stream_encoder_get_state(encoder);
if(msg)
- printf("FAILED, %s, state = %u (%s)\n", msg, (unsigned)state, FLAC__StreamEncoderStateString[state]);
+ printf("FAILED, %s", msg);
else
- printf("FAILED, state = %u (%s)\n", (unsigned)state, FLAC__StreamEncoderStateString[state]);
+ printf("FAILED");
+
+ printf(", state = %u (%s)\n", (unsigned)state, FLAC__StreamEncoderStateString[state]);
+ if(state == FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR) {
+ FLAC__StreamDecoderState dstate = FLAC__stream_encoder_get_verify_decoder_state(encoder);
+ printf(" verify decoder state = %u (%s)\n", (unsigned)dstate, FLAC__StreamDecoderStateString[dstate]);
+ }
return false;
}
-static void *malloc_or_die_(size_t size)
+static void init_metadata_blocks_(void)
+{
+ mutils__init_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_);
+}
+
+static void free_metadata_blocks_(void)
{
- void *x = malloc(size);
- if(0 == x) {
- fprintf(stderr, "ERROR: out of memory allocating %u bytes\n", (unsigned)size);
- exit(1);
+ mutils__free_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_);
+}
+
+static FLAC__StreamEncoderReadStatus stream_encoder_read_callback_(const FLAC__StreamEncoder *encoder, FLAC__byte buffer[], size_t *bytes, void *client_data)
+{
+ FILE *f = (FILE*)client_data;
+ (void)encoder;
+ if(*bytes > 0) {
+ *bytes = fread(buffer, sizeof(FLAC__byte), *bytes, f);
+ if(ferror(f))
+ return FLAC__STREAM_ENCODER_READ_STATUS_ABORT;
+ else if(*bytes == 0)
+ return FLAC__STREAM_ENCODER_READ_STATUS_END_OF_STREAM;
+ else
+ return FLAC__STREAM_ENCODER_READ_STATUS_CONTINUE;
}
- return x;
+ else
+ return FLAC__STREAM_ENCODER_READ_STATUS_ABORT;
}
-static void init_metadata_blocks_()
+static FLAC__StreamEncoderWriteStatus stream_encoder_write_callback_(const FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], size_t bytes, unsigned samples, unsigned current_frame, void *client_data)
{
- /*
- most of the actual numbers and data in the blocks don't matter,
- we just want to make sure the encoder encodes them correctly
-
- remember, the metadata interface gets tested after the encoders,
- so we do all the metadata manipulation here without it.
- */
-
- /* min/max_framesize and md5sum don't get written at first, so we have to leave them 0 */
- streaminfo_.is_last = false;
- streaminfo_.type = FLAC__METADATA_TYPE_STREAMINFO;
- streaminfo_.length = FLAC__STREAM_METADATA_STREAMINFO_LENGTH;
- streaminfo_.data.stream_info.min_blocksize = 576;
- streaminfo_.data.stream_info.max_blocksize = 576;
- streaminfo_.data.stream_info.min_framesize = 0;
- streaminfo_.data.stream_info.max_framesize = 0;
- streaminfo_.data.stream_info.sample_rate = 44100;
- streaminfo_.data.stream_info.channels = 1;
- streaminfo_.data.stream_info.bits_per_sample = 8;
- streaminfo_.data.stream_info.total_samples = 0;
- memset(streaminfo_.data.stream_info.md5sum, 0, 16);
-
- padding_.is_last = false;
- padding_.type = FLAC__METADATA_TYPE_PADDING;
- padding_.length = 1234;
-
- seektable_.is_last = false;
- seektable_.type = FLAC__METADATA_TYPE_SEEKTABLE;
- seektable_.data.seek_table.num_points = 2;
- seektable_.length = seektable_.data.seek_table.num_points * FLAC__STREAM_METADATA_SEEKPOINT_LENGTH;
- seektable_.data.seek_table.points = malloc_or_die_(seektable_.data.seek_table.num_points * sizeof(FLAC__StreamMetadata_SeekPoint));
- seektable_.data.seek_table.points[0].sample_number = 0;
- seektable_.data.seek_table.points[0].stream_offset = 0;
- seektable_.data.seek_table.points[0].frame_samples = streaminfo_.data.stream_info.min_blocksize;
- seektable_.data.seek_table.points[1].sample_number = FLAC__STREAM_METADATA_SEEKPOINT_PLACEHOLDER;
- seektable_.data.seek_table.points[1].stream_offset = 1000;
- seektable_.data.seek_table.points[1].frame_samples = streaminfo_.data.stream_info.min_blocksize;
-
- application1_.is_last = false;
- application1_.type = FLAC__METADATA_TYPE_APPLICATION;
- application1_.length = 8;
- memcpy(application1_.data.application.id, "\xfe\xdc\xba\x98", 4);
- application1_.data.application.data = malloc_or_die_(4);
- memcpy(application1_.data.application.data, "\xf0\xe1\xd2\xc3", 4);
-
- application2_.is_last = false;
- application2_.type = FLAC__METADATA_TYPE_APPLICATION;
- application2_.length = 4;
- memcpy(application2_.data.application.id, "\x76\x54\x32\x10", 4);
- application2_.data.application.data = 0;
-
- vorbiscomment_.is_last = true;
- vorbiscomment_.type = FLAC__METADATA_TYPE_VORBIS_COMMENT;
- vorbiscomment_.length = (4 + 8) + 4 + (4 + 5) + (4 + 0);
- vorbiscomment_.data.vorbis_comment.vendor_string.length = 8;
- vorbiscomment_.data.vorbis_comment.vendor_string.entry = malloc_or_die_(8);
- memcpy(vorbiscomment_.data.vorbis_comment.vendor_string.entry, "flac 1.x", 8);
- vorbiscomment_.data.vorbis_comment.num_comments = 2;
- vorbiscomment_.data.vorbis_comment.comments = malloc_or_die_(vorbiscomment_.data.vorbis_comment.num_comments * sizeof(FLAC__StreamMetadata_VorbisComment_Entry));
- vorbiscomment_.data.vorbis_comment.comments[0].length = 5;
- vorbiscomment_.data.vorbis_comment.comments[0].entry = malloc_or_die_(5);
- memcpy(vorbiscomment_.data.vorbis_comment.comments[0].entry, "ab=cd", 5);
- vorbiscomment_.data.vorbis_comment.comments[1].length = 0;
- vorbiscomment_.data.vorbis_comment.comments[1].entry = 0;
+ FILE *f = (FILE*)client_data;
+ (void)encoder, (void)samples, (void)current_frame;
+ if(fwrite(buffer, 1, bytes, f) != bytes)
+ return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
+ else
+ return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
}
-static void free_metadata_blocks_()
+static FLAC__StreamEncoderSeekStatus stream_encoder_seek_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 absolute_byte_offset, void *client_data)
{
- free(seektable_.data.seek_table.points);
- free(application1_.data.application.data);
- free(vorbiscomment_.data.vorbis_comment.vendor_string.entry);
- free(vorbiscomment_.data.vorbis_comment.comments[0].entry);
- free(vorbiscomment_.data.vorbis_comment.comments);
+ FILE *f = (FILE*)client_data;
+ (void)encoder;
+ if(fseek(f, (long)absolute_byte_offset, SEEK_SET) < 0)
+ return FLAC__STREAM_ENCODER_SEEK_STATUS_ERROR;
+ else
+ return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
}
-static FLAC__StreamEncoderWriteStatus encoder_write_callback_(const FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data)
+static FLAC__StreamEncoderTellStatus stream_encoder_tell_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 *absolute_byte_offset, void *client_data)
{
- (void)encoder, (void)buffer, (void)bytes, (void)samples, (void)current_frame, (void)client_data;
- return FLAC__STREAM_ENCODER_WRITE_OK;
+ FILE *f = (FILE*)client_data;
+ long pos;
+ (void)encoder;
+ if((pos = ftell(f)) < 0)
+ return FLAC__STREAM_ENCODER_TELL_STATUS_ERROR;
+ else {
+ *absolute_byte_offset = (FLAC__uint64)pos;
+ return FLAC__STREAM_ENCODER_TELL_STATUS_OK;
+ }
}
-static void encoder_metadata_callback_(const FLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data)
+static void stream_encoder_metadata_callback_(const FLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data)
{
(void)encoder, (void)metadata, (void)client_data;
}
-static FLAC__bool test_stream_encoder()
+static void stream_encoder_progress_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data)
+{
+ (void)encoder, (void)bytes_written, (void)samples_written, (void)frames_written, (void)total_frames_estimate, (void)client_data;
+}
+
+static FLAC__bool test_stream_encoder(Layer layer, FLAC__bool is_ogg)
{
FLAC__StreamEncoder *encoder;
+ FLAC__StreamEncoderInitStatus init_status;
FLAC__StreamEncoderState state;
+ FLAC__StreamDecoderState dstate;
+ FILE *file = 0;
FLAC__int32 samples[1024];
- FLAC__int32 *samples_array[1] = { samples };
+ FLAC__int32 *samples_array[1];
unsigned i;
- printf("\n+++ libFLAC unit test: FLAC__StreamEncoder\n\n");
+ samples_array[0] = samples;
+
+ printf("\n+++ libFLAC unit test: FLAC__StreamEncoder (layer: %s, format: %s)\n\n", LayerString[layer], is_ogg? "Ogg FLAC":"FLAC");
printf("testing FLAC__stream_encoder_new()... ");
encoder = FLAC__stream_encoder_new();
}
printf("OK\n");
- printf("testing FLAC__stream_encoder_set_streamable_subset()... ");
- if(!FLAC__stream_encoder_set_streamable_subset(encoder, true))
- return die_s_("returned false", encoder);
- printf("OK\n");
+ if(is_ogg) {
+ printf("testing FLAC__stream_encoder_set_ogg_serial_number()... ");
+ if(!FLAC__stream_encoder_set_ogg_serial_number(encoder, file_utils__ogg_serial_number))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+ }
- printf("testing FLAC__stream_encoder_set_do_mid_side_stereo()... ");
- if(!FLAC__stream_encoder_set_do_mid_side_stereo(encoder, false))
+ printf("testing FLAC__stream_encoder_set_verify()... ");
+ if(!FLAC__stream_encoder_set_verify(encoder, true))
return die_s_("returned false", encoder);
printf("OK\n");
- printf("testing FLAC__stream_encoder_set_loose_mid_side_stereo()... ");
- if(!FLAC__stream_encoder_set_loose_mid_side_stereo(encoder, false))
+ printf("testing FLAC__stream_encoder_set_streamable_subset()... ");
+ if(!FLAC__stream_encoder_set_streamable_subset(encoder, true))
return die_s_("returned false", encoder);
printf("OK\n");
return die_s_("returned false", encoder);
printf("OK\n");
+ printf("testing FLAC__stream_encoder_set_compression_level()... ");
+ if(!FLAC__stream_encoder_set_compression_level(encoder, (unsigned)(-1)))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+
printf("testing FLAC__stream_encoder_set_blocksize()... ");
if(!FLAC__stream_encoder_set_blocksize(encoder, streaminfo_.data.stream_info.min_blocksize))
return die_s_("returned false", encoder);
printf("OK\n");
+ printf("testing FLAC__stream_encoder_set_do_mid_side_stereo()... ");
+ if(!FLAC__stream_encoder_set_do_mid_side_stereo(encoder, false))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+
+ printf("testing FLAC__stream_encoder_set_loose_mid_side_stereo()... ");
+ if(!FLAC__stream_encoder_set_loose_mid_side_stereo(encoder, false))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+
printf("testing FLAC__stream_encoder_set_max_lpc_order()... ");
if(!FLAC__stream_encoder_set_max_lpc_order(encoder, 0))
return die_s_("returned false", encoder);
return die_s_("returned false", encoder);
printf("OK\n");
- printf("testing FLAC__stream_encoder_set_write_callback()... ");
- if(!FLAC__stream_encoder_set_write_callback(encoder, encoder_write_callback_))
- return die_s_("returned false", encoder);
- printf("OK\n");
-
- printf("testing FLAC__stream_encoder_set_metadata_callback()... ");
- if(!FLAC__stream_encoder_set_metadata_callback(encoder, encoder_metadata_callback_))
- return die_s_("returned false", encoder);
- printf("OK\n");
-
- printf("testing FLAC__stream_encoder_set_client_data()... ");
- if(!FLAC__stream_encoder_set_client_data(encoder, 0))
- return die_s_("returned false", encoder);
- printf("OK\n");
+ if(layer < LAYER_FILENAME) {
+ printf("opening file for FLAC output... ");
+ file = fopen(flacfilename(is_ogg), "w+b");
+ if(0 == file) {
+ printf("ERROR (%s)\n", strerror(errno));
+ return false;
+ }
+ printf("OK\n");
+ }
- printf("testing FLAC__stream_encoder_init()... ");
- if(FLAC__stream_encoder_init(encoder) != FLAC__STREAM_ENCODER_OK)
+ switch(layer) {
+ case LAYER_STREAM:
+ printf("testing FLAC__stream_encoder_init_%sstream()... ", is_ogg? "ogg_":"");
+ init_status = is_ogg?
+ FLAC__stream_encoder_init_ogg_stream(encoder, /*read_callback=*/0, stream_encoder_write_callback_, /*seek_callback=*/0, /*tell_callback=*/0, stream_encoder_metadata_callback_, /*client_data=*/file) :
+ FLAC__stream_encoder_init_stream(encoder, stream_encoder_write_callback_, /*seek_callback=*/0, /*tell_callback=*/0, stream_encoder_metadata_callback_, /*client_data=*/file);
+ break;
+ case LAYER_SEEKABLE_STREAM:
+ printf("testing FLAC__stream_encoder_init_%sstream()... ", is_ogg? "ogg_":"");
+ init_status = is_ogg?
+ FLAC__stream_encoder_init_ogg_stream(encoder, stream_encoder_read_callback_, stream_encoder_write_callback_, stream_encoder_seek_callback_, stream_encoder_tell_callback_, /*metadata_callback=*/0, /*client_data=*/file) :
+ FLAC__stream_encoder_init_stream(encoder, stream_encoder_write_callback_, stream_encoder_seek_callback_, stream_encoder_tell_callback_, /*metadata_callback=*/0, /*client_data=*/file);
+ break;
+ case LAYER_FILE:
+ printf("testing FLAC__stream_encoder_init_%sFILE()... ", is_ogg? "ogg_":"");
+ init_status = is_ogg?
+ FLAC__stream_encoder_init_ogg_FILE(encoder, file, stream_encoder_progress_callback_, /*client_data=*/0) :
+ FLAC__stream_encoder_init_FILE(encoder, file, stream_encoder_progress_callback_, /*client_data=*/0);
+ break;
+ case LAYER_FILENAME:
+ printf("testing FLAC__stream_encoder_init_%sfile()... ", is_ogg? "ogg_":"");
+ init_status = is_ogg?
+ FLAC__stream_encoder_init_ogg_file(encoder, flacfilename(is_ogg), stream_encoder_progress_callback_, /*client_data=*/0) :
+ FLAC__stream_encoder_init_file(encoder, flacfilename(is_ogg), stream_encoder_progress_callback_, /*client_data=*/0);
+ break;
+ default:
+ die_("internal error 001");
+ return false;
+ }
+ if(init_status != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
return die_s_(0, encoder);
printf("OK\n");
state = FLAC__stream_encoder_get_state(encoder);
printf("returned state = %u (%s)... OK\n", (unsigned)state, FLAC__StreamEncoderStateString[state]);
+ printf("testing FLAC__stream_encoder_get_verify_decoder_state()... ");
+ dstate = FLAC__stream_encoder_get_verify_decoder_state(encoder);
+ printf("returned state = %u (%s)... OK\n", (unsigned)dstate, FLAC__StreamDecoderStateString[dstate]);
+
+ {
+ FLAC__uint64 absolute_sample;
+ unsigned frame_number;
+ unsigned channel;
+ unsigned sample;
+ FLAC__int32 expected;
+ FLAC__int32 got;
+
+ printf("testing FLAC__stream_encoder_get_verify_decoder_error_stats()... ");
+ FLAC__stream_encoder_get_verify_decoder_error_stats(encoder, &absolute_sample, &frame_number, &channel, &sample, &expected, &got);
+ printf("OK\n");
+ }
+
+ printf("testing FLAC__stream_encoder_get_verify()... ");
+ if(FLAC__stream_encoder_get_verify(encoder) != true) {
+ printf("FAILED, expected true, got false\n");
+ return false;
+ }
+ printf("OK\n");
+
printf("testing FLAC__stream_encoder_get_streamable_subset()... ");
if(FLAC__stream_encoder_get_streamable_subset(encoder) != true) {
printf("FAILED, expected true, got false\n");
}
printf("OK\n");
+ printf("testing FLAC__stream_encoder_get_total_samples_estimate()... ");
+ if(FLAC__stream_encoder_get_total_samples_estimate(encoder) != streaminfo_.data.stream_info.total_samples) {
+ printf("FAILED, expected %" PRIu64 ", got %" PRIu64 "\n", streaminfo_.data.stream_info.total_samples, FLAC__stream_encoder_get_total_samples_estimate(encoder));
+ return false;
+ }
+ printf("OK\n");
+
/* init the dummy sample buffer */
for(i = 0; i < sizeof(samples) / sizeof(FLAC__int32); i++)
samples[i] = i & 7;
printf("OK\n");
printf("testing FLAC__stream_encoder_finish()... ");
- FLAC__stream_encoder_finish(encoder);
+ if(!FLAC__stream_encoder_finish(encoder))
+ return die_s_("returned false", encoder);
printf("OK\n");
+ if(layer < LAYER_FILE)
+ fclose(file);
+
printf("testing FLAC__stream_encoder_delete()... ");
FLAC__stream_encoder_delete(encoder);
printf("OK\n");
return true;
}
-FLAC__bool test_encoders()
+FLAC__bool test_encoders(void)
{
- init_metadata_blocks_();
+ FLAC__bool is_ogg = false;
- if(!test_stream_encoder())
- return false;
+ while(1) {
+ init_metadata_blocks_();
+
+ if(!test_stream_encoder(LAYER_STREAM, is_ogg))
+ return false;
+
+ if(!test_stream_encoder(LAYER_SEEKABLE_STREAM, is_ogg))
+ return false;
- (void) file_utils__remove_file(flacfilename_);
- free_metadata_blocks_();
+ if(!test_stream_encoder(LAYER_FILE, is_ogg))
+ return false;
+
+ if(!test_stream_encoder(LAYER_FILENAME, is_ogg))
+ return false;
+
+ (void) grabbag__file_remove_file(flacfilename(is_ogg));
+
+ free_metadata_blocks_();
+
+ if(!FLAC_API_SUPPORTS_OGG_FLAC || is_ogg)
+ break;
+ is_ogg = true;
+ }
return true;
}