Statistics
| Branch: | Revision:

root / savevm.c @ 37628f11

History | View | Annotate | Download (54.5 kB)

1 a672b469 aliguori
/*
2 a672b469 aliguori
 * QEMU System Emulator
3 a672b469 aliguori
 *
4 a672b469 aliguori
 * Copyright (c) 2003-2008 Fabrice Bellard
5 a672b469 aliguori
 *
6 a672b469 aliguori
 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 a672b469 aliguori
 * of this software and associated documentation files (the "Software"), to deal
8 a672b469 aliguori
 * in the Software without restriction, including without limitation the rights
9 a672b469 aliguori
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 a672b469 aliguori
 * copies of the Software, and to permit persons to whom the Software is
11 a672b469 aliguori
 * furnished to do so, subject to the following conditions:
12 a672b469 aliguori
 *
13 a672b469 aliguori
 * The above copyright notice and this permission notice shall be included in
14 a672b469 aliguori
 * all copies or substantial portions of the Software.
15 a672b469 aliguori
 *
16 a672b469 aliguori
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 a672b469 aliguori
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 a672b469 aliguori
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 a672b469 aliguori
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 a672b469 aliguori
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 a672b469 aliguori
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 a672b469 aliguori
 * THE SOFTWARE.
23 a672b469 aliguori
 */
24 a672b469 aliguori
#include <unistd.h>
25 a672b469 aliguori
#include <fcntl.h>
26 a672b469 aliguori
#include <signal.h>
27 a672b469 aliguori
#include <time.h>
28 a672b469 aliguori
#include <errno.h>
29 a672b469 aliguori
#include <sys/time.h>
30 a672b469 aliguori
#include <zlib.h>
31 a672b469 aliguori
32 71e72a19 Juan Quintela
/* Needed early for CONFIG_BSD etc. */
33 d40cdb10 blueswir1
#include "config-host.h"
34 d40cdb10 blueswir1
35 a672b469 aliguori
#ifndef _WIN32
36 a672b469 aliguori
#include <sys/times.h>
37 a672b469 aliguori
#include <sys/wait.h>
38 a672b469 aliguori
#include <termios.h>
39 a672b469 aliguori
#include <sys/mman.h>
40 a672b469 aliguori
#include <sys/ioctl.h>
41 a672b469 aliguori
#include <sys/resource.h>
42 a672b469 aliguori
#include <sys/socket.h>
43 a672b469 aliguori
#include <netinet/in.h>
44 a672b469 aliguori
#include <net/if.h>
45 a672b469 aliguori
#include <arpa/inet.h>
46 a672b469 aliguori
#include <dirent.h>
47 a672b469 aliguori
#include <netdb.h>
48 a672b469 aliguori
#include <sys/select.h>
49 71e72a19 Juan Quintela
#ifdef CONFIG_BSD
50 a672b469 aliguori
#include <sys/stat.h>
51 a167ba50 Aurelien Jarno
#if defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || defined(__DragonFly__)
52 a672b469 aliguori
#include <libutil.h>
53 a672b469 aliguori
#else
54 a672b469 aliguori
#include <util.h>
55 a672b469 aliguori
#endif
56 a672b469 aliguori
#ifdef __linux__
57 a672b469 aliguori
#include <pty.h>
58 a672b469 aliguori
#include <malloc.h>
59 a672b469 aliguori
#include <linux/rtc.h>
60 a672b469 aliguori
#endif
61 a672b469 aliguori
#endif
62 a672b469 aliguori
#endif
63 a672b469 aliguori
64 a672b469 aliguori
#ifdef _WIN32
65 49dc768d aliguori
#include <windows.h>
66 a672b469 aliguori
#include <malloc.h>
67 a672b469 aliguori
#include <sys/timeb.h>
68 a672b469 aliguori
#include <mmsystem.h>
69 a672b469 aliguori
#define getopt_long_only getopt_long
70 a672b469 aliguori
#define memalign(align, size) malloc(size)
71 a672b469 aliguori
#endif
72 a672b469 aliguori
73 511d2b14 blueswir1
#include "qemu-common.h"
74 511d2b14 blueswir1
#include "hw/hw.h"
75 7685ee6a Alex Williamson
#include "hw/qdev.h"
76 511d2b14 blueswir1
#include "net.h"
77 511d2b14 blueswir1
#include "monitor.h"
78 511d2b14 blueswir1
#include "sysemu.h"
79 511d2b14 blueswir1
#include "qemu-timer.h"
80 511d2b14 blueswir1
#include "qemu-char.h"
81 511d2b14 blueswir1
#include "audio/audio.h"
82 511d2b14 blueswir1
#include "migration.h"
83 511d2b14 blueswir1
#include "qemu_socket.h"
84 72cf2d4f Blue Swirl
#include "qemu-queue.h"
85 17a4663e Blue Swirl
#include "cpus.h"
86 511d2b14 blueswir1
87 a672b469 aliguori
#define SELF_ANNOUNCE_ROUNDS 5
88 a672b469 aliguori
89 18995b98 Nolan
#ifndef ETH_P_RARP
90 f8778a77 Stefan Berger
#define ETH_P_RARP 0x8035
91 18995b98 Nolan
#endif
92 18995b98 Nolan
#define ARP_HTYPE_ETH 0x0001
93 18995b98 Nolan
#define ARP_PTYPE_IP 0x0800
94 18995b98 Nolan
#define ARP_OP_REQUEST_REV 0x3
95 18995b98 Nolan
96 18995b98 Nolan
static int announce_self_create(uint8_t *buf,
97 a672b469 aliguori
                                uint8_t *mac_addr)
98 a672b469 aliguori
{
99 18995b98 Nolan
    /* Ethernet header. */
100 18995b98 Nolan
    memset(buf, 0xff, 6);         /* destination MAC addr */
101 18995b98 Nolan
    memcpy(buf + 6, mac_addr, 6); /* source MAC addr */
102 18995b98 Nolan
    *(uint16_t *)(buf + 12) = htons(ETH_P_RARP); /* ethertype */
103 18995b98 Nolan
104 18995b98 Nolan
    /* RARP header. */
105 18995b98 Nolan
    *(uint16_t *)(buf + 14) = htons(ARP_HTYPE_ETH); /* hardware addr space */
106 18995b98 Nolan
    *(uint16_t *)(buf + 16) = htons(ARP_PTYPE_IP); /* protocol addr space */
107 18995b98 Nolan
    *(buf + 18) = 6; /* hardware addr length (ethernet) */
108 18995b98 Nolan
    *(buf + 19) = 4; /* protocol addr length (IPv4) */
109 18995b98 Nolan
    *(uint16_t *)(buf + 20) = htons(ARP_OP_REQUEST_REV); /* opcode */
110 18995b98 Nolan
    memcpy(buf + 22, mac_addr, 6); /* source hw addr */
111 18995b98 Nolan
    memset(buf + 28, 0x00, 4);     /* source protocol addr */
112 18995b98 Nolan
    memcpy(buf + 32, mac_addr, 6); /* target hw addr */
113 18995b98 Nolan
    memset(buf + 38, 0x00, 4);     /* target protocol addr */
114 18995b98 Nolan
115 18995b98 Nolan
    /* Padding to get up to 60 bytes (ethernet min packet size, minus FCS). */
116 18995b98 Nolan
    memset(buf + 42, 0x00, 18);
117 18995b98 Nolan
118 18995b98 Nolan
    return 60; /* len (FCS will be added by hardware) */
119 a672b469 aliguori
}
120 a672b469 aliguori
121 f401ca22 Mark McLoughlin
static void qemu_announce_self_iter(NICState *nic, void *opaque)
122 a672b469 aliguori
{
123 18995b98 Nolan
    uint8_t buf[60];
124 f401ca22 Mark McLoughlin
    int len;
125 f401ca22 Mark McLoughlin
126 f401ca22 Mark McLoughlin
    len = announce_self_create(buf, nic->conf->macaddr.a);
127 f401ca22 Mark McLoughlin
128 f401ca22 Mark McLoughlin
    qemu_send_packet_raw(&nic->nc, buf, len);
129 f401ca22 Mark McLoughlin
}
130 f401ca22 Mark McLoughlin
131 f401ca22 Mark McLoughlin
132 f401ca22 Mark McLoughlin
static void qemu_announce_self_once(void *opaque)
133 f401ca22 Mark McLoughlin
{
134 ed8b330b Gleb Natapov
    static int count = SELF_ANNOUNCE_ROUNDS;
135 ed8b330b Gleb Natapov
    QEMUTimer *timer = *(QEMUTimer **)opaque;
136 a672b469 aliguori
137 f401ca22 Mark McLoughlin
    qemu_foreach_nic(qemu_announce_self_iter, NULL);
138 f401ca22 Mark McLoughlin
139 18995b98 Nolan
    if (--count) {
140 18995b98 Nolan
        /* delay 50ms, 150ms, 250ms, ... */
141 7bd427d8 Paolo Bonzini
        qemu_mod_timer(timer, qemu_get_clock_ms(rt_clock) +
142 18995b98 Nolan
                       50 + (SELF_ANNOUNCE_ROUNDS - count - 1) * 100);
143 ed8b330b Gleb Natapov
    } else {
144 ed8b330b Gleb Natapov
            qemu_del_timer(timer);
145 ed8b330b Gleb Natapov
            qemu_free_timer(timer);
146 ed8b330b Gleb Natapov
    }
147 ed8b330b Gleb Natapov
}
148 ed8b330b Gleb Natapov
149 ed8b330b Gleb Natapov
void qemu_announce_self(void)
150 ed8b330b Gleb Natapov
{
151 ed8b330b Gleb Natapov
        static QEMUTimer *timer;
152 7bd427d8 Paolo Bonzini
        timer = qemu_new_timer_ms(rt_clock, qemu_announce_self_once, &timer);
153 ed8b330b Gleb Natapov
        qemu_announce_self_once(&timer);
154 a672b469 aliguori
}
155 a672b469 aliguori
156 a672b469 aliguori
/***********************************************************/
157 a672b469 aliguori
/* savevm/loadvm support */
158 a672b469 aliguori
159 a672b469 aliguori
#define IO_BUF_SIZE 32768
160 a672b469 aliguori
161 a672b469 aliguori
struct QEMUFile {
162 a672b469 aliguori
    QEMUFilePutBufferFunc *put_buffer;
163 a672b469 aliguori
    QEMUFileGetBufferFunc *get_buffer;
164 a672b469 aliguori
    QEMUFileCloseFunc *close;
165 a672b469 aliguori
    QEMUFileRateLimit *rate_limit;
166 19629537 Glauber Costa
    QEMUFileSetRateLimit *set_rate_limit;
167 c163b5ca lirans@il.ibm.com
    QEMUFileGetRateLimit *get_rate_limit;
168 a672b469 aliguori
    void *opaque;
169 a672b469 aliguori
    int is_write;
170 a672b469 aliguori
171 a672b469 aliguori
    int64_t buf_offset; /* start of buffer when writing, end of buffer
172 a672b469 aliguori
                           when reading */
173 a672b469 aliguori
    int buf_index;
174 a672b469 aliguori
    int buf_size; /* 0 when writing */
175 a672b469 aliguori
    uint8_t buf[IO_BUF_SIZE];
176 a672b469 aliguori
177 a672b469 aliguori
    int has_error;
178 a672b469 aliguori
};
179 a672b469 aliguori
180 7f79dd28 Paolo Bonzini
typedef struct QEMUFileStdio
181 a672b469 aliguori
{
182 7f79dd28 Paolo Bonzini
    FILE *stdio_file;
183 a672b469 aliguori
    QEMUFile *file;
184 7f79dd28 Paolo Bonzini
} QEMUFileStdio;
185 a672b469 aliguori
186 a672b469 aliguori
typedef struct QEMUFileSocket
187 a672b469 aliguori
{
188 a672b469 aliguori
    int fd;
189 a672b469 aliguori
    QEMUFile *file;
190 a672b469 aliguori
} QEMUFileSocket;
191 a672b469 aliguori
192 a672b469 aliguori
static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
193 a672b469 aliguori
{
194 a672b469 aliguori
    QEMUFileSocket *s = opaque;
195 a672b469 aliguori
    ssize_t len;
196 a672b469 aliguori
197 a672b469 aliguori
    do {
198 c5b76b38 Blue Swirl
        len = recv(s->fd, (void *)buf, size, 0);
199 a672b469 aliguori
    } while (len == -1 && socket_error() == EINTR);
200 a672b469 aliguori
201 a672b469 aliguori
    if (len == -1)
202 a672b469 aliguori
        len = -socket_error();
203 a672b469 aliguori
204 a672b469 aliguori
    return len;
205 a672b469 aliguori
}
206 a672b469 aliguori
207 a672b469 aliguori
static int socket_close(void *opaque)
208 a672b469 aliguori
{
209 a672b469 aliguori
    QEMUFileSocket *s = opaque;
210 a672b469 aliguori
    qemu_free(s);
211 a672b469 aliguori
    return 0;
212 a672b469 aliguori
}
213 a672b469 aliguori
214 7f79dd28 Paolo Bonzini
static int stdio_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
215 a672b469 aliguori
{
216 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
217 7f79dd28 Paolo Bonzini
    return fwrite(buf, 1, size, s->stdio_file);
218 a672b469 aliguori
}
219 a672b469 aliguori
220 7f79dd28 Paolo Bonzini
static int stdio_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
221 a672b469 aliguori
{
222 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
223 7f79dd28 Paolo Bonzini
    FILE *fp = s->stdio_file;
224 8a67ec4d Uri Lublin
    int bytes;
225 8a67ec4d Uri Lublin
226 8a67ec4d Uri Lublin
    do {
227 8a67ec4d Uri Lublin
        clearerr(fp);
228 8a67ec4d Uri Lublin
        bytes = fread(buf, 1, size, fp);
229 8a67ec4d Uri Lublin
    } while ((bytes == 0) && ferror(fp) && (errno == EINTR));
230 8a67ec4d Uri Lublin
    return bytes;
231 a672b469 aliguori
}
232 a672b469 aliguori
233 7f79dd28 Paolo Bonzini
static int stdio_pclose(void *opaque)
234 7f79dd28 Paolo Bonzini
{
235 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
236 41ef56e6 Anthony Liguori
    int ret;
237 41ef56e6 Anthony Liguori
    ret = pclose(s->stdio_file);
238 7f79dd28 Paolo Bonzini
    qemu_free(s);
239 41ef56e6 Anthony Liguori
    return ret;
240 7f79dd28 Paolo Bonzini
}
241 7f79dd28 Paolo Bonzini
242 7f79dd28 Paolo Bonzini
static int stdio_fclose(void *opaque)
243 a672b469 aliguori
{
244 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
245 7f79dd28 Paolo Bonzini
    fclose(s->stdio_file);
246 a672b469 aliguori
    qemu_free(s);
247 a672b469 aliguori
    return 0;
248 a672b469 aliguori
}
249 a672b469 aliguori
250 7f79dd28 Paolo Bonzini
QEMUFile *qemu_popen(FILE *stdio_file, const char *mode)
251 a672b469 aliguori
{
252 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s;
253 a672b469 aliguori
254 7f79dd28 Paolo Bonzini
    if (stdio_file == NULL || mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) {
255 a672b469 aliguori
        fprintf(stderr, "qemu_popen: Argument validity check failed\n");
256 a672b469 aliguori
        return NULL;
257 a672b469 aliguori
    }
258 a672b469 aliguori
259 7f79dd28 Paolo Bonzini
    s = qemu_mallocz(sizeof(QEMUFileStdio));
260 a672b469 aliguori
261 7f79dd28 Paolo Bonzini
    s->stdio_file = stdio_file;
262 a672b469 aliguori
263 a672b469 aliguori
    if(mode[0] == 'r') {
264 c163b5ca lirans@il.ibm.com
        s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_pclose, 
265 c163b5ca lirans@il.ibm.com
                                 NULL, NULL, NULL);
266 a672b469 aliguori
    } else {
267 c163b5ca lirans@il.ibm.com
        s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_pclose, 
268 c163b5ca lirans@il.ibm.com
                                 NULL, NULL, NULL);
269 a672b469 aliguori
    }
270 a672b469 aliguori
    return s->file;
