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>
This commit is contained in:
parent
3665b07cd8
commit
6b93dcbee1
|
@ -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"
|
||||
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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);
|
||||
}
|
Loading…
Reference in New Issue