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
    //
Gavin Andresen's avatar
Gavin Andresen committed
153
    // If Qt is used, parameters/bitcoin.conf are parsed in qt/bitcoin.cpp's main()
154
#if !defined(QT_GUI)
155
    ParseParameters(argc, argv);
Gavin Andresen's avatar
Gavin Andresen committed
156
    if (!ReadConfigFile(mapArgs, mapMultiArgs))
157
    {
Gavin Andresen's avatar
Gavin Andresen committed
158
159
        fprintf(stderr, "Error: Specified directory does not exist\n");
        Shutdown(NULL);
160
    }
Gavin Andresen's avatar
Gavin Andresen committed
161
#endif
162
163
164
165
166
167

    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" +
168
            "  bitcoind [options]                   \t  " + "\n" +
169
170
171
172
173
174
175
176
177
            "  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" +
178
            "  -splash          \t\t  " + _("Show splash screen on startup (default: 1)") + "\n" +
179
180
181
182
183
184
            "  -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" +
185
            "  -addnode=<ip>    \t  "   + _("Add a node to connect to and attempt to keep the connection open") + "\n" +
186
            "  -connect=<ip>    \t\t  " + _("Connect only to the specified node") + "\n" +
187
188
            "  -irc             \t  "   + _("Find peers using internet relay chat (default: 0)") + "\n" +
            "  -listen          \t  "   + _("Accept connections from outside (default: 1)") + "\n" +
189
190
191
#ifdef QT_GUI
            "  -lang=<lang>     \t\t  " + _("Set language, for example \"de_DE\" (default: system locale)") + "\n" +
#endif
192
            "  -dnsseed         \t  "   + _("Find peers using DNS lookup (default: 1)") + "\n" +
193
194
195
196
            "  -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" +
197
198
#ifdef USE_UPNP
#if USE_UPNP
199
            "  -upnp            \t  "   + _("Use Universal Plug and Play to map the listening port (default: 1)") + "\n" +
200
#else
201
            "  -upnp            \t  "   + _("Use Universal Plug and Play to map the listening port (default: 0)") + "\n" +
202
203
#endif
#endif
204
            "  -paytxfee=<amt>  \t  "   + _("Fee per KB to add to transactions you send") + "\n" +
205
#ifdef QT_GUI
206
            "  -server          \t\t  " + _("Accept command line and JSON-RPC commands") + "\n" +
207
#endif
208
#if !defined(WIN32) && !defined(QT_GUI)
209
            "  -daemon          \t\t  " + _("Run in the background as a daemon and accept commands") + "\n" +
210
#endif
211
212
213
214
            "  -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" +
215
#ifdef WIN32
216
            "  -printtodebugger \t  "   + _("Send trace/debug info to debugger") + "\n" +
217
#endif
218
219
220
221
222
            "  -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" +
223
            "  -blocknotify=<cmd> "     + _("Execute command when the best block changes (%s in cmd is replaced by block hash)") + "\n" +
224
225
            "  -keypool=<n>     \t  "   + _("Set key pool size to <n> (default: 100)") + "\n" +
            "  -rescan          \t  "   + _("Rescan the block chain for missing wallet transactions") + "\n";
226
227
228

#ifdef USE_SSL
        strUsage += string() +
229
230
231
232
233
            _("\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";
234
235
236
#endif

        strUsage += string() +
237
            "  -?               \t\t  " + _("This help message") + "\n";
238
239
240

        // Remove tabs
        strUsage.erase(std::remove(strUsage.begin(), strUsage.end(), '\t'), strUsage.end());
241
242
243
244
#if defined(QT_GUI) && defined(WIN32)
        // On windows, show a message box, as there is no stderr
        wxMessageBox(strUsage, "Usage");
#else
245
        fprintf(stderr, "%s", strUsage.c_str());
246
#endif
247
248
249
        return false;
    }

250
    fTestNet = GetBoolArg("-testnet");
251
252
253
254
255
    if (fTestNet)
    {
        SoftSetBoolArg("-irc", true);
    }

256
257
    fDebug = GetBoolArg("-debug");

258
#if !defined(WIN32) && !defined(QT_GUI)
259
260
261
262
263
264
265
266
267
268
269
    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
270
#if !defined(QT_GUI)
271
    fServer = true;
272
#endif
273
274
275
276
    fPrintToConsole = GetBoolArg("-printtoconsole");
    fPrintToDebugger = GetBoolArg("-printtodebugger");
    fLogTimestamps = GetBoolArg("-logtimestamps");

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

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

289
#if !defined(WIN32) && !defined(QT_GUI)
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
334
335
    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;
    }

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

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

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

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

383
384
    RegisterWallet(pwalletMain);

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

404
    InitMessage(_("Done loading"));
405
406
    printf("Done loading\n");

407
408
409
410
411
412
    //// 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());
413

414
    if (!strErrors.str().empty())
415
    {
416
        wxMessageBox(strErrors.str(), "Bitcoin", wxOK | wxICON_ERROR);
417
418
419
420
        return false;
    }

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

423
424
425
426
    // 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.

427
428
429
430
431
432
433
434
435
    //
    // Parameters
    //
    if (GetBoolArg("-printblockindex") || GetBoolArg("-printblocktree"))
    {
        PrintBlockTree();
        return false;
    }

436
437
438
439
440
441
442
    if (mapArgs.count("-timeout"))
    {
        int nNewTimeout = GetArg("-timeout", 5000);
        if (nNewTimeout > 0 && nNewTimeout < 600000)
            nConnectTimeout = nNewTimeout;
    }

443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
    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;
    }

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

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

    fAllowDNS = GetBoolArg("-dns");
490
    fNoListen = !GetBoolArg("-listen", true);
491

492
493
494
495
496
497
    // 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
498
            SoftSetArg("-paytoscripthashtime", "1333238400"); // April 1 2012
499
500
501
502
503
504
505
506
507
508
509
510

        // 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));
    }

511
512
513
514
515
516
517
518
519
520
    if (!fNoListen)
    {
        std::string strError;
        if (!BindListenPort(strError))
        {
            wxMessageBox(strError, "Bitcoin");
            return false;
        }
    }

521
522
523
524
    if (mapArgs.count("-addnode"))
    {
        BOOST_FOREACH(string strAddr, mapMultiArgs["-addnode"])
        {
Pieter Wuille's avatar
Pieter Wuille committed
525
            CAddress addr(CService(strAddr, GetDefaultPort(), fAllowDNS));
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
            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
544
    // Start the node
545
546
547
548
549
550
551
    //
    if (!CheckDiskSpace())
        return false;

    RandAddSeedPerfmon();

    if (!CreateThread(StartNode, NULL))
552
        wxMessageBox(_("Error: CreateThread(StartNode) failed"), "Bitcoin");
553
554
555
556

    if (fServer)
        CreateThread(ThreadRPCServer, NULL);

557
558
559
560
561
#ifdef QT_GUI
    if(GetStartOnSystemStartup())
        SetStartOnSystemStartup(true); // Remove startup links to bitcoin-wx
#endif

Gavin Andresen's avatar
Gavin Andresen committed
562
#if !defined(QT_GUI)
563
564
565
566
567
568
    while (1)
        Sleep(5000);
#endif

    return true;
}
569
570
571
572
573
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

#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