Skip to content

Commit 37e864e

Browse files
committed
Add FastRandomContext::rand256() and ::randbytes()
FastRandomContext now provides all functionality that the real Rand* functions provide.
1 parent 9fec4da commit 37e864e

File tree

3 files changed

+33
-0
lines changed

3 files changed

+33
-0
lines changed

src/random.cpp

+20
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,26 @@ void FastRandomContext::RandomSeed()
304304
requires_seed = false;
305305
}
306306

307+
uint256 FastRandomContext::rand256()
308+
{
309+
if (bytebuf_size < 32) {
310+
FillByteBuffer();
311+
}
312+
uint256 ret;
313+
memcpy(ret.begin(), bytebuf + 64 - bytebuf_size, 32);
314+
bytebuf_size -= 32;
315+
return ret;
316+
}
317+
318+
std::vector<unsigned char> FastRandomContext::randbytes(size_t len)
319+
{
320+
std::vector<unsigned char> ret(len);
321+
if (len > 0) {
322+
rng.Output(&ret[0], len);
323+
}
324+
return ret;
325+
}
326+
307327
FastRandomContext::FastRandomContext(const uint256& seed) : requires_seed(false), bytebuf_size(0), bitbuf_size(0)
308328
{
309329
rng.SetKey(seed.begin(), 32);

src/random.h

+6
Original file line numberDiff line numberDiff line change
@@ -110,9 +110,15 @@ class FastRandomContext {
110110
}
111111
}
112112

113+
/** Generate random bytes. */
114+
std::vector<unsigned char> randbytes(size_t len);
115+
113116
/** Generate a random 32-bit integer. */
114117
uint32_t rand32() { return randbits(32); }
115118

119+
/** generate a random uint256. */
120+
uint256 rand256();
121+
116122
/** Generate a random boolean. */
117123
bool randbool() { return randbits(1); }
118124
};

src/test/random_tests.cpp

+7
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,21 @@ BOOST_AUTO_TEST_CASE(fastrandom_tests)
2525
BOOST_CHECK_EQUAL(ctx1.rand32(), ctx2.rand32());
2626
BOOST_CHECK_EQUAL(ctx1.rand64(), ctx2.rand64());
2727
BOOST_CHECK_EQUAL(ctx1.randbits(3), ctx2.randbits(3));
28+
BOOST_CHECK(ctx1.randbytes(17) == ctx2.randbytes(17));
29+
BOOST_CHECK(ctx1.rand256() == ctx2.rand256());
2830
BOOST_CHECK_EQUAL(ctx1.randbits(7), ctx2.randbits(7));
31+
BOOST_CHECK(ctx1.randbytes(128) == ctx2.randbytes(128));
2932
BOOST_CHECK_EQUAL(ctx1.rand32(), ctx2.rand32());
3033
BOOST_CHECK_EQUAL(ctx1.randbits(3), ctx2.randbits(3));
34+
BOOST_CHECK(ctx1.rand256() == ctx2.rand256());
35+
BOOST_CHECK(ctx1.randbytes(50) == ctx2.randbytes(50));
3136

3237
// Check that a nondeterministic ones are not
3338
FastRandomContext ctx3;
3439
FastRandomContext ctx4;
3540
BOOST_CHECK(ctx3.rand64() != ctx4.rand64()); // extremely unlikely to be equal
41+
BOOST_CHECK(ctx3.rand256() != ctx4.rand256());
42+
BOOST_CHECK(ctx3.randbytes(7) != ctx4.randbytes(7));
3643
}
3744

3845
BOOST_AUTO_TEST_CASE(fastrandom_randbits)

0 commit comments

Comments
 (0)