rand: add prng

This commit adds the ability to use random numbers with a seed, via the known methods rand() and srand(). As generator, a simple linear congruential generator is used, with the same parameters as used in musl or newlib. Signed-off-by: Norbert Manthey <[email protected]>
KernelTestFramework · Sep 29, 2020 · 8e03420 · 8e03420
1 parent ed189f8
commit 8e03420
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/include/lib.h b/include/lib.h
@@ -385,4 +385,8 @@ static inline unsigned int next_power_of_two(unsigned int n) {
 
 extern void halt(void);
 
+extern void srand(unsigned s);
+
+extern int rand(void);
+
 #endif /* KTF_LIB_H */
diff --git a/lib/lib.c b/lib/lib.c
@@ -33,3 +33,16 @@ void __noreturn halt(void) {
         pause();
     }
 }
+
+static uint64_t seed;
+
+void srand(unsigned s)
+{
+	seed = s-1;
+}
+
+int rand(void)
+{
+	seed = 6364136223846793005ULL*seed + 1;
+	return seed>>33;
+}