271 a672b469 aliguori
}
272 a672b469 aliguori
273 a672b469 aliguori
QEMUFile *qemu_popen_cmd(const char *command, const char *mode)
274 a672b469 aliguori
{
275 a672b469 aliguori
    FILE *popen_file;
276 a672b469 aliguori
277 a672b469 aliguori
    popen_file = popen(command, mode);
278 a672b469 aliguori
    if(popen_file == NULL) {
279 a672b469 aliguori
        return NULL;
280 a672b469 aliguori
    }
281 a672b469 aliguori
282 a672b469 aliguori
    return qemu_popen(popen_file, mode);
283 a672b469 aliguori
}
284 a672b469 aliguori
285 7f79dd28 Paolo Bonzini
int qemu_stdio_fd(QEMUFile *f)
286 8a43b1ea Chris Lalancette
{
287 7f79dd28 Paolo Bonzini
    QEMUFileStdio *p;
288 8a43b1ea Chris Lalancette
    int fd;
289 8a43b1ea Chris Lalancette
290 7f79dd28 Paolo Bonzini
    p = (QEMUFileStdio *)f->opaque;
291 7f79dd28 Paolo Bonzini
    fd = fileno(p->stdio_file);
292 8a43b1ea Chris Lalancette
293 8a43b1ea Chris Lalancette
    return fd;
294 8a43b1ea Chris Lalancette
}
295 8a43b1ea Chris Lalancette
296 5ac1fad3 Paolo Bonzini
QEMUFile *qemu_fdopen(int fd, const char *mode)
297 5ac1fad3 Paolo Bonzini
{
298 5ac1fad3 Paolo Bonzini
    QEMUFileStdio *s;
299 5ac1fad3 Paolo Bonzini
300 5ac1fad3 Paolo Bonzini
    if (mode == NULL ||
301 5ac1fad3 Paolo Bonzini
        (mode[0] != 'r' && mode[0] != 'w') ||
302 5ac1fad3 Paolo Bonzini
        mode[1] != 'b' || mode[2] != 0) {
303 5ac1fad3 Paolo Bonzini
        fprintf(stderr, "qemu_fdopen: Argument validity check failed\n");
304 5ac1fad3 Paolo Bonzini
        return NULL;
305 5ac1fad3 Paolo Bonzini
    }
306 5ac1fad3 Paolo Bonzini
307 5ac1fad3 Paolo Bonzini
    s = qemu_mallocz(sizeof(QEMUFileStdio));
308 5ac1fad3 Paolo Bonzini
    s->stdio_file = fdopen(fd, mode);
309 5ac1fad3 Paolo Bonzini
    if (!s->stdio_file)
310 5ac1fad3 Paolo Bonzini
        goto fail;
311 5ac1fad3 Paolo Bonzini
312 5ac1fad3 Paolo Bonzini
    if(mode[0] == 'r') {
313 c163b5ca lirans@il.ibm.com
        s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_fclose, 
314 c163b5ca lirans@il.ibm.com
                                 NULL, NULL, NULL);
315 5ac1fad3 Paolo Bonzini
    } else {
316 c163b5ca lirans@il.ibm.com
        s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_fclose, 
317 c163b5ca lirans@il.ibm.com
                                 NULL, NULL, NULL);
318 5ac1fad3 Paolo Bonzini
    }
319 5ac1fad3 Paolo Bonzini
    return s->file;
320 5ac1fad3 Paolo Bonzini
321 5ac1fad3 Paolo Bonzini
fail:
322 5ac1fad3 Paolo Bonzini
    qemu_free(s);
323 5ac1fad3 Paolo Bonzini
    return NULL;
324 5ac1fad3 Paolo Bonzini
}
325 5ac1fad3 Paolo Bonzini
326 a672b469 aliguori
QEMUFile *qemu_fopen_socket(int fd)
327 a672b469 aliguori
{
328 a672b469 aliguori
    QEMUFileSocket *s = qemu_mallocz(sizeof(QEMUFileSocket));
329 a672b469 aliguori
330 a672b469 aliguori
    s->fd = fd;
331 c163b5ca lirans@il.ibm.com
    s->file = qemu_fopen_ops(s, NULL, socket_get_buffer, socket_close, 
332 c163b5ca lirans@il.ibm.com
                             NULL, NULL, NULL);
333 a672b469 aliguori
    return s->file;
334 a672b469 aliguori
}
335 a672b469 aliguori
336 a672b469 aliguori
static int file_put_buffer(void *opaque, const uint8_t *buf,
337 a672b469 aliguori
                            int64_t pos, int size)
338 a672b469 aliguori
{
339 a672b469 aliguori
    QEMUFileStdio *s = opaque;
340 7f79dd28 Paolo Bonzini
    fseek(s->stdio_file, pos, SEEK_SET);
341 5fdb3aa1 Kirill A. Shutemov
    return fwrite(buf, 1, size, s->stdio_file);
342 a672b469 aliguori
}
343 a672b469 aliguori
344 a672b469 aliguori
static int file_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
345 a672b469 aliguori
{
346 a672b469 aliguori
    QEMUFileStdio *s = opaque;
347 7f79dd28 Paolo Bonzini
    fseek(s->stdio_file, pos, SEEK_SET);
348 7f79dd28 Paolo Bonzini
    return fread(buf, 1, size, s->stdio_file);
349 a672b469 aliguori
}
350 a672b469 aliguori
351 a672b469 aliguori
QEMUFile *qemu_fopen(const char *filename, const char *mode)
352 a672b469 aliguori
{
353 a672b469 aliguori
    QEMUFileStdio *s;
354 a672b469 aliguori
355 7f79dd28 Paolo Bonzini
    if (mode == NULL ||
356 7f79dd28 Paolo Bonzini
        (mode[0] != 'r' && mode[0] != 'w') ||
357 7f79dd28 Paolo Bonzini
        mode[1] != 'b' || mode[2] != 0) {
358 090414a3 Blue Swirl
        fprintf(stderr, "qemu_fopen: Argument validity check failed\n");
359 7f79dd28 Paolo Bonzini
        return NULL;
360 7f79dd28 Paolo Bonzini
    }
361 7f79dd28 Paolo Bonzini
362 a672b469 aliguori
    s = qemu_mallocz(sizeof(QEMUFileStdio));
363 a672b469 aliguori
364 7f79dd28 Paolo Bonzini
    s->stdio_file = fopen(filename, mode);
365 7f79dd28 Paolo Bonzini
    if (!s->stdio_file)
366 a672b469 aliguori
        goto fail;
367 c163b5ca lirans@il.ibm.com
    
368 7f79dd28 Paolo Bonzini
    if(mode[0] == 'w') {
369 c163b5ca lirans@il.ibm.com
        s->file = qemu_fopen_ops(s, file_put_buffer, NULL, stdio_fclose, 
370 c163b5ca lirans@il.ibm.com
                                 NULL, NULL, NULL);
371 7f79dd28 Paolo Bonzini
    } else {
372 c163b5ca lirans@il.ibm.com
        s->file = qemu_fopen_ops(s, NULL, file_get_buffer, stdio_fclose, 
373 c163b5ca lirans@il.ibm.com
                               NULL, NULL, NULL);
374 7f79dd28 Paolo Bonzini
    }
375 7f79dd28 Paolo Bonzini
    return s->file;
376 a672b469 aliguori
fail:
377 a672b469 aliguori
    qemu_free(s);
378 a672b469 aliguori
    return NULL;
379 a672b469 aliguori
}
380 a672b469 aliguori
381 178e08a5 aliguori
static int block_put_buffer(void *opaque, const uint8_t *buf,
382 a672b469 aliguori
                           int64_t pos, int size)
383 a672b469 aliguori
{
384 45566e9c Christoph Hellwig
    bdrv_save_vmstate(opaque, buf, pos, size);
385 a672b469 aliguori
    return size;
386 a672b469 aliguori
}
387 a672b469 aliguori
388 178e08a5 aliguori
static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
389 a672b469 aliguori
{
390 45566e9c Christoph Hellwig
    return bdrv_load_vmstate(opaque, buf, pos, size);
391 a672b469 aliguori
}
392 a672b469 aliguori
393 a672b469 aliguori
static int bdrv_fclose(void *opaque)
394 a672b469 aliguori
{
395 a672b469 aliguori
    return 0;
396 a672b469 aliguori
}
397 a672b469 aliguori
398 45566e9c Christoph Hellwig
static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int is_writable)
399 a672b469 aliguori
{
400 a672b469 aliguori
    if (is_writable)
401 c163b5ca lirans@il.ibm.com
        return qemu_fopen_ops(bs, block_put_buffer, NULL, bdrv_fclose, 
402 c163b5ca lirans@il.ibm.com
                              NULL, NULL, NULL);
403 c163b5ca lirans@il.ibm.com
    return qemu_fopen_ops(bs, NULL, block_get_buffer, bdrv_fclose, NULL, NULL, NULL);
404 a672b469 aliguori
}
405 a672b469 aliguori
406 a672b469 aliguori
QEMUFile *qemu_fopen_ops(void *opaque, QEMUFilePutBufferFunc *put_buffer,
407 a672b469 aliguori
                         QEMUFileGetBufferFunc *get_buffer,
408 a672b469 aliguori
                         QEMUFileCloseFunc *close,
409 19629537 Glauber Costa
                         QEMUFileRateLimit *rate_limit,
410 c163b5ca lirans@il.ibm.com
                         QEMUFileSetRateLimit *set_rate_limit,
411 c163b5ca lirans@il.ibm.com
                         QEMUFileGetRateLimit *get_rate_limit)
412 a672b469 aliguori
{
413 a672b469 aliguori
    QEMUFile *f;
414 a672b469 aliguori
415 a672b469 aliguori
    f = qemu_mallocz(sizeof(QEMUFile));
416 a672b469 aliguori
417 a672b469 aliguori
    f->opaque = opaque;
418 a672b469 aliguori
    f->put_buffer = put_buffer;
419 a672b469 aliguori
    f->get_buffer = get_buffer;
420 a672b469 aliguori
    f->close = close;
421 a672b469 aliguori
    f->rate_limit = rate_limit;
422 19629537 Glauber Costa
    f->set_rate_limit = set_rate_limit;
423 c163b5ca lirans@il.ibm.com
    f->get_rate_limit = get_rate_limit;
424 a672b469 aliguori
    f->is_write = 0;
425 a672b469 aliguori
426 a672b469 aliguori
    return f;
427 a672b469 aliguori
}
428 a672b469 aliguori
429 a672b469 aliguori
int qemu_file_has_error(QEMUFile *f)
430 a672b469 aliguori
{
431 a672b469 aliguori
    return f->has_error;
432 a672b469 aliguori
}
433 a672b469 aliguori
434 4dabe248 aliguori
void qemu_file_set_error(QEMUFile *f)
435 4dabe248 aliguori
{
436 4dabe248 aliguori
    f->has_error = 1;
437 4dabe248 aliguori
}
438 4dabe248 aliguori
439 a672b469 aliguori
void qemu_fflush(QEMUFile *f)
440 a672b469 aliguori
{
441 a672b469 aliguori
    if (!f->put_buffer)
442 a672b469 aliguori
        return;
443 a672b469 aliguori
444 a672b469 aliguori
    if (f->is_write && f->buf_index > 0) {
445 a672b469 aliguori
        int len;
446 a672b469 aliguori
447 a672b469 aliguori
        len = f->put_buffer(f->opaque, f->buf, f->buf_offset, f->buf_index);
448 a672b469 aliguori
        if (len > 0)
449 a672b469 aliguori
            f->buf_offset += f->buf_index;
450 a672b469 aliguori
        else
451 a672b469 aliguori
            f->has_error = 1;
452 a672b469 aliguori
        f->buf_index = 0;
453 a672b469 aliguori
    }
454 a672b469 aliguori
}
455 a672b469 aliguori
456 a672b469 aliguori
static void qemu_fill_buffer(QEMUFile *f)
457 a672b469 aliguori
{
458 a672b469 aliguori
    int len;
459 a672b469 aliguori
460 a672b469 aliguori
    if (!f->get_buffer)
461 a672b469 aliguori
        return;
462 a672b469 aliguori
463 a672b469 aliguori
    if (f->is_write)
464 a672b469 aliguori
        abort();
465 a672b469 aliguori
466 a672b469 aliguori
    len = f->get_buffer(f->opaque, f->buf, f->buf_offset, IO_BUF_SIZE);
467 a672b469 aliguori
    if (len > 0) {
468 a672b469 aliguori
        f->buf_index = 0;
469 a672b469 aliguori
        f->buf_size = len;
470 a672b469 aliguori
        f->buf_offset += len;
471 a672b469 aliguori
    } else if (len != -EAGAIN)
472 a672b469 aliguori
        f->has_error = 1;
473 a672b469 aliguori
}
474 a672b469 aliguori
475 a672b469 aliguori
int qemu_fclose(QEMUFile *f)
476 a672b469 aliguori
{
477 a672b469 aliguori
    int ret = 0;
478 a672b469 aliguori
    qemu_fflush(f);
479 a672b469 aliguori
    if (f->close)
480 a672b469 aliguori
        ret = f->close(f->opaque);
481 a672b469 aliguori
    qemu_free(f);
482 a672b469 aliguori
    return ret;
483 a672b469 aliguori
}
484 a672b469 aliguori
485 a672b469 aliguori
void qemu_file_put_notify(QEMUFile *f)
486 a672b469 aliguori
{
487 a672b469 aliguori
    f->put_buffer(f->opaque, NULL, 0, 0);
488 a672b469 aliguori
}
489 a672b469 aliguori
490 a672b469 aliguori
void qemu_put_buffer(QEMUFile *f, const uint8_t *buf, int size)
491 a672b469 aliguori
{
492 a672b469 aliguori
    int l;
493 a672b469 aliguori
494 a672b469 aliguori
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
495 a672b469 aliguori
        fprintf(stderr,
496 a672b469 aliguori
                "Attempted to write to buffer while read buffer is not empty\n");
497 a672b469 aliguori
        abort();
498 a672b469 aliguori
    }
499 a672b469 aliguori
500 a672b469 aliguori
    while (!f->has_error && size > 0) {
501 a672b469 aliguori
        l = IO_BUF_SIZE - f->buf_index;
502 a672b469 aliguori
        if (l > size)
503 a672b469 aliguori
            l = size;
504 a672b469 aliguori
        memcpy(f->buf + f->buf_index, buf, l);
505 a672b469 aliguori
        f->is_write = 1;
506 a672b469 aliguori
        f->buf_index += l;
507 a672b469 aliguori
        buf += l;
508 a672b469 aliguori
        size -= l;
509 a672b469 aliguori
        if (f->buf_index >= IO_BUF_SIZE)
510 a672b469 aliguori
            qemu_fflush(f);
511 a672b469 aliguori
    }
512 a672b469 aliguori
}
513 a672b469 aliguori
514 a672b469 aliguori
void qemu_put_byte(QEMUFile *f, int v)
515 a672b469 aliguori
{
516 a672b469 aliguori
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
517 a672b469 aliguori
        fprintf(stderr,
518 a672b469 aliguori
                "Attempted to write to buffer while read buffer is not empty\n");
519 a672b469 aliguori
        abort();
520 a672b469 aliguori
    }
521 a672b469 aliguori
522 a672b469 aliguori
    f->buf[f->buf_index++] = v;
523 a672b469 aliguori
    f->is_write = 1;
524 a672b469 aliguori
    if (f->buf_index >= IO_BUF_SIZE)
525 a672b469 aliguori
        qemu_fflush(f);
