xref: /openbmc/qemu/qga/commands-posix.c (revision d9fe4f0f)
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 
144459bf38SPeter Maydell #include "qemu/osdep.h"
15c216e5adSMichael Roth #include <sys/ioctl.h>
169848f797STomáš Golembiovský #include <sys/utsname.h>
172c02cbf6SLuiz Capitulino #include <sys/wait.h>
1846d4c572STomoki Sekiyama #include <dirent.h>
19a8d25326SMarkus Armbruster #include "qemu-common.h"
20dc03272dSMichael S. Tsirkin #include "guest-agent-core.h"
21eb815e24SMarkus Armbruster #include "qga-qapi-commands.h"
22e688df6bSMarkus Armbruster #include "qapi/error.h"
237b1b5d19SPaolo Bonzini #include "qapi/qmp/qerror.h"
241de7afc9SPaolo Bonzini #include "qemu/queue.h"
251de7afc9SPaolo Bonzini #include "qemu/host-utils.h"
2612505396SDenis V. Lunev #include "qemu/sockets.h"
27920639caSDaniel P. Berrange #include "qemu/base64.h"
28f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
295d3586b8SPhilippe Mathieu-Daudé #include "commands-common.h"
30c216e5adSMichael Roth 
31e674605fSTomáš Golembiovský #ifdef HAVE_UTMPX
32e674605fSTomáš Golembiovský #include <utmpx.h>
33e674605fSTomáš Golembiovský #endif
34e674605fSTomáš Golembiovský 
352c02cbf6SLuiz Capitulino #ifndef CONFIG_HAS_ENVIRON
36eecae147SAndreas Färber #ifdef __APPLE__
37eecae147SAndreas Färber #include <crt_externs.h>
38eecae147SAndreas Färber #define environ (*_NSGetEnviron())
39eecae147SAndreas Färber #else
402c02cbf6SLuiz Capitulino extern char **environ;
412c02cbf6SLuiz Capitulino #endif
42eecae147SAndreas Färber #endif
432c02cbf6SLuiz Capitulino 
44e72c3f2eSMichael Roth #if defined(__linux__)
45e72c3f2eSMichael Roth #include <mntent.h>
46e72c3f2eSMichael Roth #include <linux/fs.h>
47e72c3f2eSMichael Roth #include <ifaddrs.h>
48e72c3f2eSMichael Roth #include <arpa/inet.h>
49e72c3f2eSMichael Roth #include <sys/socket.h>
50e72c3f2eSMichael Roth #include <net/if.h>
5125b5ff1aSChen Hanxiao #include <sys/statvfs.h>
52e72c3f2eSMichael Roth 
53b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
54b616105aSTomáš Golembiovský #include <libudev.h>
55b616105aSTomáš Golembiovský #endif
56b616105aSTomáš Golembiovský 
57eab5fd59SPaolo Bonzini #ifdef FIFREEZE
58e72c3f2eSMichael Roth #define CONFIG_FSFREEZE
59e72c3f2eSMichael Roth #endif
60eab5fd59SPaolo Bonzini #ifdef FITRIM
61eab5fd59SPaolo Bonzini #define CONFIG_FSTRIM
62eab5fd59SPaolo Bonzini #endif
63e72c3f2eSMichael Roth #endif
64e72c3f2eSMichael Roth 
6577dbc81bSMarkus Armbruster static void ga_wait_child(pid_t pid, int *status, Error **errp)
66d220a6dfSLuiz Capitulino {
67d220a6dfSLuiz Capitulino     pid_t rpid;
68d220a6dfSLuiz Capitulino 
69d220a6dfSLuiz Capitulino     *status = 0;
70d220a6dfSLuiz Capitulino 
71d220a6dfSLuiz Capitulino     do {
72d220a6dfSLuiz Capitulino         rpid = waitpid(pid, status, 0);
73d220a6dfSLuiz Capitulino     } while (rpid == -1 && errno == EINTR);
74d220a6dfSLuiz Capitulino 
75d220a6dfSLuiz Capitulino     if (rpid == -1) {
7677dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to wait for child (pid: %d)",
7777dbc81bSMarkus Armbruster                          pid);
78d220a6dfSLuiz Capitulino         return;
79d220a6dfSLuiz Capitulino     }
80d220a6dfSLuiz Capitulino 
81d220a6dfSLuiz Capitulino     g_assert(rpid == pid);
82d220a6dfSLuiz Capitulino }
83d220a6dfSLuiz Capitulino 
8477dbc81bSMarkus Armbruster void qmp_guest_shutdown(bool has_mode, const char *mode, Error **errp)
85c216e5adSMichael Roth {
86c216e5adSMichael Roth     const char *shutdown_flag;
87d220a6dfSLuiz Capitulino     Error *local_err = NULL;
88d220a6dfSLuiz Capitulino     pid_t pid;
893674838cSLuiz Capitulino     int status;
90c216e5adSMichael Roth 
91c216e5adSMichael Roth     slog("guest-shutdown called, mode: %s", mode);
92c216e5adSMichael Roth     if (!has_mode || strcmp(mode, "powerdown") == 0) {
93c216e5adSMichael Roth         shutdown_flag = "-P";
94c216e5adSMichael Roth     } else if (strcmp(mode, "halt") == 0) {
95c216e5adSMichael Roth         shutdown_flag = "-H";
96c216e5adSMichael Roth     } else if (strcmp(mode, "reboot") == 0) {
97c216e5adSMichael Roth         shutdown_flag = "-r";
98c216e5adSMichael Roth     } else {
9977dbc81bSMarkus Armbruster         error_setg(errp,
100d220a6dfSLuiz Capitulino                    "mode is invalid (valid values are: halt|powerdown|reboot");
101c216e5adSMichael Roth         return;
102c216e5adSMichael Roth     }
103c216e5adSMichael Roth 
104d5dd3498SLuiz Capitulino     pid = fork();
105d5dd3498SLuiz Capitulino     if (pid == 0) {
106c216e5adSMichael Roth         /* child, start the shutdown */
107c216e5adSMichael Roth         setsid();
1083674838cSLuiz Capitulino         reopen_fd_to_null(0);
1093674838cSLuiz Capitulino         reopen_fd_to_null(1);
1103674838cSLuiz Capitulino         reopen_fd_to_null(2);
111c216e5adSMichael Roth 
112485e741cSwhitearchey         execle("/sbin/shutdown", "shutdown", "-h", shutdown_flag, "+0",
1133674838cSLuiz Capitulino                "hypervisor initiated shutdown", (char*)NULL, environ);
1143674838cSLuiz Capitulino         _exit(EXIT_FAILURE);
115d5dd3498SLuiz Capitulino     } else if (pid < 0) {
11677dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
117d5dd3498SLuiz Capitulino         return;
118d5dd3498SLuiz Capitulino     }
119d5dd3498SLuiz Capitulino 
120d220a6dfSLuiz Capitulino     ga_wait_child(pid, &status, &local_err);
12184d18f06SMarkus Armbruster     if (local_err) {
12277dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
123d220a6dfSLuiz Capitulino         return;
124d220a6dfSLuiz Capitulino     }
125d220a6dfSLuiz Capitulino 
126d220a6dfSLuiz Capitulino     if (!WIFEXITED(status)) {
12777dbc81bSMarkus Armbruster         error_setg(errp, "child process has terminated abnormally");
128d220a6dfSLuiz Capitulino         return;
129d220a6dfSLuiz Capitulino     }
130d220a6dfSLuiz Capitulino 
131d220a6dfSLuiz Capitulino     if (WEXITSTATUS(status)) {
13277dbc81bSMarkus Armbruster         error_setg(errp, "child process has failed to shutdown");
133d220a6dfSLuiz Capitulino         return;
134d220a6dfSLuiz Capitulino     }
135d220a6dfSLuiz Capitulino 
136085d8134SPeter Maydell     /* succeeded */
137c216e5adSMichael Roth }
138c216e5adSMichael Roth 
1396912e6a9SLei Li int64_t qmp_guest_get_time(Error **errp)
1406912e6a9SLei Li {
1416912e6a9SLei Li    int ret;
1426912e6a9SLei Li    qemu_timeval tq;
1436912e6a9SLei Li 
1446912e6a9SLei Li    ret = qemu_gettimeofday(&tq);
1456912e6a9SLei Li    if (ret < 0) {
1466912e6a9SLei Li        error_setg_errno(errp, errno, "Failed to get time");
1476912e6a9SLei Li        return -1;
1486912e6a9SLei Li    }
1496912e6a9SLei Li 
1509be38598SEduardo Habkost    return tq.tv_sec * 1000000000LL + tq.tv_usec * 1000;
1516912e6a9SLei Li }
1526912e6a9SLei Li 
1532c958923SMichal Privoznik void qmp_guest_set_time(bool has_time, int64_t time_ns, Error **errp)
154a1bca57fSLei Li {
155a1bca57fSLei Li     int ret;
156a1bca57fSLei Li     int status;
157a1bca57fSLei Li     pid_t pid;
158a1bca57fSLei Li     Error *local_err = NULL;
159a1bca57fSLei Li     struct timeval tv;
1605c6096e5SCornelia Huck     static const char hwclock_path[] = "/sbin/hwclock";
1615c6096e5SCornelia Huck     static int hwclock_available = -1;
1625c6096e5SCornelia Huck 
1635c6096e5SCornelia Huck     if (hwclock_available < 0) {
1645c6096e5SCornelia Huck         hwclock_available = (access(hwclock_path, X_OK) == 0);
1655c6096e5SCornelia Huck     }
1665c6096e5SCornelia Huck 
1675c6096e5SCornelia Huck     if (!hwclock_available) {
1685c6096e5SCornelia Huck         error_setg(errp, QERR_UNSUPPORTED);
1695c6096e5SCornelia Huck         return;
1705c6096e5SCornelia Huck     }
171a1bca57fSLei Li 
1722c958923SMichal Privoznik     /* If user has passed a time, validate and set it. */
1732c958923SMichal Privoznik     if (has_time) {
17400d2f370SMarc-André Lureau         GDate date = { 0, };
17500d2f370SMarc-André Lureau 
176a1bca57fSLei Li         /* year-2038 will overflow in case time_t is 32bit */
177a1bca57fSLei Li         if (time_ns / 1000000000 != (time_t)(time_ns / 1000000000)) {
178a1bca57fSLei Li             error_setg(errp, "Time %" PRId64 " is too large", time_ns);
179a1bca57fSLei Li             return;
180a1bca57fSLei Li         }
181a1bca57fSLei Li 
182a1bca57fSLei Li         tv.tv_sec = time_ns / 1000000000;
183a1bca57fSLei Li         tv.tv_usec = (time_ns % 1000000000) / 1000;
18400d2f370SMarc-André Lureau         g_date_set_time_t(&date, tv.tv_sec);
18500d2f370SMarc-André Lureau         if (date.year < 1970 || date.year >= 2070) {
18600d2f370SMarc-André Lureau             error_setg_errno(errp, errno, "Invalid time");
18700d2f370SMarc-André Lureau             return;
18800d2f370SMarc-André Lureau         }
189a1bca57fSLei Li 
190a1bca57fSLei Li         ret = settimeofday(&tv, NULL);
191a1bca57fSLei Li         if (ret < 0) {
192a1bca57fSLei Li             error_setg_errno(errp, errno, "Failed to set time to guest");
193a1bca57fSLei Li             return;
194a1bca57fSLei Li         }
1952c958923SMichal Privoznik     }
196a1bca57fSLei Li 
1972c958923SMichal Privoznik     /* Now, if user has passed a time to set and the system time is set, we
1982c958923SMichal Privoznik      * just need to synchronize the hardware clock. However, if no time was
1992c958923SMichal Privoznik      * passed, user is requesting the opposite: set the system time from the
2001634df56SAmos Kong      * hardware clock (RTC). */
201a1bca57fSLei Li     pid = fork();
202a1bca57fSLei Li     if (pid == 0) {
203a1bca57fSLei Li         setsid();
204a1bca57fSLei Li         reopen_fd_to_null(0);
205a1bca57fSLei Li         reopen_fd_to_null(1);
206a1bca57fSLei Li         reopen_fd_to_null(2);
207a1bca57fSLei Li 
2082c958923SMichal Privoznik         /* Use '/sbin/hwclock -w' to set RTC from the system time,
2092c958923SMichal Privoznik          * or '/sbin/hwclock -s' to set the system time from RTC. */
2105c6096e5SCornelia Huck         execle(hwclock_path, "hwclock", has_time ? "-w" : "-s",
2112c958923SMichal Privoznik                NULL, environ);
212a1bca57fSLei Li         _exit(EXIT_FAILURE);
213a1bca57fSLei Li     } else if (pid < 0) {
214a1bca57fSLei Li         error_setg_errno(errp, errno, "failed to create child process");
215a1bca57fSLei Li         return;
216a1bca57fSLei Li     }
217a1bca57fSLei Li 
218a1bca57fSLei Li     ga_wait_child(pid, &status, &local_err);
21984d18f06SMarkus Armbruster     if (local_err) {
220a1bca57fSLei Li         error_propagate(errp, local_err);
221a1bca57fSLei Li         return;
222a1bca57fSLei Li     }
223a1bca57fSLei Li 
224a1bca57fSLei Li     if (!WIFEXITED(status)) {
225a1bca57fSLei Li         error_setg(errp, "child process has terminated abnormally");
226a1bca57fSLei Li         return;
227a1bca57fSLei Li     }
228a1bca57fSLei Li 
229a1bca57fSLei Li     if (WEXITSTATUS(status)) {
230a1bca57fSLei Li         error_setg(errp, "hwclock failed to set hardware clock to system time");
231a1bca57fSLei Li         return;
232a1bca57fSLei Li     }
233a1bca57fSLei Li }
234a1bca57fSLei Li 
235895b00f6SMarc-André Lureau typedef enum {
236895b00f6SMarc-André Lureau     RW_STATE_NEW,
237895b00f6SMarc-André Lureau     RW_STATE_READING,
238895b00f6SMarc-André Lureau     RW_STATE_WRITING,
239895b00f6SMarc-André Lureau } RwState;
240895b00f6SMarc-André Lureau 
2415d3586b8SPhilippe Mathieu-Daudé struct GuestFileHandle {
242c216e5adSMichael Roth     uint64_t id;
243c216e5adSMichael Roth     FILE *fh;
244895b00f6SMarc-André Lureau     RwState state;
245c216e5adSMichael Roth     QTAILQ_ENTRY(GuestFileHandle) next;
2465d3586b8SPhilippe Mathieu-Daudé };
247c216e5adSMichael Roth 
248c216e5adSMichael Roth static struct {
249c216e5adSMichael Roth     QTAILQ_HEAD(, GuestFileHandle) filehandles;
250b4fe97c8SDenis V. Lunev } guest_file_state = {
251b4fe97c8SDenis V. Lunev     .filehandles = QTAILQ_HEAD_INITIALIZER(guest_file_state.filehandles),
252b4fe97c8SDenis V. Lunev };
253c216e5adSMichael Roth 
25439097dafSMichael Roth static int64_t guest_file_handle_add(FILE *fh, Error **errp)
255c216e5adSMichael Roth {
256c216e5adSMichael Roth     GuestFileHandle *gfh;
25739097dafSMichael Roth     int64_t handle;
25839097dafSMichael Roth 
25939097dafSMichael Roth     handle = ga_get_fd_handle(ga_state, errp);
260a903f40cSMarkus Armbruster     if (handle < 0) {
261a903f40cSMarkus Armbruster         return -1;
26239097dafSMichael Roth     }
263c216e5adSMichael Roth 
264f3a06403SMarkus Armbruster     gfh = g_new0(GuestFileHandle, 1);
26539097dafSMichael Roth     gfh->id = handle;
266c216e5adSMichael Roth     gfh->fh = fh;
267c216e5adSMichael Roth     QTAILQ_INSERT_TAIL(&guest_file_state.filehandles, gfh, next);
26839097dafSMichael Roth 
26939097dafSMichael Roth     return handle;
270c216e5adSMichael Roth }
271c216e5adSMichael Roth 
2725d3586b8SPhilippe Mathieu-Daudé GuestFileHandle *guest_file_handle_find(int64_t id, Error **errp)
273c216e5adSMichael Roth {
274c216e5adSMichael Roth     GuestFileHandle *gfh;
275c216e5adSMichael Roth 
276c216e5adSMichael Roth     QTAILQ_FOREACH(gfh, &guest_file_state.filehandles, next)
277c216e5adSMichael Roth     {
278c216e5adSMichael Roth         if (gfh->id == id) {
279c216e5adSMichael Roth             return gfh;
280c216e5adSMichael Roth         }
281c216e5adSMichael Roth     }
282c216e5adSMichael Roth 
28377dbc81bSMarkus Armbruster     error_setg(errp, "handle '%" PRId64 "' has not been found", id);
284c216e5adSMichael Roth     return NULL;
285c216e5adSMichael Roth }
286c216e5adSMichael Roth 
287c689b4f1SLaszlo Ersek typedef const char * const ccpc;
288c689b4f1SLaszlo Ersek 
2898fe6bbcaSLaszlo Ersek #ifndef O_BINARY
2908fe6bbcaSLaszlo Ersek #define O_BINARY 0
2918fe6bbcaSLaszlo Ersek #endif
2928fe6bbcaSLaszlo Ersek 
293c689b4f1SLaszlo Ersek /* http://pubs.opengroup.org/onlinepubs/9699919799/functions/fopen.html */
294c689b4f1SLaszlo Ersek static const struct {
295c689b4f1SLaszlo Ersek     ccpc *forms;
296c689b4f1SLaszlo Ersek     int oflag_base;
297c689b4f1SLaszlo Ersek } guest_file_open_modes[] = {
2988fe6bbcaSLaszlo Ersek     { (ccpc[]){ "r",          NULL }, O_RDONLY                                 },
2998fe6bbcaSLaszlo Ersek     { (ccpc[]){ "rb",         NULL }, O_RDONLY                      | O_BINARY },
3008fe6bbcaSLaszlo Ersek     { (ccpc[]){ "w",          NULL }, O_WRONLY | O_CREAT | O_TRUNC             },
3018fe6bbcaSLaszlo Ersek     { (ccpc[]){ "wb",         NULL }, O_WRONLY | O_CREAT | O_TRUNC  | O_BINARY },
3028fe6bbcaSLaszlo Ersek     { (ccpc[]){ "a",          NULL }, O_WRONLY | O_CREAT | O_APPEND            },
3038fe6bbcaSLaszlo Ersek     { (ccpc[]){ "ab",         NULL }, O_WRONLY | O_CREAT | O_APPEND | O_BINARY },
3048fe6bbcaSLaszlo Ersek     { (ccpc[]){ "r+",         NULL }, O_RDWR                                   },
3058fe6bbcaSLaszlo Ersek     { (ccpc[]){ "rb+", "r+b", NULL }, O_RDWR                        | O_BINARY },
3068fe6bbcaSLaszlo Ersek     { (ccpc[]){ "w+",         NULL }, O_RDWR   | O_CREAT | O_TRUNC             },
3078fe6bbcaSLaszlo Ersek     { (ccpc[]){ "wb+", "w+b", NULL }, O_RDWR   | O_CREAT | O_TRUNC  | O_BINARY },
3088fe6bbcaSLaszlo Ersek     { (ccpc[]){ "a+",         NULL }, O_RDWR   | O_CREAT | O_APPEND            },
3098fe6bbcaSLaszlo Ersek     { (ccpc[]){ "ab+", "a+b", NULL }, O_RDWR   | O_CREAT | O_APPEND | O_BINARY }
310c689b4f1SLaszlo Ersek };
311c689b4f1SLaszlo Ersek 
312c689b4f1SLaszlo Ersek static int
31377dbc81bSMarkus Armbruster find_open_flag(const char *mode_str, Error **errp)
314c689b4f1SLaszlo Ersek {
315c689b4f1SLaszlo Ersek     unsigned mode;
316c689b4f1SLaszlo Ersek 
317c689b4f1SLaszlo Ersek     for (mode = 0; mode < ARRAY_SIZE(guest_file_open_modes); ++mode) {
318c689b4f1SLaszlo Ersek         ccpc *form;
319c689b4f1SLaszlo Ersek 
320c689b4f1SLaszlo Ersek         form = guest_file_open_modes[mode].forms;
321c689b4f1SLaszlo Ersek         while (*form != NULL && strcmp(*form, mode_str) != 0) {
322c689b4f1SLaszlo Ersek             ++form;
323c689b4f1SLaszlo Ersek         }
324c689b4f1SLaszlo Ersek         if (*form != NULL) {
325c689b4f1SLaszlo Ersek             break;
326c689b4f1SLaszlo Ersek         }
327c689b4f1SLaszlo Ersek     }
328c689b4f1SLaszlo Ersek 
329c689b4f1SLaszlo Ersek     if (mode == ARRAY_SIZE(guest_file_open_modes)) {
33077dbc81bSMarkus Armbruster         error_setg(errp, "invalid file open mode '%s'", mode_str);
331c689b4f1SLaszlo Ersek         return -1;
332c689b4f1SLaszlo Ersek     }
333c689b4f1SLaszlo Ersek     return guest_file_open_modes[mode].oflag_base | O_NOCTTY | O_NONBLOCK;
334c689b4f1SLaszlo Ersek }
335c689b4f1SLaszlo Ersek 
336c689b4f1SLaszlo Ersek #define DEFAULT_NEW_FILE_MODE (S_IRUSR | S_IWUSR | \
337c689b4f1SLaszlo Ersek                                S_IRGRP | S_IWGRP | \
338c689b4f1SLaszlo Ersek                                S_IROTH | S_IWOTH)
339c689b4f1SLaszlo Ersek 
340c689b4f1SLaszlo Ersek static FILE *
34177dbc81bSMarkus Armbruster safe_open_or_create(const char *path, const char *mode, Error **errp)
342c689b4f1SLaszlo Ersek {
343c689b4f1SLaszlo Ersek     Error *local_err = NULL;
344c689b4f1SLaszlo Ersek     int oflag;
345c689b4f1SLaszlo Ersek 
346c689b4f1SLaszlo Ersek     oflag = find_open_flag(mode, &local_err);
347c689b4f1SLaszlo Ersek     if (local_err == NULL) {
348c689b4f1SLaszlo Ersek         int fd;
349c689b4f1SLaszlo Ersek 
350c689b4f1SLaszlo Ersek         /* If the caller wants / allows creation of a new file, we implement it
351c689b4f1SLaszlo Ersek          * with a two step process: open() + (open() / fchmod()).
352c689b4f1SLaszlo Ersek          *
353c689b4f1SLaszlo Ersek          * First we insist on creating the file exclusively as a new file. If
354c689b4f1SLaszlo Ersek          * that succeeds, we're free to set any file-mode bits on it. (The
355c689b4f1SLaszlo Ersek          * motivation is that we want to set those file-mode bits independently
356c689b4f1SLaszlo Ersek          * of the current umask.)
357c689b4f1SLaszlo Ersek          *
358c689b4f1SLaszlo Ersek          * If the exclusive creation fails because the file already exists
359c689b4f1SLaszlo Ersek          * (EEXIST is not possible for any other reason), we just attempt to
360c689b4f1SLaszlo Ersek          * open the file, but in this case we won't be allowed to change the
361c689b4f1SLaszlo Ersek          * file-mode bits on the preexistent file.
362c689b4f1SLaszlo Ersek          *
363c689b4f1SLaszlo Ersek          * The pathname should never disappear between the two open()s in
364c689b4f1SLaszlo Ersek          * practice. If it happens, then someone very likely tried to race us.
365c689b4f1SLaszlo Ersek          * In this case just go ahead and report the ENOENT from the second
366c689b4f1SLaszlo Ersek          * open() to the caller.
367c689b4f1SLaszlo Ersek          *
368c689b4f1SLaszlo Ersek          * If the caller wants to open a preexistent file, then the first
369c689b4f1SLaszlo Ersek          * open() is decisive and its third argument is ignored, and the second
370c689b4f1SLaszlo Ersek          * open() and the fchmod() are never called.
371c689b4f1SLaszlo Ersek          */
372c689b4f1SLaszlo Ersek         fd = open(path, oflag | ((oflag & O_CREAT) ? O_EXCL : 0), 0);
373c689b4f1SLaszlo Ersek         if (fd == -1 && errno == EEXIST) {
374c689b4f1SLaszlo Ersek             oflag &= ~(unsigned)O_CREAT;
375c689b4f1SLaszlo Ersek             fd = open(path, oflag);
376c689b4f1SLaszlo Ersek         }
377c689b4f1SLaszlo Ersek 
378c689b4f1SLaszlo Ersek         if (fd == -1) {
379c689b4f1SLaszlo Ersek             error_setg_errno(&local_err, errno, "failed to open file '%s' "
380c689b4f1SLaszlo Ersek                              "(mode: '%s')", path, mode);
381c689b4f1SLaszlo Ersek         } else {
382c689b4f1SLaszlo Ersek             qemu_set_cloexec(fd);
383c689b4f1SLaszlo Ersek 
384c689b4f1SLaszlo Ersek             if ((oflag & O_CREAT) && fchmod(fd, DEFAULT_NEW_FILE_MODE) == -1) {
385c689b4f1SLaszlo Ersek                 error_setg_errno(&local_err, errno, "failed to set permission "
386c689b4f1SLaszlo Ersek                                  "0%03o on new file '%s' (mode: '%s')",
387c689b4f1SLaszlo Ersek                                  (unsigned)DEFAULT_NEW_FILE_MODE, path, mode);
388c689b4f1SLaszlo Ersek             } else {
389c689b4f1SLaszlo Ersek                 FILE *f;
390c689b4f1SLaszlo Ersek 
391c689b4f1SLaszlo Ersek                 f = fdopen(fd, mode);
392c689b4f1SLaszlo Ersek                 if (f == NULL) {
393c689b4f1SLaszlo Ersek                     error_setg_errno(&local_err, errno, "failed to associate "
394c689b4f1SLaszlo Ersek                                      "stdio stream with file descriptor %d, "
395c689b4f1SLaszlo Ersek                                      "file '%s' (mode: '%s')", fd, path, mode);
396c689b4f1SLaszlo Ersek                 } else {
397c689b4f1SLaszlo Ersek                     return f;
398c689b4f1SLaszlo Ersek                 }
399c689b4f1SLaszlo Ersek             }
400c689b4f1SLaszlo Ersek 
401c689b4f1SLaszlo Ersek             close(fd);
4022b720018SLaszlo Ersek             if (oflag & O_CREAT) {
4032b720018SLaszlo Ersek                 unlink(path);
4042b720018SLaszlo Ersek             }
405c689b4f1SLaszlo Ersek         }
406c689b4f1SLaszlo Ersek     }
407c689b4f1SLaszlo Ersek 
40877dbc81bSMarkus Armbruster     error_propagate(errp, local_err);
409c689b4f1SLaszlo Ersek     return NULL;
410c689b4f1SLaszlo Ersek }
411c689b4f1SLaszlo Ersek 
41277dbc81bSMarkus Armbruster int64_t qmp_guest_file_open(const char *path, bool has_mode, const char *mode,
41377dbc81bSMarkus Armbruster                             Error **errp)
414c216e5adSMichael Roth {
415c216e5adSMichael Roth     FILE *fh;
416c689b4f1SLaszlo Ersek     Error *local_err = NULL;
41785b6f6f5SSimon Zolin     int64_t handle;
418c216e5adSMichael Roth 
419c216e5adSMichael Roth     if (!has_mode) {
420c216e5adSMichael Roth         mode = "r";
421c216e5adSMichael Roth     }
422c216e5adSMichael Roth     slog("guest-file-open called, filepath: %s, mode: %s", path, mode);
423c689b4f1SLaszlo Ersek     fh = safe_open_or_create(path, mode, &local_err);
424c689b4f1SLaszlo Ersek     if (local_err != NULL) {
42577dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
426c216e5adSMichael Roth         return -1;
427c216e5adSMichael Roth     }
428c216e5adSMichael Roth 
429c216e5adSMichael Roth     /* set fd non-blocking to avoid common use cases (like reading from a
430c216e5adSMichael Roth      * named pipe) from hanging the agent
431c216e5adSMichael Roth      */
43212505396SDenis V. Lunev     qemu_set_nonblock(fileno(fh));
433c216e5adSMichael Roth 
43477dbc81bSMarkus Armbruster     handle = guest_file_handle_add(fh, errp);
435a903f40cSMarkus Armbruster     if (handle < 0) {
43639097dafSMichael Roth         fclose(fh);
43739097dafSMichael Roth         return -1;
43839097dafSMichael Roth     }
43939097dafSMichael Roth 
440d607a523SStefan Weil     slog("guest-file-open, handle: %" PRId64, handle);
44139097dafSMichael Roth     return handle;
442c216e5adSMichael Roth }
443c216e5adSMichael Roth 
44477dbc81bSMarkus Armbruster void qmp_guest_file_close(int64_t handle, Error **errp)
445c216e5adSMichael Roth {
44677dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
447c216e5adSMichael Roth     int ret;
448c216e5adSMichael Roth 
449d607a523SStefan Weil     slog("guest-file-close called, handle: %" PRId64, handle);
450c216e5adSMichael Roth     if (!gfh) {
451c216e5adSMichael Roth         return;
452c216e5adSMichael Roth     }
453c216e5adSMichael Roth 
454c216e5adSMichael Roth     ret = fclose(gfh->fh);
4553ac4b7c5SLuiz Capitulino     if (ret == EOF) {
45677dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to close handle");
457c216e5adSMichael Roth         return;
458c216e5adSMichael Roth     }
459c216e5adSMichael Roth 
460c216e5adSMichael Roth     QTAILQ_REMOVE(&guest_file_state.filehandles, gfh, next);
461c216e5adSMichael Roth     g_free(gfh);
462c216e5adSMichael Roth }
463c216e5adSMichael Roth 
464ead83a13SPhilippe Mathieu-Daudé GuestFileRead *guest_file_read_unsafe(GuestFileHandle *gfh,
46577dbc81bSMarkus Armbruster                                       int64_t count, Error **errp)
466c216e5adSMichael Roth {
467c216e5adSMichael Roth     GuestFileRead *read_data = NULL;
468c216e5adSMichael Roth     guchar *buf;
469ead83a13SPhilippe Mathieu-Daudé     FILE *fh = gfh->fh;
470c216e5adSMichael Roth     size_t read_count;
471c216e5adSMichael Roth 
472895b00f6SMarc-André Lureau     /* explicitly flush when switching from writing to reading */
473895b00f6SMarc-André Lureau     if (gfh->state == RW_STATE_WRITING) {
474895b00f6SMarc-André Lureau         int ret = fflush(fh);
475895b00f6SMarc-André Lureau         if (ret == EOF) {
476895b00f6SMarc-André Lureau             error_setg_errno(errp, errno, "failed to flush file");
477895b00f6SMarc-André Lureau             return NULL;
478895b00f6SMarc-André Lureau         }
479895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
480895b00f6SMarc-André Lureau     }
481895b00f6SMarc-André Lureau 
482c216e5adSMichael Roth     buf = g_malloc0(count+1);
483c216e5adSMichael Roth     read_count = fread(buf, 1, count, fh);
484c216e5adSMichael Roth     if (ferror(fh)) {
48577dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to read file");
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         }
494895b00f6SMarc-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;
518895b00f6SMarc-André Lureau 
519895b00f6SMarc-André Lureau     if (gfh->state == RW_STATE_READING) {
520895b00f6SMarc-André Lureau         int ret = fseek(fh, 0, SEEK_CUR);
521895b00f6SMarc-André Lureau         if (ret == -1) {
522895b00f6SMarc-André Lureau             error_setg_errno(errp, errno, "failed to seek file");
523895b00f6SMarc-André Lureau             return NULL;
524895b00f6SMarc-André Lureau         }
525895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
526895b00f6SMarc-André Lureau     }
527895b00f6SMarc-André Lureau 
528920639caSDaniel P. Berrange     buf = qbase64_decode(buf_b64, -1, &buf_len, errp);
529920639caSDaniel P. Berrange     if (!buf) {
530920639caSDaniel P. Berrange         return NULL;
531920639caSDaniel P. Berrange     }
532c216e5adSMichael Roth 
533c216e5adSMichael Roth     if (!has_count) {
534c216e5adSMichael Roth         count = buf_len;
535c216e5adSMichael Roth     } else if (count < 0 || count > buf_len) {
53677dbc81bSMarkus Armbruster         error_setg(errp, "value '%" PRId64 "' is invalid for argument count",
537db3edb66SLuiz Capitulino                    count);
538c216e5adSMichael Roth         g_free(buf);
539c216e5adSMichael Roth         return NULL;
540c216e5adSMichael Roth     }
541c216e5adSMichael Roth 
542c216e5adSMichael Roth     write_count = fwrite(buf, 1, count, fh);
543c216e5adSMichael Roth     if (ferror(fh)) {
54477dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to write to file");
545d607a523SStefan Weil         slog("guest-file-write failed, handle: %" PRId64, handle);
546c216e5adSMichael Roth     } else {
547f3a06403SMarkus Armbruster         write_data = g_new0(GuestFileWrite, 1);
548c216e5adSMichael Roth         write_data->count = write_count;
549c216e5adSMichael Roth         write_data->eof = feof(fh);
550895b00f6SMarc-André Lureau         gfh->state = RW_STATE_WRITING;
551c216e5adSMichael Roth     }
552c216e5adSMichael Roth     g_free(buf);
553c216e5adSMichael Roth     clearerr(fh);
554c216e5adSMichael Roth 
555c216e5adSMichael Roth     return write_data;
556c216e5adSMichael Roth }
557c216e5adSMichael Roth 
558c216e5adSMichael Roth struct GuestFileSeek *qmp_guest_file_seek(int64_t handle, int64_t offset,
5590b4b4938SEric Blake                                           GuestFileWhence *whence_code,
5600b4b4938SEric Blake                                           Error **errp)
561c216e5adSMichael Roth {
56277dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
563c216e5adSMichael Roth     GuestFileSeek *seek_data = NULL;
564c216e5adSMichael Roth     FILE *fh;
565c216e5adSMichael Roth     int ret;
5660a982b1bSEric Blake     int whence;
5670b4b4938SEric Blake     Error *err = NULL;
568c216e5adSMichael Roth 
569c216e5adSMichael Roth     if (!gfh) {
570c216e5adSMichael Roth         return NULL;
571c216e5adSMichael Roth     }
572c216e5adSMichael Roth 
5730a982b1bSEric Blake     /* We stupidly exposed 'whence':'int' in our qapi */
5740b4b4938SEric Blake     whence = ga_parse_whence(whence_code, &err);
5750b4b4938SEric Blake     if (err) {
5760b4b4938SEric Blake         error_propagate(errp, err);
5770a982b1bSEric Blake         return NULL;
5780a982b1bSEric Blake     }
5790a982b1bSEric Blake 
580c216e5adSMichael Roth     fh = gfh->fh;
581c216e5adSMichael Roth     ret = fseek(fh, offset, whence);
582c216e5adSMichael Roth     if (ret == -1) {
58377dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to seek file");
584895b00f6SMarc-André Lureau         if (errno == ESPIPE) {
585895b00f6SMarc-André Lureau             /* file is non-seekable, stdio shouldn't be buffering anyways */
586895b00f6SMarc-André Lureau             gfh->state = RW_STATE_NEW;
587895b00f6SMarc-André Lureau         }
588c216e5adSMichael Roth     } else {
58910b7c5ddSMarkus Armbruster         seek_data = g_new0(GuestFileSeek, 1);
590c216e5adSMichael Roth         seek_data->position = ftell(fh);
591c216e5adSMichael Roth         seek_data->eof = feof(fh);
592895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
593c216e5adSMichael Roth     }
594c216e5adSMichael Roth     clearerr(fh);
595c216e5adSMichael Roth 
596c216e5adSMichael Roth     return seek_data;
597c216e5adSMichael Roth }
598c216e5adSMichael Roth 
59977dbc81bSMarkus Armbruster void qmp_guest_file_flush(int64_t handle, Error **errp)
600c216e5adSMichael Roth {
60177dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
602c216e5adSMichael Roth     FILE *fh;
603c216e5adSMichael Roth     int ret;
604c216e5adSMichael Roth 
605c216e5adSMichael Roth     if (!gfh) {
606c216e5adSMichael Roth         return;
607c216e5adSMichael Roth     }
608c216e5adSMichael Roth 
609c216e5adSMichael Roth     fh = gfh->fh;
610c216e5adSMichael Roth     ret = fflush(fh);
611c216e5adSMichael Roth     if (ret == EOF) {
61277dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to flush file");
613895b00f6SMarc-André Lureau     } else {
614895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
615c216e5adSMichael Roth     }
616c216e5adSMichael Roth }
617c216e5adSMichael Roth 
618e72c3f2eSMichael Roth /* linux-specific implementations. avoid this if at all possible. */
619e72c3f2eSMichael Roth #if defined(__linux__)
620e72c3f2eSMichael Roth 
621eab5fd59SPaolo Bonzini #if defined(CONFIG_FSFREEZE) || defined(CONFIG_FSTRIM)
622af02203fSPaolo Bonzini typedef struct FsMount {
623c216e5adSMichael Roth     char *dirname;
624c216e5adSMichael Roth     char *devtype;
62546d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
626af02203fSPaolo Bonzini     QTAILQ_ENTRY(FsMount) next;
627af02203fSPaolo Bonzini } FsMount;
628c216e5adSMichael Roth 
629e5d9adbdSTomoki Sekiyama typedef QTAILQ_HEAD(FsMountList, FsMount) FsMountList;
6309e8aded4SMichael Roth 
631af02203fSPaolo Bonzini static void free_fs_mount_list(FsMountList *mounts)
632c216e5adSMichael Roth {
633af02203fSPaolo Bonzini      FsMount *mount, *temp;
634c216e5adSMichael Roth 
6359e8aded4SMichael Roth      if (!mounts) {
6369e8aded4SMichael Roth          return;
6379e8aded4SMichael Roth      }
6389e8aded4SMichael Roth 
6399e8aded4SMichael Roth      QTAILQ_FOREACH_SAFE(mount, mounts, next, temp) {
6409e8aded4SMichael Roth          QTAILQ_REMOVE(mounts, mount, next);
641c216e5adSMichael Roth          g_free(mount->dirname);
642c216e5adSMichael Roth          g_free(mount->devtype);
643c216e5adSMichael Roth          g_free(mount);
644c216e5adSMichael Roth      }
6459e8aded4SMichael Roth }
6469e8aded4SMichael Roth 
64746d4c572STomoki Sekiyama static int dev_major_minor(const char *devpath,
64846d4c572STomoki Sekiyama                            unsigned int *devmajor, unsigned int *devminor)
64946d4c572STomoki Sekiyama {
65046d4c572STomoki Sekiyama     struct stat st;
65146d4c572STomoki Sekiyama 
65246d4c572STomoki Sekiyama     *devmajor = 0;
65346d4c572STomoki Sekiyama     *devminor = 0;
65446d4c572STomoki Sekiyama 
65546d4c572STomoki Sekiyama     if (stat(devpath, &st) < 0) {
65646d4c572STomoki Sekiyama         slog("failed to stat device file '%s': %s", devpath, strerror(errno));
65746d4c572STomoki Sekiyama         return -1;
65846d4c572STomoki Sekiyama     }
65946d4c572STomoki Sekiyama     if (S_ISDIR(st.st_mode)) {
66046d4c572STomoki Sekiyama         /* It is bind mount */
66146d4c572STomoki Sekiyama         return -2;
66246d4c572STomoki Sekiyama     }
66346d4c572STomoki Sekiyama     if (S_ISBLK(st.st_mode)) {
66446d4c572STomoki Sekiyama         *devmajor = major(st.st_rdev);
66546d4c572STomoki Sekiyama         *devminor = minor(st.st_rdev);
66646d4c572STomoki Sekiyama         return 0;
66746d4c572STomoki Sekiyama     }
66846d4c572STomoki Sekiyama     return -1;
66946d4c572STomoki Sekiyama }
67046d4c572STomoki Sekiyama 
6719e8aded4SMichael Roth /*
6729e8aded4SMichael Roth  * Walk the mount table and build a list of local file systems
6739e8aded4SMichael Roth  */
67446d4c572STomoki Sekiyama static void build_fs_mount_list_from_mtab(FsMountList *mounts, Error **errp)
6759e8aded4SMichael Roth {
6769e8aded4SMichael Roth     struct mntent *ment;
677af02203fSPaolo Bonzini     FsMount *mount;
6789e2fa418SMichael Roth     char const *mtab = "/proc/self/mounts";
6799e8aded4SMichael Roth     FILE *fp;
68046d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
681c216e5adSMichael Roth 
682c216e5adSMichael Roth     fp = setmntent(mtab, "r");
683c216e5adSMichael Roth     if (!fp) {
68477dbc81bSMarkus Armbruster         error_setg(errp, "failed to open mtab file: '%s'", mtab);
685261551d1SLuiz Capitulino         return;
686c216e5adSMichael Roth     }
687c216e5adSMichael Roth 
688c216e5adSMichael Roth     while ((ment = getmntent(fp))) {
689c216e5adSMichael Roth         /*
690c216e5adSMichael Roth          * An entry which device name doesn't start with a '/' is
691c216e5adSMichael Roth          * either a dummy file system or a network file system.
692c216e5adSMichael Roth          * Add special handling for smbfs and cifs as is done by
693c216e5adSMichael Roth          * coreutils as well.
694c216e5adSMichael Roth          */
695c216e5adSMichael Roth         if ((ment->mnt_fsname[0] != '/') ||
696c216e5adSMichael Roth             (strcmp(ment->mnt_type, "smbfs") == 0) ||
697c216e5adSMichael Roth             (strcmp(ment->mnt_type, "cifs") == 0)) {
698c216e5adSMichael Roth             continue;
699c216e5adSMichael Roth         }
70046d4c572STomoki Sekiyama         if (dev_major_minor(ment->mnt_fsname, &devmajor, &devminor) == -2) {
70146d4c572STomoki Sekiyama             /* Skip bind mounts */
70246d4c572STomoki Sekiyama             continue;
70346d4c572STomoki Sekiyama         }
704c216e5adSMichael Roth 
705f3a06403SMarkus Armbruster         mount = g_new0(FsMount, 1);
706c216e5adSMichael Roth         mount->dirname = g_strdup(ment->mnt_dir);
707c216e5adSMichael Roth         mount->devtype = g_strdup(ment->mnt_type);
70846d4c572STomoki Sekiyama         mount->devmajor = devmajor;
70946d4c572STomoki Sekiyama         mount->devminor = devminor;
710c216e5adSMichael Roth 
7119e8aded4SMichael Roth         QTAILQ_INSERT_TAIL(mounts, mount, next);
712c216e5adSMichael Roth     }
713c216e5adSMichael Roth 
714c216e5adSMichael Roth     endmntent(fp);
715c216e5adSMichael Roth }
71646d4c572STomoki Sekiyama 
71746d4c572STomoki Sekiyama static void decode_mntname(char *name, int len)
71846d4c572STomoki Sekiyama {
71946d4c572STomoki Sekiyama     int i, j = 0;
72046d4c572STomoki Sekiyama     for (i = 0; i <= len; i++) {
72146d4c572STomoki Sekiyama         if (name[i] != '\\') {
72246d4c572STomoki Sekiyama             name[j++] = name[i];
72346d4c572STomoki Sekiyama         } else if (name[i + 1] == '\\') {
72446d4c572STomoki Sekiyama             name[j++] = '\\';
72546d4c572STomoki Sekiyama             i++;
72646d4c572STomoki Sekiyama         } else if (name[i + 1] >= '0' && name[i + 1] <= '3' &&
72746d4c572STomoki Sekiyama                    name[i + 2] >= '0' && name[i + 2] <= '7' &&
72846d4c572STomoki Sekiyama                    name[i + 3] >= '0' && name[i + 3] <= '7') {
72946d4c572STomoki Sekiyama             name[j++] = (name[i + 1] - '0') * 64 +
73046d4c572STomoki Sekiyama                         (name[i + 2] - '0') * 8 +
73146d4c572STomoki Sekiyama                         (name[i + 3] - '0');
73246d4c572STomoki Sekiyama             i += 3;
73346d4c572STomoki Sekiyama         } else {
73446d4c572STomoki Sekiyama             name[j++] = name[i];
73546d4c572STomoki Sekiyama         }
73646d4c572STomoki Sekiyama     }
73746d4c572STomoki Sekiyama }
73846d4c572STomoki Sekiyama 
73946d4c572STomoki Sekiyama static void build_fs_mount_list(FsMountList *mounts, Error **errp)
74046d4c572STomoki Sekiyama {
74146d4c572STomoki Sekiyama     FsMount *mount;
74246d4c572STomoki Sekiyama     char const *mountinfo = "/proc/self/mountinfo";
74346d4c572STomoki Sekiyama     FILE *fp;
74446d4c572STomoki Sekiyama     char *line = NULL, *dash;
74546d4c572STomoki Sekiyama     size_t n;
74646d4c572STomoki Sekiyama     char check;
74746d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
74846d4c572STomoki Sekiyama     int ret, dir_s, dir_e, type_s, type_e, dev_s, dev_e;
74946d4c572STomoki Sekiyama 
75046d4c572STomoki Sekiyama     fp = fopen(mountinfo, "r");
75146d4c572STomoki Sekiyama     if (!fp) {
75246d4c572STomoki Sekiyama         build_fs_mount_list_from_mtab(mounts, errp);
75346d4c572STomoki Sekiyama         return;
75446d4c572STomoki Sekiyama     }
75546d4c572STomoki Sekiyama 
75646d4c572STomoki Sekiyama     while (getline(&line, &n, fp) != -1) {
75746d4c572STomoki Sekiyama         ret = sscanf(line, "%*u %*u %u:%u %*s %n%*s%n%c",
75846d4c572STomoki Sekiyama                      &devmajor, &devminor, &dir_s, &dir_e, &check);
75946d4c572STomoki Sekiyama         if (ret < 3) {
76046d4c572STomoki Sekiyama             continue;
76146d4c572STomoki Sekiyama         }
76246d4c572STomoki Sekiyama         dash = strstr(line + dir_e, " - ");
76346d4c572STomoki Sekiyama         if (!dash) {
76446d4c572STomoki Sekiyama             continue;
76546d4c572STomoki Sekiyama         }
76646d4c572STomoki Sekiyama         ret = sscanf(dash, " - %n%*s%n %n%*s%n%c",
76746d4c572STomoki Sekiyama                      &type_s, &type_e, &dev_s, &dev_e, &check);
76846d4c572STomoki Sekiyama         if (ret < 1) {
76946d4c572STomoki Sekiyama             continue;
77046d4c572STomoki Sekiyama         }
77146d4c572STomoki Sekiyama         line[dir_e] = 0;
77246d4c572STomoki Sekiyama         dash[type_e] = 0;
77346d4c572STomoki Sekiyama         dash[dev_e] = 0;
77446d4c572STomoki Sekiyama         decode_mntname(line + dir_s, dir_e - dir_s);
77546d4c572STomoki Sekiyama         decode_mntname(dash + dev_s, dev_e - dev_s);
77646d4c572STomoki Sekiyama         if (devmajor == 0) {
77746d4c572STomoki Sekiyama             /* btrfs reports major number = 0 */
77846d4c572STomoki Sekiyama             if (strcmp("btrfs", dash + type_s) != 0 ||
77946d4c572STomoki Sekiyama                 dev_major_minor(dash + dev_s, &devmajor, &devminor) < 0) {
78046d4c572STomoki Sekiyama                 continue;
78146d4c572STomoki Sekiyama             }
78246d4c572STomoki Sekiyama         }
78346d4c572STomoki Sekiyama 
784f3a06403SMarkus Armbruster         mount = g_new0(FsMount, 1);
78546d4c572STomoki Sekiyama         mount->dirname = g_strdup(line + dir_s);
78646d4c572STomoki Sekiyama         mount->devtype = g_strdup(dash + type_s);
78746d4c572STomoki Sekiyama         mount->devmajor = devmajor;
78846d4c572STomoki Sekiyama         mount->devminor = devminor;
78946d4c572STomoki Sekiyama 
79046d4c572STomoki Sekiyama         QTAILQ_INSERT_TAIL(mounts, mount, next);
79146d4c572STomoki Sekiyama     }
79246d4c572STomoki Sekiyama     free(line);
79346d4c572STomoki Sekiyama 
79446d4c572STomoki Sekiyama     fclose(fp);
79546d4c572STomoki Sekiyama }
796eab5fd59SPaolo Bonzini #endif
797eab5fd59SPaolo Bonzini 
798eab5fd59SPaolo Bonzini #if defined(CONFIG_FSFREEZE)
799c216e5adSMichael Roth 
80046d4c572STomoki Sekiyama static char *get_pci_driver(char const *syspath, int pathlen, Error **errp)
80146d4c572STomoki Sekiyama {
80246d4c572STomoki Sekiyama     char *path;
80346d4c572STomoki Sekiyama     char *dpath;
80446d4c572STomoki Sekiyama     char *driver = NULL;
80546d4c572STomoki Sekiyama     char buf[PATH_MAX];
80646d4c572STomoki Sekiyama     ssize_t len;
80746d4c572STomoki Sekiyama 
80846d4c572STomoki Sekiyama     path = g_strndup(syspath, pathlen);
80946d4c572STomoki Sekiyama     dpath = g_strdup_printf("%s/driver", path);
81046d4c572STomoki Sekiyama     len = readlink(dpath, buf, sizeof(buf) - 1);
81146d4c572STomoki Sekiyama     if (len != -1) {
81246d4c572STomoki Sekiyama         buf[len] = 0;
8133e015d81SJulia Suvorova         driver = g_path_get_basename(buf);
81446d4c572STomoki Sekiyama     }
81546d4c572STomoki Sekiyama     g_free(dpath);
81646d4c572STomoki Sekiyama     g_free(path);
81746d4c572STomoki Sekiyama     return driver;
81846d4c572STomoki Sekiyama }
81946d4c572STomoki Sekiyama 
82046d4c572STomoki Sekiyama static int compare_uint(const void *_a, const void *_b)
82146d4c572STomoki Sekiyama {
82246d4c572STomoki Sekiyama     unsigned int a = *(unsigned int *)_a;
82346d4c572STomoki Sekiyama     unsigned int b = *(unsigned int *)_b;
82446d4c572STomoki Sekiyama 
82546d4c572STomoki Sekiyama     return a < b ? -1 : a > b ? 1 : 0;
82646d4c572STomoki Sekiyama }
82746d4c572STomoki Sekiyama 
82846d4c572STomoki Sekiyama /* Walk the specified sysfs and build a sorted list of host or ata numbers */
82946d4c572STomoki Sekiyama static int build_hosts(char const *syspath, char const *host, bool ata,
83046d4c572STomoki Sekiyama                        unsigned int *hosts, int hosts_max, Error **errp)
83146d4c572STomoki Sekiyama {
83246d4c572STomoki Sekiyama     char *path;
83346d4c572STomoki Sekiyama     DIR *dir;
83446d4c572STomoki Sekiyama     struct dirent *entry;
83546d4c572STomoki Sekiyama     int i = 0;
83646d4c572STomoki Sekiyama 
83746d4c572STomoki Sekiyama     path = g_strndup(syspath, host - syspath);
83846d4c572STomoki Sekiyama     dir = opendir(path);
83946d4c572STomoki Sekiyama     if (!dir) {
84046d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "opendir(\"%s\")", path);
84146d4c572STomoki Sekiyama         g_free(path);
84246d4c572STomoki Sekiyama         return -1;
84346d4c572STomoki Sekiyama     }
84446d4c572STomoki Sekiyama 
84546d4c572STomoki Sekiyama     while (i < hosts_max) {
84646d4c572STomoki Sekiyama         entry = readdir(dir);
84746d4c572STomoki Sekiyama         if (!entry) {
84846d4c572STomoki Sekiyama             break;
84946d4c572STomoki Sekiyama         }
85046d4c572STomoki Sekiyama         if (ata && sscanf(entry->d_name, "ata%d", hosts + i) == 1) {
85146d4c572STomoki Sekiyama             ++i;
85246d4c572STomoki Sekiyama         } else if (!ata && sscanf(entry->d_name, "host%d", hosts + i) == 1) {
85346d4c572STomoki Sekiyama             ++i;
85446d4c572STomoki Sekiyama         }
85546d4c572STomoki Sekiyama     }
85646d4c572STomoki Sekiyama 
85746d4c572STomoki Sekiyama     qsort(hosts, i, sizeof(hosts[0]), compare_uint);
85846d4c572STomoki Sekiyama 
85946d4c572STomoki Sekiyama     g_free(path);
86046d4c572STomoki Sekiyama     closedir(dir);
86146d4c572STomoki Sekiyama     return i;
86246d4c572STomoki Sekiyama }
86346d4c572STomoki Sekiyama 
864*d9fe4f0fSThomas Huth /*
865*d9fe4f0fSThomas Huth  * Store disk device info for devices on the PCI bus.
866*d9fe4f0fSThomas Huth  * Returns true if information has been stored, or false for failure.
867*d9fe4f0fSThomas Huth  */
868*d9fe4f0fSThomas Huth static bool build_guest_fsinfo_for_pci_dev(char const *syspath,
869*d9fe4f0fSThomas Huth                                            GuestDiskAddress *disk,
87046d4c572STomoki Sekiyama                                            Error **errp)
87146d4c572STomoki Sekiyama {
87246d4c572STomoki Sekiyama     unsigned int pci[4], host, hosts[8], tgt[3];
87346d4c572STomoki Sekiyama     int i, nhosts = 0, pcilen;
874*d9fe4f0fSThomas Huth     GuestPCIAddress *pciaddr = disk->pci_controller;
87546d4c572STomoki Sekiyama     bool has_ata = false, has_host = false, has_tgt = false;
87646d4c572STomoki Sekiyama     char *p, *q, *driver = NULL;
877b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
878b616105aSTomáš Golembiovský     struct udev *udev = NULL;
879b616105aSTomáš Golembiovský     struct udev_device *udevice = NULL;
880b616105aSTomáš Golembiovský #endif
881*d9fe4f0fSThomas Huth     bool ret = false;
88246d4c572STomoki Sekiyama 
88346d4c572STomoki Sekiyama     p = strstr(syspath, "/devices/pci");
88446d4c572STomoki Sekiyama     if (!p || sscanf(p + 12, "%*x:%*x/%x:%x:%x.%x%n",
88546d4c572STomoki Sekiyama                      pci, pci + 1, pci + 2, pci + 3, &pcilen) < 4) {
886743c71d0SMarc-André Lureau         g_debug("only pci device is supported: sysfs path '%s'", syspath);
887*d9fe4f0fSThomas Huth         return false;
88846d4c572STomoki Sekiyama     }
88946d4c572STomoki Sekiyama 
890743c71d0SMarc-André Lureau     p += 12 + pcilen;
891743c71d0SMarc-André Lureau     while (true) {
892743c71d0SMarc-André Lureau         driver = get_pci_driver(syspath, p - syspath, errp);
893743c71d0SMarc-André Lureau         if (driver && (g_str_equal(driver, "ata_piix") ||
894743c71d0SMarc-André Lureau                        g_str_equal(driver, "sym53c8xx") ||
895743c71d0SMarc-André Lureau                        g_str_equal(driver, "virtio-pci") ||
896743c71d0SMarc-André Lureau                        g_str_equal(driver, "ahci"))) {
897743c71d0SMarc-André Lureau             break;
898743c71d0SMarc-André Lureau         }
899743c71d0SMarc-André Lureau 
900bb23a736SMarc-André Lureau         g_free(driver);
901743c71d0SMarc-André Lureau         if (sscanf(p, "/%x:%x:%x.%x%n",
902743c71d0SMarc-André Lureau                           pci, pci + 1, pci + 2, pci + 3, &pcilen) == 4) {
903743c71d0SMarc-André Lureau             p += pcilen;
904743c71d0SMarc-André Lureau             continue;
905743c71d0SMarc-André Lureau         }
906743c71d0SMarc-André Lureau 
907743c71d0SMarc-André Lureau         g_debug("unsupported driver or sysfs path '%s'", syspath);
908*d9fe4f0fSThomas Huth         return false;
90946d4c572STomoki Sekiyama     }
91046d4c572STomoki Sekiyama 
91146d4c572STomoki Sekiyama     p = strstr(syspath, "/target");
91246d4c572STomoki Sekiyama     if (p && sscanf(p + 7, "%*u:%*u:%*u/%*u:%u:%u:%u",
91346d4c572STomoki Sekiyama                     tgt, tgt + 1, tgt + 2) == 3) {
91446d4c572STomoki Sekiyama         has_tgt = true;
91546d4c572STomoki Sekiyama     }
91646d4c572STomoki Sekiyama 
91746d4c572STomoki Sekiyama     p = strstr(syspath, "/ata");
91846d4c572STomoki Sekiyama     if (p) {
91946d4c572STomoki Sekiyama         q = p + 4;
92046d4c572STomoki Sekiyama         has_ata = true;
92146d4c572STomoki Sekiyama     } else {
92246d4c572STomoki Sekiyama         p = strstr(syspath, "/host");
92346d4c572STomoki Sekiyama         q = p + 5;
92446d4c572STomoki Sekiyama     }
92546d4c572STomoki Sekiyama     if (p && sscanf(q, "%u", &host) == 1) {
92646d4c572STomoki Sekiyama         has_host = true;
92746d4c572STomoki Sekiyama         nhosts = build_hosts(syspath, p, has_ata, hosts,
92801a6df1bSPhilippe Mathieu-Daudé                              ARRAY_SIZE(hosts), errp);
92946d4c572STomoki Sekiyama         if (nhosts < 0) {
93046d4c572STomoki Sekiyama             goto cleanup;
93146d4c572STomoki Sekiyama         }
93246d4c572STomoki Sekiyama     }
93346d4c572STomoki Sekiyama 
93446d4c572STomoki Sekiyama     pciaddr->domain = pci[0];
93546d4c572STomoki Sekiyama     pciaddr->bus = pci[1];
93646d4c572STomoki Sekiyama     pciaddr->slot = pci[2];
93746d4c572STomoki Sekiyama     pciaddr->function = pci[3];
93846d4c572STomoki Sekiyama 
939b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
940b616105aSTomáš Golembiovský     udev = udev_new();
941b616105aSTomáš Golembiovský     udevice = udev_device_new_from_syspath(udev, syspath);
942b616105aSTomáš Golembiovský     if (udev == NULL || udevice == NULL) {
943b616105aSTomáš Golembiovský         g_debug("failed to query udev");
944b616105aSTomáš Golembiovský     } else {
9456589ce35STomáš Golembiovský         const char *devnode, *serial;
9466589ce35STomáš Golembiovský         devnode = udev_device_get_devnode(udevice);
9476589ce35STomáš Golembiovský         if (devnode != NULL) {
9486589ce35STomáš Golembiovský             disk->dev = g_strdup(devnode);
9496589ce35STomáš Golembiovský             disk->has_dev = true;
9506589ce35STomáš Golembiovský         }
951b616105aSTomáš Golembiovský         serial = udev_device_get_property_value(udevice, "ID_SERIAL");
952b616105aSTomáš Golembiovský         if (serial != NULL && *serial != 0) {
953b616105aSTomáš Golembiovský             disk->serial = g_strdup(serial);
954b616105aSTomáš Golembiovský             disk->has_serial = true;
955b616105aSTomáš Golembiovský         }
956b616105aSTomáš Golembiovský     }
957b616105aSTomáš Golembiovský #endif
958b616105aSTomáš Golembiovský 
95946d4c572STomoki Sekiyama     if (strcmp(driver, "ata_piix") == 0) {
96046d4c572STomoki Sekiyama         /* a host per ide bus, target*:0:<unit>:0 */
96146d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
96246d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
96346d4c572STomoki Sekiyama             goto cleanup;
96446d4c572STomoki Sekiyama         }
96546d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
96646d4c572STomoki Sekiyama             if (host == hosts[i]) {
96746d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_IDE;
96846d4c572STomoki Sekiyama                 disk->bus = i;
96946d4c572STomoki Sekiyama                 disk->unit = tgt[1];
97046d4c572STomoki Sekiyama                 break;
97146d4c572STomoki Sekiyama             }
97246d4c572STomoki Sekiyama         }
97346d4c572STomoki Sekiyama         if (i >= nhosts) {
97446d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
97546d4c572STomoki Sekiyama             goto cleanup;
97646d4c572STomoki Sekiyama         }
97746d4c572STomoki Sekiyama     } else if (strcmp(driver, "sym53c8xx") == 0) {
97846d4c572STomoki Sekiyama         /* scsi(LSI Logic): target*:0:<unit>:0 */
97946d4c572STomoki Sekiyama         if (!has_tgt) {
98046d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
98146d4c572STomoki Sekiyama             goto cleanup;
98246d4c572STomoki Sekiyama         }
98346d4c572STomoki Sekiyama         disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
98446d4c572STomoki Sekiyama         disk->unit = tgt[1];
98546d4c572STomoki Sekiyama     } else if (strcmp(driver, "virtio-pci") == 0) {
98646d4c572STomoki Sekiyama         if (has_tgt) {
98746d4c572STomoki Sekiyama             /* virtio-scsi: target*:0:0:<unit> */
98846d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
98946d4c572STomoki Sekiyama             disk->unit = tgt[2];
99046d4c572STomoki Sekiyama         } else {
99146d4c572STomoki Sekiyama             /* virtio-blk: 1 disk per 1 device */
99246d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_VIRTIO;
99346d4c572STomoki Sekiyama         }
99446d4c572STomoki Sekiyama     } else if (strcmp(driver, "ahci") == 0) {
99546d4c572STomoki Sekiyama         /* ahci: 1 host per 1 unit */
99646d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
99746d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
99846d4c572STomoki Sekiyama             goto cleanup;
99946d4c572STomoki Sekiyama         }
100046d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
100146d4c572STomoki Sekiyama             if (host == hosts[i]) {
100246d4c572STomoki Sekiyama                 disk->unit = i;
100346d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_SATA;
100446d4c572STomoki Sekiyama                 break;
100546d4c572STomoki Sekiyama             }
100646d4c572STomoki Sekiyama         }
100746d4c572STomoki Sekiyama         if (i >= nhosts) {
100846d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
100946d4c572STomoki Sekiyama             goto cleanup;
101046d4c572STomoki Sekiyama         }
101146d4c572STomoki Sekiyama     } else {
101246d4c572STomoki Sekiyama         g_debug("unknown driver '%s' (sysfs path '%s')", driver, syspath);
101346d4c572STomoki Sekiyama         goto cleanup;
101446d4c572STomoki Sekiyama     }
101546d4c572STomoki Sekiyama 
1016*d9fe4f0fSThomas Huth     ret = true;
101746d4c572STomoki Sekiyama 
101846d4c572STomoki Sekiyama cleanup:
101946d4c572STomoki Sekiyama     g_free(driver);
1020b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
1021b616105aSTomáš Golembiovský     udev_unref(udev);
1022b616105aSTomáš Golembiovský     udev_device_unref(udevice);
1023b616105aSTomáš Golembiovský #endif
1024*d9fe4f0fSThomas Huth     return ret;
1025*d9fe4f0fSThomas Huth }
1026*d9fe4f0fSThomas Huth 
1027*d9fe4f0fSThomas Huth /* Store disk device info specified by @sysfs into @fs */
1028*d9fe4f0fSThomas Huth static void build_guest_fsinfo_for_real_device(char const *syspath,
1029*d9fe4f0fSThomas Huth                                                GuestFilesystemInfo *fs,
1030*d9fe4f0fSThomas Huth                                                Error **errp)
1031*d9fe4f0fSThomas Huth {
1032*d9fe4f0fSThomas Huth     GuestDiskAddress *disk;
1033*d9fe4f0fSThomas Huth     GuestPCIAddress *pciaddr;
1034*d9fe4f0fSThomas Huth     GuestDiskAddressList *list = NULL;
1035*d9fe4f0fSThomas Huth     bool has_hwinf;
1036*d9fe4f0fSThomas Huth 
1037*d9fe4f0fSThomas Huth     pciaddr = g_new0(GuestPCIAddress, 1);
1038*d9fe4f0fSThomas Huth 
1039*d9fe4f0fSThomas Huth     disk = g_new0(GuestDiskAddress, 1);
1040*d9fe4f0fSThomas Huth     disk->pci_controller = pciaddr;
1041*d9fe4f0fSThomas Huth 
1042*d9fe4f0fSThomas Huth     list = g_new0(GuestDiskAddressList, 1);
1043*d9fe4f0fSThomas Huth     list->value = disk;
1044*d9fe4f0fSThomas Huth 
1045*d9fe4f0fSThomas Huth     has_hwinf = build_guest_fsinfo_for_pci_dev(syspath, disk, errp);
1046*d9fe4f0fSThomas Huth 
1047*d9fe4f0fSThomas Huth     if (has_hwinf) {
1048*d9fe4f0fSThomas Huth         list->next = fs->disk;
1049*d9fe4f0fSThomas Huth         fs->disk = list;
1050*d9fe4f0fSThomas Huth     } else {
1051*d9fe4f0fSThomas Huth         qapi_free_GuestDiskAddressList(list);
1052*d9fe4f0fSThomas Huth     }
105346d4c572STomoki Sekiyama }
105446d4c572STomoki Sekiyama 
105546d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
105646d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
105746d4c572STomoki Sekiyama                                           Error **errp);
105846d4c572STomoki Sekiyama 
105946d4c572STomoki Sekiyama /* Store a list of slave devices of virtual volume specified by @syspath into
106046d4c572STomoki Sekiyama  * @fs */
106146d4c572STomoki Sekiyama static void build_guest_fsinfo_for_virtual_device(char const *syspath,
106246d4c572STomoki Sekiyama                                                   GuestFilesystemInfo *fs,
106346d4c572STomoki Sekiyama                                                   Error **errp)
106446d4c572STomoki Sekiyama {
1065292743d9SMarkus Armbruster     Error *err = NULL;
106646d4c572STomoki Sekiyama     DIR *dir;
106746d4c572STomoki Sekiyama     char *dirpath;
1068e668d1b8Szhanghailiang     struct dirent *entry;
106946d4c572STomoki Sekiyama 
107046d4c572STomoki Sekiyama     dirpath = g_strdup_printf("%s/slaves", syspath);
107146d4c572STomoki Sekiyama     dir = opendir(dirpath);
107246d4c572STomoki Sekiyama     if (!dir) {
10738251a72fSMichael Roth         if (errno != ENOENT) {
107446d4c572STomoki Sekiyama             error_setg_errno(errp, errno, "opendir(\"%s\")", dirpath);
10758251a72fSMichael Roth         }
107646d4c572STomoki Sekiyama         g_free(dirpath);
107746d4c572STomoki Sekiyama         return;
107846d4c572STomoki Sekiyama     }
107946d4c572STomoki Sekiyama 
108046d4c572STomoki Sekiyama     for (;;) {
1081e668d1b8Szhanghailiang         errno = 0;
1082e668d1b8Szhanghailiang         entry = readdir(dir);
1083e668d1b8Szhanghailiang         if (entry == NULL) {
1084e668d1b8Szhanghailiang             if (errno) {
1085e668d1b8Szhanghailiang                 error_setg_errno(errp, errno, "readdir(\"%s\")", dirpath);
108646d4c572STomoki Sekiyama             }
108746d4c572STomoki Sekiyama             break;
108846d4c572STomoki Sekiyama         }
108946d4c572STomoki Sekiyama 
1090e668d1b8Szhanghailiang         if (entry->d_type == DT_LNK) {
1091e668d1b8Szhanghailiang             char *path;
1092e668d1b8Szhanghailiang 
1093e668d1b8Szhanghailiang             g_debug(" slave device '%s'", entry->d_name);
1094e668d1b8Szhanghailiang             path = g_strdup_printf("%s/slaves/%s", syspath, entry->d_name);
1095292743d9SMarkus Armbruster             build_guest_fsinfo_for_device(path, fs, &err);
1096e668d1b8Szhanghailiang             g_free(path);
109746d4c572STomoki Sekiyama 
1098292743d9SMarkus Armbruster             if (err) {
1099292743d9SMarkus Armbruster                 error_propagate(errp, err);
110046d4c572STomoki Sekiyama                 break;
110146d4c572STomoki Sekiyama             }
110246d4c572STomoki Sekiyama         }
110346d4c572STomoki Sekiyama     }
110446d4c572STomoki Sekiyama 
1105e668d1b8Szhanghailiang     g_free(dirpath);
110646d4c572STomoki Sekiyama     closedir(dir);
110746d4c572STomoki Sekiyama }
110846d4c572STomoki Sekiyama 
110946d4c572STomoki Sekiyama /* Dispatch to functions for virtual/real device */
111046d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
111146d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
111246d4c572STomoki Sekiyama                                           Error **errp)
111346d4c572STomoki Sekiyama {
111446d4c572STomoki Sekiyama     char *syspath = realpath(devpath, NULL);
111546d4c572STomoki Sekiyama 
111646d4c572STomoki Sekiyama     if (!syspath) {
111746d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "realpath(\"%s\")", devpath);
111846d4c572STomoki Sekiyama         return;
111946d4c572STomoki Sekiyama     }
112046d4c572STomoki Sekiyama 
112146d4c572STomoki Sekiyama     if (!fs->name) {
11223e015d81SJulia Suvorova         fs->name = g_path_get_basename(syspath);
112346d4c572STomoki Sekiyama     }
112446d4c572STomoki Sekiyama 
112546d4c572STomoki Sekiyama     g_debug("  parse sysfs path '%s'", syspath);
112646d4c572STomoki Sekiyama 
112746d4c572STomoki Sekiyama     if (strstr(syspath, "/devices/virtual/block/")) {
112846d4c572STomoki Sekiyama         build_guest_fsinfo_for_virtual_device(syspath, fs, errp);
112946d4c572STomoki Sekiyama     } else {
113046d4c572STomoki Sekiyama         build_guest_fsinfo_for_real_device(syspath, fs, errp);
113146d4c572STomoki Sekiyama     }
113246d4c572STomoki Sekiyama 
113346d4c572STomoki Sekiyama     free(syspath);
113446d4c572STomoki Sekiyama }
113546d4c572STomoki Sekiyama 
113646d4c572STomoki Sekiyama /* Return a list of the disk device(s)' info which @mount lies on */
113746d4c572STomoki Sekiyama static GuestFilesystemInfo *build_guest_fsinfo(struct FsMount *mount,
113846d4c572STomoki Sekiyama                                                Error **errp)
113946d4c572STomoki Sekiyama {
114046d4c572STomoki Sekiyama     GuestFilesystemInfo *fs = g_malloc0(sizeof(*fs));
114125b5ff1aSChen Hanxiao     struct statvfs buf;
114225b5ff1aSChen Hanxiao     unsigned long used, nonroot_total, fr_size;
114346d4c572STomoki Sekiyama     char *devpath = g_strdup_printf("/sys/dev/block/%u:%u",
114446d4c572STomoki Sekiyama                                     mount->devmajor, mount->devminor);
114546d4c572STomoki Sekiyama 
114646d4c572STomoki Sekiyama     fs->mountpoint = g_strdup(mount->dirname);
114746d4c572STomoki Sekiyama     fs->type = g_strdup(mount->devtype);
114846d4c572STomoki Sekiyama     build_guest_fsinfo_for_device(devpath, fs, errp);
114946d4c572STomoki Sekiyama 
115025b5ff1aSChen Hanxiao     if (statvfs(fs->mountpoint, &buf) == 0) {
115125b5ff1aSChen Hanxiao         fr_size = buf.f_frsize;
115225b5ff1aSChen Hanxiao         used = buf.f_blocks - buf.f_bfree;
115325b5ff1aSChen Hanxiao         nonroot_total = used + buf.f_bavail;
115425b5ff1aSChen Hanxiao         fs->used_bytes = used * fr_size;
115525b5ff1aSChen Hanxiao         fs->total_bytes = nonroot_total * fr_size;
115625b5ff1aSChen Hanxiao 
115725b5ff1aSChen Hanxiao         fs->has_total_bytes = true;
115825b5ff1aSChen Hanxiao         fs->has_used_bytes = true;
115925b5ff1aSChen Hanxiao     }
116025b5ff1aSChen Hanxiao 
116146d4c572STomoki Sekiyama     g_free(devpath);
116225b5ff1aSChen Hanxiao 
116346d4c572STomoki Sekiyama     return fs;
116446d4c572STomoki Sekiyama }
116546d4c572STomoki Sekiyama 
116646d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
116746d4c572STomoki Sekiyama {
116846d4c572STomoki Sekiyama     FsMountList mounts;
116946d4c572STomoki Sekiyama     struct FsMount *mount;
117046d4c572STomoki Sekiyama     GuestFilesystemInfoList *new, *ret = NULL;
117146d4c572STomoki Sekiyama     Error *local_err = NULL;
117246d4c572STomoki Sekiyama 
117346d4c572STomoki Sekiyama     QTAILQ_INIT(&mounts);
117446d4c572STomoki Sekiyama     build_fs_mount_list(&mounts, &local_err);
117546d4c572STomoki Sekiyama     if (local_err) {
117646d4c572STomoki Sekiyama         error_propagate(errp, local_err);
117746d4c572STomoki Sekiyama         return NULL;
117846d4c572STomoki Sekiyama     }
117946d4c572STomoki Sekiyama 
118046d4c572STomoki Sekiyama     QTAILQ_FOREACH(mount, &mounts, next) {
118146d4c572STomoki Sekiyama         g_debug("Building guest fsinfo for '%s'", mount->dirname);
118246d4c572STomoki Sekiyama 
118346d4c572STomoki Sekiyama         new = g_malloc0(sizeof(*ret));
118446d4c572STomoki Sekiyama         new->value = build_guest_fsinfo(mount, &local_err);
118546d4c572STomoki Sekiyama         new->next = ret;
118646d4c572STomoki Sekiyama         ret = new;
118746d4c572STomoki Sekiyama         if (local_err) {
118846d4c572STomoki Sekiyama             error_propagate(errp, local_err);
118946d4c572STomoki Sekiyama             qapi_free_GuestFilesystemInfoList(ret);
119046d4c572STomoki Sekiyama             ret = NULL;
119146d4c572STomoki Sekiyama             break;
119246d4c572STomoki Sekiyama         }
119346d4c572STomoki Sekiyama     }
119446d4c572STomoki Sekiyama 
119546d4c572STomoki Sekiyama     free_fs_mount_list(&mounts);
119646d4c572STomoki Sekiyama     return ret;
119746d4c572STomoki Sekiyama }
119846d4c572STomoki Sekiyama 
119946d4c572STomoki Sekiyama 
1200ec0f694cSTomoki Sekiyama typedef enum {
1201ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_THAW = 0,
1202ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_FREEZE,
1203ec0f694cSTomoki Sekiyama } FsfreezeHookArg;
1204ec0f694cSTomoki Sekiyama 
120513a439ecSStefan Weil static const char *fsfreeze_hook_arg_string[] = {
1206ec0f694cSTomoki Sekiyama     "thaw",
1207ec0f694cSTomoki Sekiyama     "freeze",
1208ec0f694cSTomoki Sekiyama };
1209ec0f694cSTomoki Sekiyama 
121077dbc81bSMarkus Armbruster static void execute_fsfreeze_hook(FsfreezeHookArg arg, Error **errp)
1211ec0f694cSTomoki Sekiyama {
1212ec0f694cSTomoki Sekiyama     int status;
1213ec0f694cSTomoki Sekiyama     pid_t pid;
1214ec0f694cSTomoki Sekiyama     const char *hook;
1215ec0f694cSTomoki Sekiyama     const char *arg_str = fsfreeze_hook_arg_string[arg];
1216ec0f694cSTomoki Sekiyama     Error *local_err = NULL;
1217ec0f694cSTomoki Sekiyama 
1218ec0f694cSTomoki Sekiyama     hook = ga_fsfreeze_hook(ga_state);
1219ec0f694cSTomoki Sekiyama     if (!hook) {
1220ec0f694cSTomoki Sekiyama         return;
1221ec0f694cSTomoki Sekiyama     }
1222ec0f694cSTomoki Sekiyama     if (access(hook, X_OK) != 0) {
122377dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "can't access fsfreeze hook '%s'", hook);
1224ec0f694cSTomoki Sekiyama         return;
1225ec0f694cSTomoki Sekiyama     }
1226ec0f694cSTomoki Sekiyama 
1227ec0f694cSTomoki Sekiyama     slog("executing fsfreeze hook with arg '%s'", arg_str);
1228ec0f694cSTomoki Sekiyama     pid = fork();
1229ec0f694cSTomoki Sekiyama     if (pid == 0) {
1230ec0f694cSTomoki Sekiyama         setsid();
1231ec0f694cSTomoki Sekiyama         reopen_fd_to_null(0);
1232ec0f694cSTomoki Sekiyama         reopen_fd_to_null(1);
1233ec0f694cSTomoki Sekiyama         reopen_fd_to_null(2);
1234ec0f694cSTomoki Sekiyama 
1235ec0f694cSTomoki Sekiyama         execle(hook, hook, arg_str, NULL, environ);
1236ec0f694cSTomoki Sekiyama         _exit(EXIT_FAILURE);
1237ec0f694cSTomoki Sekiyama     } else if (pid < 0) {
123877dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
1239ec0f694cSTomoki Sekiyama         return;
1240ec0f694cSTomoki Sekiyama     }
1241ec0f694cSTomoki Sekiyama 
1242ec0f694cSTomoki Sekiyama     ga_wait_child(pid, &status, &local_err);
124384d18f06SMarkus Armbruster     if (local_err) {
124477dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1245ec0f694cSTomoki Sekiyama         return;
1246ec0f694cSTomoki Sekiyama     }
1247ec0f694cSTomoki Sekiyama 
1248ec0f694cSTomoki Sekiyama     if (!WIFEXITED(status)) {
124977dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has terminated abnormally");
1250ec0f694cSTomoki Sekiyama         return;
1251ec0f694cSTomoki Sekiyama     }
1252ec0f694cSTomoki Sekiyama 
1253ec0f694cSTomoki Sekiyama     status = WEXITSTATUS(status);
1254ec0f694cSTomoki Sekiyama     if (status) {
125577dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has failed with status %d", status);
1256ec0f694cSTomoki Sekiyama         return;
1257ec0f694cSTomoki Sekiyama     }
1258ec0f694cSTomoki Sekiyama }
1259ec0f694cSTomoki Sekiyama 
1260c216e5adSMichael Roth /*
1261c216e5adSMichael Roth  * Return status of freeze/thaw
1262c216e5adSMichael Roth  */
126377dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
1264c216e5adSMichael Roth {
1265f22d85e9SMichael Roth     if (ga_is_frozen(ga_state)) {
1266f22d85e9SMichael Roth         return GUEST_FSFREEZE_STATUS_FROZEN;
1267f22d85e9SMichael Roth     }
1268f22d85e9SMichael Roth 
1269f22d85e9SMichael Roth     return GUEST_FSFREEZE_STATUS_THAWED;
1270c216e5adSMichael Roth }
1271c216e5adSMichael Roth 
1272e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze(Error **errp)
1273e99bce20STomoki Sekiyama {
1274e99bce20STomoki Sekiyama     return qmp_guest_fsfreeze_freeze_list(false, NULL, errp);
1275e99bce20STomoki Sekiyama }
1276e99bce20STomoki Sekiyama 
1277c216e5adSMichael Roth /*
1278c216e5adSMichael Roth  * Walk list of mounted file systems in the guest, and freeze the ones which
1279c216e5adSMichael Roth  * are real local file systems.
1280c216e5adSMichael Roth  */
1281e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
1282e99bce20STomoki Sekiyama                                        strList *mountpoints,
1283e99bce20STomoki Sekiyama                                        Error **errp)
1284c216e5adSMichael Roth {
1285c216e5adSMichael Roth     int ret = 0, i = 0;
1286e99bce20STomoki Sekiyama     strList *list;
1287af02203fSPaolo Bonzini     FsMountList mounts;
1288af02203fSPaolo Bonzini     struct FsMount *mount;
1289261551d1SLuiz Capitulino     Error *local_err = NULL;
1290c216e5adSMichael Roth     int fd;
1291c216e5adSMichael Roth 
1292c216e5adSMichael Roth     slog("guest-fsfreeze called");
1293c216e5adSMichael Roth 
1294ec0f694cSTomoki Sekiyama     execute_fsfreeze_hook(FSFREEZE_HOOK_FREEZE, &local_err);
129584d18f06SMarkus Armbruster     if (local_err) {
129677dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1297ec0f694cSTomoki Sekiyama         return -1;
1298ec0f694cSTomoki Sekiyama     }
1299ec0f694cSTomoki Sekiyama 
13009e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1301261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
130284d18f06SMarkus Armbruster     if (local_err) {
130377dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1304261551d1SLuiz Capitulino         return -1;
1305c216e5adSMichael Roth     }
1306c216e5adSMichael Roth 
1307c216e5adSMichael Roth     /* cannot risk guest agent blocking itself on a write in this state */
1308f22d85e9SMichael Roth     ga_set_frozen(ga_state);
1309c216e5adSMichael Roth 
1310eae3eb3eSPaolo Bonzini     QTAILQ_FOREACH_REVERSE(mount, &mounts, next) {
1311e99bce20STomoki Sekiyama         /* To issue fsfreeze in the reverse order of mounts, check if the
1312e99bce20STomoki Sekiyama          * mount is listed in the list here */
1313e99bce20STomoki Sekiyama         if (has_mountpoints) {
1314e99bce20STomoki Sekiyama             for (list = mountpoints; list; list = list->next) {
1315e99bce20STomoki Sekiyama                 if (strcmp(list->value, mount->dirname) == 0) {
1316e99bce20STomoki Sekiyama                     break;
1317e99bce20STomoki Sekiyama                 }
1318e99bce20STomoki Sekiyama             }
1319e99bce20STomoki Sekiyama             if (!list) {
1320e99bce20STomoki Sekiyama                 continue;
1321e99bce20STomoki Sekiyama             }
1322e99bce20STomoki Sekiyama         }
1323e99bce20STomoki Sekiyama 
1324c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1325c216e5adSMichael Roth         if (fd == -1) {
132677dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "failed to open %s", mount->dirname);
1327c216e5adSMichael Roth             goto error;
1328c216e5adSMichael Roth         }
1329c216e5adSMichael Roth 
1330e35916acSMichael Tokarev         /* we try to cull filesystems we know won't work in advance, but other
1331e35916acSMichael Tokarev          * filesystems may not implement fsfreeze for less obvious reasons.
13329e8aded4SMichael Roth          * these will report EOPNOTSUPP. we simply ignore these when tallying
13339e8aded4SMichael Roth          * the number of frozen filesystems.
1334ce2eb6c4SPeter Lieven          * if a filesystem is mounted more than once (aka bind mount) a
1335ce2eb6c4SPeter Lieven          * consecutive attempt to freeze an already frozen filesystem will
1336ce2eb6c4SPeter Lieven          * return EBUSY.
13379e8aded4SMichael Roth          *
13389e8aded4SMichael Roth          * any other error means a failure to freeze a filesystem we
13399e8aded4SMichael Roth          * expect to be freezable, so return an error in those cases
13409e8aded4SMichael Roth          * and return system to thawed state.
1341c216e5adSMichael Roth          */
1342c216e5adSMichael Roth         ret = ioctl(fd, FIFREEZE);
13439e8aded4SMichael Roth         if (ret == -1) {
1344ce2eb6c4SPeter Lieven             if (errno != EOPNOTSUPP && errno != EBUSY) {
134577dbc81bSMarkus Armbruster                 error_setg_errno(errp, errno, "failed to freeze %s",
1346617fbbc1SLuiz Capitulino                                  mount->dirname);
1347c216e5adSMichael Roth                 close(fd);
1348c216e5adSMichael Roth                 goto error;
1349c216e5adSMichael Roth             }
13509e8aded4SMichael Roth         } else {
1351c216e5adSMichael Roth             i++;
1352c216e5adSMichael Roth         }
13539e8aded4SMichael Roth         close(fd);
13549e8aded4SMichael Roth     }
1355c216e5adSMichael Roth 
1356af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
135765650f01SChen Hanxiao     /* We may not issue any FIFREEZE here.
135865650f01SChen Hanxiao      * Just unset ga_state here and ready for the next call.
135965650f01SChen Hanxiao      */
136065650f01SChen Hanxiao     if (i == 0) {
136165650f01SChen Hanxiao         ga_unset_frozen(ga_state);
136265650f01SChen Hanxiao     }
1363c216e5adSMichael Roth     return i;
1364c216e5adSMichael Roth 
1365c216e5adSMichael Roth error:
1366af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1367c216e5adSMichael Roth     qmp_guest_fsfreeze_thaw(NULL);
1368c216e5adSMichael Roth     return 0;
1369c216e5adSMichael Roth }
1370c216e5adSMichael Roth 
1371c216e5adSMichael Roth /*
1372c216e5adSMichael Roth  * Walk list of frozen file systems in the guest, and thaw them.
1373c216e5adSMichael Roth  */
137477dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
1375c216e5adSMichael Roth {
1376c216e5adSMichael Roth     int ret;
1377af02203fSPaolo Bonzini     FsMountList mounts;
1378af02203fSPaolo Bonzini     FsMount *mount;
13799e8aded4SMichael Roth     int fd, i = 0, logged;
1380261551d1SLuiz Capitulino     Error *local_err = NULL;
1381c216e5adSMichael Roth 
13829e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1383261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
138484d18f06SMarkus Armbruster     if (local_err) {
138577dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
13869e8aded4SMichael Roth         return 0;
13879e8aded4SMichael Roth     }
13889e8aded4SMichael Roth 
13899e8aded4SMichael Roth     QTAILQ_FOREACH(mount, &mounts, next) {
13909e8aded4SMichael Roth         logged = false;
1391c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1392c216e5adSMichael Roth         if (fd == -1) {
1393c216e5adSMichael Roth             continue;
1394c216e5adSMichael Roth         }
13959e8aded4SMichael Roth         /* we have no way of knowing whether a filesystem was actually unfrozen
13969e8aded4SMichael Roth          * as a result of a successful call to FITHAW, only that if an error
13979e8aded4SMichael Roth          * was returned the filesystem was *not* unfrozen by that particular
13989e8aded4SMichael Roth          * call.
13999e8aded4SMichael Roth          *
1400a31f0531SJim Meyering          * since multiple preceding FIFREEZEs require multiple calls to FITHAW
14019e8aded4SMichael Roth          * to unfreeze, continuing issuing FITHAW until an error is returned,
14029e8aded4SMichael Roth          * in which case either the filesystem is in an unfreezable state, or,
14039e8aded4SMichael Roth          * more likely, it was thawed previously (and remains so afterward).
14049e8aded4SMichael Roth          *
14059e8aded4SMichael Roth          * also, since the most recent successful call is the one that did
14069e8aded4SMichael Roth          * the actual unfreeze, we can use this to provide an accurate count
14079e8aded4SMichael Roth          * of the number of filesystems unfrozen by guest-fsfreeze-thaw, which
14089e8aded4SMichael Roth          * may * be useful for determining whether a filesystem was unfrozen
14099e8aded4SMichael Roth          * during the freeze/thaw phase by a process other than qemu-ga.
14109e8aded4SMichael Roth          */
14119e8aded4SMichael Roth         do {
1412c216e5adSMichael Roth             ret = ioctl(fd, FITHAW);
14139e8aded4SMichael Roth             if (ret == 0 && !logged) {
1414c216e5adSMichael Roth                 i++;
14159e8aded4SMichael Roth                 logged = true;
14169e8aded4SMichael Roth             }
14179e8aded4SMichael Roth         } while (ret == 0);
14189e8aded4SMichael Roth         close(fd);
1419c216e5adSMichael Roth     }
1420c216e5adSMichael Roth 
1421f22d85e9SMichael Roth     ga_unset_frozen(ga_state);
1422af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1423ec0f694cSTomoki Sekiyama 
142477dbc81bSMarkus Armbruster     execute_fsfreeze_hook(FSFREEZE_HOOK_THAW, errp);
1425ec0f694cSTomoki Sekiyama 
1426c216e5adSMichael Roth     return i;
1427c216e5adSMichael Roth }
1428c216e5adSMichael Roth 
1429c216e5adSMichael Roth static void guest_fsfreeze_cleanup(void)
1430c216e5adSMichael Roth {
1431c216e5adSMichael Roth     Error *err = NULL;
1432c216e5adSMichael Roth 
1433f22d85e9SMichael Roth     if (ga_is_frozen(ga_state) == GUEST_FSFREEZE_STATUS_FROZEN) {
14346f686749SMarkus Armbruster         qmp_guest_fsfreeze_thaw(&err);
14356f686749SMarkus Armbruster         if (err) {
14366f686749SMarkus Armbruster             slog("failed to clean up frozen filesystems: %s",
14376f686749SMarkus Armbruster                  error_get_pretty(err));
14386f686749SMarkus Armbruster             error_free(err);
1439c216e5adSMichael Roth         }
1440c216e5adSMichael Roth     }
1441c216e5adSMichael Roth }
1442e72c3f2eSMichael Roth #endif /* CONFIG_FSFREEZE */
1443c216e5adSMichael Roth 
1444eab5fd59SPaolo Bonzini #if defined(CONFIG_FSTRIM)
1445eab5fd59SPaolo Bonzini /*
1446eab5fd59SPaolo Bonzini  * Walk list of mounted file systems in the guest, and trim them.
1447eab5fd59SPaolo Bonzini  */
1448e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
1449e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
1450eab5fd59SPaolo Bonzini {
1451e82855d9SJustin Ossevoort     GuestFilesystemTrimResponse *response;
1452e82855d9SJustin Ossevoort     GuestFilesystemTrimResultList *list;
1453e82855d9SJustin Ossevoort     GuestFilesystemTrimResult *result;
1454eab5fd59SPaolo Bonzini     int ret = 0;
1455eab5fd59SPaolo Bonzini     FsMountList mounts;
1456eab5fd59SPaolo Bonzini     struct FsMount *mount;
1457eab5fd59SPaolo Bonzini     int fd;
1458261551d1SLuiz Capitulino     Error *local_err = NULL;
145973a652a1SJustin Ossevoort     struct fstrim_range r;
1460eab5fd59SPaolo Bonzini 
1461eab5fd59SPaolo Bonzini     slog("guest-fstrim called");
1462eab5fd59SPaolo Bonzini 
1463eab5fd59SPaolo Bonzini     QTAILQ_INIT(&mounts);
1464261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
146584d18f06SMarkus Armbruster     if (local_err) {
146677dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1467e82855d9SJustin Ossevoort         return NULL;
1468eab5fd59SPaolo Bonzini     }
1469eab5fd59SPaolo Bonzini 
1470e82855d9SJustin Ossevoort     response = g_malloc0(sizeof(*response));
1471e82855d9SJustin Ossevoort 
1472eab5fd59SPaolo Bonzini     QTAILQ_FOREACH(mount, &mounts, next) {
1473e82855d9SJustin Ossevoort         result = g_malloc0(sizeof(*result));
1474e82855d9SJustin Ossevoort         result->path = g_strdup(mount->dirname);
1475e82855d9SJustin Ossevoort 
1476e82855d9SJustin Ossevoort         list = g_malloc0(sizeof(*list));
1477e82855d9SJustin Ossevoort         list->value = result;
1478e82855d9SJustin Ossevoort         list->next = response->paths;
1479e82855d9SJustin Ossevoort         response->paths = list;
1480e82855d9SJustin Ossevoort 
1481eab5fd59SPaolo Bonzini         fd = qemu_open(mount->dirname, O_RDONLY);
1482eab5fd59SPaolo Bonzini         if (fd == -1) {
1483e82855d9SJustin Ossevoort             result->error = g_strdup_printf("failed to open: %s",
1484e82855d9SJustin Ossevoort                                             strerror(errno));
1485e82855d9SJustin Ossevoort             result->has_error = true;
1486e82855d9SJustin Ossevoort             continue;
1487eab5fd59SPaolo Bonzini         }
1488eab5fd59SPaolo Bonzini 
1489e35916acSMichael Tokarev         /* We try to cull filesystems we know won't work in advance, but other
1490e35916acSMichael Tokarev          * filesystems may not implement fstrim for less obvious reasons.
1491e35916acSMichael Tokarev          * These will report EOPNOTSUPP; while in some other cases ENOTTY
1492e35916acSMichael Tokarev          * will be reported (e.g. CD-ROMs).
1493e82855d9SJustin Ossevoort          * Any other error means an unexpected error.
1494eab5fd59SPaolo Bonzini          */
149573a652a1SJustin Ossevoort         r.start = 0;
149673a652a1SJustin Ossevoort         r.len = -1;
149773a652a1SJustin Ossevoort         r.minlen = has_minimum ? minimum : 0;
1498eab5fd59SPaolo Bonzini         ret = ioctl(fd, FITRIM, &r);
1499eab5fd59SPaolo Bonzini         if (ret == -1) {
1500e82855d9SJustin Ossevoort             result->has_error = true;
1501e82855d9SJustin Ossevoort             if (errno == ENOTTY || errno == EOPNOTSUPP) {
1502e82855d9SJustin Ossevoort                 result->error = g_strdup("trim not supported");
1503e82855d9SJustin Ossevoort             } else {
1504e82855d9SJustin Ossevoort                 result->error = g_strdup_printf("failed to trim: %s",
1505e82855d9SJustin Ossevoort                                                 strerror(errno));
1506e82855d9SJustin Ossevoort             }
1507eab5fd59SPaolo Bonzini             close(fd);
1508e82855d9SJustin Ossevoort             continue;
1509eab5fd59SPaolo Bonzini         }
1510e82855d9SJustin Ossevoort 
1511e82855d9SJustin Ossevoort         result->has_minimum = true;
1512e82855d9SJustin Ossevoort         result->minimum = r.minlen;
1513e82855d9SJustin Ossevoort         result->has_trimmed = true;
1514e82855d9SJustin Ossevoort         result->trimmed = r.len;
1515eab5fd59SPaolo Bonzini         close(fd);
1516eab5fd59SPaolo Bonzini     }
1517eab5fd59SPaolo Bonzini 
1518eab5fd59SPaolo Bonzini     free_fs_mount_list(&mounts);
1519e82855d9SJustin Ossevoort     return response;
1520eab5fd59SPaolo Bonzini }
1521eab5fd59SPaolo Bonzini #endif /* CONFIG_FSTRIM */
1522eab5fd59SPaolo Bonzini 
1523eab5fd59SPaolo Bonzini 
152411d0f125SLuiz Capitulino #define LINUX_SYS_STATE_FILE "/sys/power/state"
152511d0f125SLuiz Capitulino #define SUSPEND_SUPPORTED 0
152611d0f125SLuiz Capitulino #define SUSPEND_NOT_SUPPORTED 1
152711d0f125SLuiz Capitulino 
15288b020b5eSDaniel Henrique Barboza typedef enum {
15298b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_DISK = 0,
15308b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_RAM = 1,
15318b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_HYBRID = 2,
15328b020b5eSDaniel Henrique Barboza } SuspendMode;
15338b020b5eSDaniel Henrique Barboza 
15348b020b5eSDaniel Henrique Barboza /*
15358b020b5eSDaniel Henrique Barboza  * Executes a command in a child process using g_spawn_sync,
15368b020b5eSDaniel Henrique Barboza  * returning an int >= 0 representing the exit status of the
15378b020b5eSDaniel Henrique Barboza  * process.
15388b020b5eSDaniel Henrique Barboza  *
15398b020b5eSDaniel Henrique Barboza  * If the program wasn't found in path, returns -1.
15408b020b5eSDaniel Henrique Barboza  *
15418b020b5eSDaniel Henrique Barboza  * If a problem happened when creating the child process,
15428b020b5eSDaniel Henrique Barboza  * returns -1 and errp is set.
15438b020b5eSDaniel Henrique Barboza  */
15448b020b5eSDaniel Henrique Barboza static int run_process_child(const char *command[], Error **errp)
15458b020b5eSDaniel Henrique Barboza {
15468b020b5eSDaniel Henrique Barboza     int exit_status, spawn_flag;
15478b020b5eSDaniel Henrique Barboza     GError *g_err = NULL;
15488b020b5eSDaniel Henrique Barboza     bool success;
15498b020b5eSDaniel Henrique Barboza 
15508b020b5eSDaniel Henrique Barboza     spawn_flag = G_SPAWN_SEARCH_PATH | G_SPAWN_STDOUT_TO_DEV_NULL |
15518b020b5eSDaniel Henrique Barboza                  G_SPAWN_STDERR_TO_DEV_NULL;
15528b020b5eSDaniel Henrique Barboza 
15538b020b5eSDaniel Henrique Barboza     success =  g_spawn_sync(NULL, (char **)command, environ, spawn_flag,
15548b020b5eSDaniel Henrique Barboza                             NULL, NULL, NULL, NULL,
15558b020b5eSDaniel Henrique Barboza                             &exit_status, &g_err);
15568b020b5eSDaniel Henrique Barboza 
15578b020b5eSDaniel Henrique Barboza     if (success) {
15588b020b5eSDaniel Henrique Barboza         return WEXITSTATUS(exit_status);
15598b020b5eSDaniel Henrique Barboza     }
15608b020b5eSDaniel Henrique Barboza 
15618b020b5eSDaniel Henrique Barboza     if (g_err && (g_err->code != G_SPAWN_ERROR_NOENT)) {
15628b020b5eSDaniel Henrique Barboza         error_setg(errp, "failed to create child process, error '%s'",
15638b020b5eSDaniel Henrique Barboza                    g_err->message);
15648b020b5eSDaniel Henrique Barboza     }
15658b020b5eSDaniel Henrique Barboza 
15668b020b5eSDaniel Henrique Barboza     g_error_free(g_err);
15678b020b5eSDaniel Henrique Barboza     return -1;
15688b020b5eSDaniel Henrique Barboza }
15698b020b5eSDaniel Henrique Barboza 
1570067927d6SDaniel Henrique Barboza static bool systemd_supports_mode(SuspendMode mode, Error **errp)
1571067927d6SDaniel Henrique Barboza {
1572067927d6SDaniel Henrique Barboza     const char *systemctl_args[3] = {"systemd-hibernate", "systemd-suspend",
1573067927d6SDaniel Henrique Barboza                                      "systemd-hybrid-sleep"};
1574067927d6SDaniel Henrique Barboza     const char *cmd[4] = {"systemctl", "status", systemctl_args[mode], NULL};
1575067927d6SDaniel Henrique Barboza     int status;
1576067927d6SDaniel Henrique Barboza 
1577992861fbSMarkus Armbruster     status = run_process_child(cmd, errp);
1578067927d6SDaniel Henrique Barboza 
1579067927d6SDaniel Henrique Barboza     /*
1580067927d6SDaniel Henrique Barboza      * systemctl status uses LSB return codes so we can expect
1581067927d6SDaniel Henrique Barboza      * status > 0 and be ok. To assert if the guest has support
1582067927d6SDaniel Henrique Barboza      * for the selected suspend mode, status should be < 4. 4 is
1583067927d6SDaniel Henrique Barboza      * the code for unknown service status, the return value when
1584067927d6SDaniel Henrique Barboza      * the service does not exist. A common value is status = 3
1585067927d6SDaniel Henrique Barboza      * (program is not running).
1586067927d6SDaniel Henrique Barboza      */
1587067927d6SDaniel Henrique Barboza     if (status > 0 && status < 4) {
1588067927d6SDaniel Henrique Barboza         return true;
1589067927d6SDaniel Henrique Barboza     }
1590067927d6SDaniel Henrique Barboza 
1591067927d6SDaniel Henrique Barboza     return false;
1592067927d6SDaniel Henrique Barboza }
1593067927d6SDaniel Henrique Barboza 
1594067927d6SDaniel Henrique Barboza static void systemd_suspend(SuspendMode mode, Error **errp)
1595067927d6SDaniel Henrique Barboza {
1596067927d6SDaniel Henrique Barboza     Error *local_err = NULL;
1597067927d6SDaniel Henrique Barboza     const char *systemctl_args[3] = {"hibernate", "suspend", "hybrid-sleep"};
1598067927d6SDaniel Henrique Barboza     const char *cmd[3] = {"systemctl", systemctl_args[mode], NULL};
1599067927d6SDaniel Henrique Barboza     int status;
1600067927d6SDaniel Henrique Barboza 
1601067927d6SDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1602067927d6SDaniel Henrique Barboza 
1603067927d6SDaniel Henrique Barboza     if (status == 0) {
1604067927d6SDaniel Henrique Barboza         return;
1605067927d6SDaniel Henrique Barboza     }
1606067927d6SDaniel Henrique Barboza 
1607067927d6SDaniel Henrique Barboza     if ((status == -1) && !local_err) {
1608067927d6SDaniel Henrique Barboza         error_setg(errp, "the helper program 'systemctl %s' was not found",
1609067927d6SDaniel Henrique Barboza                    systemctl_args[mode]);
1610067927d6SDaniel Henrique Barboza         return;
1611067927d6SDaniel Henrique Barboza     }
1612067927d6SDaniel Henrique Barboza 
1613067927d6SDaniel Henrique Barboza     if (local_err) {
1614067927d6SDaniel Henrique Barboza         error_propagate(errp, local_err);
1615067927d6SDaniel Henrique Barboza     } else {
1616067927d6SDaniel Henrique Barboza         error_setg(errp, "the helper program 'systemctl %s' returned an "
1617067927d6SDaniel Henrique Barboza                    "unexpected exit status code (%d)",
1618067927d6SDaniel Henrique Barboza                    systemctl_args[mode], status);
1619067927d6SDaniel Henrique Barboza     }
1620067927d6SDaniel Henrique Barboza }
1621067927d6SDaniel Henrique Barboza 
16228b020b5eSDaniel Henrique Barboza static bool pmutils_supports_mode(SuspendMode mode, Error **errp)
162311d0f125SLuiz Capitulino {
16246b26e837SLuiz Capitulino     Error *local_err = NULL;
16258b020b5eSDaniel Henrique Barboza     const char *pmutils_args[3] = {"--hibernate", "--suspend",
16268b020b5eSDaniel Henrique Barboza                                    "--suspend-hybrid"};
16278b020b5eSDaniel Henrique Barboza     const char *cmd[3] = {"pm-is-supported", pmutils_args[mode], NULL};
1628dc8764f0SLuiz Capitulino     int status;
162911d0f125SLuiz Capitulino 
16308b020b5eSDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1631304a0fcbSDaniel Henrique Barboza 
16328b020b5eSDaniel Henrique Barboza     if (status == SUSPEND_SUPPORTED) {
16338b020b5eSDaniel Henrique Barboza         return true;
1634304a0fcbSDaniel Henrique Barboza     }
1635304a0fcbSDaniel Henrique Barboza 
16368b020b5eSDaniel Henrique Barboza     if ((status == -1) && !local_err) {
16378b020b5eSDaniel Henrique Barboza         return false;
1638a5fcf0e3SDaniel Henrique Barboza     }
163911d0f125SLuiz Capitulino 
164084d18f06SMarkus Armbruster     if (local_err) {
164177dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
16428b020b5eSDaniel Henrique Barboza     } else {
164377dbc81bSMarkus Armbruster         error_setg(errp,
16448b020b5eSDaniel Henrique Barboza                    "the helper program '%s' returned an unexpected exit"
16458b020b5eSDaniel Henrique Barboza                    " status code (%d)", "pm-is-supported", status);
1646dc8764f0SLuiz Capitulino     }
164711d0f125SLuiz Capitulino 
16488b020b5eSDaniel Henrique Barboza     return false;
1649a5fcf0e3SDaniel Henrique Barboza }
1650a5fcf0e3SDaniel Henrique Barboza 
16518b020b5eSDaniel Henrique Barboza static void pmutils_suspend(SuspendMode mode, Error **errp)
1652246d76ebSDaniel Henrique Barboza {
1653246d76ebSDaniel Henrique Barboza     Error *local_err = NULL;
16548b020b5eSDaniel Henrique Barboza     const char *pmutils_binaries[3] = {"pm-hibernate", "pm-suspend",
16558b020b5eSDaniel Henrique Barboza                                        "pm-suspend-hybrid"};
16568b020b5eSDaniel Henrique Barboza     const char *cmd[2] = {pmutils_binaries[mode], NULL};
1657246d76ebSDaniel Henrique Barboza     int status;
1658246d76ebSDaniel Henrique Barboza 
16598b020b5eSDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1660246d76ebSDaniel Henrique Barboza 
16618b020b5eSDaniel Henrique Barboza     if (status == 0) {
1662246d76ebSDaniel Henrique Barboza         return;
1663246d76ebSDaniel Henrique Barboza     }
1664246d76ebSDaniel Henrique Barboza 
16658b020b5eSDaniel Henrique Barboza     if ((status == -1) && !local_err) {
16668b020b5eSDaniel Henrique Barboza         error_setg(errp, "the helper program '%s' was not found",
16678b020b5eSDaniel Henrique Barboza                    pmutils_binaries[mode]);
1668246d76ebSDaniel Henrique Barboza         return;
1669246d76ebSDaniel Henrique Barboza     }
1670246d76ebSDaniel Henrique Barboza 
1671246d76ebSDaniel Henrique Barboza     if (local_err) {
1672246d76ebSDaniel Henrique Barboza         error_propagate(errp, local_err);
16738b020b5eSDaniel Henrique Barboza     } else {
1674246d76ebSDaniel Henrique Barboza         error_setg(errp,
16758b020b5eSDaniel Henrique Barboza                    "the helper program '%s' returned an unexpected exit"
16768b020b5eSDaniel Henrique Barboza                    " status code (%d)", pmutils_binaries[mode], status);
16778b020b5eSDaniel Henrique Barboza     }
1678246d76ebSDaniel Henrique Barboza }
1679246d76ebSDaniel Henrique Barboza 
16808b020b5eSDaniel Henrique Barboza static bool linux_sys_state_supports_mode(SuspendMode mode, Error **errp)
1681a5fcf0e3SDaniel Henrique Barboza {
16828b020b5eSDaniel Henrique Barboza     const char *sysfile_strs[3] = {"disk", "mem", NULL};
16838b020b5eSDaniel Henrique Barboza     const char *sysfile_str = sysfile_strs[mode];
1684a5fcf0e3SDaniel Henrique Barboza     char buf[32]; /* hopefully big enough */
1685a5fcf0e3SDaniel Henrique Barboza     int fd;
1686a5fcf0e3SDaniel Henrique Barboza     ssize_t ret;
1687a5fcf0e3SDaniel Henrique Barboza 
16888b020b5eSDaniel Henrique Barboza     if (!sysfile_str) {
16898b020b5eSDaniel Henrique Barboza         error_setg(errp, "unknown guest suspend mode");
1690a5fcf0e3SDaniel Henrique Barboza         return false;
1691a5fcf0e3SDaniel Henrique Barboza     }
1692a5fcf0e3SDaniel Henrique Barboza 
1693a5fcf0e3SDaniel Henrique Barboza     fd = open(LINUX_SYS_STATE_FILE, O_RDONLY);
1694a5fcf0e3SDaniel Henrique Barboza     if (fd < 0) {
1695a5fcf0e3SDaniel Henrique Barboza         return false;
1696a5fcf0e3SDaniel Henrique Barboza     }
1697a5fcf0e3SDaniel Henrique Barboza 
1698a5fcf0e3SDaniel Henrique Barboza     ret = read(fd, buf, sizeof(buf) - 1);
1699d9c745c1SPaolo Bonzini     close(fd);
1700a5fcf0e3SDaniel Henrique Barboza     if (ret <= 0) {
1701a5fcf0e3SDaniel Henrique Barboza         return false;
1702a5fcf0e3SDaniel Henrique Barboza     }
1703a5fcf0e3SDaniel Henrique Barboza     buf[ret] = '\0';
1704a5fcf0e3SDaniel Henrique Barboza 
1705a5fcf0e3SDaniel Henrique Barboza     if (strstr(buf, sysfile_str)) {
1706a5fcf0e3SDaniel Henrique Barboza         return true;
1707a5fcf0e3SDaniel Henrique Barboza     }
1708a5fcf0e3SDaniel Henrique Barboza     return false;
1709a5fcf0e3SDaniel Henrique Barboza }
1710a5fcf0e3SDaniel Henrique Barboza 
17118b020b5eSDaniel Henrique Barboza static void linux_sys_state_suspend(SuspendMode mode, Error **errp)
1712246d76ebSDaniel Henrique Barboza {
1713246d76ebSDaniel Henrique Barboza     Error *local_err = NULL;
17148b020b5eSDaniel Henrique Barboza     const char *sysfile_strs[3] = {"disk", "mem", NULL};
17158b020b5eSDaniel Henrique Barboza     const char *sysfile_str = sysfile_strs[mode];
1716246d76ebSDaniel Henrique Barboza     pid_t pid;
1717246d76ebSDaniel Henrique Barboza     int status;
1718246d76ebSDaniel Henrique Barboza 
17198b020b5eSDaniel Henrique Barboza     if (!sysfile_str) {
1720246d76ebSDaniel Henrique Barboza         error_setg(errp, "unknown guest suspend mode");
1721246d76ebSDaniel Henrique Barboza         return;
1722246d76ebSDaniel Henrique Barboza     }
1723246d76ebSDaniel Henrique Barboza 
1724246d76ebSDaniel Henrique Barboza     pid = fork();
1725246d76ebSDaniel Henrique Barboza     if (!pid) {
1726246d76ebSDaniel Henrique Barboza         /* child */
1727246d76ebSDaniel Henrique Barboza         int fd;
1728246d76ebSDaniel Henrique Barboza 
1729246d76ebSDaniel Henrique Barboza         setsid();
1730246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(0);
1731246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(1);
1732246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(2);
1733246d76ebSDaniel Henrique Barboza 
1734246d76ebSDaniel Henrique Barboza         fd = open(LINUX_SYS_STATE_FILE, O_WRONLY);
1735246d76ebSDaniel Henrique Barboza         if (fd < 0) {
1736246d76ebSDaniel Henrique Barboza             _exit(EXIT_FAILURE);
1737246d76ebSDaniel Henrique Barboza         }
1738246d76ebSDaniel Henrique Barboza 
1739246d76ebSDaniel Henrique Barboza         if (write(fd, sysfile_str, strlen(sysfile_str)) < 0) {
1740246d76ebSDaniel Henrique Barboza             _exit(EXIT_FAILURE);
1741246d76ebSDaniel Henrique Barboza         }
1742246d76ebSDaniel Henrique Barboza 
1743246d76ebSDaniel Henrique Barboza         _exit(EXIT_SUCCESS);
1744246d76ebSDaniel Henrique Barboza     } else if (pid < 0) {
1745246d76ebSDaniel Henrique Barboza         error_setg_errno(errp, errno, "failed to create child process");
1746246d76ebSDaniel Henrique Barboza         return;
1747246d76ebSDaniel Henrique Barboza     }
1748246d76ebSDaniel Henrique Barboza 
1749246d76ebSDaniel Henrique Barboza     ga_wait_child(pid, &status, &local_err);
1750246d76ebSDaniel Henrique Barboza     if (local_err) {
1751246d76ebSDaniel Henrique Barboza         error_propagate(errp, local_err);
1752246d76ebSDaniel Henrique Barboza         return;
1753246d76ebSDaniel Henrique Barboza     }
1754246d76ebSDaniel Henrique Barboza 
1755246d76ebSDaniel Henrique Barboza     if (WEXITSTATUS(status)) {
1756246d76ebSDaniel Henrique Barboza         error_setg(errp, "child process has failed to suspend");
1757246d76ebSDaniel Henrique Barboza     }
1758246d76ebSDaniel Henrique Barboza 
1759246d76ebSDaniel Henrique Barboza }
1760246d76ebSDaniel Henrique Barboza 
17618b020b5eSDaniel Henrique Barboza static void guest_suspend(SuspendMode mode, Error **errp)
176211d0f125SLuiz Capitulino {
17637b376087SLuiz Capitulino     Error *local_err = NULL;
176473e1d8ebSDaniel Henrique Barboza     bool mode_supported = false;
176511d0f125SLuiz Capitulino 
176673e1d8ebSDaniel Henrique Barboza     if (systemd_supports_mode(mode, &local_err)) {
176773e1d8ebSDaniel Henrique Barboza         mode_supported = true;
1768067927d6SDaniel Henrique Barboza         systemd_suspend(mode, &local_err);
176973e1d8ebSDaniel Henrique Barboza     }
177073e1d8ebSDaniel Henrique Barboza 
1771067927d6SDaniel Henrique Barboza     if (!local_err) {
1772067927d6SDaniel Henrique Barboza         return;
1773067927d6SDaniel Henrique Barboza     }
1774067927d6SDaniel Henrique Barboza 
1775067927d6SDaniel Henrique Barboza     error_free(local_err);
17766a4a3853SVladimir Sementsov-Ogievskiy     local_err = NULL;
1777067927d6SDaniel Henrique Barboza 
177873e1d8ebSDaniel Henrique Barboza     if (pmutils_supports_mode(mode, &local_err)) {
177973e1d8ebSDaniel Henrique Barboza         mode_supported = true;
17808b020b5eSDaniel Henrique Barboza         pmutils_suspend(mode, &local_err);
178173e1d8ebSDaniel Henrique Barboza     }
178273e1d8ebSDaniel Henrique Barboza 
1783246d76ebSDaniel Henrique Barboza     if (!local_err) {
1784304a0fcbSDaniel Henrique Barboza         return;
1785304a0fcbSDaniel Henrique Barboza     }
1786304a0fcbSDaniel Henrique Barboza 
1787246d76ebSDaniel Henrique Barboza     error_free(local_err);
17886a4a3853SVladimir Sementsov-Ogievskiy     local_err = NULL;
178911d0f125SLuiz Capitulino 
179073e1d8ebSDaniel Henrique Barboza     if (linux_sys_state_supports_mode(mode, &local_err)) {
179173e1d8ebSDaniel Henrique Barboza         mode_supported = true;
17928b020b5eSDaniel Henrique Barboza         linux_sys_state_suspend(mode, &local_err);
179373e1d8ebSDaniel Henrique Barboza     }
179473e1d8ebSDaniel Henrique Barboza 
179573e1d8ebSDaniel Henrique Barboza     if (!mode_supported) {
17966a4a3853SVladimir Sementsov-Ogievskiy         error_free(local_err);
179773e1d8ebSDaniel Henrique Barboza         error_setg(errp,
179873e1d8ebSDaniel Henrique Barboza                    "the requested suspend mode is not supported by the guest");
1799b2322003SMarkus Armbruster     } else {
180077dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
18017b376087SLuiz Capitulino     }
180211d0f125SLuiz Capitulino }
180311d0f125SLuiz Capitulino 
180477dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
180511d0f125SLuiz Capitulino {
1806304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_DISK, errp);
180711d0f125SLuiz Capitulino }
180811d0f125SLuiz Capitulino 
180977dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
1810fbf42210SLuiz Capitulino {
1811304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_RAM, errp);
1812fbf42210SLuiz Capitulino }
1813fbf42210SLuiz Capitulino 
181477dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
181595f4f404SLuiz Capitulino {
1816304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_HYBRID, errp);
181795f4f404SLuiz Capitulino }
181895f4f404SLuiz Capitulino 
18193424fc9fSMichal Privoznik static GuestNetworkInterfaceList *
18203424fc9fSMichal Privoznik guest_find_interface(GuestNetworkInterfaceList *head,
18213424fc9fSMichal Privoznik                      const char *name)
18223424fc9fSMichal Privoznik {
18233424fc9fSMichal Privoznik     for (; head; head = head->next) {
18243424fc9fSMichal Privoznik         if (strcmp(head->value->name, name) == 0) {
18253424fc9fSMichal Privoznik             break;
18263424fc9fSMichal Privoznik         }
18273424fc9fSMichal Privoznik     }
18283424fc9fSMichal Privoznik 
18293424fc9fSMichal Privoznik     return head;
18303424fc9fSMichal Privoznik }
18313424fc9fSMichal Privoznik 
183253f9fcb2SZhiPeng Lu static int guest_get_network_stats(const char *name,
183353f9fcb2SZhiPeng Lu                        GuestNetworkInterfaceStat *stats)
183453f9fcb2SZhiPeng Lu {
183553f9fcb2SZhiPeng Lu     int name_len;
183653f9fcb2SZhiPeng Lu     char const *devinfo = "/proc/net/dev";
183753f9fcb2SZhiPeng Lu     FILE *fp;
183853f9fcb2SZhiPeng Lu     char *line = NULL, *colon;
183953f9fcb2SZhiPeng Lu     size_t n = 0;
184053f9fcb2SZhiPeng Lu     fp = fopen(devinfo, "r");
184153f9fcb2SZhiPeng Lu     if (!fp) {
184253f9fcb2SZhiPeng Lu         return -1;
184353f9fcb2SZhiPeng Lu     }
184453f9fcb2SZhiPeng Lu     name_len = strlen(name);
184553f9fcb2SZhiPeng Lu     while (getline(&line, &n, fp) != -1) {
184653f9fcb2SZhiPeng Lu         long long dummy;
184753f9fcb2SZhiPeng Lu         long long rx_bytes;
184853f9fcb2SZhiPeng Lu         long long rx_packets;
184953f9fcb2SZhiPeng Lu         long long rx_errs;
185053f9fcb2SZhiPeng Lu         long long rx_dropped;
185153f9fcb2SZhiPeng Lu         long long tx_bytes;
185253f9fcb2SZhiPeng Lu         long long tx_packets;
185353f9fcb2SZhiPeng Lu         long long tx_errs;
185453f9fcb2SZhiPeng Lu         long long tx_dropped;
185553f9fcb2SZhiPeng Lu         char *trim_line;
185653f9fcb2SZhiPeng Lu         trim_line = g_strchug(line);
185753f9fcb2SZhiPeng Lu         if (trim_line[0] == '\0') {
185853f9fcb2SZhiPeng Lu             continue;
185953f9fcb2SZhiPeng Lu         }
186053f9fcb2SZhiPeng Lu         colon = strchr(trim_line, ':');
186153f9fcb2SZhiPeng Lu         if (!colon) {
186253f9fcb2SZhiPeng Lu             continue;
186353f9fcb2SZhiPeng Lu         }
186453f9fcb2SZhiPeng Lu         if (colon - name_len  == trim_line &&
186553f9fcb2SZhiPeng Lu            strncmp(trim_line, name, name_len) == 0) {
186653f9fcb2SZhiPeng Lu             if (sscanf(colon + 1,
186753f9fcb2SZhiPeng Lu                 "%lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld",
186853f9fcb2SZhiPeng Lu                   &rx_bytes, &rx_packets, &rx_errs, &rx_dropped,
186953f9fcb2SZhiPeng Lu                   &dummy, &dummy, &dummy, &dummy,
187053f9fcb2SZhiPeng Lu                   &tx_bytes, &tx_packets, &tx_errs, &tx_dropped,
187153f9fcb2SZhiPeng Lu                   &dummy, &dummy, &dummy, &dummy) != 16) {
187253f9fcb2SZhiPeng Lu                 continue;
187353f9fcb2SZhiPeng Lu             }
187453f9fcb2SZhiPeng Lu             stats->rx_bytes = rx_bytes;
187553f9fcb2SZhiPeng Lu             stats->rx_packets = rx_packets;
187653f9fcb2SZhiPeng Lu             stats->rx_errs = rx_errs;
187753f9fcb2SZhiPeng Lu             stats->rx_dropped = rx_dropped;
187853f9fcb2SZhiPeng Lu             stats->tx_bytes = tx_bytes;
187953f9fcb2SZhiPeng Lu             stats->tx_packets = tx_packets;
188053f9fcb2SZhiPeng Lu             stats->tx_errs = tx_errs;
188153f9fcb2SZhiPeng Lu             stats->tx_dropped = tx_dropped;
188253f9fcb2SZhiPeng Lu             fclose(fp);
188353f9fcb2SZhiPeng Lu             g_free(line);
188453f9fcb2SZhiPeng Lu             return 0;
188553f9fcb2SZhiPeng Lu         }
188653f9fcb2SZhiPeng Lu     }
188753f9fcb2SZhiPeng Lu     fclose(fp);
188853f9fcb2SZhiPeng Lu     g_free(line);
188953f9fcb2SZhiPeng Lu     g_debug("/proc/net/dev: Interface '%s' not found", name);
189053f9fcb2SZhiPeng Lu     return -1;
189153f9fcb2SZhiPeng Lu }
189253f9fcb2SZhiPeng Lu 
18933424fc9fSMichal Privoznik /*
18943424fc9fSMichal Privoznik  * Build information about guest interfaces
18953424fc9fSMichal Privoznik  */
18963424fc9fSMichal Privoznik GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
18973424fc9fSMichal Privoznik {
18983424fc9fSMichal Privoznik     GuestNetworkInterfaceList *head = NULL, *cur_item = NULL;
18993424fc9fSMichal Privoznik     struct ifaddrs *ifap, *ifa;
19003424fc9fSMichal Privoznik 
19013424fc9fSMichal Privoznik     if (getifaddrs(&ifap) < 0) {
1902878a0ae0SLuiz Capitulino         error_setg_errno(errp, errno, "getifaddrs failed");
19033424fc9fSMichal Privoznik         goto error;
19043424fc9fSMichal Privoznik     }
19053424fc9fSMichal Privoznik 
19063424fc9fSMichal Privoznik     for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
19073424fc9fSMichal Privoznik         GuestNetworkInterfaceList *info;
19083424fc9fSMichal Privoznik         GuestIpAddressList **address_list = NULL, *address_item = NULL;
190953f9fcb2SZhiPeng Lu         GuestNetworkInterfaceStat  *interface_stat = NULL;
19103424fc9fSMichal Privoznik         char addr4[INET_ADDRSTRLEN];
19113424fc9fSMichal Privoznik         char addr6[INET6_ADDRSTRLEN];
19123424fc9fSMichal Privoznik         int sock;
19133424fc9fSMichal Privoznik         struct ifreq ifr;
19143424fc9fSMichal Privoznik         unsigned char *mac_addr;
19153424fc9fSMichal Privoznik         void *p;
19163424fc9fSMichal Privoznik 
19173424fc9fSMichal Privoznik         g_debug("Processing %s interface", ifa->ifa_name);
19183424fc9fSMichal Privoznik 
19193424fc9fSMichal Privoznik         info = guest_find_interface(head, ifa->ifa_name);
19203424fc9fSMichal Privoznik 
19213424fc9fSMichal Privoznik         if (!info) {
19223424fc9fSMichal Privoznik             info = g_malloc0(sizeof(*info));
19233424fc9fSMichal Privoznik             info->value = g_malloc0(sizeof(*info->value));
19243424fc9fSMichal Privoznik             info->value->name = g_strdup(ifa->ifa_name);
19253424fc9fSMichal Privoznik 
19263424fc9fSMichal Privoznik             if (!cur_item) {
19273424fc9fSMichal Privoznik                 head = cur_item = info;
19283424fc9fSMichal Privoznik             } else {
19293424fc9fSMichal Privoznik                 cur_item->next = info;
19303424fc9fSMichal Privoznik                 cur_item = info;
19313424fc9fSMichal Privoznik             }
19323424fc9fSMichal Privoznik         }
19333424fc9fSMichal Privoznik 
19343424fc9fSMichal Privoznik         if (!info->value->has_hardware_address &&
19353424fc9fSMichal Privoznik             ifa->ifa_flags & SIOCGIFHWADDR) {
19363424fc9fSMichal Privoznik             /* we haven't obtained HW address yet */
19373424fc9fSMichal Privoznik             sock = socket(PF_INET, SOCK_STREAM, 0);
19383424fc9fSMichal Privoznik             if (sock == -1) {
1939878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "failed to create socket");
19403424fc9fSMichal Privoznik                 goto error;
19413424fc9fSMichal Privoznik             }
19423424fc9fSMichal Privoznik 
19433424fc9fSMichal Privoznik             memset(&ifr, 0, sizeof(ifr));
19441ab516edSJim Meyering             pstrcpy(ifr.ifr_name, IF_NAMESIZE, info->value->name);
19453424fc9fSMichal Privoznik             if (ioctl(sock, SIOCGIFHWADDR, &ifr) == -1) {
1946878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno,
1947878a0ae0SLuiz Capitulino                                  "failed to get MAC address of %s",
1948878a0ae0SLuiz Capitulino                                  ifa->ifa_name);
194910a2158fSMarkus Armbruster                 close(sock);
19503424fc9fSMichal Privoznik                 goto error;
19513424fc9fSMichal Privoznik             }
19523424fc9fSMichal Privoznik 
195310a2158fSMarkus Armbruster             close(sock);
19543424fc9fSMichal Privoznik             mac_addr = (unsigned char *) &ifr.ifr_hwaddr.sa_data;
19553424fc9fSMichal Privoznik 
1956e4ada482SStefan Weil             info->value->hardware_address =
1957e4ada482SStefan Weil                 g_strdup_printf("%02x:%02x:%02x:%02x:%02x:%02x",
19583424fc9fSMichal Privoznik                                 (int) mac_addr[0], (int) mac_addr[1],
19593424fc9fSMichal Privoznik                                 (int) mac_addr[2], (int) mac_addr[3],
1960e4ada482SStefan Weil                                 (int) mac_addr[4], (int) mac_addr[5]);
19613424fc9fSMichal Privoznik 
19623424fc9fSMichal Privoznik             info->value->has_hardware_address = true;
19633424fc9fSMichal Privoznik         }
19643424fc9fSMichal Privoznik 
19653424fc9fSMichal Privoznik         if (ifa->ifa_addr &&
19663424fc9fSMichal Privoznik             ifa->ifa_addr->sa_family == AF_INET) {
19673424fc9fSMichal Privoznik             /* interface with IPv4 address */
19683424fc9fSMichal Privoznik             p = &((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
19693424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET, p, addr4, sizeof(addr4))) {
1970878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
19713424fc9fSMichal Privoznik                 goto error;
19723424fc9fSMichal Privoznik             }
19733424fc9fSMichal Privoznik 
197410a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
197510a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
19763424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr4);
19773424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV4;
19783424fc9fSMichal Privoznik 
19793424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
19803424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
19813424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
19823424fc9fSMichal Privoznik                 p = &((struct sockaddr_in *)ifa->ifa_netmask)->sin_addr;
19833424fc9fSMichal Privoznik                 address_item->value->prefix = ctpop32(((uint32_t *) p)[0]);
19843424fc9fSMichal Privoznik             }
19853424fc9fSMichal Privoznik         } else if (ifa->ifa_addr &&
19863424fc9fSMichal Privoznik                    ifa->ifa_addr->sa_family == AF_INET6) {
19873424fc9fSMichal Privoznik             /* interface with IPv6 address */
19883424fc9fSMichal Privoznik             p = &((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
19893424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET6, p, addr6, sizeof(addr6))) {
1990878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
19913424fc9fSMichal Privoznik                 goto error;
19923424fc9fSMichal Privoznik             }
19933424fc9fSMichal Privoznik 
199410a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
199510a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
19963424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr6);
19973424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV6;
19983424fc9fSMichal Privoznik 
19993424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
20003424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
20013424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
20023424fc9fSMichal Privoznik                 p = &((struct sockaddr_in6 *)ifa->ifa_netmask)->sin6_addr;
20033424fc9fSMichal Privoznik                 address_item->value->prefix =
20043424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[0]) +
20053424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[1]) +
20063424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[2]) +
20073424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[3]);
20083424fc9fSMichal Privoznik             }
20093424fc9fSMichal Privoznik         }
20103424fc9fSMichal Privoznik 
20113424fc9fSMichal Privoznik         if (!address_item) {
20123424fc9fSMichal Privoznik             continue;
20133424fc9fSMichal Privoznik         }
20143424fc9fSMichal Privoznik 
20153424fc9fSMichal Privoznik         address_list = &info->value->ip_addresses;
20163424fc9fSMichal Privoznik 
20173424fc9fSMichal Privoznik         while (*address_list && (*address_list)->next) {
20183424fc9fSMichal Privoznik             address_list = &(*address_list)->next;
20193424fc9fSMichal Privoznik         }
20203424fc9fSMichal Privoznik 
20213424fc9fSMichal Privoznik         if (!*address_list) {
20223424fc9fSMichal Privoznik             *address_list = address_item;
20233424fc9fSMichal Privoznik         } else {
20243424fc9fSMichal Privoznik             (*address_list)->next = address_item;
20253424fc9fSMichal Privoznik         }
20263424fc9fSMichal Privoznik 
20273424fc9fSMichal Privoznik         info->value->has_ip_addresses = true;
20283424fc9fSMichal Privoznik 
202953f9fcb2SZhiPeng Lu         if (!info->value->has_statistics) {
203053f9fcb2SZhiPeng Lu             interface_stat = g_malloc0(sizeof(*interface_stat));
203153f9fcb2SZhiPeng Lu             if (guest_get_network_stats(info->value->name,
203253f9fcb2SZhiPeng Lu                 interface_stat) == -1) {
203353f9fcb2SZhiPeng Lu                 info->value->has_statistics = false;
203453f9fcb2SZhiPeng Lu                 g_free(interface_stat);
203553f9fcb2SZhiPeng Lu             } else {
203653f9fcb2SZhiPeng Lu                 info->value->statistics = interface_stat;
203753f9fcb2SZhiPeng Lu                 info->value->has_statistics = true;
203853f9fcb2SZhiPeng Lu             }
203953f9fcb2SZhiPeng Lu         }
20403424fc9fSMichal Privoznik     }
20413424fc9fSMichal Privoznik 
20423424fc9fSMichal Privoznik     freeifaddrs(ifap);
20433424fc9fSMichal Privoznik     return head;
20443424fc9fSMichal Privoznik 
20453424fc9fSMichal Privoznik error:
20463424fc9fSMichal Privoznik     freeifaddrs(ifap);
20473424fc9fSMichal Privoznik     qapi_free_GuestNetworkInterfaceList(head);
20483424fc9fSMichal Privoznik     return NULL;
20493424fc9fSMichal Privoznik }
20503424fc9fSMichal Privoznik 
205177dbc81bSMarkus Armbruster #define SYSCONF_EXACT(name, errp) sysconf_exact((name), #name, (errp))
2052d2baff62SLaszlo Ersek 
205377dbc81bSMarkus Armbruster static long sysconf_exact(int name, const char *name_str, Error **errp)
2054d2baff62SLaszlo Ersek {
2055d2baff62SLaszlo Ersek     long ret;
2056d2baff62SLaszlo Ersek 
2057d2baff62SLaszlo Ersek     errno = 0;
2058d2baff62SLaszlo Ersek     ret = sysconf(name);
2059d2baff62SLaszlo Ersek     if (ret == -1) {
2060d2baff62SLaszlo Ersek         if (errno == 0) {
206177dbc81bSMarkus Armbruster             error_setg(errp, "sysconf(%s): value indefinite", name_str);
2062d2baff62SLaszlo Ersek         } else {
206377dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "sysconf(%s)", name_str);
2064d2baff62SLaszlo Ersek         }
2065d2baff62SLaszlo Ersek     }
2066d2baff62SLaszlo Ersek     return ret;
2067d2baff62SLaszlo Ersek }
2068d2baff62SLaszlo Ersek 
2069d2baff62SLaszlo Ersek /* Transfer online/offline status between @vcpu and the guest system.
2070d2baff62SLaszlo Ersek  *
2071d2baff62SLaszlo Ersek  * On input either @errp or *@errp must be NULL.
2072d2baff62SLaszlo Ersek  *
2073d2baff62SLaszlo Ersek  * In system-to-@vcpu direction, the following @vcpu fields are accessed:
2074d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
2075d2baff62SLaszlo Ersek  * - W: vcpu->online
2076d2baff62SLaszlo Ersek  * - W: vcpu->can_offline
2077d2baff62SLaszlo Ersek  *
2078d2baff62SLaszlo Ersek  * In @vcpu-to-system direction, the following @vcpu fields are accessed:
2079d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
2080d2baff62SLaszlo Ersek  * - R: vcpu->online
2081d2baff62SLaszlo Ersek  *
2082d2baff62SLaszlo Ersek  * Written members remain unmodified on error.
2083d2baff62SLaszlo Ersek  */
2084d2baff62SLaszlo Ersek static void transfer_vcpu(GuestLogicalProcessor *vcpu, bool sys2vcpu,
2085b4bf912aSIgor Mammedov                           char *dirpath, Error **errp)
2086d2baff62SLaszlo Ersek {
2087b4bf912aSIgor Mammedov     int fd;
2088b4bf912aSIgor Mammedov     int res;
2089d2baff62SLaszlo Ersek     int dirfd;
2090b4bf912aSIgor Mammedov     static const char fn[] = "online";
2091d2baff62SLaszlo Ersek 
2092d2baff62SLaszlo Ersek     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2093d2baff62SLaszlo Ersek     if (dirfd == -1) {
2094d2baff62SLaszlo Ersek         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2095b4bf912aSIgor Mammedov         return;
2096b4bf912aSIgor Mammedov     }
2097d2baff62SLaszlo Ersek 
2098d2baff62SLaszlo Ersek     fd = openat(dirfd, fn, sys2vcpu ? O_RDONLY : O_RDWR);
2099d2baff62SLaszlo Ersek     if (fd == -1) {
2100d2baff62SLaszlo Ersek         if (errno != ENOENT) {
2101d2baff62SLaszlo Ersek             error_setg_errno(errp, errno, "open(\"%s/%s\")", dirpath, fn);
2102d2baff62SLaszlo Ersek         } else if (sys2vcpu) {
2103d2baff62SLaszlo Ersek             vcpu->online = true;
2104d2baff62SLaszlo Ersek             vcpu->can_offline = false;
2105d2baff62SLaszlo Ersek         } else if (!vcpu->online) {
2106d2baff62SLaszlo Ersek             error_setg(errp, "logical processor #%" PRId64 " can't be "
2107d2baff62SLaszlo Ersek                        "offlined", vcpu->logical_id);
2108d2baff62SLaszlo Ersek         } /* otherwise pretend successful re-onlining */
2109d2baff62SLaszlo Ersek     } else {
2110d2baff62SLaszlo Ersek         unsigned char status;
2111d2baff62SLaszlo Ersek 
2112d2baff62SLaszlo Ersek         res = pread(fd, &status, 1, 0);
2113d2baff62SLaszlo Ersek         if (res == -1) {
2114d2baff62SLaszlo Ersek             error_setg_errno(errp, errno, "pread(\"%s/%s\")", dirpath, fn);
2115d2baff62SLaszlo Ersek         } else if (res == 0) {
2116d2baff62SLaszlo Ersek             error_setg(errp, "pread(\"%s/%s\"): unexpected EOF", dirpath,
2117d2baff62SLaszlo Ersek                        fn);
2118d2baff62SLaszlo Ersek         } else if (sys2vcpu) {
2119d2baff62SLaszlo Ersek             vcpu->online = (status != '0');
2120d2baff62SLaszlo Ersek             vcpu->can_offline = true;
2121d2baff62SLaszlo Ersek         } else if (vcpu->online != (status != '0')) {
2122d2baff62SLaszlo Ersek             status = '0' + vcpu->online;
2123d2baff62SLaszlo Ersek             if (pwrite(fd, &status, 1, 0) == -1) {
2124d2baff62SLaszlo Ersek                 error_setg_errno(errp, errno, "pwrite(\"%s/%s\")", dirpath,
2125d2baff62SLaszlo Ersek                                  fn);
2126d2baff62SLaszlo Ersek             }
2127d2baff62SLaszlo Ersek         } /* otherwise pretend successful re-(on|off)-lining */
2128d2baff62SLaszlo Ersek 
2129d2baff62SLaszlo Ersek         res = close(fd);
2130d2baff62SLaszlo Ersek         g_assert(res == 0);
2131d2baff62SLaszlo Ersek     }
2132d2baff62SLaszlo Ersek 
2133d2baff62SLaszlo Ersek     res = close(dirfd);
2134d2baff62SLaszlo Ersek     g_assert(res == 0);
2135d2baff62SLaszlo Ersek }
2136d2baff62SLaszlo Ersek 
2137d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2138d2baff62SLaszlo Ersek {
2139d2baff62SLaszlo Ersek     int64_t current;
2140d2baff62SLaszlo Ersek     GuestLogicalProcessorList *head, **link;
2141d2baff62SLaszlo Ersek     long sc_max;
2142d2baff62SLaszlo Ersek     Error *local_err = NULL;
2143d2baff62SLaszlo Ersek 
2144d2baff62SLaszlo Ersek     current = 0;
2145d2baff62SLaszlo Ersek     head = NULL;
2146d2baff62SLaszlo Ersek     link = &head;
2147d2baff62SLaszlo Ersek     sc_max = SYSCONF_EXACT(_SC_NPROCESSORS_CONF, &local_err);
2148d2baff62SLaszlo Ersek 
2149d2baff62SLaszlo Ersek     while (local_err == NULL && current < sc_max) {
2150d2baff62SLaszlo Ersek         GuestLogicalProcessor *vcpu;
2151d2baff62SLaszlo Ersek         GuestLogicalProcessorList *entry;
2152b4bf912aSIgor Mammedov         int64_t id = current++;
2153b4bf912aSIgor Mammedov         char *path = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
2154b4bf912aSIgor Mammedov                                      id);
2155d2baff62SLaszlo Ersek 
2156b4bf912aSIgor Mammedov         if (g_file_test(path, G_FILE_TEST_EXISTS)) {
2157d2baff62SLaszlo Ersek             vcpu = g_malloc0(sizeof *vcpu);
2158b4bf912aSIgor Mammedov             vcpu->logical_id = id;
2159d2baff62SLaszlo Ersek             vcpu->has_can_offline = true; /* lolspeak ftw */
2160b4bf912aSIgor Mammedov             transfer_vcpu(vcpu, true, path, &local_err);
2161d2baff62SLaszlo Ersek             entry = g_malloc0(sizeof *entry);
2162d2baff62SLaszlo Ersek             entry->value = vcpu;
2163d2baff62SLaszlo Ersek             *link = entry;
2164d2baff62SLaszlo Ersek             link = &entry->next;
2165d2baff62SLaszlo Ersek         }
2166b4bf912aSIgor Mammedov         g_free(path);
2167b4bf912aSIgor Mammedov     }
2168d2baff62SLaszlo Ersek 
2169d2baff62SLaszlo Ersek     if (local_err == NULL) {
2170d2baff62SLaszlo Ersek         /* there's no guest with zero VCPUs */
2171d2baff62SLaszlo Ersek         g_assert(head != NULL);
2172d2baff62SLaszlo Ersek         return head;
2173d2baff62SLaszlo Ersek     }
2174d2baff62SLaszlo Ersek 
2175d2baff62SLaszlo Ersek     qapi_free_GuestLogicalProcessorList(head);
2176d2baff62SLaszlo Ersek     error_propagate(errp, local_err);
2177d2baff62SLaszlo Ersek     return NULL;
2178d2baff62SLaszlo Ersek }
2179d2baff62SLaszlo Ersek 
2180cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2181cbb65fc2SLaszlo Ersek {
2182cbb65fc2SLaszlo Ersek     int64_t processed;
2183cbb65fc2SLaszlo Ersek     Error *local_err = NULL;
2184cbb65fc2SLaszlo Ersek 
2185cbb65fc2SLaszlo Ersek     processed = 0;
2186cbb65fc2SLaszlo Ersek     while (vcpus != NULL) {
2187b4bf912aSIgor Mammedov         char *path = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
2188b4bf912aSIgor Mammedov                                      vcpus->value->logical_id);
2189b4bf912aSIgor Mammedov 
2190b4bf912aSIgor Mammedov         transfer_vcpu(vcpus->value, false, path, &local_err);
2191b4bf912aSIgor Mammedov         g_free(path);
2192cbb65fc2SLaszlo Ersek         if (local_err != NULL) {
2193cbb65fc2SLaszlo Ersek             break;
2194cbb65fc2SLaszlo Ersek         }
2195cbb65fc2SLaszlo Ersek         ++processed;
2196cbb65fc2SLaszlo Ersek         vcpus = vcpus->next;
2197cbb65fc2SLaszlo Ersek     }
2198cbb65fc2SLaszlo Ersek 
2199cbb65fc2SLaszlo Ersek     if (local_err != NULL) {
2200cbb65fc2SLaszlo Ersek         if (processed == 0) {
2201cbb65fc2SLaszlo Ersek             error_propagate(errp, local_err);
2202cbb65fc2SLaszlo Ersek         } else {
2203cbb65fc2SLaszlo Ersek             error_free(local_err);
2204cbb65fc2SLaszlo Ersek         }
2205cbb65fc2SLaszlo Ersek     }
2206cbb65fc2SLaszlo Ersek 
2207cbb65fc2SLaszlo Ersek     return processed;
2208cbb65fc2SLaszlo Ersek }
2209cbb65fc2SLaszlo Ersek 
2210215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2211215a2771SDaniel P. Berrange                                  const char *password,
2212215a2771SDaniel P. Berrange                                  bool crypted,
2213215a2771SDaniel P. Berrange                                  Error **errp)
2214215a2771SDaniel P. Berrange {
2215215a2771SDaniel P. Berrange     Error *local_err = NULL;
2216215a2771SDaniel P. Berrange     char *passwd_path = NULL;
2217215a2771SDaniel P. Berrange     pid_t pid;
2218215a2771SDaniel P. Berrange     int status;
2219215a2771SDaniel P. Berrange     int datafd[2] = { -1, -1 };
2220215a2771SDaniel P. Berrange     char *rawpasswddata = NULL;
2221215a2771SDaniel P. Berrange     size_t rawpasswdlen;
2222215a2771SDaniel P. Berrange     char *chpasswddata = NULL;
2223215a2771SDaniel P. Berrange     size_t chpasswdlen;
2224215a2771SDaniel P. Berrange 
2225920639caSDaniel P. Berrange     rawpasswddata = (char *)qbase64_decode(password, -1, &rawpasswdlen, errp);
2226920639caSDaniel P. Berrange     if (!rawpasswddata) {
2227920639caSDaniel P. Berrange         return;
2228920639caSDaniel P. Berrange     }
2229215a2771SDaniel P. Berrange     rawpasswddata = g_renew(char, rawpasswddata, rawpasswdlen + 1);
2230215a2771SDaniel P. Berrange     rawpasswddata[rawpasswdlen] = '\0';
2231215a2771SDaniel P. Berrange 
2232215a2771SDaniel P. Berrange     if (strchr(rawpasswddata, '\n')) {
2233215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in raw password");
2234215a2771SDaniel P. Berrange         goto out;
2235215a2771SDaniel P. Berrange     }
2236215a2771SDaniel P. Berrange 
2237215a2771SDaniel P. Berrange     if (strchr(username, '\n') ||
2238215a2771SDaniel P. Berrange         strchr(username, ':')) {
2239215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in username");
2240215a2771SDaniel P. Berrange         goto out;
2241215a2771SDaniel P. Berrange     }
2242215a2771SDaniel P. Berrange 
2243215a2771SDaniel P. Berrange     chpasswddata = g_strdup_printf("%s:%s\n", username, rawpasswddata);
2244215a2771SDaniel P. Berrange     chpasswdlen = strlen(chpasswddata);
2245215a2771SDaniel P. Berrange 
2246215a2771SDaniel P. Berrange     passwd_path = g_find_program_in_path("chpasswd");
2247215a2771SDaniel P. Berrange 
2248215a2771SDaniel P. Berrange     if (!passwd_path) {
2249215a2771SDaniel P. Berrange         error_setg(errp, "cannot find 'passwd' program in PATH");
2250215a2771SDaniel P. Berrange         goto out;
2251215a2771SDaniel P. Berrange     }
2252215a2771SDaniel P. Berrange 
2253215a2771SDaniel P. Berrange     if (pipe(datafd) < 0) {
2254215a2771SDaniel P. Berrange         error_setg(errp, "cannot create pipe FDs");
2255215a2771SDaniel P. Berrange         goto out;
2256215a2771SDaniel P. Berrange     }
2257215a2771SDaniel P. Berrange 
2258215a2771SDaniel P. Berrange     pid = fork();
2259215a2771SDaniel P. Berrange     if (pid == 0) {
2260215a2771SDaniel P. Berrange         close(datafd[1]);
2261215a2771SDaniel P. Berrange         /* child */
2262215a2771SDaniel P. Berrange         setsid();
2263215a2771SDaniel P. Berrange         dup2(datafd[0], 0);
2264215a2771SDaniel P. Berrange         reopen_fd_to_null(1);
2265215a2771SDaniel P. Berrange         reopen_fd_to_null(2);
2266215a2771SDaniel P. Berrange 
2267215a2771SDaniel P. Berrange         if (crypted) {
2268215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", "-e", NULL, environ);
2269215a2771SDaniel P. Berrange         } else {
2270215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", NULL, environ);
2271215a2771SDaniel P. Berrange         }
2272215a2771SDaniel P. Berrange         _exit(EXIT_FAILURE);
2273215a2771SDaniel P. Berrange     } else if (pid < 0) {
2274215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "failed to create child process");
2275215a2771SDaniel P. Berrange         goto out;
2276215a2771SDaniel P. Berrange     }
2277215a2771SDaniel P. Berrange     close(datafd[0]);
2278215a2771SDaniel P. Berrange     datafd[0] = -1;
2279215a2771SDaniel P. Berrange 
2280215a2771SDaniel P. Berrange     if (qemu_write_full(datafd[1], chpasswddata, chpasswdlen) != chpasswdlen) {
2281215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "cannot write new account password");
2282215a2771SDaniel P. Berrange         goto out;
2283215a2771SDaniel P. Berrange     }
2284215a2771SDaniel P. Berrange     close(datafd[1]);
2285215a2771SDaniel P. Berrange     datafd[1] = -1;
2286215a2771SDaniel P. Berrange 
2287215a2771SDaniel P. Berrange     ga_wait_child(pid, &status, &local_err);
2288215a2771SDaniel P. Berrange     if (local_err) {
2289215a2771SDaniel P. Berrange         error_propagate(errp, local_err);
2290215a2771SDaniel P. Berrange         goto out;
2291215a2771SDaniel P. Berrange     }
2292215a2771SDaniel P. Berrange 
2293215a2771SDaniel P. Berrange     if (!WIFEXITED(status)) {
2294215a2771SDaniel P. Berrange         error_setg(errp, "child process has terminated abnormally");
2295215a2771SDaniel P. Berrange         goto out;
2296215a2771SDaniel P. Berrange     }
2297215a2771SDaniel P. Berrange 
2298215a2771SDaniel P. Berrange     if (WEXITSTATUS(status)) {
2299215a2771SDaniel P. Berrange         error_setg(errp, "child process has failed to set user password");
2300215a2771SDaniel P. Berrange         goto out;
2301215a2771SDaniel P. Berrange     }
2302215a2771SDaniel P. Berrange 
2303215a2771SDaniel P. Berrange out:
2304215a2771SDaniel P. Berrange     g_free(chpasswddata);
2305215a2771SDaniel P. Berrange     g_free(rawpasswddata);
2306215a2771SDaniel P. Berrange     g_free(passwd_path);
2307215a2771SDaniel P. Berrange     if (datafd[0] != -1) {
2308215a2771SDaniel P. Berrange         close(datafd[0]);
2309215a2771SDaniel P. Berrange     }
2310215a2771SDaniel P. Berrange     if (datafd[1] != -1) {
2311215a2771SDaniel P. Berrange         close(datafd[1]);
2312215a2771SDaniel P. Berrange     }
2313215a2771SDaniel P. Berrange }
2314215a2771SDaniel P. Berrange 
2315bd240fcaSzhanghailiang static void ga_read_sysfs_file(int dirfd, const char *pathname, char *buf,
2316bd240fcaSzhanghailiang                                int size, Error **errp)
2317bd240fcaSzhanghailiang {
2318bd240fcaSzhanghailiang     int fd;
2319bd240fcaSzhanghailiang     int res;
2320bd240fcaSzhanghailiang 
2321bd240fcaSzhanghailiang     errno = 0;
2322bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_RDONLY);
2323bd240fcaSzhanghailiang     if (fd == -1) {
2324bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2325bd240fcaSzhanghailiang         return;
2326bd240fcaSzhanghailiang     }
2327bd240fcaSzhanghailiang 
2328bd240fcaSzhanghailiang     res = pread(fd, buf, size, 0);
2329bd240fcaSzhanghailiang     if (res == -1) {
2330bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pread sysfs file \"%s\"", pathname);
2331bd240fcaSzhanghailiang     } else if (res == 0) {
2332bd240fcaSzhanghailiang         error_setg(errp, "pread sysfs file \"%s\": unexpected EOF", pathname);
2333bd240fcaSzhanghailiang     }
2334bd240fcaSzhanghailiang     close(fd);
2335bd240fcaSzhanghailiang }
2336bd240fcaSzhanghailiang 
2337bd240fcaSzhanghailiang static void ga_write_sysfs_file(int dirfd, const char *pathname,
2338bd240fcaSzhanghailiang                                 const char *buf, int size, Error **errp)
2339bd240fcaSzhanghailiang {
2340bd240fcaSzhanghailiang     int fd;
2341bd240fcaSzhanghailiang 
2342bd240fcaSzhanghailiang     errno = 0;
2343bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_WRONLY);
2344bd240fcaSzhanghailiang     if (fd == -1) {
2345bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2346bd240fcaSzhanghailiang         return;
2347bd240fcaSzhanghailiang     }
2348bd240fcaSzhanghailiang 
2349bd240fcaSzhanghailiang     if (pwrite(fd, buf, size, 0) == -1) {
2350bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pwrite sysfs file \"%s\"", pathname);
2351bd240fcaSzhanghailiang     }
2352bd240fcaSzhanghailiang 
2353bd240fcaSzhanghailiang     close(fd);
2354bd240fcaSzhanghailiang }
2355bd240fcaSzhanghailiang 
2356bd240fcaSzhanghailiang /* Transfer online/offline status between @mem_blk and the guest system.
2357bd240fcaSzhanghailiang  *
2358bd240fcaSzhanghailiang  * On input either @errp or *@errp must be NULL.
2359bd240fcaSzhanghailiang  *
2360bd240fcaSzhanghailiang  * In system-to-@mem_blk direction, the following @mem_blk fields are accessed:
2361bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2362bd240fcaSzhanghailiang  * - W: mem_blk->online
2363bd240fcaSzhanghailiang  * - W: mem_blk->can_offline
2364bd240fcaSzhanghailiang  *
2365bd240fcaSzhanghailiang  * In @mem_blk-to-system direction, the following @mem_blk fields are accessed:
2366bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2367bd240fcaSzhanghailiang  * - R: mem_blk->online
2368bd240fcaSzhanghailiang  *-  R: mem_blk->can_offline
2369bd240fcaSzhanghailiang  * Written members remain unmodified on error.
2370bd240fcaSzhanghailiang  */
2371bd240fcaSzhanghailiang static void transfer_memory_block(GuestMemoryBlock *mem_blk, bool sys2memblk,
2372bd240fcaSzhanghailiang                                   GuestMemoryBlockResponse *result,
2373bd240fcaSzhanghailiang                                   Error **errp)
2374bd240fcaSzhanghailiang {
2375bd240fcaSzhanghailiang     char *dirpath;
2376bd240fcaSzhanghailiang     int dirfd;
2377bd240fcaSzhanghailiang     char *status;
2378bd240fcaSzhanghailiang     Error *local_err = NULL;
2379bd240fcaSzhanghailiang 
2380bd240fcaSzhanghailiang     if (!sys2memblk) {
2381bd240fcaSzhanghailiang         DIR *dp;
2382bd240fcaSzhanghailiang 
2383bd240fcaSzhanghailiang         if (!result) {
2384bd240fcaSzhanghailiang             error_setg(errp, "Internal error, 'result' should not be NULL");
2385bd240fcaSzhanghailiang             return;
2386bd240fcaSzhanghailiang         }
2387bd240fcaSzhanghailiang         errno = 0;
2388bd240fcaSzhanghailiang         dp = opendir("/sys/devices/system/memory/");
2389bd240fcaSzhanghailiang          /* if there is no 'memory' directory in sysfs,
2390bd240fcaSzhanghailiang          * we think this VM does not support online/offline memory block,
2391bd240fcaSzhanghailiang          * any other solution?
2392bd240fcaSzhanghailiang          */
23939879f5acSPhilippe Mathieu-Daudé         if (!dp) {
23949879f5acSPhilippe Mathieu-Daudé             if (errno == ENOENT) {
2395bd240fcaSzhanghailiang                 result->response =
2396bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
23979879f5acSPhilippe Mathieu-Daudé             }
2398bd240fcaSzhanghailiang             goto out1;
2399bd240fcaSzhanghailiang         }
2400bd240fcaSzhanghailiang         closedir(dp);
2401bd240fcaSzhanghailiang     }
2402bd240fcaSzhanghailiang 
2403bd240fcaSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/memory%" PRId64 "/",
2404bd240fcaSzhanghailiang                               mem_blk->phys_index);
2405bd240fcaSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2406bd240fcaSzhanghailiang     if (dirfd == -1) {
2407bd240fcaSzhanghailiang         if (sys2memblk) {
2408bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2409bd240fcaSzhanghailiang         } else {
2410bd240fcaSzhanghailiang             if (errno == ENOENT) {
2411bd240fcaSzhanghailiang                 result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_NOT_FOUND;
2412bd240fcaSzhanghailiang             } else {
2413bd240fcaSzhanghailiang                 result->response =
2414bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2415bd240fcaSzhanghailiang             }
2416bd240fcaSzhanghailiang         }
2417bd240fcaSzhanghailiang         g_free(dirpath);
2418bd240fcaSzhanghailiang         goto out1;
2419bd240fcaSzhanghailiang     }
2420bd240fcaSzhanghailiang     g_free(dirpath);
2421bd240fcaSzhanghailiang 
2422bd240fcaSzhanghailiang     status = g_malloc0(10);
2423bd240fcaSzhanghailiang     ga_read_sysfs_file(dirfd, "state", status, 10, &local_err);
2424bd240fcaSzhanghailiang     if (local_err) {
2425bd240fcaSzhanghailiang         /* treat with sysfs file that not exist in old kernel */
2426bd240fcaSzhanghailiang         if (errno == ENOENT) {
2427bd240fcaSzhanghailiang             error_free(local_err);
2428bd240fcaSzhanghailiang             if (sys2memblk) {
2429bd240fcaSzhanghailiang                 mem_blk->online = true;
2430bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2431bd240fcaSzhanghailiang             } else if (!mem_blk->online) {
2432bd240fcaSzhanghailiang                 result->response =
2433bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
2434bd240fcaSzhanghailiang             }
2435bd240fcaSzhanghailiang         } else {
2436bd240fcaSzhanghailiang             if (sys2memblk) {
2437bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2438bd240fcaSzhanghailiang             } else {
2439b368123dSMarkus Armbruster                 error_free(local_err);
2440bd240fcaSzhanghailiang                 result->response =
2441bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2442bd240fcaSzhanghailiang             }
2443bd240fcaSzhanghailiang         }
2444bd240fcaSzhanghailiang         goto out2;
2445bd240fcaSzhanghailiang     }
2446bd240fcaSzhanghailiang 
2447bd240fcaSzhanghailiang     if (sys2memblk) {
2448bd240fcaSzhanghailiang         char removable = '0';
2449bd240fcaSzhanghailiang 
2450bd240fcaSzhanghailiang         mem_blk->online = (strncmp(status, "online", 6) == 0);
2451bd240fcaSzhanghailiang 
2452bd240fcaSzhanghailiang         ga_read_sysfs_file(dirfd, "removable", &removable, 1, &local_err);
2453bd240fcaSzhanghailiang         if (local_err) {
245467cc32ebSVeres Lajos             /* if no 'removable' file, it doesn't support offline mem blk */
2455bd240fcaSzhanghailiang             if (errno == ENOENT) {
2456bd240fcaSzhanghailiang                 error_free(local_err);
2457bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2458bd240fcaSzhanghailiang             } else {
2459bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2460bd240fcaSzhanghailiang             }
2461bd240fcaSzhanghailiang         } else {
2462bd240fcaSzhanghailiang             mem_blk->can_offline = (removable != '0');
2463bd240fcaSzhanghailiang         }
2464bd240fcaSzhanghailiang     } else {
2465bd240fcaSzhanghailiang         if (mem_blk->online != (strncmp(status, "online", 6) == 0)) {
24667064024dSMarc-André Lureau             const char *new_state = mem_blk->online ? "online" : "offline";
2467bd240fcaSzhanghailiang 
2468bd240fcaSzhanghailiang             ga_write_sysfs_file(dirfd, "state", new_state, strlen(new_state),
2469bd240fcaSzhanghailiang                                 &local_err);
2470bd240fcaSzhanghailiang             if (local_err) {
2471bd240fcaSzhanghailiang                 error_free(local_err);
2472bd240fcaSzhanghailiang                 result->response =
2473bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2474bd240fcaSzhanghailiang                 goto out2;
2475bd240fcaSzhanghailiang             }
2476bd240fcaSzhanghailiang 
2477bd240fcaSzhanghailiang             result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_SUCCESS;
2478bd240fcaSzhanghailiang             result->has_error_code = false;
2479bd240fcaSzhanghailiang         } /* otherwise pretend successful re-(on|off)-lining */
2480bd240fcaSzhanghailiang     }
2481bd240fcaSzhanghailiang     g_free(status);
2482bd240fcaSzhanghailiang     close(dirfd);
2483bd240fcaSzhanghailiang     return;
2484bd240fcaSzhanghailiang 
2485bd240fcaSzhanghailiang out2:
2486bd240fcaSzhanghailiang     g_free(status);
2487bd240fcaSzhanghailiang     close(dirfd);
2488bd240fcaSzhanghailiang out1:
2489bd240fcaSzhanghailiang     if (!sys2memblk) {
2490bd240fcaSzhanghailiang         result->has_error_code = true;
2491bd240fcaSzhanghailiang         result->error_code = errno;
2492bd240fcaSzhanghailiang     }
2493bd240fcaSzhanghailiang }
2494bd240fcaSzhanghailiang 
2495a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2496a065aaa9Szhanghailiang {
2497bd240fcaSzhanghailiang     GuestMemoryBlockList *head, **link;
2498bd240fcaSzhanghailiang     Error *local_err = NULL;
2499bd240fcaSzhanghailiang     struct dirent *de;
2500bd240fcaSzhanghailiang     DIR *dp;
2501bd240fcaSzhanghailiang 
2502bd240fcaSzhanghailiang     head = NULL;
2503bd240fcaSzhanghailiang     link = &head;
2504bd240fcaSzhanghailiang 
2505bd240fcaSzhanghailiang     dp = opendir("/sys/devices/system/memory/");
2506bd240fcaSzhanghailiang     if (!dp) {
2507f693fe6eSMichael Roth         /* it's ok if this happens to be a system that doesn't expose
2508f693fe6eSMichael Roth          * memory blocks via sysfs, but otherwise we should report
2509f693fe6eSMichael Roth          * an error
2510f693fe6eSMichael Roth          */
2511f693fe6eSMichael Roth         if (errno != ENOENT) {
2512bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "Can't open directory"
25139af9e0feSMarkus Armbruster                              "\"/sys/devices/system/memory/\"");
2514f693fe6eSMichael Roth         }
2515bd240fcaSzhanghailiang         return NULL;
2516bd240fcaSzhanghailiang     }
2517bd240fcaSzhanghailiang 
2518bd240fcaSzhanghailiang     /* Note: the phys_index of memory block may be discontinuous,
2519bd240fcaSzhanghailiang      * this is because a memblk is the unit of the Sparse Memory design, which
2520bd240fcaSzhanghailiang      * allows discontinuous memory ranges (ex. NUMA), so here we should
2521bd240fcaSzhanghailiang      * traverse the memory block directory.
2522bd240fcaSzhanghailiang      */
2523bd240fcaSzhanghailiang     while ((de = readdir(dp)) != NULL) {
2524bd240fcaSzhanghailiang         GuestMemoryBlock *mem_blk;
2525bd240fcaSzhanghailiang         GuestMemoryBlockList *entry;
2526bd240fcaSzhanghailiang 
2527bd240fcaSzhanghailiang         if ((strncmp(de->d_name, "memory", 6) != 0) ||
2528bd240fcaSzhanghailiang             !(de->d_type & DT_DIR)) {
2529bd240fcaSzhanghailiang             continue;
2530bd240fcaSzhanghailiang         }
2531bd240fcaSzhanghailiang 
2532bd240fcaSzhanghailiang         mem_blk = g_malloc0(sizeof *mem_blk);
2533bd240fcaSzhanghailiang         /* The d_name is "memoryXXX",  phys_index is block id, same as XXX */
2534bd240fcaSzhanghailiang         mem_blk->phys_index = strtoul(&de->d_name[6], NULL, 10);
2535bd240fcaSzhanghailiang         mem_blk->has_can_offline = true; /* lolspeak ftw */
2536bd240fcaSzhanghailiang         transfer_memory_block(mem_blk, true, NULL, &local_err);
25374155c998SMarkus Armbruster         if (local_err) {
25384155c998SMarkus Armbruster             break;
25394155c998SMarkus Armbruster         }
2540bd240fcaSzhanghailiang 
2541bd240fcaSzhanghailiang         entry = g_malloc0(sizeof *entry);
2542bd240fcaSzhanghailiang         entry->value = mem_blk;
2543bd240fcaSzhanghailiang 
2544bd240fcaSzhanghailiang         *link = entry;
2545bd240fcaSzhanghailiang         link = &entry->next;
2546bd240fcaSzhanghailiang     }
2547bd240fcaSzhanghailiang 
2548bd240fcaSzhanghailiang     closedir(dp);
2549bd240fcaSzhanghailiang     if (local_err == NULL) {
2550bd240fcaSzhanghailiang         /* there's no guest with zero memory blocks */
2551bd240fcaSzhanghailiang         if (head == NULL) {
2552bd240fcaSzhanghailiang             error_setg(errp, "guest reported zero memory blocks!");
2553bd240fcaSzhanghailiang         }
2554bd240fcaSzhanghailiang         return head;
2555bd240fcaSzhanghailiang     }
2556bd240fcaSzhanghailiang 
2557bd240fcaSzhanghailiang     qapi_free_GuestMemoryBlockList(head);
2558bd240fcaSzhanghailiang     error_propagate(errp, local_err);
2559a065aaa9Szhanghailiang     return NULL;
2560a065aaa9Szhanghailiang }
2561a065aaa9Szhanghailiang 
2562a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2563a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2564a065aaa9Szhanghailiang {
256532ca7927Szhanghailiang     GuestMemoryBlockResponseList *head, **link;
256632ca7927Szhanghailiang     Error *local_err = NULL;
256732ca7927Szhanghailiang 
256832ca7927Szhanghailiang     head = NULL;
256932ca7927Szhanghailiang     link = &head;
257032ca7927Szhanghailiang 
257132ca7927Szhanghailiang     while (mem_blks != NULL) {
257232ca7927Szhanghailiang         GuestMemoryBlockResponse *result;
257332ca7927Szhanghailiang         GuestMemoryBlockResponseList *entry;
257432ca7927Szhanghailiang         GuestMemoryBlock *current_mem_blk = mem_blks->value;
257532ca7927Szhanghailiang 
257632ca7927Szhanghailiang         result = g_malloc0(sizeof(*result));
257732ca7927Szhanghailiang         result->phys_index = current_mem_blk->phys_index;
257832ca7927Szhanghailiang         transfer_memory_block(current_mem_blk, false, result, &local_err);
257932ca7927Szhanghailiang         if (local_err) { /* should never happen */
258032ca7927Szhanghailiang             goto err;
258132ca7927Szhanghailiang         }
258232ca7927Szhanghailiang         entry = g_malloc0(sizeof *entry);
258332ca7927Szhanghailiang         entry->value = result;
258432ca7927Szhanghailiang 
258532ca7927Szhanghailiang         *link = entry;
258632ca7927Szhanghailiang         link = &entry->next;
258732ca7927Szhanghailiang         mem_blks = mem_blks->next;
258832ca7927Szhanghailiang     }
258932ca7927Szhanghailiang 
259032ca7927Szhanghailiang     return head;
259132ca7927Szhanghailiang err:
259232ca7927Szhanghailiang     qapi_free_GuestMemoryBlockResponseList(head);
259332ca7927Szhanghailiang     error_propagate(errp, local_err);
2594a065aaa9Szhanghailiang     return NULL;
2595a065aaa9Szhanghailiang }
2596a065aaa9Szhanghailiang 
2597a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2598a065aaa9Szhanghailiang {
2599ef82b60bSzhanghailiang     Error *local_err = NULL;
2600ef82b60bSzhanghailiang     char *dirpath;
2601ef82b60bSzhanghailiang     int dirfd;
2602ef82b60bSzhanghailiang     char *buf;
2603ef82b60bSzhanghailiang     GuestMemoryBlockInfo *info;
2604ef82b60bSzhanghailiang 
2605ef82b60bSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/");
2606ef82b60bSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2607ef82b60bSzhanghailiang     if (dirfd == -1) {
2608ef82b60bSzhanghailiang         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2609ef82b60bSzhanghailiang         g_free(dirpath);
2610a065aaa9Szhanghailiang         return NULL;
2611a065aaa9Szhanghailiang     }
2612ef82b60bSzhanghailiang     g_free(dirpath);
2613ef82b60bSzhanghailiang 
2614ef82b60bSzhanghailiang     buf = g_malloc0(20);
2615ef82b60bSzhanghailiang     ga_read_sysfs_file(dirfd, "block_size_bytes", buf, 20, &local_err);
26168ce1ee46SShannon Zhao     close(dirfd);
2617ef82b60bSzhanghailiang     if (local_err) {
2618ef82b60bSzhanghailiang         g_free(buf);
2619ef82b60bSzhanghailiang         error_propagate(errp, local_err);
2620ef82b60bSzhanghailiang         return NULL;
2621ef82b60bSzhanghailiang     }
2622ef82b60bSzhanghailiang 
2623ef82b60bSzhanghailiang     info = g_new0(GuestMemoryBlockInfo, 1);
2624ef82b60bSzhanghailiang     info->size = strtol(buf, NULL, 16); /* the unit is bytes */
2625ef82b60bSzhanghailiang 
2626ef82b60bSzhanghailiang     g_free(buf);
2627ef82b60bSzhanghailiang 
2628ef82b60bSzhanghailiang     return info;
2629ef82b60bSzhanghailiang }
2630a065aaa9Szhanghailiang 
2631e72c3f2eSMichael Roth #else /* defined(__linux__) */
2632e72c3f2eSMichael Roth 
263377dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
2634e72c3f2eSMichael Roth {
2635c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2636e72c3f2eSMichael Roth }
2637e72c3f2eSMichael Roth 
263877dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
2639e72c3f2eSMichael Roth {
2640c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2641e72c3f2eSMichael Roth }
2642e72c3f2eSMichael Roth 
264377dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
2644e72c3f2eSMichael Roth {
2645c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2646e72c3f2eSMichael Roth }
2647e72c3f2eSMichael Roth 
2648e72c3f2eSMichael Roth GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
2649e72c3f2eSMichael Roth {
2650c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2651e72c3f2eSMichael Roth     return NULL;
2652e72c3f2eSMichael Roth }
2653e72c3f2eSMichael Roth 
2654d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2655d2baff62SLaszlo Ersek {
2656c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2657d2baff62SLaszlo Ersek     return NULL;
2658d2baff62SLaszlo Ersek }
2659d2baff62SLaszlo Ersek 
2660cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2661cbb65fc2SLaszlo Ersek {
2662c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2663cbb65fc2SLaszlo Ersek     return -1;
2664cbb65fc2SLaszlo Ersek }
2665cbb65fc2SLaszlo Ersek 
2666215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2667215a2771SDaniel P. Berrange                                  const char *password,
2668215a2771SDaniel P. Berrange                                  bool crypted,
2669215a2771SDaniel P. Berrange                                  Error **errp)
2670215a2771SDaniel P. Berrange {
2671c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2672215a2771SDaniel P. Berrange }
2673215a2771SDaniel P. Berrange 
2674a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2675a065aaa9Szhanghailiang {
2676c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2677a065aaa9Szhanghailiang     return NULL;
2678a065aaa9Szhanghailiang }
2679a065aaa9Szhanghailiang 
2680a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2681a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2682a065aaa9Szhanghailiang {
2683c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2684a065aaa9Szhanghailiang     return NULL;
2685a065aaa9Szhanghailiang }
2686a065aaa9Szhanghailiang 
2687a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2688a065aaa9Szhanghailiang {
2689c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2690a065aaa9Szhanghailiang     return NULL;
2691a065aaa9Szhanghailiang }
2692a065aaa9Szhanghailiang 
2693e72c3f2eSMichael Roth #endif
2694e72c3f2eSMichael Roth 
2695d35d4cb5SMichael Roth #if !defined(CONFIG_FSFREEZE)
2696d35d4cb5SMichael Roth 
269746d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
269846d4c572STomoki Sekiyama {
2699c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
270046d4c572STomoki Sekiyama     return NULL;
270146d4c572STomoki Sekiyama }
270246d4c572STomoki Sekiyama 
270377dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
2704d35d4cb5SMichael Roth {
2705c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2706d35d4cb5SMichael Roth 
2707d35d4cb5SMichael Roth     return 0;
2708d35d4cb5SMichael Roth }
2709d35d4cb5SMichael Roth 
271077dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_freeze(Error **errp)
2711d35d4cb5SMichael Roth {
2712c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2713d35d4cb5SMichael Roth 
2714d35d4cb5SMichael Roth     return 0;
2715d35d4cb5SMichael Roth }
2716d35d4cb5SMichael Roth 
2717e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
2718e99bce20STomoki Sekiyama                                        strList *mountpoints,
2719e99bce20STomoki Sekiyama                                        Error **errp)
2720e99bce20STomoki Sekiyama {
2721c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2722e99bce20STomoki Sekiyama 
2723e99bce20STomoki Sekiyama     return 0;
2724e99bce20STomoki Sekiyama }
2725e99bce20STomoki Sekiyama 
272677dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
2727d35d4cb5SMichael Roth {
2728c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2729d35d4cb5SMichael Roth 
2730d35d4cb5SMichael Roth     return 0;
2731d35d4cb5SMichael Roth }
2732eab5fd59SPaolo Bonzini #endif /* CONFIG_FSFREEZE */
2733d35d4cb5SMichael Roth 
2734eab5fd59SPaolo Bonzini #if !defined(CONFIG_FSTRIM)
2735e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
2736e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
2737eab5fd59SPaolo Bonzini {
2738c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2739e82855d9SJustin Ossevoort     return NULL;
2740eab5fd59SPaolo Bonzini }
2741d35d4cb5SMichael Roth #endif
2742d35d4cb5SMichael Roth 
27431281c08aSTomoki Sekiyama /* add unsupported commands to the blacklist */
27441281c08aSTomoki Sekiyama GList *ga_command_blacklist_init(GList *blacklist)
27451281c08aSTomoki Sekiyama {
27461281c08aSTomoki Sekiyama #if !defined(__linux__)
27471281c08aSTomoki Sekiyama     {
27481281c08aSTomoki Sekiyama         const char *list[] = {
27491281c08aSTomoki Sekiyama             "guest-suspend-disk", "guest-suspend-ram",
27501281c08aSTomoki Sekiyama             "guest-suspend-hybrid", "guest-network-get-interfaces",
27510dd38a03Szhanghailiang             "guest-get-vcpus", "guest-set-vcpus",
27520dd38a03Szhanghailiang             "guest-get-memory-blocks", "guest-set-memory-blocks",
275328d8dd35SBasil Salman             "guest-get-memory-block-size", "guest-get-memory-block-info",
275428d8dd35SBasil Salman             NULL};
27551281c08aSTomoki Sekiyama         char **p = (char **)list;
27561281c08aSTomoki Sekiyama 
27571281c08aSTomoki Sekiyama         while (*p) {
27584bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
27591281c08aSTomoki Sekiyama         }
27601281c08aSTomoki Sekiyama     }
27611281c08aSTomoki Sekiyama #endif
27621281c08aSTomoki Sekiyama 
27631281c08aSTomoki Sekiyama #if !defined(CONFIG_FSFREEZE)
27641281c08aSTomoki Sekiyama     {
27651281c08aSTomoki Sekiyama         const char *list[] = {
27661281c08aSTomoki Sekiyama             "guest-get-fsinfo", "guest-fsfreeze-status",
27671281c08aSTomoki Sekiyama             "guest-fsfreeze-freeze", "guest-fsfreeze-freeze-list",
27681281c08aSTomoki Sekiyama             "guest-fsfreeze-thaw", "guest-get-fsinfo", NULL};
27691281c08aSTomoki Sekiyama         char **p = (char **)list;
27701281c08aSTomoki Sekiyama 
27711281c08aSTomoki Sekiyama         while (*p) {
27724bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
27731281c08aSTomoki Sekiyama         }
27741281c08aSTomoki Sekiyama     }
27751281c08aSTomoki Sekiyama #endif
27761281c08aSTomoki Sekiyama 
27771281c08aSTomoki Sekiyama #if !defined(CONFIG_FSTRIM)
27784bca81ceSMarc-André Lureau     blacklist = g_list_append(blacklist, g_strdup("guest-fstrim"));
27791281c08aSTomoki Sekiyama #endif
27801281c08aSTomoki Sekiyama 
27811281c08aSTomoki Sekiyama     return blacklist;
27821281c08aSTomoki Sekiyama }
27831281c08aSTomoki Sekiyama 
2784c216e5adSMichael Roth /* register init/cleanup routines for stateful command groups */
2785c216e5adSMichael Roth void ga_command_state_init(GAState *s, GACommandState *cs)
2786c216e5adSMichael Roth {
2787c216e5adSMichael Roth #if defined(CONFIG_FSFREEZE)
2788f22d85e9SMichael Roth     ga_command_state_add(cs, NULL, guest_fsfreeze_cleanup);
2789c216e5adSMichael Roth #endif
2790c216e5adSMichael Roth }
2791161a56a9SVinzenz Feenstra 
2792e674605fSTomáš Golembiovský #ifdef HAVE_UTMPX
2793e674605fSTomáš Golembiovský 
2794161a56a9SVinzenz Feenstra #define QGA_MICRO_SECOND_TO_SECOND 1000000
2795161a56a9SVinzenz Feenstra 
2796161a56a9SVinzenz Feenstra static double ga_get_login_time(struct utmpx *user_info)
2797161a56a9SVinzenz Feenstra {
2798161a56a9SVinzenz Feenstra     double seconds = (double)user_info->ut_tv.tv_sec;
2799161a56a9SVinzenz Feenstra     double useconds = (double)user_info->ut_tv.tv_usec;
2800161a56a9SVinzenz Feenstra     useconds /= QGA_MICRO_SECOND_TO_SECOND;
2801161a56a9SVinzenz Feenstra     return seconds + useconds;
2802161a56a9SVinzenz Feenstra }
2803161a56a9SVinzenz Feenstra 
2804b90abbacSVladimir Sementsov-Ogievskiy GuestUserList *qmp_guest_get_users(Error **errp)
2805161a56a9SVinzenz Feenstra {
2806161a56a9SVinzenz Feenstra     GHashTable *cache = NULL;
2807161a56a9SVinzenz Feenstra     GuestUserList *head = NULL, *cur_item = NULL;
2808161a56a9SVinzenz Feenstra     struct utmpx *user_info = NULL;
2809161a56a9SVinzenz Feenstra     gpointer value = NULL;
2810161a56a9SVinzenz Feenstra     GuestUser *user = NULL;
2811161a56a9SVinzenz Feenstra     GuestUserList *item = NULL;
2812161a56a9SVinzenz Feenstra     double login_time = 0;
2813161a56a9SVinzenz Feenstra 
2814161a56a9SVinzenz Feenstra     cache = g_hash_table_new(g_str_hash, g_str_equal);
2815161a56a9SVinzenz Feenstra     setutxent();
2816161a56a9SVinzenz Feenstra 
2817161a56a9SVinzenz Feenstra     for (;;) {
2818161a56a9SVinzenz Feenstra         user_info = getutxent();
2819161a56a9SVinzenz Feenstra         if (user_info == NULL) {
2820161a56a9SVinzenz Feenstra             break;
2821161a56a9SVinzenz Feenstra         } else if (user_info->ut_type != USER_PROCESS) {
2822161a56a9SVinzenz Feenstra             continue;
2823161a56a9SVinzenz Feenstra         } else if (g_hash_table_contains(cache, user_info->ut_user)) {
2824161a56a9SVinzenz Feenstra             value = g_hash_table_lookup(cache, user_info->ut_user);
2825161a56a9SVinzenz Feenstra             user = (GuestUser *)value;
2826161a56a9SVinzenz Feenstra             login_time = ga_get_login_time(user_info);
2827161a56a9SVinzenz Feenstra             /* We're ensuring the earliest login time to be sent */
2828161a56a9SVinzenz Feenstra             if (login_time < user->login_time) {
2829161a56a9SVinzenz Feenstra                 user->login_time = login_time;
2830161a56a9SVinzenz Feenstra             }
2831161a56a9SVinzenz Feenstra             continue;
2832161a56a9SVinzenz Feenstra         }
2833161a56a9SVinzenz Feenstra 
2834161a56a9SVinzenz Feenstra         item = g_new0(GuestUserList, 1);
2835161a56a9SVinzenz Feenstra         item->value = g_new0(GuestUser, 1);
2836161a56a9SVinzenz Feenstra         item->value->user = g_strdup(user_info->ut_user);
2837161a56a9SVinzenz Feenstra         item->value->login_time = ga_get_login_time(user_info);
2838161a56a9SVinzenz Feenstra 
2839161a56a9SVinzenz Feenstra         g_hash_table_insert(cache, item->value->user, item->value);
2840161a56a9SVinzenz Feenstra 
2841161a56a9SVinzenz Feenstra         if (!cur_item) {
2842161a56a9SVinzenz Feenstra             head = cur_item = item;
2843161a56a9SVinzenz Feenstra         } else {
2844161a56a9SVinzenz Feenstra             cur_item->next = item;
2845161a56a9SVinzenz Feenstra             cur_item = item;
2846161a56a9SVinzenz Feenstra         }
2847161a56a9SVinzenz Feenstra     }
2848161a56a9SVinzenz Feenstra     endutxent();
2849161a56a9SVinzenz Feenstra     g_hash_table_destroy(cache);
2850161a56a9SVinzenz Feenstra     return head;
2851161a56a9SVinzenz Feenstra }
2852e674605fSTomáš Golembiovský 
2853e674605fSTomáš Golembiovský #else
2854e674605fSTomáš Golembiovský 
2855e674605fSTomáš Golembiovský GuestUserList *qmp_guest_get_users(Error **errp)
2856e674605fSTomáš Golembiovský {
2857e674605fSTomáš Golembiovský     error_setg(errp, QERR_UNSUPPORTED);
2858e674605fSTomáš Golembiovský     return NULL;
2859e674605fSTomáš Golembiovský }
2860e674605fSTomáš Golembiovský 
2861e674605fSTomáš Golembiovský #endif
28629848f797STomáš Golembiovský 
28639848f797STomáš Golembiovský /* Replace escaped special characters with theire real values. The replacement
28649848f797STomáš Golembiovský  * is done in place -- returned value is in the original string.
28659848f797STomáš Golembiovský  */
28669848f797STomáš Golembiovský static void ga_osrelease_replace_special(gchar *value)
28679848f797STomáš Golembiovský {
28689848f797STomáš Golembiovský     gchar *p, *p2, quote;
28699848f797STomáš Golembiovský 
28709848f797STomáš Golembiovský     /* Trim the string at first space or semicolon if it is not enclosed in
28719848f797STomáš Golembiovský      * single or double quotes. */
28729848f797STomáš Golembiovský     if ((value[0] != '"') || (value[0] == '\'')) {
28739848f797STomáš Golembiovský         p = strchr(value, ' ');
28749848f797STomáš Golembiovský         if (p != NULL) {
28759848f797STomáš Golembiovský             *p = 0;
28769848f797STomáš Golembiovský         }
28779848f797STomáš Golembiovský         p = strchr(value, ';');
28789848f797STomáš Golembiovský         if (p != NULL) {
28799848f797STomáš Golembiovský             *p = 0;
28809848f797STomáš Golembiovský         }
28819848f797STomáš Golembiovský         return;
28829848f797STomáš Golembiovský     }
28839848f797STomáš Golembiovský 
28849848f797STomáš Golembiovský     quote = value[0];
28859848f797STomáš Golembiovský     p2 = value;
28869848f797STomáš Golembiovský     p = value + 1;
28879848f797STomáš Golembiovský     while (*p != 0) {
28889848f797STomáš Golembiovský         if (*p == '\\') {
28899848f797STomáš Golembiovský             p++;
28909848f797STomáš Golembiovský             switch (*p) {
28919848f797STomáš Golembiovský             case '$':
28929848f797STomáš Golembiovský             case '\'':
28939848f797STomáš Golembiovský             case '"':
28949848f797STomáš Golembiovský             case '\\':
28959848f797STomáš Golembiovský             case '`':
28969848f797STomáš Golembiovský                 break;
28979848f797STomáš Golembiovský             default:
28989848f797STomáš Golembiovský                 /* Keep literal backslash followed by whatever is there */
28999848f797STomáš Golembiovský                 p--;
29009848f797STomáš Golembiovský                 break;
29019848f797STomáš Golembiovský             }
29029848f797STomáš Golembiovský         } else if (*p == quote) {
29039848f797STomáš Golembiovský             *p2 = 0;
29049848f797STomáš Golembiovský             break;
29059848f797STomáš Golembiovský         }
29069848f797STomáš Golembiovský         *(p2++) = *(p++);
29079848f797STomáš Golembiovský     }
29089848f797STomáš Golembiovský }
29099848f797STomáš Golembiovský 
29109848f797STomáš Golembiovský static GKeyFile *ga_parse_osrelease(const char *fname)
29119848f797STomáš Golembiovský {
29129848f797STomáš Golembiovský     gchar *content = NULL;
29139848f797STomáš Golembiovský     gchar *content2 = NULL;
29149848f797STomáš Golembiovský     GError *err = NULL;
29159848f797STomáš Golembiovský     GKeyFile *keys = g_key_file_new();
29169848f797STomáš Golembiovský     const char *group = "[os-release]\n";
29179848f797STomáš Golembiovský 
29189848f797STomáš Golembiovský     if (!g_file_get_contents(fname, &content, NULL, &err)) {
29199848f797STomáš Golembiovský         slog("failed to read '%s', error: %s", fname, err->message);
29209848f797STomáš Golembiovský         goto fail;
29219848f797STomáš Golembiovský     }
29229848f797STomáš Golembiovský 
29239848f797STomáš Golembiovský     if (!g_utf8_validate(content, -1, NULL)) {
29249848f797STomáš Golembiovský         slog("file is not utf-8 encoded: %s", fname);
29259848f797STomáš Golembiovský         goto fail;
29269848f797STomáš Golembiovský     }
29279848f797STomáš Golembiovský     content2 = g_strdup_printf("%s%s", group, content);
29289848f797STomáš Golembiovský 
29299848f797STomáš Golembiovský     if (!g_key_file_load_from_data(keys, content2, -1, G_KEY_FILE_NONE,
29309848f797STomáš Golembiovský                                    &err)) {
29319848f797STomáš Golembiovský         slog("failed to parse file '%s', error: %s", fname, err->message);
29329848f797STomáš Golembiovský         goto fail;
29339848f797STomáš Golembiovský     }
29349848f797STomáš Golembiovský 
29359848f797STomáš Golembiovský     g_free(content);
29369848f797STomáš Golembiovský     g_free(content2);
29379848f797STomáš Golembiovský     return keys;
29389848f797STomáš Golembiovský 
29399848f797STomáš Golembiovský fail:
29409848f797STomáš Golembiovský     g_error_free(err);
29419848f797STomáš Golembiovský     g_free(content);
29429848f797STomáš Golembiovský     g_free(content2);
29439848f797STomáš Golembiovský     g_key_file_free(keys);
29449848f797STomáš Golembiovský     return NULL;
29459848f797STomáš Golembiovský }
29469848f797STomáš Golembiovský 
29479848f797STomáš Golembiovský GuestOSInfo *qmp_guest_get_osinfo(Error **errp)
29489848f797STomáš Golembiovský {
29499848f797STomáš Golembiovský     GuestOSInfo *info = NULL;
29509848f797STomáš Golembiovský     struct utsname kinfo;
2951339ca68bSTomáš Golembiovský     GKeyFile *osrelease = NULL;
2952339ca68bSTomáš Golembiovský     const char *qga_os_release = g_getenv("QGA_OS_RELEASE");
29539848f797STomáš Golembiovský 
29549848f797STomáš Golembiovský     info = g_new0(GuestOSInfo, 1);
29559848f797STomáš Golembiovský 
29569848f797STomáš Golembiovský     if (uname(&kinfo) != 0) {
29579848f797STomáš Golembiovský         error_setg_errno(errp, errno, "uname failed");
29589848f797STomáš Golembiovský     } else {
29599848f797STomáš Golembiovský         info->has_kernel_version = true;
29609848f797STomáš Golembiovský         info->kernel_version = g_strdup(kinfo.version);
29619848f797STomáš Golembiovský         info->has_kernel_release = true;
29629848f797STomáš Golembiovský         info->kernel_release = g_strdup(kinfo.release);
29639848f797STomáš Golembiovský         info->has_machine = true;
29649848f797STomáš Golembiovský         info->machine = g_strdup(kinfo.machine);
29659848f797STomáš Golembiovský     }
29669848f797STomáš Golembiovský 
2967339ca68bSTomáš Golembiovský     if (qga_os_release != NULL) {
2968339ca68bSTomáš Golembiovský         osrelease = ga_parse_osrelease(qga_os_release);
2969339ca68bSTomáš Golembiovský     } else {
29709848f797STomáš Golembiovský         osrelease = ga_parse_osrelease("/etc/os-release");
29719848f797STomáš Golembiovský         if (osrelease == NULL) {
29729848f797STomáš Golembiovský             osrelease = ga_parse_osrelease("/usr/lib/os-release");
29739848f797STomáš Golembiovský         }
2974339ca68bSTomáš Golembiovský     }
29759848f797STomáš Golembiovský 
29769848f797STomáš Golembiovský     if (osrelease != NULL) {
29779848f797STomáš Golembiovský         char *value;
29789848f797STomáš Golembiovský 
29799848f797STomáš Golembiovský #define GET_FIELD(field, osfield) do { \
29809848f797STomáš Golembiovský     value = g_key_file_get_value(osrelease, "os-release", osfield, NULL); \
29819848f797STomáš Golembiovský     if (value != NULL) { \
29829848f797STomáš Golembiovský         ga_osrelease_replace_special(value); \
29839848f797STomáš Golembiovský         info->has_ ## field = true; \
29849848f797STomáš Golembiovský         info->field = value; \
29859848f797STomáš Golembiovský     } \
29869848f797STomáš Golembiovský } while (0)
29879848f797STomáš Golembiovský         GET_FIELD(id, "ID");
29889848f797STomáš Golembiovský         GET_FIELD(name, "NAME");
29899848f797STomáš Golembiovský         GET_FIELD(pretty_name, "PRETTY_NAME");
29909848f797STomáš Golembiovský         GET_FIELD(version, "VERSION");
29919848f797STomáš Golembiovský         GET_FIELD(version_id, "VERSION_ID");
29929848f797STomáš Golembiovský         GET_FIELD(variant, "VARIANT");
29939848f797STomáš Golembiovský         GET_FIELD(variant_id, "VARIANT_ID");
29949848f797STomáš Golembiovský #undef GET_FIELD
29959848f797STomáš Golembiovský 
29969848f797STomáš Golembiovský         g_key_file_free(osrelease);
29979848f797STomáš Golembiovský     }
29989848f797STomáš Golembiovský 
29999848f797STomáš Golembiovský     return info;
30009848f797STomáš Golembiovský }
3001