init.cpp 23.7 KB
Newer Older
1
// Copyright (c) 2009-2010 Satoshi Nakamoto
Gavin Andresen's avatar
Gavin Andresen committed
2
// Copyright (c) 2009-2012 The Bitcoin developers
3
4
5
// Distributed under the MIT/X11 software license, see the accompanying
// file license.txt or http://www.opensource.org/licenses/mit-license.php.
#include "headers.h"
6
#include "db.h"
Wladimir J. van der Laan's avatar
Wladimir J. van der Laan committed
7
#include "bitcoinrpc.h"
8
9
10
#include "net.h"
#include "init.h"
#include "strlcpy.h"
11
#include <boost/filesystem.hpp>
12
#include <boost/filesystem/fstream.hpp>
13
#include <boost/filesystem/convenience.hpp>
14
#include <boost/interprocess/sync/file_lock.hpp>
15

16
17
18
19
20
21
22
23
#if defined(BITCOIN_NEED_QT_PLUGINS) && !defined(_BITCOIN_QT_PLUGINS_INCLUDED)
#define _BITCOIN_QT_PLUGINS_INCLUDED
#define __INSURE__
#include <QtPlugin>
Q_IMPORT_PLUGIN(qcncodecs)
Q_IMPORT_PLUGIN(qjpcodecs)
Q_IMPORT_PLUGIN(qtwcodecs)
Q_IMPORT_PLUGIN(qkrcodecs)
24
Q_IMPORT_PLUGIN(qtaccessiblewidgets)
25
26
#endif

27
28
29
using namespace std;
using namespace boost;

30
31
CWallet* pwalletMain;

32
33
34
35
36
37
38
//////////////////////////////////////////////////////////////////////////////
//
// Shutdown
//

void ExitTimeout(void* parg)
{
Gavin Andresen's avatar
Gavin Andresen committed
39
#ifdef WIN32
40
41
42
43
44
45
46
47
48
    Sleep(5000);
    ExitProcess(0);
#endif
}

void Shutdown(void* parg)
{
    static CCriticalSection cs_Shutdown;
    static bool fTaken;
49
50
    bool fFirstThread = false;
    TRY_CRITICAL_BLOCK(cs_Shutdown)
51
52
53
54
55
56
57
58
59
    {
        fFirstThread = !fTaken;
        fTaken = true;
    }
    static bool fExit;
    if (fFirstThread)
    {
        fShutdown = true;
        nTransactionsUpdated++;
60
        DBFlush(false);
61
        StopNode();
62
        DBFlush(true);
63
        boost::filesystem::remove(GetPidFile());
64
65
        UnregisterWallet(pwalletMain);
        delete pwalletMain;
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
        CreateThread(ExitTimeout, NULL);
        Sleep(50);
        printf("Bitcoin exiting\n\n");
        fExit = true;
        exit(0);
    }
    else
    {
        while (!fExit)
            Sleep(500);
        Sleep(100);
        ExitThread(0);
    }
}

void HandleSIGTERM(int)
{
    fRequestShutdown = true;
}






//////////////////////////////////////////////////////////////////////////////
//
// Start
//
Gavin Andresen's avatar
Gavin Andresen committed
95
#if !defined(QT_GUI)
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
int main(int argc, char* argv[])
{
    bool fRet = false;
    fRet = AppInit(argc, argv);

    if (fRet && fDaemon)
        return 0;

    return 1;
}
#endif

bool AppInit(int argc, char* argv[])
{
    bool fRet = false;
    try
    {
        fRet = AppInit2(argc, argv);
    }
    catch (std::exception& e) {
        PrintException(&e, "AppInit()");
    } catch (...) {
        PrintException(NULL, "AppInit()");
    }
    if (!fRet)
        Shutdown(NULL);
    return fRet;
}