526 a672b469 aliguori
}
527 a672b469 aliguori
528 a672b469 aliguori
int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size1)
529 a672b469 aliguori
{
530 a672b469 aliguori
    int size, l;
531 a672b469 aliguori
532 a672b469 aliguori
    if (f->is_write)
533 a672b469 aliguori
        abort();
534 a672b469 aliguori
535 a672b469 aliguori
    size = size1;
536 a672b469 aliguori
    while (size > 0) {
537 a672b469 aliguori
        l = f->buf_size - f->buf_index;
538 a672b469 aliguori
        if (l == 0) {
539 a672b469 aliguori
            qemu_fill_buffer(f);
540 a672b469 aliguori
            l = f->buf_size - f->buf_index;
541 a672b469 aliguori
            if (l == 0)
542 a672b469 aliguori
                break;
543 a672b469 aliguori
        }
544 a672b469 aliguori
        if (l > size)
545 a672b469 aliguori
            l = size;
546 a672b469 aliguori
        memcpy(buf, f->buf + f->buf_index, l);
547 a672b469 aliguori
        f->buf_index += l;
548 a672b469 aliguori
        buf += l;
549 a672b469 aliguori
        size -= l;
550 a672b469 aliguori
    }
551 a672b469 aliguori
    return size1 - size;
552 a672b469 aliguori
}
553 a672b469 aliguori
554 811814bd Juan Quintela
static int qemu_peek_byte(QEMUFile *f)
555 811814bd Juan Quintela
{
556 811814bd Juan Quintela
    if (f->is_write)
557 811814bd Juan Quintela
        abort();
558 811814bd Juan Quintela
559 811814bd Juan Quintela
    if (f->buf_index >= f->buf_size) {
560 811814bd Juan Quintela
        qemu_fill_buffer(f);
561 811814bd Juan Quintela
        if (f->buf_index >= f->buf_size)
562 811814bd Juan Quintela
            return 0;
563 811814bd Juan Quintela
    }
564 811814bd Juan Quintela
    return f->buf[f->buf_index];
565 811814bd Juan Quintela
}
566 811814bd Juan Quintela
567 a672b469 aliguori
int qemu_get_byte(QEMUFile *f)
568 a672b469 aliguori
{
569 a672b469 aliguori
    if (f->is_write)
570 a672b469 aliguori
        abort();
571 a672b469 aliguori
572 a672b469 aliguori
    if (f->buf_index >= f->buf_size) {
573 a672b469 aliguori
        qemu_fill_buffer(f);
574 a672b469 aliguori
        if (f->buf_index >= f->buf_size)
575 a672b469 aliguori
            return 0;
576 a672b469 aliguori
    }
577 a672b469 aliguori
    return f->buf[f->buf_index++];
578 a672b469 aliguori
}
579 a672b469 aliguori
580 a672b469 aliguori
int64_t qemu_ftell(QEMUFile *f)
581 a672b469 aliguori
{
582 a672b469 aliguori
    return f->buf_offset - f->buf_size + f->buf_index;
583 a672b469 aliguori
}
584 a672b469 aliguori
585 a672b469 aliguori
int64_t qemu_fseek(QEMUFile *f, int64_t pos, int whence)
586 a672b469 aliguori
{
587 a672b469 aliguori
    if (whence == SEEK_SET) {
588 a672b469 aliguori
        /* nothing to do */
589 a672b469 aliguori
    } else if (whence == SEEK_CUR) {
590 a672b469 aliguori
        pos += qemu_ftell(f);
591 a672b469 aliguori
    } else {
592 a672b469 aliguori
        /* SEEK_END not supported */
593 a672b469 aliguori
        return -1;
594 a672b469 aliguori
    }
595 a672b469 aliguori
    if (f->put_buffer) {
596 a672b469 aliguori
        qemu_fflush(f);
597 a672b469 aliguori
        f->buf_offset = pos;
598 a672b469 aliguori
    } else {
599 a672b469 aliguori
        f->buf_offset = pos;
600 a672b469 aliguori
        f->buf_index = 0;
601 a672b469 aliguori
        f->buf_size = 0;
602 a672b469 aliguori
    }
603 a672b469 aliguori
    return pos;
604 a672b469 aliguori
}
605 a672b469 aliguori
606 a672b469 aliguori
int qemu_file_rate_limit(QEMUFile *f)
607 a672b469 aliguori
{
608 a672b469 aliguori
    if (f->rate_limit)
609 a672b469 aliguori
        return f->rate_limit(f->opaque);
610 a672b469 aliguori
611 a672b469 aliguori
    return 0;
612 a672b469 aliguori
}
613 a672b469 aliguori
614 3d002df3 Michael S. Tsirkin
int64_t qemu_file_get_rate_limit(QEMUFile *f)
615 c163b5ca lirans@il.ibm.com
{
616 c163b5ca lirans@il.ibm.com
    if (f->get_rate_limit)
617 c163b5ca lirans@il.ibm.com
        return f->get_rate_limit(f->opaque);
618 c163b5ca lirans@il.ibm.com
619 c163b5ca lirans@il.ibm.com
    return 0;
620 c163b5ca lirans@il.ibm.com
}
621 c163b5ca lirans@il.ibm.com
622 3d002df3 Michael S. Tsirkin
int64_t qemu_file_set_rate_limit(QEMUFile *f, int64_t new_rate)
623 19629537 Glauber Costa
{
624 0bb05eaf Glauber Costa
    /* any failed or completed migration keeps its state to allow probing of
625 0bb05eaf Glauber Costa
     * migration data, but has no associated file anymore */
626 0bb05eaf Glauber Costa
    if (f && f->set_rate_limit)
627 19629537 Glauber Costa
        return f->set_rate_limit(f->opaque, new_rate);
628 19629537 Glauber Costa
629 19629537 Glauber Costa
    return 0;
630 19629537 Glauber Costa
}
631 19629537 Glauber Costa
632 a672b469 aliguori
void qemu_put_be16(QEMUFile *f, unsigned int v)
633 a672b469 aliguori
{
634 a672b469 aliguori
    qemu_put_byte(f, v >> 8);
635 a672b469 aliguori
    qemu_put_byte(f, v);
636 a672b469 aliguori
}
637 a672b469 aliguori
638 a672b469 aliguori
void qemu_put_be32(QEMUFile *f, unsigned int v)
639 a672b469 aliguori
{
640 a672b469 aliguori
    qemu_put_byte(f, v >> 24);
641 a672b469 aliguori
    qemu_put_byte(f, v >> 16);
642 a672b469 aliguori
    qemu_put_byte(f, v >> 8);
643 a672b469 aliguori
    qemu_put_byte(f, v);
644 a672b469 aliguori
}
645 a672b469 aliguori
646 a672b469 aliguori
void qemu_put_be64(QEMUFile *f, uint64_t v)
647 a672b469 aliguori
{
648 a672b469 aliguori
    qemu_put_be32(f, v >> 32);
649 a672b469 aliguori
    qemu_put_be32(f, v);
650 a672b469 aliguori
}
651 a672b469 aliguori
652 a672b469 aliguori
unsigned int qemu_get_be16(QEMUFile *f)
653 a672b469 aliguori
{
654 a672b469 aliguori
    unsigned int v;
655 a672b469 aliguori
    v = qemu_get_byte(f) << 8;
656 a672b469 aliguori
    v |= qemu_get_byte(f);
657 a672b469 aliguori
    return v;
658 a672b469 aliguori
}
659 a672b469 aliguori
660 a672b469 aliguori
unsigned int qemu_get_be32(QEMUFile *f)
661 a672b469 aliguori
{
662 a672b469 aliguori
    unsigned int v;
663 a672b469 aliguori
    v = qemu_get_byte(f) << 24;
664 a672b469 aliguori
    v |= qemu_get_byte(f) << 16;
665 a672b469 aliguori
    v |= qemu_get_byte(f) << 8;
666 a672b469 aliguori
    v |= qemu_get_byte(f);
667 a672b469 aliguori
    return v;
668 a672b469 aliguori
}
669 a672b469 aliguori
670 a672b469 aliguori
uint64_t qemu_get_be64(QEMUFile *f)
671 a672b469 aliguori
{
672 a672b469 aliguori
    uint64_t v;
673 a672b469 aliguori
    v = (uint64_t)qemu_get_be32(f) << 32;
674 a672b469 aliguori
    v |= qemu_get_be32(f);
675 a672b469 aliguori
    return v;
676 a672b469 aliguori
}
677 a672b469 aliguori
678 cdae5cfb Gerd Hoffmann
/* bool */
679 cdae5cfb Gerd Hoffmann
680 cdae5cfb Gerd Hoffmann
static int get_bool(QEMUFile *f, void *pv, size_t size)
681 cdae5cfb Gerd Hoffmann
{
682 cdae5cfb Gerd Hoffmann
    bool *v = pv;
683 cdae5cfb Gerd Hoffmann
    *v = qemu_get_byte(f);
684 cdae5cfb Gerd Hoffmann
    return 0;
685 cdae5cfb Gerd Hoffmann
}
686 cdae5cfb Gerd Hoffmann
687 cdae5cfb Gerd Hoffmann
static void put_bool(QEMUFile *f, void *pv, size_t size)
688 cdae5cfb Gerd Hoffmann
{
689 cdae5cfb Gerd Hoffmann
    bool *v = pv;
690 cdae5cfb Gerd Hoffmann
    qemu_put_byte(f, *v);
691 cdae5cfb Gerd Hoffmann
}
692 cdae5cfb Gerd Hoffmann
693 cdae5cfb Gerd Hoffmann
const VMStateInfo vmstate_info_bool = {
694 cdae5cfb Gerd Hoffmann
    .name = "bool",
695 cdae5cfb Gerd Hoffmann
    .get  = get_bool,
696 cdae5cfb Gerd Hoffmann
    .put  = put_bool,
697 cdae5cfb Gerd Hoffmann
};
698 cdae5cfb Gerd Hoffmann
699 9ed7d6ae Juan Quintela
/* 8 bit int */
700 9ed7d6ae Juan Quintela
701 9ed7d6ae Juan Quintela
static int get_int8(QEMUFile *f, void *pv, size_t size)
702 9ed7d6ae Juan Quintela
{
703 9ed7d6ae Juan Quintela
    int8_t *v = pv;
704 9ed7d6ae Juan Quintela
    qemu_get_s8s(f, v);
705 9ed7d6ae Juan Quintela
    return 0;
706 9ed7d6ae Juan Quintela
}
707 9ed7d6ae Juan Quintela
708 84e2e3eb Juan Quintela
static void put_int8(QEMUFile *f, void *pv, size_t size)
709 9ed7d6ae Juan Quintela
{
710 84e2e3eb Juan Quintela
    int8_t *v = pv;
711 9ed7d6ae Juan Quintela
    qemu_put_s8s(f, v);
712 9ed7d6ae Juan Quintela
}
713 9ed7d6ae Juan Quintela
714 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int8 = {
715 9ed7d6ae Juan Quintela
    .name = "int8",
716 9ed7d6ae Juan Quintela
    .get  = get_int8,
717 9ed7d6ae Juan Quintela
    .put  = put_int8,
718 9ed7d6ae Juan Quintela
};
719 9ed7d6ae Juan Quintela
720 9ed7d6ae Juan Quintela
/* 16 bit int */
721 9ed7d6ae Juan Quintela
722 9ed7d6ae Juan Quintela
static int get_int16(QEMUFile *f, void *pv, size_t size)
723 9ed7d6ae Juan Quintela
{
724 9ed7d6ae Juan Quintela
    int16_t *v = pv;
725 9ed7d6ae Juan Quintela
    qemu_get_sbe16s(f, v);
726 9ed7d6ae Juan Quintela
    return 0;
727 9ed7d6ae Juan Quintela
}
728 9ed7d6ae Juan Quintela
729 84e2e3eb Juan Quintela
static void put_int16(QEMUFile *f, void *pv, size_t size)
730 9ed7d6ae Juan Quintela
{
731 84e2e3eb Juan Quintela
    int16_t *v = pv;
732 9ed7d6ae Juan Quintela
    qemu_put_sbe16s(f, v);
733 9ed7d6ae Juan Quintela
}
734 9ed7d6ae Juan Quintela
735 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int16 = {
736 9ed7d6ae Juan Quintela
    .name = "int16",
737 9ed7d6ae Juan Quintela
    .get  = get_int16,
738 9ed7d6ae Juan Quintela
    .put  = put_int16,
739 9ed7d6ae Juan Quintela
};
740 9ed7d6ae Juan Quintela
741 9ed7d6ae Juan Quintela
/* 32 bit int */
742 9ed7d6ae Juan Quintela
743 9ed7d6ae Juan Quintela
static int get_int32(QEMUFile *f, void *pv, size_t size)
744 9ed7d6ae Juan Quintela
{
745 9ed7d6ae Juan Quintela
    int32_t *v = pv;
746 9ed7d6ae Juan Quintela
    qemu_get_sbe32s(f, v);
747 9ed7d6ae Juan Quintela
    return 0;
748 9ed7d6ae Juan Quintela
}
749 9ed7d6ae Juan Quintela
750 84e2e3eb Juan Quintela
static void put_int32(QEMUFile *f, void *pv, size_t size)
751 9ed7d6ae Juan Quintela
{
752 84e2e3eb Juan Quintela
    int32_t *v = pv;
753 9ed7d6ae Juan Quintela
    qemu_put_sbe32s(f, v);
754 9ed7d6ae Juan Quintela
}
755 9ed7d6ae Juan Quintela
756 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int32 = {
757 9ed7d6ae Juan Quintela
    .name = "int32",
758 9ed7d6ae Juan Quintela
    .get  = get_int32,
759 9ed7d6ae Juan Quintela
    .put  = put_int32,
760 9ed7d6ae Juan Quintela
};
761 9ed7d6ae Juan Quintela
762 82501660 Juan Quintela
/* 32 bit int. See that the received value is the same than the one
763 82501660 Juan Quintela
   in the field */
764 82501660 Juan Quintela
765 82501660 Juan Quintela
static int get_int32_equal(QEMUFile *f, void *pv, size_t size)
766 82501660 Juan Quintela
{
767 82501660 Juan Quintela
    int32_t *v = pv;
768 82501660 Juan Quintela
    int32_t v2;
769 82501660 Juan Quintela
    qemu_get_sbe32s(f, &v2);
770 82501660 Juan Quintela
771 82501660 Juan Quintela
    if (*v == v2)
772 82501660 Juan Quintela
        return 0;
773 82501660 Juan Quintela
    return -EINVAL;
774 82501660 Juan Quintela
}
775 82501660 Juan Quintela
776 82501660 Juan Quintela
const VMStateInfo vmstate_info_int32_equal = {
777 82501660 Juan Quintela
    .name = "int32 equal",
778 82501660 Juan Quintela
    .get  = get_int32_equal,
779 82501660 Juan Quintela
    .put  = put_int32,
780 82501660 Juan Quintela
};
781 82501660 Juan Quintela
782 0a031e0a Juan Quintela
/* 32 bit int. See that the received value is the less or the same
783 0a031e0a Juan Quintela
   than the one in the field */
784 0a031e0a Juan Quintela
785 0a031e0a Juan Quintela
static int get_int32_le(QEMUFile *f, void *pv, size_t size)
786 0a031e0a Juan Quintela
{
787 0a031e0a Juan Quintela
    int32_t *old = pv;
788 0a031e0a Juan Quintela
    int32_t new;
789 0a031e0a Juan Quintela
    qemu_get_sbe32s(f, &new);
790 0a031e0a Juan Quintela
791 0a031e0a Juan Quintela
    if (*old <= new)
792 0a031e0a Juan Quintela
        return 0;
793 0a031e0a Juan Quintela
    return -EINVAL;
794 0a031e0a Juan Quintela
}
795 0a031e0a Juan Quintela
796 0a031e0a Juan Quintela
const VMStateInfo vmstate_info_int32_le = {
797 0a031e0a Juan Quintela
    .name = "int32 equal",
798 0a031e0a Juan Quintela
    .get  = get_int32_le,
799 0a031e0a Juan Quintela
    .put  = put_int32,
800 0a031e0a Juan Quintela
};
801 0a031e0a Juan Quintela
802 9ed7d6ae Juan Quintela
/* 64 bit int */
803 9ed7d6ae Juan Quintela
804 9ed7d6ae Juan Quintela
static int get_int64(QEMUFile *f, void *pv, size_t size)
805 9ed7d6ae Juan Quintela
{
806 9ed7d6ae Juan Quintela
    int64_t *v = pv;
807 9ed7d6ae Juan Quintela
    qemu_get_sbe64s(f, v);
808 9ed7d6ae Juan Quintela
    return 0;
809 9ed7d6ae Juan Quintela
}
810 9ed7d6ae Juan Quintela
811 84e2e3eb Juan Quintela
static void put_int64(QEMUFile *f, void *pv, size_t size)
812 9ed7d6ae Juan Quintela
{
813 84e2e3eb Juan Quintela
    int64_t *v = pv;
814 9ed7d6ae Juan Quintela
    qemu_put_sbe64s(f, v);
815 9ed7d6ae Juan Quintela
}
816 9ed7d6ae Juan Quintela
817 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int64 = {
818 9ed7d6ae Juan Quintela
    .name = "int64",
819 9ed7d6ae Juan Quintela
    .get  = get_int64,
820 9ed7d6ae Juan Quintela
    .put  = put_int64,
821 9ed7d6ae Juan Quintela
};
822 9ed7d6ae Juan Quintela
823 9ed7d6ae Juan Quintela
/* 8 bit unsigned int */
824 9ed7d6ae Juan Quintela
825 9ed7d6ae Juan Quintela
static int get_uint8(QEMUFile *f, void *pv, size_t size)
826 9ed7d6ae Juan Quintela
{
827 9ed7d6ae Juan Quintela
    uint8_t *v = pv;
828 9ed7d6ae Juan Quintela
    qemu_get_8s(f, v);
829 9ed7d6ae Juan Quintela
    return 0;
830 9ed7d6ae Juan Quintela
}
831 9ed7d6ae Juan Quintela
832 84e2e3eb Juan Quintela
static void put_uint8(QEMUFile *f, void *pv, size_t size)
833 9ed7d6ae Juan Quintela
{
834 84e2e3eb Juan Quintela
    uint8_t *v = pv;
835 9ed7d6ae Juan Quintela
    qemu_put_8s(f, v);
836 9ed7d6ae Juan Quintela
}
837 9ed7d6ae Juan Quintela
838 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint8 = {
839 9ed7d6ae Juan Quintela
    .name = "uint8",
840 9ed7d6ae Juan Quintela
    .get  = get_uint8,
841 9ed7d6ae Juan Quintela
    .put  = put_uint8,
842 9ed7d6ae Juan Quintela
};
843 9ed7d6ae Juan Quintela
844 9ed7d6ae Juan Quintela
/* 16 bit unsigned int */
845 9ed7d6ae Juan Quintela
846 9ed7d6ae Juan Quintela
static int get_uint16(QEMUFile *f, void *pv, size_t size)
847 9ed7d6ae Juan Quintela
{
848 9ed7d6ae Juan Quintela
    uint16_t *v = pv;
849 9ed7d6ae Juan Quintela
    qemu_get_be16s(f, v);
850 9ed7d6ae Juan Quintela
    return 0;
851 9ed7d6ae Juan Quintela
}
852 9ed7d6ae Juan Quintela
853 84e2e3eb Juan Quintela
static void put_uint16(QEMUFile *f, void *pv, size_t size)
854 9ed7d6ae Juan Quintela
{
855 84e2e3eb Juan Quintela
    uint16_t *v = pv;
856 9ed7d6ae Juan Quintela
    qemu_put_be16s(f, v);
857 9ed7d6ae Juan Quintela
}
858 9ed7d6ae Juan Quintela
859 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint16 = {
860 9ed7d6ae Juan Quintela
    .name = "uint16",
861 9ed7d6ae Juan Quintela
    .get  = get_uint16,
862 9ed7d6ae Juan Quintela
    .put  = put_uint16,
863 9ed7d6ae Juan Quintela
};
864 9ed7d6ae Juan Quintela
865 9ed7d6ae Juan Quintela
/* 32 bit unsigned int */
866 9ed7d6ae Juan Quintela
867 9ed7d6ae Juan Quintela
static int get_uint32(QEMUFile *f, void *pv, size_t size)
868 9ed7d6ae Juan Quintela
{
869 9ed7d6ae Juan Quintela
    uint32_t *v = pv;
870 9ed7d6ae Juan Quintela
    qemu_get_be32s(f, v);
871 9ed7d6ae Juan Quintela
    return 0;
872 9ed7d6ae Juan Quintela
}
873 9ed7d6ae Juan Quintela
874 84e2e3eb Juan Quintela
static void put_uint32(QEMUFile *f, void *pv, size_t size)
875 9ed7d6ae Juan Quintela
{
876 84e2e3eb Juan Quintela
    uint32_t *v = pv;
877 9ed7d6ae Juan Quintela
    qemu_put_be32s(f, v);
878 9ed7d6ae Juan Quintela
}
879 9ed7d6ae Juan Quintela
880 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint32 = {
881 9ed7d6ae Juan Quintela
    .name = "uint32",
882 9ed7d6ae Juan Quintela
    .get  = get_uint32,
883 9ed7d6ae Juan Quintela
    .put  = put_uint32,
884 9ed7d6ae Juan Quintela
};
885 9ed7d6ae Juan Quintela
886 9122a8fe Juan Quintela
/* 32 bit uint. See that the received value is the same than the one
887 9122a8fe Juan Quintela
   in the field */
