Differential D530 Diff 1344 src/script/scriptcache.cpp

Changeset View

Standalone View

src/script/scriptcache.cpp

This file was added.

				// Copyright (c) 2017 The Bitcoin developers
				// Distributed under the MIT software license, see the accompanying
				// file COPYING or http://www.opensource.org/licenses/mit-license.php.

				#include "scriptcache.h"

				#include "crypto/sha256.h"
				#include "cuckoocache.h"
				#include "primitives/transaction.h"
				#include "random.h"
				#include "script/sigcache.h"
				#include "sync.h"
				#include "util.h"
				#include "validation.h"

				static CuckooCache::cache<uint256, SignatureCacheHasher> scriptExecutionCache;
				static uint256 scriptExecutionCacheNonce(GetRandHash());

				void InitScriptExecutionCache() {
				// nMaxCacheSize is unsigned. If -maxscriptcachesize is set to zero,
				// setup_bytes creates the minimum possible cache (2 elements).
				size_t nMaxCacheSize =
				std::min(std::max(int64_t(0), GetArg("-maxscriptcachesize",
				DEFAULT_MAX_SCRIPT_CACHE_SIZE)),
				MAX_MAX_SCRIPT_CACHE_SIZE) *
				(size_t(1) << 20);
				size_t nElems = scriptExecutionCache.setup_bytes(nMaxCacheSize);
				LogPrintf("Using %zu MiB out of %zu requested for script execution cache, "
				"able to store %zu elements\n",
				(nElems * sizeof(uint256)) >> 20, nMaxCacheSize >> 20, nElems);
				}

				uint256 GetScriptCacheKey(const CTransaction &tx, uint32_t flags) {
				uint256 key;
				// We only use the first 19 bytes of nonce to avoid a second SHA round -
				// giving us 19 + 32 + 4 = 55 bytes (+ 8 + 1 = 64)
				static_assert(55 - sizeof(flags) - 32 >= 128 / 8,
				"Want at least 128 bits of nonce for script execution cache");
				CSHA256()
				.Write(scriptExecutionCacheNonce.begin(), 55 - sizeof(flags) - 32)
				.Write(tx.GetHash().begin(), 32)
				.Write((uint8_t *)&flags, sizeof(flags))
				.Finalize(key.begin());

				return key;
				}

				bool IsKeyInScriptCache(uint256 key, bool erase) {
				// TODO: Remove this requirement by making CuckooCache not require external
				// locks
				AssertLockHeld(cs_main);
				return scriptExecutionCache.contains(key, erase);
				}

				void AddKeyInScriptCache(uint256 key) {
				dagurvalUnsubmitted Not Done Inline Actions If IsKeyInScriptCache requires cs_main, shouldn't inserting also require cs_main? (AssertLockHeld(cs_main)) dagurval: If IsKeyInScriptCache requires cs_main, shouldn't inserting also require cs_main?
				deadalnixAuthorUnsubmitted Not Done Inline Actions Yes. deadalnix: Yes.
				scriptExecutionCache.insert(key);
				}