1 /*
2  * Random number generator
3  * Copyright (c) 2010-2011, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  *
8  * This random number generator is used to provide additional entropy to the
9  * one provided by the operating system (os_get_random()) for session key
10  * generation. The os_get_random() output is expected to be secure and the
11  * implementation here is expected to provide only limited protection against
12  * cases where os_get_random() cannot provide strong randomness. This
13  * implementation shall not be assumed to be secure as the sole source of
14  * randomness. The random_get_bytes() function mixes in randomness from
15  * os_get_random() and as such, calls to os_get_random() can be replaced with
16  * calls to random_get_bytes() without reducing security.
17  *
18  * The design here follows partially the design used in the Linux
19  * drivers/char/random.c, but the implementation here is simpler and not as
20  * strong. This is a compromise to reduce duplicated CPU effort and to avoid
21  * extra code/memory size. As pointed out above, os_get_random() needs to be
22  * guaranteed to be secure for any of the security assumptions to hold.
23  */
24 
25 #include "utils/includes.h"
26 #ifdef __linux__
27 #include <fcntl.h>
28 #endif /* __linux__ */
29 
30 #include "utils/common.h"
31 #include "utils/eloop.h"
32 #include "crypto/crypto.h"
33 #include "sha1.h"
34 #include "random.h"
35 
36 #define POOL_WORDS 32
37 #define POOL_WORDS_MASK (POOL_WORDS - 1)
38 #define POOL_TAP1 26
39 #define POOL_TAP2 20
40 #define POOL_TAP3 14
41 #define POOL_TAP4 7
42 #define POOL_TAP5 1
43 #define EXTRACT_LEN 16
44 #define MIN_READY_MARK 2
45 
46 static u32 pool[POOL_WORDS];
47 static unsigned int input_rotate = 0;
48 static unsigned int pool_pos = 0;
49 static u8 dummy_key[20];
50 #ifdef __linux__
51 static size_t dummy_key_avail = 0;
52 static int random_fd = -1;
53 #endif /* __linux__ */
54 static unsigned int own_pool_ready = 0;
55 #define RANDOM_ENTROPY_SIZE 20
56 static char *random_entropy_file = NULL;
57 static int random_entropy_file_read = 0;
58 
59 #define MIN_COLLECT_ENTROPY 1000
60 static unsigned int entropy = 0;
61 static unsigned int total_collected = 0;
62 
63 
64 static void random_write_entropy(void);
65 
66 
67 static u32 __ROL32(u32 x, u32 y)
68 {
69 	if (y == 0)
70 		return x;
71 
72 	return (x << (y & 31)) | (x >> (32 - (y & 31)));
73 }
74 
75 
76 static void random_mix_pool(const void *buf, size_t len)
77 {
78 	static const u32 twist[8] = {
79 		0x00000000, 0x3b6e20c8, 0x76dc4190, 0x4db26158,
80 		0xedb88320, 0xd6d6a3e8, 0x9b64c2b0, 0xa00ae278
81 	};
82 	const u8 *pos = buf;
83 	u32 w;
84 
85 	wpa_hexdump_key(MSG_EXCESSIVE, "random_mix_pool", buf, len);
86 
87 	while (len--) {
88 		w = __ROL32(*pos++, input_rotate & 31);
89 		input_rotate += pool_pos ? 7 : 14;
90 		pool_pos = (pool_pos - 1) & POOL_WORDS_MASK;
91 		w ^= pool[pool_pos];
92 		w ^= pool[(pool_pos + POOL_TAP1) & POOL_WORDS_MASK];
93 		w ^= pool[(pool_pos + POOL_TAP2) & POOL_WORDS_MASK];
94 		w ^= pool[(pool_pos + POOL_TAP3) & POOL_WORDS_MASK];
95 		w ^= pool[(pool_pos + POOL_TAP4) & POOL_WORDS_MASK];
96 		w ^= pool[(pool_pos + POOL_TAP5) & POOL_WORDS_MASK];
97 		pool[pool_pos] = (w >> 3) ^ twist[w & 7];
98 	}
99 }
100 
101 
102 static void random_extract(u8 *out)
103 {
104 	unsigned int i;
105 	u8 hash[SHA1_MAC_LEN];
106 	u32 *hash_ptr;
107 	u32 buf[POOL_WORDS / 2];
108 
109 	/* First, add hash back to pool to make backtracking more difficult. */
110 	hmac_sha1(dummy_key, sizeof(dummy_key), (const u8 *) pool,
111 		  sizeof(pool), hash);
112 	random_mix_pool(hash, sizeof(hash));
113 	/* Hash half the pool to extra data */
114 	for (i = 0; i < POOL_WORDS / 2; i++)
115 		buf[i] = pool[(pool_pos - i) & POOL_WORDS_MASK];
116 	hmac_sha1(dummy_key, sizeof(dummy_key), (const u8 *) buf,
117 		  sizeof(buf), hash);
118 	/*
119 	 * Fold the hash to further reduce any potential output pattern.
120 	 * Though, compromise this to reduce CPU use for the most common output
121 	 * length (32) and return 16 bytes from instead of only half.
122 	 */
123 	hash_ptr = (u32 *) hash;
124 	hash_ptr[0] ^= hash_ptr[4];
125 	os_memcpy(out, hash, EXTRACT_LEN);
126 }
127 
128 
129 void random_add_randomness(const void *buf, size_t len)
130 {
131 	struct os_time t;
132 	static unsigned int count = 0;
133 
134 	count++;
135 	if (entropy > MIN_COLLECT_ENTROPY && (count & 0x3ff) != 0) {
136 		/*
137 		 * No need to add more entropy at this point, so save CPU and
138 		 * skip the update.
139 		 */
140 		return;
141 	}
142 	wpa_printf(MSG_EXCESSIVE, "Add randomness: count=%u entropy=%u",
143 		   count, entropy);
144 
145 	os_get_time(&t);
146 	wpa_hexdump_key(MSG_EXCESSIVE, "random pool",
147 			(const u8 *) pool, sizeof(pool));
148 	random_mix_pool(&t, sizeof(t));
149 	random_mix_pool(buf, len);
150 	wpa_hexdump_key(MSG_EXCESSIVE, "random pool",
151 			(const u8 *) pool, sizeof(pool));
152 	entropy++;
153 	total_collected++;
154 }
155 
156 
157 int random_get_bytes(void *buf, size_t len)
158 {
159 	int ret;
160 	u8 *bytes = buf;
161 	size_t left;
162 
163 	wpa_printf(MSG_MSGDUMP, "Get randomness: len=%u entropy=%u",
164 		   (unsigned int) len, entropy);
165 
166 	/* Start with assumed strong randomness from OS */
167 	ret = os_get_random(buf, len);
168 	wpa_hexdump_key(MSG_EXCESSIVE, "random from os_get_random",
169 			buf, len);
170 
171 	/* Mix in additional entropy extracted from the internal pool */
172 	left = len;
173 	while (left) {
174 		size_t siz, i;
175 		u8 tmp[EXTRACT_LEN];
176 		random_extract(tmp);
177 		wpa_hexdump_key(MSG_EXCESSIVE, "random from internal pool",
178 				tmp, sizeof(tmp));
179 		siz = left > EXTRACT_LEN ? EXTRACT_LEN : left;
180 		for (i = 0; i < siz; i++)
181 			*bytes++ ^= tmp[i];
182 		left -= siz;
183 	}
184 
185 #ifdef CONFIG_FIPS
186 	/* Mix in additional entropy from the crypto module */
187 	bytes = buf;
188 	left = len;
189 	while (left) {
190 		size_t siz, i;
191 		u8 tmp[EXTRACT_LEN];
192 		if (crypto_get_random(tmp, sizeof(tmp)) < 0) {
193 			wpa_printf(MSG_ERROR, "random: No entropy available "
194 				   "for generating strong random bytes");
195 			return -1;
196 		}
197 		wpa_hexdump_key(MSG_EXCESSIVE, "random from crypto module",
198 				tmp, sizeof(tmp));
199 		siz = left > EXTRACT_LEN ? EXTRACT_LEN : left;
200 		for (i = 0; i < siz; i++)
201 			*bytes++ ^= tmp[i];
202 		left -= siz;
203 	}
204 #endif /* CONFIG_FIPS */
205 
206 	wpa_hexdump_key(MSG_EXCESSIVE, "mixed random", buf, len);
207 
208 	if (entropy < len)
209 		entropy = 0;
210 	else
211 		entropy -= len;
212 
213 	return ret;
214 }
215 
216 
217 int random_pool_ready(void)
218 {
219 #ifdef __linux__
220 	int fd;
221 	ssize_t res;
222 
223 	/*
224 	 * Make sure that there is reasonable entropy available before allowing
225 	 * some key derivation operations to proceed.
226 	 */
227 
228 	if (dummy_key_avail == sizeof(dummy_key))
229 		return 1; /* Already initialized - good to continue */
230 
231 	/*
232 	 * Try to fetch some more data from the kernel high quality
233 	 * /dev/random. There may not be enough data available at this point,
234 	 * so use non-blocking read to avoid blocking the application
235 	 * completely.
236 	 */
237 	fd = open("/dev/random", O_RDONLY | O_NONBLOCK);
238 	if (fd < 0) {
239 		wpa_printf(MSG_ERROR, "random: Cannot open /dev/random: %s",
240 			   strerror(errno));
241 		return -1;
242 	}
243 
244 	res = read(fd, dummy_key + dummy_key_avail,
245 		   sizeof(dummy_key) - dummy_key_avail);
246 	if (res < 0) {
247 		wpa_printf(MSG_ERROR, "random: Cannot read from /dev/random: "
248 			   "%s", strerror(errno));
249 		res = 0;
250 	}
251 	wpa_printf(MSG_DEBUG, "random: Got %u/%u bytes from "
252 		   "/dev/random", (unsigned) res,
253 		   (unsigned) (sizeof(dummy_key) - dummy_key_avail));
254 	dummy_key_avail += res;
255 	close(fd);
256 
257 	if (dummy_key_avail == sizeof(dummy_key)) {
258 		if (own_pool_ready < MIN_READY_MARK)
259 			own_pool_ready = MIN_READY_MARK;
260 		random_write_entropy();
261 		return 1;
262 	}
263 
264 	wpa_printf(MSG_INFO, "random: Only %u/%u bytes of strong "
265 		   "random data available from /dev/random",
266 		   (unsigned) dummy_key_avail, (unsigned) sizeof(dummy_key));
267 
268 	if (own_pool_ready >= MIN_READY_MARK ||
269 	    total_collected + 10 * own_pool_ready > MIN_COLLECT_ENTROPY) {
270 		wpa_printf(MSG_INFO, "random: Allow operation to proceed "
271 			   "based on internal entropy");
272 		return 1;
273 	}
274 
275 	wpa_printf(MSG_INFO, "random: Not enough entropy pool available for "
276 		   "secure operations");
277 	return 0;
278 #else /* __linux__ */
279 	/* TODO: could do similar checks on non-Linux platforms */
280 	return 1;
281 #endif /* __linux__ */
282 }
283 
284 
285 void random_mark_pool_ready(void)
286 {
287 	own_pool_ready++;
288 	wpa_printf(MSG_DEBUG, "random: Mark internal entropy pool to be "
289 		   "ready (count=%u/%u)", own_pool_ready, MIN_READY_MARK);
290 	random_write_entropy();
291 }
292 
293 
294 #ifdef __linux__
295 
296 static void random_close_fd(void)
297 {
298 	if (random_fd >= 0) {
299 		eloop_unregister_read_sock(random_fd);
300 		close(random_fd);
301 		random_fd = -1;
302 	}
303 }
304 
305 
306 static void random_read_fd(int sock, void *eloop_ctx, void *sock_ctx)
307 {
308 	ssize_t res;
309 
310 	if (dummy_key_avail == sizeof(dummy_key)) {
311 		random_close_fd();
312 		return;
313 	}
314 
315 	res = read(sock, dummy_key + dummy_key_avail,
316 		   sizeof(dummy_key) - dummy_key_avail);
317 	if (res < 0) {
318 		wpa_printf(MSG_ERROR, "random: Cannot read from /dev/random: "
319 			   "%s", strerror(errno));
320 		return;
321 	}
322 
323 	wpa_printf(MSG_DEBUG, "random: Got %u/%u bytes from /dev/random",
324 		   (unsigned) res,
325 		   (unsigned) (sizeof(dummy_key) - dummy_key_avail));
326 	dummy_key_avail += res;
327 
328 	if (dummy_key_avail == sizeof(dummy_key)) {
329 		random_close_fd();
330 		if (own_pool_ready < MIN_READY_MARK)
331 			own_pool_ready = MIN_READY_MARK;
332 		random_write_entropy();
333 	}
334 }
335 
336 #endif /* __linux__ */
337 
338 
339 static void random_read_entropy(void)
340 {
341 	char *buf;
342 	size_t len;
343 
344 	if (!random_entropy_file)
345 		return;
346 
347 	buf = os_readfile(random_entropy_file, &len);
348 	if (buf == NULL)
349 		return; /* entropy file not yet available */
350 
351 	if (len != 1 + RANDOM_ENTROPY_SIZE) {
352 		wpa_printf(MSG_DEBUG, "random: Invalid entropy file %s",
353 			   random_entropy_file);
354 		os_free(buf);
355 		return;
356 	}
357 
358 	own_pool_ready = (u8) buf[0];
359 	random_add_randomness(buf + 1, RANDOM_ENTROPY_SIZE);
360 	random_entropy_file_read = 1;
361 	os_free(buf);
362 	wpa_printf(MSG_DEBUG, "random: Added entropy from %s "
363 		   "(own_pool_ready=%u)",
364 		   random_entropy_file, own_pool_ready);
365 }
366 
367 
368 static void random_write_entropy(void)
369 {
370 	char buf[RANDOM_ENTROPY_SIZE];
371 	FILE *f;
372 	u8 opr;
373 	int fail = 0;
374 
375 	if (!random_entropy_file)
376 		return;
377 
378 	if (random_get_bytes(buf, RANDOM_ENTROPY_SIZE) < 0)
379 		return;
380 
381 	f = fopen(random_entropy_file, "wb");
382 	if (f == NULL) {
383 		wpa_printf(MSG_ERROR, "random: Could not open entropy file %s "
384 			   "for writing", random_entropy_file);
385 		return;
386 	}
387 
388 	opr = own_pool_ready > 0xff ? 0xff : own_pool_ready;
389 	if (fwrite(&opr, 1, 1, f) != 1 ||
390 	    fwrite(buf, RANDOM_ENTROPY_SIZE, 1, f) != 1)
391 		fail = 1;
392 	fclose(f);
393 	if (fail) {
394 		wpa_printf(MSG_ERROR, "random: Could not write entropy data "
395 			   "to %s", random_entropy_file);
396 		return;
397 	}
398 
399 	wpa_printf(MSG_DEBUG, "random: Updated entropy file %s "
400 		   "(own_pool_ready=%u)",
401 		   random_entropy_file, own_pool_ready);
402 }
403 
404 
405 void random_init(const char *entropy_file)
406 {
407 	os_free(random_entropy_file);
408 	if (entropy_file)
409 		random_entropy_file = os_strdup(entropy_file);
410 	else
411 		random_entropy_file = NULL;
412 	random_read_entropy();
413 
414 #ifdef __linux__
415 	if (random_fd >= 0)
416 		return;
417 
418 	random_fd = open("/dev/random", O_RDONLY | O_NONBLOCK);
419 	if (random_fd < 0) {
420 		wpa_printf(MSG_ERROR, "random: Cannot open /dev/random: %s",
421 			   strerror(errno));
422 		return;
423 	}
424 	wpa_printf(MSG_DEBUG, "random: Trying to read entropy from "
425 		   "/dev/random");
426 
427 	eloop_register_read_sock(random_fd, random_read_fd, NULL, NULL);
428 #endif /* __linux__ */
429 
430 	random_write_entropy();
431 }
432 
433 
434 void random_deinit(void)
435 {
436 #ifdef __linux__
437 	random_close_fd();
438 #endif /* __linux__ */
439 	random_write_entropy();
440 	os_free(random_entropy_file);
441 	random_entropy_file = NULL;
442 }
443