src/util.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#include "util.h"

#include <linux/random.h>

void
random_perm(uint32_t *arr, uint32_t arr_len)
{
	uint32_t i, idx, tmp;

	for (i = arr_len - 1; i > 0; --i) {
		get_random_bytes(&idx, 4);
		idx = idx % i;

		tmp = arr[idx];
		arr[idx] = arr[i];
		arr[i] = tmp;
	}
}

void
gen_random_indices(uint32_t *arr, uint32_t arr_len)
{
	uint32_t i;

	for (i = 0; i < arr_len; ++i)
		arr[i] = i;
	random_perm(arr, arr_len);
}


bool is_in_arr(uint32_t elem, uint32_t *arr, uint32_t arr_len) {
	uint32_t i;

	for (i = 0; i < arr_len; ++i) {
		if (arr[i] == elem)
			return true;
	}

	return false;
}