| // 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"; |