1/*
2 * QEMU I/O channels external command driver
3 *
4 * Copyright (c) 2015 Red Hat, Inc.
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
18 *
19 */
20
21#include "qemu/osdep.h"
22#include "io/channel-command.h"
23#include "io/channel-watch.h"
24#include "qapi/error.h"
25#include "qemu/module.h"
26#include "qemu/sockets.h"
27#include "trace.h"
28
29
30QIOChannelCommand *
31qio_channel_command_new_pid(int writefd,
32 int readfd,
33 pid_t pid)
34{
35 QIOChannelCommand *ioc;
36
37 ioc = QIO_CHANNEL_COMMAND(object_new(TYPE_QIO_CHANNEL_COMMAND));
38
39 ioc->readfd = readfd;
40 ioc->writefd = writefd;
41 ioc->pid = pid;
42
43 trace_qio_channel_command_new_pid(ioc, writefd, readfd, pid);
44 return ioc;
45}
46
47
48#ifndef WIN32
49QIOChannelCommand *
50qio_channel_command_new_spawn(const char *const argv[],
51 int flags,
52 Error **errp)
53{
54 pid_t pid = -1;
55 int stdinfd[2] = { -1, -1 };
56 int stdoutfd[2] = { -1, -1 };
57 int devnull = -1;
58 bool stdinnull = false, stdoutnull = false;
59 QIOChannelCommand *ioc;
60
61 flags = flags & O_ACCMODE;
62
63 if (flags == O_RDONLY) {
64 stdinnull = true;
65 }
66 if (flags == O_WRONLY) {
67 stdoutnull = true;
68 }
69
70 if (stdinnull || stdoutnull) {
71 devnull = open("/dev/null", O_RDWR);
72 if (devnull < 0) {
73 error_setg_errno(errp, errno,
74 "Unable to open /dev/null");
75 goto error;
76 }
77 }
78
79 if ((!stdinnull && pipe(stdinfd) < 0) ||
80 (!stdoutnull && pipe(stdoutfd) < 0)) {
81 error_setg_errno(errp, errno,
82 "Unable to open pipe");
83 goto error;
84 }
85
86 pid = qemu_fork(errp);
87 if (pid < 0) {
88 goto error;
89 }
90
91 if (pid == 0) { /* child */
92 dup2(stdinnull ? devnull : stdinfd[0], STDIN_FILENO);
93 dup2(stdoutnull ? devnull : stdoutfd[1], STDOUT_FILENO);
94 /* Leave stderr connected to qemu's stderr */
95
96 if (!stdinnull) {
97 close(stdinfd[0]);
98 close(stdinfd[1]);
99 }
100 if (!stdoutnull) {
101 close(stdoutfd[0]);
102 close(stdoutfd[1]);
103 }
104 if (devnull != -1) {
105 close(devnull);
106 }
107
108 execv(argv[0], (char * const *)argv);
109 _exit(1);
110 }
111
112 if (!stdinnull) {
113 close(stdinfd[0]);
114 }
115 if (!stdoutnull) {
116 close(stdoutfd[1]);
117 }
118
119 ioc = qio_channel_command_new_pid(stdinnull ? devnull : stdinfd[1],
120 stdoutnull ? devnull : stdoutfd[0],
121 pid);
122 trace_qio_channel_command_new_spawn(ioc, argv[0], flags);
123 return ioc;
124
125 error:
126 if (devnull != -1) {
127 close(devnull);
128 }
129 if (stdinfd[0] != -1) {
130 close(stdinfd[0]);
131 }
132 if (stdinfd[1] != -1) {
133 close(stdinfd[1]);
134 }
135 if (stdoutfd[0] != -1) {
136 close(stdoutfd[0]);
137 }
138 if (stdoutfd[1] != -1) {
139 close(stdoutfd[1]);
140 }
141 return NULL;
142}
143
144#else /* WIN32 */
145QIOChannelCommand *
146qio_channel_command_new_spawn(const char *const argv[],
147 int flags,
148 Error **errp)
149{
150 error_setg_errno(errp, ENOSYS,
151 "Command spawn not supported on this platform");
152 return NULL;
153}
154#endif /* WIN32 */
155
156#ifndef WIN32
157static int qio_channel_command_abort(QIOChannelCommand *ioc,
158 Error **errp)
159{
160 pid_t ret;
161 int status;
162 int step = 0;
163
164 /* See if intermediate process has exited; if not, try a nice
165 * SIGTERM followed by a more severe SIGKILL.
166 */
167 rewait:
168 trace_qio_channel_command_abort(ioc, ioc->pid);
169 ret = waitpid(ioc->pid, &status, WNOHANG);
170 trace_qio_channel_command_wait(ioc, ioc->pid, ret, status);
171 if (ret == (pid_t)-1) {
172 if (errno == EINTR) {
173 goto rewait;
174 } else {
175 error_setg_errno(errp, errno,
176 "Cannot wait on pid %llu",
177 (unsigned long long)ioc->pid);
178 return -1;
179 }
180 } else if (ret == 0) {
181 if (step == 0) {
182 kill(ioc->pid, SIGTERM);
183 } else if (step == 1) {
184 kill(ioc->pid, SIGKILL);
185 } else {
186 error_setg(errp,
187 "Process %llu refused to die",
188 (unsigned long long)ioc->pid);
189 return -1;
190 }
191 step++;
192 usleep(10 * 1000);
193 goto rewait;
194 }
195
196 return 0;
197}
198#endif /* ! WIN32 */
199
200
201static void qio_channel_command_init(Object *obj)
202{
203 QIOChannelCommand *ioc = QIO_CHANNEL_COMMAND(obj);
204 ioc->readfd = -1;
205 ioc->writefd = -1;
206 ioc->pid = -1;
207}
208
209static void qio_channel_command_finalize(Object *obj)
210{
211 QIOChannelCommand *ioc = QIO_CHANNEL_COMMAND(obj);
212 if (ioc->readfd != -1) {
213 close(ioc->readfd);
214 }
215 if (ioc->writefd != -1 &&
216 ioc->writefd != ioc->readfd) {
217 close(ioc->writefd);
218 }
219 ioc->writefd = ioc->readfd = -1;
220 if (ioc->pid > 0) {
221#ifndef WIN32
222 qio_channel_command_abort(ioc, NULL);
223#endif
224 }
225}
226
227
228static ssize_t qio_channel_command_readv(QIOChannel *ioc,
229 const struct iovec *iov,
230 size_t niov,
231 int **fds,
232 size_t *nfds,
233 Error **errp)
234{
235 QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc);
236 ssize_t ret;
237
238 retry:
239 ret = readv(cioc->readfd, iov, niov);
240 if (ret < 0) {
241 if (errno == EAGAIN) {
242 return QIO_CHANNEL_ERR_BLOCK;
243 }
244 if (errno == EINTR) {
245 goto retry;
246 }
247
248 error_setg_errno(errp, errno,
249 "Unable to read from command");
250 return -1;
251 }
252
253 return ret;
254}
255
256static ssize_t qio_channel_command_writev(QIOChannel *ioc,
257 const struct iovec *iov,
258 size_t niov,
259 int *fds,
260 size_t nfds,
261 Error **errp)
262{
263 QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc);
264 ssize_t ret;
265
266 retry:
267 ret = writev(cioc->writefd, iov, niov);
268 if (ret <= 0) {
269 if (errno == EAGAIN) {
270 return QIO_CHANNEL_ERR_BLOCK;
271 }
272 if (errno == EINTR) {
273 goto retry;
274 }
275 error_setg_errno(errp, errno, "%s",
276 "Unable to write to command");
277 return -1;
278 }
279 return ret;
280}
281
282static int qio_channel_command_set_blocking(QIOChannel *ioc,
283 bool enabled,
284 Error **errp)
285{
286 QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc);
287
288 if (enabled) {
289 qemu_set_block(cioc->writefd);
290 qemu_set_block(cioc->readfd);
291 } else {
292 qemu_set_nonblock(cioc->writefd);
293 qemu_set_nonblock(cioc->readfd);
294 }
295
296 return 0;
297}
298
299
300static int qio_channel_command_close(QIOChannel *ioc,
301 Error **errp)
302{
303 QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc);
304 int rv = 0;
305#ifndef WIN32
306 pid_t wp;
307#endif
308
309 /* We close FDs before killing, because that
310 * gives a better chance of clean shutdown
311 */
312 if (cioc->readfd != -1 &&
313 close(cioc->readfd) < 0) {
314 rv = -1;
315 }
316 if (cioc->writefd != -1 &&
317 cioc->writefd != cioc->readfd &&
318 close(cioc->writefd) < 0) {
319 rv = -1;
320 }
321 cioc->writefd = cioc->readfd = -1;
322
323#ifndef WIN32
324 do {
325 wp = waitpid(cioc->pid, NULL, 0);
326 } while (wp == (pid_t)-1 && errno == EINTR);
327 if (wp == (pid_t)-1) {
328 error_setg_errno(errp, errno, "Failed to wait for pid %llu",
329 (unsigned long long)cioc->pid);
330 return -1;
331 }
332#endif
333
334 if (rv < 0) {
335 error_setg_errno(errp, errno, "%s",
336 "Unable to close command");
337 }
338 return rv;
339}
340
341
342static void qio_channel_command_set_aio_fd_handler(QIOChannel *ioc,
343 AioContext *ctx,
344 IOHandler *io_read,
345 IOHandler *io_write,
346 void *opaque)
347{
348 QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc);
349 aio_set_fd_handler(ctx, cioc->readfd, false, io_read, NULL, NULL, opaque);
350 aio_set_fd_handler(ctx, cioc->writefd, false, NULL, io_write, NULL, opaque);
351}
352
353
354static GSource *qio_channel_command_create_watch(QIOChannel *ioc,
355 GIOCondition condition)
356{
357 QIOChannelCommand *cioc = QIO_CHANNEL_COMMAND(ioc);
358 return qio_channel_create_fd_pair_watch(ioc,
359 cioc->readfd,
360 cioc->writefd,
361 condition);
362}
363
364
365static void qio_channel_command_class_init(ObjectClass *klass,
366 void *class_data G_GNUC_UNUSED)
367{
368 QIOChannelClass *ioc_klass = QIO_CHANNEL_CLASS(klass);
369
370 ioc_klass->io_writev = qio_channel_command_writev;
371 ioc_klass->io_readv = qio_channel_command_readv;
372 ioc_klass->io_set_blocking = qio_channel_command_set_blocking;
373 ioc_klass->io_close = qio_channel_command_close;
374 ioc_klass->io_create_watch = qio_channel_command_create_watch;
375 ioc_klass->io_set_aio_fd_handler = qio_channel_command_set_aio_fd_handler;
376}
377
378static const TypeInfo qio_channel_command_info = {
379 .parent = TYPE_QIO_CHANNEL,
380 .name = TYPE_QIO_CHANNEL_COMMAND,
381 .instance_size = sizeof(QIOChannelCommand),
382 .instance_init = qio_channel_command_init,
383 .instance_finalize = qio_channel_command_finalize,
384 .class_init = qio_channel_command_class_init,
385};
386
387static void qio_channel_command_register_types(void)
388{
389 type_register_static(&qio_channel_command_info);
390}
391
392type_init(qio_channel_command_register_types);
393