Merge branch 'nodosbip16' of https://github.com/sipa/bitcoin
[novacoin.git] / src / main.cpp
1 // Copyright (c) 2009-2010 Satoshi Nakamoto
2 // Copyright (c) 2009-2012 The Bitcoin developers
3 // Distributed under the MIT/X11 software license, see the accompanying
4 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
5 #include "headers.h"
6 #include "checkpoints.h"
7 #include "db.h"
8 #include "net.h"
9 #include "init.h"
10 #include <boost/algorithm/string/replace.hpp>
11 #include <boost/filesystem.hpp>
12 #include <boost/filesystem/fstream.hpp>
13
14 using namespace std;
15 using namespace boost;
16
17 //
18 // Global state
19 //
20
21 // Name of client reported in the 'version' message. Report the same name
22 // for both bitcoind and bitcoin-qt, to make it harder for attackers to
23 // target servers or GUI users specifically.
24 const std::string CLIENT_NAME("Satoshi");
25
26 CCriticalSection cs_setpwalletRegistered;
27 set<CWallet*> setpwalletRegistered;
28
29 CCriticalSection cs_main;
30
31 static map<uint256, CTransaction> mapTransactions;
32 CCriticalSection cs_mapTransactions;
33 unsigned int nTransactionsUpdated = 0;
34 map<COutPoint, CInPoint> mapNextTx;
35
36 map<uint256, CBlockIndex*> mapBlockIndex;
37 uint256 hashGenesisBlock("0x000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f");
38 static CBigNum bnProofOfWorkLimit(~uint256(0) >> 32);
39 CBlockIndex* pindexGenesisBlock = NULL;
40 int nBestHeight = -1;
41 CBigNum bnBestChainWork = 0;
42 CBigNum bnBestInvalidWork = 0;
43 uint256 hashBestChain = 0;
44 CBlockIndex* pindexBest = NULL;
45 int64 nTimeBestReceived = 0;
46
47 CMedianFilter<int> cPeerBlockCounts(5, 0); // Amount of blocks that other nodes claim to have
48
49 map<uint256, CBlock*> mapOrphanBlocks;
50 multimap<uint256, CBlock*> mapOrphanBlocksByPrev;
51
52 map<uint256, CDataStream*> mapOrphanTransactions;
53 multimap<uint256, CDataStream*> mapOrphanTransactionsByPrev;
54
55 // Constant stuff for coinbase transactions we create:
56 CScript COINBASE_FLAGS;
57
58 const string strMessageMagic = "Bitcoin Signed Message:\n";
59
60 double dHashesPerSec;
61 int64 nHPSTimerStart;
62
63 // Settings
64 int64 nTransactionFee = 0;
65
66
67
68 //////////////////////////////////////////////////////////////////////////////
69 //
70 // dispatching functions
71 //
72
73 // These functions dispatch to one or all registered wallets
74
75
76 void RegisterWallet(CWallet* pwalletIn)
77 {
78     CRITICAL_BLOCK(cs_setpwalletRegistered)
79     {
80         setpwalletRegistered.insert(pwalletIn);
81     }
82 }
83
84 void UnregisterWallet(CWallet* pwalletIn)
85 {
86     CRITICAL_BLOCK(cs_setpwalletRegistered)
87     {
88         setpwalletRegistered.erase(pwalletIn);
89     }
90 }
91
92 // check whether the passed transaction is from us
93 bool static IsFromMe(CTransaction& tx)
94 {
95     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
96         if (pwallet->IsFromMe(tx))
97             return true;
98     return false;
99 }
100
101 // get the wallet transaction with the given hash (if it exists)
102 bool static GetTransaction(const uint256& hashTx, CWalletTx& wtx)
103 {
104     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
105         if (pwallet->GetTransaction(hashTx,wtx))
106             return true;
107     return false;
108 }
109
110 // erases transaction with the given hash from all wallets
111 void static EraseFromWallets(uint256 hash)
112 {
113     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
114         pwallet->EraseFromWallet(hash);
115 }
116
117 // make sure all wallets know about the given transaction, in the given block
118 void static SyncWithWallets(const CTransaction& tx, const CBlock* pblock = NULL, bool fUpdate = false)
119 {
120     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
121         pwallet->AddToWalletIfInvolvingMe(tx, pblock, fUpdate);
122 }
123
124 // notify wallets about a new best chain
125 void static SetBestChain(const CBlockLocator& loc)
126 {
127     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
128         pwallet->SetBestChain(loc);
129 }
130
131 // notify wallets about an updated transaction
132 void static UpdatedTransaction(const uint256& hashTx)
133 {
134     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
135         pwallet->UpdatedTransaction(hashTx);
136 }
137
138 // dump all wallets
139 void static PrintWallets(const CBlock& block)
140 {
141     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
142         pwallet->PrintWallet(block);
143 }
144
145 // notify wallets about an incoming inventory (for request counts)
146 void static Inventory(const uint256& hash)
147 {
148     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
149         pwallet->Inventory(hash);
150 }
151
152 // ask wallets to resend their transactions
153 void static ResendWalletTransactions()
154 {
155     BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
156         pwallet->ResendWalletTransactions();
157 }
158
159
160
161
162
163
164
165 //////////////////////////////////////////////////////////////////////////////
166 //
167 // mapOrphanTransactions
168 //
169
170 void static AddOrphanTx(const CDataStream& vMsg)
171 {
172     CTransaction tx;
173     CDataStream(vMsg) >> tx;
174     uint256 hash = tx.GetHash();
175     if (mapOrphanTransactions.count(hash))
176         return;
177     CDataStream* pvMsg = mapOrphanTransactions[hash] = new CDataStream(vMsg);
178     BOOST_FOREACH(const CTxIn& txin, tx.vin)
179         mapOrphanTransactionsByPrev.insert(make_pair(txin.prevout.hash, pvMsg));
180 }
181
182 void static EraseOrphanTx(uint256 hash)
183 {
184     if (!mapOrphanTransactions.count(hash))
185         return;
186     const CDataStream* pvMsg = mapOrphanTransactions[hash];
187     CTransaction tx;
188     CDataStream(*pvMsg) >> tx;
189     BOOST_FOREACH(const CTxIn& txin, tx.vin)
190     {
191         for (multimap<uint256, CDataStream*>::iterator mi = mapOrphanTransactionsByPrev.lower_bound(txin.prevout.hash);
192              mi != mapOrphanTransactionsByPrev.upper_bound(txin.prevout.hash);)
193         {
194             if ((*mi).second == pvMsg)
195                 mapOrphanTransactionsByPrev.erase(mi++);
196             else
197                 mi++;
198         }
199     }
200     delete pvMsg;
201     mapOrphanTransactions.erase(hash);
202 }
203
204
205
206
207
208
209
210
211 //////////////////////////////////////////////////////////////////////////////
212 //
213 // CTransaction and CTxIndex
214 //
215
216 bool CTransaction::ReadFromDisk(CTxDB& txdb, COutPoint prevout, CTxIndex& txindexRet)
217 {
218     SetNull();
219     if (!txdb.ReadTxIndex(prevout.hash, txindexRet))
220         return false;
221     if (!ReadFromDisk(txindexRet.pos))
222         return false;
223     if (prevout.n >= vout.size())
224     {
225         SetNull();
226         return false;
227     }
228     return true;
229 }
230
231 bool CTransaction::ReadFromDisk(CTxDB& txdb, COutPoint prevout)
232 {
233     CTxIndex txindex;
234     return ReadFromDisk(txdb, prevout, txindex);
235 }
236
237 bool CTransaction::ReadFromDisk(COutPoint prevout)
238 {
239     CTxDB txdb("r");
240     CTxIndex txindex;
241     return ReadFromDisk(txdb, prevout, txindex);
242 }
243
244 bool CTransaction::IsStandard() const
245 {
246     BOOST_FOREACH(const CTxIn& txin, vin)
247     {
248         // Biggest 'standard' txin is a 3-signature 3-of-3 CHECKMULTISIG
249         // pay-to-script-hash, which is 3 ~80-byte signatures, 3
250         // ~65-byte public keys, plus a few script ops.
251         if (txin.scriptSig.size() > 500)
252             return false;
253         if (!txin.scriptSig.IsPushOnly())
254             return false;
255     }
256     BOOST_FOREACH(const CTxOut& txout, vout)
257         if (!::IsStandard(txout.scriptPubKey))
258             return false;
259     return true;
260 }
261
262 //
263 // Check transaction inputs, and make sure any
264 // pay-to-script-hash transactions are evaluating IsStandard scripts
265 //
266 // Why bother? To avoid denial-of-service attacks; an attacker
267 // can submit a standard HASH... OP_EQUAL transaction,
268 // which will get accepted into blocks. The redemption
269 // script can be anything; an attacker could use a very
270 // expensive-to-check-upon-redemption script like:
271 //   DUP CHECKSIG DROP ... repeated 100 times... OP_1
272 //
273 bool CTransaction::AreInputsStandard(const MapPrevTx& mapInputs) const
274 {
275     if (IsCoinBase())
276         return true; // Coinbases don't use vin normally
277
278     for (int i = 0; i < vin.size(); i++)
279     {
280         const CTxOut& prev = GetOutputFor(vin[i], mapInputs);
281
282         vector<vector<unsigned char> > vSolutions;
283         txnouttype whichType;
284         // get the scriptPubKey corresponding to this input:
285         const CScript& prevScript = prev.scriptPubKey;
286         if (!Solver(prevScript, whichType, vSolutions))
287             return false;
288         int nArgsExpected = ScriptSigArgsExpected(whichType, vSolutions);
289
290         // Transactions with extra stuff in their scriptSigs are
291         // non-standard. Note that this EvalScript() call will
292         // be quick, because if there are any operations
293         // beside "push data" in the scriptSig the
294         // IsStandard() call returns false
295         vector<vector<unsigned char> > stack;
296         if (!EvalScript(stack, vin[i].scriptSig, *this, i, 0))
297             return false;
298
299         if (whichType == TX_SCRIPTHASH)
300         {
301             if (stack.empty())
302                 return false;
303             CScript subscript(stack.back().begin(), stack.back().end());
304             vector<vector<unsigned char> > vSolutions2;
305             txnouttype whichType2;
306             if (!Solver(subscript, whichType2, vSolutions2))
307                 return false;
308             if (whichType2 == TX_SCRIPTHASH)
309                 return false;
310             nArgsExpected += ScriptSigArgsExpected(whichType2, vSolutions2);
311         }
312
313         if (stack.size() != nArgsExpected)
314             return false;
315     }
316
317     return true;
318 }
319
320 int
321 CTransaction::GetLegacySigOpCount() const
322 {
323     int nSigOps = 0;
324     BOOST_FOREACH(const CTxIn& txin, vin)
325     {
326         nSigOps += txin.scriptSig.GetSigOpCount(false);
327     }
328     BOOST_FOREACH(const CTxOut& txout, vout)
329     {
330         nSigOps += txout.scriptPubKey.GetSigOpCount(false);
331     }
332     return nSigOps;
333 }
334
335
336 int CMerkleTx::SetMerkleBranch(const CBlock* pblock)
337 {
338     if (fClient)
339     {
340         if (hashBlock == 0)
341             return 0;
342     }
343     else
344     {
345         CBlock blockTmp;
346         if (pblock == NULL)
347         {
348             // Load the block this tx is in
349             CTxIndex txindex;
350             if (!CTxDB("r").ReadTxIndex(GetHash(), txindex))
351                 return 0;
352             if (!blockTmp.ReadFromDisk(txindex.pos.nFile, txindex.pos.nBlockPos))
353                 return 0;
354             pblock = &blockTmp;
355         }
356
357         // Update the tx's hashBlock
358         hashBlock = pblock->GetHash();
359
360         // Locate the transaction
361         for (nIndex = 0; nIndex < pblock->vtx.size(); nIndex++)
362             if (pblock->vtx[nIndex] == *(CTransaction*)this)
363                 break;
364         if (nIndex == pblock->vtx.size())
365         {
366             vMerkleBranch.clear();
367             nIndex = -1;
368             printf("ERROR: SetMerkleBranch() : couldn't find tx in block\n");
369             return 0;
370         }
371
372         // Fill in merkle branch
373         vMerkleBranch = pblock->GetMerkleBranch(nIndex);
374     }
375
376     // Is the tx in a block that's in the main chain
377     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashBlock);
378     if (mi == mapBlockIndex.end())
379         return 0;
380     CBlockIndex* pindex = (*mi).second;
381     if (!pindex || !pindex->IsInMainChain())
382         return 0;
383
384     return pindexBest->nHeight - pindex->nHeight + 1;
385 }
386
387
388
389
390
391
392
393 bool CTransaction::CheckTransaction() const
394 {
395     // Basic checks that don't depend on any context
396     if (vin.empty())
397         return DoS(10, error("CTransaction::CheckTransaction() : vin empty"));
398     if (vout.empty())
399         return DoS(10, error("CTransaction::CheckTransaction() : vout empty"));
400     // Size limits
401     if (::GetSerializeSize(*this, SER_NETWORK) > MAX_BLOCK_SIZE)
402         return DoS(100, error("CTransaction::CheckTransaction() : size limits failed"));
403
404     // Check for negative or overflow output values
405     int64 nValueOut = 0;
406     BOOST_FOREACH(const CTxOut& txout, vout)
407     {
408         if (txout.nValue < 0)
409             return DoS(100, error("CTransaction::CheckTransaction() : txout.nValue negative"));
410         if (txout.nValue > MAX_MONEY)
411             return DoS(100, error("CTransaction::CheckTransaction() : txout.nValue too high"));
412         nValueOut += txout.nValue;
413         if (!MoneyRange(nValueOut))
414             return DoS(100, error("CTransaction::CheckTransaction() : txout total out of range"));
415     }
416
417     // Check for duplicate inputs
418     set<COutPoint> vInOutPoints;
419     BOOST_FOREACH(const CTxIn& txin, vin)
420     {
421         if (vInOutPoints.count(txin.prevout))
422             return false;
423         vInOutPoints.insert(txin.prevout);
424     }
425
426     if (IsCoinBase())
427     {
428         if (vin[0].scriptSig.size() < 2 || vin[0].scriptSig.size() > 100)
429             return DoS(100, error("CTransaction::CheckTransaction() : coinbase script size"));
430     }
431     else
432     {
433         BOOST_FOREACH(const CTxIn& txin, vin)
434             if (txin.prevout.IsNull())
435                 return DoS(10, error("CTransaction::CheckTransaction() : prevout is null"));
436     }
437
438     return true;
439 }
440
441 bool CTransaction::AcceptToMemoryPool(CTxDB& txdb, bool fCheckInputs, bool* pfMissingInputs)
442 {
443     if (pfMissingInputs)
444         *pfMissingInputs = false;
445
446     if (!CheckTransaction())
447         return error("AcceptToMemoryPool() : CheckTransaction failed");
448
449     // Coinbase is only valid in a block, not as a loose transaction
450     if (IsCoinBase())
451         return DoS(100, error("AcceptToMemoryPool() : coinbase as individual tx"));
452
453     // To help v0.1.5 clients who would see it as a negative number
454     if ((int64)nLockTime > std::numeric_limits<int>::max())
455         return error("AcceptToMemoryPool() : not accepting nLockTime beyond 2038 yet");
456
457     // Rather not work on nonstandard transactions (unless -testnet)
458     if (!fTestNet && !IsStandard())
459         return error("AcceptToMemoryPool() : nonstandard transaction type");
460
461     // Do we already have it?
462     uint256 hash = GetHash();
463     CRITICAL_BLOCK(cs_mapTransactions)
464         if (mapTransactions.count(hash))
465             return false;
466     if (fCheckInputs)
467         if (txdb.ContainsTx(hash))
468             return false;
469
470     // Check for conflicts with in-memory transactions
471     CTransaction* ptxOld = NULL;
472     for (int i = 0; i < vin.size(); i++)
473     {
474         COutPoint outpoint = vin[i].prevout;
475         if (mapNextTx.count(outpoint))
476         {
477             // Disable replacement feature for now
478             return false;
479
480             // Allow replacing with a newer version of the same transaction
481             if (i != 0)
482                 return false;
483             ptxOld = mapNextTx[outpoint].ptx;
484             if (ptxOld->IsFinal())
485                 return false;
486             if (!IsNewerThan(*ptxOld))
487                 return false;
488             for (int i = 0; i < vin.size(); i++)
489             {
490                 COutPoint outpoint = vin[i].prevout;
491                 if (!mapNextTx.count(outpoint) || mapNextTx[outpoint].ptx != ptxOld)
492                     return false;
493             }
494             break;
495         }
496     }
497
498     if (fCheckInputs)
499     {
500         MapPrevTx mapInputs;
501         map<uint256, CTxIndex> mapUnused;
502         bool fInvalid = false;
503         if (!FetchInputs(txdb, mapUnused, false, false, mapInputs, fInvalid))
504         {
505             if (fInvalid)
506                 return error("AcceptToMemoryPool() : FetchInputs found invalid tx %s", hash.ToString().substr(0,10).c_str());
507             if (pfMissingInputs)
508                 *pfMissingInputs = true;
509             return error("AcceptToMemoryPool() : FetchInputs failed %s", hash.ToString().substr(0,10).c_str());
510         }
511
512         // Check for non-standard pay-to-script-hash in inputs
513         if (!AreInputsStandard(mapInputs) && !fTestNet)
514             return error("AcceptToMemoryPool() : nonstandard transaction input");
515
516         // Note: if you modify this code to accept non-standard transactions, then
517         // you should add code here to check that the transaction does a
518         // reasonable number of ECDSA signature verifications.
519
520         int64 nFees = GetValueIn(mapInputs)-GetValueOut();
521         unsigned int nSize = ::GetSerializeSize(*this, SER_NETWORK);
522
523         // Don't accept it if it can't get into a block
524         if (nFees < GetMinFee(1000, true, GMF_RELAY))
525             return error("AcceptToMemoryPool() : not enough fees");
526
527         // Continuously rate-limit free transactions
528         // This mitigates 'penny-flooding' -- sending thousands of free transactions just to
529         // be annoying or make other's transactions take longer to confirm.
530         if (nFees < MIN_RELAY_TX_FEE)
531         {
532             static CCriticalSection cs;
533             static double dFreeCount;
534             static int64 nLastTime;
535             int64 nNow = GetTime();
536
537             CRITICAL_BLOCK(cs)
538             {
539                 // Use an exponentially decaying ~10-minute window:
540                 dFreeCount *= pow(1.0 - 1.0/600.0, (double)(nNow - nLastTime));
541                 nLastTime = nNow;
542                 // -limitfreerelay unit is thousand-bytes-per-minute
543                 // At default rate it would take over a month to fill 1GB
544                 if (dFreeCount > GetArg("-limitfreerelay", 15)*10*1000 && !IsFromMe(*this))
545                     return error("AcceptToMemoryPool() : free transaction rejected by rate limiter");
546                 if (fDebug)
547                     printf("Rate limit dFreeCount: %g => %g\n", dFreeCount, dFreeCount+nSize);
548                 dFreeCount += nSize;
549             }
550         }
551
552         // Check against previous transactions
553         // This is done last to help prevent CPU exhaustion denial-of-service attacks.
554         if (!ConnectInputs(mapInputs, mapUnused, CDiskTxPos(1,1,1), pindexBest, false, false))
555         {
556             return error("AcceptToMemoryPool() : ConnectInputs failed %s", hash.ToString().substr(0,10).c_str());
557         }
558     }
559
560     // Store transaction in memory
561     CRITICAL_BLOCK(cs_mapTransactions)
562     {
563         if (ptxOld)
564         {
565             printf("AcceptToMemoryPool() : replacing tx %s with new version\n", ptxOld->GetHash().ToString().c_str());
566             ptxOld->RemoveFromMemoryPool();
567         }
568         AddToMemoryPoolUnchecked();
569     }
570
571     ///// are we sure this is ok when loading transactions or restoring block txes
572     // If updated, erase old tx from wallet
573     if (ptxOld)
574         EraseFromWallets(ptxOld->GetHash());
575
576     printf("AcceptToMemoryPool(): accepted %s\n", hash.ToString().substr(0,10).c_str());
577     return true;
578 }
579
580 bool CTransaction::AcceptToMemoryPool(bool fCheckInputs, bool* pfMissingInputs)
581 {
582     CTxDB txdb("r");
583     return AcceptToMemoryPool(txdb, fCheckInputs, pfMissingInputs);
584 }
585
586 uint64 nPooledTx = 0;
587
588 bool CTransaction::AddToMemoryPoolUnchecked()
589 {
590     printf("AcceptToMemoryPoolUnchecked(): size %lu\n",  mapTransactions.size());
591     // Add to memory pool without checking anything.  Don't call this directly,
592     // call AcceptToMemoryPool to properly check the transaction first.
593     CRITICAL_BLOCK(cs_mapTransactions)
594     {
595         uint256 hash = GetHash();
596         mapTransactions[hash] = *this;
597         for (int i = 0; i < vin.size(); i++)
598             mapNextTx[vin[i].prevout] = CInPoint(&mapTransactions[hash], i);
599         nTransactionsUpdated++;
600         ++nPooledTx;
601     }
602     return true;
603 }
604
605
606 bool CTransaction::RemoveFromMemoryPool()
607 {
608     // Remove transaction from memory pool
609     CRITICAL_BLOCK(cs_mapTransactions)
610     {
611         uint256 hash = GetHash();
612         if (mapTransactions.count(hash))
613         {
614             BOOST_FOREACH(const CTxIn& txin, vin)
615                 mapNextTx.erase(txin.prevout);
616             mapTransactions.erase(hash);
617             nTransactionsUpdated++;
618             --nPooledTx;
619         }
620     }
621     return true;
622 }
623
624
625
626
627
628
629 int CMerkleTx::GetDepthInMainChain(CBlockIndex* &pindexRet) const
630 {
631     if (hashBlock == 0 || nIndex == -1)
632         return 0;
633
634     // Find the block it claims to be in
635     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashBlock);
636     if (mi == mapBlockIndex.end())
637         return 0;
638     CBlockIndex* pindex = (*mi).second;
639     if (!pindex || !pindex->IsInMainChain())
640         return 0;
641
642     // Make sure the merkle branch connects to this block
643     if (!fMerkleVerified)
644     {
645         if (CBlock::CheckMerkleBranch(GetHash(), vMerkleBranch, nIndex) != pindex->hashMerkleRoot)
646             return 0;
647         fMerkleVerified = true;
648     }
649
650     pindexRet = pindex;
651     return pindexBest->nHeight - pindex->nHeight + 1;
652 }
653
654
655 int CMerkleTx::GetBlocksToMaturity() const
656 {
657     if (!IsCoinBase())
658         return 0;
659     return max(0, (COINBASE_MATURITY+20) - GetDepthInMainChain());
660 }
661
662
663 bool CMerkleTx::AcceptToMemoryPool(CTxDB& txdb, bool fCheckInputs)
664 {
665     if (fClient)
666     {
667         if (!IsInMainChain() && !ClientConnectInputs())
668             return false;
669         return CTransaction::AcceptToMemoryPool(txdb, false);
670     }
671     else
672     {
673         return CTransaction::AcceptToMemoryPool(txdb, fCheckInputs);
674     }
675 }
676
677 bool CMerkleTx::AcceptToMemoryPool()
678 {
679     CTxDB txdb("r");
680     return AcceptToMemoryPool(txdb);
681 }
682
683
684
685 bool CWalletTx::AcceptWalletTransaction(CTxDB& txdb, bool fCheckInputs)
686 {
687     CRITICAL_BLOCK(cs_mapTransactions)
688     {
689         // Add previous supporting transactions first
690         BOOST_FOREACH(CMerkleTx& tx, vtxPrev)
691         {
692             if (!tx.IsCoinBase())
693             {
694                 uint256 hash = tx.GetHash();
695                 if (!mapTransactions.count(hash) && !txdb.ContainsTx(hash))
696                     tx.AcceptToMemoryPool(txdb, fCheckInputs);
697             }
698         }
699         return AcceptToMemoryPool(txdb, fCheckInputs);
700     }
701     return false;
702 }
703
704 bool CWalletTx::AcceptWalletTransaction() 
705 {
706     CTxDB txdb("r");
707     return AcceptWalletTransaction(txdb);
708 }
709
710 int CTxIndex::GetDepthInMainChain() const
711 {
712     // Read block header
713     CBlock block;
714     if (!block.ReadFromDisk(pos.nFile, pos.nBlockPos, false))
715         return 0;
716     // Find the block in the index
717     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(block.GetHash());
718     if (mi == mapBlockIndex.end())
719         return 0;
720     CBlockIndex* pindex = (*mi).second;
721     if (!pindex || !pindex->IsInMainChain())
722         return 0;
723     return 1 + nBestHeight - pindex->nHeight;
724 }
725
726
727
728
729
730
731
732
733
734
735 //////////////////////////////////////////////////////////////////////////////
736 //
737 // CBlock and CBlockIndex
738 //
739
740 bool CBlock::ReadFromDisk(const CBlockIndex* pindex, bool fReadTransactions)
741 {
742     if (!fReadTransactions)
743     {
744         *this = pindex->GetBlockHeader();
745         return true;
746     }
747     if (!ReadFromDisk(pindex->nFile, pindex->nBlockPos, fReadTransactions))
748         return false;
749     if (GetHash() != pindex->GetBlockHash())
750         return error("CBlock::ReadFromDisk() : GetHash() doesn't match index");
751     return true;
752 }
753
754 uint256 static GetOrphanRoot(const CBlock* pblock)
755 {
756     // Work back to the first block in the orphan chain
757     while (mapOrphanBlocks.count(pblock->hashPrevBlock))
758         pblock = mapOrphanBlocks[pblock->hashPrevBlock];
759     return pblock->GetHash();
760 }
761
762 int64 static GetBlockValue(int nHeight, int64 nFees)
763 {
764     int64 nSubsidy = 50 * COIN;
765
766     // Subsidy is cut in half every 4 years
767     nSubsidy >>= (nHeight / 210000);
768
769     return nSubsidy + nFees;
770 }
771
772 static const int64 nTargetTimespan = 14 * 24 * 60 * 60; // two weeks
773 static const int64 nTargetSpacing = 10 * 60;
774 static const int64 nInterval = nTargetTimespan / nTargetSpacing;
775
776 //
777 // minimum amount of work that could possibly be required nTime after
778 // minimum work required was nBase
779 //
780 unsigned int ComputeMinWork(unsigned int nBase, int64 nTime)
781 {
782     // Testnet has min-difficulty blocks
783     // after nTargetSpacing*2 time between blocks:
784     if (fTestNet && nTime > nTargetSpacing*2)
785         return bnProofOfWorkLimit.GetCompact();
786
787     CBigNum bnResult;
788     bnResult.SetCompact(nBase);
789     while (nTime > 0 && bnResult < bnProofOfWorkLimit)
790     {
791         // Maximum 400% adjustment...
792         bnResult *= 4;
793         // ... in best-case exactly 4-times-normal target time
794         nTime -= nTargetTimespan*4;
795     }
796     if (bnResult > bnProofOfWorkLimit)
797         bnResult = bnProofOfWorkLimit;
798     return bnResult.GetCompact();
799 }
800
801 unsigned int static GetNextWorkRequired(const CBlockIndex* pindexLast, const CBlock *pblock)
802 {
803     unsigned int nProofOfWorkLimit = bnProofOfWorkLimit.GetCompact();
804
805     // Genesis block
806     if (pindexLast == NULL)
807         return nProofOfWorkLimit;
808
809     // Only change once per interval
810     if ((pindexLast->nHeight+1) % nInterval != 0)
811     {
812         // Special rules for testnet after 15 Feb 2012:
813         if (fTestNet && pblock->nTime > 1329264000)
814         {
815             // If the new block's timestamp is more than 2* 10 minutes
816             // then allow mining of a min-difficulty block.
817             if (pblock->nTime - pindexLast->nTime > nTargetSpacing*2)
818                 return nProofOfWorkLimit;
819             else
820             {
821                 // Return the last non-special-min-difficulty-rules-block
822                 const CBlockIndex* pindex = pindexLast;
823                 while (pindex->pprev && pindex->nHeight % nInterval != 0 && pindex->nBits == nProofOfWorkLimit)
824                     pindex = pindex->pprev;
825                 return pindex->nBits;
826             }
827         }
828
829         return pindexLast->nBits;
830     }
831
832     // Go back by what we want to be 14 days worth of blocks
833     const CBlockIndex* pindexFirst = pindexLast;
834     for (int i = 0; pindexFirst && i < nInterval-1; i++)
835         pindexFirst = pindexFirst->pprev;
836     assert(pindexFirst);
837
838     // Limit adjustment step
839     int64 nActualTimespan = pindexLast->GetBlockTime() - pindexFirst->GetBlockTime();
840     printf("  nActualTimespan = %"PRI64d"  before bounds\n", nActualTimespan);
841     if (nActualTimespan < nTargetTimespan/4)
842         nActualTimespan = nTargetTimespan/4;
843     if (nActualTimespan > nTargetTimespan*4)
844         nActualTimespan = nTargetTimespan*4;
845
846     // Retarget
847     CBigNum bnNew;
848     bnNew.SetCompact(pindexLast->nBits);
849     bnNew *= nActualTimespan;
850     bnNew /= nTargetTimespan;
851
852     if (bnNew > bnProofOfWorkLimit)
853         bnNew = bnProofOfWorkLimit;
854
855     /// debug print
856     printf("GetNextWorkRequired RETARGET\n");
857     printf("nTargetTimespan = %"PRI64d"    nActualTimespan = %"PRI64d"\n", nTargetTimespan, nActualTimespan);
858     printf("Before: %08x  %s\n", pindexLast->nBits, CBigNum().SetCompact(pindexLast->nBits).getuint256().ToString().c_str());
859     printf("After:  %08x  %s\n", bnNew.GetCompact(), bnNew.getuint256().ToString().c_str());
860
861     return bnNew.GetCompact();
862 }
863
864 bool CheckProofOfWork(uint256 hash, unsigned int nBits)
865 {
866     CBigNum bnTarget;
867     bnTarget.SetCompact(nBits);
868
869     // Check range
870     if (bnTarget <= 0 || bnTarget > bnProofOfWorkLimit)
871         return error("CheckProofOfWork() : nBits below minimum work");
872
873     // Check proof of work matches claimed amount
874     if (hash > bnTarget.getuint256())
875         return error("CheckProofOfWork() : hash doesn't match nBits");
876
877     return true;
878 }
879
880 // Return maximum amount of blocks that other nodes claim to have
881 int GetNumBlocksOfPeers()
882 {
883     return std::max(cPeerBlockCounts.median(), Checkpoints::GetTotalBlocksEstimate());
884 }
885
886 bool IsInitialBlockDownload()
887 {
888     if (pindexBest == NULL || nBestHeight < Checkpoints::GetTotalBlocksEstimate())
889         return true;
890     static int64 nLastUpdate;
891     static CBlockIndex* pindexLastBest;
892     if (pindexBest != pindexLastBest)
893     {
894         pindexLastBest = pindexBest;
895         nLastUpdate = GetTime();
896     }
897     return (GetTime() - nLastUpdate < 10 &&
898             pindexBest->GetBlockTime() < GetTime() - 24 * 60 * 60);
899 }
900
901 void static InvalidChainFound(CBlockIndex* pindexNew)
902 {
903     if (pindexNew->bnChainWork > bnBestInvalidWork)
904     {
905         bnBestInvalidWork = pindexNew->bnChainWork;
906         CTxDB().WriteBestInvalidWork(bnBestInvalidWork);
907         MainFrameRepaint();
908     }
909     printf("InvalidChainFound: invalid block=%s  height=%d  work=%s\n", pindexNew->GetBlockHash().ToString().substr(0,20).c_str(), pindexNew->nHeight, pindexNew->bnChainWork.ToString().c_str());
910     printf("InvalidChainFound:  current best=%s  height=%d  work=%s\n", hashBestChain.ToString().substr(0,20).c_str(), nBestHeight, bnBestChainWork.ToString().c_str());
911     if (pindexBest && bnBestInvalidWork > bnBestChainWork + pindexBest->GetBlockWork() * 6)
912         printf("InvalidChainFound: WARNING: Displayed transactions may not be correct!  You may need to upgrade, or other nodes may need to upgrade.\n");
913 }
914
915 void CBlock::UpdateTime(const CBlockIndex* pindexPrev)
916 {
917     nTime = max(pindexPrev->GetMedianTimePast()+1, GetAdjustedTime());
918
919     // Updating time can change work required on testnet:
920     if (fTestNet)
921         nBits = GetNextWorkRequired(pindexPrev, this);
922 }
923
924
925
926
927
928
929
930
931
932
933
934 bool CTransaction::DisconnectInputs(CTxDB& txdb)
935 {
936     // Relinquish previous transactions' spent pointers
937     if (!IsCoinBase())
938     {
939         BOOST_FOREACH(const CTxIn& txin, vin)
940         {
941             COutPoint prevout = txin.prevout;
942
943             // Get prev txindex from disk
944             CTxIndex txindex;
945             if (!txdb.ReadTxIndex(prevout.hash, txindex))
946                 return error("DisconnectInputs() : ReadTxIndex failed");
947
948             if (prevout.n >= txindex.vSpent.size())
949                 return error("DisconnectInputs() : prevout.n out of range");
950
951             // Mark outpoint as not spent
952             txindex.vSpent[prevout.n].SetNull();
953
954             // Write back
955             if (!txdb.UpdateTxIndex(prevout.hash, txindex))
956                 return error("DisconnectInputs() : UpdateTxIndex failed");
957         }
958     }
959
960     // Remove transaction from index
961     if (!txdb.EraseTxIndex(*this))
962         return error("DisconnectInputs() : EraseTxPos failed");
963
964     return true;
965 }
966
967
968 bool CTransaction::FetchInputs(CTxDB& txdb, const map<uint256, CTxIndex>& mapTestPool,
969                                bool fBlock, bool fMiner, MapPrevTx& inputsRet, bool& fInvalid)
970 {
971     // FetchInputs can return false either because we just haven't seen some inputs
972     // (in which case the transaction should be stored as an orphan)
973     // or because the transaction is malformed (in which case the transaction should
974     // be dropped).  If tx is definitely invalid, fInvalid will be set to true.
975     fInvalid = false;
976
977     if (IsCoinBase())
978         return true; // Coinbase transactions have no inputs to fetch.
979
980     for (int i = 0; i < vin.size(); i++)
981     {
982         COutPoint prevout = vin[i].prevout;
983         if (inputsRet.count(prevout.hash))
984             continue; // Got it already
985
986         // Read txindex
987         CTxIndex& txindex = inputsRet[prevout.hash].first;
988         bool fFound = true;
989         if ((fBlock || fMiner) && mapTestPool.count(prevout.hash))
990         {
991             // Get txindex from current proposed changes
992             txindex = mapTestPool.find(prevout.hash)->second;
993         }
994         else
995         {
996             // Read txindex from txdb
997             fFound = txdb.ReadTxIndex(prevout.hash, txindex);
998         }
999         if (!fFound && (fBlock || fMiner))
1000             return fMiner ? false : error("FetchInputs() : %s prev tx %s index entry not found", GetHash().ToString().substr(0,10).c_str(),  prevout.hash.ToString().substr(0,10).c_str());
1001
1002         // Read txPrev
1003         CTransaction& txPrev = inputsRet[prevout.hash].second;
1004         if (!fFound || txindex.pos == CDiskTxPos(1,1,1))
1005         {
1006             // Get prev tx from single transactions in memory
1007             CRITICAL_BLOCK(cs_mapTransactions)
1008             {
1009                 if (!mapTransactions.count(prevout.hash))
1010                     return error("FetchInputs() : %s mapTransactions prev not found %s", GetHash().ToString().substr(0,10).c_str(),  prevout.hash.ToString().substr(0,10).c_str());
1011                 txPrev = mapTransactions[prevout.hash];
1012             }
1013             if (!fFound)
1014                 txindex.vSpent.resize(txPrev.vout.size());
1015         }
1016         else
1017         {
1018             // Get prev tx from disk
1019             if (!txPrev.ReadFromDisk(txindex.pos))
1020                 return error("FetchInputs() : %s ReadFromDisk prev tx %s failed", GetHash().ToString().substr(0,10).c_str(),  prevout.hash.ToString().substr(0,10).c_str());
1021         }
1022     }
1023
1024     // Make sure all prevout.n's are valid:
1025     for (int i = 0; i < vin.size(); i++)
1026     {
1027         const COutPoint prevout = vin[i].prevout;
1028         assert(inputsRet.count(prevout.hash) != 0);
1029         const CTxIndex& txindex = inputsRet[prevout.hash].first;
1030         const CTransaction& txPrev = inputsRet[prevout.hash].second;
1031         if (prevout.n >= txPrev.vout.size() || prevout.n >= txindex.vSpent.size())
1032         {
1033             // Revisit this if/when transaction replacement is implemented and allows
1034             // adding inputs:
1035             fInvalid = true;
1036             return DoS(100, error("FetchInputs() : %s prevout.n out of range %d %d %d prev tx %s\n%s", GetHash().ToString().substr(0,10).c_str(), prevout.n, txPrev.vout.size(), txindex.vSpent.size(), prevout.hash.ToString().substr(0,10).c_str(), txPrev.ToString().c_str()));
1037         }
1038     }
1039
1040     return true;
1041 }
1042
1043 const CTxOut& CTransaction::GetOutputFor(const CTxIn& input, const MapPrevTx& inputs) const
1044 {
1045     MapPrevTx::const_iterator mi = inputs.find(input.prevout.hash);
1046     if (mi == inputs.end())
1047         throw std::runtime_error("CTransaction::GetOutputFor() : prevout.hash not found");
1048
1049     const CTransaction& txPrev = (mi->second).second;
1050     if (input.prevout.n >= txPrev.vout.size())
1051         throw std::runtime_error("CTransaction::GetOutputFor() : prevout.n out of range");
1052
1053     return txPrev.vout[input.prevout.n];
1054 }
1055
1056 int64 CTransaction::GetValueIn(const MapPrevTx& inputs) const
1057 {
1058     if (IsCoinBase())
1059         return 0;
1060
1061     int64 nResult = 0;
1062     for (int i = 0; i < vin.size(); i++)
1063     {
1064         nResult += GetOutputFor(vin[i], inputs).nValue;
1065     }
1066     return nResult;
1067
1068 }
1069
1070 int CTransaction::GetP2SHSigOpCount(const MapPrevTx& inputs) const
1071 {
1072     if (IsCoinBase())
1073         return 0;
1074
1075     int nSigOps = 0;
1076     for (int i = 0; i < vin.size(); i++)
1077     {
1078         const CTxOut& prevout = GetOutputFor(vin[i], inputs);
1079         if (prevout.scriptPubKey.IsPayToScriptHash())
1080             nSigOps += prevout.scriptPubKey.GetSigOpCount(vin[i].scriptSig);
1081     }
1082     return nSigOps;
1083 }
1084
1085 bool CTransaction::ConnectInputs(MapPrevTx inputs,
1086                                  map<uint256, CTxIndex>& mapTestPool, const CDiskTxPos& posThisTx,
1087                                  const CBlockIndex* pindexBlock, bool fBlock, bool fMiner, bool fStrictPayToScriptHash)
1088 {
1089     // Take over previous transactions' spent pointers
1090     // fBlock is true when this is called from AcceptBlock when a new best-block is added to the blockchain
1091     // fMiner is true when called from the internal bitcoin miner
1092     // ... both are false when called from CTransaction::AcceptToMemoryPool
1093     if (!IsCoinBase())
1094     {
1095         int64 nValueIn = 0;
1096         int64 nFees = 0;
1097         for (int i = 0; i < vin.size(); i++)
1098         {
1099             COutPoint prevout = vin[i].prevout;
1100             assert(inputs.count(prevout.hash) > 0);
1101             CTxIndex& txindex = inputs[prevout.hash].first;
1102             CTransaction& txPrev = inputs[prevout.hash].second;
1103
1104             if (prevout.n >= txPrev.vout.size() || prevout.n >= txindex.vSpent.size())
1105                 return DoS(100, error("ConnectInputs() : %s prevout.n out of range %d %d %d prev tx %s\n%s", GetHash().ToString().substr(0,10).c_str(), prevout.n, txPrev.vout.size(), txindex.vSpent.size(), prevout.hash.ToString().substr(0,10).c_str(), txPrev.ToString().c_str()));
1106
1107             // If prev is coinbase, check that it's matured
1108             if (txPrev.IsCoinBase())
1109                 for (const CBlockIndex* pindex = pindexBlock; pindex && pindexBlock->nHeight - pindex->nHeight < COINBASE_MATURITY; pindex = pindex->pprev)
1110                     if (pindex->nBlockPos == txindex.pos.nBlockPos && pindex->nFile == txindex.pos.nFile)
1111                         return error("ConnectInputs() : tried to spend coinbase at depth %d", pindexBlock->nHeight - pindex->nHeight);
1112
1113             // Check for conflicts (double-spend)
1114             // This doesn't trigger the DoS code on purpose; if it did, it would make it easier
1115             // for an attacker to attempt to split the network.
1116             if (!txindex.vSpent[prevout.n].IsNull())
1117                 return fMiner ? false : error("ConnectInputs() : %s prev tx already used at %s", GetHash().ToString().substr(0,10).c_str(), txindex.vSpent[prevout.n].ToString().c_str());
1118
1119             // Check for negative or overflow input values
1120             nValueIn += txPrev.vout[prevout.n].nValue;
1121             if (!MoneyRange(txPrev.vout[prevout.n].nValue) || !MoneyRange(nValueIn))
1122                 return DoS(100, error("ConnectInputs() : txin values out of range"));
1123
1124             // Skip ECDSA signature verification when connecting blocks (fBlock=true)
1125             // before the last blockchain checkpoint. This is safe because block merkle hashes are
1126             // still computed and checked, and any change will be caught at the next checkpoint.
1127             if (!(fBlock && (nBestHeight < Checkpoints::GetTotalBlocksEstimate())))
1128             {
1129                 // Verify signature
1130                 if (!VerifySignature(txPrev, *this, i, fStrictPayToScriptHash, 0))
1131                 {
1132                     // only during transition phase for P2SH: do not invoke anti-DoS code for
1133                     // potentially old clients relaying bad P2SH transactions
1134                     if (fStrictPayToScriptHash && VerifySignature(txPrev, *this, i, false, 0))
1135                         return error("ConnectInputs() : %s P2SH VerifySignature failed", GetHash().ToString().substr(0,10).c_str());
1136
1137                     return DoS(100,error("ConnectInputs() : %s VerifySignature failed", GetHash().ToString().substr(0,10).c_str()));
1138                 }
1139             }
1140
1141             // Mark outpoints as spent
1142             txindex.vSpent[prevout.n] = posThisTx;
1143
1144             // Write back
1145             if (fBlock || fMiner)
1146             {
1147                 mapTestPool[prevout.hash] = txindex;
1148             }
1149         }
1150
1151         if (nValueIn < GetValueOut())
1152             return DoS(100, error("ConnectInputs() : %s value in < value out", GetHash().ToString().substr(0,10).c_str()));
1153
1154         // Tally transaction fees
1155         int64 nTxFee = nValueIn - GetValueOut();
1156         if (nTxFee < 0)
1157             return DoS(100, error("ConnectInputs() : %s nTxFee < 0", GetHash().ToString().substr(0,10).c_str()));
1158         nFees += nTxFee;
1159         if (!MoneyRange(nFees))
1160             return DoS(100, error("ConnectInputs() : nFees out of range"));
1161     }
1162
1163     return true;
1164 }
1165
1166
1167 bool CTransaction::ClientConnectInputs()
1168 {
1169     if (IsCoinBase())
1170         return false;
1171
1172     // Take over previous transactions' spent pointers
1173     CRITICAL_BLOCK(cs_mapTransactions)
1174     {
1175         int64 nValueIn = 0;
1176         for (int i = 0; i < vin.size(); i++)
1177         {
1178             // Get prev tx from single transactions in memory
1179             COutPoint prevout = vin[i].prevout;
1180             if (!mapTransactions.count(prevout.hash))
1181                 return false;
1182             CTransaction& txPrev = mapTransactions[prevout.hash];
1183
1184             if (prevout.n >= txPrev.vout.size())
1185                 return false;
1186
1187             // Verify signature
1188             if (!VerifySignature(txPrev, *this, i, true, 0))
1189                 return error("ConnectInputs() : VerifySignature failed");
1190
1191             ///// this is redundant with the mapNextTx stuff, not sure which I want to get rid of
1192             ///// this has to go away now that posNext is gone
1193             // // Check for conflicts
1194             // if (!txPrev.vout[prevout.n].posNext.IsNull())
1195             //     return error("ConnectInputs() : prev tx already used");
1196             //
1197             // // Flag outpoints as used
1198             // txPrev.vout[prevout.n].posNext = posThisTx;
1199
1200             nValueIn += txPrev.vout[prevout.n].nValue;
1201
1202             if (!MoneyRange(txPrev.vout[prevout.n].nValue) || !MoneyRange(nValueIn))
1203                 return error("ClientConnectInputs() : txin values out of range");
1204         }
1205         if (GetValueOut() > nValueIn)
1206             return false;
1207     }
1208
1209     return true;
1210 }
1211
1212
1213
1214
1215 bool CBlock::DisconnectBlock(CTxDB& txdb, CBlockIndex* pindex)
1216 {
1217     // Disconnect in reverse order
1218     for (int i = vtx.size()-1; i >= 0; i--)
1219         if (!vtx[i].DisconnectInputs(txdb))
1220             return false;
1221
1222     // Update block index on disk without changing it in memory.
1223     // The memory index structure will be changed after the db commits.
1224     if (pindex->pprev)
1225     {
1226         CDiskBlockIndex blockindexPrev(pindex->pprev);
1227         blockindexPrev.hashNext = 0;
1228         if (!txdb.WriteBlockIndex(blockindexPrev))
1229             return error("DisconnectBlock() : WriteBlockIndex failed");
1230     }
1231
1232     return true;
1233 }
1234
1235 bool CBlock::ConnectBlock(CTxDB& txdb, CBlockIndex* pindex)
1236 {
1237     // Check it again in case a previous version let a bad block in
1238     if (!CheckBlock())
1239         return false;
1240
1241     // To avoid being on the short end of a block-chain split,
1242     // don't do secondary validation of pay-to-script-hash transactions
1243     // until blocks with timestamps after paytoscripthashtime (see init.cpp for default).
1244     // This code can be removed once a super-majority of the network has upgraded.
1245     int64 nEvalSwitchTime = GetArg("-paytoscripthashtime", std::numeric_limits<int64_t>::max());
1246     bool fStrictPayToScriptHash = (pindex->nTime >= nEvalSwitchTime);
1247
1248     //// issue here: it doesn't know the version
1249     unsigned int nTxPos = pindex->nBlockPos + ::GetSerializeSize(CBlock(), SER_DISK) - 1 + GetSizeOfCompactSize(vtx.size());
1250
1251     map<uint256, CTxIndex> mapQueuedChanges;
1252     int64 nFees = 0;
1253     int nSigOps = 0;
1254     BOOST_FOREACH(CTransaction& tx, vtx)
1255     {
1256         nSigOps += tx.GetLegacySigOpCount();
1257         if (nSigOps > MAX_BLOCK_SIGOPS)
1258             return DoS(100, error("ConnectBlock() : too many sigops"));
1259
1260         CDiskTxPos posThisTx(pindex->nFile, pindex->nBlockPos, nTxPos);
1261         nTxPos += ::GetSerializeSize(tx, SER_DISK);
1262
1263         MapPrevTx mapInputs;
1264         if (!tx.IsCoinBase())
1265         {
1266             bool fInvalid;
1267             if (!tx.FetchInputs(txdb, mapQueuedChanges, true, false, mapInputs, fInvalid))
1268                 return false;
1269
1270             if (fStrictPayToScriptHash)
1271             {
1272                 // Add in sigops done by pay-to-script-hash inputs;
1273                 // this is to prevent a "rogue miner" from creating
1274                 // an incredibly-expensive-to-validate block.
1275                 nSigOps += tx.GetP2SHSigOpCount(mapInputs);
1276                 if (nSigOps > MAX_BLOCK_SIGOPS)
1277                     return DoS(100, error("ConnectBlock() : too many sigops"));
1278             }
1279
1280             nFees += tx.GetValueIn(mapInputs)-tx.GetValueOut();
1281
1282             if (!tx.ConnectInputs(mapInputs, mapQueuedChanges, posThisTx, pindex, true, false, fStrictPayToScriptHash))
1283                 return false;
1284         }
1285
1286         mapQueuedChanges[tx.GetHash()] = CTxIndex(posThisTx, tx.vout.size());
1287     }
1288
1289     // Write queued txindex changes
1290     for (map<uint256, CTxIndex>::iterator mi = mapQueuedChanges.begin(); mi != mapQueuedChanges.end(); ++mi)
1291     {
1292         if (!txdb.UpdateTxIndex((*mi).first, (*mi).second))
1293             return error("ConnectBlock() : UpdateTxIndex failed");
1294     }
1295
1296     if (vtx[0].GetValueOut() > GetBlockValue(pindex->nHeight, nFees))
1297         return false;
1298
1299     // Update block index on disk without changing it in memory.
1300     // The memory index structure will be changed after the db commits.
1301     if (pindex->pprev)
1302     {
1303         CDiskBlockIndex blockindexPrev(pindex->pprev);
1304         blockindexPrev.hashNext = pindex->GetBlockHash();
1305         if (!txdb.WriteBlockIndex(blockindexPrev))
1306             return error("ConnectBlock() : WriteBlockIndex failed");
1307     }
1308
1309     // Watch for transactions paying to me
1310     BOOST_FOREACH(CTransaction& tx, vtx)
1311         SyncWithWallets(tx, this, true);
1312
1313     return true;
1314 }
1315
1316 bool static Reorganize(CTxDB& txdb, CBlockIndex* pindexNew)
1317 {
1318     printf("REORGANIZE\n");
1319
1320     // Find the fork
1321     CBlockIndex* pfork = pindexBest;
1322     CBlockIndex* plonger = pindexNew;
1323     while (pfork != plonger)
1324     {
1325         while (plonger->nHeight > pfork->nHeight)
1326             if (!(plonger = plonger->pprev))
1327                 return error("Reorganize() : plonger->pprev is null");
1328         if (pfork == plonger)
1329             break;
1330         if (!(pfork = pfork->pprev))
1331             return error("Reorganize() : pfork->pprev is null");
1332     }
1333
1334     // List of what to disconnect
1335     vector<CBlockIndex*> vDisconnect;
1336     for (CBlockIndex* pindex = pindexBest; pindex != pfork; pindex = pindex->pprev)
1337         vDisconnect.push_back(pindex);
1338
1339     // List of what to connect
1340     vector<CBlockIndex*> vConnect;
1341     for (CBlockIndex* pindex = pindexNew; pindex != pfork; pindex = pindex->pprev)
1342         vConnect.push_back(pindex);
1343     reverse(vConnect.begin(), vConnect.end());
1344
1345     // Disconnect shorter branch
1346     vector<CTransaction> vResurrect;
1347     BOOST_FOREACH(CBlockIndex* pindex, vDisconnect)
1348     {
1349         CBlock block;
1350         if (!block.ReadFromDisk(pindex))
1351             return error("Reorganize() : ReadFromDisk for disconnect failed");
1352         if (!block.DisconnectBlock(txdb, pindex))
1353             return error("Reorganize() : DisconnectBlock failed");
1354
1355         // Queue memory transactions to resurrect
1356         BOOST_FOREACH(const CTransaction& tx, block.vtx)
1357             if (!tx.IsCoinBase())
1358                 vResurrect.push_back(tx);
1359     }
1360
1361     // Connect longer branch
1362     vector<CTransaction> vDelete;
1363     for (int i = 0; i < vConnect.size(); i++)
1364     {
1365         CBlockIndex* pindex = vConnect[i];
1366         CBlock block;
1367         if (!block.ReadFromDisk(pindex))
1368             return error("Reorganize() : ReadFromDisk for connect failed");
1369         if (!block.ConnectBlock(txdb, pindex))
1370         {
1371             // Invalid block
1372             txdb.TxnAbort();
1373             return error("Reorganize() : ConnectBlock failed");
1374         }
1375
1376         // Queue memory transactions to delete
1377         BOOST_FOREACH(const CTransaction& tx, block.vtx)
1378             vDelete.push_back(tx);
1379     }
1380     if (!txdb.WriteHashBestChain(pindexNew->GetBlockHash()))
1381         return error("Reorganize() : WriteHashBestChain failed");
1382
1383     // Make sure it's successfully written to disk before changing memory structure
1384     if (!txdb.TxnCommit())
1385         return error("Reorganize() : TxnCommit failed");
1386
1387     // Disconnect shorter branch
1388     BOOST_FOREACH(CBlockIndex* pindex, vDisconnect)
1389         if (pindex->pprev)
1390             pindex->pprev->pnext = NULL;
1391
1392     // Connect longer branch
1393     BOOST_FOREACH(CBlockIndex* pindex, vConnect)
1394         if (pindex->pprev)
1395             pindex->pprev->pnext = pindex;
1396
1397     // Resurrect memory transactions that were in the disconnected branch
1398     BOOST_FOREACH(CTransaction& tx, vResurrect)
1399         tx.AcceptToMemoryPool(txdb, false);
1400
1401     // Delete redundant memory transactions that are in the connected branch
1402     BOOST_FOREACH(CTransaction& tx, vDelete)
1403         tx.RemoveFromMemoryPool();
1404
1405     printf("REORGANIZE: Disconnected %i blocks; %s..%s\n", vDisconnect.size(), pfork->GetBlockHash().ToString().substr(0,20).c_str(), pindexBest->GetBlockHash().ToString().substr(0,20).c_str());
1406     printf("REORGANIZE: Connected %i blocks; %s..%s\n", vConnect.size(), pfork->GetBlockHash().ToString().substr(0,20).c_str(), pindexNew->GetBlockHash().ToString().substr(0,20).c_str());
1407
1408     return true;
1409 }
1410
1411
1412 static void
1413 runCommand(std::string strCommand)
1414 {
1415     int nErr = ::system(strCommand.c_str());
1416     if (nErr)
1417         printf("runCommand error: system(%s) returned %d\n", strCommand.c_str(), nErr);
1418 }
1419
1420 bool CBlock::SetBestChain(CTxDB& txdb, CBlockIndex* pindexNew)
1421 {
1422     uint256 hash = GetHash();
1423
1424     txdb.TxnBegin();
1425     if (pindexGenesisBlock == NULL && hash == hashGenesisBlock)
1426     {
1427         txdb.WriteHashBestChain(hash);
1428         if (!txdb.TxnCommit())
1429             return error("SetBestChain() : TxnCommit failed");
1430         pindexGenesisBlock = pindexNew;
1431     }
1432     else if (hashPrevBlock == hashBestChain)
1433     {
1434         // Adding to current best branch
1435         if (!ConnectBlock(txdb, pindexNew) || !txdb.WriteHashBestChain(hash))
1436         {
1437             txdb.TxnAbort();
1438             InvalidChainFound(pindexNew);
1439             return error("SetBestChain() : ConnectBlock failed");
1440         }
1441         if (!txdb.TxnCommit())
1442             return error("SetBestChain() : TxnCommit failed");
1443
1444         // Add to current best branch
1445         pindexNew->pprev->pnext = pindexNew;
1446
1447         // Delete redundant memory transactions
1448         BOOST_FOREACH(CTransaction& tx, vtx)
1449             tx.RemoveFromMemoryPool();
1450     }
1451     else
1452     {
1453         // New best branch
1454         if (!Reorganize(txdb, pindexNew))
1455         {
1456             txdb.TxnAbort();
1457             InvalidChainFound(pindexNew);
1458             return error("SetBestChain() : Reorganize failed");
1459         }
1460     }
1461
1462     // Update best block in wallet (so we can detect restored wallets)
1463     bool fIsInitialDownload = IsInitialBlockDownload();
1464     if (!fIsInitialDownload)
1465     {
1466         const CBlockLocator locator(pindexNew);
1467         ::SetBestChain(locator);
1468     }
1469
1470     // New best block
1471     hashBestChain = hash;
1472     pindexBest = pindexNew;
1473     nBestHeight = pindexBest->nHeight;
1474     bnBestChainWork = pindexNew->bnChainWork;
1475     nTimeBestReceived = GetTime();
1476     nTransactionsUpdated++;
1477     printf("SetBestChain: new best=%s  height=%d  work=%s\n", hashBestChain.ToString().substr(0,20).c_str(), nBestHeight, bnBestChainWork.ToString().c_str());
1478
1479     std::string strCmd = GetArg("-blocknotify", "");
1480
1481     if (!fIsInitialDownload && !strCmd.empty())
1482     {
1483         boost::replace_all(strCmd, "%s", hashBestChain.GetHex());
1484         boost::thread t(runCommand, strCmd); // thread runs free
1485     }
1486
1487     return true;
1488 }
1489
1490
1491 bool CBlock::AddToBlockIndex(unsigned int nFile, unsigned int nBlockPos)
1492 {
1493     // Check for duplicate
1494     uint256 hash = GetHash();
1495     if (mapBlockIndex.count(hash))
1496         return error("AddToBlockIndex() : %s already exists", hash.ToString().substr(0,20).c_str());
1497
1498     // Construct new block index object
1499     CBlockIndex* pindexNew = new CBlockIndex(nFile, nBlockPos, *this);
1500     if (!pindexNew)
1501         return error("AddToBlockIndex() : new CBlockIndex failed");
1502     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.insert(make_pair(hash, pindexNew)).first;
1503     pindexNew->phashBlock = &((*mi).first);
1504     map<uint256, CBlockIndex*>::iterator miPrev = mapBlockIndex.find(hashPrevBlock);
1505     if (miPrev != mapBlockIndex.end())
1506     {
1507         pindexNew->pprev = (*miPrev).second;
1508         pindexNew->nHeight = pindexNew->pprev->nHeight + 1;
1509     }
1510     pindexNew->bnChainWork = (pindexNew->pprev ? pindexNew->pprev->bnChainWork : 0) + pindexNew->GetBlockWork();
1511
1512     CTxDB txdb;
1513     txdb.TxnBegin();
1514     txdb.WriteBlockIndex(CDiskBlockIndex(pindexNew));
1515     if (!txdb.TxnCommit())
1516         return false;
1517
1518     // New best
1519     if (pindexNew->bnChainWork > bnBestChainWork)
1520         if (!SetBestChain(txdb, pindexNew))
1521             return false;
1522
1523     txdb.Close();
1524
1525     if (pindexNew == pindexBest)
1526     {
1527         // Notify UI to display prev block's coinbase if it was ours
1528         static uint256 hashPrevBestCoinBase;
1529         UpdatedTransaction(hashPrevBestCoinBase);
1530         hashPrevBestCoinBase = vtx[0].GetHash();
1531     }
1532
1533     MainFrameRepaint();
1534     return true;
1535 }
1536
1537
1538
1539
1540 bool CBlock::CheckBlock() const
1541 {
1542     // These are checks that are independent of context
1543     // that can be verified before saving an orphan block.
1544
1545     // Size limits
1546     if (vtx.empty() || vtx.size() > MAX_BLOCK_SIZE || ::GetSerializeSize(*this, SER_NETWORK) > MAX_BLOCK_SIZE)
1547         return DoS(100, error("CheckBlock() : size limits failed"));
1548
1549     // Check proof of work matches claimed amount
1550     if (!CheckProofOfWork(GetHash(), nBits))
1551         return DoS(50, error("CheckBlock() : proof of work failed"));
1552
1553     // Check timestamp
1554     if (GetBlockTime() > GetAdjustedTime() + 2 * 60 * 60)
1555         return error("CheckBlock() : block timestamp too far in the future");
1556
1557     // First transaction must be coinbase, the rest must not be
1558     if (vtx.empty() || !vtx[0].IsCoinBase())
1559         return DoS(100, error("CheckBlock() : first tx is not coinbase"));
1560     for (int i = 1; i < vtx.size(); i++)
1561         if (vtx[i].IsCoinBase())
1562             return DoS(100, error("CheckBlock() : more than one coinbase"));
1563
1564     // Check transactions
1565     BOOST_FOREACH(const CTransaction& tx, vtx)
1566         if (!tx.CheckTransaction())
1567             return DoS(tx.nDoS, error("CheckBlock() : CheckTransaction failed"));
1568
1569     int nSigOps = 0;
1570     BOOST_FOREACH(const CTransaction& tx, vtx)
1571     {
1572         nSigOps += tx.GetLegacySigOpCount();
1573     }
1574     if (nSigOps > MAX_BLOCK_SIGOPS)
1575         return DoS(100, error("CheckBlock() : out-of-bounds SigOpCount"));
1576
1577     // Check merkleroot
1578     if (hashMerkleRoot != BuildMerkleTree())
1579         return DoS(100, error("CheckBlock() : hashMerkleRoot mismatch"));
1580
1581     return true;
1582 }
1583
1584 bool CBlock::AcceptBlock()
1585 {
1586     // Check for duplicate
1587     uint256 hash = GetHash();
1588     if (mapBlockIndex.count(hash))
1589         return error("AcceptBlock() : block already in mapBlockIndex");
1590
1591     // Get prev block index
1592     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashPrevBlock);
1593     if (mi == mapBlockIndex.end())
1594         return DoS(10, error("AcceptBlock() : prev block not found"));
1595     CBlockIndex* pindexPrev = (*mi).second;
1596     int nHeight = pindexPrev->nHeight+1;
1597
1598     // Check proof of work
1599     if (nBits != GetNextWorkRequired(pindexPrev, this))
1600         return DoS(100, error("AcceptBlock() : incorrect proof of work"));
1601
1602     // Check timestamp against prev
1603     if (GetBlockTime() <= pindexPrev->GetMedianTimePast())
1604         return error("AcceptBlock() : block's timestamp is too early");
1605
1606     // Check that all transactions are finalized
1607     BOOST_FOREACH(const CTransaction& tx, vtx)
1608         if (!tx.IsFinal(nHeight, GetBlockTime()))
1609             return DoS(10, error("AcceptBlock() : contains a non-final transaction"));
1610
1611     // Check that the block chain matches the known block chain up to a checkpoint
1612     if (!Checkpoints::CheckBlock(nHeight, hash))
1613         return DoS(100, error("AcceptBlock() : rejected by checkpoint lockin at %d", nHeight));
1614
1615     // Write block to history file
1616     if (!CheckDiskSpace(::GetSerializeSize(*this, SER_DISK)))
1617         return error("AcceptBlock() : out of disk space");
1618     unsigned int nFile = -1;
1619     unsigned int nBlockPos = 0;
1620     if (!WriteToDisk(nFile, nBlockPos))
1621         return error("AcceptBlock() : WriteToDisk failed");
1622     if (!AddToBlockIndex(nFile, nBlockPos))
1623         return error("AcceptBlock() : AddToBlockIndex failed");
1624
1625     // Relay inventory, but don't relay old inventory during initial block download
1626     if (hashBestChain == hash)
1627         CRITICAL_BLOCK(cs_vNodes)
1628             BOOST_FOREACH(CNode* pnode, vNodes)
1629                 if (nBestHeight > (pnode->nStartingHeight != -1 ? pnode->nStartingHeight - 2000 : 140700))
1630                     pnode->PushInventory(CInv(MSG_BLOCK, hash));
1631
1632     return true;
1633 }
1634
1635 bool ProcessBlock(CNode* pfrom, CBlock* pblock)
1636 {
1637     // Check for duplicate
1638     uint256 hash = pblock->GetHash();
1639     if (mapBlockIndex.count(hash))
1640         return error("ProcessBlock() : already have block %d %s", mapBlockIndex[hash]->nHeight, hash.ToString().substr(0,20).c_str());
1641     if (mapOrphanBlocks.count(hash))
1642         return error("ProcessBlock() : already have block (orphan) %s", hash.ToString().substr(0,20).c_str());
1643
1644     // Preliminary checks
1645     if (!pblock->CheckBlock())
1646         return error("ProcessBlock() : CheckBlock FAILED");
1647
1648     CBlockIndex* pcheckpoint = Checkpoints::GetLastCheckpoint(mapBlockIndex);
1649     if (pcheckpoint && pblock->hashPrevBlock != hashBestChain)
1650     {
1651         // Extra checks to prevent "fill up memory by spamming with bogus blocks"
1652         int64 deltaTime = pblock->GetBlockTime() - pcheckpoint->nTime;
1653         if (deltaTime < 0)
1654         {
1655             if (pfrom)
1656                 pfrom->Misbehaving(100);
1657             return error("ProcessBlock() : block with timestamp before last checkpoint");
1658         }
1659         CBigNum bnNewBlock;
1660         bnNewBlock.SetCompact(pblock->nBits);
1661         CBigNum bnRequired;
1662         bnRequired.SetCompact(ComputeMinWork(pcheckpoint->nBits, deltaTime));
1663         if (bnNewBlock > bnRequired)
1664         {
1665             if (pfrom)
1666                 pfrom->Misbehaving(100);
1667             return error("ProcessBlock() : block with too little proof-of-work");
1668         }
1669     }
1670
1671
1672     // If don't already have its previous block, shunt it off to holding area until we get it
1673     if (!mapBlockIndex.count(pblock->hashPrevBlock))
1674     {
1675         printf("ProcessBlock: ORPHAN BLOCK, prev=%s\n", pblock->hashPrevBlock.ToString().substr(0,20).c_str());
1676         CBlock* pblock2 = new CBlock(*pblock);
1677         mapOrphanBlocks.insert(make_pair(hash, pblock2));
1678         mapOrphanBlocksByPrev.insert(make_pair(pblock2->hashPrevBlock, pblock2));
1679
1680         // Ask this guy to fill in what we're missing
1681         if (pfrom)
1682             pfrom->PushGetBlocks(pindexBest, GetOrphanRoot(pblock2));
1683         return true;
1684     }
1685
1686     // Store to disk
1687     if (!pblock->AcceptBlock())
1688         return error("ProcessBlock() : AcceptBlock FAILED");
1689
1690     // Recursively process any orphan blocks that depended on this one
1691     vector<uint256> vWorkQueue;
1692     vWorkQueue.push_back(hash);
1693     for (int i = 0; i < vWorkQueue.size(); i++)
1694     {
1695         uint256 hashPrev = vWorkQueue[i];
1696         for (multimap<uint256, CBlock*>::iterator mi = mapOrphanBlocksByPrev.lower_bound(hashPrev);
1697              mi != mapOrphanBlocksByPrev.upper_bound(hashPrev);
1698              ++mi)
1699         {
1700             CBlock* pblockOrphan = (*mi).second;
1701             if (pblockOrphan->AcceptBlock())
1702                 vWorkQueue.push_back(pblockOrphan->GetHash());
1703             mapOrphanBlocks.erase(pblockOrphan->GetHash());
1704             delete pblockOrphan;
1705         }
1706         mapOrphanBlocksByPrev.erase(hashPrev);
1707     }
1708
1709     printf("ProcessBlock: ACCEPTED\n");
1710     return true;
1711 }
1712
1713
1714
1715
1716
1717
1718
1719
1720 bool CheckDiskSpace(uint64 nAdditionalBytes)
1721 {
1722     uint64 nFreeBytesAvailable = filesystem::space(GetDataDir()).available;
1723
1724     // Check for 15MB because database could create another 10MB log file at any time
1725     if (nFreeBytesAvailable < (uint64)15000000 + nAdditionalBytes)
1726     {
1727         fShutdown = true;
1728         string strMessage = _("Warning: Disk space is low  ");
1729         strMiscWarning = strMessage;
1730         printf("*** %s\n", strMessage.c_str());
1731         ThreadSafeMessageBox(strMessage, "Bitcoin", wxOK | wxICON_EXCLAMATION);
1732         CreateThread(Shutdown, NULL);
1733         return false;
1734     }
1735     return true;
1736 }
1737
1738 FILE* OpenBlockFile(unsigned int nFile, unsigned int nBlockPos, const char* pszMode)
1739 {
1740     if (nFile == -1)
1741         return NULL;
1742     FILE* file = fopen(strprintf("%s/blk%04d.dat", GetDataDir().c_str(), nFile).c_str(), pszMode);
1743     if (!file)
1744         return NULL;
1745     if (nBlockPos != 0 && !strchr(pszMode, 'a') && !strchr(pszMode, 'w'))
1746     {
1747         if (fseek(file, nBlockPos, SEEK_SET) != 0)
1748         {
1749             fclose(file);
1750             return NULL;
1751         }
1752     }
1753     return file;
1754 }
1755
1756 static unsigned int nCurrentBlockFile = 1;
1757
1758 FILE* AppendBlockFile(unsigned int& nFileRet)
1759 {
1760     nFileRet = 0;
1761     loop
1762     {
1763         FILE* file = OpenBlockFile(nCurrentBlockFile, 0, "ab");
1764         if (!file)
1765             return NULL;
1766         if (fseek(file, 0, SEEK_END) != 0)
1767             return NULL;
1768         // FAT32 filesize max 4GB, fseek and ftell max 2GB, so we must stay under 2GB
1769         if (ftell(file) < 0x7F000000 - MAX_SIZE)
1770         {
1771             nFileRet = nCurrentBlockFile;
1772             return file;
1773         }
1774         fclose(file);
1775         nCurrentBlockFile++;
1776     }
1777 }
1778
1779 bool LoadBlockIndex(bool fAllowNew)
1780 {
1781     if (fTestNet)
1782     {
1783         hashGenesisBlock = uint256("0x00000007199508e34a9ff81e6ec0c477a4cccff2a4767a8eee39c11db367b008");
1784         bnProofOfWorkLimit = CBigNum(~uint256(0) >> 28);
1785         pchMessageStart[0] = 0xfa;
1786         pchMessageStart[1] = 0xbf;
1787         pchMessageStart[2] = 0xb5;
1788         pchMessageStart[3] = 0xda;
1789     }
1790
1791     //
1792     // Load block index
1793     //
1794     CTxDB txdb("cr");
1795     if (!txdb.LoadBlockIndex())
1796         return false;
1797     txdb.Close();
1798
1799     //
1800     // Init with genesis block
1801     //
1802     if (mapBlockIndex.empty())
1803     {
1804         if (!fAllowNew)
1805             return false;
1806
1807         // Genesis Block:
1808         // CBlock(hash=000000000019d6, ver=1, hashPrevBlock=00000000000000, hashMerkleRoot=4a5e1e, nTime=1231006505, nBits=1d00ffff, nNonce=2083236893, vtx=1)
1809         //   CTransaction(hash=4a5e1e, ver=1, vin.size=1, vout.size=1, nLockTime=0)
1810         //     CTxIn(COutPoint(000000, -1), coinbase 04ffff001d0104455468652054696d65732030332f4a616e2f32303039204368616e63656c6c6f72206f6e206272696e6b206f66207365636f6e64206261696c6f757420666f722062616e6b73)
1811         //     CTxOut(nValue=50.00000000, scriptPubKey=0x5F1DF16B2B704C8A578D0B)
1812         //   vMerkleTree: 4a5e1e
1813
1814         // Genesis block
1815         const char* pszTimestamp = "The Times 03/Jan/2009 Chancellor on brink of second bailout for banks";
1816         CTransaction txNew;
1817         txNew.vin.resize(1);
1818         txNew.vout.resize(1);
1819         txNew.vin[0].scriptSig = CScript() << 486604799 << CBigNum(4) << vector<unsigned char>((const unsigned char*)pszTimestamp, (const unsigned char*)pszTimestamp + strlen(pszTimestamp));
1820         txNew.vout[0].nValue = 50 * COIN;
1821         txNew.vout[0].scriptPubKey = CScript() << ParseHex("04678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5f") << OP_CHECKSIG;
1822         CBlock block;
1823         block.vtx.push_back(txNew);
1824         block.hashPrevBlock = 0;
1825         block.hashMerkleRoot = block.BuildMerkleTree();
1826         block.nVersion = 1;
1827         block.nTime    = 1231006505;
1828         block.nBits    = 0x1d00ffff;
1829         block.nNonce   = 2083236893;
1830
1831         if (fTestNet)
1832         {
1833             block.nTime    = 1296688602;
1834             block.nBits    = 0x1d07fff8;
1835             block.nNonce   = 384568319;
1836         }
1837
1838         //// debug print
1839         printf("%s\n", block.GetHash().ToString().c_str());
1840         printf("%s\n", hashGenesisBlock.ToString().c_str());
1841         printf("%s\n", block.hashMerkleRoot.ToString().c_str());
1842         assert(block.hashMerkleRoot == uint256("0x4a5e1e4baab89f3a32518a88c31bc87f618f76673e2cc77ab2127b7afdeda33b"));
1843         block.print();
1844         assert(block.GetHash() == hashGenesisBlock);
1845
1846         // Start new block file
1847         unsigned int nFile;
1848         unsigned int nBlockPos;
1849         if (!block.WriteToDisk(nFile, nBlockPos))
1850             return error("LoadBlockIndex() : writing genesis block to disk failed");
1851         if (!block.AddToBlockIndex(nFile, nBlockPos))
1852             return error("LoadBlockIndex() : genesis block not accepted");
1853     }
1854
1855     return true;
1856 }
1857
1858
1859
1860 void PrintBlockTree()
1861 {
1862     // precompute tree structure
1863     map<CBlockIndex*, vector<CBlockIndex*> > mapNext;
1864     for (map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.begin(); mi != mapBlockIndex.end(); ++mi)
1865     {
1866         CBlockIndex* pindex = (*mi).second;
1867         mapNext[pindex->pprev].push_back(pindex);
1868         // test
1869         //while (rand() % 3 == 0)
1870         //    mapNext[pindex->pprev].push_back(pindex);
1871     }
1872
1873     vector<pair<int, CBlockIndex*> > vStack;
1874     vStack.push_back(make_pair(0, pindexGenesisBlock));
1875
1876     int nPrevCol = 0;
1877     while (!vStack.empty())
1878     {
1879         int nCol = vStack.back().first;
1880         CBlockIndex* pindex = vStack.back().second;
1881         vStack.pop_back();
1882
1883         // print split or gap
1884         if (nCol > nPrevCol)
1885         {
1886             for (int i = 0; i < nCol-1; i++)
1887                 printf("| ");
1888             printf("|\\\n");
1889         }
1890         else if (nCol < nPrevCol)
1891         {
1892             for (int i = 0; i < nCol; i++)
1893                 printf("| ");
1894             printf("|\n");
1895        }
1896         nPrevCol = nCol;
1897
1898         // print columns
1899         for (int i = 0; i < nCol; i++)
1900             printf("| ");
1901
1902         // print item
1903         CBlock block;
1904         block.ReadFromDisk(pindex);
1905         printf("%d (%u,%u) %s  %s  tx %d",
1906             pindex->nHeight,
1907             pindex->nFile,
1908             pindex->nBlockPos,
1909             block.GetHash().ToString().substr(0,20).c_str(),
1910             DateTimeStrFormat("%x %H:%M:%S", block.GetBlockTime()).c_str(),
1911             block.vtx.size());
1912
1913         PrintWallets(block);
1914
1915         // put the main timechain first
1916         vector<CBlockIndex*>& vNext = mapNext[pindex];
1917         for (int i = 0; i < vNext.size(); i++)
1918         {
1919             if (vNext[i]->pnext)
1920             {
1921                 swap(vNext[0], vNext[i]);
1922                 break;
1923             }
1924         }
1925
1926         // iterate children
1927         for (int i = 0; i < vNext.size(); i++)
1928             vStack.push_back(make_pair(nCol+i, vNext[i]));
1929     }
1930 }
1931
1932
1933
1934
1935
1936
1937
1938
1939
1940
1941 //////////////////////////////////////////////////////////////////////////////
1942 //
1943 // CAlert
1944 //
1945
1946 map<uint256, CAlert> mapAlerts;
1947 CCriticalSection cs_mapAlerts;
1948
1949 string GetWarnings(string strFor)
1950 {
1951     int nPriority = 0;
1952     string strStatusBar;
1953     string strRPC;
1954     if (GetBoolArg("-testsafemode"))
1955         strRPC = "test";
1956
1957     // Misc warnings like out of disk space and clock is wrong
1958     if (strMiscWarning != "")
1959     {
1960         nPriority = 1000;
1961         strStatusBar = strMiscWarning;
1962     }
1963
1964     // Longer invalid proof-of-work chain
1965     if (pindexBest && bnBestInvalidWork > bnBestChainWork + pindexBest->GetBlockWork() * 6)
1966     {
1967         nPriority = 2000;
1968         strStatusBar = strRPC = "WARNING: Displayed transactions may not be correct!  You may need to upgrade, or other nodes may need to upgrade.";
1969     }
1970
1971     // Alerts
1972     CRITICAL_BLOCK(cs_mapAlerts)
1973     {
1974         BOOST_FOREACH(PAIRTYPE(const uint256, CAlert)& item, mapAlerts)
1975         {
1976             const CAlert& alert = item.second;
1977             if (alert.AppliesToMe() && alert.nPriority > nPriority)
1978             {
1979                 nPriority = alert.nPriority;
1980                 strStatusBar = alert.strStatusBar;
1981             }
1982         }
1983     }
1984
1985     if (strFor == "statusbar")
1986         return strStatusBar;
1987     else if (strFor == "rpc")
1988         return strRPC;
1989     assert(!"GetWarnings() : invalid parameter");
1990     return "error";
1991 }
1992
1993 bool CAlert::ProcessAlert()
1994 {
1995     if (!CheckSignature())
1996         return false;
1997     if (!IsInEffect())
1998         return false;
1999
2000     CRITICAL_BLOCK(cs_mapAlerts)
2001     {
2002         // Cancel previous alerts
2003         for (map<uint256, CAlert>::iterator mi = mapAlerts.begin(); mi != mapAlerts.end();)
2004         {
2005             const CAlert& alert = (*mi).second;
2006             if (Cancels(alert))
2007             {
2008                 printf("cancelling alert %d\n", alert.nID);
2009                 mapAlerts.erase(mi++);
2010             }
2011             else if (!alert.IsInEffect())
2012             {
2013                 printf("expiring alert %d\n", alert.nID);
2014                 mapAlerts.erase(mi++);
2015             }
2016             else
2017                 mi++;
2018         }
2019
2020         // Check if this alert has been cancelled
2021         BOOST_FOREACH(PAIRTYPE(const uint256, CAlert)& item, mapAlerts)
2022         {
2023             const CAlert& alert = item.second;
2024             if (alert.Cancels(*this))
2025             {
2026                 printf("alert already cancelled by %d\n", alert.nID);
2027                 return false;
2028             }
2029         }
2030
2031         // Add to mapAlerts
2032         mapAlerts.insert(make_pair(GetHash(), *this));
2033     }
2034
2035     printf("accepted alert %d, AppliesToMe()=%d\n", nID, AppliesToMe());
2036     MainFrameRepaint();
2037     return true;
2038 }
2039
2040
2041
2042
2043
2044
2045
2046
2047 //////////////////////////////////////////////////////////////////////////////
2048 //
2049 // Messages
2050 //
2051
2052
2053 bool static AlreadyHave(CTxDB& txdb, const CInv& inv)
2054 {
2055     switch (inv.type)
2056     {
2057     case MSG_TX:    return mapTransactions.count(inv.hash) || mapOrphanTransactions.count(inv.hash) || txdb.ContainsTx(inv.hash);
2058     case MSG_BLOCK: return mapBlockIndex.count(inv.hash) || mapOrphanBlocks.count(inv.hash);
2059     }
2060     // Don't know what it is, just say we already got one
2061     return true;
2062 }
2063
2064
2065
2066
2067 // The message start string is designed to be unlikely to occur in normal data.
2068 // The characters are rarely used upper ascii, not valid as UTF-8, and produce
2069 // a large 4-byte int at any alignment.
2070 unsigned char pchMessageStart[4] = { 0xf9, 0xbe, 0xb4, 0xd9 };
2071
2072
2073 bool static ProcessMessage(CNode* pfrom, string strCommand, CDataStream& vRecv)
2074 {
2075     static map<CService, vector<unsigned char> > mapReuseKey;
2076     RandAddSeedPerfmon();
2077     if (fDebug) {
2078         printf("%s ", DateTimeStrFormat("%x %H:%M:%S", GetTime()).c_str());
2079         printf("received: %s (%d bytes)\n", strCommand.c_str(), vRecv.size());
2080     }
2081     if (mapArgs.count("-dropmessagestest") && GetRand(atoi(mapArgs["-dropmessagestest"])) == 0)
2082     {
2083         printf("dropmessagestest DROPPING RECV MESSAGE\n");
2084         return true;
2085     }
2086
2087
2088
2089
2090
2091     if (strCommand == "version")
2092     {
2093         // Each connection can only send one version message
2094         if (pfrom->nVersion != 0)
2095         {
2096             pfrom->Misbehaving(1);
2097             return false;
2098         }
2099
2100         int64 nTime;
2101         CAddress addrMe;
2102         CAddress addrFrom;
2103         uint64 nNonce = 1;
2104         vRecv >> pfrom->nVersion >> pfrom->nServices >> nTime >> addrMe;
2105         if (pfrom->nVersion < 209)
2106         {
2107             // Since Februari 20, 2012, the protocol is initiated at version 209,
2108             // and earlier versions are no longer supported
2109             printf("partner %s using obsolete version %i; disconnecting\n", pfrom->addr.ToString().c_str(), pfrom->nVersion);
2110             pfrom->fDisconnect = true;
2111             return false;
2112         }
2113
2114         if (pfrom->nVersion == 10300)
2115             pfrom->nVersion = 300;
2116         if (!vRecv.empty())
2117             vRecv >> addrFrom >> nNonce;
2118         if (!vRecv.empty())
2119             vRecv >> pfrom->strSubVer;
2120         if (!vRecv.empty())
2121             vRecv >> pfrom->nStartingHeight;
2122
2123         // Disconnect if we connected to ourself
2124         if (nNonce == nLocalHostNonce && nNonce > 1)
2125         {
2126             printf("connected to self at %s, disconnecting\n", pfrom->addr.ToString().c_str());
2127             pfrom->fDisconnect = true;
2128             return true;
2129         }
2130
2131         // Be shy and don't send version until we hear
2132         if (pfrom->fInbound)
2133             pfrom->PushVersion();
2134
2135         pfrom->fClient = !(pfrom->nServices & NODE_NETWORK);
2136
2137         AddTimeData(pfrom->addr, nTime);
2138
2139         // Change version
2140         pfrom->PushMessage("verack");
2141         pfrom->vSend.SetVersion(min(pfrom->nVersion, PROTOCOL_VERSION));
2142
2143         if (!pfrom->fInbound)
2144         {
2145             // Advertise our address
2146             if (!fNoListen && !fUseProxy && addrLocalHost.IsRoutable() &&
2147                 !IsInitialBlockDownload())
2148             {
2149                 CAddress addr(addrLocalHost);
2150                 addr.nTime = GetAdjustedTime();
2151                 pfrom->PushAddress(addr);
2152             }
2153
2154             // Get recent addresses
2155             if (pfrom->nVersion >= 31402 || mapAddresses.size() < 1000)
2156             {
2157                 pfrom->PushMessage("getaddr");
2158                 pfrom->fGetAddr = true;
2159             }
2160         }
2161
2162         // Ask the first connected node for block updates
2163         static int nAskedForBlocks = 0;
2164         if (!pfrom->fClient &&
2165             (pfrom->nVersion < 32000 || pfrom->nVersion >= 32400) &&
2166              (nAskedForBlocks < 1 || vNodes.size() <= 1))
2167         {
2168             nAskedForBlocks++;
2169             pfrom->PushGetBlocks(pindexBest, uint256(0));
2170         }
2171
2172         // Relay alerts
2173         CRITICAL_BLOCK(cs_mapAlerts)
2174             BOOST_FOREACH(PAIRTYPE(const uint256, CAlert)& item, mapAlerts)
2175                 item.second.RelayTo(pfrom);
2176
2177         pfrom->fSuccessfullyConnected = true;
2178
2179         printf("version message: version %d, blocks=%d\n", pfrom->nVersion, pfrom->nStartingHeight);
2180
2181         cPeerBlockCounts.input(pfrom->nStartingHeight);
2182     }
2183
2184
2185     else if (pfrom->nVersion == 0)
2186     {
2187         // Must have a version message before anything else
2188         pfrom->Misbehaving(1);
2189         return false;
2190     }
2191
2192
2193     else if (strCommand == "verack")
2194     {
2195         pfrom->vRecv.SetVersion(min(pfrom->nVersion, PROTOCOL_VERSION));
2196     }
2197
2198
2199     else if (strCommand == "addr")
2200     {
2201         vector<CAddress> vAddr;
2202         vRecv >> vAddr;
2203
2204         // Don't want addr from older versions unless seeding
2205         if (pfrom->nVersion < 31402 && mapAddresses.size() > 1000)
2206             return true;
2207         if (vAddr.size() > 1000)
2208         {
2209             pfrom->Misbehaving(20);
2210             return error("message addr size() = %d", vAddr.size());
2211         }
2212
2213         // Store the new addresses
2214         CAddrDB addrDB;
2215         addrDB.TxnBegin();
2216         int64 nNow = GetAdjustedTime();
2217         int64 nSince = nNow - 10 * 60;
2218         BOOST_FOREACH(CAddress& addr, vAddr)
2219         {
2220             if (fShutdown)
2221                 return true;
2222             // ignore IPv6 for now, since it isn't implemented anyway
2223             if (!addr.IsIPv4())
2224                 continue;
2225             if (addr.nTime <= 100000000 || addr.nTime > nNow + 10 * 60)
2226                 addr.nTime = nNow - 5 * 24 * 60 * 60;
2227             AddAddress(addr, 2 * 60 * 60, &addrDB);
2228             pfrom->AddAddressKnown(addr);
2229             if (addr.nTime > nSince && !pfrom->fGetAddr && vAddr.size() <= 10 && addr.IsRoutable())
2230             {
2231                 // Relay to a limited number of other nodes
2232                 CRITICAL_BLOCK(cs_vNodes)
2233                 {
2234                     // Use deterministic randomness to send to the same nodes for 24 hours
2235                     // at a time so the setAddrKnowns of the chosen nodes prevent repeats
2236                     static uint256 hashSalt;
2237                     if (hashSalt == 0)
2238                         RAND_bytes((unsigned char*)&hashSalt, sizeof(hashSalt));
2239                     int64 hashAddr = addr.GetHash();
2240                     uint256 hashRand = hashSalt ^ (hashAddr<<32) ^ ((GetTime()+hashAddr)/(24*60*60));
2241                     hashRand = Hash(BEGIN(hashRand), END(hashRand));
2242                     multimap<uint256, CNode*> mapMix;
2243                     BOOST_FOREACH(CNode* pnode, vNodes)
2244                     {
2245                         if (pnode->nVersion < 31402)
2246                             continue;
2247                         unsigned int nPointer;
2248                         memcpy(&nPointer, &pnode, sizeof(nPointer));
2249                         uint256 hashKey = hashRand ^ nPointer;
2250                         hashKey = Hash(BEGIN(hashKey), END(hashKey));
2251                         mapMix.insert(make_pair(hashKey, pnode));
2252                     }
2253                     int nRelayNodes = 2;
2254                     for (multimap<uint256, CNode*>::iterator mi = mapMix.begin(); mi != mapMix.end() && nRelayNodes-- > 0; ++mi)
2255                         ((*mi).second)->PushAddress(addr);
2256                 }
2257             }
2258         }
2259         addrDB.TxnCommit();  // Save addresses (it's ok if this fails)
2260         if (vAddr.size() < 1000)
2261             pfrom->fGetAddr = false;
2262     }
2263
2264
2265     else if (strCommand == "inv")
2266     {
2267         vector<CInv> vInv;
2268         vRecv >> vInv;
2269         if (vInv.size() > 50000)
2270         {
2271             pfrom->Misbehaving(20);
2272             return error("message inv size() = %d", vInv.size());
2273         }
2274
2275         CTxDB txdb("r");
2276         BOOST_FOREACH(const CInv& inv, vInv)
2277         {
2278             if (fShutdown)
2279                 return true;
2280             pfrom->AddInventoryKnown(inv);
2281
2282             bool fAlreadyHave = AlreadyHave(txdb, inv);
2283             if (fDebug)
2284                 printf("  got inventory: %s  %s\n", inv.ToString().c_str(), fAlreadyHave ? "have" : "new");
2285
2286             if (!fAlreadyHave)
2287                 pfrom->AskFor(inv);
2288             else if (inv.type == MSG_BLOCK && mapOrphanBlocks.count(inv.hash))
2289                 pfrom->PushGetBlocks(pindexBest, GetOrphanRoot(mapOrphanBlocks[inv.hash]));
2290
2291             // Track requests for our stuff
2292             Inventory(inv.hash);
2293         }
2294     }
2295
2296
2297     else if (strCommand == "getdata")
2298     {
2299         vector<CInv> vInv;
2300         vRecv >> vInv;
2301         if (vInv.size() > 50000)
2302         {
2303             pfrom->Misbehaving(20);
2304             return error("message getdata size() = %d", vInv.size());
2305         }
2306
2307         BOOST_FOREACH(const CInv& inv, vInv)
2308         {
2309             if (fShutdown)
2310                 return true;
2311             printf("received getdata for: %s\n", inv.ToString().c_str());
2312
2313             if (inv.type == MSG_BLOCK)
2314             {
2315                 // Send block from disk
2316                 map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(inv.hash);
2317                 if (mi != mapBlockIndex.end())
2318                 {
2319                     CBlock block;
2320                     block.ReadFromDisk((*mi).second);
2321                     pfrom->PushMessage("block", block);
2322
2323                     // Trigger them to send a getblocks request for the next batch of inventory
2324                     if (inv.hash == pfrom->hashContinue)
2325                     {
2326                         // Bypass PushInventory, this must send even if redundant,
2327                         // and we want it right after the last block so they don't
2328                         // wait for other stuff first.
2329                         vector<CInv> vInv;
2330                         vInv.push_back(CInv(MSG_BLOCK, hashBestChain));
2331                         pfrom->PushMessage("inv", vInv);
2332                         pfrom->hashContinue = 0;
2333                     }
2334                 }
2335             }
2336             else if (inv.IsKnownType())
2337             {
2338                 // Send stream from relay memory
2339                 CRITICAL_BLOCK(cs_mapRelay)
2340                 {
2341                     map<CInv, CDataStream>::iterator mi = mapRelay.find(inv);
2342                     if (mi != mapRelay.end())
2343                         pfrom->PushMessage(inv.GetCommand(), (*mi).second);
2344                 }
2345             }
2346
2347             // Track requests for our stuff
2348             Inventory(inv.hash);
2349         }
2350     }
2351
2352
2353     else if (strCommand == "getblocks")
2354     {
2355         CBlockLocator locator;
2356         uint256 hashStop;
2357         vRecv >> locator >> hashStop;
2358
2359         // Find the last block the caller has in the main chain
2360         CBlockIndex* pindex = locator.GetBlockIndex();
2361
2362         // Send the rest of the chain
2363         if (pindex)
2364             pindex = pindex->pnext;
2365         int nLimit = 500 + locator.GetDistanceBack();
2366         unsigned int nBytes = 0;
2367         printf("getblocks %d to %s limit %d\n", (pindex ? pindex->nHeight : -1), hashStop.ToString().substr(0,20).c_str(), nLimit);
2368         for (; pindex; pindex = pindex->pnext)
2369         {
2370             if (pindex->GetBlockHash() == hashStop)
2371             {
2372                 printf("  getblocks stopping at %d %s (%u bytes)\n", pindex->nHeight, pindex->GetBlockHash().ToString().substr(0,20).c_str(), nBytes);
2373                 break;
2374             }
2375             pfrom->PushInventory(CInv(MSG_BLOCK, pindex->GetBlockHash()));
2376             CBlock block;
2377             block.ReadFromDisk(pindex, true);
2378             nBytes += block.GetSerializeSize(SER_NETWORK);
2379             if (--nLimit <= 0 || nBytes >= SendBufferSize()/2)
2380             {
2381                 // When this block is requested, we'll send an inv that'll make them
2382                 // getblocks the next batch of inventory.
2383                 printf("  getblocks stopping at limit %d %s (%u bytes)\n", pindex->nHeight, pindex->GetBlockHash().ToString().substr(0,20).c_str(), nBytes);
2384                 pfrom->hashContinue = pindex->GetBlockHash();
2385                 break;
2386             }
2387         }
2388     }
2389
2390
2391     else if (strCommand == "getheaders")
2392     {
2393         CBlockLocator locator;
2394         uint256 hashStop;
2395         vRecv >> locator >> hashStop;
2396
2397         CBlockIndex* pindex = NULL;
2398         if (locator.IsNull())
2399         {
2400             // If locator is null, return the hashStop block
2401             map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashStop);
2402             if (mi == mapBlockIndex.end())
2403                 return true;
2404             pindex = (*mi).second;
2405         }
2406         else
2407         {
2408             // Find the last block the caller has in the main chain
2409             pindex = locator.GetBlockIndex();
2410             if (pindex)
2411                 pindex = pindex->pnext;
2412         }
2413
2414         vector<CBlock> vHeaders;
2415         int nLimit = 2000 + locator.GetDistanceBack();
2416         printf("getheaders %d to %s limit %d\n", (pindex ? pindex->nHeight : -1), hashStop.ToString().substr(0,20).c_str(), nLimit);
2417         for (; pindex; pindex = pindex->pnext)
2418         {
2419             vHeaders.push_back(pindex->GetBlockHeader());
2420             if (--nLimit <= 0 || pindex->GetBlockHash() == hashStop)
2421                 break;
2422         }
2423         pfrom->PushMessage("headers", vHeaders);
2424     }
2425
2426
2427     else if (strCommand == "tx")
2428     {
2429         vector<uint256> vWorkQueue;
2430         CDataStream vMsg(vRecv);
2431         CTransaction tx;
2432         vRecv >> tx;
2433
2434         CInv inv(MSG_TX, tx.GetHash());
2435         pfrom->AddInventoryKnown(inv);
2436
2437         bool fMissingInputs = false;
2438         if (tx.AcceptToMemoryPool(true, &fMissingInputs))
2439         {
2440             SyncWithWallets(tx, NULL, true);
2441             RelayMessage(inv, vMsg);
2442             mapAlreadyAskedFor.erase(inv);
2443             vWorkQueue.push_back(inv.hash);
2444
2445             // Recursively process any orphan transactions that depended on this one
2446             for (int i = 0; i < vWorkQueue.size(); i++)
2447             {
2448                 uint256 hashPrev = vWorkQueue[i];
2449                 for (multimap<uint256, CDataStream*>::iterator mi = mapOrphanTransactionsByPrev.lower_bound(hashPrev);
2450                      mi != mapOrphanTransactionsByPrev.upper_bound(hashPrev);
2451                      ++mi)
2452                 {
2453                     const CDataStream& vMsg = *((*mi).second);
2454                     CTransaction tx;
2455                     CDataStream(vMsg) >> tx;
2456                     CInv inv(MSG_TX, tx.GetHash());
2457
2458                     if (tx.AcceptToMemoryPool(true))
2459                     {
2460                         printf("   accepted orphan tx %s\n", inv.hash.ToString().substr(0,10).c_str());
2461                         SyncWithWallets(tx, NULL, true);
2462                         RelayMessage(inv, vMsg);
2463                         mapAlreadyAskedFor.erase(inv);
2464                         vWorkQueue.push_back(inv.hash);
2465                     }
2466                 }
2467             }
2468
2469             BOOST_FOREACH(uint256 hash, vWorkQueue)
2470                 EraseOrphanTx(hash);
2471         }
2472         else if (fMissingInputs)
2473         {
2474             printf("storing orphan tx %s\n", inv.hash.ToString().substr(0,10).c_str());
2475             AddOrphanTx(vMsg);
2476         }
2477         if (tx.nDoS) pfrom->Misbehaving(tx.nDoS);
2478     }
2479
2480
2481     else if (strCommand == "block")
2482     {
2483         CBlock block;
2484         vRecv >> block;
2485
2486         printf("received block %s\n", block.GetHash().ToString().substr(0,20).c_str());
2487         // block.print();
2488
2489         CInv inv(MSG_BLOCK, block.GetHash());
2490         pfrom->AddInventoryKnown(inv);
2491
2492         if (ProcessBlock(pfrom, &block))
2493             mapAlreadyAskedFor.erase(inv);
2494         if (block.nDoS) pfrom->Misbehaving(block.nDoS);
2495     }
2496
2497
2498     else if (strCommand == "getaddr")
2499     {
2500         // Nodes rebroadcast an addr every 24 hours
2501         pfrom->vAddrToSend.clear();
2502         int64 nSince = GetAdjustedTime() - 3 * 60 * 60; // in the last 3 hours
2503         CRITICAL_BLOCK(cs_mapAddresses)
2504         {
2505             unsigned int nCount = 0;
2506             BOOST_FOREACH(const PAIRTYPE(vector<unsigned char>, CAddress)& item, mapAddresses)
2507             {
2508                 const CAddress& addr = item.second;
2509                 if (addr.nTime > nSince)
2510                     nCount++;
2511             }
2512             BOOST_FOREACH(const PAIRTYPE(vector<unsigned char>, CAddress)& item, mapAddresses)
2513             {
2514                 const CAddress& addr = item.second;
2515                 if (addr.nTime > nSince && GetRand(nCount) < 2500)
2516                     pfrom->PushAddress(addr);
2517             }
2518         }
2519     }
2520
2521
2522     else if (strCommand == "checkorder")
2523     {
2524         uint256 hashReply;
2525         vRecv >> hashReply;
2526
2527         if (!GetBoolArg("-allowreceivebyip"))
2528         {
2529             pfrom->PushMessage("reply", hashReply, (int)2, string(""));
2530             return true;
2531         }
2532
2533         CWalletTx order;
2534         vRecv >> order;
2535
2536         /// we have a chance to check the order here
2537
2538         // Keep giving the same key to the same ip until they use it
2539         if (!mapReuseKey.count(pfrom->addr))
2540             pwalletMain->GetKeyFromPool(mapReuseKey[pfrom->addr], true);
2541
2542         // Send back approval of order and pubkey to use
2543         CScript scriptPubKey;
2544         scriptPubKey << mapReuseKey[pfrom->addr] << OP_CHECKSIG;
2545         pfrom->PushMessage("reply", hashReply, (int)0, scriptPubKey);
2546     }
2547
2548
2549     else if (strCommand == "reply")
2550     {
2551         uint256 hashReply;
2552         vRecv >> hashReply;
2553
2554         CRequestTracker tracker;
2555         CRITICAL_BLOCK(pfrom->cs_mapRequests)
2556         {
2557             map<uint256, CRequestTracker>::iterator mi = pfrom->mapRequests.find(hashReply);
2558             if (mi != pfrom->mapRequests.end())
2559             {
2560                 tracker = (*mi).second;
2561                 pfrom->mapRequests.erase(mi);
2562             }
2563         }
2564         if (!tracker.IsNull())
2565             tracker.fn(tracker.param1, vRecv);
2566     }
2567
2568
2569     else if (strCommand == "ping")
2570     {
2571     }
2572
2573
2574     else if (strCommand == "alert")
2575     {
2576         CAlert alert;
2577         vRecv >> alert;
2578
2579         if (alert.ProcessAlert())
2580         {
2581             // Relay
2582             pfrom->setKnown.insert(alert.GetHash());
2583             CRITICAL_BLOCK(cs_vNodes)
2584                 BOOST_FOREACH(CNode* pnode, vNodes)
2585                     alert.RelayTo(pnode);
2586         }
2587     }
2588
2589
2590     else
2591     {
2592         // Ignore unknown commands for extensibility
2593     }
2594
2595
2596     // Update the last seen time for this node's address
2597     if (pfrom->fNetworkNode)
2598         if (strCommand == "version" || strCommand == "addr" || strCommand == "inv" || strCommand == "getdata" || strCommand == "ping")
2599             AddressCurrentlyConnected(pfrom->addr);
2600
2601
2602     return true;
2603 }
2604
2605 bool ProcessMessages(CNode* pfrom)
2606 {
2607     CDataStream& vRecv = pfrom->vRecv;
2608     if (vRecv.empty())
2609         return true;
2610     //if (fDebug)
2611     //    printf("ProcessMessages(%u bytes)\n", vRecv.size());
2612
2613     //
2614     // Message format
2615     //  (4) message start
2616     //  (12) command
2617     //  (4) size
2618     //  (4) checksum
2619     //  (x) data
2620     //
2621
2622     loop
2623     {
2624         // Scan for message start
2625         CDataStream::iterator pstart = search(vRecv.begin(), vRecv.end(), BEGIN(pchMessageStart), END(pchMessageStart));
2626         int nHeaderSize = vRecv.GetSerializeSize(CMessageHeader());
2627         if (vRecv.end() - pstart < nHeaderSize)
2628         {
2629             if (vRecv.size() > nHeaderSize)
2630             {
2631                 printf("\n\nPROCESSMESSAGE MESSAGESTART NOT FOUND\n\n");
2632                 vRecv.erase(vRecv.begin(), vRecv.end() - nHeaderSize);
2633             }
2634             break;
2635         }
2636         if (pstart - vRecv.begin() > 0)
2637             printf("\n\nPROCESSMESSAGE SKIPPED %d BYTES\n\n", pstart - vRecv.begin());
2638         vRecv.erase(vRecv.begin(), pstart);
2639
2640         // Read header
2641         vector<char> vHeaderSave(vRecv.begin(), vRecv.begin() + nHeaderSize);
2642         CMessageHeader hdr;
2643         vRecv >> hdr;
2644         if (!hdr.IsValid())
2645         {
2646             printf("\n\nPROCESSMESSAGE: ERRORS IN HEADER %s\n\n\n", hdr.GetCommand().c_str());
2647             continue;
2648         }
2649         string strCommand = hdr.GetCommand();
2650
2651         // Message size
2652         unsigned int nMessageSize = hdr.nMessageSize;
2653         if (nMessageSize > MAX_SIZE)
2654         {
2655             printf("ProcessMessage(%s, %u bytes) : nMessageSize > MAX_SIZE\n", strCommand.c_str(), nMessageSize);
2656             continue;
2657         }
2658         if (nMessageSize > vRecv.size())
2659         {
2660             // Rewind and wait for rest of message
2661             vRecv.insert(vRecv.begin(), vHeaderSave.begin(), vHeaderSave.end());
2662             break;
2663         }
2664
2665         // Checksum
2666         uint256 hash = Hash(vRecv.begin(), vRecv.begin() + nMessageSize);
2667         unsigned int nChecksum = 0;
2668         memcpy(&nChecksum, &hash, sizeof(nChecksum));
2669         if (nChecksum != hdr.nChecksum)
2670         {
2671             printf("ProcessMessage(%s, %u bytes) : CHECKSUM ERROR nChecksum=%08x hdr.nChecksum=%08x\n",
2672                strCommand.c_str(), nMessageSize, nChecksum, hdr.nChecksum);
2673             continue;
2674         }
2675
2676         // Copy message to its own buffer
2677         CDataStream vMsg(vRecv.begin(), vRecv.begin() + nMessageSize, vRecv.nType, vRecv.nVersion);
2678         vRecv.ignore(nMessageSize);
2679
2680         // Process message
2681         bool fRet = false;
2682         try
2683         {
2684             CRITICAL_BLOCK(cs_main)
2685                 fRet = ProcessMessage(pfrom, strCommand, vMsg);
2686             if (fShutdown)
2687                 return true;
2688         }
2689         catch (std::ios_base::failure& e)
2690         {
2691             if (strstr(e.what(), "end of data"))
2692             {
2693                 // Allow exceptions from underlength message on vRecv
2694                 printf("ProcessMessage(%s, %u bytes) : Exception '%s' caught, normally caused by a message being shorter than its stated length\n", strCommand.c_str(), nMessageSize, e.what());
2695             }
2696             else if (strstr(e.what(), "size too large"))
2697             {
2698                 // Allow exceptions from overlong size
2699                 printf("ProcessMessage(%s, %u bytes) : Exception '%s' caught\n", strCommand.c_str(), nMessageSize, e.what());
2700             }
2701             else
2702             {
2703                 PrintExceptionContinue(&e, "ProcessMessage()");
2704             }
2705         }
2706         catch (std::exception& e) {
2707             PrintExceptionContinue(&e, "ProcessMessage()");
2708         } catch (...) {
2709             PrintExceptionContinue(NULL, "ProcessMessage()");
2710         }
2711
2712         if (!fRet)
2713             printf("ProcessMessage(%s, %u bytes) FAILED\n", strCommand.c_str(), nMessageSize);
2714     }
2715
2716     vRecv.Compact();
2717     return true;
2718 }
2719
2720
2721 bool SendMessages(CNode* pto, bool fSendTrickle)
2722 {
2723     CRITICAL_BLOCK(cs_main)
2724     {
2725         // Don't send anything until we get their version message
2726         if (pto->nVersion == 0)
2727             return true;
2728
2729         // Keep-alive ping
2730         if (pto->nLastSend && GetTime() - pto->nLastSend > 30 * 60 && pto->vSend.empty())
2731             pto->PushMessage("ping");
2732
2733         // Resend wallet transactions that haven't gotten in a block yet
2734         ResendWalletTransactions();
2735
2736         // Address refresh broadcast
2737         static int64 nLastRebroadcast;
2738         if (!IsInitialBlockDownload() && (GetTime() - nLastRebroadcast > 24 * 60 * 60))
2739         {
2740             CRITICAL_BLOCK(cs_vNodes)
2741             {
2742                 BOOST_FOREACH(CNode* pnode, vNodes)
2743                 {
2744                     // Periodically clear setAddrKnown to allow refresh broadcasts
2745                     if (nLastRebroadcast)
2746                         pnode->setAddrKnown.clear();
2747
2748                     // Rebroadcast our address
2749                     if (!fNoListen && !fUseProxy && addrLocalHost.IsRoutable())
2750                     {
2751                         CAddress addr(addrLocalHost);
2752                         addr.nTime = GetAdjustedTime();
2753                         pnode->PushAddress(addr);
2754                     }
2755                 }
2756             }
2757             nLastRebroadcast = GetTime();
2758         }
2759
2760         // Clear out old addresses periodically so it's not too much work at once
2761         static int64 nLastClear;
2762         if (nLastClear == 0)
2763             nLastClear = GetTime();
2764         if (GetTime() - nLastClear > 10 * 60 && vNodes.size() >= 3)
2765         {
2766             nLastClear = GetTime();
2767             CRITICAL_BLOCK(cs_mapAddresses)
2768             {
2769                 CAddrDB addrdb;
2770                 int64 nSince = GetAdjustedTime() - 14 * 24 * 60 * 60;
2771                 for (map<vector<unsigned char>, CAddress>::iterator mi = mapAddresses.begin();
2772                      mi != mapAddresses.end();)
2773                 {
2774                     const CAddress& addr = (*mi).second;
2775                     if (addr.nTime < nSince)
2776                     {
2777                         if (mapAddresses.size() < 1000 || GetTime() > nLastClear + 20)
2778                             break;
2779                         addrdb.EraseAddress(addr);
2780                         mapAddresses.erase(mi++);
2781                     }
2782                     else
2783                         mi++;
2784                 }
2785             }
2786         }
2787
2788
2789         //
2790         // Message: addr
2791         //
2792         if (fSendTrickle)
2793         {
2794             vector<CAddress> vAddr;
2795             vAddr.reserve(pto->vAddrToSend.size());
2796             BOOST_FOREACH(const CAddress& addr, pto->vAddrToSend)
2797             {
2798                 // returns true if wasn't already contained in the set
2799                 if (pto->setAddrKnown.insert(addr).second)
2800                 {
2801                     vAddr.push_back(addr);
2802                     // receiver rejects addr messages larger than 1000
2803                     if (vAddr.size() >= 1000)
2804                     {
2805                         pto->PushMessage("addr", vAddr);
2806                         vAddr.clear();
2807                     }
2808                 }
2809             }
2810             pto->vAddrToSend.clear();
2811             if (!vAddr.empty())
2812                 pto->PushMessage("addr", vAddr);
2813         }
2814
2815
2816         //
2817         // Message: inventory
2818         //
2819         vector<CInv> vInv;
2820         vector<CInv> vInvWait;
2821         CRITICAL_BLOCK(pto->cs_inventory)
2822         {
2823             vInv.reserve(pto->vInventoryToSend.size());
2824             vInvWait.reserve(pto->vInventoryToSend.size());
2825             BOOST_FOREACH(const CInv& inv, pto->vInventoryToSend)
2826             {
2827                 if (pto->setInventoryKnown.count(inv))
2828                     continue;
2829
2830                 // trickle out tx inv to protect privacy
2831                 if (inv.type == MSG_TX && !fSendTrickle)
2832                 {
2833                     // 1/4 of tx invs blast to all immediately
2834                     static uint256 hashSalt;
2835                     if (hashSalt == 0)
2836                         RAND_bytes((unsigned char*)&hashSalt, sizeof(hashSalt));
2837                     uint256 hashRand = inv.hash ^ hashSalt;
2838                     hashRand = Hash(BEGIN(hashRand), END(hashRand));
2839                     bool fTrickleWait = ((hashRand & 3) != 0);
2840
2841                     // always trickle our own transactions
2842                     if (!fTrickleWait)
2843                     {
2844                         CWalletTx wtx;
2845                         if (GetTransaction(inv.hash, wtx))
2846                             if (wtx.fFromMe)
2847                                 fTrickleWait = true;
2848                     }
2849
2850                     if (fTrickleWait)
2851                     {
2852                         vInvWait.push_back(inv);
2853                         continue;
2854                     }
2855                 }
2856
2857                 // returns true if wasn't already contained in the set
2858                 if (pto->setInventoryKnown.insert(inv).second)
2859                 {
2860                     vInv.push_back(inv);
2861                     if (vInv.size() >= 1000)
2862                     {
2863                         pto->PushMessage("inv", vInv);
2864                         vInv.clear();
2865                     }
2866                 }
2867             }
2868             pto->vInventoryToSend = vInvWait;
2869         }
2870         if (!vInv.empty())
2871             pto->PushMessage("inv", vInv);
2872
2873
2874         //
2875         // Message: getdata
2876         //
2877         vector<CInv> vGetData;
2878         int64 nNow = GetTime() * 1000000;
2879         CTxDB txdb("r");
2880         while (!pto->mapAskFor.empty() && (*pto->mapAskFor.begin()).first <= nNow)
2881         {
2882             const CInv& inv = (*pto->mapAskFor.begin()).second;
2883             if (!AlreadyHave(txdb, inv))
2884             {
2885                 printf("sending getdata: %s\n", inv.ToString().c_str());
2886                 vGetData.push_back(inv);
2887                 if (vGetData.size() >= 1000)
2888                 {
2889                     pto->PushMessage("getdata", vGetData);
2890                     vGetData.clear();
2891                 }
2892             }
2893             mapAlreadyAskedFor[inv] = nNow;
2894             pto->mapAskFor.erase(pto->mapAskFor.begin());
2895         }
2896         if (!vGetData.empty())
2897             pto->PushMessage("getdata", vGetData);
2898
2899     }
2900     return true;
2901 }
2902
2903
2904
2905
2906
2907
2908
2909
2910
2911
2912
2913
2914
2915
2916 //////////////////////////////////////////////////////////////////////////////
2917 //
2918 // BitcoinMiner
2919 //
2920
2921 int static FormatHashBlocks(void* pbuffer, unsigned int len)
2922 {
2923     unsigned char* pdata = (unsigned char*)pbuffer;
2924     unsigned int blocks = 1 + ((len + 8) / 64);
2925     unsigned char* pend = pdata + 64 * blocks;
2926     memset(pdata + len, 0, 64 * blocks - len);
2927     pdata[len] = 0x80;
2928     unsigned int bits = len * 8;
2929     pend[-1] = (bits >> 0) & 0xff;
2930     pend[-2] = (bits >> 8) & 0xff;
2931     pend[-3] = (bits >> 16) & 0xff;
2932     pend[-4] = (bits >> 24) & 0xff;
2933     return blocks;
2934 }
2935
2936 static const unsigned int pSHA256InitState[8] =
2937 {0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19};
2938
2939 void SHA256Transform(void* pstate, void* pinput, const void* pinit)
2940 {
2941     SHA256_CTX ctx;
2942     unsigned char data[64];
2943
2944     SHA256_Init(&ctx);
2945
2946     for (int i = 0; i < 16; i++)
2947         ((uint32_t*)data)[i] = ByteReverse(((uint32_t*)pinput)[i]);
2948
2949     for (int i = 0; i < 8; i++)
2950         ctx.h[i] = ((uint32_t*)pinit)[i];
2951
2952     SHA256_Update(&ctx, data, sizeof(data));
2953     for (int i = 0; i < 8; i++) 
2954         ((uint32_t*)pstate)[i] = ctx.h[i];
2955 }
2956
2957 //
2958 // ScanHash scans nonces looking for a hash with at least some zero bits.
2959 // It operates on big endian data.  Caller does the byte reversing.
2960 // All input buffers are 16-byte aligned.  nNonce is usually preserved
2961 // between calls, but periodically or if nNonce is 0xffff0000 or above,
2962 // the block is rebuilt and nNonce starts over at zero.
2963 //
2964 unsigned int static ScanHash_CryptoPP(char* pmidstate, char* pdata, char* phash1, char* phash, unsigned int& nHashesDone)
2965 {
2966     unsigned int& nNonce = *(unsigned int*)(pdata + 12);
2967     for (;;)
2968     {
2969         // Crypto++ SHA-256
2970         // Hash pdata using pmidstate as the starting state into
2971         // preformatted buffer phash1, then hash phash1 into phash
2972         nNonce++;
2973         SHA256Transform(phash1, pdata, pmidstate);
2974         SHA256Transform(phash, phash1, pSHA256InitState);
2975
2976         // Return the nonce if the hash has at least some zero bits,
2977         // caller will check if it has enough to reach the target
2978         if (((unsigned short*)phash)[14] == 0)
2979             return nNonce;
2980
2981         // If nothing found after trying for a while, return -1
2982         if ((nNonce & 0xffff) == 0)
2983         {
2984             nHashesDone = 0xffff+1;
2985             return -1;
2986         }
2987     }
2988 }
2989
2990 // Some explaining would be appreciated
2991 class COrphan
2992 {
2993 public:
2994     CTransaction* ptx;
2995     set<uint256> setDependsOn;
2996     double dPriority;
2997
2998     COrphan(CTransaction* ptxIn)
2999     {
3000         ptx = ptxIn;
3001         dPriority = 0;
3002     }
3003
3004     void print() const
3005     {
3006         printf("COrphan(hash=%s, dPriority=%.1f)\n", ptx->GetHash().ToString().substr(0,10).c_str(), dPriority);
3007         BOOST_FOREACH(uint256 hash, setDependsOn)
3008             printf("   setDependsOn %s\n", hash.ToString().substr(0,10).c_str());
3009     }
3010 };
3011
3012
3013 uint64 nLastBlockTx = 0;
3014 uint64 nLastBlockSize = 0;
3015
3016 CBlock* CreateNewBlock(CReserveKey& reservekey)
3017 {
3018     CBlockIndex* pindexPrev = pindexBest;
3019
3020     // Create new block
3021     auto_ptr<CBlock> pblock(new CBlock());
3022     if (!pblock.get())
3023         return NULL;
3024
3025     // Create coinbase tx
3026     CTransaction txNew;
3027     txNew.vin.resize(1);
3028     txNew.vin[0].prevout.SetNull();
3029     txNew.vout.resize(1);
3030     txNew.vout[0].scriptPubKey << reservekey.GetReservedKey() << OP_CHECKSIG;
3031
3032     // Add our coinbase tx as first transaction
3033     pblock->vtx.push_back(txNew);
3034
3035     // Collect memory pool transactions into the block
3036     int64 nFees = 0;
3037     CRITICAL_BLOCK(cs_main)
3038     CRITICAL_BLOCK(cs_mapTransactions)
3039     {
3040         CTxDB txdb("r");
3041
3042         // Priority order to process transactions
3043         list<COrphan> vOrphan; // list memory doesn't move
3044         map<uint256, vector<COrphan*> > mapDependers;
3045         multimap<double, CTransaction*> mapPriority;
3046         for (map<uint256, CTransaction>::iterator mi = mapTransactions.begin(); mi != mapTransactions.end(); ++mi)
3047         {
3048             CTransaction& tx = (*mi).second;
3049             if (tx.IsCoinBase() || !tx.IsFinal())
3050                 continue;
3051
3052             COrphan* porphan = NULL;
3053             double dPriority = 0;
3054             BOOST_FOREACH(const CTxIn& txin, tx.vin)
3055             {
3056                 // Read prev transaction
3057                 CTransaction txPrev;
3058                 CTxIndex txindex;
3059                 if (!txPrev.ReadFromDisk(txdb, txin.prevout, txindex))
3060                 {
3061                     // Has to wait for dependencies
3062                     if (!porphan)
3063                     {
3064                         // Use list for automatic deletion
3065                         vOrphan.push_back(COrphan(&tx));
3066                         porphan = &vOrphan.back();
3067                     }
3068                     mapDependers[txin.prevout.hash].push_back(porphan);
3069                     porphan->setDependsOn.insert(txin.prevout.hash);
3070                     continue;
3071                 }
3072                 int64 nValueIn = txPrev.vout[txin.prevout.n].nValue;
3073
3074                 // Read block header
3075                 int nConf = txindex.GetDepthInMainChain();
3076
3077                 dPriority += (double)nValueIn * nConf;
3078
3079                 if (fDebug && GetBoolArg("-printpriority"))
3080                     printf("priority     nValueIn=%-12I64d nConf=%-5d dPriority=%-20.1f\n", nValueIn, nConf, dPriority);
3081             }
3082
3083             // Priority is sum(valuein * age) / txsize
3084             dPriority /= ::GetSerializeSize(tx, SER_NETWORK);
3085
3086             if (porphan)
3087                 porphan->dPriority = dPriority;
3088             else
3089                 mapPriority.insert(make_pair(-dPriority, &(*mi).second));
3090
3091             if (fDebug && GetBoolArg("-printpriority"))
3092             {
3093                 printf("priority %-20.1f %s\n%s", dPriority, tx.GetHash().ToString().substr(0,10).c_str(), tx.ToString().c_str());
3094                 if (porphan)
3095                     porphan->print();
3096                 printf("\n");
3097             }
3098         }
3099
3100         // Collect transactions into block
3101         map<uint256, CTxIndex> mapTestPool;
3102         uint64 nBlockSize = 1000;
3103         uint64 nBlockTx = 0;
3104         int nBlockSigOps = 100;
3105         while (!mapPriority.empty())
3106         {
3107             // Take highest priority transaction off priority queue
3108             double dPriority = -(*mapPriority.begin()).first;
3109             CTransaction& tx = *(*mapPriority.begin()).second;
3110             mapPriority.erase(mapPriority.begin());
3111
3112             // Size limits
3113             unsigned int nTxSize = ::GetSerializeSize(tx, SER_NETWORK);
3114             if (nBlockSize + nTxSize >= MAX_BLOCK_SIZE_GEN)
3115                 continue;
3116
3117             // Legacy limits on sigOps:
3118             int nTxSigOps = tx.GetLegacySigOpCount();
3119             if (nBlockSigOps + nTxSigOps >= MAX_BLOCK_SIGOPS)
3120                 continue;
3121
3122             // Transaction fee required depends on block size
3123             bool fAllowFree = (nBlockSize + nTxSize < 4000 || CTransaction::AllowFree(dPriority));
3124             int64 nMinFee = tx.GetMinFee(nBlockSize, fAllowFree, GMF_BLOCK);
3125
3126             // Connecting shouldn't fail due to dependency on other memory pool transactions
3127             // because we're already processing them in order of dependency
3128             map<uint256, CTxIndex> mapTestPoolTmp(mapTestPool);
3129             MapPrevTx mapInputs;
3130             bool fInvalid;
3131             if (!tx.FetchInputs(txdb, mapTestPoolTmp, false, true, mapInputs, fInvalid))
3132                 continue;
3133
3134             int64 nTxFees = tx.GetValueIn(mapInputs)-tx.GetValueOut();
3135             if (nTxFees < nMinFee)
3136                 continue;
3137
3138             nTxSigOps += tx.GetP2SHSigOpCount(mapInputs);
3139             if (nBlockSigOps + nTxSigOps >= MAX_BLOCK_SIGOPS)
3140                 continue;
3141
3142             if (!tx.ConnectInputs(mapInputs, mapTestPoolTmp, CDiskTxPos(1,1,1), pindexPrev, false, true))
3143                 continue;
3144             mapTestPoolTmp[tx.GetHash()] = CTxIndex(CDiskTxPos(1,1,1), tx.vout.size());
3145             swap(mapTestPool, mapTestPoolTmp);
3146
3147             // Added
3148             pblock->vtx.push_back(tx);
3149             nBlockSize += nTxSize;
3150             ++nBlockTx;
3151             nBlockSigOps += nTxSigOps;
3152             nFees += nTxFees;
3153
3154             // Add transactions that depend on this one to the priority queue
3155             uint256 hash = tx.GetHash();
3156             if (mapDependers.count(hash))
3157             {
3158                 BOOST_FOREACH(COrphan* porphan, mapDependers[hash])
3159                 {
3160                     if (!porphan->setDependsOn.empty())
3161                     {
3162                         porphan->setDependsOn.erase(hash);
3163                         if (porphan->setDependsOn.empty())
3164                             mapPriority.insert(make_pair(-porphan->dPriority, porphan->ptx));
3165                     }
3166                 }
3167             }
3168         }
3169
3170         nLastBlockTx = nBlockTx;
3171         nLastBlockSize = nBlockSize;
3172         printf("CreateNewBlock(): total size %lu\n", nBlockSize);
3173
3174     }
3175     pblock->vtx[0].vout[0].nValue = GetBlockValue(pindexPrev->nHeight+1, nFees);
3176
3177     // Fill in header
3178     pblock->hashPrevBlock  = pindexPrev->GetBlockHash();
3179     pblock->hashMerkleRoot = pblock->BuildMerkleTree();
3180     pblock->UpdateTime(pindexPrev);
3181     pblock->nBits          = GetNextWorkRequired(pindexPrev, pblock.get());
3182     pblock->nNonce         = 0;
3183
3184     return pblock.release();
3185 }
3186
3187
3188 void IncrementExtraNonce(CBlock* pblock, CBlockIndex* pindexPrev, unsigned int& nExtraNonce)
3189 {
3190     // Update nExtraNonce
3191     static uint256 hashPrevBlock;
3192     if (hashPrevBlock != pblock->hashPrevBlock)
3193     {
3194         nExtraNonce = 0;
3195         hashPrevBlock = pblock->hashPrevBlock;
3196     }
3197     ++nExtraNonce;
3198     pblock->vtx[0].vin[0].scriptSig = (CScript() << pblock->nTime << CBigNum(nExtraNonce)) + COINBASE_FLAGS;
3199     assert(pblock->vtx[0].vin[0].scriptSig.size() <= 100);
3200
3201     pblock->hashMerkleRoot = pblock->BuildMerkleTree();
3202 }
3203
3204
3205 void FormatHashBuffers(CBlock* pblock, char* pmidstate, char* pdata, char* phash1)
3206 {
3207     //
3208     // Prebuild hash buffers
3209     //
3210     struct
3211     {
3212         struct unnamed2
3213         {
3214             int nVersion;
3215             uint256 hashPrevBlock;
3216             uint256 hashMerkleRoot;
3217             unsigned int nTime;
3218             unsigned int nBits;
3219             unsigned int nNonce;
3220         }
3221         block;
3222         unsigned char pchPadding0[64];
3223         uint256 hash1;
3224         unsigned char pchPadding1[64];
3225     }
3226     tmp;
3227     memset(&tmp, 0, sizeof(tmp));
3228
3229     tmp.block.nVersion       = pblock->nVersion;
3230     tmp.block.hashPrevBlock  = pblock->hashPrevBlock;
3231     tmp.block.hashMerkleRoot = pblock->hashMerkleRoot;
3232     tmp.block.nTime          = pblock->nTime;
3233     tmp.block.nBits          = pblock->nBits;
3234     tmp.block.nNonce         = pblock->nNonce;
3235
3236     FormatHashBlocks(&tmp.block, sizeof(tmp.block));
3237     FormatHashBlocks(&tmp.hash1, sizeof(tmp.hash1));
3238
3239     // Byte swap all the input buffer
3240     for (int i = 0; i < sizeof(tmp)/4; i++)
3241         ((unsigned int*)&tmp)[i] = ByteReverse(((unsigned int*)&tmp)[i]);
3242
3243     // Precalc the first half of the first hash, which stays constant
3244     SHA256Transform(pmidstate, &tmp.block, pSHA256InitState);
3245
3246     memcpy(pdata, &tmp.block, 128);
3247     memcpy(phash1, &tmp.hash1, 64);
3248 }
3249
3250
3251 bool CheckWork(CBlock* pblock, CWallet& wallet, CReserveKey& reservekey)
3252 {
3253     uint256 hash = pblock->GetHash();
3254     uint256 hashTarget = CBigNum().SetCompact(pblock->nBits).getuint256();
3255
3256     if (hash > hashTarget)
3257         return false;
3258
3259     //// debug print
3260     printf("BitcoinMiner:\n");
3261     printf("proof-of-work found  \n  hash: %s  \ntarget: %s\n", hash.GetHex().c_str(), hashTarget.GetHex().c_str());
3262     pblock->print();
3263     printf("%s ", DateTimeStrFormat("%x %H:%M", GetTime()).c_str());
3264     printf("generated %s\n", FormatMoney(pblock->vtx[0].vout[0].nValue).c_str());
3265
3266     // Found a solution
3267     CRITICAL_BLOCK(cs_main)
3268     {
3269         if (pblock->hashPrevBlock != hashBestChain)
3270             return error("BitcoinMiner : generated block is stale");
3271
3272         // Remove key from key pool
3273         reservekey.KeepKey();
3274
3275         // Track how many getdata requests this block gets
3276         CRITICAL_BLOCK(wallet.cs_wallet)
3277             wallet.mapRequestCount[pblock->GetHash()] = 0;
3278
3279         // Process this block the same as if we had received it from another node
3280         if (!ProcessBlock(NULL, pblock))
3281             return error("BitcoinMiner : ProcessBlock, block not accepted");
3282     }
3283
3284     return true;
3285 }
3286
3287 void static ThreadBitcoinMiner(void* parg);
3288
3289 static bool fGenerateBitcoins = false;
3290 static bool fLimitProcessors = false;
3291 static int nLimitProcessors = -1;
3292
3293 void static BitcoinMiner(CWallet *pwallet)
3294 {
3295     printf("BitcoinMiner started\n");
3296     SetThreadPriority(THREAD_PRIORITY_LOWEST);
3297
3298     // Each thread has its own key and counter
3299     CReserveKey reservekey(pwallet);
3300     unsigned int nExtraNonce = 0;
3301
3302     while (fGenerateBitcoins)
3303     {
3304         if (AffinityBugWorkaround(ThreadBitcoinMiner))
3305             return;
3306         if (fShutdown)
3307             return;
3308         while (vNodes.empty() || IsInitialBlockDownload())
3309         {
3310             Sleep(1000);
3311             if (fShutdown)
3312                 return;
3313             if (!fGenerateBitcoins)
3314                 return;
3315         }
3316
3317
3318         //
3319         // Create new block
3320         //
3321         unsigned int nTransactionsUpdatedLast = nTransactionsUpdated;
3322         CBlockIndex* pindexPrev = pindexBest;
3323
3324         auto_ptr<CBlock> pblock(CreateNewBlock(reservekey));
3325         if (!pblock.get())
3326             return;
3327         IncrementExtraNonce(pblock.get(), pindexPrev, nExtraNonce);
3328
3329         printf("Running BitcoinMiner with %d transactions in block\n", pblock->vtx.size());
3330
3331
3332         //
3333         // Prebuild hash buffers
3334         //
3335         char pmidstatebuf[32+16]; char* pmidstate = alignup<16>(pmidstatebuf);
3336         char pdatabuf[128+16];    char* pdata     = alignup<16>(pdatabuf);
3337         char phash1buf[64+16];    char* phash1    = alignup<16>(phash1buf);
3338
3339         FormatHashBuffers(pblock.get(), pmidstate, pdata, phash1);
3340
3341         unsigned int& nBlockTime = *(unsigned int*)(pdata + 64 + 4);
3342         unsigned int& nBlockBits = *(unsigned int*)(pdata + 64 + 8);
3343         unsigned int& nBlockNonce = *(unsigned int*)(pdata + 64 + 12);
3344
3345
3346         //
3347         // Search
3348         //
3349         int64 nStart = GetTime();
3350         uint256 hashTarget = CBigNum().SetCompact(pblock->nBits).getuint256();
3351         uint256 hashbuf[2];
3352         uint256& hash = *alignup<16>(hashbuf);
3353         loop
3354         {
3355             unsigned int nHashesDone = 0;
3356             unsigned int nNonceFound;
3357
3358             // Crypto++ SHA-256
3359             nNonceFound = ScanHash_CryptoPP(pmidstate, pdata + 64, phash1,
3360                                             (char*)&hash, nHashesDone);
3361
3362             // Check if something found
3363             if (nNonceFound != -1)
3364             {
3365                 for (int i = 0; i < sizeof(hash)/4; i++)
3366                     ((unsigned int*)&hash)[i] = ByteReverse(((unsigned int*)&hash)[i]);
3367
3368                 if (hash <= hashTarget)
3369                 {
3370                     // Found a solution
3371                     pblock->nNonce = ByteReverse(nNonceFound);
3372                     assert(hash == pblock->GetHash());
3373
3374                     SetThreadPriority(THREAD_PRIORITY_NORMAL);
3375                     CheckWork(pblock.get(), *pwalletMain, reservekey);
3376                     SetThreadPriority(THREAD_PRIORITY_LOWEST);
3377                     break;
3378                 }
3379             }
3380
3381             // Meter hashes/sec
3382             static int64 nHashCounter;
3383             if (nHPSTimerStart == 0)
3384             {
3385                 nHPSTimerStart = GetTimeMillis();
3386                 nHashCounter = 0;
3387             }
3388             else
3389                 nHashCounter += nHashesDone;
3390             if (GetTimeMillis() - nHPSTimerStart > 4000)
3391             {
3392                 static CCriticalSection cs;
3393                 CRITICAL_BLOCK(cs)
3394                 {
3395                     if (GetTimeMillis() - nHPSTimerStart > 4000)
3396                     {
3397                         dHashesPerSec = 1000.0 * nHashCounter / (GetTimeMillis() - nHPSTimerStart);
3398                         nHPSTimerStart = GetTimeMillis();
3399                         nHashCounter = 0;
3400                         string strStatus = strprintf("    %.0f khash/s", dHashesPerSec/1000.0);
3401                         UIThreadCall(boost::bind(CalledSetStatusBar, strStatus, 0));
3402                         static int64 nLogTime;
3403                         if (GetTime() - nLogTime > 30 * 60)
3404                         {
3405                             nLogTime = GetTime();
3406                             printf("%s ", DateTimeStrFormat("%x %H:%M", GetTime()).c_str());
3407                             printf("hashmeter %3d CPUs %6.0f khash/s\n", vnThreadsRunning[THREAD_MINER], dHashesPerSec/1000.0);
3408                         }
3409                     }
3410                 }
3411             }
3412
3413             // Check for stop or if block needs to be rebuilt
3414             if (fShutdown)
3415                 return;
3416             if (!fGenerateBitcoins)
3417                 return;
3418             if (fLimitProcessors && vnThreadsRunning[THREAD_MINER] > nLimitProcessors)
3419                 return;
3420             if (vNodes.empty())
3421                 break;
3422             if (nBlockNonce >= 0xffff0000)
3423                 break;
3424             if (nTransactionsUpdated != nTransactionsUpdatedLast && GetTime() - nStart > 60)
3425                 break;
3426             if (pindexPrev != pindexBest)
3427                 break;
3428
3429             // Update nTime every few seconds
3430             pblock->UpdateTime(pindexPrev);
3431             nBlockTime = ByteReverse(pblock->nTime);
3432             if (fTestNet)
3433             {
3434                 // Changing pblock->nTime can change work required on testnet:
3435                 nBlockBits = ByteReverse(pblock->nBits);
3436                 hashTarget = CBigNum().SetCompact(pblock->nBits).getuint256();
3437             }
3438         }
3439     }
3440 }
3441
3442 void static ThreadBitcoinMiner(void* parg)
3443 {
3444     CWallet* pwallet = (CWallet*)parg;
3445     try
3446     {
3447         vnThreadsRunning[THREAD_MINER]++;
3448         BitcoinMiner(pwallet);
3449         vnThreadsRunning[THREAD_MINER]--;
3450     }
3451     catch (std::exception& e) {
3452         vnThreadsRunning[THREAD_MINER]--;
3453         PrintException(&e, "ThreadBitcoinMiner()");
3454     } catch (...) {
3455         vnThreadsRunning[THREAD_MINER]--;
3456         PrintException(NULL, "ThreadBitcoinMiner()");
3457     }
3458     UIThreadCall(boost::bind(CalledSetStatusBar, "", 0));
3459     nHPSTimerStart = 0;
3460     if (vnThreadsRunning[THREAD_MINER] == 0)
3461         dHashesPerSec = 0;
3462     printf("ThreadBitcoinMiner exiting, %d threads remaining\n", vnThreadsRunning[THREAD_MINER]);
3463 }
3464
3465
3466 void GenerateBitcoins(bool fGenerate, CWallet* pwallet)
3467 {
3468     fGenerateBitcoins = fGenerate;
3469     nLimitProcessors = GetArg("-genproclimit", -1);
3470     if (nLimitProcessors == 0)
3471         fGenerateBitcoins = false;
3472     fLimitProcessors = (nLimitProcessors != -1);
3473
3474     if (fGenerate)
3475     {
3476         int nProcessors = boost::thread::hardware_concurrency();
3477         printf("%d processors\n", nProcessors);
3478         if (nProcessors < 1)
3479             nProcessors = 1;
3480         if (fLimitProcessors && nProcessors > nLimitProcessors)
3481             nProcessors = nLimitProcessors;
3482         int nAddThreads = nProcessors - vnThreadsRunning[THREAD_MINER];
3483         printf("Starting %d BitcoinMiner threads\n", nAddThreads);
3484         for (int i = 0; i < nAddThreads; i++)
3485         {
3486             if (!CreateThread(ThreadBitcoinMiner, pwallet))
3487                 printf("Error: CreateThread(ThreadBitcoinMiner) failed\n");
3488             Sleep(10);
3489         }
3490     }
3491 }