1 /* Copyright (c) 2016 Facebook
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of version 2 of the GNU General Public
5 * License as published by the Free Software Foundation.
7 #include <linux/skbuff.h>
8 #include <linux/netdevice.h>
9 #include <linux/version.h>
10 #include <uapi/linux/bpf.h>
11 #include "bpf_helpers.h"
13 #define MAX_ENTRIES 1000
15 struct bpf_map_def
SEC("maps") hash_map
= {
16 .type
= BPF_MAP_TYPE_HASH
,
17 .key_size
= sizeof(u32
),
18 .value_size
= sizeof(long),
19 .max_entries
= MAX_ENTRIES
,
22 struct bpf_map_def
SEC("maps") lru_hash_map
= {
23 .type
= BPF_MAP_TYPE_LRU_HASH
,
24 .key_size
= sizeof(u32
),
25 .value_size
= sizeof(long),
29 struct bpf_map_def
SEC("maps") percpu_lru_hash_map
= {
30 .type
= BPF_MAP_TYPE_LRU_HASH
,
31 .key_size
= sizeof(u32
),
32 .value_size
= sizeof(long),
34 .map_flags
= BPF_F_NO_COMMON_LRU
,
37 struct bpf_map_def
SEC("maps") percpu_hash_map
= {
38 .type
= BPF_MAP_TYPE_PERCPU_HASH
,
39 .key_size
= sizeof(u32
),
40 .value_size
= sizeof(long),
41 .max_entries
= MAX_ENTRIES
,
44 struct bpf_map_def
SEC("maps") hash_map_alloc
= {
45 .type
= BPF_MAP_TYPE_HASH
,
46 .key_size
= sizeof(u32
),
47 .value_size
= sizeof(long),
48 .max_entries
= MAX_ENTRIES
,
49 .map_flags
= BPF_F_NO_PREALLOC
,
52 struct bpf_map_def
SEC("maps") percpu_hash_map_alloc
= {
53 .type
= BPF_MAP_TYPE_PERCPU_HASH
,
54 .key_size
= sizeof(u32
),
55 .value_size
= sizeof(long),
56 .max_entries
= MAX_ENTRIES
,
57 .map_flags
= BPF_F_NO_PREALLOC
,
60 struct bpf_map_def
SEC("maps") lpm_trie_map_alloc
= {
61 .type
= BPF_MAP_TYPE_LPM_TRIE
,
63 .value_size
= sizeof(long),
65 .map_flags
= BPF_F_NO_PREALLOC
,
68 SEC("kprobe/sys_getuid")
69 int stress_hmap(struct pt_regs
*ctx
)
71 u32 key
= bpf_get_current_pid_tgid();
75 bpf_map_update_elem(&hash_map
, &key
, &init_val
, BPF_ANY
);
76 value
= bpf_map_lookup_elem(&hash_map
, &key
);
78 bpf_map_delete_elem(&hash_map
, &key
);
83 SEC("kprobe/sys_geteuid")
84 int stress_percpu_hmap(struct pt_regs
*ctx
)
86 u32 key
= bpf_get_current_pid_tgid();
90 bpf_map_update_elem(&percpu_hash_map
, &key
, &init_val
, BPF_ANY
);
91 value
= bpf_map_lookup_elem(&percpu_hash_map
, &key
);
93 bpf_map_delete_elem(&percpu_hash_map
, &key
);
96 SEC("kprobe/sys_getgid")
97 int stress_hmap_alloc(struct pt_regs
*ctx
)
99 u32 key
= bpf_get_current_pid_tgid();
103 bpf_map_update_elem(&hash_map_alloc
, &key
, &init_val
, BPF_ANY
);
104 value
= bpf_map_lookup_elem(&hash_map_alloc
, &key
);
106 bpf_map_delete_elem(&hash_map_alloc
, &key
);
110 SEC("kprobe/sys_getegid")
111 int stress_percpu_hmap_alloc(struct pt_regs
*ctx
)
113 u32 key
= bpf_get_current_pid_tgid();
117 bpf_map_update_elem(&percpu_hash_map_alloc
, &key
, &init_val
, BPF_ANY
);
118 value
= bpf_map_lookup_elem(&percpu_hash_map_alloc
, &key
);
120 bpf_map_delete_elem(&percpu_hash_map_alloc
, &key
);
124 SEC("kprobe/sys_getpid")
125 int stress_lru_hmap_alloc(struct pt_regs
*ctx
)
127 u32 key
= bpf_get_prandom_u32();
130 bpf_map_update_elem(&lru_hash_map
, &key
, &val
, BPF_ANY
);
135 SEC("kprobe/sys_getppid")
136 int stress_percpu_lru_hmap_alloc(struct pt_regs
*ctx
)
138 u32 key
= bpf_get_prandom_u32();
141 bpf_map_update_elem(&percpu_lru_hash_map
, &key
, &val
, BPF_ANY
);
146 SEC("kprobe/sys_gettid")
147 int stress_lpm_trie_map_alloc(struct pt_regs
*ctx
)
161 #pragma clang loop unroll(full)
162 for (i
= 0; i
< 32; ++i
)
163 bpf_map_lookup_elem(&lpm_trie_map_alloc
, &key
);
168 char _license
[] SEC("license") = "GPL";
169 u32 _version
SEC("version") = LINUX_VERSION_CODE
;