2022-10-21 13:01:39 +00:00
|
|
|
import
|
2022-11-23 23:58:17 +00:00
|
|
|
std/[times, random],
|
2023-02-13 10:43:49 +00:00
|
|
|
bearssl/rand,
|
|
|
|
|
libp2p/crypto/crypto
|
2022-10-21 13:01:39 +00:00
|
|
|
|
|
|
|
|
|
2023-02-13 10:43:49 +00:00
|
|
|
## Randomization
|
2022-11-23 23:58:17 +00:00
|
|
|
|
|
|
|
|
proc randomize*() =
|
|
|
|
|
## Initializes the default random number generator with the given seed.
|
|
|
|
|
## From: https://nim-lang.org/docs/random.html#randomize,int64
|
|
|
|
|
let now = getTime()
|
|
|
|
|
randomize(now.toUnix() * 1_000_000_000 + now.nanosecond)
|
2023-02-13 10:43:49 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
## RNG
|
|
|
|
|
# Copied from here: https://github.com/status-im/nim-libp2p/blob/d522537b19a532bc4af94fcd146f779c1f23bad0/tests/helpers.nim#L28
|
|
|
|
|
|
|
|
|
|
type Rng = object
|
|
|
|
|
rng: ref HmacDrbgContext
|
|
|
|
|
|
|
|
|
|
# Typically having a module variable is considered bad design. This case should
|
|
|
|
|
# be considered as an exception and it should be used only in the tests.
|
|
|
|
|
var rngVar: Rng
|
|
|
|
|
|
|
|
|
|
proc getRng(): ref HmacDrbgContext =
|
|
|
|
|
# TODO: if `rngVar` is a threadvar like it should be, there are random and
|
|
|
|
|
# spurious compile failures on mac - this is not gcsafe but for the
|
|
|
|
|
# purpose of the tests, it's ok as long as we only use a single thread
|
|
|
|
|
{.gcsafe.}:
|
|
|
|
|
if rngVar.rng.isNil():
|
|
|
|
|
rngVar.rng = crypto.newRng()
|
|
|
|
|
|
|
|
|
|
rngVar.rng
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
template rng*(): ref HmacDrbgContext = getRng()
|