888 9122a8fe Juan Quintela
889 9122a8fe Juan Quintela
static int get_uint32_equal(QEMUFile *f, void *pv, size_t size)
890 9122a8fe Juan Quintela
{
891 9122a8fe Juan Quintela
    uint32_t *v = pv;
892 9122a8fe Juan Quintela
    uint32_t v2;
893 9122a8fe Juan Quintela
    qemu_get_be32s(f, &v2);
894 9122a8fe Juan Quintela
895 9122a8fe Juan Quintela
    if (*v == v2) {
896 9122a8fe Juan Quintela
        return 0;
897 9122a8fe Juan Quintela
    }
898 9122a8fe Juan Quintela
    return -EINVAL;
899 9122a8fe Juan Quintela
}
900 9122a8fe Juan Quintela
901 9122a8fe Juan Quintela
const VMStateInfo vmstate_info_uint32_equal = {
902 9122a8fe Juan Quintela
    .name = "uint32 equal",
903 9122a8fe Juan Quintela
    .get  = get_uint32_equal,
904 9122a8fe Juan Quintela
    .put  = put_uint32,
905 9122a8fe Juan Quintela
};
906 9122a8fe Juan Quintela
907 9ed7d6ae Juan Quintela
/* 64 bit unsigned int */
908 9ed7d6ae Juan Quintela
909 9ed7d6ae Juan Quintela
static int get_uint64(QEMUFile *f, void *pv, size_t size)
910 9ed7d6ae Juan Quintela
{
911 9ed7d6ae Juan Quintela
    uint64_t *v = pv;
912 9ed7d6ae Juan Quintela
    qemu_get_be64s(f, v);
913 9ed7d6ae Juan Quintela
    return 0;
914 9ed7d6ae Juan Quintela
}
915 9ed7d6ae Juan Quintela
916 84e2e3eb Juan Quintela
static void put_uint64(QEMUFile *f, void *pv, size_t size)
917 9ed7d6ae Juan Quintela
{
918 84e2e3eb Juan Quintela
    uint64_t *v = pv;
919 9ed7d6ae Juan Quintela
    qemu_put_be64s(f, v);
920 9ed7d6ae Juan Quintela
}
921 9ed7d6ae Juan Quintela
922 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint64 = {
923 9ed7d6ae Juan Quintela
    .name = "uint64",
924 9ed7d6ae Juan Quintela
    .get  = get_uint64,
925 9ed7d6ae Juan Quintela
    .put  = put_uint64,
926 9ed7d6ae Juan Quintela
};
927 9ed7d6ae Juan Quintela
928 80cd83e7 Juan Quintela
/* 8 bit int. See that the received value is the same than the one
929 80cd83e7 Juan Quintela
   in the field */
930 80cd83e7 Juan Quintela
931 80cd83e7 Juan Quintela
static int get_uint8_equal(QEMUFile *f, void *pv, size_t size)
932 80cd83e7 Juan Quintela
{
933 80cd83e7 Juan Quintela
    uint8_t *v = pv;
934 80cd83e7 Juan Quintela
    uint8_t v2;
935 80cd83e7 Juan Quintela
    qemu_get_8s(f, &v2);
936 80cd83e7 Juan Quintela
937 80cd83e7 Juan Quintela
    if (*v == v2)
938 80cd83e7 Juan Quintela
        return 0;
939 80cd83e7 Juan Quintela
    return -EINVAL;
940 80cd83e7 Juan Quintela
}
941 80cd83e7 Juan Quintela
942 80cd83e7 Juan Quintela
const VMStateInfo vmstate_info_uint8_equal = {
943 aa1cce69 Juan Quintela
    .name = "uint8 equal",
944 80cd83e7 Juan Quintela
    .get  = get_uint8_equal,
945 80cd83e7 Juan Quintela
    .put  = put_uint8,
946 80cd83e7 Juan Quintela
};
947 80cd83e7 Juan Quintela
948 dc3b83a0 Juan Quintela
/* 16 bit unsigned int int. See that the received value is the same than the one
949 dc3b83a0 Juan Quintela
   in the field */
950 dc3b83a0 Juan Quintela
951 dc3b83a0 Juan Quintela
static int get_uint16_equal(QEMUFile *f, void *pv, size_t size)
952 dc3b83a0 Juan Quintela
{
953 dc3b83a0 Juan Quintela
    uint16_t *v = pv;
954 dc3b83a0 Juan Quintela
    uint16_t v2;
955 dc3b83a0 Juan Quintela
    qemu_get_be16s(f, &v2);
956 dc3b83a0 Juan Quintela
957 dc3b83a0 Juan Quintela
    if (*v == v2)
958 dc3b83a0 Juan Quintela
        return 0;
959 dc3b83a0 Juan Quintela
    return -EINVAL;
960 dc3b83a0 Juan Quintela
}
961 dc3b83a0 Juan Quintela
962 dc3b83a0 Juan Quintela
const VMStateInfo vmstate_info_uint16_equal = {
963 dc3b83a0 Juan Quintela
    .name = "uint16 equal",
964 dc3b83a0 Juan Quintela
    .get  = get_uint16_equal,
965 dc3b83a0 Juan Quintela
    .put  = put_uint16,
966 dc3b83a0 Juan Quintela
};
967 dc3b83a0 Juan Quintela
968 dde0463b Juan Quintela
/* timers  */
969 dde0463b Juan Quintela
970 dde0463b Juan Quintela
static int get_timer(QEMUFile *f, void *pv, size_t size)
971 dde0463b Juan Quintela
{
972 dde0463b Juan Quintela
    QEMUTimer *v = pv;
973 dde0463b Juan Quintela
    qemu_get_timer(f, v);
974 dde0463b Juan Quintela
    return 0;
975 dde0463b Juan Quintela
}
976 dde0463b Juan Quintela
977 84e2e3eb Juan Quintela
static void put_timer(QEMUFile *f, void *pv, size_t size)
978 dde0463b Juan Quintela
{
979 84e2e3eb Juan Quintela
    QEMUTimer *v = pv;
980 dde0463b Juan Quintela
    qemu_put_timer(f, v);
981 dde0463b Juan Quintela
}
982 dde0463b Juan Quintela
983 dde0463b Juan Quintela
const VMStateInfo vmstate_info_timer = {
984 dde0463b Juan Quintela
    .name = "timer",
985 dde0463b Juan Quintela
    .get  = get_timer,
986 dde0463b Juan Quintela
    .put  = put_timer,
987 dde0463b Juan Quintela
};
988 dde0463b Juan Quintela
989 6f67c50f Juan Quintela
/* uint8_t buffers */
990 6f67c50f Juan Quintela
991 6f67c50f Juan Quintela
static int get_buffer(QEMUFile *f, void *pv, size_t size)
992 6f67c50f Juan Quintela
{
993 6f67c50f Juan Quintela
    uint8_t *v = pv;
994 6f67c50f Juan Quintela
    qemu_get_buffer(f, v, size);
995 6f67c50f Juan Quintela
    return 0;
996 6f67c50f Juan Quintela
}
997 6f67c50f Juan Quintela
998 84e2e3eb Juan Quintela
static void put_buffer(QEMUFile *f, void *pv, size_t size)
999 6f67c50f Juan Quintela
{
1000 84e2e3eb Juan Quintela
    uint8_t *v = pv;
1001 6f67c50f Juan Quintela
    qemu_put_buffer(f, v, size);
1002 6f67c50f Juan Quintela
}
1003 6f67c50f Juan Quintela
1004 6f67c50f Juan Quintela
const VMStateInfo vmstate_info_buffer = {
1005 6f67c50f Juan Quintela
    .name = "buffer",
1006 6f67c50f Juan Quintela
    .get  = get_buffer,
1007 6f67c50f Juan Quintela
    .put  = put_buffer,
1008 6f67c50f Juan Quintela
};
1009 6f67c50f Juan Quintela
1010 76507c75 Juan Quintela
/* unused buffers: space that was used for some fields that are
1011 61cc8701 Stefan Weil
   not useful anymore */
1012 76507c75 Juan Quintela
1013 76507c75 Juan Quintela
static int get_unused_buffer(QEMUFile *f, void *pv, size_t size)
1014 76507c75 Juan Quintela
{
1015 21174c34 Jan Kiszka
    uint8_t buf[1024];
1016 21174c34 Jan Kiszka
    int block_len;
1017 21174c34 Jan Kiszka
1018 21174c34 Jan Kiszka
    while (size > 0) {
1019 21174c34 Jan Kiszka
        block_len = MIN(sizeof(buf), size);
1020 21174c34 Jan Kiszka
        size -= block_len;
1021 21174c34 Jan Kiszka
        qemu_get_buffer(f, buf, block_len);
1022 21174c34 Jan Kiszka
    }
1023 21174c34 Jan Kiszka
   return 0;
1024 76507c75 Juan Quintela
}
1025 76507c75 Juan Quintela
1026 76507c75 Juan Quintela
static void put_unused_buffer(QEMUFile *f, void *pv, size_t size)
1027 76507c75 Juan Quintela
{
1028 21174c34 Jan Kiszka
    static const uint8_t buf[1024];
1029 21174c34 Jan Kiszka
    int block_len;
1030 21174c34 Jan Kiszka
1031 21174c34 Jan Kiszka
    while (size > 0) {
1032 21174c34 Jan Kiszka
        block_len = MIN(sizeof(buf), size);
1033 21174c34 Jan Kiszka
        size -= block_len;
1034 21174c34 Jan Kiszka
        qemu_put_buffer(f, buf, block_len);
1035 21174c34 Jan Kiszka
    }
1036 76507c75 Juan Quintela
}
1037 76507c75 Juan Quintela
1038 76507c75 Juan Quintela
const VMStateInfo vmstate_info_unused_buffer = {
1039 76507c75 Juan Quintela
    .name = "unused_buffer",
1040 76507c75 Juan Quintela
    .get  = get_unused_buffer,
1041 76507c75 Juan Quintela
    .put  = put_unused_buffer,
1042 76507c75 Juan Quintela
};
1043 76507c75 Juan Quintela
1044 7685ee6a Alex Williamson
typedef struct CompatEntry {
1045 7685ee6a Alex Williamson
    char idstr[256];
1046 7685ee6a Alex Williamson
    int instance_id;
1047 7685ee6a Alex Williamson
} CompatEntry;
1048 7685ee6a Alex Williamson
1049 a672b469 aliguori
typedef struct SaveStateEntry {
1050 72cf2d4f Blue Swirl
    QTAILQ_ENTRY(SaveStateEntry) entry;
1051 a672b469 aliguori
    char idstr[256];
1052 a672b469 aliguori
    int instance_id;
1053 4d2ffa08 Jan Kiszka
    int alias_id;
1054 a672b469 aliguori
    int version_id;
1055 a672b469 aliguori
    int section_id;
1056 c163b5ca lirans@il.ibm.com
    SaveSetParamsHandler *set_params;
1057 a672b469 aliguori
    SaveLiveStateHandler *save_live_state;
1058 a672b469 aliguori
    SaveStateHandler *save_state;
1059 a672b469 aliguori
    LoadStateHandler *load_state;
1060 9ed7d6ae Juan Quintela
    const VMStateDescription *vmsd;
1061 a672b469 aliguori
    void *opaque;
1062 7685ee6a Alex Williamson
    CompatEntry *compat;
1063 24312968 Cam Macdonell
    int no_migrate;
1064 a672b469 aliguori
} SaveStateEntry;
1065 a672b469 aliguori
1066 c163b5ca lirans@il.ibm.com
1067 72cf2d4f Blue Swirl
static QTAILQ_HEAD(savevm_handlers, SaveStateEntry) savevm_handlers =
1068 72cf2d4f Blue Swirl
    QTAILQ_HEAD_INITIALIZER(savevm_handlers);
1069 9ed7d6ae Juan Quintela
static int global_section_id;
1070 a672b469 aliguori
1071 8718e999 Juan Quintela
static int calculate_new_instance_id(const char *idstr)
1072 8718e999 Juan Quintela
{
1073 8718e999 Juan Quintela
    SaveStateEntry *se;
1074 8718e999 Juan Quintela
    int instance_id = 0;
1075 8718e999 Juan Quintela
1076 72cf2d4f Blue Swirl
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1077 8718e999 Juan Quintela
        if (strcmp(idstr, se->idstr) == 0
1078 8718e999 Juan Quintela
            && instance_id <= se->instance_id) {
1079 8718e999 Juan Quintela
            instance_id = se->instance_id + 1;
1080 8718e999 Juan Quintela
        }
1081 8718e999 Juan Quintela
    }
1082 8718e999 Juan Quintela
    return instance_id;
1083 8718e999 Juan Quintela
}
1084 8718e999 Juan Quintela
1085 7685ee6a Alex Williamson
static int calculate_compat_instance_id(const char *idstr)
1086 7685ee6a Alex Williamson
{
1087 7685ee6a Alex Williamson
    SaveStateEntry *se;
1088 7685ee6a Alex Williamson
    int instance_id = 0;
1089 7685ee6a Alex Williamson
1090 7685ee6a Alex Williamson
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1091 7685ee6a Alex Williamson
        if (!se->compat)
1092 7685ee6a Alex Williamson
            continue;
1093 7685ee6a Alex Williamson
1094 7685ee6a Alex Williamson
        if (strcmp(idstr, se->compat->idstr) == 0
1095 7685ee6a Alex Williamson
            && instance_id <= se->compat->instance_id) {
1096 7685ee6a Alex Williamson
            instance_id = se->compat->instance_id + 1;
1097 7685ee6a Alex Williamson
        }
1098 7685ee6a Alex Williamson
    }
1099 7685ee6a Alex Williamson
    return instance_id;
1100 7685ee6a Alex Williamson
}
1101 7685ee6a Alex Williamson
1102 a672b469 aliguori
/* TODO: Individual devices generally have very little idea about the rest
1103 a672b469 aliguori
   of the system, so instance_id should be removed/replaced.
1104 a672b469 aliguori
   Meanwhile pass -1 as instance_id if you do not already have a clearly
1105 a672b469 aliguori
   distinguishing id for all instances of your device class. */
1106 0be71e32 Alex Williamson
int register_savevm_live(DeviceState *dev,
1107 0be71e32 Alex Williamson
                         const char *idstr,
1108 a672b469 aliguori
                         int instance_id,
1109 a672b469 aliguori
                         int version_id,
1110 c163b5ca lirans@il.ibm.com
                         SaveSetParamsHandler *set_params,
1111 a672b469 aliguori
                         SaveLiveStateHandler *save_live_state,
1112 a672b469 aliguori
                         SaveStateHandler *save_state,
1113 a672b469 aliguori
                         LoadStateHandler *load_state,
1114 a672b469 aliguori
                         void *opaque)
1115 a672b469 aliguori
{
1116 8718e999 Juan Quintela
    SaveStateEntry *se;
1117 a672b469 aliguori
1118 c163b5ca lirans@il.ibm.com
    se = qemu_mallocz(sizeof(SaveStateEntry));
1119 a672b469 aliguori
    se->version_id = version_id;
1120 a672b469 aliguori
    se->section_id = global_section_id++;
1121 c163b5ca lirans@il.ibm.com
    se->set_params = set_params;
1122 a672b469 aliguori
    se->save_live_state = save_live_state;
1123 a672b469 aliguori
    se->save_state = save_state;
1124 a672b469 aliguori
    se->load_state = load_state;
1125 a672b469 aliguori
    se->opaque = opaque;
1126 9ed7d6ae Juan Quintela
    se->vmsd = NULL;
1127 24312968 Cam Macdonell
    se->no_migrate = 0;
1128 a672b469 aliguori
1129 7685ee6a Alex Williamson
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1130 7685ee6a Alex Williamson
        char *id = dev->parent_bus->info->get_dev_path(dev);
1131 7685ee6a Alex Williamson
        if (id) {
1132 7685ee6a Alex Williamson
            pstrcpy(se->idstr, sizeof(se->idstr), id);
1133 7685ee6a Alex Williamson
            pstrcat(se->idstr, sizeof(se->idstr), "/");
1134 7685ee6a Alex Williamson
            qemu_free(id);
1135 7685ee6a Alex Williamson
1136 7685ee6a Alex Williamson
            se->compat = qemu_mallocz(sizeof(CompatEntry));
1137 7685ee6a Alex Williamson
            pstrcpy(se->compat->idstr, sizeof(se->compat->idstr), idstr);
1138 7685ee6a Alex Williamson
            se->compat->instance_id = instance_id == -1 ?
1139 7685ee6a Alex Williamson
                         calculate_compat_instance_id(idstr) : instance_id;
1140 7685ee6a Alex Williamson
            instance_id = -1;
1141 7685ee6a Alex Williamson
        }
1142 7685ee6a Alex Williamson
    }
1143 7685ee6a Alex Williamson
    pstrcat(se->idstr, sizeof(se->idstr), idstr);
