t/rng.c

   1 /*
   2  * Simple random number generator for testing.
   3  * Copyright © 2022-2024 Nick Bowler
   4  *
   5  * This program is free software: you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation, either version 3 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  17  *
  18  * The RNG implementation is adapted from xoshiro256+ and splitmix64
  19  * by David Blackman and Sebastiano Vigna, originally distributed under
  20  * the Creative Commons Zero public domain dedication.
  21  */
  22
  23 #include <config.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <string.h>
  27 #include <errno.h>
  28 #include <limits.h>
  29
  30 #if !TEST_RNG_NO_EXTERNAL_API
  31 #  include "test.h"
  32 #endif
  33
  34 #define B64(x) ((x) & 0xffffffffffffffffull)
  35
  36 struct test_rng {
  37         unsigned long long state[4];
  38 };
  39
  40 /* Rotate val left by n bits.  The behaviour is undefined if n is zero. */
  41 static unsigned long long rot_left64(unsigned long long val, int n)
  42 {
  43         return B64( (val << n) | (val >> (64 - n)) );
  44 }
  45
  46 static unsigned long long xoshiro256p(unsigned long long *s)
  47 {
  48         unsigned long long tmp, ret;
  49
  50         ret = B64(s[0] + s[3]);
  51         tmp = B64(s[1] << 17);
  52
  53         s[2] ^= s[0];
  54         s[3] ^= s[1];
  55         s[1] ^= s[2];
  56         s[0] ^= s[3];
  57         s[2] ^= tmp;
  58         s[3] = rot_left64(s[3], 45);
  59
  60         return ret;
  61 }
  62
  63 static unsigned long long splitmix64(unsigned long long *state)
  64 {
  65         unsigned long long z;
  66
  67         z = B64(*state += 0x9e3779b97f4a7c15ull);
  68         z = B64((z ^ (z >> 30)) * 0xbf58476d1ce4e5b9ull);
  69         z = B64((z ^ (z >> 27)) * 0x94d049bb133111ebull);
  70
  71         return z ^ (z >> 31);
  72 }
  73
  74 #if !TEST_RNG_NO_EXTERNAL_API
  75 struct test_rng *test_rng_alloc(const char *seed_str)
  76 {
  77         test_uintmax limit, seed_val;
  78         unsigned long long seed;
  79         struct test_rng *rng;
  80
  81         limit  = 0xffffffff;
  82         limit |= (limit << 16 << 16);
  83
  84         if (!test_strtoumax(&seed_val, seed_str, limit)) {
  85                 print_error("%s: invalid seed", seed_str);
  86                 return NULL;
  87         }
  88         seed = seed_val;
  89
  90         rng = malloc_nofail(sizeof *rng);
  91         rng->state[0] = splitmix64(&seed);
  92         rng->state[1] = splitmix64(&seed);
  93         rng->state[2] = splitmix64(&seed);
  94         rng->state[3] = splitmix64(&seed);
  95
  96         return rng;
  97 }
  98
  99 void test_rng_free(struct test_rng *rng)
 100 {
 101         free(rng);
 102 }
 103
 104 /* Calculate the least power of two greater than val, minus 1. */
 105 static unsigned rng_mask(unsigned val)
 106 {
 107         val |= val >> 1;
 108         val |= val >> 2;
 109         val |= val >> 4;
 110         val |= val >> 8;
 111
 112         if (UINT_MAX >= 65536)
 113                 val |= val >> 16;
 114
 115         return val;
 116 }
 117
 118 unsigned test_rng_uniform_int(struct test_rng *rng, unsigned max)
 119 {
 120         unsigned mask = rng_mask(max-1);
 121         unsigned long long val;
 122
 123         do {
 124                 val = ( xoshiro256p(rng->state) >> 32 ) & mask;
 125         } while (val >= max);
 126
 127         return val;
 128 }
 129 #endif