tests/forkbomb.cc

   1 // $Id: forkbomb.cc,v 1.1 2003/09/29 22:01:57 ensc Exp $
   2
   3 // Copyright (C) 2003 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
   4 // based on tests/forkbomb.cc by Jacques Gelinas
   5 //
   6 // This program is free software; you can redistribute it and/or modify
   7 // it under the terms of the GNU General Public License as published by
   8 // the Free Software Foundation; either version 2, or (at your option)
   9 // any later version.
  10 //
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU 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 02111-1307, USA.
  19
  20 #include <stdlib.h>
  21 #include <errno.h>
  22 #include <string.h>
  23 #include <sys/types.h>
  24 #include <sys/wait.h>
  25 #include <unistd.h>
  26 #include <stdio.h>
  27
  28 enum MODE {MODE_SLEEP,MODE_LOOP,MODE_FORK, MODE_FORKSHELL};
  29
  30
  31 static void forkbomb_userfork (MODE mode)
  32 {
  33         pid_t pid = fork();
  34         if (pid==-1){
  35                 fprintf (stderr,"Fork failed (%s)\n",strerror(errno));
  36         }else if (pid == 0){
  37                 if (mode == MODE_SLEEP){
  38                         sleep(20);
  39                 }else if (mode == MODE_LOOP){
  40                         int k=0;
  41                         while (1) k++;
  42                 }else if (mode == MODE_FORKSHELL){
  43                         system ("/bin/false");
  44                 }
  45                 _exit (0);
  46         }
  47 }
  48
  49
  50 int main (int argc, char *argv[])
  51 {
  52         if (argc != 4){
  53                 fprintf (stderr,"formboom N M mode\n"
  54                         "where N is the number of process to start\n"
  55                         "and M is the number of user to start\n"
  56                         "Each user will try to start N process\n"
  57                         "\n"
  58                         "mode is:\n"
  59                         "    sleep: Each process sleeps for 20 seconds and exits\n"
  60                         "    loop:  Each process loops forever\n"
  61                         "    fork:  Each process exits immediatly and is restarted\n"
  62                         "           by the parent\n"
  63                         "    forkshell: Each process runs /bin/false in a shell and\n"
  64                         "           exits, then the parent start a new one\n"
  65                         );
  66         }else{
  67                 MODE mode;
  68                 if (strcmp(argv[3],"sleep")==0){
  69                         mode = MODE_SLEEP;
  70                 }else if (strcmp(argv[3],"loop")==0){
  71                         mode = MODE_LOOP;
  72                 }else if (strcmp(argv[3],"fork")==0){
  73                         mode = MODE_FORK;
  74                 }else if (strcmp(argv[3],"forkshell")==0){
  75                         mode = MODE_FORKSHELL;
  76                 }else{
  77                         fprintf (stderr,"Invalid mode\n");
  78                         exit (-1);
  79                 }
  80                 for (int i=0; i<atoi(argv[2]); i++){
  81                         if (fork()==0){
  82                                 if (setuid (i+1)==-1){
  83                                         fprintf (stderr,"Can't setuid to uid %d (%s)\n",i+1
  84                                                 ,strerror(errno));
  85                                 }else{
  86                                         for (int j=0; j<atoi(argv[1]); j++){
  87                                                 forkbomb_userfork (mode);
  88                                         }
  89                                         if (mode == MODE_FORK || mode == MODE_FORKSHELL){
  90                                                 // Ok, all processes are started, in MODE_FORK
  91                                                 // we create a new one all the time
  92                                                 int status;
  93                                                 while (wait(&status)!=-1) forkbomb_userfork(mode);
  94                                         }
  95                                 }
  96                                 _exit (0);
  97                         }
  98                 }
  99                 system ("ps ax | wc -l");
 100                 printf ("All the process are running now\n");
 101                 printf ("Exit to end all processes\n");
 102                 system ("/bin/sh");
 103                 system ("killall forkbomb");
 104         }
 105         return 0;
 106 }
 107