tests: add signal receiver program

Target to kill with a check that will be written later.

CC: Bernhard Voelker <mail@bernhard-voelker.de>
Signed-off-by: Sami Kerola <kerolasa@iki.fi>

diff --git a/tests/commands.sh b/tests/commands.sh
index 68048fa4ebe3496c4a62d1afa29c20e74fa12689..0c8cfc946d15dc8f7167fa4fbc147aadd3515536 100644 (file)
--- a/tests/commands.sh
+++ b/tests/commands.sh
@@ -21,6 +21,7 @@ TS_HELPER_MD5="$top_builddir/test_md5"
 TS_HELPER_MORE=${TS_HELPER_MORE-"$top_builddir/test_more"}
 TS_HELPER_PARTITIONS="$top_builddir/sample-partitions"
 TS_HELPER_PATHS="$top_builddir/test_pathnames"
+TS_HELPER_SIGRECEIVE="$top_builddir/test_sigreceive"
 TS_HELPER_STRUTILS="$top_builddir/test_strutils"
 TS_HELPER_SYSINFO="$top_builddir/test_sysinfo"
 

diff --git a/tests/helpers/Makemodule.am b/tests/helpers/Makemodule.am
index 9724dae0928423a0f770f4c2f8eb195641268547..6ac5b7f01d752acf5aaf681c26376772a90a29af 100644 (file)
--- a/tests/helpers/Makemodule.am
+++ b/tests/helpers/Makemodule.am
@@ -10,3 +10,7 @@ test_pathnames_SOURCES = tests/helpers/test_pathnames.c
 
 check_PROGRAMS += test_sysinfo
 test_sysinfo_SOURCES = tests/helpers/test_sysinfo.c
+
+check_PROGRAMS += test_sigreceive
+test_sigreceive_SOURCES = tests/helpers/test_sigreceive.c
+test_sigreceive_LDADD = $(LDADD) libcommon.la

