xref: /openbmc/qemu/qga/commands-posix.c (revision 292743d9)
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"
29c216e5adSMichael Roth 
30e674605fSTomáš Golembiovský #ifdef HAVE_UTMPX
31e674605fSTomáš Golembiovský #include <utmpx.h>
32e674605fSTomáš Golembiovský #endif
33e674605fSTomáš Golembiovský 
342c02cbf6SLuiz Capitulino #ifndef CONFIG_HAS_ENVIRON
35eecae147SAndreas Färber #ifdef __APPLE__
36eecae147SAndreas Färber #include <crt_externs.h>
37eecae147SAndreas Färber #define environ (*_NSGetEnviron())
38eecae147SAndreas Färber #else
392c02cbf6SLuiz Capitulino extern char **environ;
402c02cbf6SLuiz Capitulino #endif
41eecae147SAndreas Färber #endif
422c02cbf6SLuiz Capitulino 
43e72c3f2eSMichael Roth #if defined(__linux__)
44e72c3f2eSMichael Roth #include <mntent.h>
45e72c3f2eSMichael Roth #include <linux/fs.h>
46e72c3f2eSMichael Roth #include <ifaddrs.h>
47e72c3f2eSMichael Roth #include <arpa/inet.h>
48e72c3f2eSMichael Roth #include <sys/socket.h>
49e72c3f2eSMichael Roth #include <net/if.h>
5025b5ff1aSChen Hanxiao #include <sys/statvfs.h>
51e72c3f2eSMichael Roth 
52b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
53b616105aSTomáš Golembiovský #include <libudev.h>
54b616105aSTomáš Golembiovský #endif
55b616105aSTomáš Golembiovský 
56eab5fd59SPaolo Bonzini #ifdef FIFREEZE
57e72c3f2eSMichael Roth #define CONFIG_FSFREEZE
58e72c3f2eSMichael Roth #endif
59eab5fd59SPaolo Bonzini #ifdef FITRIM
60eab5fd59SPaolo Bonzini #define CONFIG_FSTRIM
61eab5fd59SPaolo Bonzini #endif
62e72c3f2eSMichael Roth #endif
63e72c3f2eSMichael Roth 
6477dbc81bSMarkus Armbruster static void ga_wait_child(pid_t pid, int *status, Error **errp)
65d220a6dfSLuiz Capitulino {
66d220a6dfSLuiz Capitulino     pid_t rpid;
67d220a6dfSLuiz Capitulino 
68d220a6dfSLuiz Capitulino     *status = 0;
69d220a6dfSLuiz Capitulino 
70d220a6dfSLuiz Capitulino     do {
71d220a6dfSLuiz Capitulino         rpid = waitpid(pid, status, 0);
72d220a6dfSLuiz Capitulino     } while (rpid == -1 && errno == EINTR);
73d220a6dfSLuiz Capitulino 
74d220a6dfSLuiz Capitulino     if (rpid == -1) {
7577dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to wait for child (pid: %d)",
7677dbc81bSMarkus Armbruster                          pid);
77d220a6dfSLuiz Capitulino         return;
78d220a6dfSLuiz Capitulino     }
79d220a6dfSLuiz Capitulino 
80d220a6dfSLuiz Capitulino     g_assert(rpid == pid);
81d220a6dfSLuiz Capitulino }
82d220a6dfSLuiz Capitulino 
8377dbc81bSMarkus Armbruster void qmp_guest_shutdown(bool has_mode, const char *mode, Error **errp)
84c216e5adSMichael Roth {
85c216e5adSMichael Roth     const char *shutdown_flag;
86d220a6dfSLuiz Capitulino     Error *local_err = NULL;
87d220a6dfSLuiz Capitulino     pid_t pid;
883674838cSLuiz Capitulino     int status;
89c216e5adSMichael Roth 
90c216e5adSMichael Roth     slog("guest-shutdown called, mode: %s", mode);
91c216e5adSMichael Roth     if (!has_mode || strcmp(mode, "powerdown") == 0) {
92c216e5adSMichael Roth         shutdown_flag = "-P";
93c216e5adSMichael Roth     } else if (strcmp(mode, "halt") == 0) {
94c216e5adSMichael Roth         shutdown_flag = "-H";
95c216e5adSMichael Roth     } else if (strcmp(mode, "reboot") == 0) {
96c216e5adSMichael Roth         shutdown_flag = "-r";
97c216e5adSMichael Roth     } else {
9877dbc81bSMarkus Armbruster         error_setg(errp,
99d220a6dfSLuiz Capitulino                    "mode is invalid (valid values are: halt|powerdown|reboot");
100c216e5adSMichael Roth         return;
101c216e5adSMichael Roth     }
102c216e5adSMichael Roth 
103d5dd3498SLuiz Capitulino     pid = fork();
104d5dd3498SLuiz Capitulino     if (pid == 0) {
105c216e5adSMichael Roth         /* child, start the shutdown */
106c216e5adSMichael Roth         setsid();
1073674838cSLuiz Capitulino         reopen_fd_to_null(0);
1083674838cSLuiz Capitulino         reopen_fd_to_null(1);
1093674838cSLuiz Capitulino         reopen_fd_to_null(2);
110c216e5adSMichael Roth 
111485e741cSwhitearchey         execle("/sbin/shutdown", "shutdown", "-h", shutdown_flag, "+0",
1123674838cSLuiz Capitulino                "hypervisor initiated shutdown", (char*)NULL, environ);
1133674838cSLuiz Capitulino         _exit(EXIT_FAILURE);
114d5dd3498SLuiz Capitulino     } else if (pid < 0) {
11577dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
116d5dd3498SLuiz Capitulino         return;
117d5dd3498SLuiz Capitulino     }
118d5dd3498SLuiz Capitulino 
119d220a6dfSLuiz Capitulino     ga_wait_child(pid, &status, &local_err);
12084d18f06SMarkus Armbruster     if (local_err) {
12177dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
122d220a6dfSLuiz Capitulino         return;
123d220a6dfSLuiz Capitulino     }
124d220a6dfSLuiz Capitulino 
125d220a6dfSLuiz Capitulino     if (!WIFEXITED(status)) {
12677dbc81bSMarkus Armbruster         error_setg(errp, "child process has terminated abnormally");
127d220a6dfSLuiz Capitulino         return;
128d220a6dfSLuiz Capitulino     }
129d220a6dfSLuiz Capitulino 
130d220a6dfSLuiz Capitulino     if (WEXITSTATUS(status)) {
13177dbc81bSMarkus Armbruster         error_setg(errp, "child process has failed to shutdown");
132d220a6dfSLuiz Capitulino         return;
133d220a6dfSLuiz Capitulino     }
134d220a6dfSLuiz Capitulino 
135085d8134SPeter Maydell     /* succeeded */
136c216e5adSMichael Roth }
137c216e5adSMichael Roth 
1386912e6a9SLei Li int64_t qmp_guest_get_time(Error **errp)
1396912e6a9SLei Li {
1406912e6a9SLei Li    int ret;
1416912e6a9SLei Li    qemu_timeval tq;
1426912e6a9SLei Li 
1436912e6a9SLei Li    ret = qemu_gettimeofday(&tq);
1446912e6a9SLei Li    if (ret < 0) {
1456912e6a9SLei Li        error_setg_errno(errp, errno, "Failed to get time");
1466912e6a9SLei Li        return -1;
1476912e6a9SLei Li    }
1486912e6a9SLei Li 
1499be38598SEduardo Habkost    return tq.tv_sec * 1000000000LL + tq.tv_usec * 1000;
1506912e6a9SLei Li }
1516912e6a9SLei Li 
1522c958923SMichal Privoznik void qmp_guest_set_time(bool has_time, int64_t time_ns, Error **errp)
153a1bca57fSLei Li {
154a1bca57fSLei Li     int ret;
155a1bca57fSLei Li     int status;
156a1bca57fSLei Li     pid_t pid;
157a1bca57fSLei Li     Error *local_err = NULL;
158a1bca57fSLei Li     struct timeval tv;
1595c6096e5SCornelia Huck     static const char hwclock_path[] = "/sbin/hwclock";
1605c6096e5SCornelia Huck     static int hwclock_available = -1;
1615c6096e5SCornelia Huck 
1625c6096e5SCornelia Huck     if (hwclock_available < 0) {
1635c6096e5SCornelia Huck         hwclock_available = (access(hwclock_path, X_OK) == 0);
1645c6096e5SCornelia Huck     }
1655c6096e5SCornelia Huck 
1665c6096e5SCornelia Huck     if (!hwclock_available) {
1675c6096e5SCornelia Huck         error_setg(errp, QERR_UNSUPPORTED);
1685c6096e5SCornelia Huck         return;
1695c6096e5SCornelia Huck     }
170a1bca57fSLei Li 
1712c958923SMichal Privoznik     /* If user has passed a time, validate and set it. */
1722c958923SMichal Privoznik     if (has_time) {
17300d2f370SMarc-André Lureau         GDate date = { 0, };
17400d2f370SMarc-André Lureau 
175a1bca57fSLei Li         /* year-2038 will overflow in case time_t is 32bit */
176a1bca57fSLei Li         if (time_ns / 1000000000 != (time_t)(time_ns / 1000000000)) {
177a1bca57fSLei Li             error_setg(errp, "Time %" PRId64 " is too large", time_ns);
178a1bca57fSLei Li             return;
179a1bca57fSLei Li         }
180a1bca57fSLei Li 
181a1bca57fSLei Li         tv.tv_sec = time_ns / 1000000000;
182a1bca57fSLei Li         tv.tv_usec = (time_ns % 1000000000) / 1000;
18300d2f370SMarc-André Lureau         g_date_set_time_t(&date, tv.tv_sec);
18400d2f370SMarc-André Lureau         if (date.year < 1970 || date.year >= 2070) {
18500d2f370SMarc-André Lureau             error_setg_errno(errp, errno, "Invalid time");
18600d2f370SMarc-André Lureau             return;
18700d2f370SMarc-André Lureau         }
188a1bca57fSLei Li 
189a1bca57fSLei Li         ret = settimeofday(&tv, NULL);
190a1bca57fSLei Li         if (ret < 0) {
191a1bca57fSLei Li             error_setg_errno(errp, errno, "Failed to set time to guest");
192a1bca57fSLei Li             return;
193a1bca57fSLei Li         }
1942c958923SMichal Privoznik     }
195a1bca57fSLei Li 
1962c958923SMichal Privoznik     /* Now, if user has passed a time to set and the system time is set, we
1972c958923SMichal Privoznik      * just need to synchronize the hardware clock. However, if no time was
1982c958923SMichal Privoznik      * passed, user is requesting the opposite: set the system time from the
1991634df56SAmos Kong      * hardware clock (RTC). */
200a1bca57fSLei Li     pid = fork();
201a1bca57fSLei Li     if (pid == 0) {
202a1bca57fSLei Li         setsid();
203a1bca57fSLei Li         reopen_fd_to_null(0);
204a1bca57fSLei Li         reopen_fd_to_null(1);
205a1bca57fSLei Li         reopen_fd_to_null(2);
206a1bca57fSLei Li 
2072c958923SMichal Privoznik         /* Use '/sbin/hwclock -w' to set RTC from the system time,
2082c958923SMichal Privoznik          * or '/sbin/hwclock -s' to set the system time from RTC. */
2095c6096e5SCornelia Huck         execle(hwclock_path, "hwclock", has_time ? "-w" : "-s",
2102c958923SMichal Privoznik                NULL, environ);
211a1bca57fSLei Li         _exit(EXIT_FAILURE);
212a1bca57fSLei Li     } else if (pid < 0) {
213a1bca57fSLei Li         error_setg_errno(errp, errno, "failed to create child process");
214a1bca57fSLei Li         return;
215a1bca57fSLei Li     }
216a1bca57fSLei Li 
217a1bca57fSLei Li     ga_wait_child(pid, &status, &local_err);
21884d18f06SMarkus Armbruster     if (local_err) {
219a1bca57fSLei Li         error_propagate(errp, local_err);
220a1bca57fSLei Li         return;
221a1bca57fSLei Li     }
222a1bca57fSLei Li 
223a1bca57fSLei Li     if (!WIFEXITED(status)) {
224a1bca57fSLei Li         error_setg(errp, "child process has terminated abnormally");
225a1bca57fSLei Li         return;
226a1bca57fSLei Li     }
227a1bca57fSLei Li 
228a1bca57fSLei Li     if (WEXITSTATUS(status)) {
229a1bca57fSLei Li         error_setg(errp, "hwclock failed to set hardware clock to system time");
230a1bca57fSLei Li         return;
231a1bca57fSLei Li     }
232a1bca57fSLei Li }
233a1bca57fSLei Li 
234895b00f6SMarc-André Lureau typedef enum {
235895b00f6SMarc-André Lureau     RW_STATE_NEW,
236895b00f6SMarc-André Lureau     RW_STATE_READING,
237895b00f6SMarc-André Lureau     RW_STATE_WRITING,
238895b00f6SMarc-André Lureau } RwState;
239895b00f6SMarc-André Lureau 
240c216e5adSMichael Roth typedef struct GuestFileHandle {
241c216e5adSMichael Roth     uint64_t id;
242c216e5adSMichael Roth     FILE *fh;
243895b00f6SMarc-André Lureau     RwState state;
244c216e5adSMichael Roth     QTAILQ_ENTRY(GuestFileHandle) next;
245c216e5adSMichael Roth } GuestFileHandle;
246c216e5adSMichael Roth 
247c216e5adSMichael Roth static struct {
248c216e5adSMichael Roth     QTAILQ_HEAD(, GuestFileHandle) filehandles;
249b4fe97c8SDenis V. Lunev } guest_file_state = {
250b4fe97c8SDenis V. Lunev     .filehandles = QTAILQ_HEAD_INITIALIZER(guest_file_state.filehandles),
251b4fe97c8SDenis V. Lunev };
252c216e5adSMichael Roth 
25339097dafSMichael Roth static int64_t guest_file_handle_add(FILE *fh, Error **errp)
254c216e5adSMichael Roth {
255c216e5adSMichael Roth     GuestFileHandle *gfh;
25639097dafSMichael Roth     int64_t handle;
25739097dafSMichael Roth 
25839097dafSMichael Roth     handle = ga_get_fd_handle(ga_state, errp);
259a903f40cSMarkus Armbruster     if (handle < 0) {
260a903f40cSMarkus Armbruster         return -1;
26139097dafSMichael Roth     }
262c216e5adSMichael Roth 
263f3a06403SMarkus Armbruster     gfh = g_new0(GuestFileHandle, 1);
26439097dafSMichael Roth     gfh->id = handle;
265c216e5adSMichael Roth     gfh->fh = fh;
266c216e5adSMichael Roth     QTAILQ_INSERT_TAIL(&guest_file_state.filehandles, gfh, next);
26739097dafSMichael Roth 
26839097dafSMichael Roth     return handle;
269c216e5adSMichael Roth }
270c216e5adSMichael Roth 
27177dbc81bSMarkus Armbruster static GuestFileHandle *guest_file_handle_find(int64_t id, Error **errp)
272c216e5adSMichael Roth {
273c216e5adSMichael Roth     GuestFileHandle *gfh;
274c216e5adSMichael Roth 
275c216e5adSMichael Roth     QTAILQ_FOREACH(gfh, &guest_file_state.filehandles, next)
276c216e5adSMichael Roth     {
277c216e5adSMichael Roth         if (gfh->id == id) {
278c216e5adSMichael Roth             return gfh;
279c216e5adSMichael Roth         }
280c216e5adSMichael Roth     }
281c216e5adSMichael Roth 
28277dbc81bSMarkus Armbruster     error_setg(errp, "handle '%" PRId64 "' has not been found", id);
283c216e5adSMichael Roth     return NULL;
284c216e5adSMichael Roth }
285c216e5adSMichael Roth 
286c689b4f1SLaszlo Ersek typedef const char * const ccpc;
287c689b4f1SLaszlo Ersek 
2888fe6bbcaSLaszlo Ersek #ifndef O_BINARY
2898fe6bbcaSLaszlo Ersek #define O_BINARY 0
2908fe6bbcaSLaszlo Ersek #endif
2918fe6bbcaSLaszlo Ersek 
292c689b4f1SLaszlo Ersek /* http://pubs.opengroup.org/onlinepubs/9699919799/functions/fopen.html */
293c689b4f1SLaszlo Ersek static const struct {
294c689b4f1SLaszlo Ersek     ccpc *forms;
295c689b4f1SLaszlo Ersek     int oflag_base;
296c689b4f1SLaszlo Ersek } guest_file_open_modes[] = {
2978fe6bbcaSLaszlo Ersek     { (ccpc[]){ "r",          NULL }, O_RDONLY                                 },
2988fe6bbcaSLaszlo Ersek     { (ccpc[]){ "rb",         NULL }, O_RDONLY                      | O_BINARY },
2998fe6bbcaSLaszlo Ersek     { (ccpc[]){ "w",          NULL }, O_WRONLY | O_CREAT | O_TRUNC             },
3008fe6bbcaSLaszlo Ersek     { (ccpc[]){ "wb",         NULL }, O_WRONLY | O_CREAT | O_TRUNC  | O_BINARY },
3018fe6bbcaSLaszlo Ersek     { (ccpc[]){ "a",          NULL }, O_WRONLY | O_CREAT | O_APPEND            },
3028fe6bbcaSLaszlo Ersek     { (ccpc[]){ "ab",         NULL }, O_WRONLY | O_CREAT | O_APPEND | O_BINARY },
3038fe6bbcaSLaszlo Ersek     { (ccpc[]){ "r+",         NULL }, O_RDWR                                   },
3048fe6bbcaSLaszlo Ersek     { (ccpc[]){ "rb+", "r+b", NULL }, O_RDWR                        | O_BINARY },
3058fe6bbcaSLaszlo Ersek     { (ccpc[]){ "w+",         NULL }, O_RDWR   | O_CREAT | O_TRUNC             },
3068fe6bbcaSLaszlo Ersek     { (ccpc[]){ "wb+", "w+b", NULL }, O_RDWR   | O_CREAT | O_TRUNC  | O_BINARY },
3078fe6bbcaSLaszlo Ersek     { (ccpc[]){ "a+",         NULL }, O_RDWR   | O_CREAT | O_APPEND            },
3088fe6bbcaSLaszlo Ersek     { (ccpc[]){ "ab+", "a+b", NULL }, O_RDWR   | O_CREAT | O_APPEND | O_BINARY }
309c689b4f1SLaszlo Ersek };
310c689b4f1SLaszlo Ersek 
311c689b4f1SLaszlo Ersek static int
31277dbc81bSMarkus Armbruster find_open_flag(const char *mode_str, Error **errp)
313c689b4f1SLaszlo Ersek {
314c689b4f1SLaszlo Ersek     unsigned mode;
315c689b4f1SLaszlo Ersek 
316c689b4f1SLaszlo Ersek     for (mode = 0; mode < ARRAY_SIZE(guest_file_open_modes); ++mode) {
317c689b4f1SLaszlo Ersek         ccpc *form;
318c689b4f1SLaszlo Ersek 
319c689b4f1SLaszlo Ersek         form = guest_file_open_modes[mode].forms;
320c689b4f1SLaszlo Ersek         while (*form != NULL && strcmp(*form, mode_str) != 0) {
321c689b4f1SLaszlo Ersek             ++form;
322c689b4f1SLaszlo Ersek         }
323c689b4f1SLaszlo Ersek         if (*form != NULL) {
324c689b4f1SLaszlo Ersek             break;
325c689b4f1SLaszlo Ersek         }
326c689b4f1SLaszlo Ersek     }
327c689b4f1SLaszlo Ersek 
328c689b4f1SLaszlo Ersek     if (mode == ARRAY_SIZE(guest_file_open_modes)) {
32977dbc81bSMarkus Armbruster         error_setg(errp, "invalid file open mode '%s'", mode_str);
330c689b4f1SLaszlo Ersek         return -1;
331c689b4f1SLaszlo Ersek     }
332c689b4f1SLaszlo Ersek     return guest_file_open_modes[mode].oflag_base | O_NOCTTY | O_NONBLOCK;
333c689b4f1SLaszlo Ersek }
334c689b4f1SLaszlo Ersek 
335c689b4f1SLaszlo Ersek #define DEFAULT_NEW_FILE_MODE (S_IRUSR | S_IWUSR | \
336c689b4f1SLaszlo Ersek                                S_IRGRP | S_IWGRP | \
337c689b4f1SLaszlo Ersek                                S_IROTH | S_IWOTH)
338c689b4f1SLaszlo Ersek 
339c689b4f1SLaszlo Ersek static FILE *
34077dbc81bSMarkus Armbruster safe_open_or_create(const char *path, const char *mode, Error **errp)
341c689b4f1SLaszlo Ersek {
342c689b4f1SLaszlo Ersek     Error *local_err = NULL;
343c689b4f1SLaszlo Ersek     int oflag;
344c689b4f1SLaszlo Ersek 
345c689b4f1SLaszlo Ersek     oflag = find_open_flag(mode, &local_err);
346c689b4f1SLaszlo Ersek     if (local_err == NULL) {
347c689b4f1SLaszlo Ersek         int fd;
348c689b4f1SLaszlo Ersek 
349c689b4f1SLaszlo Ersek         /* If the caller wants / allows creation of a new file, we implement it
350c689b4f1SLaszlo Ersek          * with a two step process: open() + (open() / fchmod()).
351c689b4f1SLaszlo Ersek          *
352c689b4f1SLaszlo Ersek          * First we insist on creating the file exclusively as a new file. If
353c689b4f1SLaszlo Ersek          * that succeeds, we're free to set any file-mode bits on it. (The
354c689b4f1SLaszlo Ersek          * motivation is that we want to set those file-mode bits independently
355c689b4f1SLaszlo Ersek          * of the current umask.)
356c689b4f1SLaszlo Ersek          *
357c689b4f1SLaszlo Ersek          * If the exclusive creation fails because the file already exists
358c689b4f1SLaszlo Ersek          * (EEXIST is not possible for any other reason), we just attempt to
359c689b4f1SLaszlo Ersek          * open the file, but in this case we won't be allowed to change the
360c689b4f1SLaszlo Ersek          * file-mode bits on the preexistent file.
361c689b4f1SLaszlo Ersek          *
362c689b4f1SLaszlo Ersek          * The pathname should never disappear between the two open()s in
363c689b4f1SLaszlo Ersek          * practice. If it happens, then someone very likely tried to race us.
364c689b4f1SLaszlo Ersek          * In this case just go ahead and report the ENOENT from the second
365c689b4f1SLaszlo Ersek          * open() to the caller.
366c689b4f1SLaszlo Ersek          *
367c689b4f1SLaszlo Ersek          * If the caller wants to open a preexistent file, then the first
368c689b4f1SLaszlo Ersek          * open() is decisive and its third argument is ignored, and the second
369c689b4f1SLaszlo Ersek          * open() and the fchmod() are never called.
370c689b4f1SLaszlo Ersek          */
371c689b4f1SLaszlo Ersek         fd = open(path, oflag | ((oflag & O_CREAT) ? O_EXCL : 0), 0);
372c689b4f1SLaszlo Ersek         if (fd == -1 && errno == EEXIST) {
373c689b4f1SLaszlo Ersek             oflag &= ~(unsigned)O_CREAT;
374c689b4f1SLaszlo Ersek             fd = open(path, oflag);
375c689b4f1SLaszlo Ersek         }
376c689b4f1SLaszlo Ersek 
377c689b4f1SLaszlo Ersek         if (fd == -1) {
378c689b4f1SLaszlo Ersek             error_setg_errno(&local_err, errno, "failed to open file '%s' "
379c689b4f1SLaszlo Ersek                              "(mode: '%s')", path, mode);
380c689b4f1SLaszlo Ersek         } else {
381c689b4f1SLaszlo Ersek             qemu_set_cloexec(fd);
382c689b4f1SLaszlo Ersek 
383c689b4f1SLaszlo Ersek             if ((oflag & O_CREAT) && fchmod(fd, DEFAULT_NEW_FILE_MODE) == -1) {
384c689b4f1SLaszlo Ersek                 error_setg_errno(&local_err, errno, "failed to set permission "
385c689b4f1SLaszlo Ersek                                  "0%03o on new file '%s' (mode: '%s')",
386c689b4f1SLaszlo Ersek                                  (unsigned)DEFAULT_NEW_FILE_MODE, path, mode);
387c689b4f1SLaszlo Ersek             } else {
388c689b4f1SLaszlo Ersek                 FILE *f;
389c689b4f1SLaszlo Ersek 
390c689b4f1SLaszlo Ersek                 f = fdopen(fd, mode);
391c689b4f1SLaszlo Ersek                 if (f == NULL) {
392c689b4f1SLaszlo Ersek                     error_setg_errno(&local_err, errno, "failed to associate "
393c689b4f1SLaszlo Ersek                                      "stdio stream with file descriptor %d, "
394c689b4f1SLaszlo Ersek                                      "file '%s' (mode: '%s')", fd, path, mode);
395c689b4f1SLaszlo Ersek                 } else {
396c689b4f1SLaszlo Ersek                     return f;
397c689b4f1SLaszlo Ersek                 }
398c689b4f1SLaszlo Ersek             }
399c689b4f1SLaszlo Ersek 
400c689b4f1SLaszlo Ersek             close(fd);
4012b720018SLaszlo Ersek             if (oflag & O_CREAT) {
4022b720018SLaszlo Ersek                 unlink(path);
4032b720018SLaszlo Ersek             }
404c689b4f1SLaszlo Ersek         }
405c689b4f1SLaszlo Ersek     }
406c689b4f1SLaszlo Ersek 
40777dbc81bSMarkus Armbruster     error_propagate(errp, local_err);
408c689b4f1SLaszlo Ersek     return NULL;
409c689b4f1SLaszlo Ersek }
410c689b4f1SLaszlo Ersek 
41177dbc81bSMarkus Armbruster int64_t qmp_guest_file_open(const char *path, bool has_mode, const char *mode,
41277dbc81bSMarkus Armbruster                             Error **errp)
413c216e5adSMichael Roth {
414c216e5adSMichael Roth     FILE *fh;
415c689b4f1SLaszlo Ersek     Error *local_err = NULL;
41685b6f6f5SSimon Zolin     int64_t handle;
417c216e5adSMichael Roth 
418c216e5adSMichael Roth     if (!has_mode) {
419c216e5adSMichael Roth         mode = "r";
420c216e5adSMichael Roth     }
421c216e5adSMichael Roth     slog("guest-file-open called, filepath: %s, mode: %s", path, mode);
422c689b4f1SLaszlo Ersek     fh = safe_open_or_create(path, mode, &local_err);
423c689b4f1SLaszlo Ersek     if (local_err != NULL) {
42477dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
425c216e5adSMichael Roth         return -1;
426c216e5adSMichael Roth     }
427c216e5adSMichael Roth 
428c216e5adSMichael Roth     /* set fd non-blocking to avoid common use cases (like reading from a
429c216e5adSMichael Roth      * named pipe) from hanging the agent
430c216e5adSMichael Roth      */
43112505396SDenis V. Lunev     qemu_set_nonblock(fileno(fh));
432c216e5adSMichael Roth 
43377dbc81bSMarkus Armbruster     handle = guest_file_handle_add(fh, errp);
434a903f40cSMarkus Armbruster     if (handle < 0) {
43539097dafSMichael Roth         fclose(fh);
43639097dafSMichael Roth         return -1;
43739097dafSMichael Roth     }
43839097dafSMichael Roth 
439d607a523SStefan Weil     slog("guest-file-open, handle: %" PRId64, handle);
44039097dafSMichael Roth     return handle;
441c216e5adSMichael Roth }
442c216e5adSMichael Roth 
44377dbc81bSMarkus Armbruster void qmp_guest_file_close(int64_t handle, Error **errp)
444c216e5adSMichael Roth {
44577dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
446c216e5adSMichael Roth     int ret;
447c216e5adSMichael Roth 
448d607a523SStefan Weil     slog("guest-file-close called, handle: %" PRId64, handle);
449c216e5adSMichael Roth     if (!gfh) {
450c216e5adSMichael Roth         return;
451c216e5adSMichael Roth     }
452c216e5adSMichael Roth 
453c216e5adSMichael Roth     ret = fclose(gfh->fh);
4543ac4b7c5SLuiz Capitulino     if (ret == EOF) {
45577dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to close handle");
456c216e5adSMichael Roth         return;
457c216e5adSMichael Roth     }
458c216e5adSMichael Roth 
459c216e5adSMichael Roth     QTAILQ_REMOVE(&guest_file_state.filehandles, gfh, next);
460c216e5adSMichael Roth     g_free(gfh);
461c216e5adSMichael Roth }
462c216e5adSMichael Roth 
463c216e5adSMichael Roth struct GuestFileRead *qmp_guest_file_read(int64_t handle, bool has_count,
46477dbc81bSMarkus Armbruster                                           int64_t count, Error **errp)
465c216e5adSMichael Roth {
46677dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
467c216e5adSMichael Roth     GuestFileRead *read_data = NULL;
468c216e5adSMichael Roth     guchar *buf;
469c216e5adSMichael Roth     FILE *fh;
470c216e5adSMichael Roth     size_t read_count;
471c216e5adSMichael Roth 
472c216e5adSMichael Roth     if (!gfh) {
473c216e5adSMichael Roth         return NULL;
474c216e5adSMichael Roth     }
475c216e5adSMichael Roth 
476c216e5adSMichael Roth     if (!has_count) {
477c216e5adSMichael Roth         count = QGA_READ_COUNT_DEFAULT;
478141b1974SPrasad J Pandit     } else if (count < 0 || count >= UINT32_MAX) {
47977dbc81bSMarkus Armbruster         error_setg(errp, "value '%" PRId64 "' is invalid for argument count",
480db3edb66SLuiz Capitulino                    count);
481c216e5adSMichael Roth         return NULL;
482c216e5adSMichael Roth     }
483c216e5adSMichael Roth 
484c216e5adSMichael Roth     fh = gfh->fh;
485895b00f6SMarc-André Lureau 
486895b00f6SMarc-André Lureau     /* explicitly flush when switching from writing to reading */
487895b00f6SMarc-André Lureau     if (gfh->state == RW_STATE_WRITING) {
488895b00f6SMarc-André Lureau         int ret = fflush(fh);
489895b00f6SMarc-André Lureau         if (ret == EOF) {
490895b00f6SMarc-André Lureau             error_setg_errno(errp, errno, "failed to flush file");
491895b00f6SMarc-André Lureau             return NULL;
492895b00f6SMarc-André Lureau         }
493895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
494895b00f6SMarc-André Lureau     }
495895b00f6SMarc-André Lureau 
496c216e5adSMichael Roth     buf = g_malloc0(count+1);
497c216e5adSMichael Roth     read_count = fread(buf, 1, count, fh);
498c216e5adSMichael Roth     if (ferror(fh)) {
49977dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to read file");
500d607a523SStefan Weil         slog("guest-file-read failed, handle: %" PRId64, handle);
501c216e5adSMichael Roth     } else {
502c216e5adSMichael Roth         buf[read_count] = 0;
503f3a06403SMarkus Armbruster         read_data = g_new0(GuestFileRead, 1);
504c216e5adSMichael Roth         read_data->count = read_count;
505c216e5adSMichael Roth         read_data->eof = feof(fh);
506c216e5adSMichael Roth         if (read_count) {
507c216e5adSMichael Roth             read_data->buf_b64 = g_base64_encode(buf, read_count);
508c216e5adSMichael Roth         }
509895b00f6SMarc-André Lureau         gfh->state = RW_STATE_READING;
510c216e5adSMichael Roth     }
511c216e5adSMichael Roth     g_free(buf);
512c216e5adSMichael Roth     clearerr(fh);
513c216e5adSMichael Roth 
514c216e5adSMichael Roth     return read_data;
515c216e5adSMichael Roth }
516c216e5adSMichael Roth 
517c216e5adSMichael Roth GuestFileWrite *qmp_guest_file_write(int64_t handle, const char *buf_b64,
51877dbc81bSMarkus Armbruster                                      bool has_count, int64_t count,
51977dbc81bSMarkus Armbruster                                      Error **errp)
520c216e5adSMichael Roth {
521c216e5adSMichael Roth     GuestFileWrite *write_data = NULL;
522c216e5adSMichael Roth     guchar *buf;
523c216e5adSMichael Roth     gsize buf_len;
524c216e5adSMichael Roth     int write_count;
52577dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
526c216e5adSMichael Roth     FILE *fh;
527c216e5adSMichael Roth 
528c216e5adSMichael Roth     if (!gfh) {
529c216e5adSMichael Roth         return NULL;
530c216e5adSMichael Roth     }
531c216e5adSMichael Roth 
532c216e5adSMichael Roth     fh = gfh->fh;
533895b00f6SMarc-André Lureau 
534895b00f6SMarc-André Lureau     if (gfh->state == RW_STATE_READING) {
535895b00f6SMarc-André Lureau         int ret = fseek(fh, 0, SEEK_CUR);
536895b00f6SMarc-André Lureau         if (ret == -1) {
537895b00f6SMarc-André Lureau             error_setg_errno(errp, errno, "failed to seek file");
538895b00f6SMarc-André Lureau             return NULL;
539895b00f6SMarc-André Lureau         }
540895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
541895b00f6SMarc-André Lureau     }
542895b00f6SMarc-André Lureau 
543920639caSDaniel P. Berrange     buf = qbase64_decode(buf_b64, -1, &buf_len, errp);
544920639caSDaniel P. Berrange     if (!buf) {
545920639caSDaniel P. Berrange         return NULL;
546920639caSDaniel P. Berrange     }
547c216e5adSMichael Roth 
548c216e5adSMichael Roth     if (!has_count) {
549c216e5adSMichael Roth         count = buf_len;
550c216e5adSMichael Roth     } else if (count < 0 || count > buf_len) {
55177dbc81bSMarkus Armbruster         error_setg(errp, "value '%" PRId64 "' is invalid for argument count",
552db3edb66SLuiz Capitulino                    count);
553c216e5adSMichael Roth         g_free(buf);
554c216e5adSMichael Roth         return NULL;
555c216e5adSMichael Roth     }
556c216e5adSMichael Roth 
557c216e5adSMichael Roth     write_count = fwrite(buf, 1, count, fh);
558c216e5adSMichael Roth     if (ferror(fh)) {
55977dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to write to file");
560d607a523SStefan Weil         slog("guest-file-write failed, handle: %" PRId64, handle);
561c216e5adSMichael Roth     } else {
562f3a06403SMarkus Armbruster         write_data = g_new0(GuestFileWrite, 1);
563c216e5adSMichael Roth         write_data->count = write_count;
564c216e5adSMichael Roth         write_data->eof = feof(fh);
565895b00f6SMarc-André Lureau         gfh->state = RW_STATE_WRITING;
566c216e5adSMichael Roth     }
567c216e5adSMichael Roth     g_free(buf);
568c216e5adSMichael Roth     clearerr(fh);
569c216e5adSMichael Roth 
570c216e5adSMichael Roth     return write_data;
571c216e5adSMichael Roth }
572c216e5adSMichael Roth 
573c216e5adSMichael Roth struct GuestFileSeek *qmp_guest_file_seek(int64_t handle, int64_t offset,
5740b4b4938SEric Blake                                           GuestFileWhence *whence_code,
5750b4b4938SEric Blake                                           Error **errp)
576c216e5adSMichael Roth {
57777dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
578c216e5adSMichael Roth     GuestFileSeek *seek_data = NULL;
579c216e5adSMichael Roth     FILE *fh;
580c216e5adSMichael Roth     int ret;
5810a982b1bSEric Blake     int whence;
5820b4b4938SEric Blake     Error *err = NULL;
583c216e5adSMichael Roth 
584c216e5adSMichael Roth     if (!gfh) {
585c216e5adSMichael Roth         return NULL;
586c216e5adSMichael Roth     }
587c216e5adSMichael Roth 
5880a982b1bSEric Blake     /* We stupidly exposed 'whence':'int' in our qapi */
5890b4b4938SEric Blake     whence = ga_parse_whence(whence_code, &err);
5900b4b4938SEric Blake     if (err) {
5910b4b4938SEric Blake         error_propagate(errp, err);
5920a982b1bSEric Blake         return NULL;
5930a982b1bSEric Blake     }
5940a982b1bSEric Blake 
595c216e5adSMichael Roth     fh = gfh->fh;
596c216e5adSMichael Roth     ret = fseek(fh, offset, whence);
597c216e5adSMichael Roth     if (ret == -1) {
59877dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to seek file");
599895b00f6SMarc-André Lureau         if (errno == ESPIPE) {
600895b00f6SMarc-André Lureau             /* file is non-seekable, stdio shouldn't be buffering anyways */
601895b00f6SMarc-André Lureau             gfh->state = RW_STATE_NEW;
602895b00f6SMarc-André Lureau         }
603c216e5adSMichael Roth     } else {
60410b7c5ddSMarkus Armbruster         seek_data = g_new0(GuestFileSeek, 1);
605c216e5adSMichael Roth         seek_data->position = ftell(fh);
606c216e5adSMichael Roth         seek_data->eof = feof(fh);
607895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
608c216e5adSMichael Roth     }
609c216e5adSMichael Roth     clearerr(fh);
610c216e5adSMichael Roth 
611c216e5adSMichael Roth     return seek_data;
612c216e5adSMichael Roth }
613c216e5adSMichael Roth 
61477dbc81bSMarkus Armbruster void qmp_guest_file_flush(int64_t handle, Error **errp)
615c216e5adSMichael Roth {
61677dbc81bSMarkus Armbruster     GuestFileHandle *gfh = guest_file_handle_find(handle, errp);
617c216e5adSMichael Roth     FILE *fh;
618c216e5adSMichael Roth     int ret;
619c216e5adSMichael Roth 
620c216e5adSMichael Roth     if (!gfh) {
621c216e5adSMichael Roth         return;
622c216e5adSMichael Roth     }
623c216e5adSMichael Roth 
624c216e5adSMichael Roth     fh = gfh->fh;
625c216e5adSMichael Roth     ret = fflush(fh);
626c216e5adSMichael Roth     if (ret == EOF) {
62777dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to flush file");
628895b00f6SMarc-André Lureau     } else {
629895b00f6SMarc-André Lureau         gfh->state = RW_STATE_NEW;
630c216e5adSMichael Roth     }
631c216e5adSMichael Roth }
632c216e5adSMichael Roth 
633e72c3f2eSMichael Roth /* linux-specific implementations. avoid this if at all possible. */
634e72c3f2eSMichael Roth #if defined(__linux__)
635e72c3f2eSMichael Roth 
636eab5fd59SPaolo Bonzini #if defined(CONFIG_FSFREEZE) || defined(CONFIG_FSTRIM)
637af02203fSPaolo Bonzini typedef struct FsMount {
638c216e5adSMichael Roth     char *dirname;
639c216e5adSMichael Roth     char *devtype;
64046d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
641af02203fSPaolo Bonzini     QTAILQ_ENTRY(FsMount) next;
642af02203fSPaolo Bonzini } FsMount;
643c216e5adSMichael Roth 
644e5d9adbdSTomoki Sekiyama typedef QTAILQ_HEAD(FsMountList, FsMount) FsMountList;
6459e8aded4SMichael Roth 
646af02203fSPaolo Bonzini static void free_fs_mount_list(FsMountList *mounts)
647c216e5adSMichael Roth {
648af02203fSPaolo Bonzini      FsMount *mount, *temp;
649c216e5adSMichael Roth 
6509e8aded4SMichael Roth      if (!mounts) {
6519e8aded4SMichael Roth          return;
6529e8aded4SMichael Roth      }
6539e8aded4SMichael Roth 
6549e8aded4SMichael Roth      QTAILQ_FOREACH_SAFE(mount, mounts, next, temp) {
6559e8aded4SMichael Roth          QTAILQ_REMOVE(mounts, mount, next);
656c216e5adSMichael Roth          g_free(mount->dirname);
657c216e5adSMichael Roth          g_free(mount->devtype);
658c216e5adSMichael Roth          g_free(mount);
659c216e5adSMichael Roth      }
6609e8aded4SMichael Roth }
6619e8aded4SMichael Roth 
66246d4c572STomoki Sekiyama static int dev_major_minor(const char *devpath,
66346d4c572STomoki Sekiyama                            unsigned int *devmajor, unsigned int *devminor)
66446d4c572STomoki Sekiyama {
66546d4c572STomoki Sekiyama     struct stat st;
66646d4c572STomoki Sekiyama 
66746d4c572STomoki Sekiyama     *devmajor = 0;
66846d4c572STomoki Sekiyama     *devminor = 0;
66946d4c572STomoki Sekiyama 
67046d4c572STomoki Sekiyama     if (stat(devpath, &st) < 0) {
67146d4c572STomoki Sekiyama         slog("failed to stat device file '%s': %s", devpath, strerror(errno));
67246d4c572STomoki Sekiyama         return -1;
67346d4c572STomoki Sekiyama     }
67446d4c572STomoki Sekiyama     if (S_ISDIR(st.st_mode)) {
67546d4c572STomoki Sekiyama         /* It is bind mount */
67646d4c572STomoki Sekiyama         return -2;
67746d4c572STomoki Sekiyama     }
67846d4c572STomoki Sekiyama     if (S_ISBLK(st.st_mode)) {
67946d4c572STomoki Sekiyama         *devmajor = major(st.st_rdev);
68046d4c572STomoki Sekiyama         *devminor = minor(st.st_rdev);
68146d4c572STomoki Sekiyama         return 0;
68246d4c572STomoki Sekiyama     }
68346d4c572STomoki Sekiyama     return -1;
68446d4c572STomoki Sekiyama }
68546d4c572STomoki Sekiyama 
6869e8aded4SMichael Roth /*
6879e8aded4SMichael Roth  * Walk the mount table and build a list of local file systems
6889e8aded4SMichael Roth  */
68946d4c572STomoki Sekiyama static void build_fs_mount_list_from_mtab(FsMountList *mounts, Error **errp)
6909e8aded4SMichael Roth {
6919e8aded4SMichael Roth     struct mntent *ment;
692af02203fSPaolo Bonzini     FsMount *mount;
6939e2fa418SMichael Roth     char const *mtab = "/proc/self/mounts";
6949e8aded4SMichael Roth     FILE *fp;
69546d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
696c216e5adSMichael Roth 
697c216e5adSMichael Roth     fp = setmntent(mtab, "r");
698c216e5adSMichael Roth     if (!fp) {
69977dbc81bSMarkus Armbruster         error_setg(errp, "failed to open mtab file: '%s'", mtab);
700261551d1SLuiz Capitulino         return;
701c216e5adSMichael Roth     }
702c216e5adSMichael Roth 
703c216e5adSMichael Roth     while ((ment = getmntent(fp))) {
704c216e5adSMichael Roth         /*
705c216e5adSMichael Roth          * An entry which device name doesn't start with a '/' is
706c216e5adSMichael Roth          * either a dummy file system or a network file system.
707c216e5adSMichael Roth          * Add special handling for smbfs and cifs as is done by
708c216e5adSMichael Roth          * coreutils as well.
709c216e5adSMichael Roth          */
710c216e5adSMichael Roth         if ((ment->mnt_fsname[0] != '/') ||
711c216e5adSMichael Roth             (strcmp(ment->mnt_type, "smbfs") == 0) ||
712c216e5adSMichael Roth             (strcmp(ment->mnt_type, "cifs") == 0)) {
713c216e5adSMichael Roth             continue;
714c216e5adSMichael Roth         }
71546d4c572STomoki Sekiyama         if (dev_major_minor(ment->mnt_fsname, &devmajor, &devminor) == -2) {
71646d4c572STomoki Sekiyama             /* Skip bind mounts */
71746d4c572STomoki Sekiyama             continue;
71846d4c572STomoki Sekiyama         }
719c216e5adSMichael Roth 
720f3a06403SMarkus Armbruster         mount = g_new0(FsMount, 1);
721c216e5adSMichael Roth         mount->dirname = g_strdup(ment->mnt_dir);
722c216e5adSMichael Roth         mount->devtype = g_strdup(ment->mnt_type);
72346d4c572STomoki Sekiyama         mount->devmajor = devmajor;
72446d4c572STomoki Sekiyama         mount->devminor = devminor;
725c216e5adSMichael Roth 
7269e8aded4SMichael Roth         QTAILQ_INSERT_TAIL(mounts, mount, next);
727c216e5adSMichael Roth     }
728c216e5adSMichael Roth 
729c216e5adSMichael Roth     endmntent(fp);
730c216e5adSMichael Roth }
73146d4c572STomoki Sekiyama 
73246d4c572STomoki Sekiyama static void decode_mntname(char *name, int len)
73346d4c572STomoki Sekiyama {
73446d4c572STomoki Sekiyama     int i, j = 0;
73546d4c572STomoki Sekiyama     for (i = 0; i <= len; i++) {
73646d4c572STomoki Sekiyama         if (name[i] != '\\') {
73746d4c572STomoki Sekiyama             name[j++] = name[i];
73846d4c572STomoki Sekiyama         } else if (name[i + 1] == '\\') {
73946d4c572STomoki Sekiyama             name[j++] = '\\';
74046d4c572STomoki Sekiyama             i++;
74146d4c572STomoki Sekiyama         } else if (name[i + 1] >= '0' && name[i + 1] <= '3' &&
74246d4c572STomoki Sekiyama                    name[i + 2] >= '0' && name[i + 2] <= '7' &&
74346d4c572STomoki Sekiyama                    name[i + 3] >= '0' && name[i + 3] <= '7') {
74446d4c572STomoki Sekiyama             name[j++] = (name[i + 1] - '0') * 64 +
74546d4c572STomoki Sekiyama                         (name[i + 2] - '0') * 8 +
74646d4c572STomoki Sekiyama                         (name[i + 3] - '0');
74746d4c572STomoki Sekiyama             i += 3;
74846d4c572STomoki Sekiyama         } else {
74946d4c572STomoki Sekiyama             name[j++] = name[i];
75046d4c572STomoki Sekiyama         }
75146d4c572STomoki Sekiyama     }
75246d4c572STomoki Sekiyama }
75346d4c572STomoki Sekiyama 
75446d4c572STomoki Sekiyama static void build_fs_mount_list(FsMountList *mounts, Error **errp)
75546d4c572STomoki Sekiyama {
75646d4c572STomoki Sekiyama     FsMount *mount;
75746d4c572STomoki Sekiyama     char const *mountinfo = "/proc/self/mountinfo";
75846d4c572STomoki Sekiyama     FILE *fp;
75946d4c572STomoki Sekiyama     char *line = NULL, *dash;
76046d4c572STomoki Sekiyama     size_t n;
76146d4c572STomoki Sekiyama     char check;
76246d4c572STomoki Sekiyama     unsigned int devmajor, devminor;
76346d4c572STomoki Sekiyama     int ret, dir_s, dir_e, type_s, type_e, dev_s, dev_e;
76446d4c572STomoki Sekiyama 
76546d4c572STomoki Sekiyama     fp = fopen(mountinfo, "r");
76646d4c572STomoki Sekiyama     if (!fp) {
76746d4c572STomoki Sekiyama         build_fs_mount_list_from_mtab(mounts, errp);
76846d4c572STomoki Sekiyama         return;
76946d4c572STomoki Sekiyama     }
77046d4c572STomoki Sekiyama 
77146d4c572STomoki Sekiyama     while (getline(&line, &n, fp) != -1) {
77246d4c572STomoki Sekiyama         ret = sscanf(line, "%*u %*u %u:%u %*s %n%*s%n%c",
77346d4c572STomoki Sekiyama                      &devmajor, &devminor, &dir_s, &dir_e, &check);
77446d4c572STomoki Sekiyama         if (ret < 3) {
77546d4c572STomoki Sekiyama             continue;
77646d4c572STomoki Sekiyama         }
77746d4c572STomoki Sekiyama         dash = strstr(line + dir_e, " - ");
77846d4c572STomoki Sekiyama         if (!dash) {
77946d4c572STomoki Sekiyama             continue;
78046d4c572STomoki Sekiyama         }
78146d4c572STomoki Sekiyama         ret = sscanf(dash, " - %n%*s%n %n%*s%n%c",
78246d4c572STomoki Sekiyama                      &type_s, &type_e, &dev_s, &dev_e, &check);
78346d4c572STomoki Sekiyama         if (ret < 1) {
78446d4c572STomoki Sekiyama             continue;
78546d4c572STomoki Sekiyama         }
78646d4c572STomoki Sekiyama         line[dir_e] = 0;
78746d4c572STomoki Sekiyama         dash[type_e] = 0;
78846d4c572STomoki Sekiyama         dash[dev_e] = 0;
78946d4c572STomoki Sekiyama         decode_mntname(line + dir_s, dir_e - dir_s);
79046d4c572STomoki Sekiyama         decode_mntname(dash + dev_s, dev_e - dev_s);
79146d4c572STomoki Sekiyama         if (devmajor == 0) {
79246d4c572STomoki Sekiyama             /* btrfs reports major number = 0 */
79346d4c572STomoki Sekiyama             if (strcmp("btrfs", dash + type_s) != 0 ||
79446d4c572STomoki Sekiyama                 dev_major_minor(dash + dev_s, &devmajor, &devminor) < 0) {
79546d4c572STomoki Sekiyama                 continue;
79646d4c572STomoki Sekiyama             }
79746d4c572STomoki Sekiyama         }
79846d4c572STomoki Sekiyama 
799f3a06403SMarkus Armbruster         mount = g_new0(FsMount, 1);
80046d4c572STomoki Sekiyama         mount->dirname = g_strdup(line + dir_s);
80146d4c572STomoki Sekiyama         mount->devtype = g_strdup(dash + type_s);
80246d4c572STomoki Sekiyama         mount->devmajor = devmajor;
80346d4c572STomoki Sekiyama         mount->devminor = devminor;
80446d4c572STomoki Sekiyama 
80546d4c572STomoki Sekiyama         QTAILQ_INSERT_TAIL(mounts, mount, next);
80646d4c572STomoki Sekiyama     }
80746d4c572STomoki Sekiyama     free(line);
80846d4c572STomoki Sekiyama 
80946d4c572STomoki Sekiyama     fclose(fp);
81046d4c572STomoki Sekiyama }
811eab5fd59SPaolo Bonzini #endif
812eab5fd59SPaolo Bonzini 
813eab5fd59SPaolo Bonzini #if defined(CONFIG_FSFREEZE)
814c216e5adSMichael Roth 
81546d4c572STomoki Sekiyama static char *get_pci_driver(char const *syspath, int pathlen, Error **errp)
81646d4c572STomoki Sekiyama {
81746d4c572STomoki Sekiyama     char *path;
81846d4c572STomoki Sekiyama     char *dpath;
81946d4c572STomoki Sekiyama     char *driver = NULL;
82046d4c572STomoki Sekiyama     char buf[PATH_MAX];
82146d4c572STomoki Sekiyama     ssize_t len;
82246d4c572STomoki Sekiyama 
82346d4c572STomoki Sekiyama     path = g_strndup(syspath, pathlen);
82446d4c572STomoki Sekiyama     dpath = g_strdup_printf("%s/driver", path);
82546d4c572STomoki Sekiyama     len = readlink(dpath, buf, sizeof(buf) - 1);
82646d4c572STomoki Sekiyama     if (len != -1) {
82746d4c572STomoki Sekiyama         buf[len] = 0;
8283e015d81SJulia Suvorova         driver = g_path_get_basename(buf);
82946d4c572STomoki Sekiyama     }
83046d4c572STomoki Sekiyama     g_free(dpath);
83146d4c572STomoki Sekiyama     g_free(path);
83246d4c572STomoki Sekiyama     return driver;
83346d4c572STomoki Sekiyama }
83446d4c572STomoki Sekiyama 
83546d4c572STomoki Sekiyama static int compare_uint(const void *_a, const void *_b)
83646d4c572STomoki Sekiyama {
83746d4c572STomoki Sekiyama     unsigned int a = *(unsigned int *)_a;
83846d4c572STomoki Sekiyama     unsigned int b = *(unsigned int *)_b;
83946d4c572STomoki Sekiyama 
84046d4c572STomoki Sekiyama     return a < b ? -1 : a > b ? 1 : 0;
84146d4c572STomoki Sekiyama }
84246d4c572STomoki Sekiyama 
84346d4c572STomoki Sekiyama /* Walk the specified sysfs and build a sorted list of host or ata numbers */
84446d4c572STomoki Sekiyama static int build_hosts(char const *syspath, char const *host, bool ata,
84546d4c572STomoki Sekiyama                        unsigned int *hosts, int hosts_max, Error **errp)
84646d4c572STomoki Sekiyama {
84746d4c572STomoki Sekiyama     char *path;
84846d4c572STomoki Sekiyama     DIR *dir;
84946d4c572STomoki Sekiyama     struct dirent *entry;
85046d4c572STomoki Sekiyama     int i = 0;
85146d4c572STomoki Sekiyama 
85246d4c572STomoki Sekiyama     path = g_strndup(syspath, host - syspath);
85346d4c572STomoki Sekiyama     dir = opendir(path);
85446d4c572STomoki Sekiyama     if (!dir) {
85546d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "opendir(\"%s\")", path);
85646d4c572STomoki Sekiyama         g_free(path);
85746d4c572STomoki Sekiyama         return -1;
85846d4c572STomoki Sekiyama     }
85946d4c572STomoki Sekiyama 
86046d4c572STomoki Sekiyama     while (i < hosts_max) {
86146d4c572STomoki Sekiyama         entry = readdir(dir);
86246d4c572STomoki Sekiyama         if (!entry) {
86346d4c572STomoki Sekiyama             break;
86446d4c572STomoki Sekiyama         }
86546d4c572STomoki Sekiyama         if (ata && sscanf(entry->d_name, "ata%d", hosts + i) == 1) {
86646d4c572STomoki Sekiyama             ++i;
86746d4c572STomoki Sekiyama         } else if (!ata && sscanf(entry->d_name, "host%d", hosts + i) == 1) {
86846d4c572STomoki Sekiyama             ++i;
86946d4c572STomoki Sekiyama         }
87046d4c572STomoki Sekiyama     }
87146d4c572STomoki Sekiyama 
87246d4c572STomoki Sekiyama     qsort(hosts, i, sizeof(hosts[0]), compare_uint);
87346d4c572STomoki Sekiyama 
87446d4c572STomoki Sekiyama     g_free(path);
87546d4c572STomoki Sekiyama     closedir(dir);
87646d4c572STomoki Sekiyama     return i;
87746d4c572STomoki Sekiyama }
87846d4c572STomoki Sekiyama 
87946d4c572STomoki Sekiyama /* Store disk device info specified by @sysfs into @fs */
88046d4c572STomoki Sekiyama static void build_guest_fsinfo_for_real_device(char const *syspath,
88146d4c572STomoki Sekiyama                                                GuestFilesystemInfo *fs,
88246d4c572STomoki Sekiyama                                                Error **errp)
88346d4c572STomoki Sekiyama {
88446d4c572STomoki Sekiyama     unsigned int pci[4], host, hosts[8], tgt[3];
88546d4c572STomoki Sekiyama     int i, nhosts = 0, pcilen;
88646d4c572STomoki Sekiyama     GuestDiskAddress *disk;
88746d4c572STomoki Sekiyama     GuestPCIAddress *pciaddr;
88846d4c572STomoki Sekiyama     GuestDiskAddressList *list = NULL;
88946d4c572STomoki Sekiyama     bool has_ata = false, has_host = false, has_tgt = false;
89046d4c572STomoki Sekiyama     char *p, *q, *driver = NULL;
891b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
892b616105aSTomáš Golembiovský     struct udev *udev = NULL;
893b616105aSTomáš Golembiovský     struct udev_device *udevice = NULL;
894b616105aSTomáš Golembiovský #endif
89546d4c572STomoki Sekiyama 
89646d4c572STomoki Sekiyama     p = strstr(syspath, "/devices/pci");
89746d4c572STomoki Sekiyama     if (!p || sscanf(p + 12, "%*x:%*x/%x:%x:%x.%x%n",
89846d4c572STomoki Sekiyama                      pci, pci + 1, pci + 2, pci + 3, &pcilen) < 4) {
899743c71d0SMarc-André Lureau         g_debug("only pci device is supported: sysfs path '%s'", syspath);
90046d4c572STomoki Sekiyama         return;
90146d4c572STomoki Sekiyama     }
90246d4c572STomoki Sekiyama 
903743c71d0SMarc-André Lureau     p += 12 + pcilen;
904743c71d0SMarc-André Lureau     while (true) {
905743c71d0SMarc-André Lureau         driver = get_pci_driver(syspath, p - syspath, errp);
906743c71d0SMarc-André Lureau         if (driver && (g_str_equal(driver, "ata_piix") ||
907743c71d0SMarc-André Lureau                        g_str_equal(driver, "sym53c8xx") ||
908743c71d0SMarc-André Lureau                        g_str_equal(driver, "virtio-pci") ||
909743c71d0SMarc-André Lureau                        g_str_equal(driver, "ahci"))) {
910743c71d0SMarc-André Lureau             break;
911743c71d0SMarc-André Lureau         }
912743c71d0SMarc-André Lureau 
913bb23a736SMarc-André Lureau         g_free(driver);
914743c71d0SMarc-André Lureau         if (sscanf(p, "/%x:%x:%x.%x%n",
915743c71d0SMarc-André Lureau                           pci, pci + 1, pci + 2, pci + 3, &pcilen) == 4) {
916743c71d0SMarc-André Lureau             p += pcilen;
917743c71d0SMarc-André Lureau             continue;
918743c71d0SMarc-André Lureau         }
919743c71d0SMarc-André Lureau 
920743c71d0SMarc-André Lureau         g_debug("unsupported driver or sysfs path '%s'", syspath);
921743c71d0SMarc-André Lureau         return;
92246d4c572STomoki Sekiyama     }
92346d4c572STomoki Sekiyama 
92446d4c572STomoki Sekiyama     p = strstr(syspath, "/target");
92546d4c572STomoki Sekiyama     if (p && sscanf(p + 7, "%*u:%*u:%*u/%*u:%u:%u:%u",
92646d4c572STomoki Sekiyama                     tgt, tgt + 1, tgt + 2) == 3) {
92746d4c572STomoki Sekiyama         has_tgt = true;
92846d4c572STomoki Sekiyama     }
92946d4c572STomoki Sekiyama 
93046d4c572STomoki Sekiyama     p = strstr(syspath, "/ata");
93146d4c572STomoki Sekiyama     if (p) {
93246d4c572STomoki Sekiyama         q = p + 4;
93346d4c572STomoki Sekiyama         has_ata = true;
93446d4c572STomoki Sekiyama     } else {
93546d4c572STomoki Sekiyama         p = strstr(syspath, "/host");
93646d4c572STomoki Sekiyama         q = p + 5;
93746d4c572STomoki Sekiyama     }
93846d4c572STomoki Sekiyama     if (p && sscanf(q, "%u", &host) == 1) {
93946d4c572STomoki Sekiyama         has_host = true;
94046d4c572STomoki Sekiyama         nhosts = build_hosts(syspath, p, has_ata, hosts,
94101a6df1bSPhilippe Mathieu-Daudé                              ARRAY_SIZE(hosts), errp);
94246d4c572STomoki Sekiyama         if (nhosts < 0) {
94346d4c572STomoki Sekiyama             goto cleanup;
94446d4c572STomoki Sekiyama         }
94546d4c572STomoki Sekiyama     }
94646d4c572STomoki Sekiyama 
94746d4c572STomoki Sekiyama     pciaddr = g_malloc0(sizeof(*pciaddr));
94846d4c572STomoki Sekiyama     pciaddr->domain = pci[0];
94946d4c572STomoki Sekiyama     pciaddr->bus = pci[1];
95046d4c572STomoki Sekiyama     pciaddr->slot = pci[2];
95146d4c572STomoki Sekiyama     pciaddr->function = pci[3];
95246d4c572STomoki Sekiyama 
95346d4c572STomoki Sekiyama     disk = g_malloc0(sizeof(*disk));
95446d4c572STomoki Sekiyama     disk->pci_controller = pciaddr;
95546d4c572STomoki Sekiyama 
95646d4c572STomoki Sekiyama     list = g_malloc0(sizeof(*list));
95746d4c572STomoki Sekiyama     list->value = disk;
95846d4c572STomoki Sekiyama 
959b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
960b616105aSTomáš Golembiovský     udev = udev_new();
961b616105aSTomáš Golembiovský     udevice = udev_device_new_from_syspath(udev, syspath);
962b616105aSTomáš Golembiovský     if (udev == NULL || udevice == NULL) {
963b616105aSTomáš Golembiovský         g_debug("failed to query udev");
964b616105aSTomáš Golembiovský     } else {
9656589ce35STomáš Golembiovský         const char *devnode, *serial;
9666589ce35STomáš Golembiovský         devnode = udev_device_get_devnode(udevice);
9676589ce35STomáš Golembiovský         if (devnode != NULL) {
9686589ce35STomáš Golembiovský             disk->dev = g_strdup(devnode);
9696589ce35STomáš Golembiovský             disk->has_dev = true;
9706589ce35STomáš Golembiovský         }
971b616105aSTomáš Golembiovský         serial = udev_device_get_property_value(udevice, "ID_SERIAL");
972b616105aSTomáš Golembiovský         if (serial != NULL && *serial != 0) {
973b616105aSTomáš Golembiovský             disk->serial = g_strdup(serial);
974b616105aSTomáš Golembiovský             disk->has_serial = true;
975b616105aSTomáš Golembiovský         }
976b616105aSTomáš Golembiovský     }
977b616105aSTomáš Golembiovský #endif
978b616105aSTomáš Golembiovský 
97946d4c572STomoki Sekiyama     if (strcmp(driver, "ata_piix") == 0) {
98046d4c572STomoki Sekiyama         /* a host per ide bus, target*:0:<unit>:0 */
98146d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
98246d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
98346d4c572STomoki Sekiyama             goto cleanup;
98446d4c572STomoki Sekiyama         }
98546d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
98646d4c572STomoki Sekiyama             if (host == hosts[i]) {
98746d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_IDE;
98846d4c572STomoki Sekiyama                 disk->bus = i;
98946d4c572STomoki Sekiyama                 disk->unit = tgt[1];
99046d4c572STomoki Sekiyama                 break;
99146d4c572STomoki Sekiyama             }
99246d4c572STomoki Sekiyama         }
99346d4c572STomoki Sekiyama         if (i >= nhosts) {
99446d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
99546d4c572STomoki Sekiyama             goto cleanup;
99646d4c572STomoki Sekiyama         }
99746d4c572STomoki Sekiyama     } else if (strcmp(driver, "sym53c8xx") == 0) {
99846d4c572STomoki Sekiyama         /* scsi(LSI Logic): target*:0:<unit>:0 */
99946d4c572STomoki Sekiyama         if (!has_tgt) {
100046d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
100146d4c572STomoki Sekiyama             goto cleanup;
100246d4c572STomoki Sekiyama         }
100346d4c572STomoki Sekiyama         disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
100446d4c572STomoki Sekiyama         disk->unit = tgt[1];
100546d4c572STomoki Sekiyama     } else if (strcmp(driver, "virtio-pci") == 0) {
100646d4c572STomoki Sekiyama         if (has_tgt) {
100746d4c572STomoki Sekiyama             /* virtio-scsi: target*:0:0:<unit> */
100846d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_SCSI;
100946d4c572STomoki Sekiyama             disk->unit = tgt[2];
101046d4c572STomoki Sekiyama         } else {
101146d4c572STomoki Sekiyama             /* virtio-blk: 1 disk per 1 device */
101246d4c572STomoki Sekiyama             disk->bus_type = GUEST_DISK_BUS_TYPE_VIRTIO;
101346d4c572STomoki Sekiyama         }
101446d4c572STomoki Sekiyama     } else if (strcmp(driver, "ahci") == 0) {
101546d4c572STomoki Sekiyama         /* ahci: 1 host per 1 unit */
101646d4c572STomoki Sekiyama         if (!has_host || !has_tgt) {
101746d4c572STomoki Sekiyama             g_debug("invalid sysfs path '%s' (driver '%s')", syspath, driver);
101846d4c572STomoki Sekiyama             goto cleanup;
101946d4c572STomoki Sekiyama         }
102046d4c572STomoki Sekiyama         for (i = 0; i < nhosts; i++) {
102146d4c572STomoki Sekiyama             if (host == hosts[i]) {
102246d4c572STomoki Sekiyama                 disk->unit = i;
102346d4c572STomoki Sekiyama                 disk->bus_type = GUEST_DISK_BUS_TYPE_SATA;
102446d4c572STomoki Sekiyama                 break;
102546d4c572STomoki Sekiyama             }
102646d4c572STomoki Sekiyama         }
102746d4c572STomoki Sekiyama         if (i >= nhosts) {
102846d4c572STomoki Sekiyama             g_debug("no host for '%s' (driver '%s')", syspath, driver);
102946d4c572STomoki Sekiyama             goto cleanup;
103046d4c572STomoki Sekiyama         }
103146d4c572STomoki Sekiyama     } else {
103246d4c572STomoki Sekiyama         g_debug("unknown driver '%s' (sysfs path '%s')", driver, syspath);
103346d4c572STomoki Sekiyama         goto cleanup;
103446d4c572STomoki Sekiyama     }
103546d4c572STomoki Sekiyama 
103646d4c572STomoki Sekiyama     list->next = fs->disk;
103746d4c572STomoki Sekiyama     fs->disk = list;
1038b616105aSTomáš Golembiovský     goto out;
103946d4c572STomoki Sekiyama 
104046d4c572STomoki Sekiyama cleanup:
104146d4c572STomoki Sekiyama     if (list) {
104246d4c572STomoki Sekiyama         qapi_free_GuestDiskAddressList(list);
104346d4c572STomoki Sekiyama     }
1044b616105aSTomáš Golembiovský out:
104546d4c572STomoki Sekiyama     g_free(driver);
1046b616105aSTomáš Golembiovský #ifdef CONFIG_LIBUDEV
1047b616105aSTomáš Golembiovský     udev_unref(udev);
1048b616105aSTomáš Golembiovský     udev_device_unref(udevice);
1049b616105aSTomáš Golembiovský #endif
1050b616105aSTomáš Golembiovský     return;
105146d4c572STomoki Sekiyama }
105246d4c572STomoki Sekiyama 
105346d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
105446d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
105546d4c572STomoki Sekiyama                                           Error **errp);
105646d4c572STomoki Sekiyama 
105746d4c572STomoki Sekiyama /* Store a list of slave devices of virtual volume specified by @syspath into
105846d4c572STomoki Sekiyama  * @fs */
105946d4c572STomoki Sekiyama static void build_guest_fsinfo_for_virtual_device(char const *syspath,
106046d4c572STomoki Sekiyama                                                   GuestFilesystemInfo *fs,
106146d4c572STomoki Sekiyama                                                   Error **errp)
106246d4c572STomoki Sekiyama {
1063*292743d9SMarkus Armbruster     Error *err = NULL;
106446d4c572STomoki Sekiyama     DIR *dir;
106546d4c572STomoki Sekiyama     char *dirpath;
1066e668d1b8Szhanghailiang     struct dirent *entry;
106746d4c572STomoki Sekiyama 
106846d4c572STomoki Sekiyama     dirpath = g_strdup_printf("%s/slaves", syspath);
106946d4c572STomoki Sekiyama     dir = opendir(dirpath);
107046d4c572STomoki Sekiyama     if (!dir) {
10718251a72fSMichael Roth         if (errno != ENOENT) {
107246d4c572STomoki Sekiyama             error_setg_errno(errp, errno, "opendir(\"%s\")", dirpath);
10738251a72fSMichael Roth         }
107446d4c572STomoki Sekiyama         g_free(dirpath);
107546d4c572STomoki Sekiyama         return;
107646d4c572STomoki Sekiyama     }
107746d4c572STomoki Sekiyama 
107846d4c572STomoki Sekiyama     for (;;) {
1079e668d1b8Szhanghailiang         errno = 0;
1080e668d1b8Szhanghailiang         entry = readdir(dir);
1081e668d1b8Szhanghailiang         if (entry == NULL) {
1082e668d1b8Szhanghailiang             if (errno) {
1083e668d1b8Szhanghailiang                 error_setg_errno(errp, errno, "readdir(\"%s\")", dirpath);
108446d4c572STomoki Sekiyama             }
108546d4c572STomoki Sekiyama             break;
108646d4c572STomoki Sekiyama         }
108746d4c572STomoki Sekiyama 
1088e668d1b8Szhanghailiang         if (entry->d_type == DT_LNK) {
1089e668d1b8Szhanghailiang             char *path;
1090e668d1b8Szhanghailiang 
1091e668d1b8Szhanghailiang             g_debug(" slave device '%s'", entry->d_name);
1092e668d1b8Szhanghailiang             path = g_strdup_printf("%s/slaves/%s", syspath, entry->d_name);
1093*292743d9SMarkus Armbruster             build_guest_fsinfo_for_device(path, fs, &err);
1094e668d1b8Szhanghailiang             g_free(path);
109546d4c572STomoki Sekiyama 
1096*292743d9SMarkus Armbruster             if (err) {
1097*292743d9SMarkus Armbruster                 error_propagate(errp, err);
109846d4c572STomoki Sekiyama                 break;
109946d4c572STomoki Sekiyama             }
110046d4c572STomoki Sekiyama         }
110146d4c572STomoki Sekiyama     }
110246d4c572STomoki Sekiyama 
1103e668d1b8Szhanghailiang     g_free(dirpath);
110446d4c572STomoki Sekiyama     closedir(dir);
110546d4c572STomoki Sekiyama }
110646d4c572STomoki Sekiyama 
110746d4c572STomoki Sekiyama /* Dispatch to functions for virtual/real device */
110846d4c572STomoki Sekiyama static void build_guest_fsinfo_for_device(char const *devpath,
110946d4c572STomoki Sekiyama                                           GuestFilesystemInfo *fs,
111046d4c572STomoki Sekiyama                                           Error **errp)
111146d4c572STomoki Sekiyama {
111246d4c572STomoki Sekiyama     char *syspath = realpath(devpath, NULL);
111346d4c572STomoki Sekiyama 
111446d4c572STomoki Sekiyama     if (!syspath) {
111546d4c572STomoki Sekiyama         error_setg_errno(errp, errno, "realpath(\"%s\")", devpath);
111646d4c572STomoki Sekiyama         return;
111746d4c572STomoki Sekiyama     }
111846d4c572STomoki Sekiyama 
111946d4c572STomoki Sekiyama     if (!fs->name) {
11203e015d81SJulia Suvorova         fs->name = g_path_get_basename(syspath);
112146d4c572STomoki Sekiyama     }
112246d4c572STomoki Sekiyama 
112346d4c572STomoki Sekiyama     g_debug("  parse sysfs path '%s'", syspath);
112446d4c572STomoki Sekiyama 
112546d4c572STomoki Sekiyama     if (strstr(syspath, "/devices/virtual/block/")) {
112646d4c572STomoki Sekiyama         build_guest_fsinfo_for_virtual_device(syspath, fs, errp);
112746d4c572STomoki Sekiyama     } else {
112846d4c572STomoki Sekiyama         build_guest_fsinfo_for_real_device(syspath, fs, errp);
112946d4c572STomoki Sekiyama     }
113046d4c572STomoki Sekiyama 
113146d4c572STomoki Sekiyama     free(syspath);
113246d4c572STomoki Sekiyama }
113346d4c572STomoki Sekiyama 
113446d4c572STomoki Sekiyama /* Return a list of the disk device(s)' info which @mount lies on */
113546d4c572STomoki Sekiyama static GuestFilesystemInfo *build_guest_fsinfo(struct FsMount *mount,
113646d4c572STomoki Sekiyama                                                Error **errp)
113746d4c572STomoki Sekiyama {
113846d4c572STomoki Sekiyama     GuestFilesystemInfo *fs = g_malloc0(sizeof(*fs));
113925b5ff1aSChen Hanxiao     struct statvfs buf;
114025b5ff1aSChen Hanxiao     unsigned long used, nonroot_total, fr_size;
114146d4c572STomoki Sekiyama     char *devpath = g_strdup_printf("/sys/dev/block/%u:%u",
114246d4c572STomoki Sekiyama                                     mount->devmajor, mount->devminor);
114346d4c572STomoki Sekiyama 
114446d4c572STomoki Sekiyama     fs->mountpoint = g_strdup(mount->dirname);
114546d4c572STomoki Sekiyama     fs->type = g_strdup(mount->devtype);
114646d4c572STomoki Sekiyama     build_guest_fsinfo_for_device(devpath, fs, errp);
114746d4c572STomoki Sekiyama 
114825b5ff1aSChen Hanxiao     if (statvfs(fs->mountpoint, &buf) == 0) {
114925b5ff1aSChen Hanxiao         fr_size = buf.f_frsize;
115025b5ff1aSChen Hanxiao         used = buf.f_blocks - buf.f_bfree;
115125b5ff1aSChen Hanxiao         nonroot_total = used + buf.f_bavail;
115225b5ff1aSChen Hanxiao         fs->used_bytes = used * fr_size;
115325b5ff1aSChen Hanxiao         fs->total_bytes = nonroot_total * fr_size;
115425b5ff1aSChen Hanxiao 
115525b5ff1aSChen Hanxiao         fs->has_total_bytes = true;
115625b5ff1aSChen Hanxiao         fs->has_used_bytes = true;
115725b5ff1aSChen Hanxiao     }
115825b5ff1aSChen Hanxiao 
115946d4c572STomoki Sekiyama     g_free(devpath);
116025b5ff1aSChen Hanxiao 
116146d4c572STomoki Sekiyama     return fs;
116246d4c572STomoki Sekiyama }
116346d4c572STomoki Sekiyama 
116446d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
116546d4c572STomoki Sekiyama {
116646d4c572STomoki Sekiyama     FsMountList mounts;
116746d4c572STomoki Sekiyama     struct FsMount *mount;
116846d4c572STomoki Sekiyama     GuestFilesystemInfoList *new, *ret = NULL;
116946d4c572STomoki Sekiyama     Error *local_err = NULL;
117046d4c572STomoki Sekiyama 
117146d4c572STomoki Sekiyama     QTAILQ_INIT(&mounts);
117246d4c572STomoki Sekiyama     build_fs_mount_list(&mounts, &local_err);
117346d4c572STomoki Sekiyama     if (local_err) {
117446d4c572STomoki Sekiyama         error_propagate(errp, local_err);
117546d4c572STomoki Sekiyama         return NULL;
117646d4c572STomoki Sekiyama     }
117746d4c572STomoki Sekiyama 
117846d4c572STomoki Sekiyama     QTAILQ_FOREACH(mount, &mounts, next) {
117946d4c572STomoki Sekiyama         g_debug("Building guest fsinfo for '%s'", mount->dirname);
118046d4c572STomoki Sekiyama 
118146d4c572STomoki Sekiyama         new = g_malloc0(sizeof(*ret));
118246d4c572STomoki Sekiyama         new->value = build_guest_fsinfo(mount, &local_err);
118346d4c572STomoki Sekiyama         new->next = ret;
118446d4c572STomoki Sekiyama         ret = new;
118546d4c572STomoki Sekiyama         if (local_err) {
118646d4c572STomoki Sekiyama             error_propagate(errp, local_err);
118746d4c572STomoki Sekiyama             qapi_free_GuestFilesystemInfoList(ret);
118846d4c572STomoki Sekiyama             ret = NULL;
118946d4c572STomoki Sekiyama             break;
119046d4c572STomoki Sekiyama         }
119146d4c572STomoki Sekiyama     }
119246d4c572STomoki Sekiyama 
119346d4c572STomoki Sekiyama     free_fs_mount_list(&mounts);
119446d4c572STomoki Sekiyama     return ret;
119546d4c572STomoki Sekiyama }
119646d4c572STomoki Sekiyama 
119746d4c572STomoki Sekiyama 
1198ec0f694cSTomoki Sekiyama typedef enum {
1199ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_THAW = 0,
1200ec0f694cSTomoki Sekiyama     FSFREEZE_HOOK_FREEZE,
1201ec0f694cSTomoki Sekiyama } FsfreezeHookArg;
1202ec0f694cSTomoki Sekiyama 
120313a439ecSStefan Weil static const char *fsfreeze_hook_arg_string[] = {
1204ec0f694cSTomoki Sekiyama     "thaw",
1205ec0f694cSTomoki Sekiyama     "freeze",
1206ec0f694cSTomoki Sekiyama };
1207ec0f694cSTomoki Sekiyama 
120877dbc81bSMarkus Armbruster static void execute_fsfreeze_hook(FsfreezeHookArg arg, Error **errp)
1209ec0f694cSTomoki Sekiyama {
1210ec0f694cSTomoki Sekiyama     int status;
1211ec0f694cSTomoki Sekiyama     pid_t pid;
1212ec0f694cSTomoki Sekiyama     const char *hook;
1213ec0f694cSTomoki Sekiyama     const char *arg_str = fsfreeze_hook_arg_string[arg];
1214ec0f694cSTomoki Sekiyama     Error *local_err = NULL;
1215ec0f694cSTomoki Sekiyama 
1216ec0f694cSTomoki Sekiyama     hook = ga_fsfreeze_hook(ga_state);
1217ec0f694cSTomoki Sekiyama     if (!hook) {
1218ec0f694cSTomoki Sekiyama         return;
1219ec0f694cSTomoki Sekiyama     }
1220ec0f694cSTomoki Sekiyama     if (access(hook, X_OK) != 0) {
122177dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "can't access fsfreeze hook '%s'", hook);
1222ec0f694cSTomoki Sekiyama         return;
1223ec0f694cSTomoki Sekiyama     }
1224ec0f694cSTomoki Sekiyama 
1225ec0f694cSTomoki Sekiyama     slog("executing fsfreeze hook with arg '%s'", arg_str);
1226ec0f694cSTomoki Sekiyama     pid = fork();
1227ec0f694cSTomoki Sekiyama     if (pid == 0) {
1228ec0f694cSTomoki Sekiyama         setsid();
1229ec0f694cSTomoki Sekiyama         reopen_fd_to_null(0);
1230ec0f694cSTomoki Sekiyama         reopen_fd_to_null(1);
1231ec0f694cSTomoki Sekiyama         reopen_fd_to_null(2);
1232ec0f694cSTomoki Sekiyama 
1233ec0f694cSTomoki Sekiyama         execle(hook, hook, arg_str, NULL, environ);
1234ec0f694cSTomoki Sekiyama         _exit(EXIT_FAILURE);
1235ec0f694cSTomoki Sekiyama     } else if (pid < 0) {
123677dbc81bSMarkus Armbruster         error_setg_errno(errp, errno, "failed to create child process");
1237ec0f694cSTomoki Sekiyama         return;
1238ec0f694cSTomoki Sekiyama     }
1239ec0f694cSTomoki Sekiyama 
1240ec0f694cSTomoki Sekiyama     ga_wait_child(pid, &status, &local_err);
124184d18f06SMarkus Armbruster     if (local_err) {
124277dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1243ec0f694cSTomoki Sekiyama         return;
1244ec0f694cSTomoki Sekiyama     }
1245ec0f694cSTomoki Sekiyama 
1246ec0f694cSTomoki Sekiyama     if (!WIFEXITED(status)) {
124777dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has terminated abnormally");
1248ec0f694cSTomoki Sekiyama         return;
1249ec0f694cSTomoki Sekiyama     }
1250ec0f694cSTomoki Sekiyama 
1251ec0f694cSTomoki Sekiyama     status = WEXITSTATUS(status);
1252ec0f694cSTomoki Sekiyama     if (status) {
125377dbc81bSMarkus Armbruster         error_setg(errp, "fsfreeze hook has failed with status %d", status);
1254ec0f694cSTomoki Sekiyama         return;
1255ec0f694cSTomoki Sekiyama     }
1256ec0f694cSTomoki Sekiyama }
1257ec0f694cSTomoki Sekiyama 
1258c216e5adSMichael Roth /*
1259c216e5adSMichael Roth  * Return status of freeze/thaw
1260c216e5adSMichael Roth  */
126177dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
1262c216e5adSMichael Roth {
1263f22d85e9SMichael Roth     if (ga_is_frozen(ga_state)) {
1264f22d85e9SMichael Roth         return GUEST_FSFREEZE_STATUS_FROZEN;
1265f22d85e9SMichael Roth     }
1266f22d85e9SMichael Roth 
1267f22d85e9SMichael Roth     return GUEST_FSFREEZE_STATUS_THAWED;
1268c216e5adSMichael Roth }
1269c216e5adSMichael Roth 
1270e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze(Error **errp)
1271e99bce20STomoki Sekiyama {
1272e99bce20STomoki Sekiyama     return qmp_guest_fsfreeze_freeze_list(false, NULL, errp);
1273e99bce20STomoki Sekiyama }
1274e99bce20STomoki Sekiyama 
1275c216e5adSMichael Roth /*
1276c216e5adSMichael Roth  * Walk list of mounted file systems in the guest, and freeze the ones which
1277c216e5adSMichael Roth  * are real local file systems.
1278c216e5adSMichael Roth  */
1279e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
1280e99bce20STomoki Sekiyama                                        strList *mountpoints,
1281e99bce20STomoki Sekiyama                                        Error **errp)
1282c216e5adSMichael Roth {
1283c216e5adSMichael Roth     int ret = 0, i = 0;
1284e99bce20STomoki Sekiyama     strList *list;
1285af02203fSPaolo Bonzini     FsMountList mounts;
1286af02203fSPaolo Bonzini     struct FsMount *mount;
1287261551d1SLuiz Capitulino     Error *local_err = NULL;
1288c216e5adSMichael Roth     int fd;
1289c216e5adSMichael Roth 
1290c216e5adSMichael Roth     slog("guest-fsfreeze called");
1291c216e5adSMichael Roth 
1292ec0f694cSTomoki Sekiyama     execute_fsfreeze_hook(FSFREEZE_HOOK_FREEZE, &local_err);
129384d18f06SMarkus Armbruster     if (local_err) {
129477dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1295ec0f694cSTomoki Sekiyama         return -1;
1296ec0f694cSTomoki Sekiyama     }
1297ec0f694cSTomoki Sekiyama 
12989e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1299261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
130084d18f06SMarkus Armbruster     if (local_err) {
130177dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1302261551d1SLuiz Capitulino         return -1;
1303c216e5adSMichael Roth     }
1304c216e5adSMichael Roth 
1305c216e5adSMichael Roth     /* cannot risk guest agent blocking itself on a write in this state */
1306f22d85e9SMichael Roth     ga_set_frozen(ga_state);
1307c216e5adSMichael Roth 
1308eae3eb3eSPaolo Bonzini     QTAILQ_FOREACH_REVERSE(mount, &mounts, next) {
1309e99bce20STomoki Sekiyama         /* To issue fsfreeze in the reverse order of mounts, check if the
1310e99bce20STomoki Sekiyama          * mount is listed in the list here */
1311e99bce20STomoki Sekiyama         if (has_mountpoints) {
1312e99bce20STomoki Sekiyama             for (list = mountpoints; list; list = list->next) {
1313e99bce20STomoki Sekiyama                 if (strcmp(list->value, mount->dirname) == 0) {
1314e99bce20STomoki Sekiyama                     break;
1315e99bce20STomoki Sekiyama                 }
1316e99bce20STomoki Sekiyama             }
1317e99bce20STomoki Sekiyama             if (!list) {
1318e99bce20STomoki Sekiyama                 continue;
1319e99bce20STomoki Sekiyama             }
1320e99bce20STomoki Sekiyama         }
1321e99bce20STomoki Sekiyama 
1322c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1323c216e5adSMichael Roth         if (fd == -1) {
132477dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "failed to open %s", mount->dirname);
1325c216e5adSMichael Roth             goto error;
1326c216e5adSMichael Roth         }
1327c216e5adSMichael Roth 
1328e35916acSMichael Tokarev         /* we try to cull filesystems we know won't work in advance, but other
1329e35916acSMichael Tokarev          * filesystems may not implement fsfreeze for less obvious reasons.
13309e8aded4SMichael Roth          * these will report EOPNOTSUPP. we simply ignore these when tallying
13319e8aded4SMichael Roth          * the number of frozen filesystems.
1332ce2eb6c4SPeter Lieven          * if a filesystem is mounted more than once (aka bind mount) a
1333ce2eb6c4SPeter Lieven          * consecutive attempt to freeze an already frozen filesystem will
1334ce2eb6c4SPeter Lieven          * return EBUSY.
13359e8aded4SMichael Roth          *
13369e8aded4SMichael Roth          * any other error means a failure to freeze a filesystem we
13379e8aded4SMichael Roth          * expect to be freezable, so return an error in those cases
13389e8aded4SMichael Roth          * and return system to thawed state.
1339c216e5adSMichael Roth          */
1340c216e5adSMichael Roth         ret = ioctl(fd, FIFREEZE);
13419e8aded4SMichael Roth         if (ret == -1) {
1342ce2eb6c4SPeter Lieven             if (errno != EOPNOTSUPP && errno != EBUSY) {
134377dbc81bSMarkus Armbruster                 error_setg_errno(errp, errno, "failed to freeze %s",
1344617fbbc1SLuiz Capitulino                                  mount->dirname);
1345c216e5adSMichael Roth                 close(fd);
1346c216e5adSMichael Roth                 goto error;
1347c216e5adSMichael Roth             }
13489e8aded4SMichael Roth         } else {
1349c216e5adSMichael Roth             i++;
1350c216e5adSMichael Roth         }
13519e8aded4SMichael Roth         close(fd);
13529e8aded4SMichael Roth     }
1353c216e5adSMichael Roth 
1354af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
135565650f01SChen Hanxiao     /* We may not issue any FIFREEZE here.
135665650f01SChen Hanxiao      * Just unset ga_state here and ready for the next call.
135765650f01SChen Hanxiao      */
135865650f01SChen Hanxiao     if (i == 0) {
135965650f01SChen Hanxiao         ga_unset_frozen(ga_state);
136065650f01SChen Hanxiao     }
1361c216e5adSMichael Roth     return i;
1362c216e5adSMichael Roth 
1363c216e5adSMichael Roth error:
1364af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1365c216e5adSMichael Roth     qmp_guest_fsfreeze_thaw(NULL);
1366c216e5adSMichael Roth     return 0;
1367c216e5adSMichael Roth }
1368c216e5adSMichael Roth 
1369c216e5adSMichael Roth /*
1370c216e5adSMichael Roth  * Walk list of frozen file systems in the guest, and thaw them.
1371c216e5adSMichael Roth  */
137277dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
1373c216e5adSMichael Roth {
1374c216e5adSMichael Roth     int ret;
1375af02203fSPaolo Bonzini     FsMountList mounts;
1376af02203fSPaolo Bonzini     FsMount *mount;
13779e8aded4SMichael Roth     int fd, i = 0, logged;
1378261551d1SLuiz Capitulino     Error *local_err = NULL;
1379c216e5adSMichael Roth 
13809e8aded4SMichael Roth     QTAILQ_INIT(&mounts);
1381261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
138284d18f06SMarkus Armbruster     if (local_err) {
138377dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
13849e8aded4SMichael Roth         return 0;
13859e8aded4SMichael Roth     }
13869e8aded4SMichael Roth 
13879e8aded4SMichael Roth     QTAILQ_FOREACH(mount, &mounts, next) {
13889e8aded4SMichael Roth         logged = false;
1389c216e5adSMichael Roth         fd = qemu_open(mount->dirname, O_RDONLY);
1390c216e5adSMichael Roth         if (fd == -1) {
1391c216e5adSMichael Roth             continue;
1392c216e5adSMichael Roth         }
13939e8aded4SMichael Roth         /* we have no way of knowing whether a filesystem was actually unfrozen
13949e8aded4SMichael Roth          * as a result of a successful call to FITHAW, only that if an error
13959e8aded4SMichael Roth          * was returned the filesystem was *not* unfrozen by that particular
13969e8aded4SMichael Roth          * call.
13979e8aded4SMichael Roth          *
1398a31f0531SJim Meyering          * since multiple preceding FIFREEZEs require multiple calls to FITHAW
13999e8aded4SMichael Roth          * to unfreeze, continuing issuing FITHAW until an error is returned,
14009e8aded4SMichael Roth          * in which case either the filesystem is in an unfreezable state, or,
14019e8aded4SMichael Roth          * more likely, it was thawed previously (and remains so afterward).
14029e8aded4SMichael Roth          *
14039e8aded4SMichael Roth          * also, since the most recent successful call is the one that did
14049e8aded4SMichael Roth          * the actual unfreeze, we can use this to provide an accurate count
14059e8aded4SMichael Roth          * of the number of filesystems unfrozen by guest-fsfreeze-thaw, which
14069e8aded4SMichael Roth          * may * be useful for determining whether a filesystem was unfrozen
14079e8aded4SMichael Roth          * during the freeze/thaw phase by a process other than qemu-ga.
14089e8aded4SMichael Roth          */
14099e8aded4SMichael Roth         do {
1410c216e5adSMichael Roth             ret = ioctl(fd, FITHAW);
14119e8aded4SMichael Roth             if (ret == 0 && !logged) {
1412c216e5adSMichael Roth                 i++;
14139e8aded4SMichael Roth                 logged = true;
14149e8aded4SMichael Roth             }
14159e8aded4SMichael Roth         } while (ret == 0);
14169e8aded4SMichael Roth         close(fd);
1417c216e5adSMichael Roth     }
1418c216e5adSMichael Roth 
1419f22d85e9SMichael Roth     ga_unset_frozen(ga_state);
1420af02203fSPaolo Bonzini     free_fs_mount_list(&mounts);
1421ec0f694cSTomoki Sekiyama 
142277dbc81bSMarkus Armbruster     execute_fsfreeze_hook(FSFREEZE_HOOK_THAW, errp);
1423ec0f694cSTomoki Sekiyama 
1424c216e5adSMichael Roth     return i;
1425c216e5adSMichael Roth }
1426c216e5adSMichael Roth 
1427c216e5adSMichael Roth static void guest_fsfreeze_cleanup(void)
1428c216e5adSMichael Roth {
1429c216e5adSMichael Roth     Error *err = NULL;
1430c216e5adSMichael Roth 
1431f22d85e9SMichael Roth     if (ga_is_frozen(ga_state) == GUEST_FSFREEZE_STATUS_FROZEN) {
14326f686749SMarkus Armbruster         qmp_guest_fsfreeze_thaw(&err);
14336f686749SMarkus Armbruster         if (err) {
14346f686749SMarkus Armbruster             slog("failed to clean up frozen filesystems: %s",
14356f686749SMarkus Armbruster                  error_get_pretty(err));
14366f686749SMarkus Armbruster             error_free(err);
1437c216e5adSMichael Roth         }
1438c216e5adSMichael Roth     }
1439c216e5adSMichael Roth }
1440e72c3f2eSMichael Roth #endif /* CONFIG_FSFREEZE */
1441c216e5adSMichael Roth 
1442eab5fd59SPaolo Bonzini #if defined(CONFIG_FSTRIM)
1443eab5fd59SPaolo Bonzini /*
1444eab5fd59SPaolo Bonzini  * Walk list of mounted file systems in the guest, and trim them.
1445eab5fd59SPaolo Bonzini  */
1446e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
1447e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
1448eab5fd59SPaolo Bonzini {
1449e82855d9SJustin Ossevoort     GuestFilesystemTrimResponse *response;
1450e82855d9SJustin Ossevoort     GuestFilesystemTrimResultList *list;
1451e82855d9SJustin Ossevoort     GuestFilesystemTrimResult *result;
1452eab5fd59SPaolo Bonzini     int ret = 0;
1453eab5fd59SPaolo Bonzini     FsMountList mounts;
1454eab5fd59SPaolo Bonzini     struct FsMount *mount;
1455eab5fd59SPaolo Bonzini     int fd;
1456261551d1SLuiz Capitulino     Error *local_err = NULL;
145773a652a1SJustin Ossevoort     struct fstrim_range r;
1458eab5fd59SPaolo Bonzini 
1459eab5fd59SPaolo Bonzini     slog("guest-fstrim called");
1460eab5fd59SPaolo Bonzini 
1461eab5fd59SPaolo Bonzini     QTAILQ_INIT(&mounts);
1462261551d1SLuiz Capitulino     build_fs_mount_list(&mounts, &local_err);
146384d18f06SMarkus Armbruster     if (local_err) {
146477dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
1465e82855d9SJustin Ossevoort         return NULL;
1466eab5fd59SPaolo Bonzini     }
1467eab5fd59SPaolo Bonzini 
1468e82855d9SJustin Ossevoort     response = g_malloc0(sizeof(*response));
1469e82855d9SJustin Ossevoort 
1470eab5fd59SPaolo Bonzini     QTAILQ_FOREACH(mount, &mounts, next) {
1471e82855d9SJustin Ossevoort         result = g_malloc0(sizeof(*result));
1472e82855d9SJustin Ossevoort         result->path = g_strdup(mount->dirname);
1473e82855d9SJustin Ossevoort 
1474e82855d9SJustin Ossevoort         list = g_malloc0(sizeof(*list));
1475e82855d9SJustin Ossevoort         list->value = result;
1476e82855d9SJustin Ossevoort         list->next = response->paths;
1477e82855d9SJustin Ossevoort         response->paths = list;
1478e82855d9SJustin Ossevoort 
1479eab5fd59SPaolo Bonzini         fd = qemu_open(mount->dirname, O_RDONLY);
1480eab5fd59SPaolo Bonzini         if (fd == -1) {
1481e82855d9SJustin Ossevoort             result->error = g_strdup_printf("failed to open: %s",
1482e82855d9SJustin Ossevoort                                             strerror(errno));
1483e82855d9SJustin Ossevoort             result->has_error = true;
1484e82855d9SJustin Ossevoort             continue;
1485eab5fd59SPaolo Bonzini         }
1486eab5fd59SPaolo Bonzini 
1487e35916acSMichael Tokarev         /* We try to cull filesystems we know won't work in advance, but other
1488e35916acSMichael Tokarev          * filesystems may not implement fstrim for less obvious reasons.
1489e35916acSMichael Tokarev          * These will report EOPNOTSUPP; while in some other cases ENOTTY
1490e35916acSMichael Tokarev          * will be reported (e.g. CD-ROMs).
1491e82855d9SJustin Ossevoort          * Any other error means an unexpected error.
1492eab5fd59SPaolo Bonzini          */
149373a652a1SJustin Ossevoort         r.start = 0;
149473a652a1SJustin Ossevoort         r.len = -1;
149573a652a1SJustin Ossevoort         r.minlen = has_minimum ? minimum : 0;
1496eab5fd59SPaolo Bonzini         ret = ioctl(fd, FITRIM, &r);
1497eab5fd59SPaolo Bonzini         if (ret == -1) {
1498e82855d9SJustin Ossevoort             result->has_error = true;
1499e82855d9SJustin Ossevoort             if (errno == ENOTTY || errno == EOPNOTSUPP) {
1500e82855d9SJustin Ossevoort                 result->error = g_strdup("trim not supported");
1501e82855d9SJustin Ossevoort             } else {
1502e82855d9SJustin Ossevoort                 result->error = g_strdup_printf("failed to trim: %s",
1503e82855d9SJustin Ossevoort                                                 strerror(errno));
1504e82855d9SJustin Ossevoort             }
1505eab5fd59SPaolo Bonzini             close(fd);
1506e82855d9SJustin Ossevoort             continue;
1507eab5fd59SPaolo Bonzini         }
1508e82855d9SJustin Ossevoort 
1509e82855d9SJustin Ossevoort         result->has_minimum = true;
1510e82855d9SJustin Ossevoort         result->minimum = r.minlen;
1511e82855d9SJustin Ossevoort         result->has_trimmed = true;
1512e82855d9SJustin Ossevoort         result->trimmed = r.len;
1513eab5fd59SPaolo Bonzini         close(fd);
1514eab5fd59SPaolo Bonzini     }
1515eab5fd59SPaolo Bonzini 
1516eab5fd59SPaolo Bonzini     free_fs_mount_list(&mounts);
1517e82855d9SJustin Ossevoort     return response;
1518eab5fd59SPaolo Bonzini }
1519eab5fd59SPaolo Bonzini #endif /* CONFIG_FSTRIM */
1520eab5fd59SPaolo Bonzini 
1521eab5fd59SPaolo Bonzini 
152211d0f125SLuiz Capitulino #define LINUX_SYS_STATE_FILE "/sys/power/state"
152311d0f125SLuiz Capitulino #define SUSPEND_SUPPORTED 0
152411d0f125SLuiz Capitulino #define SUSPEND_NOT_SUPPORTED 1
152511d0f125SLuiz Capitulino 
15268b020b5eSDaniel Henrique Barboza typedef enum {
15278b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_DISK = 0,
15288b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_RAM = 1,
15298b020b5eSDaniel Henrique Barboza     SUSPEND_MODE_HYBRID = 2,
15308b020b5eSDaniel Henrique Barboza } SuspendMode;
15318b020b5eSDaniel Henrique Barboza 
15328b020b5eSDaniel Henrique Barboza /*
15338b020b5eSDaniel Henrique Barboza  * Executes a command in a child process using g_spawn_sync,
15348b020b5eSDaniel Henrique Barboza  * returning an int >= 0 representing the exit status of the
15358b020b5eSDaniel Henrique Barboza  * process.
15368b020b5eSDaniel Henrique Barboza  *
15378b020b5eSDaniel Henrique Barboza  * If the program wasn't found in path, returns -1.
15388b020b5eSDaniel Henrique Barboza  *
15398b020b5eSDaniel Henrique Barboza  * If a problem happened when creating the child process,
15408b020b5eSDaniel Henrique Barboza  * returns -1 and errp is set.
15418b020b5eSDaniel Henrique Barboza  */
15428b020b5eSDaniel Henrique Barboza static int run_process_child(const char *command[], Error **errp)
15438b020b5eSDaniel Henrique Barboza {
15448b020b5eSDaniel Henrique Barboza     int exit_status, spawn_flag;
15458b020b5eSDaniel Henrique Barboza     GError *g_err = NULL;
15468b020b5eSDaniel Henrique Barboza     bool success;
15478b020b5eSDaniel Henrique Barboza 
15488b020b5eSDaniel Henrique Barboza     spawn_flag = G_SPAWN_SEARCH_PATH | G_SPAWN_STDOUT_TO_DEV_NULL |
15498b020b5eSDaniel Henrique Barboza                  G_SPAWN_STDERR_TO_DEV_NULL;
15508b020b5eSDaniel Henrique Barboza 
15518b020b5eSDaniel Henrique Barboza     success =  g_spawn_sync(NULL, (char **)command, environ, spawn_flag,
15528b020b5eSDaniel Henrique Barboza                             NULL, NULL, NULL, NULL,
15538b020b5eSDaniel Henrique Barboza                             &exit_status, &g_err);
15548b020b5eSDaniel Henrique Barboza 
15558b020b5eSDaniel Henrique Barboza     if (success) {
15568b020b5eSDaniel Henrique Barboza         return WEXITSTATUS(exit_status);
15578b020b5eSDaniel Henrique Barboza     }
15588b020b5eSDaniel Henrique Barboza 
15598b020b5eSDaniel Henrique Barboza     if (g_err && (g_err->code != G_SPAWN_ERROR_NOENT)) {
15608b020b5eSDaniel Henrique Barboza         error_setg(errp, "failed to create child process, error '%s'",
15618b020b5eSDaniel Henrique Barboza                    g_err->message);
15628b020b5eSDaniel Henrique Barboza     }
15638b020b5eSDaniel Henrique Barboza 
15648b020b5eSDaniel Henrique Barboza     g_error_free(g_err);
15658b020b5eSDaniel Henrique Barboza     return -1;
15668b020b5eSDaniel Henrique Barboza }
15678b020b5eSDaniel Henrique Barboza 
1568067927d6SDaniel Henrique Barboza static bool systemd_supports_mode(SuspendMode mode, Error **errp)
1569067927d6SDaniel Henrique Barboza {
1570067927d6SDaniel Henrique Barboza     Error *local_err = NULL;
1571067927d6SDaniel Henrique Barboza     const char *systemctl_args[3] = {"systemd-hibernate", "systemd-suspend",
1572067927d6SDaniel Henrique Barboza                                      "systemd-hybrid-sleep"};
1573067927d6SDaniel Henrique Barboza     const char *cmd[4] = {"systemctl", "status", systemctl_args[mode], NULL};
1574067927d6SDaniel Henrique Barboza     int status;
1575067927d6SDaniel Henrique Barboza 
1576067927d6SDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1577067927d6SDaniel Henrique Barboza 
1578067927d6SDaniel Henrique Barboza     /*
1579067927d6SDaniel Henrique Barboza      * systemctl status uses LSB return codes so we can expect
1580067927d6SDaniel Henrique Barboza      * status > 0 and be ok. To assert if the guest has support
1581067927d6SDaniel Henrique Barboza      * for the selected suspend mode, status should be < 4. 4 is
1582067927d6SDaniel Henrique Barboza      * the code for unknown service status, the return value when
1583067927d6SDaniel Henrique Barboza      * the service does not exist. A common value is status = 3
1584067927d6SDaniel Henrique Barboza      * (program is not running).
1585067927d6SDaniel Henrique Barboza      */
1586067927d6SDaniel Henrique Barboza     if (status > 0 && status < 4) {
1587067927d6SDaniel Henrique Barboza         return true;
1588067927d6SDaniel Henrique Barboza     }
1589067927d6SDaniel Henrique Barboza 
1590067927d6SDaniel Henrique Barboza     error_propagate(errp, local_err);
1591067927d6SDaniel Henrique Barboza     return false;
1592067927d6SDaniel Henrique Barboza }
1593067927d6SDaniel Henrique Barboza 
1594067927d6SDaniel Henrique Barboza static void systemd_suspend(SuspendMode mode, Error **errp)
1595067927d6SDaniel Henrique Barboza {
1596067927d6SDaniel Henrique Barboza     Error *local_err = NULL;
1597067927d6SDaniel Henrique Barboza     const char *systemctl_args[3] = {"hibernate", "suspend", "hybrid-sleep"};
1598067927d6SDaniel Henrique Barboza     const char *cmd[3] = {"systemctl", systemctl_args[mode], NULL};
1599067927d6SDaniel Henrique Barboza     int status;
1600067927d6SDaniel Henrique Barboza 
1601067927d6SDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1602067927d6SDaniel Henrique Barboza 
1603067927d6SDaniel Henrique Barboza     if (status == 0) {
1604067927d6SDaniel Henrique Barboza         return;
1605067927d6SDaniel Henrique Barboza     }
1606067927d6SDaniel Henrique Barboza 
1607067927d6SDaniel Henrique Barboza     if ((status == -1) && !local_err) {
1608067927d6SDaniel Henrique Barboza         error_setg(errp, "the helper program 'systemctl %s' was not found",
1609067927d6SDaniel Henrique Barboza                    systemctl_args[mode]);
1610067927d6SDaniel Henrique Barboza         return;
1611067927d6SDaniel Henrique Barboza     }
1612067927d6SDaniel Henrique Barboza 
1613067927d6SDaniel Henrique Barboza     if (local_err) {
1614067927d6SDaniel Henrique Barboza         error_propagate(errp, local_err);
1615067927d6SDaniel Henrique Barboza     } else {
1616067927d6SDaniel Henrique Barboza         error_setg(errp, "the helper program 'systemctl %s' returned an "
1617067927d6SDaniel Henrique Barboza                    "unexpected exit status code (%d)",
1618067927d6SDaniel Henrique Barboza                    systemctl_args[mode], status);
1619067927d6SDaniel Henrique Barboza     }
1620067927d6SDaniel Henrique Barboza }
1621067927d6SDaniel Henrique Barboza 
16228b020b5eSDaniel Henrique Barboza static bool pmutils_supports_mode(SuspendMode mode, Error **errp)
162311d0f125SLuiz Capitulino {
16246b26e837SLuiz Capitulino     Error *local_err = NULL;
16258b020b5eSDaniel Henrique Barboza     const char *pmutils_args[3] = {"--hibernate", "--suspend",
16268b020b5eSDaniel Henrique Barboza                                    "--suspend-hybrid"};
16278b020b5eSDaniel Henrique Barboza     const char *cmd[3] = {"pm-is-supported", pmutils_args[mode], NULL};
1628dc8764f0SLuiz Capitulino     int status;
162911d0f125SLuiz Capitulino 
16308b020b5eSDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1631304a0fcbSDaniel Henrique Barboza 
16328b020b5eSDaniel Henrique Barboza     if (status == SUSPEND_SUPPORTED) {
16338b020b5eSDaniel Henrique Barboza         return true;
1634304a0fcbSDaniel Henrique Barboza     }
1635304a0fcbSDaniel Henrique Barboza 
16368b020b5eSDaniel Henrique Barboza     if ((status == -1) && !local_err) {
16378b020b5eSDaniel Henrique Barboza         return false;
1638a5fcf0e3SDaniel Henrique Barboza     }
163911d0f125SLuiz Capitulino 
164084d18f06SMarkus Armbruster     if (local_err) {
164177dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
16428b020b5eSDaniel Henrique Barboza     } else {
164377dbc81bSMarkus Armbruster         error_setg(errp,
16448b020b5eSDaniel Henrique Barboza                    "the helper program '%s' returned an unexpected exit"
16458b020b5eSDaniel Henrique Barboza                    " status code (%d)", "pm-is-supported", status);
1646dc8764f0SLuiz Capitulino     }
164711d0f125SLuiz Capitulino 
16488b020b5eSDaniel Henrique Barboza     return false;
1649a5fcf0e3SDaniel Henrique Barboza }
1650a5fcf0e3SDaniel Henrique Barboza 
16518b020b5eSDaniel Henrique Barboza static void pmutils_suspend(SuspendMode mode, Error **errp)
1652246d76ebSDaniel Henrique Barboza {
1653246d76ebSDaniel Henrique Barboza     Error *local_err = NULL;
16548b020b5eSDaniel Henrique Barboza     const char *pmutils_binaries[3] = {"pm-hibernate", "pm-suspend",
16558b020b5eSDaniel Henrique Barboza                                        "pm-suspend-hybrid"};
16568b020b5eSDaniel Henrique Barboza     const char *cmd[2] = {pmutils_binaries[mode], NULL};
1657246d76ebSDaniel Henrique Barboza     int status;
1658246d76ebSDaniel Henrique Barboza 
16598b020b5eSDaniel Henrique Barboza     status = run_process_child(cmd, &local_err);
1660246d76ebSDaniel Henrique Barboza 
16618b020b5eSDaniel Henrique Barboza     if (status == 0) {
1662246d76ebSDaniel Henrique Barboza         return;
1663246d76ebSDaniel Henrique Barboza     }
1664246d76ebSDaniel Henrique Barboza 
16658b020b5eSDaniel Henrique Barboza     if ((status == -1) && !local_err) {
16668b020b5eSDaniel Henrique Barboza         error_setg(errp, "the helper program '%s' was not found",
16678b020b5eSDaniel Henrique Barboza                    pmutils_binaries[mode]);
1668246d76ebSDaniel Henrique Barboza         return;
1669246d76ebSDaniel Henrique Barboza     }
1670246d76ebSDaniel Henrique Barboza 
1671246d76ebSDaniel Henrique Barboza     if (local_err) {
1672246d76ebSDaniel Henrique Barboza         error_propagate(errp, local_err);
16738b020b5eSDaniel Henrique Barboza     } else {
1674246d76ebSDaniel Henrique Barboza         error_setg(errp,
16758b020b5eSDaniel Henrique Barboza                    "the helper program '%s' returned an unexpected exit"
16768b020b5eSDaniel Henrique Barboza                    " status code (%d)", pmutils_binaries[mode], status);
16778b020b5eSDaniel Henrique Barboza     }
1678246d76ebSDaniel Henrique Barboza }
1679246d76ebSDaniel Henrique Barboza 
16808b020b5eSDaniel Henrique Barboza static bool linux_sys_state_supports_mode(SuspendMode mode, Error **errp)
1681a5fcf0e3SDaniel Henrique Barboza {
16828b020b5eSDaniel Henrique Barboza     const char *sysfile_strs[3] = {"disk", "mem", NULL};
16838b020b5eSDaniel Henrique Barboza     const char *sysfile_str = sysfile_strs[mode];
1684a5fcf0e3SDaniel Henrique Barboza     char buf[32]; /* hopefully big enough */
1685a5fcf0e3SDaniel Henrique Barboza     int fd;
1686a5fcf0e3SDaniel Henrique Barboza     ssize_t ret;
1687a5fcf0e3SDaniel Henrique Barboza 
16888b020b5eSDaniel Henrique Barboza     if (!sysfile_str) {
16898b020b5eSDaniel Henrique Barboza         error_setg(errp, "unknown guest suspend mode");
1690a5fcf0e3SDaniel Henrique Barboza         return false;
1691a5fcf0e3SDaniel Henrique Barboza     }
1692a5fcf0e3SDaniel Henrique Barboza 
1693a5fcf0e3SDaniel Henrique Barboza     fd = open(LINUX_SYS_STATE_FILE, O_RDONLY);
1694a5fcf0e3SDaniel Henrique Barboza     if (fd < 0) {
1695a5fcf0e3SDaniel Henrique Barboza         return false;
1696a5fcf0e3SDaniel Henrique Barboza     }
1697a5fcf0e3SDaniel Henrique Barboza 
1698a5fcf0e3SDaniel Henrique Barboza     ret = read(fd, buf, sizeof(buf) - 1);
1699d9c745c1SPaolo Bonzini     close(fd);
1700a5fcf0e3SDaniel Henrique Barboza     if (ret <= 0) {
1701a5fcf0e3SDaniel Henrique Barboza         return false;
1702a5fcf0e3SDaniel Henrique Barboza     }
1703a5fcf0e3SDaniel Henrique Barboza     buf[ret] = '\0';
1704a5fcf0e3SDaniel Henrique Barboza 
1705a5fcf0e3SDaniel Henrique Barboza     if (strstr(buf, sysfile_str)) {
1706a5fcf0e3SDaniel Henrique Barboza         return true;
1707a5fcf0e3SDaniel Henrique Barboza     }
1708a5fcf0e3SDaniel Henrique Barboza     return false;
1709a5fcf0e3SDaniel Henrique Barboza }
1710a5fcf0e3SDaniel Henrique Barboza 
17118b020b5eSDaniel Henrique Barboza static void linux_sys_state_suspend(SuspendMode mode, Error **errp)
1712246d76ebSDaniel Henrique Barboza {
1713246d76ebSDaniel Henrique Barboza     Error *local_err = NULL;
17148b020b5eSDaniel Henrique Barboza     const char *sysfile_strs[3] = {"disk", "mem", NULL};
17158b020b5eSDaniel Henrique Barboza     const char *sysfile_str = sysfile_strs[mode];
1716246d76ebSDaniel Henrique Barboza     pid_t pid;
1717246d76ebSDaniel Henrique Barboza     int status;
1718246d76ebSDaniel Henrique Barboza 
17198b020b5eSDaniel Henrique Barboza     if (!sysfile_str) {
1720246d76ebSDaniel Henrique Barboza         error_setg(errp, "unknown guest suspend mode");
1721246d76ebSDaniel Henrique Barboza         return;
1722246d76ebSDaniel Henrique Barboza     }
1723246d76ebSDaniel Henrique Barboza 
1724246d76ebSDaniel Henrique Barboza     pid = fork();
1725246d76ebSDaniel Henrique Barboza     if (!pid) {
1726246d76ebSDaniel Henrique Barboza         /* child */
1727246d76ebSDaniel Henrique Barboza         int fd;
1728246d76ebSDaniel Henrique Barboza 
1729246d76ebSDaniel Henrique Barboza         setsid();
1730246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(0);
1731246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(1);
1732246d76ebSDaniel Henrique Barboza         reopen_fd_to_null(2);
1733246d76ebSDaniel Henrique Barboza 
1734246d76ebSDaniel Henrique Barboza         fd = open(LINUX_SYS_STATE_FILE, O_WRONLY);
1735246d76ebSDaniel Henrique Barboza         if (fd < 0) {
1736246d76ebSDaniel Henrique Barboza             _exit(EXIT_FAILURE);
1737246d76ebSDaniel Henrique Barboza         }
1738246d76ebSDaniel Henrique Barboza 
1739246d76ebSDaniel Henrique Barboza         if (write(fd, sysfile_str, strlen(sysfile_str)) < 0) {
1740246d76ebSDaniel Henrique Barboza             _exit(EXIT_FAILURE);
1741246d76ebSDaniel Henrique Barboza         }
1742246d76ebSDaniel Henrique Barboza 
1743246d76ebSDaniel Henrique Barboza         _exit(EXIT_SUCCESS);
1744246d76ebSDaniel Henrique Barboza     } else if (pid < 0) {
1745246d76ebSDaniel Henrique Barboza         error_setg_errno(errp, errno, "failed to create child process");
1746246d76ebSDaniel Henrique Barboza         return;
1747246d76ebSDaniel Henrique Barboza     }
1748246d76ebSDaniel Henrique Barboza 
1749246d76ebSDaniel Henrique Barboza     ga_wait_child(pid, &status, &local_err);
1750246d76ebSDaniel Henrique Barboza     if (local_err) {
1751246d76ebSDaniel Henrique Barboza         error_propagate(errp, local_err);
1752246d76ebSDaniel Henrique Barboza         return;
1753246d76ebSDaniel Henrique Barboza     }
1754246d76ebSDaniel Henrique Barboza 
1755246d76ebSDaniel Henrique Barboza     if (WEXITSTATUS(status)) {
1756246d76ebSDaniel Henrique Barboza         error_setg(errp, "child process has failed to suspend");
1757246d76ebSDaniel Henrique Barboza     }
1758246d76ebSDaniel Henrique Barboza 
1759246d76ebSDaniel Henrique Barboza }
1760246d76ebSDaniel Henrique Barboza 
17618b020b5eSDaniel Henrique Barboza static void guest_suspend(SuspendMode mode, Error **errp)
176211d0f125SLuiz Capitulino {
17637b376087SLuiz Capitulino     Error *local_err = NULL;
176473e1d8ebSDaniel Henrique Barboza     bool mode_supported = false;
176511d0f125SLuiz Capitulino 
176673e1d8ebSDaniel Henrique Barboza     if (systemd_supports_mode(mode, &local_err)) {
176773e1d8ebSDaniel Henrique Barboza         mode_supported = true;
1768067927d6SDaniel Henrique Barboza         systemd_suspend(mode, &local_err);
176973e1d8ebSDaniel Henrique Barboza     }
177073e1d8ebSDaniel Henrique Barboza 
1771067927d6SDaniel Henrique Barboza     if (!local_err) {
1772067927d6SDaniel Henrique Barboza         return;
1773067927d6SDaniel Henrique Barboza     }
1774067927d6SDaniel Henrique Barboza 
1775067927d6SDaniel Henrique Barboza     error_free(local_err);
1776067927d6SDaniel Henrique Barboza 
177773e1d8ebSDaniel Henrique Barboza     if (pmutils_supports_mode(mode, &local_err)) {
177873e1d8ebSDaniel Henrique Barboza         mode_supported = true;
17798b020b5eSDaniel Henrique Barboza         pmutils_suspend(mode, &local_err);
178073e1d8ebSDaniel Henrique Barboza     }
178173e1d8ebSDaniel Henrique Barboza 
1782246d76ebSDaniel Henrique Barboza     if (!local_err) {
1783304a0fcbSDaniel Henrique Barboza         return;
1784304a0fcbSDaniel Henrique Barboza     }
1785304a0fcbSDaniel Henrique Barboza 
1786246d76ebSDaniel Henrique Barboza     error_free(local_err);
178711d0f125SLuiz Capitulino 
178873e1d8ebSDaniel Henrique Barboza     if (linux_sys_state_supports_mode(mode, &local_err)) {
178973e1d8ebSDaniel Henrique Barboza         mode_supported = true;
17908b020b5eSDaniel Henrique Barboza         linux_sys_state_suspend(mode, &local_err);
179173e1d8ebSDaniel Henrique Barboza     }
179273e1d8ebSDaniel Henrique Barboza 
179373e1d8ebSDaniel Henrique Barboza     if (!mode_supported) {
179473e1d8ebSDaniel Henrique Barboza         error_setg(errp,
179573e1d8ebSDaniel Henrique Barboza                    "the requested suspend mode is not supported by the guest");
1796b2322003SMarkus Armbruster     } else {
179777dbc81bSMarkus Armbruster         error_propagate(errp, local_err);
17987b376087SLuiz Capitulino     }
179911d0f125SLuiz Capitulino }
180011d0f125SLuiz Capitulino 
180177dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
180211d0f125SLuiz Capitulino {
1803304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_DISK, errp);
180411d0f125SLuiz Capitulino }
180511d0f125SLuiz Capitulino 
180677dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
1807fbf42210SLuiz Capitulino {
1808304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_RAM, errp);
1809fbf42210SLuiz Capitulino }
1810fbf42210SLuiz Capitulino 
181177dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
181295f4f404SLuiz Capitulino {
1813304a0fcbSDaniel Henrique Barboza     guest_suspend(SUSPEND_MODE_HYBRID, errp);
181495f4f404SLuiz Capitulino }
181595f4f404SLuiz Capitulino 
18163424fc9fSMichal Privoznik static GuestNetworkInterfaceList *
18173424fc9fSMichal Privoznik guest_find_interface(GuestNetworkInterfaceList *head,
18183424fc9fSMichal Privoznik                      const char *name)
18193424fc9fSMichal Privoznik {
18203424fc9fSMichal Privoznik     for (; head; head = head->next) {
18213424fc9fSMichal Privoznik         if (strcmp(head->value->name, name) == 0) {
18223424fc9fSMichal Privoznik             break;
18233424fc9fSMichal Privoznik         }
18243424fc9fSMichal Privoznik     }
18253424fc9fSMichal Privoznik 
18263424fc9fSMichal Privoznik     return head;
18273424fc9fSMichal Privoznik }
18283424fc9fSMichal Privoznik 
182953f9fcb2SZhiPeng Lu static int guest_get_network_stats(const char *name,
183053f9fcb2SZhiPeng Lu                        GuestNetworkInterfaceStat *stats)
183153f9fcb2SZhiPeng Lu {
183253f9fcb2SZhiPeng Lu     int name_len;
183353f9fcb2SZhiPeng Lu     char const *devinfo = "/proc/net/dev";
183453f9fcb2SZhiPeng Lu     FILE *fp;
183553f9fcb2SZhiPeng Lu     char *line = NULL, *colon;
183653f9fcb2SZhiPeng Lu     size_t n = 0;
183753f9fcb2SZhiPeng Lu     fp = fopen(devinfo, "r");
183853f9fcb2SZhiPeng Lu     if (!fp) {
183953f9fcb2SZhiPeng Lu         return -1;
184053f9fcb2SZhiPeng Lu     }
184153f9fcb2SZhiPeng Lu     name_len = strlen(name);
184253f9fcb2SZhiPeng Lu     while (getline(&line, &n, fp) != -1) {
184353f9fcb2SZhiPeng Lu         long long dummy;
184453f9fcb2SZhiPeng Lu         long long rx_bytes;
184553f9fcb2SZhiPeng Lu         long long rx_packets;
184653f9fcb2SZhiPeng Lu         long long rx_errs;
184753f9fcb2SZhiPeng Lu         long long rx_dropped;
184853f9fcb2SZhiPeng Lu         long long tx_bytes;
184953f9fcb2SZhiPeng Lu         long long tx_packets;
185053f9fcb2SZhiPeng Lu         long long tx_errs;
185153f9fcb2SZhiPeng Lu         long long tx_dropped;
185253f9fcb2SZhiPeng Lu         char *trim_line;
185353f9fcb2SZhiPeng Lu         trim_line = g_strchug(line);
185453f9fcb2SZhiPeng Lu         if (trim_line[0] == '\0') {
185553f9fcb2SZhiPeng Lu             continue;
185653f9fcb2SZhiPeng Lu         }
185753f9fcb2SZhiPeng Lu         colon = strchr(trim_line, ':');
185853f9fcb2SZhiPeng Lu         if (!colon) {
185953f9fcb2SZhiPeng Lu             continue;
186053f9fcb2SZhiPeng Lu         }
186153f9fcb2SZhiPeng Lu         if (colon - name_len  == trim_line &&
186253f9fcb2SZhiPeng Lu            strncmp(trim_line, name, name_len) == 0) {
186353f9fcb2SZhiPeng Lu             if (sscanf(colon + 1,
186453f9fcb2SZhiPeng Lu                 "%lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld %lld",
186553f9fcb2SZhiPeng Lu                   &rx_bytes, &rx_packets, &rx_errs, &rx_dropped,
186653f9fcb2SZhiPeng Lu                   &dummy, &dummy, &dummy, &dummy,
186753f9fcb2SZhiPeng Lu                   &tx_bytes, &tx_packets, &tx_errs, &tx_dropped,
186853f9fcb2SZhiPeng Lu                   &dummy, &dummy, &dummy, &dummy) != 16) {
186953f9fcb2SZhiPeng Lu                 continue;
187053f9fcb2SZhiPeng Lu             }
187153f9fcb2SZhiPeng Lu             stats->rx_bytes = rx_bytes;
187253f9fcb2SZhiPeng Lu             stats->rx_packets = rx_packets;
187353f9fcb2SZhiPeng Lu             stats->rx_errs = rx_errs;
187453f9fcb2SZhiPeng Lu             stats->rx_dropped = rx_dropped;
187553f9fcb2SZhiPeng Lu             stats->tx_bytes = tx_bytes;
187653f9fcb2SZhiPeng Lu             stats->tx_packets = tx_packets;
187753f9fcb2SZhiPeng Lu             stats->tx_errs = tx_errs;
187853f9fcb2SZhiPeng Lu             stats->tx_dropped = tx_dropped;
187953f9fcb2SZhiPeng Lu             fclose(fp);
188053f9fcb2SZhiPeng Lu             g_free(line);
188153f9fcb2SZhiPeng Lu             return 0;
188253f9fcb2SZhiPeng Lu         }
188353f9fcb2SZhiPeng Lu     }
188453f9fcb2SZhiPeng Lu     fclose(fp);
188553f9fcb2SZhiPeng Lu     g_free(line);
188653f9fcb2SZhiPeng Lu     g_debug("/proc/net/dev: Interface '%s' not found", name);
188753f9fcb2SZhiPeng Lu     return -1;
188853f9fcb2SZhiPeng Lu }
188953f9fcb2SZhiPeng Lu 
18903424fc9fSMichal Privoznik /*
18913424fc9fSMichal Privoznik  * Build information about guest interfaces
18923424fc9fSMichal Privoznik  */
18933424fc9fSMichal Privoznik GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
18943424fc9fSMichal Privoznik {
18953424fc9fSMichal Privoznik     GuestNetworkInterfaceList *head = NULL, *cur_item = NULL;
18963424fc9fSMichal Privoznik     struct ifaddrs *ifap, *ifa;
18973424fc9fSMichal Privoznik 
18983424fc9fSMichal Privoznik     if (getifaddrs(&ifap) < 0) {
1899878a0ae0SLuiz Capitulino         error_setg_errno(errp, errno, "getifaddrs failed");
19003424fc9fSMichal Privoznik         goto error;
19013424fc9fSMichal Privoznik     }
19023424fc9fSMichal Privoznik 
19033424fc9fSMichal Privoznik     for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
19043424fc9fSMichal Privoznik         GuestNetworkInterfaceList *info;
19053424fc9fSMichal Privoznik         GuestIpAddressList **address_list = NULL, *address_item = NULL;
190653f9fcb2SZhiPeng Lu         GuestNetworkInterfaceStat  *interface_stat = NULL;
19073424fc9fSMichal Privoznik         char addr4[INET_ADDRSTRLEN];
19083424fc9fSMichal Privoznik         char addr6[INET6_ADDRSTRLEN];
19093424fc9fSMichal Privoznik         int sock;
19103424fc9fSMichal Privoznik         struct ifreq ifr;
19113424fc9fSMichal Privoznik         unsigned char *mac_addr;
19123424fc9fSMichal Privoznik         void *p;
19133424fc9fSMichal Privoznik 
19143424fc9fSMichal Privoznik         g_debug("Processing %s interface", ifa->ifa_name);
19153424fc9fSMichal Privoznik 
19163424fc9fSMichal Privoznik         info = guest_find_interface(head, ifa->ifa_name);
19173424fc9fSMichal Privoznik 
19183424fc9fSMichal Privoznik         if (!info) {
19193424fc9fSMichal Privoznik             info = g_malloc0(sizeof(*info));
19203424fc9fSMichal Privoznik             info->value = g_malloc0(sizeof(*info->value));
19213424fc9fSMichal Privoznik             info->value->name = g_strdup(ifa->ifa_name);
19223424fc9fSMichal Privoznik 
19233424fc9fSMichal Privoznik             if (!cur_item) {
19243424fc9fSMichal Privoznik                 head = cur_item = info;
19253424fc9fSMichal Privoznik             } else {
19263424fc9fSMichal Privoznik                 cur_item->next = info;
19273424fc9fSMichal Privoznik                 cur_item = info;
19283424fc9fSMichal Privoznik             }
19293424fc9fSMichal Privoznik         }
19303424fc9fSMichal Privoznik 
19313424fc9fSMichal Privoznik         if (!info->value->has_hardware_address &&
19323424fc9fSMichal Privoznik             ifa->ifa_flags & SIOCGIFHWADDR) {
19333424fc9fSMichal Privoznik             /* we haven't obtained HW address yet */
19343424fc9fSMichal Privoznik             sock = socket(PF_INET, SOCK_STREAM, 0);
19353424fc9fSMichal Privoznik             if (sock == -1) {
1936878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "failed to create socket");
19373424fc9fSMichal Privoznik                 goto error;
19383424fc9fSMichal Privoznik             }
19393424fc9fSMichal Privoznik 
19403424fc9fSMichal Privoznik             memset(&ifr, 0, sizeof(ifr));
19411ab516edSJim Meyering             pstrcpy(ifr.ifr_name, IF_NAMESIZE, info->value->name);
19423424fc9fSMichal Privoznik             if (ioctl(sock, SIOCGIFHWADDR, &ifr) == -1) {
1943878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno,
1944878a0ae0SLuiz Capitulino                                  "failed to get MAC address of %s",
1945878a0ae0SLuiz Capitulino                                  ifa->ifa_name);
194610a2158fSMarkus Armbruster                 close(sock);
19473424fc9fSMichal Privoznik                 goto error;
19483424fc9fSMichal Privoznik             }
19493424fc9fSMichal Privoznik 
195010a2158fSMarkus Armbruster             close(sock);
19513424fc9fSMichal Privoznik             mac_addr = (unsigned char *) &ifr.ifr_hwaddr.sa_data;
19523424fc9fSMichal Privoznik 
1953e4ada482SStefan Weil             info->value->hardware_address =
1954e4ada482SStefan Weil                 g_strdup_printf("%02x:%02x:%02x:%02x:%02x:%02x",
19553424fc9fSMichal Privoznik                                 (int) mac_addr[0], (int) mac_addr[1],
19563424fc9fSMichal Privoznik                                 (int) mac_addr[2], (int) mac_addr[3],
1957e4ada482SStefan Weil                                 (int) mac_addr[4], (int) mac_addr[5]);
19583424fc9fSMichal Privoznik 
19593424fc9fSMichal Privoznik             info->value->has_hardware_address = true;
19603424fc9fSMichal Privoznik         }
19613424fc9fSMichal Privoznik 
19623424fc9fSMichal Privoznik         if (ifa->ifa_addr &&
19633424fc9fSMichal Privoznik             ifa->ifa_addr->sa_family == AF_INET) {
19643424fc9fSMichal Privoznik             /* interface with IPv4 address */
19653424fc9fSMichal Privoznik             p = &((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
19663424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET, p, addr4, sizeof(addr4))) {
1967878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
19683424fc9fSMichal Privoznik                 goto error;
19693424fc9fSMichal Privoznik             }
19703424fc9fSMichal Privoznik 
197110a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
197210a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
19733424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr4);
19743424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV4;
19753424fc9fSMichal Privoznik 
19763424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
19773424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
19783424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
19793424fc9fSMichal Privoznik                 p = &((struct sockaddr_in *)ifa->ifa_netmask)->sin_addr;
19803424fc9fSMichal Privoznik                 address_item->value->prefix = ctpop32(((uint32_t *) p)[0]);
19813424fc9fSMichal Privoznik             }
19823424fc9fSMichal Privoznik         } else if (ifa->ifa_addr &&
19833424fc9fSMichal Privoznik                    ifa->ifa_addr->sa_family == AF_INET6) {
19843424fc9fSMichal Privoznik             /* interface with IPv6 address */
19853424fc9fSMichal Privoznik             p = &((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
19863424fc9fSMichal Privoznik             if (!inet_ntop(AF_INET6, p, addr6, sizeof(addr6))) {
1987878a0ae0SLuiz Capitulino                 error_setg_errno(errp, errno, "inet_ntop failed");
19883424fc9fSMichal Privoznik                 goto error;
19893424fc9fSMichal Privoznik             }
19903424fc9fSMichal Privoznik 
199110a2158fSMarkus Armbruster             address_item = g_malloc0(sizeof(*address_item));
199210a2158fSMarkus Armbruster             address_item->value = g_malloc0(sizeof(*address_item->value));
19933424fc9fSMichal Privoznik             address_item->value->ip_address = g_strdup(addr6);
19943424fc9fSMichal Privoznik             address_item->value->ip_address_type = GUEST_IP_ADDRESS_TYPE_IPV6;
19953424fc9fSMichal Privoznik 
19963424fc9fSMichal Privoznik             if (ifa->ifa_netmask) {
19973424fc9fSMichal Privoznik                 /* Count the number of set bits in netmask.
19983424fc9fSMichal Privoznik                  * This is safe as '1' and '0' cannot be shuffled in netmask. */
19993424fc9fSMichal Privoznik                 p = &((struct sockaddr_in6 *)ifa->ifa_netmask)->sin6_addr;
20003424fc9fSMichal Privoznik                 address_item->value->prefix =
20013424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[0]) +
20023424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[1]) +
20033424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[2]) +
20043424fc9fSMichal Privoznik                     ctpop32(((uint32_t *) p)[3]);
20053424fc9fSMichal Privoznik             }
20063424fc9fSMichal Privoznik         }
20073424fc9fSMichal Privoznik 
20083424fc9fSMichal Privoznik         if (!address_item) {
20093424fc9fSMichal Privoznik             continue;
20103424fc9fSMichal Privoznik         }
20113424fc9fSMichal Privoznik 
20123424fc9fSMichal Privoznik         address_list = &info->value->ip_addresses;
20133424fc9fSMichal Privoznik 
20143424fc9fSMichal Privoznik         while (*address_list && (*address_list)->next) {
20153424fc9fSMichal Privoznik             address_list = &(*address_list)->next;
20163424fc9fSMichal Privoznik         }
20173424fc9fSMichal Privoznik 
20183424fc9fSMichal Privoznik         if (!*address_list) {
20193424fc9fSMichal Privoznik             *address_list = address_item;
20203424fc9fSMichal Privoznik         } else {
20213424fc9fSMichal Privoznik             (*address_list)->next = address_item;
20223424fc9fSMichal Privoznik         }
20233424fc9fSMichal Privoznik 
20243424fc9fSMichal Privoznik         info->value->has_ip_addresses = true;
20253424fc9fSMichal Privoznik 
202653f9fcb2SZhiPeng Lu         if (!info->value->has_statistics) {
202753f9fcb2SZhiPeng Lu             interface_stat = g_malloc0(sizeof(*interface_stat));
202853f9fcb2SZhiPeng Lu             if (guest_get_network_stats(info->value->name,
202953f9fcb2SZhiPeng Lu                 interface_stat) == -1) {
203053f9fcb2SZhiPeng Lu                 info->value->has_statistics = false;
203153f9fcb2SZhiPeng Lu                 g_free(interface_stat);
203253f9fcb2SZhiPeng Lu             } else {
203353f9fcb2SZhiPeng Lu                 info->value->statistics = interface_stat;
203453f9fcb2SZhiPeng Lu                 info->value->has_statistics = true;
203553f9fcb2SZhiPeng Lu             }
203653f9fcb2SZhiPeng Lu         }
20373424fc9fSMichal Privoznik     }
20383424fc9fSMichal Privoznik 
20393424fc9fSMichal Privoznik     freeifaddrs(ifap);
20403424fc9fSMichal Privoznik     return head;
20413424fc9fSMichal Privoznik 
20423424fc9fSMichal Privoznik error:
20433424fc9fSMichal Privoznik     freeifaddrs(ifap);
20443424fc9fSMichal Privoznik     qapi_free_GuestNetworkInterfaceList(head);
20453424fc9fSMichal Privoznik     return NULL;
20463424fc9fSMichal Privoznik }
20473424fc9fSMichal Privoznik 
204877dbc81bSMarkus Armbruster #define SYSCONF_EXACT(name, errp) sysconf_exact((name), #name, (errp))
2049d2baff62SLaszlo Ersek 
205077dbc81bSMarkus Armbruster static long sysconf_exact(int name, const char *name_str, Error **errp)
2051d2baff62SLaszlo Ersek {
2052d2baff62SLaszlo Ersek     long ret;
2053d2baff62SLaszlo Ersek 
2054d2baff62SLaszlo Ersek     errno = 0;
2055d2baff62SLaszlo Ersek     ret = sysconf(name);
2056d2baff62SLaszlo Ersek     if (ret == -1) {
2057d2baff62SLaszlo Ersek         if (errno == 0) {
205877dbc81bSMarkus Armbruster             error_setg(errp, "sysconf(%s): value indefinite", name_str);
2059d2baff62SLaszlo Ersek         } else {
206077dbc81bSMarkus Armbruster             error_setg_errno(errp, errno, "sysconf(%s)", name_str);
2061d2baff62SLaszlo Ersek         }
2062d2baff62SLaszlo Ersek     }
2063d2baff62SLaszlo Ersek     return ret;
2064d2baff62SLaszlo Ersek }
2065d2baff62SLaszlo Ersek 
2066d2baff62SLaszlo Ersek /* Transfer online/offline status between @vcpu and the guest system.
2067d2baff62SLaszlo Ersek  *
2068d2baff62SLaszlo Ersek  * On input either @errp or *@errp must be NULL.
2069d2baff62SLaszlo Ersek  *
2070d2baff62SLaszlo Ersek  * In system-to-@vcpu direction, the following @vcpu fields are accessed:
2071d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
2072d2baff62SLaszlo Ersek  * - W: vcpu->online
2073d2baff62SLaszlo Ersek  * - W: vcpu->can_offline
2074d2baff62SLaszlo Ersek  *
2075d2baff62SLaszlo Ersek  * In @vcpu-to-system direction, the following @vcpu fields are accessed:
2076d2baff62SLaszlo Ersek  * - R: vcpu->logical_id
2077d2baff62SLaszlo Ersek  * - R: vcpu->online
2078d2baff62SLaszlo Ersek  *
2079d2baff62SLaszlo Ersek  * Written members remain unmodified on error.
2080d2baff62SLaszlo Ersek  */
2081d2baff62SLaszlo Ersek static void transfer_vcpu(GuestLogicalProcessor *vcpu, bool sys2vcpu,
2082b4bf912aSIgor Mammedov                           char *dirpath, Error **errp)
2083d2baff62SLaszlo Ersek {
2084b4bf912aSIgor Mammedov     int fd;
2085b4bf912aSIgor Mammedov     int res;
2086d2baff62SLaszlo Ersek     int dirfd;
2087b4bf912aSIgor Mammedov     static const char fn[] = "online";
2088d2baff62SLaszlo Ersek 
2089d2baff62SLaszlo Ersek     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2090d2baff62SLaszlo Ersek     if (dirfd == -1) {
2091d2baff62SLaszlo Ersek         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2092b4bf912aSIgor Mammedov         return;
2093b4bf912aSIgor Mammedov     }
2094d2baff62SLaszlo Ersek 
2095d2baff62SLaszlo Ersek     fd = openat(dirfd, fn, sys2vcpu ? O_RDONLY : O_RDWR);
2096d2baff62SLaszlo Ersek     if (fd == -1) {
2097d2baff62SLaszlo Ersek         if (errno != ENOENT) {
2098d2baff62SLaszlo Ersek             error_setg_errno(errp, errno, "open(\"%s/%s\")", dirpath, fn);
2099d2baff62SLaszlo Ersek         } else if (sys2vcpu) {
2100d2baff62SLaszlo Ersek             vcpu->online = true;
2101d2baff62SLaszlo Ersek             vcpu->can_offline = false;
2102d2baff62SLaszlo Ersek         } else if (!vcpu->online) {
2103d2baff62SLaszlo Ersek             error_setg(errp, "logical processor #%" PRId64 " can't be "
2104d2baff62SLaszlo Ersek                        "offlined", vcpu->logical_id);
2105d2baff62SLaszlo Ersek         } /* otherwise pretend successful re-onlining */
2106d2baff62SLaszlo Ersek     } else {
2107d2baff62SLaszlo Ersek         unsigned char status;
2108d2baff62SLaszlo Ersek 
2109d2baff62SLaszlo Ersek         res = pread(fd, &status, 1, 0);
2110d2baff62SLaszlo Ersek         if (res == -1) {
2111d2baff62SLaszlo Ersek             error_setg_errno(errp, errno, "pread(\"%s/%s\")", dirpath, fn);
2112d2baff62SLaszlo Ersek         } else if (res == 0) {
2113d2baff62SLaszlo Ersek             error_setg(errp, "pread(\"%s/%s\"): unexpected EOF", dirpath,
2114d2baff62SLaszlo Ersek                        fn);
2115d2baff62SLaszlo Ersek         } else if (sys2vcpu) {
2116d2baff62SLaszlo Ersek             vcpu->online = (status != '0');
2117d2baff62SLaszlo Ersek             vcpu->can_offline = true;
2118d2baff62SLaszlo Ersek         } else if (vcpu->online != (status != '0')) {
2119d2baff62SLaszlo Ersek             status = '0' + vcpu->online;
2120d2baff62SLaszlo Ersek             if (pwrite(fd, &status, 1, 0) == -1) {
2121d2baff62SLaszlo Ersek                 error_setg_errno(errp, errno, "pwrite(\"%s/%s\")", dirpath,
2122d2baff62SLaszlo Ersek                                  fn);
2123d2baff62SLaszlo Ersek             }
2124d2baff62SLaszlo Ersek         } /* otherwise pretend successful re-(on|off)-lining */
2125d2baff62SLaszlo Ersek 
2126d2baff62SLaszlo Ersek         res = close(fd);
2127d2baff62SLaszlo Ersek         g_assert(res == 0);
2128d2baff62SLaszlo Ersek     }
2129d2baff62SLaszlo Ersek 
2130d2baff62SLaszlo Ersek     res = close(dirfd);
2131d2baff62SLaszlo Ersek     g_assert(res == 0);
2132d2baff62SLaszlo Ersek }
2133d2baff62SLaszlo Ersek 
2134d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2135d2baff62SLaszlo Ersek {
2136d2baff62SLaszlo Ersek     int64_t current;
2137d2baff62SLaszlo Ersek     GuestLogicalProcessorList *head, **link;
2138d2baff62SLaszlo Ersek     long sc_max;
2139d2baff62SLaszlo Ersek     Error *local_err = NULL;
2140d2baff62SLaszlo Ersek 
2141d2baff62SLaszlo Ersek     current = 0;
2142d2baff62SLaszlo Ersek     head = NULL;
2143d2baff62SLaszlo Ersek     link = &head;
2144d2baff62SLaszlo Ersek     sc_max = SYSCONF_EXACT(_SC_NPROCESSORS_CONF, &local_err);
2145d2baff62SLaszlo Ersek 
2146d2baff62SLaszlo Ersek     while (local_err == NULL && current < sc_max) {
2147d2baff62SLaszlo Ersek         GuestLogicalProcessor *vcpu;
2148d2baff62SLaszlo Ersek         GuestLogicalProcessorList *entry;
2149b4bf912aSIgor Mammedov         int64_t id = current++;
2150b4bf912aSIgor Mammedov         char *path = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
2151b4bf912aSIgor Mammedov                                      id);
2152d2baff62SLaszlo Ersek 
2153b4bf912aSIgor Mammedov         if (g_file_test(path, G_FILE_TEST_EXISTS)) {
2154d2baff62SLaszlo Ersek             vcpu = g_malloc0(sizeof *vcpu);
2155b4bf912aSIgor Mammedov             vcpu->logical_id = id;
2156d2baff62SLaszlo Ersek             vcpu->has_can_offline = true; /* lolspeak ftw */
2157b4bf912aSIgor Mammedov             transfer_vcpu(vcpu, true, path, &local_err);
2158d2baff62SLaszlo Ersek             entry = g_malloc0(sizeof *entry);
2159d2baff62SLaszlo Ersek             entry->value = vcpu;
2160d2baff62SLaszlo Ersek             *link = entry;
2161d2baff62SLaszlo Ersek             link = &entry->next;
2162d2baff62SLaszlo Ersek         }
2163b4bf912aSIgor Mammedov         g_free(path);
2164b4bf912aSIgor Mammedov     }
2165d2baff62SLaszlo Ersek 
2166d2baff62SLaszlo Ersek     if (local_err == NULL) {
2167d2baff62SLaszlo Ersek         /* there's no guest with zero VCPUs */
2168d2baff62SLaszlo Ersek         g_assert(head != NULL);
2169d2baff62SLaszlo Ersek         return head;
2170d2baff62SLaszlo Ersek     }
2171d2baff62SLaszlo Ersek 
2172d2baff62SLaszlo Ersek     qapi_free_GuestLogicalProcessorList(head);
2173d2baff62SLaszlo Ersek     error_propagate(errp, local_err);
2174d2baff62SLaszlo Ersek     return NULL;
2175d2baff62SLaszlo Ersek }
2176d2baff62SLaszlo Ersek 
2177cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2178cbb65fc2SLaszlo Ersek {
2179cbb65fc2SLaszlo Ersek     int64_t processed;
2180cbb65fc2SLaszlo Ersek     Error *local_err = NULL;
2181cbb65fc2SLaszlo Ersek 
2182cbb65fc2SLaszlo Ersek     processed = 0;
2183cbb65fc2SLaszlo Ersek     while (vcpus != NULL) {
2184b4bf912aSIgor Mammedov         char *path = g_strdup_printf("/sys/devices/system/cpu/cpu%" PRId64 "/",
2185b4bf912aSIgor Mammedov                                      vcpus->value->logical_id);
2186b4bf912aSIgor Mammedov 
2187b4bf912aSIgor Mammedov         transfer_vcpu(vcpus->value, false, path, &local_err);
2188b4bf912aSIgor Mammedov         g_free(path);
2189cbb65fc2SLaszlo Ersek         if (local_err != NULL) {
2190cbb65fc2SLaszlo Ersek             break;
2191cbb65fc2SLaszlo Ersek         }
2192cbb65fc2SLaszlo Ersek         ++processed;
2193cbb65fc2SLaszlo Ersek         vcpus = vcpus->next;
2194cbb65fc2SLaszlo Ersek     }
2195cbb65fc2SLaszlo Ersek 
2196cbb65fc2SLaszlo Ersek     if (local_err != NULL) {
2197cbb65fc2SLaszlo Ersek         if (processed == 0) {
2198cbb65fc2SLaszlo Ersek             error_propagate(errp, local_err);
2199cbb65fc2SLaszlo Ersek         } else {
2200cbb65fc2SLaszlo Ersek             error_free(local_err);
2201cbb65fc2SLaszlo Ersek         }
2202cbb65fc2SLaszlo Ersek     }
2203cbb65fc2SLaszlo Ersek 
2204cbb65fc2SLaszlo Ersek     return processed;
2205cbb65fc2SLaszlo Ersek }
2206cbb65fc2SLaszlo Ersek 
2207215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2208215a2771SDaniel P. Berrange                                  const char *password,
2209215a2771SDaniel P. Berrange                                  bool crypted,
2210215a2771SDaniel P. Berrange                                  Error **errp)
2211215a2771SDaniel P. Berrange {
2212215a2771SDaniel P. Berrange     Error *local_err = NULL;
2213215a2771SDaniel P. Berrange     char *passwd_path = NULL;
2214215a2771SDaniel P. Berrange     pid_t pid;
2215215a2771SDaniel P. Berrange     int status;
2216215a2771SDaniel P. Berrange     int datafd[2] = { -1, -1 };
2217215a2771SDaniel P. Berrange     char *rawpasswddata = NULL;
2218215a2771SDaniel P. Berrange     size_t rawpasswdlen;
2219215a2771SDaniel P. Berrange     char *chpasswddata = NULL;
2220215a2771SDaniel P. Berrange     size_t chpasswdlen;
2221215a2771SDaniel P. Berrange 
2222920639caSDaniel P. Berrange     rawpasswddata = (char *)qbase64_decode(password, -1, &rawpasswdlen, errp);
2223920639caSDaniel P. Berrange     if (!rawpasswddata) {
2224920639caSDaniel P. Berrange         return;
2225920639caSDaniel P. Berrange     }
2226215a2771SDaniel P. Berrange     rawpasswddata = g_renew(char, rawpasswddata, rawpasswdlen + 1);
2227215a2771SDaniel P. Berrange     rawpasswddata[rawpasswdlen] = '\0';
2228215a2771SDaniel P. Berrange 
2229215a2771SDaniel P. Berrange     if (strchr(rawpasswddata, '\n')) {
2230215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in raw password");
2231215a2771SDaniel P. Berrange         goto out;
2232215a2771SDaniel P. Berrange     }
2233215a2771SDaniel P. Berrange 
2234215a2771SDaniel P. Berrange     if (strchr(username, '\n') ||
2235215a2771SDaniel P. Berrange         strchr(username, ':')) {
2236215a2771SDaniel P. Berrange         error_setg(errp, "forbidden characters in username");
2237215a2771SDaniel P. Berrange         goto out;
2238215a2771SDaniel P. Berrange     }
2239215a2771SDaniel P. Berrange 
2240215a2771SDaniel P. Berrange     chpasswddata = g_strdup_printf("%s:%s\n", username, rawpasswddata);
2241215a2771SDaniel P. Berrange     chpasswdlen = strlen(chpasswddata);
2242215a2771SDaniel P. Berrange 
2243215a2771SDaniel P. Berrange     passwd_path = g_find_program_in_path("chpasswd");
2244215a2771SDaniel P. Berrange 
2245215a2771SDaniel P. Berrange     if (!passwd_path) {
2246215a2771SDaniel P. Berrange         error_setg(errp, "cannot find 'passwd' program in PATH");
2247215a2771SDaniel P. Berrange         goto out;
2248215a2771SDaniel P. Berrange     }
2249215a2771SDaniel P. Berrange 
2250215a2771SDaniel P. Berrange     if (pipe(datafd) < 0) {
2251215a2771SDaniel P. Berrange         error_setg(errp, "cannot create pipe FDs");
2252215a2771SDaniel P. Berrange         goto out;
2253215a2771SDaniel P. Berrange     }
2254215a2771SDaniel P. Berrange 
2255215a2771SDaniel P. Berrange     pid = fork();
2256215a2771SDaniel P. Berrange     if (pid == 0) {
2257215a2771SDaniel P. Berrange         close(datafd[1]);
2258215a2771SDaniel P. Berrange         /* child */
2259215a2771SDaniel P. Berrange         setsid();
2260215a2771SDaniel P. Berrange         dup2(datafd[0], 0);
2261215a2771SDaniel P. Berrange         reopen_fd_to_null(1);
2262215a2771SDaniel P. Berrange         reopen_fd_to_null(2);
2263215a2771SDaniel P. Berrange 
2264215a2771SDaniel P. Berrange         if (crypted) {
2265215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", "-e", NULL, environ);
2266215a2771SDaniel P. Berrange         } else {
2267215a2771SDaniel P. Berrange             execle(passwd_path, "chpasswd", NULL, environ);
2268215a2771SDaniel P. Berrange         }
2269215a2771SDaniel P. Berrange         _exit(EXIT_FAILURE);
2270215a2771SDaniel P. Berrange     } else if (pid < 0) {
2271215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "failed to create child process");
2272215a2771SDaniel P. Berrange         goto out;
2273215a2771SDaniel P. Berrange     }
2274215a2771SDaniel P. Berrange     close(datafd[0]);
2275215a2771SDaniel P. Berrange     datafd[0] = -1;
2276215a2771SDaniel P. Berrange 
2277215a2771SDaniel P. Berrange     if (qemu_write_full(datafd[1], chpasswddata, chpasswdlen) != chpasswdlen) {
2278215a2771SDaniel P. Berrange         error_setg_errno(errp, errno, "cannot write new account password");
2279215a2771SDaniel P. Berrange         goto out;
2280215a2771SDaniel P. Berrange     }
2281215a2771SDaniel P. Berrange     close(datafd[1]);
2282215a2771SDaniel P. Berrange     datafd[1] = -1;
2283215a2771SDaniel P. Berrange 
2284215a2771SDaniel P. Berrange     ga_wait_child(pid, &status, &local_err);
2285215a2771SDaniel P. Berrange     if (local_err) {
2286215a2771SDaniel P. Berrange         error_propagate(errp, local_err);
2287215a2771SDaniel P. Berrange         goto out;
2288215a2771SDaniel P. Berrange     }
2289215a2771SDaniel P. Berrange 
2290215a2771SDaniel P. Berrange     if (!WIFEXITED(status)) {
2291215a2771SDaniel P. Berrange         error_setg(errp, "child process has terminated abnormally");
2292215a2771SDaniel P. Berrange         goto out;
2293215a2771SDaniel P. Berrange     }
2294215a2771SDaniel P. Berrange 
2295215a2771SDaniel P. Berrange     if (WEXITSTATUS(status)) {
2296215a2771SDaniel P. Berrange         error_setg(errp, "child process has failed to set user password");
2297215a2771SDaniel P. Berrange         goto out;
2298215a2771SDaniel P. Berrange     }
2299215a2771SDaniel P. Berrange 
2300215a2771SDaniel P. Berrange out:
2301215a2771SDaniel P. Berrange     g_free(chpasswddata);
2302215a2771SDaniel P. Berrange     g_free(rawpasswddata);
2303215a2771SDaniel P. Berrange     g_free(passwd_path);
2304215a2771SDaniel P. Berrange     if (datafd[0] != -1) {
2305215a2771SDaniel P. Berrange         close(datafd[0]);
2306215a2771SDaniel P. Berrange     }
2307215a2771SDaniel P. Berrange     if (datafd[1] != -1) {
2308215a2771SDaniel P. Berrange         close(datafd[1]);
2309215a2771SDaniel P. Berrange     }
2310215a2771SDaniel P. Berrange }
2311215a2771SDaniel P. Berrange 
2312bd240fcaSzhanghailiang static void ga_read_sysfs_file(int dirfd, const char *pathname, char *buf,
2313bd240fcaSzhanghailiang                                int size, Error **errp)
2314bd240fcaSzhanghailiang {
2315bd240fcaSzhanghailiang     int fd;
2316bd240fcaSzhanghailiang     int res;
2317bd240fcaSzhanghailiang 
2318bd240fcaSzhanghailiang     errno = 0;
2319bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_RDONLY);
2320bd240fcaSzhanghailiang     if (fd == -1) {
2321bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2322bd240fcaSzhanghailiang         return;
2323bd240fcaSzhanghailiang     }
2324bd240fcaSzhanghailiang 
2325bd240fcaSzhanghailiang     res = pread(fd, buf, size, 0);
2326bd240fcaSzhanghailiang     if (res == -1) {
2327bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pread sysfs file \"%s\"", pathname);
2328bd240fcaSzhanghailiang     } else if (res == 0) {
2329bd240fcaSzhanghailiang         error_setg(errp, "pread sysfs file \"%s\": unexpected EOF", pathname);
2330bd240fcaSzhanghailiang     }
2331bd240fcaSzhanghailiang     close(fd);
2332bd240fcaSzhanghailiang }
2333bd240fcaSzhanghailiang 
2334bd240fcaSzhanghailiang static void ga_write_sysfs_file(int dirfd, const char *pathname,
2335bd240fcaSzhanghailiang                                 const char *buf, int size, Error **errp)
2336bd240fcaSzhanghailiang {
2337bd240fcaSzhanghailiang     int fd;
2338bd240fcaSzhanghailiang 
2339bd240fcaSzhanghailiang     errno = 0;
2340bd240fcaSzhanghailiang     fd = openat(dirfd, pathname, O_WRONLY);
2341bd240fcaSzhanghailiang     if (fd == -1) {
2342bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "open sysfs file \"%s\"", pathname);
2343bd240fcaSzhanghailiang         return;
2344bd240fcaSzhanghailiang     }
2345bd240fcaSzhanghailiang 
2346bd240fcaSzhanghailiang     if (pwrite(fd, buf, size, 0) == -1) {
2347bd240fcaSzhanghailiang         error_setg_errno(errp, errno, "pwrite sysfs file \"%s\"", pathname);
2348bd240fcaSzhanghailiang     }
2349bd240fcaSzhanghailiang 
2350bd240fcaSzhanghailiang     close(fd);
2351bd240fcaSzhanghailiang }
2352bd240fcaSzhanghailiang 
2353bd240fcaSzhanghailiang /* Transfer online/offline status between @mem_blk and the guest system.
2354bd240fcaSzhanghailiang  *
2355bd240fcaSzhanghailiang  * On input either @errp or *@errp must be NULL.
2356bd240fcaSzhanghailiang  *
2357bd240fcaSzhanghailiang  * In system-to-@mem_blk direction, the following @mem_blk fields are accessed:
2358bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2359bd240fcaSzhanghailiang  * - W: mem_blk->online
2360bd240fcaSzhanghailiang  * - W: mem_blk->can_offline
2361bd240fcaSzhanghailiang  *
2362bd240fcaSzhanghailiang  * In @mem_blk-to-system direction, the following @mem_blk fields are accessed:
2363bd240fcaSzhanghailiang  * - R: mem_blk->phys_index
2364bd240fcaSzhanghailiang  * - R: mem_blk->online
2365bd240fcaSzhanghailiang  *-  R: mem_blk->can_offline
2366bd240fcaSzhanghailiang  * Written members remain unmodified on error.
2367bd240fcaSzhanghailiang  */
2368bd240fcaSzhanghailiang static void transfer_memory_block(GuestMemoryBlock *mem_blk, bool sys2memblk,
2369bd240fcaSzhanghailiang                                   GuestMemoryBlockResponse *result,
2370bd240fcaSzhanghailiang                                   Error **errp)
2371bd240fcaSzhanghailiang {
2372bd240fcaSzhanghailiang     char *dirpath;
2373bd240fcaSzhanghailiang     int dirfd;
2374bd240fcaSzhanghailiang     char *status;
2375bd240fcaSzhanghailiang     Error *local_err = NULL;
2376bd240fcaSzhanghailiang 
2377bd240fcaSzhanghailiang     if (!sys2memblk) {
2378bd240fcaSzhanghailiang         DIR *dp;
2379bd240fcaSzhanghailiang 
2380bd240fcaSzhanghailiang         if (!result) {
2381bd240fcaSzhanghailiang             error_setg(errp, "Internal error, 'result' should not be NULL");
2382bd240fcaSzhanghailiang             return;
2383bd240fcaSzhanghailiang         }
2384bd240fcaSzhanghailiang         errno = 0;
2385bd240fcaSzhanghailiang         dp = opendir("/sys/devices/system/memory/");
2386bd240fcaSzhanghailiang          /* if there is no 'memory' directory in sysfs,
2387bd240fcaSzhanghailiang          * we think this VM does not support online/offline memory block,
2388bd240fcaSzhanghailiang          * any other solution?
2389bd240fcaSzhanghailiang          */
23909879f5acSPhilippe Mathieu-Daudé         if (!dp) {
23919879f5acSPhilippe Mathieu-Daudé             if (errno == ENOENT) {
2392bd240fcaSzhanghailiang                 result->response =
2393bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
23949879f5acSPhilippe Mathieu-Daudé             }
2395bd240fcaSzhanghailiang             goto out1;
2396bd240fcaSzhanghailiang         }
2397bd240fcaSzhanghailiang         closedir(dp);
2398bd240fcaSzhanghailiang     }
2399bd240fcaSzhanghailiang 
2400bd240fcaSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/memory%" PRId64 "/",
2401bd240fcaSzhanghailiang                               mem_blk->phys_index);
2402bd240fcaSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2403bd240fcaSzhanghailiang     if (dirfd == -1) {
2404bd240fcaSzhanghailiang         if (sys2memblk) {
2405bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2406bd240fcaSzhanghailiang         } else {
2407bd240fcaSzhanghailiang             if (errno == ENOENT) {
2408bd240fcaSzhanghailiang                 result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_NOT_FOUND;
2409bd240fcaSzhanghailiang             } else {
2410bd240fcaSzhanghailiang                 result->response =
2411bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2412bd240fcaSzhanghailiang             }
2413bd240fcaSzhanghailiang         }
2414bd240fcaSzhanghailiang         g_free(dirpath);
2415bd240fcaSzhanghailiang         goto out1;
2416bd240fcaSzhanghailiang     }
2417bd240fcaSzhanghailiang     g_free(dirpath);
2418bd240fcaSzhanghailiang 
2419bd240fcaSzhanghailiang     status = g_malloc0(10);
2420bd240fcaSzhanghailiang     ga_read_sysfs_file(dirfd, "state", status, 10, &local_err);
2421bd240fcaSzhanghailiang     if (local_err) {
2422bd240fcaSzhanghailiang         /* treat with sysfs file that not exist in old kernel */
2423bd240fcaSzhanghailiang         if (errno == ENOENT) {
2424bd240fcaSzhanghailiang             error_free(local_err);
2425bd240fcaSzhanghailiang             if (sys2memblk) {
2426bd240fcaSzhanghailiang                 mem_blk->online = true;
2427bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2428bd240fcaSzhanghailiang             } else if (!mem_blk->online) {
2429bd240fcaSzhanghailiang                 result->response =
2430bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_NOT_SUPPORTED;
2431bd240fcaSzhanghailiang             }
2432bd240fcaSzhanghailiang         } else {
2433bd240fcaSzhanghailiang             if (sys2memblk) {
2434bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2435bd240fcaSzhanghailiang             } else {
2436bd240fcaSzhanghailiang                 result->response =
2437bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2438bd240fcaSzhanghailiang             }
2439bd240fcaSzhanghailiang         }
2440bd240fcaSzhanghailiang         goto out2;
2441bd240fcaSzhanghailiang     }
2442bd240fcaSzhanghailiang 
2443bd240fcaSzhanghailiang     if (sys2memblk) {
2444bd240fcaSzhanghailiang         char removable = '0';
2445bd240fcaSzhanghailiang 
2446bd240fcaSzhanghailiang         mem_blk->online = (strncmp(status, "online", 6) == 0);
2447bd240fcaSzhanghailiang 
2448bd240fcaSzhanghailiang         ga_read_sysfs_file(dirfd, "removable", &removable, 1, &local_err);
2449bd240fcaSzhanghailiang         if (local_err) {
245067cc32ebSVeres Lajos             /* if no 'removable' file, it doesn't support offline mem blk */
2451bd240fcaSzhanghailiang             if (errno == ENOENT) {
2452bd240fcaSzhanghailiang                 error_free(local_err);
2453bd240fcaSzhanghailiang                 mem_blk->can_offline = false;
2454bd240fcaSzhanghailiang             } else {
2455bd240fcaSzhanghailiang                 error_propagate(errp, local_err);
2456bd240fcaSzhanghailiang             }
2457bd240fcaSzhanghailiang         } else {
2458bd240fcaSzhanghailiang             mem_blk->can_offline = (removable != '0');
2459bd240fcaSzhanghailiang         }
2460bd240fcaSzhanghailiang     } else {
2461bd240fcaSzhanghailiang         if (mem_blk->online != (strncmp(status, "online", 6) == 0)) {
24627064024dSMarc-André Lureau             const char *new_state = mem_blk->online ? "online" : "offline";
2463bd240fcaSzhanghailiang 
2464bd240fcaSzhanghailiang             ga_write_sysfs_file(dirfd, "state", new_state, strlen(new_state),
2465bd240fcaSzhanghailiang                                 &local_err);
2466bd240fcaSzhanghailiang             if (local_err) {
2467bd240fcaSzhanghailiang                 error_free(local_err);
2468bd240fcaSzhanghailiang                 result->response =
2469bd240fcaSzhanghailiang                     GUEST_MEMORY_BLOCK_RESPONSE_TYPE_OPERATION_FAILED;
2470bd240fcaSzhanghailiang                 goto out2;
2471bd240fcaSzhanghailiang             }
2472bd240fcaSzhanghailiang 
2473bd240fcaSzhanghailiang             result->response = GUEST_MEMORY_BLOCK_RESPONSE_TYPE_SUCCESS;
2474bd240fcaSzhanghailiang             result->has_error_code = false;
2475bd240fcaSzhanghailiang         } /* otherwise pretend successful re-(on|off)-lining */
2476bd240fcaSzhanghailiang     }
2477bd240fcaSzhanghailiang     g_free(status);
2478bd240fcaSzhanghailiang     close(dirfd);
2479bd240fcaSzhanghailiang     return;
2480bd240fcaSzhanghailiang 
2481bd240fcaSzhanghailiang out2:
2482bd240fcaSzhanghailiang     g_free(status);
2483bd240fcaSzhanghailiang     close(dirfd);
2484bd240fcaSzhanghailiang out1:
2485bd240fcaSzhanghailiang     if (!sys2memblk) {
2486bd240fcaSzhanghailiang         result->has_error_code = true;
2487bd240fcaSzhanghailiang         result->error_code = errno;
2488bd240fcaSzhanghailiang     }
2489bd240fcaSzhanghailiang }
2490bd240fcaSzhanghailiang 
2491a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2492a065aaa9Szhanghailiang {
2493bd240fcaSzhanghailiang     GuestMemoryBlockList *head, **link;
2494bd240fcaSzhanghailiang     Error *local_err = NULL;
2495bd240fcaSzhanghailiang     struct dirent *de;
2496bd240fcaSzhanghailiang     DIR *dp;
2497bd240fcaSzhanghailiang 
2498bd240fcaSzhanghailiang     head = NULL;
2499bd240fcaSzhanghailiang     link = &head;
2500bd240fcaSzhanghailiang 
2501bd240fcaSzhanghailiang     dp = opendir("/sys/devices/system/memory/");
2502bd240fcaSzhanghailiang     if (!dp) {
2503f693fe6eSMichael Roth         /* it's ok if this happens to be a system that doesn't expose
2504f693fe6eSMichael Roth          * memory blocks via sysfs, but otherwise we should report
2505f693fe6eSMichael Roth          * an error
2506f693fe6eSMichael Roth          */
2507f693fe6eSMichael Roth         if (errno != ENOENT) {
2508bd240fcaSzhanghailiang             error_setg_errno(errp, errno, "Can't open directory"
25099af9e0feSMarkus Armbruster                              "\"/sys/devices/system/memory/\"");
2510f693fe6eSMichael Roth         }
2511bd240fcaSzhanghailiang         return NULL;
2512bd240fcaSzhanghailiang     }
2513bd240fcaSzhanghailiang 
2514bd240fcaSzhanghailiang     /* Note: the phys_index of memory block may be discontinuous,
2515bd240fcaSzhanghailiang      * this is because a memblk is the unit of the Sparse Memory design, which
2516bd240fcaSzhanghailiang      * allows discontinuous memory ranges (ex. NUMA), so here we should
2517bd240fcaSzhanghailiang      * traverse the memory block directory.
2518bd240fcaSzhanghailiang      */
2519bd240fcaSzhanghailiang     while ((de = readdir(dp)) != NULL) {
2520bd240fcaSzhanghailiang         GuestMemoryBlock *mem_blk;
2521bd240fcaSzhanghailiang         GuestMemoryBlockList *entry;
2522bd240fcaSzhanghailiang 
2523bd240fcaSzhanghailiang         if ((strncmp(de->d_name, "memory", 6) != 0) ||
2524bd240fcaSzhanghailiang             !(de->d_type & DT_DIR)) {
2525bd240fcaSzhanghailiang             continue;
2526bd240fcaSzhanghailiang         }
2527bd240fcaSzhanghailiang 
2528bd240fcaSzhanghailiang         mem_blk = g_malloc0(sizeof *mem_blk);
2529bd240fcaSzhanghailiang         /* The d_name is "memoryXXX",  phys_index is block id, same as XXX */
2530bd240fcaSzhanghailiang         mem_blk->phys_index = strtoul(&de->d_name[6], NULL, 10);
2531bd240fcaSzhanghailiang         mem_blk->has_can_offline = true; /* lolspeak ftw */
2532bd240fcaSzhanghailiang         transfer_memory_block(mem_blk, true, NULL, &local_err);
2533bd240fcaSzhanghailiang 
2534bd240fcaSzhanghailiang         entry = g_malloc0(sizeof *entry);
2535bd240fcaSzhanghailiang         entry->value = mem_blk;
2536bd240fcaSzhanghailiang 
2537bd240fcaSzhanghailiang         *link = entry;
2538bd240fcaSzhanghailiang         link = &entry->next;
2539bd240fcaSzhanghailiang     }
2540bd240fcaSzhanghailiang 
2541bd240fcaSzhanghailiang     closedir(dp);
2542bd240fcaSzhanghailiang     if (local_err == NULL) {
2543bd240fcaSzhanghailiang         /* there's no guest with zero memory blocks */
2544bd240fcaSzhanghailiang         if (head == NULL) {
2545bd240fcaSzhanghailiang             error_setg(errp, "guest reported zero memory blocks!");
2546bd240fcaSzhanghailiang         }
2547bd240fcaSzhanghailiang         return head;
2548bd240fcaSzhanghailiang     }
2549bd240fcaSzhanghailiang 
2550bd240fcaSzhanghailiang     qapi_free_GuestMemoryBlockList(head);
2551bd240fcaSzhanghailiang     error_propagate(errp, local_err);
2552a065aaa9Szhanghailiang     return NULL;
2553a065aaa9Szhanghailiang }
2554a065aaa9Szhanghailiang 
2555a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2556a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2557a065aaa9Szhanghailiang {
255832ca7927Szhanghailiang     GuestMemoryBlockResponseList *head, **link;
255932ca7927Szhanghailiang     Error *local_err = NULL;
256032ca7927Szhanghailiang 
256132ca7927Szhanghailiang     head = NULL;
256232ca7927Szhanghailiang     link = &head;
256332ca7927Szhanghailiang 
256432ca7927Szhanghailiang     while (mem_blks != NULL) {
256532ca7927Szhanghailiang         GuestMemoryBlockResponse *result;
256632ca7927Szhanghailiang         GuestMemoryBlockResponseList *entry;
256732ca7927Szhanghailiang         GuestMemoryBlock *current_mem_blk = mem_blks->value;
256832ca7927Szhanghailiang 
256932ca7927Szhanghailiang         result = g_malloc0(sizeof(*result));
257032ca7927Szhanghailiang         result->phys_index = current_mem_blk->phys_index;
257132ca7927Szhanghailiang         transfer_memory_block(current_mem_blk, false, result, &local_err);
257232ca7927Szhanghailiang         if (local_err) { /* should never happen */
257332ca7927Szhanghailiang             goto err;
257432ca7927Szhanghailiang         }
257532ca7927Szhanghailiang         entry = g_malloc0(sizeof *entry);
257632ca7927Szhanghailiang         entry->value = result;
257732ca7927Szhanghailiang 
257832ca7927Szhanghailiang         *link = entry;
257932ca7927Szhanghailiang         link = &entry->next;
258032ca7927Szhanghailiang         mem_blks = mem_blks->next;
258132ca7927Szhanghailiang     }
258232ca7927Szhanghailiang 
258332ca7927Szhanghailiang     return head;
258432ca7927Szhanghailiang err:
258532ca7927Szhanghailiang     qapi_free_GuestMemoryBlockResponseList(head);
258632ca7927Szhanghailiang     error_propagate(errp, local_err);
2587a065aaa9Szhanghailiang     return NULL;
2588a065aaa9Szhanghailiang }
2589a065aaa9Szhanghailiang 
2590a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2591a065aaa9Szhanghailiang {
2592ef82b60bSzhanghailiang     Error *local_err = NULL;
2593ef82b60bSzhanghailiang     char *dirpath;
2594ef82b60bSzhanghailiang     int dirfd;
2595ef82b60bSzhanghailiang     char *buf;
2596ef82b60bSzhanghailiang     GuestMemoryBlockInfo *info;
2597ef82b60bSzhanghailiang 
2598ef82b60bSzhanghailiang     dirpath = g_strdup_printf("/sys/devices/system/memory/");
2599ef82b60bSzhanghailiang     dirfd = open(dirpath, O_RDONLY | O_DIRECTORY);
2600ef82b60bSzhanghailiang     if (dirfd == -1) {
2601ef82b60bSzhanghailiang         error_setg_errno(errp, errno, "open(\"%s\")", dirpath);
2602ef82b60bSzhanghailiang         g_free(dirpath);
2603a065aaa9Szhanghailiang         return NULL;
2604a065aaa9Szhanghailiang     }
2605ef82b60bSzhanghailiang     g_free(dirpath);
2606ef82b60bSzhanghailiang 
2607ef82b60bSzhanghailiang     buf = g_malloc0(20);
2608ef82b60bSzhanghailiang     ga_read_sysfs_file(dirfd, "block_size_bytes", buf, 20, &local_err);
26098ce1ee46SShannon Zhao     close(dirfd);
2610ef82b60bSzhanghailiang     if (local_err) {
2611ef82b60bSzhanghailiang         g_free(buf);
2612ef82b60bSzhanghailiang         error_propagate(errp, local_err);
2613ef82b60bSzhanghailiang         return NULL;
2614ef82b60bSzhanghailiang     }
2615ef82b60bSzhanghailiang 
2616ef82b60bSzhanghailiang     info = g_new0(GuestMemoryBlockInfo, 1);
2617ef82b60bSzhanghailiang     info->size = strtol(buf, NULL, 16); /* the unit is bytes */
2618ef82b60bSzhanghailiang 
2619ef82b60bSzhanghailiang     g_free(buf);
2620ef82b60bSzhanghailiang 
2621ef82b60bSzhanghailiang     return info;
2622ef82b60bSzhanghailiang }
2623a065aaa9Szhanghailiang 
2624e72c3f2eSMichael Roth #else /* defined(__linux__) */
2625e72c3f2eSMichael Roth 
262677dbc81bSMarkus Armbruster void qmp_guest_suspend_disk(Error **errp)
2627e72c3f2eSMichael Roth {
2628c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2629e72c3f2eSMichael Roth }
2630e72c3f2eSMichael Roth 
263177dbc81bSMarkus Armbruster void qmp_guest_suspend_ram(Error **errp)
2632e72c3f2eSMichael Roth {
2633c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2634e72c3f2eSMichael Roth }
2635e72c3f2eSMichael Roth 
263677dbc81bSMarkus Armbruster void qmp_guest_suspend_hybrid(Error **errp)
2637e72c3f2eSMichael Roth {
2638c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2639e72c3f2eSMichael Roth }
2640e72c3f2eSMichael Roth 
2641e72c3f2eSMichael Roth GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp)
2642e72c3f2eSMichael Roth {
2643c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2644e72c3f2eSMichael Roth     return NULL;
2645e72c3f2eSMichael Roth }
2646e72c3f2eSMichael Roth 
2647d2baff62SLaszlo Ersek GuestLogicalProcessorList *qmp_guest_get_vcpus(Error **errp)
2648d2baff62SLaszlo Ersek {
2649c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2650d2baff62SLaszlo Ersek     return NULL;
2651d2baff62SLaszlo Ersek }
2652d2baff62SLaszlo Ersek 
2653cbb65fc2SLaszlo Ersek int64_t qmp_guest_set_vcpus(GuestLogicalProcessorList *vcpus, Error **errp)
2654cbb65fc2SLaszlo Ersek {
2655c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2656cbb65fc2SLaszlo Ersek     return -1;
2657cbb65fc2SLaszlo Ersek }
2658cbb65fc2SLaszlo Ersek 
2659215a2771SDaniel P. Berrange void qmp_guest_set_user_password(const char *username,
2660215a2771SDaniel P. Berrange                                  const char *password,
2661215a2771SDaniel P. Berrange                                  bool crypted,
2662215a2771SDaniel P. Berrange                                  Error **errp)
2663215a2771SDaniel P. Berrange {
2664c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2665215a2771SDaniel P. Berrange }
2666215a2771SDaniel P. Berrange 
2667a065aaa9Szhanghailiang GuestMemoryBlockList *qmp_guest_get_memory_blocks(Error **errp)
2668a065aaa9Szhanghailiang {
2669c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2670a065aaa9Szhanghailiang     return NULL;
2671a065aaa9Szhanghailiang }
2672a065aaa9Szhanghailiang 
2673a065aaa9Szhanghailiang GuestMemoryBlockResponseList *
2674a065aaa9Szhanghailiang qmp_guest_set_memory_blocks(GuestMemoryBlockList *mem_blks, Error **errp)
2675a065aaa9Szhanghailiang {
2676c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2677a065aaa9Szhanghailiang     return NULL;
2678a065aaa9Szhanghailiang }
2679a065aaa9Szhanghailiang 
2680a065aaa9Szhanghailiang GuestMemoryBlockInfo *qmp_guest_get_memory_block_info(Error **errp)
2681a065aaa9Szhanghailiang {
2682c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2683a065aaa9Szhanghailiang     return NULL;
2684a065aaa9Szhanghailiang }
2685a065aaa9Szhanghailiang 
2686e72c3f2eSMichael Roth #endif
2687e72c3f2eSMichael Roth 
2688d35d4cb5SMichael Roth #if !defined(CONFIG_FSFREEZE)
2689d35d4cb5SMichael Roth 
269046d4c572STomoki Sekiyama GuestFilesystemInfoList *qmp_guest_get_fsinfo(Error **errp)
269146d4c572STomoki Sekiyama {
2692c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
269346d4c572STomoki Sekiyama     return NULL;
269446d4c572STomoki Sekiyama }
269546d4c572STomoki Sekiyama 
269677dbc81bSMarkus Armbruster GuestFsfreezeStatus qmp_guest_fsfreeze_status(Error **errp)
2697d35d4cb5SMichael Roth {
2698c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2699d35d4cb5SMichael Roth 
2700d35d4cb5SMichael Roth     return 0;
2701d35d4cb5SMichael Roth }
2702d35d4cb5SMichael Roth 
270377dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_freeze(Error **errp)
2704d35d4cb5SMichael Roth {
2705c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2706d35d4cb5SMichael Roth 
2707d35d4cb5SMichael Roth     return 0;
2708d35d4cb5SMichael Roth }
2709d35d4cb5SMichael Roth 
2710e99bce20STomoki Sekiyama int64_t qmp_guest_fsfreeze_freeze_list(bool has_mountpoints,
2711e99bce20STomoki Sekiyama                                        strList *mountpoints,
2712e99bce20STomoki Sekiyama                                        Error **errp)
2713e99bce20STomoki Sekiyama {
2714c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2715e99bce20STomoki Sekiyama 
2716e99bce20STomoki Sekiyama     return 0;
2717e99bce20STomoki Sekiyama }
2718e99bce20STomoki Sekiyama 
271977dbc81bSMarkus Armbruster int64_t qmp_guest_fsfreeze_thaw(Error **errp)
2720d35d4cb5SMichael Roth {
2721c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2722d35d4cb5SMichael Roth 
2723d35d4cb5SMichael Roth     return 0;
2724d35d4cb5SMichael Roth }
2725eab5fd59SPaolo Bonzini #endif /* CONFIG_FSFREEZE */
2726d35d4cb5SMichael Roth 
2727eab5fd59SPaolo Bonzini #if !defined(CONFIG_FSTRIM)
2728e82855d9SJustin Ossevoort GuestFilesystemTrimResponse *
2729e82855d9SJustin Ossevoort qmp_guest_fstrim(bool has_minimum, int64_t minimum, Error **errp)
2730eab5fd59SPaolo Bonzini {
2731c6bd8c70SMarkus Armbruster     error_setg(errp, QERR_UNSUPPORTED);
2732e82855d9SJustin Ossevoort     return NULL;
2733eab5fd59SPaolo Bonzini }
2734d35d4cb5SMichael Roth #endif
2735d35d4cb5SMichael Roth 
27361281c08aSTomoki Sekiyama /* add unsupported commands to the blacklist */
27371281c08aSTomoki Sekiyama GList *ga_command_blacklist_init(GList *blacklist)
27381281c08aSTomoki Sekiyama {
27391281c08aSTomoki Sekiyama #if !defined(__linux__)
27401281c08aSTomoki Sekiyama     {
27411281c08aSTomoki Sekiyama         const char *list[] = {
27421281c08aSTomoki Sekiyama             "guest-suspend-disk", "guest-suspend-ram",
27431281c08aSTomoki Sekiyama             "guest-suspend-hybrid", "guest-network-get-interfaces",
27440dd38a03Szhanghailiang             "guest-get-vcpus", "guest-set-vcpus",
27450dd38a03Szhanghailiang             "guest-get-memory-blocks", "guest-set-memory-blocks",
274628d8dd35SBasil Salman             "guest-get-memory-block-size", "guest-get-memory-block-info",
274728d8dd35SBasil Salman             NULL};
27481281c08aSTomoki Sekiyama         char **p = (char **)list;
27491281c08aSTomoki Sekiyama 
27501281c08aSTomoki Sekiyama         while (*p) {
27514bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
27521281c08aSTomoki Sekiyama         }
27531281c08aSTomoki Sekiyama     }
27541281c08aSTomoki Sekiyama #endif
27551281c08aSTomoki Sekiyama 
27561281c08aSTomoki Sekiyama #if !defined(CONFIG_FSFREEZE)
27571281c08aSTomoki Sekiyama     {
27581281c08aSTomoki Sekiyama         const char *list[] = {
27591281c08aSTomoki Sekiyama             "guest-get-fsinfo", "guest-fsfreeze-status",
27601281c08aSTomoki Sekiyama             "guest-fsfreeze-freeze", "guest-fsfreeze-freeze-list",
27611281c08aSTomoki Sekiyama             "guest-fsfreeze-thaw", "guest-get-fsinfo", NULL};
27621281c08aSTomoki Sekiyama         char **p = (char **)list;
27631281c08aSTomoki Sekiyama 
27641281c08aSTomoki Sekiyama         while (*p) {
27654bca81ceSMarc-André Lureau             blacklist = g_list_append(blacklist, g_strdup(*p++));
27661281c08aSTomoki Sekiyama         }
27671281c08aSTomoki Sekiyama     }
27681281c08aSTomoki Sekiyama #endif
27691281c08aSTomoki Sekiyama 
27701281c08aSTomoki Sekiyama #if !defined(CONFIG_FSTRIM)
27714bca81ceSMarc-André Lureau     blacklist = g_list_append(blacklist, g_strdup("guest-fstrim"));
27721281c08aSTomoki Sekiyama #endif
27731281c08aSTomoki Sekiyama 
27741281c08aSTomoki Sekiyama     return blacklist;
27751281c08aSTomoki Sekiyama }
27761281c08aSTomoki Sekiyama 
2777c216e5adSMichael Roth /* register init/cleanup routines for stateful command groups */
2778c216e5adSMichael Roth void ga_command_state_init(GAState *s, GACommandState *cs)
2779c216e5adSMichael Roth {
2780c216e5adSMichael Roth #if defined(CONFIG_FSFREEZE)
2781f22d85e9SMichael Roth     ga_command_state_add(cs, NULL, guest_fsfreeze_cleanup);
2782c216e5adSMichael Roth #endif
2783c216e5adSMichael Roth }
2784161a56a9SVinzenz Feenstra 
2785e674605fSTomáš Golembiovský #ifdef HAVE_UTMPX
2786e674605fSTomáš Golembiovský 
2787161a56a9SVinzenz Feenstra #define QGA_MICRO_SECOND_TO_SECOND 1000000
2788161a56a9SVinzenz Feenstra 
2789161a56a9SVinzenz Feenstra static double ga_get_login_time(struct utmpx *user_info)
2790161a56a9SVinzenz Feenstra {
2791161a56a9SVinzenz Feenstra     double seconds = (double)user_info->ut_tv.tv_sec;
2792161a56a9SVinzenz Feenstra     double useconds = (double)user_info->ut_tv.tv_usec;
2793161a56a9SVinzenz Feenstra     useconds /= QGA_MICRO_SECOND_TO_SECOND;
2794161a56a9SVinzenz Feenstra     return seconds + useconds;
2795161a56a9SVinzenz Feenstra }
2796161a56a9SVinzenz Feenstra 
2797161a56a9SVinzenz Feenstra GuestUserList *qmp_guest_get_users(Error **err)
2798161a56a9SVinzenz Feenstra {
2799161a56a9SVinzenz Feenstra     GHashTable *cache = NULL;
2800161a56a9SVinzenz Feenstra     GuestUserList *head = NULL, *cur_item = NULL;
2801161a56a9SVinzenz Feenstra     struct utmpx *user_info = NULL;
2802161a56a9SVinzenz Feenstra     gpointer value = NULL;
2803161a56a9SVinzenz Feenstra     GuestUser *user = NULL;
2804161a56a9SVinzenz Feenstra     GuestUserList *item = NULL;
2805161a56a9SVinzenz Feenstra     double login_time = 0;
2806161a56a9SVinzenz Feenstra 
2807161a56a9SVinzenz Feenstra     cache = g_hash_table_new(g_str_hash, g_str_equal);
2808161a56a9SVinzenz Feenstra     setutxent();
2809161a56a9SVinzenz Feenstra 
2810161a56a9SVinzenz Feenstra     for (;;) {
2811161a56a9SVinzenz Feenstra         user_info = getutxent();
2812161a56a9SVinzenz Feenstra         if (user_info == NULL) {
2813161a56a9SVinzenz Feenstra             break;
2814161a56a9SVinzenz Feenstra         } else if (user_info->ut_type != USER_PROCESS) {
2815161a56a9SVinzenz Feenstra             continue;
2816161a56a9SVinzenz Feenstra         } else if (g_hash_table_contains(cache, user_info->ut_user)) {
2817161a56a9SVinzenz Feenstra             value = g_hash_table_lookup(cache, user_info->ut_user);
2818161a56a9SVinzenz Feenstra             user = (GuestUser *)value;
2819161a56a9SVinzenz Feenstra             login_time = ga_get_login_time(user_info);
2820161a56a9SVinzenz Feenstra             /* We're ensuring the earliest login time to be sent */
2821161a56a9SVinzenz Feenstra             if (login_time < user->login_time) {
2822161a56a9SVinzenz Feenstra                 user->login_time = login_time;
2823161a56a9SVinzenz Feenstra             }
2824161a56a9SVinzenz Feenstra             continue;
2825161a56a9SVinzenz Feenstra         }
2826161a56a9SVinzenz Feenstra 
2827161a56a9SVinzenz Feenstra         item = g_new0(GuestUserList, 1);
2828161a56a9SVinzenz Feenstra         item->value = g_new0(GuestUser, 1);
2829161a56a9SVinzenz Feenstra         item->value->user = g_strdup(user_info->ut_user);
2830161a56a9SVinzenz Feenstra         item->value->login_time = ga_get_login_time(user_info);
2831161a56a9SVinzenz Feenstra 
2832161a56a9SVinzenz Feenstra         g_hash_table_insert(cache, item->value->user, item->value);
2833161a56a9SVinzenz Feenstra 
2834161a56a9SVinzenz Feenstra         if (!cur_item) {
2835161a56a9SVinzenz Feenstra             head = cur_item = item;
2836161a56a9SVinzenz Feenstra         } else {
2837161a56a9SVinzenz Feenstra             cur_item->next = item;
2838161a56a9SVinzenz Feenstra             cur_item = item;
2839161a56a9SVinzenz Feenstra         }
2840161a56a9SVinzenz Feenstra     }
2841161a56a9SVinzenz Feenstra     endutxent();
2842161a56a9SVinzenz Feenstra     g_hash_table_destroy(cache);
2843161a56a9SVinzenz Feenstra     return head;
2844161a56a9SVinzenz Feenstra }
2845e674605fSTomáš Golembiovský 
2846e674605fSTomáš Golembiovský #else
2847e674605fSTomáš Golembiovský 
2848e674605fSTomáš Golembiovský GuestUserList *qmp_guest_get_users(Error **errp)
2849e674605fSTomáš Golembiovský {
2850e674605fSTomáš Golembiovský     error_setg(errp, QERR_UNSUPPORTED);
2851e674605fSTomáš Golembiovský     return NULL;
2852e674605fSTomáš Golembiovský }
2853e674605fSTomáš Golembiovský 
2854e674605fSTomáš Golembiovský #endif
28559848f797STomáš Golembiovský 
28569848f797STomáš Golembiovský /* Replace escaped special characters with theire real values. The replacement
28579848f797STomáš Golembiovský  * is done in place -- returned value is in the original string.
28589848f797STomáš Golembiovský  */
28599848f797STomáš Golembiovský static void ga_osrelease_replace_special(gchar *value)
28609848f797STomáš Golembiovský {
28619848f797STomáš Golembiovský     gchar *p, *p2, quote;
28629848f797STomáš Golembiovský 
28639848f797STomáš Golembiovský     /* Trim the string at first space or semicolon if it is not enclosed in
28649848f797STomáš Golembiovský      * single or double quotes. */
28659848f797STomáš Golembiovský     if ((value[0] != '"') || (value[0] == '\'')) {
28669848f797STomáš Golembiovský         p = strchr(value, ' ');
28679848f797STomáš Golembiovský         if (p != NULL) {
28689848f797STomáš Golembiovský             *p = 0;
28699848f797STomáš Golembiovský         }
28709848f797STomáš Golembiovský         p = strchr(value, ';');
28719848f797STomáš Golembiovský         if (p != NULL) {
28729848f797STomáš Golembiovský             *p = 0;
28739848f797STomáš Golembiovský         }
28749848f797STomáš Golembiovský         return;
28759848f797STomáš Golembiovský     }
28769848f797STomáš Golembiovský 
28779848f797STomáš Golembiovský     quote = value[0];
28789848f797STomáš Golembiovský     p2 = value;
28799848f797STomáš Golembiovský     p = value + 1;
28809848f797STomáš Golembiovský     while (*p != 0) {
28819848f797STomáš Golembiovský         if (*p == '\\') {
28829848f797STomáš Golembiovský             p++;
28839848f797STomáš Golembiovský             switch (*p) {
28849848f797STomáš Golembiovský             case '$':
28859848f797STomáš Golembiovský             case '\'':
28869848f797STomáš Golembiovský             case '"':
28879848f797STomáš Golembiovský             case '\\':
28889848f797STomáš Golembiovský             case '`':
28899848f797STomáš Golembiovský                 break;
28909848f797STomáš Golembiovský             default:
28919848f797STomáš Golembiovský                 /* Keep literal backslash followed by whatever is there */
28929848f797STomáš Golembiovský                 p--;
28939848f797STomáš Golembiovský                 break;
28949848f797STomáš Golembiovský             }
28959848f797STomáš Golembiovský         } else if (*p == quote) {
28969848f797STomáš Golembiovský             *p2 = 0;
28979848f797STomáš Golembiovský             break;
28989848f797STomáš Golembiovský         }
28999848f797STomáš Golembiovský         *(p2++) = *(p++);
29009848f797STomáš Golembiovský     }
29019848f797STomáš Golembiovský }
29029848f797STomáš Golembiovský 
29039848f797STomáš Golembiovský static GKeyFile *ga_parse_osrelease(const char *fname)
29049848f797STomáš Golembiovský {
29059848f797STomáš Golembiovský     gchar *content = NULL;
29069848f797STomáš Golembiovský     gchar *content2 = NULL;
29079848f797STomáš Golembiovský     GError *err = NULL;
29089848f797STomáš Golembiovský     GKeyFile *keys = g_key_file_new();
29099848f797STomáš Golembiovský     const char *group = "[os-release]\n";
29109848f797STomáš Golembiovský 
29119848f797STomáš Golembiovský     if (!g_file_get_contents(fname, &content, NULL, &err)) {
29129848f797STomáš Golembiovský         slog("failed to read '%s', error: %s", fname, err->message);
29139848f797STomáš Golembiovský         goto fail;
29149848f797STomáš Golembiovský     }
29159848f797STomáš Golembiovský 
29169848f797STomáš Golembiovský     if (!g_utf8_validate(content, -1, NULL)) {
29179848f797STomáš Golembiovský         slog("file is not utf-8 encoded: %s", fname);
29189848f797STomáš Golembiovský         goto fail;
29199848f797STomáš Golembiovský     }
29209848f797STomáš Golembiovský     content2 = g_strdup_printf("%s%s", group, content);
29219848f797STomáš Golembiovský 
29229848f797STomáš Golembiovský     if (!g_key_file_load_from_data(keys, content2, -1, G_KEY_FILE_NONE,
29239848f797STomáš Golembiovský                                    &err)) {
29249848f797STomáš Golembiovský         slog("failed to parse file '%s', error: %s", fname, err->message);
29259848f797STomáš Golembiovský         goto fail;
29269848f797STomáš Golembiovský     }
29279848f797STomáš Golembiovský 
29289848f797STomáš Golembiovský     g_free(content);
29299848f797STomáš Golembiovský     g_free(content2);
29309848f797STomáš Golembiovský     return keys;
29319848f797STomáš Golembiovský 
29329848f797STomáš Golembiovský fail:
29339848f797STomáš Golembiovský     g_error_free(err);
29349848f797STomáš Golembiovský     g_free(content);
29359848f797STomáš Golembiovský     g_free(content2);
29369848f797STomáš Golembiovský     g_key_file_free(keys);
29379848f797STomáš Golembiovský     return NULL;
29389848f797STomáš Golembiovský }
29399848f797STomáš Golembiovský 
29409848f797STomáš Golembiovský GuestOSInfo *qmp_guest_get_osinfo(Error **errp)
29419848f797STomáš Golembiovský {
29429848f797STomáš Golembiovský     GuestOSInfo *info = NULL;
29439848f797STomáš Golembiovský     struct utsname kinfo;
2944339ca68bSTomáš Golembiovský     GKeyFile *osrelease = NULL;
2945339ca68bSTomáš Golembiovský     const char *qga_os_release = g_getenv("QGA_OS_RELEASE");
29469848f797STomáš Golembiovský 
29479848f797STomáš Golembiovský     info = g_new0(GuestOSInfo, 1);
29489848f797STomáš Golembiovský 
29499848f797STomáš Golembiovský     if (uname(&kinfo) != 0) {
29509848f797STomáš Golembiovský         error_setg_errno(errp, errno, "uname failed");
29519848f797STomáš Golembiovský     } else {
29529848f797STomáš Golembiovský         info->has_kernel_version = true;
29539848f797STomáš Golembiovský         info->kernel_version = g_strdup(kinfo.version);
29549848f797STomáš Golembiovský         info->has_kernel_release = true;
29559848f797STomáš Golembiovský         info->kernel_release = g_strdup(kinfo.release);
29569848f797STomáš Golembiovský         info->has_machine = true;
29579848f797STomáš Golembiovský         info->machine = g_strdup(kinfo.machine);
29589848f797STomáš Golembiovský     }
29599848f797STomáš Golembiovský 
2960339ca68bSTomáš Golembiovský     if (qga_os_release != NULL) {
2961339ca68bSTomáš Golembiovský         osrelease = ga_parse_osrelease(qga_os_release);
2962339ca68bSTomáš Golembiovský     } else {
29639848f797STomáš Golembiovský         osrelease = ga_parse_osrelease("/etc/os-release");
29649848f797STomáš Golembiovský         if (osrelease == NULL) {
29659848f797STomáš Golembiovský             osrelease = ga_parse_osrelease("/usr/lib/os-release");
29669848f797STomáš Golembiovský         }
2967339ca68bSTomáš Golembiovský     }
29689848f797STomáš Golembiovský 
29699848f797STomáš Golembiovský     if (osrelease != NULL) {
29709848f797STomáš Golembiovský         char *value;
29719848f797STomáš Golembiovský 
29729848f797STomáš Golembiovský #define GET_FIELD(field, osfield) do { \
29739848f797STomáš Golembiovský     value = g_key_file_get_value(osrelease, "os-release", osfield, NULL); \
29749848f797STomáš Golembiovský     if (value != NULL) { \
29759848f797STomáš Golembiovský         ga_osrelease_replace_special(value); \
29769848f797STomáš Golembiovský         info->has_ ## field = true; \
29779848f797STomáš Golembiovský         info->field = value; \
29789848f797STomáš Golembiovský     } \
29799848f797STomáš Golembiovský } while (0)
29809848f797STomáš Golembiovský         GET_FIELD(id, "ID");
29819848f797STomáš Golembiovský         GET_FIELD(name, "NAME");
29829848f797STomáš Golembiovský         GET_FIELD(pretty_name, "PRETTY_NAME");
29839848f797STomáš Golembiovský         GET_FIELD(version, "VERSION");
29849848f797STomáš Golembiovský         GET_FIELD(version_id, "VERSION_ID");
29859848f797STomáš Golembiovský         GET_FIELD(variant, "VARIANT");
29869848f797STomáš Golembiovský         GET_FIELD(variant_id, "VARIANT_ID");
29879848f797STomáš Golembiovský #undef GET_FIELD
29889848f797STomáš Golembiovský 
29899848f797STomáš Golembiovský         g_key_file_free(osrelease);
29909848f797STomáš Golembiovský     }
29919848f797STomáš Golembiovský 
29929848f797STomáš Golembiovský     return info;
29939848f797STomáš Golembiovský }
2994