mirror of
git://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2025-08-05 16:54:27 +00:00
dedf56d775
Improve arena based tests and add several C and asm tests with specific pattern. These tests would have failed without add_const verifier support. Also add several loop_inside_iter*() tests that are not related to add_const, but nice to have. Signed-off-by: Alexei Starovoitov <ast@kernel.org> Signed-off-by: Daniel Borkmann <daniel@iogearbox.net> Link: https://lore.kernel.org/bpf/20240613013815.953-5-alexei.starovoitov@gmail.com
59 lines
1.3 KiB
C
59 lines
1.3 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */
|
|
#define BPF_NO_KFUNC_PROTOTYPES
|
|
#include <vmlinux.h>
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
#include <bpf/bpf_core_read.h>
|
|
#include "bpf_experimental.h"
|
|
|
|
struct {
|
|
__uint(type, BPF_MAP_TYPE_ARENA);
|
|
__uint(map_flags, BPF_F_MMAPABLE);
|
|
__uint(max_entries, 100); /* number of pages */
|
|
} arena SEC(".maps");
|
|
|
|
#include "bpf_arena_htab.h"
|
|
|
|
void __arena *htab_for_user;
|
|
bool skip = false;
|
|
|
|
int zero = 0;
|
|
char __arena arr1[100000];
|
|
char arr2[1000];
|
|
|
|
SEC("syscall")
|
|
int arena_htab_llvm(void *ctx)
|
|
{
|
|
#if defined(__BPF_FEATURE_ADDR_SPACE_CAST) || defined(BPF_ARENA_FORCE_ASM)
|
|
struct htab __arena *htab;
|
|
char __arena *arr = arr1;
|
|
__u64 i;
|
|
|
|
htab = bpf_alloc(sizeof(*htab));
|
|
cast_kern(htab);
|
|
htab_init(htab);
|
|
|
|
cast_kern(arr);
|
|
|
|
/* first run. No old elems in the table */
|
|
for (i = zero; i < 100000 && can_loop; i++) {
|
|
htab_update_elem(htab, i, i);
|
|
arr[i] = i;
|
|
}
|
|
|
|
/* should replace some elems with new ones */
|
|
for (i = zero; i < 1000 && can_loop; i++) {
|
|
htab_update_elem(htab, i, i);
|
|
/* Access mem to make the verifier use bounded loop logic */
|
|
arr2[i] = i;
|
|
}
|
|
cast_user(htab);
|
|
htab_for_user = htab;
|
|
#else
|
|
skip = true;
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|