RSA.h

Go to the documentation of this file.

/*
 * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef TRINITYCORE_RSA_H
#define TRINITYCORE_RSA_H
 
#include "Define.h"
#include <openssl/evp.h>
#include <array>
#include <memory>
#include <string>
#include <vector>
 
namespace Trinity
{
namespace Crypto
{
class TC_COMMON_API RsaSignature
{
public:
    class TC_COMMON_API DigestGenerator
    {
    public:
        struct EVP_MD_Deleter
        {
            void operator()(EVP_MD* md) const;
        };
 
        virtual ~DigestGenerator() = default;
        virtual std::unique_ptr<EVP_MD, EVP_MD_Deleter> GetGenerator() const = 0;
 
        virtual OSSL_LIB_CTX* GetLib() const = 0;
        virtual std::unique_ptr<OSSL_PARAM[]> GetParams() const = 0;
    };
 
    class TC_COMMON_API SHA256 : public DigestGenerator
    {
    public:
        std::unique_ptr<EVP_MD, EVP_MD_Deleter> GetGenerator() const override;
 
        OSSL_LIB_CTX* GetLib() const override;
        std::unique_ptr<OSSL_PARAM[]> GetParams() const override;
    };
 
    class TC_COMMON_API HMAC_SHA256 : public DigestGenerator
    {
    public:
        explicit HMAC_SHA256(uint8 const* key, size_t keyLength) : _key(key), _keyLength(keyLength) { }
 
        std::unique_ptr<EVP_MD, EVP_MD_Deleter> GetGenerator() const override;
 
        OSSL_LIB_CTX* GetLib() const override;
        std::unique_ptr<OSSL_PARAM[]> GetParams() const override;
 
    private:
        uint8 const* _key;
        size_t _keyLength;
    };
 
    RsaSignature();
    RsaSignature(RsaSignature const& other);
    RsaSignature(RsaSignature&& other) noexcept;
    ~RsaSignature();
 
    RsaSignature& operator=(RsaSignature const& right);
    RsaSignature& operator=(RsaSignature&& right) noexcept;
 
    bool LoadKeyFromFile(std::string const& fileName);
 
    bool LoadKeyFromString(std::string const& keyPem);
 
    template <std::size_t N>
    bool Sign(std::array<uint8, N> const& message, DigestGenerator& generator, std::vector<uint8>& output)
    {
        return this->Sign(message.data(), message.size(), generator, output);
    }
 
    bool Sign(uint8 const* message, std::size_t messageLength, DigestGenerator& generator, std::vector<uint8>& output);
 
private:
    EVP_MD_CTX* _ctx = nullptr;
    EVP_PKEY* _key = nullptr;
};
}
}
 
#endif // TRINITYCORE_RSA_H