raw
genesis                 1 // Copyright (c) 2009-2010 Satoshi Nakamoto
genesis                 2 // Copyright (c) 2009-2012 The Bitcoin developers
genesis                 3 // Distributed under the MIT/X11 software license, see the accompanying
genesis                 4 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
programmable-vers...    5 #include "knobs.h"
genesis                 6 #include "headers.h"
genesis                 7 #include "strlcpy.h"
genesis                 8 #include <boost/program_options/detail/config_file.hpp>
genesis                 9 #include <boost/program_options/parsers.hpp>
genesis                10 #include <boost/filesystem.hpp>
genesis                11 #include <boost/filesystem/fstream.hpp>
genesis                12 #include <boost/interprocess/sync/interprocess_mutex.hpp>
genesis                13 #include <boost/interprocess/sync/interprocess_recursive_mutex.hpp>
genesis                14 #include <boost/foreach.hpp>
genesis                15 
genesis                16 using namespace std;
genesis                17 using namespace boost;
genesis                18 
genesis                19 map<string, string> mapArgs;
genesis                20 map<string, vector<string> > mapMultiArgs;
genesis                21 bool fDebug = false;
genesis                22 bool fPrintToConsole = false;
genesis                23 bool fPrintToDebugger = false;
asciilifeform_and...   24 bool fCanEat = false;
asciilifeform_add...   25 bool fVerifyAll = false;
asciilifeform_shi...   26 bool fShiva = false;
genesis                27 char pszSetDataDir[MAX_PATH] = "";
genesis                28 bool fRequestShutdown = false;
genesis                29 bool fShutdown = false;
genesis                30 bool fDaemon = false;
genesis                31 bool fServer = false;
genesis                32 bool fCommandLine = false;
genesis                33 string strMiscWarning;
genesis                34 bool fNoListen = false;
genesis                35 bool fLogTimestamps = false;
mod6_der_high_low_s    36 bool fLowS = false;
mod6_der_high_low_s    37 bool fHighS = false;
genesis                38 
programmable-vers...   39 std::string CLIENT_NAME(DEFAULT_CLIENT_NAME);
genesis                40 
genesis                41 
genesis                42 // Workaround for "multiple definition of `_tls_used'"
genesis                43 // http://svn.boost.org/trac/boost/ticket/4258
genesis                44 extern "C" void tss_cleanup_implemented() { }
genesis                45 
genesis                46 
genesis                47 
genesis                48 
genesis                49 
genesis                50 // Init openssl library multithreading support
genesis                51 static boost::interprocess::interprocess_mutex** ppmutexOpenSSL;
genesis                52 void locking_callback(int mode, int i, const char* file, int line)
genesis                53 {
genesis                54     if (mode & CRYPTO_LOCK)
genesis                55         ppmutexOpenSSL[i]->lock();
genesis                56     else
genesis                57         ppmutexOpenSSL[i]->unlock();
genesis                58 }
genesis                59 
genesis                60 // Init
genesis                61 class CInit
genesis                62 {
genesis                63 public:
genesis                64     CInit()
genesis                65     {
genesis                66         // Init openssl library multithreading support
genesis                67         ppmutexOpenSSL = (boost::interprocess::interprocess_mutex**)OPENSSL_malloc(CRYPTO_num_locks() * sizeof(boost::interprocess::interprocess_mutex*));
genesis                68         for (int i = 0; i < CRYPTO_num_locks(); i++)
genesis                69             ppmutexOpenSSL[i] = new boost::interprocess::interprocess_mutex();
genesis                70         CRYPTO_set_locking_callback(locking_callback);
genesis                71 
genesis                72         // Seed random number generator with performance counter
genesis                73         RandAddSeed();
genesis                74     }
genesis                75     ~CInit()
genesis                76     {
genesis                77         // Shutdown openssl library multithreading support
genesis                78         CRYPTO_set_locking_callback(NULL);
genesis                79         for (int i = 0; i < CRYPTO_num_locks(); i++)
genesis                80             delete ppmutexOpenSSL[i];
genesis                81         OPENSSL_free(ppmutexOpenSSL);
genesis                82     }
genesis                83 }
genesis                84 instance_of_cinit;
genesis                85 
genesis                86 
genesis                87 
genesis                88 
genesis                89 
genesis                90 
genesis                91 
genesis                92 
genesis                93 void RandAddSeed()
genesis                94 {
genesis                95     // Seed with CPU performance counter
genesis                96     int64 nCounter = GetPerformanceCounter();
genesis                97     RAND_add(&nCounter, sizeof(nCounter), 1.5);
genesis                98     memset(&nCounter, 0, sizeof(nCounter));
genesis                99 }
genesis               100 
genesis               101 void RandAddSeedPerfmon()
genesis               102 {
genesis               103     RandAddSeed();
genesis               104 
genesis               105     // This can take up to 2 seconds, so only do it every 10 minutes
genesis               106     static int64 nLastPerfmon;
genesis               107     if (GetTime() < nLastPerfmon + 10 * 60)
genesis               108         return;
genesis               109     nLastPerfmon = GetTime();
genesis               110 
genesis               111 }
genesis               112 
genesis               113 uint64 GetRand(uint64 nMax)
genesis               114 {
genesis               115     if (nMax == 0)
genesis               116         return 0;
genesis               117 
genesis               118     // The range of the random source must be a multiple of the modulus
genesis               119     // to give every possible output value an equal possibility
genesis               120     uint64 nRange = (UINT64_MAX / nMax) * nMax;
genesis               121     uint64 nRand = 0;
genesis               122     do
genesis               123         RAND_bytes((unsigned char*)&nRand, sizeof(nRand));
genesis               124     while (nRand >= nRange);
genesis               125     return (nRand % nMax);
genesis               126 }
genesis               127 
genesis               128 int GetRandInt(int nMax)
genesis               129 {
genesis               130     return GetRand(nMax);
genesis               131 }
genesis               132 
genesis               133 
genesis               134 
genesis               135 
genesis               136 
genesis               137 
genesis               138 
genesis               139 
genesis               140 
genesis               141 
genesis               142 
genesis               143 inline int OutputDebugStringF(const char* pszFormat, ...)
genesis               144 {
genesis               145     int ret = 0;
genesis               146     if (fPrintToConsole)
genesis               147     {
genesis               148         // print to console
genesis               149         va_list arg_ptr;
genesis               150         va_start(arg_ptr, pszFormat);
genesis               151         ret = vprintf(pszFormat, arg_ptr);
genesis               152         va_end(arg_ptr);
genesis               153     }
genesis               154     else
genesis               155     {
genesis               156         // print to debug.log
genesis               157         static FILE* fileout = NULL;
genesis               158 
genesis               159         if (!fileout)
genesis               160         {
genesis               161             char pszFile[MAX_PATH+100];
genesis               162             GetDataDir(pszFile);
genesis               163             strlcat(pszFile, "/debug.log", sizeof(pszFile));
genesis               164             fileout = fopen(pszFile, "a");
genesis               165             if (fileout) setbuf(fileout, NULL); // unbuffered
genesis               166         }
genesis               167         if (fileout)
genesis               168         {
genesis               169             static bool fStartedNewLine = true;
genesis               170 
genesis               171             // Debug print useful for profiling
genesis               172             if (fLogTimestamps && fStartedNewLine)
genesis               173                 fprintf(fileout, "%s ", DateTimeStrFormat("%x %H:%M:%S", GetTime()).c_str());
genesis               174             if (pszFormat[strlen(pszFormat) - 1] == '\n')
genesis               175                 fStartedNewLine = true;
genesis               176             else
genesis               177                 fStartedNewLine = false;
genesis               178 
genesis               179             va_list arg_ptr;
genesis               180             va_start(arg_ptr, pszFormat);
genesis               181             ret = vfprintf(fileout, pszFormat, arg_ptr);
genesis               182             va_end(arg_ptr);
genesis               183         }
genesis               184     }
genesis               185 
genesis               186     return ret;
genesis               187 }
genesis               188 
genesis               189 
genesis               190 // Safer snprintf
genesis               191 //  - prints up to limit-1 characters
genesis               192 //  - output string is always null terminated even if limit reached
genesis               193 //  - return value is the number of characters actually printed
genesis               194 int my_snprintf(char* buffer, size_t limit, const char* format, ...)
genesis               195 {
genesis               196     if (limit == 0)
genesis               197         return 0;
genesis               198     va_list arg_ptr;
genesis               199     va_start(arg_ptr, format);
genesis               200     int ret = _vsnprintf(buffer, limit, format, arg_ptr);
genesis               201     va_end(arg_ptr);
genesis               202     if (ret < 0 || ret >= limit)
genesis               203     {
genesis               204         ret = limit - 1;
genesis               205         buffer[limit-1] = 0;
genesis               206     }
genesis               207     return ret;
genesis               208 }
genesis               209 
genesis               210 string strprintf(const std::string &format, ...)
genesis               211 {
genesis               212     char buffer[50000];
genesis               213     char* p = buffer;
genesis               214     int limit = sizeof(buffer);
genesis               215     int ret;
genesis               216     loop
genesis               217     {
genesis               218         va_list arg_ptr;
genesis               219         va_start(arg_ptr, format);
genesis               220         ret = _vsnprintf(p, limit, format.c_str(), arg_ptr);
genesis               221         va_end(arg_ptr);
genesis               222         if (ret >= 0 && ret < limit)
genesis               223             break;
genesis               224         if (p != buffer)
genesis               225             delete[] p;
genesis               226         limit *= 2;
genesis               227         p = new char[limit];
genesis               228         if (p == NULL)
genesis               229             throw std::bad_alloc();
genesis               230     }
genesis               231     string str(p, p+ret);
genesis               232     if (p != buffer)
genesis               233         delete[] p;
genesis               234     return str;
genesis               235 }
genesis               236 
genesis               237 bool error(const std::string &format, ...)
genesis               238 {
genesis               239     char buffer[50000];
genesis               240     int limit = sizeof(buffer);
genesis               241     va_list arg_ptr;
genesis               242     va_start(arg_ptr, format);
genesis               243     int ret = _vsnprintf(buffer, limit, format.c_str(), arg_ptr);
genesis               244     va_end(arg_ptr);
genesis               245     if (ret < 0 || ret >= limit)
genesis               246     {
genesis               247         ret = limit - 1;
genesis               248         buffer[limit-1] = 0;
genesis               249     }
genesis               250     printf("ERROR: %s\n", buffer);
genesis               251     return false;
genesis               252 }
genesis               253 
genesis               254 
genesis               255 void ParseString(const string& str, char c, vector<string>& v)
genesis               256 {
genesis               257     if (str.empty())
genesis               258         return;
genesis               259     string::size_type i1 = 0;
genesis               260     string::size_type i2;
genesis               261     loop
genesis               262     {
genesis               263         i2 = str.find(c, i1);
genesis               264         if (i2 == str.npos)
genesis               265         {
genesis               266             v.push_back(str.substr(i1));
genesis               267             return;
genesis               268         }
genesis               269         v.push_back(str.substr(i1, i2-i1));
genesis               270         i1 = i2+1;
genesis               271     }
genesis               272 }
genesis               273 
genesis               274 
genesis               275 string FormatMoney(int64 n, bool fPlus)
genesis               276 {
genesis               277     // Note: not using straight sprintf here because we do NOT want
genesis               278     // localized number formatting.
genesis               279     int64 n_abs = (n > 0 ? n : -n);
genesis               280     int64 quotient = n_abs/COIN;
genesis               281     int64 remainder = n_abs%COIN;
genesis               282     string str = strprintf("%"PRI64d".%08"PRI64d, quotient, remainder);
genesis               283 
genesis               284     // Right-trim excess 0's before the decimal point:
genesis               285     int nTrim = 0;
genesis               286     for (int i = str.size()-1; (str[i] == '0' && isdigit(str[i-2])); --i)
genesis               287         ++nTrim;
genesis               288     if (nTrim)
genesis               289         str.erase(str.size()-nTrim, nTrim);
genesis               290 
genesis               291     if (n < 0)
genesis               292         str.insert((unsigned int)0, 1, '-');
genesis               293     else if (fPlus && n > 0)
genesis               294         str.insert((unsigned int)0, 1, '+');
genesis               295     return str;
genesis               296 }
genesis               297 
genesis               298 
genesis               299 bool ParseMoney(const string& str, int64& nRet)
genesis               300 {
genesis               301     return ParseMoney(str.c_str(), nRet);
genesis               302 }
genesis               303 
genesis               304 bool ParseMoney(const char* pszIn, int64& nRet)
genesis               305 {
genesis               306     string strWhole;
genesis               307     int64 nUnits = 0;
genesis               308     const char* p = pszIn;
genesis               309     while (isspace(*p))
genesis               310         p++;
genesis               311     for (; *p; p++)
genesis               312     {
genesis               313         if (*p == '.')
genesis               314         {
genesis               315             p++;
genesis               316             int64 nMult = CENT*10;
genesis               317             while (isdigit(*p) && (nMult > 0))
genesis               318             {
genesis               319                 nUnits += nMult * (*p++ - '0');
genesis               320                 nMult /= 10;
genesis               321             }
genesis               322             break;
genesis               323         }
genesis               324         if (isspace(*p))
genesis               325             break;
genesis               326         if (!isdigit(*p))
genesis               327             return false;
genesis               328         strWhole.insert(strWhole.end(), *p);
genesis               329     }
genesis               330     for (; *p; p++)
genesis               331         if (!isspace(*p))
genesis               332             return false;
genesis               333     if (strWhole.size() > 10) // guard against 63 bit overflow
genesis               334         return false;
genesis               335     if (nUnits < 0 || nUnits > COIN)
genesis               336         return false;
genesis               337     int64 nWhole = atoi64(strWhole);
genesis               338     int64 nValue = nWhole*COIN + nUnits;
genesis               339 
genesis               340     nRet = nValue;
genesis               341     return true;
genesis               342 }
genesis               343 
genesis               344 
genesis               345 vector<unsigned char> ParseHex(const char* psz)
genesis               346 {
genesis               347     static char phexdigit[256] =
genesis               348     { -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               349       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               350       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               351       0,1,2,3,4,5,6,7,8,9,-1,-1,-1,-1,-1,-1,
genesis               352       -1,0xa,0xb,0xc,0xd,0xe,0xf,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               353       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               354       -1,0xa,0xb,0xc,0xd,0xe,0xf,-1,-1,-1,-1,-1,-1,-1,-1,-1
genesis               355       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               356       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               357       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               358       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               359       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               360       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               361       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               362       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
genesis               363       -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1, };
genesis               364 
genesis               365     // convert hex dump to vector
genesis               366     vector<unsigned char> vch;
genesis               367     loop
genesis               368     {
genesis               369         while (isspace(*psz))
genesis               370             psz++;
genesis               371         char c = phexdigit[(unsigned char)*psz++];
genesis               372         if (c == (char)-1)
genesis               373             break;
genesis               374         unsigned char n = (c << 4);
genesis               375         c = phexdigit[(unsigned char)*psz++];
genesis               376         if (c == (char)-1)
genesis               377             break;
genesis               378         n |= c;
genesis               379         vch.push_back(n);
genesis               380     }
genesis               381     return vch;
genesis               382 }
genesis               383 
genesis               384 vector<unsigned char> ParseHex(const string& str)
genesis               385 {
genesis               386     return ParseHex(str.c_str());
genesis               387 }
genesis               388 
genesis               389 void ParseParameters(int argc, char* argv[])
genesis               390 {
genesis               391     mapArgs.clear();
genesis               392     mapMultiArgs.clear();
genesis               393     for (int i = 1; i < argc; i++)
genesis               394     {
genesis               395         char psz[10000];
genesis               396         strlcpy(psz, argv[i], sizeof(psz));
genesis               397         char* pszValue = (char*)"";
genesis               398         if (strchr(psz, '='))
genesis               399         {
genesis               400             pszValue = strchr(psz, '=');
genesis               401             *pszValue++ = '\0';
genesis               402         }
genesis               403         if (psz[0] != '-')
genesis               404             break;
genesis               405         mapArgs[psz] = pszValue;
genesis               406         mapMultiArgs[psz].push_back(pszValue);
genesis               407     }
genesis               408 }
genesis               409 
genesis               410 bool SoftSetArg(const std::string& strArg, const std::string& strValue)
genesis               411 {
genesis               412     if (mapArgs.count(strArg))
genesis               413         return false;
genesis               414     mapArgs[strArg] = strValue;
genesis               415     return true;
genesis               416 }
genesis               417 
genesis               418 bool SoftSetArg(const std::string& strArg, bool fValue)
genesis               419 {
genesis               420     if (fValue)
genesis               421         return SoftSetArg(strArg, std::string("1"));
genesis               422     else
genesis               423         return SoftSetArg(strArg, std::string("0"));
genesis               424 }
genesis               425 
genesis               426 
genesis               427 string EncodeBase64(const unsigned char* pch, size_t len)
genesis               428 {
genesis               429     static const char *pbase64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
genesis               430 
genesis               431     string strRet="";
genesis               432     strRet.reserve((len+2)/3*4);
genesis               433 
genesis               434     int mode=0, left=0;
genesis               435     const unsigned char *pchEnd = pch+len;
genesis               436 
genesis               437     while (pch<pchEnd)
genesis               438     {
genesis               439         int enc = *(pch++);
genesis               440         switch (mode)
genesis               441         {
genesis               442             case 0: // we have no bits
genesis               443                 strRet += pbase64[enc >> 2];
genesis               444                 left = (enc & 3) << 4;
genesis               445                 mode = 1;
genesis               446                 break;
genesis               447 
genesis               448             case 1: // we have two bits
genesis               449                 strRet += pbase64[left | (enc >> 4)];
genesis               450                 left = (enc & 15) << 2;
genesis               451                 mode = 2;
genesis               452                 break;
genesis               453 
genesis               454             case 2: // we have four bits
genesis               455                 strRet += pbase64[left | (enc >> 6)];
genesis               456                 strRet += pbase64[enc & 63];
genesis               457                 mode = 0;
genesis               458                 break;
genesis               459         }
genesis               460     }
genesis               461 
genesis               462     if (mode)
genesis               463     {
genesis               464         strRet += pbase64[left];
genesis               465         strRet += '=';
genesis               466         if (mode == 1)
genesis               467             strRet += '=';
genesis               468     }
genesis               469 
genesis               470     return strRet;
genesis               471 }
genesis               472 
genesis               473 string EncodeBase64(const string& str)
genesis               474 {
genesis               475     return EncodeBase64((const unsigned char*)str.c_str(), str.size());
genesis               476 }
genesis               477 
genesis               478 vector<unsigned char> DecodeBase64(const char* p, bool* pfInvalid)
genesis               479 {
genesis               480     static const int decode64_table[256] =
genesis               481     {
genesis               482         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               483         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               484         -1, -1, -1, 62, -1, -1, -1, 63, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1,
genesis               485         -1, -1, -1, -1, -1,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14,
genesis               486         15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1, -1, 26, 27, 28,
genesis               487         29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48,
genesis               488         49, 50, 51, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               489         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               490         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               491         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               492         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               493         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
genesis               494         -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
genesis               495     };
genesis               496 
genesis               497     if (pfInvalid)
genesis               498         *pfInvalid = false;
genesis               499 
genesis               500     vector<unsigned char> vchRet;
genesis               501     vchRet.reserve(strlen(p)*3/4);
genesis               502 
genesis               503     int mode = 0;
genesis               504     int left = 0;
genesis               505 
genesis               506     while (1)
genesis               507     {
genesis               508          int dec = decode64_table[*p];
genesis               509          if (dec == -1) break;
genesis               510          p++;
genesis               511          switch (mode)
genesis               512          {
genesis               513              case 0: // we have no bits and get 6
genesis               514                  left = dec;
genesis               515                  mode = 1;
genesis               516                  break;
genesis               517 
genesis               518               case 1: // we have 6 bits and keep 4
genesis               519                   vchRet.push_back((left<<2) | (dec>>4));
genesis               520                   left = dec & 15;
genesis               521                   mode = 2;
genesis               522                   break;
genesis               523 
genesis               524              case 2: // we have 4 bits and get 6, we keep 2
genesis               525                  vchRet.push_back((left<<4) | (dec>>2));
genesis               526                  left = dec & 3;
genesis               527                  mode = 3;
genesis               528                  break;
genesis               529 
genesis               530              case 3: // we have 2 bits and get 6
genesis               531                  vchRet.push_back((left<<6) | dec);
genesis               532                  mode = 0;
genesis               533                  break;
genesis               534          }
genesis               535     }
genesis               536 
genesis               537     if (pfInvalid)
genesis               538         switch (mode)
genesis               539         {
genesis               540             case 0: // 4n base64 characters processed: ok
genesis               541                 break;
genesis               542 
genesis               543             case 1: // 4n+1 base64 character processed: impossible
genesis               544                 *pfInvalid = true;
genesis               545                 break;
genesis               546 
genesis               547             case 2: // 4n+2 base64 characters processed: require '=='
genesis               548                 if (left || p[0] != '=' || p[1] != '=' || decode64_table[p[2]] != -1)
genesis               549                     *pfInvalid = true;
genesis               550                 break;
genesis               551 
genesis               552             case 3: // 4n+3 base64 characters processed: require '='
genesis               553                 if (left || p[0] != '=' || decode64_table[p[1]] != -1)
genesis               554                     *pfInvalid = true;
genesis               555                 break;
genesis               556         }
genesis               557 
genesis               558     return vchRet;
genesis               559 }
genesis               560 
genesis               561 string DecodeBase64(const string& str)
genesis               562 {
genesis               563     vector<unsigned char> vchRet = DecodeBase64(str.c_str());
genesis               564     return string((const char*)&vchRet[0], vchRet.size());
genesis               565 }
genesis               566 
genesis               567 
genesis               568 bool WildcardMatch(const char* psz, const char* mask)
genesis               569 {
genesis               570     loop
genesis               571     {
genesis               572         switch (*mask)
genesis               573         {
genesis               574         case '\0':
genesis               575             return (*psz == '\0');
genesis               576         case '*':
genesis               577             return WildcardMatch(psz, mask+1) || (*psz && WildcardMatch(psz+1, mask));
genesis               578         case '?':
genesis               579             if (*psz == '\0')
genesis               580                 return false;
genesis               581             break;
genesis               582         default:
genesis               583             if (*psz != *mask)
genesis               584                 return false;
genesis               585             break;
genesis               586         }
genesis               587         psz++;
genesis               588         mask++;
genesis               589     }
genesis               590 }
genesis               591 
genesis               592 bool WildcardMatch(const string& str, const string& mask)
genesis               593 {
genesis               594     return WildcardMatch(str.c_str(), mask.c_str());
genesis               595 }
genesis               596 
genesis               597 
genesis               598 
genesis               599 
genesis               600 
genesis               601 
genesis               602 
genesis               603 
genesis               604 void FormatException(char* pszMessage, std::exception* pex, const char* pszThread)
genesis               605 {
genesis               606     const char* pszModule = "bitcoin";
genesis               607     if (pex)
genesis               608         snprintf(pszMessage, 1000,
genesis               609             "EXCEPTION: %s       \n%s       \n%s in %s       \n", typeid(*pex).name(), pex->what(), pszModule, pszThread);
genesis               610     else
genesis               611         snprintf(pszMessage, 1000,
genesis               612             "UNKNOWN EXCEPTION       \n%s in %s       \n", pszModule, pszThread);
genesis               613 }
genesis               614 
genesis               615 void LogException(std::exception* pex, const char* pszThread)
genesis               616 {
genesis               617     char pszMessage[10000];
genesis               618     FormatException(pszMessage, pex, pszThread);
genesis               619     printf("\n%s", pszMessage);
genesis               620 }
genesis               621 
genesis               622 void PrintException(std::exception* pex, const char* pszThread)
genesis               623 {
genesis               624     char pszMessage[10000];
genesis               625     FormatException(pszMessage, pex, pszThread);
genesis               626     printf("\n\n************************\n%s\n", pszMessage);
genesis               627     fprintf(stderr, "\n\n************************\n%s\n", pszMessage);
genesis               628     strMiscWarning = pszMessage;
genesis               629     throw;
genesis               630 }
genesis               631 
genesis               632 void ThreadOneMessageBox(string strMessage)
genesis               633 {
genesis               634     // Skip message boxes if one is already open
genesis               635     static bool fMessageBoxOpen;
genesis               636     if (fMessageBoxOpen)
genesis               637         return;
genesis               638     fMessageBoxOpen = true;
genesis               639     ThreadSafeMessageBox(strMessage, "Bitcoin", wxOK | wxICON_EXCLAMATION);
genesis               640     fMessageBoxOpen = false;
genesis               641 }
genesis               642 
genesis               643 void PrintExceptionContinue(std::exception* pex, const char* pszThread)
genesis               644 {
genesis               645     char pszMessage[10000];
genesis               646     FormatException(pszMessage, pex, pszThread);
genesis               647     printf("\n\n************************\n%s\n", pszMessage);
genesis               648     fprintf(stderr, "\n\n************************\n%s\n", pszMessage);
genesis               649     strMiscWarning = pszMessage;
genesis               650 }
genesis               651 
genesis               652 string GetDefaultDataDir()
genesis               653 {
genesis               654     // Mac: ~/Library/Application Support/Bitcoin
genesis               655     // Unix: ~/.bitcoin
genesis               656     char* pszHome = getenv("HOME");
genesis               657     if (pszHome == NULL || strlen(pszHome) == 0)
genesis               658         pszHome = (char*)"/";
genesis               659     string strHome = pszHome;
genesis               660     if (strHome[strHome.size()-1] != '/')
genesis               661         strHome += '/';
genesis               662 #ifdef MAC_OSX
genesis               663     // Mac
genesis               664     strHome += "Library/Application Support/";
genesis               665     filesystem::create_directory(strHome.c_str());
genesis               666     return strHome + "Bitcoin";
genesis               667 #else
genesis               668     // Unix
genesis               669     return strHome + ".bitcoin";
genesis               670 #endif
genesis               671 }
genesis               672 
genesis               673 void GetDataDir(char* pszDir)
genesis               674 {
genesis               675     // pszDir must be at least MAX_PATH length.
genesis               676     int nVariation;
genesis               677     if (pszSetDataDir[0] != 0)
genesis               678     {
genesis               679         strlcpy(pszDir, pszSetDataDir, MAX_PATH);
genesis               680         nVariation = 0;
genesis               681     }
genesis               682     else
genesis               683     {
genesis               684         // This can be called during exceptions by printf, so we cache the
genesis               685         // value so we don't have to do memory allocations after that.
genesis               686         static char pszCachedDir[MAX_PATH];
genesis               687         if (pszCachedDir[0] == 0)
genesis               688             strlcpy(pszCachedDir, GetDefaultDataDir().c_str(), sizeof(pszCachedDir));
genesis               689         strlcpy(pszDir, pszCachedDir, MAX_PATH);
genesis               690         nVariation = 1;
genesis               691     }
genesis               692     static bool pfMkdir[4];
genesis               693     if (!pfMkdir[nVariation])
genesis               694     {
genesis               695         pfMkdir[nVariation] = true;
genesis               696         boost::filesystem::create_directory(pszDir);
genesis               697     }
genesis               698 }
genesis               699 
genesis               700 string GetDataDir()
genesis               701 {
genesis               702     char pszDir[MAX_PATH];
genesis               703     GetDataDir(pszDir);
genesis               704     return pszDir;
genesis               705 }
genesis               706 
genesis               707 string GetConfigFile()
genesis               708 {
genesis               709     namespace fs = boost::filesystem;
genesis               710     fs::path pathConfig(GetArg("-conf", "bitcoin.conf"));
genesis               711     if (!pathConfig.is_complete())
genesis               712         pathConfig = fs::path(GetDataDir()) / pathConfig;
genesis               713     return pathConfig.string();
genesis               714 }
genesis               715 
genesis               716 void ReadConfigFile(map<string, string>& mapSettingsRet,
genesis               717                     map<string, vector<string> >& mapMultiSettingsRet)
genesis               718 {
genesis               719     namespace fs = boost::filesystem;
genesis               720     namespace pod = boost::program_options::detail;
genesis               721 
genesis               722     fs::ifstream streamConfig(GetConfigFile());
genesis               723     if (!streamConfig.good())
genesis               724         return;
genesis               725 
genesis               726     set<string> setOptions;
genesis               727     setOptions.insert("*");
genesis               728     
genesis               729     for (pod::config_file_iterator it(streamConfig, setOptions), end; it != end; ++it)
genesis               730     {
genesis               731         // Don't overwrite existing settings so command line settings override bitcoin.conf
genesis               732         string strKey = string("-") + it->string_key;
genesis               733         if (mapSettingsRet.count(strKey) == 0)
genesis               734             mapSettingsRet[strKey] = it->value[0];
genesis               735         mapMultiSettingsRet[strKey].push_back(it->value[0]);
genesis               736     }
genesis               737 }
genesis               738 
genesis               739 string GetPidFile()
genesis               740 {
genesis               741     namespace fs = boost::filesystem;
genesis               742     fs::path pathConfig(GetArg("-pid", "bitcoind.pid"));
genesis               743     if (!pathConfig.is_complete())
genesis               744         pathConfig = fs::path(GetDataDir()) / pathConfig;
genesis               745     return pathConfig.string();
genesis               746 }
genesis               747 
genesis               748 void CreatePidFile(string pidFile, pid_t pid)
genesis               749 {
genesis               750     FILE* file = fopen(pidFile.c_str(), "w");
genesis               751     if (file)
genesis               752     {
genesis               753         fprintf(file, "%d\n", pid);
genesis               754         fclose(file);
genesis               755     }
genesis               756 }
genesis               757 
genesis               758 int GetFilesize(FILE* file)
genesis               759 {
genesis               760     int nSavePos = ftell(file);
genesis               761     int nFilesize = -1;
genesis               762     if (fseek(file, 0, SEEK_END) == 0)
genesis               763         nFilesize = ftell(file);
genesis               764     fseek(file, nSavePos, SEEK_SET);
genesis               765     return nFilesize;
genesis               766 }
genesis               767 
genesis               768 void ShrinkDebugFile()
genesis               769 {
genesis               770     // Scroll debug.log if it's getting too big
genesis               771     string strFile = GetDataDir() + "/debug.log";
genesis               772     FILE* file = fopen(strFile.c_str(), "r");
genesis               773     if (file && GetFilesize(file) > 10 * 1000000)
genesis               774     {
genesis               775         // Restart the file with some of the end
genesis               776         char pch[200000];
genesis               777         fseek(file, -sizeof(pch), SEEK_END);
genesis               778         int nBytes = fread(pch, 1, sizeof(pch), file);
genesis               779         fclose(file);
genesis               780 
genesis               781         file = fopen(strFile.c_str(), "w");
genesis               782         if (file)
genesis               783         {
genesis               784             fwrite(pch, 1, nBytes, file);
genesis               785             fclose(file);
genesis               786         }
genesis               787     }
genesis               788 }
genesis               789 
genesis               790 
asciilifeform_shi...  791 bool FileExists(const char *path) {
asciilifeform_shi...  792   FILE *file = fopen(path, "r");
asciilifeform_shi...  793   if (file) {
asciilifeform_shi...  794     fclose(file);
asciilifeform_shi...  795     return true;
asciilifeform_shi...  796   }
asciilifeform_shi...  797   return false;
asciilifeform_shi...  798 }
genesis               799 
genesis               800 
genesis               801 
genesis               802 
genesis               803 
genesis               804 //
genesis               805 // "Never go to sea with two chronometers; take one or three."
genesis               806 // Our three time sources are:
genesis               807 //  - System clock
genesis               808 //  - Median of other nodes's clocks
genesis               809 //  - The user (asking the user to fix the system clock if the first two disagree)
genesis               810 //
genesis               811 static int64 nMockTime = 0;  // For unit testing
genesis               812 
genesis               813 int64 GetTime()
genesis               814 {
genesis               815     if (nMockTime) return nMockTime;
genesis               816 
genesis               817     return time(NULL);
genesis               818 }
genesis               819 
genesis               820 void SetMockTime(int64 nMockTimeIn)
genesis               821 {
genesis               822     nMockTime = nMockTimeIn;
genesis               823 }
genesis               824 
genesis               825 static int64 nTimeOffset = 0;
genesis               826 
genesis               827 int64 GetAdjustedTime()
genesis               828 {
genesis               829     return GetTime() + nTimeOffset;
genesis               830 }
genesis               831 
genesis               832 void AddTimeData(unsigned int ip, int64 nTime)
genesis               833 {
genesis               834     int64 nOffsetSample = nTime - GetTime();
genesis               835 
genesis               836     // Ignore duplicates
genesis               837     static set<unsigned int> setKnown;
genesis               838     if (!setKnown.insert(ip).second)
genesis               839         return;
genesis               840 
genesis               841     // Add data
genesis               842     static vector<int64> vTimeOffsets;
genesis               843     if (vTimeOffsets.empty())
genesis               844         vTimeOffsets.push_back(0);
genesis               845     vTimeOffsets.push_back(nOffsetSample);
genesis               846     printf("Added time data, samples %d, offset %+"PRI64d" (%+"PRI64d" minutes)\n", vTimeOffsets.size(), vTimeOffsets.back(), vTimeOffsets.back()/60);
genesis               847     if (vTimeOffsets.size() >= 5 && vTimeOffsets.size() % 2 == 1)
genesis               848     {
genesis               849         sort(vTimeOffsets.begin(), vTimeOffsets.end());
genesis               850         int64 nMedian = vTimeOffsets[vTimeOffsets.size()/2];
genesis               851         // Only let other nodes change our time by so much
genesis               852         if (abs64(nMedian) < 70 * 60)
genesis               853         {
genesis               854             nTimeOffset = nMedian;
genesis               855         }
genesis               856         else
genesis               857         {
genesis               858             nTimeOffset = 0;
genesis               859 
genesis               860             static bool fDone;
genesis               861             if (!fDone)
genesis               862             {
genesis               863                 // If nobody has a time different than ours but within 5 minutes of ours, give a warning
genesis               864                 bool fMatch = false;
genesis               865                 BOOST_FOREACH(int64 nOffset, vTimeOffsets)
genesis               866                     if (nOffset != 0 && abs64(nOffset) < 5 * 60)
genesis               867                         fMatch = true;
genesis               868 
genesis               869                 if (!fMatch)
genesis               870                 {
genesis               871                     fDone = true;
genesis               872                     string strMessage = _("Warning: Please check that your computer's date and time are correct.  If your clock is wrong Bitcoin will not work properly.");
genesis               873                     strMiscWarning = strMessage;
genesis               874                     printf("*** %s\n", strMessage.c_str());
genesis               875                     boost::thread(boost::bind(ThreadSafeMessageBox, strMessage+" ", string("Bitcoin"), wxOK | wxICON_EXCLAMATION, (wxWindow*)NULL, -1, -1));
genesis               876                 }
genesis               877             }
genesis               878         }
genesis               879         BOOST_FOREACH(int64 n, vTimeOffsets)
genesis               880             printf("%+"PRI64d"  ", n);
genesis               881         printf("|  nTimeOffset = %+"PRI64d"  (%+"PRI64d" minutes)\n", nTimeOffset, nTimeOffset/60);
genesis               882     }
genesis               883 }
genesis               884 
genesis               885 
genesis               886 
genesis               887 
genesis               888 
genesis               889 
genesis               890 
genesis               891 
genesis               892 
genesis               893 string FormatVersion(int nVersion)
genesis               894 {
genesis               895     if (nVersion%100 == 0)
genesis               896         return strprintf("%d.%d.%d", nVersion/1000000, (nVersion/10000)%100, (nVersion/100)%100);
genesis               897     else
genesis               898         return strprintf("%d.%d.%d.%d", nVersion/1000000, (nVersion/10000)%100, (nVersion/100)%100, nVersion%100);
genesis               899 }
genesis               900 
genesis               901 string FormatFullVersion()
genesis               902 {
programmable-vers...  903     string s = FormatVersion(VERSION);
genesis               904     return s;
genesis               905 }
genesis               906 
programmable-vers...  907 std::string FormatSubVersion(const std::string& name, int nClientVersion)
programmable-vers...  908 {
programmable-vers...  909     std::ostringstream ss;
programmable-vers...  910     ss << "/";
programmable-vers...  911     ss << name << ":" << FormatVersion(nClientVersion);
programmable-vers...  912     ss << "/";
programmable-vers...  913     return ss.str();
programmable-vers...  914 }
genesis               915 
genesis               916 
genesis               917 
genesis               918 #ifdef DEBUG_LOCKORDER
genesis               919 //
genesis               920 // Early deadlock detection.
genesis               921 // Problem being solved:
genesis               922 //    Thread 1 locks  A, then B, then C
genesis               923 //    Thread 2 locks  D, then C, then A
genesis               924 //     --> may result in deadlock between the two threads, depending on when they run.
genesis               925 // Solution implemented here:
genesis               926 // Keep track of pairs of locks: (A before B), (A before C), etc.
genesis               927 // Complain if any thread trys to lock in a different order.
genesis               928 //
genesis               929 
genesis               930 struct CLockLocation
genesis               931 {
genesis               932     CLockLocation(const char* pszName, const char* pszFile, int nLine)
genesis               933     {
genesis               934         mutexName = pszName;
genesis               935         sourceFile = pszFile;
genesis               936         sourceLine = nLine;
genesis               937     }
genesis               938 
genesis               939     std::string ToString() const
genesis               940     {
genesis               941         return mutexName+"  "+sourceFile+":"+itostr(sourceLine);
genesis               942     }
genesis               943 
genesis               944 private:
genesis               945     std::string mutexName;
genesis               946     std::string sourceFile;
genesis               947     int sourceLine;
genesis               948 };
genesis               949 
genesis               950 typedef std::vector< std::pair<CCriticalSection*, CLockLocation> > LockStack;
genesis               951 
genesis               952 static boost::interprocess::interprocess_mutex dd_mutex;
genesis               953 static std::map<std::pair<CCriticalSection*, CCriticalSection*>, LockStack> lockorders;
genesis               954 static boost::thread_specific_ptr<LockStack> lockstack;
genesis               955 
genesis               956 
genesis               957 static void potential_deadlock_detected(const std::pair<CCriticalSection*, CCriticalSection*>& mismatch, const LockStack& s1, const LockStack& s2)
genesis               958 {
genesis               959     printf("POTENTIAL DEADLOCK DETECTED\n");
genesis               960     printf("Previous lock order was:\n");
genesis               961     BOOST_FOREACH(const PAIRTYPE(CCriticalSection*, CLockLocation)& i, s2)
genesis               962     {
genesis               963         if (i.first == mismatch.first) printf(" (1)");
genesis               964         if (i.first == mismatch.second) printf(" (2)");
genesis               965         printf(" %s\n", i.second.ToString().c_str());
genesis               966     }
genesis               967     printf("Current lock order is:\n");
genesis               968     BOOST_FOREACH(const PAIRTYPE(CCriticalSection*, CLockLocation)& i, s1)
genesis               969     {
genesis               970         if (i.first == mismatch.first) printf(" (1)");
genesis               971         if (i.first == mismatch.second) printf(" (2)");
genesis               972         printf(" %s\n", i.second.ToString().c_str());
genesis               973     }
genesis               974 }
genesis               975 
genesis               976 static void push_lock(CCriticalSection* c, const CLockLocation& locklocation)
genesis               977 {
genesis               978     bool fOrderOK = true;
genesis               979     if (lockstack.get() == NULL)
genesis               980         lockstack.reset(new LockStack);
genesis               981 
genesis               982     if (fDebug) printf("Locking: %s\n", locklocation.ToString().c_str());
genesis               983     dd_mutex.lock();
genesis               984 
genesis               985     (*lockstack).push_back(std::make_pair(c, locklocation));
genesis               986 
genesis               987     BOOST_FOREACH(const PAIRTYPE(CCriticalSection*, CLockLocation)& i, (*lockstack))
genesis               988     {
genesis               989         if (i.first == c) break;
genesis               990 
genesis               991         std::pair<CCriticalSection*, CCriticalSection*> p1 = std::make_pair(i.first, c);
genesis               992         if (lockorders.count(p1))
genesis               993             continue;
genesis               994         lockorders[p1] = (*lockstack);
genesis               995 
genesis               996         std::pair<CCriticalSection*, CCriticalSection*> p2 = std::make_pair(c, i.first);
genesis               997         if (lockorders.count(p2))
genesis               998         {
genesis               999             potential_deadlock_detected(p1, lockorders[p2], lockorders[p1]);
genesis              1000             break;
genesis              1001         }
genesis              1002     }
genesis              1003     dd_mutex.unlock();
genesis              1004 }
genesis              1005 
genesis              1006 static void pop_lock()
genesis              1007 {
genesis              1008     if (fDebug) 
genesis              1009     {
genesis              1010         const CLockLocation& locklocation = (*lockstack).rbegin()->second;
genesis              1011         printf("Unlocked: %s\n", locklocation.ToString().c_str());
genesis              1012     }
genesis              1013     dd_mutex.lock();
genesis              1014     (*lockstack).pop_back();
genesis              1015     dd_mutex.unlock();
genesis              1016 }
genesis              1017 
genesis              1018 void CCriticalSection::Enter(const char* pszName, const char* pszFile, int nLine)
genesis              1019 {
genesis              1020     push_lock(this, CLockLocation(pszName, pszFile, nLine));
genesis              1021     mutex.lock();
genesis              1022 }
genesis              1023 void CCriticalSection::Leave()
genesis              1024 {
genesis              1025     mutex.unlock();
genesis              1026     pop_lock();
genesis              1027 }
genesis              1028 bool CCriticalSection::TryEnter(const char* pszName, const char* pszFile, int nLine)
genesis              1029 {
genesis              1030     push_lock(this, CLockLocation(pszName, pszFile, nLine));
genesis              1031     bool result = mutex.try_lock();
genesis              1032     if (!result) pop_lock();
genesis              1033     return result;
genesis              1034 }
genesis              1035 
genesis              1036 #else
genesis              1037 
genesis              1038 void CCriticalSection::Enter(const char*, const char*, int)
genesis              1039 {
genesis              1040     mutex.lock();
genesis              1041 }
genesis              1042 
genesis              1043 void CCriticalSection::Leave()
genesis              1044 {
genesis              1045     mutex.unlock();
genesis              1046 }
genesis              1047 
genesis              1048 bool CCriticalSection::TryEnter(const char*, const char*, int)
genesis              1049 {
genesis              1050     bool result = mutex.try_lock();
genesis              1051     return result;
genesis              1052 }
genesis              1053 
genesis              1054 #endif /* DEBUG_LOCKORDER */