204 lines
11 KiB
C++
204 lines
11 KiB
C++
#include "prx/libc/include/general/VabiMacros.hpp"
|
|
#include <csignal>
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <stdexcept>
|
|
#include <chrono>
|
|
#include <thread>
|
|
using Handler = void (APS5_VABI *)(int);
|
|
extern "C" {
|
|
Handler APS5_VABI signal_nid_postfix(int, Handler);
|
|
int APS5_VABI raise_nid_postfix(int);
|
|
int APS5_VABI pthread_kill_nid_postfix(void*, int);
|
|
void* APS5_VABI scePthreadSelf();
|
|
int APS5_VABI sigaction_nid_postfix(int, const void*, void*);
|
|
int APS5_VABI sigprocmask_nid_postfix(int, const void*, void*);
|
|
int APS5_VABI pthread_sigmask_nid_postfix(int, const void*, void*);
|
|
int* APS5_VABI __error_nid_postfix();
|
|
int APS5_VABI _is_signal_return_nid_postfix(std::uint64_t);
|
|
unsigned int APS5_VABI GuestAlarm_nid_no_patch(unsigned int);
|
|
}
|
|
struct GuestSignalSet {
|
|
std::uint32_t bits[4];
|
|
};
|
|
struct GuestSigaction {
|
|
std::uintptr_t handler;
|
|
int flags;
|
|
GuestSignalSet mask;
|
|
};
|
|
static_assert(sizeof(GuestSigaction) == 32);
|
|
static bool Same(const GuestSigaction& left, const GuestSigaction& right) {
|
|
return left.handler == right.handler && left.flags == right.flags && std::memcmp(&left.mask, &right.mask, sizeof(left.mask)) == 0;
|
|
}
|
|
volatile std::sig_atomic_t received = 0;
|
|
volatile std::uintptr_t handlerReturn = 0;
|
|
void APS5_VABI InfoCallback(int, void*, void*) {}
|
|
void APS5_VABI Callback(int value) {
|
|
received = value;
|
|
handlerReturn = reinterpret_cast<std::uintptr_t>(__builtin_return_address(0));
|
|
}
|
|
static void Require(bool value) { if (!value) std::abort(); }
|
|
int main() {
|
|
const auto invalid = reinterpret_cast<Handler>(static_cast<std::uintptr_t>(-1));
|
|
const auto ignore = reinterpret_cast<Handler>(std::uintptr_t{1});
|
|
Require(signal_nid_postfix(9, Callback) == invalid);
|
|
Require(*__error_nid_postfix() == 22);
|
|
Require(signal_nid_postfix(15, Callback) != invalid);
|
|
Require(raise_nid_postfix(15) == 0 && received == 15);
|
|
Require(handlerReturn != 0 && _is_signal_return_nid_postfix(handlerReturn) == 0);
|
|
Require(_is_signal_return_nid_postfix(reinterpret_cast<std::uintptr_t>(Callback)) == 0);
|
|
Require(_is_signal_return_nid_postfix(0) == 0);
|
|
received = 0;
|
|
Require(raise_nid_postfix(15) == 0 && received == 15);
|
|
received = 0;
|
|
Require(pthread_kill_nid_postfix(scePthreadSelf(), 15) == 0 && received == 15);
|
|
received = 0;
|
|
Require(pthread_kill_nid_postfix(scePthreadSelf(), 0) == 0 && received == 0);
|
|
Require(pthread_kill_nid_postfix(scePthreadSelf(), 200) == 22);
|
|
Require(pthread_kill_nid_postfix(scePthreadSelf(), -1) == 22);
|
|
Require(pthread_kill_nid_postfix(nullptr, 15) == 3);
|
|
Require(pthread_kill_nid_postfix(nullptr, 128) == 3);
|
|
Require(pthread_kill_nid_postfix(nullptr, 129) == 22);
|
|
Require(pthread_kill_nid_postfix(reinterpret_cast<void*>(std::uintptr_t{0x10}), 15) == 3);
|
|
Require(received == 0);
|
|
Require(signal_nid_postfix(15, ignore) != invalid);
|
|
received = 0;
|
|
Require(raise_nid_postfix(15) == 0 && received == 0);
|
|
Require(signal_nid_postfix(15, nullptr) == ignore);
|
|
Require(raise_nid_postfix(100) == -1 && *__error_nid_postfix() == 22);
|
|
GuestSignalSet blocked{{0x20, 0, 0, 0}};
|
|
GuestSignalSet previous{{}};
|
|
Require(sigprocmask_nid_postfix(3, &blocked, &previous) == 0);
|
|
Require(previous.bits[0] == 0);
|
|
Require(sigprocmask_nid_postfix(1, nullptr, &previous) == 0);
|
|
Require(previous.bits[0] == 0x20);
|
|
Require(sigprocmask_nid_postfix(2, &blocked, nullptr) == 0);
|
|
Require(sigprocmask_nid_postfix(1, nullptr, &previous) == 0);
|
|
Require(previous.bits[0] == 0);
|
|
*__error_nid_postfix() = 0;
|
|
Require(pthread_sigmask_nid_postfix(1, &blocked, nullptr) == 0);
|
|
Require(pthread_sigmask_nid_postfix(0, &blocked, &previous) == 22);
|
|
Require(pthread_sigmask_nid_postfix(4, &blocked, &previous) == 22);
|
|
Require(*__error_nid_postfix() == 0);
|
|
Require(pthread_sigmask_nid_postfix(0, nullptr, &previous) == 0);
|
|
Require(previous.bits[0] == 0x20);
|
|
Require(sigprocmask_nid_postfix(1, nullptr, &previous) == 0 && previous.bits[0] == 0x20);
|
|
Require(pthread_sigmask_nid_postfix(2, &blocked, &previous) == 0 && previous.bits[0] == 0x20);
|
|
Require(pthread_sigmask_nid_postfix(1, nullptr, &previous) == 0 && previous.bits[0] == 0);
|
|
|
|
const auto callback = reinterpret_cast<std::uintptr_t>(Callback);
|
|
const auto infoCallback = reinterpret_cast<std::uintptr_t>(InfoCallback);
|
|
GuestSigaction current{};
|
|
std::memset(¤t, 0xa5, sizeof(current));
|
|
Require(sigaction_nid_postfix(15, nullptr, ¤t) == 0);
|
|
Require(current.handler == 0 && current.flags == 0x2 && current.mask.bits[0] == 0 && current.mask.bits[3] == 0);
|
|
const GuestSigaction term{callback, 0, {{0x4000, 0, 0, 0x80000000u}}};
|
|
Require(sigaction_nid_postfix(15, &term, ¤t) == 0 && current.handler == 0);
|
|
received = 0;
|
|
Require(raise_nid_postfix(15) == 0 && received == 15);
|
|
Require(sigaction_nid_postfix(15, nullptr, ¤t) == 0);
|
|
Require(Same(current, term));
|
|
const GuestSigaction defaults{0, 0, {}};
|
|
Require(sigaction_nid_postfix(15, &defaults, nullptr) == 0);
|
|
Require(signal_nid_postfix(15, Callback) == nullptr);
|
|
Require(sigaction_nid_postfix(15, nullptr, ¤t) == 0 && current.handler == callback && current.flags == 0x2);
|
|
Require(signal_nid_postfix(15, nullptr) == Callback);
|
|
|
|
const GuestSigaction crash{infoCallback, 0x40 | 0x10, {}};
|
|
for (int recorded : {10, 1, 64, 128}) {
|
|
Require(sigaction_nid_postfix(recorded, &crash, nullptr) == 0);
|
|
Require(sigaction_nid_postfix(recorded, nullptr, ¤t) == 0);
|
|
Require(Same(current, crash));
|
|
Require(sigaction_nid_postfix(recorded, &defaults, nullptr) == 0);
|
|
}
|
|
Require(sigaction_nid_postfix(6, &crash, nullptr) == 0);
|
|
bool rejected = false;
|
|
try {
|
|
raise_nid_postfix(6);
|
|
} catch (const std::runtime_error&) {
|
|
rejected = true;
|
|
}
|
|
Require(rejected);
|
|
Require(sigaction_nid_postfix(6, &defaults, ¤t) == 0 && current.handler == infoCallback && current.flags == 0x50);
|
|
|
|
*__error_nid_postfix() = 0;
|
|
Require(sigaction_nid_postfix(0, nullptr, ¤t) == -1 && *__error_nid_postfix() == 22);
|
|
*__error_nid_postfix() = 0;
|
|
Require(sigaction_nid_postfix(129, nullptr, ¤t) == -1 && *__error_nid_postfix() == 22);
|
|
for (int fixed : {9, 17}) {
|
|
*__error_nid_postfix() = 0;
|
|
Require(sigaction_nid_postfix(fixed, &term, ¤t) == -1 && *__error_nid_postfix() == 22);
|
|
Require(sigaction_nid_postfix(fixed, &defaults, ¤t) == 0 && current.handler == 0);
|
|
}
|
|
Require(sigaction_nid_postfix(15, nullptr, nullptr) == 0);
|
|
Require(signal_nid_postfix(15, Callback) != invalid);
|
|
GuestSignalSet termMask{{1u << 14, 0, 0, 0}};
|
|
received = 0;
|
|
Require(sigprocmask_nid_postfix(1, &termMask, nullptr) == 0);
|
|
Require(raise_nid_postfix(15) == 0 && received == 0);
|
|
Require(sigprocmask_nid_postfix(2, &termMask, nullptr) == 0);
|
|
Require(raise_nid_postfix(15) == 0 && received == 15);
|
|
GuestSignalSet urgMask{{1u << 15, 0, 0, 0}};
|
|
received = 0;
|
|
Require(sigprocmask_nid_postfix(1, &urgMask, nullptr) == 0);
|
|
Require(raise_nid_postfix(15) == 0 && received == 15);
|
|
Require(sigprocmask_nid_postfix(2, &urgMask, nullptr) == 0);
|
|
|
|
Require(GuestAlarm_nid_no_patch(0) == 0);
|
|
Require(GuestAlarm_nid_no_patch(5) == 0);
|
|
Require(GuestAlarm_nid_no_patch(3) == 5);
|
|
Require(GuestAlarm_nid_no_patch(0) == 3);
|
|
Require(GuestAlarm_nid_no_patch(0) == 0);
|
|
const GuestSigaction ignoreAlarm{1, 0, {}};
|
|
Require(sigaction_nid_postfix(14, &ignoreAlarm, nullptr) == 0);
|
|
Require(GuestAlarm_nid_no_patch(1) == 0);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(1500));
|
|
Require(GuestAlarm_nid_no_patch(0) == 0);
|
|
for (int unmapped : {1, 13, 30, 31, 32, 128}) {
|
|
*__error_nid_postfix() = 0;
|
|
Require(signal_nid_postfix(unmapped, ignore) == nullptr);
|
|
Require(sigaction_nid_postfix(unmapped, nullptr, ¤t) == 0 && current.handler == 1 && current.flags == 0x2);
|
|
Require(signal_nid_postfix(unmapped, Callback) == ignore);
|
|
Require(sigaction_nid_postfix(unmapped, nullptr, ¤t) == 0 && current.handler == callback && current.flags == 0x2);
|
|
Require(signal_nid_postfix(unmapped, nullptr) == Callback);
|
|
}
|
|
for (int outOfRange : {0, 129, -1}) {
|
|
*__error_nid_postfix() = 0;
|
|
Require(signal_nid_postfix(outOfRange, nullptr) == invalid && *__error_nid_postfix() == 22);
|
|
*__error_nid_postfix() = 0;
|
|
Require(signal_nid_postfix(outOfRange, Callback) == invalid && *__error_nid_postfix() == 22);
|
|
}
|
|
for (int fixed : {9, 17}) {
|
|
const GuestSigaction flagged{0, 0x1 | 0x4, {{0x5, 0x6, 0x7, 0x80000000u}}};
|
|
Require(sigaction_nid_postfix(fixed, &flagged, nullptr) == 0);
|
|
*__error_nid_postfix() = 0;
|
|
Require(signal_nid_postfix(fixed, Callback) == invalid && *__error_nid_postfix() == 22);
|
|
*__error_nid_postfix() = 0;
|
|
Require(signal_nid_postfix(fixed, ignore) == invalid && *__error_nid_postfix() == 22);
|
|
*__error_nid_postfix() = 0;
|
|
Require(signal_nid_postfix(fixed, invalid) == invalid && *__error_nid_postfix() == 22);
|
|
Require(sigaction_nid_postfix(fixed, nullptr, ¤t) == 0 && current.handler == 0);
|
|
*__error_nid_postfix() = 12345;
|
|
Require(signal_nid_postfix(fixed, nullptr) == nullptr && *__error_nid_postfix() == 12345);
|
|
Require(sigaction_nid_postfix(fixed, nullptr, ¤t) == 0 && current.handler == 0 && current.flags == 0x2 && current.mask.bits[0] == 0);
|
|
Require(sigaction_nid_postfix(fixed, &defaults, nullptr) == 0);
|
|
}
|
|
{
|
|
const GuestSigaction masked{callback, 0x1 | 0x4, {{0x5, 0x6, 0x7, 0x80000000u}}};
|
|
Require(sigaction_nid_postfix(13, &masked, nullptr) == 0);
|
|
*__error_nid_postfix() = 12345;
|
|
Require(signal_nid_postfix(13, ignore) == Callback && *__error_nid_postfix() == 12345);
|
|
Require(sigaction_nid_postfix(13, nullptr, ¤t) == 0);
|
|
Require(current.handler == 1 && current.flags == 0x2 && current.mask.bits[0] == 0 && current.mask.bits[1] == 0 && current.mask.bits[2] == 0 && current.mask.bits[3] == 0);
|
|
Require(sigaction_nid_postfix(13, &masked, nullptr) == 0);
|
|
Require(signal_nid_postfix(13, Callback) == Callback);
|
|
Require(sigaction_nid_postfix(13, nullptr, ¤t) == 0);
|
|
Require(current.handler == callback && current.flags == 0x2 && current.mask.bits[3] == 0);
|
|
}
|
|
Require(signal_nid_postfix(13, invalid) == Callback);
|
|
Require(sigaction_nid_postfix(13, nullptr, ¤t) == 0 && current.handler == static_cast<std::uintptr_t>(-1) && current.flags == 0x2);
|
|
Require(signal_nid_postfix(13, nullptr) == invalid);
|
|
Require(sigaction_nid_postfix(13, nullptr, ¤t) == 0 && current.handler == 0);
|
|
}
|