cgroups/fork_bomb.c

/* fork_bomb.c

   A fork-bomb program that can be useful when experimenting with
   the cgroups 'pids' controller.

   Usage: fork_bomb num-children [parent-sleep-secs [child-sleep-secs]]

   This program uses fork(2) to create 'num-children' child processes that
   each sleep (i.e., stay in existence) for 'child-sleep-secs' (default:
   1000) seconds.

   The 'parent-sleep-secs' argument specifies an a number of seconds
   (default: 0) that the program should sleep before creating any children.
   The intention of this sleep is to allow the user some time to manipulate
   the cgroup membership of the parent process before any child processes
   are created.
*/
#include <sys/wait.h>
#include "tlpi_hdr.h"

int
main(int argc, char *argv[])
{
    int numChildren, failed;
    pid_t childPid;
    int childSleepTime, parentSleepTime;

    if (argc < 2) {
        usageErr("%s num-children [parent-sleep-secs [child-sleep-secs]]\n",
                argv[0]);
    }

    numChildren = atoi(argv[1]);
    parentSleepTime = (argc > 2) ? atoi(argv[2]) : 0;
    childSleepTime = (argc > 3) ? atoi(argv[3]) : 300;

    printf("Parent PID = %ld\n", (long) getpid());

    /* If a parent sleep time was specified, show the user the parent's
       PID in case they want to write it to a cgroup.procs file */

    if (parentSleepTime > 0) {
        printf("Parent sleeping for %d seconds\n", parentSleepTime);
        sleep(parentSleepTime);
    }

    printf("Creating %d children that will sleep %d seconds\n",
            numChildren, childSleepTime);

    failed = 0;
    for (int j = 1; j <= numChildren && !failed; j++) {
        switch (childPid = fork()) {
        case -1:
            errMsg("fork");
            failed = 1;
            break;
        case 0:
            sleep(childSleepTime);
            exit(EXIT_SUCCESS);
        default:
            printf("Child %d: PID = %ld\n", j, (long) childPid);
            break;
        }
    }

    printf("Waiting for all children to terminate\n");

    while (waitpid(-1, NULL, 0) > 0)
        continue;

    printf("All children terminated; bye!\n");

    exit(EXIT_SUCCESS);
}