1 | // SPDX-License-Identifier: GPL-2.0 |
2 | /* Copyright (c) 2021 Facebook */ |
3 | #include <linux/bpf.h> |
4 | #include <time.h> |
5 | #include <errno.h> |
6 | #include <bpf/bpf_helpers.h> |
7 | #include "bpf_tcp_helpers.h" |
8 | |
9 | char _license[] SEC("license" ) = "GPL" ; |
10 | struct hmap_elem { |
11 | int pad; /* unused */ |
12 | struct bpf_timer timer; |
13 | }; |
14 | |
15 | struct inner_map { |
16 | __uint(type, BPF_MAP_TYPE_HASH); |
17 | __uint(max_entries, 1024); |
18 | __type(key, int); |
19 | __type(value, struct hmap_elem); |
20 | } inner_htab SEC(".maps" ); |
21 | |
22 | #define ARRAY_KEY 1 |
23 | #define HASH_KEY 1234 |
24 | |
25 | struct outer_arr { |
26 | __uint(type, BPF_MAP_TYPE_ARRAY_OF_MAPS); |
27 | __uint(max_entries, 2); |
28 | __uint(key_size, sizeof(int)); |
29 | __uint(value_size, sizeof(int)); |
30 | __array(values, struct inner_map); |
31 | } outer_arr SEC(".maps" ) = { |
32 | .values = { [ARRAY_KEY] = &inner_htab }, |
33 | }; |
34 | |
35 | __u64 err; |
36 | __u64 ok; |
37 | __u64 cnt; |
38 | |
39 | static int timer_cb1(void *map, int *key, struct hmap_elem *val); |
40 | |
41 | static int timer_cb2(void *map, int *key, struct hmap_elem *val) |
42 | { |
43 | cnt++; |
44 | bpf_timer_set_callback(&val->timer, timer_cb1); |
45 | if (bpf_timer_start(&val->timer, 1000, 0)) |
46 | err |= 1; |
47 | ok |= 1; |
48 | return 0; |
49 | } |
50 | |
51 | /* callback for inner hash map */ |
52 | static int timer_cb1(void *map, int *key, struct hmap_elem *val) |
53 | { |
54 | cnt++; |
55 | bpf_timer_set_callback(&val->timer, timer_cb2); |
56 | if (bpf_timer_start(&val->timer, 1000, 0)) |
57 | err |= 2; |
58 | /* Do a lookup to make sure 'map' and 'key' pointers are correct */ |
59 | bpf_map_lookup_elem(map, key); |
60 | ok |= 2; |
61 | return 0; |
62 | } |
63 | |
64 | SEC("fentry/bpf_fentry_test1" ) |
65 | int BPF_PROG(test1, int a) |
66 | { |
67 | struct hmap_elem init = {}; |
68 | struct bpf_map *inner_map; |
69 | struct hmap_elem *val; |
70 | int array_key = ARRAY_KEY; |
71 | int hash_key = HASH_KEY; |
72 | |
73 | inner_map = bpf_map_lookup_elem(&outer_arr, &array_key); |
74 | if (!inner_map) |
75 | return 0; |
76 | |
77 | bpf_map_update_elem(inner_map, &hash_key, &init, 0); |
78 | val = bpf_map_lookup_elem(inner_map, &hash_key); |
79 | if (!val) |
80 | return 0; |
81 | |
82 | bpf_timer_init(&val->timer, inner_map, CLOCK_MONOTONIC); |
83 | if (bpf_timer_set_callback(&val->timer, timer_cb1)) |
84 | err |= 4; |
85 | if (bpf_timer_start(&val->timer, 0, 0)) |
86 | err |= 8; |
87 | return 0; |
88 | } |
89 | |