Use the mbedtls_ecjpake_write_shared_key() to input raw shared key material as derivation secret

Signed-off-by: Neil Armstrong <narmstrong@baylibre.com>
diff --git a/library/psa_crypto_pake.c b/library/psa_crypto_pake.c
index 556acd9..dd1a91e 100644
--- a/library/psa_crypto_pake.c
+++ b/library/psa_crypto_pake.c
@@ -690,12 +690,12 @@
 #if defined(MBEDTLS_PSA_BUILTIN_ALG_JPAKE)
     if( operation->alg == PSA_ALG_JPAKE )
     {
-        ret = mbedtls_ecjpake_derive_secret( &operation->ctx.ecjpake,
-                                             operation->buffer,
-                                             PSA_PAKE_BUFFER_SIZE,
-                                             &operation->buffer_length,
-                                             mbedtls_psa_get_random,
-                                             MBEDTLS_PSA_RANDOM_STATE );
+        ret = mbedtls_ecjpake_write_shared_key( &operation->ctx.ecjpake,
+                                                operation->buffer,
+                                                PSA_PAKE_BUFFER_SIZE,
+                                                &operation->buffer_length,
+                                                mbedtls_psa_get_random,
+                                                MBEDTLS_PSA_RANDOM_STATE );
         if( ret != 0)
         {
             psa_pake_abort( operation );