// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2026 Meta Platforms, Inc. and affiliates. */

#include <vmlinux.h>
#include <bpf/bpf_helpers.h>
#include "../test_kmods/bpf_testmod_kfunc.h"
#include "bpf_misc.h"

#if defined(__BPF_FEATURE_STACK_ARGUMENT)

SEC("tc")
__failure __msg("Unrecognized *(R11-8) type STRUCT")
int test_stack_arg_big(struct __sk_buff *skb)
{
	struct prog_test_big_arg s = { .a = 1, .b = 2 };

	return bpf_kfunc_call_stack_arg_big(1, 2, 3, 4, 5, s);
}

SEC("socket")
__description("r11 in ALU instruction")
__failure __msg("R11 is invalid")
__naked void r11_alu_reject(void)
{
	asm volatile (
	"r11 += 1;"
	"r0 = 0;"
	"exit;"
	::: __clobber_all);
}

SEC("socket")
__description("r11 store with non-DW size")
__failure __msg("R11 is invalid")
__naked void r11_store_non_dw(void)
{
	asm volatile (
	"*(u32 *)(r11 - 8) = r1;"
	"r0 = 0;"
	"exit;"
	::: __clobber_all);
}

SEC("socket")
__description("r11 store with unaligned offset")
__failure __msg("R11 is invalid")
__naked void r11_store_unaligned(void)
{
	asm volatile (
	"*(u64 *)(r11 - 4) = r1;"
	"r0 = 0;"
	"exit;"
	::: __clobber_all);
}

SEC("socket")
__description("r11 store with positive offset")
__failure __msg("R11 is invalid")
__naked void r11_store_positive_off(void)
{
	asm volatile (
	"*(u64 *)(r11 + 8) = r1;"
	"r0 = 0;"
	"exit;"
	::: __clobber_all);
}

SEC("socket")
__description("r11 load with negative offset")
__failure __msg("R11 is invalid")
__naked void r11_load_negative_off(void)
{
	asm volatile (
	"r0 = *(u64 *)(r11 - 8);"
	"exit;"
	::: __clobber_all);
}

SEC("socket")
__description("r11 load with non-DW size")
__failure __msg("R11 is invalid")
__naked void r11_load_non_dw(void)
{
	asm volatile (
	"r0 = *(u32 *)(r11 + 8);"
	"exit;"
	::: __clobber_all);
}

SEC("socket")
__description("r11 store with zero offset")
__failure __msg("R11 is invalid")
__naked void r11_store_zero_off(void)
{
	asm volatile (
	"*(u64 *)(r11 + 0) = r1;"
	"r0 = 0;"
	"exit;"
	::: __clobber_all);
}

#else

SEC("tc")
__description("stack_arg_fail: not supported, dummy test")
__success
int test_stack_arg_big(struct __sk_buff *skb)
{
	return 0;
}

#endif

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