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
28static int do_test (void);
29static void do_prepare (void);
30char *tmpdirname;
31
32#define TEST_FUNCTION do_test ()
33#define PREPARE(argc, argv) do_prepare ()
34#include "../test-skeleton.c"
35
36static void
37run_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
65static int
66do_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
129static void
130create_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
140static void
141do_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

source code of glibc/posix/tst-vfork3.c