Philip Kaufmann
11 years ago
16 changed files with 220 additions and 185 deletions
@ -0,0 +1,140 @@
@@ -0,0 +1,140 @@
|
||||
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
||||
// Copyright (c) 2009-2014 The Bitcoin developers
|
||||
// Distributed under the MIT/X11 software license, see the accompanying
|
||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
#include "random.h" |
||||
|
||||
#ifdef WIN32 |
||||
#include "compat.h" // for Windows API |
||||
#endif |
||||
#include "util.h" // for LogPrint() |
||||
|
||||
#ifndef WIN32 |
||||
#include <sys/time.h> |
||||
#endif |
||||
#include <cstring> // for memset() |
||||
|
||||
#include <openssl/crypto.h> |
||||
#include <openssl/err.h> |
||||
#include <openssl/rand.h> |
||||
|
||||
static inline int64_t GetPerformanceCounter() |
||||
{ |
||||
int64_t nCounter = 0; |
||||
#ifdef WIN32 |
||||
QueryPerformanceCounter((LARGE_INTEGER*)&nCounter); |
||||
#else |
||||
timeval t; |
||||
gettimeofday(&t, NULL); |
||||
nCounter = (int64_t)(t.tv_sec * 1000000 + t.tv_usec); |
||||
#endif |
||||
return nCounter; |
||||
} |
||||
|
||||
void RandAddSeed() |
||||
{ |
||||
// Seed with CPU performance counter
|
||||
int64_t nCounter = GetPerformanceCounter(); |
||||
RAND_add(&nCounter, sizeof(nCounter), 1.5); |
||||
memset(&nCounter, 0, sizeof(nCounter)); |
||||
} |
||||
|
||||
void RandAddSeedPerfmon() |
||||
{ |
||||
RandAddSeed(); |
||||
|
||||
// This can take up to 2 seconds, so only do it every 10 minutes
|
||||
static int64_t nLastPerfmon; |
||||
if (GetTime() < nLastPerfmon + 10 * 60) |
||||
return; |
||||
nLastPerfmon = GetTime(); |
||||
|
||||
#ifdef WIN32 |
||||
// Don't need this on Linux, OpenSSL automatically uses /dev/urandom
|
||||
// Seed with the entire set of perfmon data
|
||||
std::vector <unsigned char> vData(250000,0); |
||||
long ret = 0; |
||||
unsigned long nSize = 0; |
||||
const size_t nMaxSize = 10000000; // Bail out at more than 10MB of performance data
|
||||
while (true) |
||||
{ |
||||
nSize = vData.size(); |
||||
ret = RegQueryValueExA(HKEY_PERFORMANCE_DATA, "Global", NULL, NULL, begin_ptr(vData), &nSize); |
||||
if (ret != ERROR_MORE_DATA || vData.size() >= nMaxSize) |
||||
break; |
||||
vData.resize(std::max((vData.size()*3)/2, nMaxSize)); // Grow size of buffer exponentially
|
||||
} |
||||
RegCloseKey(HKEY_PERFORMANCE_DATA); |
||||
if (ret == ERROR_SUCCESS) |
||||
{ |
||||
RAND_add(begin_ptr(vData), nSize, nSize/100.0); |
||||
OPENSSL_cleanse(begin_ptr(vData), nSize); |
||||
LogPrint("rand", "%s: %lu bytes\n", __func__, nSize); |
||||
} else { |
||||
static bool warned = false; // Warn only once
|
||||
if (!warned) |
||||
{ |
||||
LogPrintf("%s: Warning: RegQueryValueExA(HKEY_PERFORMANCE_DATA) failed with code %i\n", __func__, ret); |
||||
warned = true; |
||||
} |
||||
} |
||||
#endif |
||||
} |
||||
|
||||
bool GetRandBytes(unsigned char *buf, int num) |
||||
{ |
||||
if (RAND_bytes(buf, num) != 1) { |
||||
LogPrintf("%s: OpenSSL RAND_bytes() failed with error: %s\n", __func__, ERR_error_string(ERR_get_error(), NULL)); |
||||
return false; |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
uint64_t GetRand(uint64_t nMax) |
||||
{ |
||||
if (nMax == 0) |
||||
return 0; |
||||
|
||||
// The range of the random source must be a multiple of the modulus
|
||||
// to give every possible output value an equal possibility
|
||||
uint64_t nRange = (std::numeric_limits<uint64_t>::max() / nMax) * nMax; |
||||
uint64_t nRand = 0; |
||||
do { |
||||
GetRandBytes((unsigned char*)&nRand, sizeof(nRand)); |
||||
} while (nRand >= nRange); |
||||
return (nRand % nMax); |
||||
} |
||||
|
||||
int GetRandInt(int nMax) |
||||
{ |
||||
return GetRand(nMax); |
||||
} |
||||
|
||||
uint256 GetRandHash() |
||||
{ |
||||
uint256 hash; |
||||
GetRandBytes((unsigned char*)&hash, sizeof(hash)); |
||||
return hash; |
||||
} |
||||
|
||||
uint32_t insecure_rand_Rz = 11; |
||||
uint32_t insecure_rand_Rw = 11; |
||||
void seed_insecure_rand(bool fDeterministic) |
||||
{ |
||||
// The seed values have some unlikely fixed points which we avoid.
|
||||
if(fDeterministic) |
||||
{ |
||||
insecure_rand_Rz = insecure_rand_Rw = 11; |
||||
} else { |
||||
uint32_t tmp; |
||||
do { |
||||
GetRandBytes((unsigned char*)&tmp, 4); |
||||
} while(tmp == 0 || tmp == 0x9068ffffU); |
||||
insecure_rand_Rz = tmp; |
||||
do { |
||||
GetRandBytes((unsigned char*)&tmp, 4); |
||||
} while(tmp == 0 || tmp == 0x464fffffU); |
||||
insecure_rand_Rw = tmp; |
||||
} |
||||
} |
||||
@ -0,0 +1,49 @@
@@ -0,0 +1,49 @@
|
||||
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
||||
// Copyright (c) 2009-2014 The Bitcoin developers
|
||||
// Distributed under the MIT/X11 software license, see the accompanying
|
||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
#ifndef BITCOIN_RANDOM_H |
||||
#define BITCOIN_RANDOM_H |
||||
|
||||
#include "uint256.h" |
||||
|
||||
#include <stdint.h> |
||||
|
||||
/**
|
||||
* Seed OpenSSL PRNG with additional entropy data |
||||
*/ |
||||
void RandAddSeed(); |
||||
void RandAddSeedPerfmon(); |
||||
|
||||
/**
|
||||
* Functions to gather random data via the OpenSSL PRNG |
||||
*/ |
||||
bool GetRandBytes(unsigned char *buf, int num); |
||||
uint64_t GetRand(uint64_t nMax); |
||||
int GetRandInt(int nMax); |
||||
uint256 GetRandHash(); |
||||
|
||||
/**
|
||||
* Seed insecure_rand using the random pool. |
||||
* @param Deterministic Use a determinstic seed |
||||
*/ |
||||
void seed_insecure_rand(bool fDeterministic = false); |
||||
|
||||
/**
|
||||
* MWC RNG of George Marsaglia |
||||
* This is intended to be fast. It has a period of 2^59.3, though the |
||||
* least significant 16 bits only have a period of about 2^30.1. |
||||
* |
||||
* @return random value |
||||
*/ |
||||
extern uint32_t insecure_rand_Rz; |
||||
extern uint32_t insecure_rand_Rw; |
||||
static inline uint32_t insecure_rand(void) |
||||
{ |
||||
insecure_rand_Rz = 36969 * (insecure_rand_Rz & 65535) + (insecure_rand_Rz >> 16); |
||||
insecure_rand_Rw = 18000 * (insecure_rand_Rw & 65535) + (insecure_rand_Rw >> 16); |
||||
return (insecure_rand_Rw << 16) + insecure_rand_Rz; |
||||
} |
||||
|
||||
#endif // BITCOIN_RANDOM_H
|
||||
Loading…
Reference in new issue