Statistics
| Branch: | Revision:

root / savevm.c @ bcbdc4d3

History | View | Annotate | Download (41.4 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
#if defined(__NetBSD__)
46 a672b469 aliguori
#include <net/if_tap.h>
47 a672b469 aliguori
#endif
48 a672b469 aliguori
#ifdef __linux__
49 a672b469 aliguori
#include <linux/if_tun.h>
50 a672b469 aliguori
#endif
51 a672b469 aliguori
#include <arpa/inet.h>
52 a672b469 aliguori
#include <dirent.h>
53 a672b469 aliguori
#include <netdb.h>
54 a672b469 aliguori
#include <sys/select.h>
55 71e72a19 Juan Quintela
#ifdef CONFIG_BSD
56 a672b469 aliguori
#include <sys/stat.h>
57 c5e97233 blueswir1
#if defined(__FreeBSD__) || defined(__DragonFly__)
58 a672b469 aliguori
#include <libutil.h>
59 a672b469 aliguori
#else
60 a672b469 aliguori
#include <util.h>
61 a672b469 aliguori
#endif
62 a672b469 aliguori
#elif defined (__GLIBC__) && defined (__FreeBSD_kernel__)
63 a672b469 aliguori
#include <freebsd/stdlib.h>
64 a672b469 aliguori
#else
65 a672b469 aliguori
#ifdef __linux__
66 a672b469 aliguori
#include <pty.h>
67 a672b469 aliguori
#include <malloc.h>
68 a672b469 aliguori
#include <linux/rtc.h>
69 a672b469 aliguori
#endif
70 a672b469 aliguori
#endif
71 a672b469 aliguori
#endif
72 a672b469 aliguori
73 a672b469 aliguori
#ifdef _WIN32
74 49dc768d aliguori
#include <windows.h>
75 a672b469 aliguori
#include <malloc.h>
76 a672b469 aliguori
#include <sys/timeb.h>
77 a672b469 aliguori
#include <mmsystem.h>
78 a672b469 aliguori
#define getopt_long_only getopt_long
79 a672b469 aliguori
#define memalign(align, size) malloc(size)
80 a672b469 aliguori
#endif
81 a672b469 aliguori
82 511d2b14 blueswir1
#include "qemu-common.h"
83 511d2b14 blueswir1
#include "hw/hw.h"
84 511d2b14 blueswir1
#include "net.h"
85 511d2b14 blueswir1
#include "monitor.h"
86 511d2b14 blueswir1
#include "sysemu.h"
87 511d2b14 blueswir1
#include "qemu-timer.h"
88 511d2b14 blueswir1
#include "qemu-char.h"
89 511d2b14 blueswir1
#include "block.h"
90 511d2b14 blueswir1
#include "audio/audio.h"
91 511d2b14 blueswir1
#include "migration.h"
92 511d2b14 blueswir1
#include "qemu_socket.h"
93 511d2b14 blueswir1
94 a672b469 aliguori
/* point to the block driver where the snapshots are managed */
95 a672b469 aliguori
static BlockDriverState *bs_snapshots;
96 a672b469 aliguori
97 a672b469 aliguori
#define SELF_ANNOUNCE_ROUNDS 5
98 a672b469 aliguori
#define ETH_P_EXPERIMENTAL 0x01F1 /* just a number */
99 a672b469 aliguori
//#define ETH_P_EXPERIMENTAL 0x0012 /* make it the size of the packet */
100 a672b469 aliguori
#define EXPERIMENTAL_MAGIC 0xf1f23f4f
101 a672b469 aliguori
102 a672b469 aliguori
static int announce_self_create(uint8_t *buf, 
103 a672b469 aliguori
                                uint8_t *mac_addr)
104 a672b469 aliguori
{
105 a672b469 aliguori
    uint32_t magic = EXPERIMENTAL_MAGIC;
106 a672b469 aliguori
    uint16_t proto = htons(ETH_P_EXPERIMENTAL);
107 a672b469 aliguori
108 a672b469 aliguori
    /* FIXME: should we send a different packet (arp/rarp/ping)? */
109 a672b469 aliguori
110 976305b7 Gleb Natapov
    memset(buf, 0, 64);
111 a672b469 aliguori
    memset(buf, 0xff, 6);         /* h_dst */
112 a672b469 aliguori
    memcpy(buf + 6, mac_addr, 6); /* h_src */
113 a672b469 aliguori
    memcpy(buf + 12, &proto, 2);  /* h_proto */
114 a672b469 aliguori
    memcpy(buf + 14, &magic, 4);  /* magic */
115 a672b469 aliguori
116 976305b7 Gleb Natapov
    return 64; /* len */
117 a672b469 aliguori
}
118 a672b469 aliguori
119 ed8b330b Gleb Natapov
static void qemu_announce_self_once(void *opaque)
120 a672b469 aliguori
{
121 ed8b330b Gleb Natapov
    int i, len;
122 a672b469 aliguori
    VLANState *vlan;
123 a672b469 aliguori
    VLANClientState *vc;
124 a672b469 aliguori
    uint8_t buf[256];
125 ed8b330b Gleb Natapov
    static int count = SELF_ANNOUNCE_ROUNDS;
126 ed8b330b Gleb Natapov
    QEMUTimer *timer = *(QEMUTimer **)opaque;
127 a672b469 aliguori
128 3450df30 aliguori
    for (i = 0; i < MAX_NICS; i++) {
129 3450df30 aliguori
        if (!nd_table[i].used)
130 3450df30 aliguori
            continue;
131 a672b469 aliguori
        len = announce_self_create(buf, nd_table[i].macaddr);
132 a672b469 aliguori
        vlan = nd_table[i].vlan;
133 ed8b330b Gleb Natapov
        for(vc = vlan->first_client; vc != NULL; vc = vc->next) {
134 e3f5ec2b Mark McLoughlin
            vc->receive(vc, buf, len);
135 a672b469 aliguori
        }
136 a672b469 aliguori
    }
137 ed8b330b Gleb Natapov
    if (count--) {
138 ed8b330b Gleb Natapov
            qemu_mod_timer(timer, qemu_get_clock(rt_clock) + 100);
139 ed8b330b Gleb Natapov
    } else {
140 ed8b330b Gleb Natapov
            qemu_del_timer(timer);
141 ed8b330b Gleb Natapov
            qemu_free_timer(timer);
142 ed8b330b Gleb Natapov
    }
143 ed8b330b Gleb Natapov
}
144 ed8b330b Gleb Natapov
145 ed8b330b Gleb Natapov
void qemu_announce_self(void)
146 ed8b330b Gleb Natapov
{
147 ed8b330b Gleb Natapov
        static QEMUTimer *timer;
148 ed8b330b Gleb Natapov
        timer = qemu_new_timer(rt_clock, qemu_announce_self_once, &timer);
149 ed8b330b Gleb Natapov
        qemu_announce_self_once(&timer);
150 a672b469 aliguori
}
151 a672b469 aliguori
152 a672b469 aliguori
/***********************************************************/
153 a672b469 aliguori
/* savevm/loadvm support */
154 a672b469 aliguori
155 a672b469 aliguori
#define IO_BUF_SIZE 32768
156 a672b469 aliguori
157 a672b469 aliguori
struct QEMUFile {
158 a672b469 aliguori
    QEMUFilePutBufferFunc *put_buffer;
159 a672b469 aliguori
    QEMUFileGetBufferFunc *get_buffer;
160 a672b469 aliguori
    QEMUFileCloseFunc *close;
161 a672b469 aliguori
    QEMUFileRateLimit *rate_limit;
162 19629537 Glauber Costa
    QEMUFileSetRateLimit *set_rate_limit;
163 a672b469 aliguori
    void *opaque;
164 a672b469 aliguori
    int is_write;
165 a672b469 aliguori
166 a672b469 aliguori
    int64_t buf_offset; /* start of buffer when writing, end of buffer
167 a672b469 aliguori
                           when reading */
168 a672b469 aliguori
    int buf_index;
169 a672b469 aliguori
    int buf_size; /* 0 when writing */
170 a672b469 aliguori
    uint8_t buf[IO_BUF_SIZE];
171 a672b469 aliguori
172 a672b469 aliguori
    int has_error;
173 a672b469 aliguori
};
174 a672b469 aliguori
175 7f79dd28 Paolo Bonzini
typedef struct QEMUFileStdio
176 a672b469 aliguori
{
177 7f79dd28 Paolo Bonzini
    FILE *stdio_file;
178 a672b469 aliguori
    QEMUFile *file;
179 7f79dd28 Paolo Bonzini
} QEMUFileStdio;
180 a672b469 aliguori
181 a672b469 aliguori
typedef struct QEMUFileSocket
182 a672b469 aliguori
{
183 a672b469 aliguori
    int fd;
184 a672b469 aliguori
    QEMUFile *file;
185 a672b469 aliguori
} QEMUFileSocket;
186 a672b469 aliguori
187 a672b469 aliguori
static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
188 a672b469 aliguori
{
189 a672b469 aliguori
    QEMUFileSocket *s = opaque;
190 a672b469 aliguori
    ssize_t len;
191 a672b469 aliguori
192 a672b469 aliguori
    do {
193 c5b76b38 Blue Swirl
        len = recv(s->fd, (void *)buf, size, 0);
194 a672b469 aliguori
    } while (len == -1 && socket_error() == EINTR);
195 a672b469 aliguori
196 a672b469 aliguori
    if (len == -1)
197 a672b469 aliguori
        len = -socket_error();
198 a672b469 aliguori
199 a672b469 aliguori
    return len;
200 a672b469 aliguori
}
201 a672b469 aliguori
202 a672b469 aliguori
static int socket_close(void *opaque)
203 a672b469 aliguori
{
204 a672b469 aliguori
    QEMUFileSocket *s = opaque;
205 a672b469 aliguori
    qemu_free(s);
206 a672b469 aliguori
    return 0;
207 a672b469 aliguori
}
208 a672b469 aliguori
209 7f79dd28 Paolo Bonzini
static int stdio_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
210 a672b469 aliguori
{
211 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
212 7f79dd28 Paolo Bonzini
    return fwrite(buf, 1, size, s->stdio_file);
213 a672b469 aliguori
}
214 a672b469 aliguori
215 7f79dd28 Paolo Bonzini
static int stdio_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
216 a672b469 aliguori
{
217 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
218 7f79dd28 Paolo Bonzini
    FILE *fp = s->stdio_file;
219 8a67ec4d Uri Lublin
    int bytes;
220 8a67ec4d Uri Lublin
221 8a67ec4d Uri Lublin
    do {
222 8a67ec4d Uri Lublin
        clearerr(fp);
223 8a67ec4d Uri Lublin
        bytes = fread(buf, 1, size, fp);
224 8a67ec4d Uri Lublin
    } while ((bytes == 0) && ferror(fp) && (errno == EINTR));
225 8a67ec4d Uri Lublin
    return bytes;
226 a672b469 aliguori
}
227 a672b469 aliguori
228 7f79dd28 Paolo Bonzini
static int stdio_pclose(void *opaque)
229 7f79dd28 Paolo Bonzini
{
230 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
231 7f79dd28 Paolo Bonzini
    pclose(s->stdio_file);
232 7f79dd28 Paolo Bonzini
    qemu_free(s);
233 7f79dd28 Paolo Bonzini
    return 0;
234 7f79dd28 Paolo Bonzini
}
235 7f79dd28 Paolo Bonzini
236 7f79dd28 Paolo Bonzini
static int stdio_fclose(void *opaque)
237 a672b469 aliguori
{
238 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s = opaque;
239 7f79dd28 Paolo Bonzini
    fclose(s->stdio_file);
240 a672b469 aliguori
    qemu_free(s);
241 a672b469 aliguori
    return 0;
242 a672b469 aliguori
}
243 a672b469 aliguori
244 7f79dd28 Paolo Bonzini
QEMUFile *qemu_popen(FILE *stdio_file, const char *mode)
245 a672b469 aliguori
{
246 7f79dd28 Paolo Bonzini
    QEMUFileStdio *s;
247 a672b469 aliguori
248 7f79dd28 Paolo Bonzini
    if (stdio_file == NULL || mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) {
249 a672b469 aliguori
        fprintf(stderr, "qemu_popen: Argument validity check failed\n");
250 a672b469 aliguori
        return NULL;
251 a672b469 aliguori
    }
252 a672b469 aliguori
253 7f79dd28 Paolo Bonzini
    s = qemu_mallocz(sizeof(QEMUFileStdio));
254 a672b469 aliguori
255 7f79dd28 Paolo Bonzini
    s->stdio_file = stdio_file;
256 a672b469 aliguori
257 a672b469 aliguori
    if(mode[0] == 'r') {
258 7f79dd28 Paolo Bonzini
        s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_pclose, NULL, NULL);
259 a672b469 aliguori
    } else {
260 7f79dd28 Paolo Bonzini
        s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_pclose, NULL, NULL);
261 a672b469 aliguori
    }
