/****************************************************************************
*																			*
*								cryptlib Test Code							*
*						Copyright Peter Gutmann 1995-2009					*
*																			*
****************************************************************************/

#include "cryptlib.h"
#include "test/test.h"

#if defined( __MVS__ ) || defined( __VMCMS__ )
  /* Suspend conversion of literals to ASCII. */
  #pragma convlit( suspend )
#endif /* IBM big iron */
#if defined( __ILEC400__ )
  #pragma convert( 0 )
#endif /* IBM medium iron */

/****************************************************************************
*																			*
*								Utility Routines							*
*																			*
****************************************************************************/

/* The tests that use databases and certificate stores require that the user 
   set up a suitable ODBC data source (at least when running under Windows), 
   to automate this process we try and create the data source if it isn't 
   present.
   
   This is complicated by the fact that the universal default MSJET database 
   driver doesn't have a 64-bit version so it's not possible to use it under 
   Vista/Windows 7 x64.  To work around this we fall back to the SQL Server
   driver, which replaces MSJET on x64 systems */

#if defined( _MSC_VER ) && defined( _WIN32 ) && !defined( _WIN32_WCE )

#define DATABASE_AUTOCONFIG

#include <odbcinst.h>

#define DRIVER_NAME				TEXT( "Microsoft Access Driver (*.MDB)" )
#define DATABASE_ATTR_NAME		"DSN=" DATABASE_KEYSET_NAME_ASCII "#" \
								"DESCRIPTION=cryptlib test key database#" \
								"DBQ="
#define DATABASE_ATTR_CREATE	"DSN=" DATABASE_KEYSET_NAME_ASCII "#" \
								"DESCRIPTION=cryptlib test key database#" \
								"CREATE_DB="
#define DATABASE_ATTR_TAIL		DATABASE_KEYSET_NAME_ASCII ".mdb#"
#define CERTSTORE_ATTR_NAME		"DSN=" CERTSTORE_KEYSET_NAME_ASCII "#" \
								"DESCRIPTION=cryptlib test key database#" \
								"DBQ="
#define CERTSTORE_ATTR_CREATE	"DSN=" CERTSTORE_KEYSET_NAME_ASCII "#" \
								"DESCRIPTION=cryptlib test key database#" \
								"CREATE_DB="
#define CERTSTORE_ATTR_TAIL		CERTSTORE_KEYSET_NAME_ASCII ".mdb#"

#define DRIVER_NAME_ALT			TEXT( "SQL Server" )
#define DATABASE_ATTR_NAME_ALT	"DSN=" DATABASE_KEYSET_NAME_ASCII "#" \
								"DESCRIPTION=cryptlib test key database#" \
								"Server=localhost#" \
								"Database="
#define DATABASE_ATTR_CREATE_ALT ""
#define DATABASE_ATTR_TAIL_ALT	DATABASE_KEYSET_NAME_ASCII "#"
#define CERTSTORE_ATTR_NAME_ALT	"DSN=" CERTSTORE_KEYSET_NAME_ASCII "#" \
								"DESCRIPTION=cryptlib test key database#" \
								"Server=localhost#" \
								"Database="
#define CERTSTORE_ATTR_CREATE_ALT ""
#define CERTSTORE_ATTR_TAIL_ALT	CERTSTORE_KEYSET_NAME_ASCII "#"

static void buildDBString( char *buffer, const char *attrName,
						   const char *attrTail, const char *path )
	{
	const int attrNameSize = strlen( attrName );
	const int attrTailSize = strlen( attrTail ) + 1;
	const int pathSize = strlen( path );
	int dbStringLen, i;

	/* Build up the data-source control string */
	memcpy( buffer, attrName, attrNameSize + 1 );
	memcpy( buffer + attrNameSize, path, pathSize );
	if( attrTailSize > 0 )
		{
		memcpy( buffer + attrNameSize + pathSize, attrTail, 
				attrTailSize );
		}
	buffer[ attrNameSize + pathSize + attrTailSize ] = '\0';

	/* Finally, convert the strings to the weird embedded-null strings 
	   required by SQLConfigDataSource() */
	dbStringLen = strlen( buffer );
	for( i = 0; i < dbStringLen; i++ )
		{
		if( buffer[ i ] == '#' )
			buffer[ i ] = '\0';
		}
	}

