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