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