(_SC_IOV_MAX): New definition.
[kopensolaris-gnu/glibc.git] / linuxthreads / ptfork.c
1 /* Linuxthreads - a simple clone()-based implementation of Posix        */
2 /* threads for Linux.                                                   */
3 /* Copyright (C) 1996 Xavier Leroy (Xavier.Leroy@inria.fr)              */
4 /*                                                                      */
5 /* This program is free software; you can redistribute it and/or        */
6 /* modify it under the terms of the GNU Library General Public License  */
7 /* as published by the Free Software Foundation; either version 2       */
8 /* of the License, or (at your option) any later version.               */
9 /*                                                                      */
10 /* This program is distributed in the hope that it will be useful,      */
11 /* but WITHOUT ANY WARRANTY; without even the implied warranty of       */
12 /* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        */
13 /* GNU Library General Public License for more details.                 */
14
15 /* The "atfork" stuff */
16
17 #include <errno.h>
18 #include <stddef.h>
19 #include <stdlib.h>
20 #include <unistd.h>
21 #include "pthread.h"
22 #include "internals.h"
23 #include <bits/libc-lock.h>
24
25 struct handler_list {
26   void (*handler)(void);
27   struct handler_list * next;
28 };
29
30 static pthread_mutex_t pthread_atfork_lock = PTHREAD_MUTEX_INITIALIZER;
31 static struct handler_list * pthread_atfork_prepare = NULL;
32 static struct handler_list * pthread_atfork_parent = NULL;
33 static struct handler_list * pthread_atfork_child = NULL;
34
35 static void pthread_insert_list(struct handler_list ** list,
36                                 void (*handler)(void),
37                                 struct handler_list * newlist,
38                                 int at_end)
39 {
40   if (handler == NULL) return;
41   if (at_end) {
42     while(*list != NULL) list = &((*list)->next);
43   }
44   newlist->handler = handler;
45   newlist->next = *list;
46   *list = newlist;
47 }
48
49 struct handler_list_block {
50   struct handler_list prepare, parent, child;
51 };
52
53 int __pthread_atfork(void (*prepare)(void),
54                      void (*parent)(void),
55                      void (*child)(void))
56 {
57   struct handler_list_block * block =
58     (struct handler_list_block *) malloc(sizeof(struct handler_list_block));
59   if (block == NULL) return ENOMEM;
60   pthread_mutex_lock(&pthread_atfork_lock);
61   /* "prepare" handlers are called in LIFO */
62   pthread_insert_list(&pthread_atfork_prepare, prepare, &block->prepare, 0);
63   /* "parent" handlers are called in FIFO */
64   pthread_insert_list(&pthread_atfork_parent, parent, &block->parent, 1);
65   /* "child" handlers are called in FIFO */
66   pthread_insert_list(&pthread_atfork_child, child, &block->child, 1);
67   pthread_mutex_unlock(&pthread_atfork_lock);
68   return 0;
69 }
70 strong_alias (__pthread_atfork, pthread_atfork)
71
72 static inline void pthread_call_handlers(struct handler_list * list)
73 {
74   for (/*nothing*/; list != NULL; list = list->next) (list->handler)();
75 }
76
77 extern int __libc_fork(void);
78
79 pid_t __fork(void)
80 {
81   pid_t pid;
82
83   pthread_mutex_lock(&pthread_atfork_lock);
84
85   pthread_call_handlers(pthread_atfork_prepare);
86   __pthread_once_fork_prepare();
87   __flockfilelist();
88
89   pid = __libc_fork();
90
91   if (pid == 0) {
92     __pthread_reset_main_thread();
93
94     __fresetlockfiles();
95     __pthread_once_fork_child();
96     pthread_call_handlers(pthread_atfork_child);
97
98     pthread_mutex_init(&pthread_atfork_lock, NULL);
99   } else {
100     __funlockfilelist();
101     __pthread_once_fork_parent();
102     pthread_call_handlers(pthread_atfork_parent);
103
104     pthread_mutex_unlock(&pthread_atfork_lock);
105   }
106
107   return pid;
108 }
109
110 weak_alias (__fork, fork);
111
112 pid_t __vfork(void)
113 {
114   return __fork();
115 }
116 weak_alias (__vfork, vfork);