1 // Copyright (c) 2011-2019 The Bitcoin Core developers
2 // Distributed under the MIT software license, see the accompanying
3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4 
5 #include <bench/bench.h>
6 #include <policy/policy.h>
7 #include <test/util/setup_common.h>
8 #include <txmempool.h>
9 
10 
AddTx(const CTransactionRef & tx,const CAmount & nFee,CTxMemPool & pool)11 static void AddTx(const CTransactionRef& tx, const CAmount& nFee, CTxMemPool& pool) EXCLUSIVE_LOCKS_REQUIRED(cs_main, pool.cs)
12 {
13     int64_t nTime = 0;
14     unsigned int nHeight = 1;
15     bool spendsCoinbase = false;
16     unsigned int sigOpCost = 4;
17     LockPoints lp;
18     pool.addUnchecked(CTxMemPoolEntry(
19         tx, nFee, nTime, nHeight,
20         spendsCoinbase, sigOpCost, lp));
21 }
22 
23 // Right now this is only testing eviction performance in an extremely small
24 // mempool. Code needs to be written to generate a much wider variety of
25 // unique transactions for a more meaningful performance measurement.
MempoolEviction(benchmark::Bench & bench)26 static void MempoolEviction(benchmark::Bench& bench)
27 {
28     TestingSetup test_setup{
29         CBaseChainParams::REGTEST,
30         /* extra_args */ {
31             "-nodebuglogfile",
32             "-nodebug",
33         },
34     };
35 
36     CMutableTransaction tx1 = CMutableTransaction();
37     tx1.vin.resize(1);
38     tx1.vin[0].scriptSig = CScript() << OP_1;
39     tx1.vin[0].scriptWitness.stack.push_back({1});
40     tx1.vout.resize(1);
41     tx1.vout[0].scriptPubKey = CScript() << OP_1 << OP_EQUAL;
42     tx1.vout[0].nValue = 10 * COIN;
43 
44     CMutableTransaction tx2 = CMutableTransaction();
45     tx2.vin.resize(1);
46     tx2.vin[0].scriptSig = CScript() << OP_2;
47     tx2.vin[0].scriptWitness.stack.push_back({2});
48     tx2.vout.resize(1);
49     tx2.vout[0].scriptPubKey = CScript() << OP_2 << OP_EQUAL;
50     tx2.vout[0].nValue = 10 * COIN;
51 
52     CMutableTransaction tx3 = CMutableTransaction();
53     tx3.vin.resize(1);
54     tx3.vin[0].prevout = COutPoint(tx2.GetHash(), 0);
55     tx3.vin[0].scriptSig = CScript() << OP_2;
56     tx3.vin[0].scriptWitness.stack.push_back({3});
57     tx3.vout.resize(1);
58     tx3.vout[0].scriptPubKey = CScript() << OP_3 << OP_EQUAL;
59     tx3.vout[0].nValue = 10 * COIN;
60 
61     CMutableTransaction tx4 = CMutableTransaction();
62     tx4.vin.resize(2);
63     tx4.vin[0].prevout.SetNull();
64     tx4.vin[0].scriptSig = CScript() << OP_4;
65     tx4.vin[0].scriptWitness.stack.push_back({4});
66     tx4.vin[1].prevout.SetNull();
67     tx4.vin[1].scriptSig = CScript() << OP_4;
68     tx4.vin[1].scriptWitness.stack.push_back({4});
69     tx4.vout.resize(2);
70     tx4.vout[0].scriptPubKey = CScript() << OP_4 << OP_EQUAL;
71     tx4.vout[0].nValue = 10 * COIN;
72     tx4.vout[1].scriptPubKey = CScript() << OP_4 << OP_EQUAL;
73     tx4.vout[1].nValue = 10 * COIN;
74 
75     CMutableTransaction tx5 = CMutableTransaction();
76     tx5.vin.resize(2);
77     tx5.vin[0].prevout = COutPoint(tx4.GetHash(), 0);
78     tx5.vin[0].scriptSig = CScript() << OP_4;
79     tx5.vin[0].scriptWitness.stack.push_back({4});
80     tx5.vin[1].prevout.SetNull();
81     tx5.vin[1].scriptSig = CScript() << OP_5;
82     tx5.vin[1].scriptWitness.stack.push_back({5});
83     tx5.vout.resize(2);
84     tx5.vout[0].scriptPubKey = CScript() << OP_5 << OP_EQUAL;
85     tx5.vout[0].nValue = 10 * COIN;
86     tx5.vout[1].scriptPubKey = CScript() << OP_5 << OP_EQUAL;
87     tx5.vout[1].nValue = 10 * COIN;
88 
89     CMutableTransaction tx6 = CMutableTransaction();
90     tx6.vin.resize(2);
91     tx6.vin[0].prevout = COutPoint(tx4.GetHash(), 1);
92     tx6.vin[0].scriptSig = CScript() << OP_4;
93     tx6.vin[0].scriptWitness.stack.push_back({4});
94     tx6.vin[1].prevout.SetNull();
95     tx6.vin[1].scriptSig = CScript() << OP_6;
96     tx6.vin[1].scriptWitness.stack.push_back({6});
97     tx6.vout.resize(2);
98     tx6.vout[0].scriptPubKey = CScript() << OP_6 << OP_EQUAL;
99     tx6.vout[0].nValue = 10 * COIN;
100     tx6.vout[1].scriptPubKey = CScript() << OP_6 << OP_EQUAL;
101     tx6.vout[1].nValue = 10 * COIN;
102 
103     CMutableTransaction tx7 = CMutableTransaction();
104     tx7.vin.resize(2);
105     tx7.vin[0].prevout = COutPoint(tx5.GetHash(), 0);
106     tx7.vin[0].scriptSig = CScript() << OP_5;
107     tx7.vin[0].scriptWitness.stack.push_back({5});
108     tx7.vin[1].prevout = COutPoint(tx6.GetHash(), 0);
109     tx7.vin[1].scriptSig = CScript() << OP_6;
110     tx7.vin[1].scriptWitness.stack.push_back({6});
111     tx7.vout.resize(2);
112     tx7.vout[0].scriptPubKey = CScript() << OP_7 << OP_EQUAL;
113     tx7.vout[0].nValue = 10 * COIN;
114     tx7.vout[1].scriptPubKey = CScript() << OP_7 << OP_EQUAL;
115     tx7.vout[1].nValue = 10 * COIN;
116 
117     CTxMemPool pool;
118     LOCK2(cs_main, pool.cs);
119     // Create transaction references outside the "hot loop"
120     const CTransactionRef tx1_r{MakeTransactionRef(tx1)};
121     const CTransactionRef tx2_r{MakeTransactionRef(tx2)};
122     const CTransactionRef tx3_r{MakeTransactionRef(tx3)};
123     const CTransactionRef tx4_r{MakeTransactionRef(tx4)};
124     const CTransactionRef tx5_r{MakeTransactionRef(tx5)};
125     const CTransactionRef tx6_r{MakeTransactionRef(tx6)};
126     const CTransactionRef tx7_r{MakeTransactionRef(tx7)};
127 
128     bench.run([&]() NO_THREAD_SAFETY_ANALYSIS {
129         AddTx(tx1_r, 10000LL, pool);
130         AddTx(tx2_r, 5000LL, pool);
131         AddTx(tx3_r, 20000LL, pool);
132         AddTx(tx4_r, 7000LL, pool);
133         AddTx(tx5_r, 1000LL, pool);
134         AddTx(tx6_r, 1100LL, pool);
135         AddTx(tx7_r, 9000LL, pool);
136         pool.TrimToSize(pool.DynamicMemoryUsage() * 3 / 4);
137         pool.TrimToSize(GetVirtualTransactionSize(*tx1_r));
138     });
139 }
140 
141 BENCHMARK(MempoolEviction);
142