| // SPDX-License-Identifier: GPL-2.0 |
| /* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */ |
| |
| #include <vmlinux.h> |
| #include <bpf/bpf_helpers.h> |
| #include <bpf/bpf_tracing.h> |
| #include "bpf_misc.h" |
| #include "bpf_experimental.h" |
| #include "bpf_arena_common.h" |
| |
| #define ARENA_SIZE (1ull << 32) |
| |
| struct { |
| __uint(type, BPF_MAP_TYPE_ARENA); |
| __uint(map_flags, BPF_F_MMAPABLE); |
| __uint(max_entries, ARENA_SIZE / PAGE_SIZE); |
| } arena SEC(".maps"); |
| |
| SEC("syscall") |
| __success __retval(0) |
| int big_alloc1(void *ctx) |
| { |
| #if defined(__BPF_FEATURE_ADDR_SPACE_CAST) |
| volatile char __arena *page1, *page2, *no_page, *page3; |
| void __arena *base; |
| |
| page1 = base = bpf_arena_alloc_pages(&arena, NULL, 1, NUMA_NO_NODE, 0); |
| if (!page1) |
| return 1; |
| *page1 = 1; |
| page2 = bpf_arena_alloc_pages(&arena, base + ARENA_SIZE - PAGE_SIZE, |
| 1, NUMA_NO_NODE, 0); |
| if (!page2) |
| return 2; |
| *page2 = 2; |
| no_page = bpf_arena_alloc_pages(&arena, base + ARENA_SIZE, |
| 1, NUMA_NO_NODE, 0); |
| if (no_page) |
| return 3; |
| if (*page1 != 1) |
| return 4; |
| if (*page2 != 2) |
| return 5; |
| bpf_arena_free_pages(&arena, (void __arena *)page1, 1); |
| if (*page2 != 2) |
| return 6; |
| if (*page1 != 0) /* use-after-free should return 0 */ |
| return 7; |
| page3 = bpf_arena_alloc_pages(&arena, NULL, 1, NUMA_NO_NODE, 0); |
| if (!page3) |
| return 8; |
| *page3 = 3; |
| if (page1 != page3) |
| return 9; |
| if (*page2 != 2) |
| return 10; |
| if (*(page1 + PAGE_SIZE) != 0) |
| return 11; |
| if (*(page1 - PAGE_SIZE) != 0) |
| return 12; |
| if (*(page2 + PAGE_SIZE) != 0) |
| return 13; |
| if (*(page2 - PAGE_SIZE) != 0) |
| return 14; |
| #endif |
| return 0; |
| } |
| char _license[] SEC("license") = "GPL"; |
