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