void sfmt_init_stable_r(sfmt_t *sfmt){
uint32_t init_array[SFMT_N32];
struct timeval tp;
union lcg_state lcg_init;
lcg_init.X_i=time_seed();
int i;
for(i=0;i<SFMT_N32;i++){
init_array[i]=nrand48(lcg_init.state);
}
sfmt_init_by_array(sfmt,init_array,SFMT_N32);
}
inline void init_mt(char *seed) {
char *pEnd; //!< Used to check if user seed is parsable
/* Seed from
* http://www.random.org/integers/?num=9&min=1&max=1000000000&col=9&base=16&format=plain&rnd=date.2014-05-01
*/
// Seed the MT using 9 random integers and 1 from user input
uint32_t init[10] = { 0x012d661f, 0x399ba1e1, 0x01d033e9,
0x24ab21da, 0x3a1b142b, 0x21109a50,
0x20d005a4, 0x35215de9, 0x091c98f1};
// Parse user input
init[9] = strtol(seed, &pEnd, 10);
// Check if all was parsed
if(pEnd[0] != '\0') {
fprintf(stderr, "Could not convert user seed to an integer ('%c' is not in base 10)\n", pEnd[0]);
exit(1);
}
// Init SFMT
sfmt_init_by_array(&sfmt, init, 10);
}
int main(int argc, char * const argv[])
{
char *output = NULL;
char buffer[1024];
unsigned long gen = 100;
long seed[33];
char c;
int mul = 0, seedpos = 0;
FILE *out;
int verbosity = 1, running = 0;
struct timespec tp;
char mode[] = "w";
sfmt_t sfmt;
int cores =
#if defined(WIN32) || defined(__WIN32)
1;
#else
sysconf (_SC_NPROCESSORS_CONF);
#endif
unsigned long blocksize;
struct process *process;
#ifdef _WIN32
seed[0] = time(NULL);
#else
clock_gettime(CLOCK_REALTIME, &tp);
seed[0] = tp.tv_nsec + tp.tv_sec * 1000 * 1000 * 1000;
#endif
while ((c = getopt(argc, argv, "hc:a:o:s:g:qv")) != -1) {
switch (c) {
default:
usage(argv[0], 1, "ERROR: Unknown parameter %c\n", c);
break;
case 'h':
case '?':
usage(argv[0], 0, NULL);
break;
case 'g':
gen = atoi(optarg);
if (gen > INT_MAX)
usage(argv[0], 2, "ERROR: Generate %lu is over max %d\n", gen, INT_MAX);
break;
case 'q':
verbosity = 0;
break;
case 'c':
cores = atoi(optarg);
if (cores <= 0)
usage(argv[0], 2, "ERROR: %d cores must be over zero", cores);
break;
case 'v':
verbosity++;
break;
case 's':
{
char *startptr = optarg;
/* Read seed from /dev/urandom */
if (strcmp("dev", optarg) == 0) {
FILE *dev = fopen("/dev/urandom", "r");
int r;
if (!dev) {
perror("opening /dev/urandom");
return 12;
}
do {
r = fread(seed, sizeof seed[0], sizeof seed/sizeof seed[0], dev);
} while (r != sizeof seed/sizeof seed[0] && errno == EINTR);
if (r != sizeof seed/sizeof seed[0]) {
perror("fread /dev/random");
return 13;
}
seedpos = sizeof seed/sizeof seed[0];
fclose(dev);
break;
}
for (; seedpos < 33; seedpos++) {
char *endptr;
unsigned long s;
errno = 0;
s = strtol(startptr, &endptr, 10);
if (endptr == startptr) {
break;
}
if (errno != 0) {
perror("sttrol");
return errno;
}
startptr = endptr + 1;
seed[seedpos] = s;
if (*endptr == '\0') {
seedpos++;
break;
}
}
}
break;
case 'a':
mode[0] = 'a';
case 'o':
output = strdup(optarg);
break;
}
}
freopen("/dev/null", "r", stdin);
process = alloca(sizeof(process[0])*cores);
memset(process, 0, sizeof(process[0])*cores);
/* Calculate reasonable block sizes for the cpu configuration */
blocksize = gen;
for (mul = 20; mul > 0; mul--) {
long high = 1 << mul;
if (gen >= (unsigned)(cores*high)) {
blocksize = gen / (cores * mul);
break;
}
}
if (blocksize >= (1 << 20)/20)
blocksize = (1 << 20)/20;
int i, pos = 0;
sfmt_init_by_array(&sfmt, (uint32_t*)&seed[0], (seedpos - 1)*(sizeof(seed[0])/sizeof(uint32_t)));
pos = sprintf(buffer, "%ld", seed[0]);
for (i = 1; i < seedpos; i++)
pos += sprintf(&buffer[pos], ",%ld", seed[i]);
if (verbosity >= 1)
fprintf(stderr, "Generating %ld deals with %s seed. Each subprocess does %ld deals.\n",
gen, buffer, blocksize);
out = output ? fopen(output, mode) : stdout;
if (!out) {
perror("Can't open output file");
return 10;
}
{
int fd, maxfd = -1;
int c;
fd_set fds;
fd_set rfds;
FD_ZERO(&fds);
for (c = 0; c < cores; c++) {
unsigned long b = gen - scheduled > blocksize ?
blocksize : gen - scheduled;
if (b == 0)
break;
if ((fd = makeprocess(&process[c], b, sfmt_genrand_uint64(&sfmt), verbosity)) < 0)
return 20;
running++;
if (fd > maxfd)
maxfd = fd;
FD_SET(fd, &fds);
}
/* Loop as long as we have subprocess active */
while (running > 0) {
int active;
int p = 0;
rfds = fds;
/* Wait for any input */
do {
active = select(maxfd + 1, &rfds, NULL, NULL, NULL);
} while (active == -1 && errno == EINTR);
if (active <= 0) {
perror("select");
return 60;
}
/* Check which subprocess provided the input */
for (; p < cores; p++) {
if (FD_ISSET(process[p].fd, &rfds)) {
errno = 0;
while (fgets(buffer, sizeof buffer, process[p].f))
parseLine(buffer, out, &process[p], gen, verbosity);
/* Has the process exited? */
if (feof(process[p].f)) {
FD_CLR(process[p].fd, &fds);
closeprocess(&process[p]);
running--;
/* Is there more blocks to shedule? */
if (scheduled < gen) {
unsigned long b = gen - scheduled > blocksize ?
blocksize : gen - scheduled;
if ((fd = makeprocess(&process[p], b, sfmt_genrand_uint64(&sfmt), verbosity)) < 0)
return 20;
running++;
if (fd > maxfd)
maxfd = fd;
FD_SET(fd, &fds);
}
continue;
}
/* EWOULDBLOCK and EINTR can happen during normal operation */
if (errno != EWOULDBLOCK && errno != EINTR) {
perror("fgets");
return 50;
}
}
}
}
}
fclose(out);
return 0;
}
