1/* Check execvpe script argument handling.
2 Copyright (C) 2016-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 <stdio.h>
20#include <stdlib.h>
21#include <string.h>
22#include <unistd.h>
23#include <sys/stat.h>
24#include <sys/param.h>
25
26static char *fname1;
27static char *fname2;
28static char *logname;
29
30static void do_prepare (void);
31#define PREPARE(argc, argv) do_prepare ()
32static int do_test (void);
33#define TEST_FUNCTION do_test ()
34
35#include "../test-skeleton.c"
36
37static void
38do_prepare (void)
39{
40 int logfd = create_temp_file (base: "logfile", filename: &logname);
41 close (fd: logfd);
42
43 int fd1 = create_temp_file (base: "testscript", filename: &fname1);
44 dprintf (fd1, "echo foo $1 $2 $3 > %s\n", logname);
45 fchmod (fd: fd1, mode: 0700);
46 close (fd: fd1);
47
48 int fd2 = create_temp_file (base: "testscript", filename: &fname2);
49 dprintf (fd2, "echo foo > %s\n", logname);
50 fchmod (fd: fd2, mode: 0700);
51 close (fd: fd2);
52}
53
54static int
55run_script (const char *fname, char *args[])
56{
57 /* We want to test the `execvpe' function. To do this we restart the
58 program with an additional parameter. */
59 int status;
60 pid_t pid = fork ();
61 if (pid == 0)
62 {
63 execvpe (file: fname, argv: args, NULL);
64
65 puts (s: "Cannot exec");
66 exit (EXIT_FAILURE);
67 }
68 else if (pid == (pid_t) -1)
69 {
70 puts (s: "Cannot fork");
71 return 1;
72 }
73
74 /* Wait for the child. */
75 if (waitpid (pid: pid, stat_loc: &status, options: 0) != pid)
76 {
77 puts (s: "Wrong child");
78 return 1;
79 }
80
81 if (WTERMSIG (status) != 0)
82 {
83 puts (s: "Child terminated incorrectly");
84 return 1;
85 }
86
87 return 0;
88}
89
90static int
91check_output (const char *expected)
92{
93 /* Check log output. */
94 FILE *arq = fopen (logname, "r");
95 if (arq == NULL)
96 {
97 puts (s: "Error opening output file");
98 return 1;
99 }
100
101 char line[128];
102 if (fgets (s: line, n: sizeof (line), stream: arq) == NULL)
103 {
104 puts (s: "Error reading output file");
105 return 1;
106 }
107 fclose (arq);
108
109 if (strcmp (line, expected) != 0)
110 {
111 puts (s: "Output file different than expected");
112 return 1;
113 }
114
115 return 0;
116}
117
118static int
119do_test (void)
120{
121 if (setenv (name: "PATH", value: test_dir, replace: 1) != 0)
122 {
123 puts (s: "setenv failed");
124 return 1;
125 }
126
127 /* First check resulting script run with some arguments results in correct
128 output file. */
129 char *args1[] = { fname1, (char*) "1", (char *) "2", (char *) "3", NULL };
130 if (run_script (fname: fname1,args: args1))
131 return 1;
132 if (check_output (expected: "foo 1 2 3\n"))
133 return 1;
134
135 /* Same as before but with an expected empty argument list. */
136 char *args2[] = { fname2, NULL };
137 if (run_script (fname: fname2, args: args2))
138 return 1;
139 if (check_output (expected: "foo\n"))
140 return 1;
141
142 /* Same as before but with an empty argument list. */
143 char *args3[] = { NULL };
144 if (run_script (fname: fname2, args: args3))
145 return 1;
146 if (check_output (expected: "foo\n"))
147 return 1;
148
149 return 0;
150}
151

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