Add Intel rdrand support
[fio.git] / lib / rand.h
1 #ifndef FIO_RAND_H
2 #define FIO_RAND_H
3
4 #include "../arch/arch.h"
5
6 #define FRAND_MAX       (-1U)
7
8 struct frand_state {
9         unsigned int s1, s2, s3;
10 };
11
12 static inline void frand_copy(struct frand_state *dst,
13                               struct frand_state *src)
14 {
15         dst->s1 = src->s1;
16         dst->s2 = src->s2;
17         dst->s3 = src->s3;
18 }
19
20 static inline unsigned int __rand(struct frand_state *state)
21 {
22 #define TAUSWORTHE(s,a,b,c,d) ((s&c)<<d) ^ (((s <<a) ^ s)>>b)
23
24         state->s1 = TAUSWORTHE(state->s1, 13, 19, 4294967294UL, 12);
25         state->s2 = TAUSWORTHE(state->s2, 2, 25, 4294967288UL, 4);
26         state->s3 = TAUSWORTHE(state->s3, 3, 11, 4294967280UL, 17);
27
28         return (state->s1 ^ state->s2 ^ state->s3);
29 }
30
31 extern void init_rand(struct frand_state *);
32 extern void init_rand_seed(struct frand_state *, unsigned int seed);
33 extern void __fill_random_buf(void *buf, unsigned int len, unsigned long seed);
34 extern unsigned long fill_random_buf(struct frand_state *, void *buf, unsigned int len);
35 extern void __fill_random_buf_percentage(unsigned long, void *, unsigned int, unsigned int, unsigned int, char *, unsigned int);
36 extern unsigned long fill_random_buf_percentage(struct frand_state *, void *, unsigned int, unsigned int, unsigned int, char *, unsigned int);
37 extern void fill_pattern(void *p, unsigned int len, char *pattern, unsigned int pattern_bytes);
38
39 #endif