262 a672b469 aliguori
    return s->file;
263 a672b469 aliguori
}
264 a672b469 aliguori
265 a672b469 aliguori
QEMUFile *qemu_popen_cmd(const char *command, const char *mode)
266 a672b469 aliguori
{
267 a672b469 aliguori
    FILE *popen_file;
268 a672b469 aliguori
269 a672b469 aliguori
    popen_file = popen(command, mode);
270 a672b469 aliguori
    if(popen_file == NULL) {
271 a672b469 aliguori
        return NULL;
272 a672b469 aliguori
    }
273 a672b469 aliguori
274 a672b469 aliguori
    return qemu_popen(popen_file, mode);
275 a672b469 aliguori
}
276 a672b469 aliguori
277 7f79dd28 Paolo Bonzini
int qemu_stdio_fd(QEMUFile *f)
278 8a43b1ea Chris Lalancette
{
279 7f79dd28 Paolo Bonzini
    QEMUFileStdio *p;
280 8a43b1ea Chris Lalancette
    int fd;
281 8a43b1ea Chris Lalancette
282 7f79dd28 Paolo Bonzini
    p = (QEMUFileStdio *)f->opaque;
283 7f79dd28 Paolo Bonzini
    fd = fileno(p->stdio_file);
284 8a43b1ea Chris Lalancette
285 8a43b1ea Chris Lalancette
    return fd;
286 8a43b1ea Chris Lalancette
}
287 8a43b1ea Chris Lalancette
288 5ac1fad3 Paolo Bonzini
QEMUFile *qemu_fdopen(int fd, const char *mode)
289 5ac1fad3 Paolo Bonzini
{
290 5ac1fad3 Paolo Bonzini
    QEMUFileStdio *s;
291 5ac1fad3 Paolo Bonzini
292 5ac1fad3 Paolo Bonzini
    if (mode == NULL ||
293 5ac1fad3 Paolo Bonzini
        (mode[0] != 'r' && mode[0] != 'w') ||
294 5ac1fad3 Paolo Bonzini
        mode[1] != 'b' || mode[2] != 0) {
295 5ac1fad3 Paolo Bonzini
        fprintf(stderr, "qemu_fdopen: Argument validity check failed\n");
296 5ac1fad3 Paolo Bonzini
        return NULL;
297 5ac1fad3 Paolo Bonzini
    }
298 5ac1fad3 Paolo Bonzini
299 5ac1fad3 Paolo Bonzini
    s = qemu_mallocz(sizeof(QEMUFileStdio));
300 5ac1fad3 Paolo Bonzini
    s->stdio_file = fdopen(fd, mode);
301 5ac1fad3 Paolo Bonzini
    if (!s->stdio_file)
302 5ac1fad3 Paolo Bonzini
        goto fail;
303 5ac1fad3 Paolo Bonzini
304 5ac1fad3 Paolo Bonzini
    if(mode[0] == 'r') {
305 5ac1fad3 Paolo Bonzini
        s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_fclose, NULL, NULL);
306 5ac1fad3 Paolo Bonzini
    } else {
307 5ac1fad3 Paolo Bonzini
        s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_fclose, NULL, NULL);
308 5ac1fad3 Paolo Bonzini
    }
309 5ac1fad3 Paolo Bonzini
    return s->file;
310 5ac1fad3 Paolo Bonzini
311 5ac1fad3 Paolo Bonzini
fail:
312 5ac1fad3 Paolo Bonzini
    qemu_free(s);
313 5ac1fad3 Paolo Bonzini
    return NULL;
314 5ac1fad3 Paolo Bonzini
}
315 5ac1fad3 Paolo Bonzini
316 a672b469 aliguori
QEMUFile *qemu_fopen_socket(int fd)
317 a672b469 aliguori
{
318 a672b469 aliguori
    QEMUFileSocket *s = qemu_mallocz(sizeof(QEMUFileSocket));
319 a672b469 aliguori
320 a672b469 aliguori
    s->fd = fd;
321 19629537 Glauber Costa
    s->file = qemu_fopen_ops(s, NULL, socket_get_buffer, socket_close, NULL, NULL);
322 a672b469 aliguori
    return s->file;
323 a672b469 aliguori
}
324 a672b469 aliguori
325 a672b469 aliguori
static int file_put_buffer(void *opaque, const uint8_t *buf,
326 a672b469 aliguori
                            int64_t pos, int size)
327 a672b469 aliguori
{
328 a672b469 aliguori
    QEMUFileStdio *s = opaque;
329 7f79dd28 Paolo Bonzini
    fseek(s->stdio_file, pos, SEEK_SET);
330 7f79dd28 Paolo Bonzini
    fwrite(buf, 1, size, s->stdio_file);
331 a672b469 aliguori
    return size;
332 a672b469 aliguori
}
333 a672b469 aliguori
334 a672b469 aliguori
static int file_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
335 a672b469 aliguori
{
336 a672b469 aliguori
    QEMUFileStdio *s = opaque;
337 7f79dd28 Paolo Bonzini
    fseek(s->stdio_file, pos, SEEK_SET);
338 7f79dd28 Paolo Bonzini
    return fread(buf, 1, size, s->stdio_file);
339 a672b469 aliguori
}
340 a672b469 aliguori
341 a672b469 aliguori
QEMUFile *qemu_fopen(const char *filename, const char *mode)
342 a672b469 aliguori
{
343 a672b469 aliguori
    QEMUFileStdio *s;
344 a672b469 aliguori
345 7f79dd28 Paolo Bonzini
    if (mode == NULL ||
346 7f79dd28 Paolo Bonzini
        (mode[0] != 'r' && mode[0] != 'w') ||
347 7f79dd28 Paolo Bonzini
        mode[1] != 'b' || mode[2] != 0) {
348 7f79dd28 Paolo Bonzini
        fprintf(stderr, "qemu_fdopen: Argument validity check failed\n");
349 7f79dd28 Paolo Bonzini
        return NULL;
350 7f79dd28 Paolo Bonzini
    }
351 7f79dd28 Paolo Bonzini
352 a672b469 aliguori
    s = qemu_mallocz(sizeof(QEMUFileStdio));
353 a672b469 aliguori
354 7f79dd28 Paolo Bonzini
    s->stdio_file = fopen(filename, mode);
355 7f79dd28 Paolo Bonzini
    if (!s->stdio_file)
356 a672b469 aliguori
        goto fail;
357 a672b469 aliguori
358 7f79dd28 Paolo Bonzini
    if(mode[0] == 'w') {
359 7f79dd28 Paolo Bonzini
        s->file = qemu_fopen_ops(s, file_put_buffer, NULL, stdio_fclose, NULL, NULL);
360 7f79dd28 Paolo Bonzini
    } else {
361 7f79dd28 Paolo Bonzini
        s->file = qemu_fopen_ops(s, NULL, file_get_buffer, stdio_fclose, NULL, NULL);
362 7f79dd28 Paolo Bonzini
    }
363 7f79dd28 Paolo Bonzini
    return s->file;
364 a672b469 aliguori
fail:
365 a672b469 aliguori
    qemu_free(s);
366 a672b469 aliguori
    return NULL;
367 a672b469 aliguori
}
368 a672b469 aliguori
369 178e08a5 aliguori
static int block_put_buffer(void *opaque, const uint8_t *buf,
370 a672b469 aliguori
                           int64_t pos, int size)
371 a672b469 aliguori
{
372 45566e9c Christoph Hellwig
    bdrv_save_vmstate(opaque, buf, pos, size);
373 a672b469 aliguori
    return size;
374 a672b469 aliguori
}
375 a672b469 aliguori
376 178e08a5 aliguori
static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
377 a672b469 aliguori
{
378 45566e9c Christoph Hellwig
    return bdrv_load_vmstate(opaque, buf, pos, size);
379 a672b469 aliguori
}
380 a672b469 aliguori
381 a672b469 aliguori
static int bdrv_fclose(void *opaque)
382 a672b469 aliguori
{
383 a672b469 aliguori
    return 0;
384 a672b469 aliguori
}
385 a672b469 aliguori
386 45566e9c Christoph Hellwig
static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int is_writable)
387 a672b469 aliguori
{
388 a672b469 aliguori
    if (is_writable)
389 45566e9c Christoph Hellwig
        return qemu_fopen_ops(bs, block_put_buffer, NULL, bdrv_fclose, NULL, NULL);
390 45566e9c Christoph Hellwig
    return qemu_fopen_ops(bs, NULL, block_get_buffer, bdrv_fclose, NULL, NULL);
391 a672b469 aliguori
}
392 a672b469 aliguori
393 a672b469 aliguori
QEMUFile *qemu_fopen_ops(void *opaque, QEMUFilePutBufferFunc *put_buffer,
394 a672b469 aliguori
                         QEMUFileGetBufferFunc *get_buffer,
395 a672b469 aliguori
                         QEMUFileCloseFunc *close,
396 19629537 Glauber Costa
                         QEMUFileRateLimit *rate_limit,
397 19629537 Glauber Costa
                         QEMUFileSetRateLimit *set_rate_limit)
398 a672b469 aliguori
{
399 a672b469 aliguori
    QEMUFile *f;
400 a672b469 aliguori
401 a672b469 aliguori
    f = qemu_mallocz(sizeof(QEMUFile));
402 a672b469 aliguori
403 a672b469 aliguori
    f->opaque = opaque;
404 a672b469 aliguori
    f->put_buffer = put_buffer;
405 a672b469 aliguori
    f->get_buffer = get_buffer;
406 a672b469 aliguori
    f->close = close;
407 a672b469 aliguori
    f->rate_limit = rate_limit;
408 19629537 Glauber Costa
    f->set_rate_limit = set_rate_limit;
409 a672b469 aliguori
    f->is_write = 0;
410 a672b469 aliguori
411 a672b469 aliguori
    return f;
412 a672b469 aliguori
}
413 a672b469 aliguori
414 a672b469 aliguori
int qemu_file_has_error(QEMUFile *f)
415 a672b469 aliguori
{
416 a672b469 aliguori
    return f->has_error;
417 a672b469 aliguori
}
418 a672b469 aliguori
419 4dabe248 aliguori
void qemu_file_set_error(QEMUFile *f)
420 4dabe248 aliguori
{
421 4dabe248 aliguori
    f->has_error = 1;
422 4dabe248 aliguori
}
423 4dabe248 aliguori
424 a672b469 aliguori
void qemu_fflush(QEMUFile *f)
425 a672b469 aliguori
{
426 a672b469 aliguori
    if (!f->put_buffer)
427 a672b469 aliguori
        return;
428 a672b469 aliguori
429 a672b469 aliguori
    if (f->is_write && f->buf_index > 0) {
430 a672b469 aliguori
        int len;
431 a672b469 aliguori
432 a672b469 aliguori
        len = f->put_buffer(f->opaque, f->buf, f->buf_offset, f->buf_index);
433 a672b469 aliguori
        if (len > 0)
434 a672b469 aliguori
            f->buf_offset += f->buf_index;
435 a672b469 aliguori
        else
436 a672b469 aliguori
            f->has_error = 1;
437 a672b469 aliguori
        f->buf_index = 0;
438 a672b469 aliguori
    }
439 a672b469 aliguori
}
440 a672b469 aliguori
441 a672b469 aliguori
static void qemu_fill_buffer(QEMUFile *f)
442 a672b469 aliguori
{
443 a672b469 aliguori
    int len;
444 a672b469 aliguori
445 a672b469 aliguori
    if (!f->get_buffer)
446 a672b469 aliguori
        return;
447 a672b469 aliguori
448 a672b469 aliguori
    if (f->is_write)
449 a672b469 aliguori
        abort();
450 a672b469 aliguori
451 a672b469 aliguori
    len = f->get_buffer(f->opaque, f->buf, f->buf_offset, IO_BUF_SIZE);
452 a672b469 aliguori
    if (len > 0) {
453 a672b469 aliguori
        f->buf_index = 0;
454 a672b469 aliguori
        f->buf_size = len;
455 a672b469 aliguori
        f->buf_offset += len;
456 a672b469 aliguori
    } else if (len != -EAGAIN)
457 a672b469 aliguori
        f->has_error = 1;
458 a672b469 aliguori
}
459 a672b469 aliguori
460 a672b469 aliguori
int qemu_fclose(QEMUFile *f)
461 a672b469 aliguori
{
462 a672b469 aliguori
    int ret = 0;
463 a672b469 aliguori
    qemu_fflush(f);
464 a672b469 aliguori
    if (f->close)
465 a672b469 aliguori
        ret = f->close(f->opaque);
466 a672b469 aliguori
    qemu_free(f);
467 a672b469 aliguori
    return ret;
468 a672b469 aliguori
}
469 a672b469 aliguori
470 a672b469 aliguori
void qemu_file_put_notify(QEMUFile *f)
471 a672b469 aliguori
{
472 a672b469 aliguori
    f->put_buffer(f->opaque, NULL, 0, 0);
473 a672b469 aliguori
}
474 a672b469 aliguori
475 a672b469 aliguori
void qemu_put_buffer(QEMUFile *f, const uint8_t *buf, int size)
476 a672b469 aliguori
{
477 a672b469 aliguori
    int l;
478 a672b469 aliguori
479 a672b469 aliguori
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
480 a672b469 aliguori
        fprintf(stderr,
481 a672b469 aliguori
                "Attempted to write to buffer while read buffer is not empty\n");
482 a672b469 aliguori
        abort();
483 a672b469 aliguori
    }
