xref: /openbmc/qemu/qga/commands-posix.c (revision 895b00f6)
1c216e5adSMichael Roth /*
2c216e5adSMichael Roth  * QEMU Guest Agent POSIX-specific command implementations
3c216e5adSMichael Roth  *
4c216e5adSMichael Roth  * Copyright IBM Corp. 2011
5c216e5adSMichael Roth  *
6c216e5adSMichael Roth  * Authors:
7c216e5adSMichael Roth  *  Michael Roth      <mdroth@linux.vnet.ibm.com>
83424fc9fSMichal Privoznik  *  Michal Privoznik  <mprivozn@redhat.com>
9c216e5adSMichael Roth  *
10c216e5adSMichael Roth  * This work is licensed under the terms of the GNU GPL, version 2 or later.
11c216e5adSMichael Roth  * See the COPYING file in the top-level directory.
12c216e5adSMichael Roth  */
13c216e5adSMichael Roth 
14c216e5adSMichael Roth #include <glib.h>
15c216e5adSMichael Roth #include <sys/types.h>
16c216e5adSMichael Roth #include <sys/ioctl.h>
172c02cbf6SLuiz Capitulino #include <sys/wait.h>
18d2baff62SLaszlo Ersek #include <unistd.h>
19d2baff62SLaszlo Ersek #include <errno.h>
20d2baff62SLaszlo Ersek #include <fcntl.h>
2146d4c572STomoki Sekiyama #include <dirent.h>
22c689b4f1SLaszlo Ersek #include <stdio.h>
23c689b4f1SLaszlo Ersek #include <string.h>
24c689b4f1SLaszlo Ersek #include <sys/stat.h>
25d2baff62SLaszlo Ersek #include <inttypes.h>
26c216e5adSMichael Roth #include "qga/guest-agent-core.h"
27c216e5adSMichael Roth #include "qga-qmp-commands.h"
287b1b5d19SPaolo Bonzini #include "qapi/qmp/qerror.h"
291de7afc9SPaolo Bonzini #include "qemu/queue.h"
301de7afc9SPaolo Bonzini #include "qemu/host-utils.h"
3112505396SDenis V. Lunev #include "qemu/sockets.h"
32c216e5adSMichael Roth 
332c02cbf6SLuiz Capitulino #ifndef CONFIG_HAS_ENVIRON
34eecae147SAndreas Färber #ifdef __APPLE__
35eecae147SAndreas Färber #include <crt_externs.h>
36eecae147SAndreas Färber #define environ (*_NSGetEnviron())
37eecae147SAndreas Färber #else
382c02cbf6SLuiz Capitulino extern char **environ;
392c02cbf6SLuiz Capitulino #endif
40eecae147SAndreas Färber #endif
412c02cbf6SLuiz Capitulino 
42e72c3f2eSMichael Roth #if defined(__linux__)
43e72c3f2eSMichael Roth #include <mntent.h>
44e72c3f2eSMichael Roth #include <linux/fs.h>
45e72c3f2eSMichael Roth #include <ifaddrs.h>
46e72c3f2eSMichael Roth #include <arpa/inet.h>
47e72c3f2eSMichael Roth #include <sys/socket.h>
48e72c3f2eSMichael Roth #include <net/if.h>
49e72c3f2eSMichael Roth 
50eab5fd59SPaolo Bonzini #ifdef FIFREEZE
51e72c3f2eSMichael Roth #define CONFIG_FSFREEZE
52e72c3f2eSMichael Roth #endif
53eab5fd59SPaolo Bonzini #ifdef FITRIM
54eab5fd59SPaolo Bonzini #define CONFIG_FSTRIM
55eab5fd59SPaolo Bonzini #endif
56e72c3f2eSMichael Roth #endif
57e72c3f2eSMichael Roth 
5877dbc81bSMarkus Armbruster static void ga_wait_child(pid_t pid, int *status, Error **errp)
59d220a6dfSLuiz Capitulino {
60d220a6dfSLuiz Capitulino     pid_t rpid;
61d220a6dfSLuiz Capitulino 
62d220a6dfSLuiz Capitulino     *status = 0;
63d220a6dfSLuiz Capitulino 
64d220a6dfSLuiz Capitulino     do {
65d220a6dfSLuiz Capitulino         rpid = waitpid(pid, status, 0);
66d220a6dfSLuiz Capitulino     } while (rpid == -1 && errno == EINTR);
67d220a6dfSLuiz Capitulino 
68d220a6dfSLuiz Capitulino     if (rpid == -1) {
6977dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to wait for child (pid: %d)",
7077dbc81bSMarkus Armbruster                          pid);
71d220a6dfSLuiz Capitulino         return;
72d220a6dfSLuiz Capitulino     }
73d220a6dfSLuiz Capitulino 
74d220a6dfSLuiz Capitulino     g_assert(rpid == pid);
75d220a6dfSLuiz Capitulino }
76d220a6dfSLuiz Capitulino 
7777dbc81bSMarkus Armbruster void qmp_guest_shutdown(bool has_mode, const char *mode, Error **errp)
78c216e5adSMichael Roth {
79c216e5adSMichael Roth     const char *shutdown_flag;
80d220a6dfSLuiz Capitulino     Error *local_err = NULL;
81d220a6dfSLuiz Capitulino     pid_t pid;
823674838cSLuiz Capitulino     int status;
83c216e5adSMichael Roth 
84c216e5adSMichael Roth     slog("guest-shutdown called, mode: %s", mode);
85c216e5adSMichael Roth     if (!has_mode || strcmp(mode, "powerdown") == 0) {
86c216e5adSMichael Roth         shutdown_flag = "-P";
87c216e5adSMichael Roth     } else if (strcmp(mode, "halt") == 0) {
88c216e5adSMichael Roth         shutdown_flag = "-H";
89c216e5adSMichael Roth     } else if (strcmp(mode, "reboot") == 0) {
90c216e5adSMichael Roth         shutdown_flag = "-r";
91c216e5adSMichael Roth     } else {
9277dbc81bSMarkus Armbruster         error_setg(errp,
93d220a6dfSLuiz Capitulino                    "mode is invalid (valid values are: halt|powerdown|reboot");
94c216e5adSMichael Roth         return;
95c216e5adSMichael Roth     }
96c216e5adSMichael Roth 
97d5dd3498SLuiz Capitulino     pid = fork();
98d5dd3498SLuiz Capitulino     if (pid == 0) {
99c216e5adSMichael Roth         /* child, start the shutdown */
100c216e5adSMichael Roth         setsid();
1013674838cSLuiz Capitulino         reopen_fd_to_null(0);
1023674838cSLuiz Capitulino         reopen_fd_to_null(1);
1033674838cSLuiz Capitulino         reopen_fd_to_null(2);
104c216e5adSMichael Roth 
105485e741cSwhitearchey         execle("/sbin/shutdown", "shutdown", "-h", shutdown_flag, "+0",
1063674838cSLuiz Capitulino                "hypervisor initiated shutdown", (char*)NULL, environ);
1073674838cSLuiz Capitulino         _exit(EXIT_FAILURE);
108d5dd3498SLuiz Capitulino     } else if (pid < 0) {
10977dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
110d5dd3498SLuiz Capitulino         return;
111d5dd3498SLuiz Capitulino     }
112d5dd3498SLuiz Capitulino 
113d220a6dfSLuiz Capitulino     ga_wait_child(pid, &status, &local_err);
11484d18f06SMarkus Armbruster     if (local_err) {
11577dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
116d220a6dfSLuiz Capitulino         return;
117d220a6dfSLuiz Capitulino     }
118d220a6dfSLuiz Capitulino 
119d220a6dfSLuiz Capitulino     if (!WIFEXITED(status)) {
12077dbc81bSMarkus Armbruster         error_setg(errp, "child process has terminated abnormally");
121d220a6dfSLuiz Capitulino         return;
122d220a6dfSLuiz Capitulino     }
123d220a6dfSLuiz Capitulino 
124d220a6dfSLuiz Capitulino     if (WEXITSTATUS(status)) {
12577dbc81bSMarkus Armbruster         error_setg(errp, "child process has failed to shutdown");
126d220a6dfSLuiz Capitulino         return;
127d220a6dfSLuiz Capitulino     }
128d220a6dfSLuiz Capitulino 
129085d8134SPeter Maydell     /* succeeded */
130c216e5adSMichael Roth }
131c216e5adSMichael Roth 
1326912e6a9SLei Li int64_t qmp_guest_get_time(Error **errp)
1336912e6a9SLei Li {
1346912e6a9SLei Li    int ret;
1356912e6a9SLei Li    qemu_timeval tq;
1366912e6a9SLei Li    int64_t time_ns;
1376912e6a9SLei Li 
1386912e6a9SLei Li    ret = qemu_gettimeofday(&tq);
1396912e6a9SLei Li    if (ret < 0) {
1406912e6a9SLei Li        error_setg_errno(errp, errno, "Failed to get time");
1416912e6a9SLei Li        return -1;
1426912e6a9SLei Li    }
1436912e6a9SLei Li 
1446912e6a9SLei Li    time_ns = tq.tv_sec * 1000000000LL + tq.tv_usec * 1000;
1456912e6a9SLei Li    return time_ns;
1466912e6a9SLei Li }
1476912e6a9SLei Li 
1482c958923SMichal Privoznik void qmp_guest_set_time(bool has_time, int64_t time_ns, Error **errp)
149a1bca57fSLei Li {
150a1bca57fSLei Li     int ret;
151a1bca57fSLei Li     int status;
152a1bca57fSLei Li     pid_t pid;
153a1bca57fSLei Li     Error *local_err = NULL;
154a1bca57fSLei Li     struct timeval tv;
155a1bca57fSLei Li 
1562c958923SMichal Privoznik     /* If user has passed a time, validate and set it. */
1572c958923SMichal Privoznik     if (has_time) {
15800d2f370SMarc-André Lureau         GDate date = { 0, };
15900d2f370SMarc-André Lureau 
160a1bca57fSLei Li         /* year-2038 will overflow in case time_t is 32bit */
161a1bca57fSLei Li         if (time_ns / 1000000000 != (time_t)(time_ns / 1000000000)) {
162a1bca57fSLei Li             error_setg(errp, "Time %" PRId64 " is too large", time_ns);
163a1bca57fSLei Li             return;
164a1bca57fSLei Li         }
165a1bca57fSLei Li 
166a1bca57fSLei Li         tv.tv_sec = time_ns / 1000000000;
167a1bca57fSLei Li         tv.tv_usec = (time_ns % 1000000000) / 1000;
16800d2f370SMarc-André Lureau         g_date_set_time_t(&date, tv.tv_sec);
16900d2f370SMarc-André Lureau         if (date.year < 1970 || date.year >= 2070) {
17000d2f370SMarc-André Lureau             error_setg_errno(errp, errno, "Invalid time");
17100d2f370SMarc-André Lureau             return;
17200d2f370SMarc-André Lureau         }
173a1bca57fSLei Li 
174a1bca57fSLei Li         ret = settimeofday(&tv, NULL);
175a1bca57fSLei Li         if (ret < 0) {
176a1bca57fSLei Li             error_setg_errno(errp, errno, "Failed to set time to guest");
177a1bca57fSLei Li             return;
178a1bca57fSLei Li         }
1792c958923SMichal Privoznik     }
180a1bca57fSLei Li 
1812c958923SMichal Privoznik     /* Now, if user has passed a time to set and the system time is set, we
1822c958923SMichal Privoznik      * just need to synchronize the hardware clock. However, if no time was
1832c958923SMichal Privoznik      * passed, user is requesting the opposite: set the system time from the
1841634df56SAmos Kong      * hardware clock (RTC). */
185a1bca57fSLei Li     pid = fork();
186a1bca57fSLei Li     if (pid == 0) {
187a1bca57fSLei Li         setsid();
188a1bca57fSLei Li         reopen_fd_to_null(0);
189a1bca57fSLei Li         reopen_fd_to_null(1);
190a1bca57fSLei Li         reopen_fd_to_null(2);
191a1bca57fSLei Li 
1922c958923SMichal Privoznik         /* Use '/sbin/hwclock -w' to set RTC from the system time,
1932c958923SMichal Privoznik          * or '/sbin/hwclock -s' to set the system time from RTC. */
1942c958923SMichal Privoznik         execle("/sbin/hwclock", "hwclock", has_time ? "-w" : "-s",
1952c958923SMichal Privoznik                NULL, environ);
196a1bca57fSLei Li         _exit(EXIT_FAILURE);
197a1bca57fSLei Li     } else if (pid < 0) {
198a1bca57fSLei Li         error_setg_errno(errp, errno, "failed to create child process");
199a1bca57fSLei Li         return;
200a1bca57fSLei Li     }
201a1bca57fSLei Li 
202a1bca57fSLei Li     ga_wait_child(pid, &status, &local_err);
20384d18f06SMarkus Armbruster     if (local_err) {
204a1bca57fSLei Li         error_propagate(errp, local_err);
205a1bca57fSLei Li         return;
206a1bca57fSLei Li     }
207a1bca57fSLei Li 
208a1bca57fSLei Li     if (!WIFEXITED(status)) {
209a1bca57fSLei Li         error_setg(errp, "child process has terminated abnormally");
210a1bca57fSLei Li         return;
211a1bca57fSLei Li     }
212a1bca57fSLei Li 
213a1bca57fSLei Li     if (WEXITSTATUS(status)) {
214a1bca57fSLei Li         error_setg(errp, "hwclock failed to set hardware clock to system time");
215a1bca57fSLei Li         return;
216a1bca57fSLei Li     }
217a1bca57fSLei Li }
218a1bca57fSLei Li 
219*895b00f6SMarc-André Lureau typedef enum {
220*895b00f6SMarc-André Lureau     RW_STATE_NEW,
221*895b00f6SMarc-André Lureau     RW_STATE_READING,
222*895b00f6SMarc-André Lureau     RW_STATE_WRITING,
223*895b00f6SMarc-André Lureau } RwState;
224*895b00f6SMarc-André Lureau 
225c216e5adSMichael Roth typedef struct GuestFileHandle {
226c216e5adSMichael Roth     uint64_t id;
227c216e5adSMichael Roth     FILE *fh;
228*895b00f6SMarc-André Lureau     RwState state;
229c216e5adSMichael Roth     QTAILQ_ENTRY(GuestFileHandle) next;
230c216e5adSMichael Roth } GuestFileHandle;
231c216e5adSMichael Roth 
232c216e5adSMichael Roth static struct {
233c216e5adSMichael Roth     QTAILQ_HEAD(, GuestFileHandle) filehandles;
234b4fe97c8SDenis V. Lunev } guest_file_state = {
235b4fe97c8SDenis V. Lunev     .filehandles = QTAILQ_HEAD_INITIALIZER(guest_file_state.filehandles),
236b4fe97c8SDenis V. Lunev };
237c216e5adSMichael Roth 
23839097dafSMichael Roth static int64_t guest_file_handle_add(FILE *fh, Error **errp)
239c216e5adSMichael Roth {
240c216e5adSMichael Roth     GuestFileHandle *gfh;
24139097dafSMichael Roth     int64_t handle;
24239097dafSMichael Roth 
24339097dafSMichael Roth     handle = ga_get_fd_handle(ga_state, errp);
244a903f40cSMarkus Armbruster     if (handle < 0) {
245a903f40cSMarkus Armbruster         return -1;
24639097dafSMichael Roth     }
247c216e5adSMichael Roth 
248f3a06403SMarkus Armbruster     gfh = g_new0(GuestFileHandle, 1);
24939097dafSMichael Roth     gfh->id = handle;
250c216e5adSMichael Roth     gfh->fh = fh;
251c216e5adSMichael Roth     QTAILQ_INSERT_TAIL(&guest_file_state.filehandles, gfh, next);
25239097dafSMichael Roth 
25339097dafSMichael Roth     return handle;
254c216e5adSMichael Roth }
255c216e5adSMichael Roth 
25677dbc81bSMarkus Armbruster static GuestFileHandle *guest_file_handle_find(int64_t id, Error **errp)
257c216e5adSMichael Roth {
258c216e5adSMichael Roth     GuestFileHandle *gfh;
259c216e5adSMichael Roth 
260c216e5adSMichael Roth     QTAILQ_FOREACH(gfh, &guest_file_state.filehandles, next)
261c216e5adSMichael Roth     {
262c216e5adSMichael Roth         if (gfh->id == id) {
263c216e5adSMichael Roth             return gfh;
264c216e5adSMichael Roth         }
265c216e5adSMichael Roth     }
266c216e5adSMichael Roth 
26777dbc81bSMarkus Armbruster     error_setg(errp, "handle '%" PRId64 "' has not been found", id);
268c216e5adSMichael Roth     return NULL;
269c216e5adSMichael Roth }
270c216e5adSMichael Roth 
271c689b4f1SLaszlo Ersek typedef const char * const ccpc;
272c689b4f1SLaszlo Ersek 
2738fe6bbcaSLaszlo Ersek #ifndef O_BINARY
2748fe6bbcaSLaszlo Ersek #define O_BINARY 0
2758fe6bbcaSLaszlo Ersek #endif
2768fe6bbcaSLaszlo Ersek 
277c689b4f1SLaszlo Ersek /* http://pubs.opengroup.org/onlinepubs/9699919799/functions/fopen.html */
278c689b4f1SLaszlo Ersek static const struct {
279c689b4f1SLaszlo Ersek     ccpc *forms;
280c689b4f1SLaszlo Ersek     int oflag_base;
281c689b4f1SLaszlo Ersek } guest_file_open_modes[] = {
2828fe6bbcaSLaszlo Ersek     { (ccpc[]){ "r",          NULL }, O_RDONLY                                 },
2838fe6bbcaSLaszlo Ersek     { (ccpc[]){ "rb",         NULL }, O_RDONLY                      | O_BINARY },
2848fe6bbcaSLaszlo Ersek     { (ccpc[]){ "w",          NULL }, O_WRONLY | O_CREAT | O_TRUNC             },
2858fe6bbcaSLaszlo Ersek     { (ccpc[]){ "wb",         NULL }, O_WRONLY | O_CREAT | O_TRUNC  | O_BINARY },
2868fe6bbcaSLaszlo Ersek     { (ccpc[]){ "a",          NULL }, O_WRONLY | O_CREAT | O_APPEND            },
2878fe6bbcaSLaszlo Ersek     { (ccpc[]){ "ab",         NULL }, O_WRONLY | O_CREAT | O_APPEND | O_BINARY },
2888fe6bbcaSLaszlo Ersek     { (ccpc[]){ "r+",         NULL }, O_RDWR                                   },
2898fe6bbcaSLaszlo Ersek     { (ccpc[]){ "rb+", "r+b", NULL }, O_RDWR                        | O_BINARY },
2908fe6bbcaSLaszlo Ersek     { (ccpc[]){ "w+",         NULL }, O_RDWR   | O_CREAT | O_TRUNC             },
2918fe6bbcaSLaszlo Ersek     { (ccpc[]){ "wb+", "w+b", NULL }, O_RDWR   | O_CREAT | O_TRUNC  | O_BINARY },
2928fe6bbcaSLaszlo Ersek     { (ccpc[]){ "a+",         NULL }, O_RDWR   | O_CREAT | O_APPEND            },
2938fe6bbcaSLaszlo Ersek     { (ccpc[]){ "ab+", "a+b", NULL }, O_RDWR   | O_CREAT | O_APPEND | O_BINARY }
294c689b4f1SLaszlo Ersek };
295c689b4f1SLaszlo Ersek 
296c689b4f1SLaszlo Ersek static int
29777dbc81bSMarkus Armbruster find_open_flag(const char *mode_str, Error **errp)
298c689b4f1SLaszlo Ersek {
299c689b4f1SLaszlo Ersek     unsigned mode;
300c689b4f1SLaszlo Ersek 
301c689b4f1SLaszlo Ersek     for (mode = 0; mode < ARRAY_SIZE(guest_file_open_modes); ++mode) {
302c689b4f1SLaszlo Ersek         ccpc *form;
303c689b4f1SLaszlo Ersek 
304c689b4f1SLaszlo Ersek         form = guest_file_open_modes[mode].forms;
305c689b4f1SLaszlo Ersek         while (*form != NULL && strcmp(*form, mode_str) != 0) {
306c689b4f1SLaszlo Ersek             ++form;
307c689b4f1SLaszlo Ersek         }
308c689b4f1SLaszlo Ersek         if (*form != NULL) {
309c689b4f1SLaszlo Ersek             break;
310c689b4f1SLaszlo Ersek         }
311c689b4f1SLaszlo Ersek     }
312c689b4f1SLaszlo Ersek 
313c689b4f1SLaszlo Ersek     if (mode == ARRAY_SIZE(guest_file_open_modes)) {
31477dbc81bSMarkus Armbruster         error_setg(errp, "invalid file open mode '%s'", mode_str);
315c689b4f1SLaszlo Ersek         return -1;
316c689b4f1SLaszlo Ersek     }
317c689b4f1SLaszlo Ersek     return guest_file_open_modes[mode].oflag_base | O_NOCTTY | O_NONBLOCK;
318c689b4f1SLaszlo Ersek }
319c689b4f1SLaszlo Ersek 
320c689b4f1SLaszlo Ersek #define DEFAULT_NEW_FILE_MODE (S_IRUSR | S_IWUSR | \
321c689b4f1SLaszlo Ersek                                S_IRGRP | S_IWGRP | \
322c689b4f1SLaszlo Ersek                                S_IROTH | S_IWOTH)
323c689b4f1SLaszlo Ersek 
324c689b4f1SLaszlo Ersek static FILE *
32577dbc81bSMarkus Armbruster safe_open_or_create(const char *path, const char *mode, Error **errp)
326c689b4f1SLaszlo Ersek {
327c689b4f1SLaszlo Ersek     Error *local_err = NULL;
328c689b4f1SLaszlo Ersek     int oflag;
329c689b4f1SLaszlo Ersek 
330c689b4f1SLaszlo Ersek     oflag = find_open_flag(mode, &local_err);
331c689b4f1SLaszlo Ersek     if (local_err == NULL) {
332c689b4f1SLaszlo Ersek         int fd;
333c689b4f1SLaszlo Ersek 
334c689b4f1SLaszlo Ersek         /* If the caller wants / allows creation of a new file, we implement it
335c689b4f1SLaszlo Ersek          * with a two step process: open() + (open() / fchmod()).
336c689b4f1SLaszlo Ersek          *
337c689b4f1SLaszlo Ersek          * First we insist on creating the file exclusively as a new file. If
338c689b4f1SLaszlo Ersek          * that succeeds, we're free to set any file-mode bits on it. (The
339c689b4f1SLaszlo Ersek          * motivation is that we want to set those file-mode bits independently
340c689b4f1SLaszlo Ersek          * of the current umask.)
341c689b4f1SLaszlo Ersek          *
342c689b4f1SLaszlo Ersek          * If the exclusive creation fails because the file already exists
343c689b4f1SLaszlo Ersek          * (EEXIST is not possible for any other reason), we just attempt to
344c689b4f1SLaszlo Ersek          * open the file, but in this case we won't be allowed to change the
345c689b4f1SLaszlo Ersek          * file-mode bits on the preexistent file.
346c689b4f1SLaszlo Ersek          *
347c689b4f1SLaszlo Ersek          * The pathname should never disappear between the two open()s in
348c689b4f1SLaszlo Ersek          * practice. If it happens, then someone very likely tried to race us.
349c689b4f1SLaszlo Ersek          * In this case just go ahead and report the ENOENT from the second
350c689b4f1SLaszlo Ersek          * open() to the caller.
351c689b4f1SLaszlo Ersek          *
352c689b4f1SLaszlo Ersek          * If the caller wants to open a preexistent file, then the first
353c689b4f1SLaszlo Ersek          * open() is decisive and its third argument is ignored, and the second
354c689b4f1SLaszlo Ersek          * open() and the fchmod() are never called.
355c689b4f1SLaszlo Ersek          */
356c689b4f1SLaszlo Ersek         fd = open(path, oflag | ((oflag & O_CREAT) ? O_EXCL : 0), 0);
357c689b4f1SLaszlo Ersek         if (fd == -1 && errno == EEXIST) {
358c689b4f1SLaszlo Ersek             oflag &= ~(unsigned)O_CREAT;
359c689b4f1SLaszlo Ersek             fd = open(path, oflag);
360c689b4f1SLaszlo Ersek         }
361c689b4f1SLaszlo Ersek 
362c689b4f1SLaszlo Ersek         if (fd == -1) {
363c689b4f1SLaszlo Ersek             error_setg_errno(&local_err, errno, "failed to open file '%s' "
364c689b4f1SLaszlo Ersek                              "(mode: '%s')", path, mode);
365c689b4f1SLaszlo Ersek         } else {
366c689b4f1SLaszlo Ersek             qemu_set_cloexec(fd);
367c689b4f1SLaszlo Ersek 
368c689b4f1SLaszlo Ersek             if ((oflag & O_CREAT) && fchmod(fd, DEFAULT_NEW_FILE_MODE) == -1) {
369c689b4f1SLaszlo Ersek                 error_setg_errno(&local_err, errno, "failed to set permission "
370c689b4f1SLaszlo Ersek                                  "0%03o on new file '%s' (mode: '%s')",
371c689b4f1SLaszlo Ersek                                  (unsigned)DEFAULT_NEW_FILE_MODE, path, mode);
372c689b4f1SLaszlo Ersek             } else {
373c689b4f1SLaszlo Ersek                 FILE *f;
374c689b4f1SLaszlo Ersek 
375c689b4f1SLaszlo Ersek                 f = fdopen(fd, mode);
376c689b4f1SLaszlo Ersek                 if (f == NULL) {
377c689b4f1SLaszlo Ersek                     error_setg_errno(&local_err, errno, "failed to associate "
378c689b4f1SLaszlo Ersek                                      "stdio stream with file descriptor %d, "
379c689b4f1SLaszlo Ersek                                      "file '%s' (mode: '%s')", fd, path, mode);
380c689b4f1SLaszlo Ersek                 } else {
381c689b4f1SLaszlo Ersek                     return f;
382c689b4f1SLaszlo Ersek                 }
383c689b4f1SLaszlo Ersek             }
384c689b4f1SLaszlo Ersek 
385c689b4f1SLaszlo Ersek             close(fd);
3862b720018SLaszlo Ersek             if (oflag & O_CREAT) {
3872b720018SLaszlo Ersek                 unlink(path);
3882b720018SLaszlo Ersek             }
389c689b4f1SLaszlo Ersek         }
390c689b4f1SLaszlo Ersek     }
391c689b4f1SLaszlo Ersek 
39277dbc81bSMarkus Armbruster     error_propagate(errp, local_err);
393c689b4f1SLaszlo Ersek     return NULL;
394c689b4f1SLaszlo Ersek }
395c689b4f1SLaszlo Ersek 
39677dbc81bSMarkus Armbruster int64_t qmp_guest_file_open(const char *path, bool has_mode, const char *mode,
39777dbc81bSMarkus Armbruster                             Error **errp)
398c216e5adSMichael Roth {
399c216e5adSMichael Roth     FILE *fh;
400c689b4f1SLaszlo Ersek     Error *local_err = NULL;
40185b6f6f5SSimon Zolin     int64_t handle;
402c216e5adSMichael Roth 
403c216e5adSMichael Roth     if (!has_mode) {
404c216e5adSMichael Roth         mode = "r";
405c216e5adSMichael Roth     }
406c216e5adSMichael Roth     slog("guest-file-open called, filepath: %s, mode: %s", path, mode);
407c689b4f1SLaszlo Ersek     fh = safe_open_or_create(path, mode, &local_err);
408c689b4f1SLaszlo Ersek     if (local_err != NULL) {
40977dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
410c216e5adSMichael Roth         return -1;
411c216e5adSMichael Roth     }
412c216e5adSMichael Roth 
413c216e5adSMichael Roth     /* set fd non-blocking to avoid common use cases (like reading from a
414c216e5adSMichael Roth      * named pipe) from hanging the agent
415c216e5adSMichael Roth      */
41612505396SDenis V. Lunev     qemu_set_nonblock(fileno(fh));
417c216e5adSMichael Roth 
41877dbc81bSMarkus Armbruster     handle = guest_file_handle_add(fh, errp);
419a903f40cSMarkus Armbruster     if (handle < 0) {
42039097dafSMichael Roth         fclose(fh);
42139097dafSMichael Roth         return -1;
42239097dafSMichael Roth     }
42339097dafSMichael Roth 
424d607a523SStefan Weil     slog("guest-file-open, handle: %" PRId64, handle);
42539097dafSMichael Roth     return handle;
426c216e5adSMichael Roth }
427c216e5adSMichael Roth 
42877dbc81bSMarkus Armbruster void qmp_guest_file_close(int64_t handle, Error **errp)
429c216e5adSMichael Roth {
43077dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
431c216e5adSMichael Roth     int ret;
432c216e5adSMichael Roth 
433d607a523SStefan Weil     slog("guest-file-close called, handle: %" PRId64, handle);
434c216e5adSMichael Roth     if (!gfh) {
435c216e5adSMichael Roth         return;
436c216e5adSMichael Roth     }
437c216e5adSMichael Roth 
438c216e5adSMichael Roth     ret = fclose(gfh->fh);
4393ac4b7c5SLuiz Capitulino     if (ret == EOF) {
44077dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to close handle");
441c216e5adSMichael Roth         return;
442c216e5adSMichael Roth     }
443c216e5adSMichael Roth 
444c216e5adSMichael Roth     QTAILQ_REMOVE(&guest_file_state.filehandles, gfh, next);
445c216e5adSMichael Roth     g_free(gfh);
446c216e5adSMichael Roth }
447c216e5adSMichael Roth 
448c216e5adSMichael Roth struct GuestFileRead *qmp_guest_file_read(int64_t handle, bool has_count,
44977dbc81bSMarkus Armbruster                                           int64_t count, Error **errp)
450c216e5adSMichael Roth {
45177dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
452c216e5adSMichael Roth     GuestFileRead *read_data = NULL;
453c216e5adSMichael Roth     guchar *buf;
454c216e5adSMichael Roth     FILE *fh;
455c216e5adSMichael Roth     size_t read_count;
456c216e5adSMichael Roth 
457c216e5adSMichael Roth     if (!gfh) {
458c216e5adSMichael Roth         return NULL;
459c216e5adSMichael Roth     }
460c216e5adSMichael Roth 
461c216e5adSMichael Roth     if (!has_count) {
462c216e5adSMichael Roth         count = QGA_READ_COUNT_DEFAULT;
463c216e5adSMichael Roth     } else if (count < 0) {
46477dbc81bSMarkus Armbruster         error_setg(errp, "value '%" PRId64 "' is invalid for argument count",
465db3edb66SLuiz Capitulino                    count);
466c216e5adSMichael Roth         return NULL;
467c216e5adSMichael Roth     }
468c216e5adSMichael Roth 
469c216e5adSMichael Roth     fh = gfh->fh;
470*895b00f6SMarc-André Lureau 
471*895b00f6SMarc-André Lureau     /* explicitly flush when switching from writing to reading */
472*895b00f6SMarc-André Lureau     if (gfh->state == RW_STATE_WRITING) {
473*895b00f6SMarc-André Lureau         int ret = fflush(fh);
474*895b00f6SMarc-André Lureau         if (ret == EOF) {
475*895b00f6SMarc-André Lureau             error_setg_errno(errp, errno, "failed to flush file");
476*895b00f6SMarc-André Lureau             return NULL;
477*895b00f6SMarc-André Lureau         }
478*895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
479*895b00f6SMarc-André Lureau     }
480*895b00f6SMarc-André Lureau 
481c216e5adSMichael Roth     buf = g_malloc0(count+1);
482c216e5adSMichael Roth     read_count = fread(buf, 1, count, fh);
483c216e5adSMichael Roth     if (ferror(fh)) {
48477dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to read file");
485d607a523SStefan Weil         slog("guest-file-read failed, handle: %" PRId64, handle);
486c216e5adSMichael Roth     } else {
487c216e5adSMichael Roth         buf[read_count] = 0;
488f3a06403SMarkus Armbruster         read_data = g_new0(GuestFileRead, 1);
489c216e5adSMichael Roth         read_data->count = read_count;
490c216e5adSMichael Roth         read_data->eof = feof(fh);
491c216e5adSMichael Roth         if (read_count) {
492c216e5adSMichael Roth             read_data->buf_b64 = g_base64_encode(buf, read_count);
493c216e5adSMichael Roth         }
494*895b00f6SMarc-André Lureau         gfh->state = RW_STATE_READING;
495c216e5adSMichael Roth     }
496c216e5adSMichael Roth     g_free(buf);
497c216e5adSMichael Roth     clearerr(fh);
498c216e5adSMichael Roth 
499c216e5adSMichael Roth     return read_data;
500c216e5adSMichael Roth }
501c216e5adSMichael Roth 
502c216e5adSMichael Roth GuestFileWrite *qmp_guest_file_write(int64_t handle, const char *buf_b64,
50377dbc81bSMarkus Armbruster                                      bool has_count, int64_t count,
50477dbc81bSMarkus Armbruster                                      Error **errp)
505c216e5adSMichael Roth {
506c216e5adSMichael Roth     GuestFileWrite *write_data = NULL;
507c216e5adSMichael Roth     guchar *buf;
508c216e5adSMichael Roth     gsize buf_len;
509c216e5adSMichael Roth     int write_count;
51077dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
511c216e5adSMichael Roth     FILE *fh;
512c216e5adSMichael Roth 
513c216e5adSMichael Roth     if (!gfh) {
514c216e5adSMichael Roth         return NULL;
515c216e5adSMichael Roth     }
516c216e5adSMichael Roth 
517c216e5adSMichael Roth     fh = gfh->fh;
518*895b00f6SMarc-André Lureau 
519*895b00f6SMarc-André Lureau     if (gfh->state == RW_STATE_READING) {
520*895b00f6SMarc-André Lureau         int ret = fseek(fh, 0, SEEK_CUR);
521*895b00f6SMarc-André Lureau         if (ret == -1) {
522*895b00f6SMarc-André Lureau             error_setg_errno(errp, errno, "failed to seek file");
523*895b00f6SMarc-André Lureau             return NULL;
524*895b00f6SMarc-André Lureau         }
525*895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
526*895b00f6SMarc-André Lureau     }
527*895b00f6SMarc-André Lureau 
528c216e5adSMichael Roth     buf = g_base64_decode(buf_b64, &buf_len);
529c216e5adSMichael Roth 
530c216e5adSMichael Roth     if (!has_count) {
531c216e5adSMichael Roth         count = buf_len;
532c216e5adSMichael Roth     } else if (count < 0 || count > buf_len) {
53377dbc81bSMarkus Armbruster         error_setg(errp, "value '%" PRId64 "' is invalid for argument count",
534db3edb66SLuiz Capitulino                    count);
535c216e5adSMichael Roth         g_free(buf);
536c216e5adSMichael Roth         return NULL;
537c216e5adSMichael Roth     }
538c216e5adSMichael Roth 
539c216e5adSMichael Roth     write_count = fwrite(buf, 1, count, fh);
540c216e5adSMichael Roth     if (ferror(fh)) {
54177dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to write to file");
542d607a523SStefan Weil         slog("guest-file-write failed, handle: %" PRId64, handle);
543c216e5adSMichael Roth     } else {
544f3a06403SMarkus Armbruster         write_data = g_new0(GuestFileWrite, 1);
545c216e5adSMichael Roth         write_data->count = write_count;
546c216e5adSMichael Roth         write_data->eof = feof(fh);
547*895b00f6SMarc-André Lureau         gfh->state = RW_STATE_WRITING;
548c216e5adSMichael Roth     }
549c216e5adSMichael Roth     g_free(buf);
550c216e5adSMichael Roth     clearerr(fh);
551c216e5adSMichael Roth 
552c216e5adSMichael Roth     return write_data;
553c216e5adSMichael Roth }
554c216e5adSMichael Roth 
555c216e5adSMichael Roth struct GuestFileSeek *qmp_guest_file_seek(int64_t handle, int64_t offset,
55677dbc81bSMarkus Armbruster                                           int64_t whence, Error **errp)
557c216e5adSMichael Roth {
55877dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
559c216e5adSMichael Roth     GuestFileSeek *seek_data = NULL;
560c216e5adSMichael Roth     FILE *fh;
561c216e5adSMichael Roth     int ret;
562c216e5adSMichael Roth 
563c216e5adSMichael Roth     if (!gfh) {
564c216e5adSMichael Roth         return NULL;
565c216e5adSMichael Roth     }
566c216e5adSMichael Roth 
567c216e5adSMichael Roth     fh = gfh->fh;
568c216e5adSMichael Roth     ret = fseek(fh, offset, whence);
569c216e5adSMichael Roth     if (ret == -1) {
57077dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to seek file");
571*895b00f6SMarc-André Lureau         if (errno == ESPIPE) {
572*895b00f6SMarc-André Lureau             /* file is non-seekable, stdio shouldn't be buffering anyways */
573*895b00f6SMarc-André Lureau             gfh->state = RW_STATE_NEW;
574*895b00f6SMarc-André Lureau         }
575c216e5adSMichael Roth     } else {
57610b7c5ddSMarkus Armbruster         seek_data = g_new0(GuestFileSeek, 1);
577c216e5adSMichael Roth         seek_data->position = ftell(fh);
578c216e5adSMichael Roth         seek_data->eof = feof(fh);
579*895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
580c216e5adSMichael Roth     }
581c216e5adSMichael Roth     clearerr(fh);
582c216e5adSMichael Roth 
583c216e5adSMichael Roth     return seek_data;
584c216e5adSMichael Roth }
585c216e5adSMichael Roth 
58677dbc81bSMarkus Armbruster void qmp_guest_file_flush(int64_t handle, Error **errp)
587c216e5adSMichael Roth {
58877dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
589c216e5adSMichael Roth     FILE *fh;
590c216e5adSMichael Roth     int ret;
591c216e5adSMichael Roth 
592c216e5adSMichael Roth     if (!gfh) {
593c216e5adSMichael Roth         return;
594c216e5adSMichael Roth     }
595c216e5adSMichael Roth 
596c216e5adSMichael Roth     fh = gfh->fh;
597c216e5adSMichael Roth     ret = fflush(fh);
598c216e5adSMichael Roth     if (ret == EOF) {
59977dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to flush file");
600*895b00f6SMarc-André Lureau     } else {
601*895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
602c216e5adSMichael Roth     }
603c216e5adSMichael Roth }
604c216e5adSMichael Roth 
605e72c3f2eSMichael Roth /* linux-specific implementations. avoid this if at all possible. */
606e72c3f2eSMichael Roth #if defined(__linux__)
607e72c3f2eSMichael Roth 
608eab5fd59SPaolo Bonzini #if defined(CONFIG_FSFREEZE) || defined(CONFIG_FSTRIM)
609af02203fSPaolo Bonzini typedef struct FsMount {
610c216e5adSMichael Roth     char *dirname;
611c216e5adSMichael Roth     char *devtype;
61246d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
613af02203fSPaolo Bonzini     QTAILQ_ENTRY(FsMount) next;
614af02203fSPaolo Bonzini } FsMount;
615c216e5adSMichael Roth 
616e5d9adbdSTomoki Sekiyama typedef QTAILQ_HEAD(FsMountList, FsMount) FsMountList;
6179e8aded4SMichael Roth 
618af02203fSPaolo Bonzini static void free_fs_mount_list(FsMountList *mounts)
619c216e5adSMichael Roth {
620af02203fSPaolo Bonzini      FsMount *mount, *temp;
621c216e5adSMichael Roth 
6229e8aded4SMichael Roth      if (!mounts) {
6239e8aded4SMichael Roth          return;
6249e8aded4SMichael Roth      }
6259e8aded4SMichael Roth 
6269e8aded4SMichael Roth      QTAILQ_FOREACH_SAFE(mount, mounts, next, temp) {
6279e8aded4SMichael Roth          QTAILQ_REMOVE(mounts, mount, next);
628c216e5adSMichael Roth          g_free(mount->dirname);
629c216e5adSMichael Roth          g_free(mount->devtype);
630c216e5adSMichael Roth          g_free(mount);
631c216e5adSMichael Roth      }
6329e8aded4SMichael Roth }
6339e8aded4SMichael Roth 
63446d4c572STomoki Sekiyama static int dev_major_minor(const char *devpath,
63546d4c572STomoki Sekiyama                            unsigned int *devmajor, unsigned int *devminor)
63646d4c572STomoki Sekiyama {
63746d4c572STomoki Sekiyama     struct stat st;
63846d4c572STomoki Sekiyama 
63946d4c572STomoki Sekiyama     *devmajor = 0;
64046d4c572STomoki Sekiyama     *devminor = 0;
64146d4c572STomoki Sekiyama 
64246d4c572STomoki Sekiyama     if (stat(devpath, &st) < 0) {
64346d4c572STomoki Sekiyama         slog("failed to stat device file '%s': %s", devpath, strerror(errno));
64446d4c572STomoki Sekiyama         return -1;
64546d4c572STomoki Sekiyama     }
64646d4c572STomoki Sekiyama     if (S_ISDIR(st.st_mode)) {
64746d4c572STomoki Sekiyama         /* It is bind mount */
64846d4c572STomoki Sekiyama         return -2;
64946d4c572STomoki Sekiyama     }
65046d4c572STomoki Sekiyama     if (S_ISBLK(st.st_mode)) {
65146d4c572STomoki Sekiyama         *devmajor = major(st.st_rdev);
65246d4c572STomoki Sekiyama         *devminor = minor(st.st_rdev);
65346d4c572STomoki Sekiyama         return 0;
65446d4c572STomoki Sekiyama     }
65546d4c572STomoki Sekiyama     return -1;
65646d4c572STomoki Sekiyama }
65746d4c572STomoki Sekiyama 
6589e8aded4SMichael Roth /*
6599e8aded4SMichael Roth  * Walk the mount table and build a list of local file systems
6609e8aded4SMichael Roth  */
66146d4c572STomoki Sekiyama static void build_fs_mount_list_from_mtab(FsMountList *mounts, Error **errp)
6629e8aded4SMichael Roth {
6639e8aded4SMichael Roth     struct mntent *ment;
664af02203fSPaolo Bonzini     FsMount *mount;
6659e2fa418SMichael Roth     char const *mtab = "/proc/self/mounts";
6669e8aded4SMichael Roth     FILE *fp;
66746d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
668c216e5adSMichael Roth 
669c216e5adSMichael Roth     fp = setmntent(mtab, "r");
670c216e5adSMichael Roth     if (!fp) {
67177dbc81bSMarkus Armbruster         error_setg(errp, "failed to open mtab file: '%s'", mtab);
672261551d1SLuiz Capitulino         return;
673c216e5adSMichael Roth     }
674c216e5adSMichael Roth 
675c216e5adSMichael Roth     while ((ment = getmntent(fp))) {
676c216e5adSMichael Roth         /*
677c216e5adSMichael Roth          * An entry which device name doesn't start with a '/' is
678c216e5adSMichael Roth          * either a dummy file system or a network file system.
679c216e5adSMichael Roth          * Add special handling for smbfs and cifs as is done by
680c216e5adSMichael Roth          * coreutils as well.
681c216e5adSMichael Roth          */
682c216e5adSMichael Roth         if ((ment->mnt_fsname[0] != '/') ||
683c216e5adSMichael Roth             (strcmp(ment->mnt_type, "smbfs") == 0) ||
684c216e5adSMichael Roth             (strcmp(ment->mnt_type, "cifs") == 0)) {
685c216e5adSMichael Roth             continue;
686c216e5adSMichael Roth         }
68746d4c572STomoki Sekiyama         if (dev_major_minor(ment->mnt_fsname, &devmajor, &devminor) == -2) {
68846d4c572STomoki Sekiyama             /* Skip bind mounts */
68946d4c572STomoki Sekiyama             continue;
69046d4c572STomoki Sekiyama         }
691c216e5adSMichael Roth 
692f3a06403SMarkus Armbruster         mount = g_new0(FsMount, 1);
693c216e5adSMichael Roth         mount->dirname = g_strdup(ment->mnt_dir);
694c216e5adSMichael Roth         mount->devtype = g_strdup(ment->mnt_type);
69546d4c572STomoki Sekiyama         mount->devmajor = devmajor;
69646d4c572STomoki Sekiyama         mount->devminor = devminor;
697c216e5adSMichael Roth 
6989e8aded4SMichael Roth         QTAILQ_INSERT_TAIL(mounts, mount, next);
699c216e5adSMichael Roth     }
700c216e5adSMichael Roth 
701c216e5adSMichael Roth     endmntent(fp);
702c216e5adSMichael Roth }
70346d4c572STomoki Sekiyama 
70446d4c572STomoki Sekiyama static void decode_mntname(char *name, int len)
70546d4c572STomoki Sekiyama {
70646d4c572STomoki Sekiyama     int i, j = 0;
70746d4c572STomoki Sekiyama     for (i = 0; i <= len; i++) {
70846d4c572STomoki Sekiyama         if (name[i] != '\\') {
70946d4c572STomoki Sekiyama             name[j++] = name[i];
71046d4c572STomoki Sekiyama         } else if (name[i + 1] == '\\') {
71146d4c572STomoki Sekiyama             name[j++] = '\\';
71246d4c572STomoki Sekiyama             i++;
71346d4c572STomoki Sekiyama         } else if (name[i + 1] >= '0' && name[i + 1] <= '3' &&
71446d4c572STomoki Sekiyama                    name[i + 2] >= '0' && name[i + 2] <= '7' &&
71546d4c572STomoki Sekiyama                    name[i + 3] >= '0' && name[i + 3] <= '7') {
71646d4c572STomoki Sekiyama             name[j++] = (name[i + 1] - '0') * 64 +
71746d4c572STomoki Sekiyama                         (name[i + 2] - '0') * 8 +
71846d4c572STomoki Sekiyama                         (name[i + 3] - '0');
71946d4c572STomoki Sekiyama             i += 3;
72046d4c572STomoki Sekiyama         } else {
72146d4c572STomoki Sekiyama             name[j++] = name[i];
72246d4c572STomoki Sekiyama         }
72346d4c572STomoki Sekiyama     }
72446d4c572STomoki Sekiyama }
72546d4c572STomoki Sekiyama 
72646d4c572STomoki Sekiyama static void build_fs_mount_list(FsMountList *mounts, Error **errp)
72746d4c572STomoki Sekiyama {
72846d4c572STomoki Sekiyama     FsMount *mount;
72946d4c572STomoki Sekiyama     char const *mountinfo = "/proc/self/mountinfo";
73046d4c572STomoki Sekiyama     FILE *fp;
73146d4c572STomoki Sekiyama     char *line = NULL, *dash;
73246d4c572STomoki Sekiyama     size_t n;
73346d4c572STomoki Sekiyama     char check;
73446d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
73546d4c572STomoki Sekiyama     int ret, dir_s, dir_e, type_s, type_e, dev_s, dev_e;
73646d4c572STomoki Sekiyama 
73746d4c572STomoki Sekiyama     fp = fopen(mountinfo, "r");
73846d4c572STomoki Sekiyama     if (!fp) {
73946d4c572STomoki Sekiyama         build_fs_mount_list_from_mtab(mounts, errp);
74046d4c572STomoki Sekiyama         return;
74146d4c572STomoki Sekiyama     }
74246d4c572STomoki Sekiyama 
74346d4c572STomoki Sekiyama     while (getline(&line, &n, fp) != -1) {
74446d4c572STomoki Sekiyama         ret = sscanf(line, "%*u %*u %u:%u %*s %n%*s%n%c",
74546d4c572STomoki Sekiyama                      &devmajor, &devminor, &dir_s, &dir_e, &check);
74646d4c572STomoki Sekiyama         if (ret < 3) {
74746d4c572STomoki Sekiyama             continue;
74846d4c572STomoki Sekiyama         }
74946d4c572STomoki Sekiyama         dash = strstr(line + dir_e, " - ");
75046d4c572STomoki Sekiyama         if (!dash) {
75146d4c572STomoki Sekiyama             continue;
75246d4c572STomoki Sekiyama         }
75346d4c572STomoki Sekiyama         ret = sscanf(dash, " - %n%*s%n %n%*s%n%c",
75446d4c572STomoki Sekiyama                      &type_s, &type_e, &dev_s, &dev_e, &check);
75546d4c572STomoki Sekiyama         if (ret < 1) {
75646d4c572STomoki Sekiyama             continue;
75746d4c572STomoki Sekiyama         }
75846d4c572STomoki Sekiyama         line[dir_e] = 0;
75946d4c572STomoki Sekiyama         dash[type_e] = 0;
76046d4c572STomoki Sekiyama         dash[dev_e] = 0;
76146d4c572STomoki Sekiyama         decode_mntname(line + dir_s, dir_e - dir_s);
76246d4c572STomoki Sekiyama         decode_mntname(dash + dev_s, dev_e - dev_s);
76346d4c572STomoki Sekiyama         if (devmajor == 0) {
76446d4c572STomoki Sekiyama             /* btrfs reports major number = 0 */
76546d4c572STomoki Sekiyama             if (strcmp("btrfs", dash + type_s) != 0 ||
76646d4c572STomoki Sekiyama                 dev_major_minor(dash + dev_s, &devmajor, &devminor) < 0) {
76746d4c572STomoki Sekiyama                 continue;
76846d4c572STomoki Sekiyama             }
76946d4c572STomoki Sekiyama         }
77046d4c572STomoki Sekiyama 
771f3a06403SMarkus Armbruster         mount = g_new0(FsMount, 1);
77246d4c572STomoki Sekiyama         mount->dirname = g_strdup(line + dir_s);
77346d4c572STomoki Sekiyama         mount->devtype = g_strdup(dash + type_s);
77446d4c572STomoki Sekiyama         mount->devmajor = devmajor;
77546d4c572STomoki Sekiyama         mount->devminor = devminor;
77646d4c572STomoki Sekiyama 
77746d4c572STomoki Sekiyama         QTAILQ_INSERT_TAIL(mounts, mount, next);
77846d4c572STomoki Sekiyama     }
77946d4c572STomoki Sekiyama     free(line);
78046d4c572STomoki Sekiyama 
78146d4c572STomoki Sekiyama     fclose(fp);
78246d4c572STomoki Sekiyama }
783eab5fd59SPaolo Bonzini #endif
784eab5fd59SPaolo Bonzini 
785eab5fd59SPaolo Bonzini #if defined(CONFIG_FSFREEZE)
786c216e5adSMichael Roth 
78746d4c572STomoki Sekiyama static char *get_pci_driver(char const *syspath, int pathlen, Error **errp)
78846d4c572STomoki Sekiyama {
78946d4c572STomoki Sekiyama     char *path;
79046d4c572STomoki Sekiyama     char *dpath;
79146d4c572STomoki Sekiyama     char *driver = NULL;
79246d4c572STomoki Sekiyama     char buf[PATH_MAX];
79346d4c572STomoki Sekiyama     ssize_t len;
79446d4c572STomoki Sekiyama 
79546d4c572STomoki Sekiyama     path = g_strndup(syspath, pathlen);
79646d4c572STomoki Sekiyama     dpath = g_strdup_printf("%s/driver", path);
79746d4c572STomoki Sekiyama     len = readlink(dpath, buf, sizeof(buf) - 1);
79846d4c572STomoki Sekiyama     if (len != -1) {
79946d4c572STomoki Sekiyama         buf[len] = 0;
80046d4c572STomoki Sekiyama         driver = g_strdup(basename(buf));
80146d4c572STomoki Sekiyama     }
80246d4c572STomoki Sekiyama     g_free(dpath);
80346d4c572STomoki Sekiyama     g_free(path);
80446d4c572STomoki Sekiyama     return driver;
80546d4c572STomoki Sekiyama }
80646d4c572STomoki Sekiyama 
80746d4c572STomoki Sekiyama static int compare_uint(const void *_a, const void *_b)
80846d4c572STomoki Sekiyama {
80946d4c572STomoki Sekiyama     unsigned int a = *(unsigned int *)_a;
81046d4c572STomoki Sekiyama     unsigned int b = *(unsigned int *)_b;
81146d4c572STomoki Sekiyama 
81246d4c572STomoki Sekiyama     return a < b ? -1 : a > b ? 1 : 0;
81346d4c572STomoki Sekiyama }
81446d4c572STomoki Sekiyama 
81546d4c572STomoki Sekiyama /* Walk the specified sysfs and build a sorted list of host or ata numbers */
81646d4c572STomoki Sekiyama static int build_hosts(char const *syspath, char const *host, bool ata,
81746d4c572STomoki Sekiyama                        unsigned int *hosts, int hosts_max, Error **errp)
81846d4c572STomoki Sekiyama {
81946d4c572STomoki Sekiyama     char *path;
82046d4c572STomoki Sekiyama     DIR *dir;
82146d4c572STomoki Sekiyama     struct dirent *entry;
82246d4c572STomoki Sekiyama     int i = 0;
82346d4c572STomoki Sekiyama 
82446d4c572STomoki Sekiyama     path = g_strndup(syspath, host - syspath);
82546d4c572STomoki Sekiyama     dir = opendir(path);
82646d4c572STomoki Sekiyama     if (!dir) {
82746d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "opendir(\"%s\")", path);
82846d4c572STomoki Sekiyama         g_free(path);
82946d4c572STomoki Sekiyama         return -1;
83046d4c572STomoki Sekiyama     }
83146d4c572STomoki Sekiyama 
83246d4c572STomoki Sekiyama     while (i < hosts_max) {
83346d4c572STomoki Sekiyama         entry = readdir(dir);
83446d4c572STomoki Sekiyama         if (!entry) {
83546d4c572STomoki Sekiyama             break;
83646d4c572STomoki Sekiyama         }
83746d4c572STomoki Sekiyama         if (ata && sscanf(entry->d_name, "ata%d", hosts + i) == 1) {
83846d4c572STomoki Sekiyama             ++i;
83946d4c572STomoki Sekiyama         } else if (!ata && sscanf(entry->d_name, "host%d", hosts + i) == 1) {
84046d4c572STomoki Sekiyama             ++i;
84146d4c572STomoki Sekiyama         }
84246d4c572STomoki Sekiyama     }
84346d4c572STomoki Sekiyama 
84446d4c572STomoki Sekiyama     qsort(hosts, i, sizeof(hosts[0]), compare_uint);
84546d4c572STomoki Sekiyama 
84646d4c572STomoki Sekiyama     g_free(path);
84746d4c572STomoki Sekiyama     closedir(dir);
84846d4c572STomoki Sekiyama     return i;
84946d4c572STomoki Sekiyama }
85046d4c572STomoki Sekiyama 
85146d4c572STomoki Sekiyama /* Store disk device info specified by @sysfs into @fs */
85246d4c572STomoki Sekiyama static void build_guest_fsinfo_for_real_device(char const *syspath,
85346d4c572STomoki Sekiyama                                                GuestFilesystemInfo *fs,
85446d4c572STomoki Sekiyama                                                Error **errp)
85546d4c572STomoki Sekiyama {
85646d4c572STomoki Sekiyama     unsigned int pci[4], host, hosts[8], tgt[3];
85746d4c572STomoki Sekiyama     int i, nhosts = 0, pcilen;
85846d4c572STomoki Sekiyama     GuestDiskAddress *disk;
85946d4c572STomoki Sekiyama     GuestPCIAddress *pciaddr;
86046d4c572STomoki Sekiyama     GuestDiskAddressList *list = NULL;
86146d4c572STomoki Sekiyama     bool has_ata = false, has_host = false, has_tgt = false;
86246d4c572STomoki Sekiyama     char *p, *q, *driver = NULL;
86346d4c572STomoki Sekiyama 
86446d4c572STomoki Sekiyama     p = strstr(syspath, "/devices/pci");
86546d4c572STomoki Sekiyama     if (!p || sscanf(p + 12, "%*x:%*x/%x:%x:%x.%x%n",
86646d4c572STomoki Sekiyama                      pci, pci + 1, pci + 2, pci + 3, &pcilen) < 4) {
86746d4c572STomoki Sekiyama         g_debug("only pci device is supported: sysfs path \"%s\"", syspath);
86846d4c572STomoki Sekiyama         return;
86946d4c572STomoki Sekiyama     }
87046d4c572STomoki Sekiyama 
87146d4c572STomoki Sekiyama     driver = get_pci_driver(syspath, (p + 12 + pcilen) - syspath, errp);
87246d4c572STomoki Sekiyama     if (!driver) {
87346d4c572STomoki Sekiyama         goto cleanup;
87446d4c572STomoki Sekiyama     }
87546d4c572STomoki Sekiyama 
87646d4c572STomoki Sekiyama     p = strstr(syspath, "/target");
87746d4c572STomoki Sekiyama     if (p && sscanf(p + 7, "%*u:%*u:%*u/%*u:%u:%u:%u",
87846d4c572STomoki Sekiyama                     tgt, tgt + 1, tgt + 2) == 3) {
87946d4c572STomoki Sekiyama         has_tgt = true;
88046d4c572STomoki Sekiyama     }
88146d4c572STomoki Sekiyama 
88246d4c572STomoki Sekiyama     p = strstr(syspath, "/ata");
88346d4c572STomoki Sekiyama     if (p) {
88446d4c572STomoki Sekiyama         q = p + 4;
88546d4c572STomoki Sekiyama         has_ata = true;
88646d4c572STomoki Sekiyama     } else {
88746d4c572STomoki Sekiyama         p = strstr(syspath, "/host");
88846d4c572STomoki Sekiyama         q = p + 5;
88946d4c572STomoki Sekiyama     }
89046d4c572STomoki Sekiyama     if (p && sscanf(q, "%u", &host) == 1) {
89146d4c572STomoki Sekiyama         has_host = true;
89246d4c572STomoki Sekiyama         nhosts = build_hosts(syspath, p, has_ata, hosts,
89346d4c572STomoki Sekiyama                              sizeof(hosts) / sizeof(hosts[0]), errp);
89446d4c572STomoki Sekiyama         if (nhosts < 0) {
89546d4c572STomoki Sekiyama             goto cleanup;
89646d4c572STomoki Sekiyama         }
89746d4c572STomoki Sekiyama     }
89846d4c572STomoki Sekiyama 
89946d4c572STomoki Sekiyama     pciaddr = g_malloc0(sizeof(*pciaddr));
90046d4c572STomoki Sekiyama     pciaddr->domain = pci[0];
90146d4c572STomoki Sekiyama     pciaddr->bus = pci[1];
90246d4c572STomoki Sekiyama     pciaddr->slot = pci[2];
90346d4c572STomoki Sekiyama     pciaddr->function = pci[3];
90446d4c572STomoki Sekiyama 
90546d4c572STomoki Sekiyama     disk = g_malloc0(sizeof(*disk));
90646d4c572STomoki Sekiyama     disk->pci_controller = pciaddr;
90746d4c572STomoki Sekiyama 
90846d4c572STomoki Sekiyama     list = g_malloc0(sizeof(*list));
90946d4c572STomoki Sekiyama     list->value = disk;
91046d4c572STomoki Sekiyama 
91146d4c572STomoki Sekiyama     if (strcmp(driver, "ata_piix") == 0) {
91246d4c572STomoki Sekiyama         /* a host per ide bus, target*:0:<unit>:0 */
91346d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
91446d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
91546d4c572STomoki Sekiyama             goto cleanup;
91646d4c572STomoki Sekiyama         }
91746d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
91846d4c572STomoki Sekiyama             if (host == hosts[i]) {
91946d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_IDE;
92046d4c572STomoki Sekiyama                 disk->bus = i;
92146d4c572STomoki Sekiyama                 disk->unit = tgt[1];
92246d4c572STomoki Sekiyama                 break;
92346d4c572STomoki Sekiyama             }
92446d4c572STomoki Sekiyama         }
92546d4c572STomoki Sekiyama         if (i >= nhosts) {
92646d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
92746d4c572STomoki Sekiyama             goto cleanup;
92846d4c572STomoki Sekiyama         }
92946d4c572STomoki Sekiyama     } else if (strcmp(driver, "sym53c8xx") == 0) {
93046d4c572STomoki Sekiyama         /* scsi(LSI Logic): target*:0:<unit>:0 */
93146d4c572STomoki Sekiyama         if (!has_tgt) {
93246d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
93346d4c572STomoki Sekiyama             goto cleanup;
93446d4c572STomoki Sekiyama         }
93546d4c572STomoki Sekiyama         disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
93646d4c572STomoki Sekiyama         disk->unit = tgt[1];
93746d4c572STomoki Sekiyama     } else if (strcmp(driver, "virtio-pci") == 0) {
93846d4c572STomoki Sekiyama         if (has_tgt) {
93946d4c572STomoki Sekiyama             /* virtio-scsi: target*:0:0:<unit> */
94046d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
94146d4c572STomoki Sekiyama             disk->unit = tgt[2];
94246d4c572STomoki Sekiyama         } else {
94346d4c572STomoki Sekiyama             /* virtio-blk: 1 disk per 1 device */
94446d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_VIRTIO;
94546d4c572STomoki Sekiyama         }
94646d4c572STomoki Sekiyama     } else if (strcmp(driver, "ahci") == 0) {
94746d4c572STomoki Sekiyama         /* ahci: 1 host per 1 unit */
94846d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
94946d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
95046d4c572STomoki Sekiyama             goto cleanup;
95146d4c572STomoki Sekiyama         }
95246d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
95346d4c572STomoki Sekiyama             if (host == hosts[i]) {
95446d4c572STomoki Sekiyama                 disk->unit = i;
95546d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_SATA;
95646d4c572STomoki Sekiyama                 break;
95746d4c572STomoki Sekiyama             }
95846d4c572STomoki Sekiyama         }
95946d4c572STomoki Sekiyama         if (i >= nhosts) {
96046d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
96146d4c572STomoki Sekiyama             goto cleanup;
96246d4c572STomoki Sekiyama         }
96346d4c572STomoki Sekiyama     } else {
96446d4c572STomoki Sekiyama         g_debug("unknown driver '%s' (sysfs path '%s')", driver, syspath);
96546d4c572STomoki Sekiyama         goto cleanup;
96646d4c572STomoki Sekiyama     }
96746d4c572STomoki Sekiyama 
96846d4c572STomoki Sekiyama     list->next = fs->disk;
96946d4c572STomoki Sekiyama     fs->disk = list;
97046d4c572STomoki Sekiyama     g_free(driver);
97146d4c572STomoki Sekiyama     return;
97246d4c572STomoki Sekiyama 
97346d4c572STomoki Sekiyama cleanup:
97446d4c572STomoki Sekiyama     if (list) {
97546d4c572STomoki Sekiyama         qapi_free_GuestDiskAddressList(list);
97646d4c572STomoki Sekiyama     }
97746d4c572STomoki Sekiyama     g_free(driver);
97846d4c572STomoki Sekiyama }
97946d4c572STomoki Sekiyama 
98046d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
98146d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
98246d4c572STomoki Sekiyama                                           Error **errp);
98346d4c572STomoki Sekiyama 
98446d4c572STomoki Sekiyama /* Store a list of slave devices of virtual volume specified by @syspath into
98546d4c572STomoki Sekiyama  * @fs */
98646d4c572STomoki Sekiyama static void build_guest_fsinfo_for_virtual_device(char const *syspath,
98746d4c572STomoki Sekiyama                                                   GuestFilesystemInfo *fs,
98846d4c572STomoki Sekiyama                                                   Error **errp)
98946d4c572STomoki Sekiyama {
99046d4c572STomoki Sekiyama     DIR *dir;
99146d4c572STomoki Sekiyama     char *dirpath;
992e668d1b8Szhanghailiang     struct dirent *entry;
99346d4c572STomoki Sekiyama 
99446d4c572STomoki Sekiyama     dirpath = g_strdup_printf("%s/slaves", syspath);
99546d4c572STomoki Sekiyama     dir = opendir(dirpath);
99646d4c572STomoki Sekiyama     if (!dir) {
99746d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "opendir(\"%s\")", dirpath);
99846d4c572STomoki Sekiyama         g_free(dirpath);
99946d4c572STomoki Sekiyama         return;
100046d4c572STomoki Sekiyama     }
100146d4c572STomoki Sekiyama 
100246d4c572STomoki Sekiyama     for (;;) {
1003e668d1b8Szhanghailiang         errno = 0;
1004e668d1b8Szhanghailiang         entry = readdir(dir);
1005e668d1b8Szhanghailiang         if (entry == NULL) {
1006e668d1b8Szhanghailiang             if (errno) {
1007e668d1b8Szhanghailiang                 error_setg_errno(errp, errno, "readdir(\"%s\")", dirpath);
100846d4c572STomoki Sekiyama             }
100946d4c572STomoki Sekiyama             break;
101046d4c572STomoki Sekiyama         }
101146d4c572STomoki Sekiyama 
1012e668d1b8Szhanghailiang         if (entry->d_type == DT_LNK) {
1013e668d1b8Szhanghailiang             char *path;
1014e668d1b8Szhanghailiang 
1015e668d1b8Szhanghailiang             g_debug(" slave device '%s'", entry->d_name);
1016e668d1b8Szhanghailiang             path = g_strdup_printf("%s/slaves/%s", syspath, entry->d_name);
1017e668d1b8Szhanghailiang             build_guest_fsinfo_for_device(path, fs, errp);
1018e668d1b8Szhanghailiang             g_free(path);
101946d4c572STomoki Sekiyama 
102046d4c572STomoki Sekiyama             if (*errp) {
102146d4c572STomoki Sekiyama                 break;
102246d4c572STomoki Sekiyama             }
102346d4c572STomoki Sekiyama         }
102446d4c572STomoki Sekiyama     }
102546d4c572STomoki Sekiyama 
1026e668d1b8Szhanghailiang     g_free(dirpath);
102746d4c572STomoki Sekiyama     closedir(dir);
102846d4c572STomoki Sekiyama }
102946d4c572STomoki Sekiyama 
103046d4c572STomoki Sekiyama /* Dispatch to functions for virtual/real device */
103146d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
103246d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
103346d4c572STomoki Sekiyama                                           Error **errp)
103446d4c572STomoki Sekiyama {
103546d4c572STomoki Sekiyama     char *syspath = realpath(devpath, NULL);
103646d4c572STomoki Sekiyama 
103746d4c572STomoki Sekiyama     if (!syspath) {
103846d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "realpath(\"%s\")", devpath);
103946d4c572STomoki Sekiyama         return;
104046d4c572STomoki Sekiyama     }
104146d4c572STomoki Sekiyama 
104246d4c572STomoki Sekiyama     if (!fs->name) {
104346d4c572STomoki Sekiyama         fs->name = g_strdup(basename(syspath));
104446d4c572STomoki Sekiyama     }
104546d4c572STomoki Sekiyama 
104646d4c572STomoki Sekiyama     g_debug("  parse sysfs path '%s'", syspath);
104746d4c572STomoki Sekiyama 
104846d4c572STomoki Sekiyama     if (strstr(syspath, "/devices/virtual/block/")) {
104946d4c572STomoki Sekiyama         build_guest_fsinfo_for_virtual_device(syspath, fs, errp);
105046d4c572STomoki Sekiyama     } else {
105146d4c572STomoki Sekiyama         build_guest_fsinfo_for_real_device(syspath, fs, errp);
105246d4c572STomoki Sekiyama     }
105346d4c572STomoki Sekiyama 
105446d4c572STomoki Sekiyama     free(syspath);
105546d4c572STomoki Sekiyama }
105646d4c572STomoki Sekiyama 
105746d4c572STomoki Sekiyama /* Return a list of the disk device(s)' info which @mount lies on */
105846d4c572STomoki Sekiyama static GuestFilesystemInfo *build_guest_fsinfo(struct FsMount *mount,
105946d4c572STomoki Sekiyama                                                Error **errp)
106046d4c572STomoki Sekiyama {
106146d4c572STomoki Sekiyama     GuestFilesystemInfo *fs = g_malloc0(sizeof(*fs));
106246d4c572STomoki Sekiyama     char *devpath = g_strdup_printf("/sys/dev/block/%u:%u",
106346d4c572STomoki Sekiyama                                     mount->devmajor, mount->devminor);
106446d4c572STomoki Sekiyama 
106546d4c572STomoki Sekiyama     fs->mountpoint = g_strdup(mount->dirname);
106646d4c572STomoki Sekiyama     fs->type = g_strdup(mount->devtype);
106746d4c572STomoki Sekiyama     build_guest_fsinfo_for_device(devpath, fs, errp);
106846d4c572STomoki Sekiyama 
106946d4c572STomoki Sekiyama     g_free(devpath);
107046d4c572STomoki Sekiyama     return fs;
107146d4c572STomoki Sekiyama }
107246d4c572STomoki Sekiyama 
107346d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
107446d4c572STomoki Sekiyama {
107546d4c572STomoki Sekiyama     FsMountList mounts;
107646d4c572STomoki Sekiyama     struct FsMount *mount;
107746d4c572STomoki Sekiyama     GuestFilesystemInfoList *new, *ret = NULL;
107846d4c572STomoki Sekiyama     Error *local_err = NULL;
107946d4c572STomoki Sekiyama 
108046d4c572STomoki Sekiyama     QTAILQ_INIT(&mounts);
108146d4c572STomoki Sekiyama     build_fs_mount_list(&mounts, &local_err);
108246d4c572STomoki Sekiyama     if (local_err) {
108346d4c572STomoki Sekiyama         error_propagate(errp, local_err);
108446d4c572STomoki Sekiyama         return NULL;
108546d4c572STomoki Sekiyama     }
108646d4c572STomoki Sekiyama 
108746d4c572STomoki Sekiyama     QTAILQ_FOREACH(mount, &mounts, next) {
108846d4c572STomoki Sekiyama         g_debug("Building guest fsinfo for '%s'", mount->dirname);
108946d4c572STomoki Sekiyama 
109046d4c572STomoki Sekiyama         new = g_malloc0(sizeof(*ret));
109146d4c572STomoki Sekiyama         new->value = build_guest_fsinfo(mount, &local_err);
109246d4c572STomoki Sekiyama         new->next = ret;
109346d4c572STomoki Sekiyama         ret = new;
109446d4c572STomoki Sekiyama         if (local_err) {
109546d4c572STomoki Sekiyama             error_propagate(errp, local_err);
109646d4c572STomoki Sekiyama             qapi_free_GuestFilesystemInfoList(ret);
109746d4c572STomoki Sekiyama             ret = NULL;
109846d4c572STomoki Sekiyama             break;
109946d4c572STomoki Sekiyama         }
110046d4c572STomoki Sekiyama     }
110146d4c572STomoki Sekiyama 
110246d4c572STomoki Sekiyama     free_fs_mount_list(&mounts);
110346d4c572STomoki Sekiyama     return ret;
110446d4c572STomoki Sekiyama }
110546d4c572STomoki Sekiyama 
110646d4c572STomoki Sekiyama 
1107ec0f694cSTomoki Sekiyama typedef enum {
1108ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_THAW = 0,
1109ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_FREEZE,
1110ec0f694cSTomoki Sekiyama } FsfreezeHookArg;
1111ec0f694cSTomoki Sekiyama 
111213a439ecSStefan Weil static const char *fsfreeze_hook_arg_string[] = {
1113ec0f694cSTomoki Sekiyama     "thaw",
1114ec0f694cSTomoki Sekiyama     "freeze",
1115ec0f694cSTomoki Sekiyama };
1116ec0f694cSTomoki Sekiyama 
111777dbc81bSMarkus Armbruster static void execute_fsfreeze_hook(FsfreezeHookArg arg, Error **errp)
1118ec0f694cSTomoki Sekiyama {
1119ec0f694cSTomoki Sekiyama     int status;
1120ec0f694cSTomoki Sekiyama     pid_t pid;
1121ec0f694cSTomoki Sekiyama     const char *hook;
1122ec0f694cSTomoki Sekiyama     const char *arg_str = fsfreeze_hook_arg_string[arg];
1123ec0f694cSTomoki Sekiyama     Error *local_err = NULL;
1124ec0f694cSTomoki Sekiyama 
1125ec0f694cSTomoki Sekiyama     hook = ga_fsfreeze_hook(ga_state);
1126ec0f694cSTomoki Sekiyama     if (!hook) {
1127ec0f694cSTomoki Sekiyama         return;
1128ec0f694cSTomoki Sekiyama     }
1129ec0f694cSTomoki Sekiyama     if (access(hook, X_OK) != 0) {
113077dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "can't access fsfreeze hook '%s'", hook);
1131ec0f694cSTomoki Sekiyama         return;
1132ec0f694cSTomoki Sekiyama     }
1133ec0f694cSTomoki Sekiyama 
1134ec0f694cSTomoki Sekiyama     slog("executing fsfreeze hook with arg '%s'", arg_str);
1135ec0f694cSTomoki Sekiyama     pid = fork();
1136ec0f694cSTomoki Sekiyama     if (pid == 0) {
1137ec0f694cSTomoki Sekiyama         setsid();
1138ec0f694cSTomoki Sekiyama         reopen_fd_to_null(0);
1139ec0f694cSTomoki Sekiyama         reopen_fd_to_null(1);
1140ec0f694cSTomoki Sekiyama         reopen_fd_to_null(2);
1141ec0f694cSTomoki Sekiyama 
1142ec0f694cSTomoki Sekiyama         execle(hook, hook, arg_str, NULL, environ);
1143ec0f694cSTomoki Sekiyama         _exit(EXIT_FAILURE);
1144ec0f694cSTomoki Sekiyama     } else if (pid < 0) {
114577dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
1146ec0f694cSTomoki Sekiyama         return;
1147ec0f694cSTomoki Sekiyama     }
1148ec0f694cSTomoki Sekiyama 
1149ec0f694cSTomoki Sekiyama     ga_wait_child(pid, &status, &local_err);
115084d18f06SMarkus Armbruster     if (local_err) {
115177dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1152ec0f694cSTomoki Sekiyama         return;
1153ec0f694cSTomoki Sekiyama     }
1154ec0f694cSTomoki Sekiyama 
1155ec0f694cSTomoki Sekiyama     if (!WIFEXITED(status)) {
115677dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has terminated abnormally");
1157ec0f694cSTomoki Sekiyama         return;
1158ec0f694cSTomoki Sekiyama     }
1159ec0f694cSTomoki Sekiyama 
1160ec0f694cSTomoki Sekiyama     status = WEXITSTATUS(status);
1161ec0f694cSTomoki Sekiyama     if (status) {
116277dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has failed with status %d", status);
1163ec0f694cSTomoki Sekiyama         return;
1164ec0f694cSTomoki Sekiyama     }
1165ec0f694cSTomoki Sekiyama }
1166ec0f694cSTomoki Sekiyama 
1167c216e5adSMichael Roth /*
1168c216e5adSMichael Roth  * Return status of freeze/thaw
1169c216e5adSMichael Roth  */
117077dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
1171c216e5adSMichael Roth {
1172f22d85e9SMichael Roth     if (ga_is_frozen(ga_state)) {
1173f22d85e9SMichael Roth         return GUEST_FSFREEZE_STATUS_FROZEN;
1174f22d85e9SMichael Roth     }
1175f22d85e9SMichael Roth 
1176f22d85e9SMichael Roth     return GUEST_FSFREEZE_STATUS_THAWED;
1177c216e5adSMichael Roth }
1178c216e5adSMichael Roth 
1179e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze(Error **errp)
1180e99bce20STomoki Sekiyama {
1181e99bce20STomoki Sekiyama     return qmp_guest_fsfreeze_freeze_list(false, NULL, errp);
1182e99bce20STomoki Sekiyama }
1183e99bce20STomoki Sekiyama 
1184c216e5adSMichael Roth /*
1185c216e5adSMichael Roth  * Walk list of mounted file systems in the guest, and freeze the ones which
1186c216e5adSMichael Roth  * are real local file systems.
1187c216e5adSMichael Roth  */
1188e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
1189e99bce20STomoki Sekiyama                                        strList *mountpoints,
1190e99bce20STomoki Sekiyama                                        Error **errp)
1191c216e5adSMichael Roth {
1192c216e5adSMichael Roth     int ret = 0, i = 0;
1193e99bce20STomoki Sekiyama     strList *list;
1194af02203fSPaolo Bonzini     FsMountList mounts;
1195af02203fSPaolo Bonzini     struct FsMount *mount;
1196261551d1SLuiz Capitulino     Error *local_err = NULL;
1197c216e5adSMichael Roth     int fd;
1198c216e5adSMichael Roth 
1199c216e5adSMichael Roth     slog("guest-fsfreeze called");
1200c216e5adSMichael Roth 
1201ec0f694cSTomoki Sekiyama     execute_fsfreeze_hook(FSFREEZE_HOOK_FREEZE, &local_err);
120284d18f06SMarkus Armbruster     if (local_err) {
120377dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1204ec0f694cSTomoki Sekiyama         return -1;
1205ec0f694cSTomoki Sekiyama     }
1206ec0f694cSTomoki Sekiyama 
12079e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1208261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
120984d18f06SMarkus Armbruster     if (local_err) {
121077dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1211261551d1SLuiz Capitulino         return -1;
1212c216e5adSMichael Roth     }
1213c216e5adSMichael Roth 
1214c216e5adSMichael Roth     /* cannot risk guest agent blocking itself on a write in this state */
1215f22d85e9SMichael Roth     ga_set_frozen(ga_state);
1216c216e5adSMichael Roth 
1217e5d9adbdSTomoki Sekiyama     QTAILQ_FOREACH_REVERSE(mount, &mounts, FsMountList, next) {
1218e99bce20STomoki Sekiyama         /* To issue fsfreeze in the reverse order of mounts, check if the
1219e99bce20STomoki Sekiyama          * mount is listed in the list here */
1220e99bce20STomoki Sekiyama         if (has_mountpoints) {
1221e99bce20STomoki Sekiyama             for (list = mountpoints; list; list = list->next) {
1222e99bce20STomoki Sekiyama                 if (strcmp(list->value, mount->dirname) == 0) {
1223e99bce20STomoki Sekiyama                     break;
1224e99bce20STomoki Sekiyama                 }
1225e99bce20STomoki Sekiyama             }
1226e99bce20STomoki Sekiyama             if (!list) {
1227e99bce20STomoki Sekiyama                 continue;
1228e99bce20STomoki Sekiyama             }
1229e99bce20STomoki Sekiyama         }
1230e99bce20STomoki Sekiyama 
1231c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1232c216e5adSMichael Roth         if (fd == -1) {
123377dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "failed to open %s", mount->dirname);
1234c216e5adSMichael Roth             goto error;
1235c216e5adSMichael Roth         }
1236c216e5adSMichael Roth 
1237c216e5adSMichael Roth         /* we try to cull filesytems we know won't work in advance, but other
1238c216e5adSMichael Roth          * filesytems may not implement fsfreeze for less obvious reasons.
12399e8aded4SMichael Roth          * these will report EOPNOTSUPP. we simply ignore these when tallying
12409e8aded4SMichael Roth          * the number of frozen filesystems.
12419e8aded4SMichael Roth          *
12429e8aded4SMichael Roth          * any other error means a failure to freeze a filesystem we
12439e8aded4SMichael Roth          * expect to be freezable, so return an error in those cases
12449e8aded4SMichael Roth          * and return system to thawed state.
1245c216e5adSMichael Roth          */
1246c216e5adSMichael Roth         ret = ioctl(fd, FIFREEZE);
12479e8aded4SMichael Roth         if (ret == -1) {
12489e8aded4SMichael Roth             if (errno != EOPNOTSUPP) {
124977dbc81bSMarkus Armbruster                 error_setg_errno(errp, errno, "failed to freeze %s",
1250617fbbc1SLuiz Capitulino                                  mount->dirname);
1251c216e5adSMichael Roth                 close(fd);
1252c216e5adSMichael Roth                 goto error;
1253c216e5adSMichael Roth             }
12549e8aded4SMichael Roth         } else {
1255c216e5adSMichael Roth             i++;
1256c216e5adSMichael Roth         }
12579e8aded4SMichael Roth         close(fd);
12589e8aded4SMichael Roth     }
1259c216e5adSMichael Roth 
1260af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1261c216e5adSMichael Roth     return i;
1262c216e5adSMichael Roth 
1263c216e5adSMichael Roth error:
1264af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1265c216e5adSMichael Roth     qmp_guest_fsfreeze_thaw(NULL);
1266c216e5adSMichael Roth     return 0;
1267c216e5adSMichael Roth }
1268c216e5adSMichael Roth 
1269c216e5adSMichael Roth /*
1270c216e5adSMichael Roth  * Walk list of frozen file systems in the guest, and thaw them.
1271c216e5adSMichael Roth  */
127277dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
1273c216e5adSMichael Roth {
1274c216e5adSMichael Roth     int ret;
1275af02203fSPaolo Bonzini     FsMountList mounts;
1276af02203fSPaolo Bonzini     FsMount *mount;
12779e8aded4SMichael Roth     int fd, i = 0, logged;
1278261551d1SLuiz Capitulino     Error *local_err = NULL;
1279c216e5adSMichael Roth 
12809e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1281261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
128284d18f06SMarkus Armbruster     if (local_err) {
128377dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
12849e8aded4SMichael Roth         return 0;
12859e8aded4SMichael Roth     }
12869e8aded4SMichael Roth 
12879e8aded4SMichael Roth     QTAILQ_FOREACH(mount, &mounts, next) {
12889e8aded4SMichael Roth         logged = false;
1289c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1290c216e5adSMichael Roth         if (fd == -1) {
1291c216e5adSMichael Roth             continue;
1292c216e5adSMichael Roth         }
12939e8aded4SMichael Roth         /* we have no way of knowing whether a filesystem was actually unfrozen
12949e8aded4SMichael Roth          * as a result of a successful call to FITHAW, only that if an error
12959e8aded4SMichael Roth          * was returned the filesystem was *not* unfrozen by that particular
12969e8aded4SMichael Roth          * call.
12979e8aded4SMichael Roth          *
1298a31f0531SJim Meyering          * since multiple preceding FIFREEZEs require multiple calls to FITHAW
12999e8aded4SMichael Roth          * to unfreeze, continuing issuing FITHAW until an error is returned,
13009e8aded4SMichael Roth          * in which case either the filesystem is in an unfreezable state, or,
13019e8aded4SMichael Roth          * more likely, it was thawed previously (and remains so afterward).
13029e8aded4SMichael Roth          *
13039e8aded4SMichael Roth          * also, since the most recent successful call is the one that did
13049e8aded4SMichael Roth          * the actual unfreeze, we can use this to provide an accurate count
13059e8aded4SMichael Roth          * of the number of filesystems unfrozen by guest-fsfreeze-thaw, which
13069e8aded4SMichael Roth          * may * be useful for determining whether a filesystem was unfrozen
13079e8aded4SMichael Roth          * during the freeze/thaw phase by a process other than qemu-ga.
13089e8aded4SMichael Roth          */
13099e8aded4SMichael Roth         do {
1310c216e5adSMichael Roth             ret = ioctl(fd, FITHAW);
13119e8aded4SMichael Roth             if (ret == 0 && !logged) {
1312c216e5adSMichael Roth                 i++;
13139e8aded4SMichael Roth                 logged = true;
13149e8aded4SMichael Roth             }
13159e8aded4SMichael Roth         } while (ret == 0);
13169e8aded4SMichael Roth         close(fd);
1317c216e5adSMichael Roth     }
1318c216e5adSMichael Roth 
1319f22d85e9SMichael Roth     ga_unset_frozen(ga_state);
1320af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1321ec0f694cSTomoki Sekiyama 
132277dbc81bSMarkus Armbruster     execute_fsfreeze_hook(FSFREEZE_HOOK_THAW, errp);
1323ec0f694cSTomoki Sekiyama 
1324c216e5adSMichael Roth     return i;
1325c216e5adSMichael Roth }
1326c216e5adSMichael Roth 
1327c216e5adSMichael Roth static void guest_fsfreeze_cleanup(void)
1328c216e5adSMichael Roth {
1329c216e5adSMichael Roth     Error *err = NULL;
1330c216e5adSMichael Roth 
1331f22d85e9SMichael Roth     if (ga_is_frozen(ga_state) == GUEST_FSFREEZE_STATUS_FROZEN) {
13326f686749SMarkus Armbruster         qmp_guest_fsfreeze_thaw(&err);
13336f686749SMarkus Armbruster         if (err) {
13346f686749SMarkus Armbruster             slog("failed to clean up frozen filesystems: %s",
13356f686749SMarkus Armbruster                  error_get_pretty(err));
13366f686749SMarkus Armbruster             error_free(err);
1337c216e5adSMichael Roth         }
1338c216e5adSMichael Roth     }
1339c216e5adSMichael Roth }
1340e72c3f2eSMichael Roth #endif /* CONFIG_FSFREEZE */
1341c216e5adSMichael Roth 
1342eab5fd59SPaolo Bonzini #if defined(CONFIG_FSTRIM)
1343eab5fd59SPaolo Bonzini /*
1344eab5fd59SPaolo Bonzini  * Walk list of mounted file systems in the guest, and trim them.
1345eab5fd59SPaolo Bonzini  */
1346e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
1347e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
1348eab5fd59SPaolo Bonzini {
1349e82855d9SJustin Ossevoort     GuestFilesystemTrimResponse *response;
1350e82855d9SJustin Ossevoort     GuestFilesystemTrimResultList *list;
1351e82855d9SJustin Ossevoort     GuestFilesystemTrimResult *result;
1352eab5fd59SPaolo Bonzini     int ret = 0;
1353eab5fd59SPaolo Bonzini     FsMountList mounts;
1354eab5fd59SPaolo Bonzini     struct FsMount *mount;
1355eab5fd59SPaolo Bonzini     int fd;
1356261551d1SLuiz Capitulino     Error *local_err = NULL;
135773a652a1SJustin Ossevoort     struct fstrim_range r;
1358eab5fd59SPaolo Bonzini 
1359eab5fd59SPaolo Bonzini     slog("guest-fstrim called");
1360eab5fd59SPaolo Bonzini 
1361eab5fd59SPaolo Bonzini     QTAILQ_INIT(&mounts);
1362261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
136384d18f06SMarkus Armbruster     if (local_err) {
136477dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1365e82855d9SJustin Ossevoort         return NULL;
1366eab5fd59SPaolo Bonzini     }
1367eab5fd59SPaolo Bonzini 
1368e82855d9SJustin Ossevoort     response = g_malloc0(sizeof(*response));
1369e82855d9SJustin Ossevoort 
1370eab5fd59SPaolo Bonzini     QTAILQ_FOREACH(mount, &mounts, next) {
1371e82855d9SJustin Ossevoort         result = g_malloc0(sizeof(*result));
1372e82855d9SJustin Ossevoort         result->path = g_strdup(mount->dirname);
1373e82855d9SJustin Ossevoort 
1374e82855d9SJustin Ossevoort         list = g_malloc0(sizeof(*list));
1375e82855d9SJustin Ossevoort         list->value = result;
1376e82855d9SJustin Ossevoort         list->next = response->paths;
1377e82855d9SJustin Ossevoort         response->paths = list;
1378e82855d9SJustin Ossevoort 
1379eab5fd59SPaolo Bonzini         fd = qemu_open(mount->dirname, O_RDONLY);
1380eab5fd59SPaolo Bonzini         if (fd == -1) {
1381e82855d9SJustin Ossevoort             result->error = g_strdup_printf("failed to open: %s",
1382e82855d9SJustin Ossevoort                                             strerror(errno));
1383e82855d9SJustin Ossevoort             result->has_error = true;
1384e82855d9SJustin Ossevoort             continue;
1385eab5fd59SPaolo Bonzini         }
1386eab5fd59SPaolo Bonzini 
1387eab5fd59SPaolo Bonzini         /* We try to cull filesytems we know won't work in advance, but other
1388eab5fd59SPaolo Bonzini          * filesytems may not implement fstrim for less obvious reasons.  These
1389e82855d9SJustin Ossevoort          * will report EOPNOTSUPP; while in some other cases ENOTTY will be
1390e82855d9SJustin Ossevoort          * reported (e.g. CD-ROMs).
1391e82855d9SJustin Ossevoort          * Any other error means an unexpected error.
1392eab5fd59SPaolo Bonzini          */
139373a652a1SJustin Ossevoort         r.start = 0;
139473a652a1SJustin Ossevoort         r.len = -1;
139573a652a1SJustin Ossevoort         r.minlen = has_minimum ? minimum : 0;
1396eab5fd59SPaolo Bonzini         ret = ioctl(fd, FITRIM, &r);
1397eab5fd59SPaolo Bonzini         if (ret == -1) {
1398e82855d9SJustin Ossevoort             result->has_error = true;
1399e82855d9SJustin Ossevoort             if (errno == ENOTTY || errno == EOPNOTSUPP) {
1400e82855d9SJustin Ossevoort                 result->error = g_strdup("trim not supported");
1401e82855d9SJustin Ossevoort             } else {
1402e82855d9SJustin Ossevoort                 result->error = g_strdup_printf("failed to trim: %s",
1403e82855d9SJustin Ossevoort                                                 strerror(errno));
1404e82855d9SJustin Ossevoort             }
1405eab5fd59SPaolo Bonzini             close(fd);
1406e82855d9SJustin Ossevoort             continue;
1407eab5fd59SPaolo Bonzini         }
1408e82855d9SJustin Ossevoort 
1409e82855d9SJustin Ossevoort         result->has_minimum = true;
1410e82855d9SJustin Ossevoort         result->minimum = r.minlen;
1411e82855d9SJustin Ossevoort         result->has_trimmed = true;
1412e82855d9SJustin Ossevoort         result->trimmed = r.len;
1413eab5fd59SPaolo Bonzini         close(fd);
1414eab5fd59SPaolo Bonzini     }
1415eab5fd59SPaolo Bonzini 
1416eab5fd59SPaolo Bonzini     free_fs_mount_list(&mounts);
1417e82855d9SJustin Ossevoort     return response;
1418eab5fd59SPaolo Bonzini }
1419eab5fd59SPaolo Bonzini #endif /* CONFIG_FSTRIM */
1420eab5fd59SPaolo Bonzini 
1421eab5fd59SPaolo Bonzini 
142211d0f125SLuiz Capitulino #define LINUX_SYS_STATE_FILE "/sys/power/state"
142311d0f125SLuiz Capitulino #define SUSPEND_SUPPORTED 0
142411d0f125SLuiz Capitulino #define SUSPEND_NOT_SUPPORTED 1
142511d0f125SLuiz Capitulino 
142611d0f125SLuiz Capitulino static void bios_supports_mode(const char *pmutils_bin, const char *pmutils_arg,
142777dbc81bSMarkus Armbruster                                const char *sysfile_str, Error **errp)
142811d0f125SLuiz Capitulino {
14296b26e837SLuiz Capitulino     Error *local_err = NULL;
143011d0f125SLuiz Capitulino     char *pmutils_path;
14316b26e837SLuiz Capitulino     pid_t pid;
1432dc8764f0SLuiz Capitulino     int status;
143311d0f125SLuiz Capitulino 
143411d0f125SLuiz Capitulino     pmutils_path = g_find_program_in_path(pmutils_bin);
143511d0f125SLuiz Capitulino 
143611d0f125SLuiz Capitulino     pid = fork();
143711d0f125SLuiz Capitulino     if (!pid) {
1438dc8764f0SLuiz Capitulino         char buf[32]; /* hopefully big enough */
1439dc8764f0SLuiz Capitulino         ssize_t ret;
1440dc8764f0SLuiz Capitulino         int fd;
144111d0f125SLuiz Capitulino 
144211d0f125SLuiz Capitulino         setsid();
144311d0f125SLuiz Capitulino         reopen_fd_to_null(0);
144411d0f125SLuiz Capitulino         reopen_fd_to_null(1);
144511d0f125SLuiz Capitulino         reopen_fd_to_null(2);
144611d0f125SLuiz Capitulino 
144711d0f125SLuiz Capitulino         if (pmutils_path) {
144811d0f125SLuiz Capitulino             execle(pmutils_path, pmutils_bin, pmutils_arg, NULL, environ);
144911d0f125SLuiz Capitulino         }
145011d0f125SLuiz Capitulino 
145111d0f125SLuiz Capitulino         /*
145211d0f125SLuiz Capitulino          * If we get here either pm-utils is not installed or execle() has
145311d0f125SLuiz Capitulino          * failed. Let's try the manual method if the caller wants it.
145411d0f125SLuiz Capitulino          */
145511d0f125SLuiz Capitulino 
145611d0f125SLuiz Capitulino         if (!sysfile_str) {
145711d0f125SLuiz Capitulino             _exit(SUSPEND_NOT_SUPPORTED);
145811d0f125SLuiz Capitulino         }
145911d0f125SLuiz Capitulino 
146011d0f125SLuiz Capitulino         fd = open(LINUX_SYS_STATE_FILE, O_RDONLY);
146111d0f125SLuiz Capitulino         if (fd < 0) {
146211d0f125SLuiz Capitulino             _exit(SUSPEND_NOT_SUPPORTED);
146311d0f125SLuiz Capitulino         }
146411d0f125SLuiz Capitulino 
146511d0f125SLuiz Capitulino         ret = read(fd, buf, sizeof(buf)-1);
146611d0f125SLuiz Capitulino         if (ret <= 0) {
146711d0f125SLuiz Capitulino             _exit(SUSPEND_NOT_SUPPORTED);
146811d0f125SLuiz Capitulino         }
146911d0f125SLuiz Capitulino         buf[ret] = '\0';
147011d0f125SLuiz Capitulino 
147111d0f125SLuiz Capitulino         if (strstr(buf, sysfile_str)) {
147211d0f125SLuiz Capitulino             _exit(SUSPEND_SUPPORTED);
147311d0f125SLuiz Capitulino         }
147411d0f125SLuiz Capitulino 
147511d0f125SLuiz Capitulino         _exit(SUSPEND_NOT_SUPPORTED);
14766b26e837SLuiz Capitulino     } else if (pid < 0) {
147777dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
14786b26e837SLuiz Capitulino         goto out;
147911d0f125SLuiz Capitulino     }
148011d0f125SLuiz Capitulino 
14816b26e837SLuiz Capitulino     ga_wait_child(pid, &status, &local_err);
148284d18f06SMarkus Armbruster     if (local_err) {
148377dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
14846b26e837SLuiz Capitulino         goto out;
148511d0f125SLuiz Capitulino     }
148611d0f125SLuiz Capitulino 
14876b26e837SLuiz Capitulino     if (!WIFEXITED(status)) {
148877dbc81bSMarkus Armbruster         error_setg(errp, "child process has terminated abnormally");
14896b26e837SLuiz Capitulino         goto out;
14906b26e837SLuiz Capitulino     }
14916b26e837SLuiz Capitulino 
1492dc8764f0SLuiz Capitulino     switch (WEXITSTATUS(status)) {
1493dc8764f0SLuiz Capitulino     case SUSPEND_SUPPORTED:
14946b26e837SLuiz Capitulino         goto out;
1495dc8764f0SLuiz Capitulino     case SUSPEND_NOT_SUPPORTED:
149677dbc81bSMarkus Armbruster         error_setg(errp,
14976b26e837SLuiz Capitulino                    "the requested suspend mode is not supported by the guest");
14986b26e837SLuiz Capitulino         goto out;
1499dc8764f0SLuiz Capitulino     default:
150077dbc81bSMarkus Armbruster         error_setg(errp,
15016b26e837SLuiz Capitulino                    "the helper program '%s' returned an unexpected exit status"
15026b26e837SLuiz Capitulino                    " code (%d)", pmutils_path, WEXITSTATUS(status));
15036b26e837SLuiz Capitulino         goto out;
1504dc8764f0SLuiz Capitulino     }
150511d0f125SLuiz Capitulino 
15066b26e837SLuiz Capitulino out:
15076b26e837SLuiz Capitulino     g_free(pmutils_path);
150811d0f125SLuiz Capitulino }
150911d0f125SLuiz Capitulino 
151011d0f125SLuiz Capitulino static void guest_suspend(const char *pmutils_bin, const char *sysfile_str,
151177dbc81bSMarkus Armbruster                           Error **errp)
151211d0f125SLuiz Capitulino {
15137b376087SLuiz Capitulino     Error *local_err = NULL;
151411d0f125SLuiz Capitulino     char *pmutils_path;
15157b376087SLuiz Capitulino     pid_t pid;
1516dc8764f0SLuiz Capitulino     int status;
151711d0f125SLuiz Capitulino 
151811d0f125SLuiz Capitulino     pmutils_path = g_find_program_in_path(pmutils_bin);
151911d0f125SLuiz Capitulino 
152011d0f125SLuiz Capitulino     pid = fork();
152111d0f125SLuiz Capitulino     if (pid == 0) {
152211d0f125SLuiz Capitulino         /* child */
152311d0f125SLuiz Capitulino         int fd;
152411d0f125SLuiz Capitulino 
152511d0f125SLuiz Capitulino         setsid();
152611d0f125SLuiz Capitulino         reopen_fd_to_null(0);
152711d0f125SLuiz Capitulino         reopen_fd_to_null(1);
152811d0f125SLuiz Capitulino         reopen_fd_to_null(2);
152911d0f125SLuiz Capitulino 
153011d0f125SLuiz Capitulino         if (pmutils_path) {
153111d0f125SLuiz Capitulino             execle(pmutils_path, pmutils_bin, NULL, environ);
153211d0f125SLuiz Capitulino         }
153311d0f125SLuiz Capitulino 
153411d0f125SLuiz Capitulino         /*
153511d0f125SLuiz Capitulino          * If we get here either pm-utils is not installed or execle() has
153611d0f125SLuiz Capitulino          * failed. Let's try the manual method if the caller wants it.
153711d0f125SLuiz Capitulino          */
153811d0f125SLuiz Capitulino 
153911d0f125SLuiz Capitulino         if (!sysfile_str) {
154011d0f125SLuiz Capitulino             _exit(EXIT_FAILURE);
154111d0f125SLuiz Capitulino         }
154211d0f125SLuiz Capitulino 
154311d0f125SLuiz Capitulino         fd = open(LINUX_SYS_STATE_FILE, O_WRONLY);
154411d0f125SLuiz Capitulino         if (fd < 0) {
154511d0f125SLuiz Capitulino             _exit(EXIT_FAILURE);
154611d0f125SLuiz Capitulino         }
154711d0f125SLuiz Capitulino 
154811d0f125SLuiz Capitulino         if (write(fd, sysfile_str, strlen(sysfile_str)) < 0) {
154911d0f125SLuiz Capitulino             _exit(EXIT_FAILURE);
155011d0f125SLuiz Capitulino         }
155111d0f125SLuiz Capitulino 
155211d0f125SLuiz Capitulino         _exit(EXIT_SUCCESS);
15537b376087SLuiz Capitulino     } else if (pid < 0) {
155477dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
15557b376087SLuiz Capitulino         goto out;
155611d0f125SLuiz Capitulino     }
155711d0f125SLuiz Capitulino 
15587b376087SLuiz Capitulino     ga_wait_child(pid, &status, &local_err);
155984d18f06SMarkus Armbruster     if (local_err) {
156077dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
15617b376087SLuiz Capitulino         goto out;
15627b376087SLuiz Capitulino     }
15637b376087SLuiz Capitulino 
15647b376087SLuiz Capitulino     if (!WIFEXITED(status)) {
156577dbc81bSMarkus Armbruster         error_setg(errp, "child process has terminated abnormally");
15667b376087SLuiz Capitulino         goto out;
15677b376087SLuiz Capitulino     }
15687b376087SLuiz Capitulino 
15697b376087SLuiz Capitulino     if (WEXITSTATUS(status)) {
157077dbc81bSMarkus Armbruster         error_setg(errp, "child process has failed to suspend");
15717b376087SLuiz Capitulino         goto out;
15727b376087SLuiz Capitulino     }
15737b376087SLuiz Capitulino 
15747b376087SLuiz Capitulino out:
157511d0f125SLuiz Capitulino     g_free(pmutils_path);
157611d0f125SLuiz Capitulino }
157711d0f125SLuiz Capitulino 
157877dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
157911d0f125SLuiz Capitulino {
15800f230bf7SMarkus Armbruster     Error *local_err = NULL;
15810f230bf7SMarkus Armbruster 
15820f230bf7SMarkus Armbruster     bios_supports_mode("pm-is-supported", "--hibernate", "disk", &local_err);
15830f230bf7SMarkus Armbruster     if (local_err) {
15840f230bf7SMarkus Armbruster         error_propagate(errp, local_err);
158511d0f125SLuiz Capitulino         return;
158611d0f125SLuiz Capitulino     }
158711d0f125SLuiz Capitulino 
158877dbc81bSMarkus Armbruster     guest_suspend("pm-hibernate", "disk", errp);
158911d0f125SLuiz Capitulino }
159011d0f125SLuiz Capitulino 
159177dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
1592fbf42210SLuiz Capitulino {
15930f230bf7SMarkus Armbruster     Error *local_err = NULL;
15940f230bf7SMarkus Armbruster 
15950f230bf7SMarkus Armbruster     bios_supports_mode("pm-is-supported", "--suspend", "mem", &local_err);
15960f230bf7SMarkus Armbruster     if (local_err) {
15970f230bf7SMarkus Armbruster         error_propagate(errp, local_err);
1598fbf42210SLuiz Capitulino         return;
1599fbf42210SLuiz Capitulino     }
1600fbf42210SLuiz Capitulino 
160177dbc81bSMarkus Armbruster     guest_suspend("pm-suspend", "mem", errp);
1602fbf42210SLuiz Capitulino }
1603fbf42210SLuiz Capitulino 
160477dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
160595f4f404SLuiz Capitulino {
16060f230bf7SMarkus Armbruster     Error *local_err = NULL;
16070f230bf7SMarkus Armbruster 
16080f230bf7SMarkus Armbruster     bios_supports_mode("pm-is-supported", "--suspend-hybrid", NULL,
16090f230bf7SMarkus Armbruster                        &local_err);
16100f230bf7SMarkus Armbruster     if (local_err) {
16110f230bf7SMarkus Armbruster         error_propagate(errp, local_err);
161295f4f404SLuiz Capitulino         return;
161395f4f404SLuiz Capitulino     }
161495f4f404SLuiz Capitulino 
161577dbc81bSMarkus Armbruster     guest_suspend("pm-suspend-hybrid", NULL, errp);
161695f4f404SLuiz Capitulino }
161795f4f404SLuiz Capitulino 
16183424fc9fSMichal Privoznik static GuestNetworkInterfaceList *
16193424fc9fSMichal Privoznik guest_find_interface(GuestNetworkInterfaceList *head,
16203424fc9fSMichal Privoznik                      const char *name)
16213424fc9fSMichal Privoznik {
16223424fc9fSMichal Privoznik     for (; head; head = head->next) {
16233424fc9fSMichal Privoznik         if (strcmp(head->value->name, name) == 0) {
16243424fc9fSMichal Privoznik             break;
16253424fc9fSMichal Privoznik         }
16263424fc9fSMichal Privoznik     }
16273424fc9fSMichal Privoznik 
16283424fc9fSMichal Privoznik     return head;
16293424fc9fSMichal Privoznik }
16303424fc9fSMichal Privoznik 
16313424fc9fSMichal Privoznik /*
16323424fc9fSMichal Privoznik  * Build information about guest interfaces
16333424fc9fSMichal Privoznik  */
16343424fc9fSMichal Privoznik GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
16353424fc9fSMichal Privoznik {
16363424fc9fSMichal Privoznik     GuestNetworkInterfaceList *head = NULL, *cur_item = NULL;
16373424fc9fSMichal Privoznik     struct ifaddrs *ifap, *ifa;
16383424fc9fSMichal Privoznik 
16393424fc9fSMichal Privoznik     if (getifaddrs(&ifap) < 0) {
1640878a0ae0SLuiz Capitulino         error_setg_errno(errp, errno, "getifaddrs failed");
16413424fc9fSMichal Privoznik         goto error;
16423424fc9fSMichal Privoznik     }
16433424fc9fSMichal Privoznik 
16443424fc9fSMichal Privoznik     for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
16453424fc9fSMichal Privoznik         GuestNetworkInterfaceList *info;
16463424fc9fSMichal Privoznik         GuestIpAddressList **address_list = NULL, *address_item = NULL;
16473424fc9fSMichal Privoznik         char addr4[INET_ADDRSTRLEN];
16483424fc9fSMichal Privoznik         char addr6[INET6_ADDRSTRLEN];
16493424fc9fSMichal Privoznik         int sock;
16503424fc9fSMichal Privoznik         struct ifreq ifr;
16513424fc9fSMichal Privoznik         unsigned char *mac_addr;
16523424fc9fSMichal Privoznik         void *p;
16533424fc9fSMichal Privoznik 
16543424fc9fSMichal Privoznik         g_debug("Processing %s interface", ifa->ifa_name);
16553424fc9fSMichal Privoznik 
16563424fc9fSMichal Privoznik         info = guest_find_interface(head, ifa->ifa_name);
16573424fc9fSMichal Privoznik 
16583424fc9fSMichal Privoznik         if (!info) {
16593424fc9fSMichal Privoznik             info = g_malloc0(sizeof(*info));
16603424fc9fSMichal Privoznik             info->value = g_malloc0(sizeof(*info->value));
16613424fc9fSMichal Privoznik             info->value->name = g_strdup(ifa->ifa_name);
16623424fc9fSMichal Privoznik 
16633424fc9fSMichal Privoznik             if (!cur_item) {
16643424fc9fSMichal Privoznik                 head = cur_item = info;
16653424fc9fSMichal Privoznik             } else {
16663424fc9fSMichal Privoznik                 cur_item->next = info;
16673424fc9fSMichal Privoznik                 cur_item = info;
16683424fc9fSMichal Privoznik             }
16693424fc9fSMichal Privoznik         }
16703424fc9fSMichal Privoznik 
16713424fc9fSMichal Privoznik         if (!info->value->has_hardware_address &&
16723424fc9fSMichal Privoznik             ifa->ifa_flags & SIOCGIFHWADDR) {
16733424fc9fSMichal Privoznik             /* we haven't obtained HW address yet */
16743424fc9fSMichal Privoznik             sock = socket(PF_INET, SOCK_STREAM, 0);
16753424fc9fSMichal Privoznik             if (sock == -1) {
1676878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "failed to create socket");
16773424fc9fSMichal Privoznik                 goto error;
16783424fc9fSMichal Privoznik             }
16793424fc9fSMichal Privoznik 
16803424fc9fSMichal Privoznik             memset(&ifr, 0, sizeof(ifr));
16811ab516edSJim Meyering             pstrcpy(ifr.ifr_name, IF_NAMESIZE, info->value->name);
16823424fc9fSMichal Privoznik             if (ioctl(sock, SIOCGIFHWADDR, &ifr) == -1) {
1683878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno,
1684878a0ae0SLuiz Capitulino                                  "failed to get MAC address of %s",
1685878a0ae0SLuiz Capitulino                                  ifa->ifa_name);
168610a2158fSMarkus Armbruster                 close(sock);
16873424fc9fSMichal Privoznik                 goto error;
16883424fc9fSMichal Privoznik             }
16893424fc9fSMichal Privoznik 
169010a2158fSMarkus Armbruster             close(sock);
16913424fc9fSMichal Privoznik             mac_addr = (unsigned char *) &ifr.ifr_hwaddr.sa_data;
16923424fc9fSMichal Privoznik 
1693e4ada482SStefan Weil             info->value->hardware_address =
1694e4ada482SStefan Weil                 g_strdup_printf("%02x:%02x:%02x:%02x:%02x:%02x",
16953424fc9fSMichal Privoznik                                 (int) mac_addr[0], (int) mac_addr[1],
16963424fc9fSMichal Privoznik                                 (int) mac_addr[2], (int) mac_addr[3],
1697e4ada482SStefan Weil                                 (int) mac_addr[4], (int) mac_addr[5]);
16983424fc9fSMichal Privoznik 
16993424fc9fSMichal Privoznik             info->value->has_hardware_address = true;
17003424fc9fSMichal Privoznik         }
17013424fc9fSMichal Privoznik 
17023424fc9fSMichal Privoznik         if (ifa->ifa_addr &&
17033424fc9fSMichal Privoznik             ifa->ifa_addr->sa_family == AF_INET) {
17043424fc9fSMichal Privoznik             /* interface with IPv4 address */
17053424fc9fSMichal Privoznik             p = &((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
17063424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET, p, addr4, sizeof(addr4))) {
1707878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
17083424fc9fSMichal Privoznik                 goto error;
17093424fc9fSMichal Privoznik             }
17103424fc9fSMichal Privoznik 
171110a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
171210a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
17133424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr4);
17143424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV4;
17153424fc9fSMichal Privoznik 
17163424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
17173424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
17183424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
17193424fc9fSMichal Privoznik                 p = &((struct sockaddr_in *)ifa->ifa_netmask)->sin_addr;
17203424fc9fSMichal Privoznik                 address_item->value->prefix = ctpop32(((uint32_t *) p)[0]);
17213424fc9fSMichal Privoznik             }
17223424fc9fSMichal Privoznik         } else if (ifa->ifa_addr &&
17233424fc9fSMichal Privoznik                    ifa->ifa_addr->sa_family == AF_INET6) {
17243424fc9fSMichal Privoznik             /* interface with IPv6 address */
17253424fc9fSMichal Privoznik             p = &((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
17263424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET6, p, addr6, sizeof(addr6))) {
1727878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
17283424fc9fSMichal Privoznik                 goto error;
17293424fc9fSMichal Privoznik             }
17303424fc9fSMichal Privoznik 
173110a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
173210a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
17333424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr6);
17343424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV6;
17353424fc9fSMichal Privoznik 
17363424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
17373424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
17383424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
17393424fc9fSMichal Privoznik                 p = &((struct sockaddr_in6 *)ifa->ifa_netmask)->sin6_addr;
17403424fc9fSMichal Privoznik                 address_item->value->prefix =
17413424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[0]) +
17423424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[1]) +
17433424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[2]) +
17443424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[3]);
17453424fc9fSMichal Privoznik             }
17463424fc9fSMichal Privoznik         }
17473424fc9fSMichal Privoznik 
17483424fc9fSMichal Privoznik         if (!address_item) {
17493424fc9fSMichal Privoznik             continue;
17503424fc9fSMichal Privoznik         }
17513424fc9fSMichal Privoznik 
17523424fc9fSMichal Privoznik         address_list = &info->value->ip_addresses;
17533424fc9fSMichal Privoznik 
17543424fc9fSMichal Privoznik         while (*address_list && (*address_list)->next) {
17553424fc9fSMichal Privoznik             address_list = &(*address_list)->next;
17563424fc9fSMichal Privoznik         }
17573424fc9fSMichal Privoznik 
17583424fc9fSMichal Privoznik         if (!*address_list) {
17593424fc9fSMichal Privoznik             *address_list = address_item;
17603424fc9fSMichal Privoznik         } else {
17613424fc9fSMichal Privoznik             (*address_list)->next = address_item;
17623424fc9fSMichal Privoznik         }
17633424fc9fSMichal Privoznik 
17643424fc9fSMichal Privoznik         info->value->has_ip_addresses = true;
17653424fc9fSMichal Privoznik 
17663424fc9fSMichal Privoznik 
17673424fc9fSMichal Privoznik     }
17683424fc9fSMichal Privoznik 
17693424fc9fSMichal Privoznik     freeifaddrs(ifap);
17703424fc9fSMichal Privoznik     return head;
17713424fc9fSMichal Privoznik 
17723424fc9fSMichal Privoznik error:
17733424fc9fSMichal Privoznik     freeifaddrs(ifap);
17743424fc9fSMichal Privoznik     qapi_free_GuestNetworkInterfaceList(head);
17753424fc9fSMichal Privoznik     return NULL;
17763424fc9fSMichal Privoznik }
17773424fc9fSMichal Privoznik 
177877dbc81bSMarkus Armbruster #define SYSCONF_EXACT(name, errp) sysconf_exact((name), #name, (errp))
1779d2baff62SLaszlo Ersek 
178077dbc81bSMarkus Armbruster static long sysconf_exact(int name, const char *name_str, Error **errp)
1781d2baff62SLaszlo Ersek {
1782d2baff62SLaszlo Ersek     long ret;
1783d2baff62SLaszlo Ersek 
1784d2baff62SLaszlo Ersek     errno = 0;
1785d2baff62SLaszlo Ersek     ret = sysconf(name);
1786d2baff62SLaszlo Ersek     if (ret == -1) {
1787d2baff62SLaszlo Ersek         if (errno == 0) {
178877dbc81bSMarkus Armbruster             error_setg(errp, "sysconf(%s): value indefinite", name_str);
1789d2baff62SLaszlo Ersek         } else {
179077dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "sysconf(%s)", name_str);
1791d2baff62SLaszlo Ersek         }
1792d2baff62SLaszlo Ersek     }
1793d2baff62SLaszlo Ersek     return ret;
1794d2baff62SLaszlo Ersek }
1795d2baff62SLaszlo Ersek 
1796d2baff62SLaszlo Ersek /* Transfer online/offline status between @vcpu and the guest system.
1797d2baff62SLaszlo Ersek  *
1798d2baff62SLaszlo Ersek  * On input either @errp or *@errp must be NULL.
1799d2baff62SLaszlo Ersek  *
1800d2baff62SLaszlo Ersek  * In system-to-@vcpu direction, the following @vcpu fields are accessed:
1801d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
1802d2baff62SLaszlo Ersek  * - W: vcpu->online
1803d2baff62SLaszlo Ersek  * - W: vcpu->can_offline
1804d2baff62SLaszlo Ersek  *
1805d2baff62SLaszlo Ersek  * In @vcpu-to-system direction, the following @vcpu fields are accessed:
1806d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
1807d2baff62SLaszlo Ersek  * - R: vcpu->online
1808d2baff62SLaszlo Ersek  *
1809d2baff62SLaszlo Ersek  * Written members remain unmodified on error.
1810d2baff62SLaszlo Ersek  */
1811d2baff62SLaszlo Ersek static void transfer_vcpu(GuestLogicalProcessor *vcpu, bool sys2vcpu,
1812d2baff62SLaszlo Ersek                           Error **errp)
1813d2baff62SLaszlo Ersek {
1814d2baff62SLaszlo Ersek     char *dirpath;
1815d2baff62SLaszlo Ersek     int dirfd;
1816d2baff62SLaszlo Ersek 
1817d2baff62SLaszlo Ersek     dirpath = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
1818d2baff62SLaszlo Ersek                               vcpu->logical_id);
1819d2baff62SLaszlo Ersek     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
1820d2baff62SLaszlo Ersek     if (dirfd == -1) {
1821d2baff62SLaszlo Ersek         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
1822d2baff62SLaszlo Ersek     } else {
1823d2baff62SLaszlo Ersek         static const char fn[] = "online";
1824d2baff62SLaszlo Ersek         int fd;
1825d2baff62SLaszlo Ersek         int res;
1826d2baff62SLaszlo Ersek 
1827d2baff62SLaszlo Ersek         fd = openat(dirfd, fn, sys2vcpu ? O_RDONLY : O_RDWR);
1828d2baff62SLaszlo Ersek         if (fd == -1) {
1829d2baff62SLaszlo Ersek             if (errno != ENOENT) {
1830d2baff62SLaszlo Ersek                 error_setg_errno(errp, errno, "open(\"%s/%s\")", dirpath, fn);
1831d2baff62SLaszlo Ersek             } else if (sys2vcpu) {
1832d2baff62SLaszlo Ersek                 vcpu->online = true;
1833d2baff62SLaszlo Ersek                 vcpu->can_offline = false;
1834d2baff62SLaszlo Ersek             } else if (!vcpu->online) {
1835d2baff62SLaszlo Ersek                 error_setg(errp, "logical processor #%" PRId64 " can't be "
1836d2baff62SLaszlo Ersek                            "offlined", vcpu->logical_id);
1837d2baff62SLaszlo Ersek             } /* otherwise pretend successful re-onlining */
1838d2baff62SLaszlo Ersek         } else {
1839d2baff62SLaszlo Ersek             unsigned char status;
1840d2baff62SLaszlo Ersek 
1841d2baff62SLaszlo Ersek             res = pread(fd, &status, 1, 0);
1842d2baff62SLaszlo Ersek             if (res == -1) {
1843d2baff62SLaszlo Ersek                 error_setg_errno(errp, errno, "pread(\"%s/%s\")", dirpath, fn);
1844d2baff62SLaszlo Ersek             } else if (res == 0) {
1845d2baff62SLaszlo Ersek                 error_setg(errp, "pread(\"%s/%s\"): unexpected EOF", dirpath,
1846d2baff62SLaszlo Ersek                            fn);
1847d2baff62SLaszlo Ersek             } else if (sys2vcpu) {
1848d2baff62SLaszlo Ersek                 vcpu->online = (status != '0');
1849d2baff62SLaszlo Ersek                 vcpu->can_offline = true;
1850d2baff62SLaszlo Ersek             } else if (vcpu->online != (status != '0')) {
1851d2baff62SLaszlo Ersek                 status = '0' + vcpu->online;
1852d2baff62SLaszlo Ersek                 if (pwrite(fd, &status, 1, 0) == -1) {
1853d2baff62SLaszlo Ersek                     error_setg_errno(errp, errno, "pwrite(\"%s/%s\")", dirpath,
1854d2baff62SLaszlo Ersek                                      fn);
1855d2baff62SLaszlo Ersek                 }
1856d2baff62SLaszlo Ersek             } /* otherwise pretend successful re-(on|off)-lining */
1857d2baff62SLaszlo Ersek 
1858d2baff62SLaszlo Ersek             res = close(fd);
1859d2baff62SLaszlo Ersek             g_assert(res == 0);
1860d2baff62SLaszlo Ersek         }
1861d2baff62SLaszlo Ersek 
1862d2baff62SLaszlo Ersek         res = close(dirfd);
1863d2baff62SLaszlo Ersek         g_assert(res == 0);
1864d2baff62SLaszlo Ersek     }
1865d2baff62SLaszlo Ersek 
1866d2baff62SLaszlo Ersek     g_free(dirpath);
1867d2baff62SLaszlo Ersek }
1868d2baff62SLaszlo Ersek 
1869d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
1870d2baff62SLaszlo Ersek {
1871d2baff62SLaszlo Ersek     int64_t current;
1872d2baff62SLaszlo Ersek     GuestLogicalProcessorList *head, **link;
1873d2baff62SLaszlo Ersek     long sc_max;
1874d2baff62SLaszlo Ersek     Error *local_err = NULL;
1875d2baff62SLaszlo Ersek 
1876d2baff62SLaszlo Ersek     current = 0;
1877d2baff62SLaszlo Ersek     head = NULL;
1878d2baff62SLaszlo Ersek     link = &head;
1879d2baff62SLaszlo Ersek     sc_max = SYSCONF_EXACT(_SC_NPROCESSORS_CONF, &local_err);
1880d2baff62SLaszlo Ersek 
1881d2baff62SLaszlo Ersek     while (local_err == NULL && current < sc_max) {
1882d2baff62SLaszlo Ersek         GuestLogicalProcessor *vcpu;
1883d2baff62SLaszlo Ersek         GuestLogicalProcessorList *entry;
1884d2baff62SLaszlo Ersek 
1885d2baff62SLaszlo Ersek         vcpu = g_malloc0(sizeof *vcpu);
1886d2baff62SLaszlo Ersek         vcpu->logical_id = current++;
1887d2baff62SLaszlo Ersek         vcpu->has_can_offline = true; /* lolspeak ftw */
1888d2baff62SLaszlo Ersek         transfer_vcpu(vcpu, true, &local_err);
1889d2baff62SLaszlo Ersek 
1890d2baff62SLaszlo Ersek         entry = g_malloc0(sizeof *entry);
1891d2baff62SLaszlo Ersek         entry->value = vcpu;
1892d2baff62SLaszlo Ersek 
1893d2baff62SLaszlo Ersek         *link = entry;
1894d2baff62SLaszlo Ersek         link = &entry->next;
1895d2baff62SLaszlo Ersek     }
1896d2baff62SLaszlo Ersek 
1897d2baff62SLaszlo Ersek     if (local_err == NULL) {
1898d2baff62SLaszlo Ersek         /* there's no guest with zero VCPUs */
1899d2baff62SLaszlo Ersek         g_assert(head != NULL);
1900d2baff62SLaszlo Ersek         return head;
1901d2baff62SLaszlo Ersek     }
1902d2baff62SLaszlo Ersek 
1903d2baff62SLaszlo Ersek     qapi_free_GuestLogicalProcessorList(head);
1904d2baff62SLaszlo Ersek     error_propagate(errp, local_err);
1905d2baff62SLaszlo Ersek     return NULL;
1906d2baff62SLaszlo Ersek }
1907d2baff62SLaszlo Ersek 
1908cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
1909cbb65fc2SLaszlo Ersek {
1910cbb65fc2SLaszlo Ersek     int64_t processed;
1911cbb65fc2SLaszlo Ersek     Error *local_err = NULL;
1912cbb65fc2SLaszlo Ersek 
1913cbb65fc2SLaszlo Ersek     processed = 0;
1914cbb65fc2SLaszlo Ersek     while (vcpus != NULL) {
1915cbb65fc2SLaszlo Ersek         transfer_vcpu(vcpus->value, false, &local_err);
1916cbb65fc2SLaszlo Ersek         if (local_err != NULL) {
1917cbb65fc2SLaszlo Ersek             break;
1918cbb65fc2SLaszlo Ersek         }
1919cbb65fc2SLaszlo Ersek         ++processed;
1920cbb65fc2SLaszlo Ersek         vcpus = vcpus->next;
1921cbb65fc2SLaszlo Ersek     }
1922cbb65fc2SLaszlo Ersek 
1923cbb65fc2SLaszlo Ersek     if (local_err != NULL) {
1924cbb65fc2SLaszlo Ersek         if (processed == 0) {
1925cbb65fc2SLaszlo Ersek             error_propagate(errp, local_err);
1926cbb65fc2SLaszlo Ersek         } else {
1927cbb65fc2SLaszlo Ersek             error_free(local_err);
1928cbb65fc2SLaszlo Ersek         }
1929cbb65fc2SLaszlo Ersek     }
1930cbb65fc2SLaszlo Ersek 
1931cbb65fc2SLaszlo Ersek     return processed;
1932cbb65fc2SLaszlo Ersek }
1933cbb65fc2SLaszlo Ersek 
1934215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
1935215a2771SDaniel P. Berrange                                  const char *password,
1936215a2771SDaniel P. Berrange                                  bool crypted,
1937215a2771SDaniel P. Berrange                                  Error **errp)
1938215a2771SDaniel P. Berrange {
1939215a2771SDaniel P. Berrange     Error *local_err = NULL;
1940215a2771SDaniel P. Berrange     char *passwd_path = NULL;
1941215a2771SDaniel P. Berrange     pid_t pid;
1942215a2771SDaniel P. Berrange     int status;
1943215a2771SDaniel P. Berrange     int datafd[2] = { -1, -1 };
1944215a2771SDaniel P. Berrange     char *rawpasswddata = NULL;
1945215a2771SDaniel P. Berrange     size_t rawpasswdlen;
1946215a2771SDaniel P. Berrange     char *chpasswddata = NULL;
1947215a2771SDaniel P. Berrange     size_t chpasswdlen;
1948215a2771SDaniel P. Berrange 
1949215a2771SDaniel P. Berrange     rawpasswddata = (char *)g_base64_decode(password, &rawpasswdlen);
1950215a2771SDaniel P. Berrange     rawpasswddata = g_renew(char, rawpasswddata, rawpasswdlen + 1);
1951215a2771SDaniel P. Berrange     rawpasswddata[rawpasswdlen] = '\0';
1952215a2771SDaniel P. Berrange 
1953215a2771SDaniel P. Berrange     if (strchr(rawpasswddata, '\n')) {
1954215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in raw password");
1955215a2771SDaniel P. Berrange         goto out;
1956215a2771SDaniel P. Berrange     }
1957215a2771SDaniel P. Berrange 
1958215a2771SDaniel P. Berrange     if (strchr(username, '\n') ||
1959215a2771SDaniel P. Berrange         strchr(username, ':')) {
1960215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in username");
1961215a2771SDaniel P. Berrange         goto out;
1962215a2771SDaniel P. Berrange     }
1963215a2771SDaniel P. Berrange 
1964215a2771SDaniel P. Berrange     chpasswddata = g_strdup_printf("%s:%s\n", username, rawpasswddata);
1965215a2771SDaniel P. Berrange     chpasswdlen = strlen(chpasswddata);
1966215a2771SDaniel P. Berrange 
1967215a2771SDaniel P. Berrange     passwd_path = g_find_program_in_path("chpasswd");
1968215a2771SDaniel P. Berrange 
1969215a2771SDaniel P. Berrange     if (!passwd_path) {
1970215a2771SDaniel P. Berrange         error_setg(errp, "cannot find 'passwd' program in PATH");
1971215a2771SDaniel P. Berrange         goto out;
1972215a2771SDaniel P. Berrange     }
1973215a2771SDaniel P. Berrange 
1974215a2771SDaniel P. Berrange     if (pipe(datafd) < 0) {
1975215a2771SDaniel P. Berrange         error_setg(errp, "cannot create pipe FDs");
1976215a2771SDaniel P. Berrange         goto out;
1977215a2771SDaniel P. Berrange     }
1978215a2771SDaniel P. Berrange 
1979215a2771SDaniel P. Berrange     pid = fork();
1980215a2771SDaniel P. Berrange     if (pid == 0) {
1981215a2771SDaniel P. Berrange         close(datafd[1]);
1982215a2771SDaniel P. Berrange         /* child */
1983215a2771SDaniel P. Berrange         setsid();
1984215a2771SDaniel P. Berrange         dup2(datafd[0], 0);
1985215a2771SDaniel P. Berrange         reopen_fd_to_null(1);
1986215a2771SDaniel P. Berrange         reopen_fd_to_null(2);
1987215a2771SDaniel P. Berrange 
1988215a2771SDaniel P. Berrange         if (crypted) {
1989215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", "-e", NULL, environ);
1990215a2771SDaniel P. Berrange         } else {
1991215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", NULL, environ);
1992215a2771SDaniel P. Berrange         }
1993215a2771SDaniel P. Berrange         _exit(EXIT_FAILURE);
1994215a2771SDaniel P. Berrange     } else if (pid < 0) {
1995215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "failed to create child process");
1996215a2771SDaniel P. Berrange         goto out;
1997215a2771SDaniel P. Berrange     }
1998215a2771SDaniel P. Berrange     close(datafd[0]);
1999215a2771SDaniel P. Berrange     datafd[0] = -1;
2000215a2771SDaniel P. Berrange 
2001215a2771SDaniel P. Berrange     if (qemu_write_full(datafd[1], chpasswddata, chpasswdlen) != chpasswdlen) {
2002215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "cannot write new account password");
2003215a2771SDaniel P. Berrange         goto out;
2004215a2771SDaniel P. Berrange     }
2005215a2771SDaniel P. Berrange     close(datafd[1]);
2006215a2771SDaniel P. Berrange     datafd[1] = -1;
2007215a2771SDaniel P. Berrange 
2008215a2771SDaniel P. Berrange     ga_wait_child(pid, &status, &local_err);
2009215a2771SDaniel P. Berrange     if (local_err) {
2010215a2771SDaniel P. Berrange         error_propagate(errp, local_err);
2011215a2771SDaniel P. Berrange         goto out;
2012215a2771SDaniel P. Berrange     }
2013215a2771SDaniel P. Berrange 
2014215a2771SDaniel P. Berrange     if (!WIFEXITED(status)) {
2015215a2771SDaniel P. Berrange         error_setg(errp, "child process has terminated abnormally");
2016215a2771SDaniel P. Berrange         goto out;
2017215a2771SDaniel P. Berrange     }
2018215a2771SDaniel P. Berrange 
2019215a2771SDaniel P. Berrange     if (WEXITSTATUS(status)) {
2020215a2771SDaniel P. Berrange         error_setg(errp, "child process has failed to set user password");
2021215a2771SDaniel P. Berrange         goto out;
2022215a2771SDaniel P. Berrange     }
2023215a2771SDaniel P. Berrange 
2024215a2771SDaniel P. Berrange out:
2025215a2771SDaniel P. Berrange     g_free(chpasswddata);
2026215a2771SDaniel P. Berrange     g_free(rawpasswddata);
2027215a2771SDaniel P. Berrange     g_free(passwd_path);
2028215a2771SDaniel P. Berrange     if (datafd[0] != -1) {
2029215a2771SDaniel P. Berrange         close(datafd[0]);
2030215a2771SDaniel P. Berrange     }
2031215a2771SDaniel P. Berrange     if (datafd[1] != -1) {
2032215a2771SDaniel P. Berrange         close(datafd[1]);
2033215a2771SDaniel P. Berrange     }
2034215a2771SDaniel P. Berrange }
2035215a2771SDaniel P. Berrange 
2036bd240fcaSzhanghailiang static void ga_read_sysfs_file(int dirfd, const char *pathname, char *buf,
2037bd240fcaSzhanghailiang                                int size, Error **errp)
2038bd240fcaSzhanghailiang {
2039bd240fcaSzhanghailiang     int fd;
2040bd240fcaSzhanghailiang     int res;
2041bd240fcaSzhanghailiang 
2042bd240fcaSzhanghailiang     errno = 0;
2043bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_RDONLY);
2044bd240fcaSzhanghailiang     if (fd == -1) {
2045bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2046bd240fcaSzhanghailiang         return;
2047bd240fcaSzhanghailiang     }
2048bd240fcaSzhanghailiang 
2049bd240fcaSzhanghailiang     res = pread(fd, buf, size, 0);
2050bd240fcaSzhanghailiang     if (res == -1) {
2051bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pread sysfs file \"%s\"", pathname);
2052bd240fcaSzhanghailiang     } else if (res == 0) {
2053bd240fcaSzhanghailiang         error_setg(errp, "pread sysfs file \"%s\": unexpected EOF", pathname);
2054bd240fcaSzhanghailiang     }
2055bd240fcaSzhanghailiang     close(fd);
2056bd240fcaSzhanghailiang }
2057bd240fcaSzhanghailiang 
2058bd240fcaSzhanghailiang static void ga_write_sysfs_file(int dirfd, const char *pathname,
2059bd240fcaSzhanghailiang                                 const char *buf, int size, Error **errp)
2060bd240fcaSzhanghailiang {
2061bd240fcaSzhanghailiang     int fd;
2062bd240fcaSzhanghailiang 
2063bd240fcaSzhanghailiang     errno = 0;
2064bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_WRONLY);
2065bd240fcaSzhanghailiang     if (fd == -1) {
2066bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2067bd240fcaSzhanghailiang         return;
2068bd240fcaSzhanghailiang     }
2069bd240fcaSzhanghailiang 
2070bd240fcaSzhanghailiang     if (pwrite(fd, buf, size, 0) == -1) {
2071bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pwrite sysfs file \"%s\"", pathname);
2072bd240fcaSzhanghailiang     }
2073bd240fcaSzhanghailiang 
2074bd240fcaSzhanghailiang     close(fd);
2075bd240fcaSzhanghailiang }
2076bd240fcaSzhanghailiang 
2077bd240fcaSzhanghailiang /* Transfer online/offline status between @mem_blk and the guest system.
2078bd240fcaSzhanghailiang  *
2079bd240fcaSzhanghailiang  * On input either @errp or *@errp must be NULL.
2080bd240fcaSzhanghailiang  *
2081bd240fcaSzhanghailiang  * In system-to-@mem_blk direction, the following @mem_blk fields are accessed:
2082bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2083bd240fcaSzhanghailiang  * - W: mem_blk->online
2084bd240fcaSzhanghailiang  * - W: mem_blk->can_offline
2085bd240fcaSzhanghailiang  *
2086bd240fcaSzhanghailiang  * In @mem_blk-to-system direction, the following @mem_blk fields are accessed:
2087bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2088bd240fcaSzhanghailiang  * - R: mem_blk->online
2089bd240fcaSzhanghailiang  *-  R: mem_blk->can_offline
2090bd240fcaSzhanghailiang  * Written members remain unmodified on error.
2091bd240fcaSzhanghailiang  */
2092bd240fcaSzhanghailiang static void transfer_memory_block(GuestMemoryBlock *mem_blk, bool sys2memblk,
2093bd240fcaSzhanghailiang                                   GuestMemoryBlockResponse *result,
2094bd240fcaSzhanghailiang                                   Error **errp)
2095bd240fcaSzhanghailiang {
2096bd240fcaSzhanghailiang     char *dirpath;
2097bd240fcaSzhanghailiang     int dirfd;
2098bd240fcaSzhanghailiang     char *status;
2099bd240fcaSzhanghailiang     Error *local_err = NULL;
2100bd240fcaSzhanghailiang 
2101bd240fcaSzhanghailiang     if (!sys2memblk) {
2102bd240fcaSzhanghailiang         DIR *dp;
2103bd240fcaSzhanghailiang 
2104bd240fcaSzhanghailiang         if (!result) {
2105bd240fcaSzhanghailiang             error_setg(errp, "Internal error, 'result' should not be NULL");
2106bd240fcaSzhanghailiang             return;
2107bd240fcaSzhanghailiang         }
2108bd240fcaSzhanghailiang         errno = 0;
2109bd240fcaSzhanghailiang         dp = opendir("/sys/devices/system/memory/");
2110bd240fcaSzhanghailiang          /* if there is no 'memory' directory in sysfs,
2111bd240fcaSzhanghailiang          * we think this VM does not support online/offline memory block,
2112bd240fcaSzhanghailiang          * any other solution?
2113bd240fcaSzhanghailiang          */
2114bd240fcaSzhanghailiang         if (!dp && errno == ENOENT) {
2115bd240fcaSzhanghailiang             result->response =
2116bd240fcaSzhanghailiang                 GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
2117bd240fcaSzhanghailiang             goto out1;
2118bd240fcaSzhanghailiang         }
2119bd240fcaSzhanghailiang         closedir(dp);
2120bd240fcaSzhanghailiang     }
2121bd240fcaSzhanghailiang 
2122bd240fcaSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/memory%" PRId64 "/",
2123bd240fcaSzhanghailiang                               mem_blk->phys_index);
2124bd240fcaSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2125bd240fcaSzhanghailiang     if (dirfd == -1) {
2126bd240fcaSzhanghailiang         if (sys2memblk) {
2127bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2128bd240fcaSzhanghailiang         } else {
2129bd240fcaSzhanghailiang             if (errno == ENOENT) {
2130bd240fcaSzhanghailiang                 result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_NOT_FOUND;
2131bd240fcaSzhanghailiang             } else {
2132bd240fcaSzhanghailiang                 result->response =
2133bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2134bd240fcaSzhanghailiang             }
2135bd240fcaSzhanghailiang         }
2136bd240fcaSzhanghailiang         g_free(dirpath);
2137bd240fcaSzhanghailiang         goto out1;
2138bd240fcaSzhanghailiang     }
2139bd240fcaSzhanghailiang     g_free(dirpath);
2140bd240fcaSzhanghailiang 
2141bd240fcaSzhanghailiang     status = g_malloc0(10);
2142bd240fcaSzhanghailiang     ga_read_sysfs_file(dirfd, "state", status, 10, &local_err);
2143bd240fcaSzhanghailiang     if (local_err) {
2144bd240fcaSzhanghailiang         /* treat with sysfs file that not exist in old kernel */
2145bd240fcaSzhanghailiang         if (errno == ENOENT) {
2146bd240fcaSzhanghailiang             error_free(local_err);
2147bd240fcaSzhanghailiang             if (sys2memblk) {
2148bd240fcaSzhanghailiang                 mem_blk->online = true;
2149bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2150bd240fcaSzhanghailiang             } else if (!mem_blk->online) {
2151bd240fcaSzhanghailiang                 result->response =
2152bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
2153bd240fcaSzhanghailiang             }
2154bd240fcaSzhanghailiang         } else {
2155bd240fcaSzhanghailiang             if (sys2memblk) {
2156bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2157bd240fcaSzhanghailiang             } else {
2158bd240fcaSzhanghailiang                 result->response =
2159bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2160bd240fcaSzhanghailiang             }
2161bd240fcaSzhanghailiang         }
2162bd240fcaSzhanghailiang         goto out2;
2163bd240fcaSzhanghailiang     }
2164bd240fcaSzhanghailiang 
2165bd240fcaSzhanghailiang     if (sys2memblk) {
2166bd240fcaSzhanghailiang         char removable = '0';
2167bd240fcaSzhanghailiang 
2168bd240fcaSzhanghailiang         mem_blk->online = (strncmp(status, "online", 6) == 0);
2169bd240fcaSzhanghailiang 
2170bd240fcaSzhanghailiang         ga_read_sysfs_file(dirfd, "removable", &removable, 1, &local_err);
2171bd240fcaSzhanghailiang         if (local_err) {
217267cc32ebSVeres Lajos             /* if no 'removable' file, it doesn't support offline mem blk */
2173bd240fcaSzhanghailiang             if (errno == ENOENT) {
2174bd240fcaSzhanghailiang                 error_free(local_err);
2175bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2176bd240fcaSzhanghailiang             } else {
2177bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2178bd240fcaSzhanghailiang             }
2179bd240fcaSzhanghailiang         } else {
2180bd240fcaSzhanghailiang             mem_blk->can_offline = (removable != '0');
2181bd240fcaSzhanghailiang         }
2182bd240fcaSzhanghailiang     } else {
2183bd240fcaSzhanghailiang         if (mem_blk->online != (strncmp(status, "online", 6) == 0)) {
2184bd240fcaSzhanghailiang             char *new_state = mem_blk->online ? g_strdup("online") :
2185bd240fcaSzhanghailiang                                                 g_strdup("offline");
2186bd240fcaSzhanghailiang 
2187bd240fcaSzhanghailiang             ga_write_sysfs_file(dirfd, "state", new_state, strlen(new_state),
2188bd240fcaSzhanghailiang                                 &local_err);
2189bd240fcaSzhanghailiang             g_free(new_state);
2190bd240fcaSzhanghailiang             if (local_err) {
2191bd240fcaSzhanghailiang                 error_free(local_err);
2192bd240fcaSzhanghailiang                 result->response =
2193bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2194bd240fcaSzhanghailiang                 goto out2;
2195bd240fcaSzhanghailiang             }
2196bd240fcaSzhanghailiang 
2197bd240fcaSzhanghailiang             result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_SUCCESS;
2198bd240fcaSzhanghailiang             result->has_error_code = false;
2199bd240fcaSzhanghailiang         } /* otherwise pretend successful re-(on|off)-lining */
2200bd240fcaSzhanghailiang     }
2201bd240fcaSzhanghailiang     g_free(status);
2202bd240fcaSzhanghailiang     close(dirfd);
2203bd240fcaSzhanghailiang     return;
2204bd240fcaSzhanghailiang 
2205bd240fcaSzhanghailiang out2:
2206bd240fcaSzhanghailiang     g_free(status);
2207bd240fcaSzhanghailiang     close(dirfd);
2208bd240fcaSzhanghailiang out1:
2209bd240fcaSzhanghailiang     if (!sys2memblk) {
2210bd240fcaSzhanghailiang         result->has_error_code = true;
2211bd240fcaSzhanghailiang         result->error_code = errno;
2212bd240fcaSzhanghailiang     }
2213bd240fcaSzhanghailiang }
2214bd240fcaSzhanghailiang 
2215a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2216a065aaa9Szhanghailiang {
2217bd240fcaSzhanghailiang     GuestMemoryBlockList *head, **link;
2218bd240fcaSzhanghailiang     Error *local_err = NULL;
2219bd240fcaSzhanghailiang     struct dirent *de;
2220bd240fcaSzhanghailiang     DIR *dp;
2221bd240fcaSzhanghailiang 
2222bd240fcaSzhanghailiang     head = NULL;
2223bd240fcaSzhanghailiang     link = &head;
2224bd240fcaSzhanghailiang 
2225bd240fcaSzhanghailiang     dp = opendir("/sys/devices/system/memory/");
2226bd240fcaSzhanghailiang     if (!dp) {
2227f693fe6eSMichael Roth         /* it's ok if this happens to be a system that doesn't expose
2228f693fe6eSMichael Roth          * memory blocks via sysfs, but otherwise we should report
2229f693fe6eSMichael Roth          * an error
2230f693fe6eSMichael Roth          */
2231f693fe6eSMichael Roth         if (errno != ENOENT) {
2232bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "Can't open directory"
2233bd240fcaSzhanghailiang                              "\"/sys/devices/system/memory/\"\n");
2234f693fe6eSMichael Roth         }
2235bd240fcaSzhanghailiang         return NULL;
2236bd240fcaSzhanghailiang     }
2237bd240fcaSzhanghailiang 
2238bd240fcaSzhanghailiang     /* Note: the phys_index of memory block may be discontinuous,
2239bd240fcaSzhanghailiang      * this is because a memblk is the unit of the Sparse Memory design, which
2240bd240fcaSzhanghailiang      * allows discontinuous memory ranges (ex. NUMA), so here we should
2241bd240fcaSzhanghailiang      * traverse the memory block directory.
2242bd240fcaSzhanghailiang      */
2243bd240fcaSzhanghailiang     while ((de = readdir(dp)) != NULL) {
2244bd240fcaSzhanghailiang         GuestMemoryBlock *mem_blk;
2245bd240fcaSzhanghailiang         GuestMemoryBlockList *entry;
2246bd240fcaSzhanghailiang 
2247bd240fcaSzhanghailiang         if ((strncmp(de->d_name, "memory", 6) != 0) ||
2248bd240fcaSzhanghailiang             !(de->d_type & DT_DIR)) {
2249bd240fcaSzhanghailiang             continue;
2250bd240fcaSzhanghailiang         }
2251bd240fcaSzhanghailiang 
2252bd240fcaSzhanghailiang         mem_blk = g_malloc0(sizeof *mem_blk);
2253bd240fcaSzhanghailiang         /* The d_name is "memoryXXX",  phys_index is block id, same as XXX */
2254bd240fcaSzhanghailiang         mem_blk->phys_index = strtoul(&de->d_name[6], NULL, 10);
2255bd240fcaSzhanghailiang         mem_blk->has_can_offline = true; /* lolspeak ftw */
2256bd240fcaSzhanghailiang         transfer_memory_block(mem_blk, true, NULL, &local_err);
2257bd240fcaSzhanghailiang 
2258bd240fcaSzhanghailiang         entry = g_malloc0(sizeof *entry);
2259bd240fcaSzhanghailiang         entry->value = mem_blk;
2260bd240fcaSzhanghailiang 
2261bd240fcaSzhanghailiang         *link = entry;
2262bd240fcaSzhanghailiang         link = &entry->next;
2263bd240fcaSzhanghailiang     }
2264bd240fcaSzhanghailiang 
2265bd240fcaSzhanghailiang     closedir(dp);
2266bd240fcaSzhanghailiang     if (local_err == NULL) {
2267bd240fcaSzhanghailiang         /* there's no guest with zero memory blocks */
2268bd240fcaSzhanghailiang         if (head == NULL) {
2269bd240fcaSzhanghailiang             error_setg(errp, "guest reported zero memory blocks!");
2270bd240fcaSzhanghailiang         }
2271bd240fcaSzhanghailiang         return head;
2272bd240fcaSzhanghailiang     }
2273bd240fcaSzhanghailiang 
2274bd240fcaSzhanghailiang     qapi_free_GuestMemoryBlockList(head);
2275bd240fcaSzhanghailiang     error_propagate(errp, local_err);
2276a065aaa9Szhanghailiang     return NULL;
2277a065aaa9Szhanghailiang }
2278a065aaa9Szhanghailiang 
2279a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2280a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2281a065aaa9Szhanghailiang {
228232ca7927Szhanghailiang     GuestMemoryBlockResponseList *head, **link;
228332ca7927Szhanghailiang     Error *local_err = NULL;
228432ca7927Szhanghailiang 
228532ca7927Szhanghailiang     head = NULL;
228632ca7927Szhanghailiang     link = &head;
228732ca7927Szhanghailiang 
228832ca7927Szhanghailiang     while (mem_blks != NULL) {
228932ca7927Szhanghailiang         GuestMemoryBlockResponse *result;
229032ca7927Szhanghailiang         GuestMemoryBlockResponseList *entry;
229132ca7927Szhanghailiang         GuestMemoryBlock *current_mem_blk = mem_blks->value;
229232ca7927Szhanghailiang 
229332ca7927Szhanghailiang         result = g_malloc0(sizeof(*result));
229432ca7927Szhanghailiang         result->phys_index = current_mem_blk->phys_index;
229532ca7927Szhanghailiang         transfer_memory_block(current_mem_blk, false, result, &local_err);
229632ca7927Szhanghailiang         if (local_err) { /* should never happen */
229732ca7927Szhanghailiang             goto err;
229832ca7927Szhanghailiang         }
229932ca7927Szhanghailiang         entry = g_malloc0(sizeof *entry);
230032ca7927Szhanghailiang         entry->value = result;
230132ca7927Szhanghailiang 
230232ca7927Szhanghailiang         *link = entry;
230332ca7927Szhanghailiang         link = &entry->next;
230432ca7927Szhanghailiang         mem_blks = mem_blks->next;
230532ca7927Szhanghailiang     }
230632ca7927Szhanghailiang 
230732ca7927Szhanghailiang     return head;
230832ca7927Szhanghailiang err:
230932ca7927Szhanghailiang     qapi_free_GuestMemoryBlockResponseList(head);
231032ca7927Szhanghailiang     error_propagate(errp, local_err);
2311a065aaa9Szhanghailiang     return NULL;
2312a065aaa9Szhanghailiang }
2313a065aaa9Szhanghailiang 
2314a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2315a065aaa9Szhanghailiang {
2316ef82b60bSzhanghailiang     Error *local_err = NULL;
2317ef82b60bSzhanghailiang     char *dirpath;
2318ef82b60bSzhanghailiang     int dirfd;
2319ef82b60bSzhanghailiang     char *buf;
2320ef82b60bSzhanghailiang     GuestMemoryBlockInfo *info;
2321ef82b60bSzhanghailiang 
2322ef82b60bSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/");
2323ef82b60bSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2324ef82b60bSzhanghailiang     if (dirfd == -1) {
2325ef82b60bSzhanghailiang         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2326ef82b60bSzhanghailiang         g_free(dirpath);
2327a065aaa9Szhanghailiang         return NULL;
2328a065aaa9Szhanghailiang     }
2329ef82b60bSzhanghailiang     g_free(dirpath);
2330ef82b60bSzhanghailiang 
2331ef82b60bSzhanghailiang     buf = g_malloc0(20);
2332ef82b60bSzhanghailiang     ga_read_sysfs_file(dirfd, "block_size_bytes", buf, 20, &local_err);
23338ce1ee46SShannon Zhao     close(dirfd);
2334ef82b60bSzhanghailiang     if (local_err) {
2335ef82b60bSzhanghailiang         g_free(buf);
2336ef82b60bSzhanghailiang         error_propagate(errp, local_err);
2337ef82b60bSzhanghailiang         return NULL;
2338ef82b60bSzhanghailiang     }
2339ef82b60bSzhanghailiang 
2340ef82b60bSzhanghailiang     info = g_new0(GuestMemoryBlockInfo, 1);
2341ef82b60bSzhanghailiang     info->size = strtol(buf, NULL, 16); /* the unit is bytes */
2342ef82b60bSzhanghailiang 
2343ef82b60bSzhanghailiang     g_free(buf);
2344ef82b60bSzhanghailiang 
2345ef82b60bSzhanghailiang     return info;
2346ef82b60bSzhanghailiang }
2347a065aaa9Szhanghailiang 
2348e72c3f2eSMichael Roth #else /* defined(__linux__) */
2349e72c3f2eSMichael Roth 
235077dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
2351e72c3f2eSMichael Roth {
2352c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2353e72c3f2eSMichael Roth }
2354e72c3f2eSMichael Roth 
235577dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
2356e72c3f2eSMichael Roth {
2357c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2358e72c3f2eSMichael Roth }
2359e72c3f2eSMichael Roth 
236077dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
2361e72c3f2eSMichael Roth {
2362c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2363e72c3f2eSMichael Roth }
2364e72c3f2eSMichael Roth 
2365e72c3f2eSMichael Roth GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
2366e72c3f2eSMichael Roth {
2367c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2368e72c3f2eSMichael Roth     return NULL;
2369e72c3f2eSMichael Roth }
2370e72c3f2eSMichael Roth 
2371d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2372d2baff62SLaszlo Ersek {
2373c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2374d2baff62SLaszlo Ersek     return NULL;
2375d2baff62SLaszlo Ersek }
2376d2baff62SLaszlo Ersek 
2377cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2378cbb65fc2SLaszlo Ersek {
2379c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2380cbb65fc2SLaszlo Ersek     return -1;
2381cbb65fc2SLaszlo Ersek }
2382cbb65fc2SLaszlo Ersek 
2383215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2384215a2771SDaniel P. Berrange                                  const char *password,
2385215a2771SDaniel P. Berrange                                  bool crypted,
2386215a2771SDaniel P. Berrange                                  Error **errp)
2387215a2771SDaniel P. Berrange {
2388c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2389215a2771SDaniel P. Berrange }
2390215a2771SDaniel P. Berrange 
2391a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2392a065aaa9Szhanghailiang {
2393c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2394a065aaa9Szhanghailiang     return NULL;
2395a065aaa9Szhanghailiang }
2396a065aaa9Szhanghailiang 
2397a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2398a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2399a065aaa9Szhanghailiang {
2400c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2401a065aaa9Szhanghailiang     return NULL;
2402a065aaa9Szhanghailiang }
2403a065aaa9Szhanghailiang 
2404a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2405a065aaa9Szhanghailiang {
2406c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2407a065aaa9Szhanghailiang     return NULL;
2408a065aaa9Szhanghailiang }
2409a065aaa9Szhanghailiang 
2410e72c3f2eSMichael Roth #endif
2411e72c3f2eSMichael Roth 
2412d35d4cb5SMichael Roth #if !defined(CONFIG_FSFREEZE)
2413d35d4cb5SMichael Roth 
241446d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
241546d4c572STomoki Sekiyama {
2416c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
241746d4c572STomoki Sekiyama     return NULL;
241846d4c572STomoki Sekiyama }
241946d4c572STomoki Sekiyama 
242077dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
2421d35d4cb5SMichael Roth {
2422c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2423d35d4cb5SMichael Roth 
2424d35d4cb5SMichael Roth     return 0;
2425d35d4cb5SMichael Roth }
2426d35d4cb5SMichael Roth 
242777dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_freeze(Error **errp)
2428d35d4cb5SMichael Roth {
2429c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2430d35d4cb5SMichael Roth 
2431d35d4cb5SMichael Roth     return 0;
2432d35d4cb5SMichael Roth }
2433d35d4cb5SMichael Roth 
2434e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
2435e99bce20STomoki Sekiyama                                        strList *mountpoints,
2436e99bce20STomoki Sekiyama                                        Error **errp)
2437e99bce20STomoki Sekiyama {
2438c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2439e99bce20STomoki Sekiyama 
2440e99bce20STomoki Sekiyama     return 0;
2441e99bce20STomoki Sekiyama }
2442e99bce20STomoki Sekiyama 
244377dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
2444d35d4cb5SMichael Roth {
2445c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2446d35d4cb5SMichael Roth 
2447d35d4cb5SMichael Roth     return 0;
2448d35d4cb5SMichael Roth }
2449eab5fd59SPaolo Bonzini #endif /* CONFIG_FSFREEZE */
2450d35d4cb5SMichael Roth 
2451eab5fd59SPaolo Bonzini #if !defined(CONFIG_FSTRIM)
2452e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
2453e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
2454eab5fd59SPaolo Bonzini {
2455c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2456e82855d9SJustin Ossevoort     return NULL;
2457eab5fd59SPaolo Bonzini }
2458d35d4cb5SMichael Roth #endif
2459d35d4cb5SMichael Roth 
24601281c08aSTomoki Sekiyama /* add unsupported commands to the blacklist */
24611281c08aSTomoki Sekiyama GList *ga_command_blacklist_init(GList *blacklist)
24621281c08aSTomoki Sekiyama {
24631281c08aSTomoki Sekiyama #if !defined(__linux__)
24641281c08aSTomoki Sekiyama     {
24651281c08aSTomoki Sekiyama         const char *list[] = {
24661281c08aSTomoki Sekiyama             "guest-suspend-disk", "guest-suspend-ram",
24671281c08aSTomoki Sekiyama             "guest-suspend-hybrid", "guest-network-get-interfaces",
24680dd38a03Szhanghailiang             "guest-get-vcpus", "guest-set-vcpus",
24690dd38a03Szhanghailiang             "guest-get-memory-blocks", "guest-set-memory-blocks",
24700dd38a03Szhanghailiang             "guest-get-memory-block-size", NULL};
24711281c08aSTomoki Sekiyama         char **p = (char **)list;
24721281c08aSTomoki Sekiyama 
24731281c08aSTomoki Sekiyama         while (*p) {
24744bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
24751281c08aSTomoki Sekiyama         }
24761281c08aSTomoki Sekiyama     }
24771281c08aSTomoki Sekiyama #endif
24781281c08aSTomoki Sekiyama 
24791281c08aSTomoki Sekiyama #if !defined(CONFIG_FSFREEZE)
24801281c08aSTomoki Sekiyama     {
24811281c08aSTomoki Sekiyama         const char *list[] = {
24821281c08aSTomoki Sekiyama             "guest-get-fsinfo", "guest-fsfreeze-status",
24831281c08aSTomoki Sekiyama             "guest-fsfreeze-freeze", "guest-fsfreeze-freeze-list",
24841281c08aSTomoki Sekiyama             "guest-fsfreeze-thaw", "guest-get-fsinfo", NULL};
24851281c08aSTomoki Sekiyama         char **p = (char **)list;
24861281c08aSTomoki Sekiyama 
24871281c08aSTomoki Sekiyama         while (*p) {
24884bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
24891281c08aSTomoki Sekiyama         }
24901281c08aSTomoki Sekiyama     }
24911281c08aSTomoki Sekiyama #endif
24921281c08aSTomoki Sekiyama 
24931281c08aSTomoki Sekiyama #if !defined(CONFIG_FSTRIM)
24944bca81ceSMarc-André Lureau     blacklist = g_list_append(blacklist, g_strdup("guest-fstrim"));
24951281c08aSTomoki Sekiyama #endif
24961281c08aSTomoki Sekiyama 
24971281c08aSTomoki Sekiyama     return blacklist;
24981281c08aSTomoki Sekiyama }
24991281c08aSTomoki Sekiyama 
2500c216e5adSMichael Roth /* register init/cleanup routines for stateful command groups */
2501c216e5adSMichael Roth void ga_command_state_init(GAState *s, GACommandState *cs)
2502c216e5adSMichael Roth {
2503c216e5adSMichael Roth #if defined(CONFIG_FSFREEZE)
2504f22d85e9SMichael Roth     ga_command_state_add(cs, NULL, guest_fsfreeze_cleanup);
2505c216e5adSMichael Roth #endif
2506c216e5adSMichael Roth }
2507