mirror of
https://github.com/boykopovar/AnyPS5.git
synced 2026-10-22 00:22:57 -05:00
Adelmo Junior
72211b70fe
Read the optional aligned budget from param.json and apply it consistently to configured, available and mapped flexible memory. Add parser and allocation regression coverage.
90 lines
4.5 KiB
C++
90 lines
4.5 KiB
C++
#include "prx/libc/include/general/VabiMacros.hpp"
|
|
#include "prx/libc/include/General.hpp"
|
|
#include "prx/libkernel/DirectMemory/DirectMemory.hpp"
|
|
#include "prx/libkernel/KernelErrors.hpp"
|
|
#include <cstdint>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <filesystem>
|
|
#include <source_location>
|
|
#include <stdexcept>
|
|
#include <string_view>
|
|
|
|
extern "C" {
|
|
int APS5_VABI sceKernelConfiguredFlexibleMemorySize(std::size_t*);
|
|
int APS5_VABI sceKernelAvailableFlexibleMemorySize(std::size_t*);
|
|
int APS5_VABI sceKernelMapFlexibleMemory(void**, std::size_t, int, int);
|
|
int APS5_VABI sceKernelReleaseFlexibleMemory(void*, std::size_t);
|
|
std::uint64_t GuestImageSegmentBytes_nid_no_patch();
|
|
void* APS5_VABI dlopen_nid_postfix(const char*, int);
|
|
char* APS5_VABI dlerror_nid_postfix();
|
|
int APS5_VABI dlclose_nid_postfix(void*);
|
|
}
|
|
|
|
static void Require(bool condition, std::source_location location = std::source_location::current()) {
|
|
if (!condition) {
|
|
std::fprintf(stderr, "Flexible memory budget check failed at %s:%u\n", location.file_name(), static_cast<unsigned>(location.line()));
|
|
std::abort();
|
|
}
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
Require(argc == 2);
|
|
const std::string_view mode = argv[1];
|
|
Require(sceKernelConfiguredFlexibleMemorySize(nullptr) == SCE_KERNEL_ERROR_EINVAL);
|
|
Require(sceKernelAvailableFlexibleMemorySize(nullptr) == SCE_KERNEL_ERROR_EINVAL);
|
|
std::size_t configured = 0;
|
|
if (mode == "invalid") {
|
|
bool rejected = false;
|
|
try { sceKernelConfiguredFlexibleMemorySize(&configured); }
|
|
catch (const std::runtime_error&) { rejected = true; }
|
|
Require(rejected && configured == 0);
|
|
return 0;
|
|
}
|
|
const std::size_t expected = (mode == "declared" ? 800u : mode == "limited" ? 16u : 448u) * 1024u * 1024u;
|
|
Require(sceKernelConfiguredFlexibleMemorySize(&configured) == 0 && configured == expected);
|
|
std::size_t before = 0;
|
|
Require(sceKernelAvailableFlexibleMemorySize(&before) == 0);
|
|
Require(before + GuestImageSegmentBytes_nid_no_patch() == configured);
|
|
void* module = nullptr;
|
|
if (mode == "declared") {
|
|
const std::filesystem::path fixture = GUEST_FLEXIBLE_MEMORY_MODULE;
|
|
AddPathAlias_nid_no_patch("budget_module", fixture.parent_path().string().c_str());
|
|
const auto path = "/budget_module/" + fixture.filename().generic_string();
|
|
module = dlopen_nid_postfix(path.c_str(), 2);
|
|
if (!module) std::fprintf(stderr, "Cannot load budget fixture: %s\n", dlerror_nid_postfix());
|
|
Require(module != nullptr);
|
|
Require(sceKernelAvailableFlexibleMemorySize(&before) == 0);
|
|
Require(GuestImageSegmentBytes_nid_no_patch() > 448u * 1024u * 1024u);
|
|
Require(before + GuestImageSegmentBytes_nid_no_patch() == configured && before > 0x80000);
|
|
}
|
|
constexpr std::size_t length = 0x80000;
|
|
void* mapped = nullptr;
|
|
Require(sceKernelMapFlexibleMemory(&mapped, length, 3, 0) == 0 && mapped != nullptr);
|
|
static_cast<volatile unsigned char*>(mapped)[length - 1] = 7;
|
|
std::size_t available = 0;
|
|
Require(sceKernelAvailableFlexibleMemorySize(&available) == 0 && available == before - length);
|
|
Require(sceKernelReleaseFlexibleMemory(mapped, length / 2) == 0);
|
|
Require(sceKernelAvailableFlexibleMemorySize(&available) == 0 && available == before - length / 2);
|
|
Require(sceKernelReleaseFlexibleMemory(static_cast<unsigned char*>(mapped) + length / 2, length / 2) == 0);
|
|
Require(sceKernelAvailableFlexibleMemorySize(&available) == 0 && available == before);
|
|
void* oversized = nullptr;
|
|
Require(sceKernelMapFlexibleMemory(&oversized, before + PS5_PAGE_SIZE, 3, 0) == SCE_KERNEL_ERROR_ENOMEM && oversized == nullptr);
|
|
Require(sceKernelAvailableFlexibleMemorySize(&available) == 0 && available == before);
|
|
if (mode == "limited") {
|
|
const auto length = before & ~(PS5_PAGE_SIZE - 1);
|
|
void* full = nullptr;
|
|
Require(sceKernelMapFlexibleMemory(&full, length, 3, 0) == 0);
|
|
void* extra = nullptr;
|
|
Require(sceKernelMapFlexibleMemory(&extra, PS5_PAGE_SIZE, 3, 0) == SCE_KERNEL_ERROR_ENOMEM && extra == nullptr);
|
|
Require(sceKernelReleaseFlexibleMemory(full, length) == 0);
|
|
Require(sceKernelAvailableFlexibleMemorySize(&available) == 0 && available == before);
|
|
}
|
|
if (module) {
|
|
Require(dlclose_nid_postfix(module) == 0);
|
|
Require(sceKernelAvailableFlexibleMemorySize(&available) == 0);
|
|
Require(available > before && available + GuestImageSegmentBytes_nid_no_patch() == configured);
|
|
}
|
|
std::puts("Flexible memory budget checks passed");
|
|
}
|