bool AppInit2(int argc, char* argv[])
{
#ifdef _MSC_VER
    // Turn off microsoft heap dump noise
    _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
    _CrtSetReportFile(_CRT_WARN, CreateFileA("NUL", GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, 0));
#endif
#if _MSC_VER >= 1400
    // Disable confusing "helpful" text message on abort, ctrl-c
    _set_abort_behavior(0, _WRITE_ABORT_MSG | _CALL_REPORTFAULT);
#endif
Gavin Andresen's avatar
Gavin Andresen committed
136
#ifndef WIN32
137
138
    umask(077);
#endif
Gavin Andresen's avatar
Gavin Andresen committed
139
#ifndef WIN32
140
141
142
143
144
145
146
147
148
149
150
151
152
    // Clean shutdown on SIGTERM
    struct sigaction sa;
    sa.sa_handler = HandleSIGTERM;
    sigemptyset(&sa.sa_mask);
    sa.sa_flags = 0;
    sigaction(SIGTERM, &sa, NULL);
    sigaction(SIGINT, &sa, NULL);
    sigaction(SIGHUP, &sa, NULL);
#endif

    //
    // Parameters
    //
153
154
    // If Qt is used, parameters are parsed in qt/bitcoin.cpp's main()
#if !defined(QT_GUI)
155
    ParseParameters(argc, argv);
156
#endif
157
158
159

    if (mapArgs.count("-datadir"))
    {
160
161
162
163
164
165
166
167
168
169
        if (filesystem::is_directory(filesystem::system_complete(mapArgs["-datadir"])))
        {
            filesystem::path pathDataDir = filesystem::system_complete(mapArgs["-datadir"]);
            strlcpy(pszSetDataDir, pathDataDir.string().c_str(), sizeof(pszSetDataDir));
        }
        else
        {
            fprintf(stderr, "Error: Specified directory does not exist\n");
            Shutdown(NULL);
        }
170
171
    }

172

173
174
175
176
177
178
179
    ReadConfigFile(mapArgs, mapMultiArgs); // Must be done after processing datadir

    if (mapArgs.count("-?") || mapArgs.count("--help"))
    {
        string strUsage = string() +
          _("Bitcoin version") + " " + FormatFullVersion() + "\n\n" +
          _("Usage:") + "\t\t\t\t\t\t\t\t\t\t\n" +
180
            "  bitcoind [options]                   \t  " + "\n" +
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
            "  bitcoind [options] <command> [params]\t  " + _("Send command to -server or bitcoind") + "\n" +
            "  bitcoind [options] help              \t\t  " + _("List commands") + "\n" +
            "  bitcoind [options] help <command>    \t\t  " + _("Get help for a command") + "\n" +
          _("Options:") + "\n" +
            "  -conf=<file>     \t\t  " + _("Specify configuration file (default: bitcoin.conf)") + "\n" +
            "  -pid=<file>      \t\t  " + _("Specify pid file (default: bitcoind.pid)") + "\n" +
            "  -gen             \t\t  " + _("Generate coins") + "\n" +
            "  -gen=0           \t\t  " + _("Don't generate coins") + "\n" +
            "  -min             \t\t  " + _("Start minimized") + "\n" +
            "  -datadir=<dir>   \t\t  " + _("Specify data directory") + "\n" +
            "  -timeout=<n>     \t  "   + _("Specify connection timeout (in milliseconds)") + "\n" +
            "  -proxy=<ip:port> \t  "   + _("Connect through socks4 proxy") + "\n" +
            "  -dns             \t  "   + _("Allow DNS lookups for addnode and connect") + "\n" +
            "  -port=<port>     \t\t  " + _("Listen for connections on <port> (default: 8333 or testnet: 18333)") + "\n" +
            "  -maxconnections=<n>\t  " + _("Maintain at most <n> connections to peers (default: 125)") + "\n" +
196
            "  -addnode=<ip>    \t  "   + _("Add a node to connect to and attempt to keep the connection open") + "\n" +
197
            "  -connect=<ip>    \t\t  " + _("Connect only to the specified node") + "\n" +
198
199
200
            "  -irc             \t  "   + _("Find peers using internet relay chat (default: 0)") + "\n" +
            "  -listen          \t  "   + _("Accept connections from outside (default: 1)") + "\n" +
            "  -dnsseed         \t  "   + _("Find peers using DNS lookup (default: 1)") + "\n" +
201
202
203
204
            "  -banscore=<n>    \t  "   + _("Threshold for disconnecting misbehaving peers (default: 100)") + "\n" +
            "  -bantime=<n>     \t  "   + _("Number of seconds to keep misbehaving peers from reconnecting (default: 86400)") + "\n" +
            "  -maxreceivebuffer=<n>\t  " + _("Maximum per-connection receive buffer, <n>*1000 bytes (default: 10000)") + "\n" +
            "  -maxsendbuffer=<n>\t  "   + _("Maximum per-connection send buffer, <n>*1000 bytes (default: 10000)") + "\n" +
205
206
#ifdef USE_UPNP
#if USE_UPNP
207
            "  -upnp            \t  "   + _("Use Universal Plug and Play to map the listening port (default: 1)") + "\n" +
208
#else
209
            "  -upnp            \t  "   + _("Use Universal Plug and Play to map the listening port (default: 0)") + "\n" +
210
211
#endif
#endif
212
            "  -paytxfee=<amt>  \t  "   + _("Fee per KB to add to transactions you send") + "\n" +
213
#ifdef QT_GUI
214
            "  -server          \t\t  " + _("Accept command line and JSON-RPC commands") + "\n" +
215
#endif
216
#if !defined(WIN32) && !defined(QT_GUI)
217
            "  -daemon          \t\t  " + _("Run in the background as a daemon and accept commands") + "\n" +
218
#endif
219
220
221
222
            "  -testnet         \t\t  " + _("Use the test network") + "\n" +
            "  -debug           \t\t  " + _("Output extra debugging information") + "\n" +
            "  -logtimestamps   \t  "   + _("Prepend debug output with timestamp") + "\n" +
            "  -printtoconsole  \t  "   + _("Send trace/debug info to console instead of debug.log file") + "\n" +
223
#ifdef WIN32
224
            "  -printtodebugger \t  "   + _("Send trace/debug info to debugger") + "\n" +
225
#endif
226
227
228
229
230
            "  -rpcuser=<user>  \t  "   + _("Username for JSON-RPC connections") + "\n" +
            "  -rpcpassword=<pw>\t  "   + _("Password for JSON-RPC connections") + "\n" +
            "  -rpcport=<port>  \t\t  " + _("Listen for JSON-RPC connections on <port> (default: 8332)") + "\n" +
            "  -rpcallowip=<ip> \t\t  " + _("Allow JSON-RPC connections from specified IP address") + "\n" +
            "  -rpcconnect=<ip> \t  "   + _("Send commands to node running on <ip> (default: 127.0.0.1)") + "\n" +
231
            "  -blocknotify=<cmd> "     + _("Execute command when the best block changes (%s in cmd is replaced by block hash)") + "\n" +
232
233
            "  -keypool=<n>     \t  "   + _("Set key pool size to <n> (default: 100)") + "\n" +
            "  -rescan          \t  "   + _("Rescan the block chain for missing wallet transactions") + "\n";
234
235
236

#ifdef USE_SSL
        strUsage += string() +
237
238
239
240
241
            _("\nSSL options: (see the Bitcoin Wiki for SSL setup instructions)") + "\n" +
            "  -rpcssl                                \t  " + _("Use OpenSSL (https) for JSON-RPC connections") + "\n" +
            "  -rpcsslcertificatechainfile=<file.cert>\t  " + _("Server certificate file (default: server.cert)") + "\n" +
            "  -rpcsslprivatekeyfile=<file.pem>       \t  " + _("Server private key (default: server.pem)") + "\n" +
            "  -rpcsslciphers=<ciphers>               \t  " + _("Acceptable ciphers (default: TLSv1+HIGH:!SSLv2:!aNULL:!eNULL:!AH:!3DES:@STRENGTH)") + "\n";
242
243
244
#endif

        strUsage += string() +
245
            "  -?               \t\t  " + _("This help message") + "\n";
246
247
248
249
250
251
252

        // Remove tabs
        strUsage.erase(std::remove(strUsage.begin(), strUsage.end(), '\t'), strUsage.end());
        fprintf(stderr, "%s", strUsage.c_str());
        return false;
    }

253
    fTestNet = GetBoolArg("-testnet");
254
255
    fDebug = GetBoolArg("-debug");

256
#if !defined(WIN32) && !defined(QT_GUI)
257
258
259
260
261
262
263
264
265
266
267
    fDaemon = GetBoolArg("-daemon");
#else
    fDaemon = false;
#endif

    if (fDaemon)
        fServer = true;
    else
        fServer = GetBoolArg("-server");

    /* force fServer when running without GUI */
Gavin Andresen's avatar
Gavin Andresen committed
268
#if !defined(QT_GUI)
269
    fServer = true;
270
#endif
271
272
273
274
    fPrintToConsole = GetBoolArg("-printtoconsole");
    fPrintToDebugger = GetBoolArg("-printtodebugger");
    fLogTimestamps = GetBoolArg("-logtimestamps");

275
#ifndef QT_GUI
276
    for (int i = 1; i < argc; i++)
277
        if (!IsSwitchChar(argv[i][0]) && !(strlen(argv[i]) > 7 && strncasecmp(argv[i], "bitcoin:", 8) == 0))
278
279
280
281
282
283
284
            fCommandLine = true;

    if (fCommandLine)
    {
        int ret = CommandLineRPC(argc, argv);
        exit(ret);
    }
285
#endif
286

287
#if !defined(WIN32) && !defined(QT_GUI)
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
    if (fDaemon)
    {
        // Daemonize
        pid_t pid = fork();
        if (pid < 0)
        {
            fprintf(stderr, "Error: fork() returned %d errno %d\n", pid, errno);
            return false;
        }
        if (pid > 0)
        {
            CreatePidFile(GetPidFile(), pid);
            return true;
        }

        pid_t sid = setsid();
        if (sid < 0)
            fprintf(stderr, "Error: setsid() returned %d errno %d\n", sid, errno);
    }
#endif

    if (!fDebug && !pszSetDataDir[0])
        ShrinkDebugFile();
    printf("\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n");
    printf("Bitcoin version %s\n", FormatFullVersion().c_str());
    printf("Default data directory %s\n", GetDefaultDataDir().c_str());

    if (GetBoolArg("-loadblockindextest"))
    {
        CTxDB txdb("r");
        txdb.LoadBlockIndex();
        PrintBlockTree();
        return false;
    }

    // Make sure only a single bitcoin process is using the data directory.
    string strLockFile = GetDataDir() + "/.lock";
    FILE* file = fopen(strLockFile.c_str(), "a"); // empty lock file; created if it doesn't exist.
    if (file) fclose(file);
    static boost::interprocess::file_lock lock(strLockFile.c_str());
    if (!lock.try_lock())
    {
        wxMessageBox(strprintf(_("Cannot obtain a lock on data directory %s.  Bitcoin is probably already running."), GetDataDir().c_str()), "Bitcoin");
        return false;
    }

334
    std::ostringstream strErrors;
335
336
337
338
339
    //
    // Load data files
    //
    if (fDaemon)
        fprintf(stdout, "bitcoin server starting\n");
340
    int64 nStart;
341

342
    InitMessage(_("Loading addresses..."));
343
344
345
    printf("Loading addresses...\n");
    nStart = GetTimeMillis();
    if (!LoadAddresses())
346
        strErrors << _("Error loading addr.dat") << "\n";
347
348
    printf(" addresses   %15"PRI64d"ms\n", GetTimeMillis() - nStart);

349
    InitMessage(_("Loading block index..."));
350
351
352
    printf("Loading block index...\n");
    nStart = GetTimeMillis();
    if (!LoadBlockIndex())
353
        strErrors << _("Error loading blkindex.dat") << "\n";
354
355
    printf(" block index %15"PRI64d"ms\n", GetTimeMillis() - nStart);

356
    InitMessage(_("Loading wallet..."));
357
358
359
    printf("Loading wallet...\n");
    nStart = GetTimeMillis();
    bool fFirstRun;
360
    pwalletMain = new CWallet("wallet.dat");
Matt Corallo's avatar
Matt Corallo committed
361
362
363
364
    int nLoadWalletRet = pwalletMain->LoadWallet(fFirstRun);
    if (nLoadWalletRet != DB_LOAD_OK)
    {
        if (nLoadWalletRet == DB_CORRUPT)
365
            strErrors << _("Error loading wallet.dat: Wallet corrupted") << "\n";
Matt Corallo's avatar
Matt Corallo committed
366
        else if (nLoadWalletRet == DB_TOO_NEW)
367
            strErrors << _("Error loading wallet.dat: Wallet requires newer version of Bitcoin") << "\n";
368
369
        else if (nLoadWalletRet == DB_NEED_REWRITE)
        {
370
371
            strErrors << _("Wallet needed to be rewritten: restart Bitcoin to complete") << "\n";
            wxMessageBox(strErrors.str(), "Bitcoin", wxOK | wxICON_ERROR);
372
373
            return false;
        }
Matt Corallo's avatar
Matt Corallo committed
374
        else
375
            strErrors << _("Error loading wallet.dat") << "\n";
Matt Corallo's avatar
Matt Corallo committed
376
    }
377
378
    printf(" wallet      %15"PRI64d"ms\n", GetTimeMillis() - nStart);

379
380
    RegisterWallet(pwalletMain);

381
382
383
384
385
    CBlockIndex *pindexRescan = pindexBest;
    if (GetBoolArg("-rescan"))
        pindexRescan = pindexGenesisBlock;
    else
    {
386
        CWalletDB walletdb("wallet.dat");
387
388
389
390
391
392
        CBlockLocator locator;
        if (walletdb.ReadBestBlock(locator))
            pindexRescan = locator.GetBlockIndex();
    }
    if (pindexBest != pindexRescan)
    {
393
        InitMessage(_("Rescanning..."));
394
395
        printf("Rescanning last %i blocks (from block %i)...\n", pindexBest->nHeight - pindexRescan->nHeight, pindexRescan->nHeight);
        nStart = GetTimeMillis();
396
        pwalletMain->ScanForWalletTransactions(pindexRescan, true);
397
398
399
        printf(" rescan      %15"PRI64d"ms\n", GetTimeMillis() - nStart);
    }

400
    InitMessage(_("Done loading"));
401
402
    printf("Done loading\n");

403
404
405
406
407
408
    //// debug print
    printf("mapBlockIndex.size() = %d\n",   mapBlockIndex.size());
    printf("nBestHeight = %d\n",            nBestHeight);
    printf("setKeyPool.size() = %d\n",      pwalletMain->setKeyPool.size());
    printf("mapWallet.size() = %d\n",       pwalletMain->mapWallet.size());
    printf("mapAddressBook.size() = %d\n",  pwalletMain->mapAddressBook.size());
409

410
    if (!strErrors.str().empty())
411
    {
412
        wxMessageBox(strErrors.str(), "Bitcoin", wxOK | wxICON_ERROR);
413
414
415
416
        return false;
    }

    // Add wallet transactions that aren't already in a block to mapTransactions
417
    pwalletMain->ReacceptWalletTransactions();
418

419
420
421
422
    // Note: Bitcoin-QT stores several settings in the wallet, so we want
    // to load the wallet BEFORE parsing command-line arguments, so
    // the command-line/bitcoin.conf settings override GUI setting.

423
424
425
426
427
428
429
430
431
    //
    // Parameters
    //
    if (GetBoolArg("-printblockindex") || GetBoolArg("-printblocktree"))
    {
        PrintBlockTree();
        return false;
    }

432
433
434
435
436
437
438
    if (mapArgs.count("-timeout"))
    {
        int nNewTimeout = GetArg("-timeout", 5000);
        if (nNewTimeout > 0 && nNewTimeout < 600000)
            nConnectTimeout = nNewTimeout;
    }

439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
    if (mapArgs.count("-printblock"))
    {
        string strMatch = mapArgs["-printblock"];
        int nFound = 0;
        for (map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.begin(); mi != mapBlockIndex.end(); ++mi)
        {
            uint256 hash = (*mi).first;
            if (strncmp(hash.ToString().c_str(), strMatch.c_str(), strMatch.size()) == 0)
            {
                CBlockIndex* pindex = (*mi).second;
                CBlock block;
                block.ReadFromDisk(pindex);
                block.BuildMerkleTree();
                block.print();
                printf("\n");
                nFound++;
            }
        }
        if (nFound == 0)
            printf("No blocks matching %s were found\n", strMatch.c_str());
        return false;
    }

    fGenerateBitcoins = GetBoolArg("-gen");

    if (mapArgs.count("-proxy"))
    {
        fUseProxy = true;
Pieter Wuille's avatar
Pieter Wuille committed
467
        addrProxy = CService(mapArgs["-proxy"], 9050);
468
469
470
471
472
473
474
        if (!addrProxy.IsValid())
        {
            wxMessageBox(_("Invalid -proxy address"), "Bitcoin");
            return false;
        }
    }

Pieter Wuille's avatar
Pieter Wuille committed
475
    bool fTor = (fUseProxy && addrProxy.GetPort() == 9050);
476
477
    if (fTor)
    {
478
        // Use SoftSetBoolArg here so user can override any of these if they wish.
479
        // Note: the GetBoolArg() calls for all of these must happen later.
480
481
482
        SoftSetBoolArg("-listen", false);
        SoftSetBoolArg("-irc", false);
        SoftSetBoolArg("-dnsseed", false);
483
484
        SoftSetBoolArg("-upnp", false);
        SoftSetBoolArg("-dns", false);
485
486
487
    }

    fAllowDNS = GetBoolArg("-dns");
488
    fNoListen = !GetBoolArg("-listen", true);
489

490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
    // This code can be removed once a super-majority of the network has upgraded.
    if (GetBoolArg("-bip16", true))
    {
        if (fTestNet)
            SoftSetArg("-paytoscripthashtime", "1329264000"); // Feb 15
        else
            SoftSetArg("-paytoscripthashtime", "1330578000"); // Mar 1

        // Put "/P2SH/" in the coinbase so everybody can tell when
        // a majority of miners support it
        const char* pszP2SH = "/P2SH/";
        COINBASE_FLAGS << std::vector<unsigned char>(pszP2SH, pszP2SH+strlen(pszP2SH));
    }
    else
    {
        const char* pszP2SH = "NOP2SH";
        COINBASE_FLAGS << std::vector<unsigned char>(pszP2SH, pszP2SH+strlen(pszP2SH));
    }

509
    // Command-line args override in-wallet settings:
510
511
512
513
514
#if USE_UPNP
    fUseUPnP = GetBoolArg("-upnp", true);
#else
    fUseUPnP = GetBoolArg("-upnp", false);
#endif
515
516
517
518
519
520
521
522
523
524
525

    if (!fNoListen)
    {
        std::string strError;
        if (!BindListenPort(strError))
        {
            wxMessageBox(strError, "Bitcoin");
            return false;
        }
    }

526
527
528
529
    if (mapArgs.count("-addnode"))
    {
        BOOST_FOREACH(string strAddr, mapMultiArgs["-addnode"])
        {
Pieter Wuille's avatar
Pieter Wuille committed
530
            CAddress addr(CService(strAddr, GetDefaultPort(), fAllowDNS));
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
            addr.nTime = 0; // so it won't relay unless successfully connected
            if (addr.IsValid())
                AddAddress(addr);
        }
    }

    if (mapArgs.count("-paytxfee"))
    {
        if (!ParseMoney(mapArgs["-paytxfee"], nTransactionFee))
        {
            wxMessageBox(_("Invalid amount for -paytxfee=<amount>"), "Bitcoin");
            return false;
        }
        if (nTransactionFee > 0.25 * COIN)
            wxMessageBox(_("Warning: -paytxfee is set very high.  This is the transaction fee you will pay if you send a transaction."), "Bitcoin", wxOK | wxICON_EXCLAMATION);
    }

    //
Gavin Andresen's avatar
Gavin Andresen committed
549
    // Start the node
550
551
552
553
554
555
556
    //
    if (!CheckDiskSpace())
        return false;

    RandAddSeedPerfmon();

    if (!CreateThread(StartNode, NULL))
557
        wxMessageBox(_("Error: CreateThread(StartNode) failed"), "Bitcoin");
558
559
560
561

    if (fServer)
        CreateThread(ThreadRPCServer, NULL);

562
563
564
565
566
#ifdef QT_GUI
    if(GetStartOnSystemStartup())
        SetStartOnSystemStartup(true); // Remove startup links to bitcoin-wx
#endif

Gavin Andresen's avatar
Gavin Andresen committed
567
#if !defined(QT_GUI)
568
569
570
571
572
573
    while (1)
        Sleep(5000);
#endif

    return true;
}
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722

