Files
AnyPS5/core/libs/tests/GuestGetrusage.cpp

66 lines
2.4 KiB
C++

#include "prx/libkernel/include/Api.hpp"
#include "prx/libc/include/Api.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <chrono>
#include <cstdint>
#include <cstring>
#include <cstdlib>
#include <thread>
struct Timeval {
std::int64_t tv_sec;
std::int64_t tv_usec;
};
static void Require(bool value) { if (!value) std::abort(); }
static std::int64_t CpuMicros(int who) {
GuestResourceUsage usage{};
Require(getrusage_nid_postfix(who, &usage) == 0);
return (usage.ru_utime.tv_sec + usage.ru_stime.tv_sec) * 1000000 + usage.ru_utime.tv_usec + usage.ru_stime.tv_usec;
}
static void SpinUntil(int who, std::int64_t micros) {
const auto deadline = std::chrono::steady_clock::now() + std::chrono::seconds(10);
volatile std::uint64_t sink = 0;
while (CpuMicros(who) < micros) {
for (int i = 0; i < 100000; ++i) sink = sink + i;
Require(std::chrono::steady_clock::now() < deadline);
}
}
static void RequireFailure(int who, GuestResourceUsage* usage, int error) {
*__error_nid_postfix() = 0;
Require(getrusage_nid_postfix(who, usage) == -1);
Require(*__error_nid_postfix() == error);
}
int main() {
GuestResourceUsage untouched;
std::memset(&untouched, 0x5a, sizeof(untouched));
const GuestResourceUsage pattern = untouched;
RequireFailure(2, &untouched, 22);
RequireFailure(-2, &untouched, 22);
Require(std::memcmp(&untouched, &pattern, sizeof(untouched)) == 0);
RequireFailure(7, nullptr, 22);
RequireFailure(0, nullptr, 14);
RequireFailure(1, nullptr, 14);
RequireFailure(-1, nullptr, 14);
auto* const invalid = reinterpret_cast<GuestResourceUsage*>(static_cast<std::uintptr_t>(1));
RequireFailure(2, invalid, 22);
RequireFailure(0, invalid, 14);
RequireFailure(1, invalid, 14);
RequireFailure(-1, invalid, 14);
GuestResourceUsage children;
std::memset(&children, 0xff, sizeof(children));
Require(getrusage_nid_postfix(-1, &children) == 0);
const GuestResourceUsage empty{};
Require(std::memcmp(&children, &empty, sizeof(children)) == 0);
SpinUntil(0, 400000);
const std::int64_t mainThread = CpuMicros(1);
Require(mainThread >= 200000);
std::thread worker([] {
const std::int64_t fresh = CpuMicros(1);
Require(fresh < 100000);
SpinUntil(1, fresh + 100000);
Require(CpuMicros(0) >= CpuMicros(1) + 300000);
});
worker.join();
Require(CpuMicros(0) >= 500000);
}