extern "C" {
#include <sys/types.h>
#include <signal.h>
#include <unistd.h>
}
#include <cerrno>
#include <cstdlib>
#include <iostream>
#include "atf-c++/macros.hpp"
#include "atf-c++/detail/exceptions.hpp"
#include "atf-c++/detail/process.hpp"
#include "signals.hpp"
namespace sigusr1 {
static bool happened = false;
static
void
handler(int signo)
{
happened = true;
}
static
void
program(void)
{
struct sigaction sa;
sa.sa_handler = handler;
sigemptyset(&sa.sa_mask);
sa.sa_flags = 0;
if (::sigaction(SIGUSR1, &sa, NULL) == -1)
throw atf::system_error("sigusr1::program",
"sigaction(2) failed", errno);
}
}
namespace sigusr1_2 {
static bool happened = false;
static
void
handler(int signo)
{
happened = true;
}
}
ATF_TEST_CASE(signal_holder_preserve);
ATF_TEST_CASE_HEAD(signal_holder_preserve)
{
set_md_var("descr", "Tests that signal_holder preserves the original "
"signal handler and restores it upon destruction");
}
ATF_TEST_CASE_BODY(signal_holder_preserve)
{
using atf::atf_run::signal_holder;
sigusr1::program();
sigusr1::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1::happened);
{
signal_holder hld(SIGUSR1);
::kill(::getpid(), SIGUSR1);
}
sigusr1::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1::happened);
}
ATF_TEST_CASE(signal_holder_destructor);
ATF_TEST_CASE_HEAD(signal_holder_destructor)
{
set_md_var("descr", "Tests that signal_holder processes a pending "
"signal upon destruction");
}
ATF_TEST_CASE_BODY(signal_holder_destructor)
{
using atf::atf_run::signal_holder;
sigusr1::program();
sigusr1::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1::happened);
{
signal_holder hld(SIGUSR1);
sigusr1::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(!sigusr1::happened);
}
ATF_REQUIRE(sigusr1::happened);
}
ATF_TEST_CASE(signal_holder_process);
ATF_TEST_CASE_HEAD(signal_holder_process)
{
set_md_var("descr", "Tests that signal_holder's process method works "
"to process a delayed signal explicitly");
}
ATF_TEST_CASE_BODY(signal_holder_process)
{
using atf::atf_run::signal_holder;
sigusr1::program();
sigusr1::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1::happened);
{
signal_holder hld(SIGUSR1);
sigusr1::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(!sigusr1::happened);
hld.process();
ATF_REQUIRE(sigusr1::happened);
sigusr1::happened = false;
}
ATF_REQUIRE(!sigusr1::happened);
}
ATF_TEST_CASE(signal_programmer_program);
ATF_TEST_CASE_HEAD(signal_programmer_program)
{
set_md_var("descr", "Tests that signal_programmer correctly installs a "
"handler");
}
ATF_TEST_CASE_BODY(signal_programmer_program)
{
using atf::atf_run::signal_programmer;
signal_programmer sp(SIGUSR1, sigusr1_2::handler);
sigusr1_2::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1_2::happened);
}
ATF_TEST_CASE(signal_programmer_preserve);
ATF_TEST_CASE_HEAD(signal_programmer_preserve)
{
set_md_var("descr", "Tests that signal_programmer uninstalls the "
"handler during destruction");
}
ATF_TEST_CASE_BODY(signal_programmer_preserve)
{
using atf::atf_run::signal_programmer;
sigusr1::program();
sigusr1::happened = false;
{
signal_programmer sp(SIGUSR1, sigusr1_2::handler);
sigusr1_2::happened = false;
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1_2::happened);
}
ATF_REQUIRE(!sigusr1::happened);
::kill(::getpid(), SIGUSR1);
ATF_REQUIRE(sigusr1::happened);
}
static
void
reset_child(void *v)
{
sigusr1::program();
sigusr1::happened = false;
atf::atf_run::reset(SIGUSR1);
kill(::getpid(), SIGUSR1);
if (sigusr1::happened) {
std::cerr << "Signal was not resetted correctly\n";
std::abort();
} else {
std::exit(EXIT_SUCCESS);
}
}
ATF_TEST_CASE(reset);
ATF_TEST_CASE_HEAD(reset)
{
set_md_var("descr", "Tests the reset function");
}
ATF_TEST_CASE_BODY(reset)
{
atf::process::child c =
atf::process::fork(reset_child, atf::process::stream_inherit(),
atf::process::stream_inherit(), NULL);
const atf::process::status s = c.wait();
ATF_REQUIRE(s.exited() || s.signaled());
ATF_REQUIRE(!s.signaled() || s.termsig() == SIGUSR1);
}
ATF_INIT_TEST_CASES(tcs)
{
ATF_ADD_TEST_CASE(tcs, signal_holder_preserve);
ATF_ADD_TEST_CASE(tcs, signal_holder_destructor);
ATF_ADD_TEST_CASE(tcs, signal_holder_process);
ATF_ADD_TEST_CASE(tcs, signal_programmer_program);
ATF_ADD_TEST_CASE(tcs, signal_programmer_preserve);
ATF_ADD_TEST_CASE(tcs, reset);
}