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( KEY_VALUE_MAPPING_NOT_FOUND ); } /*----------------------------------------------------------------------------*/ /* 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( DEPENDENCY_NOT_SUPPORTED ); } int dispatch_test(int cnt, char *params[50]) { int ret; ((void) cnt); ((void) params); #if defined(TEST_SUITE_ACTIVE) ret = DISPATCH_TEST_SUCCESS; DISPATCH_FUNCTION { mbedtls_fprintf( stdout, "FAILED\nSkipping unknown test function '%s'\n", params[0] ); fflush( stdout ); ret = DISPATCH_TEST_FN_NOT_FOUND; } #else ret = DISPATCH_UNSUPPORTED_SUITE; #endif return( ret ); } /*----------------------------------------------------------------------------*/ /* Main Test code */ #define USAGE \ "Usage: %s [OPTIONS] files...\n\n" \ " Command line arguments:\n" \ " files... One or more test data file. If no file is specified\n" \ " the followimg default test case is used:\n" \ " %s\n\n" \ " Options:\n" \ " -v | --verbose Display full information about each test\n" \ " -h | --help Display this information\n\n", \ argv[0], \ "TEST_FILENAME" int get_line( FILE *f, char *buf, size_t len ) { char *ret; ret = fgets( buf, len, f ); if( ret == NULL ) return( -1 ); if( strlen( buf ) && buf[strlen(buf) - 1] == '\n' ) buf[strlen(buf) - 1] = '\0'; if( strlen( buf ) && buf[strlen(buf) - 1] == '\r' ) buf[strlen(buf) - 1] = '\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 argc, const char *argv[]) { /* Local Configurations and options */ const char *default_filename = "TEST_FILENAME"; const char *test_filename = NULL; const char **test_files = NULL; int testfile_count = 0; int option_verbose = 0; /* Other Local variables */ int arg_index = 1; const char *next_arg; int testfile_index, ret, i, cnt; int total_errors = 0, total_tests = 0, total_skipped = 0; 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 ); } while( arg_index < argc) { next_arg = argv[ arg_index ]; if( strcmp(next_arg, "--verbose" ) == 0 || strcmp(next_arg, "-v" ) == 0 ) { option_verbose = 1; } else if( strcmp(next_arg, "--help" ) == 0 || strcmp(next_arg, "-h" ) == 0 ) { mbedtls_fprintf( stdout, USAGE ); mbedtls_exit( EXIT_SUCCESS ); } else { /* Not an option, therefore treat all further arguments as the file * list. */ test_files = &argv[ arg_index ]; testfile_count = argc - arg_index; } arg_index++; } /* If no files were specified, assume a default */ if ( test_files == NULL || testfile_count == 0 ) { test_files = &default_filename; testfile_count = 1; } /* Now begin to execute the tests in the testfiles */ for ( testfile_index = 0; testfile_index < testfile_count; testfile_index++ ) { test_filename = test_files[ testfile_index ]; file = fopen( test_filename, "r" ); if( file == NULL ) { mbedtls_fprintf( stderr, "Failed to open test file: %s\n", test_filename ); return( 1 ); } while( !feof( file ) ) { int unmet_dep_count = 0; char *unmet_dependencies[20]; 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] ) != DEPENDENCY_SUPPORTED ) { unmet_dependencies[ i-1 ] = strdup(params[i]); if( unmet_dependencies[ i-1 ] == NULL ) { mbedtls_printf("FATAL: Out of memory\n"); mbedtls_exit( MBEDTLS_EXIT_FAILURE ); } unmet_dep_count++; } } if( ( ret = get_line( file, buf, sizeof(buf) ) ) != 0 ) break; cnt = parse_arguments( buf, strlen(buf), params ); } // If there are no unmet dependencies execute the test if( unmet_dep_count == 0 ) { test_errors = 0; ret = dispatch_test( cnt, params ); } if( unmet_dep_count > 0 || ret == DISPATCH_UNSUPPORTED_SUITE ) { total_skipped++; mbedtls_fprintf( stdout, "----\n" ); if( 1 == option_verbose && ret == DISPATCH_UNSUPPORTED_SUITE ) { mbedtls_fprintf( stdout, " Test Suite not enabled" ); } if( 1 == option_verbose && unmet_dep_count > 0 ) { mbedtls_fprintf( stdout, " Unmet dependencies: " ); while( unmet_dep_count > 0) { mbedtls_fprintf(stdout, "%s ", unmet_dependencies[unmet_dep_count - 1]); free(unmet_dependencies[unmet_dep_count - 1]); unmet_dep_count--; } mbedtls_fprintf( stdout, "\n" ); } fflush( stdout ); } else if( ret == DISPATCH_TEST_SUCCESS && test_errors == 0 ) { mbedtls_fprintf( stdout, "PASS\n" ); fflush( stdout ); } else if( ret == DISPATCH_INVALID_TEST_DATA ) { 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 ); }