blob: 4b7200093c47eb7666e1d903bb51f5a56efb7881 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* File: rng.cc
* Summary: Random number generator wrapping.
*/
#include "AppHdr.h"
#include "rng.h"
#include "mt19937ar.h"
#ifdef USE_MORE_SECURE_SEED
// for times()
#include <sys/times.h>
// for getpid()
#include <sys/types.h>
#include <unistd.h>
#endif
#ifdef MORE_HARDENED_PRNG
#include "sha256.h"
#endif
void seed_rng(unsigned long* seed_key, size_t num_keys)
{
// MT19937 -- see mt19937ar.cc for details/licence
init_by_array(seed_key, num_keys);
// Reset the sha256 generator to get predictable random numbers in case
// of a saved rng state.
#ifdef MORE_HARDENED_PRNG
reset_sha256_state();
#endif
// for std::random_shuffle()
srand(seed_key[0]);
}
void seed_rng(long seed)
{
// MT19937 -- see mt19937ar.cc for details/licence
init_genrand(seed);
// Reset the sha256 generator to get predictable random numbers in case
// of a saved rng state.
#ifdef MORE_HARDENED_PRNG
reset_sha256_state();
#endif
// for std::random_shuffle()
srand(seed);
}
void seed_rng()
{
unsigned long seed = time( NULL );
#ifdef USE_MORE_SECURE_SEED
/* (at least) 256-bit wide seed */
unsigned long seed_key[8];
struct tms buf;
seed += times( &buf ) + getpid();
seed_key[0] = seed;
/* Try opening from various system provided (hopefully) CSPRNGs */
FILE* seed_f = fopen("/dev/urandom", "rb");
if (!seed_f)
seed_f = fopen("/dev/random", "rb");
if (!seed_f)
seed_f = fopen("/dev/srandom", "rb");
if (!seed_f)
seed_f = fopen("/dev/arandom", "rb");
if (seed_f)
{
fread(&seed_key[1], sizeof(unsigned long), 7, seed_f);
fclose(seed_f);
}
seed_rng(seed_key, 8);
#else
seed_rng(seed);
#endif
}
// MT19937 -- see mt19937ar.cc for details
unsigned long random_int( void )
{
#ifndef MORE_HARDENED_PRNG
return (genrand_int32());
#else
return (sha256_genrand());
#endif
}
void push_rng_state()
{
#ifndef MORE_HARDENED_PRNG
push_mt_state();
#else
push_sha256_state();
#endif
}
void pop_rng_state()
{
#ifndef MORE_HARDENED_PRNG
pop_mt_state();
#else
pop_sha256_state();
#endif
}
|