static BOOLEAN createDatabase( const char *driverName,
							   const char *keysetName, 
							   const char *nameString, 
							   const char *createString, 
							   const char *trailerString,
							   const BOOLEAN isSqlServer )
	{
	char tempPathBuffer[ 512 ];
	char attrBuffer[ 1024 ];
#ifdef UNICODE_STRINGS
	wchar_t wcAttrBuffer[ 1024 ];
#endif /* UNICODE_STRINGS */
	int length, status;

	if( !( length = GetTempPath( 512, tempPathBuffer ) ) )
		{
		strcpy( tempPathBuffer, "C:\\Temp\\" );
		length = 8;
		}

	/* Try and create the DSN.  For the default Access driver his is a two-
	   step process, first we create the DSN and then the underlying file 
	   that contains the database.  For SQL Server it's simpler, the database
	   server already exists so all we have to do is create the database */
	if( isSqlServer )
		{
		printf( "Attempting to create keyset '%s' using alternative data "
				"source...\n", keysetName );
		}
	else
		{
		printf( "Database keyset '%s' not found, attempting to create data "
				"source...\n", keysetName );
		}
	buildDBString( attrBuffer, nameString, trailerString, tempPathBuffer );
#ifdef UNICODE_STRINGS
	mbstowcs( wcAttrBuffer, attrBuffer, strlen( attrBuffer ) + 1 );
	status = SQLConfigDataSource( NULL, ODBC_ADD_DSN, driverName, 
								  wcAttrBuffer );
#else
	status = SQLConfigDataSource( NULL, ODBC_ADD_DSN, driverName, 
								  attrBuffer );
#endif /* UNICODE_STRINGS */
	if( status != 1 )
		{
		DWORD dwErrorCode;
		WORD errorMessageLen;
		char errorMessage[ 256 ];
		
		if( SQLInstallerError( 1, &dwErrorCode, errorMessage, 256, 
							   &errorMessageLen ) != SQL_NO_DATA )
			{
			printf( "SQLConfigDataSource() returned error code %d,\n  "
					"message '%s'.\n", dwErrorCode, errorMessage );
#if defined( _M_X64 )
			if( !isSqlServer )
				{
				puts( "  (This is probably because there's no appropriate "
					  "64-bit driver present,\n  retrying the create with "
					  "an alternative driver...)." );
				}
#endif /* _M_X64 */
			}
		return( FALSE );
		}
	if( isSqlServer )
		{
		/* The server already exists and we're done */
		return( TRUE );
		}
	buildDBString( attrBuffer, createString, trailerString, tempPathBuffer );
#ifdef UNICODE_STRINGS
	mbstowcs( wcAttrBuffer, attrBuffer, strlen( attrBuffer ) + 1 );
	status = SQLConfigDataSource( NULL, ODBC_ADD_DSN, driverName, 
								  wcAttrBuffer );
#else
	status = SQLConfigDataSource( NULL, ODBC_ADD_DSN, driverName, 
								  attrBuffer );
#endif /* UNICODE_STRINGS */

	return( ( status == 1 ) ? TRUE : FALSE );
	}

static BOOLEAN createDatabaseKeyset( void )
	{
	int status;

	status = createDatabase( DRIVER_NAME, DATABASE_KEYSET_NAME_ASCII,
							 DATABASE_ATTR_NAME, DATABASE_ATTR_CREATE, 
							 DATABASE_ATTR_TAIL, FALSE );
	if( status == FALSE )
		{
		/* The create with the default (MS Access) driver failed, fall back
		   to the SQL server alternative */
		status = createDatabase( DRIVER_NAME_ALT, DATABASE_KEYSET_NAME_ASCII,
								 DATABASE_ATTR_NAME_ALT, 
								 DATABASE_ATTR_CREATE_ALT, 
								 DATABASE_ATTR_TAIL_ALT, TRUE );
		}
	puts( ( status == TRUE ) ? "Data source creation succeeded." : \
		  "Data source creation failed.\n\nYou need to create the "
		  "keyset data source as described in the cryptlib manual\n"
		  "for the database keyset tests to run." );

	return( status );
	}

static BOOLEAN createDatabaseCertstore( void )
	{
	int status;

	status = createDatabase( DRIVER_NAME, CERTSTORE_KEYSET_NAME_ASCII,
							 CERTSTORE_ATTR_NAME, CERTSTORE_ATTR_CREATE, 
							 CERTSTORE_ATTR_TAIL, FALSE );
	if( status == FALSE )
		{
		/* The create with the default (MS Access) driver failed, fall back
		   to the SQL server alternative */
		status = createDatabase( DRIVER_NAME_ALT, CERTSTORE_KEYSET_NAME_ASCII,
								 CERTSTORE_ATTR_NAME_ALT, 
								 CERTSTORE_ATTR_CREATE_ALT, 
								 CERTSTORE_ATTR_TAIL_ALT, TRUE );
		}
	puts( ( status == TRUE ) ? "Data source creation succeeded.\n" : \
		  "Data source creation failed.\n\nYou need to create the "
		  "certificate store data source as described in the\n"
		  "cryptlib manual for the certificate management tests to "
		  "run.\n" );

	return( status );
	}

static void checkCreateDatabaseKeysets( void )
	{
	CRYPT_KEYSET cryptKeyset;
	int status;

	/* Try and open the test keyset */
	status = cryptKeysetOpen( &cryptKeyset, CRYPT_UNUSED,
							  CRYPT_KEYSET_ODBC, DATABASE_KEYSET_NAME,
							  CRYPT_KEYOPT_READONLY );
	if( cryptStatusOK( status ) )
		cryptKeysetClose( cryptKeyset );
	else
		{
		if( status == CRYPT_ERROR_OPEN )
			createDatabaseKeyset();
		}

	/* Try and open the test certificate store.  This can return a
	   CRYPT_ARGERROR_PARAM3 as a normal condition since a freshly-created
	   database is empty and therefore can't be identified as a certificate 
	   store until data is written to it */
	status = cryptKeysetOpen( &cryptKeyset, CRYPT_UNUSED,
							  CRYPT_KEYSET_ODBC_STORE, CERTSTORE_KEYSET_NAME,
							  CRYPT_KEYOPT_READONLY );
	if( cryptStatusOK( status ) )
		cryptKeysetClose( cryptKeyset );
	else
		{
		if( status == CRYPT_ERROR_OPEN )
			createDatabaseCertstore();
		}
	}
#endif /* Win32 with VC++ */

/****************************************************************************
*																			*
*							Test Low-level Functions						*
*																			*
****************************************************************************/

