src/base58.cpp

   1 // Copyright (c) 2009-2010 Satoshi Nakamoto
   2 // Copyright (c) 2009-2012 The Bitcoin Developers
   3 // Distributed under the MIT/X11 software license, see the accompanying
   4 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5
   6
   7 //
   8 // Why base-58 instead of standard base-64 encoding?
   9 // - Don't want 0OIl characters that look the same in some fonts and
  10 //      could be used to create visually identical looking account numbers.
  11 // - A string with non-alphanumeric characters is not as easily accepted as an account number.
  12 // - E-mail usually won't line-break if there's no punctuation to break at.
  13 // - Double-clicking selects the whole number as one word if it's all alphanumeric.
  14 //
  15
  16 #include <string>
  17 #include <vector>
  18 #include <openssl/crypto.h> // for OPENSSL_cleanse()
  19 #include "bignum.h"
  20 #include "key.h"
  21 #include "script.h"
  22 #include "base58.h"
  23
  24 static const char* pszBase58 = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz";
  25
  26 // Encode a byte sequence as a base58-encoded string
  27 std::string EncodeBase58(const unsigned char* pbegin, const unsigned char* pend)
  28 {
  29     CAutoBN_CTX pctx;
  30     CBigNum bn58 = 58;
  31     CBigNum bn0 = 0;
  32
  33     // Convert big endian data to little endian
  34     // Extra zero at the end make sure bignum will interpret as a positive number
  35     std::vector<unsigned char> vchTmp(pend-pbegin+1, 0);
  36     reverse_copy(pbegin, pend, vchTmp.begin());
  37
  38     // Convert little endian data to bignum
  39     CBigNum bn;
  40     bn.setvch(vchTmp);
  41
  42     // Convert bignum to std::string
  43     std::string str;
  44     // Expected size increase from base58 conversion is approximately 137%
  45     // use 138% to be safe
  46     str.reserve((pend - pbegin) * 138 / 100 + 1);
  47     CBigNum dv;
  48     CBigNum rem;
  49     while (bn > bn0)
  50     {
  51         if (!BN_div(&dv, &rem, &bn, &bn58, pctx))
  52             throw bignum_error("EncodeBase58 : BN_div failed");
  53         bn = dv;
  54         unsigned int c = rem.getuint32();
  55         str += pszBase58[c];
  56     }
  57
  58     // Leading zeroes encoded as base58 zeros
  59     for (const unsigned char* p = pbegin; p < pend && *p == 0; p++)
  60         str += pszBase58[0];
  61
  62     // Convert little endian std::string to big endian
  63     reverse(str.begin(), str.end());
  64     return str;
  65 }
  66
  67 // Encode a byte vector as a base58-encoded string
  68 std::string EncodeBase58(const std::vector<unsigned char>& vch)
  69 {
  70     return EncodeBase58(&vch[0], &vch[0] + vch.size());
  71 }
  72
  73 // Decode a base58-encoded string psz into byte vector vchRet
  74 // returns true if decoding is successful
  75 bool DecodeBase58(const char* psz, std::vector<unsigned char>& vchRet)
  76 {
  77     CAutoBN_CTX pctx;
  78     vchRet.clear();
  79     CBigNum bn58 = 58;
  80     CBigNum bn = 0;
  81     CBigNum bnChar;
  82     while (isspace(*psz))
  83         psz++;
  84
  85     // Convert big endian string to bignum
  86     for (const char* p = psz; *p; p++)
  87     {
  88         const char* p1 = strchr(pszBase58, *p);
  89         if (p1 == NULL)
  90         {
  91             while (isspace(*p))
  92                 p++;
  93             if (*p != '\0')
  94                 return false;
  95             break;
  96         }
  97         bnChar.setuint32((uint32_t)(p1 - pszBase58));
  98         if (!BN_mul(&bn, &bn, &bn58, pctx))
  99             throw bignum_error("DecodeBase58 : BN_mul failed");
 100         bn += bnChar;
 101     }
 102
 103     // Get bignum as little endian data
 104     std::vector<unsigned char> vchTmp = bn.getvch();
 105
 106     // Trim off sign byte if present
 107     if (vchTmp.size() >= 2 && vchTmp.end()[-1] == 0 && vchTmp.end()[-2] >= 0x80)
 108         vchTmp.erase(vchTmp.end()-1);
 109
 110     // Restore leading zeros
 111     int nLeadingZeros = 0;
 112     for (const char* p = psz; *p == pszBase58[0]; p++)
 113         nLeadingZeros++;
 114     vchRet.assign(nLeadingZeros + vchTmp.size(), 0);
 115
 116     // Convert little endian data to big endian
 117     reverse_copy(vchTmp.begin(), vchTmp.end(), vchRet.end() - vchTmp.size());
 118     return true;
 119 }
 120
 121 // Decode a base58-encoded string str into byte vector vchRet
 122 // returns true if decoding is successful
 123 bool DecodeBase58(const std::string& str, std::vector<unsigned char>& vchRet)
 124 {
 125     return DecodeBase58(str.c_str(), vchRet);
 126 }
 127
 128 // Encode a byte vector to a base58-encoded string, including checksum
 129 std::string EncodeBase58Check(const std::vector<unsigned char>& vchIn)
 130 {
 131     // add 4-byte hash check to the end
 132     std::vector<unsigned char> vch(vchIn);
 133     uint256 hash = Hash(vch.begin(), vch.end());
 134     vch.insert(vch.end(), (unsigned char*)&hash, (unsigned char*)&hash + 4);
 135     return EncodeBase58(vch);
 136 }
 137
 138 // Decode a base58-encoded string psz that includes a checksum, into byte vector vchRet
 139 // returns true if decoding is successful
 140 bool DecodeBase58Check(const char* psz, std::vector<unsigned char>& vchRet)
 141 {
 142     if (!DecodeBase58(psz, vchRet))
 143         return false;
 144     if (vchRet.size() < 4)
 145     {
 146         vchRet.clear();
 147         return false;
 148     }
 149     uint256 hash = Hash(vchRet.begin(), vchRet.end()-4);
 150     if (memcmp(&hash, &vchRet.end()[-4], 4) != 0)
 151     {
 152         vchRet.clear();
 153         return false;
 154     }
 155     vchRet.resize(vchRet.size()-4);
 156     return true;
 157 }
 158
 159 // Decode a base58-encoded string str that includes a checksum, into byte vector vchRet
 160 // returns true if decoding is successful
 161 bool DecodeBase58Check(const std::string& str, std::vector<unsigned char>& vchRet)
 162 {
 163     return DecodeBase58Check(str.c_str(), vchRet);
 164 }
 165
 166     CBase58Data::CBase58Data()
 167     {
 168         nVersion = 0;
 169         vchData.clear();
 170     }
 171
 172     CBase58Data::~CBase58Data()
 173     {
 174         // zero the memory, as it may contain sensitive data
 175         if (!vchData.empty())
 176             OPENSSL_cleanse(&vchData[0], vchData.size());
 177     }
 178
 179     void CBase58Data::SetData(int nVersionIn, const void* pdata, size_t nSize)
 180     {
 181         nVersion = nVersionIn;
 182         vchData.resize(nSize);
 183         if (!vchData.empty())
 184             memcpy(&vchData[0], pdata, nSize);
 185     }
 186
 187     const std::vector<unsigned char> &CBase58Data::GetData() const
 188     {
 189         return vchData;
 190     }
 191
 192     void CBase58Data::SetData(int nVersionIn, const unsigned char *pbegin, const unsigned char *pend)
 193     {
 194         SetData(nVersionIn, (void*)pbegin, pend - pbegin);
 195     }
 196
 197     bool CBase58Data::SetString(const char* psz)
 198     {
 199         std::vector<unsigned char> vchTemp;
 200         DecodeBase58Check(psz, vchTemp);
 201         if (vchTemp.empty())
 202         {
 203             vchData.clear();
 204             nVersion = 0;
 205             return false;
 206         }
 207         nVersion = vchTemp[0];
 208         vchData.resize(vchTemp.size() - 1);
 209         if (!vchData.empty())
 210             memcpy(&vchData[0], &vchTemp[1], vchData.size());
 211         OPENSSL_cleanse(&vchTemp[0], vchData.size());
 212         return true;
 213     }
 214
 215     bool CBase58Data::SetString(const std::string& str)
 216     {
 217         return SetString(str.c_str());
 218     }
 219
 220     std::string CBase58Data::ToString() const
 221     {
 222         std::vector<unsigned char> vch(1, nVersion);
 223         vch.insert(vch.end(), vchData.begin(), vchData.end());
 224         return EncodeBase58Check(vch);
 225     }
 226
 227     int CBase58Data::CompareTo(const CBase58Data& b58) const
 228     {
 229         if (nVersion < b58.nVersion) return -1;
 230         if (nVersion > b58.nVersion) return  1;
 231         if (vchData < b58.vchData)   return -1;
 232         if (vchData > b58.vchData)   return  1;
 233         return 0;
 234     }
 235
 236     bool CBitcoinAddress::Set(const CKeyID &id) {
 237         SetData(fTestNet ? PUBKEY_ADDRESS_TEST : PUBKEY_ADDRESS, &id, 20);
 238         return true;
 239     }
 240
 241     bool CBitcoinAddress::Set(const CScriptID &id) {
 242         SetData(fTestNet ? SCRIPT_ADDRESS_TEST : SCRIPT_ADDRESS, &id, 20);
 243         return true;
 244     }
 245
 246     bool CBitcoinAddress::Set(const CTxDestination &dest)
 247     {
 248         return boost::apply_visitor(CBitcoinAddressVisitor(this), dest);
 249     }
 250
 251     bool CBitcoinAddress::Set(const CMalleablePubKey &mpk) {
 252         std::vector<unsigned char> vchPubkeyPair = mpk.Raw();
 253         SetData(fTestNet ? PUBKEY_PAIR_ADDRESS_TEST : PUBKEY_PAIR_ADDRESS, &vchPubkeyPair[0], 68);
 254         return true;
 255     }
 256
 257     bool CBitcoinAddress::Set(const CBitcoinAddress &dest)
 258     {
 259         nVersion = dest.nVersion;
 260         vchData = dest.vchData;
 261         return true;
 262     }
 263
 264     bool CBitcoinAddress::IsValid() const
 265     {
 266         unsigned int nExpectedSize = 20;
 267         bool fExpectTestNet = false;
 268         bool fSimple = true;
 269         switch(nVersion)
 270         {
 271             case PUBKEY_PAIR_ADDRESS:
 272                 nExpectedSize = 68; // Serialized pair of public keys
 273                 fExpectTestNet = false;
 274                 fSimple = false;
 275                 break;
 276             case PUBKEY_ADDRESS:
 277                 nExpectedSize = 20; // Hash of public key
 278                 fExpectTestNet = false;
 279                 break;
 280             case SCRIPT_ADDRESS:
 281                 nExpectedSize = 20; // Hash of CScript
 282                 fExpectTestNet = false;
 283                 break;
 284
 285             case PUBKEY_PAIR_ADDRESS_TEST:
 286                 nExpectedSize = 68;
 287                 fExpectTestNet = true;
 288                 fSimple = false;
 289                 break;
 290             case PUBKEY_ADDRESS_TEST:
 291                 nExpectedSize = 20;
 292                 fExpectTestNet = true;
 293                 break;
 294             case SCRIPT_ADDRESS_TEST:
 295                 nExpectedSize = 20;
 296                 fExpectTestNet = true;
 297                 break;
 298
 299             default:
 300                 return false;
 301         }
 302
 303         // Basic format sanity check
 304         bool fSeemsSane = (fExpectTestNet == fTestNet && vchData.size() == nExpectedSize);
 305
 306         if (fSeemsSane && !fSimple)
 307         {
 308             // Perform dditional checking
 309             //    for pubkey pair addresses
 310             CMalleablePubKey mpk;
 311             mpk.setvch(vchData);
 312             return mpk.IsValid();
 313         }
 314         else
 315             return fSeemsSane;
 316     }
 317
 318     CTxDestination CBitcoinAddress::Get() const {
 319         if (!IsValid())
 320             return CNoDestination();
 321         switch (nVersion) {
 322         case PUBKEY_ADDRESS:
 323         case PUBKEY_ADDRESS_TEST: {
 324             uint160 id;
 325             memcpy(&id, &vchData[0], 20);
 326             return CKeyID(id);
 327         }
 328         case SCRIPT_ADDRESS:
 329         case SCRIPT_ADDRESS_TEST: {
 330             uint160 id;
 331             memcpy(&id, &vchData[0], 20);
 332             return CScriptID(id);
 333         }
 334         }
 335         return CNoDestination();
 336     }
 337
 338     bool CBitcoinAddress::GetKeyID(CKeyID &keyID) const {
 339         if (!IsValid())
 340             return false;
 341         switch (nVersion) {
 342         case PUBKEY_ADDRESS:
 343         case PUBKEY_ADDRESS_TEST: {
 344             uint160 id;
 345             memcpy(&id, &vchData[0], 20);
 346             keyID = CKeyID(id);
 347             return true;
 348         }
 349         case PUBKEY_PAIR_ADDRESS:
 350         case PUBKEY_PAIR_ADDRESS_TEST:
 351         {
 352             CMalleablePubKey mPubKey;
 353             mPubKey.setvch(vchData);
 354             keyID = mPubKey.GetID();
 355             return true;
 356         }
 357         default: return false;
 358         }
 359     }
 360
 361     bool CBitcoinAddress::IsScript() const {
 362         if (!IsValid())
 363             return false;
 364         switch (nVersion) {
 365         case SCRIPT_ADDRESS:
 366         case SCRIPT_ADDRESS_TEST: {
 367             return true;
 368         }
 369         default: return false;
 370         }
 371     }
 372
 373     bool CBitcoinAddress::IsPubKey() const {
 374         if (!IsValid())
 375             return false;
 376         switch (nVersion) {
 377         case PUBKEY_ADDRESS:
 378         case PUBKEY_ADDRESS_TEST: {
 379             return true;
 380         }
 381         default: return false;
 382         }
 383     }
 384
 385     bool CBitcoinAddress::IsPair() const {
 386         if (!IsValid())
 387             return false;
 388         switch (nVersion) {
 389         case PUBKEY_PAIR_ADDRESS:
 390         case PUBKEY_PAIR_ADDRESS_TEST: {
 391             return true;
 392         }
 393         default: return false;
 394         }
 395     }
 396
 397     void CBitcoinSecret::SetSecret(const CSecret& vchSecret, bool fCompressed)
 398     {
 399         assert(vchSecret.size() == 32);
 400         SetData(128 + (fTestNet ? CBitcoinAddress::PUBKEY_ADDRESS_TEST : CBitcoinAddress::PUBKEY_ADDRESS), &vchSecret[0], vchSecret.size());
 401         if (fCompressed)
 402             vchData.push_back(1);
 403     }
 404
 405     CSecret CBitcoinSecret::GetSecret(bool &fCompressedOut)
 406     {
 407         CSecret vchSecret;
 408         vchSecret.resize(32);
 409         memcpy(&vchSecret[0], &vchData[0], 32);
 410         fCompressedOut = vchData.size() == 33;
 411         return vchSecret;
 412     }
 413
 414     bool CBitcoinSecret::IsValid() const
 415     {
 416         bool fExpectTestNet = false;
 417         switch(nVersion)
 418         {
 419             case (128 + CBitcoinAddress::PUBKEY_ADDRESS):
 420                 break;
 421
 422             case (128 + CBitcoinAddress::PUBKEY_ADDRESS_TEST):
 423                 fExpectTestNet = true;
 424                 break;
 425
 426             default:
 427                 return false;
 428         }
 429         return fExpectTestNet == fTestNet && (vchData.size() == 32 || (vchData.size() == 33 && vchData[32] == 1));
 430     }
 431
 432     bool CBitcoinSecret::SetString(const char* pszSecret)
 433     {
 434         return CBase58Data::SetString(pszSecret) && IsValid();
 435     }
 436
 437     bool CBitcoinSecret::SetString(const std::string& strSecret)
 438     {
 439         return SetString(strSecret.c_str());
 440     }
 441
 442     CBitcoinSecret::CBitcoinSecret(const CSecret& vchSecret, bool fCompressed)
 443     {
 444         SetSecret(vchSecret, fCompressed);
 445     }
 446
 447