484 a672b469 aliguori
485 a672b469 aliguori
    while (!f->has_error && size > 0) {
486 a672b469 aliguori
        l = IO_BUF_SIZE - f->buf_index;
487 a672b469 aliguori
        if (l > size)
488 a672b469 aliguori
            l = size;
489 a672b469 aliguori
        memcpy(f->buf + f->buf_index, buf, l);
490 a672b469 aliguori
        f->is_write = 1;
491 a672b469 aliguori
        f->buf_index += l;
492 a672b469 aliguori
        buf += l;
493 a672b469 aliguori
        size -= l;
494 a672b469 aliguori
        if (f->buf_index >= IO_BUF_SIZE)
495 a672b469 aliguori
            qemu_fflush(f);
496 a672b469 aliguori
    }
497 a672b469 aliguori
}
498 a672b469 aliguori
499 a672b469 aliguori
void qemu_put_byte(QEMUFile *f, int v)
500 a672b469 aliguori
{
501 a672b469 aliguori
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
502 a672b469 aliguori
        fprintf(stderr,
503 a672b469 aliguori
                "Attempted to write to buffer while read buffer is not empty\n");
504 a672b469 aliguori
        abort();
505 a672b469 aliguori
    }
506 a672b469 aliguori
507 a672b469 aliguori
    f->buf[f->buf_index++] = v;
508 a672b469 aliguori
    f->is_write = 1;
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
int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size1)
514 a672b469 aliguori
{
515 a672b469 aliguori
    int size, l;
516 a672b469 aliguori
517 a672b469 aliguori
    if (f->is_write)
518 a672b469 aliguori
        abort();
519 a672b469 aliguori
520 a672b469 aliguori
    size = size1;
521 a672b469 aliguori
    while (size > 0) {
522 a672b469 aliguori
        l = f->buf_size - f->buf_index;
523 a672b469 aliguori
        if (l == 0) {
524 a672b469 aliguori
            qemu_fill_buffer(f);
525 a672b469 aliguori
            l = f->buf_size - f->buf_index;
526 a672b469 aliguori
            if (l == 0)
527 a672b469 aliguori
                break;
528 a672b469 aliguori
        }
529 a672b469 aliguori
        if (l > size)
530 a672b469 aliguori
            l = size;
531 a672b469 aliguori
        memcpy(buf, f->buf + f->buf_index, l);
532 a672b469 aliguori
        f->buf_index += l;
533 a672b469 aliguori
        buf += l;
534 a672b469 aliguori
        size -= l;
535 a672b469 aliguori
    }
536 a672b469 aliguori
    return size1 - size;
537 a672b469 aliguori
}
538 a672b469 aliguori
539 a672b469 aliguori
int qemu_get_byte(QEMUFile *f)
540 a672b469 aliguori
{
541 a672b469 aliguori
    if (f->is_write)
542 a672b469 aliguori
        abort();
543 a672b469 aliguori
544 a672b469 aliguori
    if (f->buf_index >= f->buf_size) {
545 a672b469 aliguori
        qemu_fill_buffer(f);
546 a672b469 aliguori
        if (f->buf_index >= f->buf_size)
547 a672b469 aliguori
            return 0;
548 a672b469 aliguori
    }
549 a672b469 aliguori
    return f->buf[f->buf_index++];
550 a672b469 aliguori
}
551 a672b469 aliguori
552 a672b469 aliguori
int64_t qemu_ftell(QEMUFile *f)
553 a672b469 aliguori
{
554 a672b469 aliguori
    return f->buf_offset - f->buf_size + f->buf_index;
555 a672b469 aliguori
}
556 a672b469 aliguori
557 a672b469 aliguori
int64_t qemu_fseek(QEMUFile *f, int64_t pos, int whence)
558 a672b469 aliguori
{
559 a672b469 aliguori
    if (whence == SEEK_SET) {
560 a672b469 aliguori
        /* nothing to do */
561 a672b469 aliguori
    } else if (whence == SEEK_CUR) {
562 a672b469 aliguori
        pos += qemu_ftell(f);
563 a672b469 aliguori
    } else {
564 a672b469 aliguori
        /* SEEK_END not supported */
565 a672b469 aliguori
        return -1;
566 a672b469 aliguori
    }
567 a672b469 aliguori
    if (f->put_buffer) {
568 a672b469 aliguori
        qemu_fflush(f);
569 a672b469 aliguori
        f->buf_offset = pos;
570 a672b469 aliguori
    } else {
571 a672b469 aliguori
        f->buf_offset = pos;
572 a672b469 aliguori
        f->buf_index = 0;
573 a672b469 aliguori
        f->buf_size = 0;
574 a672b469 aliguori
    }
575 a672b469 aliguori
    return pos;
576 a672b469 aliguori
}
577 a672b469 aliguori
578 a672b469 aliguori
int qemu_file_rate_limit(QEMUFile *f)
579 a672b469 aliguori
{
580 a672b469 aliguori
    if (f->rate_limit)
581 a672b469 aliguori
        return f->rate_limit(f->opaque);
582 a672b469 aliguori
583 a672b469 aliguori
    return 0;
584 a672b469 aliguori
}
585 a672b469 aliguori
586 19629537 Glauber Costa
size_t qemu_file_set_rate_limit(QEMUFile *f, size_t new_rate)
587 19629537 Glauber Costa
{
588 0bb05eaf Glauber Costa
    /* any failed or completed migration keeps its state to allow probing of
589 0bb05eaf Glauber Costa
     * migration data, but has no associated file anymore */
590 0bb05eaf Glauber Costa
    if (f && f->set_rate_limit)
591 19629537 Glauber Costa
        return f->set_rate_limit(f->opaque, new_rate);
592 19629537 Glauber Costa
593 19629537 Glauber Costa
    return 0;
594 19629537 Glauber Costa
}
595 19629537 Glauber Costa
596 a672b469 aliguori
void qemu_put_be16(QEMUFile *f, unsigned int v)
597 a672b469 aliguori
{
598 a672b469 aliguori
    qemu_put_byte(f, v >> 8);
599 a672b469 aliguori
    qemu_put_byte(f, v);
600 a672b469 aliguori
}
601 a672b469 aliguori
602 a672b469 aliguori
void qemu_put_be32(QEMUFile *f, unsigned int v)
603 a672b469 aliguori
{
604 a672b469 aliguori
    qemu_put_byte(f, v >> 24);
605 a672b469 aliguori
    qemu_put_byte(f, v >> 16);
606 a672b469 aliguori
    qemu_put_byte(f, v >> 8);
607 a672b469 aliguori
    qemu_put_byte(f, v);
608 a672b469 aliguori
}
609 a672b469 aliguori
610 a672b469 aliguori
void qemu_put_be64(QEMUFile *f, uint64_t v)
611 a672b469 aliguori
{
612 a672b469 aliguori
    qemu_put_be32(f, v >> 32);
613 a672b469 aliguori
    qemu_put_be32(f, v);
614 a672b469 aliguori
}
615 a672b469 aliguori
616 a672b469 aliguori
unsigned int qemu_get_be16(QEMUFile *f)
617 a672b469 aliguori
{
618 a672b469 aliguori
    unsigned int v;
619 a672b469 aliguori
    v = qemu_get_byte(f) << 8;
620 a672b469 aliguori
    v |= qemu_get_byte(f);
621 a672b469 aliguori
    return v;
622 a672b469 aliguori
}
623 a672b469 aliguori
624 a672b469 aliguori
unsigned int qemu_get_be32(QEMUFile *f)
625 a672b469 aliguori
{
626 a672b469 aliguori
    unsigned int v;
627 a672b469 aliguori
    v = qemu_get_byte(f) << 24;
628 a672b469 aliguori
    v |= qemu_get_byte(f) << 16;
629 a672b469 aliguori
    v |= qemu_get_byte(f) << 8;
630 a672b469 aliguori
    v |= qemu_get_byte(f);
631 a672b469 aliguori
    return v;
632 a672b469 aliguori
}
633 a672b469 aliguori
634 a672b469 aliguori
uint64_t qemu_get_be64(QEMUFile *f)
635 a672b469 aliguori
{
636 a672b469 aliguori
    uint64_t v;
637 a672b469 aliguori
    v = (uint64_t)qemu_get_be32(f) << 32;
638 a672b469 aliguori
    v |= qemu_get_be32(f);
639 a672b469 aliguori
    return v;
640 a672b469 aliguori
}
641 a672b469 aliguori
642 9ed7d6ae Juan Quintela
/* 8 bit int */
643 9ed7d6ae Juan Quintela
644 9ed7d6ae Juan Quintela
static int get_int8(QEMUFile *f, void *pv, size_t size)
645 9ed7d6ae Juan Quintela
{
646 9ed7d6ae Juan Quintela
    int8_t *v = pv;
647 9ed7d6ae Juan Quintela
    qemu_get_s8s(f, v);
648 9ed7d6ae Juan Quintela
    return 0;
649 9ed7d6ae Juan Quintela
}
650 9ed7d6ae Juan Quintela
651 9ed7d6ae Juan Quintela
static void put_int8(QEMUFile *f, const void *pv, size_t size)
652 9ed7d6ae Juan Quintela
{
653 9ed7d6ae Juan Quintela
    const int8_t *v = pv;
654 9ed7d6ae Juan Quintela
    qemu_put_s8s(f, v);
655 9ed7d6ae Juan Quintela
}
656 9ed7d6ae Juan Quintela
657 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int8 = {
658 9ed7d6ae Juan Quintela
    .name = "int8",
659 9ed7d6ae Juan Quintela
    .get  = get_int8,
660 9ed7d6ae Juan Quintela
    .put  = put_int8,
661 9ed7d6ae Juan Quintela
};
662 9ed7d6ae Juan Quintela
663 9ed7d6ae Juan Quintela
/* 16 bit int */
664 9ed7d6ae Juan Quintela
665 9ed7d6ae Juan Quintela
static int get_int16(QEMUFile *f, void *pv, size_t size)
666 9ed7d6ae Juan Quintela
{
667 9ed7d6ae Juan Quintela
    int16_t *v = pv;
668 9ed7d6ae Juan Quintela
    qemu_get_sbe16s(f, v);
669 9ed7d6ae Juan Quintela
    return 0;
670 9ed7d6ae Juan Quintela
}
671 9ed7d6ae Juan Quintela
672 9ed7d6ae Juan Quintela
static void put_int16(QEMUFile *f, const void *pv, size_t size)
673 9ed7d6ae Juan Quintela
{
674 9ed7d6ae Juan Quintela
    const int16_t *v = pv;
675 9ed7d6ae Juan Quintela
    qemu_put_sbe16s(f, v);
676 9ed7d6ae Juan Quintela
}
677 9ed7d6ae Juan Quintela
678 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int16 = {
679 9ed7d6ae Juan Quintela
    .name = "int16",
680 9ed7d6ae Juan Quintela
    .get  = get_int16,
681 9ed7d6ae Juan Quintela
    .put  = put_int16,
682 9ed7d6ae Juan Quintela
};
683 9ed7d6ae Juan Quintela
684 9ed7d6ae Juan Quintela
/* 32 bit int */
685 9ed7d6ae Juan Quintela
686 9ed7d6ae Juan Quintela
static int get_int32(QEMUFile *f, void *pv, size_t size)
687 9ed7d6ae Juan Quintela
{
688 9ed7d6ae Juan Quintela
    int32_t *v = pv;
689 9ed7d6ae Juan Quintela
    qemu_get_sbe32s(f, v);
690 9ed7d6ae Juan Quintela
    return 0;
691 9ed7d6ae Juan Quintela
}
692 9ed7d6ae Juan Quintela
693 9ed7d6ae Juan Quintela
static void put_int32(QEMUFile *f, const void *pv, size_t size)
694 9ed7d6ae Juan Quintela
{
695 9ed7d6ae Juan Quintela
    const int32_t *v = pv;
696 9ed7d6ae Juan Quintela
    qemu_put_sbe32s(f, v);
697 9ed7d6ae Juan Quintela
}
698 9ed7d6ae Juan Quintela
699 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int32 = {
700 9ed7d6ae Juan Quintela
    .name = "int32",
701 9ed7d6ae Juan Quintela
    .get  = get_int32,
702 9ed7d6ae Juan Quintela
    .put  = put_int32,
703 9ed7d6ae Juan Quintela
};
704 9ed7d6ae Juan Quintela
705 82501660 Juan Quintela
/* 32 bit int. See that the received value is the same than the one
706 82501660 Juan Quintela
   in the field */