#ifdef WIN32
string StartupShortcutPath()
{
    return MyGetSpecialFolderPath(CSIDL_STARTUP, true) + "\\Bitcoin.lnk";
}

bool GetStartOnSystemStartup()
{
    return filesystem::exists(StartupShortcutPath().c_str());
}

bool SetStartOnSystemStartup(bool fAutoStart)
{
    // If the shortcut exists already, remove it for updating
    remove(StartupShortcutPath().c_str());

    if (fAutoStart)
    {
        CoInitialize(NULL);

        // Get a pointer to the IShellLink interface.
        IShellLink* psl = NULL;
        HRESULT hres = CoCreateInstance(CLSID_ShellLink, NULL,
                                CLSCTX_INPROC_SERVER, IID_IShellLink,
                                reinterpret_cast<void**>(&psl));

        if (SUCCEEDED(hres))
        {
            // Get the current executable path
            TCHAR pszExePath[MAX_PATH];
            GetModuleFileName(NULL, pszExePath, sizeof(pszExePath));

            TCHAR pszArgs[5] = TEXT("-min");

            // Set the path to the shortcut target
            psl->SetPath(pszExePath);
            PathRemoveFileSpec(pszExePath);
            psl->SetWorkingDirectory(pszExePath);
            psl->SetShowCmd(SW_SHOWMINNOACTIVE);
            psl->SetArguments(pszArgs);

            // Query IShellLink for the IPersistFile interface for
            // saving the shortcut in persistent storage.
            IPersistFile* ppf = NULL;
            hres = psl->QueryInterface(IID_IPersistFile,
                                       reinterpret_cast<void**>(&ppf));
            if (SUCCEEDED(hres))
            {
                WCHAR pwsz[MAX_PATH];
                // Ensure that the string is ANSI.
                MultiByteToWideChar(CP_ACP, 0, StartupShortcutPath().c_str(), -1, pwsz, MAX_PATH);
                // Save the link by calling IPersistFile::Save.
                hres = ppf->Save(pwsz, TRUE);
                ppf->Release();
                psl->Release();
                CoUninitialize();
                return true;
            }
            psl->Release();
        }
        CoUninitialize();
        return false;
    }
    return true;
}

