refactor: cryptography component

src/Cryptography/Algorithms/AlgorithmMd5.cpp

@@ -0,0 +1,54 @@
+#include "AlgorithmMd5.h"
+
+#include "Internal/CryptoLibrary.h"
+
+#include <cstdint>
+
+using namespace cryptography;
+
+namespace
+{
+    constexpr int HASH_SIZE = 16;
+
+    class AlgorithmMd5 final : public IHashFunction
+    {
+    public:
+        AlgorithmMd5()
+        {
+            internal::CryptoLibrary::Init();
+
+            Init();
+        }
+
+        size_t GetHashSize() override
+        {
+            return HASH_SIZE;
+        }
+
+        void Init() override
+        {
+            md5_init(&m_state);
+        }
+
+        void Process(const void* input, const size_t inputSize) override
+        {
+            md5_process(&m_state, static_cast<const uint8_t*>(input), static_cast<unsigned long>(inputSize));
+        }
+
+        void Finish(void* hashBuffer) override
+        {
+            md5_done(&m_state, static_cast<uint8_t*>(hashBuffer));
+        }
+
+    private:
+        hash_state m_state{};
+    };
+} // namespace
+
+namespace cryptography
+{
+    std::unique_ptr<IHashFunction> CreateMd5()
+    {
+        return std::make_unique<AlgorithmMd5>();
+    }
+} // namespace cryptography

src/Cryptography/Algorithms/AlgorithmMd5.h

@@ -0,0 +1,10 @@
+#pragma once
+
+#include "IHashFunction.h"
+
+#include <memory>
+
+namespace cryptography
+{
+    std::unique_ptr<IHashFunction> CreateMd5();
+}

src/Cryptography/Algorithms/AlgorithmRsa.cpp

@@ -0,0 +1,84 @@
+#include "AlgorithmRsa.h"
+
+#include "Internal/CryptoLibrary.h"
+
+using namespace cryptography;
+
+namespace
+{
+    class AlgorithmRsa final : public IPublicKeyAlgorithm
+    {
+    public:
+        AlgorithmRsa(const HashingAlgorithm hash, const RsaPaddingMode padding)
+        {
+            m_hash = hash;
+            m_padding = padding;
+
+            internal::CryptoLibrary::Init();
+        }
+
+        bool SetKey(const uint8_t* keyData, const size_t keySize) override
+        {
+            return rsa_import(keyData, static_cast<unsigned long>(keySize), &m_key) == CRYPT_OK;
+        }
+
+        bool Verify(const uint8_t* signedData, const size_t signedDataSize, const uint8_t* signature, const size_t signatureSize) override
+        {
+            const ltc_hash_descriptor* hashDesc = GetHashDescriptor();
+            const int hashId = register_hash(hashDesc);
+            const int padding = GetPaddingMode();
+
+            int result;
+            rsa_verify_hash_ex(signature,
+                               static_cast<unsigned long>(signatureSize),
+                               signedData,
+                               static_cast<unsigned long>(signedDataSize),
+                               padding,
+                               hashId,
+                               8,
+                               &result,
+                               &m_key);
+
+            return result == 1;
+        }
+
+    private:
+        [[nodiscard]] const ltc_hash_descriptor* GetHashDescriptor() const
+        {
+            switch (m_hash)
+            {
+            case HashingAlgorithm::RSA_HASH_SHA256:
+                return &sha256_desc;
+
+            default:
+            case HashingAlgorithm::RSA_HASH_SHA512:
+                return &sha512_desc;
+            }
+        }
+
+        [[nodiscard]] int GetPaddingMode() const
+        {
+            switch (m_padding)
+            {
+            case RsaPaddingMode::RSA_PADDING_PKS1:
+                return LTC_PKCS_1_V1_5;
+
+            default:
+            case RsaPaddingMode::RSA_PADDING_PSS:
+                return LTC_PKCS_1_PSS;
+            }
+        }
+
+        rsa_key m_key{};
+        HashingAlgorithm m_hash;
+        RsaPaddingMode m_padding;
+    };
+} // namespace
+
+namespace cryptography
+{
+    std::unique_ptr<IPublicKeyAlgorithm> CreateRsa(const HashingAlgorithm hashingAlgorithm, const RsaPaddingMode paddingMode)
+    {
+        return std::make_unique<AlgorithmRsa>(hashingAlgorithm, paddingMode);
+    }
+} // namespace cryptography

src/Cryptography/Algorithms/AlgorithmRsa.h

