277 lines
8.1 KiB
C++
277 lines
8.1 KiB
C++
/*
|
|
* This file is part of the Flowee project
|
|
* Copyright (c) 2014-2015 The Bitcoin Core developers
|
|
* Copyright (C) 2019-2021 Tom Zander <tom@flowee.org>
|
|
*
|
|
* 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 3 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/>.
|
|
*/
|
|
|
|
#include "base58.h"
|
|
#include "hash.h"
|
|
#include "uint256.h"
|
|
|
|
#include <cassert>
|
|
#include <cstring>
|
|
|
|
/** All alphanumeric characters except for "0", "I", "O", and "l" */
|
|
static const char* pszBase58 = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz";
|
|
|
|
bool DecodeBase58(const char* psz, std::vector<unsigned char>& vch)
|
|
{
|
|
// Skip leading spaces.
|
|
while (*psz && isspace(*psz))
|
|
psz++;
|
|
// Skip and count leading '1's.
|
|
int zeroes = 0;
|
|
while (*psz == '1') {
|
|
zeroes++;
|
|
psz++;
|
|
}
|
|
// Allocate enough space in big-endian base256 representation.
|
|
std::vector<unsigned char> b256(strlen(psz) * 733 / 1000 + 1); // log(58) / log(256), rounded up.
|
|
// Process the characters.
|
|
while (*psz && !isspace(*psz)) {
|
|
// Decode base58 character
|
|
const char* ch = strchr(pszBase58, *psz);
|
|
if (ch == NULL)
|
|
return false;
|
|
// Apply "b256 = b256 * 58 + ch".
|
|
int carry = ch - pszBase58;
|
|
for (std::vector<unsigned char>::reverse_iterator it = b256.rbegin(); it != b256.rend(); ++it) {
|
|
carry += 58 * (*it);
|
|
*it = carry % 256;
|
|
carry /= 256;
|
|
}
|
|
assert(carry == 0);
|
|
psz++;
|
|
}
|
|
// Skip trailing spaces.
|
|
while (isspace(*psz))
|
|
psz++;
|
|
if (*psz != 0)
|
|
return false;
|
|
// Skip leading zeroes in b256.
|
|
std::vector<unsigned char>::iterator it = b256.begin();
|
|
while (it != b256.end() && *it == 0)
|
|
++it;
|
|
// Copy result into output vector.
|
|
vch.reserve(zeroes + (b256.end() - it));
|
|
vch.assign(zeroes, 0x00);
|
|
while (it != b256.end())
|
|
vch.push_back(*(it++));
|
|
return true;
|
|
}
|
|
|
|
std::string EncodeBase58(const unsigned char* pbegin, const unsigned char* pend)
|
|
{
|
|
// Skip & count leading zeroes.
|
|
int zeroes = 0;
|
|
while (pbegin != pend && *pbegin == 0) {
|
|
pbegin++;
|
|
zeroes++;
|
|
}
|
|
// Allocate enough space in big-endian base58 representation.
|
|
std::vector<unsigned char> b58((pend - pbegin) * 138 / 100 + 1); // log(256) / log(58), rounded up.
|
|
// Process the bytes.
|
|
while (pbegin != pend) {
|
|
int carry = *pbegin;
|
|
// Apply "b58 = b58 * 256 + ch".
|
|
for (std::vector<unsigned char>::reverse_iterator it = b58.rbegin(); it != b58.rend(); ++it) {
|
|
carry += 256 * (*it);
|
|
*it = carry % 58;
|
|
carry /= 58;
|
|
}
|
|
assert(carry == 0);
|
|
pbegin++;
|
|
}
|
|
// Skip leading zeroes in base58 result.
|
|
std::vector<unsigned char>::iterator it = b58.begin();
|
|
while (it != b58.end() && *it == 0)
|
|
++it;
|
|
// Translate the result into a string.
|
|
std::string str;
|
|
str.reserve(zeroes + (b58.end() - it));
|
|
str.assign(zeroes, '1');
|
|
while (it != b58.end())
|
|
str += pszBase58[*(it++)];
|
|
return str;
|
|
}
|
|
|
|
std::string EncodeBase58(const std::vector<unsigned char>& vch)
|
|
{
|
|
return EncodeBase58(&vch[0], &vch[0] + vch.size());
|
|
}
|
|
|
|
bool DecodeBase58(const std::string& str, std::vector<unsigned char>& vchRet)
|
|
{
|
|
return DecodeBase58(str.c_str(), vchRet);
|
|
}
|
|
|
|
std::string EncodeBase58Check(const std::vector<unsigned char>& vchIn)
|
|
{
|
|
// add 4-byte hash check to the end
|
|
std::vector<unsigned char> vch(vchIn);
|
|
uint256 hash = Hash(vch.begin(), vch.end());
|
|
vch.insert(vch.end(), (unsigned char*)&hash, (unsigned char*)&hash + 4);
|
|
return EncodeBase58(vch);
|
|
}
|
|
|
|
bool DecodeBase58Check(const char* psz, std::vector<unsigned char>& vchRet)
|
|
{
|
|
if (!DecodeBase58(psz, vchRet) ||
|
|
(vchRet.size() < 4)) {
|
|
vchRet.clear();
|
|
return false;
|
|
}
|
|
// re-calculate the checksum, insure it matches the included 4-byte checksum
|
|
uint256 hash = Hash(vchRet.begin(), vchRet.end() - 4);
|
|
if (memcmp(&hash, &vchRet.end()[-4], 4) != 0) {
|
|
vchRet.clear();
|
|
return false;
|
|
}
|
|
vchRet.resize(vchRet.size() - 4);
|
|
return true;
|
|
}
|
|
|
|
bool DecodeBase58Check(const std::string& str, std::vector<unsigned char>& vchRet)
|
|
{
|
|
return DecodeBase58Check(str.c_str(), vchRet);
|
|
}
|
|
|
|
CBase58Data::CBase58Data()
|
|
{
|
|
vchVersion.clear();
|
|
vchData.clear();
|
|
}
|
|
|
|
void CBase58Data::SetData(const std::vector<unsigned char>& vchVersionIn, const void* pdata, size_t nSize)
|
|
{
|
|
vchVersion = vchVersionIn;
|
|
vchData.resize(nSize);
|
|
if (!vchData.empty())
|
|
memcpy(&vchData[0], pdata, nSize);
|
|
}
|
|
|
|
void CBase58Data::setData(const uint160 &key, CBase58Data::KeyType type, Chain chain)
|
|
{
|
|
constexpr uint8_t TestPubkeyType = 111;
|
|
constexpr uint8_t TestScriptType = 196;
|
|
constexpr uint8_t PubkeyType = 0;
|
|
constexpr uint8_t ScriptType = 5;
|
|
|
|
uint8_t ver;
|
|
if (type == PubkeyType)
|
|
ver = chain == Mainnet ? PubkeyType : TestPubkeyType;
|
|
else
|
|
ver = chain == Mainnet ? ScriptType : TestScriptType;
|
|
|
|
std::vector<uint8_t> version(1, ver);
|
|
SetData(version, key.begin(), 20);
|
|
}
|
|
|
|
#ifndef NPRIVKEY
|
|
# include <primitives/key.h>
|
|
void CBase58Data::setData(const CKey &key, Chain chain)
|
|
{
|
|
constexpr uint8_t PrivkeyType = 128;
|
|
constexpr uint8_t TestPrivkeyType = 239;
|
|
uint8_t ver = chain == Mainnet ? PrivkeyType : TestPrivkeyType;
|
|
|
|
std::vector<uint8_t> version(1, ver);
|
|
SetData(version, key.begin(), 32);
|
|
if (key.IsCompressed())
|
|
vchData.push_back(1);
|
|
}
|
|
#endif
|
|
|
|
void CBase58Data::SetData(const std::vector<unsigned char>& vchVersionIn, const unsigned char* pbegin, const unsigned char* pend)
|
|
{
|
|
SetData(vchVersionIn, (void*)pbegin, pend - pbegin);
|
|
}
|
|
|
|
bool CBase58Data::SetString(const char* psz, unsigned int nVersionBytes)
|
|
{
|
|
std::vector<unsigned char> vchTemp;
|
|
bool rc58 = DecodeBase58Check(psz, vchTemp);
|
|
if ((!rc58) || (vchTemp.size() < nVersionBytes)) {
|
|
vchData.clear();
|
|
vchVersion.clear();
|
|
return false;
|
|
}
|
|
vchVersion.assign(vchTemp.begin(), vchTemp.begin() + nVersionBytes);
|
|
vchData.resize(vchTemp.size() - nVersionBytes);
|
|
if (!vchData.empty())
|
|
memcpy(&vchData[0], &vchTemp[nVersionBytes], vchData.size());
|
|
return true;
|
|
}
|
|
|
|
bool CBase58Data::SetString(const std::string& str)
|
|
{
|
|
return SetString(str.c_str());
|
|
}
|
|
|
|
std::string CBase58Data::ToString() const
|
|
{
|
|
std::vector<unsigned char> vch = vchVersion;
|
|
vch.insert(vch.end(), vchData.begin(), vchData.end());
|
|
return EncodeBase58Check(vch);
|
|
}
|
|
|
|
int CBase58Data::CompareTo(const CBase58Data& b58) const
|
|
{
|
|
if (vchVersion < b58.vchVersion)
|
|
return -1;
|
|
if (vchVersion > b58.vchVersion)
|
|
return 1;
|
|
if (vchData < b58.vchData)
|
|
return -1;
|
|
if (vchData > b58.vchData)
|
|
return 1;
|
|
return 0;
|
|
}
|
|
|
|
bool CBase58Data::isMainnetPkh() const
|
|
{
|
|
return vchVersion.size() == 1 && vchVersion.at(0) == 0 && vchData.size() == 20;
|
|
}
|
|
|
|
bool CBase58Data::isMainnetSh() const
|
|
{
|
|
return vchVersion.size() == 1 && vchVersion.at(0) == 5 && vchData.size() == 20;
|
|
}
|
|
|
|
bool CBase58Data::isMainnetPrivKey() const
|
|
{
|
|
if (vchVersion.size() != 1 || vchVersion.at(0) != 128)
|
|
return false;
|
|
return vchData.size() == 32 || (vchData.size() == 33 && vchData.at(32) == 1);
|
|
}
|
|
|
|
bool CBase58Data::isTestnetPkh() const
|
|
{
|
|
return vchVersion.size() == 1 && vchVersion.at(0) == 111 && vchData.size() == 20;
|
|
}
|
|
|
|
bool CBase58Data::isTestnetSh() const
|
|
{
|
|
return vchVersion.size() == 1 && vchVersion.at(0) == 196 && vchData.size() == 20;
|
|
}
|
|
|
|
bool CBase58Data::isTestnetPrivKey() const
|
|
{
|
|
if (vchVersion.size() != 1 || vchVersion.at(0) != 239)
|
|
return false;
|
|
return vchData.size() == 32 || (vchData.size() == 33 && vchData.at(32) == 1);
|
|
}
|