#ifdef TEST_SELFTEST

/* Test the cryptlib self-test routines */

BOOLEAN testSelfTest( void )
	{
	int value, status;

	/* Perform the self-test.  First we write the value to true to force a
	   self-test, then we read it back to see whether it succeeded */
	status = cryptSetAttribute( CRYPT_UNUSED, CRYPT_OPTION_SELFTESTOK, 
								TRUE );
	if( cryptStatusError( status ) )
		{
		printf( "Attempt to perform cryptlib algorithm self-test failed "
				"with error code %d, line %d.\n", status, __LINE__ );
		return( FALSE );
		}
	status = cryptGetAttribute( CRYPT_UNUSED, CRYPT_OPTION_SELFTESTOK, 
								&value );
	if( cryptStatusError( status ) || value != TRUE )
		{
		/* Unfortunately all that we can report at this point is that the
		   self-test failed, we can't try each algorithm individually
		   because the self-test has disabled the failed one(s) */
		printf( "cryptlib algorithm self-test failed, line %d.\n", 
				__LINE__ );
		return( FALSE );
		}
	puts( "cryptlib algorithm self-test succeeded.\n" );

	return( TRUE );
	}
#else

BOOLEAN testSelfTest( void )
	{
	puts( "Skipping test of self-test routines...\n" );
	return( TRUE );
	}
#endif /* TEST_SELFTEST */

#ifdef TEST_LOWLEVEL

/* Test the low-level encryption routines */

BOOLEAN testLowLevel( void )
	{
	CRYPT_ALGO_TYPE cryptAlgo;
	BOOLEAN algosEnabled;

	/* Test the conventional encryption routines */
	algosEnabled = FALSE;
	for( cryptAlgo = CRYPT_ALGO_FIRST_CONVENTIONAL;
		 cryptAlgo <= CRYPT_ALGO_LAST_CONVENTIONAL; cryptAlgo++ )
		{
		if( cryptStatusOK( cryptQueryCapability( cryptAlgo, NULL ) ) )
			{
			if( !testLowlevel( CRYPT_UNUSED, cryptAlgo, FALSE ) )
				return( FALSE );
			algosEnabled = TRUE;
			}
		}
	if( !algosEnabled )
		puts( "(No conventional-encryption algorithms enabled)." );

	/* Test the public-key encryption routines */
	algosEnabled = FALSE;
	for( cryptAlgo = CRYPT_ALGO_FIRST_PKC;
		 cryptAlgo <= CRYPT_ALGO_LAST_PKC; cryptAlgo++ )
		{
		if( cryptStatusOK( cryptQueryCapability( cryptAlgo, NULL ) ) )
			{
			if( !testLowlevel( CRYPT_UNUSED, cryptAlgo, FALSE ) )
				return( FALSE );
			algosEnabled = TRUE;
			}
		}
	if( cryptStatusOK( cryptQueryCapability( CRYPT_ALGO_RSA, NULL ) ) && \
		!testRSAMinimalKey() )
		return( FALSE );
	if( !algosEnabled )
		puts( "(No public-key algorithms enabled)." );

	/* Test the hash routines */
	algosEnabled = FALSE;
	for( cryptAlgo = CRYPT_ALGO_FIRST_HASH;
		 cryptAlgo <= CRYPT_ALGO_LAST_HASH; cryptAlgo++ )
		{
		if( cryptStatusOK( cryptQueryCapability( cryptAlgo, NULL ) ) )
			{
			if( !testLowlevel( CRYPT_UNUSED, cryptAlgo, FALSE ) )
				return( FALSE );
			algosEnabled = TRUE;
			}
		}
	if( !algosEnabled )
		puts( "(No hash algorithms enabled)." );

	/* Test the MAC routines */
	algosEnabled = FALSE;
	for( cryptAlgo = CRYPT_ALGO_FIRST_MAC;
		 cryptAlgo <= CRYPT_ALGO_LAST_MAC; cryptAlgo++ )
		{
		if( cryptStatusOK( cryptQueryCapability( cryptAlgo, NULL ) ) )
			{
			if( !testLowlevel( CRYPT_UNUSED, cryptAlgo, FALSE ) )
				return( FALSE );
			algosEnabled = TRUE;
			}
		}
	if( !algosEnabled )
		puts( "(No MAC algorithms enabled)." );
	printf( "\n" );

	return( TRUE );
	}
#else

BOOLEAN testLowLevel( void )
	{
	puts( "Skipping test of low-level encryption routines...\n" );
	return( TRUE );
	}
#endif /* TEST_LOWLEVEL */

/****************************************************************************
*																			*
*					Test Randomness, Config, and Device Functions			*
*																			*
****************************************************************************/

#ifdef TEST_RANDOM

/* Test the randomness-gathering routines */

BOOLEAN testRandom( void )
	{
	if( !testRandomRoutines() )
		{
		puts( "The self-test will proceed without using a strong random "
			  "number source.\n" );

		/* Kludge the randomness routines so we can continue the self-tests */
		cryptAddRandom( "xyzzy", 5 );
		}

	return( TRUE );
	}
#else

BOOLEAN testRandom( void )
	{
	puts( "Skipping test of randomness routines...\n" );
	return( TRUE );
	}
#endif /* TEST_RANDOM */

#ifdef TEST_CONFIG

/* The names of the configuration options we check for */

