Add conditional platform context creation & usage
Add another layer of abstraction before calling platform setup and teardown.
This commit is contained in:
parent
aca09c7026
commit
32a675f032
2 changed files with 31 additions and 12 deletions
|
@ -109,7 +109,9 @@ static struct
|
||||||
}
|
}
|
||||||
test_info;
|
test_info;
|
||||||
|
|
||||||
|
#if defined(MBEDTLS_PLATFORM_C)
|
||||||
mbedtls_platform_context platform_ctx;
|
mbedtls_platform_context platform_ctx;
|
||||||
|
#endif
|
||||||
|
|
||||||
/*----------------------------------------------------------------------------*/
|
/*----------------------------------------------------------------------------*/
|
||||||
/* Helper flags for complex dependencies */
|
/* Helper flags for complex dependencies */
|
||||||
|
@ -128,6 +130,23 @@ mbedtls_platform_context platform_ctx;
|
||||||
|
|
||||||
/*----------------------------------------------------------------------------*/
|
/*----------------------------------------------------------------------------*/
|
||||||
/* Helper Functions */
|
/* Helper Functions */
|
||||||
|
static int platform_setup()
|
||||||
|
{
|
||||||
|
#if defined(MBEDTLS_PLATFORM_C)
|
||||||
|
if( mbedtls_platform_setup( &platform_ctx ) )
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
#endif /* MBEDTLS_PLATFORM_C */
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void platform_teardown()
|
||||||
|
{
|
||||||
|
#if defined(MBEDTLS_PLATFORM_C)
|
||||||
|
mbedtls_platform_teardown( &platform_ctx );
|
||||||
|
#endif /* MBEDTLS_PLATFORM_C */
|
||||||
|
}
|
||||||
|
|
||||||
#if defined(__unix__) || (defined(__APPLE__) && defined(__MACH__))
|
#if defined(__unix__) || (defined(__APPLE__) && defined(__MACH__))
|
||||||
static int redirect_output( FILE** out_stream, const char* path )
|
static int redirect_output( FILE** out_stream, const char* path )
|
||||||
|
|
|
@ -282,7 +282,7 @@ int main(int argc, const char *argv[])
|
||||||
!defined(TEST_SUITE_MEMORY_BUFFER_ALLOC)
|
!defined(TEST_SUITE_MEMORY_BUFFER_ALLOC)
|
||||||
unsigned char alloc_buf[1000000];
|
unsigned char alloc_buf[1000000];
|
||||||
#endif
|
#endif
|
||||||
if( mbedtls_platform_setup( &platform_ctx ) )
|
if( platform_setup() )
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "FATAL: Failed to initialize platform" );
|
mbedtls_fprintf( stderr, "FATAL: Failed to initialize platform" );
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -301,7 +301,7 @@ int main(int argc, const char *argv[])
|
||||||
if( pointer != NULL )
|
if( pointer != NULL )
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "all-bits-zero is not a NULL pointer\n" );
|
mbedtls_fprintf( stderr, "all-bits-zero is not a NULL pointer\n" );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
return( 1 );
|
return( 1 );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -311,7 +311,7 @@ int main(int argc, const char *argv[])
|
||||||
if( run_test_snprintf() != 0 )
|
if( run_test_snprintf() != 0 )
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "the snprintf implementation is broken\n" );
|
mbedtls_fprintf( stderr, "the snprintf implementation is broken\n" );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
return( 0 );
|
return( 0 );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -328,7 +328,7 @@ int main(int argc, const char *argv[])
|
||||||
strcmp(next_arg, "-h" ) == 0 )
|
strcmp(next_arg, "-h" ) == 0 )
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stdout, USAGE );
|
mbedtls_fprintf( stdout, USAGE );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
mbedtls_exit( EXIT_SUCCESS );
|
mbedtls_exit( EXIT_SUCCESS );
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
|
@ -368,7 +368,7 @@ int main(int argc, const char *argv[])
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "Failed to open test file: %s\n",
|
mbedtls_fprintf( stderr, "Failed to open test file: %s\n",
|
||||||
test_filename );
|
test_filename );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
return( 1 );
|
return( 1 );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -378,7 +378,7 @@ int main(int argc, const char *argv[])
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr,
|
mbedtls_fprintf( stderr,
|
||||||
"FATAL: Dep count larger than zero at start of loop\n" );
|
"FATAL: Dep count larger than zero at start of loop\n" );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
mbedtls_exit( MBEDTLS_EXIT_FAILURE );
|
mbedtls_exit( MBEDTLS_EXIT_FAILURE );
|
||||||
}
|
}
|
||||||
unmet_dep_count = 0;
|
unmet_dep_count = 0;
|
||||||
|
@ -415,7 +415,7 @@ int main(int argc, const char *argv[])
|
||||||
if( unmet_dependencies[ unmet_dep_count ] == NULL )
|
if( unmet_dependencies[ unmet_dep_count ] == NULL )
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "FATAL: Out of memory\n" );
|
mbedtls_fprintf( stderr, "FATAL: Out of memory\n" );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
mbedtls_exit( MBEDTLS_EXIT_FAILURE );
|
mbedtls_exit( MBEDTLS_EXIT_FAILURE );
|
||||||
}
|
}
|
||||||
unmet_dep_count++;
|
unmet_dep_count++;
|
||||||
|
@ -441,8 +441,8 @@ int main(int argc, const char *argv[])
|
||||||
stdout_fd = redirect_output( &stdout, "/dev/null" );
|
stdout_fd = redirect_output( &stdout, "/dev/null" );
|
||||||
if( stdout_fd == -1 )
|
if( stdout_fd == -1 )
|
||||||
{
|
{
|
||||||
|
platform_teardown();
|
||||||
/* Redirection has failed with no stdout so exit */
|
/* Redirection has failed with no stdout so exit */
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
|
||||||
exit( 1 );
|
exit( 1 );
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -454,7 +454,7 @@ int main(int argc, const char *argv[])
|
||||||
if( !option_verbose && restore_output( &stdout, stdout_fd ) )
|
if( !option_verbose && restore_output( &stdout, stdout_fd ) )
|
||||||
{
|
{
|
||||||
/* Redirection has failed with no stdout so exit */
|
/* Redirection has failed with no stdout so exit */
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
exit( 1 );
|
exit( 1 );
|
||||||
}
|
}
|
||||||
#endif /* __unix__ || __APPLE__ __MACH__ */
|
#endif /* __unix__ || __APPLE__ __MACH__ */
|
||||||
|
@ -506,7 +506,7 @@ int main(int argc, const char *argv[])
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "FAILED: FATAL PARSE ERROR\n" );
|
mbedtls_fprintf( stderr, "FAILED: FATAL PARSE ERROR\n" );
|
||||||
fclose( file );
|
fclose( file );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
mbedtls_exit( 2 );
|
mbedtls_exit( 2 );
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
|
@ -518,7 +518,7 @@ int main(int argc, const char *argv[])
|
||||||
{
|
{
|
||||||
mbedtls_fprintf( stderr, "Should be empty %d\n",
|
mbedtls_fprintf( stderr, "Should be empty %d\n",
|
||||||
(int) strlen( buf ) );
|
(int) strlen( buf ) );
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
return( 1 );
|
return( 1 );
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -551,6 +551,6 @@ int main(int argc, const char *argv[])
|
||||||
close_output( stdout );
|
close_output( stdout );
|
||||||
#endif /* __unix__ || __APPLE__ __MACH__ */
|
#endif /* __unix__ || __APPLE__ __MACH__ */
|
||||||
|
|
||||||
mbedtls_platform_teardown( &platform_ctx );
|
platform_teardown();
|
||||||
return( total_errors != 0 );
|
return( total_errors != 0 );
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue