-
-
Save billywhizz/e483ff00a7f18bdb577d15853df646d9 to your computer and use it in GitHub Desktop.
A Linux process cloning example using Clone().
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
all: default | |
default: waitpid waitpid_optimized | |
waitpid: waitpid.c Makefile | |
$(CC) -Wall -Werror -std=gnu17 -ggdb -o waitpid waitpid.c | |
waitpid_optimized: waitpid.c Makefile | |
$(CC) -Wall -Werror -std=gnu17 -Ofast -o waitpid_optimized waitpid.c | |
clean: | |
$(RM) waitpid waitpid_optimized |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#define _GNU_SOURCE | |
#include <sched.h> | |
#include <stdio.h> | |
#include <stdlib.h> | |
#include <string.h> | |
#include <sys/wait.h> | |
#include <unistd.h> | |
#define STACK_SIZE (1024 * 1024) /* Stack size for cloned child */ | |
#define ERREXIT(msg) \ | |
{ \ | |
perror(msg); \ | |
exit(EXIT_FAILURE); \ | |
} | |
#define CHECKALLOC(ptr, msg) \ | |
({ \ | |
void *p = ptr; \ | |
if (p == NULL) \ | |
ERREXIT(msg); \ | |
p; \ | |
}) | |
static int | |
childFunc(void *arg __attribute__((unused))) | |
{ | |
puts("child: start"); | |
sleep(2); | |
puts("child: terminate"); | |
return 0; /* Child terminates now */ | |
} | |
int | |
main(int argc, char *argv[]) | |
{ | |
char **stacks; /* Start of stack buffer */ | |
pid_t *pids; /* Child process's pids */ | |
size_t nproc, i; | |
if (argc != 2) { | |
puts("Wrong way to execute the program:\n" | |
"\t\t./waitpid nProcesses\n" | |
"example:\t./waitpid 2"); | |
return EXIT_FAILURE; | |
} | |
nproc = atol(argv[1]); /* Process count */ | |
stacks = CHECKALLOC(malloc(nproc * sizeof(void *)), "malloc"); | |
pids = CHECKALLOC(malloc(nproc * sizeof(pid_t)), "malloc"); | |
for (i = 0; i < nproc; i++) { | |
char *stackTop; /* End of stack buffer */ | |
stacks[i] = CHECKALLOC(malloc(STACK_SIZE), "stack malloc"); | |
stackTop = stacks[i] + STACK_SIZE; | |
pids[i] = clone(childFunc, stackTop, | |
CLONE_CHILD_CLEARTID | CLONE_CHILD_SETTID | SIGCHLD, NULL); | |
if (pids[i] == -1) | |
ERREXIT("clone"); | |
printf("clone() returned %ld\n", (long)pids[i]); | |
} | |
sleep(1); | |
for (i = 0; i < nproc; i++) { | |
if (waitpid(pids[i], NULL, 0) == -1) | |
ERREXIT("waitpid"); | |
printf("child %ld has terminated\n", (long)pids[i]); | |
} | |
for (i = 0; i < nproc; i++) | |
free(stacks[i]); | |
free(stacks); | |
free(pids); | |
return EXIT_SUCCESS; | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment