main.c 14.4 KB
Newer Older
Josh Coalson's avatar
Josh Coalson committed
1
/* test_seeking - Seeking tester for libFLAC and libOggFLAC
Josh Coalson's avatar
Josh Coalson committed
2
 * Copyright (C) 2004,2005,2006  Josh Coalson
Josh Coalson's avatar
Josh Coalson committed
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * 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.
 */

Josh Coalson's avatar
Josh Coalson committed
19
20
#if HAVE_CONFIG_H
#  include <config.h>
21
22
#endif

Josh Coalson's avatar
Josh Coalson committed
23
24
25
26
27
28
29
30
31
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#if defined _MSC_VER || defined __MINGW32__
#include <time.h>
#else
#include <sys/time.h>
#endif
32
#include <sys/stat.h> /* for stat() */
Josh Coalson's avatar
Josh Coalson committed
33
#include "FLAC/assert.h"
34
#include "FLAC/stream_decoder.h"
35
#ifdef FLAC__HAS_OGG
36
#include "OggFLAC/stream_decoder.h"
37
#endif
Josh Coalson's avatar
Josh Coalson committed
38
39

typedef struct {
40
	FLAC__bool got_data;
Josh Coalson's avatar
Josh Coalson committed
41
	FLAC__uint64 total_samples;
42
43
	unsigned channels;
	unsigned bits_per_sample;
44
	FLAC__bool quiet;
Josh Coalson's avatar
Josh Coalson committed
45
46
	FLAC__bool ignore_errors;
	FLAC__bool error_occurred;
47
} DecoderClientData;
Josh Coalson's avatar
Josh Coalson committed
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

static FLAC__bool stop_signal_ = false;

static void our_sigint_handler_(int signal)
{
	(void)signal;
	printf("(caught SIGINT) ");
	fflush(stdout);
	stop_signal_ = true;
}

static FLAC__bool die_(const char *msg)
{
	printf("ERROR: %s\n", msg);
	return false;
}

65
static FLAC__bool die_s_(const char *msg, const FLAC__StreamDecoder *decoder)
Josh Coalson's avatar
Josh Coalson committed
66
{
67
	FLAC__StreamDecoderState state = FLAC__stream_decoder_get_state(decoder);
Josh Coalson's avatar
Josh Coalson committed
68
69
70
71
72
73

	if(msg)
		printf("FAILED, %s", msg);
	else
		printf("FAILED");

74
	printf(", state = %u (%s)\n", (unsigned)state, FLAC__StreamDecoderStateString[state]);
Josh Coalson's avatar
Josh Coalson committed
75
76
77
78

	return false;
}

79
#ifdef FLAC__HAS_OGG
80
static FLAC__bool die_os_(const char *msg, const OggFLAC__StreamDecoder *decoder)
Josh Coalson's avatar
Josh Coalson committed
81
{
82
	OggFLAC__StreamDecoderState state = OggFLAC__stream_decoder_get_state(decoder);
Josh Coalson's avatar
Josh Coalson committed
83
84
85
86
87
88

	if(msg)
		printf("FAILED, %s", msg);
	else
		printf("FAILED");

89
90
91
92
	printf(", state = %u (%s)\n", (unsigned)state, OggFLAC__StreamDecoderStateString[state]);
	if(state == OggFLAC__STREAM_DECODER_FLAC_STREAM_DECODER_ERROR) {
		FLAC__StreamDecoderState state_ = OggFLAC__stream_decoder_get_FLAC_stream_decoder_state(decoder);
		printf("      FLAC stream decoder state = %u (%s)\n", (unsigned)state_, FLAC__StreamDecoderStateString[state_]);
Josh Coalson's avatar
Josh Coalson committed
93
94
95
96
	}

	return false;
}
97
#endif
Josh Coalson's avatar
Josh Coalson committed
98