static struct {
	const CRYPT_ATTRIBUTE_TYPE option;	/* Option */
	const char FAR_BSS *name;			/* Option name */
	const BOOLEAN isNumeric;			/* Whether it's a numeric option */
	} FAR_BSS configOption[] = {
	{ CRYPT_OPTION_INFO_DESCRIPTION, "CRYPT_OPTION_INFO_DESCRIPTION", FALSE },
	{ CRYPT_OPTION_INFO_COPYRIGHT, "CRYPT_OPTION_INFO_COPYRIGHT", FALSE },
	{ CRYPT_OPTION_INFO_MAJORVERSION, "CRYPT_OPTION_INFO_MAJORVERSION", TRUE },
	{ CRYPT_OPTION_INFO_MINORVERSION, "CRYPT_OPTION_INFO_MINORVERSION", TRUE },
	{ CRYPT_OPTION_INFO_STEPPING, "CRYPT_OPTION_INFO_STEPPING", TRUE },

	{ CRYPT_OPTION_ENCR_ALGO, "CRYPT_OPTION_ENCR_ALGO", TRUE },
	{ CRYPT_OPTION_ENCR_HASH, "CRYPT_OPTION_ENCR_HASH", TRUE },
	{ CRYPT_OPTION_ENCR_MAC, "CRYPT_OPTION_ENCR_MAC", TRUE },

	{ CRYPT_OPTION_PKC_ALGO, "CRYPT_OPTION_PKC_ALGO", TRUE },
	{ CRYPT_OPTION_PKC_KEYSIZE, "CRYPT_OPTION_PKC_KEYSIZE", TRUE },

	{ CRYPT_OPTION_SIG_ALGO, "CRYPT_OPTION_SIG_ALGO", TRUE },
	{ CRYPT_OPTION_SIG_KEYSIZE, "CRYPT_OPTION_SIG_KEYSIZE", TRUE },

	{ CRYPT_OPTION_KEYING_ALGO, "CRYPT_OPTION_KEYING_ALGO", TRUE },
	{ CRYPT_OPTION_KEYING_ITERATIONS, "CRYPT_OPTION_KEYING_ITERATIONS", TRUE },

	{ CRYPT_OPTION_CERT_SIGNUNRECOGNISEDATTRIBUTES, "CRYPT_OPTION_CERT_SIGNUNRECOGNISEDATTRIBUTES", TRUE },
	{ CRYPT_OPTION_CERT_VALIDITY, "CRYPT_OPTION_CERT_VALIDITY", TRUE },
	{ CRYPT_OPTION_CERT_UPDATEINTERVAL, "CRYPT_OPTION_CERT_UPDATEINTERVAL", TRUE },
	{ CRYPT_OPTION_CERT_COMPLIANCELEVEL, "CRYPT_OPTION_CERT_COMPLIANCELEVEL", TRUE },
	{ CRYPT_OPTION_CERT_REQUIREPOLICY, "CRYPT_OPTION_CERT_REQUIREPOLICY", TRUE },

	{ CRYPT_OPTION_CMS_DEFAULTATTRIBUTES, "CRYPT_OPTION_CMS_DEFAULTATTRIBUTES", TRUE },

	{ CRYPT_OPTION_KEYS_LDAP_OBJECTCLASS, "CRYPT_OPTION_KEYS_LDAP_OBJECTCLASS", FALSE },
	{ CRYPT_OPTION_KEYS_LDAP_OBJECTTYPE, "CRYPT_OPTION_KEYS_LDAP_OBJECTTYPE", TRUE },
	{ CRYPT_OPTION_KEYS_LDAP_FILTER, "CRYPT_OPTION_KEYS_LDAP_FILTER", FALSE },
	{ CRYPT_OPTION_KEYS_LDAP_CACERTNAME, "CRYPT_OPTION_KEYS_LDAP_CACERTNAME", FALSE },
	{ CRYPT_OPTION_KEYS_LDAP_CERTNAME, "CRYPT_OPTION_KEYS_LDAP_CERTNAME", FALSE },
	{ CRYPT_OPTION_KEYS_LDAP_CRLNAME, "CRYPT_OPTION_KEYS_LDAP_CRLNAME", FALSE },
	{ CRYPT_OPTION_KEYS_LDAP_EMAILNAME, "CRYPT_OPTION_KEYS_LDAP_EMAILNAME", FALSE },

	{ CRYPT_OPTION_DEVICE_PKCS11_DVR01, "CRYPT_OPTION_DEVICE_PKCS11_DVR01", FALSE },
	{ CRYPT_OPTION_DEVICE_PKCS11_DVR02, "CRYPT_OPTION_DEVICE_PKCS11_DVR02", FALSE },
	{ CRYPT_OPTION_DEVICE_PKCS11_DVR03, "CRYPT_OPTION_DEVICE_PKCS11_DVR03", FALSE },
	{ CRYPT_OPTION_DEVICE_PKCS11_DVR04, "CRYPT_OPTION_DEVICE_PKCS11_DVR04", FALSE },
	{ CRYPT_OPTION_DEVICE_PKCS11_DVR05, "CRYPT_OPTION_DEVICE_PKCS11_DVR05", FALSE },
	{ CRYPT_OPTION_DEVICE_PKCS11_HARDWAREONLY, "CRYPT_OPTION_DEVICE_PKCS11_HARDWAREONLY", TRUE },

	{ CRYPT_OPTION_NET_SOCKS_SERVER, "CRYPT_OPTION_NET_SOCKS_SERVER", FALSE },
	{ CRYPT_OPTION_NET_SOCKS_USERNAME, "CRYPT_OPTION_NET_SOCKS_USERNAME", FALSE },
	{ CRYPT_OPTION_NET_HTTP_PROXY, "CRYPT_OPTION_NET_HTTP_PROXY", FALSE },
	{ CRYPT_OPTION_NET_CONNECTTIMEOUT, "CRYPT_OPTION_NET_CONNECTTIMEOUT", TRUE },
	{ CRYPT_OPTION_NET_READTIMEOUT, "CRYPT_OPTION_NET_READTIMEOUT", TRUE },
	{ CRYPT_OPTION_NET_WRITETIMEOUT, "CRYPT_OPTION_NET_WRITETIMEOUT", TRUE },

	{ CRYPT_OPTION_MISC_ASYNCINIT, "CRYPT_OPTION_MISC_ASYNCINIT", TRUE },
	{ CRYPT_OPTION_MISC_SIDECHANNELPROTECTION, "CRYPT_OPTION_MISC_SIDECHANNELPROTECTION", TRUE },

	{ CRYPT_ATTRIBUTE_NONE, NULL, 0 }
	};