diff --git a/tests/helpers/test_sigreceive.c b/tests/helpers/test_sigreceive.c
new file mode 100644 (file)
index 0000000..3f76f06
--- /dev/null
+++ b/tests/helpers/test_sigreceive.c
@@ -0,0 +1,175 @@
+/*
+ * test_sigreceive - wait for signal and exit with value of it
+ *
+ * Written by Sami Kerola <kerolasa@iki.fi>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it would be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <err.h>
+#include <getopt.h>
+#include <pwd.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/select.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+#include "strutils.h"
+
+#define TEST_SIGRECEIVE_FAILURE 0
+
+static void __attribute__((__noreturn__)) usage(FILE *out)
+{
+       fputs("Usage: test_sigreceive [-s|--setuid <login|uid>]\n", out);
+       exit(TEST_SIGRECEIVE_FAILURE);
+}
+
+static __attribute__((__noreturn__))
+void exiter(int sig)
+{
+       _exit(sig);
+}
+
+int main(int argc, char **argv)
+{
+       struct sigaction sigact;
+       fd_set rfds;
+       struct timeval timeout;
+       char *user = NULL;
+       int c;
+
+       static const struct option longopts[] = {
+               {"setuid", required_argument, NULL, 's'},
+               {NULL, 0, NULL, 0}
+       };
+
+       while ((c = getopt_long(argc, argv, "s:h", longopts, NULL)) != -1)
+               switch (c) {
+               case 's':
+                       user = optarg;
+                       break;
+               case 'h':
+                       usage(stdout);
+               default:
+                       usage(stderr);
+               }
+
+       if (user) {
+               struct passwd *pw;
+               uid_t uid;
+
+               pw = getpwnam(user);
+               if (pw)
+                       uid = pw->pw_uid;
+               else
+                       uid = strtou32_or_err(user, "failed to parse uid");
+               if (setuid(uid) < 0)
+                       err(TEST_SIGRECEIVE_FAILURE, "setuid failed");
+       }
+
+       sigemptyset(&sigact.sa_mask);
+       sigact.sa_flags = 0;
+       sigact.sa_handler = exiter;
+       timeout.tv_sec = 5;
+       timeout.tv_usec = 0;
+
+       sigaction(SIGINT, &sigact, NULL);
+       sigaction(SIGQUIT, &sigact, NULL);
+       sigaction(SIGILL, &sigact, NULL);
+#ifdef SIGTRAP
+       sigaction(SIGTRAP, &sigact, NULL);
+#endif
+       sigaction(SIGABRT, &sigact, NULL);
+#ifdef SIGIOT
+       sigaction(SIGIOT, &sigact, NULL);
+#endif
+#ifdef SIGEMT
+       sigaction(SIGEMT, &sigact, NULL);
+#endif
+#ifdef SIGBUS
+       sigaction(SIGBUS, &sigact, NULL);
+#endif
+       sigaction(SIGFPE, &sigact, NULL);
+       sigaction(SIGUSR1, &sigact, NULL);
+       sigaction(SIGSEGV, &sigact, NULL);
+       sigaction(SIGUSR2, &sigact, NULL);
+       sigaction(SIGPIPE, &sigact, NULL);
+       sigaction(SIGALRM, &sigact, NULL);
+       sigaction(SIGTERM, &sigact, NULL);
+#ifdef SIGSTKFLT
+       sigaction(SIGSTKFLT, &sigact, NULL);
+#endif
+       sigaction(SIGCHLD, &sigact, NULL);
+#ifdef SIGCLD
+       sigaction(SIGCLD, &sigact, NULL);
+#endif
+       sigaction(SIGCONT, &sigact, NULL);
+       sigaction(SIGTSTP, &sigact, NULL);
+       sigaction(SIGTTIN, &sigact, NULL);
+       sigaction(SIGTTOU, &sigact, NULL);
+#ifdef SIGURG
+       sigaction(SIGURG, &sigact, NULL);
+#endif
+#ifdef SIGXCPU
+       sigaction(SIGXCPU, &sigact, NULL);
+#endif
+#ifdef SIGXFSZ
+       sigaction(SIGXFSZ, &sigact, NULL);
+#endif
+#ifdef SIGVTALRM
+       sigaction(SIGVTALRM, &sigact, NULL);
+#endif
+#ifdef SIGPROF
+       sigaction(SIGPROF, &sigact, NULL);
+#endif
+#ifdef SIGWINCH
+       sigaction(SIGWINCH, &sigact, NULL);
+#endif
+#ifdef SIGIO
+       sigaction(SIGIO, &sigact, NULL);
+#endif
+#ifdef SIGPOLL
+       sigaction(SIGPOLL, &sigact, NULL);
+#endif
+#ifdef SIGINFO
+       sigaction(SIGINFO, &sigact, NULL);
+#endif
+#ifdef SIGLOST
+       sigaction(SIGLOST, &sigact, NULL);
+#endif
+#ifdef SIGPWR
+       sigaction(SIGPWR, &sigact, NULL);
+#endif
+#ifdef SIGUNUSED
+       sigaction(SIGUNUSED, &sigact, NULL);
+#endif
+#ifdef SIGSYS
+       sigaction(SIGSYS, &sigact, NULL);
+#endif
+#ifdef SIGRTMIN
+       sigaction(SIGRTMIN, &sigact, NULL);
+       sigaction(SIGRTMAX, &sigact, NULL);
+#endif
+       /* Keep SIGHUP last, the bit it flips tells to check script the
+        * helper is ready to be killed.  */
+       sigaction(SIGHUP, &sigact, NULL);
+
+       FD_ZERO(&rfds);
+       FD_SET(STDIN_FILENO, &rfds);
+       select(0, &rfds, NULL, NULL, &timeout);
+
+       exiter(TEST_SIGRECEIVE_FAILURE);
+}