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