SUITE_PRE_DEP #define TEST_SUITE_ACTIVE int verify_string( char **str ) { if( (*str)[0] != '"' || (*str)[strlen( *str ) - 1] != '"' ) { mbedtls_printf( "Expected string (with \"\") for parameter and got: %s\n", *str ); return( -1 ); } (*str)++; (*str)[strlen( *str ) - 1] = '\0'; return( 0 ); } int verify_int( char *str, int *value ) { size_t i; int minus = 0; int digits = 1; int hex = 0; for( i = 0; i < strlen( str ); i++ ) { if( i == 0 && str[i] == '-' ) { minus = 1; continue; } if( ( ( minus && i == 2 ) || ( !minus && i == 1 ) ) && str[i - 1] == '0' && str[i] == 'x' ) { hex = 1; continue; } if( ! ( ( str[i] >= '0' && str[i] <= '9' ) || ( hex && ( ( str[i] >= 'a' && str[i] <= 'f' ) || ( str[i] >= 'A' && str[i] <= 'F' ) ) ) ) ) { digits = 0; break; } } if( digits ) { if( hex ) *value = strtol( str, NULL, 16 ); else *value = strtol( str, NULL, 10 ); return( 0 ); } MAPPING_CODE mbedtls_printf( "Expected integer for parameter and got: %s\n", str ); return( -1 ); } /*----------------------------------------------------------------------------*/ /* Test Case code */ FUNCTION_CODE SUITE_POST_DEP /*----------------------------------------------------------------------------*/ /* Test dispatch code */ int dep_check( char *str ) { if( str == NULL ) return( 1 ); DEP_CHECK_CODE return( 1 ); } int dispatch_test(int cnt, char *params[50]) { int ret; ((void) cnt); ((void) params); #if defined(TEST_SUITE_ACTIVE) DISPATCH_FUNCTION { mbedtls_fprintf( stdout, "FAILED\nSkipping unknown test function '%s'\n", params[0] ); fflush( stdout ); return( 1 ); } #else return( 3 ); #endif return( ret ); } /*----------------------------------------------------------------------------*/ /* Main Test code */ /** Retrieve one input line into buf, which must have room for len * bytes. The trailing line break (if any) is stripped from the result. * Lines beginning with the character '#' are skipped. Lines that are * more than len-1 bytes long including the trailing line break are * truncated; note that the following bytes remain in the input stream. * * \return 0 on success, -1 on error or end of file */ int get_line( FILE *f, char *buf, size_t len ) { char *ret; do { ret = fgets( buf, len, f ); if( ret == NULL ) return( -1 ); } while( buf[0] == '#' ); ret = buf + strlen( buf ); if( ret-- > buf && *ret == '\n' ) *ret = '\0'; if( ret-- > buf && *ret == '\r' ) *ret = '\0'; return( 0 ); } int parse_arguments( char *buf, size_t len, char *params[50] ) { int cnt = 0, i; char *cur = buf; char *p = buf, *q; params[cnt++] = cur; while( *p != '\0' && p < buf + len ) { if( *p == '\\' ) { p++; p++; continue; } if( *p == ':' ) { if( p + 1 < buf + len ) { cur = p + 1; params[cnt++] = cur; } *p = '\0'; } p++; } /* Replace newlines, question marks and colons in strings */ for( i = 0; i < cnt; i++ ) { p = params[i]; q = params[i]; while( *p != '\0' ) { if( *p == '\\' && *(p + 1) == 'n' ) { p += 2; *(q++) = '\n'; } else if( *p == '\\' && *(p + 1) == ':' ) { p += 2; *(q++) = ':'; } else if( *p == '\\' && *(p + 1) == '?' ) { p += 2; *(q++) = '?'; } else *(q++) = *(p++); } *q = '\0'; } return( cnt ); } static int test_snprintf( size_t n, const char ref_buf[10], int ref_ret ) { int ret; char buf[10] = "xxxxxxxxx"; const char ref[10] = "xxxxxxxxx"; ret = mbedtls_snprintf( buf, n, "%s", "123" ); if( ret < 0 || (size_t) ret >= n ) ret = -1; if( strncmp( ref_buf, buf, sizeof( buf ) ) != 0 || ref_ret != ret || memcmp( buf + n, ref + n, sizeof( buf ) - n ) != 0 ) { return( 1 ); } return( 0 ); } static int run_test_snprintf( void ) { return( test_snprintf( 0, "xxxxxxxxx", -1 ) != 0 || test_snprintf( 1, "", -1 ) != 0 || test_snprintf( 2, "1", -1 ) != 0 || test_snprintf( 3, "12", -1 ) != 0 || test_snprintf( 4, "123", 3 ) != 0 || test_snprintf( 5, "123", 3 ) != 0 ); } int main() { int ret, i, cnt, total_errors = 0, total_tests = 0, total_skipped = 0; const char *filename = "TEST_FILENAME"; FILE *file; char buf[5000]; char *params[50]; void *pointer; #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C) && \ !defined(TEST_SUITE_MEMORY_BUFFER_ALLOC) unsigned char alloc_buf[1000000]; mbedtls_memory_buffer_alloc_init( alloc_buf, sizeof(alloc_buf) ); #endif /* * The C standard doesn't guarantee that all-bits-0 is the representation * of a NULL pointer. We do however use that in our code for initializing * structures, which should work on every modern platform. Let's be sure. */ memset( &pointer, 0, sizeof( void * ) ); if( pointer != NULL ) { mbedtls_fprintf( stderr, "all-bits-zero is not a NULL pointer\n" ); return( 1 ); } /* * Make sure we have a snprintf that correctly zero-terminates */ if( run_test_snprintf() != 0 ) { mbedtls_fprintf( stderr, "the snprintf implementation is broken\n" ); return( 0 ); } file = fopen( filename, "r" ); if( file == NULL ) { mbedtls_fprintf( stderr, "Failed to open\n" ); return( 1 ); } while( !feof( file ) ) { int skip = 0; if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 ) break; mbedtls_fprintf( stdout, "%s%.66s", test_errors ? "\n" : "", buf ); mbedtls_fprintf( stdout, " " ); for( i = strlen( buf ) + 1; i < 67; i++ ) mbedtls_fprintf( stdout, "." ); mbedtls_fprintf( stdout, " " ); fflush( stdout ); total_tests++; if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 ) break; cnt = parse_arguments( buf, strlen(buf), params ); if( strcmp( params[0], "depends_on" ) == 0 ) { for( i = 1; i < cnt; i++ ) if( dep_check( params[i] ) != 0 ) skip = 1; if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 ) break; cnt = parse_arguments( buf, strlen(buf), params ); } if( skip == 0 ) { test_errors = 0; ret = dispatch_test( cnt, params ); } if( skip == 1 || ret == 3 ) { total_skipped++; mbedtls_fprintf( stdout, "----\n" ); fflush( stdout ); } else if( ret == 0 && test_errors == 0 ) { mbedtls_fprintf( stdout, "PASS\n" ); fflush( stdout ); } else if( ret == 2 ) { mbedtls_fprintf( stderr, "FAILED: FATAL PARSE ERROR\n" ); fclose(file); mbedtls_exit( 2 ); } else total_errors++; if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 ) break; if( strlen(buf) != 0 ) { mbedtls_fprintf( stderr, "Should be empty %d\n", (int) strlen(buf) ); return( 1 ); } } fclose(file); mbedtls_fprintf( stdout, "\n----------------------------------------------------------------------------\n\n"); if( total_errors == 0 ) mbedtls_fprintf( stdout, "PASSED" ); else mbedtls_fprintf( stdout, "FAILED" ); mbedtls_fprintf( stdout, " (%d / %d tests (%d skipped))\n", total_tests - total_errors, total_tests, total_skipped ); #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C) && \ !defined(TEST_SUITE_MEMORY_BUFFER_ALLOC) #if defined(MBEDTLS_MEMORY_DEBUG) mbedtls_memory_buffer_alloc_status(); #endif mbedtls_memory_buffer_alloc_free(); #endif return( total_errors != 0 ); }