1 /* Copyright (c) International Business Machines Corp., 2009 2 * Author: Matt Helsley <matthltc (at) us.ibm.com> 3 * 4 * This library is free software; you can redistribute it and/or 5 * modify it under the terms of the GNU Lesser General Public 6 * License as published by the Free Software Foundation; either 7 * version 2.1 of the License, or (at your option) any later version. 8 * 9 * This library is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 12 * Lesser General Public License for more details. 13 * 14 * You should have received a copy of the GNU Lesser General Public 15 * License along with this library; if not, write to the Free Software 16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 17 */ 18 #include <stdio.h> 19 #include <stdlib.h> 20 #include <string.h> 21 #include <errno.h> 22 23 #include <sys/time.h> 24 #include <sys/wait.h> 25 #include <time.h> 26 #include <unistd.h> 27 #include <signal.h> 28 29 int main(int argc, char **argv) 30 { 31 struct timeval start, now; 32 pid_t child_root; 33 unsigned int duration = 1, num_tasks = 0; 34 35 if (argc > 1) { 36 int i; 37 38 for (i = 0; i < argc; i++) { 39 if (strcmp(argv[i], "--help") == 0) { 40 printf("%s [duration]\n", argv[0]); 41 exit(EXIT_SUCCESS); 42 } 43 44 if (sscanf(argv[i], "%u", &duration) == 1) { 45 printf("%u second duration\n", duration); 46 break; 47 } 48 } 49 } 50 51 /* 52 * Wait until we're told to go. This allows the driver script to 53 * configure the test system (e.g. by putting this task in a 54 * suitable cgroup) before the timed forkbomb begins. 55 */ 56 { 57 char *word; 58 while (scanf("%as", &word) < 1) { 59 } 60 free(word); 61 } 62 if (gettimeofday(&start, NULL)) 63 exit(EXIT_FAILURE); 64 child_root = fork(); 65 if (child_root == -1) 66 exit(EXIT_FAILURE); 67 /* if (child_root == 0) 68 signal(SIGCHLD, SIG_IGN);*/ 69 do { 70 if (child_root == 0) { 71 /* signal(SIGCHLD, SIG_IGN);*/ 72 if (fork() == -1) 73 break; 74 } else { 75 if (wait(NULL) > 0) 76 num_tasks++; 77 } 78 if (gettimeofday(&now, NULL) && (errno != EINTR)) 79 break; 80 } while ((now.tv_sec - start.tv_sec) <= duration); 81 82 if (child_root != 0) 83 printf("Forked %d tasks\n", num_tasks); 84 exit(EXIT_SUCCESS); 85 } 86
