Add length checks in client keywrap request functions

Author: padelsbach

src/wh_client_keywrap.c

@@ -26,6 +26,10 @@ int wh_Client_KeyWrapRequest(whClientContext*   ctx,
         return WH_ERROR_BADARGS;
     }
 
+    if (keySz == 0 || keySz > WOLFHSM_CFG_KEYWRAP_MAX_KEY_SIZE) {
+        return WH_ERROR_BADARGS;
+    }
+
     /* Set the request pointer to the shared comm data memory region */
     req =
         (whMessageKeystore_KeyWrapRequest*)wh_CommClient_GetDataPtr(ctx->comm);
@@ -139,6 +143,10 @@ int wh_Client_KeyUnwrapAndExportRequest(whClientContext*   ctx,
         return WH_ERROR_BADARGS;
     }
 
+    if (wrappedKeySz == 0 || wrappedKeySz > WOLFHSM_CFG_KEYWRAP_MAX_KEY_SIZE) {
+        return WH_ERROR_BADARGS;
+    }
+
     /* Set the request pointer to the shared comm data memory region */
     req =
         (whMessageKeystore_KeyUnwrapAndExportRequest*)wh_CommClient_GetDataPtr(
@@ -256,6 +264,10 @@ int wh_Client_KeyUnwrapAndCacheRequest(whClientContext*   ctx,
     if (ctx == NULL || wrappedKeyIn == NULL)
         return WH_ERROR_BADARGS;
 
+    if (wrappedKeySz == 0 || wrappedKeySz > WOLFHSM_CFG_KEYWRAP_MAX_KEY_SIZE) {
+        return WH_ERROR_BADARGS;
+    }
+
     /* Set the request pointer to the shared comm data memory region */
     req = (whMessageKeystore_KeyUnwrapAndCacheRequest*)wh_CommClient_GetDataPtr(
         ctx->comm);
@@ -357,6 +369,10 @@ int wh_Client_DataWrapRequest(whClientContext*   ctx,
         return WH_ERROR_BADARGS;
     }
 
+    if (dataInSz == 0 || dataInSz > WOLFHSM_CFG_KEYWRAP_MAX_DATA_SIZE) {
+        return WH_ERROR_BADARGS;
+    }
+
     /* Set the request pointer to the shared comm data memory region */
     req =
         (whMessageKeystore_DataWrapRequest*)wh_CommClient_GetDataPtr(ctx->comm);
@@ -465,6 +481,11 @@ int wh_Client_DataUnwrapRequest(whClientContext*   ctx,
         return WH_ERROR_BADARGS;
     }
 
+    if (wrappedDataInSz == 0 ||
+        wrappedDataInSz > WOLFHSM_CFG_KEYWRAP_MAX_DATA_SIZE) {
+        return WH_ERROR_BADARGS;
+    }
+
     /* Set the request pointer to the shared comm data memory region */
     req = (whMessageKeystore_DataUnwrapRequest*)wh_CommClient_GetDataPtr(
         ctx->comm);