Files
FEX-Emu--FEX/unittests/FEXLinuxTests/tests/smc/smc-unexec-stack.cpp
T

31 lines
942 B
C++

#include <catch2/catch_test_macros.hpp>
#include <signal.h>
#include <ucontext.h>
#include <sys/mman.h>
bool got_signal = false;
void *tramp = nullptr;
static void sigsegv_handler(int signal, siginfo_t *siginfo, void* context) {
REQUIRE(siginfo->si_code == SEGV_ACCERR);
got_signal = true;
size_t page_size = sysconf(_SC_PAGESIZE);
void *fault_addr = (void *)((uintptr_t)(siginfo->si_addr) & ~(page_size - 1));
REQUIRE(mprotect(fault_addr, page_size, PROT_READ | PROT_WRITE | PROT_EXEC) == 0);
}
TEST_CASE("smc-unexec-stack: PT_GNU_STACK == RW") {
// Register signal handler
struct sigaction act {};
act.sa_sigaction = sigsegv_handler;
act.sa_flags = SA_SIGINFO;
sigaction(SIGSEGV, &act, nullptr);
// Try executing from stack
uint8_t stack_code = 0xC3; // ret instruction
tramp = (void *)((uintptr_t)&stack_code & ~(sysconf(_SC_PAGESIZE) - 1));
((void (*)())(&stack_code))();
REQUIRE(got_signal == true);
}