brintos

brintos / llvm-project-archived public Read only

0
0
Text · 2.2 KiB · a12d798 Raw
66 lines · cpp
1//===-- Unittests for sigaction -------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "hdr/errno_macros.h"10#include "hdr/signal_macros.h"11#include "src/signal/raise.h"12#include "src/signal/sigaction.h"13#include "test/UnitTest/ErrnoSetterMatcher.h"14#include "test/UnitTest/Test.h"15 16using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails;17using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds;18 19TEST(LlvmLibcSigaction, Invalid) {20  // -1 is a much larger signal that NSIG, so this should fail.21  EXPECT_THAT(LIBC_NAMESPACE::sigaction(-1, nullptr, nullptr), Fails(EINVAL));22}23 24// SIGKILL cannot have its action changed, but it can be examined.25TEST(LlvmLibcSigaction, Sigkill) {26  struct sigaction action;27  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGKILL, nullptr, &action), Succeeds());28  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGKILL, &action, nullptr),29              Fails(EINVAL));30}31 32static int sigusr1Count;33static bool correctSignal;34 35TEST(LlvmLibcSigaction, CustomAction) {36  // Zero this incase tests get run multiple times in the future.37  sigusr1Count = 0;38 39  struct sigaction action;40  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGUSR1, nullptr, &action), Succeeds());41 42  action.sa_handler = +[](int signal) {43    correctSignal = signal == SIGUSR1;44    sigusr1Count++;45  };46  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGUSR1, &action, nullptr), Succeeds());47 48  LIBC_NAMESPACE::raise(SIGUSR1);49  EXPECT_EQ(sigusr1Count, 1);50  EXPECT_TRUE(correctSignal);51 52  action.sa_handler = SIG_DFL;53  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGUSR1, &action, nullptr), Succeeds());54 55  EXPECT_DEATH([] { LIBC_NAMESPACE::raise(SIGUSR1); }, WITH_SIGNAL(SIGUSR1));56}57 58TEST(LlvmLibcSigaction, Ignore) {59  struct sigaction action;60  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGUSR1, nullptr, &action), Succeeds());61  action.sa_handler = SIG_IGN;62  EXPECT_THAT(LIBC_NAMESPACE::sigaction(SIGUSR1, &action, nullptr), Succeeds());63 64  EXPECT_EXITS([] { LIBC_NAMESPACE::raise(SIGUSR1); }, 0);65}66