@@ -0,0 +1,17 @@
+#pragma once
+
+#include "IPublicKeyAlgorithm.h"
+
+#include <cstdint>
+#include <memory>
+
+namespace cryptography
+{
+    enum class RsaPaddingMode : std::uint8_t
+    {
+        RSA_PADDING_PKS1,
+        RSA_PADDING_PSS,
+    };
+
+    std::unique_ptr<IPublicKeyAlgorithm> CreateRsa(HashingAlgorithm hashingAlgorithm, RsaPaddingMode paddingMode);
+} // namespace cryptography

src/Cryptography/Algorithms/AlgorithmSalsa20.cpp

@@ -0,0 +1,46 @@
+#include "AlgorithmSalsa20.h"
+
+#include "salsa20.h"
+
+#include <cassert>
+#include <stdexcept>
+
+using namespace cryptography;
+
+namespace
+{
+    class AlgorithmSalsa20 final : public IStreamCipher
+    {
+    public:
+        AlgorithmSalsa20(const uint8_t* keyBytes, const size_t keySize)
+        {
+            Salsa20_KeySetup(&m_context, keyBytes, static_cast<uint32_t>(keySize * 8uz));
+        }
+
+        void SetIv(const uint8_t* iv, const size_t ivSize) override
+        {
+            assert(ivSize == 8);
+
+            if (ivSize != 8)
+                throw std::invalid_argument("Salsa20 IV size must be 8");
+
+            Salsa20_IVSetup(&m_context, iv);
+        }
+
+        void Process(const void* plainText, void* cipherText, const size_t amount) override
+        {
+            Salsa20_Encrypt_Bytes(&m_context, static_cast<const uint8_t*>(plainText), static_cast<uint8_t*>(cipherText), static_cast<uint32_t>(amount));
+        }
+
+    private:
+        salsa20_ctx m_context{};
+    };
+} // namespace
+
+namespace cryptography
+{
+    std::unique_ptr<IStreamCipher> CreateSalsa20(const uint8_t* keyBytes, const size_t keySize)
+    {
+        return std::make_unique<AlgorithmSalsa20>(keyBytes, keySize);
+    }
+} // namespace cryptography

src/Cryptography/Algorithms/AlgorithmSalsa20.h

@@ -0,0 +1,10 @@
+#pragma once
+
+#include "IStreamCipher.h"
+
+#include <memory>
+
+namespace cryptography
+{
+    std::unique_ptr<IStreamCipher> CreateSalsa20(const uint8_t* keyBytes, size_t keySize);
+}

src/Cryptography/Algorithms/AlgorithmSha1.cpp

@@ -0,0 +1,54 @@
+#include "AlgorithmSha1.h"
+
+#include "Internal/CryptoLibrary.h"
+
+#include <cstdint>
+
+using namespace cryptography;
+
+namespace
+{
+    constexpr int HASH_SIZE = 20;
+
+    class AlgorithmSha1 final : public IHashFunction
+    {
+    public:
+        AlgorithmSha1()
+        {
+            internal::CryptoLibrary::Init();
+
+            Init();
+        }
+
+        size_t GetHashSize() override
+        {
+            return HASH_SIZE;
+        }
+
+        void Init() override
+        {
+            sha1_init(&m_state);
+        }
+
+        void Process(const void* input, const size_t inputSize) override
+        {
+            sha1_process(&m_state, static_cast<const uint8_t*>(input), static_cast<unsigned long>(inputSize));
+        }
+
+        void Finish(void* hashBuffer) override
+        {
+            sha1_done(&m_state, static_cast<uint8_t*>(hashBuffer));
+        }
+
+    private:
+        hash_state m_state{};
+    };
+} // namespace
+
+namespace cryptography
+{
+    std::unique_ptr<IHashFunction> CreateSha1()
+    {
+        return std::make_unique<AlgorithmSha1>();
+    }
+} // namespace cryptography

src/Cryptography/Algorithms/AlgorithmSha1.h

@@ -0,0 +1,10 @@
+#pragma once
+
+#include "IHashFunction.h"
+
+#include <memory>
+
+namespace cryptography
+{
+    std::unique_ptr<IHashFunction> CreateSha1();
+}

src/Cryptography/Algorithms/AlgorithmSha256.cpp