/* Test the configuration options routines */

BOOLEAN testConfig( void )
	{
	int i, value, status;

	for( i = 0; configOption[ i ].option != CRYPT_ATTRIBUTE_NONE; i++ )
		{
		C_CHR buffer[ 256 ];
		int length;

		if( configOption[ i ].isNumeric )
			{
			status = cryptGetAttribute( CRYPT_UNUSED, configOption[ i ].option, 
										&value );
			if( cryptStatusError( status ) )
				{
				printf( "%s appears to be disabled/unavailable in this build.\n", 
						configOption[ i ].name );
				continue;
				}
			printf( "%s = %d.\n", configOption[ i ].name, value );
			continue;
			}
		status = cryptGetAttributeString( CRYPT_UNUSED, 
										  configOption[ i ].option,
										  buffer, &length );
		if( cryptStatusError( status ) )
			{
			printf( "%s appears to be disabled/unavailable in this build.\n", 
					configOption[ i ].name );
			continue;
			}
		assert( length < 256 );
#ifdef UNICODE_STRINGS
		buffer[ length / sizeof( wchar_t ) ] = TEXT( '\0' );
		printf( "%s = %S.\n", configOption[ i ].name, buffer );
#else
		buffer[ length ] = '\0';
		printf( "%s = %s.\n", configOption[ i ].name, buffer );
#endif /* UNICODE_STRINGS */
		}
	printf( "\n" );

	return( TRUE );
	}
#else

BOOLEAN testConfig( void )
	{
	puts( "Skipping display of config options...\n" );
	return( TRUE );
	}
#endif /* TEST_CONFIG */

#ifdef TEST_DEVICE

/* Test the crypto device routines */

BOOLEAN testDevice( void )
	{
	int status;

	status = testDevices();
	if( status == CRYPT_ERROR_NOTAVAIL )
		{
		puts( "Handling for crypto devices doesn't appear to be enabled in "
			  "this build of\ncryptlib.\n" );
		return( TRUE );
		}
	if( !status )
		return( FALSE );

	return( TRUE );
	}
#else

BOOLEAN testDevice( void )
	{
	puts( "Skipping test of crypto device routines...\n" );
	return( TRUE );
	}
#endif /* TEST_DEVICE */

/****************************************************************************
*																			*
*							Test Mid/High-level Functions					*
*																			*
****************************************************************************/

#ifdef TEST_MIDLEVEL

/* Test the mid-level routines */

BOOLEAN testMidLevel( void )
	{
	if( !testLargeBufferEncrypt() )
		return( FALSE );
	if( !testDeriveKey() )
		return( FALSE );
	if( !testConventionalExportImport() )
		return( FALSE );
	if( cryptStatusOK( cryptQueryCapability( CRYPT_ALGO_HMAC_SHA1, NULL ) ) )
		{
		/* Only test the MAC functions of HMAC-SHA1 is enabled */
		if( !testMACExportImport() )
			return( FALSE );
		}
	if( cryptStatusOK( cryptQueryCapability( CRYPT_ALGO_RSA, NULL ) ) )
		{
		/* Only test the PKC functions if RSA is enabled */
		if( !testKeyExportImport() )
			return( FALSE );
		if( !testSignData() )
			return( FALSE );
		if( !testKeygen() )
			return( FALSE );
		if( !testKeygenAsync() )
			return( FALSE );
		}
	/* No need for putchar, mid-level functions leave a blank line at end */

	return( TRUE );
	}
#else

BOOLEAN testMidLevel( void )
	{
	puts( "Skipping test of mid-level encryption routines...\n" );
	return( TRUE );
	}
#endif /* TEST_MIDLEVEL */

#ifdef TEST_HIGHLEVEL

/* Test the high-level routines (these are similar to the mid-level routines 
   but rely on things like certificate management to work) */

BOOLEAN testHighLevel( void )
	{
	if( !testKeyExportImportCMS() )
		return( FALSE );
	if( !testSignDataCMS() )
		return( FALSE );

	return( TRUE );
	}
#else

BOOLEAN testHighLevel( void )
	{
	puts( "Skipping test of high-level routines...\n" );
	return( TRUE );
	}
#endif /* TEST_HIGHLEVEL */

