117 lines
2.4 KiB
C
117 lines
2.4 KiB
C
#include <stdint.h>
|
|
#include <math.h>
|
|
#include <unistd.h>
|
|
#include <time.h>
|
|
#ifdef LINUX
|
|
#include <sys/time.h>
|
|
#include <fcntl.h>
|
|
#endif
|
|
#include <SDL.h>
|
|
|
|
#include "rng.h"
|
|
#include "log.h"
|
|
|
|
/* Mersenne twister state. */
|
|
static uint32_t MT[624];
|
|
static uint32_t mt_y;
|
|
static int mt_pos = 0; /* Current number. */
|
|
|
|
static uint32_t rng_timeEntropy(void);
|
|
/* Mersenne twister. */
|
|
static void mt_initArray(uint32_t seed);
|
|
static void mt_genArray(void);
|
|
static uint32_t mt_getInt(void);
|
|
|
|
void rng_init(void) {
|
|
uint32_t i;
|
|
int need_init;
|
|
|
|
need_init = 1; /* Initialize by default. */
|
|
#ifdef LINUX
|
|
int fd;
|
|
fd = open("/dev/urandom", O_RDONLY); /* /dev/urandom is better then time seed. */
|
|
if(fd != -1) {
|
|
i = sizeof(uint32_t)*624;
|
|
if(read(fd, &MT, i) == (ssize_t)i)
|
|
need_init = 0;
|
|
else
|
|
i = rng_timeEntropy();
|
|
close(fd);
|
|
} else
|
|
i = rng_timeEntropy();
|
|
#else
|
|
i = rng_timeEntropy();
|
|
#endif
|
|
if(need_init)
|
|
mt_initArray(i);
|
|
for(i = 0; i < 10; i++)
|
|
/* Generate numbers to get away from poor initial values. */
|
|
mt_genArray();
|
|
}
|
|
|
|
/* Use the time as source of entropy. */
|
|
static uint32_t rng_timeEntropy(void) {
|
|
int i;
|
|
|
|
#ifdef WIN32
|
|
struct _timeb tb;
|
|
_ftime(&tb);
|
|
i = tb.time * 1000 + tb.millitm;
|
|
#else
|
|
struct timeval tv;
|
|
gettimeofday(&tv, NULL);
|
|
i = tv.tv_sec * 1000000 + tv.tv_usec;
|
|
#endif
|
|
return i;
|
|
}
|
|
|
|
/* Generates the initial mersenne twister based on seed. */
|
|
static void mt_initArray(uint32_t seed) {
|
|
int i;
|
|
MT[0] = seed;
|
|
for(i = 1; i < 624; i++)
|
|
MT[i] = 1812433253 * (MT[i-1] ^ (((MT[i-1])) + i) >> 30);
|
|
mt_pos = 0;
|
|
}
|
|
|
|
/* Generate an array of numbers. */
|
|
static void mt_genArray(void) {
|
|
int i;
|
|
for(i = 0; i < 624; i++) {
|
|
mt_y = (MT[i] & 0x80000000) + ((MT[i] % 624) & 0x7FFFFFFF);
|
|
if(mt_y % 2)
|
|
/* Odd. */
|
|
MT[i] = (MT[(i+397) % 624] ^ (mt_y >> 1)) ^ 2567483615;
|
|
else
|
|
/* Even. */
|
|
MT[i] = MT[(i+397) % 624] ^ (mt_y >> 1);
|
|
}
|
|
mt_pos = 0;
|
|
}
|
|
|
|
/* Get the next int. */
|
|
static uint32_t mt_getInt(void) {
|
|
if(mt_pos >= 624) mt_genArray();
|
|
|
|
mt_y = MT[mt_pos++];
|
|
mt_y ^= mt_y >> 11;
|
|
mt_y ^= (mt_y << 7) & 2636928640;
|
|
mt_y ^= (mt_y << 15) & 4022730752;
|
|
mt_y ^= (mt_y >> 18);
|
|
|
|
return mt_y;
|
|
}
|
|
|
|
/* Return a random int. */
|
|
unsigned int randint(void) {
|
|
return mt_getInt();
|
|
}
|
|
|
|
/* Return a random double. */
|
|
static double m_div = (double)(0xFFFFFFFF);
|
|
double randfp(void) {
|
|
double m = (double)mt_getInt();
|
|
return m / m_div;
|
|
}
|
|
|