1144 7685ee6a Alex Williamson
1145 8718e999 Juan Quintela
    if (instance_id == -1) {
1146 7685ee6a Alex Williamson
        se->instance_id = calculate_new_instance_id(se->idstr);
1147 8718e999 Juan Quintela
    } else {
1148 8718e999 Juan Quintela
        se->instance_id = instance_id;
1149 a672b469 aliguori
    }
1150 7685ee6a Alex Williamson
    assert(!se->compat || se->instance_id == 0);
1151 8718e999 Juan Quintela
    /* add at the end of list */
1152 72cf2d4f Blue Swirl
    QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry);
1153 a672b469 aliguori
    return 0;
1154 a672b469 aliguori
}
1155 a672b469 aliguori
1156 0be71e32 Alex Williamson
int register_savevm(DeviceState *dev,
1157 0be71e32 Alex Williamson
                    const char *idstr,
1158 a672b469 aliguori
                    int instance_id,
1159 a672b469 aliguori
                    int version_id,
1160 a672b469 aliguori
                    SaveStateHandler *save_state,
1161 a672b469 aliguori
                    LoadStateHandler *load_state,
1162 a672b469 aliguori
                    void *opaque)
1163 a672b469 aliguori
{
1164 0be71e32 Alex Williamson
    return register_savevm_live(dev, idstr, instance_id, version_id,
1165 c163b5ca lirans@il.ibm.com
                                NULL, NULL, save_state, load_state, opaque);
1166 a672b469 aliguori
}
1167 a672b469 aliguori
1168 0be71e32 Alex Williamson
void unregister_savevm(DeviceState *dev, const char *idstr, void *opaque)
1169 41bd13af aliguori
{
1170 8718e999 Juan Quintela
    SaveStateEntry *se, *new_se;
1171 7685ee6a Alex Williamson
    char id[256] = "";
1172 7685ee6a Alex Williamson
1173 7685ee6a Alex Williamson
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1174 7685ee6a Alex Williamson
        char *path = dev->parent_bus->info->get_dev_path(dev);
1175 7685ee6a Alex Williamson
        if (path) {
1176 7685ee6a Alex Williamson
            pstrcpy(id, sizeof(id), path);
1177 7685ee6a Alex Williamson
            pstrcat(id, sizeof(id), "/");
1178 7685ee6a Alex Williamson
            qemu_free(path);
1179 7685ee6a Alex Williamson
        }
1180 7685ee6a Alex Williamson
    }
1181 7685ee6a Alex Williamson
    pstrcat(id, sizeof(id), idstr);
1182 41bd13af aliguori
1183 72cf2d4f Blue Swirl
    QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) {
1184 7685ee6a Alex Williamson
        if (strcmp(se->idstr, id) == 0 && se->opaque == opaque) {
1185 72cf2d4f Blue Swirl
            QTAILQ_REMOVE(&savevm_handlers, se, entry);
1186 69e58af9 Alex Williamson
            if (se->compat) {
1187 69e58af9 Alex Williamson
                qemu_free(se->compat);
1188 69e58af9 Alex Williamson
            }
1189 8718e999 Juan Quintela
            qemu_free(se);
1190 41bd13af aliguori
        }
1191 41bd13af aliguori
    }
1192 41bd13af aliguori
}
1193 41bd13af aliguori
1194 24312968 Cam Macdonell
/* mark a device as not to be migrated, that is the device should be
1195 24312968 Cam Macdonell
   unplugged before migration */
1196 24312968 Cam Macdonell
void register_device_unmigratable(DeviceState *dev, const char *idstr,
1197 24312968 Cam Macdonell
                                                            void *opaque)
1198 24312968 Cam Macdonell
{
1199 24312968 Cam Macdonell
    SaveStateEntry *se;
1200 24312968 Cam Macdonell
    char id[256] = "";
1201 24312968 Cam Macdonell
1202 24312968 Cam Macdonell
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1203 24312968 Cam Macdonell
        char *path = dev->parent_bus->info->get_dev_path(dev);
1204 24312968 Cam Macdonell
        if (path) {
1205 24312968 Cam Macdonell
            pstrcpy(id, sizeof(id), path);
1206 24312968 Cam Macdonell
            pstrcat(id, sizeof(id), "/");
1207 24312968 Cam Macdonell
            qemu_free(path);
1208 24312968 Cam Macdonell
        }
1209 24312968 Cam Macdonell
    }
1210 24312968 Cam Macdonell
    pstrcat(id, sizeof(id), idstr);
1211 24312968 Cam Macdonell
1212 24312968 Cam Macdonell
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1213 24312968 Cam Macdonell
        if (strcmp(se->idstr, id) == 0 && se->opaque == opaque) {
1214 24312968 Cam Macdonell
            se->no_migrate = 1;
1215 24312968 Cam Macdonell
        }
1216 24312968 Cam Macdonell
    }
1217 24312968 Cam Macdonell
}
1218 24312968 Cam Macdonell
1219 0be71e32 Alex Williamson
int vmstate_register_with_alias_id(DeviceState *dev, int instance_id,
1220 4d2ffa08 Jan Kiszka
                                   const VMStateDescription *vmsd,
1221 4d2ffa08 Jan Kiszka
                                   void *opaque, int alias_id,
1222 4d2ffa08 Jan Kiszka
                                   int required_for_version)
1223 9ed7d6ae Juan Quintela
{
1224 8718e999 Juan Quintela
    SaveStateEntry *se;
1225 9ed7d6ae Juan Quintela
1226 4d2ffa08 Jan Kiszka
    /* If this triggers, alias support can be dropped for the vmsd. */
1227 4d2ffa08 Jan Kiszka
    assert(alias_id == -1 || required_for_version >= vmsd->minimum_version_id);
1228 4d2ffa08 Jan Kiszka
1229 c163b5ca lirans@il.ibm.com
    se = qemu_mallocz(sizeof(SaveStateEntry));
1230 9ed7d6ae Juan Quintela
    se->version_id = vmsd->version_id;
1231 9ed7d6ae Juan Quintela
    se->section_id = global_section_id++;
1232 9ed7d6ae Juan Quintela
    se->save_live_state = NULL;
1233 9ed7d6ae Juan Quintela
    se->save_state = NULL;
1234 9ed7d6ae Juan Quintela
    se->load_state = NULL;
1235 9ed7d6ae Juan Quintela
    se->opaque = opaque;
1236 9ed7d6ae Juan Quintela
    se->vmsd = vmsd;
1237 4d2ffa08 Jan Kiszka
    se->alias_id = alias_id;
1238 9ed7d6ae Juan Quintela
1239 7685ee6a Alex Williamson
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1240 7685ee6a Alex Williamson
        char *id = dev->parent_bus->info->get_dev_path(dev);
1241 7685ee6a Alex Williamson
        if (id) {
1242 7685ee6a Alex Williamson
            pstrcpy(se->idstr, sizeof(se->idstr), id);
1243 7685ee6a Alex Williamson
            pstrcat(se->idstr, sizeof(se->idstr), "/");
1244 7685ee6a Alex Williamson
            qemu_free(id);
1245 7685ee6a Alex Williamson
1246 7685ee6a Alex Williamson
            se->compat = qemu_mallocz(sizeof(CompatEntry));
1247 7685ee6a Alex Williamson
            pstrcpy(se->compat->idstr, sizeof(se->compat->idstr), vmsd->name);
1248 7685ee6a Alex Williamson
            se->compat->instance_id = instance_id == -1 ?
1249 7685ee6a Alex Williamson
                         calculate_compat_instance_id(vmsd->name) : instance_id;
1250 7685ee6a Alex Williamson
            instance_id = -1;
1251 7685ee6a Alex Williamson
        }
1252 7685ee6a Alex Williamson
    }
1253 7685ee6a Alex Williamson
    pstrcat(se->idstr, sizeof(se->idstr), vmsd->name);
1254 7685ee6a Alex Williamson
1255 8718e999 Juan Quintela
    if (instance_id == -1) {
1256 7685ee6a Alex Williamson
        se->instance_id = calculate_new_instance_id(se->idstr);
1257 8718e999 Juan Quintela
    } else {
1258 8718e999 Juan Quintela
        se->instance_id = instance_id;
1259 9ed7d6ae Juan Quintela
    }
1260 7685ee6a Alex Williamson
    assert(!se->compat || se->instance_id == 0);
1261 8718e999 Juan Quintela
    /* add at the end of list */
1262 72cf2d4f Blue Swirl
    QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry);
1263 9ed7d6ae Juan Quintela
    return 0;
1264 9ed7d6ae Juan Quintela
}
1265 9ed7d6ae Juan Quintela
1266 0be71e32 Alex Williamson
int vmstate_register(DeviceState *dev, int instance_id,
1267 0be71e32 Alex Williamson
                     const VMStateDescription *vmsd, void *opaque)
1268 4d2ffa08 Jan Kiszka
{
1269 0be71e32 Alex Williamson
    return vmstate_register_with_alias_id(dev, instance_id, vmsd,
1270 0be71e32 Alex Williamson
                                          opaque, -1, 0);
1271 4d2ffa08 Jan Kiszka
}
1272 4d2ffa08 Jan Kiszka
1273 0be71e32 Alex Williamson
void vmstate_unregister(DeviceState *dev, const VMStateDescription *vmsd,
1274 0be71e32 Alex Williamson
                        void *opaque)
1275 9ed7d6ae Juan Quintela
{
1276 1eb7538b Juan Quintela
    SaveStateEntry *se, *new_se;
1277 1eb7538b Juan Quintela
1278 72cf2d4f Blue Swirl
    QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) {
1279 1eb7538b Juan Quintela
        if (se->vmsd == vmsd && se->opaque == opaque) {
1280 72cf2d4f Blue Swirl
            QTAILQ_REMOVE(&savevm_handlers, se, entry);
1281 69e58af9 Alex Williamson
            if (se->compat) {
1282 69e58af9 Alex Williamson
                qemu_free(se->compat);
1283 69e58af9 Alex Williamson
            }
1284 1eb7538b Juan Quintela
            qemu_free(se);
1285 1eb7538b Juan Quintela
        }
1286 1eb7538b Juan Quintela
    }
1287 9ed7d6ae Juan Quintela
}
1288 9ed7d6ae Juan Quintela
1289 811814bd Juan Quintela
static void vmstate_subsection_save(QEMUFile *f, const VMStateDescription *vmsd,
1290 811814bd Juan Quintela
                                    void *opaque);
1291 811814bd Juan Quintela
static int vmstate_subsection_load(QEMUFile *f, const VMStateDescription *vmsd,
1292 811814bd Juan Quintela
                                   void *opaque);
1293 811814bd Juan Quintela
1294 9ed7d6ae Juan Quintela
int vmstate_load_state(QEMUFile *f, const VMStateDescription *vmsd,
1295 9ed7d6ae Juan Quintela
                       void *opaque, int version_id)
1296 9ed7d6ae Juan Quintela
{
1297 9ed7d6ae Juan Quintela
    VMStateField *field = vmsd->fields;
1298 811814bd Juan Quintela
    int ret;
1299 9ed7d6ae Juan Quintela
1300 9ed7d6ae Juan Quintela
    if (version_id > vmsd->version_id) {
1301 9ed7d6ae Juan Quintela
        return -EINVAL;
1302 9ed7d6ae Juan Quintela
    }
1303 9ed7d6ae Juan Quintela
    if (version_id < vmsd->minimum_version_id_old) {
1304 9ed7d6ae Juan Quintela
        return -EINVAL;
1305 9ed7d6ae Juan Quintela
    }
1306 9ed7d6ae Juan Quintela
    if  (version_id < vmsd->minimum_version_id) {
1307 9ed7d6ae Juan Quintela
        return vmsd->load_state_old(f, opaque, version_id);
1308 9ed7d6ae Juan Quintela
    }
1309 fd4d52de Juan Quintela
    if (vmsd->pre_load) {
1310 fd4d52de Juan Quintela
        int ret = vmsd->pre_load(opaque);
1311 fd4d52de Juan Quintela
        if (ret)
1312 fd4d52de Juan Quintela
            return ret;
1313 fd4d52de Juan Quintela
    }
1314 9ed7d6ae Juan Quintela
    while(field->name) {
1315 f11f6a5f Juan Quintela
        if ((field->field_exists &&
1316 f11f6a5f Juan Quintela
             field->field_exists(opaque, version_id)) ||
1317 f11f6a5f Juan Quintela
            (!field->field_exists &&
1318 f11f6a5f Juan Quintela
             field->version_id <= version_id)) {
1319 f752a6aa Juan Quintela
            void *base_addr = opaque + field->offset;
1320 811814bd Juan Quintela
            int i, n_elems = 1;
1321 e61a1e0a Juan Quintela
            int size = field->size;
1322 9ed7d6ae Juan Quintela
1323 e61a1e0a Juan Quintela
            if (field->flags & VMS_VBUFFER) {
1324 e61a1e0a Juan Quintela
                size = *(int32_t *)(opaque+field->size_offset);
1325 33599e2a Juan Quintela
                if (field->flags & VMS_MULTIPLY) {
1326 33599e2a Juan Quintela
                    size *= field->size;
1327 33599e2a Juan Quintela
                }
1328 e61a1e0a Juan Quintela
            }
1329 f752a6aa Juan Quintela
            if (field->flags & VMS_ARRAY) {
1330 f752a6aa Juan Quintela
                n_elems = field->num;
1331 d6698281 Juan Quintela
            } else if (field->flags & VMS_VARRAY_INT32) {
1332 d6698281 Juan Quintela
                n_elems = *(int32_t *)(opaque+field->num_offset);
1333 a624b086 Juan Quintela
            } else if (field->flags & VMS_VARRAY_UINT32) {
1334 a624b086 Juan Quintela
                n_elems = *(uint32_t *)(opaque+field->num_offset);
1335 bdb4941d Juan Quintela
            } else if (field->flags & VMS_VARRAY_UINT16) {
1336 bdb4941d Juan Quintela
                n_elems = *(uint16_t *)(opaque+field->num_offset);
1337 82fa39b7 Juan Quintela
            } else if (field->flags & VMS_VARRAY_UINT8) {
1338 82fa39b7 Juan Quintela
                n_elems = *(uint8_t *)(opaque+field->num_offset);
1339 f752a6aa Juan Quintela
            }
1340 dde0463b Juan Quintela
            if (field->flags & VMS_POINTER) {
1341 e61a1e0a Juan Quintela
                base_addr = *(void **)base_addr + field->start;
1342 dde0463b Juan Quintela
            }
1343 f752a6aa Juan Quintela
            for (i = 0; i < n_elems; i++) {
1344 e61a1e0a Juan Quintela
                void *addr = base_addr + size * i;
1345 ec245e21 Juan Quintela
1346 19df438b Juan Quintela
                if (field->flags & VMS_ARRAY_OF_POINTER) {
1347 19df438b Juan Quintela
                    addr = *(void **)addr;
1348 19df438b Juan Quintela
                }
1349 ec245e21 Juan Quintela
                if (field->flags & VMS_STRUCT) {
1350 fa3aad24 Juan Quintela
                    ret = vmstate_load_state(f, field->vmsd, addr, field->vmsd->version_id);
1351 ec245e21 Juan Quintela
                } else {
1352 e61a1e0a Juan Quintela
                    ret = field->info->get(f, addr, size);
1353 ec245e21 Juan Quintela
1354 ec245e21 Juan Quintela
                }
1355 f752a6aa Juan Quintela
                if (ret < 0) {
1356 f752a6aa Juan Quintela
                    return ret;
1357 f752a6aa Juan Quintela
                }
1358 9ed7d6ae Juan Quintela
            }
1359 9ed7d6ae Juan Quintela
        }
1360 9ed7d6ae Juan Quintela
        field++;
1361 9ed7d6ae Juan Quintela
    }
1362 811814bd Juan Quintela
    ret = vmstate_subsection_load(f, vmsd, opaque);
1363 811814bd Juan Quintela
    if (ret != 0) {
1364 811814bd Juan Quintela
        return ret;
1365 811814bd Juan Quintela
    }
1366 752ff2fa Juan Quintela
    if (vmsd->post_load) {
1367 e59fb374 Juan Quintela
        return vmsd->post_load(opaque, version_id);
1368 752ff2fa Juan Quintela
    }
1369 9ed7d6ae Juan Quintela
    return 0;
1370 9ed7d6ae Juan Quintela
}
1371 9ed7d6ae Juan Quintela
1372 9ed7d6ae Juan Quintela
void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd,
1373 84e2e3eb Juan Quintela
                        void *opaque)
