Skip to content

Commit 190b8d2

Browse files
committed
Make BaseSignatureCreator a pure interface
1 parent 8d651ae commit 190b8d2

File tree

7 files changed

+39
-45
lines changed

7 files changed

+39
-45
lines changed

src/bitcoin-tx.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -644,7 +644,7 @@ static void MutateTxSign(CMutableTransaction& tx, const std::string& flagStr)
644644
SignatureData sigdata;
645645
// Only sign SIGHASH_SINGLE if there's a corresponding output:
646646
if (!fHashSingle || (i < mergedTx.vout.size()))
647-
ProduceSignature(MutableTransactionSignatureCreator(&keystore, &mergedTx, i, amount, nHashType), prevPubKey, sigdata);
647+
ProduceSignature(keystore, MutableTransactionSignatureCreator(&mergedTx, i, amount, nHashType), prevPubKey, sigdata);
648648

649649
// ... and merge in other signatures:
650650
for (const CTransaction& txv : txVariants)

src/rpc/rawtransaction.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -831,7 +831,7 @@ UniValue SignTransaction(CMutableTransaction& mtx, const UniValue& prevTxsUnival
831831
SignatureData sigdata;
832832
// Only sign SIGHASH_SINGLE if there's a corresponding output:
833833
if (!fHashSingle || (i < mtx.vout.size())) {
834-
ProduceSignature(MutableTransactionSignatureCreator(keystore, &mtx, i, amount, nHashType), prevPubKey, sigdata);
834+
ProduceSignature(*keystore, MutableTransactionSignatureCreator(&mtx, i, amount, nHashType), prevPubKey, sigdata);
835835
}
836836
sigdata = CombineSignatures(prevPubKey, TransactionSignatureChecker(&txConst, i, amount), sigdata, DataFromTransaction(mtx, i));
837837

src/script/ismine.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ isminetype IsMine(const CKeyStore &keystore, const CScript& scriptPubKey, bool&
146146
if (keystore.HaveWatchOnly(scriptPubKey)) {
147147
// TODO: This could be optimized some by doing some work after the above solver
148148
SignatureData sigs;
149-
return ProduceSignature(DummySignatureCreator(&keystore), scriptPubKey, sigs) ? ISMINE_WATCH_SOLVABLE : ISMINE_WATCH_UNSOLVABLE;
149+
return ProduceSignature(keystore, DummySignatureCreator(), scriptPubKey, sigs) ? ISMINE_WATCH_SOLVABLE : ISMINE_WATCH_UNSOLVABLE;
150150
}
151151
return ISMINE_NO;
152152
}

src/script/sign.cpp

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,12 @@
1414

1515
typedef std::vector<unsigned char> valtype;
1616

17-
TransactionSignatureCreator::TransactionSignatureCreator(const SigningProvider* provider, const CTransaction* txToIn, unsigned int nInIn, const CAmount& amountIn, int nHashTypeIn) : BaseSignatureCreator(provider), txTo(txToIn), nIn(nInIn), nHashType(nHashTypeIn), amount(amountIn), checker(txTo, nIn, amountIn) {}
17+
TransactionSignatureCreator::TransactionSignatureCreator(const CTransaction* txToIn, unsigned int nInIn, const CAmount& amountIn, int nHashTypeIn) : txTo(txToIn), nIn(nInIn), nHashType(nHashTypeIn), amount(amountIn), checker(txTo, nIn, amountIn) {}
1818

19-
bool TransactionSignatureCreator::CreateSig(std::vector<unsigned char>& vchSig, const CKeyID& address, const CScript& scriptCode, SigVersion sigversion) const
19+
bool TransactionSignatureCreator::CreateSig(const SigningProvider& provider, std::vector<unsigned char>& vchSig, const CKeyID& address, const CScript& scriptCode, SigVersion sigversion) const
2020
{
2121
CKey key;
22-
if (!m_provider->GetKey(address, key))
22+
if (!provider.GetKey(address, key))
2323
return false;
2424

2525
// Signing with uncompressed keys is disabled in witness scripts
@@ -33,24 +33,24 @@ bool TransactionSignatureCreator::CreateSig(std::vector<unsigned char>& vchSig,
3333
return true;
3434
}
3535

36-
static bool Sign1(const CKeyID& address, const BaseSignatureCreator& creator, const CScript& scriptCode, std::vector<valtype>& ret, SigVersion sigversion)
36+
static bool Sign1(const SigningProvider& provider, const CKeyID& address, const BaseSignatureCreator& creator, const CScript& scriptCode, std::vector<valtype>& ret, SigVersion sigversion)
3737
{
3838
std::vector<unsigned char> vchSig;
39-
if (!creator.CreateSig(vchSig, address, scriptCode, sigversion))
39+
if (!creator.CreateSig(provider, vchSig, address, scriptCode, sigversion))
4040
return false;
4141
ret.push_back(vchSig);
4242
return true;
4343
}
4444

45-
static bool SignN(const std::vector<valtype>& multisigdata, const BaseSignatureCreator& creator, const CScript& scriptCode, std::vector<valtype>& ret, SigVersion sigversion)
45+
static bool SignN(const SigningProvider& provider, const std::vector<valtype>& multisigdata, const BaseSignatureCreator& creator, const CScript& scriptCode, std::vector<valtype>& ret, SigVersion sigversion)
4646
{
4747
int nSigned = 0;
4848
int nRequired = multisigdata.front()[0];
4949
for (unsigned int i = 1; i < multisigdata.size()-1 && nSigned < nRequired; i++)
5050
{
5151
const valtype& pubkey = multisigdata[i];
5252
CKeyID keyID = CPubKey(pubkey).GetID();
53-
if (Sign1(keyID, creator, scriptCode, ret, sigversion))
53+
if (Sign1(provider, keyID, creator, scriptCode, ret, sigversion))
5454
++nSigned;
5555
}
5656
return nSigned==nRequired;
@@ -62,7 +62,7 @@ static bool SignN(const std::vector<valtype>& multisigdata, const BaseSignatureC
6262
* unless whichTypeRet is TX_SCRIPTHASH, in which case scriptSigRet is the redemption script.
6363
* Returns false if scriptPubKey could not be completely satisfied.
6464
*/
65-
static bool SignStep(const BaseSignatureCreator& creator, const CScript& scriptPubKey,
65+
static bool SignStep(const SigningProvider& provider, const BaseSignatureCreator& creator, const CScript& scriptPubKey,
6666
std::vector<valtype>& ret, txnouttype& whichTypeRet, SigVersion sigversion)
6767
{
6868
CScript scriptRet;
@@ -82,36 +82,36 @@ static bool SignStep(const BaseSignatureCreator& creator, const CScript& scriptP
8282
return false;
8383
case TX_PUBKEY:
8484
keyID = CPubKey(vSolutions[0]).GetID();
85-
return Sign1(keyID, creator, scriptPubKey, ret, sigversion);
85+
return Sign1(provider, keyID, creator, scriptPubKey, ret, sigversion);
8686
case TX_PUBKEYHASH:
8787
keyID = CKeyID(uint160(vSolutions[0]));
88-
if (!Sign1(keyID, creator, scriptPubKey, ret, sigversion))
88+
if (!Sign1(provider, keyID, creator, scriptPubKey, ret, sigversion))
8989
return false;
9090
else
9191
{
9292
CPubKey vch;
93-
creator.Provider().GetPubKey(keyID, vch);
93+
provider.GetPubKey(keyID, vch);
9494
ret.push_back(ToByteVector(vch));
9595
}
9696
return true;
9797
case TX_SCRIPTHASH:
98-
if (creator.Provider().GetCScript(uint160(vSolutions[0]), scriptRet)) {
98+
if (provider.GetCScript(uint160(vSolutions[0]), scriptRet)) {
9999
ret.push_back(std::vector<unsigned char>(scriptRet.begin(), scriptRet.end()));
100100
return true;
101101
}
102102
return false;
103103

104104
case TX_MULTISIG:
105105
ret.push_back(valtype()); // workaround CHECKMULTISIG bug
106-
return (SignN(vSolutions, creator, scriptPubKey, ret, sigversion));
106+
return (SignN(provider, vSolutions, creator, scriptPubKey, ret, sigversion));
107107

108108
case TX_WITNESS_V0_KEYHASH:
109109
ret.push_back(vSolutions[0]);
110110
return true;
111111

112112
case TX_WITNESS_V0_SCRIPTHASH:
113113
CRIPEMD160().Write(&vSolutions[0][0], vSolutions[0].size()).Finalize(h160.begin());
114-
if (creator.Provider().GetCScript(h160, scriptRet)) {
114+
if (provider.GetCScript(h160, scriptRet)) {
115115
ret.push_back(std::vector<unsigned char>(scriptRet.begin(), scriptRet.end()));
116116
return true;
117117
}
@@ -137,11 +137,11 @@ static CScript PushAll(const std::vector<valtype>& values)
137137
return result;
138138
}
139139

140-
bool ProduceSignature(const BaseSignatureCreator& creator, const CScript& fromPubKey, SignatureData& sigdata)
140+
bool ProduceSignature(const SigningProvider& provider, const BaseSignatureCreator& creator, const CScript& fromPubKey, SignatureData& sigdata)
141141
{
142142
std::vector<valtype> result;
143143
txnouttype whichType;
144-
bool solved = SignStep(creator, fromPubKey, result, whichType, SigVersion::BASE);
144+
bool solved = SignStep(provider, creator, fromPubKey, result, whichType, SigVersion::BASE);
145145
bool P2SH = false;
146146
CScript subscript;
147147
sigdata.scriptWitness.stack.clear();
@@ -152,7 +152,7 @@ bool ProduceSignature(const BaseSignatureCreator& creator, const CScript& fromPu
152152
// the final scriptSig is the signatures from that
153153
// and then the serialized subscript:
154154
subscript = CScript(result[0].begin(), result[0].end());
155-
solved = solved && SignStep(creator, subscript, result, whichType, SigVersion::BASE) && whichType != TX_SCRIPTHASH;
155+
solved = solved && SignStep(provider, creator, subscript, result, whichType, SigVersion::BASE) && whichType != TX_SCRIPTHASH;
156156
P2SH = true;
157157
}
158158

@@ -161,15 +161,15 @@ bool ProduceSignature(const BaseSignatureCreator& creator, const CScript& fromPu
161161
CScript witnessscript;
162162
witnessscript << OP_DUP << OP_HASH160 << ToByteVector(result[0]) << OP_EQUALVERIFY << OP_CHECKSIG;
163163
txnouttype subType;
164-
solved = solved && SignStep(creator, witnessscript, result, subType, SigVersion::WITNESS_V0);
164+
solved = solved && SignStep(provider, creator, witnessscript, result, subType, SigVersion::WITNESS_V0);
165165
sigdata.scriptWitness.stack = result;
166166
result.clear();
167167
}
168168
else if (solved && whichType == TX_WITNESS_V0_SCRIPTHASH)
169169
{
170170
CScript witnessscript(result[0].begin(), result[0].end());
171171
txnouttype subType;
172-
solved = solved && SignStep(creator, witnessscript, result, subType, SigVersion::WITNESS_V0) && subType != TX_SCRIPTHASH && subType != TX_WITNESS_V0_SCRIPTHASH && subType != TX_WITNESS_V0_KEYHASH;
172+
solved = solved && SignStep(provider, creator, witnessscript, result, subType, SigVersion::WITNESS_V0) && subType != TX_SCRIPTHASH && subType != TX_WITNESS_V0_SCRIPTHASH && subType != TX_WITNESS_V0_KEYHASH;
173173
result.push_back(std::vector<unsigned char>(witnessscript.begin(), witnessscript.end()));
174174
sigdata.scriptWitness.stack = result;
175175
result.clear();
@@ -210,10 +210,10 @@ bool SignSignature(const SigningProvider &provider, const CScript& fromPubKey, C
210210
assert(nIn < txTo.vin.size());
211211

212212
CTransaction txToConst(txTo);
213-
TransactionSignatureCreator creator(&provider, &txToConst, nIn, amount, nHashType);
213+
TransactionSignatureCreator creator(&txToConst, nIn, amount, nHashType);
214214

215215
SignatureData sigdata;
216-
bool ret = ProduceSignature(creator, fromPubKey, sigdata);
216+
bool ret = ProduceSignature(provider, creator, fromPubKey, sigdata);
217217
UpdateTransaction(txTo, nIn, sigdata);
218218
return ret;
219219
}
@@ -410,7 +410,7 @@ const BaseSignatureChecker& DummySignatureCreator::Checker() const
410410
return dummyChecker;
411411
}
412412

413-
bool DummySignatureCreator::CreateSig(std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const
413+
bool DummySignatureCreator::CreateSig(const SigningProvider& provider, std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const
414414
{
415415
// Create a dummy signature that is a valid DER-encoding
416416
vchSig.assign(72, '\000');
@@ -432,12 +432,12 @@ bool IsSolvable(const SigningProvider& provider, const CScript& script)
432432
// if we were to have the private keys. This is just to make sure that the script is valid and that,
433433
// if found in a transaction, we would still accept and relay that transaction. In particular,
434434
// it will reject witness outputs that require signing with an uncompressed public key.
435-
DummySignatureCreator creator(&provider);
435+
static const DummySignatureCreator creator;
436436
SignatureData sigs;
437437
// Make sure that STANDARD_SCRIPT_VERIFY_FLAGS includes SCRIPT_VERIFY_WITNESS_PUBKEYTYPE, the most
438438
// important property this function is designed to test for.
439439
static_assert(STANDARD_SCRIPT_VERIFY_FLAGS & SCRIPT_VERIFY_WITNESS_PUBKEYTYPE, "IsSolvable requires standard script flags to include WITNESS_PUBKEYTYPE");
440-
if (ProduceSignature(creator, script, sigs)) {
440+
if (ProduceSignature(provider, creator, script, sigs)) {
441441
// VerifyScript check is just defensive, and should never fail.
442442
assert(VerifyScript(sigs.scriptSig, script, &sigs.scriptWitness, STANDARD_SCRIPT_VERIFY_FLAGS, creator.Checker()));
443443
return true;

src/script/sign.h

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -26,19 +26,14 @@ class SigningProvider
2626
virtual bool GetKey(const CKeyID &address, CKey& key) const =0;
2727
};
2828

29-
/** Virtual base class for signature creators. */
29+
/** Interface for signature creators. */
3030
class BaseSignatureCreator {
31-
protected:
32-
const SigningProvider* m_provider;
33-
3431
public:
35-
explicit BaseSignatureCreator(const SigningProvider* provider) : m_provider(provider) {}
36-
const SigningProvider& Provider() const { return *m_provider; }
3732
virtual ~BaseSignatureCreator() {}
3833
virtual const BaseSignatureChecker& Checker() const =0;
3934

4035
/** Create a singular (non-script) signature. */
41-
virtual bool CreateSig(std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const =0;
36+
virtual bool CreateSig(const SigningProvider& provider, std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const =0;
4237
};
4338

4439
/** A signature creator for transactions. */
@@ -50,24 +45,23 @@ class TransactionSignatureCreator : public BaseSignatureCreator {
5045
const TransactionSignatureChecker checker;
5146

5247
public:
53-
TransactionSignatureCreator(const SigningProvider* provider, const CTransaction* txToIn, unsigned int nInIn, const CAmount& amountIn, int nHashTypeIn=SIGHASH_ALL);
48+
TransactionSignatureCreator(const CTransaction* txToIn, unsigned int nInIn, const CAmount& amountIn, int nHashTypeIn=SIGHASH_ALL);
5449
const BaseSignatureChecker& Checker() const override { return checker; }
55-
bool CreateSig(std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const override;
50+
bool CreateSig(const SigningProvider& provider, std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const override;
5651
};
5752

5853
class MutableTransactionSignatureCreator : public TransactionSignatureCreator {
5954
CTransaction tx;
6055

6156
public:
62-
MutableTransactionSignatureCreator(const SigningProvider* provider, const CMutableTransaction* txToIn, unsigned int nInIn, const CAmount& amountIn, int nHashTypeIn) : TransactionSignatureCreator(provider, &tx, nInIn, amountIn, nHashTypeIn), tx(*txToIn) {}
57+
MutableTransactionSignatureCreator(const CMutableTransaction* txToIn, unsigned int nInIn, const CAmount& amountIn, int nHashTypeIn) : TransactionSignatureCreator(&tx, nInIn, amountIn, nHashTypeIn), tx(*txToIn) {}
6358
};
6459

6560
/** A signature creator that just produces 72-byte empty signatures. */
6661
class DummySignatureCreator : public BaseSignatureCreator {
6762
public:
68-
explicit DummySignatureCreator(const SigningProvider* provider) : BaseSignatureCreator(provider) {}
6963
const BaseSignatureChecker& Checker() const override;
70-
bool CreateSig(std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const override;
64+
bool CreateSig(const SigningProvider& provider, std::vector<unsigned char>& vchSig, const CKeyID& keyid, const CScript& scriptCode, SigVersion sigversion) const override;
7165
};
7266

7367
struct SignatureData {
@@ -79,7 +73,7 @@ struct SignatureData {
7973
};
8074

8175
/** Produce a script signature using a generic signature creator. */
82-
bool ProduceSignature(const BaseSignatureCreator& creator, const CScript& scriptPubKey, SignatureData& sigdata);
76+
bool ProduceSignature(const SigningProvider& provider, const BaseSignatureCreator& creator, const CScript& scriptPubKey, SignatureData& sigdata);
8377

8478
/** Produce a script signature for a transaction. */
8579
bool SignSignature(const SigningProvider &provider, const CScript& fromPubKey, CMutableTransaction& txTo, unsigned int nIn, const CAmount& amount, int nHashType);

src/test/txvalidationcache_tests.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ BOOST_FIXTURE_TEST_CASE(checkinputs_test, TestChain100Setup)
314314

315315
// Sign
316316
SignatureData sigdata;
317-
ProduceSignature(MutableTransactionSignatureCreator(&keystore, &valid_with_witness_tx, 0, 11*CENT, SIGHASH_ALL), spend_tx.vout[1].scriptPubKey, sigdata);
317+
ProduceSignature(keystore, MutableTransactionSignatureCreator(&valid_with_witness_tx, 0, 11*CENT, SIGHASH_ALL), spend_tx.vout[1].scriptPubKey, sigdata);
318318
UpdateTransaction(valid_with_witness_tx, 0, sigdata);
319319

320320
// This should be valid under all script flags.
@@ -342,7 +342,7 @@ BOOST_FIXTURE_TEST_CASE(checkinputs_test, TestChain100Setup)
342342
// Sign
343343
for (int i=0; i<2; ++i) {
344344
SignatureData sigdata;
345-
ProduceSignature(MutableTransactionSignatureCreator(&keystore, &tx, i, 11*CENT, SIGHASH_ALL), spend_tx.vout[i].scriptPubKey, sigdata);
345+
ProduceSignature(keystore, MutableTransactionSignatureCreator(&tx, i, 11*CENT, SIGHASH_ALL), spend_tx.vout[i].scriptPubKey, sigdata);
346346
UpdateTransaction(tx, i, sigdata);
347347
}
348348

src/wallet/wallet.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1535,7 +1535,7 @@ bool CWallet::DummySignInput(CTxIn &tx_in, const CTxOut &txout) const
15351535
const CScript& scriptPubKey = txout.scriptPubKey;
15361536
SignatureData sigdata;
15371537

1538-
if (!ProduceSignature(DummySignatureCreator(this), scriptPubKey, sigdata))
1538+
if (!ProduceSignature(*this, DummySignatureCreator(), scriptPubKey, sigdata))
15391539
{
15401540
return false;
15411541
} else {
@@ -2579,7 +2579,7 @@ bool CWallet::SignTransaction(CMutableTransaction &tx)
25792579
const CScript& scriptPubKey = mi->second.tx->vout[input.prevout.n].scriptPubKey;
25802580
const CAmount& amount = mi->second.tx->vout[input.prevout.n].nValue;
25812581
SignatureData sigdata;
2582-
if (!ProduceSignature(TransactionSignatureCreator(this, &txNewConst, nIn, amount, SIGHASH_ALL), scriptPubKey, sigdata)) {
2582+
if (!ProduceSignature(*this, TransactionSignatureCreator(&txNewConst, nIn, amount, SIGHASH_ALL), scriptPubKey, sigdata)) {
25832583
return false;
25842584
}
25852585
UpdateTransaction(tx, nIn, sigdata);
@@ -3008,7 +3008,7 @@ bool CWallet::CreateTransaction(const std::vector<CRecipient>& vecSend, CTransac
30083008
const CScript& scriptPubKey = coin.txout.scriptPubKey;
30093009
SignatureData sigdata;
30103010

3011-
if (!ProduceSignature(TransactionSignatureCreator(this, &txNewConst, nIn, coin.txout.nValue, SIGHASH_ALL), scriptPubKey, sigdata))
3011+
if (!ProduceSignature(*this, TransactionSignatureCreator(&txNewConst, nIn, coin.txout.nValue, SIGHASH_ALL), scriptPubKey, sigdata))
30123012
{
30133013
strFailReason = _("Signing transaction failed");
30143014
return false;

0 commit comments

Comments
 (0)