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
12SEC("raw_tp/bpf_testmod_test_read")
13int 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
22SEC("raw_tp/bpf_testmod_test_write_bare")
23int 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
32SEC("tp_btf/bpf_testmod_test_read")
33int 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
42SEC("fentry/bpf_testmod_test_read")
43int 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
53SEC("fentry/placeholder")
54int 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;
63int fexit_ret = 0;
64
65SEC("fexit/bpf_testmod_test_read")
66int 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
78SEC("fmod_ret/bpf_testmod_test_read")
79int 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
87char _license[] SEC("license") = "GPL";