graphene/LibOS/shim/test/native/fork_bomb.c

/* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
/* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/wait.h>
#include <signal.h>
#include <sys/time.h>

int firstpid;
int secondpid;

#define NTRIES 1000
#define TEST_TIMES 32
#define DO_BENCH 1

int count = 0;

void sighand1 (int signum)
{
    count++;
#ifndef DO_BENCH
    printf("%d receive a SIGUSR (count = %d)\n",getpid(), count);
#endif
    kill(secondpid, SIGUSR1);
}

void sighand2 (int signum)
{
    count++;
#ifndef DO_BENCH
    printf("%d receive a SIGUSR (count = %d)\n",getpid(), count);
#endif
    kill(firstpid, SIGUSR1);
}

int main(int argc, char ** argv)
{
    int times = TEST_TIMES;

    for (int i = 0 ; i < times; i++) {
        int pipes[2];

        pipe(pipes);

        firstpid = fork();

        if (firstpid < 0) {
            printf("fork failed\n");
            return -1;
        }

        if (firstpid == 0) {
            close(pipes[1]);

            signal(SIGUSR1, sighand1);
            read(pipes[0], &secondpid, sizeof(int));
#ifndef DO_BENCH
            printf("%d killing %d\n",getpid(), secondpid);
#endif
            struct timeval start_time;
            gettimeofday(&start_time, NULL);
            kill(secondpid, SIGUSR1);
            while (count < NTRIES - 1)
                sleep(1);

            struct timeval finish_time;
            gettimeofday(&finish_time, NULL);

            printf("%d time spent: %lu microsecond\n", getpid(),
                   (finish_time.tv_sec * 1000000L + finish_time.tv_usec)
                   - (start_time.tv_sec * 1000000L + start_time.tv_usec));

            exit(0);
        }

        close(pipes[0]);

        secondpid = fork();

        if (secondpid < 0) {
            printf("fork failed\n");
            return -1;
        }

        if (secondpid == 0) {
            signal(SIGUSR1, sighand2);
            secondpid = getpid();
            write(pipes[1], &secondpid, sizeof(int));
            struct timeval start_time;
            gettimeofday(&start_time, NULL);

            while (count < NTRIES)
                sleep(1);

            struct timeval finish_time;
            gettimeofday(&finish_time, NULL);

            printf("%d time spent: %lu microsecond\n",getpid(),
                   (finish_time.tv_sec * 1000000L + finish_time.tv_usec)
                   - (start_time.tv_sec * 1000000L + start_time.tv_usec));

            exit(0);
        }

    }
    return 0;
}