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