@@ -0,0 +1,54 @@
+#include "AlgorithmSha256.h"
+
+#include "Internal/CryptoLibrary.h"
+
+#include <cstdint>
+
+using namespace cryptography;
+
+namespace
+{
+    constexpr int HASH_SIZE = 32;
+
+    class AlgorithmSha256 final : public IHashFunction
+    {
+    public:
+        AlgorithmSha256()
+        {
+            internal::CryptoLibrary::Init();
+
+            Init();
+        }
+
+        size_t GetHashSize() override
+        {
+            return HASH_SIZE;
+        }
+
+        void Init() override
+        {
+            sha256_init(&m_state);
+        }
+
+        void Process(const void* input, const size_t inputSize) override
+        {
+            sha256_process(&m_state, static_cast<const uint8_t*>(input), static_cast<unsigned long>(inputSize));
+        }
+
+        void Finish(void* hashBuffer) override
+        {
+            sha256_done(&m_state, static_cast<uint8_t*>(hashBuffer));
+        }
+
+    private:
+        hash_state m_state{};
+    };
+} // namespace
+
+namespace cryptography
+{
+    std::unique_ptr<IHashFunction> CreateSha256()
+    {
+        return std::make_unique<AlgorithmSha256>();
+    }
+} // namespace cryptography

src/Cryptography/Algorithms/AlgorithmSha256.h

@@ -0,0 +1,10 @@
+#pragma once
+
+#include "IHashFunction.h"
+
+#include <memory>
+
+namespace cryptography
+{
+    std::unique_ptr<IHashFunction> CreateSha256();
+}

src/Cryptography/Base64.cpp

@@ -0,0 +1,72 @@
+#include "Base64.h"
+
+#define LTC_NO_PROTOTYPES
+#include <tomcrypt.h>
+
+namespace base64
+{
+    std::string EncodeBase64(const void* inputData, const size_t inputLength)
+    {
+        const auto base64BufferSize = GetBase64EncodeOutputLength(inputLength);
+
+        std::string output(base64BufferSize, '\0');
+        const auto outLength = base64BufferSize + 1u;
+
+        const auto result = EncodeBase64(inputData, inputLength, output.data(), outLength);
+        assert(result);
+
+        return output;
+    }
+
+    bool EncodeBase64(const void* inputData, const size_t inputLength, void* outputBuffer, const size_t outputBufferSize)
+    {
+        unsigned long outLength = static_cast<unsigned long>(outputBufferSize);
+        const auto result =
+            base64_encode(static_cast<const unsigned char*>(inputData), static_cast<unsigned long>(inputLength), static_cast<char*>(outputBuffer), &outLength);
+        return result == CRYPT_OK;
+    }
+
+    size_t GetBase64EncodeOutputLength(const size_t inputLength)
+    {
+        return 4u * ((inputLength + 2u) / 3u);
+    }
+
+    size_t DecodeBase64(const void* base64Data, const size_t inputLength, void* outputBuffer, const size_t outputBufferSize)
+    {
+        auto outLength = static_cast<unsigned long>(GetBase64DecodeOutputLength(base64Data, inputLength));
+        assert(outLength <= outputBufferSize);
+        if (outLength > outputBufferSize)
+            return 0u;
+
+        const auto result =
+            base64_decode(static_cast<const char*>(base64Data), static_cast<unsigned long>(inputLength), static_cast<unsigned char*>(outputBuffer), &outLength);
+        assert(result == CRYPT_OK);
+
+        return static_cast<size_t>(outLength);
+    }
+
+    size_t GetBase64DecodeOutputLength(const void* base64Data, const size_t inputLength)
+    {
+        assert(base64Data);
+        assert(inputLength);
+
+        if (!base64Data || inputLength == 0u)
+            return 0u;
+
+        auto padding = 0u;
+        if (inputLength >= 1 && static_cast<const char*>(base64Data)[inputLength - 1] == '=')
+        {
+            if (inputLength >= 2 && static_cast<const char*>(base64Data)[inputLength - 2] == '=')
+                padding = 2u;
+            else
+                padding = 1u;
+        }
+
+        return ((inputLength / 4u) * 3u) - padding;
+    }
+
+    size_t GetBase64DecodeOutputLength(const size_t inputLength)
+    {
+        return (inputLength / 4u) * 3u;
+    }
+} // namespace base64

src/Cryptography/Base64.h

