messagesigner.h

Go to the documentation of this file.

// Copyright (c) 2014-2018 The Dash Core developers
// Copyright (c) 2018-2021 The PIVX Core developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef PIVX_MESSAGESIGNER_H
#define PIVX_MESSAGESIGNER_H
 
#include "key.h"
#include "primitives/transaction.h" // for CTxIn
 
class CBLSPublicKey;
class CBLSSecretKey;
 
enum MessageVersion {
        MESS_VER_STRMESS    = 0, // old format
        MESS_VER_HASH       = 1,
};
 
class CMessageSigner
{
public:
    static bool GetKeysFromSecret(const std::string& strSecret, CKey& keyRet, CPubKey& pubkeyRet);
    static bool GetKeysFromSecret(const std::string& strSecret, CKey& keyRet, CKeyID& keyIDRet);
    static uint256 GetMessageHash(const std::string& strMessage);
    static bool SignMessage(const std::string& strMessage, std::vector<unsigned char>& vchSigRet, const CKey& key);
    static bool SignMessage(const std::string& strMessage, std::vector<unsigned char>& vchSigRet, const CBLSSecretKey& key);
    static bool VerifyMessage(const CPubKey& pubkey, const std::vector<unsigned char>& vchSig, const std::string& strMessage, std::string& strErrorRet);
    static bool VerifyMessage(const CKeyID& keyID, const std::vector<unsigned char>& vchSig, const std::string& strMessage, std::string& strErrorRet);
    static bool VerifyMessage(const CBLSPublicKey& pk, const std::vector<unsigned char>& vchSig, const std::string& strMessage);
};
 
class CHashSigner
{
public:
    static bool SignHash(const uint256& hash, const CKey& key, std::vector<unsigned char>& vchSigRet);
    static bool SignHash(const uint256& hash, const CBLSSecretKey& key, std::vector<unsigned char>& vchSigRet);
    static bool VerifyHash(const uint256& hash, const CPubKey& pubkey, const std::vector<unsigned char>& vchSig, std::string& strErrorRet);
    static bool VerifyHash(const uint256& hash, const CKeyID& keyID, const std::vector<unsigned char>& vchSig, std::string& strErrorRet);
    static bool VerifyHash(const uint256& hash, const CBLSPublicKey& pk, const std::vector<unsigned char>& vchSig);
};
 
class CSignedMessage
{
protected:
    std::vector<unsigned char> vchSig;
 
public:
    int nMessVersion;
 
    CSignedMessage() :
        vchSig(),
        nMessVersion(MessageVersion::MESS_VER_HASH)
    {}
    virtual ~CSignedMessage() {};
 
    // Sign-Verify message
    bool Sign(const CKey& key, const CKeyID& keyID);
    bool Sign(const std::string strSignKey);
    bool CheckSignature(const CKeyID& keyID) const;
 
    // Pure virtual functions (used in Sign-Verify functions)
    // Must be implemented in child classes
    virtual uint256 GetSignatureHash() const = 0;
    virtual std::string GetStrMessage() const = 0;
 
    // Setters and getters
    void SetVchSig(const std::vector<unsigned char>& vchSigIn) { vchSig = vchSigIn; }
    std::vector<unsigned char> GetVchSig() const { return vchSig; }
    std::string GetSignatureBase64() const;
 
    // Sign-Verify with BLS
    bool Sign(const CBLSSecretKey& sk);
    bool CheckSignature(const CBLSPublicKey& pk) const;
};
 
#endif // PIVX_MESSAGESIGNER_H