Merge pull request #11598 from shuffle2/std-prng

use std-provided randomness for JitArm64 unittests
This commit is contained in:
Pierre Bourdon 2023-02-24 19:47:18 +09:00 committed by GitHub
commit 88fc431dce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 61 deletions

View File

@ -10,38 +10,6 @@
namespace Common::Random namespace Common::Random
{ {
struct PRNG::Impl
{
Impl(void* seed, std::size_t size)
{
mbedtls_hmac_drbg_init(&m_context);
const int ret = mbedtls_hmac_drbg_seed_buf(
&m_context, mbedtls_md_info_from_type(MBEDTLS_MD_SHA256), static_cast<u8*>(seed), size);
ASSERT(ret == 0);
}
~Impl() { mbedtls_hmac_drbg_free(&m_context); }
void Generate(void* buffer, std::size_t size)
{
const int ret = mbedtls_hmac_drbg_random(&m_context, static_cast<u8*>(buffer), size);
ASSERT(ret == 0);
}
mbedtls_hmac_drbg_context m_context;
};
PRNG::PRNG(void* seed, std::size_t size) : m_impl(std::make_unique<Impl>(seed, size))
{
}
PRNG::~PRNG() = default;
void PRNG::Generate(void* buffer, std::size_t size)
{
m_impl->Generate(buffer, size);
}
class EntropySeededPRNG final class EntropySeededPRNG final
{ {
public: public:

View File

@ -11,30 +11,6 @@
namespace Common::Random namespace Common::Random
{ {
/// Cryptographically secure pseudo-random number generator, with explicit seed.
class PRNG final
{
public:
explicit PRNG(u64 seed) : PRNG(&seed, sizeof(u64)) {}
PRNG(void* seed, std::size_t size);
~PRNG();
void Generate(void* buffer, std::size_t size);
template <typename T>
T GenerateValue()
{
static_assert(std::is_arithmetic<T>(), "T must be an arithmetic type in GenerateValue.");
T value;
Generate(&value, sizeof(value));
return value;
}
private:
struct Impl;
std::unique_ptr<Impl> m_impl;
};
/// Fill `buffer` with random bytes using a cryptographically secure pseudo-random number generator. /// Fill `buffer` with random bytes using a cryptographically secure pseudo-random number generator.
void Generate(void* buffer, std::size_t size); void Generate(void* buffer, std::size_t size);

View File

@ -2,12 +2,12 @@
// SPDX-License-Identifier: GPL-2.0-or-later // SPDX-License-Identifier: GPL-2.0-or-later
#include <cstddef> #include <cstddef>
#include <random>
#include <type_traits> #include <type_traits>
#include "Common/Arm64Emitter.h" #include "Common/Arm64Emitter.h"
#include "Common/Assert.h" #include "Common/Assert.h"
#include "Common/BitUtils.h" #include "Common/BitUtils.h"
#include "Common/Random.h"
#include <gtest/gtest.h> #include <gtest/gtest.h>
@ -59,11 +59,12 @@ public:
TEST(JitArm64, MovI2R_32BitValues) TEST(JitArm64, MovI2R_32BitValues)
{ {
Common::Random::PRNG rng{0}; std::default_random_engine engine(0);
std::uniform_int_distribution<u32> dist;
TestMovI2R test; TestMovI2R test;
for (u64 i = 0; i < 0x100000; i++) for (u64 i = 0; i < 0x100000; i++)
{ {
const u32 value = rng.GenerateValue<u32>(); const u32 value = dist(engine);
test.Check32(value); test.Check32(value);
test.Check64(value); test.Check64(value);
} }
@ -71,11 +72,12 @@ TEST(JitArm64, MovI2R_32BitValues)
TEST(JitArm64, MovI2R_Rand) TEST(JitArm64, MovI2R_Rand)
{ {
Common::Random::PRNG rng{0}; std::default_random_engine engine(0);
std::uniform_int_distribution<u64> dist;
TestMovI2R test; TestMovI2R test;
for (u64 i = 0; i < 0x100000; i++) for (u64 i = 0; i < 0x100000; i++)
{ {
test.Check64(rng.GenerateValue<u64>()); test.Check64(dist(engine));
} }
} }