Skip to content

Commit

Permalink
Add support for native ECDSA signature
Browse files Browse the repository at this point in the history
  • Loading branch information
@KostasTsiounis
KostasTsiounis committed May 8, 2024
1 parent 915b28b commit 74d2cab
Show file tree
Hide file tree
Showing 4 changed files with 1,156 additions and 93 deletions.
13 changes: 13 additions & 0 deletions closed/src/java.base/share/classes/jdk/crypto/jniprovider/NativeCrypto.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -416,6 +416,19 @@ public final native int PBEDerive(byte[] password,
int id,
int hashAlgorithm);

/* Native ECDSA interfaces. */
public final native int ECDSASign(long key,
byte[] digest,
int digestLen,
byte[] signature,
int sigLen);

public final native int ECDSAVerify(long key,
byte[] digest,
int digestLen,
byte[] signature,
int sigLen);

/* Native XDH (X25519, X448) interfaces. */
public final native int XDHCreateKeys(byte[] privateKey,
int privateKeyLength,
Expand Down
175 changes: 173 additions & 2 deletions closed/src/java.base/share/native/libjncrypto/NativeCrypto.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,14 @@ typedef int OSSL_EC_KEY_check_key_t(const EC_KEY *);
typedef int EC_set_public_key_t(EC_KEY *, BIGNUM *, BIGNUM *, int);
typedef const BIGNUM *OSSL_EC_KEY_get0_private_key_t(const EC_KEY *);

typedef ECDSA_SIG *OSSL_ECDSA_do_sign_t(const unsigned char *, int, EC_KEY *);
typedef int OSSL_ECDSA_do_verify_t(const unsigned char *, int, const ECDSA_SIG *, EC_KEY *);
typedef ECDSA_SIG *OSSL_ECDSA_SIG_new_t(void);
typedef void OSSL_ECDSA_SIG_free_t(ECDSA_SIG *);
typedef const BIGNUM *OSSL_ECDSA_SIG_get0_r_t(const ECDSA_SIG *);
typedef const BIGNUM *OSSL_ECDSA_SIG_get0_s_t(const ECDSA_SIG *);
typedef int OSSL_ECDSA_SIG_set0_t(ECDSA_SIG *, BIGNUM *, BIGNUM *);

typedef EVP_PKEY_CTX *OSSL_EVP_PKEY_CTX_new_t(EVP_PKEY *, ENGINE *);
typedef EVP_PKEY_CTX *OSSL_EVP_PKEY_CTX_new_id_t(int, ENGINE *);
typedef int OSSL_EVP_PKEY_keygen_init_t(EVP_PKEY_CTX *);
Expand Down Expand Up @@ -282,6 +290,15 @@ OSSL_EC_KEY_check_key_t* OSSL_EC_KEY_check_key;
EC_set_public_key_t* EC_set_public_key;
OSSL_EC_KEY_get0_private_key_t *OSSL_EC_KEY_get0_private_key;

/* Define pointers for OpenSSL functions to handle ECDSA algorithm. */
OSSL_ECDSA_do_sign_t *OSSL_ECDSA_do_sign;
OSSL_ECDSA_do_verify_t *OSSL_ECDSA_do_verify;
OSSL_ECDSA_SIG_new_t *OSSL_ECDSA_SIG_new;
OSSL_ECDSA_SIG_free_t *OSSL_ECDSA_SIG_free;
OSSL_ECDSA_SIG_get0_r_t *OSSL_ECDSA_SIG_get0_r;
OSSL_ECDSA_SIG_get0_s_t *OSSL_ECDSA_SIG_get0_s;
OSSL_ECDSA_SIG_set0_t *OSSL_ECDSA_SIG_set0;

/* Define pointers for OpenSSL functions to handle XDH algorithm. */
OSSL_EVP_PKEY_CTX_new_t *OSSL_EVP_PKEY_CTX_new;
OSSL_EVP_PKEY_CTX_new_id_t *OSSL_EVP_PKEY_CTX_new_id;
Expand Down Expand Up @@ -566,7 +583,7 @@ JNIEXPORT jlong JNICALL Java_jdk_crypto_jniprovider_NativeCrypto_loadCrypto
OSSL_ECGF2M = JNI_TRUE;
}

