base58.h

Go to the documentation of this file.

// Copyright (c) 2009-2010 Satoshi Nakamoto
// Copyright (c) 2009-2014 The Bitcoin developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef BITCOIN_BASE58_H
#define BITCOIN_BASE58_H
 
#include "chainparams.h"
#include "key.h"
#include "pubkey.h"
#include "script/script.h"
#include "script/standard.h"
 
#include <string>
#include <vector>
 
std::string EncodeBase58(const unsigned char* pbegin, const unsigned char* pend);
 
std::string EncodeBase58(const std::vector<unsigned char>& vch);
 
bool DecodeBase58(const char* psz, std::vector<unsigned char>& vchRet);
 
std::string DecodeBase58(const char* psz);
 
bool DecodeBase58(const std::string& str, std::vector<unsigned char>& vchRet);
 
std::string EncodeBase58Check(const std::vector<unsigned char>& vchIn);
 
inline bool DecodeBase58Check(const char* psz, std::vector<unsigned char>& vchRet);
 
inline bool DecodeBase58Check(const std::string& str, std::vector<unsigned char>& vchRet);
 
class CBase58Data
{
protected:
    std::vector<unsigned char> vchVersion;
 
    typedef std::vector<unsigned char, zero_after_free_allocator<unsigned char> > vector_uchar;
    vector_uchar vchData;
 
    CBase58Data();
    void SetData(const std::vector<unsigned char>& vchVersionIn, const void* pdata, size_t nSize);
    void SetData(const std::vector<unsigned char>& vchVersionIn, const unsigned char* pbegin, const unsigned char* pend);
 
public:
    bool SetString(const char* psz, unsigned int nVersionBytes = 1);
    bool SetString(const std::string& str);
    std::string ToString() const;
    int CompareTo(const CBase58Data& b58) const;
 
    bool operator==(const CBase58Data& b58) const { return CompareTo(b58) == 0; }
    bool operator<=(const CBase58Data& b58) const { return CompareTo(b58) <= 0; }
    bool operator>=(const CBase58Data& b58) const { return CompareTo(b58) >= 0; }
    bool operator<(const CBase58Data& b58) const { return CompareTo(b58) < 0; }
    bool operator>(const CBase58Data& b58) const { return CompareTo(b58) > 0; }
};
 
class CBitcoinAddress : public CBase58Data
{
public:
    bool Set(const CKeyID& id);
    bool Set(const CScriptID& id);
    bool Set(const CTxDestination& dest);
    bool IsValid() const;
    bool IsValid(const CChainParams& params) const;
 
    CBitcoinAddress() {}
    CBitcoinAddress(const CTxDestination& dest) { Set(dest); }
    CBitcoinAddress(const std::string& strAddress) { SetString(strAddress); }
    CBitcoinAddress(const char* pszAddress) { SetString(pszAddress); }
 
    CTxDestination Get() const;
    bool GetKeyID(CKeyID& keyID) const;
    bool IsScript() const;
};
 
class CBitcoinSecret : public CBase58Data
{
public:
    void SetKey(const CKey& vchSecret);
    CKey GetKey();
    bool IsValid() const;
    bool SetString(const char* pszSecret);
    bool SetString(const std::string& strSecret);
 
    CBitcoinSecret(const CKey& vchSecret) { SetKey(vchSecret); }
    CBitcoinSecret() {}
};
 
template <typename K, int Size, CChainParams::Base58Type Type>
class CBitcoinExtKeyBase : public CBase58Data
{
public:
    void SetKey(const K& key)
    {
        unsigned char vch[Size];
        key.Encode(vch);
        SetData(Params().Base58Prefix(Type), vch, vch + Size);
    }
 
    K GetKey()
    {
        K ret;
        ret.Decode(&vchData[0], &vchData[Size]);
        return ret;
    }
 
    CBitcoinExtKeyBase(const K& key)
    {
        SetKey(key);
    }
 
    CBitcoinExtKeyBase() {}
};
 
typedef CBitcoinExtKeyBase<CExtKey, 74, CChainParams::EXT_SECRET_KEY> CBitcoinExtKey;
typedef CBitcoinExtKeyBase<CExtPubKey, 74, CChainParams::EXT_PUBLIC_KEY> CBitcoinExtPubKey;
 
#endif // BITCOIN_BASE58_H