2 files changed, 4 insertions, 4 deletions
diff --git a/Cryptlib/Hmac/CryptHmacMd5.c b/Cryptlib/Hmac/CryptHmacMd5.c
index 0cdab7ac..693cd322 100644
--- a/Cryptlib/Hmac/CryptHmacMd5.c
+++ b/Cryptlib/Hmac/CryptHmacMd5.c
@@ -30,7 +30,7 @@ HmacMd5GetContextSize (
   //
   // Retrieves the OpenSSL HMAC-MD5 Context Size
   //
-  return (UINTN)(sizeof (HMAC_CTX));
+  return (UINTN) (sizeof (HMAC_CTX));
 }
 
 /**
@@ -58,7 +58,7 @@ HmacMd5Init (
   //
   // Check input parameters.
   //
-  if (HmacMd5Context == NULL) {
+  if (HmacMd5Context == NULL || KeySize > INT_MAX) {
     return FALSE;
   }
 
diff --git a/Cryptlib/Hmac/CryptHmacSha1.c b/Cryptlib/Hmac/CryptHmacSha1.c
index 58da2f3a..881d26cf 100644
--- a/Cryptlib/Hmac/CryptHmacSha1.c
+++ b/Cryptlib/Hmac/CryptHmacSha1.c
@@ -30,7 +30,7 @@ HmacSha1GetContextSize (
   //
   // Retrieves the OpenSSL HMAC-SHA1 Context Size
   //
-  return (UINTN)(sizeof (HMAC_CTX));
+  return (UINTN) (sizeof (HMAC_CTX));
 }
 
 /**
@@ -58,7 +58,7 @@ HmacSha1Init (
   //
   // Check input parameters.
   //
-  if (HmacSha1Context == NULL) {
+  if (HmacSha1Context == NULL || KeySize > INT_MAX) {
     return FALSE;
   }