generic encryption for RoutingDestination

libi2pd/CryptoKey.cpp

@@ -11,11 +11,9 @@ namespace crypto
 		memcpy (m_PublicKey, pub, 256);
 	}
 
-	void ElGamalEncryptor::Encrypt (const uint8_t * data, uint8_t * encrypted)
+	void ElGamalEncryptor::Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx)
 	{
-		BN_CTX * ctx = BN_CTX_new ();
 		ElGamalEncrypt (m_PublicKey, data, encrypted, ctx, true);
-		BN_CTX_free (ctx);
 	}
 
 	ElGamalDecryptor::ElGamalDecryptor (const uint8_t * priv)
@@ -23,11 +21,9 @@ namespace crypto
 		memcpy (m_PrivateKey, priv, 256);
 	}
 
-	void ElGamalDecryptor::Decrypt (const uint8_t * encrypted, uint8_t * data)
+	bool ElGamalDecryptor::Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx)
 	{
-		BN_CTX * ctx = BN_CTX_new ();
+		return ElGamalDecrypt (m_PrivateKey, encrypted, data, ctx, true);
-		ElGamalDecrypt (m_PrivateKey, encrypted, data, ctx, true);
-		BN_CTX_free (ctx);
 	}
 
 	ECIESP256Encryptor::ECIESP256Encryptor (const uint8_t * pub)
@@ -47,14 +43,10 @@ namespace crypto
 		if (m_PublicKey) EC_POINT_free (m_PublicKey);
 	}
 
-	void ECIESP256Encryptor::Encrypt (const uint8_t * data, uint8_t * encrypted)
+	void ECIESP256Encryptor::Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx)
 	{
 		if (m_Curve && m_PublicKey)
-		{
-			BN_CTX * ctx = BN_CTX_new ();
 			ECIESEncrypt (m_Curve, m_PublicKey, data, encrypted, ctx);
-			BN_CTX_free (ctx);
-		}
 	}
 
 	ECIESP256Decryptor::ECIESP256Decryptor (const uint8_t * priv)
@@ -69,14 +61,11 @@ namespace crypto
 		if (m_PrivateKey) BN_free (m_PrivateKey);
 	}
 
-	void ECIESP256Decryptor::Decrypt (const uint8_t * encrypted, uint8_t * data)
+	bool ECIESP256Decryptor::Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx)
 	{
 		if (m_Curve && m_PrivateKey)
-		{
+			return ECIESDecrypt (m_Curve, m_PrivateKey, encrypted, data, ctx);
-			BN_CTX * ctx = BN_CTX_new ();
+		return false;;
-			ECIESDecrypt (m_Curve, m_PrivateKey, encrypted, data, ctx);
-			BN_CTX_free (ctx);
-		}
 	}
 
 	void CreateECIESP256RandomKeys (uint8_t * priv, uint8_t * pub)

libi2pd/CryptoKey.h

@@ -13,7 +13,7 @@ namespace crypto
 		public:
 
 			virtual ~CryptoKeyEncryptor () {};
-			virtual void Encrypt (const uint8_t * data, uint8_t * encrypted); // 222 bytes data, 512 bytes encrypted
+			virtual void Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx) = 0; // 222 bytes data, 512 bytes encrypted
 	};	
 
 	class CryptoKeyDecryptor 
@@ -21,40 +21,40 @@ namespace crypto
 		public:
 
 			virtual ~CryptoKeyDecryptor () {};
-			virtual void Decrypt (const uint8_t * encrypted, uint8_t * data); // 512 bytes encrypted, 222 bytes data
+			virtual bool Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx) = 0; // 512 bytes encrypted, 222 bytes data
 	};
 
-	class ElGamalEncryptor // for destination
+	class ElGamalEncryptor: public CryptoKeyEncryptor // for destination
 	{
 		public:
 
 			ElGamalEncryptor (const uint8_t * pub);
-			void Encrypt (const uint8_t * data, uint8_t * encrypted); 
+			void Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx); 
 
 		private:
 
 			uint8_t m_PublicKey[256];
 	};
 
-	class ElGamalDecryptor // for destination
+	class ElGamalDecryptor: public CryptoKeyDecryptor // for destination
 	{
 		public:
 
 			ElGamalDecryptor (const uint8_t * priv);
-			void Decrypt (const uint8_t * encrypted, uint8_t * data); 
+			bool Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx); 
 
 		private:
 
 			uint8_t m_PrivateKey[256];
 	};
 
