142074a9dSMichael Roth /*
242074a9dSMichael Roth * QEMU Guest Agent common/cross-platform command implementations
342074a9dSMichael Roth *
442074a9dSMichael Roth * Copyright IBM Corp. 2012
542074a9dSMichael Roth *
642074a9dSMichael Roth * Authors:
742074a9dSMichael Roth * Michael Roth <mdroth@linux.vnet.ibm.com>
842074a9dSMichael Roth *
942074a9dSMichael Roth * This work is licensed under the terms of the GNU GPL, version 2 or later.
1042074a9dSMichael Roth * See the COPYING file in the top-level directory.
1142074a9dSMichael Roth */
1242074a9dSMichael Roth
134459bf38SPeter Maydell #include "qemu/osdep.h"
141329651fSPhilippe Mathieu-Daudé #include "qemu/units.h"
15dc03272dSMichael S. Tsirkin #include "guest-agent-core.h"
16eb815e24SMarkus Armbruster #include "qga-qapi-commands.h"
17e688df6bSMarkus Armbruster #include "qapi/error.h"
187b1b5d19SPaolo Bonzini #include "qapi/qmp/qerror.h"
19920639caSDaniel P. Berrange #include "qemu/base64.h"
20f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
21ead83a13SPhilippe Mathieu-Daudé #include "commands-common.h"
2242074a9dSMichael Roth
23a1853dcaSYuri Pudgorodskiy /* Maximum captured guest-exec out_data/err_data - 16MB */
24a1853dcaSYuri Pudgorodskiy #define GUEST_EXEC_MAX_OUTPUT (16 * 1024 * 1024)
25a1853dcaSYuri Pudgorodskiy /* Allocation and I/O buffer for reading guest-exec out_data/err_data - 4KB */
26a1853dcaSYuri Pudgorodskiy #define GUEST_EXEC_IO_SIZE (4 * 1024)
271329651fSPhilippe Mathieu-Daudé /*
281329651fSPhilippe Mathieu-Daudé * Maximum file size to read - 48MB
291329651fSPhilippe Mathieu-Daudé *
301329651fSPhilippe Mathieu-Daudé * (48MB + Base64 3:4 overhead = JSON parser 64 MB limit)
311329651fSPhilippe Mathieu-Daudé */
321329651fSPhilippe Mathieu-Daudé #define GUEST_FILE_READ_COUNT_MAX (48 * MiB)
33a1853dcaSYuri Pudgorodskiy
3442074a9dSMichael Roth /* Note: in some situations, like with the fsfreeze, logging may be
35c4023359SMarkus Armbruster * temporarily disabled. if it is necessary that a command be able
36c4023359SMarkus Armbruster * to log for accounting purposes, check ga_logging_enabled() beforehand.
3742074a9dSMichael Roth */
slog(const gchar * fmt,...)3842074a9dSMichael Roth void slog(const gchar *fmt, ...)
3942074a9dSMichael Roth {
4042074a9dSMichael Roth va_list ap;
4142074a9dSMichael Roth
4242074a9dSMichael Roth va_start(ap, fmt);
4342074a9dSMichael Roth g_logv("syslog", G_LOG_LEVEL_INFO, fmt, ap);
4442074a9dSMichael Roth va_end(ap);
4542074a9dSMichael Roth }
4642074a9dSMichael Roth
qmp_guest_sync_delimited(int64_t id,Error ** errp)473cf0bed8SMichael Roth int64_t qmp_guest_sync_delimited(int64_t id, Error **errp)
483cf0bed8SMichael Roth {
493cf0bed8SMichael Roth ga_set_response_delimited(ga_state);
503cf0bed8SMichael Roth return id;
513cf0bed8SMichael Roth }
523cf0bed8SMichael Roth
qmp_guest_sync(int64_t id,Error ** errp)5342074a9dSMichael Roth int64_t qmp_guest_sync(int64_t id, Error **errp)
5442074a9dSMichael Roth {
5542074a9dSMichael Roth return id;
5642074a9dSMichael Roth }
5742074a9dSMichael Roth
qmp_guest_ping(Error ** errp)5877dbc81bSMarkus Armbruster void qmp_guest_ping(Error **errp)
5942074a9dSMichael Roth {
6042074a9dSMichael Roth slog("guest-ping called");
6142074a9dSMichael Roth }
6242074a9dSMichael Roth
qmp_command_info(const QmpCommand * cmd,void * opaque)63f0ccc00bSMarc-André Lureau static void qmp_command_info(const QmpCommand *cmd, void *opaque)
6442074a9dSMichael Roth {
658dc4d915SMark Wu GuestAgentInfo *info = opaque;
6642074a9dSMichael Roth GuestAgentCommandInfo *cmd_info;
6742074a9dSMichael Roth
68f3a06403SMarkus Armbruster cmd_info = g_new0(GuestAgentCommandInfo, 1);
698dc4d915SMark Wu cmd_info->name = g_strdup(qmp_command_name(cmd));
708dc4d915SMark Wu cmd_info->enabled = qmp_command_is_enabled(cmd);
710106dc4fSMark Wu cmd_info->success_response = qmp_has_success_response(cmd);
7242074a9dSMichael Roth
7354aa3de7SEric Blake QAPI_LIST_PREPEND(info->supported_commands, cmd_info);
7442074a9dSMichael Roth }
7542074a9dSMichael Roth
qmp_guest_info(Error ** errp)7677dbc81bSMarkus Armbruster struct GuestAgentInfo *qmp_guest_info(Error **errp)
778dc4d915SMark Wu {
78f3a06403SMarkus Armbruster GuestAgentInfo *info = g_new0(GuestAgentInfo, 1);
798dc4d915SMark Wu
808dc4d915SMark Wu info->version = g_strdup(QEMU_VERSION);
811527badbSMarkus Armbruster qmp_for_each_command(&ga_commands, qmp_command_info, info);
8242074a9dSMichael Roth return info;
8342074a9dSMichael Roth }
84d697e30cSYuri Pudgorodskiy
85a1853dcaSYuri Pudgorodskiy struct GuestExecIOData {
86a1853dcaSYuri Pudgorodskiy guchar *data;
87a1853dcaSYuri Pudgorodskiy gsize size;
88a1853dcaSYuri Pudgorodskiy gsize length;
89a31393e7SAlex Bennée bool closed;
90a1853dcaSYuri Pudgorodskiy bool truncated;
91a1853dcaSYuri Pudgorodskiy const char *name;
92a1853dcaSYuri Pudgorodskiy };
93a1853dcaSYuri Pudgorodskiy typedef struct GuestExecIOData GuestExecIOData;
94a1853dcaSYuri Pudgorodskiy
95d697e30cSYuri Pudgorodskiy struct GuestExecInfo {
96d697e30cSYuri Pudgorodskiy GPid pid;
97d697e30cSYuri Pudgorodskiy int64_t pid_numeric;
98d697e30cSYuri Pudgorodskiy gint status;
99a1853dcaSYuri Pudgorodskiy bool has_output;
100a31393e7SAlex Bennée bool finished;
101a1853dcaSYuri Pudgorodskiy GuestExecIOData in;
102a1853dcaSYuri Pudgorodskiy GuestExecIOData out;
103a1853dcaSYuri Pudgorodskiy GuestExecIOData err;
104d697e30cSYuri Pudgorodskiy QTAILQ_ENTRY(GuestExecInfo) next;
105d697e30cSYuri Pudgorodskiy };
106d697e30cSYuri Pudgorodskiy typedef struct GuestExecInfo GuestExecInfo;
107d697e30cSYuri Pudgorodskiy
108d697e30cSYuri Pudgorodskiy static struct {
109d697e30cSYuri Pudgorodskiy QTAILQ_HEAD(, GuestExecInfo) processes;
110d697e30cSYuri Pudgorodskiy } guest_exec_state = {
111d697e30cSYuri Pudgorodskiy .processes = QTAILQ_HEAD_INITIALIZER(guest_exec_state.processes),
112d697e30cSYuri Pudgorodskiy };
113d697e30cSYuri Pudgorodskiy
gpid_to_int64(GPid pid)114d697e30cSYuri Pudgorodskiy static int64_t gpid_to_int64(GPid pid)
115d697e30cSYuri Pudgorodskiy {
116d697e30cSYuri Pudgorodskiy #ifdef G_OS_WIN32
117d697e30cSYuri Pudgorodskiy return GetProcessId(pid);
118d697e30cSYuri Pudgorodskiy #else
119d697e30cSYuri Pudgorodskiy return (int64_t)pid;
120d697e30cSYuri Pudgorodskiy #endif
121d697e30cSYuri Pudgorodskiy }
122d697e30cSYuri Pudgorodskiy
guest_exec_info_add(GPid pid)123d697e30cSYuri Pudgorodskiy static GuestExecInfo *guest_exec_info_add(GPid pid)
124d697e30cSYuri Pudgorodskiy {
125d697e30cSYuri Pudgorodskiy GuestExecInfo *gei;
126d697e30cSYuri Pudgorodskiy
127d697e30cSYuri Pudgorodskiy gei = g_new0(GuestExecInfo, 1);
128d697e30cSYuri Pudgorodskiy gei->pid = pid;
129d697e30cSYuri Pudgorodskiy gei->pid_numeric = gpid_to_int64(pid);
130d697e30cSYuri Pudgorodskiy QTAILQ_INSERT_TAIL(&guest_exec_state.processes, gei, next);
131d697e30cSYuri Pudgorodskiy
132d697e30cSYuri Pudgorodskiy return gei;
133d697e30cSYuri Pudgorodskiy }
134d697e30cSYuri Pudgorodskiy
guest_exec_info_find(int64_t pid_numeric)135d697e30cSYuri Pudgorodskiy static GuestExecInfo *guest_exec_info_find(int64_t pid_numeric)
136d697e30cSYuri Pudgorodskiy {
137d697e30cSYuri Pudgorodskiy GuestExecInfo *gei;
138d697e30cSYuri Pudgorodskiy
139d697e30cSYuri Pudgorodskiy QTAILQ_FOREACH(gei, &guest_exec_state.processes, next) {
140d697e30cSYuri Pudgorodskiy if (gei->pid_numeric == pid_numeric) {
141d697e30cSYuri Pudgorodskiy return gei;
142d697e30cSYuri Pudgorodskiy }
143d697e30cSYuri Pudgorodskiy }
144d697e30cSYuri Pudgorodskiy
145d697e30cSYuri Pudgorodskiy return NULL;
146d697e30cSYuri Pudgorodskiy }
147d697e30cSYuri Pudgorodskiy
qmp_guest_exec_status(int64_t pid,Error ** errp)148b90abbacSVladimir Sementsov-Ogievskiy GuestExecStatus *qmp_guest_exec_status(int64_t pid, Error **errp)
149d697e30cSYuri Pudgorodskiy {
150d697e30cSYuri Pudgorodskiy GuestExecInfo *gei;
151d697e30cSYuri Pudgorodskiy GuestExecStatus *ges;
152d697e30cSYuri Pudgorodskiy
153d697e30cSYuri Pudgorodskiy slog("guest-exec-status called, pid: %u", (uint32_t)pid);
154d697e30cSYuri Pudgorodskiy
155d697e30cSYuri Pudgorodskiy gei = guest_exec_info_find(pid);
156d697e30cSYuri Pudgorodskiy if (gei == NULL) {
157*b6651659SMarkus Armbruster error_setg(errp, "PID " PRId64 " does not exist");
158d697e30cSYuri Pudgorodskiy return NULL;
159d697e30cSYuri Pudgorodskiy }
160d697e30cSYuri Pudgorodskiy
161d697e30cSYuri Pudgorodskiy ges = g_new0(GuestExecStatus, 1);
162d697e30cSYuri Pudgorodskiy
163c267d750SMarc-André Lureau bool finished = gei->finished;
164a1853dcaSYuri Pudgorodskiy
165a1853dcaSYuri Pudgorodskiy /* need to wait till output channels are closed
166a1853dcaSYuri Pudgorodskiy * to be sure we captured all output at this point */
167a1853dcaSYuri Pudgorodskiy if (gei->has_output) {
168c267d750SMarc-André Lureau finished &= gei->out.closed && gei->err.closed;
169a1853dcaSYuri Pudgorodskiy }
170a1853dcaSYuri Pudgorodskiy
171a1853dcaSYuri Pudgorodskiy ges->exited = finished;
172a1853dcaSYuri Pudgorodskiy if (finished) {
173d697e30cSYuri Pudgorodskiy /* Glib has no portable way to parse exit status.
174d697e30cSYuri Pudgorodskiy * On UNIX, we can get either exit code from normal termination
175d697e30cSYuri Pudgorodskiy * or signal number.
176d697e30cSYuri Pudgorodskiy * On Windows, it is either the same exit code or the exception
177d697e30cSYuri Pudgorodskiy * value for an unhandled exception that caused the process
178d697e30cSYuri Pudgorodskiy * to terminate.
179d697e30cSYuri Pudgorodskiy * See MSDN for GetExitCodeProcess() and ntstatus.h for possible
180d697e30cSYuri Pudgorodskiy * well-known codes, e.g. C0000005 ACCESS_DENIED - analog of SIGSEGV
181d697e30cSYuri Pudgorodskiy * References:
182d697e30cSYuri Pudgorodskiy * https://msdn.microsoft.com/en-us/library/windows/desktop/ms683189(v=vs.85).aspx
183d697e30cSYuri Pudgorodskiy * https://msdn.microsoft.com/en-us/library/aa260331(v=vs.60).aspx
184d697e30cSYuri Pudgorodskiy */
185d697e30cSYuri Pudgorodskiy #ifdef G_OS_WIN32
186d697e30cSYuri Pudgorodskiy /* Additionally WIN32 does not provide any additional information
187d697e30cSYuri Pudgorodskiy * on whether the child exited or terminated via signal.
188cb8d4c8fSStefan Weil * We use this simple range check to distinguish application exit code
189d697e30cSYuri Pudgorodskiy * (usually value less then 256) and unhandled exception code with
190d697e30cSYuri Pudgorodskiy * ntstatus (always value greater then 0xC0000005). */
191d697e30cSYuri Pudgorodskiy if ((uint32_t)gei->status < 0xC0000000U) {
192d697e30cSYuri Pudgorodskiy ges->has_exitcode = true;
193d697e30cSYuri Pudgorodskiy ges->exitcode = gei->status;
194d697e30cSYuri Pudgorodskiy } else {
195d697e30cSYuri Pudgorodskiy ges->has_signal = true;
196d697e30cSYuri Pudgorodskiy ges->signal = gei->status;
197d697e30cSYuri Pudgorodskiy }
198d697e30cSYuri Pudgorodskiy #else
199d697e30cSYuri Pudgorodskiy if (WIFEXITED(gei->status)) {
200d697e30cSYuri Pudgorodskiy ges->has_exitcode = true;
201d697e30cSYuri Pudgorodskiy ges->exitcode = WEXITSTATUS(gei->status);
202d697e30cSYuri Pudgorodskiy } else if (WIFSIGNALED(gei->status)) {
203d697e30cSYuri Pudgorodskiy ges->has_signal = true;
204d697e30cSYuri Pudgorodskiy ges->signal = WTERMSIG(gei->status);
205d697e30cSYuri Pudgorodskiy }
206d697e30cSYuri Pudgorodskiy #endif
207a1853dcaSYuri Pudgorodskiy if (gei->out.length > 0) {
208a1853dcaSYuri Pudgorodskiy ges->out_data = g_base64_encode(gei->out.data, gei->out.length);
209a1853dcaSYuri Pudgorodskiy ges->has_out_truncated = gei->out.truncated;
210a1853dcaSYuri Pudgorodskiy }
211d6f67b83SDaniel Xu g_free(gei->out.data);
212a1853dcaSYuri Pudgorodskiy
213a1853dcaSYuri Pudgorodskiy if (gei->err.length > 0) {
214a1853dcaSYuri Pudgorodskiy ges->err_data = g_base64_encode(gei->err.data, gei->err.length);
215a1853dcaSYuri Pudgorodskiy ges->has_err_truncated = gei->err.truncated;
216a1853dcaSYuri Pudgorodskiy }
217d6f67b83SDaniel Xu g_free(gei->err.data);
218a1853dcaSYuri Pudgorodskiy
219d697e30cSYuri Pudgorodskiy QTAILQ_REMOVE(&guest_exec_state.processes, gei, next);
220d697e30cSYuri Pudgorodskiy g_free(gei);
221d697e30cSYuri Pudgorodskiy }
222d697e30cSYuri Pudgorodskiy
223d697e30cSYuri Pudgorodskiy return ges;
224d697e30cSYuri Pudgorodskiy }
225d697e30cSYuri Pudgorodskiy
226d697e30cSYuri Pudgorodskiy /* Get environment variables or arguments array for execve(). */
guest_exec_get_args(const strList * entry,bool log)227d697e30cSYuri Pudgorodskiy static char **guest_exec_get_args(const strList *entry, bool log)
228d697e30cSYuri Pudgorodskiy {
229d697e30cSYuri Pudgorodskiy const strList *it;
230d697e30cSYuri Pudgorodskiy int count = 1, i = 0; /* reserve for NULL terminator */
231d697e30cSYuri Pudgorodskiy char **args;
232d697e30cSYuri Pudgorodskiy char *str; /* for logging array of arguments */
233d697e30cSYuri Pudgorodskiy size_t str_size = 1;
234d697e30cSYuri Pudgorodskiy
235d697e30cSYuri Pudgorodskiy for (it = entry; it != NULL; it = it->next) {
236d697e30cSYuri Pudgorodskiy count++;
237d697e30cSYuri Pudgorodskiy str_size += 1 + strlen(it->value);
238d697e30cSYuri Pudgorodskiy }
239d697e30cSYuri Pudgorodskiy
240d697e30cSYuri Pudgorodskiy str = g_malloc(str_size);
241d697e30cSYuri Pudgorodskiy *str = 0;
242b21e2380SMarkus Armbruster args = g_new(char *, count);
243d697e30cSYuri Pudgorodskiy for (it = entry; it != NULL; it = it->next) {
244d697e30cSYuri Pudgorodskiy args[i++] = it->value;
245d697e30cSYuri Pudgorodskiy pstrcat(str, str_size, it->value);
246d697e30cSYuri Pudgorodskiy if (it->next) {
247d697e30cSYuri Pudgorodskiy pstrcat(str, str_size, " ");
248d697e30cSYuri Pudgorodskiy }
249d697e30cSYuri Pudgorodskiy }
250d697e30cSYuri Pudgorodskiy args[i] = NULL;
251d697e30cSYuri Pudgorodskiy
252d697e30cSYuri Pudgorodskiy if (log) {
253d697e30cSYuri Pudgorodskiy slog("guest-exec called: \"%s\"", str);
254d697e30cSYuri Pudgorodskiy }
255d697e30cSYuri Pudgorodskiy g_free(str);
256d697e30cSYuri Pudgorodskiy
257d697e30cSYuri Pudgorodskiy return args;
258d697e30cSYuri Pudgorodskiy }
259d697e30cSYuri Pudgorodskiy
guest_exec_child_watch(GPid pid,gint status,gpointer data)260d697e30cSYuri Pudgorodskiy static void guest_exec_child_watch(GPid pid, gint status, gpointer data)
261d697e30cSYuri Pudgorodskiy {
262d697e30cSYuri Pudgorodskiy GuestExecInfo *gei = (GuestExecInfo *)data;
263d697e30cSYuri Pudgorodskiy
264d697e30cSYuri Pudgorodskiy g_debug("guest_exec_child_watch called, pid: %d, status: %u",
265d697e30cSYuri Pudgorodskiy (int32_t)gpid_to_int64(pid), (uint32_t)status);
266d697e30cSYuri Pudgorodskiy
267d697e30cSYuri Pudgorodskiy gei->status = status;
268c267d750SMarc-André Lureau gei->finished = true;
269d697e30cSYuri Pudgorodskiy
270d697e30cSYuri Pudgorodskiy g_spawn_close_pid(pid);
271d697e30cSYuri Pudgorodskiy }
272d697e30cSYuri Pudgorodskiy
guest_exec_task_setup(gpointer data)2734005b473SDenis V. Lunev static void guest_exec_task_setup(gpointer data)
2744005b473SDenis V. Lunev {
2754005b473SDenis V. Lunev #if !defined(G_OS_WIN32)
276810f677aSDaniel Xu bool has_merge = *(bool *)data;
2774005b473SDenis V. Lunev struct sigaction sigact;
2784005b473SDenis V. Lunev
279810f677aSDaniel Xu if (has_merge) {
280810f677aSDaniel Xu /*
281810f677aSDaniel Xu * FIXME: When `GLIB_VERSION_MIN_REQUIRED` is bumped to 2.58+, use
282810f677aSDaniel Xu * g_spawn_async_with_fds() to be portable on windows. The current
283810f677aSDaniel Xu * logic does not work on windows b/c `GSpawnChildSetupFunc` is run
284810f677aSDaniel Xu * inside the parent, not the child.
285810f677aSDaniel Xu */
286810f677aSDaniel Xu if (dup2(STDOUT_FILENO, STDERR_FILENO) != 0) {
287810f677aSDaniel Xu slog("dup2() failed to merge stderr into stdout: %s",
288810f677aSDaniel Xu strerror(errno));
289810f677aSDaniel Xu }
290810f677aSDaniel Xu }
291810f677aSDaniel Xu
292810f677aSDaniel Xu /* Reset ignored signals back to default. */
2934005b473SDenis V. Lunev memset(&sigact, 0, sizeof(struct sigaction));
2944005b473SDenis V. Lunev sigact.sa_handler = SIG_DFL;
2954005b473SDenis V. Lunev
2964005b473SDenis V. Lunev if (sigaction(SIGPIPE, &sigact, NULL) != 0) {
2974005b473SDenis V. Lunev slog("sigaction() failed to reset child process's SIGPIPE: %s",
2984005b473SDenis V. Lunev strerror(errno));
2994005b473SDenis V. Lunev }
3004005b473SDenis V. Lunev #endif
3014005b473SDenis V. Lunev }
3024005b473SDenis V. Lunev
guest_exec_input_watch(GIOChannel * ch,GIOCondition cond,gpointer p_)303a1853dcaSYuri Pudgorodskiy static gboolean guest_exec_input_watch(GIOChannel *ch,
304a1853dcaSYuri Pudgorodskiy GIOCondition cond, gpointer p_)
305a1853dcaSYuri Pudgorodskiy {
306a1853dcaSYuri Pudgorodskiy GuestExecIOData *p = (GuestExecIOData *)p_;
307a1853dcaSYuri Pudgorodskiy gsize bytes_written = 0;
308a1853dcaSYuri Pudgorodskiy GIOStatus status;
309a1853dcaSYuri Pudgorodskiy GError *gerr = NULL;
310a1853dcaSYuri Pudgorodskiy
311a1853dcaSYuri Pudgorodskiy /* nothing left to write */
312a1853dcaSYuri Pudgorodskiy if (p->size == p->length) {
313a1853dcaSYuri Pudgorodskiy goto done;
314a1853dcaSYuri Pudgorodskiy }
315a1853dcaSYuri Pudgorodskiy
316a1853dcaSYuri Pudgorodskiy status = g_io_channel_write_chars(ch, (gchar *)p->data + p->length,
317a1853dcaSYuri Pudgorodskiy p->size - p->length, &bytes_written, &gerr);
318a1853dcaSYuri Pudgorodskiy
319a1853dcaSYuri Pudgorodskiy /* can be not 0 even if not G_IO_STATUS_NORMAL */
320a1853dcaSYuri Pudgorodskiy if (bytes_written != 0) {
321a1853dcaSYuri Pudgorodskiy p->length += bytes_written;
322a1853dcaSYuri Pudgorodskiy }
323a1853dcaSYuri Pudgorodskiy
324a1853dcaSYuri Pudgorodskiy /* continue write, our callback will be called again */
325a1853dcaSYuri Pudgorodskiy if (status == G_IO_STATUS_NORMAL || status == G_IO_STATUS_AGAIN) {
326a1853dcaSYuri Pudgorodskiy return true;
327a1853dcaSYuri Pudgorodskiy }
328a1853dcaSYuri Pudgorodskiy
329a1853dcaSYuri Pudgorodskiy if (gerr) {
330a1853dcaSYuri Pudgorodskiy g_warning("qga: i/o error writing to input_data channel: %s",
331a1853dcaSYuri Pudgorodskiy gerr->message);
332a1853dcaSYuri Pudgorodskiy g_error_free(gerr);
333a1853dcaSYuri Pudgorodskiy }
334a1853dcaSYuri Pudgorodskiy
335a1853dcaSYuri Pudgorodskiy done:
336a1853dcaSYuri Pudgorodskiy g_io_channel_shutdown(ch, true, NULL);
337a1853dcaSYuri Pudgorodskiy g_io_channel_unref(ch);
338c267d750SMarc-André Lureau p->closed = true;
339a1853dcaSYuri Pudgorodskiy g_free(p->data);
340a1853dcaSYuri Pudgorodskiy
341a1853dcaSYuri Pudgorodskiy return false;
342a1853dcaSYuri Pudgorodskiy }
343a1853dcaSYuri Pudgorodskiy
guest_exec_output_watch(GIOChannel * ch,GIOCondition cond,gpointer p_)344a1853dcaSYuri Pudgorodskiy static gboolean guest_exec_output_watch(GIOChannel *ch,
345a1853dcaSYuri Pudgorodskiy GIOCondition cond, gpointer p_)
346a1853dcaSYuri Pudgorodskiy {
347a1853dcaSYuri Pudgorodskiy GuestExecIOData *p = (GuestExecIOData *)p_;
348a1853dcaSYuri Pudgorodskiy gsize bytes_read;
349a1853dcaSYuri Pudgorodskiy GIOStatus gstatus;
350a1853dcaSYuri Pudgorodskiy
351a1853dcaSYuri Pudgorodskiy if (cond == G_IO_HUP || cond == G_IO_ERR) {
352a1853dcaSYuri Pudgorodskiy goto close;
353a1853dcaSYuri Pudgorodskiy }
354a1853dcaSYuri Pudgorodskiy
355a1853dcaSYuri Pudgorodskiy if (p->size == p->length) {
356a1853dcaSYuri Pudgorodskiy gpointer t = NULL;
357a1853dcaSYuri Pudgorodskiy if (!p->truncated && p->size < GUEST_EXEC_MAX_OUTPUT) {
358a1853dcaSYuri Pudgorodskiy t = g_try_realloc(p->data, p->size + GUEST_EXEC_IO_SIZE);
359a1853dcaSYuri Pudgorodskiy }
360a1853dcaSYuri Pudgorodskiy if (t == NULL) {
361a1853dcaSYuri Pudgorodskiy /* ignore truncated output */
362a1853dcaSYuri Pudgorodskiy gchar buf[GUEST_EXEC_IO_SIZE];
363a1853dcaSYuri Pudgorodskiy
364a1853dcaSYuri Pudgorodskiy p->truncated = true;
365a1853dcaSYuri Pudgorodskiy gstatus = g_io_channel_read_chars(ch, buf, sizeof(buf),
366a1853dcaSYuri Pudgorodskiy &bytes_read, NULL);
367a1853dcaSYuri Pudgorodskiy if (gstatus == G_IO_STATUS_EOF || gstatus == G_IO_STATUS_ERROR) {
368a1853dcaSYuri Pudgorodskiy goto close;
369a1853dcaSYuri Pudgorodskiy }
370a1853dcaSYuri Pudgorodskiy
371a1853dcaSYuri Pudgorodskiy return true;
372a1853dcaSYuri Pudgorodskiy }
373a1853dcaSYuri Pudgorodskiy p->size += GUEST_EXEC_IO_SIZE;
374a1853dcaSYuri Pudgorodskiy p->data = t;
375a1853dcaSYuri Pudgorodskiy }
376a1853dcaSYuri Pudgorodskiy
377a1853dcaSYuri Pudgorodskiy /* Calling read API once.
378a1853dcaSYuri Pudgorodskiy * On next available data our callback will be called again */
379a1853dcaSYuri Pudgorodskiy gstatus = g_io_channel_read_chars(ch, (gchar *)p->data + p->length,
380a1853dcaSYuri Pudgorodskiy p->size - p->length, &bytes_read, NULL);
381a1853dcaSYuri Pudgorodskiy if (gstatus == G_IO_STATUS_EOF || gstatus == G_IO_STATUS_ERROR) {
382a1853dcaSYuri Pudgorodskiy goto close;
383a1853dcaSYuri Pudgorodskiy }
384a1853dcaSYuri Pudgorodskiy
385a1853dcaSYuri Pudgorodskiy p->length += bytes_read;
386a1853dcaSYuri Pudgorodskiy
387a1853dcaSYuri Pudgorodskiy return true;
388a1853dcaSYuri Pudgorodskiy
389a1853dcaSYuri Pudgorodskiy close:
3903005c2c2SYuriy Pudgorodskiy g_io_channel_shutdown(ch, true, NULL);
391a1853dcaSYuri Pudgorodskiy g_io_channel_unref(ch);
392c267d750SMarc-André Lureau p->closed = true;
393a1853dcaSYuri Pudgorodskiy return false;
394a1853dcaSYuri Pudgorodskiy }
395a1853dcaSYuri Pudgorodskiy
ga_parse_capture_output(GuestExecCaptureOutput * capture_output)3969c5ccc52SDaniel Xu static GuestExecCaptureOutputMode ga_parse_capture_output(
3979c5ccc52SDaniel Xu GuestExecCaptureOutput *capture_output)
3989c5ccc52SDaniel Xu {
3999c5ccc52SDaniel Xu if (!capture_output)
4009c5ccc52SDaniel Xu return GUEST_EXEC_CAPTURE_OUTPUT_MODE_NONE;
4019c5ccc52SDaniel Xu else if (capture_output->type == QTYPE_QBOOL)
4029c5ccc52SDaniel Xu return capture_output->u.flag ? GUEST_EXEC_CAPTURE_OUTPUT_MODE_SEPARATED
4039c5ccc52SDaniel Xu : GUEST_EXEC_CAPTURE_OUTPUT_MODE_NONE;
4049c5ccc52SDaniel Xu else
4059c5ccc52SDaniel Xu return capture_output->u.mode;
4069c5ccc52SDaniel Xu }
4079c5ccc52SDaniel Xu
qmp_guest_exec(const char * path,bool has_arg,strList * arg,bool has_env,strList * env,const char * input_data,GuestExecCaptureOutput * capture_output,Error ** errp)408d697e30cSYuri Pudgorodskiy GuestExec *qmp_guest_exec(const char *path,
409d697e30cSYuri Pudgorodskiy bool has_arg, strList *arg,
410d697e30cSYuri Pudgorodskiy bool has_env, strList *env,
41191eab32aSMarkus Armbruster const char *input_data,
4129c5ccc52SDaniel Xu GuestExecCaptureOutput *capture_output,
413b90abbacSVladimir Sementsov-Ogievskiy Error **errp)
414d697e30cSYuri Pudgorodskiy {
415d697e30cSYuri Pudgorodskiy GPid pid;
416d697e30cSYuri Pudgorodskiy GuestExec *ge = NULL;
417d697e30cSYuri Pudgorodskiy GuestExecInfo *gei;
418d697e30cSYuri Pudgorodskiy char **argv, **envp;
419d697e30cSYuri Pudgorodskiy strList arglist;
420d697e30cSYuri Pudgorodskiy gboolean ret;
421d697e30cSYuri Pudgorodskiy GError *gerr = NULL;
422a1853dcaSYuri Pudgorodskiy gint in_fd, out_fd, err_fd;
423a1853dcaSYuri Pudgorodskiy GIOChannel *in_ch, *out_ch, *err_ch;
424a1853dcaSYuri Pudgorodskiy GSpawnFlags flags;
4259c5ccc52SDaniel Xu bool has_output = false;
426810f677aSDaniel Xu bool has_merge = false;
4279c5ccc52SDaniel Xu GuestExecCaptureOutputMode output_mode;
428057489ddSDaniel P. Berrangé g_autofree uint8_t *input = NULL;
429920639caSDaniel P. Berrange size_t ninput = 0;
430d697e30cSYuri Pudgorodskiy
431d697e30cSYuri Pudgorodskiy arglist.value = (char *)path;
432d697e30cSYuri Pudgorodskiy arglist.next = has_arg ? arg : NULL;
433d697e30cSYuri Pudgorodskiy
43491eab32aSMarkus Armbruster if (input_data) {
435b90abbacSVladimir Sementsov-Ogievskiy input = qbase64_decode(input_data, -1, &ninput, errp);
436920639caSDaniel P. Berrange if (!input) {
437920639caSDaniel P. Berrange return NULL;
438920639caSDaniel P. Berrange }
439920639caSDaniel P. Berrange }
440920639caSDaniel P. Berrange
441d697e30cSYuri Pudgorodskiy argv = guest_exec_get_args(&arglist, true);
44202a4d82eSYuri Pudgorodskiy envp = has_env ? guest_exec_get_args(env, false) : NULL;
443d697e30cSYuri Pudgorodskiy
444e7b3af81SDaniel P. Berrangé flags = G_SPAWN_SEARCH_PATH | G_SPAWN_DO_NOT_REAP_CHILD |
445e7b3af81SDaniel P. Berrangé G_SPAWN_SEARCH_PATH_FROM_ENVP;
4469c5ccc52SDaniel Xu
4479c5ccc52SDaniel Xu output_mode = ga_parse_capture_output(capture_output);
4489c5ccc52SDaniel Xu switch (output_mode) {
4499c5ccc52SDaniel Xu case GUEST_EXEC_CAPTURE_OUTPUT_MODE_NONE:
450a1853dcaSYuri Pudgorodskiy flags |= G_SPAWN_STDOUT_TO_DEV_NULL | G_SPAWN_STDERR_TO_DEV_NULL;
4519c5ccc52SDaniel Xu break;
4529c5ccc52SDaniel Xu case GUEST_EXEC_CAPTURE_OUTPUT_MODE_STDOUT:
4539c5ccc52SDaniel Xu has_output = true;
4549c5ccc52SDaniel Xu flags |= G_SPAWN_STDERR_TO_DEV_NULL;
4559c5ccc52SDaniel Xu break;
4569c5ccc52SDaniel Xu case GUEST_EXEC_CAPTURE_OUTPUT_MODE_STDERR:
4579c5ccc52SDaniel Xu has_output = true;
4589c5ccc52SDaniel Xu flags |= G_SPAWN_STDOUT_TO_DEV_NULL;
4599c5ccc52SDaniel Xu break;
4609c5ccc52SDaniel Xu case GUEST_EXEC_CAPTURE_OUTPUT_MODE_SEPARATED:
4619c5ccc52SDaniel Xu has_output = true;
4629c5ccc52SDaniel Xu break;
463810f677aSDaniel Xu #if !defined(G_OS_WIN32)
464810f677aSDaniel Xu case GUEST_EXEC_CAPTURE_OUTPUT_MODE_MERGED:
465810f677aSDaniel Xu has_output = true;
466810f677aSDaniel Xu has_merge = true;
467810f677aSDaniel Xu break;
468810f677aSDaniel Xu #endif
4699c5ccc52SDaniel Xu case GUEST_EXEC_CAPTURE_OUTPUT_MODE__MAX:
4709c5ccc52SDaniel Xu /* Silence warning; impossible branch */
4719c5ccc52SDaniel Xu break;
472a1853dcaSYuri Pudgorodskiy }
473a1853dcaSYuri Pudgorodskiy
474a1853dcaSYuri Pudgorodskiy ret = g_spawn_async_with_pipes(NULL, argv, envp, flags,
475810f677aSDaniel Xu guest_exec_task_setup, &has_merge, &pid, input_data ? &in_fd : NULL,
476a1853dcaSYuri Pudgorodskiy has_output ? &out_fd : NULL, has_output ? &err_fd : NULL, &gerr);
477d697e30cSYuri Pudgorodskiy if (!ret) {
478b90abbacSVladimir Sementsov-Ogievskiy error_setg(errp, QERR_QGA_COMMAND_FAILED, gerr->message);
479d697e30cSYuri Pudgorodskiy g_error_free(gerr);
480d697e30cSYuri Pudgorodskiy goto done;
481d697e30cSYuri Pudgorodskiy }
482d697e30cSYuri Pudgorodskiy
483d697e30cSYuri Pudgorodskiy ge = g_new0(GuestExec, 1);
484d697e30cSYuri Pudgorodskiy ge->pid = gpid_to_int64(pid);
485d697e30cSYuri Pudgorodskiy
486d697e30cSYuri Pudgorodskiy gei = guest_exec_info_add(pid);
487a1853dcaSYuri Pudgorodskiy gei->has_output = has_output;
488d697e30cSYuri Pudgorodskiy g_child_watch_add(pid, guest_exec_child_watch, gei);
489d697e30cSYuri Pudgorodskiy
49091eab32aSMarkus Armbruster if (input_data) {
491057489ddSDaniel P. Berrangé gei->in.data = g_steal_pointer(&input);
492920639caSDaniel P. Berrange gei->in.size = ninput;
493a1853dcaSYuri Pudgorodskiy #ifdef G_OS_WIN32
494a1853dcaSYuri Pudgorodskiy in_ch = g_io_channel_win32_new_fd(in_fd);
495a1853dcaSYuri Pudgorodskiy #else
496a1853dcaSYuri Pudgorodskiy in_ch = g_io_channel_unix_new(in_fd);
497a1853dcaSYuri Pudgorodskiy #endif
498a1853dcaSYuri Pudgorodskiy g_io_channel_set_encoding(in_ch, NULL, NULL);
499a1853dcaSYuri Pudgorodskiy g_io_channel_set_buffered(in_ch, false);
500a1853dcaSYuri Pudgorodskiy g_io_channel_set_flags(in_ch, G_IO_FLAG_NONBLOCK, NULL);
5013005c2c2SYuriy Pudgorodskiy g_io_channel_set_close_on_unref(in_ch, true);
502a1853dcaSYuri Pudgorodskiy g_io_add_watch(in_ch, G_IO_OUT, guest_exec_input_watch, &gei->in);
503a1853dcaSYuri Pudgorodskiy }
504a1853dcaSYuri Pudgorodskiy
505a1853dcaSYuri Pudgorodskiy if (has_output) {
506a1853dcaSYuri Pudgorodskiy #ifdef G_OS_WIN32
507a1853dcaSYuri Pudgorodskiy out_ch = g_io_channel_win32_new_fd(out_fd);
508a1853dcaSYuri Pudgorodskiy err_ch = g_io_channel_win32_new_fd(err_fd);
509a1853dcaSYuri Pudgorodskiy #else
510a1853dcaSYuri Pudgorodskiy out_ch = g_io_channel_unix_new(out_fd);
511a1853dcaSYuri Pudgorodskiy err_ch = g_io_channel_unix_new(err_fd);
512a1853dcaSYuri Pudgorodskiy #endif
513a1853dcaSYuri Pudgorodskiy g_io_channel_set_encoding(out_ch, NULL, NULL);
514a1853dcaSYuri Pudgorodskiy g_io_channel_set_encoding(err_ch, NULL, NULL);
515a1853dcaSYuri Pudgorodskiy g_io_channel_set_buffered(out_ch, false);
516a1853dcaSYuri Pudgorodskiy g_io_channel_set_buffered(err_ch, false);
5173005c2c2SYuriy Pudgorodskiy g_io_channel_set_close_on_unref(out_ch, true);
5183005c2c2SYuriy Pudgorodskiy g_io_channel_set_close_on_unref(err_ch, true);
519a1853dcaSYuri Pudgorodskiy g_io_add_watch(out_ch, G_IO_IN | G_IO_HUP,
520a1853dcaSYuri Pudgorodskiy guest_exec_output_watch, &gei->out);
521a1853dcaSYuri Pudgorodskiy g_io_add_watch(err_ch, G_IO_IN | G_IO_HUP,
522a1853dcaSYuri Pudgorodskiy guest_exec_output_watch, &gei->err);
523a1853dcaSYuri Pudgorodskiy }
524a1853dcaSYuri Pudgorodskiy
525d697e30cSYuri Pudgorodskiy done:
526d697e30cSYuri Pudgorodskiy g_free(argv);
527d697e30cSYuri Pudgorodskiy g_free(envp);
528d697e30cSYuri Pudgorodskiy
529d697e30cSYuri Pudgorodskiy return ge;
530d697e30cSYuri Pudgorodskiy }
5310b4b4938SEric Blake
5320b4b4938SEric Blake /* Convert GuestFileWhence (either a raw integer or an enum value) into
5330b4b4938SEric Blake * the guest's SEEK_ constants. */
ga_parse_whence(GuestFileWhence * whence,Error ** errp)5340b4b4938SEric Blake int ga_parse_whence(GuestFileWhence *whence, Error **errp)
5350b4b4938SEric Blake {
536a23f38a7SEric Blake /*
537a23f38a7SEric Blake * Exploit the fact that we picked values to match QGA_SEEK_*;
538a23f38a7SEric Blake * however, we have to use a temporary variable since the union
539a23f38a7SEric Blake * members may have different size.
540a23f38a7SEric Blake */
5410b4b4938SEric Blake if (whence->type == QTYPE_QSTRING) {
542a23f38a7SEric Blake int value = whence->u.name;
54301b2ffceSMarc-André Lureau whence->type = QTYPE_QNUM;
544a23f38a7SEric Blake whence->u.value = value;
5450b4b4938SEric Blake }
5460b4b4938SEric Blake switch (whence->u.value) {
5470b4b4938SEric Blake case QGA_SEEK_SET:
5480b4b4938SEric Blake return SEEK_SET;
5490b4b4938SEric Blake case QGA_SEEK_CUR:
5500b4b4938SEric Blake return SEEK_CUR;
5510b4b4938SEric Blake case QGA_SEEK_END:
5520b4b4938SEric Blake return SEEK_END;
5530b4b4938SEric Blake }
5540b4b4938SEric Blake error_setg(errp, "invalid whence code %"PRId64, whence->u.value);
5550b4b4938SEric Blake return -1;
5560b4b4938SEric Blake }
5570a3d197aSVinzenz Feenstra
qmp_guest_get_host_name(Error ** errp)558b90abbacSVladimir Sementsov-Ogievskiy GuestHostName *qmp_guest_get_host_name(Error **errp)
5590a3d197aSVinzenz Feenstra {
5600a3d197aSVinzenz Feenstra GuestHostName *result = NULL;
561548fb0daSMarc-André Lureau g_autofree char *hostname = qga_get_host_name(errp);
5620d3a8f32SMichal Privoznik
5630d3a8f32SMichal Privoznik /*
5640d3a8f32SMichal Privoznik * We want to avoid using g_get_host_name() because that
5650d3a8f32SMichal Privoznik * caches the result and we wouldn't reflect changes in the
5660d3a8f32SMichal Privoznik * host name.
5670d3a8f32SMichal Privoznik */
5680d3a8f32SMichal Privoznik
5690d3a8f32SMichal Privoznik if (!hostname) {
5700d3a8f32SMichal Privoznik hostname = g_strdup("localhost");
5710a3d197aSVinzenz Feenstra }
5720d3a8f32SMichal Privoznik
5730d3a8f32SMichal Privoznik result = g_new0(GuestHostName, 1);
5740d3a8f32SMichal Privoznik result->host_name = g_steal_pointer(&hostname);
5750a3d197aSVinzenz Feenstra return result;
5760a3d197aSVinzenz Feenstra }
57753c58e64SVinzenz Feenstra
qmp_guest_get_timezone(Error ** errp)57853c58e64SVinzenz Feenstra GuestTimezone *qmp_guest_get_timezone(Error **errp)
57953c58e64SVinzenz Feenstra {
58053c58e64SVinzenz Feenstra GuestTimezone *info = NULL;
58153c58e64SVinzenz Feenstra GTimeZone *tz = NULL;
58253c58e64SVinzenz Feenstra gint64 now = 0;
58353c58e64SVinzenz Feenstra gint32 intv = 0;
58453c58e64SVinzenz Feenstra gchar const *name = NULL;
58553c58e64SVinzenz Feenstra
58653c58e64SVinzenz Feenstra info = g_new0(GuestTimezone, 1);
58753c58e64SVinzenz Feenstra tz = g_time_zone_new_local();
58853c58e64SVinzenz Feenstra if (tz == NULL) {
58953c58e64SVinzenz Feenstra error_setg(errp, QERR_QGA_COMMAND_FAILED,
59053c58e64SVinzenz Feenstra "Couldn't retrieve local timezone");
59153c58e64SVinzenz Feenstra goto error;
59253c58e64SVinzenz Feenstra }
59353c58e64SVinzenz Feenstra
59453c58e64SVinzenz Feenstra now = g_get_real_time() / G_USEC_PER_SEC;
59553c58e64SVinzenz Feenstra intv = g_time_zone_find_interval(tz, G_TIME_TYPE_UNIVERSAL, now);
59653c58e64SVinzenz Feenstra info->offset = g_time_zone_get_offset(tz, intv);
59753c58e64SVinzenz Feenstra name = g_time_zone_get_abbreviation(tz, intv);
59853c58e64SVinzenz Feenstra if (name != NULL) {
59953c58e64SVinzenz Feenstra info->zone = g_strdup(name);
60053c58e64SVinzenz Feenstra }
60153c58e64SVinzenz Feenstra g_time_zone_unref(tz);
60253c58e64SVinzenz Feenstra
60353c58e64SVinzenz Feenstra return info;
60453c58e64SVinzenz Feenstra
60553c58e64SVinzenz Feenstra error:
60653c58e64SVinzenz Feenstra g_free(info);
60753c58e64SVinzenz Feenstra return NULL;
60853c58e64SVinzenz Feenstra }
609ead83a13SPhilippe Mathieu-Daudé
qmp_guest_file_read(int64_t handle,bool has_count,int64_t count,Error ** errp)610ead83a13SPhilippe Mathieu-Daudé GuestFileRead *qmp_guest_file_read(int64_t handle, bool has_count,
611ead83a13SPhilippe Mathieu-Daudé int64_t count, Error **errp)
612ead83a13SPhilippe Mathieu-Daudé {
613ead83a13SPhilippe Mathieu-Daudé GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
614ead83a13SPhilippe Mathieu-Daudé GuestFileRead *read_data;
615ead83a13SPhilippe Mathieu-Daudé
616ead83a13SPhilippe Mathieu-Daudé if (!gfh) {
617ead83a13SPhilippe Mathieu-Daudé return NULL;
618ead83a13SPhilippe Mathieu-Daudé }
619ead83a13SPhilippe Mathieu-Daudé if (!has_count) {
620ead83a13SPhilippe Mathieu-Daudé count = QGA_READ_COUNT_DEFAULT;
6211329651fSPhilippe Mathieu-Daudé } else if (count < 0 || count > GUEST_FILE_READ_COUNT_MAX) {
622ead83a13SPhilippe Mathieu-Daudé error_setg(errp, "value '%" PRId64 "' is invalid for argument count",
623ead83a13SPhilippe Mathieu-Daudé count);
624ead83a13SPhilippe Mathieu-Daudé return NULL;
625ead83a13SPhilippe Mathieu-Daudé }
626ead83a13SPhilippe Mathieu-Daudé
627ead83a13SPhilippe Mathieu-Daudé read_data = guest_file_read_unsafe(gfh, count, errp);
628ead83a13SPhilippe Mathieu-Daudé if (!read_data) {
629ead83a13SPhilippe Mathieu-Daudé slog("guest-file-write failed, handle: %" PRId64, handle);
630ead83a13SPhilippe Mathieu-Daudé }
631ead83a13SPhilippe Mathieu-Daudé
632ead83a13SPhilippe Mathieu-Daudé return read_data;
633ead83a13SPhilippe Mathieu-Daudé }
634287698e5SMarc-André Lureau
qmp_guest_get_time(Error ** errp)635287698e5SMarc-André Lureau int64_t qmp_guest_get_time(Error **errp)
636287698e5SMarc-André Lureau {
637287698e5SMarc-André Lureau return g_get_real_time() * 1000;
638287698e5SMarc-André Lureau }
639