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