@@ -0,0 +1,13 @@
+#pragma once
+#include <string>
+
+namespace base64
+{
+    std::string EncodeBase64(const void* inputData, size_t inputLength);
+    bool EncodeBase64(const void* inputData, size_t inputLength, void* outputBuffer, size_t outputBufferSize);
+    size_t GetBase64EncodeOutputLength(size_t inputLength);
+
+    size_t DecodeBase64(const void* base64Data, size_t inputLength, void* outputBuffer, size_t outputBufferSize);
+    size_t GetBase64DecodeOutputLength(const void* base64Data, size_t inputLength);
+    size_t GetBase64DecodeOutputLength(size_t inputLength);
+} // namespace base64

src/Cryptography/Cryptography.h

@@ -0,0 +1,7 @@
+#pragma once
+
+#include "Algorithms/AlgorithmMd5.h"
+#include "Algorithms/AlgorithmRsa.h"
+#include "Algorithms/AlgorithmSalsa20.h"
+#include "Algorithms/AlgorithmSha1.h"
+#include "Algorithms/AlgorithmSha256.h"

src/Cryptography/IHashFunction.h

@@ -0,0 +1,22 @@
+#pragma once
+
+#include <cstdlib>
+
+namespace cryptography
+{
+    class IHashFunction
+    {
+    public:
+        IHashFunction() = default;
+        virtual ~IHashFunction() = default;
+        IHashFunction(const IHashFunction& other) = default;
+        IHashFunction(IHashFunction&& other) noexcept = default;
+        IHashFunction& operator=(const IHashFunction& other) = default;
+        IHashFunction& operator=(IHashFunction&& other) noexcept = default;
+
+        virtual size_t GetHashSize() = 0;
+        virtual void Init() = 0;
+        virtual void Process(const void* input, size_t inputSize) = 0;
+        virtual void Finish(void* hashBuffer) = 0;
+    };
+} // namespace cryptography

src/Cryptography/IPublicKeyAlgorithm.h

@@ -0,0 +1,30 @@
+#pragma once
+
+#include <cstdint>
+#include <cstdlib>
+
+namespace cryptography
+{
+    enum class HashingAlgorithm : std::uint8_t
+    {
+        RSA_HASH_SHA256,
+        RSA_HASH_SHA512
+    };
+
+    class IPublicKeyAlgorithm
+    {
+    public:
+        IPublicKeyAlgorithm() = default;
+        virtual ~IPublicKeyAlgorithm() = default;
+        IPublicKeyAlgorithm(const IPublicKeyAlgorithm& other) = default;
+        IPublicKeyAlgorithm(IPublicKeyAlgorithm&& other) noexcept = default;
+        IPublicKeyAlgorithm& operator=(const IPublicKeyAlgorithm& other) = default;
+        IPublicKeyAlgorithm& operator=(IPublicKeyAlgorithm&& other) noexcept = default;
+
+        virtual bool SetKey(const uint8_t* keyData, size_t keySize) = 0;
+
+        // If needed, add a signing method
+
+        virtual bool Verify(const uint8_t* signedData, size_t signedDataSize, const uint8_t* signature, size_t signatureSize) = 0;
+    };
+} // namespace cryptography

src/Cryptography/IStreamCipher.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include <cstdint>
+#include <cstdlib>
+
+namespace cryptography
+{
+    class IStreamCipher
+    {
+    public:
+        IStreamCipher() = default;
+        virtual ~IStreamCipher() = default;
+        IStreamCipher(const IStreamCipher& other) = default;
+        IStreamCipher(IStreamCipher&& other) noexcept = default;
+        IStreamCipher& operator=(const IStreamCipher& other) = default;
+        IStreamCipher& operator=(IStreamCipher&& other) noexcept = default;
+
+        virtual void SetIv(const uint8_t* iv, size_t ivSize) = 0;
+        virtual void Process(const void* plainText, void* cipherText, size_t amount) = 0;
+    };
+} // namespace cryptography

src/Cryptography/Internal/CryptoLibrary.cpp

@@ -0,0 +1,18 @@
+#include "CryptoLibrary.h"
+
+#include "tommath.h"
+
+namespace cryptography::internal
+{
+    void CryptoLibrary::Init()
+    {
+        static bool initialized = false;
+
+        if (!initialized)
+        {
+            initialized = true;
+
+            ltc_mp = ltm_desc;
+        }
+    }
+} // namespace cryptography::internal

src/Cryptography/Internal/CryptoLibrary.h

@@ -0,0 +1,13 @@
+#pragma once
+
+#define LTC_NO_PROTOTYPES
+#include "tomcrypt.h"
+
+namespace cryptography::internal
+{
+    class CryptoLibrary
+    {
+    public:
+        static void Init();
+    };
+} // namespace cryptography::internal