1374 9ed7d6ae Juan Quintela
{
1375 9ed7d6ae Juan Quintela
    VMStateField *field = vmsd->fields;
1376 9ed7d6ae Juan Quintela
1377 8fb0791d Juan Quintela
    if (vmsd->pre_save) {
1378 8fb0791d Juan Quintela
        vmsd->pre_save(opaque);
1379 8fb0791d Juan Quintela
    }
1380 9ed7d6ae Juan Quintela
    while(field->name) {
1381 f11f6a5f Juan Quintela
        if (!field->field_exists ||
1382 f11f6a5f Juan Quintela
            field->field_exists(opaque, vmsd->version_id)) {
1383 f11f6a5f Juan Quintela
            void *base_addr = opaque + field->offset;
1384 f11f6a5f Juan Quintela
            int i, n_elems = 1;
1385 e61a1e0a Juan Quintela
            int size = field->size;
1386 dde0463b Juan Quintela
1387 e61a1e0a Juan Quintela
            if (field->flags & VMS_VBUFFER) {
1388 e61a1e0a Juan Quintela
                size = *(int32_t *)(opaque+field->size_offset);
1389 33599e2a Juan Quintela
                if (field->flags & VMS_MULTIPLY) {
1390 33599e2a Juan Quintela
                    size *= field->size;
1391 33599e2a Juan Quintela
                }
1392 e61a1e0a Juan Quintela
            }
1393 f11f6a5f Juan Quintela
            if (field->flags & VMS_ARRAY) {
1394 f11f6a5f Juan Quintela
                n_elems = field->num;
1395 d6698281 Juan Quintela
            } else if (field->flags & VMS_VARRAY_INT32) {
1396 d6698281 Juan Quintela
                n_elems = *(int32_t *)(opaque+field->num_offset);
1397 bdb4941d Juan Quintela
            } else if (field->flags & VMS_VARRAY_UINT16) {
1398 bdb4941d Juan Quintela
                n_elems = *(uint16_t *)(opaque+field->num_offset);
1399 b784421c Juan Quintela
            } else if (field->flags & VMS_VARRAY_UINT8) {
1400 b784421c Juan Quintela
                n_elems = *(uint8_t *)(opaque+field->num_offset);
1401 f11f6a5f Juan Quintela
            }
1402 f11f6a5f Juan Quintela
            if (field->flags & VMS_POINTER) {
1403 e61a1e0a Juan Quintela
                base_addr = *(void **)base_addr + field->start;
1404 f11f6a5f Juan Quintela
            }
1405 f11f6a5f Juan Quintela
            for (i = 0; i < n_elems; i++) {
1406 e61a1e0a Juan Quintela
                void *addr = base_addr + size * i;
1407 ec245e21 Juan Quintela
1408 8595387e Juan Quintela
                if (field->flags & VMS_ARRAY_OF_POINTER) {
1409 8595387e Juan Quintela
                    addr = *(void **)addr;
1410 8595387e Juan Quintela
                }
1411 f11f6a5f Juan Quintela
                if (field->flags & VMS_STRUCT) {
1412 f11f6a5f Juan Quintela
                    vmstate_save_state(f, field->vmsd, addr);
1413 f11f6a5f Juan Quintela
                } else {
1414 e61a1e0a Juan Quintela
                    field->info->put(f, addr, size);
1415 f11f6a5f Juan Quintela
                }
1416 ec245e21 Juan Quintela
            }
1417 dde0463b Juan Quintela
        }
1418 9ed7d6ae Juan Quintela
        field++;
1419 9ed7d6ae Juan Quintela
    }
1420 811814bd Juan Quintela
    vmstate_subsection_save(f, vmsd, opaque);
1421 9ed7d6ae Juan Quintela
}
1422 9ed7d6ae Juan Quintela
1423 4082be4d Juan Quintela
static int vmstate_load(QEMUFile *f, SaveStateEntry *se, int version_id)
1424 4082be4d Juan Quintela
{
1425 9ed7d6ae Juan Quintela
    if (!se->vmsd) {         /* Old style */
1426 9ed7d6ae Juan Quintela
        return se->load_state(f, se->opaque, version_id);
1427 9ed7d6ae Juan Quintela
    }
1428 9ed7d6ae Juan Quintela
    return vmstate_load_state(f, se->vmsd, se->opaque, version_id);
1429 4082be4d Juan Quintela
}
1430 4082be4d Juan Quintela
1431 dc912121 Alex Williamson
static void vmstate_save(QEMUFile *f, SaveStateEntry *se)
1432 4082be4d Juan Quintela
{
1433 9ed7d6ae Juan Quintela
    if (!se->vmsd) {         /* Old style */
1434 9ed7d6ae Juan Quintela
        se->save_state(f, se->opaque);
1435 dc912121 Alex Williamson
        return;
1436 9ed7d6ae Juan Quintela
    }
1437 9ed7d6ae Juan Quintela
    vmstate_save_state(f,se->vmsd, se->opaque);
1438 4082be4d Juan Quintela
}
1439 4082be4d Juan Quintela
1440 a672b469 aliguori
#define QEMU_VM_FILE_MAGIC           0x5145564d
1441 a672b469 aliguori
#define QEMU_VM_FILE_VERSION_COMPAT  0x00000002
1442 a672b469 aliguori
#define QEMU_VM_FILE_VERSION         0x00000003
1443 a672b469 aliguori
1444 a672b469 aliguori
#define QEMU_VM_EOF                  0x00
1445 a672b469 aliguori
#define QEMU_VM_SECTION_START        0x01
1446 a672b469 aliguori
#define QEMU_VM_SECTION_PART         0x02
1447 a672b469 aliguori
#define QEMU_VM_SECTION_END          0x03
1448 a672b469 aliguori
#define QEMU_VM_SECTION_FULL         0x04
1449 811814bd Juan Quintela
#define QEMU_VM_SUBSECTION           0x05
1450 a672b469 aliguori
1451 dc912121 Alex Williamson
bool qemu_savevm_state_blocked(Monitor *mon)
1452 dc912121 Alex Williamson
{
1453 dc912121 Alex Williamson
    SaveStateEntry *se;
1454 dc912121 Alex Williamson
1455 dc912121 Alex Williamson
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1456 dc912121 Alex Williamson
        if (se->no_migrate) {
1457 dc912121 Alex Williamson
            monitor_printf(mon, "state blocked by non-migratable device '%s'\n",
1458 dc912121 Alex Williamson
                           se->idstr);
1459 dc912121 Alex Williamson
            return true;
1460 dc912121 Alex Williamson
        }
1461 dc912121 Alex Williamson
    }
1462 dc912121 Alex Williamson
    return false;
1463 dc912121 Alex Williamson
}
1464 dc912121 Alex Williamson
1465 f327aa0c Jan Kiszka
int qemu_savevm_state_begin(Monitor *mon, QEMUFile *f, int blk_enable,
1466 f327aa0c Jan Kiszka
                            int shared)
1467 a672b469 aliguori
{
1468 a672b469 aliguori
    SaveStateEntry *se;
1469 a672b469 aliguori
1470 c163b5ca lirans@il.ibm.com
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1471 c163b5ca lirans@il.ibm.com
        if(se->set_params == NULL) {
1472 c163b5ca lirans@il.ibm.com
            continue;
1473 c163b5ca lirans@il.ibm.com
        }
1474 c163b5ca lirans@il.ibm.com
        se->set_params(blk_enable, shared, se->opaque);
1475 c163b5ca lirans@il.ibm.com
    }
1476 c163b5ca lirans@il.ibm.com
    
1477 a672b469 aliguori
    qemu_put_be32(f, QEMU_VM_FILE_MAGIC);
1478 a672b469 aliguori
    qemu_put_be32(f, QEMU_VM_FILE_VERSION);
1479 a672b469 aliguori
1480 72cf2d4f Blue Swirl
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1481 a672b469 aliguori
        int len;
1482 a672b469 aliguori
1483 a672b469 aliguori
        if (se->save_live_state == NULL)
1484 a672b469 aliguori
            continue;
1485 a672b469 aliguori
1486 a672b469 aliguori
        /* Section type */
1487 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_START);
1488 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1489 a672b469 aliguori
1490 a672b469 aliguori
        /* ID string */
1491 a672b469 aliguori
        len = strlen(se->idstr);
1492 a672b469 aliguori
        qemu_put_byte(f, len);
1493 a672b469 aliguori
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1494 a672b469 aliguori
1495 a672b469 aliguori
        qemu_put_be32(f, se->instance_id);
1496 a672b469 aliguori
        qemu_put_be32(f, se->version_id);
1497 a672b469 aliguori
1498 f327aa0c Jan Kiszka
        se->save_live_state(mon, f, QEMU_VM_SECTION_START, se->opaque);
1499 a672b469 aliguori
    }
1500 a672b469 aliguori
1501 4ec7fcc7 Jan Kiszka
    if (qemu_file_has_error(f)) {
1502 f327aa0c Jan Kiszka
        qemu_savevm_state_cancel(mon, f);
1503 a672b469 aliguori
        return -EIO;
1504 4ec7fcc7 Jan Kiszka
    }
1505 a672b469 aliguori
1506 a672b469 aliguori
    return 0;
1507 a672b469 aliguori
}
1508 a672b469 aliguori
1509 f327aa0c Jan Kiszka
int qemu_savevm_state_iterate(Monitor *mon, QEMUFile *f)
1510 a672b469 aliguori
{
1511 a672b469 aliguori
    SaveStateEntry *se;
1512 a672b469 aliguori
    int ret = 1;
1513 a672b469 aliguori
1514 72cf2d4f Blue Swirl
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1515 a672b469 aliguori
        if (se->save_live_state == NULL)
1516 a672b469 aliguori
            continue;
1517 a672b469 aliguori
1518 a672b469 aliguori
        /* Section type */
1519 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_PART);
1520 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1521 a672b469 aliguori
1522 90697be8 Jan Kiszka
        ret = se->save_live_state(mon, f, QEMU_VM_SECTION_PART, se->opaque);
1523 90697be8 Jan Kiszka
        if (!ret) {
1524 90697be8 Jan Kiszka
            /* Do not proceed to the next vmstate before this one reported
1525 90697be8 Jan Kiszka
               completion of the current stage. This serializes the migration
1526 90697be8 Jan Kiszka
               and reduces the probability that a faster changing state is
1527 90697be8 Jan Kiszka
               synchronized over and over again. */
1528 90697be8 Jan Kiszka
            break;
1529 90697be8 Jan Kiszka
        }
1530 a672b469 aliguori
    }
1531 a672b469 aliguori
1532 a672b469 aliguori
    if (ret)
1533 a672b469 aliguori
        return 1;
1534 a672b469 aliguori
1535 4ec7fcc7 Jan Kiszka
    if (qemu_file_has_error(f)) {
1536 f327aa0c Jan Kiszka
        qemu_savevm_state_cancel(mon, f);
1537 a672b469 aliguori
        return -EIO;
1538 4ec7fcc7 Jan Kiszka
    }
1539 a672b469 aliguori
1540 a672b469 aliguori
    return 0;
1541 a672b469 aliguori
}
1542 a672b469 aliguori
1543 f327aa0c Jan Kiszka
int qemu_savevm_state_complete(Monitor *mon, QEMUFile *f)
1544 a672b469 aliguori
{
1545 a672b469 aliguori
    SaveStateEntry *se;
1546 a672b469 aliguori
1547 ea375f9a Jan Kiszka
    cpu_synchronize_all_states();
1548 ea375f9a Jan Kiszka
1549 72cf2d4f Blue Swirl
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1550 a672b469 aliguori
        if (se->save_live_state == NULL)
1551 a672b469 aliguori
            continue;
1552 a672b469 aliguori
1553 a672b469 aliguori
        /* Section type */
1554 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_END);
1555 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1556 a672b469 aliguori
1557 f327aa0c Jan Kiszka
        se->save_live_state(mon, f, QEMU_VM_SECTION_END, se->opaque);
1558 a672b469 aliguori
    }
1559 a672b469 aliguori
1560 72cf2d4f Blue Swirl
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1561 a672b469 aliguori
        int len;
1562 a672b469 aliguori
1563 9ed7d6ae Juan Quintela
        if (se->save_state == NULL && se->vmsd == NULL)
1564 a672b469 aliguori
            continue;
1565 a672b469 aliguori
1566 a672b469 aliguori
        /* Section type */
1567 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_FULL);
1568 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1569 a672b469 aliguori
1570 a672b469 aliguori
        /* ID string */
1571 a672b469 aliguori
        len = strlen(se->idstr);
1572 a672b469 aliguori
        qemu_put_byte(f, len);
1573 a672b469 aliguori
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1574 a672b469 aliguori
1575 a672b469 aliguori
        qemu_put_be32(f, se->instance_id);
1576 a672b469 aliguori
        qemu_put_be32(f, se->version_id);
1577 a672b469 aliguori
1578 dc912121 Alex Williamson
        vmstate_save(f, se);
1579 a672b469 aliguori
    }
1580 a672b469 aliguori
1581 a672b469 aliguori
    qemu_put_byte(f, QEMU_VM_EOF);
1582 a672b469 aliguori
1583 a672b469 aliguori
    if (qemu_file_has_error(f))
1584 a672b469 aliguori
        return -EIO;
1585 a672b469 aliguori
1586 a672b469 aliguori
    return 0;
1587 a672b469 aliguori
}
1588 a672b469 aliguori
1589 f327aa0c Jan Kiszka
void qemu_savevm_state_cancel(Monitor *mon, QEMUFile *f)
1590 4ec7fcc7 Jan Kiszka
{
1591 4ec7fcc7 Jan Kiszka
    SaveStateEntry *se;
1592 4ec7fcc7 Jan Kiszka
1593 4ec7fcc7 Jan Kiszka
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1594 4ec7fcc7 Jan Kiszka
        if (se->save_live_state) {
1595 f327aa0c Jan Kiszka
            se->save_live_state(mon, f, -1, se->opaque);
1596 4ec7fcc7 Jan Kiszka
        }
1597 4ec7fcc7 Jan Kiszka
    }
1598 4ec7fcc7 Jan Kiszka
}
1599 4ec7fcc7 Jan Kiszka
1600 f327aa0c Jan Kiszka
static int qemu_savevm_state(Monitor *mon, QEMUFile *f)
1601 a672b469 aliguori
{
1602 a672b469 aliguori
    int saved_vm_running;
1603 a672b469 aliguori
    int ret;
1604 a672b469 aliguori
1605 a672b469 aliguori
    saved_vm_running = vm_running;
1606 e07bbac5 Jan Kiszka
    vm_stop(VMSTOP_SAVEVM);
1607 a672b469 aliguori
1608 dc912121 Alex Williamson
    if (qemu_savevm_state_blocked(mon)) {
1609 dc912121 Alex Williamson
        ret = -EINVAL;
1610 dc912121 Alex Williamson
        goto out;
1611 dc912121 Alex Williamson
    }
1612 dc912121 Alex Williamson
1613 f327aa0c Jan Kiszka
    ret = qemu_savevm_state_begin(mon, f, 0, 0);
1614 a672b469 aliguori
    if (ret < 0)
1615 a672b469 aliguori
        goto out;
1616 a672b469 aliguori
1617 a672b469 aliguori
    do {
1618 f327aa0c Jan Kiszka
        ret = qemu_savevm_state_iterate(mon, f);
1619 a672b469 aliguori
        if (ret < 0)
1620 a672b469 aliguori
            goto out;
1621 a672b469 aliguori
    } while (ret == 0);
1622 a672b469 aliguori
1623 f327aa0c Jan Kiszka
    ret = qemu_savevm_state_complete(mon, f);
1624 a672b469 aliguori
1625 a672b469 aliguori
out:
1626 a672b469 aliguori
    if (qemu_file_has_error(f))
1627 a672b469 aliguori
        ret = -EIO;
1628 a672b469 aliguori
1629 a672b469 aliguori
    if (!ret && saved_vm_running)
1630 a672b469 aliguori
        vm_start();
1631 a672b469 aliguori
1632 a672b469 aliguori
    return ret;
1633 a672b469 aliguori
}
1634 a672b469 aliguori
1635 a672b469 aliguori
static SaveStateEntry *find_se(const char *idstr, int instance_id)
1636 a672b469 aliguori
{
1637 a672b469 aliguori
    SaveStateEntry *se;
1638 a672b469 aliguori
1639 72cf2d4f Blue Swirl
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1640 a672b469 aliguori
        if (!strcmp(se->idstr, idstr) &&
1641 4d2ffa08 Jan Kiszka
            (instance_id == se->instance_id ||
1642 4d2ffa08 Jan Kiszka
             instance_id == se->alias_id))
1643 a672b469 aliguori
            return se;
1644 7685ee6a Alex Williamson
        /* Migrating from an older version? */
1645 7685ee6a Alex Williamson
        if (strstr(se->idstr, idstr) && se->compat) {
1646 7685ee6a Alex Williamson
            if (!strcmp(se->compat->idstr, idstr) &&
1647 7685ee6a Alex Williamson
                (instance_id == se->compat->instance_id ||
1648 7685ee6a Alex Williamson
                 instance_id == se->alias_id))
1649 7685ee6a Alex Williamson
                return se;
1650 7685ee6a Alex Williamson
        }
1651 a672b469 aliguori
    }
1652 a672b469 aliguori
    return NULL;
1653 a672b469 aliguori
}
1654 a672b469 aliguori
1655 811814bd Juan Quintela
static const VMStateDescription *vmstate_get_subsection(const VMStateSubsection *sub, char *idstr)
1656 811814bd Juan Quintela
{
1657 811814bd Juan Quintela
    while(sub && sub->needed) {
1658 811814bd Juan Quintela
        if (strcmp(idstr, sub->vmsd->name) == 0) {
1659 811814bd Juan Quintela
            return sub->vmsd;
1660 811814bd Juan Quintela
        }
1661 811814bd Juan Quintela
        sub++;
1662 811814bd Juan Quintela
    }
1663 811814bd Juan Quintela
    return NULL;
1664 811814bd Juan Quintela
}
1665 811814bd Juan Quintela
1666 811814bd Juan Quintela
static int vmstate_subsection_load(QEMUFile *f, const VMStateDescription *vmsd,
1667 811814bd Juan Quintela
                                   void *opaque)
