Chris@40: /*
Chris@40: ** Copyright (C) 2003-2016 Erik de Castro Lopo <erikd@mega-nerd.com>
Chris@40: **
Chris@40: ** This program is free software; you can redistribute it and/or modify
Chris@40: ** it under the terms of the GNU General Public License as published by
Chris@40: ** the Free Software Foundation; either version 2 of the License, or
Chris@40: ** (at your option) any later version.
Chris@40: **
Chris@40: ** This program is distributed in the hope that it will be useful,
Chris@40: ** but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@40: ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@40: ** GNU General Public License for more details.
Chris@40: **
Chris@40: ** You should have received a copy of the GNU General Public License
Chris@40: ** along with this program; if not, write to the Free Software
Chris@40: ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
Chris@40: */
Chris@40: 
Chris@40: #include "sfconfig.h"
Chris@40: 
Chris@40: #include <stdio.h>
Chris@40: #include <stdlib.h>
Chris@40: #include <string.h>
Chris@40: #include <math.h>
Chris@40: 
Chris@40: #if HAVE_UNISTD_H
Chris@40: #include <unistd.h>
Chris@40: #endif
Chris@40: 
Chris@40: #include	<sndfile.h>
Chris@40: 
Chris@40: #include	"utils.h"
Chris@40: 
Chris@40: #define	BUFFER_LEN			(1 << 10)
Chris@40: #define LOG_BUFFER_SIZE		1024
Chris@40: 
Chris@40: static void	chunk_test (const char *filename, int format) ;
Chris@40: 
Chris@40: static void
Chris@40: chunk_test_helper (const char *filename, int format, const char * testdata) ;
Chris@40: 
Chris@40: int
Chris@40: main (int argc, char *argv [])
Chris@40: {	int		do_all = 0 ;
Chris@40: 	int		test_count = 0 ;
Chris@40: 
Chris@40: 	if (argc != 2)
Chris@40: 	{	printf ("Usage : %s <test>\n", argv [0]) ;
Chris@40: 		printf ("    Where <test> is one of the following:\n") ;
Chris@40: 		printf ("           wav  - test adding chunks to WAV files\n") ;
Chris@40: 		printf ("           aiff - test adding chunks to AIFF files\n") ;
Chris@40: 		printf ("           caf  - test adding chunks to CAF files\n") ;
Chris@40: 		printf ("           rf64 - test adding chunks to RF64 files\n") ;
Chris@40: 		printf ("           all  - perform all tests\n") ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	do_all = ! strcmp (argv [1], "all") ;
Chris@40: 
Chris@40: 	if (do_all || ! strcmp (argv [1], "wav"))
Chris@40: 	{	chunk_test ("chunks_pcm16.wav", SF_FORMAT_WAV | SF_FORMAT_PCM_16) ;
Chris@40: 		chunk_test ("chunks_pcm16.rifx", SF_ENDIAN_BIG | SF_FORMAT_WAV | SF_FORMAT_PCM_16) ;
Chris@40: 		chunk_test ("chunks_pcm16.wavex", SF_FORMAT_WAVEX | SF_FORMAT_PCM_16) ;
Chris@40: 		test_count++ ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (do_all || ! strcmp (argv [1], "aiff"))
Chris@40: 	{	chunk_test ("chunks_pcm16.aiff", SF_FORMAT_AIFF | SF_FORMAT_PCM_16) ;
Chris@40: 		test_count++ ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (do_all || ! strcmp (argv [1], "caf"))
Chris@40: 	{	chunk_test ("chunks_pcm16.caf", SF_FORMAT_CAF | SF_FORMAT_PCM_16) ;
Chris@40: 		chunk_test ("chunks_alac.caf", SF_FORMAT_CAF | SF_FORMAT_ALAC_16) ;
Chris@40: 		test_count++ ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (do_all || ! strcmp (argv [1], "rf64"))
Chris@40: 	{	chunk_test ("chunks_pcm16.rf64", SF_FORMAT_RF64 | SF_FORMAT_PCM_16) ;
Chris@40: 		test_count++ ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (test_count == 0)
Chris@40: 	{	printf ("Mono : ************************************\n") ;
Chris@40: 		printf ("Mono : *  No '%s' test defined.\n", argv [1]) ;
Chris@40: 		printf ("Mono : ************************************\n") ;
Chris@40: 		return 1 ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* main */
Chris@40: 
Chris@40: 
Chris@40: /*============================================================================================
Chris@40: **	Here are the test functions.
Chris@40: */
Chris@40: 
Chris@40: static void
Chris@40: chunk_test_helper (const char *filename, int format, const char * testdata)
Chris@40: {	SNDFILE			*file ;
Chris@40: 	SF_INFO			sfinfo ;
Chris@40: 	SF_CHUNK_INFO	chunk_info ;
Chris@40: 	SF_CHUNK_ITERATOR * iterator ;
Chris@40: 	uint32_t		length_before ;
Chris@40: 	int				err, allow_fd ;
Chris@40: 
Chris@40: 	switch (format & SF_FORMAT_SUBMASK)
Chris@40: 	{	case SF_FORMAT_ALAC_16 :
Chris@40: 			allow_fd = SF_FALSE ;
Chris@40: 			break ;
Chris@40: 		default :
Chris@40: 			allow_fd = SF_TRUE ;
Chris@40: 			break ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	sfinfo.samplerate	= 44100 ;
Chris@40: 	sfinfo.channels		= 1 ;
Chris@40: 	sfinfo.frames		= 0 ;
Chris@40: 	sfinfo.format		= format ;
Chris@40: 
Chris@40: 	file = test_open_file_or_die (filename, SFM_WRITE, &sfinfo, allow_fd, __LINE__) ;
Chris@40: 
Chris@40: 	/* Set up the chunk to write. */
Chris@40: 	memset (&chunk_info, 0, sizeof (chunk_info)) ;
Chris@40: 	snprintf (chunk_info.id, sizeof (chunk_info.id), "Test") ;
Chris@40: 	chunk_info.id_size = 4 ;
Chris@40: 	chunk_info.data = strdup (testdata) ;
Chris@40: 	chunk_info.datalen = strlen (chunk_info.data) ;
Chris@40: 
Chris@40: 	length_before = chunk_info.datalen ;
Chris@40: 
Chris@40: 	err = sf_set_chunk (file, &chunk_info) ;
Chris@40: 	exit_if_true (
Chris@40: 		err != SF_ERR_NO_ERROR,
Chris@40: 		"\n\nLine %d : sf_set_chunk returned for testdata '%s' : %s\n\n", __LINE__, testdata, sf_error_number (err)
Chris@40: 		) ;
Chris@40: 
Chris@40: 	memset (chunk_info.data, 0, chunk_info.datalen) ;
Chris@40: 	free (chunk_info.data) ;
Chris@40: 
Chris@40: 	sf_close (file) ;
Chris@40: 
Chris@40: 	file = test_open_file_or_die (filename, SFM_READ, &sfinfo, allow_fd, __LINE__) ;
Chris@40: 
Chris@40: 	memset (&chunk_info, 0, sizeof (chunk_info)) ;
Chris@40: 	snprintf (chunk_info.id, sizeof (chunk_info.id), "Test") ;
Chris@40: 	chunk_info.id_size = 4 ;
Chris@40: 
Chris@40: 	iterator = sf_get_chunk_iterator (file, &chunk_info) ;
Chris@40: 	err = sf_get_chunk_size (iterator, &chunk_info) ;
Chris@40: 	exit_if_true (
Chris@40: 		err != SF_ERR_NO_ERROR,
Chris@40: 		"\n\nLine %d : sf_get_chunk_size returned for testdata '%s' : %s\n\n", __LINE__, testdata, sf_error_number (err)
Chris@40: 		) ;
Chris@40: 
Chris@40: 	exit_if_true (
Chris@40: 		length_before > chunk_info.datalen || chunk_info.datalen - length_before > 4,
Chris@40: 		"\n\nLine %d : testdata '%s' : Bad chunk length %u (previous length %u)\n\n", __LINE__, testdata, chunk_info.datalen, length_before
Chris@40: 		) ;
Chris@40: 
Chris@40: 	chunk_info.data = malloc (chunk_info.datalen) ;
Chris@40: 	err = sf_get_chunk_data (iterator, &chunk_info) ;
Chris@40: 	exit_if_true (
Chris@40: 		err != SF_ERR_NO_ERROR,
Chris@40: 		"\n\nLine %d : sf_get_chunk_size returned for testdata '%s' : %s\n\n", __LINE__, testdata, sf_error_number (err)
Chris@40: 		) ;
Chris@40: 
Chris@40: 	exit_if_true (
Chris@40: 		memcmp (testdata, chunk_info.data, length_before),
Chris@40: 		"\n\nLine %d : Data compare failed.\n    %s\n    %s\n\n", __LINE__, testdata, (char*) chunk_info.data
Chris@40: 		) ;
Chris@40: 
Chris@40: 	free (chunk_info.data) ;
Chris@40: 
Chris@40: 	sf_close (file) ;
Chris@40: 	unlink (filename) ;
Chris@40: } /* chunk_test_helper */
Chris@40: 
Chris@40: static void
Chris@40: multichunk_test_helper (const char *filename, int format, const char * testdata [], size_t testdata_len)
Chris@40: {	SNDFILE			*file ;
Chris@40: 	SF_INFO			sfinfo ;
Chris@40: 	SF_CHUNK_INFO	chunk_info ;
Chris@40: 	SF_CHUNK_ITERATOR * iterator ;
Chris@40: 	uint32_t		length_before [testdata_len] ;
Chris@40: 	int				err, allow_fd ;
Chris@40: 	size_t			i ;
Chris@40: 
Chris@40: 	sfinfo.samplerate	= 44100 ;
Chris@40: 	sfinfo.channels		= 1 ;
Chris@40: 	sfinfo.frames		= 0 ;
Chris@40: 	sfinfo.format		= format ;
Chris@40: 
Chris@40: 	switch (format & SF_FORMAT_SUBMASK)
Chris@40: 	{	case SF_FORMAT_ALAC_16 :
Chris@40: 			allow_fd = SF_FALSE ;
Chris@40: 			break ;
Chris@40: 		default :
Chris@40: 			allow_fd = SF_TRUE ;
Chris@40: 			break ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	file = test_open_file_or_die (filename, SFM_WRITE, &sfinfo, allow_fd, __LINE__) ;
Chris@40: 
Chris@40: 	/* Set up the chunk to write. */
Chris@40: 	for (i = 0 ; i < testdata_len ; i++)
Chris@40: 	{	memset (&chunk_info, 0, sizeof (chunk_info)) ;
Chris@40: 		snprintf (chunk_info.id, sizeof (chunk_info.id), "Test") ;
Chris@40: 		chunk_info.id_size = 4 ;
Chris@40: 
Chris@40: 		chunk_info.data = strdup (testdata [i]) ;
Chris@40: 		chunk_info.datalen = strlen (chunk_info.data) ;
Chris@40: 
Chris@40: 		length_before [i] = chunk_info.datalen ;
Chris@40: 
Chris@40: 		err = sf_set_chunk (file, &chunk_info) ;
Chris@40: 		exit_if_true (
Chris@40: 			err != SF_ERR_NO_ERROR,
Chris@40: 			"\n\nLine %d : sf_set_chunk returned for testdata[%d] '%s' : %s\n\n", __LINE__, (int) i, testdata [i], sf_error_number (err)
Chris@40: 			) ;
Chris@40: 
Chris@40: 		memset (chunk_info.data, 0, chunk_info.datalen) ;
Chris@40: 		free (chunk_info.data) ;
Chris@40: 	}
Chris@40: 
Chris@40: 	sf_close (file) ;
Chris@40: 
Chris@40: 	file = test_open_file_or_die (filename, SFM_READ, &sfinfo, allow_fd, __LINE__) ;
Chris@40: 
Chris@40: 	memset (&chunk_info, 0, sizeof (chunk_info)) ;
Chris@40: 	snprintf (chunk_info.id, sizeof (chunk_info.id), "Test") ;
Chris@40: 	chunk_info.id_size = 4 ;
Chris@40: 
Chris@40: 	iterator = sf_get_chunk_iterator (file, &chunk_info) ;
Chris@40: 
Chris@40: 	i = 0 ;
Chris@40: 	while (iterator)
Chris@40: 	{	memset (&chunk_info, 0, sizeof (chunk_info)) ;
Chris@40: 		err = sf_get_chunk_size (iterator, &chunk_info) ;
Chris@40: 		exit_if_true (
Chris@40: 			i > testdata_len,
Chris@40: 			"\n\nLine %d : iterated to chunk #%d, but only %d chunks have been written\n\n", __LINE__, (int) i, (int) testdata_len
Chris@40: 			) ;
Chris@40: 
Chris@40: 		exit_if_true (
Chris@40: 			err != SF_ERR_NO_ERROR,
Chris@40: 			"\n\nLine %d : sf_get_chunk_size returned for testdata[%d] '%s' : %s\n\n", __LINE__, (int) i, testdata [i], sf_error_number (err)
Chris@40: 			) ;
Chris@40: 
Chris@40: 		exit_if_true (
Chris@40: 			length_before [i] > chunk_info.datalen || chunk_info.datalen - length_before [i] > 4,
Chris@40: 			"\n\nLine %d : testdata[%d] '%s' : Bad chunk length %u (previous length %u)\n\n", __LINE__, (int) i, testdata [i], chunk_info.datalen, length_before [i]
Chris@40: 			) ;
Chris@40: 
Chris@40: 		chunk_info.data = malloc (chunk_info.datalen) ;
Chris@40: 		err = sf_get_chunk_data (iterator, &chunk_info) ;
Chris@40: 		exit_if_true (
Chris@40: 			err != SF_ERR_NO_ERROR,
Chris@40: 			"\n\nLine %d : sf_get_chunk_size returned for testdata[%d] '%s' : %s\n\n", __LINE__, (int) i, testdata [i], sf_error_number (err)
Chris@40: 			) ;
Chris@40: 
Chris@40: 		exit_if_true (
Chris@40: 			4 != chunk_info.id_size,
Chris@40: 			"\n\nLine %d : testdata[%d] : Bad ID length %u (previous length %u)\n\n", __LINE__, (int) i, chunk_info.id_size, 4
Chris@40: 			) ;
Chris@40: 		exit_if_true (
Chris@40: 			memcmp ("Test", chunk_info.id, 4),
Chris@40: 			"\n\nLine %d : ID compare failed at %d.\n    %s\n    %s\n\n", __LINE__, (int) i, "Test", (char*) chunk_info.id
Chris@40: 			) ;
Chris@40: 
Chris@40: 		exit_if_true (
Chris@40: 			memcmp (testdata [i], chunk_info.data, length_before [i]),
Chris@40: 			"\n\nLine %d : Data compare failed at %d.\n    %s\n    %s\n\n", __LINE__, (int) i, testdata [i], (char*) chunk_info.data
Chris@40: 			) ;
Chris@40: 
Chris@40: 		free (chunk_info.data) ;
Chris@40: 		iterator = sf_next_chunk_iterator (iterator) ;
Chris@40: 		i++ ;
Chris@40: 	}
Chris@40: 
Chris@40: 	sf_close (file) ;
Chris@40: 	unlink (filename) ;
Chris@40: } /* multichunk_test_helper */
Chris@40: 
Chris@40: 
Chris@40: static void
Chris@40: chunk_test (const char *filename, int format)
Chris@40: {	const char*		testdata [] =
Chris@40: 	{	"There can be only one.", "", "A", "AB", "ABC", "ABCD", "ABCDE" } ;
Chris@40: 	uint32_t k ;
Chris@40: 
Chris@40: 	print_test_name (__func__, filename) ;
Chris@40: 
Chris@40: 	for (k = 0 ; k < ARRAY_LEN (testdata) ; k++)
Chris@40: 		chunk_test_helper (filename, format, testdata [k]) ;
Chris@40: 
Chris@40: 	multichunk_test_helper (filename, format, testdata, ARRAY_LEN (testdata)) ;
Chris@40: 
Chris@40: 	puts ("ok") ;
Chris@40: } /* chunk_test */