/* Load the functions symbols for OpenSSL XDH algorithm. (Need OpenSSL 1.1.x or above). */
/* Load the functions symbols for OpenSSL XDH and ECDSA algorithms. (Need OpenSSL 1.1.x or above). */
if (ossl_ver >= OPENSSL_VERSION_1_1_1) {
OSSL_EVP_PKEY_CTX_new = (OSSL_EVP_PKEY_CTX_new_t *)find_crypto_symbol(crypto_library, "EVP_PKEY_CTX_new");
OSSL_EVP_PKEY_CTX_new_id = (OSSL_EVP_PKEY_CTX_new_id_t *)find_crypto_symbol(crypto_library, "EVP_PKEY_CTX_new_id");
Expand All @@ -581,6 +598,14 @@ JNIEXPORT jlong JNICALL Java_jdk_crypto_jniprovider_NativeCrypto_loadCrypto
OSSL_EVP_PKEY_derive_set_peer = (OSSL_EVP_PKEY_derive_set_peer_t *)find_crypto_symbol(crypto_library, "EVP_PKEY_derive_set_peer");
OSSL_EVP_PKEY_derive = (OSSL_EVP_PKEY_derive_t *)find_crypto_symbol(crypto_library, "EVP_PKEY_derive");
OSSL_EVP_PKEY_free = (OSSL_EVP_PKEY_free_t *)find_crypto_symbol(crypto_library, "EVP_PKEY_free");

OSSL_ECDSA_do_sign = (OSSL_ECDSA_do_sign_t *)find_crypto_symbol(crypto_library, "ECDSA_do_sign");
OSSL_ECDSA_do_verify = (OSSL_ECDSA_do_verify_t *)find_crypto_symbol(crypto_library, "ECDSA_do_verify");
OSSL_ECDSA_SIG_new = (OSSL_ECDSA_SIG_new_t *)find_crypto_symbol(crypto_library, "ECDSA_SIG_new");
OSSL_ECDSA_SIG_free = (OSSL_ECDSA_SIG_free_t *)find_crypto_symbol(crypto_library, "ECDSA_SIG_free");
OSSL_ECDSA_SIG_get0_r = (OSSL_ECDSA_SIG_get0_r_t *)find_crypto_symbol(crypto_library, "ECDSA_SIG_get0_r");
OSSL_ECDSA_SIG_get0_s = (OSSL_ECDSA_SIG_get0_s_t *)find_crypto_symbol(crypto_library, "ECDSA_SIG_get0_s");
OSSL_ECDSA_SIG_set0 = (OSSL_ECDSA_SIG_set0_t *)find_crypto_symbol(crypto_library, "ECDSA_SIG_set0");
} else {
OSSL_EVP_PKEY_CTX_new = NULL;
OSSL_EVP_PKEY_CTX_new_id = NULL;
Expand All @@ -595,6 +620,14 @@ JNIEXPORT jlong JNICALL Java_jdk_crypto_jniprovider_NativeCrypto_loadCrypto
OSSL_EVP_PKEY_derive_set_peer = NULL;
OSSL_EVP_PKEY_derive = NULL;
OSSL_EVP_PKEY_free = NULL;

OSSL_ECDSA_do_sign = NULL;
OSSL_ECDSA_do_verify = NULL;
OSSL_ECDSA_SIG_new = NULL;
OSSL_ECDSA_SIG_free = NULL;
OSSL_ECDSA_SIG_get0_r = NULL;
OSSL_ECDSA_SIG_get0_s = NULL;
OSSL_ECDSA_SIG_set0 = NULL;
}

/* Load the functions symbols for OpenSSL PBE algorithm. */
Expand Down Expand Up @@ -679,7 +712,14 @@ JNIEXPORT jlong JNICALL Java_jdk_crypto_jniprovider_NativeCrypto_loadCrypto
(NULL == OSSL_EVP_PKEY_derive_init) ||
(NULL == OSSL_EVP_PKEY_derive_set_peer) ||
(NULL == OSSL_EVP_PKEY_derive) ||
(NULL == OSSL_EVP_PKEY_free))) ||
(NULL == OSSL_EVP_PKEY_free) ||
(NULL == OSSL_ECDSA_do_sign) ||
(NULL == OSSL_ECDSA_do_verify) ||
(NULL == OSSL_ECDSA_SIG_new) ||
(NULL == OSSL_ECDSA_SIG_free) ||
(NULL == OSSL_ECDSA_SIG_get0_r) ||
(NULL == OSSL_ECDSA_SIG_get0_s) ||
(NULL == OSSL_ECDSA_SIG_set0))) ||
/* Check symbols that are only available in OpenSSL 1.1.x and above */
((ossl_ver >= OPENSSL_VERSION_1_1_0) && ((NULL == OSSL_chacha20) || (NULL == OSSL_chacha20_poly1305))) ||
/* Check symbols that are only available in OpenSSL 1.0.x and above */
Expand Down Expand Up @@ -3101,6 +3141,137 @@ Java_jdk_crypto_jniprovider_NativeCrypto_PBEDerive
return ret;
}

