Gavin Andresen
13 years ago
21 changed files with 824 additions and 544 deletions
@ -0,0 +1,74 @@
@@ -0,0 +1,74 @@
|
||||
#include <boost/test/unit_test.hpp> |
||||
#include <boost/foreach.hpp> |
||||
|
||||
#include "base58.h" |
||||
#include "util.h" |
||||
#include "json/json_spirit_reader_template.h" |
||||
#include "json/json_spirit_writer_template.h" |
||||
#include "json/json_spirit_utils.h" |
||||
|
||||
using namespace std; |
||||
using namespace json_spirit; |
||||
|
||||
typedef Value(*rpcfn_type)(const Array& params, bool fHelp); |
||||
extern map<string, rpcfn_type> mapCallTable; |
||||
|
||||
BOOST_AUTO_TEST_SUITE(rpc_tests) |
||||
|
||||
static Array |
||||
createArgs(int nRequired, const char* address1=NULL, const char* address2=NULL) |
||||
{ |
||||
Array result; |
||||
result.push_back(nRequired); |
||||
Array addresses; |
||||
if (address1) addresses.push_back(address1); |
||||
if (address2) addresses.push_back(address1); |
||||
result.push_back(addresses); |
||||
return result; |
||||
} |
||||
|
||||
// This can be removed this when addmultisigaddress is enabled on main net:
|
||||
struct TestNetFixture |
||||
{ |
||||
TestNetFixture() { fTestNet = true; } |
||||
~TestNetFixture() { fTestNet = false; } |
||||
}; |
||||
|
||||
BOOST_FIXTURE_TEST_CASE(rpc_addmultisig, TestNetFixture) |
||||
{ |
||||
rpcfn_type addmultisig = mapCallTable["addmultisigaddress"]; |
||||
|
||||
// old, 65-byte-long:
|
||||
const char* address1Hex = "0434e3e09f49ea168c5bbf53f877ff4206923858aab7c7e1df25bc263978107c95e35065a27ef6f1b27222db0ec97e0e895eaca603d3ee0d4c060ce3d8a00286c8"; |
||||
// new, compressed:
|
||||
const char* address2Hex = "0388c2037017c62240b6b72ac1a2a5f94da790596ebd06177c8572752922165cb4"; |
||||
|
||||
Value v; |
||||
CBitcoinAddress address; |
||||
BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex), false)); |
||||
address.SetString(v.get_str()); |
||||
BOOST_CHECK(address.IsValid() && address.IsScript()); |
||||
|
||||
BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(1, address1Hex, address2Hex), false)); |
||||
address.SetString(v.get_str()); |
||||
BOOST_CHECK(address.IsValid() && address.IsScript()); |
||||
|
||||
BOOST_CHECK_NO_THROW(v = addmultisig(createArgs(2, address1Hex, address2Hex), false)); |
||||
address.SetString(v.get_str()); |
||||
BOOST_CHECK(address.IsValid() && address.IsScript()); |
||||
|
||||
BOOST_CHECK_THROW(addmultisig(createArgs(0), false), runtime_error); |
||||
BOOST_CHECK_THROW(addmultisig(createArgs(1), false), runtime_error); |
||||
BOOST_CHECK_THROW(addmultisig(createArgs(2, address1Hex), false), runtime_error); |
||||
|
||||
BOOST_CHECK_THROW(addmultisig(createArgs(1, ""), false), runtime_error); |
||||
BOOST_CHECK_THROW(addmultisig(createArgs(1, "NotAValidPubkey"), false), runtime_error); |
||||
|
||||
string short1(address1Hex, address1Hex+sizeof(address1Hex)-2); // last byte missing
|
||||
BOOST_CHECK_THROW(addmultisig(createArgs(2, short1.c_str()), false), runtime_error); |
||||
|
||||
string short2(address1Hex+2, address1Hex+sizeof(address1Hex)); // first byte missing
|
||||
BOOST_CHECK_THROW(addmultisig(createArgs(2, short2.c_str()), false), runtime_error); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_SUITE_END() |
||||
@ -0,0 +1,310 @@
@@ -0,0 +1,310 @@
|
||||
#include <boost/assert.hpp> |
||||
#include <boost/assign/list_of.hpp> |
||||
#include <boost/assign/list_inserter.hpp> |
||||
#include <boost/assign/std/vector.hpp> |
||||
#include <boost/test/unit_test.hpp> |
||||
#include <boost/foreach.hpp> |
||||
|
||||
#include "../main.h" |
||||
#include "../script.h" |
||||
#include "../wallet.h" |
||||
|
||||
using namespace std; |
||||
|
||||
// Test routines internal to script.cpp:
|
||||
extern uint256 SignatureHash(CScript scriptCode, const CTransaction& txTo, unsigned int nIn, int nHashType); |
||||
extern bool VerifyScript(const CScript& scriptSig, const CScript& scriptPubKey, const CTransaction& txTo, unsigned int nIn, |
||||
bool fValidatePayToScriptHash, int nHashType); |
||||
|
||||
// Helpers:
|
||||
static std::vector<unsigned char> |
||||
Serialize(const CScript& s) |
||||
{ |
||||
std::vector<unsigned char> sSerialized(s); |
||||
return sSerialized; |
||||
} |
||||
|
||||
static bool |
||||
Verify(const CScript& scriptSig, const CScript& scriptPubKey, bool fStrict) |
||||
{ |
||||
// Create dummy to/from transactions:
|
||||
CTransaction txFrom; |
||||
txFrom.vout.resize(1); |
||||
txFrom.vout[0].scriptPubKey = scriptPubKey; |
||||
|
||||
CTransaction txTo; |
||||
txTo.vin.resize(1); |
||||
txTo.vout.resize(1); |
||||
txTo.vin[0].prevout.n = 0; |
||||
txTo.vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo.vin[0].scriptSig = scriptSig; |
||||
txTo.vout[0].nValue = 1; |
||||
|
||||
return VerifyScript(scriptSig, scriptPubKey, txTo, 0, fStrict, 0); |
||||
} |
||||
|
||||
|
||||
BOOST_AUTO_TEST_SUITE(script_P2SH_tests) |
||||
|
||||
BOOST_AUTO_TEST_CASE(sign) |
||||
{ |
||||
// Pay-to-script-hash looks like this:
|
||||
// scriptSig: <sig> <sig...> <serialized_script>
|
||||
// scriptPubKey: HASH160 <hash> EQUAL
|
||||
|
||||
// Test SignSignature() (and therefore the version of Solver() that signs transactions)
|
||||
CBasicKeyStore keystore; |
||||
CKey key[4]; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
key[i].MakeNewKey(); |
||||
keystore.AddKey(key[i]); |
||||
} |
||||
|
||||
// 8 Scripts: checking all combinations of
|
||||
// different keys, straight/P2SH, pubkey/pubkeyhash
|
||||
CScript standardScripts[4]; |
||||
standardScripts[0] << key[0].GetPubKey() << OP_CHECKSIG; |
||||
standardScripts[1].SetBitcoinAddress(key[1].GetPubKey()); |
||||
standardScripts[2] << key[1].GetPubKey() << OP_CHECKSIG; |
||||
standardScripts[3].SetBitcoinAddress(key[2].GetPubKey()); |
||||
CScript evalScripts[4]; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
keystore.AddCScript(standardScripts[i]); |
||||
evalScripts[i].SetPayToScriptHash(standardScripts[i]); |
||||
} |
||||
|
||||
CTransaction txFrom; // Funding transaction:
|
||||
txFrom.vout.resize(8); |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
txFrom.vout[i].scriptPubKey = evalScripts[i]; |
||||
txFrom.vout[i+4].scriptPubKey = standardScripts[i]; |
||||
} |
||||
BOOST_CHECK(txFrom.IsStandard()); |
||||
|
||||
CTransaction txTo[8]; // Spending transactions
|
||||
for (int i = 0; i < 8; i++) |
||||
{ |
||||
txTo[i].vin.resize(1); |
||||
txTo[i].vout.resize(1); |
||||
txTo[i].vin[0].prevout.n = i; |
||||
txTo[i].vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo[i].vout[0].nValue = 1; |
||||
BOOST_CHECK_MESSAGE(IsMine(keystore, txFrom.vout[i].scriptPubKey), strprintf("IsMine %d", i)); |
||||
} |
||||
for (int i = 0; i < 8; i++) |
||||
{ |
||||
BOOST_CHECK_MESSAGE(SignSignature(keystore, txFrom, txTo[i], 0), strprintf("SignSignature %d", i)); |
||||
} |
||||
// All of the above should be OK, and the txTos have valid signatures
|
||||
// Check to make sure signature verification fails if we use the wrong ScriptSig:
|
||||
for (int i = 0; i < 8; i++) |
||||
for (int j = 0; j < 8; j++) |
||||
{ |
||||
CScript sigSave = txTo[i].vin[0].scriptSig; |
||||
txTo[i].vin[0].scriptSig = txTo[j].vin[0].scriptSig; |
||||
bool sigOK = VerifySignature(txFrom, txTo[i], 0, true, 0); |
||||
if (i == j) |
||||
BOOST_CHECK_MESSAGE(sigOK, strprintf("VerifySignature %d %d", i, j)); |
||||
else |
||||
BOOST_CHECK_MESSAGE(!sigOK, strprintf("VerifySignature %d %d", i, j)); |
||||
txTo[i].vin[0].scriptSig = sigSave; |
||||
} |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(norecurse) |
||||
{ |
||||
// Make sure only the outer pay-to-script-hash does the
|
||||
// extra-validation thing:
|
||||
CScript invalidAsScript; |
||||
invalidAsScript << OP_INVALIDOPCODE << OP_INVALIDOPCODE; |
||||
|
||||
CScript p2sh; |
||||
p2sh.SetPayToScriptHash(invalidAsScript); |
||||
|
||||
CScript scriptSig; |
||||
scriptSig << Serialize(invalidAsScript); |
||||
|
||||
// Should not verify, because it will try to execute OP_INVALIDOPCODE
|
||||
BOOST_CHECK(!Verify(scriptSig, p2sh, true)); |
||||
|
||||
// Try to recurse, and verification should succeed because
|
||||
// the inner HASH160 <> EQUAL should only check the hash:
|
||||
CScript p2sh2; |
||||
p2sh2.SetPayToScriptHash(p2sh); |
||||
CScript scriptSig2; |
||||
scriptSig2 << Serialize(invalidAsScript) << Serialize(p2sh); |
||||
|
||||
BOOST_CHECK(Verify(scriptSig2, p2sh2, true)); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(set) |
||||
{ |
||||
// Test the CScript::Set* methods
|
||||
CBasicKeyStore keystore; |
||||
CKey key[4]; |
||||
std::vector<CKey> keys; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
key[i].MakeNewKey(); |
||||
keystore.AddKey(key[i]); |
||||
keys.push_back(key[i]); |
||||
} |
||||
|
||||
CScript inner[4]; |
||||
inner[0].SetBitcoinAddress(key[0].GetPubKey()); |
||||
inner[1].SetMultisig(2, std::vector<CKey>(keys.begin(), keys.begin()+2)); |
||||
inner[2].SetMultisig(1, std::vector<CKey>(keys.begin(), keys.begin()+2)); |
||||
inner[3].SetMultisig(2, std::vector<CKey>(keys.begin(), keys.begin()+3)); |
||||
|
||||
CScript outer[4]; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
outer[i].SetPayToScriptHash(inner[i]); |
||||
keystore.AddCScript(inner[i]); |
||||
} |
||||
|
||||
CTransaction txFrom; // Funding transaction:
|
||||
txFrom.vout.resize(4); |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
txFrom.vout[i].scriptPubKey = outer[i]; |
||||
} |
||||
BOOST_CHECK(txFrom.IsStandard()); |
||||
|
||||
CTransaction txTo[4]; // Spending transactions
|
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
txTo[i].vin.resize(1); |
||||
txTo[i].vout.resize(1); |
||||
txTo[i].vin[0].prevout.n = i; |
||||
txTo[i].vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo[i].vout[0].nValue = 1; |
||||
txTo[i].vout[0].scriptPubKey = inner[i]; |
||||
BOOST_CHECK_MESSAGE(IsMine(keystore, txFrom.vout[i].scriptPubKey), strprintf("IsMine %d", i)); |
||||
} |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
BOOST_CHECK_MESSAGE(SignSignature(keystore, txFrom, txTo[i], 0), strprintf("SignSignature %d", i)); |
||||
BOOST_CHECK_MESSAGE(txTo[i].IsStandard(), strprintf("txTo[%d].IsStandard", i)); |
||||
} |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(is) |
||||
{ |
||||
// Test CScript::IsPayToScriptHash()
|
||||
uint160 dummy; |
||||
CScript p2sh; |
||||
p2sh << OP_HASH160 << dummy << OP_EQUAL; |
||||
BOOST_CHECK(p2sh.IsPayToScriptHash()); |
||||
|
||||
// Not considered pay-to-script-hash if using one of the OP_PUSHDATA opcodes:
|
||||
static const unsigned char direct[] = { OP_HASH160, 20, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, OP_EQUAL }; |
||||
BOOST_CHECK(CScript(direct, direct+sizeof(direct)).IsPayToScriptHash()); |
||||
static const unsigned char pushdata1[] = { OP_HASH160, OP_PUSHDATA1, 20, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, OP_EQUAL }; |
||||
BOOST_CHECK(!CScript(pushdata1, pushdata1+sizeof(pushdata1)).IsPayToScriptHash()); |
||||
static const unsigned char pushdata2[] = { OP_HASH160, OP_PUSHDATA2, 20,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, OP_EQUAL }; |
||||
BOOST_CHECK(!CScript(pushdata2, pushdata2+sizeof(pushdata2)).IsPayToScriptHash()); |
||||
static const unsigned char pushdata4[] = { OP_HASH160, OP_PUSHDATA4, 20,0,0,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, OP_EQUAL }; |
||||
BOOST_CHECK(!CScript(pushdata4, pushdata4+sizeof(pushdata4)).IsPayToScriptHash()); |
||||
|
||||
CScript not_p2sh; |
||||
BOOST_CHECK(!not_p2sh.IsPayToScriptHash()); |
||||
|
||||
not_p2sh.clear(); not_p2sh << OP_HASH160 << dummy << dummy << OP_EQUAL; |
||||
BOOST_CHECK(!not_p2sh.IsPayToScriptHash()); |
||||
|
||||
not_p2sh.clear(); not_p2sh << OP_NOP << dummy << OP_EQUAL; |
||||
BOOST_CHECK(!not_p2sh.IsPayToScriptHash()); |
||||
|
||||
not_p2sh.clear(); not_p2sh << OP_HASH160 << dummy << OP_CHECKSIG; |
||||
BOOST_CHECK(!not_p2sh.IsPayToScriptHash()); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(switchover) |
||||
{ |
||||
// Test switchover code
|
||||
CScript notValid; |
||||
notValid << OP_11 << OP_12 << OP_EQUALVERIFY; |
||||
CScript scriptSig; |
||||
scriptSig << Serialize(notValid); |
||||
|
||||
CScript fund; |
||||
fund.SetPayToScriptHash(notValid); |
||||
|
||||
|
||||
// Validation should succeed under old rules (hash is correct):
|
||||
BOOST_CHECK(Verify(scriptSig, fund, false)); |
||||
// Fail under new:
|
||||
BOOST_CHECK(!Verify(scriptSig, fund, true)); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(AreInputsStandard) |
||||
{ |
||||
std::map<uint256, std::pair<CTxIndex, CTransaction> > mapInputs; |
||||
CBasicKeyStore keystore; |
||||
CKey key[3]; |
||||
vector<CKey> keys; |
||||
for (int i = 0; i < 3; i++) |
||||
{ |
||||
key[i].MakeNewKey(); |
||||
keystore.AddKey(key[i]); |
||||
keys.push_back(key[i]); |
||||
} |
||||
|
||||
CTransaction txFrom; |
||||
txFrom.vout.resize(5); |
||||
|
||||
// First three are standard:
|
||||
CScript pay1; pay1.SetBitcoinAddress(key[0].GetPubKey()); |
||||
keystore.AddCScript(pay1); |
||||
CScript payScriptHash1; payScriptHash1.SetPayToScriptHash(pay1); |
||||
CScript pay1of3; pay1of3.SetMultisig(1, keys); |
||||
|
||||
txFrom.vout[0].scriptPubKey = payScriptHash1; |
||||
txFrom.vout[1].scriptPubKey = pay1; |
||||
txFrom.vout[2].scriptPubKey = pay1of3; |
||||
|
||||
// Last two non-standard:
|
||||
CScript empty; |
||||
keystore.AddCScript(empty); |
||||
txFrom.vout[3].scriptPubKey = empty; |
||||
// Can't use SetPayToScriptHash, it checks for the empty Script. So:
|
||||
txFrom.vout[4].scriptPubKey << OP_HASH160 << Hash160(empty) << OP_EQUAL; |
||||
|
||||
mapInputs[txFrom.GetHash()] = make_pair(CTxIndex(), txFrom); |
||||
|
||||
CTransaction txTo; |
||||
txTo.vout.resize(1); |
||||
txTo.vout[0].scriptPubKey.SetBitcoinAddress(key[1].GetPubKey()); |
||||
|
||||
txTo.vin.resize(3); |
||||
txTo.vin[0].prevout.n = 0; |
||||
txTo.vin[0].prevout.hash = txFrom.GetHash(); |
||||
BOOST_CHECK(SignSignature(keystore, txFrom, txTo, 0)); |
||||
txTo.vin[1].prevout.n = 1; |
||||
txTo.vin[1].prevout.hash = txFrom.GetHash(); |
||||
BOOST_CHECK(SignSignature(keystore, txFrom, txTo, 1)); |
||||
txTo.vin[2].prevout.n = 2; |
||||
txTo.vin[2].prevout.hash = txFrom.GetHash(); |
||||
BOOST_CHECK(SignSignature(keystore, txFrom, txTo, 2)); |
||||
|
||||
BOOST_CHECK(txTo.AreInputsStandard(mapInputs)); |
||||
|
||||
CTransaction txToNonStd; |
||||
txToNonStd.vout.resize(1); |
||||
txToNonStd.vout[0].scriptPubKey.SetBitcoinAddress(key[1].GetPubKey()); |
||||
txToNonStd.vin.resize(1); |
||||
txToNonStd.vin[0].prevout.n = 4; |
||||
txToNonStd.vin[0].prevout.hash = txFrom.GetHash(); |
||||
txToNonStd.vin[0].scriptSig << Serialize(empty); |
||||
|
||||
BOOST_CHECK(!txToNonStd.AreInputsStandard(mapInputs)); |
||||
|
||||
txToNonStd.vin[0].scriptSig.clear(); |
||||
BOOST_CHECK(!txToNonStd.AreInputsStandard(mapInputs)); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_SUITE_END() |
||||
@ -1,268 +0,0 @@
@@ -1,268 +0,0 @@
|
||||
#include <boost/assert.hpp> |
||||
#include <boost/assign/list_of.hpp> |
||||
#include <boost/assign/list_inserter.hpp> |
||||
#include <boost/assign/std/vector.hpp> |
||||
#include <boost/test/unit_test.hpp> |
||||
#include <boost/foreach.hpp> |
||||
|
||||
#include "../main.h" |
||||
#include "../script.h" |
||||
#include "../wallet.h" |
||||
|
||||
using namespace std; |
||||
|
||||
// Test routines internal to script.cpp:
|
||||
extern uint256 SignatureHash(CScript scriptCode, const CTransaction& txTo, unsigned int nIn, int nHashType); |
||||
extern bool VerifyScript(const CScript& scriptSig, const CScript& scriptPubKey, const CTransaction& txTo, unsigned int nIn, int& nSigOps, |
||||
int nHashType, bool fStrictOpEval); |
||||
|
||||
BOOST_AUTO_TEST_SUITE(script_op_eval_tests) |
||||
|
||||
BOOST_AUTO_TEST_CASE(script_op_eval1) |
||||
{ |
||||
// OP_EVAL looks like this:
|
||||
// scriptSig: <sig> <sig...> <serialized_script>
|
||||
// scriptPubKey: DUP HASH160 <hash> EQUALVERIFY EVAL
|
||||
|
||||
// Test SignSignature() (and therefore the version of Solver() that signs transactions)
|
||||
CBasicKeyStore keystore; |
||||
CKey key[4]; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
key[i].MakeNewKey(); |
||||
keystore.AddKey(key[i]); |
||||
} |
||||
|
||||
// 8 Scripts: checking all combinations of
|
||||
// different keys, straight/EVAL, pubkey/pubkeyhash
|
||||
CScript standardScripts[4]; |
||||
standardScripts[0] << key[0].GetPubKey() << OP_CHECKSIG; |
||||
standardScripts[1].SetBitcoinAddress(key[1].GetPubKey()); |
||||
standardScripts[2] << key[1].GetPubKey() << OP_CHECKSIG; |
||||
standardScripts[3].SetBitcoinAddress(key[2].GetPubKey()); |
||||
CScript evalScripts[4]; |
||||
uint160 sigScriptHashes[4]; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
sigScriptHashes[i] = Hash160(standardScripts[i]); |
||||
keystore.AddCScript(sigScriptHashes[i], standardScripts[i]); |
||||
evalScripts[i] << OP_DUP << OP_HASH160 << sigScriptHashes[i] << OP_EQUALVERIFY << OP_EVAL; |
||||
} |
||||
|
||||
CTransaction txFrom; // Funding transaction:
|
||||
txFrom.vout.resize(8); |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
txFrom.vout[i].scriptPubKey = evalScripts[i]; |
||||
txFrom.vout[i+4].scriptPubKey = standardScripts[i]; |
||||
} |
||||
BOOST_CHECK(txFrom.IsStandard()); |
||||
|
||||
CTransaction txTo[8]; // Spending transactions
|
||||
for (int i = 0; i < 8; i++) |
||||
{ |
||||
txTo[i].vin.resize(1); |
||||
txTo[i].vout.resize(1); |
||||
txTo[i].vin[0].prevout.n = i; |
||||
txTo[i].vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo[i].vout[0].nValue = 1; |
||||
BOOST_CHECK_MESSAGE(IsMine(keystore, txFrom.vout[i].scriptPubKey), strprintf("IsMine %d", i)); |
||||
} |
||||
for (int i = 0; i < 8; i++) |
||||
{ |
||||
BOOST_CHECK_MESSAGE(SignSignature(keystore, txFrom, txTo[i], 0), strprintf("SignSignature %d", i)); |
||||
} |
||||
// All of the above should be OK, and the txTos have valid signatures
|
||||
// Check to make sure signature verification fails if we use the wrong ScriptSig:
|
||||
for (int i = 0; i < 8; i++) |
||||
for (int j = 0; j < 8; j++) |
||||
{ |
||||
CScript sigSave = txTo[i].vin[0].scriptSig; |
||||
txTo[i].vin[0].scriptSig = txTo[j].vin[0].scriptSig; |
||||
int nUnused = 0; |
||||
bool sigOK = VerifySignature(txFrom, txTo[i], 0, nUnused); |
||||
if (i == j) |
||||
BOOST_CHECK_MESSAGE(sigOK, strprintf("VerifySignature %d %d", i, j)); |
||||
else |
||||
BOOST_CHECK_MESSAGE(!sigOK, strprintf("VerifySignature %d %d", i, j)); |
||||
txTo[i].vin[0].scriptSig = sigSave; |
||||
} |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(script_op_eval2) |
||||
{ |
||||
// Test OP_EVAL edge cases
|
||||
|
||||
// Make sure infinite recursion fails to validate:
|
||||
CScript infiniteRecurse; |
||||
infiniteRecurse << OP_DUP << OP_EVAL; |
||||
|
||||
uint160 infiniteRecurseHash = Hash160(infiniteRecurse); |
||||
|
||||
CScript fund1; |
||||
fund1 << OP_DUP << OP_HASH160 << infiniteRecurseHash << OP_EQUALVERIFY << OP_EVAL; |
||||
|
||||
CTransaction txFrom1; // Funding transaction:
|
||||
txFrom1.vout.resize(1); |
||||
txFrom1.vout[0].scriptPubKey = fund1; |
||||
|
||||
BOOST_CHECK(txFrom1.IsStandard()); // Looks like a standard transaction until you try to spend it
|
||||
|
||||
std::vector<unsigned char> infiniteRecurseSerialized(infiniteRecurse); |
||||
|
||||
CTransaction txTo1; |
||||
txTo1.vin.resize(1); |
||||
txTo1.vout.resize(1); |
||||
txTo1.vin[0].prevout.n = 0; |
||||
txTo1.vin[0].prevout.hash = txFrom1.GetHash(); |
||||
txTo1.vin[0].scriptSig = CScript() << infiniteRecurseSerialized << infiniteRecurseSerialized; |
||||
txTo1.vout[0].nValue = 1; |
||||
|
||||
int nUnused1 = 0; |
||||
BOOST_CHECK(!VerifyScript(txTo1.vin[0].scriptSig, txFrom1.vout[0].scriptPubKey, txTo1, 0, nUnused1, 0, true)); |
||||
BOOST_CHECK(!VerifySignature(txFrom1, txTo1, 0, nUnused1, true)); |
||||
|
||||
// Make sure 3-level-deep recursion fails to validate:
|
||||
CScript recurse3; |
||||
recurse3 << OP_EVAL; |
||||
|
||||
uint160 recurse3Hash = Hash160(recurse3); |
||||
|
||||
CScript fund2; |
||||
fund2 << OP_DUP << OP_HASH160 << recurse3Hash << OP_EQUALVERIFY << OP_EVAL; |
||||
|
||||
CTransaction txFrom2; // Funding transaction:
|
||||
txFrom2.vout.resize(1); |
||||
txFrom2.vout[0].scriptPubKey = fund2; |
||||
|
||||
BOOST_CHECK(txFrom2.IsStandard()); // Looks like a standard transaction until you try to spend it
|
||||
|
||||
std::vector<unsigned char> recurse3Serialized(recurse3); |
||||
CScript op1Script = CScript() << OP_1; |
||||
std::vector<unsigned char> op1Serialized(op1Script); |
||||
|
||||
CTransaction txTo2; |
||||
txTo2.vin.resize(1); |
||||
txTo2.vout.resize(1); |
||||
txTo2.vin[0].prevout.n = 0; |
||||
txTo2.vin[0].prevout.hash = txFrom2.GetHash(); |
||||
txTo2.vin[0].scriptSig = CScript() << op1Serialized << recurse3Serialized << recurse3Serialized; |
||||
txTo2.vout[0].nValue = 1; |
||||
|
||||
int nUnused2 = 0; |
||||
BOOST_CHECK(!VerifyScript(txTo2.vin[0].scriptSig, txFrom2.vout[0].scriptPubKey, txTo2, 0, nUnused2, 0, true)); |
||||
BOOST_CHECK(!VerifySignature(txFrom2, txTo2, 0, nUnused2, true)); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(script_op_eval3) |
||||
{ |
||||
// Test the CScript::Set* methods
|
||||
CBasicKeyStore keystore; |
||||
CKey key[4]; |
||||
std::vector<CKey> keys; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
key[i].MakeNewKey(); |
||||
keystore.AddKey(key[i]); |
||||
keys.push_back(key[i]); |
||||
} |
||||
|
||||
CScript inner[4]; |
||||
inner[0].SetBitcoinAddress(key[0].GetPubKey()); |
||||
inner[1].SetMultisig(2, std::vector<CKey>(keys.begin(), keys.begin()+2)); |
||||
inner[2].SetMultisig(1, std::vector<CKey>(keys.begin(), keys.begin()+2)); |
||||
inner[3].SetMultisig(2, std::vector<CKey>(keys.begin(), keys.begin()+3)); |
||||
|
||||
CScript outer[4]; |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
outer[i].SetEval(inner[i]); |
||||
keystore.AddCScript(Hash160(inner[i]), inner[i]); |
||||
} |
||||
|
||||
CTransaction txFrom; // Funding transaction:
|
||||
txFrom.vout.resize(4); |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
txFrom.vout[i].scriptPubKey = outer[i]; |
||||
} |
||||
BOOST_CHECK(txFrom.IsStandard()); |
||||
|
||||
CTransaction txTo[4]; // Spending transactions
|
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
txTo[i].vin.resize(1); |
||||
txTo[i].vout.resize(1); |
||||
txTo[i].vin[0].prevout.n = i; |
||||
txTo[i].vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo[i].vout[0].nValue = 1; |
||||
txTo[i].vout[0].scriptPubKey = inner[i]; |
||||
BOOST_CHECK_MESSAGE(IsMine(keystore, txFrom.vout[i].scriptPubKey), strprintf("IsMine %d", i)); |
||||
} |
||||
for (int i = 0; i < 4; i++) |
||||
{ |
||||
BOOST_CHECK_MESSAGE(SignSignature(keystore, txFrom, txTo[i], 0), strprintf("SignSignature %d", i)); |
||||
BOOST_CHECK_MESSAGE(txTo[i].IsStandard(), strprintf("txTo[%d].IsStandard", i)); |
||||
} |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(script_op_eval_backcompat1) |
||||
{ |
||||
// Check backwards-incompatibility-testing code
|
||||
CScript returnsEleven; |
||||
returnsEleven << OP_11; |
||||
|
||||
// This should validate on new clients, but will
|
||||
// be invalid on old clients (that interpret OP_EVAL as a no-op)
|
||||
// ... except there's a special rule that makes new clients reject
|
||||
// it.
|
||||
CScript fund; |
||||
fund << OP_EVAL << OP_11 << OP_EQUAL; |
||||
|
||||
CTransaction txFrom; // Funding transaction:
|
||||
txFrom.vout.resize(1); |
||||
txFrom.vout[0].scriptPubKey = fund; |
||||
|
||||
CTransaction txTo; |
||||
txTo.vin.resize(1); |
||||
txTo.vout.resize(1); |
||||
txTo.vin[0].prevout.n = 0; |
||||
txTo.vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo.vin[0].scriptSig = CScript() << static_cast<std::vector<unsigned char> >(returnsEleven); |
||||
txTo.vout[0].nValue = 1; |
||||
|
||||
int nUnused = 0; |
||||
BOOST_CHECK(!VerifyScript(txTo.vin[0].scriptSig, txFrom.vout[0].scriptPubKey, txTo, 0, nUnused, 0, true)); |
||||
BOOST_CHECK(!VerifySignature(txFrom, txTo, 0, nUnused, true)); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_CASE(script_op_eval_switchover) |
||||
{ |
||||
// Test OP_EVAL switchover code
|
||||
CScript notValid; |
||||
notValid << OP_11 << OP_12 << OP_EQUALVERIFY; |
||||
|
||||
// This will be valid under old rules, invalid under new:
|
||||
CScript fund; |
||||
fund << OP_EVAL; |
||||
|
||||
CTransaction txFrom; // Funding transaction:
|
||||
txFrom.vout.resize(1); |
||||
txFrom.vout[0].scriptPubKey = fund; |
||||
|
||||
CTransaction txTo; |
||||
txTo.vin.resize(1); |
||||
txTo.vout.resize(1); |
||||
txTo.vin[0].prevout.n = 0; |
||||
txTo.vin[0].prevout.hash = txFrom.GetHash(); |
||||
txTo.vin[0].scriptSig = CScript() << static_cast<std::vector<unsigned char> >(notValid); |
||||
txTo.vout[0].nValue = 1; |
||||
|
||||
int nUnused = 0; |
||||
BOOST_CHECK(VerifyScript(txTo.vin[0].scriptSig, txFrom.vout[0].scriptPubKey, txTo, 0, nUnused, 0, false)); |
||||
|
||||
// Under strict op_eval switchover, it should be considered invalid:
|
||||
BOOST_CHECK(!VerifyScript(txTo.vin[0].scriptSig, txFrom.vout[0].scriptPubKey, txTo, 0, nUnused, 0, true)); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_SUITE_END() |
||||
@ -0,0 +1,60 @@
@@ -0,0 +1,60 @@
|
||||
#include <vector> |
||||
#include <boost/test/unit_test.hpp> |
||||
#include <boost/foreach.hpp> |
||||
|
||||
#include "script.h" |
||||
#include "key.h" |
||||
|
||||
using namespace std; |
||||
|
||||
// Helpers:
|
||||
static std::vector<unsigned char> |
||||
Serialize(const CScript& s) |
||||
{ |
||||
std::vector<unsigned char> sSerialized(s); |
||||
return sSerialized; |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_SUITE(sigopcount_tests) |
||||
|
||||
BOOST_AUTO_TEST_CASE(GetSigOpCount) |
||||
{ |
||||
// Test CScript::GetSigOpCount()
|
||||
CScript s1; |
||||
BOOST_CHECK_EQUAL(s1.GetSigOpCount(false), 0); |
||||
BOOST_CHECK_EQUAL(s1.GetSigOpCount(true), 0); |
||||
|
||||
uint160 dummy; |
||||
s1 << OP_1 << dummy << dummy << OP_2 << OP_CHECKMULTISIG; |
||||
BOOST_CHECK_EQUAL(s1.GetSigOpCount(true), 2); |
||||
s1 << OP_IF << OP_CHECKSIG << OP_ENDIF; |
||||
BOOST_CHECK_EQUAL(s1.GetSigOpCount(true), 3); |
||||
BOOST_CHECK_EQUAL(s1.GetSigOpCount(false), 21); |
||||
|
||||
CScript p2sh; |
||||
p2sh.SetPayToScriptHash(s1); |
||||
CScript scriptSig; |
||||
scriptSig << OP_0 << Serialize(s1); |
||||
BOOST_CHECK_EQUAL(p2sh.GetSigOpCount(scriptSig), 3); |
||||
|
||||
std::vector<CKey> keys; |
||||
for (int i = 0; i < 3; i++) |
||||
{ |
||||
CKey k; |
||||
k.MakeNewKey(); |
||||
keys.push_back(k); |
||||
} |
||||
CScript s2; |
||||
s2.SetMultisig(1, keys); |
||||
BOOST_CHECK_EQUAL(s2.GetSigOpCount(true), 3); |
||||
BOOST_CHECK_EQUAL(s2.GetSigOpCount(false), 20); |
||||
|
||||
p2sh.SetPayToScriptHash(s2); |
||||
BOOST_CHECK_EQUAL(p2sh.GetSigOpCount(true), 0); |
||||
BOOST_CHECK_EQUAL(p2sh.GetSigOpCount(false), 0); |
||||
CScript scriptSig2; |
||||
scriptSig2 << OP_1 << dummy << dummy << Serialize(s2); |
||||
BOOST_CHECK_EQUAL(p2sh.GetSigOpCount(scriptSig2), 3); |
||||
} |
||||
|
||||
BOOST_AUTO_TEST_SUITE_END() |
||||
Loading…
Reference in new issue