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 | |
26 | static char *fname1; |
27 | static char *fname2; |
28 | static char *logname; |
29 | |
30 | static void do_prepare (void); |
31 | #define PREPARE(argc, argv) do_prepare () |
32 | static int do_test (void); |
33 | #define TEST_FUNCTION do_test () |
34 | |
35 | #include "../test-skeleton.c" |
36 | |
37 | static void |
38 | do_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 | |
54 | static int |
55 | run_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 | |
90 | static int |
91 | check_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 | |
118 | static int |
119 | do_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 | |