Security hardening, input validation, memory-safety, and key zeroization fixes

src/wp_aes_aead.c

@@ -1789,6 +1789,7 @@ static int wp_aesccm_init(wp_AeadCtx* ctx, const unsigned char *key,
     }
     if (ok) {
         ctx->enc = enc;
+        ctx->tlsAadLen = UNINITIALISED_SIZET;
         ok = wp_aead_set_ctx_params(ctx, params);
     }
 
@@ -1914,6 +1915,7 @@ static int wp_aesccm_tls_cipher(wp_AeadCtx* ctx, unsigned char* out,
     }
 
     *outLen = olen;
+    ctx->tlsAadLen = UNINITIALISED_SIZET;
     WOLFPROV_LEAVE(WP_LOG_COMP_AES, __FILE__ ":" WOLFPROV_STRINGIZE(__LINE__), ok);
     return ok;
 }

src/wp_aes_block.c

@@ -589,16 +589,20 @@ static int wp_aes_block_tls_dec_record(wp_AesBlockCtx *ctx,
             gd = d - 1;
         }
         ivRecLen -= gd & ((size_t)padV + 1);
-        *outLen = ivRecLen;
         /* No MAC to extract */
         ctx->tlsmac = NULL;
         ctx->tlsmacAlloced = 0;
         /* With ETM/no-MAC, bad padding is a real error (the MAC was
          * already verified by the record layer, so there is no padding
          * oracle concern).  Matches OpenSSL ssl3_cbc_copy_mac returning
          * 0 when good==0 and mac_size==0. */
