summaryrefslogtreecommitdiff
path: root/rnd.c
blob: bc72c978f4f89cee0e9ec2012ca33b56514b39ee (plain)
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
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>

#include "rnd.h"
#include "die.h"
#include "ioexact.h"
#include "ioops.h"

static int fdw = -1;

static void randombytes_weak(unsigned char *x, size_t xlen)
{
	int ret;

	if (fdw == -1) {
		for (;;) {
			fdw = open(LOW_ENTROPY_SOURCE, O_RDONLY);
			if (fdw != -1)
				break;
			sleep(1);
		}
	}

	while (xlen > 0) {
		if (xlen < 1048576)
			ret = xlen;
		else
			ret = 1048576;

		ret = read(fdw, x, ret);
		if (ret < 1) {
			sleep(1);
			continue;
		}

		x += ret;
		xlen -= ret;
	}
}

static void randombytes_strong(unsigned char *x, size_t xlen)
{
	int fds, ret;

	fds = open_or_die(HIG_ENTROPY_SOURCE, O_RDONLY);

	ret = read_exact(fds, x, xlen, 0);
	if (ret != xlen)
		panic("Error reading from entropy source!\n");

	close(fds);
}

int secrand(void)
{
	int ret;

	randombytes_weak((void *) &ret, sizeof(ret));

	return ret;
}

void gen_key_bytes(unsigned char *area, size_t len)
{
	randombytes_strong(area, len);
}