707 82501660 Juan Quintela
708 82501660 Juan Quintela
static int get_int32_equal(QEMUFile *f, void *pv, size_t size)
709 82501660 Juan Quintela
{
710 82501660 Juan Quintela
    int32_t *v = pv;
711 82501660 Juan Quintela
    int32_t v2;
712 82501660 Juan Quintela
    qemu_get_sbe32s(f, &v2);
713 82501660 Juan Quintela
714 82501660 Juan Quintela
    if (*v == v2)
715 82501660 Juan Quintela
        return 0;
716 82501660 Juan Quintela
    return -EINVAL;
717 82501660 Juan Quintela
}
718 82501660 Juan Quintela
719 82501660 Juan Quintela
const VMStateInfo vmstate_info_int32_equal = {
720 82501660 Juan Quintela
    .name = "int32 equal",
721 82501660 Juan Quintela
    .get  = get_int32_equal,
722 82501660 Juan Quintela
    .put  = put_int32,
723 82501660 Juan Quintela
};
724 82501660 Juan Quintela
725 0a031e0a Juan Quintela
/* 32 bit int. See that the received value is the less or the same
726 0a031e0a Juan Quintela
   than the one in the field */
727 0a031e0a Juan Quintela
728 0a031e0a Juan Quintela
static int get_int32_le(QEMUFile *f, void *pv, size_t size)
729 0a031e0a Juan Quintela
{
730 0a031e0a Juan Quintela
    int32_t *old = pv;
731 0a031e0a Juan Quintela
    int32_t new;
732 0a031e0a Juan Quintela
    qemu_get_sbe32s(f, &new);
733 0a031e0a Juan Quintela
734 0a031e0a Juan Quintela
    if (*old <= new)
735 0a031e0a Juan Quintela
        return 0;
736 0a031e0a Juan Quintela
    return -EINVAL;
737 0a031e0a Juan Quintela
}
738 0a031e0a Juan Quintela
739 0a031e0a Juan Quintela
const VMStateInfo vmstate_info_int32_le = {
740 0a031e0a Juan Quintela
    .name = "int32 equal",
741 0a031e0a Juan Quintela
    .get  = get_int32_le,
742 0a031e0a Juan Quintela
    .put  = put_int32,
743 0a031e0a Juan Quintela
};
744 0a031e0a Juan Quintela
745 9ed7d6ae Juan Quintela
/* 64 bit int */
746 9ed7d6ae Juan Quintela
747 9ed7d6ae Juan Quintela
static int get_int64(QEMUFile *f, void *pv, size_t size)
748 9ed7d6ae Juan Quintela
{
749 9ed7d6ae Juan Quintela
    int64_t *v = pv;
750 9ed7d6ae Juan Quintela
    qemu_get_sbe64s(f, v);
751 9ed7d6ae Juan Quintela
    return 0;
752 9ed7d6ae Juan Quintela
}
753 9ed7d6ae Juan Quintela
754 9ed7d6ae Juan Quintela
static void put_int64(QEMUFile *f, const void *pv, size_t size)
755 9ed7d6ae Juan Quintela
{
756 9ed7d6ae Juan Quintela
    const int64_t *v = pv;
757 9ed7d6ae Juan Quintela
    qemu_put_sbe64s(f, v);
758 9ed7d6ae Juan Quintela
}
759 9ed7d6ae Juan Quintela
760 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_int64 = {
761 9ed7d6ae Juan Quintela
    .name = "int64",
762 9ed7d6ae Juan Quintela
    .get  = get_int64,
763 9ed7d6ae Juan Quintela
    .put  = put_int64,
764 9ed7d6ae Juan Quintela
};
765 9ed7d6ae Juan Quintela
766 9ed7d6ae Juan Quintela
/* 8 bit unsigned int */
767 9ed7d6ae Juan Quintela
768 9ed7d6ae Juan Quintela
static int get_uint8(QEMUFile *f, void *pv, size_t size)
769 9ed7d6ae Juan Quintela
{
770 9ed7d6ae Juan Quintela
    uint8_t *v = pv;
771 9ed7d6ae Juan Quintela
    qemu_get_8s(f, v);
772 9ed7d6ae Juan Quintela
    return 0;
773 9ed7d6ae Juan Quintela
}
774 9ed7d6ae Juan Quintela
775 9ed7d6ae Juan Quintela
static void put_uint8(QEMUFile *f, const void *pv, size_t size)
776 9ed7d6ae Juan Quintela
{
777 9ed7d6ae Juan Quintela
    const uint8_t *v = pv;
778 9ed7d6ae Juan Quintela
    qemu_put_8s(f, v);
779 9ed7d6ae Juan Quintela
}
780 9ed7d6ae Juan Quintela
781 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint8 = {
782 9ed7d6ae Juan Quintela
    .name = "uint8",
783 9ed7d6ae Juan Quintela
    .get  = get_uint8,
784 9ed7d6ae Juan Quintela
    .put  = put_uint8,
785 9ed7d6ae Juan Quintela
};
786 9ed7d6ae Juan Quintela
787 9ed7d6ae Juan Quintela
/* 16 bit unsigned int */
788 9ed7d6ae Juan Quintela
789 9ed7d6ae Juan Quintela
static int get_uint16(QEMUFile *f, void *pv, size_t size)
790 9ed7d6ae Juan Quintela
{
791 9ed7d6ae Juan Quintela
    uint16_t *v = pv;
792 9ed7d6ae Juan Quintela
    qemu_get_be16s(f, v);
793 9ed7d6ae Juan Quintela
    return 0;
794 9ed7d6ae Juan Quintela
}
795 9ed7d6ae Juan Quintela
796 9ed7d6ae Juan Quintela
static void put_uint16(QEMUFile *f, const void *pv, size_t size)
797 9ed7d6ae Juan Quintela
{
798 9ed7d6ae Juan Quintela
    const uint16_t *v = pv;
799 9ed7d6ae Juan Quintela
    qemu_put_be16s(f, v);
800 9ed7d6ae Juan Quintela
}
801 9ed7d6ae Juan Quintela
802 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint16 = {
803 9ed7d6ae Juan Quintela
    .name = "uint16",
804 9ed7d6ae Juan Quintela
    .get  = get_uint16,
805 9ed7d6ae Juan Quintela
    .put  = put_uint16,
806 9ed7d6ae Juan Quintela
};
807 9ed7d6ae Juan Quintela
808 9ed7d6ae Juan Quintela
/* 32 bit unsigned int */
809 9ed7d6ae Juan Quintela
810 9ed7d6ae Juan Quintela
static int get_uint32(QEMUFile *f, void *pv, size_t size)
811 9ed7d6ae Juan Quintela
{
812 9ed7d6ae Juan Quintela
    uint32_t *v = pv;
813 9ed7d6ae Juan Quintela
    qemu_get_be32s(f, v);
814 9ed7d6ae Juan Quintela
    return 0;
815 9ed7d6ae Juan Quintela
}
816 9ed7d6ae Juan Quintela
817 9ed7d6ae Juan Quintela
static void put_uint32(QEMUFile *f, const void *pv, size_t size)
818 9ed7d6ae Juan Quintela
{
819 9ed7d6ae Juan Quintela
    const uint32_t *v = pv;
820 9ed7d6ae Juan Quintela
    qemu_put_be32s(f, v);
821 9ed7d6ae Juan Quintela
}
822 9ed7d6ae Juan Quintela
823 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint32 = {
824 9ed7d6ae Juan Quintela
    .name = "uint32",
825 9ed7d6ae Juan Quintela
    .get  = get_uint32,
826 9ed7d6ae Juan Quintela
    .put  = put_uint32,
827 9ed7d6ae Juan Quintela
};
828 9ed7d6ae Juan Quintela
829 9ed7d6ae Juan Quintela
/* 64 bit unsigned int */
830 9ed7d6ae Juan Quintela
831 9ed7d6ae Juan Quintela
static int get_uint64(QEMUFile *f, void *pv, size_t size)
832 9ed7d6ae Juan Quintela
{
833 9ed7d6ae Juan Quintela
    uint64_t *v = pv;
834 9ed7d6ae Juan Quintela
    qemu_get_be64s(f, v);
835 9ed7d6ae Juan Quintela
    return 0;
836 9ed7d6ae Juan Quintela
}
837 9ed7d6ae Juan Quintela
838 9ed7d6ae Juan Quintela
static void put_uint64(QEMUFile *f, const void *pv, size_t size)
839 9ed7d6ae Juan Quintela
{
840 9ed7d6ae Juan Quintela
    const uint64_t *v = pv;
841 9ed7d6ae Juan Quintela
    qemu_put_be64s(f, v);
842 9ed7d6ae Juan Quintela
}
843 9ed7d6ae Juan Quintela
844 9ed7d6ae Juan Quintela
const VMStateInfo vmstate_info_uint64 = {
845 9ed7d6ae Juan Quintela
    .name = "uint64",
846 9ed7d6ae Juan Quintela
    .get  = get_uint64,
847 9ed7d6ae Juan Quintela
    .put  = put_uint64,
848 9ed7d6ae Juan Quintela
};
849 9ed7d6ae Juan Quintela
850 dde0463b Juan Quintela
/* timers  */
851 dde0463b Juan Quintela
852 dde0463b Juan Quintela
static int get_timer(QEMUFile *f, void *pv, size_t size)
853 dde0463b Juan Quintela
{
854 dde0463b Juan Quintela
    QEMUTimer *v = pv;
855 dde0463b Juan Quintela
    qemu_get_timer(f, v);
856 dde0463b Juan Quintela
    return 0;
857 dde0463b Juan Quintela
}
858 dde0463b Juan Quintela
859 dde0463b Juan Quintela
static void put_timer(QEMUFile *f, const void *pv, size_t size)
860 dde0463b Juan Quintela
{
861 dde0463b Juan Quintela
    QEMUTimer *v = (void *)pv;
862 dde0463b Juan Quintela
    qemu_put_timer(f, v);
863 dde0463b Juan Quintela
}
864 dde0463b Juan Quintela
865 dde0463b Juan Quintela
const VMStateInfo vmstate_info_timer = {
866 dde0463b Juan Quintela
    .name = "timer",
867 dde0463b Juan Quintela
    .get  = get_timer,
868 dde0463b Juan Quintela
    .put  = put_timer,
869 dde0463b Juan Quintela
};
870 dde0463b Juan Quintela
871 6f67c50f Juan Quintela
/* uint8_t buffers */
872 6f67c50f Juan Quintela
873 6f67c50f Juan Quintela
static int get_buffer(QEMUFile *f, void *pv, size_t size)
874 6f67c50f Juan Quintela
{
875 6f67c50f Juan Quintela
    uint8_t *v = pv;
876 6f67c50f Juan Quintela
    qemu_get_buffer(f, v, size);
877 6f67c50f Juan Quintela
    return 0;
878 6f67c50f Juan Quintela
}
879 6f67c50f Juan Quintela
880 6f67c50f Juan Quintela
static void put_buffer(QEMUFile *f, const void *pv, size_t size)
881 6f67c50f Juan Quintela
{
882 6f67c50f Juan Quintela
    uint8_t *v = (void *)pv;
883 6f67c50f Juan Quintela
    qemu_put_buffer(f, v, size);
884 6f67c50f Juan Quintela
}
885 6f67c50f Juan Quintela
886 6f67c50f Juan Quintela
const VMStateInfo vmstate_info_buffer = {
887 6f67c50f Juan Quintela
    .name = "buffer",
888 6f67c50f Juan Quintela
    .get  = get_buffer,
889 6f67c50f Juan Quintela
    .put  = put_buffer,
890 6f67c50f Juan Quintela
};
891 6f67c50f Juan Quintela
892 a672b469 aliguori
typedef struct SaveStateEntry {
893 a672b469 aliguori
    char idstr[256];
894 a672b469 aliguori
    int instance_id;
895 a672b469 aliguori
    int version_id;
896 a672b469 aliguori
    int section_id;
897 a672b469 aliguori
    SaveLiveStateHandler *save_live_state;
898 a672b469 aliguori
    SaveStateHandler *save_state;
899 a672b469 aliguori
    LoadStateHandler *load_state;
900 9ed7d6ae Juan Quintela
    const VMStateDescription *vmsd;
901 a672b469 aliguori
    void *opaque;
902 a672b469 aliguori
    struct SaveStateEntry *next;
903 a672b469 aliguori
} SaveStateEntry;
904 a672b469 aliguori
905 a672b469 aliguori
static SaveStateEntry *first_se;
906 9ed7d6ae Juan Quintela
static int global_section_id;
907 a672b469 aliguori
908 a672b469 aliguori
/* TODO: Individual devices generally have very little idea about the rest
909 a672b469 aliguori
   of the system, so instance_id should be removed/replaced.
910 a672b469 aliguori
   Meanwhile pass -1 as instance_id if you do not already have a clearly
911 a672b469 aliguori
   distinguishing id for all instances of your device class. */