-	class ECIESP256Encryptor
+	class ECIESP256Encryptor: public CryptoKeyEncryptor 
 	{
 		public:
 
 			ECIESP256Encryptor (const uint8_t * pub);
 			~ECIESP256Encryptor ();
-			void Encrypt (const uint8_t * data, uint8_t * encrypted); 
+			void Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx); 
 
 		private:
 
@@ -63,13 +63,13 @@ namespace crypto
 	};
 
 
-	class ECIESP256Decryptor
+	class ECIESP256Decryptor: public CryptoKeyDecryptor
 	{
 		public:
 
 			ECIESP256Decryptor (const uint8_t * priv);
 			~ECIESP256Decryptor ();
-			void Decrypt (const uint8_t * encrypted, uint8_t * data); 
+			bool Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx); 
 
 		private:
 

libi2pd/Identity.cpp

@@ -1,7 +1,6 @@
 #include <time.h>
 #include <stdio.h>
 #include "Crypto.h"
-#include "CryptoKey.h"
 #include "I2PEndian.h"
 #include "Log.h"
 #include "Identity.h"
@@ -443,6 +442,23 @@ namespace data
 		m_Verifier = nullptr;
 	}
 
+	std::shared_ptr<i2p::crypto::CryptoKeyEncryptor> IdentityEx::CreateEncryptor (const uint8_t * key) const
+	{
+		if (!key) key = GetEncryptionPublicKey (); // use publicKey 
+		switch (GetCryptoKeyType ())
+		{
+			case CRYPTO_KEY_TYPE_ELGAMAL:
+				return std::make_shared<i2p::crypto::ElGamalEncryptor>(key);
+			break;
+			case CRYPTO_KEY_TYPE_ECIES_P256_SHA256_AES256CBC:
+				return std::make_shared<i2p::crypto::ECIESP256Encryptor>(key);
+			break;
+			default:
+				LogPrint (eLogError, "Identity: Unknown crypto key type ", (int)GetCryptoKeyType ());
+		};
+		return nullptr;
+	}
+
 	PrivateKeys& PrivateKeys::operator=(const Keys& keys)
 	{
 		m_Public = std::make_shared<IdentityEx>(Identity (keys));
@@ -568,6 +584,23 @@ namespace data
 			return nullptr; // TODO: implement me
 	}
 
+	std::shared_ptr<i2p::crypto::CryptoKeyDecryptor> PrivateKeys::CreateDecryptor (const uint8_t * key) const
+	{
+		if (!key) key = m_PrivateKey; // use privateKey 
+		switch (m_Public->GetCryptoKeyType ())
+		{
+			case CRYPTO_KEY_TYPE_ELGAMAL:
+				return std::make_shared<i2p::crypto::ElGamalDecryptor>(key);
+			break;
+			case CRYPTO_KEY_TYPE_ECIES_P256_SHA256_AES256CBC:
+				return std::make_shared<i2p::crypto::ECIESP256Decryptor>(key);
+			break;
+			default:
+				LogPrint (eLogError, "Identity: Unknown crypto key type ", (int)m_Public->GetCryptoKeyType ());
+		};
+		return nullptr;
+	}
+
 	PrivateKeys PrivateKeys::CreateRandomKeys (SigningKeyType type, CryptoKeyType cryptoType)
 	{
 		if (type != SIGNING_KEY_TYPE_DSA_SHA1)
@@ -627,7 +660,7 @@ namespace data
 			case CRYPTO_KEY_TYPE_ELGAMAL:
 				i2p::crypto::GenerateElGamalKeyPair(priv, pub);
 			break;
-			case CRYPTO_KEY_TYPE_ECICS_P256_SHA256_AES256CBC:
+			case CRYPTO_KEY_TYPE_ECIES_P256_SHA256_AES256CBC:
 				i2p::crypto::CreateECIESP256RandomKeys (priv, pub);
 			break;
 			default:

libi2pd/Identity.h

@@ -8,6 +8,7 @@
 #include <atomic>
 #include "Base.h"
 #include "Signature.h"
+#include "CryptoKey.h"
 
 namespace i2p
 {
@@ -52,7 +53,7 @@ namespace data
 	const size_t DEFAULT_IDENTITY_SIZE = sizeof (Identity); // 387 bytes
 
 	const uint16_t CRYPTO_KEY_TYPE_ELGAMAL = 0;
-	const uint16_t CRYPTO_KEY_TYPE_ECICS_P256_SHA256_AES256CBC = 65280; // TODO: change to actual code
+	const uint16_t CRYPTO_KEY_TYPE_ECIES_P256_SHA256_AES256CBC = 65280; // TODO: change to actual code
 
 	const uint16_t SIGNING_KEY_TYPE_DSA_SHA1 = 0;
 	const uint16_t SIGNING_KEY_TYPE_ECDSA_SHA256_P256 = 1;
@@ -93,6 +94,7 @@ namespace data
 			const IdentHash& GetIdentHash () const { return m_IdentHash; };
     		const uint8_t * GetEncryptionPublicKey () const { return m_StandardIdentity.publicKey; };
     		uint8_t * GetEncryptionPublicKeyBuffer () { return m_StandardIdentity.publicKey; };
+			std::shared_ptr<i2p::crypto::CryptoKeyEncryptor> CreateEncryptor (const uint8_t * key) const; 
 			size_t GetFullLen () const { return m_ExtendedLen + DEFAULT_IDENTITY_SIZE; };
 			size_t GetSigningPublicKeyLen () const;
 			size_t GetSigningPrivateKeyLen () const;
@@ -146,6 +148,8 @@ namespace data
 			size_t FromBase64(const std::string& s);
 			std::string ToBase64 () const;
 
+			std::shared_ptr<i2p::crypto::CryptoKeyDecryptor> CreateDecryptor (const uint8_t * key) const;
+
 			static PrivateKeys CreateRandomKeys (SigningKeyType type = SIGNING_KEY_TYPE_DSA_SHA1, CryptoKeyType cryptoType = CRYPTO_KEY_TYPE_ELGAMAL);
 			static void GenerateCryptoKeyPair (CryptoKeyType type, uint8_t * priv, uint8_t * pub); // priv and pub are 256 bytes long
 
@@ -187,7 +191,8 @@ namespace data
 			virtual ~RoutingDestination () {};
 
 			virtual std::shared_ptr<const IdentityEx> GetIdentity ()  const = 0;
-			virtual const uint8_t * GetEncryptionPublicKey () const = 0;
+			virtual const uint8_t * GetEncryptionPublicKey () const = 0; // deprecated
+			virtual void Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx) = 0; // encrypt data for
 			virtual bool IsDestination () const = 0; // for garlic
 
 			const IdentHash& GetIdentHash () const { return GetIdentity ()->GetIdentHash (); };

libi2pd/LeaseSet.cpp

@@ -208,6 +208,13 @@ namespace data
 		return ts > m_ExpirationTime;
 	}
 
