1 /*
2 * Copyright (c) 2013 Oracle and/or its affiliates. All Rights Reserved.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License as
6 * published by the Free Software Foundation; either version 2 of
7 * the License, or (at your option) any later version.
8 *
9 * This program is distributed in the hope that it would be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write the Free Software Foundation,
16 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
17 *
18 * Author: Alexey Kodanev <alexey.kodanev@oracle.com>
19 *
20 */
21
22 #include <errno.h>
23 #include <sys/types.h>
24 #include <sys/stat.h>
25 #include <sys/wait.h>
26 #include <fcntl.h>
27 #include <unistd.h>
28 #include <signal.h>
29 #include "test.h"
30
31 #define OPEN_MODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)
32 #define OPEN_FLAGS (O_WRONLY | O_APPEND | O_CREAT)
33
tst_run_cmd_fds_(void (cleanup_fn)(void),const char * const argv[],int stdout_fd,int stderr_fd,int pass_exit_val)34 int tst_run_cmd_fds_(void (cleanup_fn)(void),
35 const char *const argv[],
36 int stdout_fd,
37 int stderr_fd,
38 int pass_exit_val)
39 {
40 int rc;
41
42 if (argv == NULL || argv[0] == NULL) {
43 tst_brkm(TBROK, cleanup_fn,
44 "argument list is empty at %s:%d", __FILE__, __LINE__);
45 return -1;
46 }
47
48 /*
49 * The tst_sig() install poisoned signal handlers for all signals the
50 * test is not expected to get.
51 *
52 * So we temporarily disable the handler for sigchild we get after our
53 * child exits so that we don't have to disable it in each test that
54 * uses this interface.
55 */
56 void *old_handler = signal(SIGCHLD, SIG_DFL);
57
58 pid_t pid = vfork();
59 if (pid == -1) {
60 tst_brkm(TBROK | TERRNO, cleanup_fn, "vfork failed at %s:%d",
61 __FILE__, __LINE__);
62 return -1;
63 }
64 if (!pid) {
65 /* redirecting stdout and stderr if needed */
66 if (stdout_fd != -1) {
67 close(STDOUT_FILENO);
68 dup2(stdout_fd, STDOUT_FILENO);
69 }
70
71 if (stderr_fd != -1) {
72 close(STDERR_FILENO);
73 dup2(stderr_fd, STDERR_FILENO);
74 }
75
76 if (execvp(argv[0], (char *const *)argv)) {
77 if (errno == ENOENT)
78 _exit(255);
79 }
80 _exit(254);
81 }
82
83 int ret = -1;
84 if (waitpid(pid, &ret, 0) != pid) {
85 tst_brkm(TBROK | TERRNO, cleanup_fn, "waitpid failed at %s:%d",
86 __FILE__, __LINE__);
87 return -1;
88 }
89
90 signal(SIGCHLD, old_handler);
91
92 if (!WIFEXITED(ret)) {
93 tst_brkm(TBROK, cleanup_fn, "failed to exec cmd '%s' at %s:%d",
94 argv[0], __FILE__, __LINE__);
95 return -1;
96 }
97
98 rc = WEXITSTATUS(ret);
99
100 if ((!pass_exit_val) && rc) {
101 tst_brkm(TBROK, cleanup_fn,
102 "'%s' exited with a non-zero code %d at %s:%d",
103 argv[0], rc, __FILE__, __LINE__);
104 return -1;
105 }
106
107 return rc;
108 }
109
tst_run_cmd_(void (cleanup_fn)(void),const char * const argv[],const char * stdout_path,const char * stderr_path,int pass_exit_val)110 int tst_run_cmd_(void (cleanup_fn)(void),
111 const char *const argv[],
112 const char *stdout_path,
113 const char *stderr_path,
114 int pass_exit_val)
115 {
116 int stdout_fd = -1;
117 int stderr_fd = -1;
118 int rc;
119
120 if (stdout_path != NULL) {
121 stdout_fd = open(stdout_path,
122 OPEN_FLAGS, OPEN_MODE);
123
124 if (stdout_fd == -1)
125 tst_resm(TWARN | TERRNO,
126 "open() on %s failed at %s:%d",
127 stdout_path, __FILE__, __LINE__);
128 }
129
130 if (stderr_path != NULL) {
131 stderr_fd = open(stderr_path,
132 OPEN_FLAGS, OPEN_MODE);
133
134 if (stderr_fd == -1)
135 tst_resm(TWARN | TERRNO,
136 "open() on %s failed at %s:%d",
137 stderr_path, __FILE__, __LINE__);
138 }
139
140 rc = tst_run_cmd_fds(cleanup_fn, argv, stdout_fd, stderr_fd,
141 pass_exit_val);
142
143 if ((stdout_fd != -1) && (close(stdout_fd) == -1))
144 tst_resm(TWARN | TERRNO,
145 "close() on %s failed at %s:%d",
146 stdout_path, __FILE__, __LINE__);
147
148 if ((stderr_fd != -1) && (close(stderr_fd) == -1))
149 tst_resm(TWARN | TERRNO,
150 "close() on %s failed at %s:%d",
151 stderr_path, __FILE__, __LINE__);
152
153 return rc;
154 }
155
tst_system(const char * command)156 int tst_system(const char *command)
157 {
158 int ret = 0;
159
160 /*
161 *Temporarily disable SIGCHLD of user defined handler, so the
162 *system(3) function will not cause unexpected SIGCHLD signal
163 *callback function for test cases.
164 */
165 void *old_handler = signal(SIGCHLD, SIG_DFL);
166
167 ret = system(command);
168
169 signal(SIGCHLD, old_handler);
170 return ret;
171 }
172