Differential D3254 Diff 9727 src/test/sigcache_tests.cpp

Changeset View

Standalone View

src/test/sigcache_tests.cpp

	Show All 19 Lines
	#include <string>			#include <string>
	#include <vector>			#include <vector>

	static const std::string strSecret1 =			static const std::string strSecret1 =
	"5HxWvvfubhXpYYpS3tJkw6fq9jE9j18THftkZjHHfmFiWtmAbrj";			"5HxWvvfubhXpYYpS3tJkw6fq9jE9j18THftkZjHHfmFiWtmAbrj";
	static const std::string strSecret1C =			static const std::string strSecret1C =
	"Kwr371tjA9u2rFSMZjTNun2PXXP3WPZu2afRHTcta6KxEUdm1vEw";			"Kwr371tjA9u2rFSMZjTNun2PXXP3WPZu2afRHTcta6KxEUdm1vEw";

	/* We will be testing that these flags do not affect the cache entry.
	* This list must match the one found in script/sigcache.cpp , however
	* we duplicate it here to make sure that changes in cache behaviour also
	* require an intentional change to this test.
	*/
	static const uint32_t TEST_INVARIANT_FLAGS =
	SCRIPT_VERIFY_P2SH \| SCRIPT_VERIFY_STRICTENC \| SCRIPT_VERIFY_DERSIG \|
	SCRIPT_VERIFY_LOW_S \| SCRIPT_VERIFY_NULLDUMMY \| SCRIPT_VERIFY_SIGPUSHONLY \|
	SCRIPT_VERIFY_MINIMALDATA \| SCRIPT_VERIFY_DISCOURAGE_UPGRADABLE_NOPS \|
	SCRIPT_VERIFY_CLEANSTACK \| SCRIPT_VERIFY_CHECKLOCKTIMEVERIFY \|
	SCRIPT_VERIFY_CHECKSEQUENCEVERIFY \| SCRIPT_VERIFY_MINIMALIF \|
	SCRIPT_VERIFY_NULLFAIL \| SCRIPT_VERIFY_COMPRESSED_PUBKEYTYPE \|
	SCRIPT_ENABLE_SIGHASH_FORKID \| SCRIPT_ENABLE_REPLAY_PROTECTION \|
	SCRIPT_VERIFY_CHECKDATASIG_SIGOPS \| SCRIPT_DISALLOW_SEGWIT_RECOVERY;
	/* We will be testing that these flags DO affect the cache entry. The expected
	* behaviour is that flags which are not explicitly listed as invariant in
	* script/sigcache.cpp will affect the cache entry. Here we will thus enforce
	* that certain flags are omitted from that sigcache.cpp list.
	*/
	static const uint32_t TEST_VARIANT_FLAGS = 0;

	/**			/**
	* Sigcache is only accessible via CachingTransactionSignatureChecker			* Sigcache is only accessible via CachingTransactionSignatureChecker
	* as friend.			* as friend.
	*/			*/
	class TestCachingTransactionSignatureChecker {			class TestCachingTransactionSignatureChecker {
	CachingTransactionSignatureChecker *pchecker;			CachingTransactionSignatureChecker *pchecker;

	public:			public:
	TestCachingTransactionSignatureChecker(			TestCachingTransactionSignatureChecker(
	CachingTransactionSignatureChecker &checkerarg) {			CachingTransactionSignatureChecker &checkerarg) {
	pchecker = &checkerarg;			pchecker = &checkerarg;
	}			}

	inline bool VerifyAndStore(const std::vector<uint8_t> &vchSig,			inline bool VerifyAndStore(const std::vector<uint8_t> &vchSig,
	const CPubKey &pubkey, const uint256 &sighash,			const CPubKey &pubkey, const uint256 &sighash) {
	uint32_t flags) {			return pchecker->VerifySignature(vchSig, pubkey, sighash);
	return pchecker->VerifySignature(vchSig, pubkey, sighash, flags);
	}			}

	inline bool IsCached(const std::vector<uint8_t> &vchSig,			inline bool IsCached(const std::vector<uint8_t> &vchSig,
	const CPubKey &pubkey, const uint256 &sighash,			const CPubKey &pubkey, const uint256 &sighash) {
	uint32_t flags) {			return pchecker->IsCached(vchSig, pubkey, sighash);
	return pchecker->IsCached(vchSig, pubkey, sighash, flags);
	}			}
	};			};

	BOOST_FIXTURE_TEST_SUITE(sigcache_tests, BasicTestingSetup)			BOOST_FIXTURE_TEST_SUITE(sigcache_tests, BasicTestingSetup)

	BOOST_AUTO_TEST_CASE(sig_pubkey_hash_variations) {			BOOST_AUTO_TEST_CASE(sig_pubkey_hash_variations) {
	/**			/**
	* Making CachingTransactionSignatureChecker requires a tx. So we make a			* Making CachingTransactionSignatureChecker requires a tx. So we make a
	* dummy transaction (doesn't matter what it is) to construct it.			* dummy transaction (doesn't matter what it is) to construct it.
	*/			*/
	CDataStream stream(			CDataStream stream(
	ParseHex(			ParseHex(
	"010000000122739e70fbee987a8be1788395a2f2e6ad18ccb7ff611cd798071539"			"010000000122739e70fbee987a8be1788395a2f2e6ad18ccb7ff611cd798071539"
	"dde3c38e000000000151ffffffff010000000000000000016a00000000"),			"dde3c38e000000000151ffffffff010000000000000000016a00000000"),
	SER_NETWORK, PROTOCOL_VERSION);			SER_NETWORK, PROTOCOL_VERSION);
	CTransaction dummyTx(deserialize, stream);			CTransaction dummyTx(deserialize, stream);
	PrecomputedTransactionData txdata(dummyTx);			PrecomputedTransactionData txdata(dummyTx);
	CachingTransactionSignatureChecker checker(&dummyTx, 0, 0 * SATOSHI, true,			CachingTransactionSignatureChecker checker(&dummyTx, 0, 0 * SATOSHI, true,
	txdata);			txdata);

	TestCachingTransactionSignatureChecker testChecker(checker);			TestCachingTransactionSignatureChecker testChecker(checker);

	uint32_t flags = 0;

	CBitcoinSecret bsecret1, bsecret1C;			CBitcoinSecret bsecret1, bsecret1C;
	BOOST_CHECK(bsecret1.SetString(strSecret1));			BOOST_CHECK(bsecret1.SetString(strSecret1));
	BOOST_CHECK(bsecret1C.SetString(strSecret1C));			BOOST_CHECK(bsecret1C.SetString(strSecret1C));

	CKey key1 = bsecret1.GetKey();			CKey key1 = bsecret1.GetKey();
	BOOST_CHECK(key1.IsCompressed() == false);			BOOST_CHECK(key1.IsCompressed() == false);
	CKey key1C = bsecret1C.GetKey();			CKey key1C = bsecret1C.GetKey();
	BOOST_CHECK(key1C.IsCompressed() == true);			BOOST_CHECK(key1C.IsCompressed() == true);

	CPubKey pubkey1 = key1.GetPubKey();			CPubKey pubkey1 = key1.GetPubKey();
	CPubKey pubkey1C = key1C.GetPubKey();			CPubKey pubkey1C = key1C.GetPubKey();

	for (int n = 0; n < 16; n++) {			for (int n = 0; n < 16; n++) {
	std::string strMsg = strprintf("Sigcache test1 %i: xx", n);			std::string strMsg = strprintf("Sigcache test1 %i: xx", n);
	uint256 hashMsg = Hash(strMsg.begin(), strMsg.end());			uint256 hashMsg = Hash(strMsg.begin(), strMsg.end());
	uint256 hashMsg2 = Hash(strMsg.begin() + 1, strMsg.end());			uint256 hashMsg2 = Hash(strMsg.begin() + 1, strMsg.end());

	std::vector<uint8_t> sig;			std::vector<uint8_t> sig;
	BOOST_CHECK(key1.SignECDSA(hashMsg, sig));			BOOST_CHECK(key1.SignECDSA(hashMsg, sig));
	std::vector<uint8_t> sig2;			std::vector<uint8_t> sig2;
	BOOST_CHECK(key1.SignECDSA(hashMsg2, sig2));			BOOST_CHECK(key1.SignECDSA(hashMsg2, sig2));

	// cross-check			// cross-check
	BOOST_CHECK(!testChecker.VerifyAndStore(sig2, pubkey1, hashMsg, flags));			BOOST_CHECK(!testChecker.VerifyAndStore(sig2, pubkey1, hashMsg));
	BOOST_CHECK(!testChecker.VerifyAndStore(sig, pubkey1, hashMsg2, flags));			BOOST_CHECK(!testChecker.VerifyAndStore(sig, pubkey1, hashMsg2));
	// that should not have put them in cache...			// that should not have put them in cache...
	BOOST_CHECK(!testChecker.IsCached(sig2, pubkey1, hashMsg, flags));			BOOST_CHECK(!testChecker.IsCached(sig2, pubkey1, hashMsg));
	BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg2, flags));			BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg2));

	// check that it's not in cache at start			// check that it's not in cache at start
	BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg, flags));			BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg));
	BOOST_CHECK(!testChecker.IsCached(sig2, pubkey1, hashMsg2, flags));			BOOST_CHECK(!testChecker.IsCached(sig2, pubkey1, hashMsg2));
	// Insert into cache			// Insert into cache
	BOOST_CHECK(testChecker.VerifyAndStore(sig, pubkey1, hashMsg, flags));			BOOST_CHECK(testChecker.VerifyAndStore(sig, pubkey1, hashMsg));
	BOOST_CHECK(testChecker.VerifyAndStore(sig2, pubkey1, hashMsg2, flags));			BOOST_CHECK(testChecker.VerifyAndStore(sig2, pubkey1, hashMsg2));
	// check that it's in			// check that it's in
	BOOST_CHECK(testChecker.IsCached(sig, pubkey1, hashMsg, flags));			BOOST_CHECK(testChecker.IsCached(sig, pubkey1, hashMsg));
	BOOST_CHECK(testChecker.IsCached(sig2, pubkey1, hashMsg2, flags));			BOOST_CHECK(testChecker.IsCached(sig2, pubkey1, hashMsg2));
	// check that different signature hits different entry			// check that different signature hits different entry
	BOOST_CHECK(!testChecker.IsCached(sig2, pubkey1, hashMsg, flags));			BOOST_CHECK(!testChecker.IsCached(sig2, pubkey1, hashMsg));
	// check that compressed pubkey hits different entry			// check that compressed pubkey hits different entry
	BOOST_CHECK(!testChecker.IsCached(sig, pubkey1C, hashMsg, flags));			BOOST_CHECK(!testChecker.IsCached(sig, pubkey1C, hashMsg));
	// check that different message hits different entry			// check that different message hits different entry
	BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg2, flags));			BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg2));

	// compressed key is for same privkey, so verifying works:			// compressed key is for same privkey, so verifying works:
	BOOST_CHECK(testChecker.VerifyAndStore(sig, pubkey1C, hashMsg, flags));			BOOST_CHECK(testChecker.VerifyAndStore(sig, pubkey1C, hashMsg));
	// now we should get a hit			// now we should get a hit
	BOOST_CHECK(testChecker.IsCached(sig, pubkey1C, hashMsg, flags));			BOOST_CHECK(testChecker.IsCached(sig, pubkey1C, hashMsg));
	}
	}

	BOOST_AUTO_TEST_CASE(flag_invariants) {
	/**
	* Making CachingTransactionSignatureChecker requires a tx. So we make a
	* dummy transaction (doesn't matter what it is) to construct it.
	*/
	CDataStream stream(
	ParseHex(
	"010000000122739e70fbee987a8be1788395a2f2e6ad18ccb7ff611cd798071539"
	"dde3c38e000000000151ffffffff010000000000000000016a00000000"),
	SER_NETWORK, PROTOCOL_VERSION);
	CTransaction dummyTx(deserialize, stream);
	PrecomputedTransactionData txdata(dummyTx);
	CachingTransactionSignatureChecker checker(&dummyTx, 0, 0 * SATOSHI, true,
	txdata);

	TestCachingTransactionSignatureChecker testChecker(checker);

	CBitcoinSecret bsecret1;
	bsecret1.SetString(strSecret1);
	CKey key1 = bsecret1.GetKey();
	CPubKey pubkey1 = key1.GetPubKey();

	// there should not be any overlap
	BOOST_REQUIRE((TEST_VARIANT_FLAGS & TEST_INVARIANT_FLAGS) == 0);

	for (int n = 0; n < 16; n++) {
	std::string strMsg = strprintf("Sigcache testflags %i: xx", n);
	uint256 hashMsg = Hash(strMsg.begin(), strMsg.end());

	std::vector<uint8_t> sig;
	BOOST_CHECK(key1.SignECDSA(hashMsg, sig));

	// choose random background flagset to test
	uint32_t base_flags = insecure_rand();

	// shouldn't be in cache at start
	BOOST_CHECK(!testChecker.IsCached(sig, pubkey1, hashMsg, base_flags));
	// insert into cache
	BOOST_CHECK(
	testChecker.VerifyAndStore(sig, pubkey1, hashMsg, base_flags));
	// check that it's in
	BOOST_CHECK(testChecker.IsCached(sig, pubkey1, hashMsg, base_flags));

	// now we flip each of 32 flags one by one, checking cache
	for (uint32_t flag = 1; flag; flag <<= 1) {
	uint32_t alt_flags = base_flags ^ flag;
	BOOST_CHECK(alt_flags != base_flags);
	bool hit = testChecker.IsCached(sig, pubkey1, hashMsg, alt_flags);
	if (TEST_VARIANT_FLAGS & flag) {
	// if it's in TEST_VARIANT_FLAGS, we must miss cache
	BOOST_CHECK_MESSAGE(!hit, n << " bad cache hit 0x" << std::hex
	<< base_flags << " ^ 0x" << flag);
	} else if (TEST_INVARIANT_FLAGS & flag) {
	// if it's in TEST_INVARIANT_FLAGS, we must hit cache
	BOOST_CHECK_MESSAGE(hit, n << " bad cache miss 0x" << std::hex
	<< base_flags << " ^ 0x" << flag);
	} else {
	// if it's in neither, don't care.
	}
	}
	}			}
	}			}

	BOOST_AUTO_TEST_SUITE_END()			BOOST_AUTO_TEST_SUITE_END()