Update native method

corretto · Aug 1, 2024 · 5c55cc8 · 5c55cc8
1 parent d510787
commit 5c55cc8
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/csrc/java_evp_keys.cpp b/csrc/java_evp_keys.cpp
@@ -350,12 +350,15 @@ JNIEXPORT jbyteArray JNICALL Java_com_amazon_corretto_crypto_provider_EvpEdEcPri
         CHECK_OPENSSL(pkcs8.isInitialized());
 
         // This next line allocates memory
-        size_t bufLen = i2d_PKCS8_PRIV_KEY_INFO(pkcs8, &privateKeyBuffer);
+        int bufLen = i2d_PKCS8_PRIV_KEY_INFO(pkcs8, &privateKeyBuffer);
         CHECK_OPENSSL(bufLen > 0);
 
-        CHECK_OPENSSL(EVP_PKEY_get_raw_private_key(key, privateKeyBuffer, &bufLen) > 0);
-        result = env->NewByteArray(bufLen);
-        env->SetByteArrayRegion(result, 0, bufLen, privateKeyBuffer);
+        size_t bufLen2 = (size_t) bufLen;
+
+        CHECK_OPENSSL(EVP_PKEY_get_raw_private_key(key, privateKeyBuffer, &bufLen2) > 0);
+
+        result = env->NewByteArray(bufLen2);
+        env->SetByteArrayRegion(result, 0, bufLen2, privateKeyBuffer);
     } catch (java_ex& ex) {
         ex.throw_to_java(pEnv);
     }