#elif defined(LINUX)

// Follow the Desktop Application Autostart Spec:
//  http://standards.freedesktop.org/autostart-spec/autostart-spec-latest.html

boost::filesystem::path GetAutostartDir()
{
    namespace fs = boost::filesystem;

    char* pszConfigHome = getenv("XDG_CONFIG_HOME");
    if (pszConfigHome) return fs::path(pszConfigHome) / fs::path("autostart");
    char* pszHome = getenv("HOME");
    if (pszHome) return fs::path(pszHome) / fs::path(".config/autostart");
    return fs::path();
}

boost::filesystem::path GetAutostartFilePath()
{
    return GetAutostartDir() / boost::filesystem::path("bitcoin.desktop");
}

bool GetStartOnSystemStartup()
{
    boost::filesystem::ifstream optionFile(GetAutostartFilePath());
    if (!optionFile.good())
        return false;
    // Scan through file for "Hidden=true":
    string line;
    while (!optionFile.eof())
    {
        getline(optionFile, line);
        if (line.find("Hidden") != string::npos &&
            line.find("true") != string::npos)
            return false;
    }
    optionFile.close();

    return true;
}

bool SetStartOnSystemStartup(bool fAutoStart)
{
    if (!fAutoStart)
    {
#if defined(BOOST_FILESYSTEM_VERSION) && BOOST_FILESYSTEM_VERSION >= 3
        unlink(GetAutostartFilePath().string().c_str());
#else
        unlink(GetAutostartFilePath().native_file_string().c_str());
#endif
    }
    else
    {
        char pszExePath[MAX_PATH+1];
        memset(pszExePath, 0, sizeof(pszExePath));
        if (readlink("/proc/self/exe", pszExePath, sizeof(pszExePath)-1) == -1)
            return false;

        boost::filesystem::create_directories(GetAutostartDir());

        boost::filesystem::ofstream optionFile(GetAutostartFilePath(), ios_base::out|ios_base::trunc);
        if (!optionFile.good())
            return false;
        // Write a bitcoin.desktop file to the autostart directory:
        optionFile << "[Desktop Entry]\n";
        optionFile << "Type=Application\n";
        optionFile << "Name=Bitcoin\n";
        optionFile << "Exec=" << pszExePath << " -min\n";
        optionFile << "Terminal=false\n";
        optionFile << "Hidden=false\n";
        optionFile.close();
    }
    return true;
}
#else

// TODO: OSX startup stuff; see:
// http://developer.apple.com/mac/library/documentation/MacOSX/Conceptual/BPSystemStartup/Articles/CustomLogin.html

bool GetStartOnSystemStartup() { return false; }
bool SetStartOnSystemStartup(bool fAutoStart) { return false; }

#endif