Added 320 fresh seednodes to replace the old ones.
[novacoin.git] / main.cpp
1 // Copyright (c) 2009-2010 Satoshi Nakamoto
2 // Distributed under the MIT/X11 software license, see the accompanying
3 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
4
5 #include "headers.h"
6 #include "cryptopp/sha.h"
7
8
9
10
11
12 //
13 // Global state
14 //
15
16 CCriticalSection cs_main;
17
18 map<uint256, CTransaction> mapTransactions;
19 CCriticalSection cs_mapTransactions;
20 unsigned int nTransactionsUpdated = 0;
21 map<COutPoint, CInPoint> mapNextTx;
22
23 map<uint256, CBlockIndex*> mapBlockIndex;
24 uint256 hashGenesisBlock("0x000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f");
25 CBigNum bnProofOfWorkLimit(~uint256(0) >> 32);
26 CBlockIndex* pindexGenesisBlock = NULL;
27 int nBestHeight = -1;
28 CBigNum bnBestChainWork = 0;
29 CBigNum bnBestInvalidWork = 0;
30 uint256 hashBestChain = 0;
31 CBlockIndex* pindexBest = NULL;
32 int64 nTimeBestReceived = 0;
33
34 map<uint256, CBlock*> mapOrphanBlocks;
35 multimap<uint256, CBlock*> mapOrphanBlocksByPrev;
36
37 map<uint256, CDataStream*> mapOrphanTransactions;
38 multimap<uint256, CDataStream*> mapOrphanTransactionsByPrev;
39
40 map<uint256, CWalletTx> mapWallet;
41 vector<uint256> vWalletUpdated;
42 CCriticalSection cs_mapWallet;
43
44 map<vector<unsigned char>, CPrivKey> mapKeys;
45 map<uint160, vector<unsigned char> > mapPubKeys;
46 CCriticalSection cs_mapKeys;
47 CKey keyUser;
48
49 map<uint256, int> mapRequestCount;
50 CCriticalSection cs_mapRequestCount;
51
52 map<string, string> mapAddressBook;
53 CCriticalSection cs_mapAddressBook;
54
55 vector<unsigned char> vchDefaultKey;
56
57 double dHashesPerSec;
58 int64 nHPSTimerStart;
59
60 // Settings
61 int fGenerateBitcoins = false;
62 int64 nTransactionFee = 0;
63 CAddress addrIncoming;
64 int fLimitProcessors = false;
65 int nLimitProcessors = 1;
66 int fMinimizeToTray = true;
67 int fMinimizeOnClose = true;
68
69
70
71
72
73
74 //////////////////////////////////////////////////////////////////////////////
75 //
76 // mapKeys
77 //
78
79 bool AddKey(const CKey& key)
80 {
81     CRITICAL_BLOCK(cs_mapKeys)
82     {
83         mapKeys[key.GetPubKey()] = key.GetPrivKey();
84         mapPubKeys[Hash160(key.GetPubKey())] = key.GetPubKey();
85     }
86     return CWalletDB().WriteKey(key.GetPubKey(), key.GetPrivKey());
87 }
88
89 vector<unsigned char> GenerateNewKey()
90 {
91     RandAddSeedPerfmon();
92     CKey key;
93     key.MakeNewKey();
94     if (!AddKey(key))
95         throw runtime_error("GenerateNewKey() : AddKey failed");
96     return key.GetPubKey();
97 }
98
99
100
101
102 //////////////////////////////////////////////////////////////////////////////
103 //
104 // mapWallet
105 //
106
107 bool AddToWallet(const CWalletTx& wtxIn)
108 {
109     uint256 hash = wtxIn.GetHash();
110     CRITICAL_BLOCK(cs_mapWallet)
111     {
112         // Inserts only if not already there, returns tx inserted or tx found
113         pair<map<uint256, CWalletTx>::iterator, bool> ret = mapWallet.insert(make_pair(hash, wtxIn));
114         CWalletTx& wtx = (*ret.first).second;
115         bool fInsertedNew = ret.second;
116         if (fInsertedNew)
117             wtx.nTimeReceived = GetAdjustedTime();
118
119         bool fUpdated = false;
120         if (!fInsertedNew)
121         {
122             // Merge
123             if (wtxIn.hashBlock != 0 && wtxIn.hashBlock != wtx.hashBlock)
124             {
125                 wtx.hashBlock = wtxIn.hashBlock;
126                 fUpdated = true;
127             }
128             if (wtxIn.nIndex != -1 && (wtxIn.vMerkleBranch != wtx.vMerkleBranch || wtxIn.nIndex != wtx.nIndex))
129             {
130                 wtx.vMerkleBranch = wtxIn.vMerkleBranch;
131                 wtx.nIndex = wtxIn.nIndex;
132                 fUpdated = true;
133             }
134             if (wtxIn.fFromMe && wtxIn.fFromMe != wtx.fFromMe)
135             {
136                 wtx.fFromMe = wtxIn.fFromMe;
137                 fUpdated = true;
138             }
139             if (wtxIn.fSpent && wtxIn.fSpent != wtx.fSpent)
140             {
141                 wtx.fSpent = wtxIn.fSpent;
142                 fUpdated = true;
143             }
144         }
145
146         //// debug print
147         printf("AddToWallet %s  %s%s\n", wtxIn.GetHash().ToString().substr(0,10).c_str(), (fInsertedNew ? "new" : ""), (fUpdated ? "update" : ""));
148
149         // Write to disk
150         if (fInsertedNew || fUpdated)
151             if (!wtx.WriteToDisk())
152                 return false;
153
154         // If default receiving address gets used, replace it with a new one
155         CScript scriptDefaultKey;
156         scriptDefaultKey.SetBitcoinAddress(vchDefaultKey);
157         foreach(const CTxOut& txout, wtx.vout)
158         {
159             if (txout.scriptPubKey == scriptDefaultKey)
160             {
161                 CWalletDB walletdb;
162                 vchDefaultKey = GetKeyFromKeyPool();
163                 walletdb.WriteDefaultKey(vchDefaultKey);
164                 walletdb.WriteName(PubKeyToAddress(vchDefaultKey), "");
165             }
166         }
167
168         // Notify UI
169         vWalletUpdated.push_back(hash);
170     }
171
172     // Refresh UI
173     MainFrameRepaint();
174     return true;
175 }
176
177 bool AddToWalletIfMine(const CTransaction& tx, const CBlock* pblock)
178 {
179     if (tx.IsMine() || mapWallet.count(tx.GetHash()))
180     {
181         CWalletTx wtx(tx);
182         // Get merkle branch if transaction was found in a block
183         if (pblock)
184             wtx.SetMerkleBranch(pblock);
185         return AddToWallet(wtx);
186     }
187     return true;
188 }
189
190 bool AddToWalletIfFromMe(const CTransaction& tx, const CBlock* pblock)
191 {
192     if (tx.IsFromMe() || mapWallet.count(tx.GetHash()))
193     {
194         CWalletTx wtx(tx);
195         // Get merkle branch if transaction was found in a block
196         if (pblock)
197             wtx.SetMerkleBranch(pblock);
198         return AddToWallet(wtx);
199     }
200     return true;
201 }
202
203 bool EraseFromWallet(uint256 hash)
204 {
205     CRITICAL_BLOCK(cs_mapWallet)
206     {
207         if (mapWallet.erase(hash))
208             CWalletDB().EraseTx(hash);
209     }
210     return true;
211 }
212
213 void WalletUpdateSpent(const COutPoint& prevout)
214 {
215     // Anytime a signature is successfully verified, it's proof the outpoint is spent.
216     // Update the wallet spent flag if it doesn't know due to wallet.dat being
217     // restored from backup or the user making copies of wallet.dat.
218     CRITICAL_BLOCK(cs_mapWallet)
219     {
220         map<uint256, CWalletTx>::iterator mi = mapWallet.find(prevout.hash);
221         if (mi != mapWallet.end())
222         {
223             CWalletTx& wtx = (*mi).second;
224             if (!wtx.fSpent && wtx.vout[prevout.n].IsMine())
225             {
226                 printf("WalletUpdateSpent found spent coin %sbc %s\n", FormatMoney(wtx.GetCredit()).c_str(), wtx.GetHash().ToString().c_str());
227                 wtx.fSpent = true;
228                 wtx.WriteToDisk();
229                 vWalletUpdated.push_back(prevout.hash);
230             }
231         }
232     }
233 }
234
235
236
237
238
239
240
241
242 //////////////////////////////////////////////////////////////////////////////
243 //
244 // mapOrphanTransactions
245 //
246
247 void AddOrphanTx(const CDataStream& vMsg)
248 {
249     CTransaction tx;
250     CDataStream(vMsg) >> tx;
251     uint256 hash = tx.GetHash();
252     if (mapOrphanTransactions.count(hash))
253         return;
254     CDataStream* pvMsg = mapOrphanTransactions[hash] = new CDataStream(vMsg);
255     foreach(const CTxIn& txin, tx.vin)
256         mapOrphanTransactionsByPrev.insert(make_pair(txin.prevout.hash, pvMsg));
257 }
258
259 void EraseOrphanTx(uint256 hash)
260 {
261     if (!mapOrphanTransactions.count(hash))
262         return;
263     const CDataStream* pvMsg = mapOrphanTransactions[hash];
264     CTransaction tx;
265     CDataStream(*pvMsg) >> tx;
266     foreach(const CTxIn& txin, tx.vin)
267     {
268         for (multimap<uint256, CDataStream*>::iterator mi = mapOrphanTransactionsByPrev.lower_bound(txin.prevout.hash);
269              mi != mapOrphanTransactionsByPrev.upper_bound(txin.prevout.hash);)
270         {
271             if ((*mi).second == pvMsg)
272                 mapOrphanTransactionsByPrev.erase(mi++);
273             else
274                 mi++;
275         }
276     }
277     delete pvMsg;
278     mapOrphanTransactions.erase(hash);
279 }
280
281
282
283
284
285
286
287
288 //////////////////////////////////////////////////////////////////////////////
289 //
290 // CTransaction
291 //
292
293 bool CTransaction::ReadFromDisk(CTxDB& txdb, COutPoint prevout, CTxIndex& txindexRet)
294 {
295     SetNull();
296     if (!txdb.ReadTxIndex(prevout.hash, txindexRet))
297         return false;
298     if (!ReadFromDisk(txindexRet.pos))
299         return false;
300     if (prevout.n >= vout.size())
301     {
302         SetNull();
303         return false;
304     }
305     return true;
306 }
307
308 bool CTransaction::ReadFromDisk(CTxDB& txdb, COutPoint prevout)
309 {
310     CTxIndex txindex;
311     return ReadFromDisk(txdb, prevout, txindex);
312 }
313
314 bool CTransaction::ReadFromDisk(COutPoint prevout)
315 {
316     CTxDB txdb("r");
317     CTxIndex txindex;
318     return ReadFromDisk(txdb, prevout, txindex);
319 }
320
321 bool CTxIn::IsMine() const
322 {
323     CRITICAL_BLOCK(cs_mapWallet)
324     {
325         map<uint256, CWalletTx>::iterator mi = mapWallet.find(prevout.hash);
326         if (mi != mapWallet.end())
327         {
328             const CWalletTx& prev = (*mi).second;
329             if (prevout.n < prev.vout.size())
330                 if (prev.vout[prevout.n].IsMine())
331                     return true;
332         }
333     }
334     return false;
335 }
336
337 int64 CTxIn::GetDebit() const
338 {
339     CRITICAL_BLOCK(cs_mapWallet)
340     {
341         map<uint256, CWalletTx>::iterator mi = mapWallet.find(prevout.hash);
342         if (mi != mapWallet.end())
343         {
344             const CWalletTx& prev = (*mi).second;
345             if (prevout.n < prev.vout.size())
346                 if (prev.vout[prevout.n].IsMine())
347                     return prev.vout[prevout.n].nValue;
348         }
349     }
350     return 0;
351 }
352
353 int64 CWalletTx::GetTxTime() const
354 {
355     if (!fTimeReceivedIsTxTime && hashBlock != 0)
356     {
357         // If we did not receive the transaction directly, we rely on the block's
358         // time to figure out when it happened.  We use the median over a range
359         // of blocks to try to filter out inaccurate block times.
360         map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashBlock);
361         if (mi != mapBlockIndex.end())
362         {
363             CBlockIndex* pindex = (*mi).second;
364             if (pindex)
365                 return pindex->GetMedianTime();
366         }
367     }
368     return nTimeReceived;
369 }
370
371 int CWalletTx::GetRequestCount() const
372 {
373     // Returns -1 if it wasn't being tracked
374     int nRequests = -1;
375     CRITICAL_BLOCK(cs_mapRequestCount)
376     {
377         if (IsCoinBase())
378         {
379             // Generated block
380             if (hashBlock != 0)
381             {
382                 map<uint256, int>::iterator mi = mapRequestCount.find(hashBlock);
383                 if (mi != mapRequestCount.end())
384                     nRequests = (*mi).second;
385             }
386         }
387         else
388         {
389             // Did anyone request this transaction?
390             map<uint256, int>::iterator mi = mapRequestCount.find(GetHash());
391             if (mi != mapRequestCount.end())
392             {
393                 nRequests = (*mi).second;
394
395                 // How about the block it's in?
396                 if (nRequests == 0 && hashBlock != 0)
397                 {
398                     map<uint256, int>::iterator mi = mapRequestCount.find(hashBlock);
399                     if (mi != mapRequestCount.end())
400                         nRequests = (*mi).second;
401                     else
402                         nRequests = 1; // If it's in someone else's block it must have got out
403                 }
404             }
405         }
406     }
407     return nRequests;
408 }
409
410 void CWalletTx::GetAmounts(int64& nGenerated, list<pair<string, int64> >& listReceived,
411                            list<pair<string, int64> >& listSent, int64& nFee, string& strSentAccount) const
412 {
413     nGenerated = nFee = 0;
414     listReceived.clear();
415     listSent.clear();
416     strSentAccount = strFromAccount;
417
418     if (IsCoinBase())
419     {
420         if (GetDepthInMainChain() >= COINBASE_MATURITY)
421             nGenerated = GetCredit();
422         return;
423     }
424
425     // Compute fee:
426     int64 nDebit = GetDebit();
427     if (nDebit > 0) // debit>0 means we signed/sent this transaction
428     {
429         int64 nValueOut = GetValueOut();
430         nFee = nDebit - nValueOut;
431     }
432
433     // Sent/received.  Standard client will never generate a send-to-multiple-recipients,
434     // but non-standard clients might (so return a list of address/amount pairs)
435     foreach(const CTxOut& txout, vout)
436     {
437         string address;
438         uint160 hash160;
439         vector<unsigned char> vchPubKey;
440         if (ExtractHash160(txout.scriptPubKey, hash160))
441             address = Hash160ToAddress(hash160);
442         else if (ExtractPubKey(txout.scriptPubKey, false, vchPubKey))
443             address = PubKeyToAddress(vchPubKey);
444         else
445         {
446             printf("CWalletTx::GetAmounts: Unknown transaction type found, txid %s\n",
447                    this->GetHash().ToString().c_str());
448             address = " unknown ";
449         }
450
451         // Don't report 'change' txouts
452         if (nDebit > 0 && txout.IsChange())
453             continue;
454
455         if (nDebit > 0)
456             listSent.push_back(make_pair(address, txout.nValue));
457
458         if (txout.IsMine())
459             listReceived.push_back(make_pair(address, txout.nValue));
460     }
461
462 }
463
464 void CWalletTx::GetAccountAmounts(const string& strAccount, int64& nGenerated, int64& nReceived, 
465                                   int64& nSent, int64& nFee) const
466 {
467     nGenerated = nReceived = nSent = nFee = 0;
468
469     int64 allGenerated, allFee;
470     allGenerated = allFee = 0;
471     string strSentAccount;
472     list<pair<string, int64> > listReceived;
473     list<pair<string, int64> > listSent;
474     GetAmounts(allGenerated, listReceived, listSent, allFee, strSentAccount);
475
476     if (strAccount == "")
477         nGenerated = allGenerated;
478     if (strAccount == strSentAccount)
479     {
480         foreach(const PAIRTYPE(string,int64)& s, listSent)
481             nSent += s.second;
482         nFee = allFee;
483     }
484     CRITICAL_BLOCK(cs_mapAddressBook)
485     {
486         foreach(const PAIRTYPE(string,int64)& r, listReceived)
487         {
488             if (mapAddressBook.count(r.first))
489             {
490                 if (mapAddressBook[r.first] == strAccount)
491                 {
492                     nReceived += r.second;
493                 }
494             }
495             else if (strAccount.empty())
496             {
497                 nReceived += r.second;
498             }
499         }
500     }
501 }
502
503
504
505 int CMerkleTx::SetMerkleBranch(const CBlock* pblock)
506 {
507     if (fClient)
508     {
509         if (hashBlock == 0)
510             return 0;
511     }
512     else
513     {
514         CBlock blockTmp;
515         if (pblock == NULL)
516         {
517             // Load the block this tx is in
518             CTxIndex txindex;
519             if (!CTxDB("r").ReadTxIndex(GetHash(), txindex))
520                 return 0;
521             if (!blockTmp.ReadFromDisk(txindex.pos.nFile, txindex.pos.nBlockPos))
522                 return 0;
523             pblock = &blockTmp;
524         }
525
526         // Update the tx's hashBlock
527         hashBlock = pblock->GetHash();
528
529         // Locate the transaction
530         for (nIndex = 0; nIndex < pblock->vtx.size(); nIndex++)
531             if (pblock->vtx[nIndex] == *(CTransaction*)this)
532                 break;
533         if (nIndex == pblock->vtx.size())
534         {
535             vMerkleBranch.clear();
536             nIndex = -1;
537             printf("ERROR: SetMerkleBranch() : couldn't find tx in block\n");
538             return 0;
539         }
540
541         // Fill in merkle branch
542         vMerkleBranch = pblock->GetMerkleBranch(nIndex);
543     }
544
545     // Is the tx in a block that's in the main chain
546     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashBlock);
547     if (mi == mapBlockIndex.end())
548         return 0;
549     CBlockIndex* pindex = (*mi).second;
550     if (!pindex || !pindex->IsInMainChain())
551         return 0;
552
553     return pindexBest->nHeight - pindex->nHeight + 1;
554 }
555
556
557
558 void CWalletTx::AddSupportingTransactions(CTxDB& txdb)
559 {
560     vtxPrev.clear();
561
562     const int COPY_DEPTH = 3;
563     if (SetMerkleBranch() < COPY_DEPTH)
564     {
565         vector<uint256> vWorkQueue;
566         foreach(const CTxIn& txin, vin)
567             vWorkQueue.push_back(txin.prevout.hash);
568
569         // This critsect is OK because txdb is already open
570         CRITICAL_BLOCK(cs_mapWallet)
571         {
572             map<uint256, const CMerkleTx*> mapWalletPrev;
573             set<uint256> setAlreadyDone;
574             for (int i = 0; i < vWorkQueue.size(); i++)
575             {
576                 uint256 hash = vWorkQueue[i];
577                 if (setAlreadyDone.count(hash))
578                     continue;
579                 setAlreadyDone.insert(hash);
580
581                 CMerkleTx tx;
582                 if (mapWallet.count(hash))
583                 {
584                     tx = mapWallet[hash];
585                     foreach(const CMerkleTx& txWalletPrev, mapWallet[hash].vtxPrev)
586                         mapWalletPrev[txWalletPrev.GetHash()] = &txWalletPrev;
587                 }
588                 else if (mapWalletPrev.count(hash))
589                 {
590                     tx = *mapWalletPrev[hash];
591                 }
592                 else if (!fClient && txdb.ReadDiskTx(hash, tx))
593                 {
594                     ;
595                 }
596                 else
597                 {
598                     printf("ERROR: AddSupportingTransactions() : unsupported transaction\n");
599                     continue;
600                 }
601
602                 int nDepth = tx.SetMerkleBranch();
603                 vtxPrev.push_back(tx);
604
605                 if (nDepth < COPY_DEPTH)
606                     foreach(const CTxIn& txin, tx.vin)
607                         vWorkQueue.push_back(txin.prevout.hash);
608             }
609         }
610     }
611
612     reverse(vtxPrev.begin(), vtxPrev.end());
613 }
614
615
616
617
618
619
620
621
622
623
624
625 bool CTransaction::CheckTransaction() const
626 {
627     // Basic checks that don't depend on any context
628     if (vin.empty() || vout.empty())
629         return error("CTransaction::CheckTransaction() : vin or vout empty");
630
631     // Size limits
632     if (::GetSerializeSize(*this, SER_NETWORK) > MAX_BLOCK_SIZE)
633         return error("CTransaction::CheckTransaction() : size limits failed");
634
635     // Check for negative or overflow output values
636     int64 nValueOut = 0;
637     foreach(const CTxOut& txout, vout)
638     {
639         if (txout.nValue < 0)
640             return error("CTransaction::CheckTransaction() : txout.nValue negative");
641         if (txout.nValue > MAX_MONEY)
642             return error("CTransaction::CheckTransaction() : txout.nValue too high");
643         nValueOut += txout.nValue;
644         if (!MoneyRange(nValueOut))
645             return error("CTransaction::CheckTransaction() : txout total out of range");
646     }
647
648     if (IsCoinBase())
649     {
650         if (vin[0].scriptSig.size() < 2 || vin[0].scriptSig.size() > 100)
651             return error("CTransaction::CheckTransaction() : coinbase script size");
652     }
653     else
654     {
655         foreach(const CTxIn& txin, vin)
656             if (txin.prevout.IsNull())
657                 return error("CTransaction::CheckTransaction() : prevout is null");
658     }
659
660     return true;
661 }
662
663 bool CTransaction::AcceptToMemoryPool(CTxDB& txdb, bool fCheckInputs, bool* pfMissingInputs)
664 {
665     if (pfMissingInputs)
666         *pfMissingInputs = false;
667
668     if (!CheckTransaction())
669         return error("AcceptToMemoryPool() : CheckTransaction failed");
670
671     // Coinbase is only valid in a block, not as a loose transaction
672     if (IsCoinBase())
673         return error("AcceptToMemoryPool() : coinbase as individual tx");
674
675     // To help v0.1.5 clients who would see it as a negative number
676     if ((int64)nLockTime > INT_MAX)
677         return error("AcceptToMemoryPool() : not accepting nLockTime beyond 2038 yet");
678
679     // Safety limits
680     unsigned int nSize = ::GetSerializeSize(*this, SER_NETWORK);
681     if (GetSigOpCount() > 2 || nSize < 100)
682         return error("AcceptToMemoryPool() : nonstandard transaction");
683
684     // Rather not work on nonstandard transactions
685     if (!IsStandard())
686         return error("AcceptToMemoryPool() : nonstandard transaction type");
687
688     // Do we already have it?
689     uint256 hash = GetHash();
690     CRITICAL_BLOCK(cs_mapTransactions)
691         if (mapTransactions.count(hash))
692             return false;
693     if (fCheckInputs)
694         if (txdb.ContainsTx(hash))
695             return false;
696
697     // Check for conflicts with in-memory transactions
698     CTransaction* ptxOld = NULL;
699     for (int i = 0; i < vin.size(); i++)
700     {
701         COutPoint outpoint = vin[i].prevout;
702         if (mapNextTx.count(outpoint))
703         {
704             // Disable replacement feature for now
705             return false;
706
707             // Allow replacing with a newer version of the same transaction
708             if (i != 0)
709                 return false;
710             ptxOld = mapNextTx[outpoint].ptx;
711             if (ptxOld->IsFinal())
712                 return false;
713             if (!IsNewerThan(*ptxOld))
714                 return false;
715             for (int i = 0; i < vin.size(); i++)
716             {
717                 COutPoint outpoint = vin[i].prevout;
718                 if (!mapNextTx.count(outpoint) || mapNextTx[outpoint].ptx != ptxOld)
719                     return false;
720             }
721             break;
722         }
723     }
724
725     if (fCheckInputs)
726     {
727         // Check against previous transactions
728         map<uint256, CTxIndex> mapUnused;
729         int64 nFees = 0;
730         if (!ConnectInputs(txdb, mapUnused, CDiskTxPos(1,1,1), pindexBest, nFees, false, false))
731         {
732             if (pfMissingInputs)
733                 *pfMissingInputs = true;
734             return error("AcceptToMemoryPool() : ConnectInputs failed %s", hash.ToString().substr(0,10).c_str());
735         }
736
737         // Don't accept it if it can't get into a block
738         if (nFees < GetMinFee(1000))
739             return error("AcceptToMemoryPool() : not enough fees");
740
741         // Limit free transactions per 10 minutes
742         if (nFees < CENT && GetBoolArg("-limitfreerelay"))
743         {
744             static int64 nNextReset;
745             static int64 nFreeCount;
746             if (GetTime() > nNextReset)
747             {
748                 nNextReset = GetTime() + 10 * 60;
749                 nFreeCount = 0;
750             }
751             if (nFreeCount > 150000 && !IsFromMe())
752                 return error("AcceptToMemoryPool() : free transaction rejected by rate limiter");
753             nFreeCount += nSize;
754         }
755     }
756
757     // Store transaction in memory
758     CRITICAL_BLOCK(cs_mapTransactions)
759     {
760         if (ptxOld)
761         {
762             printf("AcceptToMemoryPool() : replacing tx %s with new version\n", ptxOld->GetHash().ToString().c_str());
763             ptxOld->RemoveFromMemoryPool();
764         }
765         AddToMemoryPoolUnchecked();
766     }
767
768     ///// are we sure this is ok when loading transactions or restoring block txes
769     // If updated, erase old tx from wallet
770     if (ptxOld)
771         EraseFromWallet(ptxOld->GetHash());
772
773     printf("AcceptToMemoryPool(): accepted %s\n", hash.ToString().substr(0,10).c_str());
774     return true;
775 }
776
777
778 bool CTransaction::AddToMemoryPoolUnchecked()
779 {
780     // Add to memory pool without checking anything.  Don't call this directly,
781     // call AcceptToMemoryPool to properly check the transaction first.
782     CRITICAL_BLOCK(cs_mapTransactions)
783     {
784         uint256 hash = GetHash();
785         mapTransactions[hash] = *this;
786         for (int i = 0; i < vin.size(); i++)
787             mapNextTx[vin[i].prevout] = CInPoint(&mapTransactions[hash], i);
788         nTransactionsUpdated++;
789     }
790     return true;
791 }
792
793
794 bool CTransaction::RemoveFromMemoryPool()
795 {
796     // Remove transaction from memory pool
797     CRITICAL_BLOCK(cs_mapTransactions)
798     {
799         foreach(const CTxIn& txin, vin)
800             mapNextTx.erase(txin.prevout);
801         mapTransactions.erase(GetHash());
802         nTransactionsUpdated++;
803     }
804     return true;
805 }
806
807
808
809
810
811
812 int CMerkleTx::GetDepthInMainChain(int& nHeightRet) const
813 {
814     if (hashBlock == 0 || nIndex == -1)
815         return 0;
816
817     // Find the block it claims to be in
818     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashBlock);
819     if (mi == mapBlockIndex.end())
820         return 0;
821     CBlockIndex* pindex = (*mi).second;
822     if (!pindex || !pindex->IsInMainChain())
823         return 0;
824
825     // Make sure the merkle branch connects to this block
826     if (!fMerkleVerified)
827     {
828         if (CBlock::CheckMerkleBranch(GetHash(), vMerkleBranch, nIndex) != pindex->hashMerkleRoot)
829             return 0;
830         fMerkleVerified = true;
831     }
832
833     nHeightRet = pindex->nHeight;
834     return pindexBest->nHeight - pindex->nHeight + 1;
835 }
836
837
838 int CMerkleTx::GetBlocksToMaturity() const
839 {
840     if (!IsCoinBase())
841         return 0;
842     return max(0, (COINBASE_MATURITY+20) - GetDepthInMainChain());
843 }
844
845
846 bool CMerkleTx::AcceptToMemoryPool(CTxDB& txdb, bool fCheckInputs)
847 {
848     if (fClient)
849     {
850         if (!IsInMainChain() && !ClientConnectInputs())
851             return false;
852         return CTransaction::AcceptToMemoryPool(txdb, false);
853     }
854     else
855     {
856         return CTransaction::AcceptToMemoryPool(txdb, fCheckInputs);
857     }
858 }
859
860
861
862 bool CWalletTx::AcceptWalletTransaction(CTxDB& txdb, bool fCheckInputs)
863 {
864     CRITICAL_BLOCK(cs_mapTransactions)
865     {
866         // Add previous supporting transactions first
867         foreach(CMerkleTx& tx, vtxPrev)
868         {
869             if (!tx.IsCoinBase())
870             {
871                 uint256 hash = tx.GetHash();
872                 if (!mapTransactions.count(hash) && !txdb.ContainsTx(hash))
873                     tx.AcceptToMemoryPool(txdb, fCheckInputs);
874             }
875         }
876         return AcceptToMemoryPool(txdb, fCheckInputs);
877     }
878     return false;
879 }
880
881 int ScanForWalletTransactions(CBlockIndex* pindexStart)
882 {
883     int ret = 0;
884
885     CBlockIndex* pindex = pindexStart;
886     CRITICAL_BLOCK(cs_mapWallet)
887     {
888         while (pindex)
889         {
890             CBlock block;
891             block.ReadFromDisk(pindex, true);
892             foreach(CTransaction& tx, block.vtx)
893             {
894                 uint256 hash = tx.GetHash();
895                 if (mapWallet.count(hash)) continue;
896                 AddToWalletIfMine(tx, &block);
897                 if (mapWallet.count(hash))
898                 {
899                     ++ret;
900                     printf("Added missing RECEIVE %s\n", hash.ToString().c_str());
901                     continue;
902                 }
903                 AddToWalletIfFromMe(tx, &block);
904                 if (mapWallet.count(hash))
905                 {
906                     ++ret;
907                     printf("Added missing SEND %s\n", hash.ToString().c_str());
908                     continue;
909                 }
910             }
911             pindex = pindex->pnext;
912         }
913     }
914     return ret;
915 }
916
917 void ReacceptWalletTransactions()
918 {
919     CTxDB txdb("r");
920     bool fRepeat = true;
921     while (fRepeat) CRITICAL_BLOCK(cs_mapWallet)
922     {
923         fRepeat = false;
924         vector<CDiskTxPos> vMissingTx;
925         foreach(PAIRTYPE(const uint256, CWalletTx)& item, mapWallet)
926         {
927             CWalletTx& wtx = item.second;
928             if (wtx.fSpent && wtx.IsCoinBase())
929                 continue;
930
931             CTxIndex txindex;
932             if (txdb.ReadTxIndex(wtx.GetHash(), txindex))
933             {
934                 // Update fSpent if a tx got spent somewhere else by a copy of wallet.dat
935                 if (!wtx.fSpent)
936                 {
937                     if (txindex.vSpent.size() != wtx.vout.size())
938                     {
939                         printf("ERROR: ReacceptWalletTransactions() : txindex.vSpent.size() %d != wtx.vout.size() %d\n", txindex.vSpent.size(), wtx.vout.size());
940                         continue;
941                     }
942                     for (int i = 0; i < txindex.vSpent.size(); i++)
943                     {
944                         if (!txindex.vSpent[i].IsNull() && wtx.vout[i].IsMine())
945                         {
946                             wtx.fSpent = true;
947                             vMissingTx.push_back(txindex.vSpent[i]);
948                         }
949                     }
950                     if (wtx.fSpent)
951                     {
952                         printf("ReacceptWalletTransactions found spent coin %sbc %s\n", FormatMoney(wtx.GetCredit()).c_str(), wtx.GetHash().ToString().c_str());
953                         wtx.WriteToDisk();
954                     }
955                 }
956             }
957             else
958             {
959                 // Reaccept any txes of ours that aren't already in a block
960                 if (!wtx.IsCoinBase())
961                     wtx.AcceptWalletTransaction(txdb, false);
962             }
963         }
964         if (!vMissingTx.empty())
965         {
966             // TODO: optimize this to scan just part of the block chain?
967             if (ScanForWalletTransactions(pindexGenesisBlock))
968                 fRepeat = true;  // Found missing transactions: re-do Reaccept.
969         }
970     }
971 }
972
973
974 void CWalletTx::RelayWalletTransaction(CTxDB& txdb)
975 {
976     foreach(const CMerkleTx& tx, vtxPrev)
977     {
978         if (!tx.IsCoinBase())
979         {
980             uint256 hash = tx.GetHash();
981             if (!txdb.ContainsTx(hash))
982                 RelayMessage(CInv(MSG_TX, hash), (CTransaction)tx);
983         }
984     }
985     if (!IsCoinBase())
986     {
987         uint256 hash = GetHash();
988         if (!txdb.ContainsTx(hash))
989         {
990             printf("Relaying wtx %s\n", hash.ToString().substr(0,10).c_str());
991             RelayMessage(CInv(MSG_TX, hash), (CTransaction)*this);
992         }
993     }
994 }
995
996 void ResendWalletTransactions()
997 {
998     // Do this infrequently and randomly to avoid giving away
999     // that these are our transactions.
1000     static int64 nNextTime;
1001     if (GetTime() < nNextTime)
1002         return;
1003     bool fFirst = (nNextTime == 0);
1004     nNextTime = GetTime() + GetRand(30 * 60);
1005     if (fFirst)
1006         return;
1007
1008     // Only do it if there's been a new block since last time
1009     static int64 nLastTime;
1010     if (nTimeBestReceived < nLastTime)
1011         return;
1012     nLastTime = GetTime();
1013
1014     // Rebroadcast any of our txes that aren't in a block yet
1015     printf("ResendWalletTransactions()\n");
1016     CTxDB txdb("r");
1017     CRITICAL_BLOCK(cs_mapWallet)
1018     {
1019         // Sort them in chronological order
1020         multimap<unsigned int, CWalletTx*> mapSorted;
1021         foreach(PAIRTYPE(const uint256, CWalletTx)& item, mapWallet)
1022         {
1023             CWalletTx& wtx = item.second;
1024             // Don't rebroadcast until it's had plenty of time that
1025             // it should have gotten in already by now.
1026             if (nTimeBestReceived - (int64)wtx.nTimeReceived > 5 * 60)
1027                 mapSorted.insert(make_pair(wtx.nTimeReceived, &wtx));
1028         }
1029         foreach(PAIRTYPE(const unsigned int, CWalletTx*)& item, mapSorted)
1030         {
1031             CWalletTx& wtx = *item.second;
1032             wtx.RelayWalletTransaction(txdb);
1033         }
1034     }
1035 }
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046 //////////////////////////////////////////////////////////////////////////////
1047 //
1048 // CBlock and CBlockIndex
1049 //
1050
1051 bool CBlock::ReadFromDisk(const CBlockIndex* pindex, bool fReadTransactions)
1052 {
1053     if (!fReadTransactions)
1054     {
1055         *this = pindex->GetBlockHeader();
1056         return true;
1057     }
1058     if (!ReadFromDisk(pindex->nFile, pindex->nBlockPos, fReadTransactions))
1059         return false;
1060     if (GetHash() != pindex->GetBlockHash())
1061         return error("CBlock::ReadFromDisk() : GetHash() doesn't match index");
1062     return true;
1063 }
1064
1065 uint256 GetOrphanRoot(const CBlock* pblock)
1066 {
1067     // Work back to the first block in the orphan chain
1068     while (mapOrphanBlocks.count(pblock->hashPrevBlock))
1069         pblock = mapOrphanBlocks[pblock->hashPrevBlock];
1070     return pblock->GetHash();
1071 }
1072
1073 int64 GetBlockValue(int nHeight, int64 nFees)
1074 {
1075     int64 nSubsidy = 50 * COIN;
1076
1077     // Subsidy is cut in half every 4 years
1078     nSubsidy >>= (nHeight / 210000);
1079
1080     return nSubsidy + nFees;
1081 }
1082
1083 unsigned int GetNextWorkRequired(const CBlockIndex* pindexLast)
1084 {
1085     const int64 nTargetTimespan = 14 * 24 * 60 * 60; // two weeks
1086     const int64 nTargetSpacing = 10 * 60;
1087     const int64 nInterval = nTargetTimespan / nTargetSpacing;
1088
1089     // Genesis block
1090     if (pindexLast == NULL)
1091         return bnProofOfWorkLimit.GetCompact();
1092
1093     // Only change once per interval
1094     if ((pindexLast->nHeight+1) % nInterval != 0)
1095         return pindexLast->nBits;
1096
1097     // Go back by what we want to be 14 days worth of blocks
1098     const CBlockIndex* pindexFirst = pindexLast;
1099     for (int i = 0; pindexFirst && i < nInterval-1; i++)
1100         pindexFirst = pindexFirst->pprev;
1101     assert(pindexFirst);
1102
1103     // Limit adjustment step
1104     int64 nActualTimespan = pindexLast->GetBlockTime() - pindexFirst->GetBlockTime();
1105     printf("  nActualTimespan = %"PRI64d"  before bounds\n", nActualTimespan);
1106     if (nActualTimespan < nTargetTimespan/4)
1107         nActualTimespan = nTargetTimespan/4;
1108     if (nActualTimespan > nTargetTimespan*4)
1109         nActualTimespan = nTargetTimespan*4;
1110
1111     // Retarget
1112     CBigNum bnNew;
1113     bnNew.SetCompact(pindexLast->nBits);
1114     bnNew *= nActualTimespan;
1115     bnNew /= nTargetTimespan;
1116
1117     if (bnNew > bnProofOfWorkLimit)
1118         bnNew = bnProofOfWorkLimit;
1119
1120     /// debug print
1121     printf("GetNextWorkRequired RETARGET\n");
1122     printf("nTargetTimespan = %"PRI64d"    nActualTimespan = %"PRI64d"\n", nTargetTimespan, nActualTimespan);
1123     printf("Before: %08x  %s\n", pindexLast->nBits, CBigNum().SetCompact(pindexLast->nBits).getuint256().ToString().c_str());
1124     printf("After:  %08x  %s\n", bnNew.GetCompact(), bnNew.getuint256().ToString().c_str());
1125
1126     return bnNew.GetCompact();
1127 }
1128
1129 bool CheckProofOfWork(uint256 hash, unsigned int nBits)
1130 {
1131     CBigNum bnTarget;
1132     bnTarget.SetCompact(nBits);
1133
1134     // Check range
1135     if (bnTarget <= 0 || bnTarget > bnProofOfWorkLimit)
1136         return error("CheckProofOfWork() : nBits below minimum work");
1137
1138     // Check proof of work matches claimed amount
1139     if (hash > bnTarget.getuint256())
1140         return error("CheckProofOfWork() : hash doesn't match nBits");
1141
1142     return true;
1143 }
1144
1145 bool IsInitialBlockDownload()
1146 {
1147     if (pindexBest == NULL || (!fTestNet && nBestHeight < 105000))
1148         return true;
1149     static int64 nLastUpdate;
1150     static CBlockIndex* pindexLastBest;
1151     if (pindexBest != pindexLastBest)
1152     {
1153         pindexLastBest = pindexBest;
1154         nLastUpdate = GetTime();
1155     }
1156     return (GetTime() - nLastUpdate < 10 &&
1157             pindexBest->GetBlockTime() < GetTime() - 24 * 60 * 60);
1158 }
1159
1160 void InvalidChainFound(CBlockIndex* pindexNew)
1161 {
1162     if (pindexNew->bnChainWork > bnBestInvalidWork)
1163     {
1164         bnBestInvalidWork = pindexNew->bnChainWork;
1165         CTxDB().WriteBestInvalidWork(bnBestInvalidWork);
1166         MainFrameRepaint();
1167     }
1168     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());
1169     printf("InvalidChainFound:  current best=%s  height=%d  work=%s\n", hashBestChain.ToString().substr(0,20).c_str(), nBestHeight, bnBestChainWork.ToString().c_str());
1170     if (pindexBest && bnBestInvalidWork > bnBestChainWork + pindexBest->GetBlockWork() * 6)
1171         printf("InvalidChainFound: WARNING: Displayed transactions may not be correct!  You may need to upgrade, or other nodes may need to upgrade.\n");
1172 }
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184 bool CTransaction::DisconnectInputs(CTxDB& txdb)
1185 {
1186     // Relinquish previous transactions' spent pointers
1187     if (!IsCoinBase())
1188     {
1189         foreach(const CTxIn& txin, vin)
1190         {
1191             COutPoint prevout = txin.prevout;
1192
1193             // Get prev txindex from disk
1194             CTxIndex txindex;
1195             if (!txdb.ReadTxIndex(prevout.hash, txindex))
1196                 return error("DisconnectInputs() : ReadTxIndex failed");
1197
1198             if (prevout.n >= txindex.vSpent.size())
1199                 return error("DisconnectInputs() : prevout.n out of range");
1200
1201             // Mark outpoint as not spent
1202             txindex.vSpent[prevout.n].SetNull();
1203
1204             // Write back
1205             if (!txdb.UpdateTxIndex(prevout.hash, txindex))
1206                 return error("DisconnectInputs() : UpdateTxIndex failed");
1207         }
1208     }
1209
1210     // Remove transaction from index
1211     if (!txdb.EraseTxIndex(*this))
1212         return error("DisconnectInputs() : EraseTxPos failed");
1213
1214     return true;
1215 }
1216
1217
1218 bool CTransaction::ConnectInputs(CTxDB& txdb, map<uint256, CTxIndex>& mapTestPool, CDiskTxPos posThisTx,
1219                                  CBlockIndex* pindexBlock, int64& nFees, bool fBlock, bool fMiner, int64 nMinFee)
1220 {
1221     // Take over previous transactions' spent pointers
1222     if (!IsCoinBase())
1223     {
1224         int64 nValueIn = 0;
1225         for (int i = 0; i < vin.size(); i++)
1226         {
1227             COutPoint prevout = vin[i].prevout;
1228
1229             // Read txindex
1230             CTxIndex txindex;
1231             bool fFound = true;
1232             if (fMiner && mapTestPool.count(prevout.hash))
1233             {
1234                 // Get txindex from current proposed changes
1235                 txindex = mapTestPool[prevout.hash];
1236             }
1237             else
1238             {
1239                 // Read txindex from txdb
1240                 fFound = txdb.ReadTxIndex(prevout.hash, txindex);
1241             }
1242             if (!fFound && (fBlock || fMiner))
1243                 return fMiner ? false : error("ConnectInputs() : %s prev tx %s index entry not found", GetHash().ToString().substr(0,10).c_str(),  prevout.hash.ToString().substr(0,10).c_str());
1244
1245             // Read txPrev
1246             CTransaction txPrev;
1247             if (!fFound || txindex.pos == CDiskTxPos(1,1,1))
1248             {
1249                 // Get prev tx from single transactions in memory
1250                 CRITICAL_BLOCK(cs_mapTransactions)
1251                 {
1252                     if (!mapTransactions.count(prevout.hash))
1253                         return error("ConnectInputs() : %s mapTransactions prev not found %s", GetHash().ToString().substr(0,10).c_str(),  prevout.hash.ToString().substr(0,10).c_str());
1254                     txPrev = mapTransactions[prevout.hash];
1255                 }
1256                 if (!fFound)
1257                     txindex.vSpent.resize(txPrev.vout.size());
1258             }
1259             else
1260             {
1261                 // Get prev tx from disk
1262                 if (!txPrev.ReadFromDisk(txindex.pos))
1263                     return error("ConnectInputs() : %s ReadFromDisk prev tx %s failed", GetHash().ToString().substr(0,10).c_str(),  prevout.hash.ToString().substr(0,10).c_str());
1264             }
1265
1266             if (prevout.n >= txPrev.vout.size() || prevout.n >= txindex.vSpent.size())
1267                 return 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());
1268
1269             // If prev is coinbase, check that it's matured
1270             if (txPrev.IsCoinBase())
1271                 for (CBlockIndex* pindex = pindexBlock; pindex && pindexBlock->nHeight - pindex->nHeight < COINBASE_MATURITY; pindex = pindex->pprev)
1272                     if (pindex->nBlockPos == txindex.pos.nBlockPos && pindex->nFile == txindex.pos.nFile)
1273                         return error("ConnectInputs() : tried to spend coinbase at depth %d", pindexBlock->nHeight - pindex->nHeight);
1274
1275             // Verify signature
1276             if (!VerifySignature(txPrev, *this, i))
1277                 return error("ConnectInputs() : %s VerifySignature failed", GetHash().ToString().substr(0,10).c_str());
1278
1279             // Check for conflicts
1280             if (!txindex.vSpent[prevout.n].IsNull())
1281                 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());
1282
1283             // Check for negative or overflow input values
1284             nValueIn += txPrev.vout[prevout.n].nValue;
1285             if (!MoneyRange(txPrev.vout[prevout.n].nValue) || !MoneyRange(nValueIn))
1286                 return error("ConnectInputs() : txin values out of range");
1287
1288             // Mark outpoints as spent
1289             txindex.vSpent[prevout.n] = posThisTx;
1290
1291             // Write back
1292             if (fBlock)
1293             {
1294                 if (!txdb.UpdateTxIndex(prevout.hash, txindex))
1295                     return error("ConnectInputs() : UpdateTxIndex failed");
1296             }
1297             else if (fMiner)
1298             {
1299                 mapTestPool[prevout.hash] = txindex;
1300             }
1301         }
1302
1303         if (nValueIn < GetValueOut())
1304             return error("ConnectInputs() : %s value in < value out", GetHash().ToString().substr(0,10).c_str());
1305
1306         // Tally transaction fees
1307         int64 nTxFee = nValueIn - GetValueOut();
1308         if (nTxFee < 0)
1309             return error("ConnectInputs() : %s nTxFee < 0", GetHash().ToString().substr(0,10).c_str());
1310         if (nTxFee < nMinFee)
1311             return false;
1312         nFees += nTxFee;
1313         if (!MoneyRange(nFees))
1314             return error("ConnectInputs() : nFees out of range");
1315     }
1316
1317     if (fBlock)
1318     {
1319         // Add transaction to disk index
1320         if (!txdb.AddTxIndex(*this, posThisTx, pindexBlock->nHeight))
1321             return error("ConnectInputs() : AddTxPos failed");
1322     }
1323     else if (fMiner)
1324     {
1325         // Add transaction to test pool
1326         mapTestPool[GetHash()] = CTxIndex(CDiskTxPos(1,1,1), vout.size());
1327     }
1328
1329     return true;
1330 }
1331
1332
1333 bool CTransaction::ClientConnectInputs()
1334 {
1335     if (IsCoinBase())
1336         return false;
1337
1338     // Take over previous transactions' spent pointers
1339     CRITICAL_BLOCK(cs_mapTransactions)
1340     {
1341         int64 nValueIn = 0;
1342         for (int i = 0; i < vin.size(); i++)
1343         {
1344             // Get prev tx from single transactions in memory
1345             COutPoint prevout = vin[i].prevout;
1346             if (!mapTransactions.count(prevout.hash))
1347                 return false;
1348             CTransaction& txPrev = mapTransactions[prevout.hash];
1349
1350             if (prevout.n >= txPrev.vout.size())
1351                 return false;
1352
1353             // Verify signature
1354             if (!VerifySignature(txPrev, *this, i))
1355                 return error("ConnectInputs() : VerifySignature failed");
1356
1357             ///// this is redundant with the mapNextTx stuff, not sure which I want to get rid of
1358             ///// this has to go away now that posNext is gone
1359             // // Check for conflicts
1360             // if (!txPrev.vout[prevout.n].posNext.IsNull())
1361             //     return error("ConnectInputs() : prev tx already used");
1362             //
1363             // // Flag outpoints as used
1364             // txPrev.vout[prevout.n].posNext = posThisTx;
1365
1366             nValueIn += txPrev.vout[prevout.n].nValue;
1367
1368             if (!MoneyRange(txPrev.vout[prevout.n].nValue) || !MoneyRange(nValueIn))
1369                 return error("ClientConnectInputs() : txin values out of range");
1370         }
1371         if (GetValueOut() > nValueIn)
1372             return false;
1373     }
1374
1375     return true;
1376 }
1377
1378
1379
1380
1381 bool CBlock::DisconnectBlock(CTxDB& txdb, CBlockIndex* pindex)
1382 {
1383     // Disconnect in reverse order
1384     for (int i = vtx.size()-1; i >= 0; i--)
1385         if (!vtx[i].DisconnectInputs(txdb))
1386             return false;
1387
1388     // Update block index on disk without changing it in memory.
1389     // The memory index structure will be changed after the db commits.
1390     if (pindex->pprev)
1391     {
1392         CDiskBlockIndex blockindexPrev(pindex->pprev);
1393         blockindexPrev.hashNext = 0;
1394         if (!txdb.WriteBlockIndex(blockindexPrev))
1395             return error("DisconnectBlock() : WriteBlockIndex failed");
1396     }
1397
1398     return true;
1399 }
1400
1401 bool CBlock::ConnectBlock(CTxDB& txdb, CBlockIndex* pindex)
1402 {
1403     // Check it again in case a previous version let a bad block in
1404     if (!CheckBlock())
1405         return false;
1406
1407     //// issue here: it doesn't know the version
1408     unsigned int nTxPos = pindex->nBlockPos + ::GetSerializeSize(CBlock(), SER_DISK) - 1 + GetSizeOfCompactSize(vtx.size());
1409
1410     map<uint256, CTxIndex> mapUnused;
1411     int64 nFees = 0;
1412     foreach(CTransaction& tx, vtx)
1413     {
1414         CDiskTxPos posThisTx(pindex->nFile, pindex->nBlockPos, nTxPos);
1415         nTxPos += ::GetSerializeSize(tx, SER_DISK);
1416
1417         if (!tx.ConnectInputs(txdb, mapUnused, posThisTx, pindex, nFees, true, false))
1418             return false;
1419     }
1420
1421     if (vtx[0].GetValueOut() > GetBlockValue(pindex->nHeight, nFees))
1422         return false;
1423
1424     // Update block index on disk without changing it in memory.
1425     // The memory index structure will be changed after the db commits.
1426     if (pindex->pprev)
1427     {
1428         CDiskBlockIndex blockindexPrev(pindex->pprev);
1429         blockindexPrev.hashNext = pindex->GetBlockHash();
1430         if (!txdb.WriteBlockIndex(blockindexPrev))
1431             return error("ConnectBlock() : WriteBlockIndex failed");
1432     }
1433
1434     // Watch for transactions paying to me
1435     foreach(CTransaction& tx, vtx)
1436         AddToWalletIfMine(tx, this);
1437
1438     return true;
1439 }
1440
1441 bool Reorganize(CTxDB& txdb, CBlockIndex* pindexNew)
1442 {
1443     printf("REORGANIZE\n");
1444
1445     // Find the fork
1446     CBlockIndex* pfork = pindexBest;
1447     CBlockIndex* plonger = pindexNew;
1448     while (pfork != plonger)
1449     {
1450         while (plonger->nHeight > pfork->nHeight)
1451             if (!(plonger = plonger->pprev))
1452                 return error("Reorganize() : plonger->pprev is null");
1453         if (pfork == plonger)
1454             break;
1455         if (!(pfork = pfork->pprev))
1456             return error("Reorganize() : pfork->pprev is null");
1457     }
1458
1459     // List of what to disconnect
1460     vector<CBlockIndex*> vDisconnect;
1461     for (CBlockIndex* pindex = pindexBest; pindex != pfork; pindex = pindex->pprev)
1462         vDisconnect.push_back(pindex);
1463
1464     // List of what to connect
1465     vector<CBlockIndex*> vConnect;
1466     for (CBlockIndex* pindex = pindexNew; pindex != pfork; pindex = pindex->pprev)
1467         vConnect.push_back(pindex);
1468     reverse(vConnect.begin(), vConnect.end());
1469
1470     // Disconnect shorter branch
1471     vector<CTransaction> vResurrect;
1472     foreach(CBlockIndex* pindex, vDisconnect)
1473     {
1474         CBlock block;
1475         if (!block.ReadFromDisk(pindex))
1476             return error("Reorganize() : ReadFromDisk for disconnect failed");
1477         if (!block.DisconnectBlock(txdb, pindex))
1478             return error("Reorganize() : DisconnectBlock failed");
1479
1480         // Queue memory transactions to resurrect
1481         foreach(const CTransaction& tx, block.vtx)
1482             if (!tx.IsCoinBase())
1483                 vResurrect.push_back(tx);
1484     }
1485
1486     // Connect longer branch
1487     vector<CTransaction> vDelete;
1488     for (int i = 0; i < vConnect.size(); i++)
1489     {
1490         CBlockIndex* pindex = vConnect[i];
1491         CBlock block;
1492         if (!block.ReadFromDisk(pindex))
1493             return error("Reorganize() : ReadFromDisk for connect failed");
1494         if (!block.ConnectBlock(txdb, pindex))
1495         {
1496             // Invalid block
1497             txdb.TxnAbort();
1498             return error("Reorganize() : ConnectBlock failed");
1499         }
1500
1501         // Queue memory transactions to delete
1502         foreach(const CTransaction& tx, block.vtx)
1503             vDelete.push_back(tx);
1504     }
1505     if (!txdb.WriteHashBestChain(pindexNew->GetBlockHash()))
1506         return error("Reorganize() : WriteHashBestChain failed");
1507
1508     // Make sure it's successfully written to disk before changing memory structure
1509     if (!txdb.TxnCommit())
1510         return error("Reorganize() : TxnCommit failed");
1511
1512     // Disconnect shorter branch
1513     foreach(CBlockIndex* pindex, vDisconnect)
1514         if (pindex->pprev)
1515             pindex->pprev->pnext = NULL;
1516
1517     // Connect longer branch
1518     foreach(CBlockIndex* pindex, vConnect)
1519         if (pindex->pprev)
1520             pindex->pprev->pnext = pindex;
1521
1522     // Resurrect memory transactions that were in the disconnected branch
1523     foreach(CTransaction& tx, vResurrect)
1524         tx.AcceptToMemoryPool(txdb, false);
1525
1526     // Delete redundant memory transactions that are in the connected branch
1527     foreach(CTransaction& tx, vDelete)
1528         tx.RemoveFromMemoryPool();
1529
1530     return true;
1531 }
1532
1533
1534 bool CBlock::SetBestChain(CTxDB& txdb, CBlockIndex* pindexNew)
1535 {
1536     uint256 hash = GetHash();
1537
1538     txdb.TxnBegin();
1539     if (pindexGenesisBlock == NULL && hash == hashGenesisBlock)
1540     {
1541         txdb.WriteHashBestChain(hash);
1542         if (!txdb.TxnCommit())
1543             return error("SetBestChain() : TxnCommit failed");
1544         pindexGenesisBlock = pindexNew;
1545     }
1546     else if (hashPrevBlock == hashBestChain)
1547     {
1548         // Adding to current best branch
1549         if (!ConnectBlock(txdb, pindexNew) || !txdb.WriteHashBestChain(hash))
1550         {
1551             txdb.TxnAbort();
1552             InvalidChainFound(pindexNew);
1553             return error("SetBestChain() : ConnectBlock failed");
1554         }
1555         if (!txdb.TxnCommit())
1556             return error("SetBestChain() : TxnCommit failed");
1557
1558         // Add to current best branch
1559         pindexNew->pprev->pnext = pindexNew;
1560
1561         // Delete redundant memory transactions
1562         foreach(CTransaction& tx, vtx)
1563             tx.RemoveFromMemoryPool();
1564     }
1565     else
1566     {
1567         // New best branch
1568         if (!Reorganize(txdb, pindexNew))
1569         {
1570             txdb.TxnAbort();
1571             InvalidChainFound(pindexNew);
1572             return error("SetBestChain() : Reorganize failed");
1573         }
1574     }
1575
1576     // New best block
1577     hashBestChain = hash;
1578     pindexBest = pindexNew;
1579     nBestHeight = pindexBest->nHeight;
1580     bnBestChainWork = pindexNew->bnChainWork;
1581     nTimeBestReceived = GetTime();
1582     nTransactionsUpdated++;
1583     printf("SetBestChain: new best=%s  height=%d  work=%s\n", hashBestChain.ToString().substr(0,20).c_str(), nBestHeight, bnBestChainWork.ToString().c_str());
1584
1585     return true;
1586 }
1587
1588
1589 bool CBlock::AddToBlockIndex(unsigned int nFile, unsigned int nBlockPos)
1590 {
1591     // Check for duplicate
1592     uint256 hash = GetHash();
1593     if (mapBlockIndex.count(hash))
1594         return error("AddToBlockIndex() : %s already exists", hash.ToString().substr(0,20).c_str());
1595
1596     // Construct new block index object
1597     CBlockIndex* pindexNew = new CBlockIndex(nFile, nBlockPos, *this);
1598     if (!pindexNew)
1599         return error("AddToBlockIndex() : new CBlockIndex failed");
1600     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.insert(make_pair(hash, pindexNew)).first;
1601     pindexNew->phashBlock = &((*mi).first);
1602     map<uint256, CBlockIndex*>::iterator miPrev = mapBlockIndex.find(hashPrevBlock);
1603     if (miPrev != mapBlockIndex.end())
1604     {
1605         pindexNew->pprev = (*miPrev).second;
1606         pindexNew->nHeight = pindexNew->pprev->nHeight + 1;
1607     }
1608     pindexNew->bnChainWork = (pindexNew->pprev ? pindexNew->pprev->bnChainWork : 0) + pindexNew->GetBlockWork();
1609
1610     CTxDB txdb;
1611     txdb.TxnBegin();
1612     txdb.WriteBlockIndex(CDiskBlockIndex(pindexNew));
1613     if (!txdb.TxnCommit())
1614         return false;
1615
1616     // New best
1617     if (pindexNew->bnChainWork > bnBestChainWork)
1618         if (!SetBestChain(txdb, pindexNew))
1619             return false;
1620
1621     txdb.Close();
1622
1623     if (pindexNew == pindexBest)
1624     {
1625         // Notify UI to display prev block's coinbase if it was ours
1626         static uint256 hashPrevBestCoinBase;
1627         CRITICAL_BLOCK(cs_mapWallet)
1628             vWalletUpdated.push_back(hashPrevBestCoinBase);
1629         hashPrevBestCoinBase = vtx[0].GetHash();
1630     }
1631
1632     MainFrameRepaint();
1633     return true;
1634 }
1635
1636
1637
1638
1639 bool CBlock::CheckBlock() const
1640 {
1641     // These are checks that are independent of context
1642     // that can be verified before saving an orphan block.
1643
1644     // Size limits
1645     if (vtx.empty() || vtx.size() > MAX_BLOCK_SIZE || ::GetSerializeSize(*this, SER_NETWORK) > MAX_BLOCK_SIZE)
1646         return error("CheckBlock() : size limits failed");
1647
1648     // Check proof of work matches claimed amount
1649     if (!CheckProofOfWork(GetHash(), nBits))
1650         return error("CheckBlock() : proof of work failed");
1651
1652     // Check timestamp
1653     if (GetBlockTime() > GetAdjustedTime() + 2 * 60 * 60)
1654         return error("CheckBlock() : block timestamp too far in the future");
1655
1656     // First transaction must be coinbase, the rest must not be
1657     if (vtx.empty() || !vtx[0].IsCoinBase())
1658         return error("CheckBlock() : first tx is not coinbase");
1659     for (int i = 1; i < vtx.size(); i++)
1660         if (vtx[i].IsCoinBase())
1661             return error("CheckBlock() : more than one coinbase");
1662
1663     // Check transactions
1664     foreach(const CTransaction& tx, vtx)
1665         if (!tx.CheckTransaction())
1666             return error("CheckBlock() : CheckTransaction failed");
1667
1668     // Check that it's not full of nonstandard transactions
1669     if (GetSigOpCount() > MAX_BLOCK_SIGOPS)
1670         return error("CheckBlock() : too many nonstandard transactions");
1671
1672     // Check merkleroot
1673     if (hashMerkleRoot != BuildMerkleTree())
1674         return error("CheckBlock() : hashMerkleRoot mismatch");
1675
1676     return true;
1677 }
1678
1679 bool CBlock::AcceptBlock()
1680 {
1681     // Check for duplicate
1682     uint256 hash = GetHash();
1683     if (mapBlockIndex.count(hash))
1684         return error("AcceptBlock() : block already in mapBlockIndex");
1685
1686     // Get prev block index
1687     map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashPrevBlock);
1688     if (mi == mapBlockIndex.end())
1689         return error("AcceptBlock() : prev block not found");
1690     CBlockIndex* pindexPrev = (*mi).second;
1691     int nHeight = pindexPrev->nHeight+1;
1692
1693     // Check proof of work
1694     if (nBits != GetNextWorkRequired(pindexPrev))
1695         return error("AcceptBlock() : incorrect proof of work");
1696
1697     // Check timestamp against prev
1698     if (GetBlockTime() <= pindexPrev->GetMedianTimePast())
1699         return error("AcceptBlock() : block's timestamp is too early");
1700
1701     // Check that all transactions are finalized
1702     foreach(const CTransaction& tx, vtx)
1703         if (!tx.IsFinal(nHeight, GetBlockTime()))
1704             return error("AcceptBlock() : contains a non-final transaction");
1705
1706     // Check that the block chain matches the known block chain up to a checkpoint
1707     if (!fTestNet)
1708         if ((nHeight ==  11111 && hash != uint256("0x0000000069e244f73d78e8fd29ba2fd2ed618bd6fa2ee92559f542fdb26e7c1d")) ||
1709             (nHeight ==  33333 && hash != uint256("0x000000002dd5588a74784eaa7ab0507a18ad16a236e7b1ce69f00d7ddfb5d0a6")) ||
1710             (nHeight ==  68555 && hash != uint256("0x00000000001e1b4903550a0b96e9a9405c8a95f387162e4944e8d9fbe501cd6a")) ||
1711             (nHeight ==  70567 && hash != uint256("0x00000000006a49b14bcf27462068f1264c961f11fa2e0eddd2be0791e1d4124a")) ||
1712             (nHeight ==  74000 && hash != uint256("0x0000000000573993a3c9e41ce34471c079dcf5f52a0e824a81e7f953b8661a20")) ||
1713             (nHeight == 105000 && hash != uint256("0x00000000000291ce28027faea320c8d2b054b2e0fe44a773f3eefb151d6bdc97")))
1714             return error("AcceptBlock() : rejected by checkpoint lockin at %d", nHeight);
1715
1716     // Write block to history file
1717     if (!CheckDiskSpace(::GetSerializeSize(*this, SER_DISK)))
1718         return error("AcceptBlock() : out of disk space");
1719     unsigned int nFile = -1;
1720     unsigned int nBlockPos = 0;
1721     if (!WriteToDisk(nFile, nBlockPos))
1722         return error("AcceptBlock() : WriteToDisk failed");
1723     if (!AddToBlockIndex(nFile, nBlockPos))
1724         return error("AcceptBlock() : AddToBlockIndex failed");
1725
1726     // Relay inventory, but don't relay old inventory during initial block download
1727     if (hashBestChain == hash)
1728         CRITICAL_BLOCK(cs_vNodes)
1729             foreach(CNode* pnode, vNodes)
1730                 if (nBestHeight > (pnode->nStartingHeight != -1 ? pnode->nStartingHeight - 2000 : 105000))
1731                     pnode->PushInventory(CInv(MSG_BLOCK, hash));
1732
1733     return true;
1734 }
1735
1736 bool ProcessBlock(CNode* pfrom, CBlock* pblock)
1737 {
1738     // Check for duplicate
1739     uint256 hash = pblock->GetHash();
1740     if (mapBlockIndex.count(hash))
1741         return error("ProcessBlock() : already have block %d %s", mapBlockIndex[hash]->nHeight, hash.ToString().substr(0,20).c_str());
1742     if (mapOrphanBlocks.count(hash))
1743         return error("ProcessBlock() : already have block (orphan) %s", hash.ToString().substr(0,20).c_str());
1744
1745     // Preliminary checks
1746     if (!pblock->CheckBlock())
1747         return error("ProcessBlock() : CheckBlock FAILED");
1748
1749     // If don't already have its previous block, shunt it off to holding area until we get it
1750     if (!mapBlockIndex.count(pblock->hashPrevBlock))
1751     {
1752         printf("ProcessBlock: ORPHAN BLOCK, prev=%s\n", pblock->hashPrevBlock.ToString().substr(0,20).c_str());
1753         CBlock* pblock2 = new CBlock(*pblock);
1754         mapOrphanBlocks.insert(make_pair(hash, pblock2));
1755         mapOrphanBlocksByPrev.insert(make_pair(pblock2->hashPrevBlock, pblock2));
1756
1757         // Ask this guy to fill in what we're missing
1758         if (pfrom)
1759             pfrom->PushGetBlocks(pindexBest, GetOrphanRoot(pblock2));
1760         return true;
1761     }
1762
1763     // Store to disk
1764     if (!pblock->AcceptBlock())
1765         return error("ProcessBlock() : AcceptBlock FAILED");
1766
1767     // Recursively process any orphan blocks that depended on this one
1768     vector<uint256> vWorkQueue;
1769     vWorkQueue.push_back(hash);
1770     for (int i = 0; i < vWorkQueue.size(); i++)
1771     {
1772         uint256 hashPrev = vWorkQueue[i];
1773         for (multimap<uint256, CBlock*>::iterator mi = mapOrphanBlocksByPrev.lower_bound(hashPrev);
1774              mi != mapOrphanBlocksByPrev.upper_bound(hashPrev);
1775              ++mi)
1776         {
1777             CBlock* pblockOrphan = (*mi).second;
1778             if (pblockOrphan->AcceptBlock())
1779                 vWorkQueue.push_back(pblockOrphan->GetHash());
1780             mapOrphanBlocks.erase(pblockOrphan->GetHash());
1781             delete pblockOrphan;
1782         }
1783         mapOrphanBlocksByPrev.erase(hashPrev);
1784     }
1785
1786     printf("ProcessBlock: ACCEPTED\n");
1787     return true;
1788 }
1789
1790
1791
1792
1793
1794
1795
1796
1797 template<typename Stream>
1798 bool ScanMessageStart(Stream& s)
1799 {
1800     // Scan ahead to the next pchMessageStart, which should normally be immediately
1801     // at the file pointer.  Leaves file pointer at end of pchMessageStart.
1802     s.clear(0);
1803     short prevmask = s.exceptions(0);
1804     const char* p = BEGIN(pchMessageStart);
1805     try
1806     {
1807         loop
1808         {
1809             char c;
1810             s.read(&c, 1);
1811             if (s.fail())
1812             {
1813                 s.clear(0);
1814                 s.exceptions(prevmask);
1815                 return false;
1816             }
1817             if (*p != c)
1818                 p = BEGIN(pchMessageStart);
1819             if (*p == c)
1820             {
1821                 if (++p == END(pchMessageStart))
1822                 {
1823                     s.clear(0);
1824                     s.exceptions(prevmask);
1825                     return true;
1826                 }
1827             }
1828         }
1829     }
1830     catch (...)
1831     {
1832         s.clear(0);
1833         s.exceptions(prevmask);
1834         return false;
1835     }
1836 }
1837
1838 bool CheckDiskSpace(uint64 nAdditionalBytes)
1839 {
1840     uint64 nFreeBytesAvailable = filesystem::space(GetDataDir()).available;
1841
1842     // Check for 15MB because database could create another 10MB log file at any time
1843     if (nFreeBytesAvailable < (uint64)15000000 + nAdditionalBytes)
1844     {
1845         fShutdown = true;
1846         string strMessage = _("Warning: Disk space is low  ");
1847         strMiscWarning = strMessage;
1848         printf("*** %s\n", strMessage.c_str());
1849         ThreadSafeMessageBox(strMessage, "Bitcoin", wxOK | wxICON_EXCLAMATION);
1850         CreateThread(Shutdown, NULL);
1851         return false;
1852     }
1853     return true;
1854 }
1855
1856 FILE* OpenBlockFile(unsigned int nFile, unsigned int nBlockPos, const char* pszMode)
1857 {
1858     if (nFile == -1)
1859         return NULL;
1860     FILE* file = fopen(strprintf("%s/blk%04d.dat", GetDataDir().c_str(), nFile).c_str(), pszMode);
1861     if (!file)
1862         return NULL;
1863     if (nBlockPos != 0 && !strchr(pszMode, 'a') && !strchr(pszMode, 'w'))
1864     {
1865         if (fseek(file, nBlockPos, SEEK_SET) != 0)
1866         {
1867             fclose(file);
1868             return NULL;
1869         }
1870     }
1871     return file;
1872 }
1873
1874 static unsigned int nCurrentBlockFile = 1;
1875
1876 FILE* AppendBlockFile(unsigned int& nFileRet)
1877 {
1878     nFileRet = 0;
1879     loop
1880     {
1881         FILE* file = OpenBlockFile(nCurrentBlockFile, 0, "ab");
1882         if (!file)
1883             return NULL;
1884         if (fseek(file, 0, SEEK_END) != 0)
1885             return NULL;
1886         // FAT32 filesize max 4GB, fseek and ftell max 2GB, so we must stay under 2GB
1887         if (ftell(file) < 0x7F000000 - MAX_SIZE)
1888         {
1889             nFileRet = nCurrentBlockFile;
1890             return file;
1891         }
1892         fclose(file);
1893         nCurrentBlockFile++;
1894     }
1895 }
1896
1897 bool LoadBlockIndex(bool fAllowNew)
1898 {
1899     if (fTestNet)
1900     {
1901         hashGenesisBlock = uint256("0x00000007199508e34a9ff81e6ec0c477a4cccff2a4767a8eee39c11db367b008");
1902         bnProofOfWorkLimit = CBigNum(~uint256(0) >> 28);
1903         pchMessageStart[0] = 0xfa;
1904         pchMessageStart[1] = 0xbf;
1905         pchMessageStart[2] = 0xb5;
1906         pchMessageStart[3] = 0xda;
1907     }
1908
1909     //
1910     // Load block index
1911     //
1912     CTxDB txdb("cr");
1913     if (!txdb.LoadBlockIndex())
1914         return false;
1915     txdb.Close();
1916
1917     //
1918     // Init with genesis block
1919     //
1920     if (mapBlockIndex.empty())
1921     {
1922         if (!fAllowNew)
1923             return false;
1924
1925         // Genesis Block:
1926         // CBlock(hash=000000000019d6, ver=1, hashPrevBlock=00000000000000, hashMerkleRoot=4a5e1e, nTime=1231006505, nBits=1d00ffff, nNonce=2083236893, vtx=1)
1927         //   CTransaction(hash=4a5e1e, ver=1, vin.size=1, vout.size=1, nLockTime=0)
1928         //     CTxIn(COutPoint(000000, -1), coinbase 04ffff001d0104455468652054696d65732030332f4a616e2f32303039204368616e63656c6c6f72206f6e206272696e6b206f66207365636f6e64206261696c6f757420666f722062616e6b73)
1929         //     CTxOut(nValue=50.00000000, scriptPubKey=0x5F1DF16B2B704C8A578D0B)
1930         //   vMerkleTree: 4a5e1e
1931
1932         // Genesis block
1933         const char* pszTimestamp = "The Times 03/Jan/2009 Chancellor on brink of second bailout for banks";
1934         CTransaction txNew;
1935         txNew.vin.resize(1);
1936         txNew.vout.resize(1);
1937         txNew.vin[0].scriptSig = CScript() << 486604799 << CBigNum(4) << vector<unsigned char>((const unsigned char*)pszTimestamp, (const unsigned char*)pszTimestamp + strlen(pszTimestamp));
1938         txNew.vout[0].nValue = 50 * COIN;
1939         txNew.vout[0].scriptPubKey = CScript() << ParseHex("04678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5f") << OP_CHECKSIG;
1940         CBlock block;
1941         block.vtx.push_back(txNew);
1942         block.hashPrevBlock = 0;
1943         block.hashMerkleRoot = block.BuildMerkleTree();
1944         block.nVersion = 1;
1945         block.nTime    = 1231006505;
1946         block.nBits    = 0x1d00ffff;
1947         block.nNonce   = 2083236893;
1948
1949         if (fTestNet)
1950         {
1951             block.nTime    = 1296688602;
1952             block.nBits    = 0x1d07fff8;
1953             block.nNonce   = 384568319;
1954         }
1955
1956         //// debug print
1957         printf("%s\n", block.GetHash().ToString().c_str());
1958         printf("%s\n", hashGenesisBlock.ToString().c_str());
1959         printf("%s\n", block.hashMerkleRoot.ToString().c_str());
1960         assert(block.hashMerkleRoot == uint256("0x4a5e1e4baab89f3a32518a88c31bc87f618f76673e2cc77ab2127b7afdeda33b"));
1961         block.print();
1962         assert(block.GetHash() == hashGenesisBlock);
1963
1964         // Start new block file
1965         unsigned int nFile;
1966         unsigned int nBlockPos;
1967         if (!block.WriteToDisk(nFile, nBlockPos))
1968             return error("LoadBlockIndex() : writing genesis block to disk failed");
1969         if (!block.AddToBlockIndex(nFile, nBlockPos))
1970             return error("LoadBlockIndex() : genesis block not accepted");
1971     }
1972
1973     return true;
1974 }
1975
1976
1977
1978 void PrintBlockTree()
1979 {
1980     // precompute tree structure
1981     map<CBlockIndex*, vector<CBlockIndex*> > mapNext;
1982     for (map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.begin(); mi != mapBlockIndex.end(); ++mi)
1983     {
1984         CBlockIndex* pindex = (*mi).second;
1985         mapNext[pindex->pprev].push_back(pindex);
1986         // test
1987         //while (rand() % 3 == 0)
1988         //    mapNext[pindex->pprev].push_back(pindex);
1989     }
1990
1991     vector<pair<int, CBlockIndex*> > vStack;
1992     vStack.push_back(make_pair(0, pindexGenesisBlock));
1993
1994     int nPrevCol = 0;
1995     while (!vStack.empty())
1996     {
1997         int nCol = vStack.back().first;
1998         CBlockIndex* pindex = vStack.back().second;
1999         vStack.pop_back();
2000
2001         // print split or gap
2002         if (nCol > nPrevCol)
2003         {
2004             for (int i = 0; i < nCol-1; i++)
2005                 printf("| ");
2006             printf("|\\\n");
2007         }
2008         else if (nCol < nPrevCol)
2009         {
2010             for (int i = 0; i < nCol; i++)
2011                 printf("| ");
2012             printf("|\n");
2013         }
2014         nPrevCol = nCol;
2015
2016         // print columns
2017         for (int i = 0; i < nCol; i++)
2018             printf("| ");
2019
2020         // print item
2021         CBlock block;
2022         block.ReadFromDisk(pindex);
2023         printf("%d (%u,%u) %s  %s  tx %d",
2024             pindex->nHeight,
2025             pindex->nFile,
2026             pindex->nBlockPos,
2027             block.GetHash().ToString().substr(0,20).c_str(),
2028             DateTimeStrFormat("%x %H:%M:%S", block.GetBlockTime()).c_str(),
2029             block.vtx.size());
2030
2031         CRITICAL_BLOCK(cs_mapWallet)
2032         {
2033             if (mapWallet.count(block.vtx[0].GetHash()))
2034             {
2035                 CWalletTx& wtx = mapWallet[block.vtx[0].GetHash()];
2036                 printf("    mine:  %d  %d  %d", wtx.GetDepthInMainChain(), wtx.GetBlocksToMaturity(), wtx.GetCredit());
2037             }
2038         }
2039         printf("\n");
2040
2041
2042         // put the main timechain first
2043         vector<CBlockIndex*>& vNext = mapNext[pindex];
2044         for (int i = 0; i < vNext.size(); i++)
2045         {
2046             if (vNext[i]->pnext)
2047             {
2048                 swap(vNext[0], vNext[i]);
2049                 break;
2050             }
2051         }
2052
2053         // iterate children
2054         for (int i = 0; i < vNext.size(); i++)
2055             vStack.push_back(make_pair(nCol+i, vNext[i]));
2056     }
2057 }
2058
2059
2060
2061
2062
2063
2064
2065
2066
2067
2068 //////////////////////////////////////////////////////////////////////////////
2069 //
2070 // CAlert
2071 //
2072
2073 map<uint256, CAlert> mapAlerts;
2074 CCriticalSection cs_mapAlerts;
2075
2076 string GetWarnings(string strFor)
2077 {
2078     int nPriority = 0;
2079     string strStatusBar;
2080     string strRPC;
2081     if (GetBoolArg("-testsafemode"))
2082         strRPC = "test";
2083
2084     // Misc warnings like out of disk space and clock is wrong
2085     if (strMiscWarning != "")
2086     {
2087         nPriority = 1000;
2088         strStatusBar = strMiscWarning;
2089     }
2090
2091     // Longer invalid proof-of-work chain
2092     if (pindexBest && bnBestInvalidWork > bnBestChainWork + pindexBest->GetBlockWork() * 6)
2093     {
2094         nPriority = 2000;
2095         strStatusBar = strRPC = "WARNING: Displayed transactions may not be correct!  You may need to upgrade, or other nodes may need to upgrade.";
2096     }
2097
2098     // Alerts
2099     CRITICAL_BLOCK(cs_mapAlerts)
2100     {
2101         foreach(PAIRTYPE(const uint256, CAlert)& item, mapAlerts)
2102         {
2103             const CAlert& alert = item.second;
2104             if (alert.AppliesToMe() && alert.nPriority > nPriority)
2105             {
2106                 nPriority = alert.nPriority;
2107                 strStatusBar = alert.strStatusBar;
2108             }
2109         }
2110     }
2111
2112     if (strFor == "statusbar")
2113         return strStatusBar;
2114     else if (strFor == "rpc")
2115         return strRPC;
2116     assert(("GetWarnings() : invalid parameter", false));
2117     return "error";
2118 }
2119
2120 bool CAlert::ProcessAlert()
2121 {
2122     if (!CheckSignature())
2123         return false;
2124     if (!IsInEffect())
2125         return false;
2126
2127     CRITICAL_BLOCK(cs_mapAlerts)
2128     {
2129         // Cancel previous alerts
2130         for (map<uint256, CAlert>::iterator mi = mapAlerts.begin(); mi != mapAlerts.end();)
2131         {
2132             const CAlert& alert = (*mi).second;
2133             if (Cancels(alert))
2134             {
2135                 printf("cancelling alert %d\n", alert.nID);
2136                 mapAlerts.erase(mi++);
2137             }
2138             else if (!alert.IsInEffect())
2139             {
2140                 printf("expiring alert %d\n", alert.nID);
2141                 mapAlerts.erase(mi++);
2142             }
2143             else
2144                 mi++;
2145         }
2146
2147         // Check if this alert has been cancelled
2148         foreach(PAIRTYPE(const uint256, CAlert)& item, mapAlerts)
2149         {
2150             const CAlert& alert = item.second;
2151             if (alert.Cancels(*this))
2152             {
2153                 printf("alert already cancelled by %d\n", alert.nID);
2154                 return false;
2155             }
2156         }
2157
2158         // Add to mapAlerts
2159         mapAlerts.insert(make_pair(GetHash(), *this));
2160     }
2161
2162     printf("accepted alert %d, AppliesToMe()=%d\n", nID, AppliesToMe());
2163     MainFrameRepaint();
2164     return true;
2165 }
2166
2167
2168
2169
2170
2171
2172
2173
2174 //////////////////////////////////////////////////////////////////////////////
2175 //
2176 // Messages
2177 //
2178
2179
2180 bool AlreadyHave(CTxDB& txdb, const CInv& inv)
2181 {
2182     switch (inv.type)
2183     {
2184     case MSG_TX:    return mapTransactions.count(inv.hash) || mapOrphanTransactions.count(inv.hash) || txdb.ContainsTx(inv.hash);
2185     case MSG_BLOCK: return mapBlockIndex.count(inv.hash) || mapOrphanBlocks.count(inv.hash);
2186     }
2187     // Don't know what it is, just say we already got one
2188     return true;
2189 }
2190
2191
2192
2193
2194 // The message start string is designed to be unlikely to occur in normal data.
2195 // The characters are rarely used upper ascii, not valid as UTF-8, and produce
2196 // a large 4-byte int at any alignment.
2197 char pchMessageStart[4] = { 0xf9, 0xbe, 0xb4, 0xd9 };
2198
2199
2200 bool ProcessMessages(CNode* pfrom)
2201 {
2202     CDataStream& vRecv = pfrom->vRecv;
2203     if (vRecv.empty())
2204         return true;
2205     //if (fDebug)
2206     //    printf("ProcessMessages(%u bytes)\n", vRecv.size());
2207
2208     //
2209     // Message format
2210     //  (4) message start
2211     //  (12) command
2212     //  (4) size
2213     //  (4) checksum
2214     //  (x) data
2215     //
2216
2217     loop
2218     {
2219         // Scan for message start
2220         CDataStream::iterator pstart = search(vRecv.begin(), vRecv.end(), BEGIN(pchMessageStart), END(pchMessageStart));
2221         int nHeaderSize = vRecv.GetSerializeSize(CMessageHeader());
2222         if (vRecv.end() - pstart < nHeaderSize)
2223         {
2224             if (vRecv.size() > nHeaderSize)
2225             {
2226                 printf("\n\nPROCESSMESSAGE MESSAGESTART NOT FOUND\n\n");
2227                 vRecv.erase(vRecv.begin(), vRecv.end() - nHeaderSize);
2228             }
2229             break;
2230         }
2231         if (pstart - vRecv.begin() > 0)
2232             printf("\n\nPROCESSMESSAGE SKIPPED %d BYTES\n\n", pstart - vRecv.begin());
2233         vRecv.erase(vRecv.begin(), pstart);
2234
2235         // Read header
2236         vector<char> vHeaderSave(vRecv.begin(), vRecv.begin() + nHeaderSize);
2237         CMessageHeader hdr;
2238         vRecv >> hdr;
2239         if (!hdr.IsValid())
2240         {
2241             printf("\n\nPROCESSMESSAGE: ERRORS IN HEADER %s\n\n\n", hdr.GetCommand().c_str());
2242             continue;
2243         }
2244         string strCommand = hdr.GetCommand();
2245
2246         // Message size
2247         unsigned int nMessageSize = hdr.nMessageSize;
2248         if (nMessageSize > MAX_SIZE)
2249         {
2250             printf("ProcessMessage(%s, %u bytes) : nMessageSize > MAX_SIZE\n", strCommand.c_str(), nMessageSize);
2251             continue;
2252         }
2253         if (nMessageSize > vRecv.size())
2254         {
2255             // Rewind and wait for rest of message
2256             vRecv.insert(vRecv.begin(), vHeaderSave.begin(), vHeaderSave.end());
2257             break;
2258         }
2259
2260         // Checksum
2261         if (vRecv.GetVersion() >= 209)
2262         {
2263             uint256 hash = Hash(vRecv.begin(), vRecv.begin() + nMessageSize);
2264             unsigned int nChecksum = 0;
2265             memcpy(&nChecksum, &hash, sizeof(nChecksum));
2266             if (nChecksum != hdr.nChecksum)
2267             {
2268                 printf("ProcessMessage(%s, %u bytes) : CHECKSUM ERROR nChecksum=%08x hdr.nChecksum=%08x\n",
2269                        strCommand.c_str(), nMessageSize, nChecksum, hdr.nChecksum);
2270                 continue;
2271             }
2272         }
2273
2274         // Copy message to its own buffer
2275         CDataStream vMsg(vRecv.begin(), vRecv.begin() + nMessageSize, vRecv.nType, vRecv.nVersion);
2276         vRecv.ignore(nMessageSize);
2277
2278         // Process message
2279         bool fRet = false;
2280         try
2281         {
2282             CRITICAL_BLOCK(cs_main)
2283                 fRet = ProcessMessage(pfrom, strCommand, vMsg);
2284             if (fShutdown)
2285                 return true;
2286         }
2287         catch (std::ios_base::failure& e)
2288         {
2289             if (strstr(e.what(), "end of data"))
2290             {
2291                 // Allow exceptions from underlength message on vRecv
2292                 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());
2293             }
2294             else if (strstr(e.what(), "size too large"))
2295             {
2296                 // Allow exceptions from overlong size
2297                 printf("ProcessMessage(%s, %u bytes) : Exception '%s' caught\n", strCommand.c_str(), nMessageSize, e.what());
2298             }
2299             else
2300             {
2301                 PrintExceptionContinue(&e, "ProcessMessage()");
2302             }
2303         }
2304         catch (std::exception& e) {
2305             PrintExceptionContinue(&e, "ProcessMessage()");
2306         } catch (...) {
2307             PrintExceptionContinue(NULL, "ProcessMessage()");
2308         }
2309
2310         if (!fRet)
2311             printf("ProcessMessage(%s, %u bytes) FAILED\n", strCommand.c_str(), nMessageSize);
2312     }
2313
2314     vRecv.Compact();
2315     return true;
2316 }
2317
2318
2319
2320
2321 bool ProcessMessage(CNode* pfrom, string strCommand, CDataStream& vRecv)
2322 {
2323     static map<unsigned int, vector<unsigned char> > mapReuseKey;
2324     RandAddSeedPerfmon();
2325     if (fDebug)
2326         printf("%s ", DateTimeStrFormat("%x %H:%M:%S", GetTime()).c_str());
2327     printf("received: %s (%d bytes)\n", strCommand.c_str(), vRecv.size());
2328     if (mapArgs.count("-dropmessagestest") && GetRand(atoi(mapArgs["-dropmessagestest"])) == 0)
2329     {
2330         printf("dropmessagestest DROPPING RECV MESSAGE\n");
2331         return true;
2332     }
2333
2334
2335
2336
2337
2338     if (strCommand == "version")
2339     {
2340         // Each connection can only send one version message
2341         if (pfrom->nVersion != 0)
2342             return false;
2343
2344         int64 nTime;
2345         CAddress addrMe;
2346         CAddress addrFrom;
2347         uint64 nNonce = 1;
2348         vRecv >> pfrom->nVersion >> pfrom->nServices >> nTime >> addrMe;
2349         if (pfrom->nVersion == 10300)
2350             pfrom->nVersion = 300;
2351         if (pfrom->nVersion >= 106 && !vRecv.empty())
2352             vRecv >> addrFrom >> nNonce;
2353         if (pfrom->nVersion >= 106 && !vRecv.empty())
2354             vRecv >> pfrom->strSubVer;
2355         if (pfrom->nVersion >= 209 && !vRecv.empty())
2356             vRecv >> pfrom->nStartingHeight;
2357
2358         if (pfrom->nVersion == 0)
2359             return false;
2360
2361         // Disconnect if we connected to ourself
2362         if (nNonce == nLocalHostNonce && nNonce > 1)
2363         {
2364             printf("connected to self at %s, disconnecting\n", pfrom->addr.ToString().c_str());
2365             pfrom->fDisconnect = true;
2366             return true;
2367         }
2368
2369         pfrom->fClient = !(pfrom->nServices & NODE_NETWORK);
2370
2371         AddTimeData(pfrom->addr.ip, nTime);
2372
2373         // Change version
2374         if (pfrom->nVersion >= 209)
2375             pfrom->PushMessage("verack");
2376         pfrom->vSend.SetVersion(min(pfrom->nVersion, VERSION));
2377         if (pfrom->nVersion < 209)
2378             pfrom->vRecv.SetVersion(min(pfrom->nVersion, VERSION));
2379
2380         if (!pfrom->fInbound)
2381         {
2382             // Advertise our address
2383             if (addrLocalHost.IsRoutable() && !fUseProxy)
2384             {
2385                 CAddress addr(addrLocalHost);
2386                 addr.nTime = GetAdjustedTime();
2387                 pfrom->PushAddress(addr);
2388             }
2389
2390             // Get recent addresses
2391             if (pfrom->nVersion >= 31402 || mapAddresses.size() < 1000)
2392             {
2393                 pfrom->PushMessage("getaddr");
2394                 pfrom->fGetAddr = true;
2395             }
2396         }
2397
2398         // Ask the first connected node for block updates
2399         static int nAskedForBlocks;
2400         if (!pfrom->fClient && (nAskedForBlocks < 1 || vNodes.size() <= 1))
2401         {
2402             nAskedForBlocks++;
2403             pfrom->PushGetBlocks(pindexBest, uint256(0));
2404         }
2405
2406         // Relay alerts
2407         CRITICAL_BLOCK(cs_mapAlerts)
2408             foreach(PAIRTYPE(const uint256, CAlert)& item, mapAlerts)
2409                 item.second.RelayTo(pfrom);
2410
2411         pfrom->fSuccessfullyConnected = true;
2412
2413         printf("version message: version %d, blocks=%d\n", pfrom->nVersion, pfrom->nStartingHeight);
2414     }
2415
2416
2417     else if (pfrom->nVersion == 0)
2418     {
2419         // Must have a version message before anything else
2420         return false;
2421     }
2422
2423
2424     else if (strCommand == "verack")
2425     {
2426         pfrom->vRecv.SetVersion(min(pfrom->nVersion, VERSION));
2427     }
2428
2429
2430     else if (strCommand == "addr")
2431     {
2432         vector<CAddress> vAddr;
2433         vRecv >> vAddr;
2434
2435         // Don't want addr from older versions unless seeding
2436         if (pfrom->nVersion < 209)
2437             return true;
2438         if (pfrom->nVersion < 31402 && mapAddresses.size() > 1000)
2439             return true;
2440         if (vAddr.size() > 1000)
2441             return error("message addr size() = %d", vAddr.size());
2442
2443         // Store the new addresses
2444         int64 nNow = GetAdjustedTime();
2445         int64 nSince = nNow - 10 * 60;
2446         foreach(CAddress& addr, vAddr)
2447         {
2448             if (fShutdown)
2449                 return true;
2450             // ignore IPv6 for now, since it isn't implemented anyway
2451             if (!addr.IsIPv4())
2452                 continue;
2453             if (addr.nTime <= 100000000 || addr.nTime > nNow + 10 * 60)
2454                 addr.nTime = nNow - 5 * 24 * 60 * 60;
2455             AddAddress(addr, 2 * 60 * 60);
2456             pfrom->AddAddressKnown(addr);
2457             if (addr.nTime > nSince && !pfrom->fGetAddr && vAddr.size() <= 10 && addr.IsRoutable())
2458             {
2459                 // Relay to a limited number of other nodes
2460                 CRITICAL_BLOCK(cs_vNodes)
2461                 {
2462                     // Use deterministic randomness to send to the same nodes for 24 hours
2463                     // at a time so the setAddrKnowns of the chosen nodes prevent repeats
2464                     static uint256 hashSalt;
2465                     if (hashSalt == 0)
2466                         RAND_bytes((unsigned char*)&hashSalt, sizeof(hashSalt));
2467                     uint256 hashRand = hashSalt ^ (((int64)addr.ip)<<32) ^ ((GetTime()+addr.ip)/(24*60*60));
2468                     hashRand = Hash(BEGIN(hashRand), END(hashRand));
2469                     multimap<uint256, CNode*> mapMix;
2470                     foreach(CNode* pnode, vNodes)
2471                     {
2472                         if (pnode->nVersion < 31402)
2473                             continue;
2474                         unsigned int nPointer;
2475                         memcpy(&nPointer, &pnode, sizeof(nPointer));
2476                         uint256 hashKey = hashRand ^ nPointer;
2477                         hashKey = Hash(BEGIN(hashKey), END(hashKey));
2478                         mapMix.insert(make_pair(hashKey, pnode));
2479                     }
2480                     int nRelayNodes = 2;
2481                     for (multimap<uint256, CNode*>::iterator mi = mapMix.begin(); mi != mapMix.end() && nRelayNodes-- > 0; ++mi)
2482                         ((*mi).second)->PushAddress(addr);
2483                 }
2484             }
2485         }
2486         if (vAddr.size() < 1000)
2487             pfrom->fGetAddr = false;
2488     }
2489
2490
2491     else if (strCommand == "inv")
2492     {
2493         vector<CInv> vInv;
2494         vRecv >> vInv;
2495         if (vInv.size() > 50000)
2496             return error("message inv size() = %d", vInv.size());
2497
2498         CTxDB txdb("r");
2499         foreach(const CInv& inv, vInv)
2500         {
2501             if (fShutdown)
2502                 return true;
2503             pfrom->AddInventoryKnown(inv);
2504
2505             bool fAlreadyHave = AlreadyHave(txdb, inv);
2506             printf("  got inventory: %s  %s\n", inv.ToString().c_str(), fAlreadyHave ? "have" : "new");
2507
2508             if (!fAlreadyHave)
2509                 pfrom->AskFor(inv);
2510             else if (inv.type == MSG_BLOCK && mapOrphanBlocks.count(inv.hash))
2511                 pfrom->PushGetBlocks(pindexBest, GetOrphanRoot(mapOrphanBlocks[inv.hash]));
2512
2513             // Track requests for our stuff
2514             CRITICAL_BLOCK(cs_mapRequestCount)
2515             {
2516                 map<uint256, int>::iterator mi = mapRequestCount.find(inv.hash);
2517                 if (mi != mapRequestCount.end())
2518                     (*mi).second++;
2519             }
2520         }
2521     }
2522
2523
2524     else if (strCommand == "getdata")
2525     {
2526         vector<CInv> vInv;
2527         vRecv >> vInv;
2528         if (vInv.size() > 50000)
2529             return error("message getdata size() = %d", vInv.size());
2530
2531         foreach(const CInv& inv, vInv)
2532         {
2533             if (fShutdown)
2534                 return true;
2535             printf("received getdata for: %s\n", inv.ToString().c_str());
2536
2537             if (inv.type == MSG_BLOCK)
2538             {
2539                 // Send block from disk
2540                 map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(inv.hash);
2541                 if (mi != mapBlockIndex.end())
2542                 {
2543                     CBlock block;
2544                     block.ReadFromDisk((*mi).second);
2545                     pfrom->PushMessage("block", block);
2546
2547                     // Trigger them to send a getblocks request for the next batch of inventory
2548                     if (inv.hash == pfrom->hashContinue)
2549                     {
2550                         // Bypass PushInventory, this must send even if redundant,
2551                         // and we want it right after the last block so they don't
2552                         // wait for other stuff first.
2553                         vector<CInv> vInv;
2554                         vInv.push_back(CInv(MSG_BLOCK, hashBestChain));
2555                         pfrom->PushMessage("inv", vInv);
2556                         pfrom->hashContinue = 0;
2557                     }
2558                 }
2559             }
2560             else if (inv.IsKnownType())
2561             {
2562                 // Send stream from relay memory
2563                 CRITICAL_BLOCK(cs_mapRelay)
2564                 {
2565                     map<CInv, CDataStream>::iterator mi = mapRelay.find(inv);
2566                     if (mi != mapRelay.end())
2567                         pfrom->PushMessage(inv.GetCommand(), (*mi).second);
2568                 }
2569             }
2570
2571             // Track requests for our stuff
2572             CRITICAL_BLOCK(cs_mapRequestCount)
2573             {
2574                 map<uint256, int>::iterator mi = mapRequestCount.find(inv.hash);
2575                 if (mi != mapRequestCount.end())
2576                     (*mi).second++;
2577             }
2578         }
2579     }
2580
2581
2582     else if (strCommand == "getblocks")
2583     {
2584         CBlockLocator locator;
2585         uint256 hashStop;
2586         vRecv >> locator >> hashStop;
2587
2588         // Find the last block the caller has in the main chain
2589         CBlockIndex* pindex = locator.GetBlockIndex();
2590
2591         // Send the rest of the chain
2592         if (pindex)
2593             pindex = pindex->pnext;
2594         int nLimit = 500 + locator.GetDistanceBack();
2595         printf("getblocks %d to %s limit %d\n", (pindex ? pindex->nHeight : -1), hashStop.ToString().substr(0,20).c_str(), nLimit);
2596         for (; pindex; pindex = pindex->pnext)
2597         {
2598             if (pindex->GetBlockHash() == hashStop)
2599             {
2600                 printf("  getblocks stopping at %d %s\n", pindex->nHeight, pindex->GetBlockHash().ToString().substr(0,20).c_str());
2601                 break;
2602             }
2603             pfrom->PushInventory(CInv(MSG_BLOCK, pindex->GetBlockHash()));
2604             if (--nLimit <= 0)
2605             {
2606                 // When this block is requested, we'll send an inv that'll make them
2607                 // getblocks the next batch of inventory.
2608                 printf("  getblocks stopping at limit %d %s\n", pindex->nHeight, pindex->GetBlockHash().ToString().substr(0,20).c_str());
2609                 pfrom->hashContinue = pindex->GetBlockHash();
2610                 break;
2611             }
2612         }
2613     }
2614
2615
2616     else if (strCommand == "getheaders")
2617     {
2618         CBlockLocator locator;
2619         uint256 hashStop;
2620         vRecv >> locator >> hashStop;
2621
2622         CBlockIndex* pindex = NULL;
2623         if (locator.IsNull())
2624         {
2625             // If locator is null, return the hashStop block
2626             map<uint256, CBlockIndex*>::iterator mi = mapBlockIndex.find(hashStop);
2627             if (mi == mapBlockIndex.end())
2628                 return true;
2629             pindex = (*mi).second;
2630         }
2631         else
2632         {
2633             // Find the last block the caller has in the main chain
2634             pindex = locator.GetBlockIndex();
2635             if (pindex)
2636                 pindex = pindex->pnext;
2637         }
2638
2639         vector<CBlock> vHeaders;
2640         int nLimit = 2000 + locator.GetDistanceBack();
2641         printf("getheaders %d to %s limit %d\n", (pindex ? pindex->nHeight : -1), hashStop.ToString().substr(0,20).c_str(), nLimit);
2642         for (; pindex; pindex = pindex->pnext)
2643         {
2644             vHeaders.push_back(pindex->GetBlockHeader());
2645             if (--nLimit <= 0 || pindex->GetBlockHash() == hashStop)
2646                 break;
2647         }
2648         pfrom->PushMessage("headers", vHeaders);
2649     }
2650
2651
2652     else if (strCommand == "tx")
2653     {
2654         vector<uint256> vWorkQueue;
2655         CDataStream vMsg(vRecv);
2656         CTransaction tx;
2657         vRecv >> tx;
2658
2659         CInv inv(MSG_TX, tx.GetHash());
2660         pfrom->AddInventoryKnown(inv);
2661
2662         bool fMissingInputs = false;
2663         if (tx.AcceptToMemoryPool(true, &fMissingInputs))
2664         {
2665             AddToWalletIfMine(tx, NULL);
2666             RelayMessage(inv, vMsg);
2667             mapAlreadyAskedFor.erase(inv);
2668             vWorkQueue.push_back(inv.hash);
2669
2670             // Recursively process any orphan transactions that depended on this one
2671             for (int i = 0; i < vWorkQueue.size(); i++)
2672             {
2673                 uint256 hashPrev = vWorkQueue[i];
2674                 for (multimap<uint256, CDataStream*>::iterator mi = mapOrphanTransactionsByPrev.lower_bound(hashPrev);
2675                      mi != mapOrphanTransactionsByPrev.upper_bound(hashPrev);
2676                      ++mi)
2677                 {
2678                     const CDataStream& vMsg = *((*mi).second);
2679                     CTransaction tx;
2680                     CDataStream(vMsg) >> tx;
2681                     CInv inv(MSG_TX, tx.GetHash());
2682
2683                     if (tx.AcceptToMemoryPool(true))
2684                     {
2685                         printf("   accepted orphan tx %s\n", inv.hash.ToString().substr(0,10).c_str());
2686                         AddToWalletIfMine(tx, NULL);
2687                         RelayMessage(inv, vMsg);
2688                         mapAlreadyAskedFor.erase(inv);
2689                         vWorkQueue.push_back(inv.hash);
2690                     }
2691                 }
2692             }
2693
2694             foreach(uint256 hash, vWorkQueue)
2695                 EraseOrphanTx(hash);
2696         }
2697         else if (fMissingInputs)
2698         {
2699             printf("storing orphan tx %s\n", inv.hash.ToString().substr(0,10).c_str());
2700             AddOrphanTx(vMsg);
2701         }
2702     }
2703
2704
2705     else if (strCommand == "block")
2706     {
2707         CBlock block;
2708         vRecv >> block;
2709
2710         printf("received block %s\n", block.GetHash().ToString().substr(0,20).c_str());
2711         // block.print();
2712
2713         CInv inv(MSG_BLOCK, block.GetHash());
2714         pfrom->AddInventoryKnown(inv);
2715
2716         if (ProcessBlock(pfrom, &block))
2717             mapAlreadyAskedFor.erase(inv);
2718     }
2719
2720
2721     else if (strCommand == "getaddr")
2722     {
2723         // Nodes rebroadcast an addr every 24 hours
2724         pfrom->vAddrToSend.clear();
2725         int64 nSince = GetAdjustedTime() - 3 * 60 * 60; // in the last 3 hours
2726         CRITICAL_BLOCK(cs_mapAddresses)
2727         {
2728             unsigned int nCount = 0;
2729             foreach(const PAIRTYPE(vector<unsigned char>, CAddress)& item, mapAddresses)
2730             {
2731                 const CAddress& addr = item.second;
2732                 if (addr.nTime > nSince)
2733                     nCount++;
2734             }
2735             foreach(const PAIRTYPE(vector<unsigned char>, CAddress)& item, mapAddresses)
2736             {
2737                 const CAddress& addr = item.second;
2738                 if (addr.nTime > nSince && GetRand(nCount) < 2500)
2739                     pfrom->PushAddress(addr);
2740             }
2741         }
2742     }
2743
2744
2745     else if (strCommand == "checkorder")
2746     {
2747         uint256 hashReply;
2748         vRecv >> hashReply;
2749
2750         if (!GetBoolArg("-allowreceivebyip"))
2751         {
2752             pfrom->PushMessage("reply", hashReply, (int)2, string(""));
2753             return true;
2754         }
2755
2756         CWalletTx order;
2757         vRecv >> order;
2758
2759         /// we have a chance to check the order here
2760
2761         // Keep giving the same key to the same ip until they use it
2762         if (!mapReuseKey.count(pfrom->addr.ip))
2763             mapReuseKey[pfrom->addr.ip] = GetKeyFromKeyPool();
2764
2765         // Send back approval of order and pubkey to use
2766         CScript scriptPubKey;
2767         scriptPubKey << mapReuseKey[pfrom->addr.ip] << OP_CHECKSIG;
2768         pfrom->PushMessage("reply", hashReply, (int)0, scriptPubKey);
2769     }
2770
2771
2772     else if (strCommand == "submitorder")
2773     {
2774         uint256 hashReply;
2775         vRecv >> hashReply;
2776
2777         if (!GetBoolArg("-allowreceivebyip"))
2778         {
2779             pfrom->PushMessage("reply", hashReply, (int)2);
2780             return true;
2781         }
2782
2783         CWalletTx wtxNew;
2784         vRecv >> wtxNew;
2785         wtxNew.fFromMe = false;
2786
2787         // Broadcast
2788         if (!wtxNew.AcceptWalletTransaction())
2789         {
2790             pfrom->PushMessage("reply", hashReply, (int)1);
2791             return error("submitorder AcceptWalletTransaction() failed, returning error 1");
2792         }
2793         wtxNew.fTimeReceivedIsTxTime = true;
2794         AddToWallet(wtxNew);
2795         wtxNew.RelayWalletTransaction();
2796         mapReuseKey.erase(pfrom->addr.ip);
2797
2798         // Send back confirmation
2799         pfrom->PushMessage("reply", hashReply, (int)0);
2800     }
2801
2802
2803     else if (strCommand == "reply")
2804     {
2805         uint256 hashReply;
2806         vRecv >> hashReply;
2807
2808         CRequestTracker tracker;
2809         CRITICAL_BLOCK(pfrom->cs_mapRequests)
2810         {
2811             map<uint256, CRequestTracker>::iterator mi = pfrom->mapRequests.find(hashReply);
2812             if (mi != pfrom->mapRequests.end())
2813             {
2814                 tracker = (*mi).second;
2815                 pfrom->mapRequests.erase(mi);
2816             }
2817         }
2818         if (!tracker.IsNull())
2819             tracker.fn(tracker.param1, vRecv);
2820     }
2821
2822
2823     else if (strCommand == "ping")
2824     {
2825     }
2826
2827
2828     else if (strCommand == "alert")
2829     {
2830         CAlert alert;
2831         vRecv >> alert;
2832
2833         if (alert.ProcessAlert())
2834         {
2835             // Relay
2836             pfrom->setKnown.insert(alert.GetHash());
2837             CRITICAL_BLOCK(cs_vNodes)
2838                 foreach(CNode* pnode, vNodes)
2839                     alert.RelayTo(pnode);
2840         }
2841     }
2842
2843
2844     else
2845     {
2846         // Ignore unknown commands for extensibility
2847     }
2848
2849
2850     // Update the last seen time for this node's address
2851     if (pfrom->fNetworkNode)
2852         if (strCommand == "version" || strCommand == "addr" || strCommand == "inv" || strCommand == "getdata" || strCommand == "ping")
2853             AddressCurrentlyConnected(pfrom->addr);
2854
2855
2856     return true;
2857 }
2858
2859
2860
2861
2862
2863
2864
2865
2866
2867 bool SendMessages(CNode* pto, bool fSendTrickle)
2868 {
2869     CRITICAL_BLOCK(cs_main)
2870     {
2871         // Don't send anything until we get their version message
2872         if (pto->nVersion == 0)
2873             return true;
2874
2875         // Keep-alive ping
2876         if (pto->nLastSend && GetTime() - pto->nLastSend > 30 * 60 && pto->vSend.empty())
2877             pto->PushMessage("ping");
2878
2879         // Resend wallet transactions that haven't gotten in a block yet
2880         ResendWalletTransactions();
2881
2882         // Address refresh broadcast
2883         static int64 nLastRebroadcast;
2884         if (GetTime() - nLastRebroadcast > 24 * 60 * 60)
2885         {
2886             nLastRebroadcast = GetTime();
2887             CRITICAL_BLOCK(cs_vNodes)
2888             {
2889                 foreach(CNode* pnode, vNodes)
2890                 {
2891                     // Periodically clear setAddrKnown to allow refresh broadcasts
2892                     pnode->setAddrKnown.clear();
2893
2894                     // Rebroadcast our address
2895                     if (addrLocalHost.IsRoutable() && !fUseProxy)
2896                     {
2897                         CAddress addr(addrLocalHost);
2898                         addr.nTime = GetAdjustedTime();
2899                         pnode->PushAddress(addr);
2900                     }
2901                 }
2902             }
2903         }
2904
2905         // Clear out old addresses periodically so it's not too much work at once
2906         static int64 nLastClear;
2907         if (nLastClear == 0)
2908             nLastClear = GetTime();
2909         if (GetTime() - nLastClear > 10 * 60 && vNodes.size() >= 3)
2910         {
2911             nLastClear = GetTime();
2912             CRITICAL_BLOCK(cs_mapAddresses)
2913             {
2914                 CAddrDB addrdb;
2915                 int64 nSince = GetAdjustedTime() - 14 * 24 * 60 * 60;
2916                 for (map<vector<unsigned char>, CAddress>::iterator mi = mapAddresses.begin();
2917                      mi != mapAddresses.end();)
2918                 {
2919                     const CAddress& addr = (*mi).second;
2920                     if (addr.nTime < nSince)
2921                     {
2922                         if (mapAddresses.size() < 1000 || GetTime() > nLastClear + 20)
2923                             break;
2924                         addrdb.EraseAddress(addr);
2925                         mapAddresses.erase(mi++);
2926                     }
2927                     else
2928                         mi++;
2929                 }
2930             }
2931         }
2932
2933
2934         //
2935         // Message: addr
2936         //
2937         if (fSendTrickle)
2938         {
2939             vector<CAddress> vAddr;
2940             vAddr.reserve(pto->vAddrToSend.size());
2941             foreach(const CAddress& addr, pto->vAddrToSend)
2942             {
2943                 // returns true if wasn't already contained in the set
2944                 if (pto->setAddrKnown.insert(addr).second)
2945                 {
2946                     vAddr.push_back(addr);
2947                     // receiver rejects addr messages larger than 1000
2948                     if (vAddr.size() >= 1000)
2949                     {
2950                         pto->PushMessage("addr", vAddr);
2951                         vAddr.clear();
2952                     }
2953                 }
2954             }
2955             pto->vAddrToSend.clear();
2956             if (!vAddr.empty())
2957                 pto->PushMessage("addr", vAddr);
2958         }
2959
2960
2961         //
2962         // Message: inventory
2963         //
2964         vector<CInv> vInv;
2965         vector<CInv> vInvWait;
2966         CRITICAL_BLOCK(pto->cs_inventory)
2967         {
2968             vInv.reserve(pto->vInventoryToSend.size());
2969             vInvWait.reserve(pto->vInventoryToSend.size());
2970             foreach(const CInv& inv, pto->vInventoryToSend)
2971             {
2972                 if (pto->setInventoryKnown.count(inv))
2973                     continue;
2974
2975                 // trickle out tx inv to protect privacy
2976                 if (inv.type == MSG_TX && !fSendTrickle)
2977                 {
2978                     // 1/4 of tx invs blast to all immediately
2979                     static uint256 hashSalt;
2980                     if (hashSalt == 0)
2981                         RAND_bytes((unsigned char*)&hashSalt, sizeof(hashSalt));
2982                     uint256 hashRand = inv.hash ^ hashSalt;
2983                     hashRand = Hash(BEGIN(hashRand), END(hashRand));
2984                     bool fTrickleWait = ((hashRand & 3) != 0);
2985
2986                     // always trickle our own transactions
2987                     if (!fTrickleWait)
2988                     {
2989                         TRY_CRITICAL_BLOCK(cs_mapWallet)
2990                         {
2991                             map<uint256, CWalletTx>::iterator mi = mapWallet.find(inv.hash);
2992                             if (mi != mapWallet.end())
2993                             {
2994                                 CWalletTx& wtx = (*mi).second;
2995                                 if (wtx.fFromMe)
2996                                     fTrickleWait = true;
2997                             }
2998                         }
2999                     }
3000
3001                     if (fTrickleWait)
3002                     {
3003                         vInvWait.push_back(inv);
3004                         continue;
3005                     }
3006                 }
3007
3008                 // returns true if wasn't already contained in the set
3009                 if (pto->setInventoryKnown.insert(inv).second)
3010                 {
3011                     vInv.push_back(inv);
3012                     if (vInv.size() >= 1000)
3013                     {
3014                         pto->PushMessage("inv", vInv);
3015                         vInv.clear();
3016                     }
3017                 }
3018             }
3019             pto->vInventoryToSend = vInvWait;
3020         }
3021         if (!vInv.empty())
3022             pto->PushMessage("inv", vInv);
3023
3024
3025         //
3026         // Message: getdata
3027         //
3028         vector<CInv> vGetData;
3029         int64 nNow = GetTime() * 1000000;
3030         CTxDB txdb("r");
3031         while (!pto->mapAskFor.empty() && (*pto->mapAskFor.begin()).first <= nNow)
3032         {
3033             const CInv& inv = (*pto->mapAskFor.begin()).second;
3034             if (!AlreadyHave(txdb, inv))
3035             {
3036                 printf("sending getdata: %s\n", inv.ToString().c_str());
3037                 vGetData.push_back(inv);
3038                 if (vGetData.size() >= 1000)
3039                 {
3040                     pto->PushMessage("getdata", vGetData);
3041                     vGetData.clear();
3042                 }
3043             }
3044             pto->mapAskFor.erase(pto->mapAskFor.begin());
3045         }
3046         if (!vGetData.empty())
3047             pto->PushMessage("getdata", vGetData);
3048
3049     }
3050     return true;
3051 }
3052
3053
3054
3055
3056
3057
3058
3059
3060
3061
3062
3063
3064
3065
3066 //////////////////////////////////////////////////////////////////////////////
3067 //
3068 // BitcoinMiner
3069 //
3070
3071 void GenerateBitcoins(bool fGenerate)
3072 {
3073     if (fGenerateBitcoins != fGenerate)
3074     {
3075         fGenerateBitcoins = fGenerate;
3076         CWalletDB().WriteSetting("fGenerateBitcoins", fGenerateBitcoins);
3077         MainFrameRepaint();
3078     }
3079     if (fGenerateBitcoins)
3080     {
3081         int nProcessors = boost::thread::hardware_concurrency();
3082         printf("%d processors\n", nProcessors);
3083         if (nProcessors < 1)
3084             nProcessors = 1;
3085         if (fLimitProcessors && nProcessors > nLimitProcessors)
3086             nProcessors = nLimitProcessors;
3087         int nAddThreads = nProcessors - vnThreadsRunning[3];
3088         printf("Starting %d BitcoinMiner threads\n", nAddThreads);
3089         for (int i = 0; i < nAddThreads; i++)
3090         {
3091             if (!CreateThread(ThreadBitcoinMiner, NULL))
3092                 printf("Error: CreateThread(ThreadBitcoinMiner) failed\n");
3093             Sleep(10);
3094         }
3095     }
3096 }
3097
3098 void ThreadBitcoinMiner(void* parg)
3099 {
3100     try
3101     {
3102         vnThreadsRunning[3]++;
3103         BitcoinMiner();
3104         vnThreadsRunning[3]--;
3105     }
3106     catch (std::exception& e) {
3107         vnThreadsRunning[3]--;
3108         PrintException(&e, "ThreadBitcoinMiner()");
3109     } catch (...) {
3110         vnThreadsRunning[3]--;
3111         PrintException(NULL, "ThreadBitcoinMiner()");
3112     }
3113     UIThreadCall(boost::bind(CalledSetStatusBar, "", 0));
3114     nHPSTimerStart = 0;
3115     if (vnThreadsRunning[3] == 0)
3116         dHashesPerSec = 0;
3117     printf("ThreadBitcoinMiner exiting, %d threads remaining\n", vnThreadsRunning[3]);
3118 }
3119
3120 #if defined(__GNUC__) && defined(CRYPTOPP_X86_ASM_AVAILABLE)
3121 void CallCPUID(int in, int& aret, int& cret)
3122 {
3123     int a, c;
3124     asm (
3125         "mov %2, %%eax; " // in into eax
3126         "cpuid;"
3127         "mov %%eax, %0;" // eax into a
3128         "mov %%ecx, %1;" // ecx into c
3129         :"=r"(a),"=r"(c) /* output */
3130         :"r"(in) /* input */
3131         :"%eax","%ebx","%ecx","%edx" /* clobbered register */
3132     );
3133     aret = a;
3134     cret = c;
3135 }
3136
3137 bool Detect128BitSSE2()
3138 {
3139     int a, c, nBrand;
3140     CallCPUID(0, a, nBrand);
3141     bool fIntel = (nBrand == 0x6c65746e); // ntel
3142     bool fAMD = (nBrand == 0x444d4163); // cAMD
3143
3144     struct
3145     {
3146         unsigned int nStepping : 4;
3147         unsigned int nModel : 4;
3148         unsigned int nFamily : 4;
3149         unsigned int nProcessorType : 2;
3150         unsigned int nUnused : 2;
3151         unsigned int nExtendedModel : 4;
3152         unsigned int nExtendedFamily : 8;
3153     }
3154     cpu;
3155     CallCPUID(1, a, c);
3156     memcpy(&cpu, &a, sizeof(cpu));
3157     int nFamily = cpu.nExtendedFamily + cpu.nFamily;
3158     int nModel = cpu.nExtendedModel*16 + cpu.nModel;
3159
3160     // We need Intel Nehalem or AMD K10 or better for 128bit SSE2
3161     // Nehalem = i3/i5/i7 and some Xeon
3162     // K10 = Opterons with 4 or more cores, Phenom, Phenom II, Athlon II
3163     //  Intel Core i5  family 6, model 26 or 30
3164     //  Intel Core i7  family 6, model 26 or 30
3165     //  Intel Core i3  family 6, model 37
3166     //  AMD Phenom    family 16, model 10
3167     bool fUseSSE2 = ((fIntel && nFamily * 10000 + nModel >=  60026) ||
3168                      (fAMD   && nFamily * 10000 + nModel >= 160010));
3169
3170     // AMD reports a lower model number in 64-bit mode
3171     if (fAMD && sizeof(void*) > 4 && nFamily * 10000 + nModel >= 160000)
3172         fUseSSE2 = true;
3173
3174     static bool fPrinted;
3175     if (!fPrinted)
3176     {
3177         fPrinted = true;
3178         printf("CPUID %08x family %d, model %d, stepping %d, fUseSSE2=%d\n", nBrand, nFamily, nModel, cpu.nStepping, fUseSSE2);
3179     }
3180     return fUseSSE2;
3181 }
3182 #else
3183 bool Detect128BitSSE2() { return false; }
3184 #endif
3185
3186 int FormatHashBlocks(void* pbuffer, unsigned int len)
3187 {
3188     unsigned char* pdata = (unsigned char*)pbuffer;
3189     unsigned int blocks = 1 + ((len + 8) / 64);
3190     unsigned char* pend = pdata + 64 * blocks;
3191     memset(pdata + len, 0, 64 * blocks - len);
3192     pdata[len] = 0x80;
3193     unsigned int bits = len * 8;
3194     pend[-1] = (bits >> 0) & 0xff;
3195     pend[-2] = (bits >> 8) & 0xff;
3196     pend[-3] = (bits >> 16) & 0xff;
3197     pend[-4] = (bits >> 24) & 0xff;
3198     return blocks;
3199 }
3200
3201 using CryptoPP::ByteReverse;
3202
3203 static const unsigned int pSHA256InitState[8] =
3204 {0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19};
3205
3206 inline void SHA256Transform(void* pstate, void* pinput, const void* pinit)
3207 {
3208     memcpy(pstate, pinit, 32);
3209     CryptoPP::SHA256::Transform((CryptoPP::word32*)pstate, (CryptoPP::word32*)pinput);
3210 }
3211
3212 //
3213 // ScanHash scans nonces looking for a hash with at least some zero bits.
3214 // It operates on big endian data.  Caller does the byte reversing.
3215 // All input buffers are 16-byte aligned.  nNonce is usually preserved
3216 // between calls, but periodically or if nNonce is 0xffff0000 or above,
3217 // the block is rebuilt and nNonce starts over at zero.
3218 //
3219 unsigned int ScanHash_CryptoPP(char* pmidstate, char* pdata, char* phash1, char* phash, unsigned int& nHashesDone)
3220 {
3221     unsigned int& nNonce = *(unsigned int*)(pdata + 12);
3222     for (;;)
3223     {
3224         // Crypto++ SHA-256
3225         // Hash pdata using pmidstate as the starting state into
3226         // preformatted buffer phash1, then hash phash1 into phash
3227         nNonce++;
3228         SHA256Transform(phash1, pdata, pmidstate);
3229         SHA256Transform(phash, phash1, pSHA256InitState);
3230
3231         // Return the nonce if the hash has at least some zero bits,
3232         // caller will check if it has enough to reach the target
3233         if (((unsigned short*)phash)[14] == 0)
3234             return nNonce;
3235
3236         // If nothing found after trying for a while, return -1
3237         if ((nNonce & 0xffff) == 0)
3238         {
3239             nHashesDone = 0xffff+1;
3240             return -1;
3241         }
3242     }
3243 }
3244
3245 extern unsigned int ScanHash_4WaySSE2(char* pmidstate, char* pblock, char* phash1, char* phash, unsigned int& nHashesDone);
3246
3247
3248
3249 class COrphan
3250 {
3251 public:
3252     CTransaction* ptx;
3253     set<uint256> setDependsOn;
3254     double dPriority;
3255
3256     COrphan(CTransaction* ptxIn)
3257     {
3258         ptx = ptxIn;
3259         dPriority = 0;
3260     }
3261
3262     void print() const
3263     {
3264         printf("COrphan(hash=%s, dPriority=%.1f)\n", ptx->GetHash().ToString().substr(0,10).c_str(), dPriority);
3265         foreach(uint256 hash, setDependsOn)
3266             printf("   setDependsOn %s\n", hash.ToString().substr(0,10).c_str());
3267     }
3268 };
3269
3270
3271 CBlock* CreateNewBlock(CReserveKey& reservekey)
3272 {
3273     CBlockIndex* pindexPrev = pindexBest;
3274
3275     // Create new block
3276     auto_ptr<CBlock> pblock(new CBlock());
3277     if (!pblock.get())
3278         return NULL;
3279
3280     // Create coinbase tx
3281     CTransaction txNew;
3282     txNew.vin.resize(1);
3283     txNew.vin[0].prevout.SetNull();
3284     txNew.vout.resize(1);
3285     txNew.vout[0].scriptPubKey << reservekey.GetReservedKey() << OP_CHECKSIG;
3286
3287     // Add our coinbase tx as first transaction
3288     pblock->vtx.push_back(txNew);
3289
3290     // Collect memory pool transactions into the block
3291     int64 nFees = 0;
3292     CRITICAL_BLOCK(cs_main)
3293     CRITICAL_BLOCK(cs_mapTransactions)
3294     {
3295         CTxDB txdb("r");
3296
3297         // Priority order to process transactions
3298         list<COrphan> vOrphan; // list memory doesn't move
3299         map<uint256, vector<COrphan*> > mapDependers;
3300         multimap<double, CTransaction*> mapPriority;
3301         for (map<uint256, CTransaction>::iterator mi = mapTransactions.begin(); mi != mapTransactions.end(); ++mi)
3302         {
3303             CTransaction& tx = (*mi).second;
3304             if (tx.IsCoinBase() || !tx.IsFinal())
3305                 continue;
3306
3307             COrphan* porphan = NULL;
3308             double dPriority = 0;
3309             foreach(const CTxIn& txin, tx.vin)
3310             {
3311                 // Read prev transaction
3312                 CTransaction txPrev;
3313                 CTxIndex txindex;
3314                 if (!txPrev.ReadFromDisk(txdb, txin.prevout, txindex))
3315                 {
3316                     // Has to wait for dependencies
3317                     if (!porphan)
3318                     {
3319                         // Use list for automatic deletion
3320                         vOrphan.push_back(COrphan(&tx));
3321                         porphan = &vOrphan.back();
3322                     }
3323                     mapDependers[txin.prevout.hash].push_back(porphan);
3324                     porphan->setDependsOn.insert(txin.prevout.hash);
3325                     continue;
3326                 }
3327                 int64 nValueIn = txPrev.vout[txin.prevout.n].nValue;
3328
3329                 // Read block header
3330                 int nConf = 0;
3331                 CBlock block;
3332                 if (block.ReadFromDisk(txindex.pos.nFile, txindex.pos.nBlockPos, false))
3333                 {
3334                     map<uint256, CBlockIndex*>::iterator it = mapBlockIndex.find(block.GetHash());
3335                     if (it != mapBlockIndex.end())
3336                     {
3337                         CBlockIndex* pindex = (*it).second;
3338                         if (pindex->IsInMainChain())
3339                             nConf = 1 + nBestHeight - pindex->nHeight;
3340                     }
3341                 }
3342
3343                 dPriority += (double)nValueIn * nConf;
3344
3345                 if (fDebug && GetBoolArg("-printpriority"))
3346                     printf("priority     nValueIn=%-12I64d nConf=%-5d dPriority=%-20.1f\n", nValueIn, nConf, dPriority);
3347             }
3348
3349             // Priority is sum(valuein * age) / txsize
3350             dPriority /= ::GetSerializeSize(tx, SER_NETWORK);
3351
3352             if (porphan)
3353                 porphan->dPriority = dPriority;
3354             else
3355                 mapPriority.insert(make_pair(-dPriority, &(*mi).second));
3356
3357             if (fDebug && GetBoolArg("-printpriority"))
3358             {
3359                 printf("priority %-20.1f %s\n%s", dPriority, tx.GetHash().ToString().substr(0,10).c_str(), tx.ToString().c_str());
3360                 if (porphan)
3361                     porphan->print();
3362                 printf("\n");
3363             }
3364         }
3365
3366         // Collect transactions into block
3367         map<uint256, CTxIndex> mapTestPool;
3368         uint64 nBlockSize = 1000;
3369         int nBlockSigOps = 100;
3370         while (!mapPriority.empty())
3371         {
3372             // Take highest priority transaction off priority queue
3373             double dPriority = -(*mapPriority.begin()).first;
3374             CTransaction& tx = *(*mapPriority.begin()).second;
3375             mapPriority.erase(mapPriority.begin());
3376
3377             // Size limits
3378             unsigned int nTxSize = ::GetSerializeSize(tx, SER_NETWORK);
3379             if (nBlockSize + nTxSize >= MAX_BLOCK_SIZE_GEN)
3380                 continue;
3381             int nTxSigOps = tx.GetSigOpCount();
3382             if (nBlockSigOps + nTxSigOps >= MAX_BLOCK_SIGOPS)
3383                 continue;
3384
3385             // Transaction fee required depends on block size
3386             bool fAllowFree = (nBlockSize + nTxSize < 4000 || dPriority > COIN * 144 / 250);
3387             int64 nMinFee = tx.GetMinFee(nBlockSize, fAllowFree);
3388
3389             // Connecting shouldn't fail due to dependency on other memory pool transactions
3390             // because we're already processing them in order of dependency
3391             map<uint256, CTxIndex> mapTestPoolTmp(mapTestPool);
3392             if (!tx.ConnectInputs(txdb, mapTestPoolTmp, CDiskTxPos(1,1,1), pindexPrev, nFees, false, true, nMinFee))
3393                 continue;
3394             swap(mapTestPool, mapTestPoolTmp);
3395
3396             // Added
3397             pblock->vtx.push_back(tx);
3398             nBlockSize += nTxSize;
3399             nBlockSigOps += nTxSigOps;
3400
3401             // Add transactions that depend on this one to the priority queue
3402             uint256 hash = tx.GetHash();
3403             if (mapDependers.count(hash))
3404             {
3405                 foreach(COrphan* porphan, mapDependers[hash])
3406                 {
3407                     if (!porphan->setDependsOn.empty())
3408                     {
3409                         porphan->setDependsOn.erase(hash);
3410                         if (porphan->setDependsOn.empty())
3411                             mapPriority.insert(make_pair(-porphan->dPriority, porphan->ptx));
3412                     }
3413                 }
3414             }
3415         }
3416     }
3417     pblock->vtx[0].vout[0].nValue = GetBlockValue(pindexPrev->nHeight+1, nFees);
3418
3419     // Fill in header
3420     pblock->hashPrevBlock  = pindexPrev->GetBlockHash();
3421     pblock->hashMerkleRoot = pblock->BuildMerkleTree();
3422     pblock->nTime          = max(pindexPrev->GetMedianTimePast()+1, GetAdjustedTime());
3423     pblock->nBits          = GetNextWorkRequired(pindexPrev);
3424     pblock->nNonce         = 0;
3425
3426     return pblock.release();
3427 }
3428
3429
3430 void IncrementExtraNonce(CBlock* pblock, CBlockIndex* pindexPrev, unsigned int& nExtraNonce, int64& nPrevTime)
3431 {
3432     // Update nExtraNonce
3433     int64 nNow = max(pindexPrev->GetMedianTimePast()+1, GetAdjustedTime());
3434     if (++nExtraNonce >= 0x7f && nNow > nPrevTime+1)
3435     {
3436         nExtraNonce = 1;
3437         nPrevTime = nNow;
3438     }
3439     pblock->vtx[0].vin[0].scriptSig = CScript() << pblock->nBits << CBigNum(nExtraNonce);
3440     pblock->hashMerkleRoot = pblock->BuildMerkleTree();
3441 }
3442
3443
3444 void FormatHashBuffers(CBlock* pblock, char* pmidstate, char* pdata, char* phash1)
3445 {
3446     //
3447     // Prebuild hash buffers
3448     //
3449     struct
3450     {
3451         struct unnamed2
3452         {
3453             int nVersion;
3454             uint256 hashPrevBlock;
3455             uint256 hashMerkleRoot;
3456             unsigned int nTime;
3457             unsigned int nBits;
3458             unsigned int nNonce;
3459         }
3460         block;
3461         unsigned char pchPadding0[64];
3462         uint256 hash1;
3463         unsigned char pchPadding1[64];
3464     }
3465     tmp;
3466     memset(&tmp, 0, sizeof(tmp));
3467
3468     tmp.block.nVersion       = pblock->nVersion;
3469     tmp.block.hashPrevBlock  = pblock->hashPrevBlock;
3470     tmp.block.hashMerkleRoot = pblock->hashMerkleRoot;
3471     tmp.block.nTime          = pblock->nTime;
3472     tmp.block.nBits          = pblock->nBits;
3473     tmp.block.nNonce         = pblock->nNonce;
3474
3475     FormatHashBlocks(&tmp.block, sizeof(tmp.block));
3476     FormatHashBlocks(&tmp.hash1, sizeof(tmp.hash1));
3477
3478     // Byte swap all the input buffer
3479     for (int i = 0; i < sizeof(tmp)/4; i++)
3480         ((unsigned int*)&tmp)[i] = ByteReverse(((unsigned int*)&tmp)[i]);
3481
3482     // Precalc the first half of the first hash, which stays constant
3483     SHA256Transform(pmidstate, &tmp.block, pSHA256InitState);
3484
3485     memcpy(pdata, &tmp.block, 128);
3486     memcpy(phash1, &tmp.hash1, 64);
3487 }
3488
3489
3490 bool CheckWork(CBlock* pblock, CReserveKey& reservekey)
3491 {
3492     uint256 hash = pblock->GetHash();
3493     uint256 hashTarget = CBigNum().SetCompact(pblock->nBits).getuint256();
3494
3495     if (hash > hashTarget)
3496         return false;
3497
3498     //// debug print
3499     printf("BitcoinMiner:\n");
3500     printf("proof-of-work found  \n  hash: %s  \ntarget: %s\n", hash.GetHex().c_str(), hashTarget.GetHex().c_str());
3501     pblock->print();
3502     printf("%s ", DateTimeStrFormat("%x %H:%M", GetTime()).c_str());
3503     printf("generated %s\n", FormatMoney(pblock->vtx[0].vout[0].nValue).c_str());
3504
3505     // Found a solution
3506     CRITICAL_BLOCK(cs_main)
3507     {
3508         if (pblock->hashPrevBlock != hashBestChain)
3509             return error("BitcoinMiner : generated block is stale");
3510
3511         // Remove key from key pool
3512         reservekey.KeepKey();
3513
3514         // Track how many getdata requests this block gets
3515         CRITICAL_BLOCK(cs_mapRequestCount)
3516             mapRequestCount[pblock->GetHash()] = 0;
3517
3518         // Process this block the same as if we had received it from another node
3519         if (!ProcessBlock(NULL, pblock))
3520             return error("BitcoinMiner : ProcessBlock, block not accepted");
3521     }
3522
3523     Sleep(2000);
3524     return true;
3525 }
3526
3527
3528 void BitcoinMiner()
3529 {
3530     printf("BitcoinMiner started\n");
3531     SetThreadPriority(THREAD_PRIORITY_LOWEST);
3532     bool f4WaySSE2 = Detect128BitSSE2();
3533     if (mapArgs.count("-4way"))
3534         f4WaySSE2 = GetBoolArg("-4way");
3535
3536     // Each thread has its own key and counter
3537     CReserveKey reservekey;
3538     unsigned int nExtraNonce = 0;
3539     int64 nPrevTime = 0;
3540
3541     while (fGenerateBitcoins)
3542     {
3543         if (AffinityBugWorkaround(ThreadBitcoinMiner))
3544             return;
3545         if (fShutdown)
3546             return;
3547         while (vNodes.empty() || IsInitialBlockDownload())
3548         {
3549             Sleep(1000);
3550             if (fShutdown)
3551                 return;
3552             if (!fGenerateBitcoins)
3553                 return;
3554         }
3555
3556
3557         //
3558         // Create new block
3559         //
3560         unsigned int nTransactionsUpdatedLast = nTransactionsUpdated;
3561         CBlockIndex* pindexPrev = pindexBest;
3562
3563         auto_ptr<CBlock> pblock(CreateNewBlock(reservekey));
3564         if (!pblock.get())
3565             return;
3566         IncrementExtraNonce(pblock.get(), pindexPrev, nExtraNonce, nPrevTime);
3567
3568         printf("Running BitcoinMiner with %d transactions in block\n", pblock->vtx.size());
3569
3570
3571         //
3572         // Prebuild hash buffers
3573         //
3574         char pmidstatebuf[32+16]; char* pmidstate = alignup<16>(pmidstatebuf);
3575         char pdatabuf[128+16];    char* pdata     = alignup<16>(pdatabuf);
3576         char phash1buf[64+16];    char* phash1    = alignup<16>(phash1buf);
3577
3578         FormatHashBuffers(pblock.get(), pmidstate, pdata, phash1);
3579
3580         unsigned int& nBlockTime = *(unsigned int*)(pdata + 64 + 4);
3581         unsigned int& nBlockNonce = *(unsigned int*)(pdata + 64 + 12);
3582
3583
3584         //
3585         // Search
3586         //
3587         int64 nStart = GetTime();
3588         uint256 hashTarget = CBigNum().SetCompact(pblock->nBits).getuint256();
3589         uint256 hashbuf[2];
3590         uint256& hash = *alignup<16>(hashbuf);
3591         loop
3592         {
3593             unsigned int nHashesDone = 0;
3594             unsigned int nNonceFound;
3595
3596 #ifdef FOURWAYSSE2
3597             if (f4WaySSE2)
3598                 // tcatm's 4-way 128-bit SSE2 SHA-256
3599                 nNonceFound = ScanHash_4WaySSE2(pmidstate, pdata + 64, phash1, (char*)&hash, nHashesDone);
3600             else
3601 #endif
3602                 // Crypto++ SHA-256
3603                 nNonceFound = ScanHash_CryptoPP(pmidstate, pdata + 64, phash1, (char*)&hash, nHashesDone);
3604
3605             // Check if something found
3606             if (nNonceFound != -1)
3607             {
3608                 for (int i = 0; i < sizeof(hash)/4; i++)
3609                     ((unsigned int*)&hash)[i] = ByteReverse(((unsigned int*)&hash)[i]);
3610
3611                 if (hash <= hashTarget)
3612                 {
3613                     // Found a solution
3614                     pblock->nNonce = ByteReverse(nNonceFound);
3615                     assert(hash == pblock->GetHash());
3616
3617                     SetThreadPriority(THREAD_PRIORITY_NORMAL);
3618                     CheckWork(pblock.get(), reservekey);
3619                     SetThreadPriority(THREAD_PRIORITY_LOWEST);
3620                     break;
3621                 }
3622             }
3623
3624             // Meter hashes/sec
3625             static int64 nHashCounter;
3626             if (nHPSTimerStart == 0)
3627             {
3628                 nHPSTimerStart = GetTimeMillis();
3629                 nHashCounter = 0;
3630             }
3631             else
3632                 nHashCounter += nHashesDone;
3633             if (GetTimeMillis() - nHPSTimerStart > 4000)
3634             {
3635                 static CCriticalSection cs;
3636                 CRITICAL_BLOCK(cs)
3637                 {
3638                     if (GetTimeMillis() - nHPSTimerStart > 4000)
3639                     {
3640                         dHashesPerSec = 1000.0 * nHashCounter / (GetTimeMillis() - nHPSTimerStart);
3641                         nHPSTimerStart = GetTimeMillis();
3642                         nHashCounter = 0;
3643                         string strStatus = strprintf("    %.0f khash/s", dHashesPerSec/1000.0);
3644                         UIThreadCall(boost::bind(CalledSetStatusBar, strStatus, 0));
3645                         static int64 nLogTime;
3646                         if (GetTime() - nLogTime > 30 * 60)
3647                         {
3648                             nLogTime = GetTime();
3649                             printf("%s ", DateTimeStrFormat("%x %H:%M", GetTime()).c_str());
3650                             printf("hashmeter %3d CPUs %6.0f khash/s\n", vnThreadsRunning[3], dHashesPerSec/1000.0);
3651                         }
3652                     }
3653                 }
3654             }
3655
3656             // Check for stop or if block needs to be rebuilt
3657             if (fShutdown)
3658                 return;
3659             if (!fGenerateBitcoins)
3660                 return;
3661             if (fLimitProcessors && vnThreadsRunning[3] > nLimitProcessors)
3662                 return;
3663             if (vNodes.empty())
3664                 break;
3665             if (nBlockNonce >= 0xffff0000)
3666                 break;
3667             if (nTransactionsUpdated != nTransactionsUpdatedLast && GetTime() - nStart > 60)
3668                 break;
3669             if (pindexPrev != pindexBest)
3670                 break;
3671
3672             // Update nTime every few seconds
3673             pblock->nTime = max(pindexPrev->GetMedianTimePast()+1, GetAdjustedTime());
3674             nBlockTime = ByteReverse(pblock->nTime);
3675         }
3676     }
3677 }
3678
3679
3680
3681
3682
3683
3684
3685
3686
3687
3688
3689
3690
3691
3692
3693
3694
3695
3696 //////////////////////////////////////////////////////////////////////////////
3697 //
3698 // Actions
3699 //
3700
3701
3702 int64 GetBalance()
3703 {
3704     int64 nStart = GetTimeMillis();
3705
3706     int64 nTotal = 0;
3707     CRITICAL_BLOCK(cs_mapWallet)
3708     {
3709         for (map<uint256, CWalletTx>::iterator it = mapWallet.begin(); it != mapWallet.end(); ++it)
3710         {
3711             CWalletTx* pcoin = &(*it).second;
3712             if (!pcoin->IsFinal() || pcoin->fSpent || !pcoin->IsConfirmed())
3713                 continue;
3714             nTotal += pcoin->GetCredit();
3715         }
3716     }
3717
3718     //printf("GetBalance() %"PRI64d"ms\n", GetTimeMillis() - nStart);
3719     return nTotal;
3720 }
3721
3722
3723 bool SelectCoinsMinConf(int64 nTargetValue, int nConfMine, int nConfTheirs, set<CWalletTx*>& setCoinsRet)
3724 {
3725     setCoinsRet.clear();
3726
3727     // List of values less than target
3728     int64 nLowestLarger = INT64_MAX;
3729     CWalletTx* pcoinLowestLarger = NULL;
3730     vector<pair<int64, CWalletTx*> > vValue;
3731     int64 nTotalLower = 0;
3732
3733     CRITICAL_BLOCK(cs_mapWallet)
3734     {
3735        vector<CWalletTx*> vCoins;
3736        vCoins.reserve(mapWallet.size());
3737        for (map<uint256, CWalletTx>::iterator it = mapWallet.begin(); it != mapWallet.end(); ++it)
3738            vCoins.push_back(&(*it).second);
3739        random_shuffle(vCoins.begin(), vCoins.end(), GetRandInt);
3740
3741        foreach(CWalletTx* pcoin, vCoins)
3742        {
3743             if (!pcoin->IsFinal() || pcoin->fSpent || !pcoin->IsConfirmed())
3744                 continue;
3745
3746             int nDepth = pcoin->GetDepthInMainChain();
3747             if (nDepth < (pcoin->IsFromMe() ? nConfMine : nConfTheirs))
3748                 continue;
3749
3750             int64 n = pcoin->GetCredit();
3751             if (n <= 0)
3752                 continue;
3753             if (n < nTargetValue)
3754             {
3755                 vValue.push_back(make_pair(n, pcoin));
3756                 nTotalLower += n;
3757             }
3758             else if (n == nTargetValue)
3759             {
3760                 setCoinsRet.insert(pcoin);
3761                 return true;
3762             }
3763             else if (n < nLowestLarger)
3764             {
3765                 nLowestLarger = n;
3766                 pcoinLowestLarger = pcoin;
3767             }
3768         }
3769     }
3770
3771     if (nTotalLower < nTargetValue)
3772     {
3773         if (pcoinLowestLarger == NULL)
3774             return false;
3775         setCoinsRet.insert(pcoinLowestLarger);
3776         return true;
3777     }
3778
3779     // Solve subset sum by stochastic approximation
3780     sort(vValue.rbegin(), vValue.rend());
3781     vector<char> vfIncluded;
3782     vector<char> vfBest(vValue.size(), true);
3783     int64 nBest = nTotalLower;
3784
3785     for (int nRep = 0; nRep < 1000 && nBest != nTargetValue; nRep++)
3786     {
3787         vfIncluded.assign(vValue.size(), false);
3788         int64 nTotal = 0;
3789         bool fReachedTarget = false;
3790         for (int nPass = 0; nPass < 2 && !fReachedTarget; nPass++)
3791         {
3792             for (int i = 0; i < vValue.size(); i++)
3793             {
3794                 if (nPass == 0 ? rand() % 2 : !vfIncluded[i])
3795                 {
3796                     nTotal += vValue[i].first;
3797                     vfIncluded[i] = true;
3798                     if (nTotal >= nTargetValue)
3799                     {
3800                         fReachedTarget = true;
3801                         if (nTotal < nBest)
3802                         {
3803                             nBest = nTotal;
3804                             vfBest = vfIncluded;
3805                         }
3806                         nTotal -= vValue[i].first;
3807                         vfIncluded[i] = false;
3808                     }
3809                 }
3810             }
3811         }
3812     }
3813
3814     // If the next larger is still closer, return it
3815     if (pcoinLowestLarger && nLowestLarger - nTargetValue <= nBest - nTargetValue)
3816         setCoinsRet.insert(pcoinLowestLarger);
3817     else
3818     {
3819         for (int i = 0; i < vValue.size(); i++)
3820             if (vfBest[i])
3821                 setCoinsRet.insert(vValue[i].second);
3822
3823         //// debug print
3824         printf("SelectCoins() best subset: ");
3825         for (int i = 0; i < vValue.size(); i++)
3826             if (vfBest[i])
3827                 printf("%s ", FormatMoney(vValue[i].first).c_str());
3828         printf("total %s\n", FormatMoney(nBest).c_str());
3829     }
3830
3831     return true;
3832 }
3833
3834 bool SelectCoins(int64 nTargetValue, set<CWalletTx*>& setCoinsRet)
3835 {
3836     return (SelectCoinsMinConf(nTargetValue, 1, 6, setCoinsRet) ||
3837             SelectCoinsMinConf(nTargetValue, 1, 1, setCoinsRet) ||
3838             SelectCoinsMinConf(nTargetValue, 0, 1, setCoinsRet));
3839 }
3840
3841
3842
3843
3844 bool CreateTransaction(CScript scriptPubKey, int64 nValue, CWalletTx& wtxNew, CReserveKey& reservekey, int64& nFeeRet)
3845 {
3846     CRITICAL_BLOCK(cs_main)
3847     {
3848         // txdb must be opened before the mapWallet lock
3849         CTxDB txdb("r");
3850         CRITICAL_BLOCK(cs_mapWallet)
3851         {
3852             nFeeRet = nTransactionFee;
3853             loop
3854             {
3855                 wtxNew.vin.clear();
3856                 wtxNew.vout.clear();
3857                 wtxNew.fFromMe = true;
3858                 if (nValue < 0)
3859                     return false;
3860                 int64 nValueOut = nValue;
3861                 int64 nTotalValue = nValue + nFeeRet;
3862
3863                 // Choose coins to use
3864                 set<CWalletTx*> setCoins;
3865                 if (!SelectCoins(nTotalValue, setCoins))
3866                     return false;
3867                 int64 nValueIn = 0;
3868                 foreach(CWalletTx* pcoin, setCoins)
3869                     nValueIn += pcoin->GetCredit();
3870
3871                 // Fill a vout to the payee
3872                 bool fChangeFirst = GetRand(2);
3873                 if (!fChangeFirst)
3874                     wtxNew.vout.push_back(CTxOut(nValueOut, scriptPubKey));
3875
3876                 // Fill a vout back to self with any change
3877                 int64 nChange = nValueIn - nTotalValue;
3878                 if (nChange >= CENT)
3879                 {
3880                     // Note: We use a new key here to keep it from being obvious which side is the change.
3881                     //  The drawback is that by not reusing a previous key, the change may be lost if a
3882                     //  backup is restored, if the backup doesn't have the new private key for the change.
3883                     //  If we reused the old key, it would be possible to add code to look for and
3884                     //  rediscover unknown transactions that were written with keys of ours to recover
3885                     //  post-backup change.
3886
3887                     // Reserve a new key pair from key pool
3888                     vector<unsigned char> vchPubKey = reservekey.GetReservedKey();
3889                     assert(mapKeys.count(vchPubKey));
3890
3891                     // Fill a vout to ourself, using same address type as the payment
3892                     CScript scriptChange;
3893                     if (scriptPubKey.GetBitcoinAddressHash160() != 0)
3894                         scriptChange.SetBitcoinAddress(vchPubKey);
3895                     else
3896                         scriptChange << vchPubKey << OP_CHECKSIG;
3897                     wtxNew.vout.push_back(CTxOut(nChange, scriptChange));
3898                 }
3899                 else
3900                     reservekey.ReturnKey();
3901
3902                 // Fill a vout to the payee
3903                 if (fChangeFirst)
3904                     wtxNew.vout.push_back(CTxOut(nValueOut, scriptPubKey));
3905
3906                 // Fill vin
3907                 foreach(CWalletTx* pcoin, setCoins)
3908                     for (int nOut = 0; nOut < pcoin->vout.size(); nOut++)
3909                         if (pcoin->vout[nOut].IsMine())
3910                             wtxNew.vin.push_back(CTxIn(pcoin->GetHash(), nOut));
3911
3912                 // Sign
3913                 int nIn = 0;
3914                 foreach(CWalletTx* pcoin, setCoins)
3915                     for (int nOut = 0; nOut < pcoin->vout.size(); nOut++)
3916                         if (pcoin->vout[nOut].IsMine())
3917                             if (!SignSignature(*pcoin, wtxNew, nIn++))
3918                                 return false;
3919
3920                 // Limit size
3921                 unsigned int nBytes = ::GetSerializeSize(*(CTransaction*)&wtxNew, SER_NETWORK);
3922                 if (nBytes >= MAX_BLOCK_SIZE_GEN/5)
3923                     return false;
3924
3925                 // Check that enough fee is included
3926                 int64 nPayFee = nTransactionFee * (1 + (int64)nBytes / 1000);
3927                 int64 nMinFee = wtxNew.GetMinFee();
3928                 if (nFeeRet < max(nPayFee, nMinFee))
3929                 {
3930                     nFeeRet = max(nPayFee, nMinFee);
3931                     continue;
3932                 }
3933
3934                 // Fill vtxPrev by copying from previous transactions vtxPrev
3935                 wtxNew.AddSupportingTransactions(txdb);
3936                 wtxNew.fTimeReceivedIsTxTime = true;
3937
3938                 break;
3939             }
3940         }
3941     }
3942     return true;
3943 }
3944
3945 // Call after CreateTransaction unless you want to abort
3946 bool CommitTransaction(CWalletTx& wtxNew, CReserveKey& reservekey)
3947 {
3948     CRITICAL_BLOCK(cs_main)
3949     {
3950         printf("CommitTransaction:\n%s", wtxNew.ToString().c_str());
3951         CRITICAL_BLOCK(cs_mapWallet)
3952         {
3953             // This is only to keep the database open to defeat the auto-flush for the
3954             // duration of this scope.  This is the only place where this optimization
3955             // maybe makes sense; please don't do it anywhere else.
3956             CWalletDB walletdb("r");
3957
3958             // Take key pair from key pool so it won't be used again
3959             reservekey.KeepKey();
3960
3961             // Add tx to wallet, because if it has change it's also ours,
3962             // otherwise just for transaction history.
3963             AddToWallet(wtxNew);
3964
3965             // Mark old coins as spent
3966             set<CWalletTx*> setCoins;
3967             foreach(const CTxIn& txin, wtxNew.vin)
3968                 setCoins.insert(&mapWallet[txin.prevout.hash]);
3969             foreach(CWalletTx* pcoin, setCoins)
3970             {
3971                 pcoin->fSpent = true;
3972                 pcoin->WriteToDisk();
3973                 vWalletUpdated.push_back(pcoin->GetHash());
3974             }
3975         }
3976
3977         // Track how many getdata requests our transaction gets
3978         CRITICAL_BLOCK(cs_mapRequestCount)
3979             mapRequestCount[wtxNew.GetHash()] = 0;
3980
3981         // Broadcast
3982         if (!wtxNew.AcceptToMemoryPool())
3983         {
3984             // This must not fail. The transaction has already been signed and recorded.
3985             printf("CommitTransaction() : Error: Transaction not valid");
3986             return false;
3987         }
3988         wtxNew.RelayWalletTransaction();
3989     }
3990     MainFrameRepaint();
3991     return true;
3992 }
3993
3994
3995
3996
3997 string SendMoney(CScript scriptPubKey, int64 nValue, CWalletTx& wtxNew, bool fAskFee)
3998 {
3999     CRITICAL_BLOCK(cs_main)
4000     {
4001         CReserveKey reservekey;
4002         int64 nFeeRequired;
4003         if (!CreateTransaction(scriptPubKey, nValue, wtxNew, reservekey, nFeeRequired))
4004         {
4005             string strError;
4006             if (nValue + nFeeRequired > GetBalance())
4007                 strError = strprintf(_("Error: This is an oversized transaction that requires a transaction fee of %s  "), FormatMoney(nFeeRequired).c_str());
4008             else
4009                 strError = _("Error: Transaction creation failed  ");
4010             printf("SendMoney() : %s", strError.c_str());
4011             return strError;
4012         }
4013
4014         if (fAskFee && !ThreadSafeAskFee(nFeeRequired, _("Sending..."), NULL))
4015             return "ABORTED";
4016
4017         if (!CommitTransaction(wtxNew, reservekey))
4018             return _("Error: The transaction was rejected.  This might happen if some of the coins in your wallet were already spent, such as if you used a copy of wallet.dat and coins were spent in the copy but not marked as spent here.");
4019     }
4020     MainFrameRepaint();
4021     return "";
4022 }
4023
4024
4025
4026 string SendMoneyToBitcoinAddress(string strAddress, int64 nValue, CWalletTx& wtxNew, bool fAskFee)
4027 {
4028     // Check amount
4029     if (nValue <= 0)
4030         return _("Invalid amount");
4031     if (nValue + nTransactionFee > GetBalance())
4032         return _("Insufficient funds");
4033
4034     // Parse bitcoin address
4035     CScript scriptPubKey;
4036     if (!scriptPubKey.SetBitcoinAddress(strAddress))
4037         return _("Invalid bitcoin address");
4038
4039     return SendMoney(scriptPubKey, nValue, wtxNew, fAskFee);
4040 }