xref: /openbmc/qemu/qga/commands-posix.c (revision 23843c12)
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 
864d9fe4f0fSThomas Huth /*
865d9fe4f0fSThomas Huth  * Store disk device info for devices on the PCI bus.
866d9fe4f0fSThomas Huth  * Returns true if information has been stored, or false for failure.
867d9fe4f0fSThomas Huth  */
868d9fe4f0fSThomas Huth static bool build_guest_fsinfo_for_pci_dev(char const *syspath,
869d9fe4f0fSThomas 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;
874d9fe4f0fSThomas 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;
877d9fe4f0fSThomas Huth     bool ret = false;
87846d4c572STomoki Sekiyama 
87946d4c572STomoki Sekiyama     p = strstr(syspath, "/devices/pci");
88046d4c572STomoki Sekiyama     if (!p || sscanf(p + 12, "%*x:%*x/%x:%x:%x.%x%n",
88146d4c572STomoki Sekiyama                      pci, pci + 1, pci + 2, pci + 3, &pcilen) < 4) {
882743c71d0SMarc-André Lureau         g_debug("only pci device is supported: sysfs path '%s'", syspath);
883d9fe4f0fSThomas Huth         return false;
88446d4c572STomoki Sekiyama     }
88546d4c572STomoki Sekiyama 
886743c71d0SMarc-André Lureau     p += 12 + pcilen;
887743c71d0SMarc-André Lureau     while (true) {
888743c71d0SMarc-André Lureau         driver = get_pci_driver(syspath, p - syspath, errp);
889743c71d0SMarc-André Lureau         if (driver && (g_str_equal(driver, "ata_piix") ||
890743c71d0SMarc-André Lureau                        g_str_equal(driver, "sym53c8xx") ||
891743c71d0SMarc-André Lureau                        g_str_equal(driver, "virtio-pci") ||
892743c71d0SMarc-André Lureau                        g_str_equal(driver, "ahci"))) {
893743c71d0SMarc-André Lureau             break;
894743c71d0SMarc-André Lureau         }
895743c71d0SMarc-André Lureau 
896bb23a736SMarc-André Lureau         g_free(driver);
897743c71d0SMarc-André Lureau         if (sscanf(p, "/%x:%x:%x.%x%n",
898743c71d0SMarc-André Lureau                           pci, pci + 1, pci + 2, pci + 3, &pcilen) == 4) {
899743c71d0SMarc-André Lureau             p += pcilen;
900743c71d0SMarc-André Lureau             continue;
901743c71d0SMarc-André Lureau         }
902743c71d0SMarc-André Lureau 
903743c71d0SMarc-André Lureau         g_debug("unsupported driver or sysfs path '%s'", syspath);
904d9fe4f0fSThomas Huth         return false;
90546d4c572STomoki Sekiyama     }
90646d4c572STomoki Sekiyama 
90746d4c572STomoki Sekiyama     p = strstr(syspath, "/target");
90846d4c572STomoki Sekiyama     if (p && sscanf(p + 7, "%*u:%*u:%*u/%*u:%u:%u:%u",
90946d4c572STomoki Sekiyama                     tgt, tgt + 1, tgt + 2) == 3) {
91046d4c572STomoki Sekiyama         has_tgt = true;
91146d4c572STomoki Sekiyama     }
91246d4c572STomoki Sekiyama 
91346d4c572STomoki Sekiyama     p = strstr(syspath, "/ata");
91446d4c572STomoki Sekiyama     if (p) {
91546d4c572STomoki Sekiyama         q = p + 4;
91646d4c572STomoki Sekiyama         has_ata = true;
91746d4c572STomoki Sekiyama     } else {
91846d4c572STomoki Sekiyama         p = strstr(syspath, "/host");
91946d4c572STomoki Sekiyama         q = p + 5;
92046d4c572STomoki Sekiyama     }
92146d4c572STomoki Sekiyama     if (p && sscanf(q, "%u", &host) == 1) {
92246d4c572STomoki Sekiyama         has_host = true;
92346d4c572STomoki Sekiyama         nhosts = build_hosts(syspath, p, has_ata, hosts,
92401a6df1bSPhilippe Mathieu-Daudé                              ARRAY_SIZE(hosts), errp);
92546d4c572STomoki Sekiyama         if (nhosts < 0) {
92646d4c572STomoki Sekiyama             goto cleanup;
92746d4c572STomoki Sekiyama         }
92846d4c572STomoki Sekiyama     }
92946d4c572STomoki Sekiyama 
93046d4c572STomoki Sekiyama     pciaddr->domain = pci[0];
93146d4c572STomoki Sekiyama     pciaddr->bus = pci[1];
93246d4c572STomoki Sekiyama     pciaddr->slot = pci[2];
93346d4c572STomoki Sekiyama     pciaddr->function = pci[3];
93446d4c572STomoki Sekiyama 
93546d4c572STomoki Sekiyama     if (strcmp(driver, "ata_piix") == 0) {
93646d4c572STomoki Sekiyama         /* a host per ide bus, target*:0:<unit>:0 */
93746d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
93846d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
93946d4c572STomoki Sekiyama             goto cleanup;
94046d4c572STomoki Sekiyama         }
94146d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
94246d4c572STomoki Sekiyama             if (host == hosts[i]) {
94346d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_IDE;
94446d4c572STomoki Sekiyama                 disk->bus = i;
94546d4c572STomoki Sekiyama                 disk->unit = tgt[1];
94646d4c572STomoki Sekiyama                 break;
94746d4c572STomoki Sekiyama             }
94846d4c572STomoki Sekiyama         }
94946d4c572STomoki Sekiyama         if (i >= nhosts) {
95046d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
95146d4c572STomoki Sekiyama             goto cleanup;
95246d4c572STomoki Sekiyama         }
95346d4c572STomoki Sekiyama     } else if (strcmp(driver, "sym53c8xx") == 0) {
95446d4c572STomoki Sekiyama         /* scsi(LSI Logic): target*:0:<unit>:0 */
95546d4c572STomoki Sekiyama         if (!has_tgt) {
95646d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
95746d4c572STomoki Sekiyama             goto cleanup;
95846d4c572STomoki Sekiyama         }
95946d4c572STomoki Sekiyama         disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
96046d4c572STomoki Sekiyama         disk->unit = tgt[1];
96146d4c572STomoki Sekiyama     } else if (strcmp(driver, "virtio-pci") == 0) {
96246d4c572STomoki Sekiyama         if (has_tgt) {
96346d4c572STomoki Sekiyama             /* virtio-scsi: target*:0:0:<unit> */
96446d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
96546d4c572STomoki Sekiyama             disk->unit = tgt[2];
96646d4c572STomoki Sekiyama         } else {
96746d4c572STomoki Sekiyama             /* virtio-blk: 1 disk per 1 device */
96846d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_VIRTIO;
96946d4c572STomoki Sekiyama         }
97046d4c572STomoki Sekiyama     } else if (strcmp(driver, "ahci") == 0) {
97146d4c572STomoki Sekiyama         /* ahci: 1 host per 1 unit */
97246d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
97346d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
97446d4c572STomoki Sekiyama             goto cleanup;
97546d4c572STomoki Sekiyama         }
97646d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
97746d4c572STomoki Sekiyama             if (host == hosts[i]) {
97846d4c572STomoki Sekiyama                 disk->unit = i;
97946d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_SATA;
98046d4c572STomoki Sekiyama                 break;
98146d4c572STomoki Sekiyama             }
98246d4c572STomoki Sekiyama         }
98346d4c572STomoki Sekiyama         if (i >= nhosts) {
98446d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
98546d4c572STomoki Sekiyama             goto cleanup;
98646d4c572STomoki Sekiyama         }
98746d4c572STomoki Sekiyama     } else {
98846d4c572STomoki Sekiyama         g_debug("unknown driver '%s' (sysfs path '%s')", driver, syspath);
98946d4c572STomoki Sekiyama         goto cleanup;
99046d4c572STomoki Sekiyama     }
99146d4c572STomoki Sekiyama 
992d9fe4f0fSThomas Huth     ret = true;
99346d4c572STomoki Sekiyama 
99446d4c572STomoki Sekiyama cleanup:
99546d4c572STomoki Sekiyama     g_free(driver);
996d9fe4f0fSThomas Huth     return ret;
997d9fe4f0fSThomas Huth }
998d9fe4f0fSThomas Huth 
999*23843c12SThomas Huth /*
1000*23843c12SThomas Huth  * Store disk device info for non-PCI virtio devices (for example s390x
1001*23843c12SThomas Huth  * channel I/O devices). Returns true if information has been stored, or
1002*23843c12SThomas Huth  * false for failure.
1003*23843c12SThomas Huth  */
1004*23843c12SThomas Huth static bool build_guest_fsinfo_for_nonpci_virtio(char const *syspath,
1005*23843c12SThomas Huth                                                  GuestDiskAddress *disk,
1006*23843c12SThomas Huth                                                  Error **errp)
1007*23843c12SThomas Huth {
1008*23843c12SThomas Huth     unsigned int tgt[3];
1009*23843c12SThomas Huth     char *p;
1010*23843c12SThomas Huth 
1011*23843c12SThomas Huth     if (!strstr(syspath, "/virtio") || !strstr(syspath, "/block")) {
1012*23843c12SThomas Huth         g_debug("Unsupported virtio device '%s'", syspath);
1013*23843c12SThomas Huth         return false;
1014*23843c12SThomas Huth     }
1015*23843c12SThomas Huth 
1016*23843c12SThomas Huth     p = strstr(syspath, "/target");
1017*23843c12SThomas Huth     if (p && sscanf(p + 7, "%*u:%*u:%*u/%*u:%u:%u:%u",
1018*23843c12SThomas Huth                     &tgt[0], &tgt[1], &tgt[2]) == 3) {
1019*23843c12SThomas Huth         /* virtio-scsi: target*:0:<target>:<unit> */
1020*23843c12SThomas Huth         disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
1021*23843c12SThomas Huth         disk->bus = tgt[0];
1022*23843c12SThomas Huth         disk->target = tgt[1];
1023*23843c12SThomas Huth         disk->unit = tgt[2];
1024*23843c12SThomas Huth     } else {
1025*23843c12SThomas Huth         /* virtio-blk: 1 disk per 1 device */
1026*23843c12SThomas Huth         disk->bus_type = GUEST_DISK_BUS_TYPE_VIRTIO;
1027*23843c12SThomas Huth     }
1028*23843c12SThomas Huth 
1029*23843c12SThomas Huth     return true;
1030*23843c12SThomas Huth }
1031*23843c12SThomas Huth 
1032d9fe4f0fSThomas Huth /* Store disk device info specified by @sysfs into @fs */
1033d9fe4f0fSThomas Huth static void build_guest_fsinfo_for_real_device(char const *syspath,
1034d9fe4f0fSThomas Huth                                                GuestFilesystemInfo *fs,
1035d9fe4f0fSThomas Huth                                                Error **errp)
1036d9fe4f0fSThomas Huth {
1037d9fe4f0fSThomas Huth     GuestDiskAddress *disk;
1038d9fe4f0fSThomas Huth     GuestPCIAddress *pciaddr;
1039d9fe4f0fSThomas Huth     GuestDiskAddressList *list = NULL;
1040d9fe4f0fSThomas Huth     bool has_hwinf;
104143dadc43SThomas Huth #ifdef CONFIG_LIBUDEV
104243dadc43SThomas Huth     struct udev *udev = NULL;
104343dadc43SThomas Huth     struct udev_device *udevice = NULL;
104443dadc43SThomas Huth #endif
1045d9fe4f0fSThomas Huth 
1046d9fe4f0fSThomas Huth     pciaddr = g_new0(GuestPCIAddress, 1);
104743dadc43SThomas Huth     pciaddr->domain = -1;                       /* -1 means field is invalid */
104843dadc43SThomas Huth     pciaddr->bus = -1;
104943dadc43SThomas Huth     pciaddr->slot = -1;
105043dadc43SThomas Huth     pciaddr->function = -1;
1051d9fe4f0fSThomas Huth 
1052d9fe4f0fSThomas Huth     disk = g_new0(GuestDiskAddress, 1);
1053d9fe4f0fSThomas Huth     disk->pci_controller = pciaddr;
105443dadc43SThomas Huth     disk->bus_type = GUEST_DISK_BUS_TYPE_UNKNOWN;
1055d9fe4f0fSThomas Huth 
1056d9fe4f0fSThomas Huth     list = g_new0(GuestDiskAddressList, 1);
1057d9fe4f0fSThomas Huth     list->value = disk;
1058d9fe4f0fSThomas Huth 
105943dadc43SThomas Huth #ifdef CONFIG_LIBUDEV
106043dadc43SThomas Huth     udev = udev_new();
106143dadc43SThomas Huth     udevice = udev_device_new_from_syspath(udev, syspath);
106243dadc43SThomas Huth     if (udev == NULL || udevice == NULL) {
106343dadc43SThomas Huth         g_debug("failed to query udev");
106443dadc43SThomas Huth     } else {
106543dadc43SThomas Huth         const char *devnode, *serial;
106643dadc43SThomas Huth         devnode = udev_device_get_devnode(udevice);
106743dadc43SThomas Huth         if (devnode != NULL) {
106843dadc43SThomas Huth             disk->dev = g_strdup(devnode);
106943dadc43SThomas Huth             disk->has_dev = true;
107043dadc43SThomas Huth         }
107143dadc43SThomas Huth         serial = udev_device_get_property_value(udevice, "ID_SERIAL");
107243dadc43SThomas Huth         if (serial != NULL && *serial != 0) {
107343dadc43SThomas Huth             disk->serial = g_strdup(serial);
107443dadc43SThomas Huth             disk->has_serial = true;
107543dadc43SThomas Huth         }
107643dadc43SThomas Huth     }
107743dadc43SThomas Huth 
107843dadc43SThomas Huth     udev_unref(udev);
107943dadc43SThomas Huth     udev_device_unref(udevice);
108043dadc43SThomas Huth #endif
108143dadc43SThomas Huth 
1082*23843c12SThomas Huth     if (strstr(syspath, "/devices/pci")) {
1083d9fe4f0fSThomas Huth         has_hwinf = build_guest_fsinfo_for_pci_dev(syspath, disk, errp);
1084*23843c12SThomas Huth     } else if (strstr(syspath, "/virtio")) {
1085*23843c12SThomas Huth         has_hwinf = build_guest_fsinfo_for_nonpci_virtio(syspath, disk, errp);
1086*23843c12SThomas Huth     } else {
1087*23843c12SThomas Huth         g_debug("Unsupported device type for '%s'", syspath);
1088*23843c12SThomas Huth         has_hwinf = false;
1089*23843c12SThomas Huth     }
1090d9fe4f0fSThomas Huth 
109143dadc43SThomas Huth     if (has_hwinf || disk->has_dev || disk->has_serial) {
1092d9fe4f0fSThomas Huth         list->next = fs->disk;
1093d9fe4f0fSThomas Huth         fs->disk = list;
1094d9fe4f0fSThomas Huth     } else {
1095d9fe4f0fSThomas Huth         qapi_free_GuestDiskAddressList(list);
1096d9fe4f0fSThomas Huth     }
109746d4c572STomoki Sekiyama }
109846d4c572STomoki Sekiyama 
109946d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
110046d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
110146d4c572STomoki Sekiyama                                           Error **errp);
110246d4c572STomoki Sekiyama 
110346d4c572STomoki Sekiyama /* Store a list of slave devices of virtual volume specified by @syspath into
110446d4c572STomoki Sekiyama  * @fs */
110546d4c572STomoki Sekiyama static void build_guest_fsinfo_for_virtual_device(char const *syspath,
110646d4c572STomoki Sekiyama                                                   GuestFilesystemInfo *fs,
110746d4c572STomoki Sekiyama                                                   Error **errp)
110846d4c572STomoki Sekiyama {
1109292743d9SMarkus Armbruster     Error *err = NULL;
111046d4c572STomoki Sekiyama     DIR *dir;
111146d4c572STomoki Sekiyama     char *dirpath;
1112e668d1b8Szhanghailiang     struct dirent *entry;
111346d4c572STomoki Sekiyama 
111446d4c572STomoki Sekiyama     dirpath = g_strdup_printf("%s/slaves", syspath);
111546d4c572STomoki Sekiyama     dir = opendir(dirpath);
111646d4c572STomoki Sekiyama     if (!dir) {
11178251a72fSMichael Roth         if (errno != ENOENT) {
111846d4c572STomoki Sekiyama             error_setg_errno(errp, errno, "opendir(\"%s\")", dirpath);
11198251a72fSMichael Roth         }
112046d4c572STomoki Sekiyama         g_free(dirpath);
112146d4c572STomoki Sekiyama         return;
112246d4c572STomoki Sekiyama     }
112346d4c572STomoki Sekiyama 
112446d4c572STomoki Sekiyama     for (;;) {
1125e668d1b8Szhanghailiang         errno = 0;
1126e668d1b8Szhanghailiang         entry = readdir(dir);
1127e668d1b8Szhanghailiang         if (entry == NULL) {
1128e668d1b8Szhanghailiang             if (errno) {
1129e668d1b8Szhanghailiang                 error_setg_errno(errp, errno, "readdir(\"%s\")", dirpath);
113046d4c572STomoki Sekiyama             }
113146d4c572STomoki Sekiyama             break;
113246d4c572STomoki Sekiyama         }
113346d4c572STomoki Sekiyama 
1134e668d1b8Szhanghailiang         if (entry->d_type == DT_LNK) {
1135e668d1b8Szhanghailiang             char *path;
1136e668d1b8Szhanghailiang 
1137e668d1b8Szhanghailiang             g_debug(" slave device '%s'", entry->d_name);
1138e668d1b8Szhanghailiang             path = g_strdup_printf("%s/slaves/%s", syspath, entry->d_name);
1139292743d9SMarkus Armbruster             build_guest_fsinfo_for_device(path, fs, &err);
1140e668d1b8Szhanghailiang             g_free(path);
114146d4c572STomoki Sekiyama 
1142292743d9SMarkus Armbruster             if (err) {
1143292743d9SMarkus Armbruster                 error_propagate(errp, err);
114446d4c572STomoki Sekiyama                 break;
114546d4c572STomoki Sekiyama             }
114646d4c572STomoki Sekiyama         }
114746d4c572STomoki Sekiyama     }
114846d4c572STomoki Sekiyama 
1149e668d1b8Szhanghailiang     g_free(dirpath);
115046d4c572STomoki Sekiyama     closedir(dir);
115146d4c572STomoki Sekiyama }
115246d4c572STomoki Sekiyama 
115346d4c572STomoki Sekiyama /* Dispatch to functions for virtual/real device */
115446d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
115546d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
115646d4c572STomoki Sekiyama                                           Error **errp)
115746d4c572STomoki Sekiyama {
115846d4c572STomoki Sekiyama     char *syspath = realpath(devpath, NULL);
115946d4c572STomoki Sekiyama 
116046d4c572STomoki Sekiyama     if (!syspath) {
116146d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "realpath(\"%s\")", devpath);
116246d4c572STomoki Sekiyama         return;
116346d4c572STomoki Sekiyama     }
116446d4c572STomoki Sekiyama 
116546d4c572STomoki Sekiyama     if (!fs->name) {
11663e015d81SJulia Suvorova         fs->name = g_path_get_basename(syspath);
116746d4c572STomoki Sekiyama     }
116846d4c572STomoki Sekiyama 
116946d4c572STomoki Sekiyama     g_debug("  parse sysfs path '%s'", syspath);
117046d4c572STomoki Sekiyama 
117146d4c572STomoki Sekiyama     if (strstr(syspath, "/devices/virtual/block/")) {
117246d4c572STomoki Sekiyama         build_guest_fsinfo_for_virtual_device(syspath, fs, errp);
117346d4c572STomoki Sekiyama     } else {
117446d4c572STomoki Sekiyama         build_guest_fsinfo_for_real_device(syspath, fs, errp);
117546d4c572STomoki Sekiyama     }
117646d4c572STomoki Sekiyama 
117746d4c572STomoki Sekiyama     free(syspath);
117846d4c572STomoki Sekiyama }
117946d4c572STomoki Sekiyama 
118046d4c572STomoki Sekiyama /* Return a list of the disk device(s)' info which @mount lies on */
118146d4c572STomoki Sekiyama static GuestFilesystemInfo *build_guest_fsinfo(struct FsMount *mount,
118246d4c572STomoki Sekiyama                                                Error **errp)
118346d4c572STomoki Sekiyama {
118446d4c572STomoki Sekiyama     GuestFilesystemInfo *fs = g_malloc0(sizeof(*fs));
118525b5ff1aSChen Hanxiao     struct statvfs buf;
118625b5ff1aSChen Hanxiao     unsigned long used, nonroot_total, fr_size;
118746d4c572STomoki Sekiyama     char *devpath = g_strdup_printf("/sys/dev/block/%u:%u",
118846d4c572STomoki Sekiyama                                     mount->devmajor, mount->devminor);
118946d4c572STomoki Sekiyama 
119046d4c572STomoki Sekiyama     fs->mountpoint = g_strdup(mount->dirname);
119146d4c572STomoki Sekiyama     fs->type = g_strdup(mount->devtype);
119246d4c572STomoki Sekiyama     build_guest_fsinfo_for_device(devpath, fs, errp);
119346d4c572STomoki Sekiyama 
119425b5ff1aSChen Hanxiao     if (statvfs(fs->mountpoint, &buf) == 0) {
119525b5ff1aSChen Hanxiao         fr_size = buf.f_frsize;
119625b5ff1aSChen Hanxiao         used = buf.f_blocks - buf.f_bfree;
119725b5ff1aSChen Hanxiao         nonroot_total = used + buf.f_bavail;
119825b5ff1aSChen Hanxiao         fs->used_bytes = used * fr_size;
119925b5ff1aSChen Hanxiao         fs->total_bytes = nonroot_total * fr_size;
120025b5ff1aSChen Hanxiao 
120125b5ff1aSChen Hanxiao         fs->has_total_bytes = true;
120225b5ff1aSChen Hanxiao         fs->has_used_bytes = true;
120325b5ff1aSChen Hanxiao     }
120425b5ff1aSChen Hanxiao 
120546d4c572STomoki Sekiyama     g_free(devpath);
120625b5ff1aSChen Hanxiao 
120746d4c572STomoki Sekiyama     return fs;
120846d4c572STomoki Sekiyama }
120946d4c572STomoki Sekiyama 
121046d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
121146d4c572STomoki Sekiyama {
121246d4c572STomoki Sekiyama     FsMountList mounts;
121346d4c572STomoki Sekiyama     struct FsMount *mount;
121446d4c572STomoki Sekiyama     GuestFilesystemInfoList *new, *ret = NULL;
121546d4c572STomoki Sekiyama     Error *local_err = NULL;
121646d4c572STomoki Sekiyama 
121746d4c572STomoki Sekiyama     QTAILQ_INIT(&mounts);
121846d4c572STomoki Sekiyama     build_fs_mount_list(&mounts, &local_err);
121946d4c572STomoki Sekiyama     if (local_err) {
122046d4c572STomoki Sekiyama         error_propagate(errp, local_err);
122146d4c572STomoki Sekiyama         return NULL;
122246d4c572STomoki Sekiyama     }
122346d4c572STomoki Sekiyama 
122446d4c572STomoki Sekiyama     QTAILQ_FOREACH(mount, &mounts, next) {
122546d4c572STomoki Sekiyama         g_debug("Building guest fsinfo for '%s'", mount->dirname);
122646d4c572STomoki Sekiyama 
122746d4c572STomoki Sekiyama         new = g_malloc0(sizeof(*ret));
122846d4c572STomoki Sekiyama         new->value = build_guest_fsinfo(mount, &local_err);
122946d4c572STomoki Sekiyama         new->next = ret;
123046d4c572STomoki Sekiyama         ret = new;
123146d4c572STomoki Sekiyama         if (local_err) {
123246d4c572STomoki Sekiyama             error_propagate(errp, local_err);
123346d4c572STomoki Sekiyama             qapi_free_GuestFilesystemInfoList(ret);
123446d4c572STomoki Sekiyama             ret = NULL;
123546d4c572STomoki Sekiyama             break;
123646d4c572STomoki Sekiyama         }
123746d4c572STomoki Sekiyama     }
123846d4c572STomoki Sekiyama 
123946d4c572STomoki Sekiyama     free_fs_mount_list(&mounts);
124046d4c572STomoki Sekiyama     return ret;
124146d4c572STomoki Sekiyama }
124246d4c572STomoki Sekiyama 
124346d4c572STomoki Sekiyama 
1244ec0f694cSTomoki Sekiyama typedef enum {
1245ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_THAW = 0,
1246ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_FREEZE,
1247ec0f694cSTomoki Sekiyama } FsfreezeHookArg;
1248ec0f694cSTomoki Sekiyama 
124913a439ecSStefan Weil static const char *fsfreeze_hook_arg_string[] = {
1250ec0f694cSTomoki Sekiyama     "thaw",
1251ec0f694cSTomoki Sekiyama     "freeze",
1252ec0f694cSTomoki Sekiyama };
1253ec0f694cSTomoki Sekiyama 
125477dbc81bSMarkus Armbruster static void execute_fsfreeze_hook(FsfreezeHookArg arg, Error **errp)
1255ec0f694cSTomoki Sekiyama {
1256ec0f694cSTomoki Sekiyama     int status;
1257ec0f694cSTomoki Sekiyama     pid_t pid;
1258ec0f694cSTomoki Sekiyama     const char *hook;
1259ec0f694cSTomoki Sekiyama     const char *arg_str = fsfreeze_hook_arg_string[arg];
1260ec0f694cSTomoki Sekiyama     Error *local_err = NULL;
1261ec0f694cSTomoki Sekiyama 
1262ec0f694cSTomoki Sekiyama     hook = ga_fsfreeze_hook(ga_state);
1263ec0f694cSTomoki Sekiyama     if (!hook) {
1264ec0f694cSTomoki Sekiyama         return;
1265ec0f694cSTomoki Sekiyama     }
1266ec0f694cSTomoki Sekiyama     if (access(hook, X_OK) != 0) {
126777dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "can't access fsfreeze hook '%s'", hook);
1268ec0f694cSTomoki Sekiyama         return;
1269ec0f694cSTomoki Sekiyama     }
1270ec0f694cSTomoki Sekiyama 
1271ec0f694cSTomoki Sekiyama     slog("executing fsfreeze hook with arg '%s'", arg_str);
1272ec0f694cSTomoki Sekiyama     pid = fork();
1273ec0f694cSTomoki Sekiyama     if (pid == 0) {
1274ec0f694cSTomoki Sekiyama         setsid();
1275ec0f694cSTomoki Sekiyama         reopen_fd_to_null(0);
1276ec0f694cSTomoki Sekiyama         reopen_fd_to_null(1);
1277ec0f694cSTomoki Sekiyama         reopen_fd_to_null(2);
1278ec0f694cSTomoki Sekiyama 
1279ec0f694cSTomoki Sekiyama         execle(hook, hook, arg_str, NULL, environ);
1280ec0f694cSTomoki Sekiyama         _exit(EXIT_FAILURE);
1281ec0f694cSTomoki Sekiyama     } else if (pid < 0) {
128277dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
1283ec0f694cSTomoki Sekiyama         return;
1284ec0f694cSTomoki Sekiyama     }
1285ec0f694cSTomoki Sekiyama 
1286ec0f694cSTomoki Sekiyama     ga_wait_child(pid, &status, &local_err);
128784d18f06SMarkus Armbruster     if (local_err) {
128877dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1289ec0f694cSTomoki Sekiyama         return;
1290ec0f694cSTomoki Sekiyama     }
1291ec0f694cSTomoki Sekiyama 
1292ec0f694cSTomoki Sekiyama     if (!WIFEXITED(status)) {
129377dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has terminated abnormally");
1294ec0f694cSTomoki Sekiyama         return;
1295ec0f694cSTomoki Sekiyama     }
1296ec0f694cSTomoki Sekiyama 
1297ec0f694cSTomoki Sekiyama     status = WEXITSTATUS(status);
1298ec0f694cSTomoki Sekiyama     if (status) {
129977dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has failed with status %d", status);
1300ec0f694cSTomoki Sekiyama         return;
1301ec0f694cSTomoki Sekiyama     }
1302ec0f694cSTomoki Sekiyama }
1303ec0f694cSTomoki Sekiyama 
1304c216e5adSMichael Roth /*
1305c216e5adSMichael Roth  * Return status of freeze/thaw
1306c216e5adSMichael Roth  */
130777dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
1308c216e5adSMichael Roth {
1309f22d85e9SMichael Roth     if (ga_is_frozen(ga_state)) {
1310f22d85e9SMichael Roth         return GUEST_FSFREEZE_STATUS_FROZEN;
1311f22d85e9SMichael Roth     }
1312f22d85e9SMichael Roth 
1313f22d85e9SMichael Roth     return GUEST_FSFREEZE_STATUS_THAWED;
1314c216e5adSMichael Roth }
1315c216e5adSMichael Roth 
1316e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze(Error **errp)
1317e99bce20STomoki Sekiyama {
1318e99bce20STomoki Sekiyama     return qmp_guest_fsfreeze_freeze_list(false, NULL, errp);
1319e99bce20STomoki Sekiyama }
1320e99bce20STomoki Sekiyama 
1321c216e5adSMichael Roth /*
1322c216e5adSMichael Roth  * Walk list of mounted file systems in the guest, and freeze the ones which
1323c216e5adSMichael Roth  * are real local file systems.
1324c216e5adSMichael Roth  */
1325e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
1326e99bce20STomoki Sekiyama                                        strList *mountpoints,
1327e99bce20STomoki Sekiyama                                        Error **errp)
1328c216e5adSMichael Roth {
1329c216e5adSMichael Roth     int ret = 0, i = 0;
1330e99bce20STomoki Sekiyama     strList *list;
1331af02203fSPaolo Bonzini     FsMountList mounts;
1332af02203fSPaolo Bonzini     struct FsMount *mount;
1333261551d1SLuiz Capitulino     Error *local_err = NULL;
1334c216e5adSMichael Roth     int fd;
1335c216e5adSMichael Roth 
1336c216e5adSMichael Roth     slog("guest-fsfreeze called");
1337c216e5adSMichael Roth 
1338ec0f694cSTomoki Sekiyama     execute_fsfreeze_hook(FSFREEZE_HOOK_FREEZE, &local_err);
133984d18f06SMarkus Armbruster     if (local_err) {
134077dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1341ec0f694cSTomoki Sekiyama         return -1;
1342ec0f694cSTomoki Sekiyama     }
1343ec0f694cSTomoki Sekiyama 
13449e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1345261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
134684d18f06SMarkus Armbruster     if (local_err) {
134777dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1348261551d1SLuiz Capitulino         return -1;
1349c216e5adSMichael Roth     }
1350c216e5adSMichael Roth 
1351c216e5adSMichael Roth     /* cannot risk guest agent blocking itself on a write in this state */
1352f22d85e9SMichael Roth     ga_set_frozen(ga_state);
1353c216e5adSMichael Roth 
1354eae3eb3eSPaolo Bonzini     QTAILQ_FOREACH_REVERSE(mount, &mounts, next) {
1355e99bce20STomoki Sekiyama         /* To issue fsfreeze in the reverse order of mounts, check if the
1356e99bce20STomoki Sekiyama          * mount is listed in the list here */
1357e99bce20STomoki Sekiyama         if (has_mountpoints) {
1358e99bce20STomoki Sekiyama             for (list = mountpoints; list; list = list->next) {
1359e99bce20STomoki Sekiyama                 if (strcmp(list->value, mount->dirname) == 0) {
1360e99bce20STomoki Sekiyama                     break;
1361e99bce20STomoki Sekiyama                 }
1362e99bce20STomoki Sekiyama             }
1363e99bce20STomoki Sekiyama             if (!list) {
1364e99bce20STomoki Sekiyama                 continue;
1365e99bce20STomoki Sekiyama             }
1366e99bce20STomoki Sekiyama         }
1367e99bce20STomoki Sekiyama 
1368c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1369c216e5adSMichael Roth         if (fd == -1) {
137077dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "failed to open %s", mount->dirname);
1371c216e5adSMichael Roth             goto error;
1372c216e5adSMichael Roth         }
1373c216e5adSMichael Roth 
1374e35916acSMichael Tokarev         /* we try to cull filesystems we know won't work in advance, but other
1375e35916acSMichael Tokarev          * filesystems may not implement fsfreeze for less obvious reasons.
13769e8aded4SMichael Roth          * these will report EOPNOTSUPP. we simply ignore these when tallying
13779e8aded4SMichael Roth          * the number of frozen filesystems.
1378ce2eb6c4SPeter Lieven          * if a filesystem is mounted more than once (aka bind mount) a
1379ce2eb6c4SPeter Lieven          * consecutive attempt to freeze an already frozen filesystem will
1380ce2eb6c4SPeter Lieven          * return EBUSY.
13819e8aded4SMichael Roth          *
13829e8aded4SMichael Roth          * any other error means a failure to freeze a filesystem we
13839e8aded4SMichael Roth          * expect to be freezable, so return an error in those cases
13849e8aded4SMichael Roth          * and return system to thawed state.
1385c216e5adSMichael Roth          */
1386c216e5adSMichael Roth         ret = ioctl(fd, FIFREEZE);
13879e8aded4SMichael Roth         if (ret == -1) {
1388ce2eb6c4SPeter Lieven             if (errno != EOPNOTSUPP && errno != EBUSY) {
138977dbc81bSMarkus Armbruster                 error_setg_errno(errp, errno, "failed to freeze %s",
1390617fbbc1SLuiz Capitulino                                  mount->dirname);
1391c216e5adSMichael Roth                 close(fd);
1392c216e5adSMichael Roth                 goto error;
1393c216e5adSMichael Roth             }
13949e8aded4SMichael Roth         } else {
1395c216e5adSMichael Roth             i++;
1396c216e5adSMichael Roth         }
13979e8aded4SMichael Roth         close(fd);
13989e8aded4SMichael Roth     }
1399c216e5adSMichael Roth 
1400af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
140165650f01SChen Hanxiao     /* We may not issue any FIFREEZE here.
140265650f01SChen Hanxiao      * Just unset ga_state here and ready for the next call.
140365650f01SChen Hanxiao      */
140465650f01SChen Hanxiao     if (i == 0) {
140565650f01SChen Hanxiao         ga_unset_frozen(ga_state);
140665650f01SChen Hanxiao     }
1407c216e5adSMichael Roth     return i;
1408c216e5adSMichael Roth 
1409c216e5adSMichael Roth error:
1410af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1411c216e5adSMichael Roth     qmp_guest_fsfreeze_thaw(NULL);
1412c216e5adSMichael Roth     return 0;
1413c216e5adSMichael Roth }
1414c216e5adSMichael Roth 
1415c216e5adSMichael Roth /*
1416c216e5adSMichael Roth  * Walk list of frozen file systems in the guest, and thaw them.
1417c216e5adSMichael Roth  */
141877dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
1419c216e5adSMichael Roth {
1420c216e5adSMichael Roth     int ret;
1421af02203fSPaolo Bonzini     FsMountList mounts;
1422af02203fSPaolo Bonzini     FsMount *mount;
14239e8aded4SMichael Roth     int fd, i = 0, logged;
1424261551d1SLuiz Capitulino     Error *local_err = NULL;
1425c216e5adSMichael Roth 
14269e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1427261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
142884d18f06SMarkus Armbruster     if (local_err) {
142977dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
14309e8aded4SMichael Roth         return 0;
14319e8aded4SMichael Roth     }
14329e8aded4SMichael Roth 
14339e8aded4SMichael Roth     QTAILQ_FOREACH(mount, &mounts, next) {
14349e8aded4SMichael Roth         logged = false;
1435c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1436c216e5adSMichael Roth         if (fd == -1) {
1437c216e5adSMichael Roth             continue;
1438c216e5adSMichael Roth         }
14399e8aded4SMichael Roth         /* we have no way of knowing whether a filesystem was actually unfrozen
14409e8aded4SMichael Roth          * as a result of a successful call to FITHAW, only that if an error
14419e8aded4SMichael Roth          * was returned the filesystem was *not* unfrozen by that particular
14429e8aded4SMichael Roth          * call.
14439e8aded4SMichael Roth          *
1444a31f0531SJim Meyering          * since multiple preceding FIFREEZEs require multiple calls to FITHAW
14459e8aded4SMichael Roth          * to unfreeze, continuing issuing FITHAW until an error is returned,
14469e8aded4SMichael Roth          * in which case either the filesystem is in an unfreezable state, or,
14479e8aded4SMichael Roth          * more likely, it was thawed previously (and remains so afterward).
14489e8aded4SMichael Roth          *
14499e8aded4SMichael Roth          * also, since the most recent successful call is the one that did
14509e8aded4SMichael Roth          * the actual unfreeze, we can use this to provide an accurate count
14519e8aded4SMichael Roth          * of the number of filesystems unfrozen by guest-fsfreeze-thaw, which
14529e8aded4SMichael Roth          * may * be useful for determining whether a filesystem was unfrozen
14539e8aded4SMichael Roth          * during the freeze/thaw phase by a process other than qemu-ga.
14549e8aded4SMichael Roth          */
14559e8aded4SMichael Roth         do {
1456c216e5adSMichael Roth             ret = ioctl(fd, FITHAW);
14579e8aded4SMichael Roth             if (ret == 0 && !logged) {
1458c216e5adSMichael Roth                 i++;
14599e8aded4SMichael Roth                 logged = true;
14609e8aded4SMichael Roth             }
14619e8aded4SMichael Roth         } while (ret == 0);
14629e8aded4SMichael Roth         close(fd);
1463c216e5adSMichael Roth     }
1464c216e5adSMichael Roth 
1465f22d85e9SMichael Roth     ga_unset_frozen(ga_state);
1466af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1467ec0f694cSTomoki Sekiyama 
146877dbc81bSMarkus Armbruster     execute_fsfreeze_hook(FSFREEZE_HOOK_THAW, errp);
1469ec0f694cSTomoki Sekiyama 
1470c216e5adSMichael Roth     return i;
1471c216e5adSMichael Roth }
1472c216e5adSMichael Roth 
1473c216e5adSMichael Roth static void guest_fsfreeze_cleanup(void)
1474c216e5adSMichael Roth {
1475c216e5adSMichael Roth     Error *err = NULL;
1476c216e5adSMichael Roth 
1477f22d85e9SMichael Roth     if (ga_is_frozen(ga_state) == GUEST_FSFREEZE_STATUS_FROZEN) {
14786f686749SMarkus Armbruster         qmp_guest_fsfreeze_thaw(&err);
14796f686749SMarkus Armbruster         if (err) {
14806f686749SMarkus Armbruster             slog("failed to clean up frozen filesystems: %s",
14816f686749SMarkus Armbruster                  error_get_pretty(err));
14826f686749SMarkus Armbruster             error_free(err);
1483c216e5adSMichael Roth         }
1484c216e5adSMichael Roth     }
1485c216e5adSMichael Roth }
1486e72c3f2eSMichael Roth #endif /* CONFIG_FSFREEZE */
1487c216e5adSMichael Roth 
1488eab5fd59SPaolo Bonzini #if defined(CONFIG_FSTRIM)
1489eab5fd59SPaolo Bonzini /*
1490eab5fd59SPaolo Bonzini  * Walk list of mounted file systems in the guest, and trim them.
1491eab5fd59SPaolo Bonzini  */
1492e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
1493e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
1494eab5fd59SPaolo Bonzini {
1495e82855d9SJustin Ossevoort     GuestFilesystemTrimResponse *response;
1496e82855d9SJustin Ossevoort     GuestFilesystemTrimResultList *list;
1497e82855d9SJustin Ossevoort     GuestFilesystemTrimResult *result;
1498eab5fd59SPaolo Bonzini     int ret = 0;
1499eab5fd59SPaolo Bonzini     FsMountList mounts;
1500eab5fd59SPaolo Bonzini     struct FsMount *mount;
1501eab5fd59SPaolo Bonzini     int fd;
1502261551d1SLuiz Capitulino     Error *local_err = NULL;
150373a652a1SJustin Ossevoort     struct fstrim_range r;
1504eab5fd59SPaolo Bonzini 
1505eab5fd59SPaolo Bonzini     slog("guest-fstrim called");
1506eab5fd59SPaolo Bonzini 
1507eab5fd59SPaolo Bonzini     QTAILQ_INIT(&mounts);
1508261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
150984d18f06SMarkus Armbruster     if (local_err) {
151077dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1511e82855d9SJustin Ossevoort         return NULL;
1512eab5fd59SPaolo Bonzini     }
1513eab5fd59SPaolo Bonzini 
1514e82855d9SJustin Ossevoort     response = g_malloc0(sizeof(*response));
1515e82855d9SJustin Ossevoort 
1516eab5fd59SPaolo Bonzini     QTAILQ_FOREACH(mount, &mounts, next) {
1517e82855d9SJustin Ossevoort         result = g_malloc0(sizeof(*result));
1518e82855d9SJustin Ossevoort         result->path = g_strdup(mount->dirname);
1519e82855d9SJustin Ossevoort 
1520e82855d9SJustin Ossevoort         list = g_malloc0(sizeof(*list));
1521e82855d9SJustin Ossevoort         list->value = result;
1522e82855d9SJustin Ossevoort         list->next = response->paths;
1523e82855d9SJustin Ossevoort         response->paths = list;
1524e82855d9SJustin Ossevoort 
1525eab5fd59SPaolo Bonzini         fd = qemu_open(mount->dirname, O_RDONLY);
1526eab5fd59SPaolo Bonzini         if (fd == -1) {
1527e82855d9SJustin Ossevoort             result->error = g_strdup_printf("failed to open: %s",
1528e82855d9SJustin Ossevoort                                             strerror(errno));
1529e82855d9SJustin Ossevoort             result->has_error = true;
1530e82855d9SJustin Ossevoort             continue;
1531eab5fd59SPaolo Bonzini         }
1532eab5fd59SPaolo Bonzini 
1533e35916acSMichael Tokarev         /* We try to cull filesystems we know won't work in advance, but other
1534e35916acSMichael Tokarev          * filesystems may not implement fstrim for less obvious reasons.
1535e35916acSMichael Tokarev          * These will report EOPNOTSUPP; while in some other cases ENOTTY
1536e35916acSMichael Tokarev          * will be reported (e.g. CD-ROMs).
1537e82855d9SJustin Ossevoort          * Any other error means an unexpected error.
1538eab5fd59SPaolo Bonzini          */
153973a652a1SJustin Ossevoort         r.start = 0;
154073a652a1SJustin Ossevoort         r.len = -1;
154173a652a1SJustin Ossevoort         r.minlen = has_minimum ? minimum : 0;
1542eab5fd59SPaolo Bonzini         ret = ioctl(fd, FITRIM, &r);
1543eab5fd59SPaolo Bonzini         if (ret == -1) {
1544e82855d9SJustin Ossevoort             result->has_error = true;
1545e82855d9SJustin Ossevoort             if (errno == ENOTTY || errno == EOPNOTSUPP) {
1546e82855d9SJustin Ossevoort                 result->error = g_strdup("trim not supported");
1547e82855d9SJustin Ossevoort             } else {
1548e82855d9SJustin Ossevoort                 result->error = g_strdup_printf("failed to trim: %s",
1549e82855d9SJustin Ossevoort                                                 strerror(errno));
1550e82855d9SJustin Ossevoort             }
1551eab5fd59SPaolo Bonzini             close(fd);
1552e82855d9SJustin Ossevoort             continue;
1553eab5fd59SPaolo Bonzini         }
1554e82855d9SJustin Ossevoort 
1555e82855d9SJustin Ossevoort         result->has_minimum = true;
1556e82855d9SJustin Ossevoort         result->minimum = r.minlen;
1557e82855d9SJustin Ossevoort         result->has_trimmed = true;
1558e82855d9SJustin Ossevoort         result->trimmed = r.len;
1559eab5fd59SPaolo Bonzini         close(fd);
1560eab5fd59SPaolo Bonzini     }
1561eab5fd59SPaolo Bonzini 
1562eab5fd59SPaolo Bonzini     free_fs_mount_list(&mounts);
1563e82855d9SJustin Ossevoort     return response;
1564eab5fd59SPaolo Bonzini }
1565eab5fd59SPaolo Bonzini #endif /* CONFIG_FSTRIM */
1566eab5fd59SPaolo Bonzini 
1567eab5fd59SPaolo Bonzini 
156811d0f125SLuiz Capitulino #define LINUX_SYS_STATE_FILE "/sys/power/state"
156911d0f125SLuiz Capitulino #define SUSPEND_SUPPORTED 0
157011d0f125SLuiz Capitulino #define SUSPEND_NOT_SUPPORTED 1
157111d0f125SLuiz Capitulino 
15728b020b5eSDaniel Henrique Barboza typedef enum {
15738b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_DISK = 0,
15748b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_RAM = 1,
15758b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_HYBRID = 2,
15768b020b5eSDaniel Henrique Barboza } SuspendMode;
15778b020b5eSDaniel Henrique Barboza 
15788b020b5eSDaniel Henrique Barboza /*
15798b020b5eSDaniel Henrique Barboza  * Executes a command in a child process using g_spawn_sync,
15808b020b5eSDaniel Henrique Barboza  * returning an int >= 0 representing the exit status of the
15818b020b5eSDaniel Henrique Barboza  * process.
15828b020b5eSDaniel Henrique Barboza  *
15838b020b5eSDaniel Henrique Barboza  * If the program wasn't found in path, returns -1.
15848b020b5eSDaniel Henrique Barboza  *
15858b020b5eSDaniel Henrique Barboza  * If a problem happened when creating the child process,
15868b020b5eSDaniel Henrique Barboza  * returns -1 and errp is set.
15878b020b5eSDaniel Henrique Barboza  */
15888b020b5eSDaniel Henrique Barboza static int run_process_child(const char *command[], Error **errp)
15898b020b5eSDaniel Henrique Barboza {
15908b020b5eSDaniel Henrique Barboza     int exit_status, spawn_flag;
15918b020b5eSDaniel Henrique Barboza     GError *g_err = NULL;
15928b020b5eSDaniel Henrique Barboza     bool success;
15938b020b5eSDaniel Henrique Barboza 
15948b020b5eSDaniel Henrique Barboza     spawn_flag = G_SPAWN_SEARCH_PATH | G_SPAWN_STDOUT_TO_DEV_NULL |
15958b020b5eSDaniel Henrique Barboza                  G_SPAWN_STDERR_TO_DEV_NULL;
15968b020b5eSDaniel Henrique Barboza 
15978b020b5eSDaniel Henrique Barboza     success =  g_spawn_sync(NULL, (char **)command, environ, spawn_flag,
15988b020b5eSDaniel Henrique Barboza                             NULL, NULL, NULL, NULL,
15998b020b5eSDaniel Henrique Barboza                             &exit_status, &g_err);
16008b020b5eSDaniel Henrique Barboza 
16018b020b5eSDaniel Henrique Barboza     if (success) {
16028b020b5eSDaniel Henrique Barboza         return WEXITSTATUS(exit_status);
16038b020b5eSDaniel Henrique Barboza     }
16048b020b5eSDaniel Henrique Barboza 
16058b020b5eSDaniel Henrique Barboza     if (g_err && (g_err->code != G_SPAWN_ERROR_NOENT)) {
16068b020b5eSDaniel Henrique Barboza         error_setg(errp, "failed to create child process, error '%s'",
16078b020b5eSDaniel Henrique Barboza                    g_err->message);
16088b020b5eSDaniel Henrique Barboza     }
16098b020b5eSDaniel Henrique Barboza 
16108b020b5eSDaniel Henrique Barboza     g_error_free(g_err);
16118b020b5eSDaniel Henrique Barboza     return -1;
16128b020b5eSDaniel Henrique Barboza }
16138b020b5eSDaniel Henrique Barboza 
1614067927d6SDaniel Henrique Barboza static bool systemd_supports_mode(SuspendMode mode, Error **errp)
1615067927d6SDaniel Henrique Barboza {
1616067927d6SDaniel Henrique Barboza     const char *systemctl_args[3] = {"systemd-hibernate", "systemd-suspend",
1617067927d6SDaniel Henrique Barboza                                      "systemd-hybrid-sleep"};
1618067927d6SDaniel Henrique Barboza     const char *cmd[4] = {"systemctl", "status", systemctl_args[mode], NULL};
1619067927d6SDaniel Henrique Barboza     int status;
1620067927d6SDaniel Henrique Barboza 
1621992861fbSMarkus Armbruster     status = run_process_child(cmd, errp);
1622067927d6SDaniel Henrique Barboza 
1623067927d6SDaniel Henrique Barboza     /*
1624067927d6SDaniel Henrique Barboza      * systemctl status uses LSB return codes so we can expect
1625067927d6SDaniel Henrique Barboza      * status > 0 and be ok. To assert if the guest has support
1626067927d6SDaniel Henrique Barboza      * for the selected suspend mode, status should be < 4. 4 is
1627067927d6SDaniel Henrique Barboza      * the code for unknown service status, the return value when
1628067927d6SDaniel Henrique Barboza      * the service does not exist. A common value is status = 3
1629067927d6SDaniel Henrique Barboza      * (program is not running).
1630067927d6SDaniel Henrique Barboza      */
1631067927d6SDaniel Henrique Barboza     if (status > 0 && status < 4) {
1632067927d6SDaniel Henrique Barboza         return true;
1633067927d6SDaniel Henrique Barboza     }
1634067927d6SDaniel Henrique Barboza 
1635067927d6SDaniel Henrique Barboza     return false;
1636067927d6SDaniel Henrique Barboza }
1637067927d6SDaniel Henrique Barboza 
1638067927d6SDaniel Henrique Barboza static void systemd_suspend(SuspendMode mode, Error **errp)
1639067927d6SDaniel Henrique Barboza {
1640067927d6SDaniel Henrique Barboza     Error *local_err = NULL;
1641067927d6SDaniel Henrique Barboza     const char *systemctl_args[3] = {"hibernate", "suspend", "hybrid-sleep"};
1642067927d6SDaniel Henrique Barboza     const char *cmd[3] = {"systemctl", systemctl_args[mode], NULL};
1643067927d6SDaniel Henrique Barboza     int status;
1644067927d6SDaniel Henrique Barboza 
1645067927d6SDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1646067927d6SDaniel Henrique Barboza 
1647067927d6SDaniel Henrique Barboza     if (status == 0) {
1648067927d6SDaniel Henrique Barboza         return;
1649067927d6SDaniel Henrique Barboza     }
1650067927d6SDaniel Henrique Barboza 
1651067927d6SDaniel Henrique Barboza     if ((status == -1) && !local_err) {
1652067927d6SDaniel Henrique Barboza         error_setg(errp, "the helper program 'systemctl %s' was not found",
1653067927d6SDaniel Henrique Barboza                    systemctl_args[mode]);
1654067927d6SDaniel Henrique Barboza         return;
1655067927d6SDaniel Henrique Barboza     }
1656067927d6SDaniel Henrique Barboza 
1657067927d6SDaniel Henrique Barboza     if (local_err) {
1658067927d6SDaniel Henrique Barboza         error_propagate(errp, local_err);
1659067927d6SDaniel Henrique Barboza     } else {
1660067927d6SDaniel Henrique Barboza         error_setg(errp, "the helper program 'systemctl %s' returned an "
1661067927d6SDaniel Henrique Barboza                    "unexpected exit status code (%d)",
1662067927d6SDaniel Henrique Barboza                    systemctl_args[mode], status);
1663067927d6SDaniel Henrique Barboza     }
1664067927d6SDaniel Henrique Barboza }
1665067927d6SDaniel Henrique Barboza 
16668b020b5eSDaniel Henrique Barboza static bool pmutils_supports_mode(SuspendMode mode, Error **errp)
166711d0f125SLuiz Capitulino {
16686b26e837SLuiz Capitulino     Error *local_err = NULL;
16698b020b5eSDaniel Henrique Barboza     const char *pmutils_args[3] = {"--hibernate", "--suspend",
16708b020b5eSDaniel Henrique Barboza                                    "--suspend-hybrid"};
16718b020b5eSDaniel Henrique Barboza     const char *cmd[3] = {"pm-is-supported", pmutils_args[mode], NULL};
1672dc8764f0SLuiz Capitulino     int status;
167311d0f125SLuiz Capitulino 
16748b020b5eSDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1675304a0fcbSDaniel Henrique Barboza 
16768b020b5eSDaniel Henrique Barboza     if (status == SUSPEND_SUPPORTED) {
16778b020b5eSDaniel Henrique Barboza         return true;
1678304a0fcbSDaniel Henrique Barboza     }
1679304a0fcbSDaniel Henrique Barboza 
16808b020b5eSDaniel Henrique Barboza     if ((status == -1) && !local_err) {
16818b020b5eSDaniel Henrique Barboza         return false;
1682a5fcf0e3SDaniel Henrique Barboza     }
168311d0f125SLuiz Capitulino 
168484d18f06SMarkus Armbruster     if (local_err) {
168577dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
16868b020b5eSDaniel Henrique Barboza     } else {
168777dbc81bSMarkus Armbruster         error_setg(errp,
16888b020b5eSDaniel Henrique Barboza                    "the helper program '%s' returned an unexpected exit"
16898b020b5eSDaniel Henrique Barboza                    " status code (%d)", "pm-is-supported", status);
1690dc8764f0SLuiz Capitulino     }
169111d0f125SLuiz Capitulino 
16928b020b5eSDaniel Henrique Barboza     return false;
1693a5fcf0e3SDaniel Henrique Barboza }
1694a5fcf0e3SDaniel Henrique Barboza 
16958b020b5eSDaniel Henrique Barboza static void pmutils_suspend(SuspendMode mode, Error **errp)
1696246d76ebSDaniel Henrique Barboza {
1697246d76ebSDaniel Henrique Barboza     Error *local_err = NULL;
16988b020b5eSDaniel Henrique Barboza     const char *pmutils_binaries[3] = {"pm-hibernate", "pm-suspend",
16998b020b5eSDaniel Henrique Barboza                                        "pm-suspend-hybrid"};
17008b020b5eSDaniel Henrique Barboza     const char *cmd[2] = {pmutils_binaries[mode], NULL};
1701246d76ebSDaniel Henrique Barboza     int status;
1702246d76ebSDaniel Henrique Barboza 
17038b020b5eSDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1704246d76ebSDaniel Henrique Barboza 
17058b020b5eSDaniel Henrique Barboza     if (status == 0) {
1706246d76ebSDaniel Henrique Barboza         return;
1707246d76ebSDaniel Henrique Barboza     }
1708246d76ebSDaniel Henrique Barboza 
17098b020b5eSDaniel Henrique Barboza     if ((status == -1) && !local_err) {
17108b020b5eSDaniel Henrique Barboza         error_setg(errp, "the helper program '%s' was not found",
17118b020b5eSDaniel Henrique Barboza                    pmutils_binaries[mode]);
1712246d76ebSDaniel Henrique Barboza         return;
1713246d76ebSDaniel Henrique Barboza     }
1714246d76ebSDaniel Henrique Barboza 
1715246d76ebSDaniel Henrique Barboza     if (local_err) {
1716246d76ebSDaniel Henrique Barboza         error_propagate(errp, local_err);
17178b020b5eSDaniel Henrique Barboza     } else {
1718246d76ebSDaniel Henrique Barboza         error_setg(errp,
17198b020b5eSDaniel Henrique Barboza                    "the helper program '%s' returned an unexpected exit"
17208b020b5eSDaniel Henrique Barboza                    " status code (%d)", pmutils_binaries[mode], status);
17218b020b5eSDaniel Henrique Barboza     }
1722246d76ebSDaniel Henrique Barboza }
1723246d76ebSDaniel Henrique Barboza 
17248b020b5eSDaniel Henrique Barboza static bool linux_sys_state_supports_mode(SuspendMode mode, Error **errp)
1725a5fcf0e3SDaniel Henrique Barboza {
17268b020b5eSDaniel Henrique Barboza     const char *sysfile_strs[3] = {"disk", "mem", NULL};
17278b020b5eSDaniel Henrique Barboza     const char *sysfile_str = sysfile_strs[mode];
1728a5fcf0e3SDaniel Henrique Barboza     char buf[32]; /* hopefully big enough */
1729a5fcf0e3SDaniel Henrique Barboza     int fd;
1730a5fcf0e3SDaniel Henrique Barboza     ssize_t ret;
1731a5fcf0e3SDaniel Henrique Barboza 
17328b020b5eSDaniel Henrique Barboza     if (!sysfile_str) {
17338b020b5eSDaniel Henrique Barboza         error_setg(errp, "unknown guest suspend mode");
1734a5fcf0e3SDaniel Henrique Barboza         return false;
1735a5fcf0e3SDaniel Henrique Barboza     }
1736a5fcf0e3SDaniel Henrique Barboza 
1737a5fcf0e3SDaniel Henrique Barboza     fd = open(LINUX_SYS_STATE_FILE, O_RDONLY);
1738a5fcf0e3SDaniel Henrique Barboza     if (fd < 0) {
1739a5fcf0e3SDaniel Henrique Barboza         return false;
1740a5fcf0e3SDaniel Henrique Barboza     }
1741a5fcf0e3SDaniel Henrique Barboza 
1742a5fcf0e3SDaniel Henrique Barboza     ret = read(fd, buf, sizeof(buf) - 1);
1743d9c745c1SPaolo Bonzini     close(fd);
1744a5fcf0e3SDaniel Henrique Barboza     if (ret <= 0) {
1745a5fcf0e3SDaniel Henrique Barboza         return false;
1746a5fcf0e3SDaniel Henrique Barboza     }
1747a5fcf0e3SDaniel Henrique Barboza     buf[ret] = '\0';
1748a5fcf0e3SDaniel Henrique Barboza 
1749a5fcf0e3SDaniel Henrique Barboza     if (strstr(buf, sysfile_str)) {
1750a5fcf0e3SDaniel Henrique Barboza         return true;
1751a5fcf0e3SDaniel Henrique Barboza     }
1752a5fcf0e3SDaniel Henrique Barboza     return false;
1753a5fcf0e3SDaniel Henrique Barboza }
1754a5fcf0e3SDaniel Henrique Barboza 
17558b020b5eSDaniel Henrique Barboza static void linux_sys_state_suspend(SuspendMode mode, Error **errp)
1756246d76ebSDaniel Henrique Barboza {
1757246d76ebSDaniel Henrique Barboza     Error *local_err = NULL;
17588b020b5eSDaniel Henrique Barboza     const char *sysfile_strs[3] = {"disk", "mem", NULL};
17598b020b5eSDaniel Henrique Barboza     const char *sysfile_str = sysfile_strs[mode];
1760246d76ebSDaniel Henrique Barboza     pid_t pid;
1761246d76ebSDaniel Henrique Barboza     int status;
1762246d76ebSDaniel Henrique Barboza 
17638b020b5eSDaniel Henrique Barboza     if (!sysfile_str) {
1764246d76ebSDaniel Henrique Barboza         error_setg(errp, "unknown guest suspend mode");
1765246d76ebSDaniel Henrique Barboza         return;
1766246d76ebSDaniel Henrique Barboza     }
1767246d76ebSDaniel Henrique Barboza 
1768246d76ebSDaniel Henrique Barboza     pid = fork();
1769246d76ebSDaniel Henrique Barboza     if (!pid) {
1770246d76ebSDaniel Henrique Barboza         /* child */
1771246d76ebSDaniel Henrique Barboza         int fd;
1772246d76ebSDaniel Henrique Barboza 
1773246d76ebSDaniel Henrique Barboza         setsid();
1774246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(0);
1775246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(1);
1776246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(2);
1777246d76ebSDaniel Henrique Barboza 
1778246d76ebSDaniel Henrique Barboza         fd = open(LINUX_SYS_STATE_FILE, O_WRONLY);
1779246d76ebSDaniel Henrique Barboza         if (fd < 0) {
1780246d76ebSDaniel Henrique Barboza             _exit(EXIT_FAILURE);
1781246d76ebSDaniel Henrique Barboza         }
1782246d76ebSDaniel Henrique Barboza 
1783246d76ebSDaniel Henrique Barboza         if (write(fd, sysfile_str, strlen(sysfile_str)) < 0) {
1784246d76ebSDaniel Henrique Barboza             _exit(EXIT_FAILURE);
1785246d76ebSDaniel Henrique Barboza         }
1786246d76ebSDaniel Henrique Barboza 
1787246d76ebSDaniel Henrique Barboza         _exit(EXIT_SUCCESS);
1788246d76ebSDaniel Henrique Barboza     } else if (pid < 0) {
1789246d76ebSDaniel Henrique Barboza         error_setg_errno(errp, errno, "failed to create child process");
1790246d76ebSDaniel Henrique Barboza         return;
1791246d76ebSDaniel Henrique Barboza     }
1792246d76ebSDaniel Henrique Barboza 
1793246d76ebSDaniel Henrique Barboza     ga_wait_child(pid, &status, &local_err);
1794246d76ebSDaniel Henrique Barboza     if (local_err) {
1795246d76ebSDaniel Henrique Barboza         error_propagate(errp, local_err);
1796246d76ebSDaniel Henrique Barboza         return;
1797246d76ebSDaniel Henrique Barboza     }
1798246d76ebSDaniel Henrique Barboza 
1799246d76ebSDaniel Henrique Barboza     if (WEXITSTATUS(status)) {
1800246d76ebSDaniel Henrique Barboza         error_setg(errp, "child process has failed to suspend");
1801246d76ebSDaniel Henrique Barboza     }
1802246d76ebSDaniel Henrique Barboza 
1803246d76ebSDaniel Henrique Barboza }
1804246d76ebSDaniel Henrique Barboza 
18058b020b5eSDaniel Henrique Barboza static void guest_suspend(SuspendMode mode, Error **errp)
180611d0f125SLuiz Capitulino {
18077b376087SLuiz Capitulino     Error *local_err = NULL;
180873e1d8ebSDaniel Henrique Barboza     bool mode_supported = false;
180911d0f125SLuiz Capitulino 
181073e1d8ebSDaniel Henrique Barboza     if (systemd_supports_mode(mode, &local_err)) {
181173e1d8ebSDaniel Henrique Barboza         mode_supported = true;
1812067927d6SDaniel Henrique Barboza         systemd_suspend(mode, &local_err);
181373e1d8ebSDaniel Henrique Barboza     }
181473e1d8ebSDaniel Henrique Barboza 
1815067927d6SDaniel Henrique Barboza     if (!local_err) {
1816067927d6SDaniel Henrique Barboza         return;
1817067927d6SDaniel Henrique Barboza     }
1818067927d6SDaniel Henrique Barboza 
1819067927d6SDaniel Henrique Barboza     error_free(local_err);
18206a4a3853SVladimir Sementsov-Ogievskiy     local_err = NULL;
1821067927d6SDaniel Henrique Barboza 
182273e1d8ebSDaniel Henrique Barboza     if (pmutils_supports_mode(mode, &local_err)) {
182373e1d8ebSDaniel Henrique Barboza         mode_supported = true;
18248b020b5eSDaniel Henrique Barboza         pmutils_suspend(mode, &local_err);
182573e1d8ebSDaniel Henrique Barboza     }
182673e1d8ebSDaniel Henrique Barboza 
1827246d76ebSDaniel Henrique Barboza     if (!local_err) {
1828304a0fcbSDaniel Henrique Barboza         return;
1829304a0fcbSDaniel Henrique Barboza     }
1830304a0fcbSDaniel Henrique Barboza 
1831246d76ebSDaniel Henrique Barboza     error_free(local_err);
18326a4a3853SVladimir Sementsov-Ogievskiy     local_err = NULL;
183311d0f125SLuiz Capitulino 
183473e1d8ebSDaniel Henrique Barboza     if (linux_sys_state_supports_mode(mode, &local_err)) {
183573e1d8ebSDaniel Henrique Barboza         mode_supported = true;
18368b020b5eSDaniel Henrique Barboza         linux_sys_state_suspend(mode, &local_err);
183773e1d8ebSDaniel Henrique Barboza     }
183873e1d8ebSDaniel Henrique Barboza 
183973e1d8ebSDaniel Henrique Barboza     if (!mode_supported) {
18406a4a3853SVladimir Sementsov-Ogievskiy         error_free(local_err);
184173e1d8ebSDaniel Henrique Barboza         error_setg(errp,
184273e1d8ebSDaniel Henrique Barboza                    "the requested suspend mode is not supported by the guest");
1843b2322003SMarkus Armbruster     } else {
184477dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
18457b376087SLuiz Capitulino     }
184611d0f125SLuiz Capitulino }
184711d0f125SLuiz Capitulino 
184877dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
184911d0f125SLuiz Capitulino {
1850304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_DISK, errp);
185111d0f125SLuiz Capitulino }
185211d0f125SLuiz Capitulino 
185377dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
1854fbf42210SLuiz Capitulino {
1855304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_RAM, errp);
1856fbf42210SLuiz Capitulino }
1857fbf42210SLuiz Capitulino 
185877dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
185995f4f404SLuiz Capitulino {
1860304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_HYBRID, errp);
186195f4f404SLuiz Capitulino }
186295f4f404SLuiz Capitulino 
18633424fc9fSMichal Privoznik static GuestNetworkInterfaceList *
18643424fc9fSMichal Privoznik guest_find_interface(GuestNetworkInterfaceList *head,
18653424fc9fSMichal Privoznik                      const char *name)
18663424fc9fSMichal Privoznik {
18673424fc9fSMichal Privoznik     for (; head; head = head->next) {
18683424fc9fSMichal Privoznik         if (strcmp(head->value->name, name) == 0) {
18693424fc9fSMichal Privoznik             break;
18703424fc9fSMichal Privoznik         }
18713424fc9fSMichal Privoznik     }
18723424fc9fSMichal Privoznik 
18733424fc9fSMichal Privoznik     return head;
18743424fc9fSMichal Privoznik }
18753424fc9fSMichal Privoznik 
187653f9fcb2SZhiPeng Lu static int guest_get_network_stats(const char *name,
187753f9fcb2SZhiPeng Lu                        GuestNetworkInterfaceStat *stats)
187853f9fcb2SZhiPeng Lu {
187953f9fcb2SZhiPeng Lu     int name_len;
188053f9fcb2SZhiPeng Lu     char const *devinfo = "/proc/net/dev";
188153f9fcb2SZhiPeng Lu     FILE *fp;
188253f9fcb2SZhiPeng Lu     char *line = NULL, *colon;
188353f9fcb2SZhiPeng Lu     size_t n = 0;
188453f9fcb2SZhiPeng Lu     fp = fopen(devinfo, "r");
188553f9fcb2SZhiPeng Lu     if (!fp) {
188653f9fcb2SZhiPeng Lu         return -1;
188753f9fcb2SZhiPeng Lu     }
188853f9fcb2SZhiPeng Lu     name_len = strlen(name);
188953f9fcb2SZhiPeng Lu     while (getline(&line, &n, fp) != -1) {
189053f9fcb2SZhiPeng Lu         long long dummy;
189153f9fcb2SZhiPeng Lu         long long rx_bytes;
189253f9fcb2SZhiPeng Lu         long long rx_packets;
189353f9fcb2SZhiPeng Lu         long long rx_errs;
189453f9fcb2SZhiPeng Lu         long long rx_dropped;
189553f9fcb2SZhiPeng Lu         long long tx_bytes;
189653f9fcb2SZhiPeng Lu         long long tx_packets;
189753f9fcb2SZhiPeng Lu         long long tx_errs;
189853f9fcb2SZhiPeng Lu         long long tx_dropped;
189953f9fcb2SZhiPeng Lu         char *trim_line;
190053f9fcb2SZhiPeng Lu         trim_line = g_strchug(line);
190153f9fcb2SZhiPeng Lu         if (trim_line[0] == '\0') {
190253f9fcb2SZhiPeng Lu             continue;
190353f9fcb2SZhiPeng Lu         }
190453f9fcb2SZhiPeng Lu         colon = strchr(trim_line, ':');
190553f9fcb2SZhiPeng Lu         if (!colon) {
190653f9fcb2SZhiPeng Lu             continue;
190753f9fcb2SZhiPeng Lu         }
190853f9fcb2SZhiPeng Lu         if (colon - name_len  == trim_line &&
190953f9fcb2SZhiPeng Lu            strncmp(trim_line, name, name_len) == 0) {
191053f9fcb2SZhiPeng Lu             if (sscanf(colon + 1,
191153f9fcb2SZhiPeng Lu                 "%lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld",
191253f9fcb2SZhiPeng Lu                   &rx_bytes, &rx_packets, &rx_errs, &rx_dropped,
191353f9fcb2SZhiPeng Lu                   &dummy, &dummy, &dummy, &dummy,
191453f9fcb2SZhiPeng Lu                   &tx_bytes, &tx_packets, &tx_errs, &tx_dropped,
191553f9fcb2SZhiPeng Lu                   &dummy, &dummy, &dummy, &dummy) != 16) {
191653f9fcb2SZhiPeng Lu                 continue;
191753f9fcb2SZhiPeng Lu             }
191853f9fcb2SZhiPeng Lu             stats->rx_bytes = rx_bytes;
191953f9fcb2SZhiPeng Lu             stats->rx_packets = rx_packets;
192053f9fcb2SZhiPeng Lu             stats->rx_errs = rx_errs;
192153f9fcb2SZhiPeng Lu             stats->rx_dropped = rx_dropped;
192253f9fcb2SZhiPeng Lu             stats->tx_bytes = tx_bytes;
192353f9fcb2SZhiPeng Lu             stats->tx_packets = tx_packets;
192453f9fcb2SZhiPeng Lu             stats->tx_errs = tx_errs;
192553f9fcb2SZhiPeng Lu             stats->tx_dropped = tx_dropped;
192653f9fcb2SZhiPeng Lu             fclose(fp);
192753f9fcb2SZhiPeng Lu             g_free(line);
192853f9fcb2SZhiPeng Lu             return 0;
192953f9fcb2SZhiPeng Lu         }
193053f9fcb2SZhiPeng Lu     }
193153f9fcb2SZhiPeng Lu     fclose(fp);
193253f9fcb2SZhiPeng Lu     g_free(line);
193353f9fcb2SZhiPeng Lu     g_debug("/proc/net/dev: Interface '%s' not found", name);
193453f9fcb2SZhiPeng Lu     return -1;
193553f9fcb2SZhiPeng Lu }
193653f9fcb2SZhiPeng Lu 
19373424fc9fSMichal Privoznik /*
19383424fc9fSMichal Privoznik  * Build information about guest interfaces
19393424fc9fSMichal Privoznik  */
19403424fc9fSMichal Privoznik GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
19413424fc9fSMichal Privoznik {
19423424fc9fSMichal Privoznik     GuestNetworkInterfaceList *head = NULL, *cur_item = NULL;
19433424fc9fSMichal Privoznik     struct ifaddrs *ifap, *ifa;
19443424fc9fSMichal Privoznik 
19453424fc9fSMichal Privoznik     if (getifaddrs(&ifap) < 0) {
1946878a0ae0SLuiz Capitulino         error_setg_errno(errp, errno, "getifaddrs failed");
19473424fc9fSMichal Privoznik         goto error;
19483424fc9fSMichal Privoznik     }
19493424fc9fSMichal Privoznik 
19503424fc9fSMichal Privoznik     for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
19513424fc9fSMichal Privoznik         GuestNetworkInterfaceList *info;
19523424fc9fSMichal Privoznik         GuestIpAddressList **address_list = NULL, *address_item = NULL;
195353f9fcb2SZhiPeng Lu         GuestNetworkInterfaceStat  *interface_stat = NULL;
19543424fc9fSMichal Privoznik         char addr4[INET_ADDRSTRLEN];
19553424fc9fSMichal Privoznik         char addr6[INET6_ADDRSTRLEN];
19563424fc9fSMichal Privoznik         int sock;
19573424fc9fSMichal Privoznik         struct ifreq ifr;
19583424fc9fSMichal Privoznik         unsigned char *mac_addr;
19593424fc9fSMichal Privoznik         void *p;
19603424fc9fSMichal Privoznik 
19613424fc9fSMichal Privoznik         g_debug("Processing %s interface", ifa->ifa_name);
19623424fc9fSMichal Privoznik 
19633424fc9fSMichal Privoznik         info = guest_find_interface(head, ifa->ifa_name);
19643424fc9fSMichal Privoznik 
19653424fc9fSMichal Privoznik         if (!info) {
19663424fc9fSMichal Privoznik             info = g_malloc0(sizeof(*info));
19673424fc9fSMichal Privoznik             info->value = g_malloc0(sizeof(*info->value));
19683424fc9fSMichal Privoznik             info->value->name = g_strdup(ifa->ifa_name);
19693424fc9fSMichal Privoznik 
19703424fc9fSMichal Privoznik             if (!cur_item) {
19713424fc9fSMichal Privoznik                 head = cur_item = info;
19723424fc9fSMichal Privoznik             } else {
19733424fc9fSMichal Privoznik                 cur_item->next = info;
19743424fc9fSMichal Privoznik                 cur_item = info;
19753424fc9fSMichal Privoznik             }
19763424fc9fSMichal Privoznik         }
19773424fc9fSMichal Privoznik 
19783424fc9fSMichal Privoznik         if (!info->value->has_hardware_address &&
19793424fc9fSMichal Privoznik             ifa->ifa_flags & SIOCGIFHWADDR) {
19803424fc9fSMichal Privoznik             /* we haven't obtained HW address yet */
19813424fc9fSMichal Privoznik             sock = socket(PF_INET, SOCK_STREAM, 0);
19823424fc9fSMichal Privoznik             if (sock == -1) {
1983878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "failed to create socket");
19843424fc9fSMichal Privoznik                 goto error;
19853424fc9fSMichal Privoznik             }
19863424fc9fSMichal Privoznik 
19873424fc9fSMichal Privoznik             memset(&ifr, 0, sizeof(ifr));
19881ab516edSJim Meyering             pstrcpy(ifr.ifr_name, IF_NAMESIZE, info->value->name);
19893424fc9fSMichal Privoznik             if (ioctl(sock, SIOCGIFHWADDR, &ifr) == -1) {
1990878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno,
1991878a0ae0SLuiz Capitulino                                  "failed to get MAC address of %s",
1992878a0ae0SLuiz Capitulino                                  ifa->ifa_name);
199310a2158fSMarkus Armbruster                 close(sock);
19943424fc9fSMichal Privoznik                 goto error;
19953424fc9fSMichal Privoznik             }
19963424fc9fSMichal Privoznik 
199710a2158fSMarkus Armbruster             close(sock);
19983424fc9fSMichal Privoznik             mac_addr = (unsigned char *) &ifr.ifr_hwaddr.sa_data;
19993424fc9fSMichal Privoznik 
2000e4ada482SStefan Weil             info->value->hardware_address =
2001e4ada482SStefan Weil                 g_strdup_printf("%02x:%02x:%02x:%02x:%02x:%02x",
20023424fc9fSMichal Privoznik                                 (int) mac_addr[0], (int) mac_addr[1],
20033424fc9fSMichal Privoznik                                 (int) mac_addr[2], (int) mac_addr[3],
2004e4ada482SStefan Weil                                 (int) mac_addr[4], (int) mac_addr[5]);
20053424fc9fSMichal Privoznik 
20063424fc9fSMichal Privoznik             info->value->has_hardware_address = true;
20073424fc9fSMichal Privoznik         }
20083424fc9fSMichal Privoznik 
20093424fc9fSMichal Privoznik         if (ifa->ifa_addr &&
20103424fc9fSMichal Privoznik             ifa->ifa_addr->sa_family == AF_INET) {
20113424fc9fSMichal Privoznik             /* interface with IPv4 address */
20123424fc9fSMichal Privoznik             p = &((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
20133424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET, p, addr4, sizeof(addr4))) {
2014878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
20153424fc9fSMichal Privoznik                 goto error;
20163424fc9fSMichal Privoznik             }
20173424fc9fSMichal Privoznik 
201810a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
201910a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
20203424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr4);
20213424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV4;
20223424fc9fSMichal Privoznik 
20233424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
20243424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
20253424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
20263424fc9fSMichal Privoznik                 p = &((struct sockaddr_in *)ifa->ifa_netmask)->sin_addr;
20273424fc9fSMichal Privoznik                 address_item->value->prefix = ctpop32(((uint32_t *) p)[0]);
20283424fc9fSMichal Privoznik             }
20293424fc9fSMichal Privoznik         } else if (ifa->ifa_addr &&
20303424fc9fSMichal Privoznik                    ifa->ifa_addr->sa_family == AF_INET6) {
20313424fc9fSMichal Privoznik             /* interface with IPv6 address */
20323424fc9fSMichal Privoznik             p = &((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
20333424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET6, p, addr6, sizeof(addr6))) {
2034878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
20353424fc9fSMichal Privoznik                 goto error;
20363424fc9fSMichal Privoznik             }
20373424fc9fSMichal Privoznik 
203810a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
203910a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
20403424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr6);
20413424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV6;
20423424fc9fSMichal Privoznik 
20433424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
20443424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
20453424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
20463424fc9fSMichal Privoznik                 p = &((struct sockaddr_in6 *)ifa->ifa_netmask)->sin6_addr;
20473424fc9fSMichal Privoznik                 address_item->value->prefix =
20483424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[0]) +
20493424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[1]) +
20503424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[2]) +
20513424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[3]);
20523424fc9fSMichal Privoznik             }
20533424fc9fSMichal Privoznik         }
20543424fc9fSMichal Privoznik 
20553424fc9fSMichal Privoznik         if (!address_item) {
20563424fc9fSMichal Privoznik             continue;
20573424fc9fSMichal Privoznik         }
20583424fc9fSMichal Privoznik 
20593424fc9fSMichal Privoznik         address_list = &info->value->ip_addresses;
20603424fc9fSMichal Privoznik 
20613424fc9fSMichal Privoznik         while (*address_list && (*address_list)->next) {
20623424fc9fSMichal Privoznik             address_list = &(*address_list)->next;
20633424fc9fSMichal Privoznik         }
20643424fc9fSMichal Privoznik 
20653424fc9fSMichal Privoznik         if (!*address_list) {
20663424fc9fSMichal Privoznik             *address_list = address_item;
20673424fc9fSMichal Privoznik         } else {
20683424fc9fSMichal Privoznik             (*address_list)->next = address_item;
20693424fc9fSMichal Privoznik         }
20703424fc9fSMichal Privoznik 
20713424fc9fSMichal Privoznik         info->value->has_ip_addresses = true;
20723424fc9fSMichal Privoznik 
207353f9fcb2SZhiPeng Lu         if (!info->value->has_statistics) {
207453f9fcb2SZhiPeng Lu             interface_stat = g_malloc0(sizeof(*interface_stat));
207553f9fcb2SZhiPeng Lu             if (guest_get_network_stats(info->value->name,
207653f9fcb2SZhiPeng Lu                 interface_stat) == -1) {
207753f9fcb2SZhiPeng Lu                 info->value->has_statistics = false;
207853f9fcb2SZhiPeng Lu                 g_free(interface_stat);
207953f9fcb2SZhiPeng Lu             } else {
208053f9fcb2SZhiPeng Lu                 info->value->statistics = interface_stat;
208153f9fcb2SZhiPeng Lu                 info->value->has_statistics = true;
208253f9fcb2SZhiPeng Lu             }
208353f9fcb2SZhiPeng Lu         }
20843424fc9fSMichal Privoznik     }
20853424fc9fSMichal Privoznik 
20863424fc9fSMichal Privoznik     freeifaddrs(ifap);
20873424fc9fSMichal Privoznik     return head;
20883424fc9fSMichal Privoznik 
20893424fc9fSMichal Privoznik error:
20903424fc9fSMichal Privoznik     freeifaddrs(ifap);
20913424fc9fSMichal Privoznik     qapi_free_GuestNetworkInterfaceList(head);
20923424fc9fSMichal Privoznik     return NULL;
20933424fc9fSMichal Privoznik }
20943424fc9fSMichal Privoznik 
209577dbc81bSMarkus Armbruster #define SYSCONF_EXACT(name, errp) sysconf_exact((name), #name, (errp))
2096d2baff62SLaszlo Ersek 
209777dbc81bSMarkus Armbruster static long sysconf_exact(int name, const char *name_str, Error **errp)
2098d2baff62SLaszlo Ersek {
2099d2baff62SLaszlo Ersek     long ret;
2100d2baff62SLaszlo Ersek 
2101d2baff62SLaszlo Ersek     errno = 0;
2102d2baff62SLaszlo Ersek     ret = sysconf(name);
2103d2baff62SLaszlo Ersek     if (ret == -1) {
2104d2baff62SLaszlo Ersek         if (errno == 0) {
210577dbc81bSMarkus Armbruster             error_setg(errp, "sysconf(%s): value indefinite", name_str);
2106d2baff62SLaszlo Ersek         } else {
210777dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "sysconf(%s)", name_str);
2108d2baff62SLaszlo Ersek         }
2109d2baff62SLaszlo Ersek     }
2110d2baff62SLaszlo Ersek     return ret;
2111d2baff62SLaszlo Ersek }
2112d2baff62SLaszlo Ersek 
2113d2baff62SLaszlo Ersek /* Transfer online/offline status between @vcpu and the guest system.
2114d2baff62SLaszlo Ersek  *
2115d2baff62SLaszlo Ersek  * On input either @errp or *@errp must be NULL.
2116d2baff62SLaszlo Ersek  *
2117d2baff62SLaszlo Ersek  * In system-to-@vcpu direction, the following @vcpu fields are accessed:
2118d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
2119d2baff62SLaszlo Ersek  * - W: vcpu->online
2120d2baff62SLaszlo Ersek  * - W: vcpu->can_offline
2121d2baff62SLaszlo Ersek  *
2122d2baff62SLaszlo Ersek  * In @vcpu-to-system direction, the following @vcpu fields are accessed:
2123d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
2124d2baff62SLaszlo Ersek  * - R: vcpu->online
2125d2baff62SLaszlo Ersek  *
2126d2baff62SLaszlo Ersek  * Written members remain unmodified on error.
2127d2baff62SLaszlo Ersek  */
2128d2baff62SLaszlo Ersek static void transfer_vcpu(GuestLogicalProcessor *vcpu, bool sys2vcpu,
2129b4bf912aSIgor Mammedov                           char *dirpath, Error **errp)
2130d2baff62SLaszlo Ersek {
2131b4bf912aSIgor Mammedov     int fd;
2132b4bf912aSIgor Mammedov     int res;
2133d2baff62SLaszlo Ersek     int dirfd;
2134b4bf912aSIgor Mammedov     static const char fn[] = "online";
2135d2baff62SLaszlo Ersek 
2136d2baff62SLaszlo Ersek     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2137d2baff62SLaszlo Ersek     if (dirfd == -1) {
2138d2baff62SLaszlo Ersek         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2139b4bf912aSIgor Mammedov         return;
2140b4bf912aSIgor Mammedov     }
2141d2baff62SLaszlo Ersek 
2142d2baff62SLaszlo Ersek     fd = openat(dirfd, fn, sys2vcpu ? O_RDONLY : O_RDWR);
2143d2baff62SLaszlo Ersek     if (fd == -1) {
2144d2baff62SLaszlo Ersek         if (errno != ENOENT) {
2145d2baff62SLaszlo Ersek             error_setg_errno(errp, errno, "open(\"%s/%s\")", dirpath, fn);
2146d2baff62SLaszlo Ersek         } else if (sys2vcpu) {
2147d2baff62SLaszlo Ersek             vcpu->online = true;
2148d2baff62SLaszlo Ersek             vcpu->can_offline = false;
2149d2baff62SLaszlo Ersek         } else if (!vcpu->online) {
2150d2baff62SLaszlo Ersek             error_setg(errp, "logical processor #%" PRId64 " can't be "
2151d2baff62SLaszlo Ersek                        "offlined", vcpu->logical_id);
2152d2baff62SLaszlo Ersek         } /* otherwise pretend successful re-onlining */
2153d2baff62SLaszlo Ersek     } else {
2154d2baff62SLaszlo Ersek         unsigned char status;
2155d2baff62SLaszlo Ersek 
2156d2baff62SLaszlo Ersek         res = pread(fd, &status, 1, 0);
2157d2baff62SLaszlo Ersek         if (res == -1) {
2158d2baff62SLaszlo Ersek             error_setg_errno(errp, errno, "pread(\"%s/%s\")", dirpath, fn);
2159d2baff62SLaszlo Ersek         } else if (res == 0) {
2160d2baff62SLaszlo Ersek             error_setg(errp, "pread(\"%s/%s\"): unexpected EOF", dirpath,
2161d2baff62SLaszlo Ersek                        fn);
2162d2baff62SLaszlo Ersek         } else if (sys2vcpu) {
2163d2baff62SLaszlo Ersek             vcpu->online = (status != '0');
2164d2baff62SLaszlo Ersek             vcpu->can_offline = true;
2165d2baff62SLaszlo Ersek         } else if (vcpu->online != (status != '0')) {
2166d2baff62SLaszlo Ersek             status = '0' + vcpu->online;
2167d2baff62SLaszlo Ersek             if (pwrite(fd, &status, 1, 0) == -1) {
2168d2baff62SLaszlo Ersek                 error_setg_errno(errp, errno, "pwrite(\"%s/%s\")", dirpath,
2169d2baff62SLaszlo Ersek                                  fn);
2170d2baff62SLaszlo Ersek             }
2171d2baff62SLaszlo Ersek         } /* otherwise pretend successful re-(on|off)-lining */
2172d2baff62SLaszlo Ersek 
2173d2baff62SLaszlo Ersek         res = close(fd);
2174d2baff62SLaszlo Ersek         g_assert(res == 0);
2175d2baff62SLaszlo Ersek     }
2176d2baff62SLaszlo Ersek 
2177d2baff62SLaszlo Ersek     res = close(dirfd);
2178d2baff62SLaszlo Ersek     g_assert(res == 0);
2179d2baff62SLaszlo Ersek }
2180d2baff62SLaszlo Ersek 
2181d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2182d2baff62SLaszlo Ersek {
2183d2baff62SLaszlo Ersek     int64_t current;
2184d2baff62SLaszlo Ersek     GuestLogicalProcessorList *head, **link;
2185d2baff62SLaszlo Ersek     long sc_max;
2186d2baff62SLaszlo Ersek     Error *local_err = NULL;
2187d2baff62SLaszlo Ersek 
2188d2baff62SLaszlo Ersek     current = 0;
2189d2baff62SLaszlo Ersek     head = NULL;
2190d2baff62SLaszlo Ersek     link = &head;
2191d2baff62SLaszlo Ersek     sc_max = SYSCONF_EXACT(_SC_NPROCESSORS_CONF, &local_err);
2192d2baff62SLaszlo Ersek 
2193d2baff62SLaszlo Ersek     while (local_err == NULL && current < sc_max) {
2194d2baff62SLaszlo Ersek         GuestLogicalProcessor *vcpu;
2195d2baff62SLaszlo Ersek         GuestLogicalProcessorList *entry;
2196b4bf912aSIgor Mammedov         int64_t id = current++;
2197b4bf912aSIgor Mammedov         char *path = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
2198b4bf912aSIgor Mammedov                                      id);
2199d2baff62SLaszlo Ersek 
2200b4bf912aSIgor Mammedov         if (g_file_test(path, G_FILE_TEST_EXISTS)) {
2201d2baff62SLaszlo Ersek             vcpu = g_malloc0(sizeof *vcpu);
2202b4bf912aSIgor Mammedov             vcpu->logical_id = id;
2203d2baff62SLaszlo Ersek             vcpu->has_can_offline = true; /* lolspeak ftw */
2204b4bf912aSIgor Mammedov             transfer_vcpu(vcpu, true, path, &local_err);
2205d2baff62SLaszlo Ersek             entry = g_malloc0(sizeof *entry);
2206d2baff62SLaszlo Ersek             entry->value = vcpu;
2207d2baff62SLaszlo Ersek             *link = entry;
2208d2baff62SLaszlo Ersek             link = &entry->next;
2209d2baff62SLaszlo Ersek         }
2210b4bf912aSIgor Mammedov         g_free(path);
2211b4bf912aSIgor Mammedov     }
2212d2baff62SLaszlo Ersek 
2213d2baff62SLaszlo Ersek     if (local_err == NULL) {
2214d2baff62SLaszlo Ersek         /* there's no guest with zero VCPUs */
2215d2baff62SLaszlo Ersek         g_assert(head != NULL);
2216d2baff62SLaszlo Ersek         return head;
2217d2baff62SLaszlo Ersek     }
2218d2baff62SLaszlo Ersek 
2219d2baff62SLaszlo Ersek     qapi_free_GuestLogicalProcessorList(head);
2220d2baff62SLaszlo Ersek     error_propagate(errp, local_err);
2221d2baff62SLaszlo Ersek     return NULL;
2222d2baff62SLaszlo Ersek }
2223d2baff62SLaszlo Ersek 
2224cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2225cbb65fc2SLaszlo Ersek {
2226cbb65fc2SLaszlo Ersek     int64_t processed;
2227cbb65fc2SLaszlo Ersek     Error *local_err = NULL;
2228cbb65fc2SLaszlo Ersek 
2229cbb65fc2SLaszlo Ersek     processed = 0;
2230cbb65fc2SLaszlo Ersek     while (vcpus != NULL) {
2231b4bf912aSIgor Mammedov         char *path = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
2232b4bf912aSIgor Mammedov                                      vcpus->value->logical_id);
2233b4bf912aSIgor Mammedov 
2234b4bf912aSIgor Mammedov         transfer_vcpu(vcpus->value, false, path, &local_err);
2235b4bf912aSIgor Mammedov         g_free(path);
2236cbb65fc2SLaszlo Ersek         if (local_err != NULL) {
2237cbb65fc2SLaszlo Ersek             break;
2238cbb65fc2SLaszlo Ersek         }
2239cbb65fc2SLaszlo Ersek         ++processed;
2240cbb65fc2SLaszlo Ersek         vcpus = vcpus->next;
2241cbb65fc2SLaszlo Ersek     }
2242cbb65fc2SLaszlo Ersek 
2243cbb65fc2SLaszlo Ersek     if (local_err != NULL) {
2244cbb65fc2SLaszlo Ersek         if (processed == 0) {
2245cbb65fc2SLaszlo Ersek             error_propagate(errp, local_err);
2246cbb65fc2SLaszlo Ersek         } else {
2247cbb65fc2SLaszlo Ersek             error_free(local_err);
2248cbb65fc2SLaszlo Ersek         }
2249cbb65fc2SLaszlo Ersek     }
2250cbb65fc2SLaszlo Ersek 
2251cbb65fc2SLaszlo Ersek     return processed;
2252cbb65fc2SLaszlo Ersek }
2253cbb65fc2SLaszlo Ersek 
2254215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2255215a2771SDaniel P. Berrange                                  const char *password,
2256215a2771SDaniel P. Berrange                                  bool crypted,
2257215a2771SDaniel P. Berrange                                  Error **errp)
2258215a2771SDaniel P. Berrange {
2259215a2771SDaniel P. Berrange     Error *local_err = NULL;
2260215a2771SDaniel P. Berrange     char *passwd_path = NULL;
2261215a2771SDaniel P. Berrange     pid_t pid;
2262215a2771SDaniel P. Berrange     int status;
2263215a2771SDaniel P. Berrange     int datafd[2] = { -1, -1 };
2264215a2771SDaniel P. Berrange     char *rawpasswddata = NULL;
2265215a2771SDaniel P. Berrange     size_t rawpasswdlen;
2266215a2771SDaniel P. Berrange     char *chpasswddata = NULL;
2267215a2771SDaniel P. Berrange     size_t chpasswdlen;
2268215a2771SDaniel P. Berrange 
2269920639caSDaniel P. Berrange     rawpasswddata = (char *)qbase64_decode(password, -1, &rawpasswdlen, errp);
2270920639caSDaniel P. Berrange     if (!rawpasswddata) {
2271920639caSDaniel P. Berrange         return;
2272920639caSDaniel P. Berrange     }
2273215a2771SDaniel P. Berrange     rawpasswddata = g_renew(char, rawpasswddata, rawpasswdlen + 1);
2274215a2771SDaniel P. Berrange     rawpasswddata[rawpasswdlen] = '\0';
2275215a2771SDaniel P. Berrange 
2276215a2771SDaniel P. Berrange     if (strchr(rawpasswddata, '\n')) {
2277215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in raw password");
2278215a2771SDaniel P. Berrange         goto out;
2279215a2771SDaniel P. Berrange     }
2280215a2771SDaniel P. Berrange 
2281215a2771SDaniel P. Berrange     if (strchr(username, '\n') ||
2282215a2771SDaniel P. Berrange         strchr(username, ':')) {
2283215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in username");
2284215a2771SDaniel P. Berrange         goto out;
2285215a2771SDaniel P. Berrange     }
2286215a2771SDaniel P. Berrange 
2287215a2771SDaniel P. Berrange     chpasswddata = g_strdup_printf("%s:%s\n", username, rawpasswddata);
2288215a2771SDaniel P. Berrange     chpasswdlen = strlen(chpasswddata);
2289215a2771SDaniel P. Berrange 
2290215a2771SDaniel P. Berrange     passwd_path = g_find_program_in_path("chpasswd");
2291215a2771SDaniel P. Berrange 
2292215a2771SDaniel P. Berrange     if (!passwd_path) {
2293215a2771SDaniel P. Berrange         error_setg(errp, "cannot find 'passwd' program in PATH");
2294215a2771SDaniel P. Berrange         goto out;
2295215a2771SDaniel P. Berrange     }
2296215a2771SDaniel P. Berrange 
2297215a2771SDaniel P. Berrange     if (pipe(datafd) < 0) {
2298215a2771SDaniel P. Berrange         error_setg(errp, "cannot create pipe FDs");
2299215a2771SDaniel P. Berrange         goto out;
2300215a2771SDaniel P. Berrange     }
2301215a2771SDaniel P. Berrange 
2302215a2771SDaniel P. Berrange     pid = fork();
2303215a2771SDaniel P. Berrange     if (pid == 0) {
2304215a2771SDaniel P. Berrange         close(datafd[1]);
2305215a2771SDaniel P. Berrange         /* child */
2306215a2771SDaniel P. Berrange         setsid();
2307215a2771SDaniel P. Berrange         dup2(datafd[0], 0);
2308215a2771SDaniel P. Berrange         reopen_fd_to_null(1);
2309215a2771SDaniel P. Berrange         reopen_fd_to_null(2);
2310215a2771SDaniel P. Berrange 
2311215a2771SDaniel P. Berrange         if (crypted) {
2312215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", "-e", NULL, environ);
2313215a2771SDaniel P. Berrange         } else {
2314215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", NULL, environ);
2315215a2771SDaniel P. Berrange         }
2316215a2771SDaniel P. Berrange         _exit(EXIT_FAILURE);
2317215a2771SDaniel P. Berrange     } else if (pid < 0) {
2318215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "failed to create child process");
2319215a2771SDaniel P. Berrange         goto out;
2320215a2771SDaniel P. Berrange     }
2321215a2771SDaniel P. Berrange     close(datafd[0]);
2322215a2771SDaniel P. Berrange     datafd[0] = -1;
2323215a2771SDaniel P. Berrange 
2324215a2771SDaniel P. Berrange     if (qemu_write_full(datafd[1], chpasswddata, chpasswdlen) != chpasswdlen) {
2325215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "cannot write new account password");
2326215a2771SDaniel P. Berrange         goto out;
2327215a2771SDaniel P. Berrange     }
2328215a2771SDaniel P. Berrange     close(datafd[1]);
2329215a2771SDaniel P. Berrange     datafd[1] = -1;
2330215a2771SDaniel P. Berrange 
2331215a2771SDaniel P. Berrange     ga_wait_child(pid, &status, &local_err);
2332215a2771SDaniel P. Berrange     if (local_err) {
2333215a2771SDaniel P. Berrange         error_propagate(errp, local_err);
2334215a2771SDaniel P. Berrange         goto out;
2335215a2771SDaniel P. Berrange     }
2336215a2771SDaniel P. Berrange 
2337215a2771SDaniel P. Berrange     if (!WIFEXITED(status)) {
2338215a2771SDaniel P. Berrange         error_setg(errp, "child process has terminated abnormally");
2339215a2771SDaniel P. Berrange         goto out;
2340215a2771SDaniel P. Berrange     }
2341215a2771SDaniel P. Berrange 
2342215a2771SDaniel P. Berrange     if (WEXITSTATUS(status)) {
2343215a2771SDaniel P. Berrange         error_setg(errp, "child process has failed to set user password");
2344215a2771SDaniel P. Berrange         goto out;
2345215a2771SDaniel P. Berrange     }
2346215a2771SDaniel P. Berrange 
2347215a2771SDaniel P. Berrange out:
2348215a2771SDaniel P. Berrange     g_free(chpasswddata);
2349215a2771SDaniel P. Berrange     g_free(rawpasswddata);
2350215a2771SDaniel P. Berrange     g_free(passwd_path);
2351215a2771SDaniel P. Berrange     if (datafd[0] != -1) {
2352215a2771SDaniel P. Berrange         close(datafd[0]);
2353215a2771SDaniel P. Berrange     }
2354215a2771SDaniel P. Berrange     if (datafd[1] != -1) {
2355215a2771SDaniel P. Berrange         close(datafd[1]);
2356215a2771SDaniel P. Berrange     }
2357215a2771SDaniel P. Berrange }
2358215a2771SDaniel P. Berrange 
2359bd240fcaSzhanghailiang static void ga_read_sysfs_file(int dirfd, const char *pathname, char *buf,
2360bd240fcaSzhanghailiang                                int size, Error **errp)
2361bd240fcaSzhanghailiang {
2362bd240fcaSzhanghailiang     int fd;
2363bd240fcaSzhanghailiang     int res;
2364bd240fcaSzhanghailiang 
2365bd240fcaSzhanghailiang     errno = 0;
2366bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_RDONLY);
2367bd240fcaSzhanghailiang     if (fd == -1) {
2368bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2369bd240fcaSzhanghailiang         return;
2370bd240fcaSzhanghailiang     }
2371bd240fcaSzhanghailiang 
2372bd240fcaSzhanghailiang     res = pread(fd, buf, size, 0);
2373bd240fcaSzhanghailiang     if (res == -1) {
2374bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pread sysfs file \"%s\"", pathname);
2375bd240fcaSzhanghailiang     } else if (res == 0) {
2376bd240fcaSzhanghailiang         error_setg(errp, "pread sysfs file \"%s\": unexpected EOF", pathname);
2377bd240fcaSzhanghailiang     }
2378bd240fcaSzhanghailiang     close(fd);
2379bd240fcaSzhanghailiang }
2380bd240fcaSzhanghailiang 
2381bd240fcaSzhanghailiang static void ga_write_sysfs_file(int dirfd, const char *pathname,
2382bd240fcaSzhanghailiang                                 const char *buf, int size, Error **errp)
2383bd240fcaSzhanghailiang {
2384bd240fcaSzhanghailiang     int fd;
2385bd240fcaSzhanghailiang 
2386bd240fcaSzhanghailiang     errno = 0;
2387bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_WRONLY);
2388bd240fcaSzhanghailiang     if (fd == -1) {
2389bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2390bd240fcaSzhanghailiang         return;
2391bd240fcaSzhanghailiang     }
2392bd240fcaSzhanghailiang 
2393bd240fcaSzhanghailiang     if (pwrite(fd, buf, size, 0) == -1) {
2394bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pwrite sysfs file \"%s\"", pathname);
2395bd240fcaSzhanghailiang     }
2396bd240fcaSzhanghailiang 
2397bd240fcaSzhanghailiang     close(fd);
2398bd240fcaSzhanghailiang }
2399bd240fcaSzhanghailiang 
2400bd240fcaSzhanghailiang /* Transfer online/offline status between @mem_blk and the guest system.
2401bd240fcaSzhanghailiang  *
2402bd240fcaSzhanghailiang  * On input either @errp or *@errp must be NULL.
2403bd240fcaSzhanghailiang  *
2404bd240fcaSzhanghailiang  * In system-to-@mem_blk direction, the following @mem_blk fields are accessed:
2405bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2406bd240fcaSzhanghailiang  * - W: mem_blk->online
2407bd240fcaSzhanghailiang  * - W: mem_blk->can_offline
2408bd240fcaSzhanghailiang  *
2409bd240fcaSzhanghailiang  * In @mem_blk-to-system direction, the following @mem_blk fields are accessed:
2410bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2411bd240fcaSzhanghailiang  * - R: mem_blk->online
2412bd240fcaSzhanghailiang  *-  R: mem_blk->can_offline
2413bd240fcaSzhanghailiang  * Written members remain unmodified on error.
2414bd240fcaSzhanghailiang  */
2415bd240fcaSzhanghailiang static void transfer_memory_block(GuestMemoryBlock *mem_blk, bool sys2memblk,
2416bd240fcaSzhanghailiang                                   GuestMemoryBlockResponse *result,
2417bd240fcaSzhanghailiang                                   Error **errp)
2418bd240fcaSzhanghailiang {
2419bd240fcaSzhanghailiang     char *dirpath;
2420bd240fcaSzhanghailiang     int dirfd;
2421bd240fcaSzhanghailiang     char *status;
2422bd240fcaSzhanghailiang     Error *local_err = NULL;
2423bd240fcaSzhanghailiang 
2424bd240fcaSzhanghailiang     if (!sys2memblk) {
2425bd240fcaSzhanghailiang         DIR *dp;
2426bd240fcaSzhanghailiang 
2427bd240fcaSzhanghailiang         if (!result) {
2428bd240fcaSzhanghailiang             error_setg(errp, "Internal error, 'result' should not be NULL");
2429bd240fcaSzhanghailiang             return;
2430bd240fcaSzhanghailiang         }
2431bd240fcaSzhanghailiang         errno = 0;
2432bd240fcaSzhanghailiang         dp = opendir("/sys/devices/system/memory/");
2433bd240fcaSzhanghailiang          /* if there is no 'memory' directory in sysfs,
2434bd240fcaSzhanghailiang          * we think this VM does not support online/offline memory block,
2435bd240fcaSzhanghailiang          * any other solution?
2436bd240fcaSzhanghailiang          */
24379879f5acSPhilippe Mathieu-Daudé         if (!dp) {
24389879f5acSPhilippe Mathieu-Daudé             if (errno == ENOENT) {
2439bd240fcaSzhanghailiang                 result->response =
2440bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
24419879f5acSPhilippe Mathieu-Daudé             }
2442bd240fcaSzhanghailiang             goto out1;
2443bd240fcaSzhanghailiang         }
2444bd240fcaSzhanghailiang         closedir(dp);
2445bd240fcaSzhanghailiang     }
2446bd240fcaSzhanghailiang 
2447bd240fcaSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/memory%" PRId64 "/",
2448bd240fcaSzhanghailiang                               mem_blk->phys_index);
2449bd240fcaSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2450bd240fcaSzhanghailiang     if (dirfd == -1) {
2451bd240fcaSzhanghailiang         if (sys2memblk) {
2452bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2453bd240fcaSzhanghailiang         } else {
2454bd240fcaSzhanghailiang             if (errno == ENOENT) {
2455bd240fcaSzhanghailiang                 result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_NOT_FOUND;
2456bd240fcaSzhanghailiang             } else {
2457bd240fcaSzhanghailiang                 result->response =
2458bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2459bd240fcaSzhanghailiang             }
2460bd240fcaSzhanghailiang         }
2461bd240fcaSzhanghailiang         g_free(dirpath);
2462bd240fcaSzhanghailiang         goto out1;
2463bd240fcaSzhanghailiang     }
2464bd240fcaSzhanghailiang     g_free(dirpath);
2465bd240fcaSzhanghailiang 
2466bd240fcaSzhanghailiang     status = g_malloc0(10);
2467bd240fcaSzhanghailiang     ga_read_sysfs_file(dirfd, "state", status, 10, &local_err);
2468bd240fcaSzhanghailiang     if (local_err) {
2469bd240fcaSzhanghailiang         /* treat with sysfs file that not exist in old kernel */
2470bd240fcaSzhanghailiang         if (errno == ENOENT) {
2471bd240fcaSzhanghailiang             error_free(local_err);
2472bd240fcaSzhanghailiang             if (sys2memblk) {
2473bd240fcaSzhanghailiang                 mem_blk->online = true;
2474bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2475bd240fcaSzhanghailiang             } else if (!mem_blk->online) {
2476bd240fcaSzhanghailiang                 result->response =
2477bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
2478bd240fcaSzhanghailiang             }
2479bd240fcaSzhanghailiang         } else {
2480bd240fcaSzhanghailiang             if (sys2memblk) {
2481bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2482bd240fcaSzhanghailiang             } else {
2483b368123dSMarkus Armbruster                 error_free(local_err);
2484bd240fcaSzhanghailiang                 result->response =
2485bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2486bd240fcaSzhanghailiang             }
2487bd240fcaSzhanghailiang         }
2488bd240fcaSzhanghailiang         goto out2;
2489bd240fcaSzhanghailiang     }
2490bd240fcaSzhanghailiang 
2491bd240fcaSzhanghailiang     if (sys2memblk) {
2492bd240fcaSzhanghailiang         char removable = '0';
2493bd240fcaSzhanghailiang 
2494bd240fcaSzhanghailiang         mem_blk->online = (strncmp(status, "online", 6) == 0);
2495bd240fcaSzhanghailiang 
2496bd240fcaSzhanghailiang         ga_read_sysfs_file(dirfd, "removable", &removable, 1, &local_err);
2497bd240fcaSzhanghailiang         if (local_err) {
249867cc32ebSVeres Lajos             /* if no 'removable' file, it doesn't support offline mem blk */
2499bd240fcaSzhanghailiang             if (errno == ENOENT) {
2500bd240fcaSzhanghailiang                 error_free(local_err);
2501bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2502bd240fcaSzhanghailiang             } else {
2503bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2504bd240fcaSzhanghailiang             }
2505bd240fcaSzhanghailiang         } else {
2506bd240fcaSzhanghailiang             mem_blk->can_offline = (removable != '0');
2507bd240fcaSzhanghailiang         }
2508bd240fcaSzhanghailiang     } else {
2509bd240fcaSzhanghailiang         if (mem_blk->online != (strncmp(status, "online", 6) == 0)) {
25107064024dSMarc-André Lureau             const char *new_state = mem_blk->online ? "online" : "offline";
2511bd240fcaSzhanghailiang 
2512bd240fcaSzhanghailiang             ga_write_sysfs_file(dirfd, "state", new_state, strlen(new_state),
2513bd240fcaSzhanghailiang                                 &local_err);
2514bd240fcaSzhanghailiang             if (local_err) {
2515bd240fcaSzhanghailiang                 error_free(local_err);
2516bd240fcaSzhanghailiang                 result->response =
2517bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2518bd240fcaSzhanghailiang                 goto out2;
2519bd240fcaSzhanghailiang             }
2520bd240fcaSzhanghailiang 
2521bd240fcaSzhanghailiang             result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_SUCCESS;
2522bd240fcaSzhanghailiang             result->has_error_code = false;
2523bd240fcaSzhanghailiang         } /* otherwise pretend successful re-(on|off)-lining */
2524bd240fcaSzhanghailiang     }
2525bd240fcaSzhanghailiang     g_free(status);
2526bd240fcaSzhanghailiang     close(dirfd);
2527bd240fcaSzhanghailiang     return;
2528bd240fcaSzhanghailiang 
2529bd240fcaSzhanghailiang out2:
2530bd240fcaSzhanghailiang     g_free(status);
2531bd240fcaSzhanghailiang     close(dirfd);
2532bd240fcaSzhanghailiang out1:
2533bd240fcaSzhanghailiang     if (!sys2memblk) {
2534bd240fcaSzhanghailiang         result->has_error_code = true;
2535bd240fcaSzhanghailiang         result->error_code = errno;
2536bd240fcaSzhanghailiang     }
2537bd240fcaSzhanghailiang }
2538bd240fcaSzhanghailiang 
2539a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2540a065aaa9Szhanghailiang {
2541bd240fcaSzhanghailiang     GuestMemoryBlockList *head, **link;
2542bd240fcaSzhanghailiang     Error *local_err = NULL;
2543bd240fcaSzhanghailiang     struct dirent *de;
2544bd240fcaSzhanghailiang     DIR *dp;
2545bd240fcaSzhanghailiang 
2546bd240fcaSzhanghailiang     head = NULL;
2547bd240fcaSzhanghailiang     link = &head;
2548bd240fcaSzhanghailiang 
2549bd240fcaSzhanghailiang     dp = opendir("/sys/devices/system/memory/");
2550bd240fcaSzhanghailiang     if (!dp) {
2551f693fe6eSMichael Roth         /* it's ok if this happens to be a system that doesn't expose
2552f693fe6eSMichael Roth          * memory blocks via sysfs, but otherwise we should report
2553f693fe6eSMichael Roth          * an error
2554f693fe6eSMichael Roth          */
2555f693fe6eSMichael Roth         if (errno != ENOENT) {
2556bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "Can't open directory"
25579af9e0feSMarkus Armbruster                              "\"/sys/devices/system/memory/\"");
2558f693fe6eSMichael Roth         }
2559bd240fcaSzhanghailiang         return NULL;
2560bd240fcaSzhanghailiang     }
2561bd240fcaSzhanghailiang 
2562bd240fcaSzhanghailiang     /* Note: the phys_index of memory block may be discontinuous,
2563bd240fcaSzhanghailiang      * this is because a memblk is the unit of the Sparse Memory design, which
2564bd240fcaSzhanghailiang      * allows discontinuous memory ranges (ex. NUMA), so here we should
2565bd240fcaSzhanghailiang      * traverse the memory block directory.
2566bd240fcaSzhanghailiang      */
2567bd240fcaSzhanghailiang     while ((de = readdir(dp)) != NULL) {
2568bd240fcaSzhanghailiang         GuestMemoryBlock *mem_blk;
2569bd240fcaSzhanghailiang         GuestMemoryBlockList *entry;
2570bd240fcaSzhanghailiang 
2571bd240fcaSzhanghailiang         if ((strncmp(de->d_name, "memory", 6) != 0) ||
2572bd240fcaSzhanghailiang             !(de->d_type & DT_DIR)) {
2573bd240fcaSzhanghailiang             continue;
2574bd240fcaSzhanghailiang         }
2575bd240fcaSzhanghailiang 
2576bd240fcaSzhanghailiang         mem_blk = g_malloc0(sizeof *mem_blk);
2577bd240fcaSzhanghailiang         /* The d_name is "memoryXXX",  phys_index is block id, same as XXX */
2578bd240fcaSzhanghailiang         mem_blk->phys_index = strtoul(&de->d_name[6], NULL, 10);
2579bd240fcaSzhanghailiang         mem_blk->has_can_offline = true; /* lolspeak ftw */
2580bd240fcaSzhanghailiang         transfer_memory_block(mem_blk, true, NULL, &local_err);
25814155c998SMarkus Armbruster         if (local_err) {
25824155c998SMarkus Armbruster             break;
25834155c998SMarkus Armbruster         }
2584bd240fcaSzhanghailiang 
2585bd240fcaSzhanghailiang         entry = g_malloc0(sizeof *entry);
2586bd240fcaSzhanghailiang         entry->value = mem_blk;
2587bd240fcaSzhanghailiang 
2588bd240fcaSzhanghailiang         *link = entry;
2589bd240fcaSzhanghailiang         link = &entry->next;
2590bd240fcaSzhanghailiang     }
2591bd240fcaSzhanghailiang 
2592bd240fcaSzhanghailiang     closedir(dp);
2593bd240fcaSzhanghailiang     if (local_err == NULL) {
2594bd240fcaSzhanghailiang         /* there's no guest with zero memory blocks */
2595bd240fcaSzhanghailiang         if (head == NULL) {
2596bd240fcaSzhanghailiang             error_setg(errp, "guest reported zero memory blocks!");
2597bd240fcaSzhanghailiang         }
2598bd240fcaSzhanghailiang         return head;
2599bd240fcaSzhanghailiang     }
2600bd240fcaSzhanghailiang 
2601bd240fcaSzhanghailiang     qapi_free_GuestMemoryBlockList(head);
2602bd240fcaSzhanghailiang     error_propagate(errp, local_err);
2603a065aaa9Szhanghailiang     return NULL;
2604a065aaa9Szhanghailiang }
2605a065aaa9Szhanghailiang 
2606a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2607a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2608a065aaa9Szhanghailiang {
260932ca7927Szhanghailiang     GuestMemoryBlockResponseList *head, **link;
261032ca7927Szhanghailiang     Error *local_err = NULL;
261132ca7927Szhanghailiang 
261232ca7927Szhanghailiang     head = NULL;
261332ca7927Szhanghailiang     link = &head;
261432ca7927Szhanghailiang 
261532ca7927Szhanghailiang     while (mem_blks != NULL) {
261632ca7927Szhanghailiang         GuestMemoryBlockResponse *result;
261732ca7927Szhanghailiang         GuestMemoryBlockResponseList *entry;
261832ca7927Szhanghailiang         GuestMemoryBlock *current_mem_blk = mem_blks->value;
261932ca7927Szhanghailiang 
262032ca7927Szhanghailiang         result = g_malloc0(sizeof(*result));
262132ca7927Szhanghailiang         result->phys_index = current_mem_blk->phys_index;
262232ca7927Szhanghailiang         transfer_memory_block(current_mem_blk, false, result, &local_err);
262332ca7927Szhanghailiang         if (local_err) { /* should never happen */
262432ca7927Szhanghailiang             goto err;
262532ca7927Szhanghailiang         }
262632ca7927Szhanghailiang         entry = g_malloc0(sizeof *entry);
262732ca7927Szhanghailiang         entry->value = result;
262832ca7927Szhanghailiang 
262932ca7927Szhanghailiang         *link = entry;
263032ca7927Szhanghailiang         link = &entry->next;
263132ca7927Szhanghailiang         mem_blks = mem_blks->next;
263232ca7927Szhanghailiang     }
263332ca7927Szhanghailiang 
263432ca7927Szhanghailiang     return head;
263532ca7927Szhanghailiang err:
263632ca7927Szhanghailiang     qapi_free_GuestMemoryBlockResponseList(head);
263732ca7927Szhanghailiang     error_propagate(errp, local_err);
2638a065aaa9Szhanghailiang     return NULL;
2639a065aaa9Szhanghailiang }
2640a065aaa9Szhanghailiang 
2641a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2642a065aaa9Szhanghailiang {
2643ef82b60bSzhanghailiang     Error *local_err = NULL;
2644ef82b60bSzhanghailiang     char *dirpath;
2645ef82b60bSzhanghailiang     int dirfd;
2646ef82b60bSzhanghailiang     char *buf;
2647ef82b60bSzhanghailiang     GuestMemoryBlockInfo *info;
2648ef82b60bSzhanghailiang 
2649ef82b60bSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/");
2650ef82b60bSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2651ef82b60bSzhanghailiang     if (dirfd == -1) {
2652ef82b60bSzhanghailiang         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2653ef82b60bSzhanghailiang         g_free(dirpath);
2654a065aaa9Szhanghailiang         return NULL;
2655a065aaa9Szhanghailiang     }
2656ef82b60bSzhanghailiang     g_free(dirpath);
2657ef82b60bSzhanghailiang 
2658ef82b60bSzhanghailiang     buf = g_malloc0(20);
2659ef82b60bSzhanghailiang     ga_read_sysfs_file(dirfd, "block_size_bytes", buf, 20, &local_err);
26608ce1ee46SShannon Zhao     close(dirfd);
2661ef82b60bSzhanghailiang     if (local_err) {
2662ef82b60bSzhanghailiang         g_free(buf);
2663ef82b60bSzhanghailiang         error_propagate(errp, local_err);
2664ef82b60bSzhanghailiang         return NULL;
2665ef82b60bSzhanghailiang     }
2666ef82b60bSzhanghailiang 
2667ef82b60bSzhanghailiang     info = g_new0(GuestMemoryBlockInfo, 1);
2668ef82b60bSzhanghailiang     info->size = strtol(buf, NULL, 16); /* the unit is bytes */
2669ef82b60bSzhanghailiang 
2670ef82b60bSzhanghailiang     g_free(buf);
2671ef82b60bSzhanghailiang 
2672ef82b60bSzhanghailiang     return info;
2673ef82b60bSzhanghailiang }
2674a065aaa9Szhanghailiang 
2675e72c3f2eSMichael Roth #else /* defined(__linux__) */
2676e72c3f2eSMichael Roth 
267777dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
2678e72c3f2eSMichael Roth {
2679c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2680e72c3f2eSMichael Roth }
2681e72c3f2eSMichael Roth 
268277dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
2683e72c3f2eSMichael Roth {
2684c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2685e72c3f2eSMichael Roth }
2686e72c3f2eSMichael Roth 
268777dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
2688e72c3f2eSMichael Roth {
2689c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2690e72c3f2eSMichael Roth }
2691e72c3f2eSMichael Roth 
2692e72c3f2eSMichael Roth GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
2693e72c3f2eSMichael Roth {
2694c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2695e72c3f2eSMichael Roth     return NULL;
2696e72c3f2eSMichael Roth }
2697e72c3f2eSMichael Roth 
2698d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2699d2baff62SLaszlo Ersek {
2700c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2701d2baff62SLaszlo Ersek     return NULL;
2702d2baff62SLaszlo Ersek }
2703d2baff62SLaszlo Ersek 
2704cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2705cbb65fc2SLaszlo Ersek {
2706c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2707cbb65fc2SLaszlo Ersek     return -1;
2708cbb65fc2SLaszlo Ersek }
2709cbb65fc2SLaszlo Ersek 
2710215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2711215a2771SDaniel P. Berrange                                  const char *password,
2712215a2771SDaniel P. Berrange                                  bool crypted,
2713215a2771SDaniel P. Berrange                                  Error **errp)
2714215a2771SDaniel P. Berrange {
2715c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2716215a2771SDaniel P. Berrange }
2717215a2771SDaniel P. Berrange 
2718a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2719a065aaa9Szhanghailiang {
2720c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2721a065aaa9Szhanghailiang     return NULL;
2722a065aaa9Szhanghailiang }
2723a065aaa9Szhanghailiang 
2724a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2725a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2726a065aaa9Szhanghailiang {
2727c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2728a065aaa9Szhanghailiang     return NULL;
2729a065aaa9Szhanghailiang }
2730a065aaa9Szhanghailiang 
2731a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2732a065aaa9Szhanghailiang {
2733c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2734a065aaa9Szhanghailiang     return NULL;
2735a065aaa9Szhanghailiang }
2736a065aaa9Szhanghailiang 
2737e72c3f2eSMichael Roth #endif
2738e72c3f2eSMichael Roth 
2739d35d4cb5SMichael Roth #if !defined(CONFIG_FSFREEZE)
2740d35d4cb5SMichael Roth 
274146d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
274246d4c572STomoki Sekiyama {
2743c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
274446d4c572STomoki Sekiyama     return NULL;
274546d4c572STomoki Sekiyama }
274646d4c572STomoki Sekiyama 
274777dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
2748d35d4cb5SMichael Roth {
2749c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2750d35d4cb5SMichael Roth 
2751d35d4cb5SMichael Roth     return 0;
2752d35d4cb5SMichael Roth }
2753d35d4cb5SMichael Roth 
275477dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_freeze(Error **errp)
2755d35d4cb5SMichael Roth {
2756c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2757d35d4cb5SMichael Roth 
2758d35d4cb5SMichael Roth     return 0;
2759d35d4cb5SMichael Roth }
2760d35d4cb5SMichael Roth 
2761e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
2762e99bce20STomoki Sekiyama                                        strList *mountpoints,
2763e99bce20STomoki Sekiyama                                        Error **errp)
2764e99bce20STomoki Sekiyama {
2765c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2766e99bce20STomoki Sekiyama 
2767e99bce20STomoki Sekiyama     return 0;
2768e99bce20STomoki Sekiyama }
2769e99bce20STomoki Sekiyama 
277077dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
2771d35d4cb5SMichael Roth {
2772c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2773d35d4cb5SMichael Roth 
2774d35d4cb5SMichael Roth     return 0;
2775d35d4cb5SMichael Roth }
2776eab5fd59SPaolo Bonzini #endif /* CONFIG_FSFREEZE */
2777d35d4cb5SMichael Roth 
2778eab5fd59SPaolo Bonzini #if !defined(CONFIG_FSTRIM)
2779e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
2780e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
2781eab5fd59SPaolo Bonzini {
2782c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2783e82855d9SJustin Ossevoort     return NULL;
2784eab5fd59SPaolo Bonzini }
2785d35d4cb5SMichael Roth #endif
2786d35d4cb5SMichael Roth 
27871281c08aSTomoki Sekiyama /* add unsupported commands to the blacklist */
27881281c08aSTomoki Sekiyama GList *ga_command_blacklist_init(GList *blacklist)
27891281c08aSTomoki Sekiyama {
27901281c08aSTomoki Sekiyama #if !defined(__linux__)
27911281c08aSTomoki Sekiyama     {
27921281c08aSTomoki Sekiyama         const char *list[] = {
27931281c08aSTomoki Sekiyama             "guest-suspend-disk", "guest-suspend-ram",
27941281c08aSTomoki Sekiyama             "guest-suspend-hybrid", "guest-network-get-interfaces",
27950dd38a03Szhanghailiang             "guest-get-vcpus", "guest-set-vcpus",
27960dd38a03Szhanghailiang             "guest-get-memory-blocks", "guest-set-memory-blocks",
279728d8dd35SBasil Salman             "guest-get-memory-block-size", "guest-get-memory-block-info",
279828d8dd35SBasil Salman             NULL};
27991281c08aSTomoki Sekiyama         char **p = (char **)list;
28001281c08aSTomoki Sekiyama 
28011281c08aSTomoki Sekiyama         while (*p) {
28024bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
28031281c08aSTomoki Sekiyama         }
28041281c08aSTomoki Sekiyama     }
28051281c08aSTomoki Sekiyama #endif
28061281c08aSTomoki Sekiyama 
28071281c08aSTomoki Sekiyama #if !defined(CONFIG_FSFREEZE)
28081281c08aSTomoki Sekiyama     {
28091281c08aSTomoki Sekiyama         const char *list[] = {
28101281c08aSTomoki Sekiyama             "guest-get-fsinfo", "guest-fsfreeze-status",
28111281c08aSTomoki Sekiyama             "guest-fsfreeze-freeze", "guest-fsfreeze-freeze-list",
28121281c08aSTomoki Sekiyama             "guest-fsfreeze-thaw", "guest-get-fsinfo", NULL};
28131281c08aSTomoki Sekiyama         char **p = (char **)list;
28141281c08aSTomoki Sekiyama 
28151281c08aSTomoki Sekiyama         while (*p) {
28164bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
28171281c08aSTomoki Sekiyama         }
28181281c08aSTomoki Sekiyama     }
28191281c08aSTomoki Sekiyama #endif
28201281c08aSTomoki Sekiyama 
28211281c08aSTomoki Sekiyama #if !defined(CONFIG_FSTRIM)
28224bca81ceSMarc-André Lureau     blacklist = g_list_append(blacklist, g_strdup("guest-fstrim"));
28231281c08aSTomoki Sekiyama #endif
28241281c08aSTomoki Sekiyama 
28251281c08aSTomoki Sekiyama     return blacklist;
28261281c08aSTomoki Sekiyama }
28271281c08aSTomoki Sekiyama 
2828c216e5adSMichael Roth /* register init/cleanup routines for stateful command groups */
2829c216e5adSMichael Roth void ga_command_state_init(GAState *s, GACommandState *cs)
2830c216e5adSMichael Roth {
2831c216e5adSMichael Roth #if defined(CONFIG_FSFREEZE)
2832f22d85e9SMichael Roth     ga_command_state_add(cs, NULL, guest_fsfreeze_cleanup);
2833c216e5adSMichael Roth #endif
2834c216e5adSMichael Roth }
2835161a56a9SVinzenz Feenstra 
2836e674605fSTomáš Golembiovský #ifdef HAVE_UTMPX
2837e674605fSTomáš Golembiovský 
2838161a56a9SVinzenz Feenstra #define QGA_MICRO_SECOND_TO_SECOND 1000000
2839161a56a9SVinzenz Feenstra 
2840161a56a9SVinzenz Feenstra static double ga_get_login_time(struct utmpx *user_info)
2841161a56a9SVinzenz Feenstra {
2842161a56a9SVinzenz Feenstra     double seconds = (double)user_info->ut_tv.tv_sec;
2843161a56a9SVinzenz Feenstra     double useconds = (double)user_info->ut_tv.tv_usec;
2844161a56a9SVinzenz Feenstra     useconds /= QGA_MICRO_SECOND_TO_SECOND;
2845161a56a9SVinzenz Feenstra     return seconds + useconds;
2846161a56a9SVinzenz Feenstra }
2847161a56a9SVinzenz Feenstra 
2848b90abbacSVladimir Sementsov-Ogievskiy GuestUserList *qmp_guest_get_users(Error **errp)
2849161a56a9SVinzenz Feenstra {
2850161a56a9SVinzenz Feenstra     GHashTable *cache = NULL;
2851161a56a9SVinzenz Feenstra     GuestUserList *head = NULL, *cur_item = NULL;
2852161a56a9SVinzenz Feenstra     struct utmpx *user_info = NULL;
2853161a56a9SVinzenz Feenstra     gpointer value = NULL;
2854161a56a9SVinzenz Feenstra     GuestUser *user = NULL;
2855161a56a9SVinzenz Feenstra     GuestUserList *item = NULL;
2856161a56a9SVinzenz Feenstra     double login_time = 0;
2857161a56a9SVinzenz Feenstra 
2858161a56a9SVinzenz Feenstra     cache = g_hash_table_new(g_str_hash, g_str_equal);
2859161a56a9SVinzenz Feenstra     setutxent();
2860161a56a9SVinzenz Feenstra 
2861161a56a9SVinzenz Feenstra     for (;;) {
2862161a56a9SVinzenz Feenstra         user_info = getutxent();
2863161a56a9SVinzenz Feenstra         if (user_info == NULL) {
2864161a56a9SVinzenz Feenstra             break;
2865161a56a9SVinzenz Feenstra         } else if (user_info->ut_type != USER_PROCESS) {
2866161a56a9SVinzenz Feenstra             continue;
2867161a56a9SVinzenz Feenstra         } else if (g_hash_table_contains(cache, user_info->ut_user)) {
2868161a56a9SVinzenz Feenstra             value = g_hash_table_lookup(cache, user_info->ut_user);
2869161a56a9SVinzenz Feenstra             user = (GuestUser *)value;
2870161a56a9SVinzenz Feenstra             login_time = ga_get_login_time(user_info);
2871161a56a9SVinzenz Feenstra             /* We're ensuring the earliest login time to be sent */
2872161a56a9SVinzenz Feenstra             if (login_time < user->login_time) {
2873161a56a9SVinzenz Feenstra                 user->login_time = login_time;
2874161a56a9SVinzenz Feenstra             }
2875161a56a9SVinzenz Feenstra             continue;
2876161a56a9SVinzenz Feenstra         }
2877161a56a9SVinzenz Feenstra 
2878161a56a9SVinzenz Feenstra         item = g_new0(GuestUserList, 1);
2879161a56a9SVinzenz Feenstra         item->value = g_new0(GuestUser, 1);
2880161a56a9SVinzenz Feenstra         item->value->user = g_strdup(user_info->ut_user);
2881161a56a9SVinzenz Feenstra         item->value->login_time = ga_get_login_time(user_info);
2882161a56a9SVinzenz Feenstra 
2883161a56a9SVinzenz Feenstra         g_hash_table_insert(cache, item->value->user, item->value);
2884161a56a9SVinzenz Feenstra 
2885161a56a9SVinzenz Feenstra         if (!cur_item) {
2886161a56a9SVinzenz Feenstra             head = cur_item = item;
2887161a56a9SVinzenz Feenstra         } else {
2888161a56a9SVinzenz Feenstra             cur_item->next = item;
2889161a56a9SVinzenz Feenstra             cur_item = item;
2890161a56a9SVinzenz Feenstra         }
2891161a56a9SVinzenz Feenstra     }
2892161a56a9SVinzenz Feenstra     endutxent();
2893161a56a9SVinzenz Feenstra     g_hash_table_destroy(cache);
2894161a56a9SVinzenz Feenstra     return head;
2895161a56a9SVinzenz Feenstra }
2896e674605fSTomáš Golembiovský 
2897e674605fSTomáš Golembiovský #else
2898e674605fSTomáš Golembiovský 
2899e674605fSTomáš Golembiovský GuestUserList *qmp_guest_get_users(Error **errp)
2900e674605fSTomáš Golembiovský {
2901e674605fSTomáš Golembiovský     error_setg(errp, QERR_UNSUPPORTED);
2902e674605fSTomáš Golembiovský     return NULL;
2903e674605fSTomáš Golembiovský }
2904e674605fSTomáš Golembiovský 
2905e674605fSTomáš Golembiovský #endif
29069848f797STomáš Golembiovský 
29079848f797STomáš Golembiovský /* Replace escaped special characters with theire real values. The replacement
29089848f797STomáš Golembiovský  * is done in place -- returned value is in the original string.
29099848f797STomáš Golembiovský  */
29109848f797STomáš Golembiovský static void ga_osrelease_replace_special(gchar *value)
29119848f797STomáš Golembiovský {
29129848f797STomáš Golembiovský     gchar *p, *p2, quote;
29139848f797STomáš Golembiovský 
29149848f797STomáš Golembiovský     /* Trim the string at first space or semicolon if it is not enclosed in
29159848f797STomáš Golembiovský      * single or double quotes. */
29169848f797STomáš Golembiovský     if ((value[0] != '"') || (value[0] == '\'')) {
29179848f797STomáš Golembiovský         p = strchr(value, ' ');
29189848f797STomáš Golembiovský         if (p != NULL) {
29199848f797STomáš Golembiovský             *p = 0;
29209848f797STomáš Golembiovský         }
29219848f797STomáš Golembiovský         p = strchr(value, ';');
29229848f797STomáš Golembiovský         if (p != NULL) {
29239848f797STomáš Golembiovský             *p = 0;
29249848f797STomáš Golembiovský         }
29259848f797STomáš Golembiovský         return;
29269848f797STomáš Golembiovský     }
29279848f797STomáš Golembiovský 
29289848f797STomáš Golembiovský     quote = value[0];
29299848f797STomáš Golembiovský     p2 = value;
29309848f797STomáš Golembiovský     p = value + 1;
29319848f797STomáš Golembiovský     while (*p != 0) {
29329848f797STomáš Golembiovský         if (*p == '\\') {
29339848f797STomáš Golembiovský             p++;
29349848f797STomáš Golembiovský             switch (*p) {
29359848f797STomáš Golembiovský             case '$':
29369848f797STomáš Golembiovský             case '\'':
29379848f797STomáš Golembiovský             case '"':
29389848f797STomáš Golembiovský             case '\\':
29399848f797STomáš Golembiovský             case '`':
29409848f797STomáš Golembiovský                 break;
29419848f797STomáš Golembiovský             default:
29429848f797STomáš Golembiovský                 /* Keep literal backslash followed by whatever is there */
29439848f797STomáš Golembiovský                 p--;
29449848f797STomáš Golembiovský                 break;
29459848f797STomáš Golembiovský             }
29469848f797STomáš Golembiovský         } else if (*p == quote) {
29479848f797STomáš Golembiovský             *p2 = 0;
29489848f797STomáš Golembiovský             break;
29499848f797STomáš Golembiovský         }
29509848f797STomáš Golembiovský         *(p2++) = *(p++);
29519848f797STomáš Golembiovský     }
29529848f797STomáš Golembiovský }
29539848f797STomáš Golembiovský 
29549848f797STomáš Golembiovský static GKeyFile *ga_parse_osrelease(const char *fname)
29559848f797STomáš Golembiovský {
29569848f797STomáš Golembiovský     gchar *content = NULL;
29579848f797STomáš Golembiovský     gchar *content2 = NULL;
29589848f797STomáš Golembiovský     GError *err = NULL;
29599848f797STomáš Golembiovský     GKeyFile *keys = g_key_file_new();
29609848f797STomáš Golembiovský     const char *group = "[os-release]\n";
29619848f797STomáš Golembiovský 
29629848f797STomáš Golembiovský     if (!g_file_get_contents(fname, &content, NULL, &err)) {
29639848f797STomáš Golembiovský         slog("failed to read '%s', error: %s", fname, err->message);
29649848f797STomáš Golembiovský         goto fail;
29659848f797STomáš Golembiovský     }
29669848f797STomáš Golembiovský 
29679848f797STomáš Golembiovský     if (!g_utf8_validate(content, -1, NULL)) {
29689848f797STomáš Golembiovský         slog("file is not utf-8 encoded: %s", fname);
29699848f797STomáš Golembiovský         goto fail;
29709848f797STomáš Golembiovský     }
29719848f797STomáš Golembiovský     content2 = g_strdup_printf("%s%s", group, content);
29729848f797STomáš Golembiovský 
29739848f797STomáš Golembiovský     if (!g_key_file_load_from_data(keys, content2, -1, G_KEY_FILE_NONE,
29749848f797STomáš Golembiovský                                    &err)) {
29759848f797STomáš Golembiovský         slog("failed to parse file '%s', error: %s", fname, err->message);
29769848f797STomáš Golembiovský         goto fail;
29779848f797STomáš Golembiovský     }
29789848f797STomáš Golembiovský 
29799848f797STomáš Golembiovský     g_free(content);
29809848f797STomáš Golembiovský     g_free(content2);
29819848f797STomáš Golembiovský     return keys;
29829848f797STomáš Golembiovský 
29839848f797STomáš Golembiovský fail:
29849848f797STomáš Golembiovský     g_error_free(err);
29859848f797STomáš Golembiovský     g_free(content);
29869848f797STomáš Golembiovský     g_free(content2);
29879848f797STomáš Golembiovský     g_key_file_free(keys);
29889848f797STomáš Golembiovský     return NULL;
29899848f797STomáš Golembiovský }
29909848f797STomáš Golembiovský 
29919848f797STomáš Golembiovský GuestOSInfo *qmp_guest_get_osinfo(Error **errp)
29929848f797STomáš Golembiovský {
29939848f797STomáš Golembiovský     GuestOSInfo *info = NULL;
29949848f797STomáš Golembiovský     struct utsname kinfo;
2995339ca68bSTomáš Golembiovský     GKeyFile *osrelease = NULL;
2996339ca68bSTomáš Golembiovský     const char *qga_os_release = g_getenv("QGA_OS_RELEASE");
29979848f797STomáš Golembiovský 
29989848f797STomáš Golembiovský     info = g_new0(GuestOSInfo, 1);
29999848f797STomáš Golembiovský 
30009848f797STomáš Golembiovský     if (uname(&kinfo) != 0) {
30019848f797STomáš Golembiovský         error_setg_errno(errp, errno, "uname failed");
30029848f797STomáš Golembiovský     } else {
30039848f797STomáš Golembiovský         info->has_kernel_version = true;
30049848f797STomáš Golembiovský         info->kernel_version = g_strdup(kinfo.version);
30059848f797STomáš Golembiovský         info->has_kernel_release = true;
30069848f797STomáš Golembiovský         info->kernel_release = g_strdup(kinfo.release);
30079848f797STomáš Golembiovský         info->has_machine = true;
30089848f797STomáš Golembiovský         info->machine = g_strdup(kinfo.machine);
30099848f797STomáš Golembiovský     }
30109848f797STomáš Golembiovský 
3011339ca68bSTomáš Golembiovský     if (qga_os_release != NULL) {
3012339ca68bSTomáš Golembiovský         osrelease = ga_parse_osrelease(qga_os_release);
3013339ca68bSTomáš Golembiovský     } else {
30149848f797STomáš Golembiovský         osrelease = ga_parse_osrelease("/etc/os-release");
30159848f797STomáš Golembiovský         if (osrelease == NULL) {
30169848f797STomáš Golembiovský             osrelease = ga_parse_osrelease("/usr/lib/os-release");
30179848f797STomáš Golembiovský         }
3018339ca68bSTomáš Golembiovský     }
30199848f797STomáš Golembiovský 
30209848f797STomáš Golembiovský     if (osrelease != NULL) {
30219848f797STomáš Golembiovský         char *value;
30229848f797STomáš Golembiovský 
30239848f797STomáš Golembiovský #define GET_FIELD(field, osfield) do { \
30249848f797STomáš Golembiovský     value = g_key_file_get_value(osrelease, "os-release", osfield, NULL); \
30259848f797STomáš Golembiovský     if (value != NULL) { \
30269848f797STomáš Golembiovský         ga_osrelease_replace_special(value); \
30279848f797STomáš Golembiovský         info->has_ ## field = true; \
30289848f797STomáš Golembiovský         info->field = value; \
30299848f797STomáš Golembiovský     } \
30309848f797STomáš Golembiovský } while (0)
30319848f797STomáš Golembiovský         GET_FIELD(id, "ID");
30329848f797STomáš Golembiovský         GET_FIELD(name, "NAME");
30339848f797STomáš Golembiovský         GET_FIELD(pretty_name, "PRETTY_NAME");
30349848f797STomáš Golembiovský         GET_FIELD(version, "VERSION");
30359848f797STomáš Golembiovský         GET_FIELD(version_id, "VERSION_ID");
30369848f797STomáš Golembiovský         GET_FIELD(variant, "VARIANT");
30379848f797STomáš Golembiovský         GET_FIELD(variant_id, "VARIANT_ID");
30389848f797STomáš Golembiovský #undef GET_FIELD
30399848f797STomáš Golembiovský 
30409848f797STomáš Golembiovský         g_key_file_free(osrelease);
30419848f797STomáš Golembiovský     }
30429848f797STomáš Golembiovský 
30439848f797STomáš Golembiovský     return info;
30449848f797STomáš Golembiovský }
3045