nsz Git - musl/blob - src/thread/pthread_atfork.c

   1 #include <pthread.h>
   2 #include "libc.h"
   3
   4 static struct atfork_funcs {
   5         void (*prepare)(void);
   6         void (*parent)(void);
   7         void (*child)(void);
   8         struct atfork_funcs *prev, *next;
   9 } *funcs;
  10
  11 static int lock;
  12
  13 static void fork_handler(int who)
  14 {
  15         struct atfork_funcs *p;
  16         if (who < 0) {
  17                 LOCK(&lock);
  18                 for (p=funcs; p; p = p->next) {
  19                         if (p->prepare) p->prepare();
  20                         funcs = p;
  21                 }
  22         } else {
  23                 for (p=funcs; p; p = p->prev) {
  24                         if (!who && p->parent) p->parent();
  25                         else if (who && p->child) p->child();
  26                         funcs = p;
  27                 }
  28                 UNLOCK(&lock);
  29         }
  30 }
  31
  32 int pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void))
  33 {
  34         struct atfork_funcs *new = malloc(sizeof *new);
  35         if (!new) return -1;
  36
  37         LOCK(&lock);
  38         libc.fork_handler = fork_handler;
  39         new->next = funcs;
  40         new->prev = 0;
  41         new->prepare = prepare;
  42         new->parent = parent;
  43         new->child = child;
  44         if (funcs) funcs->prev = new;
  45         funcs = new;
  46         UNLOCK(&lock);
  47         return 0;
  48 }