7bc9f68232
Switch from the direct use of slot numbers to handles allocated by psa_allocate_key. The general principle for each function is: * Change `psa_key_slot_t slot` to `psa_key_handle_t handle` or `psa_key_id_t key_id` depending on whether it's used as a handle to an open slot or as a persistent name for a key. * Call psa_create_key() before using a slot, instead of calling psa_set_key_lifetime to make a slot persistent. Remove the unit test persistent_key_is_configurable which is no longer relevant.
160 lines
4.5 KiB
Text
160 lines
4.5 KiB
Text
/* BEGIN_HEADER */
|
|
#include <stdint.h>
|
|
#include "psa/crypto.h"
|
|
#include "psa_crypto_storage_backend.h"
|
|
|
|
/* END_HEADER */
|
|
|
|
/* BEGIN_DEPENDENCIES
|
|
* depends_on:MBEDTLS_PSA_CRYPTO_C:MBEDTLS_PSA_CRYPTO_STORAGE_FILE_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
|
|
/* BEGIN_CASE */
|
|
void load_data_from_file( int id_to_load_arg,
|
|
data_t *data, int should_make_file,
|
|
int capacity_arg, int expected_status )
|
|
{
|
|
psa_key_id_t id_to_load = id_to_load_arg;
|
|
char slot_location[] = "psa_key_slot_1";
|
|
psa_status_t status;
|
|
int ret;
|
|
size_t file_size = 0;
|
|
uint8_t *loaded_data = NULL;
|
|
size_t capacity = (size_t) capacity_arg;
|
|
|
|
if( should_make_file == 1 )
|
|
{
|
|
/* Create a file with data contents, with mask permissions. */
|
|
FILE *file;
|
|
file = fopen( slot_location, "wb+" );
|
|
TEST_ASSERT( file != NULL );
|
|
file_size = fwrite( data->x, 1, data->len, file );
|
|
TEST_ASSERT( file_size == data->len );
|
|
ret = fclose( file );
|
|
TEST_ASSERT( ret == 0 );
|
|
}
|
|
|
|
/* Read from the file with psa_crypto_storage_load. */
|
|
loaded_data = mbedtls_calloc( 1, capacity );
|
|
TEST_ASSERT( loaded_data != NULL );
|
|
status = psa_crypto_storage_load( id_to_load, loaded_data, file_size );
|
|
|
|
/* Check we get the expected status. */
|
|
TEST_ASSERT( status == expected_status );
|
|
if( status != PSA_SUCCESS )
|
|
goto exit;
|
|
|
|
/* Check that the file data and data length is what we expect. */
|
|
ASSERT_COMPARE( data->x, data->len, loaded_data, file_size );
|
|
|
|
exit:
|
|
mbedtls_free( loaded_data );
|
|
remove( slot_location );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void write_data_to_file( data_t *data, int expected_status )
|
|
{
|
|
char slot_location[] = "psa_key_slot_1";
|
|
psa_status_t status;
|
|
int ret;
|
|
FILE *file;
|
|
size_t file_size;
|
|
size_t num_read;
|
|
uint8_t *loaded_data = NULL;
|
|
|
|
/* Write data to file. */
|
|
status = psa_crypto_storage_store( 1, data->x, data->len );
|
|
|
|
/* Check that we got the expected status. */
|
|
TEST_ASSERT( status == expected_status );
|
|
if( status != PSA_SUCCESS )
|
|
goto exit;
|
|
|
|
/* Check that the file length is what we expect */
|
|
file = fopen( slot_location, "rb" );
|
|
TEST_ASSERT( file != NULL );
|
|
fseek( file, 0, SEEK_END );
|
|
file_size = (size_t) ftell( file );
|
|
fseek( file, 0, SEEK_SET );
|
|
TEST_ASSERT( file_size == data->len );
|
|
|
|
/* Check that the file contents are what we expect */
|
|
loaded_data = mbedtls_calloc( 1, data->len );
|
|
TEST_ASSERT( loaded_data != NULL );
|
|
|
|
num_read = fread( loaded_data, 1, file_size, file );
|
|
TEST_ASSERT( num_read == file_size );
|
|
ASSERT_COMPARE( data->x, data->len, loaded_data, file_size );
|
|
ret = fclose( file );
|
|
TEST_ASSERT( ret == 0 );
|
|
|
|
exit:
|
|
mbedtls_free( loaded_data );
|
|
remove( slot_location );
|
|
}
|
|
/* END_CASE */
|
|
|
|
|
|
/* BEGIN_CASE */
|
|
void get_file_size( data_t *data, int expected_data_length,
|
|
int expected_status, int should_make_file )
|
|
{
|
|
char slot_location[] = "psa_key_slot_1";
|
|
psa_status_t status;
|
|
int ret;
|
|
size_t file_size;
|
|
|
|
if( should_make_file )
|
|
{
|
|
/* Create a file with data contents, with mask permissions. */
|
|
FILE *file;
|
|
file = fopen( slot_location, "wb+" );
|
|
TEST_ASSERT( file != NULL );
|
|
file_size = fwrite( data->x, 1, data->len, file );
|
|
TEST_ASSERT( file_size == data->len );
|
|
ret = fclose( file );
|
|
TEST_ASSERT( ret == 0 );
|
|
}
|
|
|
|
/* Check get data size is what we expect */
|
|
status = psa_crypto_storage_get_data_length( 1, &file_size );
|
|
TEST_ASSERT( status == expected_status );
|
|
if( expected_status == PSA_SUCCESS )
|
|
TEST_ASSERT( file_size == (size_t)expected_data_length );
|
|
|
|
exit:
|
|
remove( slot_location );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void write_data_to_prexisting_file( char *preexist_file_location,
|
|
data_t *data, int expected_status )
|
|
{
|
|
char slot_location[] = "psa_key_slot_1";
|
|
psa_status_t status;
|
|
int ret;
|
|
FILE *file;
|
|
|
|
/* Create file first */
|
|
file = fopen( preexist_file_location, "wb" );
|
|
TEST_ASSERT( file != NULL );
|
|
ret = fclose( file );
|
|
TEST_ASSERT( ret == 0 );
|
|
|
|
/* Write data to file. */
|
|
status = psa_crypto_storage_store( 1, data->x, data->len );
|
|
|
|
/* Check that we got the expected status. */
|
|
TEST_ASSERT( status == expected_status );
|
|
if( status != PSA_SUCCESS )
|
|
goto exit;
|
|
|
|
exit:
|
|
remove( preexist_file_location );
|
|
remove( slot_location );
|
|
}
|
|
/* END_CASE */
|