From a83cd99f58f7ecd75e335a88cc94aea13381e6a6 Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Mon, 14 Apr 2025 13:29:53 -0400
Subject: [PATCH] all ML-KEM crypto types

---
 libi2pd/Crypto.h   | 28 ++++++++++++++++++++++++----
 libi2pd/Identity.h |  2 ++
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/libi2pd/Crypto.h b/libi2pd/Crypto.h
index 9b47f806..10f65b32 100644
--- a/libi2pd/Crypto.h
+++ b/libi2pd/Crypto.h
@@ -305,10 +305,16 @@ namespace crypto
 
 	constexpr size_t MLKEM512_KEY_LENGTH = 800;
 	constexpr size_t MLKEM512_CIPHER_TEXT_LENGTH = 768;
-
-	constexpr std::array<std::tuple<std::string_view, size_t, size_t>, 1> MLKEMS =
+	constexpr size_t MLKEM768_KEY_LENGTH = 1184;
+	constexpr size_t MLKEM768_CIPHER_TEXT_LENGTH = 1088;
+	constexpr size_t MLKEM1024_KEY_LENGTH = 1568;
+	constexpr size_t MLKEM1024_CIPHER_TEXT_LENGTH = 1568;
+	
+	constexpr std::array<std::tuple<std::string_view, size_t, size_t>, 3> MLKEMS =
 	{
-		std::make_tuple ("ML-KEM-512", MLKEM512_KEY_LENGTH, MLKEM512_CIPHER_TEXT_LENGTH)
+		std::make_tuple ("ML-KEM-512", MLKEM512_KEY_LENGTH, MLKEM512_CIPHER_TEXT_LENGTH),
+		std::make_tuple ("ML-KEM-768", MLKEM768_KEY_LENGTH, MLKEM768_CIPHER_TEXT_LENGTH),
+		std::make_tuple ("ML-KEM-1024", MLKEM1024_KEY_LENGTH, MLKEM1024_CIPHER_TEXT_LENGTH)
 	};	
 	
 	class MLKEM512Keys: public MLKEMKeys
@@ -317,7 +323,21 @@ namespace crypto
 
 			MLKEM512Keys (): MLKEMKeys (std::get<0>(MLKEMS[0]), std::get<1>(MLKEMS[0]), std::get<2>(MLKEMS[0])) {}
 	};
-		
+
+	class MLKEM768Keys: public MLKEMKeys
+	{
+		public:
+
+			MLKEM768Keys (): MLKEMKeys (std::get<0>(MLKEMS[1]), std::get<1>(MLKEMS[1]), std::get<2>(MLKEMS[1])) {}
+	};
+
+	class MLKEM1024Keys: public MLKEMKeys
+	{
+		public:
+
+			MLKEM1024Keys (): MLKEMKeys (std::get<0>(MLKEMS[2]), std::get<1>(MLKEMS[2]), std::get<2>(MLKEMS[2])) {}
+	};
+	
 #endif	
 }
 }
diff --git a/libi2pd/Identity.h b/libi2pd/Identity.h
index ee736441..173559e9 100644
--- a/libi2pd/Identity.h
+++ b/libi2pd/Identity.h
@@ -71,6 +71,8 @@ namespace data
 	const uint16_t CRYPTO_KEY_TYPE_ECIES_P256_SHA256_AES256CBC = 1;
 	const uint16_t CRYPTO_KEY_TYPE_ECIES_X25519_AEAD = 4;
 	const uint16_t CRYPTO_KEY_TYPE_ECIES_MLKEM512_X25519_AEAD = 5;
+	const uint16_t CRYPTO_KEY_TYPE_ECIES_MLKEM768_X25519_AEAD = 6;
+	const uint16_t CRYPTO_KEY_TYPE_ECIES_MLKEM1024_X25519_AEAD = 7;
 	
 	const uint16_t SIGNING_KEY_TYPE_DSA_SHA1 = 0;
 	const uint16_t SIGNING_KEY_TYPE_ECDSA_SHA256_P256 = 1;