FEX-Emu--FEX/unittests/FEXLinuxTests/tests/signal/sigtest_sigmask.cpp

#include <catch2/catch_test_macros.hpp>

#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <ucontext.h>
#include <unistd.h>

volatile bool loop = false;
volatile bool inhandler = false;

#define SIGN SIGTSTP

void sig_handler(int signum, siginfo_t* info, void* context) {
  loop = false;
  printf("Inside handler function\n");
  if (inhandler) {
    printf("Signal reentering bug\n");
    exit(-1);
  }
  inhandler = true;
  raise(signum);

  auto uctx = (ucontext_t*)context;
  sigfillset(&uctx->uc_sigmask);
}

TEST_CASE("Signals: sigmask") {
  struct sigaction act = {0};

  act.sa_flags = SA_SIGINFO;
  act.sa_sigaction = &sig_handler;
  REQUIRE(sigaction(SIGN, &act, NULL) == 0);

  loop = true;
  while (loop) {
    printf("Inside main loop, raising signal\n");
    raise(SIGN);

    // Ensure the signal got indeed raised
    REQUIRE_FALSE(loop);
  }
}