GNU Linux-libre 4.9.332-gnu1
[releases.git] / samples / bpf / map_perf_test_kern.c
1 /* Copyright (c) 2016 Facebook
2  *
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.
6  */
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"
12
13 #define MAX_ENTRIES 1000
14
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,
20 };
21
22 struct bpf_map_def SEC("maps") percpu_hash_map = {
23         .type = BPF_MAP_TYPE_PERCPU_HASH,
24         .key_size = sizeof(u32),
25         .value_size = sizeof(long),
26         .max_entries = MAX_ENTRIES,
27 };
28
29 struct bpf_map_def SEC("maps") hash_map_alloc = {
30         .type = BPF_MAP_TYPE_HASH,
31         .key_size = sizeof(u32),
32         .value_size = sizeof(long),
33         .max_entries = MAX_ENTRIES,
34         .map_flags = BPF_F_NO_PREALLOC,
35 };
36
37 struct bpf_map_def SEC("maps") percpu_hash_map_alloc = {
38         .type = BPF_MAP_TYPE_PERCPU_HASH,
39         .key_size = sizeof(u32),
40         .value_size = sizeof(long),
41         .max_entries = MAX_ENTRIES,
42         .map_flags = BPF_F_NO_PREALLOC,
43 };
44
45 SEC("kprobe/sys_getuid")
46 int stress_hmap(struct pt_regs *ctx)
47 {
48         u32 key = bpf_get_current_pid_tgid();
49         long init_val = 1;
50         long *value;
51
52         bpf_map_update_elem(&hash_map, &key, &init_val, BPF_ANY);
53         value = bpf_map_lookup_elem(&hash_map, &key);
54         if (value)
55                 bpf_map_delete_elem(&hash_map, &key);
56         return 0;
57 }
58
59 SEC("kprobe/sys_geteuid")
60 int stress_percpu_hmap(struct pt_regs *ctx)
61 {
62         u32 key = bpf_get_current_pid_tgid();
63         long init_val = 1;
64         long *value;
65
66         bpf_map_update_elem(&percpu_hash_map, &key, &init_val, BPF_ANY);
67         value = bpf_map_lookup_elem(&percpu_hash_map, &key);
68         if (value)
69                 bpf_map_delete_elem(&percpu_hash_map, &key);
70         return 0;
71 }
72 SEC("kprobe/sys_getgid")
73 int stress_hmap_alloc(struct pt_regs *ctx)
74 {
75         u32 key = bpf_get_current_pid_tgid();
76         long init_val = 1;
77         long *value;
78
79         bpf_map_update_elem(&hash_map_alloc, &key, &init_val, BPF_ANY);
80         value = bpf_map_lookup_elem(&hash_map_alloc, &key);
81         if (value)
82                 bpf_map_delete_elem(&hash_map_alloc, &key);
83         return 0;
84 }
85
86 SEC("kprobe/sys_getegid")
87 int stress_percpu_hmap_alloc(struct pt_regs *ctx)
88 {
89         u32 key = bpf_get_current_pid_tgid();
90         long init_val = 1;
91         long *value;
92
93         bpf_map_update_elem(&percpu_hash_map_alloc, &key, &init_val, BPF_ANY);
94         value = bpf_map_lookup_elem(&percpu_hash_map_alloc, &key);
95         if (value)
96                 bpf_map_delete_elem(&percpu_hash_map_alloc, &key);
97         return 0;
98 }
99 char _license[] SEC("license") = "GPL";
100 u32 _version SEC("version") = LINUX_VERSION_CODE;