1 | /* Test for vfork functions. |
2 | Copyright (C) 2007-2022 Free Software Foundation, Inc. |
3 | This file is part of the GNU C Library. |
4 | |
5 | The GNU C Library is free software; you can redistribute it and/or |
6 | modify it under the terms of the GNU Lesser General Public |
7 | License as published by the Free Software Foundation; either |
8 | version 2.1 of the License, or (at your option) any later version. |
9 | |
10 | The GNU C Library 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 GNU |
13 | Lesser General Public License for more details. |
14 | |
15 | You should have received a copy of the GNU Lesser General Public |
16 | License along with the GNU C Library; if not, see |
17 | <https://www.gnu.org/licenses/>. */ |
18 | |
19 | #include <errno.h> |
20 | #include <fcntl.h> |
21 | #include <mcheck.h> |
22 | #include <stdlib.h> |
23 | #include <string.h> |
24 | #include <unistd.h> |
25 | #include <sys/stat.h> |
26 | #include <sys/wait.h> |
27 | |
28 | static int do_test (void); |
29 | static void do_prepare (void); |
30 | char *tmpdirname; |
31 | |
32 | #define TEST_FUNCTION do_test () |
33 | #define PREPARE(argc, argv) do_prepare () |
34 | #include "../test-skeleton.c" |
35 | |
36 | static void |
37 | run_script (const char *script, char *const argv[]) |
38 | { |
39 | for (size_t i = 0; i < 5; i++) |
40 | { |
41 | pid_t pid = vfork (); |
42 | if (pid < 0) |
43 | FAIL_EXIT1 ("vfork failed: %m" ); |
44 | else if (pid == 0) |
45 | { |
46 | execvp (script, argv); |
47 | _exit (errno); |
48 | } |
49 | |
50 | int status; |
51 | if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid) |
52 | FAIL_EXIT1 ("waitpid failed" ); |
53 | else if (status != 0) |
54 | { |
55 | if (WIFEXITED (status)) |
56 | FAIL_EXIT1 ("%s failed with status %d\n" , script, |
57 | WEXITSTATUS (status)); |
58 | else |
59 | FAIL_EXIT1 ("%s killed by signal %d\n" , script, |
60 | WTERMSIG (status)); |
61 | } |
62 | } |
63 | } |
64 | |
65 | static int |
66 | do_test (void) |
67 | { |
68 | mtrace (); |
69 | |
70 | const char *path = getenv ("PATH" ); |
71 | if (path == NULL) |
72 | path = "/bin" ; |
73 | char pathbuf[strlen (tmpdirname) + 1 + strlen (path) + 1]; |
74 | strcpy (stpcpy (stpcpy (pathbuf, tmpdirname), ":" ), path); |
75 | if (setenv (name: "PATH" , value: pathbuf, replace: 1) < 0) |
76 | { |
77 | puts (s: "setenv failed" ); |
78 | return 1; |
79 | } |
80 | |
81 | /* Although manual states first argument should be the script name itself, |
82 | current execv{p,e} implementation allows it. */ |
83 | char *argv00[] = { NULL }; |
84 | run_script (script: "script0.sh" , argv: argv00); |
85 | |
86 | char *argv01[] = { (char*) "script0.sh" , NULL }; |
87 | run_script (script: "script0.sh" , argv: argv01); |
88 | |
89 | char *argv1[] = { (char *) "script1.sh" , (char *) "1" , NULL }; |
90 | run_script (script: "script1.sh" , argv: argv1); |
91 | |
92 | char *argv2[] = { (char *) "script2.sh" , (char *) "2" , NULL }; |
93 | run_script (script: "script2.sh" , argv: argv2); |
94 | |
95 | /* Same as before but with execlp. */ |
96 | for (size_t i = 0; i < 5; i++) |
97 | { |
98 | pid_t pid = vfork (); |
99 | if (pid < 0) |
100 | { |
101 | printf (format: "vfork failed: %m\n" ); |
102 | return 1; |
103 | } |
104 | else if (pid == 0) |
105 | { |
106 | execlp ("script2.sh" , "script2.sh" , "3" , NULL); |
107 | _exit (errno); |
108 | } |
109 | int status; |
110 | if (TEMP_FAILURE_RETRY (waitpid (pid, &status, 0)) != pid) |
111 | { |
112 | puts (s: "waitpid failed" ); |
113 | return 1; |
114 | } |
115 | else if (status != 0) |
116 | { |
117 | printf (format: "script2.sh failed with status %d\n" , status); |
118 | return 1; |
119 | } |
120 | } |
121 | |
122 | unsetenv (name: "PATH" ); |
123 | char *argv4[] = { (char *) "echo" , (char *) "script 4" , NULL }; |
124 | run_script (script: "echo" , argv: argv4); |
125 | |
126 | return 0; |
127 | } |
128 | |
129 | static void |
130 | create_script (const char *script, const char *contents, size_t size) |
131 | { |
132 | int fd = open (file: script, O_WRONLY | O_CREAT, 0700); |
133 | if (fd < 0 |
134 | || TEMP_FAILURE_RETRY (write (fd, contents, size)) != size |
135 | || fchmod (fd: fd, S_IRUSR | S_IXUSR) < 0) |
136 | FAIL_EXIT1 ("could not write %s\n" , script); |
137 | close (fd: fd); |
138 | } |
139 | |
140 | static void |
141 | do_prepare (void) |
142 | { |
143 | size_t len = strlen (test_dir) + sizeof ("/tst-vfork3.XXXXXX" ); |
144 | tmpdirname = malloc (size: len); |
145 | if (tmpdirname == NULL) |
146 | FAIL_EXIT1 ("out of memory" ); |
147 | strcpy (stpcpy (tmpdirname, test_dir), "/tst-vfork3.XXXXXX" ); |
148 | |
149 | tmpdirname = mkdtemp (template: tmpdirname); |
150 | if (tmpdirname == NULL) |
151 | FAIL_EXIT1 ("could not create temporary directory" ); |
152 | |
153 | char script0[len + sizeof "/script0.sh" ]; |
154 | char script1[len + sizeof "/script1.sh" ]; |
155 | char script2[len + sizeof "/script2.sh" ]; |
156 | |
157 | strcpy (stpcpy (script0, tmpdirname), "/script0.sh" ); |
158 | strcpy (stpcpy (script1, tmpdirname), "/script1.sh" ); |
159 | strcpy (stpcpy (script2, tmpdirname), "/script2.sh" ); |
160 | |
161 | add_temp_file (name: tmpdirname); |
162 | add_temp_file (name: script0); |
163 | add_temp_file (name: script1); |
164 | add_temp_file (name: script2); |
165 | |
166 | const char content0[] = "#!/bin/sh\necho empty\n" ; |
167 | create_script (script: script0, contents: content0, size: sizeof content0); |
168 | |
169 | const char content1[] = "#!/bin/sh\necho script $1\n" ; |
170 | create_script (script: script1, contents: content1, size: sizeof content1); |
171 | |
172 | const char content2[] = "echo script $1\n" ; |
173 | create_script (script: script2, contents: content2, size: sizeof content2); |
174 | } |
175 | |