pintos-os.org Git - pintos-anon/blob - 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 }