/****************************************************************************
*																			*
*								Test Certificates							*
*																			*
****************************************************************************/

#ifdef TEST_CERT

/* Test the certificate routines */

BOOLEAN testCert( void )
	{
	if( !testBasicCert() )
		return( FALSE );
	if( !testCACert() )
		return( FALSE );
	if( !testXyzzyCert() )
		return( FALSE );
	if( !testTextStringCert() )
		return( FALSE );
	if( !testComplexCert() )
		return( FALSE );
	if( !testCertExtension() )
		return( FALSE );
	if( !testCustomDNCert() )
		return( FALSE );
	if( !testSETCert() )
		return( FALSE );
	if( !testAttributeCert() )
		return( FALSE );
	if( !testCertRequest() )
		return( FALSE );
	if( !testComplexCertRequest() )
		return( FALSE );
	if( !testCRMFRequest() )
		return( FALSE );
	if( !testComplexCRMFRequest() )
		return( FALSE );
	if( !testCRL() )
		return( FALSE );
	if( !testComplexCRL() )
		return( FALSE );
	if( !testRevRequest() )
		return( FALSE );
	if( !testCertChain() )
		return( FALSE );
	if( !testCMSAttributes() )
		return( FALSE );
	if( !testOCSPReqResp() )
		return( FALSE );
	if( !testCertImport() )
		return( FALSE );
	if( !testCertImportECC() )
		return( FALSE );
	if( !testCertReqImport() )
		return( FALSE );
	if( !testCRLImport() )
		return( FALSE );
	if( !testCertChainImport() )
		return( FALSE );
	if( !testOCSPImport() )
		return( FALSE );
	if( !testBase64CertImport() )
		return( FALSE );
	if( !testBase64CertChainImport() )
		return( FALSE );
	if( !testMiscImport() )
		return( FALSE );
	if( !testNonchainCert() )
		return( FALSE );
	if( !testCertComplianceLevel() )
		return( FALSE );
	if( !testPKCS1Padding() )
		return( FALSE );
#if 0	/* This takes a while to run and produces a lot of output that won't
		   be meaningful to anyone other than cryptlib developers so it's
		   disabled by default */
	if( !testPathProcessing() )
		return( FALSE );
#endif /* 0 */

	return( TRUE );
	}
#else

BOOLEAN testCert( void )
	{
	puts( "Skipping test of certificate routines...\n" );
	return( TRUE );
	}
#endif /* TEST_CERT */

#ifdef TEST_CERTPROCESS

/* Test the certificate processing and CA certificate management 
   functionality.  A side-effect of the certificate-management 
   functionality is that the OCSP EE test certificates are written 
   to the test data directory */

BOOLEAN testCertMgmt( void )
	{
	int status;

	if( !testCertProcess() )
		return( FALSE );
	status = testCertManagement();
	if( status == CRYPT_ERROR_NOTAVAIL )
		{
		puts( "Handling for CA certificate stores doesn't appear to be "
			  "enabled in this\nbuild of cryptlib, skipping the test of "
			  "the certificate management routines.\n" );
		}
	else
		{
		if( !status )
			return( FALSE );
		}

	return( TRUE );
	}
#else

BOOLEAN testCertMgmt( void )
	{
	puts( "Skipping test of certificate handling/CA management...\n" );
	return( TRUE );
	}
#endif /* TEST_CERTPROCESS */

/****************************************************************************
*																			*
*								Test Keysets								*
*																			*
****************************************************************************/

#ifdef TEST_KEYSET

/* Test the keyset read routines */

BOOLEAN testKeyset( void )
	{
	int status;

  #ifdef DATABASE_AUTOCONFIG
	checkCreateDatabaseKeysets();
  #endif /* DATABASE_AUTOCONFIG */
	if( !testGetPGPPublicKey() )
		return( FALSE );
	if( !testGetPGPPrivateKey() )
		return( FALSE );
	if( !testGetBorkenKey() )
		return( FALSE );
	if( !testReadWriteFileKey() )
		return( FALSE );
	if( !testImportFileKey() )
		return( FALSE );
	if( !testReadFilePublicKey() )
		return( FALSE );
	if( !testDeleteFileKey() )
		return( FALSE );
	if( !testUpdateFileCert() )
		return( FALSE );
	if( !testReadFileCert() )
		return( FALSE );
	if( !testReadFileCertPrivkey() )
		return( FALSE );
	if( !testWriteFileCertChain() )
		return( FALSE );
	if( !testReadFileCertChain() )
		return( FALSE );
	if( !testAddTrustedCert() )
		return( FALSE );
#if 0	/* This changes the global config file and is disabled by default */
	if( !testAddGloballyTrustedCert() )
		return( FALSE );
#endif /* 0 */
	if( !testWriteFileLongCertChain() )
		return( FALSE );
	if( !testSingleStepFileCert() )
		return( FALSE );
	if( !testDoubleCertFile() )
		return( FALSE );
	if( !testRenewedCertFile() )
		return( FALSE );
	if( !testReadMiscFile() )
		return( FALSE );
	status = testWriteCert();
	if( status == CRYPT_ERROR_NOTAVAIL )
		{
		puts( "Handling for certificate databases doesn't appear to be "
			  "enabled in this\nbuild of cryptlib, skipping the test of "
			  "the certificate database routines.\n" );
		}
	else
		{
		if( status == TRUE )
			{
			if( !testReadCert() )
				return( FALSE );
			if( !testKeysetQuery() )
				return( FALSE );
			}
		}
	/* For the following tests we may have read access but not write access,
	   so we test a read of known-present certs before trying a write -
	   unlike the local keysets we don't need to add a certificate before we 
	   can try reading it */
	status = testReadCertLDAP();
	if( status == CRYPT_ERROR_NOTAVAIL )
		{
		puts( "Handling for LDAP certificate directories doesn't appear to "
			  "be enabled in\nthis build of cryptlib, skipping the test of "
			  "the certificate directory\nroutines.\n" );
		}
	else
		{
		/* LDAP access can fail if the directory doesn't use the standard
		   du jour, so we don't treat a failure as a fatal error */
		if( status )
			{
			/* LDAP writes are even worse than LDAP reads, so we don't
			   treat failures here as fatal either */
			testWriteCertLDAP();
			}
		}
	status = testReadCertURL();
	if( status == CRYPT_ERROR_NOTAVAIL )
		{
		puts( "Handling for fetching certificates from web pages doesn't "
			  "appear to be\nenabled in this build of cryptlib, skipping "
			  "the test of the HTTP routines.\n" );
		}
	else
		{
		/* Being able to read a certificate from a web page is rather 
		   different from access to an HTTP certificate store so we don't 
		   treat an error here as fatal */
		if( status )
			testReadCertHTTP();
		}

	return( TRUE );
	}
