pintos-os.org Git - pintos-anon/blob - src/lib/random.c

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