1668 811814bd Juan Quintela
{
1669 eb60260d Yoshiaki Tamura
    const VMStateSubsection *sub = vmsd->subsections;
1670 eb60260d Yoshiaki Tamura
1671 eb60260d Yoshiaki Tamura
    if (!sub || !sub->needed) {
1672 eb60260d Yoshiaki Tamura
        return 0;
1673 eb60260d Yoshiaki Tamura
    }
1674 eb60260d Yoshiaki Tamura
1675 811814bd Juan Quintela
    while (qemu_peek_byte(f) == QEMU_VM_SUBSECTION) {
1676 811814bd Juan Quintela
        char idstr[256];
1677 811814bd Juan Quintela
        int ret;
1678 49a2942d Blue Swirl
        uint8_t version_id, len;
1679 811814bd Juan Quintela
        const VMStateDescription *sub_vmsd;
1680 811814bd Juan Quintela
1681 49a2942d Blue Swirl
        qemu_get_byte(f); /* subsection */
1682 811814bd Juan Quintela
        len = qemu_get_byte(f);
1683 811814bd Juan Quintela
        qemu_get_buffer(f, (uint8_t *)idstr, len);
1684 811814bd Juan Quintela
        idstr[len] = 0;
1685 811814bd Juan Quintela
        version_id = qemu_get_be32(f);
1686 811814bd Juan Quintela
1687 eb60260d Yoshiaki Tamura
        sub_vmsd = vmstate_get_subsection(sub, idstr);
1688 811814bd Juan Quintela
        if (sub_vmsd == NULL) {
1689 811814bd Juan Quintela
            return -ENOENT;
1690 811814bd Juan Quintela
        }
1691 eb60260d Yoshiaki Tamura
        assert(!sub_vmsd->subsections);
1692 811814bd Juan Quintela
        ret = vmstate_load_state(f, sub_vmsd, opaque, version_id);
1693 811814bd Juan Quintela
        if (ret) {
1694 811814bd Juan Quintela
            return ret;
1695 811814bd Juan Quintela
        }
1696 811814bd Juan Quintela
    }
1697 811814bd Juan Quintela
    return 0;
1698 811814bd Juan Quintela
}
1699 811814bd Juan Quintela
1700 811814bd Juan Quintela
static void vmstate_subsection_save(QEMUFile *f, const VMStateDescription *vmsd,
1701 811814bd Juan Quintela
                                    void *opaque)
1702 811814bd Juan Quintela
{
1703 811814bd Juan Quintela
    const VMStateSubsection *sub = vmsd->subsections;
1704 811814bd Juan Quintela
1705 811814bd Juan Quintela
    while (sub && sub->needed) {
1706 811814bd Juan Quintela
        if (sub->needed(opaque)) {
1707 811814bd Juan Quintela
            const VMStateDescription *vmsd = sub->vmsd;
1708 811814bd Juan Quintela
            uint8_t len;
1709 811814bd Juan Quintela
1710 811814bd Juan Quintela
            qemu_put_byte(f, QEMU_VM_SUBSECTION);
1711 811814bd Juan Quintela
            len = strlen(vmsd->name);
1712 811814bd Juan Quintela
            qemu_put_byte(f, len);
1713 811814bd Juan Quintela
            qemu_put_buffer(f, (uint8_t *)vmsd->name, len);
1714 811814bd Juan Quintela
            qemu_put_be32(f, vmsd->version_id);
1715 eb60260d Yoshiaki Tamura
            assert(!vmsd->subsections);
1716 811814bd Juan Quintela
            vmstate_save_state(f, vmsd, opaque);
1717 811814bd Juan Quintela
        }
1718 811814bd Juan Quintela
        sub++;
1719 811814bd Juan Quintela
    }
1720 811814bd Juan Quintela
}
1721 811814bd Juan Quintela
1722 a672b469 aliguori
typedef struct LoadStateEntry {
1723 72cf2d4f Blue Swirl
    QLIST_ENTRY(LoadStateEntry) entry;
1724 a672b469 aliguori
    SaveStateEntry *se;
1725 a672b469 aliguori
    int section_id;
1726 a672b469 aliguori
    int version_id;
1727 a672b469 aliguori
} LoadStateEntry;
1728 a672b469 aliguori
1729 a672b469 aliguori
int qemu_loadvm_state(QEMUFile *f)
1730 a672b469 aliguori
{
1731 72cf2d4f Blue Swirl
    QLIST_HEAD(, LoadStateEntry) loadvm_handlers =
1732 72cf2d4f Blue Swirl
        QLIST_HEAD_INITIALIZER(loadvm_handlers);
1733 f4dbb8dd Juan Quintela
    LoadStateEntry *le, *new_le;
1734 a672b469 aliguori
    uint8_t section_type;
1735 a672b469 aliguori
    unsigned int v;
1736 a672b469 aliguori
    int ret;
1737 a672b469 aliguori
1738 dc912121 Alex Williamson
    if (qemu_savevm_state_blocked(default_mon)) {
1739 dc912121 Alex Williamson
        return -EINVAL;
1740 dc912121 Alex Williamson
    }
1741 dc912121 Alex Williamson
1742 a672b469 aliguori
    v = qemu_get_be32(f);
1743 a672b469 aliguori
    if (v != QEMU_VM_FILE_MAGIC)
1744 a672b469 aliguori
        return -EINVAL;
1745 a672b469 aliguori
1746 a672b469 aliguori
    v = qemu_get_be32(f);
1747 bbfe1408 Juan Quintela
    if (v == QEMU_VM_FILE_VERSION_COMPAT) {
1748 bbfe1408 Juan Quintela
        fprintf(stderr, "SaveVM v2 format is obsolete and don't work anymore\n");
1749 bbfe1408 Juan Quintela
        return -ENOTSUP;
1750 bbfe1408 Juan Quintela
    }
1751 a672b469 aliguori
    if (v != QEMU_VM_FILE_VERSION)
1752 a672b469 aliguori
        return -ENOTSUP;
1753 a672b469 aliguori
1754 a672b469 aliguori
    while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
1755 a672b469 aliguori
        uint32_t instance_id, version_id, section_id;
1756 a672b469 aliguori
        SaveStateEntry *se;
1757 a672b469 aliguori
        char idstr[257];
1758 a672b469 aliguori
        int len;
1759 a672b469 aliguori
1760 a672b469 aliguori
        switch (section_type) {
1761 a672b469 aliguori
        case QEMU_VM_SECTION_START:
1762 a672b469 aliguori
        case QEMU_VM_SECTION_FULL:
1763 a672b469 aliguori
            /* Read section start */
1764 a672b469 aliguori
            section_id = qemu_get_be32(f);
1765 a672b469 aliguori
            len = qemu_get_byte(f);
1766 a672b469 aliguori
            qemu_get_buffer(f, (uint8_t *)idstr, len);
1767 a672b469 aliguori
            idstr[len] = 0;
1768 a672b469 aliguori
            instance_id = qemu_get_be32(f);
1769 a672b469 aliguori
            version_id = qemu_get_be32(f);
1770 a672b469 aliguori
1771 a672b469 aliguori
            /* Find savevm section */
1772 a672b469 aliguori
            se = find_se(idstr, instance_id);
1773 a672b469 aliguori
            if (se == NULL) {
1774 a672b469 aliguori
                fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
1775 a672b469 aliguori
                ret = -EINVAL;
1776 a672b469 aliguori
                goto out;
1777 a672b469 aliguori
            }
1778 a672b469 aliguori
1779 a672b469 aliguori
            /* Validate version */
1780 a672b469 aliguori
            if (version_id > se->version_id) {
1781 a672b469 aliguori
                fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
1782 a672b469 aliguori
                        version_id, idstr, se->version_id);
1783 a672b469 aliguori
                ret = -EINVAL;
1784 a672b469 aliguori
                goto out;
1785 a672b469 aliguori
            }
1786 a672b469 aliguori
1787 a672b469 aliguori
            /* Add entry */
1788 a672b469 aliguori
            le = qemu_mallocz(sizeof(*le));
1789 a672b469 aliguori
1790 a672b469 aliguori
            le->se = se;
1791 a672b469 aliguori
            le->section_id = section_id;
1792 a672b469 aliguori
            le->version_id = version_id;
1793 72cf2d4f Blue Swirl
            QLIST_INSERT_HEAD(&loadvm_handlers, le, entry);
1794 a672b469 aliguori
1795 4082be4d Juan Quintela
            ret = vmstate_load(f, le->se, le->version_id);
1796 b5a22e4a Juan Quintela
            if (ret < 0) {
1797 b5a22e4a Juan Quintela
                fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
1798 b5a22e4a Juan Quintela
                        instance_id, idstr);
1799 b5a22e4a Juan Quintela
                goto out;
1800 b5a22e4a Juan Quintela
            }
1801 a672b469 aliguori
            break;
1802 a672b469 aliguori
        case QEMU_VM_SECTION_PART:
1803 a672b469 aliguori
        case QEMU_VM_SECTION_END:
1804 a672b469 aliguori
            section_id = qemu_get_be32(f);
1805 a672b469 aliguori
1806 72cf2d4f Blue Swirl
            QLIST_FOREACH(le, &loadvm_handlers, entry) {
1807 f4dbb8dd Juan Quintela
                if (le->section_id == section_id) {
1808 f4dbb8dd Juan Quintela
                    break;
1809 f4dbb8dd Juan Quintela
                }
1810 f4dbb8dd Juan Quintela
            }
1811 a672b469 aliguori
            if (le == NULL) {
1812 a672b469 aliguori
                fprintf(stderr, "Unknown savevm section %d\n", section_id);
1813 a672b469 aliguori
                ret = -EINVAL;
1814 a672b469 aliguori
                goto out;
1815 a672b469 aliguori
            }
1816 a672b469 aliguori
1817 4082be4d Juan Quintela
            ret = vmstate_load(f, le->se, le->version_id);
1818 b5a22e4a Juan Quintela
            if (ret < 0) {
1819 b5a22e4a Juan Quintela
                fprintf(stderr, "qemu: warning: error while loading state section id %d\n",
1820 b5a22e4a Juan Quintela
                        section_id);
1821 b5a22e4a Juan Quintela
                goto out;
1822 b5a22e4a Juan Quintela
            }
1823 a672b469 aliguori
            break;
1824 a672b469 aliguori
        default:
1825 a672b469 aliguori
            fprintf(stderr, "Unknown savevm section type %d\n", section_type);
1826 a672b469 aliguori
            ret = -EINVAL;
1827 a672b469 aliguori
            goto out;
1828 a672b469 aliguori
        }
1829 a672b469 aliguori
    }
1830 a672b469 aliguori
1831 ea375f9a Jan Kiszka
    cpu_synchronize_all_post_init();
1832 ea375f9a Jan Kiszka
1833 a672b469 aliguori
    ret = 0;
1834 a672b469 aliguori
1835 a672b469 aliguori
out:
1836 72cf2d4f Blue Swirl
    QLIST_FOREACH_SAFE(le, &loadvm_handlers, entry, new_le) {
1837 72cf2d4f Blue Swirl
        QLIST_REMOVE(le, entry);
1838 a672b469 aliguori
        qemu_free(le);
1839 a672b469 aliguori
    }
1840 a672b469 aliguori
1841 a672b469 aliguori
    if (qemu_file_has_error(f))
1842 a672b469 aliguori
        ret = -EIO;
1843 a672b469 aliguori
1844 a672b469 aliguori
    return ret;
1845 a672b469 aliguori
}
1846 a672b469 aliguori
1847 a672b469 aliguori
static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
1848 a672b469 aliguori
                              const char *name)
1849 a672b469 aliguori
{
1850 a672b469 aliguori
    QEMUSnapshotInfo *sn_tab, *sn;
1851 a672b469 aliguori
    int nb_sns, i, ret;
1852 a672b469 aliguori
1853 a672b469 aliguori
    ret = -ENOENT;
1854 a672b469 aliguori
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1855 a672b469 aliguori
    if (nb_sns < 0)
1856 a672b469 aliguori
        return ret;
1857 a672b469 aliguori
    for(i = 0; i < nb_sns; i++) {
1858 a672b469 aliguori
        sn = &sn_tab[i];
1859 a672b469 aliguori
        if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
1860 a672b469 aliguori
            *sn_info = *sn;
1861 a672b469 aliguori
            ret = 0;
1862 a672b469 aliguori
            break;
1863 a672b469 aliguori
        }
1864 a672b469 aliguori
    }
1865 a672b469 aliguori
    qemu_free(sn_tab);
1866 a672b469 aliguori
    return ret;
1867 a672b469 aliguori
}
1868 a672b469 aliguori
1869 cb499fb2 Kevin Wolf
/*
1870 cb499fb2 Kevin Wolf
 * Deletes snapshots of a given name in all opened images.
1871 cb499fb2 Kevin Wolf
 */
