rand: add rand64_between()

[fio.git] / lib / rand.h

diff --git a/lib/rand.h b/lib/rand.h
index 8832c73a7687412241dbe3e9bddbdad6ba5c2b2b..6be53072a64751c4ae864a439e33a374205e3c4b 100644 (file)
--- a/lib/rand.h
+++ b/lib/rand.h
@@ -127,6 +127,24 @@ static inline int rand32_between(struct frand_state *state, int start, int end)
        return start + (int) ((double)end * (r / (FRAND32_MAX + 1.0)));
 }
 
+static inline uint64_t rand64_between(struct frand_state *state, uint64_t start,
+                                     uint64_t end)
+{
+       uint64_t r;
+
+       r = __rand64(&state->state64);
+       return start + (uint64_t) ((double)end * (r / (FRAND64_MAX + 1.0)));
+}
+
+static inline uint64_t rand_between(struct frand_state *state, uint64_t start,
+                                   uint64_t end)
+{
+       if (state->use64)
+               return rand32_between(state, start, end);
+       else
+               return rand64_between(state, start, end);
+}
+
 extern void init_rand(struct frand_state *, bool);
 extern void init_rand_seed(struct frand_state *, unsigned int seed, bool);
 extern void __fill_random_buf(void *buf, unsigned int len, unsigned long seed);