1 /* Copyright (C) 2003-2022 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public
6 License as published by the Free Software Foundation; either
7 version 2.1 of the License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 Lesser General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; if not, see
16 <https://www.gnu.org/licenses/>. */
17
18 #include <dlfcn.h>
19 #include <errno.h>
20 #include <mcheck.h>
21 #include <pthread.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <unistd.h>
25 #include <sys/wait.h>
26
27
28 /* Must be exported. */
29 int val;
30
31 static void
prepare(void)32 prepare (void)
33 {
34 val *= 2;
35 }
36
37 static void
parent(void)38 parent (void)
39 {
40 val += 4;
41 }
42
43 static void
child(void)44 child (void)
45 {
46 val += 8;
47 }
48
49
50 static int
do_test(void)51 do_test (void)
52 {
53 mtrace ();
54
55 if (pthread_atfork (prepare, parent, child) != 0)
56 {
57 puts ("do_test: atfork failed");
58 exit (1);
59 }
60
61 void *h = dlopen ("tst-atfork2mod.so", RTLD_LAZY);
62 if (h == NULL)
63 {
64 printf ("dlopen failed: %s\n", dlerror ());
65 exit (1);
66 }
67
68 /* First trial of fork. */
69 pid_t pid = fork ();
70 if (pid == -1)
71 {
72 puts ("1st fork failed");
73 exit (1);
74 }
75
76 if (pid == 0)
77 {
78 /* Child. */
79 if (val != 80)
80 {
81 printf ("1st: expected val=%d, got %d\n", 80, val);
82 exit (2);
83 }
84
85 exit (0);
86 }
87
88 /* Parent. */
89 if (val != 24)
90 {
91 printf ("1st: expected val=%d, got %d\n", 24, val);
92 exit (1);
93 }
94
95 int status;
96 if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
97 {
98 puts ("1st waitpid failed");
99 exit (1);
100 }
101
102 if (status != 0)
103 exit (status);
104
105 puts ("unloading now");
106
107 /* Unload the module. */
108 if (dlclose (h) != 0)
109 {
110 puts ("dlclose failed");
111 exit (1);
112 }
113
114 puts ("2nd fork");
115
116 /* Second fork trial. */
117 val = 1;
118 pid = fork ();
119 if (pid == -1)
120 {
121 puts ("2nd fork failed");
122 exit (1);
123 }
124
125 if (pid == 0)
126 {
127 /* Child. */
128 if (val != 10)
129 {
130 printf ("2nd: expected val=%d, got %d\n", 10, val);
131 exit (3);
132 }
133
134 exit (0);
135 }
136
137 /* Parent. */
138 if (val != 6)
139 {
140 printf ("2nd: expected val=%d, got %d\n", 6, val);
141 exit (1);
142 }
143
144 if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid)
145 {
146 puts ("2nd waitpid failed");
147 exit (1);
148 }
149
150 if (status != 0)
151 exit (status);
152
153 return 0;
154 }
155
156 #define TEST_FUNCTION do_test ()
157 #include "../test-skeleton.c"
158