commit 4fc744f8af1fef75f8d005584275dc3a6e209090
author mohammad1603 <mohammad.abomokh@arm.com> Sun Jun 03 15:10:47 2018 +0300
committer itayzafrir <itay.zafrir@arm.com> Wed Sep 05 12:41:51 2018 +0300
tree 4eb61692001f8db175ebabc7c6e793c0146c337a
parent 0f21465175662498cf3f84a92a08f8fd65d36fc7

change the check of block size for all supported algorithms

library/psa_crypto.c

diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 317417d..905b9a8 100755
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1509,8 +1509,8 @@
 
     //TODO: check key policy
 
-    if ( !( ( key_type & PSA_KEY_TYPE_CATEGORY_MASK ) == PSA_KEY_TYPE_CATEGORY_CIPHER
-            && PSA_BLOCK_CIPHER_BLOCK_SIZE( key_type ) == 16 ) )
+    if ( !( ( key_type & PSA_KEY_TYPE_CATEGORY_MASK ) == PSA_KEY_TYPE_CATEGORY_SYMMETRIC
+            && PSA_BLOCK_CIPHER_BLOCK_SIZE( key_type ) == cipher_info->block_size ) )
         return( PSA_ERROR_INVALID_ARGUMENT );
 
     if( alg == PSA_ALG_GCM )
@@ -1633,8 +1633,8 @@
             return( PSA_ERROR_NOT_SUPPORTED );
     //TODO: check key policy
 
-    if ( !( ( key_type & PSA_KEY_TYPE_CATEGORY_MASK ) == PSA_KEY_TYPE_CATEGORY_CIPHER
-            && PSA_BLOCK_CIPHER_BLOCK_SIZE( key_type ) == 16 ) )
+    if ( !( ( key_type & PSA_KEY_TYPE_CATEGORY_MASK ) == PSA_KEY_TYPE_CATEGORY_SYMMETRIC
+            && PSA_BLOCK_CIPHER_BLOCK_SIZE( key_type ) == cipher_info->block_size ) )
         return( PSA_ERROR_INVALID_ARGUMENT );
 
     if( alg == PSA_ALG_GCM )