/* Create an ECDSA Signature
*
* Class: jdk_crypto_jniprovider_NativeCrypto
* Method: ECDSASign
* Signature: (J[BI[B)I
*/
JNIEXPORT jint JNICALL
Java_jdk_crypto_jniprovider_NativeCrypto_ECDSASign
(JNIEnv *env, jclass obj, jlong key, jbyteArray digest, jint digestLen, jbyteArray sig, jint sigLen)
{
jint ret = -1;

unsigned char *nativeDigest = NULL;
unsigned char *nativeSig = NULL;
EC_KEY *privateKey = (EC_KEY *)(intptr_t)key;
ECDSA_SIG *signature = NULL;
const BIGNUM *rBN = NULL;
const BIGNUM *sBN = NULL;

nativeDigest = (unsigned char *)((*env)->GetPrimitiveArrayCritical(env, digest, 0));
if (NULL == nativeDigest) {
goto cleanup;
}

signature = (*OSSL_ECDSA_do_sign)(nativeDigest, digestLen, privateKey);
if (NULL == signature) {
printf("Failed to create an ECDSA Signature.\n");
goto cleanup;
}

rBN = (*OSSL_ECDSA_SIG_get0_r)(signature);
sBN = (*OSSL_ECDSA_SIG_get0_s)(signature);

nativeSig = (unsigned char *)((*env)->GetPrimitiveArrayCritical(env, sig, 0));
if (NULL == nativeSig) {
goto cleanup;
}

ret = getArrayFromBN(rBN, nativeSig, sigLen / 2);
if (-1 == ret) {
goto cleanup;
}

ret = getArrayFromBN(sBN, &nativeSig[sigLen / 2], sigLen / 2);
if (-1 == ret) {
goto cleanup;
}

ret = sigLen;

cleanup:
if (NULL != nativeSig) {
(*env)->ReleasePrimitiveArrayCritical(env, sig, nativeSig, 0);
}

if (NULL != signature) {
(*OSSL_ECDSA_SIG_free)(signature);
}

if (NULL != nativeDigest) {
(*env)->ReleasePrimitiveArrayCritical(env, digest, nativeDigest, JNI_ABORT);
}

return ret;
}

/* Verify an ECDSA Signature
*
* Class: jdk_crypto_jniprovider_NativeCrypto
* Method: ECDSAVerify
* Signature: (J[BI[B)I
*/
JNIEXPORT jint JNICALL
Java_jdk_crypto_jniprovider_NativeCrypto_ECDSAVerify
(JNIEnv *env, jclass obj, jlong key, jbyteArray digest, jint digestLen, jbyteArray sig, jint sigLen)
{
jint ret = -1;

unsigned char *nativeDigest = NULL;
unsigned char *nativeSig = NULL;
EC_KEY *publicKey = (EC_KEY *)(intptr_t)key;
ECDSA_SIG *signature = NULL;
BIGNUM *rBN = NULL;
BIGNUM *sBN = NULL;

nativeSig = (unsigned char *)((*env)->GetPrimitiveArrayCritical(env, sig, 0));
if (NULL == nativeSig) {
goto cleanup;
}

rBN = (*OSSL_BN_bin2bn)(nativeSig, sigLen / 2, NULL);
sBN = (*OSSL_BN_bin2bn)(&nativeSig[sigLen / 2], sigLen / 2, NULL);
signature = (*OSSL_ECDSA_SIG_new)();
if (0 == (*OSSL_ECDSA_SIG_set0)(signature, rBN, sBN)) {
goto cleanup;
}

nativeDigest = (unsigned char *)((*env)->GetPrimitiveArrayCritical(env, digest, 0));
if (NULL == nativeDigest) {
goto cleanup;
}

ret = (*OSSL_ECDSA_do_verify)(nativeDigest, digestLen, signature, publicKey);

cleanup:
if (NULL != nativeDigest) {
(*env)->ReleasePrimitiveArrayCritical(env, digest, nativeDigest, JNI_ABORT);
}

if (NULL != signature) {
// The BIGNUM structs will be freed by the signature.
sBN = NULL;
rBN = NULL;
(*OSSL_ECDSA_SIG_free)(signature);
}

// In case the BIGNUM structs weren't freed by the signature.
if (NULL != sBN) {
(*OSSL_BN_free)(sBN);
}
if (NULL != rBN) {
(*OSSL_BN_free)(rBN);
}

if (NULL != nativeSig) {
(*env)->ReleasePrimitiveArrayCritical(env, sig, nativeSig, JNI_ABORT);
}

return ret;
}

/* Create a pair of private and public keys for XDH Key Agreement.
*
* Class: jdk_crypto_jniprovider_NativeCrypto
Expand Down
Loading

0 comments on commit 74d2cab

Please sign in to comment.