Mirrors/i2pd
2
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2025-04-16 06:02:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

ML-KEM-512 encaps/decaps

Browse source
This commit is contained in:
orignal 2025-03-24 15:45:06 -04:00
parent 75d5c6036e
commit 22d854a6be
2 changed files with 63 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
libi2pd/Crypto.cpp
View file

@ -992,19 +992,71 @@ namespace crypto
void MLKEM512Keys::GenerateKeys () void MLKEM512Keys::GenerateKeys ()
{ {
if (m_Pkey) if (m_Pkey) EVP_PKEY_free (m_Pkey);
{
EVP_PKEY_free (m_Pkey);
m_Pkey = nullptr;
}
m_Pkey = EVP_PKEY_Q_keygen(NULL, NULL, "ML-KEM-512"); m_Pkey = EVP_PKEY_Q_keygen(NULL, NULL, "ML-KEM-512");
} }
void MLKEM512Keys::GetPublicKey (uint8_t * pub) const void MLKEM512Keys::GetPublicKey (uint8_t * pub) const
{ {
size_t len = MLKEM512_KEY_LENGTH; if (m_Pkey)
EVP_PKEY_get_octet_string_param (m_Pkey, OSSL_PKEY_PARAM_PUB_KEY, pub, MLKEM512_KEY_LENGTH, &len); {
size_t len = MLKEM512_KEY_LENGTH;
EVP_PKEY_get_octet_string_param (m_Pkey, OSSL_PKEY_PARAM_PUB_KEY, pub, MLKEM512_KEY_LENGTH, &len);
}
} }
void MLKEM512Keys::SetPublicKey (const uint8_t * pub)
{
if (m_Pkey)
{
EVP_PKEY_free (m_Pkey);
m_Pkey = nullptr;
}
OSSL_PARAM params[] =
{
OSSL_PARAM_octet_string (OSSL_PKEY_PARAM_PUB_KEY, (uint8_t *)pub, MLKEM512_KEY_LENGTH),
OSSL_PARAM_END
};
EVP_PKEY_CTX *ctx = EVP_PKEY_CTX_new_from_name (NULL, "ML-KEM-512", NULL);
if (ctx)
{
EVP_PKEY_fromdata_init (ctx);
EVP_PKEY_fromdata (ctx, &m_Pkey, OSSL_KEYMGMT_SELECT_PUBLIC_KEY, params);
EVP_PKEY_CTX_free (ctx);
}
else
LogPrint (eLogError, "MLKEM512 can't create PKEY context");
}
void MLKEM512Keys::Encaps (uint8_t * ciphertext, uint8_t * shared)
{
if (!m_Pkey) return;
auto ctx = EVP_PKEY_CTX_new_from_pkey (NULL, m_Pkey, NULL);
if (ctx)
{
EVP_PKEY_encapsulate_init (ctx, NULL);
size_t len = MLKEM512_CIPHER_TEXT_LENGTH, sharedLen = 32;
EVP_PKEY_encapsulate (ctx, ciphertext, &len, shared, &sharedLen);
EVP_PKEY_CTX_free (ctx);
}
else
LogPrint (eLogError, "MLKEM512 can't create PKEY context");
}
void MLKEM512Keys::Decaps (const uint8_t * ciphertext, uint8_t * shared)
{
if (!m_Pkey) return;
auto ctx = EVP_PKEY_CTX_new_from_pkey (NULL, m_Pkey, NULL);
if (ctx)
{
EVP_PKEY_decapsulate_init (ctx, NULL);
size_t sharedLen = 32;
EVP_PKEY_decapsulate (ctx, shared, &sharedLen, ciphertext, MLKEM512_CIPHER_TEXT_LENGTH);
EVP_PKEY_CTX_free (ctx);
}
else
LogPrint (eLogError, "MLKEM512 can't create PKEY context");
}
#endif #endif
} }
} }

4
libi2pd/Crypto.h
View file

@ -274,6 +274,7 @@ namespace crypto
#if OPENSSL_PQ #if OPENSSL_PQ
// Post Quantum // Post Quantum
constexpr size_t MLKEM512_KEY_LENGTH = 800; constexpr size_t MLKEM512_KEY_LENGTH = 800;
constexpr size_t MLKEM512_CIPHER_TEXT_LENGTH = 768;
class MLKEM512Keys class MLKEM512Keys
{ {
public: public:
@ -283,6 +284,9 @@ namespace crypto
void GenerateKeys (); void GenerateKeys ();
void GetPublicKey (uint8_t * pub) const; void GetPublicKey (uint8_t * pub) const;
void SetPublicKey (const uint8_t * pub);
void Encaps (uint8_t * ciphertext, uint8_t * shared);
void Decaps (const uint8_t * ciphertext, uint8_t * shared);
private: private: