#define _GNU_SOURCE

#include <sched.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

const int STACK_SIZE = 4096;

/* volé depuis le template */
volatile char counter = 0;

int child()
{
  printf("Coucou depuis l'enfant\n");
  printf("parent id: %d\n", getppid());
  printf("process id: %d\n", getpid());
  printf("thread id: %d\n", gettid());
  printf("user id: %d\n", getuid());

  while (counter < 4)
  {
    printf("[Child] Counter: %d\n", counter);
    counter += 1;
    sleep(1);
  }

  return 0;
}

void help(char *name)
{
  printf("%s - Émule différent types de fork()\n", name);
  printf("Usage: %s <mode>\n\n", name);
  printf("  fork      Fork basique.\n");
  printf("  chimera   Créer un processus enfant partageant le même espace mémoire.\n");
  printf("  thread    Créé un thread.\n");
}

int main(int argc, char *argv[])
{
  if (argc < 2)
  {
    help(argv[0]);
    return 0;
  }

  char *mode = argv[1];
  int flags;

  /*
   * Je mets SIGCHLD quand on fait un fork basique, histoire d'avoir au moins
   * un flag. En théorie, il n'y en a pas besoin vu qu'on ne souhaite pas
   * gérer la mort de l'enfant.
   *
   * Pas besoin de notifier le père pour la chimère, on met directement le
   * flag pour partager la zone mémoire.
   */
  if (!strcmp(mode, "fork")) flags = SIGCHLD;
  else if (!strcmp(mode, "chimera")) flags = CLONE_VM;
  else if (!strcmp(mode, "thread")) flags = CLONE_VM | CLONE_THREAD
    | CLONE_SIGHAND;
  else
  {
    help(argv[0]);
    return 0;
  }

  char *stack = (char *) malloc(STACK_SIZE);
  int c_pid;

  printf("Coucou depuis le parent\n");
  printf("parent id: %d\n", getppid());
  printf("process id: %d\n", getpid());
  printf("thread id: %d\n", gettid());
  printf("user id: %d\n\n", getuid());

  /*
   * On passe la dernière adresse de la stack, cf. le manuel.
   */
  c_pid = clone(child, (stack + STACK_SIZE - 1), flags, NULL);

  while (counter < 4)
  {
    printf("[Parent] Counter: %d\n", counter);
    counter += 1;
    sleep(1);
  }

  free(stack);
  return 0;
}