#else

BOOLEAN testKeyset( void )
	{
	puts( "Skipping test of keyset read routines...\n" );
	return( TRUE );
	}
#endif /* TEST_KEYSET */

/****************************************************************************
*																			*
*								Test Enveloping								*
*																			*
****************************************************************************/

#ifdef TEST_ENVELOPE

/* Test the enveloping routines */

BOOLEAN testEnveloping( void )
	{
	if( !testEnvelopeData() )
		return( FALSE );
	if( !testEnvelopeDataLargeBuffer() )
		return( FALSE );
	if( !testEnvelopeCompress() )
		return( FALSE );
	if( !testPGPEnvelopeCompressedDataImport() )
		return( FALSE );
	if( !testEnvelopeSessionCrypt() )
		return( FALSE );
	if( !testEnvelopeSessionCryptLargeBuffer() )
		return( FALSE );
	if( !testEnvelopeCrypt() )
		return( FALSE );
	if( !testEnvelopePasswordCrypt() )
		return( FALSE );
	if( !testEnvelopePasswordCryptImport() )
		return( FALSE );
	if( !testPGPEnvelopePasswordCryptImport() )
		return( FALSE );
	if( !testEnvelopePKCCrypt() )
		return( FALSE );
	if( !testEnvelopePKCCryptAlgo() )
		return( FALSE );
	if( !testPGPEnvelopePKCCryptImport() )
		return( FALSE );
	if( !testEnvelopeSign() )
		return( FALSE );
	if( !testEnvelopeSignAlgo() )
		return( FALSE );
	if( !testEnvelopeSignOverflow() )
		return( FALSE );
	if( !testPGPEnvelopeSignedDataImport() )
		return( FALSE );
	if( !testEnvelopeAuthenticate() )
		return( FALSE );
	if( !testEnvelopeAuthEnc() )
		return( FALSE );
	if( !testCMSEnvelopePKCCrypt() )
		return( FALSE );
	if( !testCMSEnvelopePKCCryptDoubleCert() )
		return( FALSE );
	if( !testCMSEnvelopePKCCryptImport() )
		return( FALSE );
	if( !testCMSEnvelopeSign() )
		return( FALSE );
	if( !testCMSEnvelopeDualSign() )
		return( FALSE );
	if( !testCMSEnvelopeDetachedSig() )
		return( FALSE );
	if( !testCMSEnvelopeSignedDataImport() )
		return( FALSE );

	return( TRUE );
	}
#else

BOOLEAN testEnveloping( void )
	{
	puts( "Skipping test of enveloping routines...\n" );
	return( TRUE );
	}
#endif /* TEST_ENVELOPE */

/****************************************************************************
*																			*
*								Test Sessions								*
*																			*
****************************************************************************/

#ifdef TEST_SESSION

/* Test the session routines */

BOOLEAN testSessions( void )
	{
	int status;

	status = testSessionUrlParse();
	if( !status )
		return( FALSE );
	if( status == CRYPT_ERROR_NOTAVAIL )
		{
		puts( "Network access doesn't appear to be enabled in this build of "
			  "cryptlib,\nskipping the test of the secure session routines.\n" );
		return( TRUE );
		}
	if( !checkNetworkAccess() )
		{
		puts( "Couldn't perform a test connect to a well-known site "
			  "(Amazon.com) which\nindicates that external network access "
			  "isn't available.  Is this machine\nsituated behind a "
			  "firewall?\n" );
		return( FALSE );
		}
	if( !testSessionAttributes() )
		return( FALSE );
	if( !testSessionSSHv1() )
		return( FALSE );
	if( !testSessionSSH() )
		return( FALSE );
	if( !testSessionSSHClientCert() )
		return( FALSE );
	if( !testSessionSSHPortforward() )
		return( FALSE );
	if( !testSessionSSHExec() )
		return( FALSE );
	if( !testSessionSSL() )
		return( FALSE );
	if( !testSessionSSLLocalSocket() )
		return( FALSE );
	if( !testSessionTLS() )
		return( FALSE );
	if( !testSessionTLS11() )
		return( FALSE );
	if( !testSessionTLS12() )
		return( FALSE );
#if 0	/* The MS test server used for the general TLS 1.2 tests requires 
		   fairly extensive custom configuration of client certs and the
		   ability to do rehandshakes due to the oddball way that schannel
		   handles client-auth, so we disable this test until another
		   server that actually does TLS 1.2 client auth appears */
	if( !testSessionTLS12ClientCert() )
		return( FALSE );
#endif /* 0 */
	if( !testSessionOCSP() )
		return( FALSE );
	if( !testSessionTSP() )
		return( FALSE );
	if( !testSessionEnvTSP() )
		return( FALSE );
	if( !testSessionCMP() )
		return( FALSE );

	return( TRUE );
	}
