openssl: Add support for ML-KEM with OpenSSL 3.5

This commit is contained in:
Tobias Brunner
2025-04-14 11:38:48 +02:00
parent f2e88b169f
commit b914333ab4
2 changed files with 103 additions and 27 deletions
+101 -26
View File
@@ -1,4 +1,5 @@
/* /*
* Copyright (C) 2025 Tobias Brunner
* Copyright (C) 2024 Gerardo Ravago * Copyright (C) 2024 Gerardo Ravago
* *
* Copyright (C) secunet Security Networks AG * Copyright (C) secunet Security Networks AG
@@ -19,9 +20,24 @@
#include <openssl/evp.h> #include <openssl/evp.h>
#if (OPENSSL_VERSION_NUMBER >= 0x30500000L && !defined(OPENSSL_NO_ML_KEM)) || \
defined(OPENSSL_IS_AWSLC)
#ifdef OPENSSL_IS_AWSLC #ifdef OPENSSL_IS_AWSLC
#include <openssl/experimental/kem_deterministic_api.h> #include <openssl/experimental/kem_deterministic_api.h>
#define NID_ML_KEM_512 NID_MLKEM512
#define NID_ML_KEM_768 NID_MLKEM768
#define NID_ML_KEM_1024 NID_MLKEM1024
#else
#include <openssl/core_names.h>
#endif
/**
* Length of seeds (d, z and m).
*/
#define ML_KEM_SEED_LEN 32
typedef struct private_key_exchange_t private_key_exchange_t; typedef struct private_key_exchange_t private_key_exchange_t;
/** /**
@@ -68,11 +84,11 @@ static int openssl_kem_get_nid(private_key_exchange_t *this)
switch (this->group) switch (this->group)
{ {
case ML_KEM_512: case ML_KEM_512:
return NID_MLKEM512; return NID_ML_KEM_512;
case ML_KEM_768: case ML_KEM_768:
return NID_MLKEM768; return NID_ML_KEM_768;
case ML_KEM_1024: case ML_KEM_1024:
return NID_MLKEM1024; return NID_ML_KEM_1024;
default: default:
return NID_undef; return NID_undef;
} }
@@ -89,10 +105,10 @@ METHOD(key_exchange_t, get_method, key_exchange_method_t,
*/ */
static bool openssl_kem_generate_pkey(private_key_exchange_t *this) static bool openssl_kem_generate_pkey(private_key_exchange_t *this)
{ {
EVP_PKEY_CTX *ctx = NULL; EVP_PKEY_CTX *ctx;
size_t seed_length = 0;
chunk_t seed = chunk_empty; chunk_t seed = chunk_empty;
#ifdef OPENSSL_IS_AWSLC
ctx = EVP_PKEY_CTX_new_id(EVP_PKEY_KEM, NULL); ctx = EVP_PKEY_CTX_new_id(EVP_PKEY_KEM, NULL);
if (!ctx) if (!ctx)
{ {
@@ -103,6 +119,13 @@ static bool openssl_kem_generate_pkey(private_key_exchange_t *this)
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
return FALSE; return FALSE;
} }
#else /* OPENSSL_IS_AWSLC */
ctx = EVP_PKEY_CTX_new_id(openssl_kem_get_nid(this), NULL);
if (!ctx)
{
return FALSE;
}
#endif /* OPENSSL_IS_AWSLC */
if (!EVP_PKEY_keygen_init(ctx)) if (!EVP_PKEY_keygen_init(ctx))
{ {
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
@@ -110,17 +133,13 @@ static bool openssl_kem_generate_pkey(private_key_exchange_t *this)
} }
if (this->drbg) if (this->drbg)
{ {
if (!EVP_PKEY_keygen_deterministic(ctx, NULL, NULL, &seed_length)) seed = chunk_alloc(2*ML_KEM_SEED_LEN);
{
EVP_PKEY_CTX_free(ctx);
return FALSE;
}
seed = chunk_alloc(seed_length);
if (!this->drbg->generate(this->drbg, seed.len, seed.ptr)) if (!this->drbg->generate(this->drbg, seed.len, seed.ptr))
{ {
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
return FALSE; return FALSE;
} }
#ifdef OPENSSL_IS_AWSLC
if (!EVP_PKEY_keygen_deterministic(ctx, &this->pkey, seed.ptr, if (!EVP_PKEY_keygen_deterministic(ctx, &this->pkey, seed.ptr,
&seed.len)) &seed.len))
{ {
@@ -128,8 +147,22 @@ static bool openssl_kem_generate_pkey(private_key_exchange_t *this)
chunk_clear(&seed); chunk_clear(&seed);
return FALSE; return FALSE;
} }
#else /* OPENSSL_IS_AWSLC */
OSSL_PARAM params[] = {
OSSL_PARAM_octet_string(OSSL_PKEY_PARAM_ML_KEM_SEED,
seed.ptr, seed.len),
OSSL_PARAM_END,
};
if (!EVP_PKEY_CTX_set_params(ctx, params) ||
EVP_PKEY_keygen(ctx, &this->pkey) <= 0)
{
EVP_PKEY_CTX_free(ctx);
chunk_clear(&seed);
return FALSE;
}
#endif /* OPENSSL_IS_AWSLC */
} }
else if (!EVP_PKEY_keygen(ctx, &this->pkey)) else if (EVP_PKEY_keygen(ctx, &this->pkey) <= 0)
{ {
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
return FALSE; return FALSE;
@@ -208,16 +241,23 @@ static bool openssl_kem_decapsulate(private_key_exchange_t *this,
{ {
return FALSE; return FALSE;
} }
if (!EVP_PKEY_decapsulate(ctx, NULL, &shared_secret_length, ciphertext.ptr, #ifndef OPENSSL_IS_AWSLC
ciphertext.len)) if (EVP_PKEY_decapsulate_init(ctx, NULL) <= 0)
{
EVP_PKEY_CTX_free(ctx);
return FALSE;
}
#endif /* !OPENSSL_IS_AWSLC */
if (EVP_PKEY_decapsulate(ctx, NULL, &shared_secret_length, ciphertext.ptr,
ciphertext.len) <= 0)
{ {
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
return FALSE; return FALSE;
} }
this->shared_secret = chunk_alloc(shared_secret_length); this->shared_secret = chunk_alloc(shared_secret_length);
if (!EVP_PKEY_decapsulate(ctx, this->shared_secret.ptr, if (EVP_PKEY_decapsulate(ctx, this->shared_secret.ptr,
&this->shared_secret.len, ciphertext.ptr, &this->shared_secret.len, ciphertext.ptr,
ciphertext.len)) ciphertext.len) <= 0)
{ {
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
chunk_clear(&this->shared_secret); chunk_clear(&this->shared_secret);
@@ -238,11 +278,16 @@ static bool openssl_kem_encapsulate(private_key_exchange_t *this,
EVP_PKEY *pkey; EVP_PKEY *pkey;
size_t shared_secret_length = 0; size_t shared_secret_length = 0;
size_t ciphertext_length = 0; size_t ciphertext_length = 0;
size_t seed_length = 0; size_t seed_length = ML_KEM_SEED_LEN;
chunk_t seed = chunk_empty; chunk_t seed = chunk_empty;
#ifdef OPENSSL_IS_AWSLC
pkey = EVP_PKEY_kem_new_raw_public_key(openssl_kem_get_nid(this), pkey = EVP_PKEY_kem_new_raw_public_key(openssl_kem_get_nid(this),
public_key.ptr, public_key.len); public_key.ptr, public_key.len);
#else
pkey = EVP_PKEY_new_raw_public_key(openssl_kem_get_nid(this), NULL,
public_key.ptr, public_key.len);
#endif
if (!pkey) if (!pkey)
{ {
return FALSE; return FALSE;
@@ -253,6 +298,7 @@ static bool openssl_kem_encapsulate(private_key_exchange_t *this,
EVP_PKEY_free(pkey); EVP_PKEY_free(pkey);
return FALSE; return FALSE;
} }
#ifdef OPENSSL_IS_AWSLC
if (this->drbg) if (this->drbg)
{ {
if (!EVP_PKEY_encapsulate_deterministic(ctx, NULL, &ciphertext_length, if (!EVP_PKEY_encapsulate_deterministic(ctx, NULL, &ciphertext_length,
@@ -279,11 +325,39 @@ static bool openssl_kem_encapsulate(private_key_exchange_t *this,
chunk_clear(&seed); chunk_clear(&seed);
return FALSE; return FALSE;
} }
chunk_clear(&seed);
} }
else else
#endif /* OPENSSL_IS_AWSLC */
{ {
if (!EVP_PKEY_encapsulate(ctx, NULL, &ciphertext_length, NULL, #ifndef OPENSSL_IS_AWSLC
&shared_secret_length)) OSSL_PARAM params[] = {
OSSL_PARAM_END,
OSSL_PARAM_END,
};
if (this->drbg)
{
seed = chunk_alloc(seed_length);
if (!this->drbg->generate(this->drbg, seed.len, seed.ptr))
{
EVP_PKEY_free(pkey);
EVP_PKEY_CTX_free(ctx);
return FALSE;
}
params[0] = OSSL_PARAM_construct_octet_string(OSSL_KEM_PARAM_IKME,
seed.ptr, seed.len);
}
if (EVP_PKEY_encapsulate_init(ctx, params) <= 0)
{
EVP_PKEY_free(pkey);
EVP_PKEY_CTX_free(ctx);
chunk_clear(&seed);
return FALSE;
}
chunk_clear(&seed);
#endif /* !OPENSSL_IS_AWSLC */
if (EVP_PKEY_encapsulate(ctx, NULL, &ciphertext_length, NULL,
&shared_secret_length) <= 0)
{ {
EVP_PKEY_free(pkey); EVP_PKEY_free(pkey);
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
@@ -291,9 +365,9 @@ static bool openssl_kem_encapsulate(private_key_exchange_t *this,
} }
this->shared_secret = chunk_alloc(shared_secret_length); this->shared_secret = chunk_alloc(shared_secret_length);
this->ciphertext = chunk_alloc(ciphertext_length); this->ciphertext = chunk_alloc(ciphertext_length);
if (!EVP_PKEY_encapsulate(ctx, this->ciphertext.ptr, &this->ciphertext.len, if (EVP_PKEY_encapsulate(ctx, this->ciphertext.ptr, &this->ciphertext.len,
this->shared_secret.ptr, this->shared_secret.ptr,
&this->shared_secret.len)) &this->shared_secret.len) <= 0)
{ {
EVP_PKEY_free(pkey); EVP_PKEY_free(pkey);
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
@@ -302,7 +376,6 @@ static bool openssl_kem_encapsulate(private_key_exchange_t *this,
} }
EVP_PKEY_free(pkey); EVP_PKEY_free(pkey);
EVP_PKEY_CTX_free(ctx); EVP_PKEY_CTX_free(ctx);
chunk_clear(&seed);
return TRUE; return TRUE;
} }
@@ -340,7 +413,8 @@ METHOD(key_exchange_t, set_seed, bool, private_key_exchange_t *this,
#endif /* TESTABLE_KE */ #endif /* TESTABLE_KE */
METHOD(key_exchange_t, destroy, void, private_key_exchange_t *this) METHOD(key_exchange_t, destroy, void,
private_key_exchange_t *this)
{ {
EVP_PKEY_free(this->pkey); EVP_PKEY_free(this->pkey);
chunk_clear(&this->shared_secret); chunk_clear(&this->shared_secret);
@@ -373,4 +447,5 @@ key_exchange_t *openssl_kem_create(key_exchange_method_t method)
return &this->public; return &this->public;
} }
#endif /* OPENSSL_IS_AWSLC */
#endif /* OPENSSL_VERSION || OPENSSL_IS_AWSLC */
@@ -553,7 +553,8 @@ METHOD(plugin_t, get_features, int,
PLUGIN_PROVIDE(KE, MODP_768_BIT), PLUGIN_PROVIDE(KE, MODP_768_BIT),
PLUGIN_PROVIDE(KE, MODP_CUSTOM), PLUGIN_PROVIDE(KE, MODP_CUSTOM),
#endif #endif
#ifdef OPENSSL_IS_AWSLC #if (OPENSSL_VERSION_NUMBER >= 0x30500000L && !defined(OPENSSL_NO_ML_KEM)) || \
defined(OPENSSL_IS_AWSLC)
/* ML-KEM key exchanges */ /* ML-KEM key exchanges */
PLUGIN_REGISTER(KE, openssl_kem_create), PLUGIN_REGISTER(KE, openssl_kem_create),
PLUGIN_PROVIDE(KE, ML_KEM_512), PLUGIN_PROVIDE(KE, ML_KEM_512),