912 a672b469 aliguori
int register_savevm_live(const char *idstr,
913 a672b469 aliguori
                         int instance_id,
914 a672b469 aliguori
                         int version_id,
915 a672b469 aliguori
                         SaveLiveStateHandler *save_live_state,
916 a672b469 aliguori
                         SaveStateHandler *save_state,
917 a672b469 aliguori
                         LoadStateHandler *load_state,
918 a672b469 aliguori
                         void *opaque)
919 a672b469 aliguori
{
920 a672b469 aliguori
    SaveStateEntry *se, **pse;
921 a672b469 aliguori
922 a672b469 aliguori
    se = qemu_malloc(sizeof(SaveStateEntry));
923 a672b469 aliguori
    pstrcpy(se->idstr, sizeof(se->idstr), idstr);
924 a672b469 aliguori
    se->instance_id = (instance_id == -1) ? 0 : instance_id;
925 a672b469 aliguori
    se->version_id = version_id;
926 a672b469 aliguori
    se->section_id = global_section_id++;
927 a672b469 aliguori
    se->save_live_state = save_live_state;
928 a672b469 aliguori
    se->save_state = save_state;
929 a672b469 aliguori
    se->load_state = load_state;
930 a672b469 aliguori
    se->opaque = opaque;
931 9ed7d6ae Juan Quintela
    se->vmsd = NULL;
932 a672b469 aliguori
    se->next = NULL;
933 a672b469 aliguori
934 a672b469 aliguori
    /* add at the end of list */
935 a672b469 aliguori
    pse = &first_se;
936 a672b469 aliguori
    while (*pse != NULL) {
937 a672b469 aliguori
        if (instance_id == -1
938 a672b469 aliguori
                && strcmp(se->idstr, (*pse)->idstr) == 0
939 a672b469 aliguori
                && se->instance_id <= (*pse)->instance_id)
940 a672b469 aliguori
            se->instance_id = (*pse)->instance_id + 1;
941 a672b469 aliguori
        pse = &(*pse)->next;
942 a672b469 aliguori
    }
943 a672b469 aliguori
    *pse = se;
944 a672b469 aliguori
    return 0;
945 a672b469 aliguori
}
946 a672b469 aliguori
947 a672b469 aliguori
int register_savevm(const char *idstr,
948 a672b469 aliguori
                    int instance_id,
949 a672b469 aliguori
                    int version_id,
950 a672b469 aliguori
                    SaveStateHandler *save_state,
951 a672b469 aliguori
                    LoadStateHandler *load_state,
952 a672b469 aliguori
                    void *opaque)
953 a672b469 aliguori
{
954 a672b469 aliguori
    return register_savevm_live(idstr, instance_id, version_id,
955 a672b469 aliguori
                                NULL, save_state, load_state, opaque);
956 a672b469 aliguori
}
957 a672b469 aliguori
958 41bd13af aliguori
void unregister_savevm(const char *idstr, void *opaque)
959 41bd13af aliguori
{
960 41bd13af aliguori
    SaveStateEntry **pse;
961 41bd13af aliguori
962 41bd13af aliguori
    pse = &first_se;
963 41bd13af aliguori
    while (*pse != NULL) {
964 41bd13af aliguori
        if (strcmp((*pse)->idstr, idstr) == 0 && (*pse)->opaque == opaque) {
965 41bd13af aliguori
            SaveStateEntry *next = (*pse)->next;
966 41bd13af aliguori
            qemu_free(*pse);
967 41bd13af aliguori
            *pse = next;
968 41bd13af aliguori
            continue;
969 41bd13af aliguori
        }
970 41bd13af aliguori
        pse = &(*pse)->next;
971 41bd13af aliguori
    }
972 41bd13af aliguori
}
973 41bd13af aliguori
974 9ed7d6ae Juan Quintela
int vmstate_register(int instance_id, const VMStateDescription *vmsd,
975 9ed7d6ae Juan Quintela
                     void *opaque)
976 9ed7d6ae Juan Quintela
{
977 9ed7d6ae Juan Quintela
    SaveStateEntry *se, **pse;
978 9ed7d6ae Juan Quintela
979 9ed7d6ae Juan Quintela
    se = qemu_malloc(sizeof(SaveStateEntry));
980 9ed7d6ae Juan Quintela
    pstrcpy(se->idstr, sizeof(se->idstr), vmsd->name);
981 9ed7d6ae Juan Quintela
    se->instance_id = (instance_id == -1) ? 0 : instance_id;
982 9ed7d6ae Juan Quintela
    se->version_id = vmsd->version_id;
983 9ed7d6ae Juan Quintela
    se->section_id = global_section_id++;
984 9ed7d6ae Juan Quintela
    se->save_live_state = NULL;
985 9ed7d6ae Juan Quintela
    se->save_state = NULL;
986 9ed7d6ae Juan Quintela
    se->load_state = NULL;
987 9ed7d6ae Juan Quintela
    se->opaque = opaque;
988 9ed7d6ae Juan Quintela
    se->vmsd = vmsd;
989 9ed7d6ae Juan Quintela
    se->next = NULL;
990 9ed7d6ae Juan Quintela
991 9ed7d6ae Juan Quintela
    /* add at the end of list */
992 9ed7d6ae Juan Quintela
    pse = &first_se;
993 9ed7d6ae Juan Quintela
    while (*pse != NULL) {
994 9ed7d6ae Juan Quintela
        if (instance_id == -1
995 9ed7d6ae Juan Quintela
                && strcmp(se->idstr, (*pse)->idstr) == 0
996 9ed7d6ae Juan Quintela
                && se->instance_id <= (*pse)->instance_id)
997 9ed7d6ae Juan Quintela
            se->instance_id = (*pse)->instance_id + 1;
998 9ed7d6ae Juan Quintela
        pse = &(*pse)->next;
999 9ed7d6ae Juan Quintela
    }
1000 9ed7d6ae Juan Quintela
    *pse = se;
1001 9ed7d6ae Juan Quintela
    return 0;
1002 9ed7d6ae Juan Quintela
}
1003 9ed7d6ae Juan Quintela
1004 9ed7d6ae Juan Quintela
void vmstate_unregister(const char *idstr,  void *opaque)
1005 9ed7d6ae Juan Quintela
{
1006 9ed7d6ae Juan Quintela
    SaveStateEntry **pse;
1007 9ed7d6ae Juan Quintela
1008 9ed7d6ae Juan Quintela
    pse = &first_se;
1009 9ed7d6ae Juan Quintela
    while (*pse != NULL) {
1010 9ed7d6ae Juan Quintela
        if (strcmp((*pse)->idstr, idstr) == 0 && (*pse)->opaque == opaque) {
1011 9ed7d6ae Juan Quintela
            SaveStateEntry *next = (*pse)->next;
1012 9ed7d6ae Juan Quintela
            qemu_free(*pse);
1013 9ed7d6ae Juan Quintela
            *pse = next;
1014 9ed7d6ae Juan Quintela
            continue;
1015 9ed7d6ae Juan Quintela
        }
1016 9ed7d6ae Juan Quintela
        pse = &(*pse)->next;
1017 9ed7d6ae Juan Quintela
    }
1018 9ed7d6ae Juan Quintela
}
1019 9ed7d6ae Juan Quintela
1020 9ed7d6ae Juan Quintela
int vmstate_load_state(QEMUFile *f, const VMStateDescription *vmsd,
1021 9ed7d6ae Juan Quintela
                       void *opaque, int version_id)
1022 9ed7d6ae Juan Quintela
{
1023 9ed7d6ae Juan Quintela
    VMStateField *field = vmsd->fields;
1024 9ed7d6ae Juan Quintela
1025 9ed7d6ae Juan Quintela
    if (version_id > vmsd->version_id) {
1026 9ed7d6ae Juan Quintela
        return -EINVAL;
1027 9ed7d6ae Juan Quintela
    }
1028 9ed7d6ae Juan Quintela
    if (version_id < vmsd->minimum_version_id_old) {
1029 9ed7d6ae Juan Quintela
        return -EINVAL;
1030 9ed7d6ae Juan Quintela
    }
1031 9ed7d6ae Juan Quintela
    if  (version_id < vmsd->minimum_version_id) {
1032 9ed7d6ae Juan Quintela
        return vmsd->load_state_old(f, opaque, version_id);
1033 9ed7d6ae Juan Quintela
    }
1034 9ed7d6ae Juan Quintela
    while(field->name) {
1035 9ed7d6ae Juan Quintela
        if (field->version_id <= version_id) {
1036 f752a6aa Juan Quintela
            void *base_addr = opaque + field->offset;
1037 f752a6aa Juan Quintela
            int ret, i, n_elems = 1;
1038 9ed7d6ae Juan Quintela
1039 f752a6aa Juan Quintela
            if (field->flags & VMS_ARRAY) {
1040 f752a6aa Juan Quintela
                n_elems = field->num;
1041 b00319a9 Juan Quintela
            } else if (field->flags & VMS_VARRAY) {
1042 b00319a9 Juan Quintela
                n_elems = *(size_t *)(opaque+field->num_offset);
1043 f752a6aa Juan Quintela
            }
1044 dde0463b Juan Quintela
            if (field->flags & VMS_POINTER) {
1045 f752a6aa Juan Quintela
                base_addr = *(void **)base_addr;
1046 dde0463b Juan Quintela
            }
1047 f752a6aa Juan Quintela
            for (i = 0; i < n_elems; i++) {
1048 f752a6aa Juan Quintela
                void *addr = base_addr + field->size * i;
1049 ec245e21 Juan Quintela
1050 ec245e21 Juan Quintela
                if (field->flags & VMS_STRUCT) {
1051 ec245e21 Juan Quintela
                    ret = vmstate_load_state(f, field->vmsd, addr, version_id);
1052 ec245e21 Juan Quintela
                } else {
1053 ec245e21 Juan Quintela
                    ret = field->info->get(f, addr, field->size);
1054 ec245e21 Juan Quintela
1055 ec245e21 Juan Quintela
                }
1056 f752a6aa Juan Quintela
                if (ret < 0) {
1057 f752a6aa Juan Quintela
                    return ret;
1058 f752a6aa Juan Quintela
                }
1059 9ed7d6ae Juan Quintela
            }
1060 9ed7d6ae Juan Quintela
        }
1061 9ed7d6ae Juan Quintela
        field++;
1062 9ed7d6ae Juan Quintela
    }
1063 cff09e92 Juan Quintela
    if (vmsd->run_after_load)
1064 cff09e92 Juan Quintela
        return vmsd->run_after_load(opaque);
1065 9ed7d6ae Juan Quintela
    return 0;
1066 9ed7d6ae Juan Quintela
}
1067 9ed7d6ae Juan Quintela
1068 9ed7d6ae Juan Quintela
void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd,
1069 9ed7d6ae Juan Quintela
                        const void *opaque)
