]>
Commit | Line | Data |
---|---|---|
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 | |
15 | using namespace std; | |
69d605f4 | 16 | using namespace boost; |
2097c09a WL |
17 | |
18 | map<string, string> mapArgs; | |
19 | map<string, vector<string> > mapMultiArgs; | |
20 | bool fDebug = false; | |
21 | bool fPrintToConsole = false; | |
22 | bool fPrintToDebugger = false; | |
23 | char pszSetDataDir[MAX_PATH] = ""; | |
24 | bool fRequestShutdown = false; | |
25 | bool fShutdown = false; | |
26 | bool fDaemon = false; | |
27 | bool fServer = false; | |
28 | bool fCommandLine = false; | |
29 | string strMiscWarning; | |
30 | bool fTestNet = false; | |
31 | bool fNoListen = false; | |
32 | bool fLogTimestamps = false; | |
33 | ||
34 | ||
35 | ||
36 | ||
37 | // Workaround for "multiple definition of `_tls_used'" | |
38 | // http://svn.boost.org/trac/boost/ticket/4258 | |
39 | extern "C" void tss_cleanup_implemented() { } | |
40 | ||
41 | ||
42 | ||
43 | ||
44 | ||
45 | // Init openssl library multithreading support | |
46 | static boost::interprocess::interprocess_mutex** ppmutexOpenSSL; | |
47 | void 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 | |
56 | class CInit | |
57 | { | |
58 | public: | |
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 | } | |
84 | instance_of_cinit; | |
85 | ||
86 | ||
87 | ||
88 | ||
89 | ||
90 | ||
91 | ||
92 | ||
93 | void 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 | ||
101 | void 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 | ||
128 | uint64 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 | ||
143 | int GetRandInt(int nMax) | |
144 | { | |
145 | return GetRand(nMax); | |
146 | } | |
147 | ||
148 | ||
149 | ||
150 | ||
151 | ||
152 | ||
153 | ||
154 | ||
155 | ||
156 | ||
157 | ||
158 | inline 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 | |
251 | int 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 | 267 | string 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 | 294 | bool 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 | ||
312 | void 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 | ||
332 | string 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 | ||
356 | bool ParseMoney(const string& str, int64& nRet) | |
357 | { | |
358 | return ParseMoney(str.c_str(), nRet); | |
359 | } | |
360 | ||
361 | bool 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 | ||
402 | vector<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 | ||
441 | vector<unsigned char> ParseHex(const string& str) | |
442 | { | |
443 | return ParseHex(str.c_str()); | |
444 | } | |
445 | ||
446 | ||
447 | void 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 |
473 | static 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 | ||
490 | std::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 | |
534 | bool 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 | ||
558 | bool 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 | ||
570 | void 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 | ||
587 | void LogException(std::exception* pex, const char* pszThread) | |
588 | { | |
589 | char pszMessage[10000]; | |
590 | FormatException(pszMessage, pex, pszThread); | |
591 | printf("\n%s", pszMessage); | |
592 | } | |
593 | ||
594 | void 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 | ||
604 | void 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 | ||
615 | void 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__ | |
632 | typedef WINSHELLAPI BOOL (WINAPI *PSHGETSPECIALFOLDERPATHA)(HWND hwndOwner, LPSTR lpszPath, int nFolder, BOOL fCreate); | |
633 | ||
634 | string 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 | ||
667 | string 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 | ||
694 | void 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 | ||
729 | string GetDataDir() | |
730 | { | |
731 | char pszDir[MAX_PATH]; | |
732 | GetDataDir(pszDir); | |
733 | return pszDir; | |
734 | } | |
735 | ||
736 | string 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 | ||
745 | void 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 | ||
768 | string 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 | ||
777 | void 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 | ||
787 | int 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 | ||
797 | void 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 |
833 | static int64 nMockTime = 0; // For unit testing |
834 | ||
2097c09a WL |
835 | int64 GetTime() |
836 | { | |
54d02f15 GA |
837 | if (nMockTime) return nMockTime; |
838 | ||
2097c09a WL |
839 | return time(NULL); |
840 | } | |
841 | ||
54d02f15 GA |
842 | void SetMockTime(int64 nMockTimeIn) |
843 | { | |
844 | nMockTime = nMockTimeIn; | |
845 | } | |
846 | ||
2097c09a WL |
847 | static int64 nTimeOffset = 0; |
848 | ||
849 | int64 GetAdjustedTime() | |
850 | { | |
851 | return GetTime() + nTimeOffset; | |
852 | } | |
853 | ||
854 | void 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 | ||
915 | string 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 | ||
923 | string 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 | ||
948 | struct 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 | ||
962 | private: | |
963 | std::string mutexName; | |
964 | std::string sourceFile; | |
965 | int sourceLine; | |
865ed8a1 GA |
966 | }; |
967 | ||
968 | typedef std::vector< std::pair<CCriticalSection*, CLockLocation> > LockStack; | |
969 | ||
970 | static boost::interprocess::interprocess_mutex dd_mutex; | |
971 | static std::map<std::pair<CCriticalSection*, CCriticalSection*>, LockStack> lockorders; | |
972 | static boost::thread_specific_ptr<LockStack> lockstack; | |
973 | ||
974 | ||
b0243da7 | 975 | static 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 | ||
994 | static 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 | ||
1024 | static 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 | ||
1036 | void CCriticalSection::Enter(const char* pszName, const char* pszFile, int nLine) | |
1037 | { | |
1038 | push_lock(this, CLockLocation(pszName, pszFile, nLine)); | |
1039 | mutex.lock(); | |
1040 | } | |
1041 | void CCriticalSection::Leave() | |
1042 | { | |
1043 | mutex.unlock(); | |
1044 | pop_lock(); | |
1045 | } | |
1046 | bool 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 | ||
1056 | void CCriticalSection::Enter(const char*, const char*, int) | |
1057 | { | |
1058 | mutex.lock(); | |
1059 | } | |
1060 | ||
1061 | void CCriticalSection::Leave() | |
1062 | { | |
1063 | mutex.unlock(); | |
1064 | } | |
1065 | ||
1066 | bool 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 */ |