Destroy PT pages in addrspace_destroy().
[pintos-anon] / src / lib / random.c
1 #include "random.h"
2 #include <stdbool.h>
3 #include <stdint.h>
4 #include "debug.h"
5
6 /* RC4-based pseudo-random state. */
7 static uint8_t s[256];
8 static uint8_t s_i, s_j;
9 static bool inited;
10
11 static inline void
12 swap_byte (uint8_t *a, uint8_t *b) 
13 {
14   uint8_t t = *a;
15   *a = *b;
16   *b = t;
17 }
18
19 void
20 random_init (unsigned seed)
21 {
22   uint8_t *seedp = (uint8_t *) &seed;
23   int i;
24   uint8_t j;
25
26   if (inited)
27     return;
28   
29   for (i = 0; i < 256; i++) 
30     s[i] = i;
31   for (i = j = 0; i < 256; i++) 
32     {
33       j += s[i] + seedp[i % sizeof seed];
34       swap_byte (s + i, s + j);
35     }
36
37   s_i = s_j = 0;
38   inited = true;
39 }
40
41 void
42 random_bytes (void *buf_, size_t size) 
43 {
44   uint8_t *buf;
45
46   ASSERT (inited);
47   for (buf = buf_; size-- > 0; buf++)
48     {
49       uint8_t s_k;
50       
51       s_i++;
52       s_j += s[s_i];
53       swap_byte (s + s_i, s + s_j);
54
55       s_k = s[s_i] + s[s_j];
56       *buf = s[s_k];
57     }
58 }
59
60 /* Returns a pseudo-random unsigned long. */
61 unsigned long
62 random_ulong (void) 
63 {
64   unsigned long ul;
65   random_bytes (&ul, sizeof ul);
66   return ul;
67 }