99
100
101
102
103
104
105
106
107
108
static off_t get_filesize_(const char *srcpath)
{
	struct stat srcstat;

	if(0 == stat(srcpath, &srcstat))
		return srcstat.st_size;
	else
		return -1;
}

109
static FLAC__StreamDecoderWriteStatus write_callback_(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data)
Josh Coalson's avatar
Josh Coalson committed
110
{
111
	DecoderClientData *dcd = (DecoderClientData*)client_data;
Josh Coalson's avatar
Josh Coalson committed
112
113
114
115
116
117
118
119
120
121
122

	(void)decoder, (void)buffer;

	if(0 == dcd) {
		printf("ERROR: client_data in write callback is NULL\n");
		return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
	}

	if(dcd->error_occurred)
		return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;

123
124
125
126
127
128
129
130
	if (frame->header.number_type == FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER) {
		if (!dcd->quiet)
			printf("frame@%uf(%u)... ", frame->header.number.frame_number, frame->header.blocksize);
	}
	else if (frame->header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER) {
		if (!dcd->quiet)
			printf("frame@%llu(%u)... ", frame->header.number.sample_number, frame->header.blocksize);
	}
Josh Coalson's avatar
Josh Coalson committed
131
132
133
134
135
136
137
138
139
140
	else {
		FLAC__ASSERT(0);
		dcd->error_occurred = true;
		return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
	}
	fflush(stdout);

	return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
}

141
static void metadata_callback_(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data)
Josh Coalson's avatar
Josh Coalson committed
142
{
143
	DecoderClientData *dcd = (DecoderClientData*)client_data;
Josh Coalson's avatar
Josh Coalson committed
144
145
146
147
148
149
150
151
152
153
154

	(void)decoder;

	if(0 == dcd) {
		printf("ERROR: client_data in metadata callback is NULL\n");
		return;
	}

	if(dcd->error_occurred)
		return;

155
156
	if (!dcd->got_data && metadata->type == FLAC__METADATA_TYPE_STREAMINFO) {
		dcd->got_data = true;
Josh Coalson's avatar
Josh Coalson committed
157
		dcd->total_samples = metadata->data.stream_info.total_samples;
158
159
160
		dcd->channels = metadata->data.stream_info.channels;
		dcd->bits_per_sample = metadata->data.stream_info.bits_per_sample;
	}
Josh Coalson's avatar
Josh Coalson committed
161
162
}

163
static void error_callback_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
Josh Coalson's avatar
Josh Coalson committed
164
{
165
	DecoderClientData *dcd = (DecoderClientData*)client_data;
Josh Coalson's avatar
Josh Coalson committed
166
167
168
169
170
171
172
173
174
175
176
177
178
179

	(void)decoder;

	if(0 == dcd) {
		printf("ERROR: client_data in error callback is NULL\n");
		return;
	}

	if(!dcd->ignore_errors) {
		printf("ERROR: got error callback: err = %u (%s)\n", (unsigned)status, FLAC__StreamDecoderErrorStatusString[status]);
		dcd->error_occurred = true;
	}
}

