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 class CBitcoinAddress
 163 {
 164 protected:
 165     unsigned char nVersion;
 166     std::vector<unsigned char> vchData;
 167
 168 public:
 169     bool SetAddress(const uint160& hash160)
 170     {
 171         nVersion = fTestNet ? 111 : 0;
 172         vchData.resize(20);
 173         memcpy(&vchData[0], &hash160, 20);
 174         return true;
 175     }
 176
 177     bool SetAddress(const char* pszAddress)
 178     {
 179         std::vector<unsigned char> vchTemp;
 180         DecodeBase58Check(pszAddress, vchTemp);
 181         if (vchTemp.empty())
 182         {
 183             vchData.clear();
 184             nVersion = 0;
 185             return false;
 186         }
 187         nVersion = vchTemp[0];
 188         vchData.resize(vchTemp.size() - 1);
 189         memcpy(&vchData[0], &vchTemp[1], vchData.size());
 190         return true;
 191     }
 192
 193     bool SetAddress(const std::string& strAddress)
 194     {
 195         return SetAddress(strAddress.c_str());
 196     }
 197
 198     bool SetAddress(const std::vector<unsigned char>& vchPubKey)
 199     {
 200         return SetAddress(Hash160(vchPubKey));
 201     }
 202
 203     bool IsValid() const
 204     {
 205         int nExpectedSize = 20;
 206         bool fExpectTestNet = false;
 207         switch(nVersion)
 208         {
 209             case 0:
 210                 break;
 211
 212             case 111:
 213                 fExpectTestNet = true;
 214                 break;
 215
 216             default:
 217                 return false;
 218         }
 219         return fExpectTestNet == fTestNet && vchData.size() == nExpectedSize;
 220     }
 221
 222     CBitcoinAddress()
 223     {
 224         nVersion = 0;
 225         vchData.clear();
 226     }
 227
 228     CBitcoinAddress(uint160 hash160In)
 229     {
 230         SetAddress(hash160In);
 231     }
 232
 233     CBitcoinAddress(const std::vector<unsigned char>& vchPubKey)
 234     {
 235         SetAddress(vchPubKey);
 236     }
 237
 238     CBitcoinAddress(const std::string& strAddress)
 239     {
 240         SetAddress(strAddress);
 241     }
 242
 243     CBitcoinAddress(const char* pszAddress)
 244     {
 245         SetAddress(pszAddress);
 246     }
 247
 248     std::string ToString() const
 249     {
 250         std::vector<unsigned char> vch(1, nVersion);
 251         vch.insert(vch.end(), vchData.begin(), vchData.end());
 252         return EncodeBase58Check(vch);
 253     }
 254
 255     uint160 GetHash160() const
 256     {
 257         assert(vchData.size() == 20);
 258         uint160 hash160;
 259         memcpy(&hash160, &vchData[0], 20);
 260         return hash160;
 261     }
 262
 263     int CompareTo(const CBitcoinAddress& address) const
 264     {
 265         if (nVersion < address.nVersion) return -1;
 266         if (nVersion < address.nVersion) return  1;
 267         if (vchData < address.vchData)   return -1;
 268         if (vchData > address.vchData)   return  1;
 269         return 0;
 270     }
 271
 272     bool operator==(const CBitcoinAddress& address) const { return CompareTo(address) == 0; }
 273     bool operator<=(const CBitcoinAddress& address) const { return CompareTo(address) <= 0; }
 274     bool operator>=(const CBitcoinAddress& address) const { return CompareTo(address) >= 0; }
 275     bool operator< (const CBitcoinAddress& address) const { return CompareTo(address) <  0; }
 276     bool operator> (const CBitcoinAddress& address) const { return CompareTo(address) >  0; }
 277 };
 278
 279 #endif