MarcoFalke
fa19af555d
The global g_insecure_rand_ctx will be removed in the future, so removing it from this helper is useful. Also, tying the two concepts of the global internal RNGState and the global test-only rng context is a bit confusing, because tests can simply use the m_rng, if it exists. Also, tests may seed more than one random context, or none at all, or a random context of a different type. Fix all issues by moving the Reseed call to the two places where it is used.
61 lines
1.5 KiB
C++
61 lines
1.5 KiB
C++
// Copyright (c) 2023 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#ifndef BITCOIN_TEST_UTIL_RANDOM_H
|
|
#define BITCOIN_TEST_UTIL_RANDOM_H
|
|
|
|
#include <consensus/amount.h>
|
|
#include <random.h>
|
|
#include <uint256.h>
|
|
|
|
#include <cstdint>
|
|
|
|
/**
|
|
* This global and the helpers that use it are not thread-safe.
|
|
*
|
|
* If thread-safety is needed, a per-thread instance could be
|
|
* used in the multi-threaded test.
|
|
*/
|
|
extern FastRandomContext g_insecure_rand_ctx;
|
|
|
|
enum class SeedRand {
|
|
ZEROS, //!< Seed with a compile time constant of zeros
|
|
SEED, //!< Use (and report) random seed from environment, or a (truly) random one.
|
|
};
|
|
|
|
/** Seed the global RNG state for testing and log the seed value. This affects all randomness, except GetStrongRandBytes(). */
|
|
void SeedRandomStateForTest(SeedRand seed);
|
|
|
|
static inline uint32_t InsecureRand32()
|
|
{
|
|
return g_insecure_rand_ctx.rand32();
|
|
}
|
|
|
|
static inline uint256 InsecureRand256()
|
|
{
|
|
return g_insecure_rand_ctx.rand256();
|
|
}
|
|
|
|
static inline uint64_t InsecureRandBits(int bits)
|
|
{
|
|
return g_insecure_rand_ctx.randbits(bits);
|
|
}
|
|
|
|
static inline uint64_t InsecureRandRange(uint64_t range)
|
|
{
|
|
return g_insecure_rand_ctx.randrange(range);
|
|
}
|
|
|
|
static inline bool InsecureRandBool()
|
|
{
|
|
return g_insecure_rand_ctx.randbool();
|
|
}
|
|
|
|
static inline CAmount InsecureRandMoneyAmount()
|
|
{
|
|
return static_cast<CAmount>(InsecureRandRange(MAX_MONEY + 1));
|
|
}
|
|
|
|
#endif // BITCOIN_TEST_UTIL_RANDOM_H
|