+	void LeaseSet::Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx)
+	{
+		auto encryptor = m_Identity->CreateEncryptor (m_EncryptionKey);
+		if (encryptor)
+			encryptor->Encrypt (data, encrypted, ctx);
+	}
+
 	LocalLeaseSet::LocalLeaseSet (std::shared_ptr<const IdentityEx> identity, const uint8_t * encryptionPublicKey, std::vector<std::shared_ptr<i2p::tunnel::InboundTunnel> > tunnels):
 		m_ExpirationTime (0), m_Identity (identity)
 	{

libi2pd/LeaseSet.h

@@ -76,7 +76,8 @@ namespace data
 
 			// implements RoutingDestination
 			std::shared_ptr<const IdentityEx> GetIdentity () const { return m_Identity; };
-			const uint8_t * GetEncryptionPublicKey () const { return m_EncryptionKey; };
+			const uint8_t * GetEncryptionPublicKey () const { return m_EncryptionKey; }; // deprecated
+			void Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx);
 			bool IsDestination () const { return true; };
 
 		private:

libi2pd/RouterInfo.cpp

@@ -828,5 +828,11 @@ namespace data
 			m_Profile = GetRouterProfile (GetIdentHash ());
 		return m_Profile;
 	}	
+
+	void RouterInfo::Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx)
+	{
+		// TODO: we always assume ElGamal for RouterInfo, might change later
+		i2p::crypto::ElGamalEncrypt (m_RouterIdentity->GetEncryptionPublicKey (), data, encrypted, ctx);
+	}
 }
 }

libi2pd/RouterInfo.h

@@ -186,7 +186,9 @@ namespace data
       
 			// implements RoutingDestination
 			std::shared_ptr<const IdentityEx> GetIdentity () const { return m_RouterIdentity; };
-			const uint8_t * GetEncryptionPublicKey () const { return m_RouterIdentity->GetStandardIdentity ().publicKey; };
+			const uint8_t * GetEncryptionPublicKey () const { return m_RouterIdentity->GetStandardIdentity ().publicKey; }; // deprecated
+			void Encrypt (const uint8_t * data, uint8_t * encrypted, BN_CTX * ctx);			
+
 			bool IsDestination () const { return false; };
 
 		private: