xref: /xnu-10002.61.3/bsd/dev/dtrace/dtrace_xoroshiro128_plus.c (revision 0f4c859e951fba394238ab619495c4e1d54d0f34) !
1*0f4c859eSApple OSS Distributions /*-
2*0f4c859eSApple OSS Distributions  * Copyright (c) 2016 (Graeme Jenkinson)
3*0f4c859eSApple OSS Distributions  * All rights reserved.
4*0f4c859eSApple OSS Distributions  *
5*0f4c859eSApple OSS Distributions  * This software was developed by BAE Systems, the University of Cambridge
6*0f4c859eSApple OSS Distributions  * Computer Laboratory, and Memorial University under DARPA/AFRL contract
7*0f4c859eSApple OSS Distributions  * FA8650-15-C-7558 ("CADETS"), as part of the DARPA Transparent Computing
8*0f4c859eSApple OSS Distributions  * (TC) research program.
9*0f4c859eSApple OSS Distributions  *
10*0f4c859eSApple OSS Distributions  * Redistribution and use in source and binary forms, with or without
11*0f4c859eSApple OSS Distributions  * modification, are permitted provided that the following conditions
12*0f4c859eSApple OSS Distributions  * are met:
13*0f4c859eSApple OSS Distributions  * 1. Redistributions of source code must retain the above copyright
14*0f4c859eSApple OSS Distributions  *    notice, this list of conditions and the following disclaimer.
15*0f4c859eSApple OSS Distributions  * 2. Redistributions in binary form must reproduce the above copyright
16*0f4c859eSApple OSS Distributions  *    notice, this list of conditions and the following disclaimer in the
17*0f4c859eSApple OSS Distributions  *    documentation and/or other materials provided with the distribution.
18*0f4c859eSApple OSS Distributions  *
19*0f4c859eSApple OSS Distributions  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20*0f4c859eSApple OSS Distributions  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21*0f4c859eSApple OSS Distributions  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22*0f4c859eSApple OSS Distributions  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23*0f4c859eSApple OSS Distributions  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24*0f4c859eSApple OSS Distributions  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25*0f4c859eSApple OSS Distributions  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26*0f4c859eSApple OSS Distributions  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27*0f4c859eSApple OSS Distributions  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28*0f4c859eSApple OSS Distributions  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29*0f4c859eSApple OSS Distributions  * SUCH DAMAGE.
30*0f4c859eSApple OSS Distributions  *
31*0f4c859eSApple OSS Distributions  */
32*0f4c859eSApple OSS Distributions 
33*0f4c859eSApple OSS Distributions #include <sys/types.h>
34*0f4c859eSApple OSS Distributions 
35*0f4c859eSApple OSS Distributions #include "dtrace_xoroshiro128_plus.h"
36*0f4c859eSApple OSS Distributions 
37*0f4c859eSApple OSS Distributions static __inline uint64_t
rotl(const uint64_t x,int k)38*0f4c859eSApple OSS Distributions rotl(const uint64_t x, int k)
39*0f4c859eSApple OSS Distributions {
40*0f4c859eSApple OSS Distributions 	return (x << k) | (x >> (64 - k));
41*0f4c859eSApple OSS Distributions }
42*0f4c859eSApple OSS Distributions 
43*0f4c859eSApple OSS Distributions /*
44*0f4c859eSApple OSS Distributions  * This is the jump function for the generator. It is equivalent to 2^64 calls
45*0f4c859eSApple OSS Distributions  * to next(); it can be used to generate 2^64 non-overlapping subsequences for
46*0f4c859eSApple OSS Distributions  * parallel computations.
47*0f4c859eSApple OSS Distributions  */
48*0f4c859eSApple OSS Distributions void
dtrace_xoroshiro128_plus_jump(uint64_t * const state,uint64_t * const jump_state)49*0f4c859eSApple OSS Distributions dtrace_xoroshiro128_plus_jump(uint64_t * const state,
50*0f4c859eSApple OSS Distributions     uint64_t * const jump_state)
51*0f4c859eSApple OSS Distributions {
52*0f4c859eSApple OSS Distributions 	static const uint64_t JUMP[] = { 0xbeac0467eba5facb,
53*0f4c859eSApple OSS Distributions 		                         0xd86b048b86aa9922 };
54*0f4c859eSApple OSS Distributions 
55*0f4c859eSApple OSS Distributions 	uint64_t s0 = 0;
56*0f4c859eSApple OSS Distributions 	uint64_t s1 = 0;
57*0f4c859eSApple OSS Distributions 	size_t i = 0;
58*0f4c859eSApple OSS Distributions 	int b = 0;
59*0f4c859eSApple OSS Distributions 	for (i = 0; i < sizeof JUMP / sizeof *JUMP; i++) {
60*0f4c859eSApple OSS Distributions 		for (b = 0; b < 64; b++) {
61*0f4c859eSApple OSS Distributions 			if (JUMP[i] & 1ULL << b) {
62*0f4c859eSApple OSS Distributions 				s0 ^= state[0];
63*0f4c859eSApple OSS Distributions 				s1 ^= state[1];
64*0f4c859eSApple OSS Distributions 			}
65*0f4c859eSApple OSS Distributions 			dtrace_xoroshiro128_plus_next(state);
66*0f4c859eSApple OSS Distributions 		}
67*0f4c859eSApple OSS Distributions 	}
68*0f4c859eSApple OSS Distributions 	jump_state[0] = s0;
69*0f4c859eSApple OSS Distributions 	jump_state[1] = s1;
70*0f4c859eSApple OSS Distributions }
71*0f4c859eSApple OSS Distributions 
72*0f4c859eSApple OSS Distributions /*
73*0f4c859eSApple OSS Distributions  * xoroshiro128+ - XOR/rotate/shift/rotate
74*0f4c859eSApple OSS Distributions  * xorshift.di.unimi.it
75*0f4c859eSApple OSS Distributions  */
76*0f4c859eSApple OSS Distributions uint64_t
dtrace_xoroshiro128_plus_next(uint64_t * const state)77*0f4c859eSApple OSS Distributions dtrace_xoroshiro128_plus_next(uint64_t * const state)
78*0f4c859eSApple OSS Distributions {
79*0f4c859eSApple OSS Distributions 	const uint64_t s0 = state[0];
80*0f4c859eSApple OSS Distributions 	uint64_t s1 = state[1];
81*0f4c859eSApple OSS Distributions 	uint64_t result;
82*0f4c859eSApple OSS Distributions 	result = s0 + s1;
83*0f4c859eSApple OSS Distributions 
84*0f4c859eSApple OSS Distributions 	s1 ^= s0;
85*0f4c859eSApple OSS Distributions 	state[0] = rotl(s0, 55) ^ s1 ^ (s1 << 14);
86*0f4c859eSApple OSS Distributions 	state[1] = rotl(s1, 36);
87*0f4c859eSApple OSS Distributions 
88*0f4c859eSApple OSS Distributions 	return result;
89*0f4c859eSApple OSS Distributions }
90