testcases/pounder21/fancy_timed_loop.c

   1 /* Repeatedly run a program with a given uid, gid and termination signal. */
   2
   3 /*
   4  * Copyright (C) 2003-2006 IBM
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  19  * 02111-1307, USA.
  20  */
  21
  22 #include <stdio.h>
  23 #include <string.h>
  24 #include <strings.h>
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <sys/types.h>
  28 #include <sys/wait.h>
  29
  30 #include "debug.h"
  31
  32 static int res = 0;
  33 static char *progname;
  34 static pid_t test_pgrp;
  35 static FILE *tty_fp;
  36
  37 static int the_signal = SIGTERM;
  38
  39 static void int_func(int signum) {
  40         pounder_fprintf(tty_fp, "%s: Killed by interrupt.  Last exit code = %d.\n",
  41                 progname, res);
  42         kill(-test_pgrp, the_signal);
  43         exit(res);
  44 }
  45 static void alarm_func(int signum) {
  46         pounder_fprintf(tty_fp, "%s: Killed by timer.  Last exit code = %d.\n",
  47                 progname, res);
  48         kill(-test_pgrp, the_signal);
  49         exit(res);
  50 }
  51
  52 int main(int argc, char *argv[]) {
  53         int secs, stat;
  54         pid_t pid;
  55         unsigned int revs = 0;
  56         struct sigaction zig;
  57         uid_t uid;
  58         gid_t gid;
  59
  60         if (argc < 5) {
  61                 printf("Usage: %s time_in_sec uid gid signal command [args]\n", argv[0]);
  62                 exit(1);
  63         }
  64
  65         tty_fp = fdopen(3, "w+");
  66         if (tty_fp == NULL) {
  67                 tty_fp = fopen("/dev/tty", "w+");
  68                 if (tty_fp == NULL) {
  69                         perror("stdout");
  70                         exit(2);
  71                 }
  72         }
  73
  74         progname = rindex(argv[5], '/');
  75         if (progname == NULL) {
  76                 progname = argv[5];
  77         } else {
  78                 progname++;
  79         }
  80
  81         /* Set up signals */
  82         memset(&zig, 0x00, sizeof(zig));
  83         zig.sa_handler = alarm_func;
  84         sigaction(SIGALRM, &zig, NULL);
  85         zig.sa_handler = int_func;
  86         sigaction(SIGINT, &zig, NULL);
  87         sigaction(SIGTERM, &zig, NULL);
  88
  89         /* set up process groups so that we can kill the
  90          * loop test and descendants easily */
  91
  92         secs = atoi(argv[1]);
  93         alarm(secs);
  94
  95         the_signal = atoi(argv[4]);
  96         uid = atoi(argv[2]);
  97         gid = atoi(argv[3]);
  98
  99         pounder_fprintf(tty_fp, "%s: uid = %d, gid = %d, sig = %d\n",
 100                 progname, uid, gid, the_signal);
 101
 102         while (1) {
 103                 pounder_fprintf(tty_fp, "%s: %s loop #%d.\n", progname,
 104                         start_msg, revs++);
 105                 pid = fork();
 106                 if (pid == 0) {
 107                         // set process group
 108                         if (setpgrp() < 0) {
 109                                 perror("setpgid");
 110                         }
 111
 112                         // set group and user id
 113                         if (setregid(gid, gid) != 0) {
 114                                 perror("setregid");
 115                                 exit(-1);
 116                         }
 117
 118                         if (setreuid(uid, uid) != 0) {
 119                                 perror("setreuid");
 120                                 exit(-1);
 121                         }
 122
 123                         // run the program
 124                         if (argc > 3) {
 125                                 stat = execvp(argv[5], &argv[5]);
 126                         } else {
 127                                 stat = execvp(argv[5], &argv[5]);
 128                         }
 129
 130                         perror(argv[5]);
 131
 132                         exit(-1);
 133                 }
 134
 135                 /* save the pgrp of the spawned process */
 136                 test_pgrp = pid;
 137
 138                 // wait for it to be done
 139                 if (waitpid(pid, &stat, 0) != pid) {
 140                         perror("waitpid");
 141                         exit(1);
 142                 }
 143
 144                 // interrogate it
 145                 if (WIFSIGNALED(stat)) {
 146                         pounder_fprintf(tty_fp, "%s: %s on signal %d.\n",
 147                                 progname, fail_msg, WTERMSIG(stat));
 148                         res = 255;
 149                 } else {
 150                         res = WEXITSTATUS(stat);
 151                         if (res == 0) {
 152                                 pounder_fprintf(tty_fp, "%s: %s.\n", progname, pass_msg);
 153                         } else if (res < 0 || res == 255) {
 154                                 pounder_fprintf(tty_fp, "%s: %s with code %d.\n",
 155                                         progname, abort_msg, res);
 156                                 exit(-1);
 157                                 // FIXME: add test to blacklist
 158                         } else {
 159                                 pounder_fprintf(tty_fp, "%s: %s with code %d.\n",
 160                                         progname, fail_msg, res);
 161                         }
 162                 }
 163         }
 164 }