crypto_se_driver: add key generation mock and test

Author: AndrzejKurek

tests/suites/test_suite_psa_crypto_se_driver_hal.data

@@ -32,3 +32,6 @@ mock_import:
 
 Key exporting mock test
 mock_export:
+
+Key generating mock test
+mock_generate:

tests/suites/test_suite_psa_crypto_se_driver_hal.function

@@ -11,6 +11,17 @@
  * This is probably a bug in the library. */
 #define PSA_ERROR_DETECTED_BY_DRIVER ((psa_status_t)( -500 ))
 
+static struct
+{
+    uint16_t called;
+    psa_key_slot_number_t key_slot;
+    psa_key_type_t type;
+    psa_key_usage_t usage;
+    size_t bits;
+    size_t extra_size;
+    size_t pubkey_out_size;
+} mock_generate_data;
+
 static struct
 {
     uint16_t called;
@@ -46,37 +57,67 @@ static void mock_teardown( void )
     memset( &mock_export_data, 0, sizeof( mock_export_data ) );
     memset( &mock_allocate_data, 0, sizeof( mock_allocate_data ) );
     memset( &mock_destroy_data, 0, sizeof( mock_destroy_data ) );
+    memset( &mock_generate_data, 0, sizeof( mock_generate_data ) );
+}
+
+static psa_status_t mock_generate( psa_key_slot_number_t key_slot,
+                                   psa_key_type_t type,
+                                   psa_key_usage_t usage,
+                                   size_t bits,
+                                   const void *extra,
+                                   size_t extra_size,
+                                   uint8_t *p_pubkey_out,
+                                   size_t pubkey_out_size,
+                                   size_t *p_pubkey_length )
+{
+    (void) extra;
+    (void) p_pubkey_out;
+    (void) p_pubkey_length;
+
+    mock_generate_data.called++;
+    mock_generate_data.key_slot = key_slot;
+    mock_generate_data.type = type;
+    mock_generate_data.usage = usage;
+    mock_generate_data.bits = bits;
+    mock_generate_data.extra_size = extra_size;
+    mock_generate_data.pubkey_out_size = pubkey_out_size;
+
+    return( PSA_SUCCESS );
 }
 
 static psa_status_t mock_import( psa_key_slot_number_t slot_number,
-                                psa_key_lifetime_t lifetime,
-                                psa_key_type_t type,
-                                psa_algorithm_t algorithm,
-                                psa_key_usage_t usage,
-                                const uint8_t *p_data,
-                                size_t data_length )
+                                 psa_key_lifetime_t lifetime,
+                                 psa_key_type_t type,
+                                 psa_algorithm_t algorithm,
+                                 psa_key_usage_t usage,
+                                 const uint8_t *p_data,
+                                 size_t data_length )
 {
     (void) p_data;
+
     mock_import_data.called++;
     mock_import_data.slot_number = slot_number;
     mock_import_data.lifetime = lifetime;
     mock_import_data.type = type;
     mock_import_data.algorithm = algorithm;
     mock_import_data.usage = usage;
     mock_import_data.data_length = data_length;
+
     return( PSA_SUCCESS );
 }
 
 psa_status_t mock_export( psa_key_slot_number_t slot_number,
-                         uint8_t *p_data,
-                         size_t data_size,
-                         size_t *p_data_length )
+                          uint8_t *p_data,
+                          size_t data_size,
+                          size_t *p_data_length )
 {
     (void) p_data;
     (void) p_data_length;
+
     mock_export_data.called++;
     mock_export_data.slot_number = slot_number;
     mock_export_data.data_size = data_size;
+
     return( PSA_SUCCESS );
 }
 
@@ -87,15 +128,18 @@ psa_status_t mock_allocate( const psa_key_attributes_t *attributes,
     (void) attributes;
     (void) slot_usage;
     (void) slot_number;
+
     mock_allocate_data.called++;
     *slot_number = 0;
+
     return( PSA_SUCCESS );
 }
 
 psa_status_t mock_destroy( psa_key_slot_number_t slot_number )
 {
     mock_destroy_data.called++;
     mock_destroy_data.slot_number = slot_number;
+
     return( PSA_SUCCESS );
 }
 
@@ -253,15 +297,50 @@ void mock_export( )
     PSA_ASSERT( psa_export_key( handle,
                                 exported, sizeof( exported ),
                                 &exported_length ) );
-    
+
     TEST_ASSERT( mock_export_data.called == 1 );
-    
+
     PSA_ASSERT( psa_destroy_key( handle ) );
 
     TEST_ASSERT( mock_destroy_data.called == 1 );
 
 exit:
     PSA_DONE( );
-    mock_teardown( ); 
+    mock_teardown( );
+}
+/* END_CASE */
+
+/* BEGIN_CASE */
+void mock_generate( )
+{
+    psa_drv_se_t driver;
+    psa_drv_se_key_management_t key_management;
+    psa_key_lifetime_t lifetime = 2;
+    psa_key_id_t id = 1;
+    psa_key_handle_t handle = 0;
+    psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
+
+    memset( &driver, 0, sizeof( driver ) );
+    memset( &key_management, 0, sizeof( key_management ) );
+    driver.hal_version = PSA_DRV_SE_HAL_VERSION;
+    driver.key_management = &key_management;
+    key_management.slot_count = 1;
+    key_management.p_generate = mock_generate;
+    key_management.p_destroy = mock_destroy;
+    key_management.p_allocate = mock_allocate;
+
+    PSA_ASSERT( psa_register_se_driver( lifetime, &driver ) );
+    PSA_ASSERT( psa_crypto_init( ) );
+
+    psa_set_key_id( &attributes, id );
+    psa_set_key_lifetime( &attributes, lifetime );
+    psa_set_key_usage_flags( &attributes, PSA_KEY_USAGE_EXPORT );
+    psa_set_key_type( &attributes, PSA_KEY_TYPE_RAW_DATA );
+    PSA_ASSERT( psa_generate_key( &attributes, &handle ) );
+    TEST_ASSERT( mock_generate_data.called == 1 );
+
+exit:
+    PSA_DONE( );
+    mock_teardown( );
 }
 /* END_CASE */