Differential D2342 Diff 6740 src/script/sigcache.cpp

Changeset View

Standalone View

src/script/sigcache.cpp

Show All 27 Lines	private:
typedef CuckooCache::cache<uint256, SignatureCacheHasher> map_type;		typedef CuckooCache::cache<uint256, SignatureCacheHasher> map_type;
map_type setValid;		map_type setValid;
boost::shared_mutex cs_sigcache;		boost::shared_mutex cs_sigcache;

public:		public:
CSignatureCache() { GetRandBytes(nonce.begin(), 32); }		CSignatureCache() { GetRandBytes(nonce.begin(), 32); }

void ComputeEntry(uint256 &entry, const uint256 &hash,		void ComputeEntry(uint256 &entry, const uint256 &hash,
const std::vector<uint8_t> &vchSig,		bool isSchnorr, const std::vector<uint8_t> &vchSig,
const CPubKey &pubkey) {		const CPubKey &pubkey) {
		uint8_t sb = isSchnorr?1:0;
CSHA256()		CSHA256()
.Write(nonce.begin(), 32)		.Write(nonce.begin(), 32)
.Write(hash.begin(), 32)		.Write(hash.begin(), 32)
.Write(&pubkey[0], pubkey.size())		.Write(&pubkey[0], pubkey.size())
		.Write(&sb, 1)
.Write(&vchSig[0], vchSig.size())		.Write(&vchSig[0], vchSig.size())
.Finalize(entry.begin());		.Finalize(entry.begin());
}		}

bool Get(const uint256 &entry, const bool erase) {		bool Get(const uint256 &entry, const bool erase) {
boost::shared_lock<boost::shared_mutex> lock(cs_sigcache);		boost::shared_lock<boost::shared_mutex> lock(cs_sigcache);
return setValid.contains(entry, erase);		return setValid.contains(entry, erase);
}		}
Show All 26 Lines	size_t nMaxCacheSize =
(size_t(1) << 20);		(size_t(1) << 20);
size_t nElems = signatureCache.setup_bytes(nMaxCacheSize);		size_t nElems = signatureCache.setup_bytes(nMaxCacheSize);
LogPrintf("Using %zu MiB out of %zu requested for signature cache, able to "		LogPrintf("Using %zu MiB out of %zu requested for signature cache, able to "
"store %zu elements\n",		"store %zu elements\n",
(nElems * sizeof(uint256)) >> 20, nMaxCacheSize >> 20, nElems);		(nElems * sizeof(uint256)) >> 20, nMaxCacheSize >> 20, nElems);
}		}

bool CachingTransactionSignatureChecker::VerifySignature(		bool CachingTransactionSignatureChecker::VerifySignature(
		bool isSchnorr,
const std::vector<uint8_t> &vchSig, const CPubKey &pubkey,		const std::vector<uint8_t> &vchSig, const CPubKey &pubkey,
const uint256 &sighash) const {		const uint256 &sighash) const {
uint256 entry;		uint256 entry;
signatureCache.ComputeEntry(entry, sighash, vchSig, pubkey);		signatureCache.ComputeEntry(entry, sighash, isSchnorr, vchSig, pubkey);
if (signatureCache.Get(entry, !store)) {		if (signatureCache.Get(entry, !store)) {
return true;		return true;
}		}
if (!TransactionSignatureChecker::VerifySignature(vchSig, pubkey,		if (!TransactionSignatureChecker::VerifySignature(isSchnorr, vchSig,
sighash)) {		pubkey, sighash)) {
return false;		return false;
}		}
if (store) {		if (store) {
signatureCache.Set(entry);		signatureCache.Set(entry);
}		}
return true;		return true;
}		}