linux/tools/testing/selftests/bpf/progs/struct_ops_private_stack.c

// SPDX-License-Identifier: GPL-2.0

#include <vmlinux.h>
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
#include "../test_kmods/bpf_testmod.h"

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

#if defined(__TARGET_ARCH_x86) || defined(__TARGET_ARCH_arm64)
bool skip __attribute((__section__(".data"))) = false;
#else
bool skip = true;
#endif

void bpf_testmod_ops3_call_test_2(void) __ksym;

int val_i, val_j;

__noinline static int subprog2(int *a, int *b)
{
	return val_i + a[10] + b[20];
}

__noinline static int subprog1(int *a)
{
	/* stack size 200 bytes */
	int b[50] = {};

	b[20] = 2;
	return subprog2(a, b);
}


SEC("struct_ops")
int BPF_PROG(test_1)
{
	/* stack size 400 bytes */
	int a[100] = {};

	a[10] = 1;
	val_i = subprog1(a);
	bpf_testmod_ops3_call_test_2();
	return 0;
}

SEC("struct_ops")
int BPF_PROG(test_2)
{
	/* stack size 200 bytes */
	int a[50] = {};

	a[10] = 3;
	val_j = subprog1(a);
	return 0;
}

SEC(".struct_ops")
struct bpf_testmod_ops3 testmod_1 = {
	.test_1 = (void *)test_1,
	.test_2 = (void *)test_2,
};