1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2020 Facebook */
3
4 #include "vmlinux.h"
5 #include <bpf/bpf_helpers.h>
6 #include <bpf/bpf_tracing.h>
7 #include <bpf/bpf_core_read.h>
8 #include "../bpf_testmod/bpf_testmod.h"
9
10 __u32 raw_tp_read_sz = 0;
11
12 SEC("raw_tp/bpf_testmod_test_read")
BPF_PROG(handle_raw_tp,struct task_struct * task,struct bpf_testmod_test_read_ctx * read_ctx)13 int BPF_PROG(handle_raw_tp,
14 struct task_struct *task, struct bpf_testmod_test_read_ctx *read_ctx)
15 {
16 raw_tp_read_sz = BPF_CORE_READ(read_ctx, len);
17 return 0;
18 }
19
20 __u32 raw_tp_bare_write_sz = 0;
21
22 SEC("raw_tp/bpf_testmod_test_write_bare")
BPF_PROG(handle_raw_tp_bare,struct task_struct * task,struct bpf_testmod_test_write_ctx * write_ctx)23 int BPF_PROG(handle_raw_tp_bare,
24 struct task_struct *task, struct bpf_testmod_test_write_ctx *write_ctx)
25 {
26 raw_tp_bare_write_sz = BPF_CORE_READ(write_ctx, len);
27 return 0;
28 }
29
30 __u32 tp_btf_read_sz = 0;
31
32 SEC("tp_btf/bpf_testmod_test_read")
BPF_PROG(handle_tp_btf,struct task_struct * task,struct bpf_testmod_test_read_ctx * read_ctx)33 int BPF_PROG(handle_tp_btf,
34 struct task_struct *task, struct bpf_testmod_test_read_ctx *read_ctx)
35 {
36 tp_btf_read_sz = read_ctx->len;
37 return 0;
38 }
39
40 __u32 fentry_read_sz = 0;
41
42 SEC("fentry/bpf_testmod_test_read")
BPF_PROG(handle_fentry,struct file * file,struct kobject * kobj,struct bin_attribute * bin_attr,char * buf,loff_t off,size_t len)43 int BPF_PROG(handle_fentry,
44 struct file *file, struct kobject *kobj,
45 struct bin_attribute *bin_attr, char *buf, loff_t off, size_t len)
46 {
47 fentry_read_sz = len;
48 return 0;
49 }
50
51 __u32 fentry_manual_read_sz = 0;
52
53 SEC("fentry/placeholder")
BPF_PROG(handle_fentry_manual,struct file * file,struct kobject * kobj,struct bin_attribute * bin_attr,char * buf,loff_t off,size_t len)54 int BPF_PROG(handle_fentry_manual,
55 struct file *file, struct kobject *kobj,
56 struct bin_attribute *bin_attr, char *buf, loff_t off, size_t len)
57 {
58 fentry_manual_read_sz = len;
59 return 0;
60 }
61
62 __u32 fexit_read_sz = 0;
63 int fexit_ret = 0;
64
65 SEC("fexit/bpf_testmod_test_read")
BPF_PROG(handle_fexit,struct file * file,struct kobject * kobj,struct bin_attribute * bin_attr,char * buf,loff_t off,size_t len,int ret)66 int BPF_PROG(handle_fexit,
67 struct file *file, struct kobject *kobj,
68 struct bin_attribute *bin_attr, char *buf, loff_t off, size_t len,
69 int ret)
70 {
71 fexit_read_sz = len;
72 fexit_ret = ret;
73 return 0;
74 }
75
76 __u32 fmod_ret_read_sz = 0;
77
78 SEC("fmod_ret/bpf_testmod_test_read")
BPF_PROG(handle_fmod_ret,struct file * file,struct kobject * kobj,struct bin_attribute * bin_attr,char * buf,loff_t off,size_t len)79 int BPF_PROG(handle_fmod_ret,
80 struct file *file, struct kobject *kobj,
81 struct bin_attribute *bin_attr, char *buf, loff_t off, size_t len)
82 {
83 fmod_ret_read_sz = len;
84 return 0; /* don't override the exit code */
85 }
86
87 char _license[] SEC("license") = "GPL";
88