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.c

diff --git a/library/pk.c b/library/pk.c
index 4213001..bb55a18 100644
--- a/library/pk.c
+++ b/library/pk.c
@@ -329,12 +329,7 @@
         return( MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE );
     }
 
-    if( prv->pk_info->type == MBEDTLS_PK_RSA_ALT )
-    {
-        if( pub->pk_info->type != MBEDTLS_PK_RSA )
-            return( MBEDTLS_ERR_PK_TYPE_MISMATCH );
-    }
-    else if( prv->pk_info->type != MBEDTLS_PK_OPAQUE )
+    if( prv->pk_info->type != MBEDTLS_PK_OPAQUE )
     {
         if( pub->pk_info != prv->pk_info )
             return( MBEDTLS_ERR_PK_TYPE_MISMATCH );

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 );