// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2025 ChinaTelecom */
#include <vmlinux.h>
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>

char _license[] SEC("license") = "GPL";

__u64 test1_entry_result = 0;
__u64 test1_exit_result = 0;

SEC("fsession/bpf_fentry_test1")
int BPF_PROG(test1, int a, int ret)
{
	bool is_exit = bpf_session_is_return(ctx);

	if (!is_exit) {
		test1_entry_result = a == 1 && ret == 0;
		return 0;
	}

	test1_exit_result = a == 1 && ret == 2;
	return 0;
}

__u64 test2_entry_result = 0;
__u64 test2_exit_result = 0;

SEC("fsession/bpf_fentry_test3")
int BPF_PROG(test2, char a, int b, __u64 c, int ret)
{
	bool is_exit = bpf_session_is_return(ctx);

	if (!is_exit) {
		test2_entry_result = a == 4 && b == 5 && c == 6 && ret == 0;
		return 0;
	}

	test2_exit_result = a == 4 && b == 5 && c == 6 && ret == 15;
	return 0;
}

__u64 test3_entry_result = 0;
__u64 test3_exit_result = 0;

SEC("fsession/bpf_fentry_test4")
int BPF_PROG(test3, void *a, char b, int c, __u64 d, int ret)
{
	bool is_exit = bpf_session_is_return(ctx);

	if (!is_exit) {
		test3_entry_result = a == (void *)7 && b == 8 && c == 9 && d == 10 && ret == 0;
		return 0;
	}

	test3_exit_result = a == (void *)7 && b == 8 && c == 9 && d == 10 && ret == 34;
	return 0;
}

__u64 test4_entry_result = 0;
__u64 test4_exit_result = 0;

SEC("fsession/bpf_fentry_test5")
int BPF_PROG(test4, __u64 a, void *b, short c, int d, __u64 e, int ret)
{
	bool is_exit = bpf_session_is_return(ctx);

	if (!is_exit) {
		test4_entry_result = a == 11 && b == (void *)12 && c == 13 && d == 14 &&
			e == 15 && ret == 0;
		return 0;
	}

	test4_exit_result = a == 11 && b == (void *)12 && c == 13 && d == 14 &&
		e == 15 && ret == 65;
	return 0;
}

__u64 test5_entry_result = 0;
__u64 test5_exit_result = 0;

SEC("fsession/bpf_fentry_test7")
int BPF_PROG(test5, struct bpf_fentry_test_t *arg, int ret)
{
	bool is_exit = bpf_session_is_return(ctx);

	if (!is_exit) {
		if (!arg)
			test5_entry_result = ret == 0;
		return 0;
	}

	if (!arg)
		test5_exit_result = 1;
	return 0;
}

__u64 test6_entry_result = 0;
__u64 test6_exit_result = 0;
SEC("fsession/bpf_fentry_test1")
int BPF_PROG(test6, int a)
{
	__u64 addr = bpf_get_func_ip(ctx);

	if (bpf_session_is_return(ctx))
		test6_exit_result = (const void *) addr == &bpf_fentry_test1;
	else
		test6_entry_result = (const void *) addr == &bpf_fentry_test1;
	return 0;
}

__u64 test7_entry_ok = 0;
__u64 test7_exit_ok = 0;
SEC("fsession/bpf_fentry_test1")
int BPF_PROG(test7, int a)
{
	volatile __u64 *cookie = bpf_session_cookie(ctx);

	if (!bpf_session_is_return(ctx)) {
		*cookie = 0xAAAABBBBCCCCDDDDull;
		test7_entry_ok = *cookie == 0xAAAABBBBCCCCDDDDull;
		return 0;
	}

	test7_exit_ok = *cookie == 0xAAAABBBBCCCCDDDDull;
	return 0;
}

__u64 test8_entry_ok = 0;
__u64 test8_exit_ok = 0;

SEC("fsession/bpf_fentry_test1")
int BPF_PROG(test8, int a)
{
	volatile __u64 *cookie = bpf_session_cookie(ctx);

	if (!bpf_session_is_return(ctx)) {
		*cookie = 0x1111222233334444ull;
		test8_entry_ok = *cookie == 0x1111222233334444ull;
		return 0;
	}

	test8_exit_ok = *cookie == 0x1111222233334444ull;
	return 0;
}

__u64 test9_entry_result = 0;
__u64 test9_exit_result = 0;

SEC("fsession/bpf_fentry_test1")
int BPF_PROG(test9, int a, int ret)
{
	__u64 *cookie = bpf_session_cookie(ctx);

	if (!bpf_session_is_return(ctx)) {
		test9_entry_result = a == 1 && ret == 0;
		*cookie = 0x123456ULL;
		return 0;
	}

	test9_exit_result = a == 1 && ret == 2 && *cookie == 0x123456ULL;
	return 0;
}

__u64 test10_result = 0;
SEC("fexit/bpf_fentry_test1")
int BPF_PROG(test10, int a, int ret)
{
	test10_result = a == 1 && ret == 2;
	return 0;
}

__u64 test11_result = 0;
SEC("fentry/bpf_fentry_test1")
int BPF_PROG(test11, int a)
{
	test11_result = a == 1;
	return 0;
}