Skip to content

Commit

Permalink
Tests: Add tests for inheriting signal handlers
Browse files Browse the repository at this point in the history
  • Loading branch information
timschumi authored and IdanHo committed Jul 5, 2022
1 parent 67f352b commit 60fc0ce
Show file tree
Hide file tree
Showing 2 changed files with 79 additions and 0 deletions.
1 change: 1 addition & 0 deletions Tests/Kernel/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ set(LIBTEST_BASED_SOURCES
TestProcFS.cpp
TestProcFSWrite.cpp
TestSigAltStack.cpp
TestSigHandler.cpp
TestSigWait.cpp
)

Expand Down
78 changes: 78 additions & 0 deletions Tests/Kernel/TestSigHandler.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
/*
* Copyright (c) 2022, Tim Schumacher <[email protected]>
*
* SPDX-License-Identifier: BSD-2-Clause
*/

#include <LibTest/TestCase.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <unistd.h>

static void signal_handler(int)
{
VERIFY_NOT_REACHED();
}

TEST_CASE(default_handlers)
{
struct sigaction current_action { };

int rc = sigaction(SIGUSR2, nullptr, &current_action);

EXPECT_EQ(rc, 0);
EXPECT_EQ(current_action.sa_handler, SIG_DFL);
}

TEST_CASE(handlers_after_fork)
{
struct sigaction new_action {
{ signal_handler }, 0, 0
};
int rc = sigaction(SIGUSR2, &new_action, nullptr);
EXPECT_EQ(rc, 0);

pid_t pid = fork();

if (pid == 0) {
struct sigaction current_action { };
rc = sigaction(SIGUSR2, nullptr, &current_action);
EXPECT_EQ(rc, 0);
EXPECT_EQ(current_action.sa_handler, signal_handler);
exit(rc == 0 && current_action.sa_handler == signal_handler ? EXIT_SUCCESS : EXIT_FAILURE);
} else {
int exit_status = 0;
rc = waitpid(pid, &exit_status, 0);
EXPECT_EQ(rc, pid);
EXPECT(WIFEXITED(exit_status));
EXPECT_EQ(WEXITSTATUS(exit_status), 0);
}
}

TEST_CASE(handlers_after_exec)
{
struct sigaction new_action {
{ signal_handler }, 0, 0
};
int rc = sigaction(SIGUSR2, &new_action, nullptr);
EXPECT_EQ(rc, 0);

pid_t pid = fork();

if (pid == 0) {
// Hide the confusing "Running 1 cases out of 3" output.
freopen("/dev/null", "w", stdout);

// This runs the 'default_handlers' test in this binary again, but after exec.
execl("/proc/self/exe", "TestSigHandler", "default_handlers", nullptr);
FAIL("Failed to exec.");
} else {
int exit_status = 0;
rc = waitpid(pid, &exit_status, 0);
EXPECT_EQ(rc, pid);
EXPECT(WIFEXITED(exit_status));
EXPECT_EQ(WEXITSTATUS(exit_status), 0);
}
}

0 comments on commit 60fc0ce

Please sign in to comment.