1070 9ed7d6ae Juan Quintela
{
1071 9ed7d6ae Juan Quintela
    VMStateField *field = vmsd->fields;
1072 9ed7d6ae Juan Quintela
1073 9ed7d6ae Juan Quintela
    while(field->name) {
1074 f752a6aa Juan Quintela
        const void *base_addr = opaque + field->offset;
1075 f752a6aa Juan Quintela
        int i, n_elems = 1;
1076 dde0463b Juan Quintela
1077 f752a6aa Juan Quintela
        if (field->flags & VMS_ARRAY) {
1078 f752a6aa Juan Quintela
            n_elems = field->num;
1079 b00319a9 Juan Quintela
        } else if (field->flags & VMS_VARRAY) {
1080 b00319a9 Juan Quintela
            n_elems = *(size_t *)(opaque+field->num_offset);
1081 f752a6aa Juan Quintela
        }
1082 dde0463b Juan Quintela
        if (field->flags & VMS_POINTER) {
1083 f752a6aa Juan Quintela
            base_addr = *(void **)base_addr;
1084 f752a6aa Juan Quintela
        }
1085 f752a6aa Juan Quintela
        for (i = 0; i < n_elems; i++) {
1086 f752a6aa Juan Quintela
            const void *addr = base_addr + field->size * i;
1087 ec245e21 Juan Quintela
1088 ec245e21 Juan Quintela
            if (field->flags & VMS_STRUCT) {
1089 ec245e21 Juan Quintela
                vmstate_save_state(f, field->vmsd, addr);
1090 ec245e21 Juan Quintela
            } else {
1091 ec245e21 Juan Quintela
                field->info->put(f, addr, field->size);
1092 ec245e21 Juan Quintela
            }
1093 dde0463b Juan Quintela
        }
1094 9ed7d6ae Juan Quintela
        field++;
1095 9ed7d6ae Juan Quintela
    }
1096 9ed7d6ae Juan Quintela
}
1097 9ed7d6ae Juan Quintela
1098 4082be4d Juan Quintela
static int vmstate_load(QEMUFile *f, SaveStateEntry *se, int version_id)
1099 4082be4d Juan Quintela
{
1100 9ed7d6ae Juan Quintela
    if (!se->vmsd) {         /* Old style */
1101 9ed7d6ae Juan Quintela
        return se->load_state(f, se->opaque, version_id);
1102 9ed7d6ae Juan Quintela
    }
1103 9ed7d6ae Juan Quintela
    return vmstate_load_state(f, se->vmsd, se->opaque, version_id);
1104 4082be4d Juan Quintela
}
1105 4082be4d Juan Quintela
1106 4082be4d Juan Quintela
static void vmstate_save(QEMUFile *f, SaveStateEntry *se)
1107 4082be4d Juan Quintela
{
1108 9ed7d6ae Juan Quintela
    if (!se->vmsd) {         /* Old style */
1109 9ed7d6ae Juan Quintela
        se->save_state(f, se->opaque);
1110 9ed7d6ae Juan Quintela
        return;
1111 9ed7d6ae Juan Quintela
    }
1112 9ed7d6ae Juan Quintela
    vmstate_save_state(f,se->vmsd, se->opaque);
1113 4082be4d Juan Quintela
}
1114 4082be4d Juan Quintela
1115 a672b469 aliguori
#define QEMU_VM_FILE_MAGIC           0x5145564d
1116 a672b469 aliguori
#define QEMU_VM_FILE_VERSION_COMPAT  0x00000002
1117 a672b469 aliguori
#define QEMU_VM_FILE_VERSION         0x00000003
1118 a672b469 aliguori
1119 a672b469 aliguori
#define QEMU_VM_EOF                  0x00
1120 a672b469 aliguori
#define QEMU_VM_SECTION_START        0x01
1121 a672b469 aliguori
#define QEMU_VM_SECTION_PART         0x02
1122 a672b469 aliguori
#define QEMU_VM_SECTION_END          0x03
1123 a672b469 aliguori
#define QEMU_VM_SECTION_FULL         0x04
1124 a672b469 aliguori
1125 a672b469 aliguori
int qemu_savevm_state_begin(QEMUFile *f)
1126 a672b469 aliguori
{
1127 a672b469 aliguori
    SaveStateEntry *se;
1128 a672b469 aliguori
1129 a672b469 aliguori
    qemu_put_be32(f, QEMU_VM_FILE_MAGIC);
1130 a672b469 aliguori
    qemu_put_be32(f, QEMU_VM_FILE_VERSION);
1131 a672b469 aliguori
1132 a672b469 aliguori
    for (se = first_se; se != NULL; se = se->next) {
1133 a672b469 aliguori
        int len;
1134 a672b469 aliguori
1135 a672b469 aliguori
        if (se->save_live_state == NULL)
1136 a672b469 aliguori
            continue;
1137 a672b469 aliguori
1138 a672b469 aliguori
        /* Section type */
1139 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_START);
1140 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1141 a672b469 aliguori
1142 a672b469 aliguori
        /* ID string */
1143 a672b469 aliguori
        len = strlen(se->idstr);
1144 a672b469 aliguori
        qemu_put_byte(f, len);
1145 a672b469 aliguori
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1146 a672b469 aliguori
1147 a672b469 aliguori
        qemu_put_be32(f, se->instance_id);
1148 a672b469 aliguori
        qemu_put_be32(f, se->version_id);
1149 a672b469 aliguori
1150 a672b469 aliguori
        se->save_live_state(f, QEMU_VM_SECTION_START, se->opaque);
1151 a672b469 aliguori
    }
1152 a672b469 aliguori
1153 a672b469 aliguori
    if (qemu_file_has_error(f))
1154 a672b469 aliguori
        return -EIO;
1155 a672b469 aliguori
1156 a672b469 aliguori
    return 0;
1157 a672b469 aliguori
}
1158 a672b469 aliguori
1159 a672b469 aliguori
int qemu_savevm_state_iterate(QEMUFile *f)
1160 a672b469 aliguori
{
1161 a672b469 aliguori
    SaveStateEntry *se;
1162 a672b469 aliguori
    int ret = 1;
1163 a672b469 aliguori
1164 a672b469 aliguori
    for (se = first_se; se != NULL; se = se->next) {
1165 a672b469 aliguori
        if (se->save_live_state == NULL)
1166 a672b469 aliguori
            continue;
1167 a672b469 aliguori
1168 a672b469 aliguori
        /* Section type */
1169 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_PART);
1170 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1171 a672b469 aliguori
1172 a672b469 aliguori
        ret &= !!se->save_live_state(f, QEMU_VM_SECTION_PART, se->opaque);
1173 a672b469 aliguori
    }
1174 a672b469 aliguori
1175 a672b469 aliguori
    if (ret)
1176 a672b469 aliguori
        return 1;
1177 a672b469 aliguori
1178 a672b469 aliguori
    if (qemu_file_has_error(f))
1179 a672b469 aliguori
        return -EIO;
1180 a672b469 aliguori
1181 a672b469 aliguori
    return 0;
1182 a672b469 aliguori
}
1183 a672b469 aliguori
1184 a672b469 aliguori
int qemu_savevm_state_complete(QEMUFile *f)
1185 a672b469 aliguori
{
1186 a672b469 aliguori
    SaveStateEntry *se;
1187 a672b469 aliguori
1188 a672b469 aliguori
    for (se = first_se; se != NULL; se = se->next) {
1189 a672b469 aliguori
        if (se->save_live_state == NULL)
1190 a672b469 aliguori
            continue;
1191 a672b469 aliguori
1192 a672b469 aliguori
        /* Section type */
1193 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_END);
1194 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1195 a672b469 aliguori
1196 a672b469 aliguori
        se->save_live_state(f, QEMU_VM_SECTION_END, se->opaque);
1197 a672b469 aliguori
    }
1198 a672b469 aliguori
1199 a672b469 aliguori
    for(se = first_se; se != NULL; se = se->next) {
1200 a672b469 aliguori
        int len;
1201 a672b469 aliguori
1202 9ed7d6ae Juan Quintela
        if (se->save_state == NULL && se->vmsd == NULL)
1203 a672b469 aliguori
            continue;
1204 a672b469 aliguori
1205 a672b469 aliguori
        /* Section type */
1206 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_FULL);
1207 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
1208 a672b469 aliguori
1209 a672b469 aliguori
        /* ID string */
1210 a672b469 aliguori
        len = strlen(se->idstr);
1211 a672b469 aliguori
        qemu_put_byte(f, len);
1212 a672b469 aliguori
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1213 a672b469 aliguori
1214 a672b469 aliguori
        qemu_put_be32(f, se->instance_id);
1215 a672b469 aliguori
        qemu_put_be32(f, se->version_id);
1216 a672b469 aliguori
1217 4082be4d Juan Quintela
        vmstate_save(f, se);
1218 a672b469 aliguori
    }
1219 a672b469 aliguori
1220 a672b469 aliguori
    qemu_put_byte(f, QEMU_VM_EOF);
1221 a672b469 aliguori
1222 a672b469 aliguori
    if (qemu_file_has_error(f))
1223 a672b469 aliguori
        return -EIO;
1224 a672b469 aliguori
1225 a672b469 aliguori
    return 0;
1226 a672b469 aliguori
}
1227 a672b469 aliguori
1228 a672b469 aliguori
int qemu_savevm_state(QEMUFile *f)
1229 a672b469 aliguori
{
1230 a672b469 aliguori
    int saved_vm_running;
1231 a672b469 aliguori
    int ret;
1232 a672b469 aliguori
1233 a672b469 aliguori
    saved_vm_running = vm_running;
1234 a672b469 aliguori
    vm_stop(0);
1235 a672b469 aliguori
1236 a672b469 aliguori
    bdrv_flush_all();
1237 a672b469 aliguori
1238 a672b469 aliguori
    ret = qemu_savevm_state_begin(f);
1239 a672b469 aliguori
    if (ret < 0)
1240 a672b469 aliguori
        goto out;
1241 a672b469 aliguori
1242 a672b469 aliguori
    do {
1243 a672b469 aliguori
        ret = qemu_savevm_state_iterate(f);
1244 a672b469 aliguori
        if (ret < 0)
1245 a672b469 aliguori
            goto out;
1246 a672b469 aliguori
    } while (ret == 0);
1247 a672b469 aliguori
1248 a672b469 aliguori
    ret = qemu_savevm_state_complete(f);
1249 a672b469 aliguori
1250 a672b469 aliguori
out:
1251 a672b469 aliguori
    if (qemu_file_has_error(f))
1252 a672b469 aliguori
        ret = -EIO;
1253 a672b469 aliguori
1254 a672b469 aliguori
    if (!ret && saved_vm_running)
1255 a672b469 aliguori
        vm_start();
1256 a672b469 aliguori
1257 a672b469 aliguori
    return ret;
1258 a672b469 aliguori
}
1259 a672b469 aliguori
1260 a672b469 aliguori
static SaveStateEntry *find_se(const char *idstr, int instance_id)
1261 a672b469 aliguori
{
1262 a672b469 aliguori
    SaveStateEntry *se;
1263 a672b469 aliguori
1264 a672b469 aliguori
    for(se = first_se; se != NULL; se = se->next) {
1265 a672b469 aliguori
        if (!strcmp(se->idstr, idstr) &&
1266 a672b469 aliguori
            instance_id == se->instance_id)
1267 a672b469 aliguori
            return se;
1268 a672b469 aliguori
    }
1269 a672b469 aliguori
    return NULL;
1270 a672b469 aliguori
}
1271 a672b469 aliguori
1272 a672b469 aliguori
typedef struct LoadStateEntry {
1273 a672b469 aliguori
    SaveStateEntry *se;
1274 a672b469 aliguori
    int section_id;
1275 a672b469 aliguori
    int version_id;
1276 a672b469 aliguori
    struct LoadStateEntry *next;
1277 a672b469 aliguori
} LoadStateEntry;
1278 a672b469 aliguori
1279 a672b469 aliguori
static int qemu_loadvm_state_v2(QEMUFile *f)
1280 a672b469 aliguori
{
1281 a672b469 aliguori
    SaveStateEntry *se;
1282 a672b469 aliguori
    int len, ret, instance_id, record_len, version_id;
1283 a672b469 aliguori
    int64_t total_len, end_pos, cur_pos;
1284 a672b469 aliguori
    char idstr[256];
1285 a672b469 aliguori
1286 a672b469 aliguori
    total_len = qemu_get_be64(f);
1287 a672b469 aliguori
    end_pos = total_len + qemu_ftell(f);
1288 a672b469 aliguori
    for(;;) {
1289 a672b469 aliguori
        if (qemu_ftell(f) >= end_pos)
1290 a672b469 aliguori
            break;
1291 a672b469 aliguori
        len = qemu_get_byte(f);
1292 a672b469 aliguori
        qemu_get_buffer(f, (uint8_t *)idstr, len);
1293 a672b469 aliguori
        idstr[len] = '\0';
1294 a672b469 aliguori
        instance_id = qemu_get_be32(f);
1295 a672b469 aliguori
        version_id = qemu_get_be32(f);
1296 a672b469 aliguori
        record_len = qemu_get_be32(f);
1297 a672b469 aliguori
        cur_pos = qemu_ftell(f);
1298 a672b469 aliguori
        se = find_se(idstr, instance_id);
1299 a672b469 aliguori
        if (!se) {
1300 a672b469 aliguori
            fprintf(stderr, "qemu: warning: instance 0x%x of device '%s' not present in current VM\n",
1301 a672b469 aliguori
                    instance_id, idstr);
1302 a672b469 aliguori
        } else {
1303 4082be4d Juan Quintela
            ret = vmstate_load(f, se, version_id);
1304 a672b469 aliguori
            if (ret < 0) {
1305 a672b469 aliguori
                fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
1306 a672b469 aliguori
                        instance_id, idstr);
1307 d02f7094 Anthony Liguori
                return ret;
1308 a672b469 aliguori
            }
1309 a672b469 aliguori
        }
1310 a672b469 aliguori
        /* always seek to exact end of record */
1311 a672b469 aliguori
        qemu_fseek(f, cur_pos + record_len, SEEK_SET);
1312 a672b469 aliguori
    }
