Chris@40: /*
Chris@40: **	Copyright (C) 2005-2011 Erik de Castro Lopo
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., 675 Mass Ave, Cambridge, MA 02139, USA.
Chris@40: */
Chris@40: 
Chris@40: #include "config.h"
Chris@40: 
Chris@40: #include <stdio.h>
Chris@40: #include <stdlib.h>
Chris@40: #include <unistd.h>
Chris@40: #include <string.h>
Chris@40: #include <fcntl.h>
Chris@40: #include <sys/stat.h>
Chris@40: 
Chris@40: #include <sndfile.h>
Chris@40: 
Chris@40: #include "regtest.h"
Chris@40: 
Chris@40: #if HAVE_SQLITE3
Chris@40: 
Chris@40: #include <sqlite3.h>
Chris@40: 
Chris@40: typedef struct
Chris@40: {	sqlite3 *sql ;
Chris@40: 
Chris@40: 	int count ;
Chris@40: 	int ekey_max ;
Chris@40: 
Chris@40: 	/* Filename and pathname for file. */
Chris@40: 	char filename [256] ;
Chris@40: 	char pathname [512] ;
Chris@40: 
Chris@40: 	/* Storage for createding SQL commands. Must be larger than logbuf below. */
Chris@40: 	char cmdbuf [1 << 15] ;
Chris@40: 
Chris@40: 	/* Storage for log buffer retrieved from SNDFILE* .*/
Chris@40: 	char logbuf [1 << 14] ;
Chris@40: 
Chris@40: } REGTEST_DB ;
Chris@40: 
Chris@40: /* In checksum.c */
Chris@40: int calc_checksum (SNDFILE * file, const SF_INFO * info) ;
Chris@40: 
Chris@40: static void get_filename_pathname (REGTEST_DB * db, const char *filepath) ;
Chris@40: static void single_quote_replace (char * buf) ;
Chris@40: 
Chris@40: static int get_ekey_from_filename (REGTEST_DB * db, const char *filepath) ;
Chris@40: static int get_filename_pathname_by_ekey (REGTEST_DB * db, int ekey) ;
Chris@40: static int check_file_by_ekey (REGTEST_DB * db, int ekey) ;
Chris@40: 
Chris@40: static int count_callback (REGTEST_DB * db, int argc, char **argv, char **colname) ;
Chris@40: static int ekey_max_callback (REGTEST_DB * db, int argc, char **argv, char **colname) ;
Chris@40: static int callback (void *unused, int argc, char **argv, char **colname) ;
Chris@40: 
Chris@40: REG_DB *
Chris@40: db_open (const char * db_name)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 	int err ;
Chris@40: 
Chris@40: 	if ((db = malloc (sizeof (REGTEST_DB))) == NULL)
Chris@40: 	{	perror ("malloc") ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if ((err = sqlite3_open (db_name, &(db->sql))) != 0)
Chris@40: 	{	printf ("Can't open database: %s\n", sqlite3_errmsg (db->sql)) ;
Chris@40:         sqlite3_close (db->sql) ;
Chris@40: 		free (db) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	return (REG_DB *) db ;
Chris@40: } /* db_open */
Chris@40: 
Chris@40: int
Chris@40: db_create (const char * db_name)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 	const char *cmd ;
Chris@40: 	char * errmsg = NULL ;
Chris@40: 	int err ;
Chris@40: 
Chris@40: 	db = (REGTEST_DB *) db_open (db_name) ;
Chris@40: 
Chris@40: 	cmd = "create table sndfile (ekey INTEGER PRIMARY KEY,"
Chris@40: 			"fname VARCHAR(1),"
Chris@40: 			"fpath VARCHAR(1),"
Chris@40: 			"srate INTEGER,"
Chris@40: 			"frames VARCHAR(1),"
Chris@40: 			"channels INTEGER,"
Chris@40: 			"format VARCHAR(1),"
Chris@40: 			"checksum VARCHAR(1),"
Chris@40: 			"logbuf VARCHAR(1)"
Chris@40: 			");" ;
Chris@40: 
Chris@40: 	err = sqlite3_exec (db->sql, cmd, callback, 0, &errmsg) ;
Chris@40: 	if (err != SQLITE_OK)
Chris@40: 		printf ("Line %d : SQL error: %s\n", __LINE__, errmsg) ;
Chris@40: 
Chris@40: 	sqlite3_close (db->sql) ;
Chris@40: 	free (db) ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* db_create */
Chris@40: 
Chris@40: int
Chris@40: db_close (REG_DB * db_handle)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 
Chris@40: 	db = (REGTEST_DB *) db_handle ;
Chris@40: 
Chris@40: 	sqlite3_close (db->sql) ;
Chris@40: 	free (db) ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* db_close */
Chris@40: 
Chris@40: /*==============================================================================
Chris@40: */
Chris@40: 
Chris@40: int
Chris@40: db_file_exists (REG_DB * db_handle, const char * filename)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 	const char * cptr ;
Chris@40: 	char * errmsg ;
Chris@40: 	int err ;
Chris@40: 
Chris@40: 	db = (REGTEST_DB *) db_handle ;
Chris@40: 
Chris@40: 	if ((cptr = strrchr (filename, '/')) != NULL)
Chris@40: 		filename = cptr + 1 ;
Chris@40: 
Chris@40: 	snprintf (db->cmdbuf, sizeof (db->cmdbuf), "select fname from sndfile where fname='%s'", filename) ;
Chris@40: 
Chris@40: 	db->count = 0 ;
Chris@40: 	err = sqlite3_exec (db->sql, db->cmdbuf, (sqlite3_callback) count_callback, db, &errmsg) ;
Chris@40: 	if (err == 0 && db->count == 1)
Chris@40: 		return 1 ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* db_file_exists */
Chris@40: 
Chris@40: int
Chris@40: db_add_file (REG_DB * db_handle, const char * filepath)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 	SNDFILE * sndfile ;
Chris@40: 	SF_INFO info ;
Chris@40: 	char * errmsg ;
Chris@40: 	int err, checksum ;
Chris@40: 
Chris@40: 	db = (REGTEST_DB *) db_handle ;
Chris@40: 
Chris@40: 	get_filename_pathname (db, filepath) ;
Chris@40: 
Chris@40: 	if (db_file_exists (db_handle, filepath))
Chris@40: 	{	printf ("    %s : already in database\n", db->filename) ;
Chris@40: 		return 0 ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	memset (&info, 0, sizeof (info)) ;
Chris@40: 	sndfile = sf_open (db->pathname, SFM_READ, &info) ;
Chris@40: 	sf_command (sndfile, SFC_GET_LOG_INFO, db->logbuf, sizeof (db->logbuf)) ;
Chris@40: 	checksum = (sndfile == NULL) ? 0 : calc_checksum (sndfile, &info) ;
Chris@40: 	sf_close (sndfile) ;
Chris@40: 
Chris@40: 	if (sndfile == NULL)
Chris@40: 	{	printf ("    %s : could not open : %s\n", db->filename, sf_strerror (NULL)) ;
Chris@40: 		puts (db->logbuf) ;
Chris@40: 		return 1 ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	single_quote_replace (db->logbuf) ;
Chris@40: 
Chris@40: 	snprintf (db->cmdbuf, sizeof (db->cmdbuf), "insert into sndfile "
Chris@40: 		"(fname, fpath, srate, frames, channels, format, checksum, logbuf) values"
Chris@40: 		"('%s','%s',%d,'%ld', %d, '0x%08x', '0x%08x', '%s');",
Chris@40: 		db->filename, db->pathname, info.samplerate, (long) info.frames, info.channels, info.format, checksum, db->logbuf) ;
Chris@40: 
Chris@40: 	if (strlen (db->cmdbuf) >= sizeof (db->cmdbuf) - 1)
Chris@40: 	{	printf ("strlen (db->cmdbuf) too long.\n") ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	err = sqlite3_exec (db->sql, db->cmdbuf, callback, 0, &errmsg) ;
Chris@40: 	if (err != SQLITE_OK)
Chris@40: 	{	printf ("Line %d : SQL error: %s\n", __LINE__, errmsg) ;
Chris@40: 		puts (db->cmdbuf) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* db_add_file */
Chris@40: 
Chris@40: int
Chris@40: db_check_file (REG_DB * db_handle, const char * filepath)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 	int ekey ;
Chris@40: 
Chris@40: 	if (db_file_exists (db_handle, filepath) == 0)
Chris@40: 	{	printf ("\nFile not in database.\n\n") ;
Chris@40: 		exit (0) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	db = (REGTEST_DB *) db_handle ;
Chris@40: 
Chris@40: 	ekey = get_ekey_from_filename (db, filepath) ;
Chris@40: 
Chris@40: 	return check_file_by_ekey (db, ekey) ;
Chris@40: } /* db_check_file */
Chris@40: 
Chris@40: /*==============================================================================
Chris@40: */
Chris@40: 
Chris@40: int
Chris@40: db_check_all (REG_DB * db_handle)
Chris@40: {	REGTEST_DB * db ;
Chris@40: 	char * errmsg ;
Chris@40: 	int err, ekey ;
Chris@40: 
Chris@40: 	db = (REGTEST_DB *) db_handle ;
Chris@40: 
Chris@40: 	db->ekey_max = 0 ;
Chris@40: 
Chris@40: 	snprintf (db->cmdbuf, sizeof (db->cmdbuf), "select ekey from sndfile") ;
Chris@40: 
Chris@40: 	err = sqlite3_exec (db->sql, db->cmdbuf, (sqlite3_callback) ekey_max_callback, db, &errmsg) ;
Chris@40: 	if (err != SQLITE_OK)
Chris@40: 	{	printf ("Line %d : SQL error: %s\n", __LINE__, errmsg) ;
Chris@40: 		puts (db->cmdbuf) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	for (ekey = 1 ; ekey <= db->ekey_max ; ekey++)
Chris@40: 		if (get_filename_pathname_by_ekey (db, ekey) != 0)
Chris@40: 			check_file_by_ekey (db, ekey) ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* db_check_all */
Chris@40: 
Chris@40: 
Chris@40: int
Chris@40: db_list_all (REG_DB * db_handle)
Chris@40: {
Chris@40: 	printf ("%s : %p\n", __func__, db_handle) ;
Chris@40: 	return 0 ;
Chris@40: } /* db_list_all */
Chris@40: 
Chris@40: int
Chris@40: db_del_entry (REG_DB * db_handle, const char * entry)
Chris@40: {
Chris@40: 	printf ("%s : %p %s\n", __func__, db_handle, entry) ;
Chris@40: 	return 0 ;
Chris@40: } /* db_del_entry */
Chris@40: 
Chris@40: /*==============================================================================
Chris@40: */
Chris@40: 
Chris@40: static int
Chris@40: get_ekey_from_filename (REGTEST_DB * db, const char *filepath)
Chris@40: {	char * errmsg, **result ;
Chris@40: 	int err, ekey = 0, rows, cols ;
Chris@40: 
Chris@40: 	get_filename_pathname (db, filepath) ;
Chris@40: 
Chris@40: 	snprintf (db->cmdbuf, sizeof (db->cmdbuf), "select ekey from sndfile where fname='%s'", db->filename) ;
Chris@40: 
Chris@40: 	err = sqlite3_get_table (db->sql, db->cmdbuf, &result, &rows, &cols, &errmsg) ;
Chris@40: 	if (err != SQLITE_OK)
Chris@40: 	{	printf ("Line %d : SQL error: %s\n", __LINE__, errmsg) ;
Chris@40: 		puts (db->cmdbuf) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (cols != 1 || rows != 1)
Chris@40: 	{	printf ("Bad juju!! rows = %d cols = %d\n", rows, cols) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	ekey = strtol (result [1], NULL, 10) ;
Chris@40: 
Chris@40: 	sqlite3_free_table (result) ;
Chris@40: 
Chris@40: 	return ekey ;
Chris@40: } /* get_ekey_from_filename */
Chris@40: 
Chris@40: static int
Chris@40: get_filename_pathname_by_ekey (REGTEST_DB * db, int ekey)
Chris@40: {	char *errmsg, **result ;
Chris@40: 	int err, rows, cols ;
Chris@40: 
Chris@40: 	snprintf (db->cmdbuf, sizeof (db->cmdbuf), "select fname,fpath from sndfile where ekey='%d'", ekey) ;
Chris@40: 
Chris@40: 	err = sqlite3_get_table (db->sql, db->cmdbuf, &result, &rows, &cols, &errmsg) ;
Chris@40: 	if (err != SQLITE_OK)
Chris@40: 	{	printf ("Line %d : SQL error: %s\n", __LINE__, errmsg) ;
Chris@40: 		puts (db->cmdbuf) ;
Chris@40: 		return 0 ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	if (cols != 2 || rows != 1)
Chris@40: 	{	printf ("\nError (%s %d) : rows = %d cols = %d\n", __func__, __LINE__, rows, cols) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	snprintf (db->filename, sizeof (db->filename), "%s", result [2]) ;
Chris@40: 	snprintf (db->pathname, sizeof (db->pathname), "%s", result [3]) ;
Chris@40: 
Chris@40: 	sqlite3_free_table (result) ;
Chris@40: 
Chris@40: 	return 1 ;
Chris@40: } /* get_filename_pathname_by_ekey */
Chris@40: 
Chris@40: static int
Chris@40: check_file_by_ekey (REGTEST_DB * db, int ekey)
Chris@40: {	SNDFILE * sndfile ;
Chris@40: 	SF_INFO info ;
Chris@40: 	char * errmsg, **result ;
Chris@40: 	int err, k, rows, cols, checksum ;
Chris@40: 
Chris@40: 	printf ("    %s : ", db->filename) ;
Chris@40: 	fflush (stdout) ;
Chris@40: 
Chris@40: 	memset (&info, 0, sizeof (info)) ;
Chris@40: 	sndfile = sf_open (db->pathname, SFM_READ, &info) ;
Chris@40: 	sf_command (sndfile, SFC_GET_LOG_INFO, db->logbuf, sizeof (db->logbuf)) ;
Chris@40: 	checksum = (sndfile == NULL) ? 0 : calc_checksum (sndfile, &info) ;
Chris@40: 	sf_close (sndfile) ;
Chris@40: 
Chris@40: 	if (sndfile == NULL)
Chris@40: 	{	printf ("\n\nError : Could not open '%s' : %s\n", db->pathname, sf_strerror (NULL)) ;
Chris@40: 		puts (db->logbuf) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	single_quote_replace (db->logbuf) ;
Chris@40: 
Chris@40: 	snprintf (db->cmdbuf, sizeof (db->cmdbuf), "select fname,srate,frames,channels,format,"
Chris@40: 			"checksum,logbuf from sndfile where ekey='%d'", ekey) ;
Chris@40: 
Chris@40: 	err = sqlite3_get_table (db->sql, db->cmdbuf, &result, &rows, &cols, &errmsg) ;
Chris@40: 	if (err != SQLITE_OK)
Chris@40: 	{	printf ("Line %d : SQL error: %s\n", __LINE__, errmsg) ;
Chris@40: 		puts (db->cmdbuf) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	for (k = 0 ; k < cols ; k++)
Chris@40: 	{	if (strcmp (result [k], "fname") == 0)
Chris@40: 		{	if (strcmp (result [k + cols], db->filename) == 0)
Chris@40: 				continue ;
Chris@40: 			printf ("\n\nError : fname doesn't match : %s != %s\n", result [k + cols], db->filename) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		if (strcmp (result [k], "srate") == 0)
Chris@40: 		{	if (strtol (result [k + cols], NULL, 10) == info.samplerate)
Chris@40: 				continue ;
Chris@40: 			printf ("\n\nError : srate doesn't match : %s == %d\n", result [k + cols], info.samplerate) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		if (strcmp (result [k], "frames") == 0)
Chris@40: 		{	if (strtoll (result [k + cols], NULL, 10) == info.frames)
Chris@40: 				continue ;
Chris@40: 			printf ("\n\nError : frames doesn't match : %s == %ld\n", result [k + cols], (long) info.frames) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		if (strcmp (result [k], "channels") == 0)
Chris@40: 		{	if (strtol (result [k + cols], NULL, 10) == info.channels)
Chris@40: 				continue ;
Chris@40: 			printf ("\n\nError : channels doesn't match : %s == %d\n", result [k + cols], info.channels) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		if (strcmp (result [k], "format") == 0)
Chris@40: 		{	if (strtol (result [k + cols], NULL, 16) == info.format)
Chris@40: 				continue ;
Chris@40: 			printf ("\n\nError : format doesn't match : %s == 0x%08x\n", result [k + cols], info.format) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		if (strcmp (result [k], "checksum") == 0)
Chris@40: 		{	int db_val = (int) strtoll (result [k + cols], NULL, 16) ;
Chris@40: 
Chris@40: 			if (db_val == checksum)
Chris@40: 				continue ;
Chris@40: 			printf ("\n\nError : checksum doesn't match : 0x%08x == 0x%08x\n", db_val, checksum) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		if (strcmp (result [k], "logbuf") == 0)
Chris@40: 			continue ;
Chris@40: 
Chris@40: 		printf ("\nHere is the old logubuffer :\n\n%s\n\nand the new :\n\n%s\n\n", result [2 * cols - 1], db->logbuf) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	sqlite3_free_table (result) ;
Chris@40: 
Chris@40: 	puts ("ok") ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* check_file_by_ekey */
Chris@40: 
Chris@40: /*==============================================================================
Chris@40: */
Chris@40: 
Chris@40: static void
Chris@40: get_filename_pathname (REGTEST_DB * db, const char *filepath)
Chris@40: {	const char * cptr ;
Chris@40: 	int slen ;
Chris@40: 
Chris@40: 	if (filepath [0] != '/')
Chris@40: 	{	memset (db->pathname, 0, sizeof (db->pathname)) ;
Chris@40: 		if (getcwd (db->pathname, sizeof (db->pathname)) == NULL)
Chris@40: 		{	perror ("\ngetcwd failed") ;
Chris@40: 			exit (1) ;
Chris@40: 			} ;
Chris@40: 
Chris@40: 		slen = strlen (db->pathname) ;
Chris@40: 		db->pathname [slen ++] = '/' ;
Chris@40: 		snprintf (db->pathname + slen, sizeof (db->pathname) - slen, "%s", filepath) ;
Chris@40: 		}
Chris@40: 	else
Chris@40: 		snprintf (db->pathname, sizeof (db->pathname), "%s", filepath) ;
Chris@40: 
Chris@40: 	if ((cptr = strrchr (db->pathname, '/')) == NULL)
Chris@40: 	{	printf ("\nError : bad pathname %s\n", filepath) ;
Chris@40: 		exit (1) ;
Chris@40: 		} ;
Chris@40: 
Chris@40: 	snprintf (db->filename, sizeof (db->filename), "%s", cptr + 1) ;
Chris@40: } /* get filename_pathname */
Chris@40: 
Chris@40: static void
Chris@40: single_quote_replace (char * buf)
Chris@40: {	while ((buf = strchr (buf, '\'')) != 0)
Chris@40: 		buf [0] = '"' ;
Chris@40: } /* single_quote_replace */
Chris@40: 
Chris@40: static int
Chris@40: count_callback (REGTEST_DB * db, int argc, char **argv, char **colname)
Chris@40: {	db->count ++ ;
Chris@40: 
Chris@40: 	(void) argc ;
Chris@40: 	(void) argv ;
Chris@40: 	(void) colname ;
Chris@40: 	return 0 ;
Chris@40: } /* count_callback */
Chris@40: 
Chris@40: static int
Chris@40: ekey_max_callback (REGTEST_DB * db, int argc, char **argv, char **unused)
Chris@40: {	int ekey ;
Chris@40: 
Chris@40: 	(void) argc ;
Chris@40: 	(void) unused ;
Chris@40: 
Chris@40: 	ekey = strtol (argv [0], NULL, 10) ;
Chris@40: 	if (ekey > db->ekey_max)
Chris@40: 		db->ekey_max = ekey ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* ekey_max_callback */
Chris@40: 
Chris@40: static int
Chris@40: callback (void *unused, int argc, char **argv, char **colname)
Chris@40: {	int k ;
Chris@40: 
Chris@40: 	(void) unused ;
Chris@40: 
Chris@40: 	for (k = 0 ; k < argc ; k++)
Chris@40: 		printf ("%s = %s\n", colname [k], argv [k] ? argv [k] : "NULL") ;
Chris@40: 
Chris@40: 	printf ("\n") ;
Chris@40: 
Chris@40: 	return 0 ;
Chris@40: } /* callback */
Chris@40: 
Chris@40: #else
Chris@40: 
Chris@40: int dummy (void) ;
Chris@40: 
Chris@40: int
Chris@40: dummy (void)
Chris@40: {	/*
Chris@40: 	**	Empty dummy fnction so tha compiler doesn't winge about an
Chris@40: 	**	empty file.
Chris@40: 	*/
Chris@40: 	return 0 ;
Chris@40: } /* dummy */
Chris@40: 
Chris@40: #endif