1872 cb499fb2 Kevin Wolf
static int del_existing_snapshots(Monitor *mon, const char *name)
1873 cb499fb2 Kevin Wolf
{
1874 cb499fb2 Kevin Wolf
    BlockDriverState *bs;
1875 cb499fb2 Kevin Wolf
    QEMUSnapshotInfo sn1, *snapshot = &sn1;
1876 cb499fb2 Kevin Wolf
    int ret;
1877 cb499fb2 Kevin Wolf
1878 dbc13590 Markus Armbruster
    bs = NULL;
1879 dbc13590 Markus Armbruster
    while ((bs = bdrv_next(bs))) {
1880 cb499fb2 Kevin Wolf
        if (bdrv_can_snapshot(bs) &&
1881 cb499fb2 Kevin Wolf
            bdrv_snapshot_find(bs, snapshot, name) >= 0)
1882 cb499fb2 Kevin Wolf
        {
1883 cb499fb2 Kevin Wolf
            ret = bdrv_snapshot_delete(bs, name);
1884 cb499fb2 Kevin Wolf
            if (ret < 0) {
1885 cb499fb2 Kevin Wolf
                monitor_printf(mon,
1886 cb499fb2 Kevin Wolf
                               "Error while deleting snapshot on '%s'\n",
1887 cb499fb2 Kevin Wolf
                               bdrv_get_device_name(bs));
1888 cb499fb2 Kevin Wolf
                return -1;
1889 cb499fb2 Kevin Wolf
            }
1890 cb499fb2 Kevin Wolf
        }
1891 cb499fb2 Kevin Wolf
    }
1892 cb499fb2 Kevin Wolf
1893 cb499fb2 Kevin Wolf
    return 0;
1894 cb499fb2 Kevin Wolf
}
1895 cb499fb2 Kevin Wolf
1896 d54908a5 Luiz Capitulino
void do_savevm(Monitor *mon, const QDict *qdict)
1897 a672b469 aliguori
{
1898 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1899 a672b469 aliguori
    QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1;
1900 cb499fb2 Kevin Wolf
    int ret;
1901 a672b469 aliguori
    QEMUFile *f;
1902 a672b469 aliguori
    int saved_vm_running;
1903 2d22b18f aliguori
    uint32_t vm_state_size;
1904 a672b469 aliguori
#ifdef _WIN32
1905 a672b469 aliguori
    struct _timeb tb;
1906 7d631a11 Miguel Di Ciurcio Filho
    struct tm *ptm;
1907 a672b469 aliguori
#else
1908 a672b469 aliguori
    struct timeval tv;
1909 7d631a11 Miguel Di Ciurcio Filho
    struct tm tm;
1910 a672b469 aliguori
#endif
1911 d54908a5 Luiz Capitulino
    const char *name = qdict_get_try_str(qdict, "name");
1912 a672b469 aliguori
1913 feeee5ac Miguel Di Ciurcio Filho
    /* Verify if there is a device that doesn't support snapshots and is writable */
1914 dbc13590 Markus Armbruster
    bs = NULL;
1915 dbc13590 Markus Armbruster
    while ((bs = bdrv_next(bs))) {
1916 feeee5ac Miguel Di Ciurcio Filho
1917 feeee5ac Miguel Di Ciurcio Filho
        if (bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
1918 feeee5ac Miguel Di Ciurcio Filho
            continue;
1919 feeee5ac Miguel Di Ciurcio Filho
        }
1920 feeee5ac Miguel Di Ciurcio Filho
1921 feeee5ac Miguel Di Ciurcio Filho
        if (!bdrv_can_snapshot(bs)) {
1922 feeee5ac Miguel Di Ciurcio Filho
            monitor_printf(mon, "Device '%s' is writable but does not support snapshots.\n",
1923 feeee5ac Miguel Di Ciurcio Filho
                               bdrv_get_device_name(bs));
1924 feeee5ac Miguel Di Ciurcio Filho
            return;
1925 feeee5ac Miguel Di Ciurcio Filho
        }
1926 feeee5ac Miguel Di Ciurcio Filho
    }
1927 feeee5ac Miguel Di Ciurcio Filho
1928 f9092b10 Markus Armbruster
    bs = bdrv_snapshots();
1929 a672b469 aliguori
    if (!bs) {
1930 376253ec aliguori
        monitor_printf(mon, "No block device can accept snapshots\n");
1931 a672b469 aliguori
        return;
1932 a672b469 aliguori
    }
1933 a672b469 aliguori
1934 a672b469 aliguori
    saved_vm_running = vm_running;
1935 e07bbac5 Jan Kiszka
    vm_stop(VMSTOP_SAVEVM);
1936 a672b469 aliguori
1937 cb499fb2 Kevin Wolf
    memset(sn, 0, sizeof(*sn));
1938 a672b469 aliguori
1939 a672b469 aliguori
    /* fill auxiliary fields */
1940 a672b469 aliguori
#ifdef _WIN32
1941 a672b469 aliguori
    _ftime(&tb);
1942 a672b469 aliguori
    sn->date_sec = tb.time;
1943 a672b469 aliguori
    sn->date_nsec = tb.millitm * 1000000;
1944 a672b469 aliguori
#else
1945 a672b469 aliguori
    gettimeofday(&tv, NULL);
1946 a672b469 aliguori
    sn->date_sec = tv.tv_sec;
1947 a672b469 aliguori
    sn->date_nsec = tv.tv_usec * 1000;
1948 a672b469 aliguori
#endif
1949 74475455 Paolo Bonzini
    sn->vm_clock_nsec = qemu_get_clock_ns(vm_clock);
1950 a672b469 aliguori
1951 7d631a11 Miguel Di Ciurcio Filho
    if (name) {
1952 7d631a11 Miguel Di Ciurcio Filho
        ret = bdrv_snapshot_find(bs, old_sn, name);
1953 7d631a11 Miguel Di Ciurcio Filho
        if (ret >= 0) {
1954 7d631a11 Miguel Di Ciurcio Filho
            pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
1955 7d631a11 Miguel Di Ciurcio Filho
            pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
1956 7d631a11 Miguel Di Ciurcio Filho
        } else {
1957 7d631a11 Miguel Di Ciurcio Filho
            pstrcpy(sn->name, sizeof(sn->name), name);
1958 7d631a11 Miguel Di Ciurcio Filho
        }
1959 7d631a11 Miguel Di Ciurcio Filho
    } else {
1960 7d631a11 Miguel Di Ciurcio Filho
#ifdef _WIN32
1961 7d631a11 Miguel Di Ciurcio Filho
        ptm = localtime(&tb.time);
1962 7d631a11 Miguel Di Ciurcio Filho
        strftime(sn->name, sizeof(sn->name), "vm-%Y%m%d%H%M%S", ptm);
1963 7d631a11 Miguel Di Ciurcio Filho
#else
1964 d7d9b528 Blue Swirl
        /* cast below needed for OpenBSD where tv_sec is still 'long' */
1965 d7d9b528 Blue Swirl
        localtime_r((const time_t *)&tv.tv_sec, &tm);
1966 7d631a11 Miguel Di Ciurcio Filho
        strftime(sn->name, sizeof(sn->name), "vm-%Y%m%d%H%M%S", &tm);
1967 7d631a11 Miguel Di Ciurcio Filho
#endif
1968 7d631a11 Miguel Di Ciurcio Filho
    }
1969 7d631a11 Miguel Di Ciurcio Filho
1970 cb499fb2 Kevin Wolf
    /* Delete old snapshots of the same name */
1971 f139a412 Marcelo Tosatti
    if (name && del_existing_snapshots(mon, name) < 0) {
1972 cb499fb2 Kevin Wolf
        goto the_end;
1973 cb499fb2 Kevin Wolf
    }
1974 cb499fb2 Kevin Wolf
1975 a672b469 aliguori
    /* save the VM state */
1976 45566e9c Christoph Hellwig
    f = qemu_fopen_bdrv(bs, 1);
1977 a672b469 aliguori
    if (!f) {
1978 376253ec aliguori
        monitor_printf(mon, "Could not open VM state file\n");
1979 a672b469 aliguori
        goto the_end;
1980 a672b469 aliguori
    }
1981 f327aa0c Jan Kiszka
    ret = qemu_savevm_state(mon, f);
1982 2d22b18f aliguori
    vm_state_size = qemu_ftell(f);
1983 a672b469 aliguori
    qemu_fclose(f);
1984 a672b469 aliguori
    if (ret < 0) {
1985 376253ec aliguori
        monitor_printf(mon, "Error %d while writing VM\n", ret);
1986 a672b469 aliguori
        goto the_end;
1987 a672b469 aliguori
    }
1988 a672b469 aliguori
1989 a672b469 aliguori
    /* create the snapshots */
1990 a672b469 aliguori
1991 dbc13590 Markus Armbruster
    bs1 = NULL;
1992 dbc13590 Markus Armbruster
    while ((bs1 = bdrv_next(bs1))) {
1993 feeee5ac Miguel Di Ciurcio Filho
        if (bdrv_can_snapshot(bs1)) {
1994 2d22b18f aliguori
            /* Write VM state size only to the image that contains the state */
1995 2d22b18f aliguori
            sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
1996 a672b469 aliguori
            ret = bdrv_snapshot_create(bs1, sn);
1997 a672b469 aliguori
            if (ret < 0) {
1998 376253ec aliguori
                monitor_printf(mon, "Error while creating snapshot on '%s'\n",
1999 376253ec aliguori
                               bdrv_get_device_name(bs1));
2000 a672b469 aliguori
            }
2001 a672b469 aliguori
        }
2002 a672b469 aliguori
    }
2003 a672b469 aliguori
2004 a672b469 aliguori
 the_end:
2005 a672b469 aliguori
    if (saved_vm_running)
2006 a672b469 aliguori
        vm_start();
2007 a672b469 aliguori
}
2008 a672b469 aliguori
2009 03cd4655 Markus Armbruster
int load_vmstate(const char *name)
2010 a672b469 aliguori
{
2011 f0aa7a8b Miguel Di Ciurcio Filho
    BlockDriverState *bs, *bs_vm_state;
2012 2d22b18f aliguori
    QEMUSnapshotInfo sn;
2013 a672b469 aliguori
    QEMUFile *f;
2014 751c6a17 Gerd Hoffmann
    int ret;
2015 a672b469 aliguori
2016 f0aa7a8b Miguel Di Ciurcio Filho
    bs_vm_state = bdrv_snapshots();
2017 f0aa7a8b Miguel Di Ciurcio Filho
    if (!bs_vm_state) {
2018 f0aa7a8b Miguel Di Ciurcio Filho
        error_report("No block device supports snapshots");
2019 f0aa7a8b Miguel Di Ciurcio Filho
        return -ENOTSUP;
2020 f0aa7a8b Miguel Di Ciurcio Filho
    }
2021 f0aa7a8b Miguel Di Ciurcio Filho
2022 f0aa7a8b Miguel Di Ciurcio Filho
    /* Don't even try to load empty VM states */
2023 f0aa7a8b Miguel Di Ciurcio Filho
    ret = bdrv_snapshot_find(bs_vm_state, &sn, name);
2024 f0aa7a8b Miguel Di Ciurcio Filho
    if (ret < 0) {
2025 f0aa7a8b Miguel Di Ciurcio Filho
        return ret;
2026 f0aa7a8b Miguel Di Ciurcio Filho
    } else if (sn.vm_state_size == 0) {
2027 e11480db Kevin Wolf
        error_report("This is a disk-only snapshot. Revert to it offline "
2028 e11480db Kevin Wolf
            "using qemu-img.");
2029 f0aa7a8b Miguel Di Ciurcio Filho
        return -EINVAL;
2030 f0aa7a8b Miguel Di Ciurcio Filho
    }
2031 f0aa7a8b Miguel Di Ciurcio Filho
2032 f0aa7a8b Miguel Di Ciurcio Filho
    /* Verify if there is any device that doesn't support snapshots and is
2033 f0aa7a8b Miguel Di Ciurcio Filho
    writable and check if the requested snapshot is available too. */
2034 dbc13590 Markus Armbruster
    bs = NULL;
2035 dbc13590 Markus Armbruster
    while ((bs = bdrv_next(bs))) {
2036 feeee5ac Miguel Di Ciurcio Filho
2037 feeee5ac Miguel Di Ciurcio Filho
        if (bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
2038 feeee5ac Miguel Di Ciurcio Filho
            continue;
2039 feeee5ac Miguel Di Ciurcio Filho
        }
2040 feeee5ac Miguel Di Ciurcio Filho
2041 feeee5ac Miguel Di Ciurcio Filho
        if (!bdrv_can_snapshot(bs)) {
2042 feeee5ac Miguel Di Ciurcio Filho
            error_report("Device '%s' is writable but does not support snapshots.",
2043 feeee5ac Miguel Di Ciurcio Filho
                               bdrv_get_device_name(bs));
2044 feeee5ac Miguel Di Ciurcio Filho
            return -ENOTSUP;
2045 feeee5ac Miguel Di Ciurcio Filho
        }
2046 feeee5ac Miguel Di Ciurcio Filho
2047 f0aa7a8b Miguel Di Ciurcio Filho
        ret = bdrv_snapshot_find(bs, &sn, name);
2048 f0aa7a8b Miguel Di Ciurcio Filho
        if (ret < 0) {
2049 f0aa7a8b Miguel Di Ciurcio Filho
            error_report("Device '%s' does not have the requested snapshot '%s'",
2050 f0aa7a8b Miguel Di Ciurcio Filho
                           bdrv_get_device_name(bs), name);
2051 f0aa7a8b Miguel Di Ciurcio Filho
            return ret;
2052 f0aa7a8b Miguel Di Ciurcio Filho
        }
2053 a672b469 aliguori
    }
2054 a672b469 aliguori
2055 a672b469 aliguori
    /* Flush all IO requests so they don't interfere with the new state.  */
2056 a672b469 aliguori
    qemu_aio_flush();
2057 a672b469 aliguori
2058 f0aa7a8b Miguel Di Ciurcio Filho
    bs = NULL;
2059 f0aa7a8b Miguel Di Ciurcio Filho
    while ((bs = bdrv_next(bs))) {
2060 f0aa7a8b Miguel Di Ciurcio Filho
        if (bdrv_can_snapshot(bs)) {
2061 f0aa7a8b Miguel Di Ciurcio Filho
            ret = bdrv_snapshot_goto(bs, name);
2062 a672b469 aliguori
            if (ret < 0) {
2063 f0aa7a8b Miguel Di Ciurcio Filho
                error_report("Error %d while activating snapshot '%s' on '%s'",
2064 f0aa7a8b Miguel Di Ciurcio Filho
                             ret, name, bdrv_get_device_name(bs));
2065 f0aa7a8b Miguel Di Ciurcio Filho
                return ret;
2066 a672b469 aliguori
            }
2067 a672b469 aliguori
        }
2068 a672b469 aliguori
    }
2069 a672b469 aliguori
2070 a672b469 aliguori
    /* restore the VM state */
2071 f0aa7a8b Miguel Di Ciurcio Filho
    f = qemu_fopen_bdrv(bs_vm_state, 0);
2072 a672b469 aliguori
    if (!f) {
2073 1ecda02b Markus Armbruster
        error_report("Could not open VM state file");
2074 05f2401e Juan Quintela
        return -EINVAL;
2075 a672b469 aliguori
    }
2076 f0aa7a8b Miguel Di Ciurcio Filho
2077 a672b469 aliguori
    ret = qemu_loadvm_state(f);
2078 f0aa7a8b Miguel Di Ciurcio Filho
2079 a672b469 aliguori
    qemu_fclose(f);
2080 a672b469 aliguori
    if (ret < 0) {
2081 1ecda02b Markus Armbruster
        error_report("Error %d while loading VM state", ret);
2082 05f2401e Juan Quintela
        return ret;
2083 a672b469 aliguori
    }
2084 f0aa7a8b Miguel Di Ciurcio Filho
2085 05f2401e Juan Quintela
    return 0;
2086 7b630349 Juan Quintela
}
2087 7b630349 Juan Quintela
2088 d54908a5 Luiz Capitulino
void do_delvm(Monitor *mon, const QDict *qdict)
2089 a672b469 aliguori
{
2090 a672b469 aliguori
    BlockDriverState *bs, *bs1;
2091 751c6a17 Gerd Hoffmann
    int ret;
2092 d54908a5 Luiz Capitulino
    const char *name = qdict_get_str(qdict, "name");
2093 a672b469 aliguori
2094 f9092b10 Markus Armbruster
    bs = bdrv_snapshots();
2095 a672b469 aliguori
    if (!bs) {
2096 376253ec aliguori
        monitor_printf(mon, "No block device supports snapshots\n");
2097 a672b469 aliguori
        return;
2098 a672b469 aliguori
    }
2099 a672b469 aliguori
2100 dbc13590 Markus Armbruster
    bs1 = NULL;
2101 dbc13590 Markus Armbruster
    while ((bs1 = bdrv_next(bs1))) {
2102 feeee5ac Miguel Di Ciurcio Filho
        if (bdrv_can_snapshot(bs1)) {
2103 a672b469 aliguori
            ret = bdrv_snapshot_delete(bs1, name);
2104 a672b469 aliguori
            if (ret < 0) {
2105 a672b469 aliguori
                if (ret == -ENOTSUP)
2106 376253ec aliguori
                    monitor_printf(mon,
2107 376253ec aliguori
                                   "Snapshots not supported on device '%s'\n",
2108 376253ec aliguori
                                   bdrv_get_device_name(bs1));
2109 a672b469 aliguori
                else
2110 376253ec aliguori
                    monitor_printf(mon, "Error %d while deleting snapshot on "
2111 376253ec aliguori
                                   "'%s'\n", ret, bdrv_get_device_name(bs1));
2112 a672b469 aliguori
            }
2113 a672b469 aliguori
        }
2114 a672b469 aliguori
    }
2115 a672b469 aliguori
}
2116 a672b469 aliguori
2117 376253ec aliguori
void do_info_snapshots(Monitor *mon)
2118 a672b469 aliguori
{
2119 a672b469 aliguori
    BlockDriverState *bs, *bs1;
2120 f9209915 Miguel Di Ciurcio Filho
    QEMUSnapshotInfo *sn_tab, *sn, s, *sn_info = &s;
2121 f9209915 Miguel Di Ciurcio Filho
    int nb_sns, i, ret, available;
2122 f9209915 Miguel Di Ciurcio Filho
    int total;
2123 f9209915 Miguel Di Ciurcio Filho
    int *available_snapshots;
2124 a672b469 aliguori
    char buf[256];
2125 a672b469 aliguori
2126 f9092b10 Markus Armbruster
    bs = bdrv_snapshots();
2127 a672b469 aliguori
    if (!bs) {
2128 376253ec aliguori
        monitor_printf(mon, "No available block device supports snapshots\n");
2129 a672b469 aliguori
        return;
2130 a672b469 aliguori
    }
2131 a672b469 aliguori
2132 a672b469 aliguori
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
2133 a672b469 aliguori
    if (nb_sns < 0) {
2134 376253ec aliguori
        monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
2135 a672b469 aliguori
        return;
2136 a672b469 aliguori
    }
2137 f9209915 Miguel Di Ciurcio Filho
2138 f9209915 Miguel Di Ciurcio Filho
    if (nb_sns == 0) {
2139 f9209915 Miguel Di Ciurcio Filho
        monitor_printf(mon, "There is no snapshot available.\n");
2140 f9209915 Miguel Di Ciurcio Filho
        return;
2141 f9209915 Miguel Di Ciurcio Filho
    }
2142 f9209915 Miguel Di Ciurcio Filho
2143 f9209915 Miguel Di Ciurcio Filho
    available_snapshots = qemu_mallocz(sizeof(int) * nb_sns);
2144 f9209915 Miguel Di Ciurcio Filho
    total = 0;
2145 f9209915 Miguel Di Ciurcio Filho
    for (i = 0; i < nb_sns; i++) {
2146 a672b469 aliguori
        sn = &sn_tab[i];
2147 f9209915 Miguel Di Ciurcio Filho
        available = 1;
2148 f9209915 Miguel Di Ciurcio Filho
        bs1 = NULL;
2149 f9209915 Miguel Di Ciurcio Filho
2150 f9209915 Miguel Di Ciurcio Filho
        while ((bs1 = bdrv_next(bs1))) {
2151 f9209915 Miguel Di Ciurcio Filho
            if (bdrv_can_snapshot(bs1) && bs1 != bs) {
2152 f9209915 Miguel Di Ciurcio Filho
                ret = bdrv_snapshot_find(bs1, sn_info, sn->id_str);
2153 f9209915 Miguel Di Ciurcio Filho
                if (ret < 0) {
2154 f9209915 Miguel Di Ciurcio Filho
                    available = 0;
2155 f9209915 Miguel Di Ciurcio Filho
                    break;
2156 f9209915 Miguel Di Ciurcio Filho
                }
2157 f9209915 Miguel Di Ciurcio Filho
            }
2158 f9209915 Miguel Di Ciurcio Filho
        }
2159 f9209915 Miguel Di Ciurcio Filho
2160 f9209915 Miguel Di Ciurcio Filho
        if (available) {
2161 f9209915 Miguel Di Ciurcio Filho
            available_snapshots[total] = i;
2162 f9209915 Miguel Di Ciurcio Filho
            total++;
2163 f9209915 Miguel Di Ciurcio Filho
        }
2164 a672b469 aliguori
    }
2165 f9209915 Miguel Di Ciurcio Filho
2166 f9209915 Miguel Di Ciurcio Filho
    if (total > 0) {
2167 f9209915 Miguel Di Ciurcio Filho
        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
2168 f9209915 Miguel Di Ciurcio Filho
        for (i = 0; i < total; i++) {
2169 f9209915 Miguel Di Ciurcio Filho
            sn = &sn_tab[available_snapshots[i]];
2170 f9209915 Miguel Di Ciurcio Filho
            monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
2171 f9209915 Miguel Di Ciurcio Filho
        }
2172 f9209915 Miguel Di Ciurcio Filho
    } else {
2173 f9209915 Miguel Di Ciurcio Filho
        monitor_printf(mon, "There is no suitable snapshot available\n");
2174 f9209915 Miguel Di Ciurcio Filho
    }
2175 f9209915 Miguel Di Ciurcio Filho
2176 a672b469 aliguori
    qemu_free(sn_tab);
2177 f9209915 Miguel Di Ciurcio Filho
    qemu_free(available_snapshots);
2178 f9209915 Miguel Di Ciurcio Filho
2179 a672b469 aliguori
}