1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/bpf.h>
3 #include <bpf/bpf_tracing.h>
4 #include <bpf/bpf_helpers.h>
5
6 char _license[] SEC("license") = "GPL";
7
8 struct net_device {
9 /* Structure does not need to contain all entries,
10 * as "preserve_access_index" will use BTF to fix this...
11 */
12 int ifindex;
13 } __attribute__((preserve_access_index));
14
15 struct xdp_rxq_info {
16 /* Structure does not need to contain all entries,
17 * as "preserve_access_index" will use BTF to fix this...
18 */
19 struct net_device *dev;
20 __u32 queue_index;
21 } __attribute__((preserve_access_index));
22
23 struct xdp_buff {
24 void *data;
25 void *data_end;
26 void *data_meta;
27 void *data_hard_start;
28 unsigned long handle;
29 struct xdp_rxq_info *rxq;
30 } __attribute__((preserve_access_index));
31
32 struct meta {
33 int ifindex;
34 int pkt_len;
35 };
36
37 struct {
38 __uint(type, BPF_MAP_TYPE_PERF_EVENT_ARRAY);
39 __type(key, int);
40 __type(value, int);
41 } perf_buf_map SEC(".maps");
42
43 __u64 test_result_fentry = 0;
44 SEC("fentry/FUNC")
BPF_PROG(trace_on_entry,struct xdp_buff * xdp)45 int BPF_PROG(trace_on_entry, struct xdp_buff *xdp)
46 {
47 struct meta meta;
48
49 meta.ifindex = xdp->rxq->dev->ifindex;
50 meta.pkt_len = bpf_xdp_get_buff_len((struct xdp_md *)xdp);
51 bpf_xdp_output(xdp, &perf_buf_map,
52 ((__u64) meta.pkt_len << 32) |
53 BPF_F_CURRENT_CPU,
54 &meta, sizeof(meta));
55
56 test_result_fentry = xdp->rxq->dev->ifindex;
57 return 0;
58 }
59
60 __u64 test_result_fexit = 0;
61 SEC("fexit/FUNC")
BPF_PROG(trace_on_exit,struct xdp_buff * xdp,int ret)62 int BPF_PROG(trace_on_exit, struct xdp_buff *xdp, int ret)
63 {
64 test_result_fexit = ret;
65 return 0;
66 }
67