commit 9653d80518f6ddb228badd7a892e7f1da787b2f7
author Hanno Becker <hanno.becker@arm.com> Wed Aug 21 13:49:13 2019 +0100
committer Hanno Becker <hanno.becker@arm.com> Wed Sep 04 16:17:25 2019 +0100
tree 5102cf16d87f0ddf037adad3af9444dbcfc0dc1a
parent 4102972d313e1d617a06f5f526b3f1e2018a7ee7

TinyCrypt PK wrap: Implement check_pair PK wrapper

library/pk_wrap.c

diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 8383604..78cee74 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -535,6 +535,23 @@
     return( (size_t) ( NUM_ECC_BYTES * 8 ) );
 }
 
+static int uecc_eckey_check_pair( const void *pub, const void *prv )
+{
+    const mbedtls_uecc_keypair *uecc_pub =
+        (const mbedtls_uecc_keypair *) pub;
+    const mbedtls_uecc_keypair *uecc_prv =
+        (const mbedtls_uecc_keypair *) prv;
+
+    if( memcmp( uecc_pub->public_key,
+                uecc_prv->public_key,
+                2 * NUM_ECC_BYTES ) == 0 )
+    {
+        return( 0 );
+    }
+
+    return( MBEDTLS_ERR_ECP_BAD_INPUT_DATA );
+}
+
 static int uecc_eckey_can_do( mbedtls_pk_type_t type )
 {
     return( type == MBEDTLS_PK_ECDSA ||
@@ -706,7 +723,7 @@
     uecc_eckey_sign_wrap,
     NULL,
     NULL,
-    NULL,
+    uecc_eckey_check_pair,
     uecc_eckey_alloc_wrap,
     uecc_eckey_free_wrap,
     NULL,