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