mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-02-01 12:03:41 +00:00
99dc1281f5
The new asynchronous signal delivery technique is now also being used for tkill(), raise(), etc. Many subtle issues have been addresesd. We now signal handling on Windows that's remarkably similar to the POSIX behaviors. However that's just across threads. We're lacking a way to have the signal semantics work well, across multiple WIN32 processes.
356 lines
12 KiB
C
356 lines
12 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
|
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi│
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Copyright 2021 Justine Alexandra Roberts Tunney │
|
|
│ │
|
|
│ Permission to use, copy, modify, and/or distribute this software for │
|
|
│ any purpose with or without fee is hereby granted, provided that the │
|
|
│ above copyright notice and this permission notice appear in all copies. │
|
|
│ │
|
|
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL │
|
|
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED │
|
|
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE │
|
|
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL │
|
|
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR │
|
|
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER │
|
|
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR │
|
|
│ PERFORMANCE OF THIS SOFTWARE. │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#include "libc/calls/struct/sigaction.h"
|
|
#include "libc/calls/calls.h"
|
|
#include "libc/calls/pledge.h"
|
|
#include "libc/calls/struct/rusage.h"
|
|
#include "libc/calls/struct/sigaction.h"
|
|
#include "libc/calls/struct/siginfo.h"
|
|
#include "libc/calls/struct/sigset.h"
|
|
#include "libc/calls/struct/sigset.internal.h"
|
|
#include "libc/calls/struct/ucontext.internal.h"
|
|
#include "libc/calls/syscall_support-sysv.internal.h"
|
|
#include "libc/calls/ucontext.h"
|
|
#include "libc/dce.h"
|
|
#include "libc/errno.h"
|
|
#include "libc/intrin/kprintf.h"
|
|
#include "libc/nexgen32e/nexgen32e.h"
|
|
#include "libc/nexgen32e/vendor.internal.h"
|
|
#include "libc/runtime/internal.h"
|
|
#include "libc/runtime/runtime.h"
|
|
#include "libc/runtime/syslib.internal.h"
|
|
#include "libc/str/str.h"
|
|
#include "libc/sysv/consts/sa.h"
|
|
#include "libc/sysv/consts/sig.h"
|
|
#include "libc/sysv/consts/uc.h"
|
|
#include "libc/testlib/testlib.h"
|
|
#include "third_party/xed/x86.h"
|
|
|
|
struct sigaction oldsa;
|
|
volatile bool gotsigint;
|
|
|
|
void SetUpOnce(void) {
|
|
__enable_threads();
|
|
ASSERT_SYS(0, 0, pledge("stdio rpath proc", 0));
|
|
}
|
|
|
|
void OnSigInt(int sig) {
|
|
CheckStackIsAligned();
|
|
gotsigint = true;
|
|
}
|
|
|
|
void SetUp(void) {
|
|
gotsigint = false;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// test that signal handlers expose cpu state, and let it be changed arbitrarily
|
|
|
|
void *Gateway(void *arg) {
|
|
__builtin_trap();
|
|
}
|
|
|
|
void PromisedLand(void *arg) {
|
|
sigset_t ss;
|
|
CheckStackIsAligned();
|
|
sigprocmask(SIG_SETMASK, 0, &ss);
|
|
ASSERT_TRUE(sigismember(&ss, SIGUSR1));
|
|
pthread_exit(arg);
|
|
}
|
|
|
|
void Teleporter(int sig, struct siginfo *si, void *ctx) {
|
|
ucontext_t *uc = ctx;
|
|
sigaddset(&uc->uc_sigmask, SIGUSR1);
|
|
uc->uc_mcontext.PC = (uintptr_t)PromisedLand;
|
|
uc->uc_mcontext.SP &= -16;
|
|
#ifdef __x86_64__
|
|
uc->uc_mcontext.SP -= 8;
|
|
#endif
|
|
}
|
|
|
|
TEST(sigaction, handlersCanMutateMachineState) {
|
|
void *rc;
|
|
sigset_t ss;
|
|
pthread_t t;
|
|
struct sigaction oldill, oldtrap;
|
|
struct sigaction sa = {.sa_sigaction = Teleporter, .sa_flags = SA_SIGINFO};
|
|
sigprocmask(SIG_SETMASK, 0, &ss);
|
|
ASSERT_FALSE(sigismember(&ss, SIGUSR1));
|
|
ASSERT_SYS(0, 0, sigaction(SIGILL, &sa, &oldill));
|
|
ASSERT_SYS(0, 0, sigaction(SIGTRAP, &sa, &oldtrap));
|
|
ASSERT_EQ(0, pthread_create(&t, 0, Gateway, (void *)42L));
|
|
ASSERT_EQ(0, pthread_join(t, &rc));
|
|
ASSERT_EQ(42, (uintptr_t)rc);
|
|
ASSERT_SYS(0, 0, sigaction(SIGILL, &oldill, 0));
|
|
ASSERT_SYS(0, 0, sigaction(SIGTRAP, &oldtrap, 0));
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// test raise()
|
|
|
|
TEST(sigaction, raise) {
|
|
struct sigaction saint = {.sa_handler = OnSigInt};
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &saint, &oldsa));
|
|
ASSERT_FALSE(gotsigint);
|
|
EXPECT_NE(-1, raise(SIGINT));
|
|
ASSERT_TRUE(gotsigint);
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &oldsa, NULL));
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// test kill()
|
|
|
|
TEST(sigaction, testPingPongParentChildWithSigint) {
|
|
if (IsNetbsd()) return; // TODO: what's up with runitd on netbsd?
|
|
int pid, status;
|
|
sigset_t blockint, oldmask;
|
|
struct sigaction oldint;
|
|
struct sigaction ignoreint = {.sa_handler = SIG_IGN};
|
|
struct sigaction catchint = {.sa_handler = OnSigInt};
|
|
if (IsWindows()) {
|
|
// this works if it's run by itself on the command prompt. but it
|
|
// doesn't currently work if it's launched as a subprocess of some
|
|
// kind of runner. todo(fixme!)
|
|
return;
|
|
}
|
|
EXPECT_SYS(0, 0, sigemptyset(&blockint));
|
|
EXPECT_SYS(0, 0, sigaddset(&blockint, SIGINT));
|
|
EXPECT_SYS(0, 0, sigprocmask(SIG_BLOCK, &blockint, &oldmask));
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &catchint, &oldint));
|
|
ASSERT_NE(-1, (pid = fork()));
|
|
if (!pid) {
|
|
// ping
|
|
EXPECT_SYS(0, 0, kill(getppid(), SIGINT));
|
|
EXPECT_FALSE(gotsigint);
|
|
// pong
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &catchint, 0));
|
|
EXPECT_SYS(EINTR, -1, sigsuspend(0));
|
|
EXPECT_TRUE(gotsigint);
|
|
_exit(0);
|
|
}
|
|
// pong
|
|
EXPECT_FALSE(gotsigint);
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &catchint, 0));
|
|
EXPECT_SYS(EINTR, -1, sigsuspend(0));
|
|
EXPECT_TRUE(gotsigint);
|
|
// ping
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &ignoreint, 0));
|
|
EXPECT_SYS(0, 0, kill(pid, SIGINT));
|
|
// cleanup
|
|
EXPECT_NE(-1, wait4(pid, &status, 0, 0));
|
|
EXPECT_EQ(1, WIFEXITED(status));
|
|
EXPECT_EQ(0, WEXITSTATUS(status));
|
|
EXPECT_EQ(0, WTERMSIG(status));
|
|
EXPECT_SYS(0, 0, sigaction(SIGINT, &oldint, 0));
|
|
EXPECT_SYS(0, 0, sigprocmask(SIG_BLOCK, &oldmask, 0));
|
|
}
|
|
|
|
#ifdef __x86_64__
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// test int3 crash
|
|
// we expect this to be recoverable by default
|
|
|
|
volatile int trapeax;
|
|
|
|
void OnTrap(int sig, struct siginfo *si, void *vctx) {
|
|
struct ucontext *ctx = vctx;
|
|
CheckStackIsAligned();
|
|
trapeax = ctx->uc_mcontext.rax;
|
|
}
|
|
|
|
TEST(sigaction, debugBreak_handlerCanReadCpuState) {
|
|
struct sigaction saint = {.sa_sigaction = OnTrap, .sa_flags = SA_SIGINFO};
|
|
EXPECT_NE(-1, sigaction(SIGTRAP, &saint, &oldsa));
|
|
asm("int3" : /* no outputs */ : "a"(0x31337));
|
|
EXPECT_EQ(0x31337, trapeax);
|
|
EXPECT_SYS(0, 0, sigaction(SIGTRAP, &oldsa, NULL));
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// test fpu crash (unrecoverable)
|
|
// test signal handler can modify cpu registers (now it's recoverable!)
|
|
|
|
void SkipOverFaultingInstruction(struct ucontext *ctx) {
|
|
CheckStackIsAligned();
|
|
struct XedDecodedInst xedd;
|
|
xed_decoded_inst_zero_set_mode(&xedd, XED_MACHINE_MODE_LONG_64);
|
|
xed_instruction_length_decode(&xedd, (void *)ctx->uc_mcontext.rip, 15);
|
|
ctx->uc_mcontext.rip += xedd.length;
|
|
}
|
|
|
|
void OnFpe(int sig, struct siginfo *si, void *vctx) {
|
|
struct ucontext *ctx = vctx;
|
|
CheckStackIsAligned();
|
|
SkipOverFaultingInstruction(ctx);
|
|
ctx->uc_mcontext.rax = 42;
|
|
ctx->uc_mcontext.rdx = 0;
|
|
}
|
|
|
|
dontubsan void ubsanTrumpsSystemsEngineering(void) {
|
|
struct sigaction saint = {.sa_sigaction = OnFpe, .sa_flags = SA_SIGINFO};
|
|
EXPECT_SYS(0, 0, sigaction(SIGFPE, &saint, &oldsa));
|
|
volatile long x = 0;
|
|
EXPECT_EQ(42, 666 / x); /* systems engineering trumps math */
|
|
EXPECT_SYS(0, 0, sigaction(SIGFPE, &oldsa, NULL));
|
|
}
|
|
|
|
TEST(sigaction, sigFpe_handlerCanEditProcessStateAndRecoverExecution) {
|
|
ubsanTrumpsSystemsEngineering();
|
|
}
|
|
|
|
#endif /* __x86_64__ */
|
|
|
|
static unsigned OnSignalCnt = 0;
|
|
void OnSignal(int sig, siginfo_t *si, void *ctx) {
|
|
OnSignalCnt++;
|
|
}
|
|
|
|
TEST(sigaction, ignoringSignalDiscardsSignal) {
|
|
struct sigaction sa = {.sa_sigaction = OnSignal, .sa_flags = SA_SIGINFO};
|
|
ASSERT_EQ(0, sigaction(SIGUSR1, &sa, NULL));
|
|
sigset_t blocked;
|
|
sigemptyset(&blocked);
|
|
sigaddset(&blocked, SIGUSR1);
|
|
ASSERT_EQ(0, sigprocmask(SIG_SETMASK, &blocked, NULL));
|
|
ASSERT_EQ(0, raise(SIGUSR1));
|
|
ASSERT_NE(SIG_ERR, signal(SIGUSR1, SIG_IGN));
|
|
ASSERT_EQ(0, sigaction(SIGUSR1, &sa, NULL));
|
|
ASSERT_EQ(0, sigprocmask(SIG_UNBLOCK, &blocked, NULL));
|
|
EXPECT_EQ(0, OnSignalCnt);
|
|
}
|
|
|
|
TEST(sigaction, autoZombieSlayer) {
|
|
if (IsWindows()) return;
|
|
if (IsCygwin()) return;
|
|
int pid;
|
|
struct sigaction sa;
|
|
// make sure we're starting in expected state
|
|
ASSERT_SYS(0, 0, sigaction(SIGCHLD, 0, &sa));
|
|
ASSERT_EQ(SIG_DFL, sa.sa_handler);
|
|
// verify child becomes zombie
|
|
ASSERT_NE(-1, (pid = fork()));
|
|
if (!pid) _Exit(0);
|
|
ASSERT_SYS(0, pid, wait(0));
|
|
// enable automatic zombie slayer
|
|
sa.sa_handler = SIG_IGN;
|
|
sa.sa_flags = 0;
|
|
sigemptyset(&sa.sa_mask);
|
|
ASSERT_SYS(0, 0, sigaction(SIGCHLD, &sa, &sa));
|
|
// verify it works
|
|
ASSERT_NE(-1, (pid = fork()));
|
|
if (!pid) _Exit(0);
|
|
// XXX: WSL does the wrong thing here.
|
|
if (__iswsl1()) usleep(10);
|
|
ASSERT_SYS(ECHILD, -1, wait(0));
|
|
// clean up
|
|
ASSERT_SYS(0, 0, sigaction(SIGCHLD, &sa, 0));
|
|
}
|
|
|
|
TEST(sigaction, enosys_returnsErrnoRatherThanSigsysByDefault) {
|
|
if (IsTiny()) return; // systemfive.S disables the fix w/ tiny
|
|
if (IsOpenbsd()) return; // TODO: Why does OpenBSD raise SIGABRT?
|
|
ASSERT_SYS(ENOSYS, -1, sys_bogus());
|
|
}
|
|
|
|
sig_atomic_t gotusr1;
|
|
|
|
void OnSigMask(int sig, struct siginfo *si, void *ctx) {
|
|
ucontext_t *uc = ctx;
|
|
#ifdef __x86_64__
|
|
ASSERT_EQ(123, uc->uc_mcontext.r15);
|
|
#endif
|
|
sigaddset(&uc->uc_sigmask, sig);
|
|
gotusr1 = true;
|
|
}
|
|
|
|
TEST(uc_sigmask, signalHandlerCanChangeSignalMaskOfTrappedThread) {
|
|
sigset_t want, got;
|
|
struct sigaction oldsa;
|
|
struct sigaction sa = {.sa_sigaction = OnSigMask, .sa_flags = SA_SIGINFO};
|
|
sigemptyset(&want);
|
|
ASSERT_SYS(0, 0, sigprocmask(SIG_SETMASK, &want, &got));
|
|
ASSERT_FALSE(sigismember(&got, SIGUSR1));
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR1, &sa, &oldsa));
|
|
#ifdef __x86_64__
|
|
asm volatile("mov\t%0,%%r15" : : "i"(123) : "r15", "memory");
|
|
#endif
|
|
ASSERT_SYS(0, 0, raise(SIGUSR1));
|
|
ASSERT_TRUE(gotusr1);
|
|
ASSERT_SYS(0, 0, sigprocmask(SIG_SETMASK, 0, &got));
|
|
ASSERT_TRUE(sigismember(&got, SIGUSR1));
|
|
sigaddset(&want, SIGUSR1);
|
|
ASSERT_STREQ(DescribeSigset(0, &want), DescribeSigset(0, &got));
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR1, &oldsa, 0));
|
|
sigdelset(&want, SIGUSR1);
|
|
ASSERT_SYS(0, 0, sigprocmask(SIG_SETMASK, &want, 0));
|
|
}
|
|
|
|
TEST(sig_ign, discardsPendingSignalsEvenIfBlocked) {
|
|
sigset_t block, oldmask;
|
|
struct sigaction sa, oldsa;
|
|
ASSERT_SYS(0, 0, sigemptyset(&block));
|
|
ASSERT_SYS(0, 0, sigaddset(&block, SIGUSR1));
|
|
ASSERT_SYS(0, 0, sigprocmask(SIG_BLOCK, &block, &oldmask));
|
|
raise(SIGUSR1); // enqueue
|
|
sa.sa_flags = 0;
|
|
sa.sa_handler = SIG_IGN;
|
|
ASSERT_SYS(0, 0, sigemptyset(&sa.sa_mask));
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR1, &sa, &oldsa)); // discard
|
|
ASSERT_SYS(0, 0, sigprocmask(SIG_UNBLOCK, &block, 0));
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR1, &oldsa, 0));
|
|
ASSERT_SYS(0, 0, sigprocmask(SIG_SETMASK, &oldmask, 0));
|
|
}
|
|
|
|
void AutoMask(int sig, struct siginfo *si, void *ctx) {
|
|
sigset_t ss;
|
|
ucontext_t *uc = ctx;
|
|
sigprocmask(SIG_SETMASK, 0, &ss);
|
|
EXPECT_FALSE(sigismember(&uc->uc_sigmask, SIGUSR2)); // original mask
|
|
EXPECT_TRUE(sigismember(&ss, SIGUSR2)); // temporary mask
|
|
}
|
|
|
|
TEST(sigaction, signalBeingDeliveredGetsAutoMasked) {
|
|
sigset_t ss;
|
|
struct sigaction os, sa = {.sa_sigaction = AutoMask, .sa_flags = SA_SIGINFO};
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR2, &sa, &os));
|
|
raise(SIGUSR2);
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR2, &os, 0));
|
|
sigprocmask(SIG_SETMASK, 0, &ss);
|
|
EXPECT_FALSE(sigismember(&ss, SIGUSR2)); // original mask
|
|
}
|
|
|
|
void NoDefer(int sig, struct siginfo *si, void *ctx) {
|
|
sigset_t ss;
|
|
ucontext_t *uc = ctx;
|
|
sigprocmask(SIG_SETMASK, 0, &ss);
|
|
EXPECT_FALSE(sigismember(&uc->uc_sigmask, SIGUSR2));
|
|
EXPECT_FALSE(sigismember(&ss, SIGUSR2));
|
|
}
|
|
|
|
TEST(sigaction, NoDefer) {
|
|
struct sigaction os;
|
|
struct sigaction sa = {
|
|
.sa_sigaction = NoDefer,
|
|
.sa_flags = SA_SIGINFO | SA_NODEFER,
|
|
};
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR2, &sa, &os));
|
|
raise(SIGUSR2);
|
|
ASSERT_SYS(0, 0, sigaction(SIGUSR2, &os, 0));
|
|
}
|