RSA signature stuff.

src/Utils/Cryptography.cpp

@@ -0,0 +1,48 @@
+#include "STDInclude.hpp"
+
+/// http://www.opensource.apple.com/source/CommonCrypto/CommonCrypto-55010/Source/libtomcrypt/doc/libTomCryptDoc.pdf
+
+namespace Utils
+{
+	namespace Cryptography
+	{
+		RSA::Key RSA::GenerateKey(int bits)
+		{
+			RSA::Key key;
+
+			register_prng(&sprng_desc);
+			register_hash(&sha1_desc);
+
+			ltc_mp = ltm_desc;
+
+			rsa_make_key(NULL, find_prng("sprng"), bits / 8, 65537, key.GetKeyPtr());
+
+			return key;
+		}
+
+		std::string RSA::SignMessage(RSA::Key key, std::string message)
+		{
+			uint8_t buffer[0x200]; // Default size is 512
+			DWORD length = sizeof(buffer);
+
+			register_prng(&sprng_desc);
+			register_hash(&sha1_desc);
+
+			ltc_mp = ltm_desc;
+			
+			rsa_sign_hash(reinterpret_cast<const uint8_t*>(message.data()), message.size(), buffer, &length, NULL, find_prng("sprng"), find_hash("sha1"), 0, key.GetKeyPtr());
+
+			return std::string(reinterpret_cast<char*>(buffer), length);
+		}
+
+		bool RSA::VerifyMessage(Key key, std::string message, std::string signature)
+		{
+			register_hash(&sha1_desc);
+
+			ltc_mp = ltm_desc;
+
+			int result = 0;
+			return (rsa_verify_hash(reinterpret_cast<const uint8_t*>(signature.data()), signature.size(), reinterpret_cast<const uint8_t*>(message.data()), message.size(), find_hash("sha1"), 0, &result, key.GetKeyPtr()) == CRYPT_OK && result != 0);
+		}
+	}
+}

src/Utils/Cryptography.hpp

@@ -0,0 +1,32 @@
+namespace Utils
+{
+	namespace Cryptography
+	{
+		class RSA
+		{
+		public:
+			class Key
+			{
+			public:
+				Key() { ZeroMemory(&this->KeyStorage, sizeof(this->KeyStorage)); };
+				Key(rsa_key* key) : Key(*key) {};
+				Key(rsa_key key) : KeyStorage(key) {};
+				Key(const Key& obj) : KeyStorage(obj.KeyStorage) {};
+
+				~Key() {}
+
+				rsa_key* GetKeyPtr()
+				{
+					return &this->KeyStorage;
+				}
+
+			private:
+				rsa_key KeyStorage;
+			};
+
+			static RSA::Key GenerateKey(int bits);
+			static std::string SignMessage(Key key, std::string message);
+			static bool VerifyMessage(Key key, std::string message, std::string signature);
+		};
+	}
+}