180
static FLAC__bool seek_barrage_native_flac(const char *filename, off_t filesize, unsigned count)
Josh Coalson's avatar
Josh Coalson committed
181
{
182
183
	FLAC__StreamDecoder *decoder;
	DecoderClientData decoder_client_data;
Josh Coalson's avatar
Josh Coalson committed
184
185
186
	unsigned i;
	long int n;

187
	decoder_client_data.got_data = false;
Josh Coalson's avatar
Josh Coalson committed
188
	decoder_client_data.total_samples = 0;
189
	decoder_client_data.quiet = false;
Josh Coalson's avatar
Josh Coalson committed
190
191
192
	decoder_client_data.ignore_errors = false;
	decoder_client_data.error_occurred = false;

193
	printf("\n+++ seek test: FLAC__StreamDecoder\n\n");
Josh Coalson's avatar
Josh Coalson committed
194

195
	decoder = FLAC__stream_decoder_new();
Josh Coalson's avatar
Josh Coalson committed
196
	if(0 == decoder)
197
		return die_("FLAC__stream_decoder_new() FAILED, returned NULL\n");
Josh Coalson's avatar
Josh Coalson committed
198

199
200
	if(FLAC__stream_decoder_init_file(decoder, filename, write_callback_, metadata_callback_, error_callback_, &decoder_client_data) != FLAC__STREAM_DECODER_INIT_STATUS_OK)
		return die_s_("FLAC__stream_decoder_init_file() FAILED", decoder);
Josh Coalson's avatar
Josh Coalson committed
201

202
203
	if(!FLAC__stream_decoder_process_until_end_of_metadata(decoder))
		return die_s_("FLAC__stream_decoder_process_until_end_of_metadata() FAILED", decoder);
Josh Coalson's avatar
Josh Coalson committed
204

205
206
207
208
209
210
211
212
213
214
	/* process until end of stream to make sure we can still seek in that state */
	decoder_client_data.quiet = true;
	if(!FLAC__stream_decoder_process_until_end_of_stream(decoder))
		return die_s_("FLAC__stream_decoder_process_until_end_of_stream() FAILED", decoder);
	decoder_client_data.quiet = false;

	printf("stream decoder state is %s\n", FLAC__stream_decoder_get_resolved_state_string(decoder));
	if(FLAC__stream_decoder_get_state(decoder) != FLAC__STREAM_DECODER_END_OF_STREAM)
		return die_s_("expected FLAC__STREAM_DECODER_END_OF_STREAM", decoder);

Josh Coalson's avatar
Josh Coalson committed
215
	printf("file's total_samples is %llu\n", decoder_client_data.total_samples);
216
#if !defined _MSC_VER && !defined __MINGW32__ && !defined __EMX__
217
218
	if (decoder_client_data.total_samples > (FLAC__uint64)RAND_MAX) {
		printf("ERROR: must be total_samples < %u\n", (unsigned)RAND_MAX);
Josh Coalson's avatar
Josh Coalson committed
219
220
		return false;
	}
Josh Coalson's avatar
Josh Coalson committed
221
#endif
Josh Coalson's avatar
Josh Coalson committed
222
223
	n = (long int)decoder_client_data.total_samples;

224
225
226
227
	/* if we don't have a total samples count, just guess based on the file size */
	if(n == 0) {
		/* 8 would imply no compression, 9 guarantees that we will get some samples off the end of the stream to test that case */
		n = 9 * filesize / (decoder_client_data.channels * decoder_client_data.bits_per_sample);
Josh Coalson's avatar
Josh Coalson committed
228
#if !defined _MSC_VER && !defined __MINGW32__
229
230
		if(n > RAND_MAX)
			n = RAND_MAX;
Josh Coalson's avatar
Josh Coalson committed
231
#endif
232
233
	}

Josh Coalson's avatar
Josh Coalson committed
234
235
236
237
238
	printf("Begin seek barrage, count=%u\n", count);

	for (i = 0; !stop_signal_ && (count == 0 || i < count); i++) {
		FLAC__uint64 pos;

239
240
241
242
243
244
245
246
		/* for the first 10, seek to the first 10 samples */
		if (n >= 10 && i < 10) {
			pos = i;
		}
		/* for the second 10, seek to the last 10 samples */
		else if (n >= 10 && i < 20) {
			pos = n - 1 - (i-10);
		}
247
248
249
250
		/* for the third 10, seek past the end and make sure we fail properly as expected */
		else if (i < 30) {
			pos = n + (i-20);
		}
251
		else {
Josh Coalson's avatar
Josh Coalson committed
252
#if !defined _MSC_VER && !defined __MINGW32__
Josh Coalson's avatar
Josh Coalson committed
253
			pos = (FLAC__uint64)(random() % n);
Josh Coalson's avatar
Josh Coalson committed
254
255
#else
			/* RAND_MAX is only 32767 in my MSVC */
Josh Coalson's avatar
Josh Coalson committed
256
			pos = (FLAC__uint64)((rand()<<15|rand()) % n);
Josh Coalson's avatar
Josh Coalson committed
257
#endif
258
		}
Josh Coalson's avatar
Josh Coalson committed
259
260
261

		printf("seek(%llu)... ", pos);
		fflush(stdout);
262
		if(!FLAC__stream_decoder_seek_absolute(decoder, pos)) {
263
			if(pos < (FLAC__uint64)n && decoder_client_data.total_samples != 0)
264
				return die_s_("FLAC__stream_decoder_seek_absolute() FAILED", decoder);
265
266
267
268
			else if(decoder_client_data.total_samples == 0)
				printf("seek failed, assuming it was past EOF... ");
			else
				printf("seek past end failed as expected... ");
269
270
			if(!FLAC__stream_decoder_flush(decoder))
				return die_s_("FLAC__stream_decoder_flush() FAILED", decoder);
271
272
273
274
		}
		else {
			printf("decode_frame... ");
			fflush(stdout);
275
276
			if(!FLAC__stream_decoder_process_single(decoder))
				return die_s_("FLAC__stream_decoder_process_single() FAILED", decoder);
277
278
279

			printf("decode_frame... ");
			fflush(stdout);
280
281
			if(!FLAC__stream_decoder_process_single(decoder))
				return die_s_("FLAC__stream_decoder_process_single() FAILED", decoder);
282
		}
Josh Coalson's avatar
Josh Coalson committed
283
284
285
286
287

		printf("OK\n");
		fflush(stdout);
	}

288
289
290
	if(FLAC__stream_decoder_get_state(decoder) != FLAC__STREAM_DECODER_UNINITIALIZED) {
		if(!FLAC__stream_decoder_finish(decoder))
			return die_s_("FLAC__stream_decoder_finish() FAILED", decoder);
291
292
	}

Josh Coalson's avatar
Josh Coalson committed
293
294
295
296
297
	printf("\nPASSED!\n");

	return true;
}

298
#ifdef FLAC__HAS_OGG
299
static FLAC__bool seek_barrage_ogg_flac(const char *filename, off_t filesize, unsigned count)
Josh Coalson's avatar
Josh Coalson committed
300
{
301
302
	OggFLAC__StreamDecoder *decoder;
	DecoderClientData decoder_client_data;
Josh Coalson's avatar
Josh Coalson committed
303
304
305
	unsigned i;
	long int n;

306
	decoder_client_data.got_data = false;
Josh Coalson's avatar
Josh Coalson committed
307
	decoder_client_data.total_samples = 0;
308
	decoder_client_data.quiet = false;
Josh Coalson's avatar
Josh Coalson committed
309
310
311
	decoder_client_data.ignore_errors = false;
	decoder_client_data.error_occurred = false;

312
	printf("\n+++ seek test: OggFLAC__StreamDecoder\n\n");
Josh Coalson's avatar
Josh Coalson committed
313

314
	decoder = OggFLAC__stream_decoder_new();
Josh Coalson's avatar
Josh Coalson committed
315
	if(0 == decoder)
316
		return die_("OggFLAC__stream_decoder_new() FAILED, returned NULL\n");
Josh Coalson's avatar
Josh Coalson committed
317

318
319
	if(OggFLAC__stream_decoder_init_file(decoder, filename, write_callback_, metadata_callback_, error_callback_, &decoder_client_data) != FLAC__STREAM_DECODER_INIT_STATUS_OK)
		return die_os_("OggFLAC__stream_decoder_init_file() FAILED", decoder);
Josh Coalson's avatar
Josh Coalson committed
320

321
322
	if(!OggFLAC__stream_decoder_process_until_end_of_metadata(decoder))
		return die_os_("OggFLAC__stream_decoder_process_until_end_of_metadata() FAILED", decoder);
Josh Coalson's avatar
Josh Coalson committed
323

324
325
326
327
328
329
330
331
332
333
334
335
336
	/* process until end of stream to make sure we can still seek in that state */
#if 0
	/* not necessary for the Ogg seeking method */
	decoder_client_data.quiet = true;
	if(!OggFLAC__stream_decoder_process_until_end_of_stream(decoder))
		return die_os_("OggFLAC__stream_decoder_process_until_end_of_stream() FAILED", decoder);
	decoder_client_data.quiet = false;

	printf("stream decoder state is %s\n", OggFLAC__stream_decoder_get_resolved_state_string(decoder));
	if(OggFLAC__stream_decoder_get_state(decoder) != OggFLAC__STREAM_DECODER_END_OF_STREAM)
		return die_os_("expected OggFLAC__STREAM_DECODER_END_OF_STREAM", decoder);
#endif

Josh Coalson's avatar
Josh Coalson committed
337
	printf("file's total_samples is %llu\n", decoder_client_data.total_samples);
Josh Coalson's avatar
Josh Coalson committed
338
#if !defined _MSC_VER && !defined __MINGW32__ && !defined __EMX__
339
340
	if (decoder_client_data.total_samples > (FLAC__uint64)RAND_MAX) {
		printf("ERROR: must be total_samples < %u\n", (unsigned)RAND_MAX);
Josh Coalson's avatar
Josh Coalson committed
341
342
		return false;
	}
Josh Coalson's avatar
Josh Coalson committed
343
#endif
Josh Coalson's avatar
Josh Coalson committed
344
345
	n = (long int)decoder_client_data.total_samples;

346
	/* if we don't have a total samples count, just guess based on the file size */
347
	/* @@@ should get it from last page's granulepos */
348
349
350
	if(n == 0) {
		/* 8 would imply no compression, 9 guarantees that we will get some samples off the end of the stream to test that case */
		n = 9 * filesize / (decoder_client_data.channels * decoder_client_data.bits_per_sample);
Josh Coalson's avatar
Josh Coalson committed
351
#if !defined _MSC_VER && !defined __MINGW32__
352
353
		if(n > RAND_MAX)
			n = RAND_MAX;
Josh Coalson's avatar
Josh Coalson committed
354
#endif
355
356
	}

Josh Coalson's avatar
Josh Coalson committed
357
358
359
360
361
	printf("Begin seek barrage, count=%u\n", count);

	for (i = 0; !stop_signal_ && (count == 0 || i < count); i++) {
		FLAC__uint64 pos;

362
363
364
365
366
367
368
369
		/* for the first 10, seek to the first 10 samples */
		if (n >= 10 && i < 10) {
			pos = i;
		}
		/* for the second 10, seek to the last 10 samples */
		else if (n >= 10 && i < 20) {
			pos = n - 1 - (i-10);
		}
370
371
372
373
		/* for the third 10, seek past the end and make sure we fail properly as expected */
		else if (i < 30) {
			pos = n + (i-20);
		}
374
		else {
Josh Coalson's avatar
Josh Coalson committed
375
#if !defined _MSC_VER && !defined __MINGW32__
Josh Coalson's avatar
Josh Coalson committed
376
			pos = (FLAC__uint64)(random() % n);
Josh Coalson's avatar
Josh Coalson committed
377
378
#else
			/* RAND_MAX is only 32767 in my MSVC */
Josh Coalson's avatar
Josh Coalson committed
379
			pos = (FLAC__uint64)((rand()<<15|rand()) % n);
Josh Coalson's avatar
Josh Coalson committed
380
#endif
381
		}
Josh Coalson's avatar
Josh Coalson committed
382
383
384

		printf("seek(%llu)... ", pos);
		fflush(stdout);
385
		if(!OggFLAC__stream_decoder_seek_absolute(decoder, pos)) {
386
			if(pos < (FLAC__uint64)n && decoder_client_data.total_samples != 0)
387
				return die_os_("OggFLAC__stream_decoder_seek_absolute() FAILED", decoder);
388
389
390
391
			else if(decoder_client_data.total_samples == 0)
				printf("seek failed, assuming it was past EOF... ");
			else
				printf("seek past end failed as expected... ");
392
393
			if(!OggFLAC__stream_decoder_flush(decoder))
				return die_os_("OggFLAC__stream_decoder_flush() FAILED", decoder);
394
395
396
397
		}
		else {
			printf("decode_frame... ");
			fflush(stdout);
398
399
			if(!OggFLAC__stream_decoder_process_single(decoder))
				return die_os_("OggFLAC__stream_decoder_process_single() FAILED", decoder);
400
401
402

			printf("decode_frame... ");
			fflush(stdout);
403
404
			if(!OggFLAC__stream_decoder_process_single(decoder))
				return die_os_("OggFLAC__stream_decoder_process_single() FAILED", decoder);
405
		}
Josh Coalson's avatar
Josh Coalson committed
406
407
408
409
410

		printf("OK\n");
		fflush(stdout);
	}

411
412
413
	if(OggFLAC__stream_decoder_get_state(decoder) != OggFLAC__STREAM_DECODER_UNINITIALIZED) {
		if(!OggFLAC__stream_decoder_finish(decoder))
			return die_os_("OggFLAC__stream_decoder_finish() FAILED", decoder);
414
415
	}

Josh Coalson's avatar
Josh Coalson committed
416
417
418
419
	printf("\nPASSED!\n");

	return true;
}
420
#endif
Josh Coalson's avatar
Josh Coalson committed
421
422
423
424
425

