commit 7607cd6e576b3c4ea0430fb0989b56975e56fd56
author Gilles Peskine <Gilles.Peskine@arm.com> Wed May 29 17:35:00 2019 +0200
committer Janos Follath <janos.follath@arm.com> Thu Jul 04 14:22:27 2019 +0100
tree a9a834d0856a3dff5ad952981ee902268275f65f
parent d6dce9f4f310efc7f60f23d1e7987fe4f7501f4b

Convert exercise_key_derivation_key to the new KDF API

tests/suites/test_suite_psa_crypto.function

diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index 48f5337..de88bfb 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -530,9 +530,9 @@
 
     if( usage & PSA_KEY_USAGE_DERIVE )
     {
+        PSA_ASSERT( psa_key_derivation_setup( &operation, alg ) );
         if( PSA_ALG_IS_HKDF( alg ) )
         {
-            PSA_ASSERT( psa_key_derivation_setup( &operation, alg ) );
             PSA_ASSERT( psa_key_derivation_input_bytes( &operation,
                                                         PSA_KEY_DERIVATION_INPUT_SALT,
                                                         label,
@@ -545,17 +545,26 @@
                                                         seed,
                                                         seed_length ) );
         }
-#if defined(PSA_PRE_1_0_KEY_DERIVATION)
+        else if( PSA_ALG_IS_TLS12_PRF( alg ) ||
+                 PSA_ALG_IS_TLS12_PSK_TO_MS( alg ) )
+        {
+            PSA_ASSERT( psa_key_derivation_input_bytes( &operation,
+                                                        PSA_KEY_DERIVATION_INPUT_SEED,
+                                                        seed,
+                                                        seed_length ) );
+            PSA_ASSERT( psa_key_derivation_input_key( &operation,
+                                                      PSA_KEY_DERIVATION_INPUT_SECRET,
+                                                      handle ) );
+            PSA_ASSERT( psa_key_derivation_input_bytes( &operation,
+                                                        PSA_KEY_DERIVATION_INPUT_LABEL,
+                                                        label,
+                                                        label_length ) );
+        }
         else
         {
-            // legacy
-            PSA_ASSERT( psa_key_derivation( &operation,
-                                            handle, alg,
-                                            label, label_length,
-                                            seed, seed_length,
-                                            sizeof( output ) ) );
+            TEST_ASSERT( ! "Key derivation algorithm not supported" );
         }
-#endif
+
         PSA_ASSERT( psa_key_derivation_output_bytes( &operation,
                                                      output,
                                                      sizeof( output ) ) );