-        if (gd == 0)
+        if (gd == 0) {
             ok = 0;
+            *outLen = 0;
+        }
+        else {
+            *outLen = ivRecLen;
+        }
     }
     else if (ok) {
         /* 64-byte aligned buffer for cache-line-aware MAC rotation */
@@ -743,6 +747,7 @@ static int wp_aes_block_update(wp_AesBlockCtx *ctx, unsigned char *out,
     int ok = 1;
     size_t oLen = 0;
     size_t nextBlocks;
+    unsigned char *outStart = out;
 
     WOLFPROV_ENTER(WP_LOG_COMP_AES, "wp_aes_block_update");
 
@@ -808,7 +813,7 @@ static int wp_aes_block_update(wp_AesBlockCtx *ctx, unsigned char *out,
         *outLen = oLen;
     }
     if (ok && (ctx->tls_version > 0) && (!ctx->enc)) {
-        ok = wp_aes_block_tls_dec_record(ctx, out, oLen, outLen);
+        ok = wp_aes_block_tls_dec_record(ctx, outStart, oLen, outLen);
     }
 
     WOLFPROV_LEAVE(WP_LOG_COMP_AES, __FILE__ ":" WOLFPROV_STRINGIZE(__LINE__), ok);

src/wp_aes_wrap.c

@@ -349,6 +349,11 @@ static int wp_aes_wrap_update(wp_AesWrapCtx *ctx, unsigned char *out,
         ok = 0;
     }
 
+    if (ok && (((uint64_t)inLen > (uint64_t)UINT32_MAX) ||
+            ((uint64_t)outSize > (uint64_t)UINT32_MAX))) {
+        ok = 0;
+    }
+
     if (ok && (inLen == 0)) {
         *outLen = 0;
     }

src/wp_dec_epki2pki.c

@@ -118,9 +118,15 @@ static int wp_DecryptPKCS8Key(byte* input, word32 sz, const char* password,
     EncryptedInfo info;
     int algoId;
 
-    pem = OPENSSL_malloc(sz * 2);
-    if (pem == NULL) {
-        ret = MEMORY_E;
+    /* sz * 2 must not overflow word32 before allocating and encoding. */
+    if (sz > (0xFFFFFFFFU / 2)) {
+        ret = BAD_FUNC_ARG;
+    }
+    if (ret == 0) {
+        pem = OPENSSL_malloc((size_t)sz * 2);
+        if (pem == NULL) {
+            ret = MEMORY_E;
+        }
     }
     if (ret == 0) {
         ret = wc_DerToPem(input, sz, pem, sz * 2, PKCS8_ENC_PRIVATEKEY_TYPE);
@@ -232,6 +238,9 @@ static int wp_epki2pki_decode(wp_Epki2Pki* ctx, OSSL_CORE_BIO* coreBio,
             pwCbArg))) {
         done = 1;
     }
+    if ((!done) && ok && (passwordLen > sizeof(password))) {
+        ok = 0;
+    }
     if ((!done) && ok) {
     #if LIBWOLFSSL_VERSION_HEX >= 0x05000000
         rc = wc_DecryptPKCS8Key(data, len, password, (int)passwordLen);

src/wp_dec_pem2der.c

@@ -124,6 +124,9 @@ static int wp_pem_password_cb(char* passwd, int sz, int rw, void* userdata)
     if (!osslCb->cb(passwd, sz, &len, NULL, osslCb->cbArg)) {
         ret = -1;
     }
+    else if (len > (size_t)sz) {
+        ret = -1;
+    }
     else {
         ret = (int)len;
     }
@@ -157,13 +160,19 @@ static int wp_pem2der_convert(const char* data, word32 len, DerBuffer** pDer,
     WOLFPROV_ENTER(WP_LOG_COMP_PK, "wp_pem2der_convert");
 
     /* Skip '-----BEGIN <name>-----\n'. */
-    base64Data = data + 16 + nameLen + 1;
-    base64Len = len - (16 + nameLen + 1);
-    footer = XSTRSTR(base64Data, "-----END ");
-    if (footer == NULL) {
+    if (len <= (word32)(16 + nameLen + 1)) {
         info->consumed = len;
         ok = 0;
     }
+    if (ok) {
+        base64Data = data + 16 + nameLen + 1;
+        base64Len = len - (16 + nameLen + 1);
+        footer = XSTRSTR(base64Data, "-----END ");
+        if (footer == NULL) {
+            info->consumed = len;
+            ok = 0;
+        }
+    }
     if (ok) {
         /* Include footer and '\n'. */
         info->consumed = (long)(footer - data) + 14 + nameLen + 1;
@@ -230,7 +239,11 @@ static int wp_pem2der_decode_data(const unsigned char* data, word32 len,
 
     /* Identify the type of object by looking at the header. */
     /* TODO: support more PEM headers. */
-    if (XMEMCMP(data, "-----BEGIN CERTIFICATE-----", 27) == 0) {
+    /* Header detection reads fixed-length prefixes; reject short input. */
+    if (len < 41) {
+        ok = 0;
+    }
+    else if (XMEMCMP(data, "-----BEGIN CERTIFICATE-----", 27) == 0) {
         type = CERT_TYPE;
         obj = OSSL_OBJECT_CERT;
     }

src/wp_des.c

@@ -420,6 +420,7 @@ static int wp_des3_block_update(wp_Des3BlockCtx *ctx, unsigned char *out,
     int ok = 1;
     size_t oLen = 0;
     size_t nextBlocks;
+    unsigned char *outStart = out;
 
     WOLFPROV_ENTER(WP_LOG_COMP_DES, "wp_des3_block_update");
 
@@ -489,19 +490,28 @@ static int wp_des3_block_update(wp_Des3BlockCtx *ctx, unsigned char *out,
         }
         *outLen = oLen;
     }
+    if (ok && (ctx->tls_version > 0) && (!ctx->enc) &&
+            (oLen < (size_t)(2 * DES_BLOCK_SIZE))) {
+        ok = 0;
+    }
     if (ok && (ctx->tls_version > 0) && (!ctx->enc)) {
-        unsigned char pad = out[oLen-1];
+        unsigned char pad = outStart[oLen-1];
         int padStart = DES_BLOCK_SIZE - pad - 1;
         unsigned char invalid = (pad < DES_BLOCK_SIZE) - 1;
         int i;
 
         for (i = DES_BLOCK_SIZE - 1; i >= 0; i--) {
             byte check = wp_ct_int_mask_gte(i, padStart);
-            check &= wp_ct_byte_mask_ne(out[oLen - DES_BLOCK_SIZE + i], pad);
+            check &= wp_ct_byte_mask_ne(outStart[oLen - DES_BLOCK_SIZE + i], pad);
             invalid |= check;
         }
-        *outLen = oLen - pad - 1 - DES_BLOCK_SIZE;
         ok = invalid == 0;
+        if (ok) {
+            *outLen = oLen - pad - 1 - DES_BLOCK_SIZE;
+        }
+        else {
+            *outLen = 0;
+        }
     }
 
     WOLFPROV_LEAVE(WP_LOG_COMP_DES, __FILE__ ":" WOLFPROV_STRINGIZE(__LINE__), ok);

src/wp_dh_kmgmt.c

@@ -1026,7 +1026,34 @@ static int wp_dh_validate(const wp_Dh* dh, int selection, int checkType)
 
     if (((selection & OSSL_KEYMGMT_SELECT_DOMAIN_PARAMETERS) != 0) &&
         (dh->id == 0)) {
-        /* TODO: check explicit parameters. */
+        int isPrime = 0;
+        int pm1Init = 0;
+        mp_int* p = (mp_int*)&dh->key.p;
+        mp_int* g = (mp_int*)&dh->key.g;
+        mp_int pm1;
+
+        /* Explicit domain parameters: p must be prime and g in [2, p-1). */
+        if (mp_init(&pm1) != 0) {
+            ok = 0;
+        }
+        else {
+            pm1Init = 1;
+        }
+        if (ok) {
+            rc = mp_prime_is_prime(p, 8, &isPrime);
+            if ((rc != 0) || (!isPrime)) {
+                ok = 0;
+            }
+        }
+        if (ok && (mp_sub_d(p, 1, &pm1) != 0)) {
+            ok = 0;
+        }
+        if (ok && ((mp_cmp_d(g, 2) == MP_LT) || (mp_cmp(g, &pm1) != MP_LT))) {
+            ok = 0;
+        }
+        if (pm1Init) {
+            mp_clear(&pm1);
+        }
     }
     if (ok && ((selection & OSSL_KEYMGMT_SELECT_PUBLIC_KEY) != 0)) {
     #if LIBWOLFSSL_VERSION_HEX >= 0x05000000
@@ -1071,13 +1098,23 @@ static int wp_dh_import_group(wp_Dh* dh, const OSSL_PARAM params[])
     int ok = 1;
     const OSSL_PARAM* p;
     const char* name = NULL;
+    char nameBuf[WP_MAX_DH_GROUP_NAME_SZ];
 
     WOLFPROV_ENTER(WP_LOG_COMP_DH, "wp_dh_import_group");
 
     /* Look for the group name first. */
-    if (!wp_params_get_utf8_string_ptr(params, OSSL_PKEY_PARAM_GROUP_NAME,
-            &name)) {
-        ok = 0;
+    p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_GROUP_NAME);
+    if (p != NULL) {
+        /* p->data may not be NUL-terminated; copy into a bounded buffer. */
+        if ((p->data_type != OSSL_PARAM_UTF8_STRING) || (p->data == NULL) ||
+                (p->data_size >= sizeof(nameBuf))) {
+            ok = 0;
+        }
+        else {
+            XMEMCPY(nameBuf, p->data, p->data_size);
+            nameBuf[p->data_size] = '\0';
+            name = nameBuf;
+        }
     }
     /* When name was found, set the parameters based on the mapping. */
     if (ok && (name != NULL) && (!wp_dh_map_group_name(dh, name))) {

src/wp_ecc_kmgmt.c

@@ -1101,12 +1101,18 @@ static int wp_ecc_import_group(wp_Ecc* ecc, const OSSL_PARAM params[])
     p = OSSL_PARAM_locate_const(params, OSSL_PKEY_PARAM_GROUP_NAME);
     if (p != NULL) {
         const char* name = NULL;
+        char nameBuf[WP_MAX_EC_GROUP_NAME_SZ];
 
         if (p->data_type == OSSL_PARAM_UTF8_STRING) {
-            name = (const char*)p->data;
-            if (name == NULL) {
+            /* p->data may not be NUL-terminated; copy into a bounded buffer. */
+            if ((p->data == NULL) || (p->data_size >= sizeof(nameBuf))) {
                 ok = 0;
             }
+            else {
+                XMEMCPY(nameBuf, p->data, p->data_size);
+                nameBuf[p->data_size] = '\0';
+                name = nameBuf;
+            }
         }
         else if (p->data_type == OSSL_PARAM_UTF8_PTR) {
             if (!OSSL_PARAM_get_utf8_ptr(p, &name)) {
@@ -1150,6 +1156,40 @@ static int wp_ecc_import_keypair(wp_Ecc* ecc, const OSSL_PARAM params[],
             NULL))) {
         ok = 0;
     }
+    if (ok && priv) {
+        int idx = wc_ecc_get_curve_idx(ecc->curveId);
+        const ecc_set_type* dp = (idx >= 0) ? wc_ecc_get_curve_params(idx) :
+            NULL;
+        int orderInit = 0;
+        mp_int order;
+#if (!defined(HAVE_FIPS) || FIPS_VERSION_GE(5,3)) && LIBWOLFSSL_VERSION_HEX >= 0x05006002
+        mp_int* d = wc_ecc_key_get_priv(&ecc->key);
+#else
+        mp_int* d = &(ecc->key.k);
+#endif
+
+        /* Reject scalar >= order n (FIPS 186-4); fail closed if n missing. */
+        if (dp == NULL) {
+            ok = 0;
+        }
+        if (ok) {
+            if (mp_init(&order) != MP_OKAY) {
+                ok = 0;
+            }
+            else {
+                orderInit = 1;
+            }
+        }
+        if (ok && (mp_read_radix(&order, dp->order, 16) != MP_OKAY)) {
+            ok = 0;
+        }
+        if (ok && (mp_cmp(d, &order) != MP_LT)) {
+            ok = 0;
+        }
+        if (orderInit) {
+            mp_clear(&order);
+        }
+    }
     if (ok &&
 #if (!defined(HAVE_FIPS) || FIPS_VERSION_GE(5,3)) && LIBWOLFSSL_VERSION_HEX >= 0x05006002
             (!mp_iszero(wc_ecc_key_get_priv(&ecc->key)))

src/wp_ecdsa_sig.c

@@ -487,42 +487,33 @@ static int wp_ecdsa_setup_md(wp_EcdsaSigCtx *ctx, const char *mdName,
 
     if (mdName != NULL) {
         int rc;
+        enum wc_HashType hashType;
 
-#if LIBWOLFSSL_VERSION_HEX >= 0x05007004
-        ctx->hash.type = wp_name_to_wc_hash_type(ctx->libCtx, mdName, mdProps);
-        if ((ctx->hash.type == WC_HASH_TYPE_NONE) ||
-            (ctx->hash.type == WC_HASH_TYPE_MD5))
-#else
-        ctx->hashType = wp_name_to_wc_hash_type(ctx->libCtx, mdName, mdProps);
-        if ((ctx->hashType == WC_HASH_TYPE_NONE) ||
-            (ctx->hashType == WC_HASH_TYPE_MD5))
-#endif
-        {
+        hashType = wp_name_to_wc_hash_type(ctx->libCtx, mdName, mdProps);
+        if ((hashType == WC_HASH_TYPE_NONE) ||
+            (hashType == WC_HASH_TYPE_MD5)) {
             ok = 0;
         }
 #ifdef HAVE_FIPS
-#if LIBWOLFSSL_VERSION_HEX >= 0x05007004
-        if ((ctx->hash.type == WC_HASH_TYPE_SHA) && (op == EVP_PKEY_OP_SIGN))
-#else
-        if ((ctx->hashType == WC_HASH_TYPE_SHA) && (op == EVP_PKEY_OP_SIGN))
-#endif
-        {
+        if (ok && (hashType == WC_HASH_TYPE_SHA) &&
+            (op == EVP_PKEY_OP_SIGN)) {
             ok = 0;
         }
 #endif
 
         if (ok) {
-#if LIBWOLFSSL_VERSION_HEX >= 0x05007004
-            rc = wc_HashInit_ex(&ctx->hash, ctx->hash.type, NULL, INVALID_DEVID);
-#else
-            rc = wc_HashInit_ex(&ctx->hash, ctx->hashType, NULL, INVALID_DEVID);
-#endif
+            rc = wc_HashInit_ex(&ctx->hash, hashType, NULL, INVALID_DEVID);
             if (rc != 0) {
                 WOLFPROV_MSG_DEBUG_RETCODE(WP_LOG_LEVEL_DEBUG, "wc_HashInit_ex", rc);
                 ok = 0;
             }
         }
         if (ok) {
+#if LIBWOLFSSL_VERSION_HEX >= 0x05007004
+            ctx->hash.type = hashType;
+#else
+            ctx->hashType = hashType;
+#endif
             OPENSSL_strlcpy(ctx->mdName, mdName, sizeof(ctx->mdName));
         }
     }

src/wp_ecx_kmgmt.c

@@ -975,6 +975,9 @@ static int wp_ecx_import(wp_Ecx* ecx, int selection, const OSSL_PARAM params[])
             &privData, &len)) {
             ok = 0;
         }
+        if (ok && (privData != NULL) && (len == 0)) {
+            ok = 0;
+        }
         if (ok && (privData != NULL)) {
             ecx->unclamped[0] = privData[0];
             ecx->unclamped[1] = privData[len - 1];