1 /* SPDX-License-Identifier: GPL-2.0 */
2 /* Copyright (c) 2018 Facebook */
3 #include <linux/bpf.h>
4 #include "bpf_helpers.h"
5 #include "bpf_legacy.h"
6
7 int _version SEC("version") = 1;
8
9 struct ipv_counts {
10 unsigned int v4;
11 unsigned int v6;
12 };
13
14 /* just to validate we can handle maps in multiple sections */
15 struct bpf_map_def SEC("maps") btf_map_legacy = {
16 .type = BPF_MAP_TYPE_ARRAY,
17 .key_size = sizeof(int),
18 .value_size = sizeof(long long),
19 .max_entries = 4,
20 };
21
22 BPF_ANNOTATE_KV_PAIR(btf_map_legacy, int, struct ipv_counts);
23
24 struct {
25 __uint(type, BPF_MAP_TYPE_ARRAY);
26 __uint(max_entries, 4);
27 __type(key, int);
28 __type(value, struct ipv_counts);
29 } btf_map SEC(".maps");
30
31 struct dummy_tracepoint_args {
32 unsigned long long pad;
33 struct sock *sock;
34 };
35
36 __attribute__((noinline))
test_long_fname_2(struct dummy_tracepoint_args * arg)37 int test_long_fname_2(struct dummy_tracepoint_args *arg)
38 {
39 struct ipv_counts *counts;
40 int key = 0;
41
42 if (!arg->sock)
43 return 0;
44
45 counts = bpf_map_lookup_elem(&btf_map, &key);
46 if (!counts)
47 return 0;
48
49 counts->v6++;
50
51 /* just verify we can reference both maps */
52 counts = bpf_map_lookup_elem(&btf_map_legacy, &key);
53 if (!counts)
54 return 0;
55
56 return 0;
57 }
58
59 __attribute__((noinline))
test_long_fname_1(struct dummy_tracepoint_args * arg)60 int test_long_fname_1(struct dummy_tracepoint_args *arg)
61 {
62 return test_long_fname_2(arg);
63 }
64
65 SEC("dummy_tracepoint")
_dummy_tracepoint(struct dummy_tracepoint_args * arg)66 int _dummy_tracepoint(struct dummy_tracepoint_args *arg)
67 {
68 return test_long_fname_1(arg);
69 }
70
71 char _license[] SEC("license") = "GPL";
72