1313 a672b469 aliguori
1314 a672b469 aliguori
    if (qemu_file_has_error(f))
1315 a672b469 aliguori
        return -EIO;
1316 a672b469 aliguori
1317 a672b469 aliguori
    return 0;
1318 a672b469 aliguori
}
1319 a672b469 aliguori
1320 a672b469 aliguori
int qemu_loadvm_state(QEMUFile *f)
1321 a672b469 aliguori
{
1322 a672b469 aliguori
    LoadStateEntry *first_le = NULL;
1323 a672b469 aliguori
    uint8_t section_type;
1324 a672b469 aliguori
    unsigned int v;
1325 a672b469 aliguori
    int ret;
1326 a672b469 aliguori
1327 a672b469 aliguori
    v = qemu_get_be32(f);
1328 a672b469 aliguori
    if (v != QEMU_VM_FILE_MAGIC)
1329 a672b469 aliguori
        return -EINVAL;
1330 a672b469 aliguori
1331 a672b469 aliguori
    v = qemu_get_be32(f);
1332 a672b469 aliguori
    if (v == QEMU_VM_FILE_VERSION_COMPAT)
1333 a672b469 aliguori
        return qemu_loadvm_state_v2(f);
1334 a672b469 aliguori
    if (v != QEMU_VM_FILE_VERSION)
1335 a672b469 aliguori
        return -ENOTSUP;
1336 a672b469 aliguori
1337 a672b469 aliguori
    while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
1338 a672b469 aliguori
        uint32_t instance_id, version_id, section_id;
1339 a672b469 aliguori
        LoadStateEntry *le;
1340 a672b469 aliguori
        SaveStateEntry *se;
1341 a672b469 aliguori
        char idstr[257];
1342 a672b469 aliguori
        int len;
1343 a672b469 aliguori
1344 a672b469 aliguori
        switch (section_type) {
1345 a672b469 aliguori
        case QEMU_VM_SECTION_START:
1346 a672b469 aliguori
        case QEMU_VM_SECTION_FULL:
1347 a672b469 aliguori
            /* Read section start */
1348 a672b469 aliguori
            section_id = qemu_get_be32(f);
1349 a672b469 aliguori
            len = qemu_get_byte(f);
1350 a672b469 aliguori
            qemu_get_buffer(f, (uint8_t *)idstr, len);
1351 a672b469 aliguori
            idstr[len] = 0;
1352 a672b469 aliguori
            instance_id = qemu_get_be32(f);
1353 a672b469 aliguori
            version_id = qemu_get_be32(f);
1354 a672b469 aliguori
1355 a672b469 aliguori
            /* Find savevm section */
1356 a672b469 aliguori
            se = find_se(idstr, instance_id);
1357 a672b469 aliguori
            if (se == NULL) {
1358 a672b469 aliguori
                fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
1359 a672b469 aliguori
                ret = -EINVAL;
1360 a672b469 aliguori
                goto out;
1361 a672b469 aliguori
            }
1362 a672b469 aliguori
1363 a672b469 aliguori
            /* Validate version */
1364 a672b469 aliguori
            if (version_id > se->version_id) {
1365 a672b469 aliguori
                fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
1366 a672b469 aliguori
                        version_id, idstr, se->version_id);
1367 a672b469 aliguori
                ret = -EINVAL;
1368 a672b469 aliguori
                goto out;
1369 a672b469 aliguori
            }
1370 a672b469 aliguori
1371 a672b469 aliguori
            /* Add entry */
1372 a672b469 aliguori
            le = qemu_mallocz(sizeof(*le));
1373 a672b469 aliguori
1374 a672b469 aliguori
            le->se = se;
1375 a672b469 aliguori
            le->section_id = section_id;
1376 a672b469 aliguori
            le->version_id = version_id;
1377 a672b469 aliguori
            le->next = first_le;
1378 a672b469 aliguori
            first_le = le;
1379 a672b469 aliguori
1380 4082be4d Juan Quintela
            ret = vmstate_load(f, le->se, le->version_id);
1381 b5a22e4a Juan Quintela
            if (ret < 0) {
1382 b5a22e4a Juan Quintela
                fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
1383 b5a22e4a Juan Quintela
                        instance_id, idstr);
1384 b5a22e4a Juan Quintela
                goto out;
1385 b5a22e4a Juan Quintela
            }
1386 a672b469 aliguori
            break;
1387 a672b469 aliguori
        case QEMU_VM_SECTION_PART:
1388 a672b469 aliguori
        case QEMU_VM_SECTION_END:
1389 a672b469 aliguori
            section_id = qemu_get_be32(f);
1390 a672b469 aliguori
1391 a672b469 aliguori
            for (le = first_le; le && le->section_id != section_id; le = le->next);
1392 a672b469 aliguori
            if (le == NULL) {
1393 a672b469 aliguori
                fprintf(stderr, "Unknown savevm section %d\n", section_id);
1394 a672b469 aliguori
                ret = -EINVAL;
1395 a672b469 aliguori
                goto out;
1396 a672b469 aliguori
            }
1397 a672b469 aliguori
1398 4082be4d Juan Quintela
            ret = vmstate_load(f, le->se, le->version_id);
1399 b5a22e4a Juan Quintela
            if (ret < 0) {
1400 b5a22e4a Juan Quintela
                fprintf(stderr, "qemu: warning: error while loading state section id %d\n",
1401 b5a22e4a Juan Quintela
                        section_id);
1402 b5a22e4a Juan Quintela
                goto out;
1403 b5a22e4a Juan Quintela
            }
1404 a672b469 aliguori
            break;
1405 a672b469 aliguori
        default:
1406 a672b469 aliguori
            fprintf(stderr, "Unknown savevm section type %d\n", section_type);
1407 a672b469 aliguori
            ret = -EINVAL;
1408 a672b469 aliguori
            goto out;
1409 a672b469 aliguori
        }
1410 a672b469 aliguori
    }
1411 a672b469 aliguori
1412 a672b469 aliguori
    ret = 0;
1413 a672b469 aliguori
1414 a672b469 aliguori
out:
1415 a672b469 aliguori
    while (first_le) {
1416 a672b469 aliguori
        LoadStateEntry *le = first_le;
1417 a672b469 aliguori
        first_le = first_le->next;
1418 a672b469 aliguori
        qemu_free(le);
1419 a672b469 aliguori
    }
1420 a672b469 aliguori
1421 a672b469 aliguori
    if (qemu_file_has_error(f))
1422 a672b469 aliguori
        ret = -EIO;
1423 a672b469 aliguori
1424 a672b469 aliguori
    return ret;
1425 a672b469 aliguori
}
1426 a672b469 aliguori
1427 a672b469 aliguori
/* device can contain snapshots */
1428 a672b469 aliguori
static int bdrv_can_snapshot(BlockDriverState *bs)
1429 a672b469 aliguori
{
1430 a672b469 aliguori
    return (bs &&
1431 a672b469 aliguori
            !bdrv_is_removable(bs) &&
1432 a672b469 aliguori
            !bdrv_is_read_only(bs));
1433 a672b469 aliguori
}
1434 a672b469 aliguori
1435 a672b469 aliguori
/* device must be snapshots in order to have a reliable snapshot */
1436 a672b469 aliguori
static int bdrv_has_snapshot(BlockDriverState *bs)
1437 a672b469 aliguori
{
1438 a672b469 aliguori
    return (bs &&
1439 a672b469 aliguori
            !bdrv_is_removable(bs) &&
1440 a672b469 aliguori
            !bdrv_is_read_only(bs));
1441 a672b469 aliguori
}
1442 a672b469 aliguori
1443 a672b469 aliguori
static BlockDriverState *get_bs_snapshots(void)
1444 a672b469 aliguori
{
1445 a672b469 aliguori
    BlockDriverState *bs;
1446 751c6a17 Gerd Hoffmann
    DriveInfo *dinfo;
1447 a672b469 aliguori
1448 a672b469 aliguori
    if (bs_snapshots)
1449 a672b469 aliguori
        return bs_snapshots;
1450 751c6a17 Gerd Hoffmann
    TAILQ_FOREACH(dinfo, &drives, next) {
1451 751c6a17 Gerd Hoffmann
        bs = dinfo->bdrv;
1452 a672b469 aliguori
        if (bdrv_can_snapshot(bs))
1453 a672b469 aliguori
            goto ok;
1454 a672b469 aliguori
    }
1455 a672b469 aliguori
    return NULL;
1456 a672b469 aliguori
 ok:
1457 a672b469 aliguori
    bs_snapshots = bs;
1458 a672b469 aliguori
    return bs;
1459 a672b469 aliguori
}
1460 a672b469 aliguori
1461 a672b469 aliguori
static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
1462 a672b469 aliguori
                              const char *name)
1463 a672b469 aliguori
{
1464 a672b469 aliguori
    QEMUSnapshotInfo *sn_tab, *sn;
1465 a672b469 aliguori
    int nb_sns, i, ret;
1466 a672b469 aliguori
1467 a672b469 aliguori
    ret = -ENOENT;
1468 a672b469 aliguori
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1469 a672b469 aliguori
    if (nb_sns < 0)
1470 a672b469 aliguori
        return ret;
1471 a672b469 aliguori
    for(i = 0; i < nb_sns; i++) {
1472 a672b469 aliguori
        sn = &sn_tab[i];
1473 a672b469 aliguori
        if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
1474 a672b469 aliguori
            *sn_info = *sn;
1475 a672b469 aliguori
            ret = 0;
1476 a672b469 aliguori
            break;
1477 a672b469 aliguori
        }
1478 a672b469 aliguori
    }
1479 a672b469 aliguori
    qemu_free(sn_tab);
1480 a672b469 aliguori
    return ret;