#else

BOOLEAN testSessions( void )
	{
	puts( "Skipping test of session routines...\n" );
	return( TRUE );
	}
#endif /* TEST_SESSION */

#ifdef TEST_SESSION_LOOPBACK

/* Test loopback client/server sessions.  These require a threaded OS and 
   are aliased to no-ops on non-threaded systems.  In addition there can be 
   synchronisation problems between the two threads if the server is delayed 
   for some reason, resulting in the client waiting for a socket that isn't 
   opened yet.  This isn't easy to fix without a lot of explicit intra-
   thread synchronisation, if there's a problem it's easier to just re-run 
   the tests */

BOOLEAN testSessionsLoopback( void )
	{
  #ifdef DATABASE_AUTOCONFIG
	checkCreateDatabaseKeysets();	/* Needed for PKI tests */
  #endif /* DATABASE_AUTOCONFIG */
	if( !testSessionSSHv1ClientServer() )
		return( FALSE );
	if( !testSessionSSHClientServer() )
		return( FALSE );
	if( !testSessionSSHClientServerDsaKey() )
		return( FALSE );
	if( !testSessionSSHClientServerEccKey() )
		return( FALSE );
	if( !testSessionSSHClientServerFingerprint() )
		return( FALSE );
	if( !testSessionSSHClientServerPortForward() )
		return( FALSE );
	if( !testSessionSSHClientServerExec() )
		return( FALSE );
	if( !testSessionSSHClientServerMultichannel() )
		return( FALSE );
	if( !testSessionSSLClientServer() )
		return( FALSE );
	if( !testSessionSSLClientCertClientServer() )
		return( FALSE );
	if( !testSessionTLSClientServer() )
		return( FALSE );
	if( !testSessionTLSSharedKeyClientServer() )
		return( FALSE );
	if( !testSessionTLSNoSharedKeyClientServer() )
		return( FALSE );
	if( !testSessionTLSBulkTransferClientServer() )
		return( FALSE );
	if( !testSessionTLS11ClientServer() )
		return( FALSE );
	if( !testSessionTLS12ClientServer() )
		return( FALSE );
	if( !testSessionTLS12ClientCertClientServer() )
		return( FALSE );
	if( !testSessionHTTPCertstoreClientServer() )
		return( FALSE );
	if( !testSessionRTCSClientServer() )
		return( FALSE );
	if( !testSessionOCSPClientServer() )
		return( FALSE );
	if( !testSessionTSPClientServer() )
		return( FALSE );
	if( !testSessionTSPClientServerPersistent() )
		return( FALSE );
	if( !testSessionSCEPClientServer() )
		return( FALSE );
	if( !testSessionSCEPCACertClientServer() )
		return( FALSE );
	if( !testSessionCMPClientServer() )
		return( FALSE );
	if( !testSessionCMPSHA2ClientServer() )
		return( FALSE );
	if( !testSessionCMPPKIBootClientServer() )
		return( FALSE );
	if( !testSessionPNPPKIClientServer() )
		return( FALSE );
	if( !testSessionPNPPKICAClientServer() )
		return( FALSE );
	if( !testSessionCMPFailClientServer() )
		return( FALSE );

	/* The final set of loopback tests, which spawn a large number of 
	   threads, can be somewhat alarming due to the amount of message spew 
	   that they produce so we only run them on one specific development 
	   test machine */
#if defined( __WINDOWS__ ) && !defined( _WIN32_WCE )
	{
	char name[ MAX_COMPUTERNAME_LENGTH + 1 ];
	int length = MAX_COMPUTERNAME_LENGTH + 1;

	if( GetComputerName( name, &length ) && length == 7 && \
		!memcmp( name, "WHISPER", length ) && 0 )
		{
		if( !testSessionSSHClientServerDualThread() )
			return( FALSE );
		if( !testSessionSSHClientServerMultiThread() )
			return( FALSE );
		if( !testSessionTLSClientServerMultiThread() )
			return( FALSE );
		}
	}
#endif /* __WINDOWS__ && !WinCE */
	return( TRUE );
	}
#else

BOOLEAN testSessionsLoopback( void )
	{
	puts( "Skipping test of session routines...\n" );
	return( TRUE );
	}
#endif /* TEST_SESSION_LOOPBACK */

/****************************************************************************
*																			*
*								Test Users									*
*																			*
****************************************************************************/

#ifdef TEST_USER

/* Test the user routines */

BOOLEAN testUsers( void )
	{
	if( !testUser() )
		return( FALSE );

	return( TRUE );
	}
#else

BOOLEAN testUsers( void )
	{
	puts( "Skipping test of user routines...\n" );
	return( TRUE );
	}
#endif /* TEST_USER */
