src/base58.h

   1 // Copyright (c) 2009-2010 Satoshi Nakamoto
   2 // Distributed under the MIT/X11 software license, see the accompanying
   3 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
   4
   5
   6 //
   7 // Why base-58 instead of standard base-64 encoding?
   8 // - Don't want 0OIl characters that look the same in some fonts and
   9 //      could be used to create visually identical looking account numbers.
  10 // - A string with non-alphanumeric characters is not as easily accepted as an account number.
  11 // - E-mail usually won't line-break if there's no punctuation to break at.
  12 // - Doubleclicking selects the whole number as one word if it's all alphanumeric.
  13 //
  14 #ifndef BITCOIN_BASE58_H
  15 #define BITCOIN_BASE58_H
  16
  17 #include <string>
  18 #include <vector>
  19 #include "bignum.h"
  20
  21 static const char* pszBase58 = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz";
  22
  23
  24 inline std::string EncodeBase58(const unsigned char* pbegin, const unsigned char* pend)
  25 {
  26     CAutoBN_CTX pctx;
  27     CBigNum bn58 = 58;
  28     CBigNum bn0 = 0;
  29
  30     // Convert big endian data to little endian
  31     // Extra zero at the end make sure bignum will interpret as a positive number
  32     std::vector<unsigned char> vchTmp(pend-pbegin+1, 0);
  33     reverse_copy(pbegin, pend, vchTmp.begin());
  34
  35     // Convert little endian data to bignum
  36     CBigNum bn;
  37     bn.setvch(vchTmp);
  38
  39     // Convert bignum to std::string
  40     std::string str;
  41     // Expected size increase from base58 conversion is approximately 137%
  42     // use 138% to be safe
  43     str.reserve((pend - pbegin) * 138 / 100 + 1);
  44     CBigNum dv;
  45     CBigNum rem;
  46     while (bn > bn0)
  47     {
  48         if (!BN_div(&dv, &rem, &bn, &bn58, pctx))
  49             throw bignum_error("EncodeBase58 : BN_div failed");
  50         bn = dv;
  51         unsigned int c = rem.getulong();
  52         str += pszBase58[c];
  53     }
  54
  55     // Leading zeroes encoded as base58 zeros
  56     for (const unsigned char* p = pbegin; p < pend && *p == 0; p++)
  57         str += pszBase58[0];
  58
  59     // Convert little endian std::string to big endian
  60     reverse(str.begin(), str.end());
  61     return str;
  62 }
  63
  64 inline std::string EncodeBase58(const std::vector<unsigned char>& vch)
  65 {
  66     return EncodeBase58(&vch[0], &vch[0] + vch.size());
  67 }
  68
  69 inline bool DecodeBase58(const char* psz, std::vector<unsigned char>& vchRet)
  70 {
  71     CAutoBN_CTX pctx;
  72     vchRet.clear();
  73     CBigNum bn58 = 58;
  74     CBigNum bn = 0;
  75     CBigNum bnChar;
  76     while (isspace(*psz))
  77         psz++;
  78
  79     // Convert big endian string to bignum
  80     for (const char* p = psz; *p; p++)
  81     {
  82         const char* p1 = strchr(pszBase58, *p);
  83         if (p1 == NULL)
  84         {
  85             while (isspace(*p))
  86                 p++;
  87             if (*p != '\0')
  88                 return false;
  89             break;
  90         }
  91         bnChar.setulong(p1 - pszBase58);
  92         if (!BN_mul(&bn, &bn, &bn58, pctx))
  93             throw bignum_error("DecodeBase58 : BN_mul failed");
  94         bn += bnChar;
  95     }
  96
  97     // Get bignum as little endian data
  98     std::vector<unsigned char> vchTmp = bn.getvch();
  99
 100     // Trim off sign byte if present
 101     if (vchTmp.size() >= 2 && vchTmp.end()[-1] == 0 && vchTmp.end()[-2] >= 0x80)
 102         vchTmp.erase(vchTmp.end()-1);
 103
 104     // Restore leading zeros
 105     int nLeadingZeros = 0;
 106     for (const char* p = psz; *p == pszBase58[0]; p++)
 107         nLeadingZeros++;
 108     vchRet.assign(nLeadingZeros + vchTmp.size(), 0);
 109
 110     // Convert little endian data to big endian
 111     reverse_copy(vchTmp.begin(), vchTmp.end(), vchRet.end() - vchTmp.size());
 112     return true;
 113 }
 114
 115 inline bool DecodeBase58(const std::string& str, std::vector<unsigned char>& vchRet)
 116 {
 117     return DecodeBase58(str.c_str(), vchRet);
 118 }
 119
 120
 121
 122
 123
 124 inline std::string EncodeBase58Check(const std::vector<unsigned char>& vchIn)
 125 {
 126     // add 4-byte hash check to the end
 127     std::vector<unsigned char> vch(vchIn);
 128     uint256 hash = Hash(vch.begin(), vch.end());
 129     vch.insert(vch.end(), (unsigned char*)&hash, (unsigned char*)&hash + 4);
 130     return EncodeBase58(vch);
 131 }
 132
 133 inline bool DecodeBase58Check(const char* psz, std::vector<unsigned char>& vchRet)
 134 {
 135     if (!DecodeBase58(psz, vchRet))
 136         return false;
 137     if (vchRet.size() < 4)
 138     {
 139         vchRet.clear();
 140         return false;
 141     }
 142     uint256 hash = Hash(vchRet.begin(), vchRet.end()-4);
 143     if (memcmp(&hash, &vchRet.end()[-4], 4) != 0)
 144     {
 145         vchRet.clear();
 146         return false;
 147     }
 148     vchRet.resize(vchRet.size()-4);
 149     return true;
 150 }
 151
 152 inline bool DecodeBase58Check(const std::string& str, std::vector<unsigned char>& vchRet)
 153 {
 154     return DecodeBase58Check(str.c_str(), vchRet);
 155 }
 156
 157
 158
 159
 160
 161
 162 #define ADDRESSVERSION   ((unsigned char)(fTestNet ? 111 : 0))
 163
 164 inline std::string Hash160ToAddress(uint160 hash160)
 165 {
 166     // add 1-byte version number to the front
 167     std::vector<unsigned char> vch(1, ADDRESSVERSION);
 168     vch.insert(vch.end(), UBEGIN(hash160), UEND(hash160));
 169     return EncodeBase58Check(vch);
 170 }
 171
 172 inline bool AddressToHash160(const char* psz, uint160& hash160Ret)
 173 {
 174     std::vector<unsigned char> vch;
 175     if (!DecodeBase58Check(psz, vch))
 176         return false;
 177     if (vch.empty())
 178         return false;
 179     unsigned char nVersion = vch[0];
 180     if (vch.size() != sizeof(hash160Ret) + 1)
 181         return false;
 182     memcpy(&hash160Ret, &vch[1], sizeof(hash160Ret));
 183     return (nVersion <= ADDRESSVERSION);
 184 }
 185
 186 inline bool AddressToHash160(const std::string& str, uint160& hash160Ret)
 187 {
 188     return AddressToHash160(str.c_str(), hash160Ret);
 189 }
 190
 191 inline bool IsValidBitcoinAddress(const char* psz)
 192 {
 193     uint160 hash160;
 194     return AddressToHash160(psz, hash160);
 195 }
 196
 197 inline bool IsValidBitcoinAddress(const std::string& str)
 198 {
 199     return IsValidBitcoinAddress(str.c_str());
 200 }
 201
 202
 203
 204
 205 inline std::string PubKeyToAddress(const std::vector<unsigned char>& vchPubKey)
 206 {
 207     return Hash160ToAddress(Hash160(vchPubKey));
 208 }
 209
 210 #endif