int main(int argc, char *argv[])
{
	const char *filename;
	unsigned count = 0;
426
	off_t filesize;
Josh Coalson's avatar
Josh Coalson committed
427
428
429
430
431
432
433
434
435
436
437
438
439

	static const char * const usage = "usage: test_seeking file.flac [#seeks]\n";

	if (argc < 1 || argc > 3) {
		fprintf(stderr, usage);
		return 1;
	}

	filename = argv[1];

	if (argc > 2)
		count = strtoul(argv[2], 0, 10);

440
441
	if (count < 30)
		fprintf(stderr, "WARNING: random seeks don't kick in until after 30 preprogrammed ones\n");
442

Josh Coalson's avatar
Josh Coalson committed
443
444
445
446
#if !defined _MSC_VER && !defined __MINGW32__
	{
		struct timeval tv;

447
		if (gettimeofday(&tv, 0) < 0) {
Josh Coalson's avatar
Josh Coalson committed
448
449
450
451
452
453
454
455
456
			fprintf(stderr, "WARNING: couldn't seed RNG with time\n");
			tv.tv_usec = 4321;
		}
		srandom(tv.tv_usec);
	}
#else
	srand(time(0));
#endif

457
458
459
460
461
462
	filesize = get_filesize_(filename);
	if (filesize < 0) {
		fprintf(stderr, "ERROR: can't determine filesize for %s\n", filename);
		return 1;
	}

Josh Coalson's avatar
Josh Coalson committed
463
464
465
466
	(void) signal(SIGINT, our_sigint_handler_);

	{
		FLAC__bool ok;
467
468
		if (strlen(filename) > 4 && 0 == strcmp(filename+strlen(filename)-4, ".ogg")) {
#ifdef FLAC__HAS_OGG
469
			ok = seek_barrage_ogg_flac(filename, filesize, count);
470
471
472
473
474
475
#else
			fprintf(stderr, "ERROR: Ogg FLAC not supported\n");
			ok = false;
#endif
		}
		else {
476
			ok = seek_barrage_native_flac(filename, filesize, count);
477
		}
Josh Coalson's avatar
Josh Coalson committed
478
479
480
		return ok? 0 : 2;
	}
}