| b.liu | e958203 | 2025-04-17 19:18:16 +0800 | [diff] [blame^] | 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
| 2 | #ifndef __BPF_RAND__ |
| 3 | #define __BPF_RAND__ |
| 4 | |
| 5 | #include <stdint.h> |
| 6 | #include <stdlib.h> |
| 7 | #include <time.h> |
| 8 | |
| 9 | static inline uint64_t bpf_rand_mask(uint64_t mask) |
| 10 | { |
| 11 | return (((uint64_t)(uint32_t)rand()) | |
| 12 | ((uint64_t)(uint32_t)rand() << 32)) & mask; |
| 13 | } |
| 14 | |
| 15 | #define bpf_rand_ux(x, m) \ |
| 16 | static inline uint64_t bpf_rand_u##x(int shift) \ |
| 17 | { \ |
| 18 | return bpf_rand_mask((m)) << shift; \ |
| 19 | } |
| 20 | |
| 21 | bpf_rand_ux( 8, 0xffULL) |
| 22 | bpf_rand_ux(16, 0xffffULL) |
| 23 | bpf_rand_ux(24, 0xffffffULL) |
| 24 | bpf_rand_ux(32, 0xffffffffULL) |
| 25 | bpf_rand_ux(40, 0xffffffffffULL) |
| 26 | bpf_rand_ux(48, 0xffffffffffffULL) |
| 27 | bpf_rand_ux(56, 0xffffffffffffffULL) |
| 28 | bpf_rand_ux(64, 0xffffffffffffffffULL) |
| 29 | |
| 30 | static inline void bpf_semi_rand_init(void) |
| 31 | { |
| 32 | srand(time(NULL)); |
| 33 | } |
| 34 | |
| 35 | static inline uint64_t bpf_semi_rand_get(void) |
| 36 | { |
| 37 | switch (rand() % 39) { |
| 38 | case 0: return 0x000000ff00000000ULL | bpf_rand_u8(0); |
| 39 | case 1: return 0xffffffff00000000ULL | bpf_rand_u16(0); |
| 40 | case 2: return 0x00000000ffff0000ULL | bpf_rand_u16(0); |
| 41 | case 3: return 0x8000000000000000ULL | bpf_rand_u32(0); |
| 42 | case 4: return 0x00000000f0000000ULL | bpf_rand_u32(0); |
| 43 | case 5: return 0x0000000100000000ULL | bpf_rand_u24(0); |
| 44 | case 6: return 0x800ff00000000000ULL | bpf_rand_u32(0); |
| 45 | case 7: return 0x7fffffff00000000ULL | bpf_rand_u32(0); |
| 46 | case 8: return 0xffffffffffffff00ULL ^ bpf_rand_u32(24); |
| 47 | case 9: return 0xffffffffffffff00ULL | bpf_rand_u8(0); |
| 48 | case 10: return 0x0000000010000000ULL | bpf_rand_u32(0); |
| 49 | case 11: return 0xf000000000000000ULL | bpf_rand_u8(0); |
| 50 | case 12: return 0x0000f00000000000ULL | bpf_rand_u8(8); |
| 51 | case 13: return 0x000000000f000000ULL | bpf_rand_u8(16); |
| 52 | case 14: return 0x0000000000000f00ULL | bpf_rand_u8(32); |
| 53 | case 15: return 0x00fff00000000f00ULL | bpf_rand_u8(48); |
| 54 | case 16: return 0x00007fffffffffffULL ^ bpf_rand_u32(1); |
| 55 | case 17: return 0xffff800000000000ULL | bpf_rand_u8(4); |
| 56 | case 18: return 0xffff800000000000ULL | bpf_rand_u8(20); |
| 57 | case 19: return (0xffffffc000000000ULL + 0x80000ULL) | bpf_rand_u32(0); |
| 58 | case 20: return (0xffffffc000000000ULL - 0x04000000ULL) | bpf_rand_u32(0); |
| 59 | case 21: return 0x0000000000000000ULL | bpf_rand_u8(55) | bpf_rand_u32(20); |
| 60 | case 22: return 0xffffffffffffffffULL ^ bpf_rand_u8(3) ^ bpf_rand_u32(40); |
| 61 | case 23: return 0x0000000000000000ULL | bpf_rand_u8(bpf_rand_u8(0) % 64); |
| 62 | case 24: return 0x0000000000000000ULL | bpf_rand_u16(bpf_rand_u8(0) % 64); |
| 63 | case 25: return 0xffffffffffffffffULL ^ bpf_rand_u8(bpf_rand_u8(0) % 64); |
| 64 | case 26: return 0xffffffffffffffffULL ^ bpf_rand_u40(bpf_rand_u8(0) % 64); |
| 65 | case 27: return 0x0000800000000000ULL; |
| 66 | case 28: return 0x8000000000000000ULL; |
| 67 | case 29: return 0x0000000000000000ULL; |
| 68 | case 30: return 0xffffffffffffffffULL; |
| 69 | case 31: return bpf_rand_u16(bpf_rand_u8(0) % 64); |
| 70 | case 32: return bpf_rand_u24(bpf_rand_u8(0) % 64); |
| 71 | case 33: return bpf_rand_u32(bpf_rand_u8(0) % 64); |
| 72 | case 34: return bpf_rand_u40(bpf_rand_u8(0) % 64); |
| 73 | case 35: return bpf_rand_u48(bpf_rand_u8(0) % 64); |
| 74 | case 36: return bpf_rand_u56(bpf_rand_u8(0) % 64); |
| 75 | case 37: return bpf_rand_u64(bpf_rand_u8(0) % 64); |
| 76 | default: return bpf_rand_u64(0); |
| 77 | } |
| 78 | } |
| 79 | |
| 80 | #endif /* __BPF_RAND__ */ |