commit 7f4b10b5bd48a7cda4d69d5f16d571b31ba6be1c
author Unknown <andrzej.kurek@mobica.com> Tue Feb 06 03:28:23 2018 -0500
committer Unknown <andrzej.kurek@mobica.com> Tue Feb 06 03:28:23 2018 -0500
tree b17aaece738bb6c122b62f34c0354154ebbf2633
parent 3b450a1798192271df89c511b6838c2610139d2e

Move RSA_ALT pk_info type check to the respective check_pair function

library/pk_wrap.c

diff --git a/library/pk_wrap.c b/library/pk_wrap.c
index 17df304..6098ac1 100644
--- a/library/pk_wrap.c
+++ b/library/pk_wrap.c
@@ -486,6 +486,13 @@
     unsigned char hash[32];
     size_t sig_len = 0;
     int ret;
+    const mbedtls_pk_context* prv_context = prv;
+
+    if( prv_context->pk_info->type == MBEDTLS_PK_RSA_ALT )
+    {
+        if( pub->pk_info->type != MBEDTLS_PK_RSA )
+            return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
+    }
 
     if( rsa_alt_get_bitlen( prv ) != rsa_get_bitlen( pub->pk_ctx ) )
         return( MBEDTLS_ERR_RSA_KEY_CHECK_FAILED );