src/test/test-signal-util.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <signal.h>
   4 #include <unistd.h>
   5
   6 #include "log.h"
   7 #include "macro.h"
   8 #include "signal-util.h"
   9 #include "stdio-util.h"
  10 #include "string-util.h"
  11 #include "process-util.h"
  12
  13 #define info(sig) log_info(#sig " = " STRINGIFY(sig) " = %d", sig)
  14
  15 static void test_rt_signals(void) {
  16         info(SIGRTMIN);
  17         info(SIGRTMAX);
  18
  19         /* We use signals SIGRTMIN+0 to SIGRTMIN+24 unconditionally */
  20         assert(SIGRTMAX - SIGRTMIN >= 24);
  21 }
  22
  23 static void test_signal_to_string_one(int val) {
  24         const char *p;
  25
  26         assert_se(p = signal_to_string(val));
  27
  28         assert_se(signal_from_string(p) == val);
  29
  30         p = strjoina("SIG", p);
  31         assert_se(signal_from_string(p) == val);
  32 }
  33
  34 static void test_signal_from_string_one(const char *s, int val) {
  35         const char *p;
  36
  37         assert_se(signal_from_string(s) == val);
  38
  39         p = strjoina("SIG", s);
  40         assert_se(signal_from_string(p) == val);
  41 }
  42
  43 static void test_signal_from_string_number(const char *s, int val) {
  44         const char *p;
  45
  46         assert_se(signal_from_string(s) == val);
  47
  48         p = strjoina("SIG", s);
  49         assert_se(signal_from_string(p) == -EINVAL);
  50 }
  51
  52 static void test_signal_from_string(void) {
  53         char buf[STRLEN("RTMIN+") + DECIMAL_STR_MAX(int) + 1];
  54
  55         test_signal_to_string_one(SIGHUP);
  56         test_signal_to_string_one(SIGTERM);
  57         test_signal_to_string_one(SIGRTMIN);
  58         test_signal_to_string_one(SIGRTMIN+3);
  59         test_signal_to_string_one(SIGRTMAX-4);
  60
  61         test_signal_from_string_one("RTMIN", SIGRTMIN);
  62         test_signal_from_string_one("RTMAX", SIGRTMAX);
  63
  64         xsprintf(buf, "RTMIN+%d", SIGRTMAX-SIGRTMIN);
  65         test_signal_from_string_one(buf, SIGRTMAX);
  66
  67         xsprintf(buf, "RTMIN+%d", INT_MAX);
  68         test_signal_from_string_one(buf, -ERANGE);
  69
  70         xsprintf(buf, "RTMAX-%d", SIGRTMAX-SIGRTMIN);
  71         test_signal_from_string_one(buf, SIGRTMIN);
  72
  73         xsprintf(buf, "RTMAX-%d", INT_MAX);
  74         test_signal_from_string_one(buf, -ERANGE);
  75
  76         test_signal_from_string_one("", -EINVAL);
  77         test_signal_from_string_one("hup", -EINVAL);
  78         test_signal_from_string_one("HOGEHOGE", -EINVAL);
  79
  80         test_signal_from_string_one("RTMIN-5", -EINVAL);
  81         test_signal_from_string_one("RTMIN-    5", -EINVAL);
  82         test_signal_from_string_one("RTMIN    -5", -EINVAL);
  83         test_signal_from_string_one("RTMIN+    5", -EINVAL);
  84         test_signal_from_string_one("RTMIN    +5", -EINVAL);
  85         test_signal_from_string_one("RTMIN+100", -ERANGE);
  86         test_signal_from_string_one("RTMIN+-3", -EINVAL);
  87         test_signal_from_string_one("RTMIN++3", -EINVAL);
  88         test_signal_from_string_one("RTMIN+HUP", -EINVAL);
  89         test_signal_from_string_one("RTMIN3", -EINVAL);
  90
  91         test_signal_from_string_one("RTMAX+5", -EINVAL);
  92         test_signal_from_string_one("RTMAX+    5", -EINVAL);
  93         test_signal_from_string_one("RTMAX    +5", -EINVAL);
  94         test_signal_from_string_one("RTMAX-    5", -EINVAL);
  95         test_signal_from_string_one("RTMAX    -5", -EINVAL);
  96         test_signal_from_string_one("RTMAX-100", -ERANGE);
  97         test_signal_from_string_one("RTMAX-+3", -EINVAL);
  98         test_signal_from_string_one("RTMAX--3", -EINVAL);
  99         test_signal_from_string_one("RTMAX-HUP", -EINVAL);
 100
 101         test_signal_from_string_number("3", 3);
 102         test_signal_from_string_number("+5", 5);
 103         test_signal_from_string_number("  +5", 5);
 104         test_signal_from_string_number("10000", -ERANGE);
 105         test_signal_from_string_number("-2", -ERANGE);
 106 }
 107
 108 static void test_block_signals(void) {
 109         sigset_t ss;
 110
 111         assert_se(sigprocmask(0, NULL, &ss) >= 0);
 112
 113         assert_se(sigismember(&ss, SIGUSR1) == 0);
 114         assert_se(sigismember(&ss, SIGALRM) == 0);
 115         assert_se(sigismember(&ss, SIGVTALRM) == 0);
 116
 117         {
 118                 BLOCK_SIGNALS(SIGUSR1, SIGVTALRM);
 119
 120                 assert_se(sigprocmask(0, NULL, &ss) >= 0);
 121                 assert_se(sigismember(&ss, SIGUSR1) == 1);
 122                 assert_se(sigismember(&ss, SIGALRM) == 0);
 123                 assert_se(sigismember(&ss, SIGVTALRM) == 1);
 124
 125         }
 126
 127         assert_se(sigprocmask(0, NULL, &ss) >= 0);
 128         assert_se(sigismember(&ss, SIGUSR1) == 0);
 129         assert_se(sigismember(&ss, SIGALRM) == 0);
 130         assert_se(sigismember(&ss, SIGVTALRM) == 0);
 131 }
 132
 133 static void test_ignore_signals(void) {
 134         assert_se(ignore_signals(SIGINT, -1) >= 0);
 135         assert_se(kill(getpid_cached(), SIGINT) >= 0);
 136         assert_se(ignore_signals(SIGUSR1, SIGUSR2, SIGTERM, SIGPIPE, -1) >= 0);
 137         assert_se(kill(getpid_cached(), SIGUSR1) >= 0);
 138         assert_se(kill(getpid_cached(), SIGUSR2) >= 0);
 139         assert_se(kill(getpid_cached(), SIGTERM) >= 0);
 140         assert_se(kill(getpid_cached(), SIGPIPE) >= 0);
 141         assert_se(default_signals(SIGINT, SIGUSR1, SIGUSR2, SIGTERM, SIGPIPE, -1) >= 0);
 142 }
 143
 144 int main(int argc, char *argv[]) {
 145         test_rt_signals();
 146         test_signal_from_string();
 147         test_block_signals();
 148         test_ignore_signals();
 149
 150         return 0;
 151 }