diff --git a/src/cuckoocache.h b/src/cuckoocache.h index 5fc852439fe..b5da87136d9 100644 --- a/src/cuckoocache.h +++ b/src/cuckoocache.h @@ -12,7 +12,9 @@ #include #include #include +#include #include +#include #include #include @@ -359,10 +361,15 @@ public: * structure * @returns A pair of the maximum number of elements storable (see setup() * documentation for more detail) and the approxmiate total size of these - * elements in bytes. + * elements in bytes or std::nullopt if the size requested is too large. */ - std::pair setup_bytes(size_t bytes) + std::optional> setup_bytes(size_t bytes) { + size_t requested_num_elems = bytes / sizeof(Element); + if (std::numeric_limits::max() < requested_num_elems) { + return std::nullopt; + } + auto num_elems = setup(bytes/sizeof(Element)); size_t approx_size_bytes = num_elems * sizeof(Element); diff --git a/src/script/sigcache.cpp b/src/script/sigcache.cpp index 43b594f3cb6..507754ad7d9 100644 --- a/src/script/sigcache.cpp +++ b/src/script/sigcache.cpp @@ -14,6 +14,7 @@ #include #include +#include #include #include @@ -75,7 +76,7 @@ public: std::unique_lock lock(cs_sigcache); setValid.insert(entry); } - std::pair setup_bytes(size_t n) + std::optional> setup_bytes(size_t n) { return setValid.setup_bytes(n); } @@ -99,8 +100,9 @@ bool InitSignatureCache() size_t nMaxCacheSize = std::max((int64_t)0, gArgs.GetIntArg("-maxsigcachesize", DEFAULT_MAX_SIG_CACHE_SIZE) / 2) * ((size_t) 1 << 20); auto setup_results = signatureCache.setup_bytes(nMaxCacheSize); + if (!setup_results) return false; - const auto [num_elems, approx_size_bytes] = setup_results; + const auto [num_elems, approx_size_bytes] = *setup_results; LogPrintf("Using %zu MiB out of %zu/2 requested for signature cache, able to store %zu elements\n", approx_size_bytes >> 20, (nMaxCacheSize * 2) >> 20, num_elems); return true; diff --git a/src/validation.cpp b/src/validation.cpp index 4d174a4b166..73a6e46cc49 100644 --- a/src/validation.cpp +++ b/src/validation.cpp @@ -1669,8 +1669,9 @@ bool InitScriptExecutionCache() { size_t nMaxCacheSize = std::max((int64_t)0, gArgs.GetIntArg("-maxsigcachesize", DEFAULT_MAX_SIG_CACHE_SIZE) / 2) * ((size_t) 1 << 20); auto setup_results = g_scriptExecutionCache.setup_bytes(nMaxCacheSize); + if (!setup_results) return false; - const auto [num_elems, approx_size_bytes] = setup_results; + const auto [num_elems, approx_size_bytes] = *setup_results; LogPrintf("Using %zu MiB out of %zu/2 requested for script execution cache, able to store %zu elements\n", approx_size_bytes >> 20, (nMaxCacheSize * 2) >> 20, num_elems); return true;