1481 a672b469 aliguori
}
1482 a672b469 aliguori
1483 376253ec aliguori
void do_savevm(Monitor *mon, const char *name)
1484 a672b469 aliguori
{
1485 751c6a17 Gerd Hoffmann
    DriveInfo *dinfo;
1486 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1487 a672b469 aliguori
    QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1;
1488 751c6a17 Gerd Hoffmann
    int must_delete, ret;
1489 a672b469 aliguori
    QEMUFile *f;
1490 a672b469 aliguori
    int saved_vm_running;
1491 2d22b18f aliguori
    uint32_t vm_state_size;
1492 a672b469 aliguori
#ifdef _WIN32
1493 a672b469 aliguori
    struct _timeb tb;
1494 a672b469 aliguori
#else
1495 a672b469 aliguori
    struct timeval tv;
1496 a672b469 aliguori
#endif
1497 a672b469 aliguori
1498 a672b469 aliguori
    bs = get_bs_snapshots();
1499 a672b469 aliguori
    if (!bs) {
1500 376253ec aliguori
        monitor_printf(mon, "No block device can accept snapshots\n");
1501 a672b469 aliguori
        return;
1502 a672b469 aliguori
    }
1503 a672b469 aliguori
1504 a672b469 aliguori
    /* ??? Should this occur after vm_stop?  */
1505 a672b469 aliguori
    qemu_aio_flush();
1506 a672b469 aliguori
1507 a672b469 aliguori
    saved_vm_running = vm_running;
1508 a672b469 aliguori
    vm_stop(0);
1509 a672b469 aliguori
1510 a672b469 aliguori
    must_delete = 0;
1511 a672b469 aliguori
    if (name) {
1512 a672b469 aliguori
        ret = bdrv_snapshot_find(bs, old_sn, name);
1513 a672b469 aliguori
        if (ret >= 0) {
1514 a672b469 aliguori
            must_delete = 1;
1515 a672b469 aliguori
        }
1516 a672b469 aliguori
    }
1517 a672b469 aliguori
    memset(sn, 0, sizeof(*sn));
1518 a672b469 aliguori
    if (must_delete) {
1519 a672b469 aliguori
        pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
1520 a672b469 aliguori
        pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
1521 a672b469 aliguori
    } else {
1522 a672b469 aliguori
        if (name)
1523 a672b469 aliguori
            pstrcpy(sn->name, sizeof(sn->name), name);
1524 a672b469 aliguori
    }
1525 a672b469 aliguori
1526 a672b469 aliguori
    /* fill auxiliary fields */
1527 a672b469 aliguori
#ifdef _WIN32
1528 a672b469 aliguori
    _ftime(&tb);
1529 a672b469 aliguori
    sn->date_sec = tb.time;
1530 a672b469 aliguori
    sn->date_nsec = tb.millitm * 1000000;
1531 a672b469 aliguori
#else
1532 a672b469 aliguori
    gettimeofday(&tv, NULL);
1533 a672b469 aliguori
    sn->date_sec = tv.tv_sec;
1534 a672b469 aliguori
    sn->date_nsec = tv.tv_usec * 1000;
1535 a672b469 aliguori
#endif
1536 a672b469 aliguori
    sn->vm_clock_nsec = qemu_get_clock(vm_clock);
1537 a672b469 aliguori
1538 a672b469 aliguori
    /* save the VM state */
1539 45566e9c Christoph Hellwig
    f = qemu_fopen_bdrv(bs, 1);
1540 a672b469 aliguori
    if (!f) {
1541 376253ec aliguori
        monitor_printf(mon, "Could not open VM state file\n");
1542 a672b469 aliguori
        goto the_end;
1543 a672b469 aliguori
    }
1544 a672b469 aliguori
    ret = qemu_savevm_state(f);
1545 2d22b18f aliguori
    vm_state_size = qemu_ftell(f);
1546 a672b469 aliguori
    qemu_fclose(f);
1547 a672b469 aliguori
    if (ret < 0) {
1548 376253ec aliguori
        monitor_printf(mon, "Error %d while writing VM\n", ret);
1549 a672b469 aliguori
        goto the_end;
1550 a672b469 aliguori
    }
1551 a672b469 aliguori
1552 a672b469 aliguori
    /* create the snapshots */
1553 a672b469 aliguori
1554 751c6a17 Gerd Hoffmann
    TAILQ_FOREACH(dinfo, &drives, next) {
1555 751c6a17 Gerd Hoffmann
        bs1 = dinfo->bdrv;
1556 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1557 a672b469 aliguori
            if (must_delete) {
1558 a672b469 aliguori
                ret = bdrv_snapshot_delete(bs1, old_sn->id_str);
1559 a672b469 aliguori
                if (ret < 0) {
1560 376253ec aliguori
                    monitor_printf(mon,
1561 376253ec aliguori
                                   "Error while deleting snapshot on '%s'\n",
1562 376253ec aliguori
                                   bdrv_get_device_name(bs1));
1563 a672b469 aliguori
                }
1564 a672b469 aliguori
            }
1565 2d22b18f aliguori
            /* Write VM state size only to the image that contains the state */
1566 2d22b18f aliguori
            sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
1567 a672b469 aliguori
            ret = bdrv_snapshot_create(bs1, sn);
1568 a672b469 aliguori
            if (ret < 0) {
1569 376253ec aliguori
                monitor_printf(mon, "Error while creating snapshot on '%s'\n",
1570 376253ec aliguori
                               bdrv_get_device_name(bs1));
1571 a672b469 aliguori
            }
1572 a672b469 aliguori
        }
1573 a672b469 aliguori
    }
1574 a672b469 aliguori
1575 a672b469 aliguori
 the_end:
1576 a672b469 aliguori
    if (saved_vm_running)
1577 a672b469 aliguori
        vm_start();
1578 a672b469 aliguori
}
1579 a672b469 aliguori
1580 05f2401e Juan Quintela
int load_vmstate(Monitor *mon, const char *name)
1581 a672b469 aliguori
{
1582 751c6a17 Gerd Hoffmann
    DriveInfo *dinfo;
1583 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1584 2d22b18f aliguori
    QEMUSnapshotInfo sn;
1585 a672b469 aliguori
    QEMUFile *f;
1586 751c6a17 Gerd Hoffmann
    int ret;
1587 a672b469 aliguori
1588 a672b469 aliguori
    bs = get_bs_snapshots();
1589 a672b469 aliguori
    if (!bs) {
1590 376253ec aliguori
        monitor_printf(mon, "No block device supports snapshots\n");
1591 05f2401e Juan Quintela
        return -EINVAL;
1592 a672b469 aliguori
    }
1593 a672b469 aliguori
1594 a672b469 aliguori
    /* Flush all IO requests so they don't interfere with the new state.  */
1595 a672b469 aliguori
    qemu_aio_flush();
1596 a672b469 aliguori
1597 751c6a17 Gerd Hoffmann
    TAILQ_FOREACH(dinfo, &drives, next) {
1598 751c6a17 Gerd Hoffmann
        bs1 = dinfo->bdrv;
1599 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1600 a672b469 aliguori
            ret = bdrv_snapshot_goto(bs1, name);
1601 a672b469 aliguori
            if (ret < 0) {
1602 a672b469 aliguori
                if (bs != bs1)
1603 376253ec aliguori
                    monitor_printf(mon, "Warning: ");
1604 a672b469 aliguori
                switch(ret) {
1605 a672b469 aliguori
                case -ENOTSUP:
1606 376253ec aliguori
                    monitor_printf(mon,
1607 376253ec aliguori
                                   "Snapshots not supported on device '%s'\n",
1608 376253ec aliguori
                                   bdrv_get_device_name(bs1));
1609 a672b469 aliguori
                    break;
1610 a672b469 aliguori
                case -ENOENT:
1611 376253ec aliguori
                    monitor_printf(mon, "Could not find snapshot '%s' on "
1612 376253ec aliguori
                                   "device '%s'\n",
1613 376253ec aliguori
                                   name, bdrv_get_device_name(bs1));
1614 a672b469 aliguori
                    break;
1615 a672b469 aliguori
                default:
1616 376253ec aliguori
                    monitor_printf(mon, "Error %d while activating snapshot on"
1617 376253ec aliguori
                                   " '%s'\n", ret, bdrv_get_device_name(bs1));
1618 a672b469 aliguori
                    break;
1619 a672b469 aliguori
                }
1620 a672b469 aliguori
                /* fatal on snapshot block device */
1621 a672b469 aliguori
                if (bs == bs1)
1622 05f2401e Juan Quintela
                    return 0;
1623 a672b469 aliguori
            }
1624 a672b469 aliguori
        }
1625 a672b469 aliguori
    }
1626 a672b469 aliguori
1627 2d22b18f aliguori
    /* Don't even try to load empty VM states */
1628 2d22b18f aliguori
    ret = bdrv_snapshot_find(bs, &sn, name);
1629 2d22b18f aliguori
    if ((ret >= 0) && (sn.vm_state_size == 0))
1630 05f2401e Juan Quintela
        return -EINVAL;
1631 2d22b18f aliguori
1632 a672b469 aliguori
    /* restore the VM state */
1633 45566e9c Christoph Hellwig
    f = qemu_fopen_bdrv(bs, 0);
1634 a672b469 aliguori
    if (!f) {
1635 376253ec aliguori
        monitor_printf(mon, "Could not open VM state file\n");
1636 05f2401e Juan Quintela
        return -EINVAL;
1637 a672b469 aliguori
    }
1638 a672b469 aliguori
    ret = qemu_loadvm_state(f);
1639 a672b469 aliguori
    qemu_fclose(f);
1640 a672b469 aliguori
    if (ret < 0) {
1641 376253ec aliguori
        monitor_printf(mon, "Error %d while loading VM state\n", ret);
1642 05f2401e Juan Quintela
        return ret;
1643 a672b469 aliguori
    }
1644 05f2401e Juan Quintela
    return 0;
1645 7b630349 Juan Quintela
}
1646 7b630349 Juan Quintela
1647 376253ec aliguori
void do_delvm(Monitor *mon, const char *name)
1648 a672b469 aliguori
{
1649 751c6a17 Gerd Hoffmann
    DriveInfo *dinfo;
1650 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1651 751c6a17 Gerd Hoffmann
    int ret;
1652 a672b469 aliguori
1653 a672b469 aliguori
    bs = get_bs_snapshots();
1654 a672b469 aliguori
    if (!bs) {
1655 376253ec aliguori
        monitor_printf(mon, "No block device supports snapshots\n");
1656 a672b469 aliguori
        return;
1657 a672b469 aliguori
    }
1658 a672b469 aliguori
1659 751c6a17 Gerd Hoffmann
    TAILQ_FOREACH(dinfo, &drives, next) {
1660 751c6a17 Gerd Hoffmann
        bs1 = dinfo->bdrv;
1661 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1662 a672b469 aliguori
            ret = bdrv_snapshot_delete(bs1, name);
1663 a672b469 aliguori
            if (ret < 0) {
1664 a672b469 aliguori
                if (ret == -ENOTSUP)
1665 376253ec aliguori
                    monitor_printf(mon,
1666 376253ec aliguori
                                   "Snapshots not supported on device '%s'\n",
1667 376253ec aliguori
                                   bdrv_get_device_name(bs1));
1668 a672b469 aliguori
                else
1669 376253ec aliguori
                    monitor_printf(mon, "Error %d while deleting snapshot on "
1670 376253ec aliguori
                                   "'%s'\n", ret, bdrv_get_device_name(bs1));
1671 a672b469 aliguori
            }
1672 a672b469 aliguori
        }
1673 a672b469 aliguori
    }
1674 a672b469 aliguori
}
1675 a672b469 aliguori
1676 376253ec aliguori
void do_info_snapshots(Monitor *mon)
1677 a672b469 aliguori
{
1678 751c6a17 Gerd Hoffmann
    DriveInfo *dinfo;
1679 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1680 a672b469 aliguori
    QEMUSnapshotInfo *sn_tab, *sn;
1681 a672b469 aliguori
    int nb_sns, i;
1682 a672b469 aliguori
    char buf[256];
1683 a672b469 aliguori
1684 a672b469 aliguori
    bs = get_bs_snapshots();
1685 a672b469 aliguori
    if (!bs) {
1686 376253ec aliguori
        monitor_printf(mon, "No available block device supports snapshots\n");
1687 a672b469 aliguori
        return;
1688 a672b469 aliguori
    }
1689 376253ec aliguori
    monitor_printf(mon, "Snapshot devices:");
1690 751c6a17 Gerd Hoffmann
    TAILQ_FOREACH(dinfo, &drives, next) {
1691 751c6a17 Gerd Hoffmann
        bs1 = dinfo->bdrv;
1692 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1693 a672b469 aliguori
            if (bs == bs1)
1694 376253ec aliguori
                monitor_printf(mon, " %s", bdrv_get_device_name(bs1));
1695 a672b469 aliguori
        }
1696 a672b469 aliguori
    }
1697 376253ec aliguori
    monitor_printf(mon, "\n");
1698 a672b469 aliguori
1699 a672b469 aliguori
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1700 a672b469 aliguori
    if (nb_sns < 0) {
1701 376253ec aliguori
        monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
1702 a672b469 aliguori
        return;
1703 a672b469 aliguori
    }
1704 376253ec aliguori
    monitor_printf(mon, "Snapshot list (from %s):\n",
1705 376253ec aliguori
                   bdrv_get_device_name(bs));
1706 376253ec aliguori
    monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
1707 a672b469 aliguori
    for(i = 0; i < nb_sns; i++) {
1708 a672b469 aliguori
        sn = &sn_tab[i];
1709 376253ec aliguori
        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
1710 a672b469 aliguori
    }
1711 a672b469 aliguori
    qemu_free(sn_tab);
1712 a672b469 aliguori
}