#ifndef MASSDNS_FLOW_H
#define MASSDNS_FLOW_H
#include <signal.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/types.h>
static void kill_process_group(int sig)
{
static int received_termination = 0;
if(received_termination)
{
return;
}
received_termination = 1;
kill(0, sig);
exit(0);
}
static void handle_termination()
{
signal(SIGINT, kill_process_group);
signal(SIGTERM, kill_process_group);
}
// times is the number of resulting processes, i.e. if times is two, the process will fork once
size_t split_process(size_t times, pid_t *pids)
{
if(pids != NULL)
{
pids[0] = getpid();
}
for (size_t i = 0; i < times - 1; i++)
{
pid_t child = fork();
switch (child)
{
case -1:
{
perror("Failed to fork");
exit(EXIT_FAILURE);
}
case 0:
{
handle_termination();
return i + 1;
}
default:
if(pids != NULL)
{
pids[i + 1] = child;
}
break;
}
}
if(times > 1)
{
handle_termination();
}
return 0;
}
#endif