Merge branch 'stable-0.11' of git://git.sv.gnu.org/qemu into stable-0.11
[qemu-kvm/fedora.git] / savevm.c
blob181c0884940827ba2cc6be16c13de8aec0cad670
1 /*
2 * QEMU System Emulator
4 * Copyright (c) 2003-2008 Fabrice Bellard
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:
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
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.
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>
32 /* Needed early for HOST_BSD etc. */
33 #include "config-host.h"
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 #if defined(__NetBSD__)
46 #include <net/if_tap.h>
47 #endif
48 #ifdef __linux__
49 #include <linux/if_tun.h>
50 #endif
51 #include <arpa/inet.h>
52 #include <dirent.h>
53 #include <netdb.h>
54 #include <sys/select.h>
55 #ifdef HOST_BSD
56 #include <sys/stat.h>
57 #if defined(__FreeBSD__) || defined(__DragonFly__)
58 #include <libutil.h>
59 #else
60 #include <util.h>
61 #endif
62 #elif defined (__GLIBC__) && defined (__FreeBSD_kernel__)
63 #include <freebsd/stdlib.h>
64 #else
65 #ifdef __linux__
66 #include <pty.h>
67 #include <malloc.h>
68 #include <linux/rtc.h>
69 #endif
70 #endif
71 #endif
73 #ifdef _WIN32
74 #include <windows.h>
75 #include <malloc.h>
76 #include <sys/timeb.h>
77 #include <mmsystem.h>
78 #define getopt_long_only getopt_long
79 #define memalign(align, size) malloc(size)
80 #endif
82 #include "qemu-common.h"
83 #include "hw/hw.h"
84 #include "net.h"
85 #include "monitor.h"
86 #include "sysemu.h"
87 #include "qemu-timer.h"
88 #include "qemu-char.h"
89 #include "block.h"
90 #include "audio/audio.h"
91 #include "migration.h"
92 #include "qemu_socket.h"
94 /* point to the block driver where the snapshots are managed */
95 static BlockDriverState *bs_snapshots;
97 #define SELF_ANNOUNCE_ROUNDS 5
98 #define ETH_P_EXPERIMENTAL 0x01F1 /* just a number */
99 //#define ETH_P_EXPERIMENTAL 0x0012 /* make it the size of the packet */
100 #define EXPERIMENTAL_MAGIC 0xf1f23f4f
102 static int announce_self_create(uint8_t *buf,
103 uint8_t *mac_addr)
105 uint32_t magic = EXPERIMENTAL_MAGIC;
106 uint16_t proto = htons(ETH_P_EXPERIMENTAL);
108 /* FIXME: should we send a different packet (arp/rarp/ping)? */
110 memset(buf, 0, 64);
111 memset(buf, 0xff, 6); /* h_dst */
112 memcpy(buf + 6, mac_addr, 6); /* h_src */
113 memcpy(buf + 12, &proto, 2); /* h_proto */
114 memcpy(buf + 14, &magic, 4); /* magic */
116 return 64; /* len */
119 static void qemu_announce_self_once(void *opaque)
121 int i, len;
122 VLANState *vlan;
123 VLANClientState *vc;
124 uint8_t buf[256];
125 static int count = SELF_ANNOUNCE_ROUNDS;
126 QEMUTimer *timer = *(QEMUTimer **)opaque;
128 for (i = 0; i < MAX_NICS; i++) {
129 if (!nd_table[i].used)
130 continue;
131 len = announce_self_create(buf, nd_table[i].macaddr);
132 vlan = nd_table[i].vlan;
133 for(vc = vlan->first_client; vc != NULL; vc = vc->next) {
134 vc->receive(vc, buf, len);
137 if (count--) {
138 qemu_mod_timer(timer, qemu_get_clock(rt_clock) + 100);
139 } else {
140 qemu_del_timer(timer);
141 qemu_free_timer(timer);
145 void qemu_announce_self(void)
147 static QEMUTimer *timer;
148 timer = qemu_new_timer(rt_clock, qemu_announce_self_once, &timer);
149 qemu_announce_self_once(&timer);
152 /***********************************************************/
153 /* savevm/loadvm support */
155 #define IO_BUF_SIZE 32768
157 struct QEMUFile {
158 QEMUFilePutBufferFunc *put_buffer;
159 QEMUFileGetBufferFunc *get_buffer;
160 QEMUFileCloseFunc *close;
161 QEMUFileRateLimit *rate_limit;
162 QEMUFileSetRateLimit *set_rate_limit;
163 void *opaque;
164 int is_write;
166 int64_t buf_offset; /* start of buffer when writing, end of buffer
167 when reading */
168 int buf_index;
169 int buf_size; /* 0 when writing */
170 uint8_t buf[IO_BUF_SIZE];
172 int has_error;
175 typedef struct QEMUFilePopen
177 FILE *popen_file;
178 QEMUFile *file;
179 } QEMUFilePopen;
181 typedef struct QEMUFileSocket
183 int fd;
184 QEMUFile *file;
185 } QEMUFileSocket;
187 static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
189 QEMUFileSocket *s = opaque;
190 ssize_t len;
192 do {
193 len = recv(s->fd, (void *)buf, size, 0);
194 } while (len == -1 && socket_error() == EINTR);
196 if (len == -1)
197 len = -socket_error();
199 return len;
202 static int socket_close(void *opaque)
204 QEMUFileSocket *s = opaque;
205 qemu_free(s);
206 return 0;
209 static int popen_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
211 QEMUFilePopen *s = opaque;
212 return fwrite(buf, 1, size, s->popen_file);
215 static int popen_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
217 QEMUFilePopen *s = opaque;
218 FILE *fp = s->popen_file;
219 int bytes;
221 do {
222 clearerr(fp);
223 bytes = fread(buf, 1, size, fp);
224 } while ((bytes == 0) && ferror(fp) && (errno == EINTR));
225 return bytes;
228 static int popen_close(void *opaque)
230 QEMUFilePopen *s = opaque;
231 pclose(s->popen_file);
232 qemu_free(s);
233 return 0;
236 QEMUFile *qemu_popen(FILE *popen_file, const char *mode)
238 QEMUFilePopen *s;
240 if (popen_file == NULL || mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) {
241 fprintf(stderr, "qemu_popen: Argument validity check failed\n");
242 return NULL;
245 s = qemu_mallocz(sizeof(QEMUFilePopen));
247 s->popen_file = popen_file;
249 if(mode[0] == 'r') {
250 s->file = qemu_fopen_ops(s, NULL, popen_get_buffer, popen_close, NULL, NULL);
251 } else {
252 s->file = qemu_fopen_ops(s, popen_put_buffer, NULL, popen_close, NULL, NULL);
254 return s->file;
257 QEMUFile *qemu_popen_cmd(const char *command, const char *mode)
259 FILE *popen_file;
261 popen_file = popen(command, mode);
262 if(popen_file == NULL) {
263 return NULL;
266 return qemu_popen(popen_file, mode);
269 int qemu_popen_fd(QEMUFile *f)
271 QEMUFilePopen *p;
272 int fd;
274 p = (QEMUFilePopen *)f->opaque;
275 fd = fileno(p->popen_file);
277 return fd;
280 QEMUFile *qemu_fopen_socket(int fd)
282 QEMUFileSocket *s = qemu_mallocz(sizeof(QEMUFileSocket));
284 s->fd = fd;
285 s->file = qemu_fopen_ops(s, NULL, socket_get_buffer, socket_close, NULL, NULL);
286 return s->file;
289 typedef struct QEMUFileStdio
291 FILE *outfile;
292 } QEMUFileStdio;
294 static int file_put_buffer(void *opaque, const uint8_t *buf,
295 int64_t pos, int size)
297 QEMUFileStdio *s = opaque;
298 fseek(s->outfile, pos, SEEK_SET);
299 fwrite(buf, 1, size, s->outfile);
300 return size;
303 static int file_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
305 QEMUFileStdio *s = opaque;
306 fseek(s->outfile, pos, SEEK_SET);
307 return fread(buf, 1, size, s->outfile);
310 static int file_close(void *opaque)
312 QEMUFileStdio *s = opaque;
313 fclose(s->outfile);
314 qemu_free(s);
315 return 0;
318 QEMUFile *qemu_fopen(const char *filename, const char *mode)
320 QEMUFileStdio *s;
322 s = qemu_mallocz(sizeof(QEMUFileStdio));
324 s->outfile = fopen(filename, mode);
325 if (!s->outfile)
326 goto fail;
328 if (!strcmp(mode, "wb"))
329 return qemu_fopen_ops(s, file_put_buffer, NULL, file_close, NULL, NULL);
330 else if (!strcmp(mode, "rb"))
331 return qemu_fopen_ops(s, NULL, file_get_buffer, file_close, NULL, NULL);
333 fail:
334 if (s->outfile)
335 fclose(s->outfile);
336 qemu_free(s);
337 return NULL;
340 static int block_put_buffer(void *opaque, const uint8_t *buf,
341 int64_t pos, int size)
343 bdrv_save_vmstate(opaque, buf, pos, size);
344 return size;
347 static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
349 return bdrv_load_vmstate(opaque, buf, pos, size);
352 static int bdrv_fclose(void *opaque)
354 return 0;
357 static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int is_writable)
359 if (is_writable)
360 return qemu_fopen_ops(bs, block_put_buffer, NULL, bdrv_fclose, NULL, NULL);
361 return qemu_fopen_ops(bs, NULL, block_get_buffer, bdrv_fclose, NULL, NULL);
364 QEMUFile *qemu_fopen_ops(void *opaque, QEMUFilePutBufferFunc *put_buffer,
365 QEMUFileGetBufferFunc *get_buffer,
366 QEMUFileCloseFunc *close,
367 QEMUFileRateLimit *rate_limit,
368 QEMUFileSetRateLimit *set_rate_limit)
370 QEMUFile *f;
372 f = qemu_mallocz(sizeof(QEMUFile));
374 f->opaque = opaque;
375 f->put_buffer = put_buffer;
376 f->get_buffer = get_buffer;
377 f->close = close;
378 f->rate_limit = rate_limit;
379 f->set_rate_limit = set_rate_limit;
380 f->is_write = 0;
382 return f;
385 int qemu_file_has_error(QEMUFile *f)
387 return f->has_error;
390 void qemu_file_set_error(QEMUFile *f)
392 f->has_error = 1;
395 void qemu_fflush(QEMUFile *f)
397 if (!f->put_buffer)
398 return;
400 if (f->is_write && f->buf_index > 0) {
401 int len;
403 len = f->put_buffer(f->opaque, f->buf, f->buf_offset, f->buf_index);
404 if (len > 0)
405 f->buf_offset += f->buf_index;
406 else
407 f->has_error = 1;
408 f->buf_index = 0;
412 static void qemu_fill_buffer(QEMUFile *f)
414 int len;
416 if (!f->get_buffer)
417 return;
419 if (f->is_write)
420 abort();
422 len = f->get_buffer(f->opaque, f->buf, f->buf_offset, IO_BUF_SIZE);
423 if (len > 0) {
424 f->buf_index = 0;
425 f->buf_size = len;
426 f->buf_offset += len;
427 } else if (len != -EAGAIN)
428 f->has_error = 1;
431 int qemu_fclose(QEMUFile *f)
433 int ret = 0;
434 qemu_fflush(f);
435 if (f->close)
436 ret = f->close(f->opaque);
437 qemu_free(f);
438 return ret;
441 void qemu_file_put_notify(QEMUFile *f)
443 f->put_buffer(f->opaque, NULL, 0, 0);
446 void qemu_put_buffer(QEMUFile *f, const uint8_t *buf, int size)
448 int l;
450 if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
451 fprintf(stderr,
452 "Attempted to write to buffer while read buffer is not empty\n");
453 abort();
456 while (!f->has_error && size > 0) {
457 l = IO_BUF_SIZE - f->buf_index;
458 if (l > size)
459 l = size;
460 memcpy(f->buf + f->buf_index, buf, l);
461 f->is_write = 1;
462 f->buf_index += l;
463 buf += l;
464 size -= l;
465 if (f->buf_index >= IO_BUF_SIZE)
466 qemu_fflush(f);
470 void qemu_put_byte(QEMUFile *f, int v)
472 if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
473 fprintf(stderr,
474 "Attempted to write to buffer while read buffer is not empty\n");
475 abort();
478 f->buf[f->buf_index++] = v;
479 f->is_write = 1;
480 if (f->buf_index >= IO_BUF_SIZE)
481 qemu_fflush(f);
484 int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size1)
486 int size, l;
488 if (f->is_write)
489 abort();
491 size = size1;
492 while (size > 0) {
493 l = f->buf_size - f->buf_index;
494 if (l == 0) {
495 qemu_fill_buffer(f);
496 l = f->buf_size - f->buf_index;
497 if (l == 0)
498 break;
500 if (l > size)
501 l = size;
502 memcpy(buf, f->buf + f->buf_index, l);
503 f->buf_index += l;
504 buf += l;
505 size -= l;
507 return size1 - size;
510 int qemu_get_byte(QEMUFile *f)
512 if (f->is_write)
513 abort();
515 if (f->buf_index >= f->buf_size) {
516 qemu_fill_buffer(f);
517 if (f->buf_index >= f->buf_size)
518 return 0;
520 return f->buf[f->buf_index++];
523 int64_t qemu_ftell(QEMUFile *f)
525 return f->buf_offset - f->buf_size + f->buf_index;
528 int64_t qemu_fseek(QEMUFile *f, int64_t pos, int whence)
530 if (whence == SEEK_SET) {
531 /* nothing to do */
532 } else if (whence == SEEK_CUR) {
533 pos += qemu_ftell(f);
534 } else {
535 /* SEEK_END not supported */
536 return -1;
538 if (f->put_buffer) {
539 qemu_fflush(f);
540 f->buf_offset = pos;
541 } else {
542 f->buf_offset = pos;
543 f->buf_index = 0;
544 f->buf_size = 0;
546 return pos;
549 int qemu_file_rate_limit(QEMUFile *f)
551 if (f->rate_limit)
552 return f->rate_limit(f->opaque);
554 return 0;
557 size_t qemu_file_set_rate_limit(QEMUFile *f, size_t new_rate)
559 /* any failed or completed migration keeps its state to allow probing of
560 * migration data, but has no associated file anymore */
561 if (f && f->set_rate_limit)
562 return f->set_rate_limit(f->opaque, new_rate);
564 return 0;
567 void qemu_put_be16(QEMUFile *f, unsigned int v)
569 qemu_put_byte(f, v >> 8);
570 qemu_put_byte(f, v);
573 void qemu_put_be32(QEMUFile *f, unsigned int v)
575 qemu_put_byte(f, v >> 24);
576 qemu_put_byte(f, v >> 16);
577 qemu_put_byte(f, v >> 8);
578 qemu_put_byte(f, v);
581 void qemu_put_be64(QEMUFile *f, uint64_t v)
583 qemu_put_be32(f, v >> 32);
584 qemu_put_be32(f, v);
587 unsigned int qemu_get_be16(QEMUFile *f)
589 unsigned int v;
590 v = qemu_get_byte(f) << 8;
591 v |= qemu_get_byte(f);
592 return v;
595 unsigned int qemu_get_be32(QEMUFile *f)
597 unsigned int v;
598 v = qemu_get_byte(f) << 24;
599 v |= qemu_get_byte(f) << 16;
600 v |= qemu_get_byte(f) << 8;
601 v |= qemu_get_byte(f);
602 return v;
605 uint64_t qemu_get_be64(QEMUFile *f)
607 uint64_t v;
608 v = (uint64_t)qemu_get_be32(f) << 32;
609 v |= qemu_get_be32(f);
610 return v;
613 typedef struct SaveStateEntry {
614 char idstr[256];
615 int instance_id;
616 int version_id;
617 int section_id;
618 SaveLiveStateHandler *save_live_state;
619 SaveStateHandler *save_state;
620 LoadStateHandler *load_state;
621 void *opaque;
622 struct SaveStateEntry *next;
623 } SaveStateEntry;
625 static SaveStateEntry *first_se;
627 /* TODO: Individual devices generally have very little idea about the rest
628 of the system, so instance_id should be removed/replaced.
629 Meanwhile pass -1 as instance_id if you do not already have a clearly
630 distinguishing id for all instances of your device class. */
631 int register_savevm_live(const char *idstr,
632 int instance_id,
633 int version_id,
634 SaveLiveStateHandler *save_live_state,
635 SaveStateHandler *save_state,
636 LoadStateHandler *load_state,
637 void *opaque)
639 SaveStateEntry *se, **pse;
640 static int global_section_id;
642 se = qemu_malloc(sizeof(SaveStateEntry));
643 pstrcpy(se->idstr, sizeof(se->idstr), idstr);
644 se->instance_id = (instance_id == -1) ? 0 : instance_id;
645 se->version_id = version_id;
646 se->section_id = global_section_id++;
647 se->save_live_state = save_live_state;
648 se->save_state = save_state;
649 se->load_state = load_state;
650 se->opaque = opaque;
651 se->next = NULL;
653 /* add at the end of list */
654 pse = &first_se;
655 while (*pse != NULL) {
656 if (instance_id == -1
657 && strcmp(se->idstr, (*pse)->idstr) == 0
658 && se->instance_id <= (*pse)->instance_id)
659 se->instance_id = (*pse)->instance_id + 1;
660 pse = &(*pse)->next;
662 *pse = se;
663 return 0;
666 int register_savevm(const char *idstr,
667 int instance_id,
668 int version_id,
669 SaveStateHandler *save_state,
670 LoadStateHandler *load_state,
671 void *opaque)
673 return register_savevm_live(idstr, instance_id, version_id,
674 NULL, save_state, load_state, opaque);
677 void unregister_savevm(const char *idstr, void *opaque)
679 SaveStateEntry **pse;
681 pse = &first_se;
682 while (*pse != NULL) {
683 if (strcmp((*pse)->idstr, idstr) == 0 && (*pse)->opaque == opaque) {
684 SaveStateEntry *next = (*pse)->next;
685 qemu_free(*pse);
686 *pse = next;
687 continue;
689 pse = &(*pse)->next;
693 #define QEMU_VM_FILE_MAGIC 0x5145564d
694 #define QEMU_VM_FILE_VERSION_COMPAT 0x00000002
695 #define QEMU_VM_FILE_VERSION 0x00000003
697 #define QEMU_VM_EOF 0x00
698 #define QEMU_VM_SECTION_START 0x01
699 #define QEMU_VM_SECTION_PART 0x02
700 #define QEMU_VM_SECTION_END 0x03
701 #define QEMU_VM_SECTION_FULL 0x04
703 int qemu_savevm_state_begin(QEMUFile *f)
705 SaveStateEntry *se;
707 qemu_put_be32(f, QEMU_VM_FILE_MAGIC);
708 qemu_put_be32(f, QEMU_VM_FILE_VERSION);
710 for (se = first_se; se != NULL; se = se->next) {
711 int len;
713 if (se->save_live_state == NULL)
714 continue;
716 /* Section type */
717 qemu_put_byte(f, QEMU_VM_SECTION_START);
718 qemu_put_be32(f, se->section_id);
720 /* ID string */
721 len = strlen(se->idstr);
722 qemu_put_byte(f, len);
723 qemu_put_buffer(f, (uint8_t *)se->idstr, len);
725 qemu_put_be32(f, se->instance_id);
726 qemu_put_be32(f, se->version_id);
728 se->save_live_state(f, QEMU_VM_SECTION_START, se->opaque);
731 if (qemu_file_has_error(f))
732 return -EIO;
734 return 0;
737 int qemu_savevm_state_iterate(QEMUFile *f)
739 SaveStateEntry *se;
740 int ret = 1;
742 for (se = first_se; se != NULL; se = se->next) {
743 if (se->save_live_state == NULL)
744 continue;
746 /* Section type */
747 qemu_put_byte(f, QEMU_VM_SECTION_PART);
748 qemu_put_be32(f, se->section_id);
750 ret &= !!se->save_live_state(f, QEMU_VM_SECTION_PART, se->opaque);
753 if (ret)
754 return 1;
756 if (qemu_file_has_error(f))
757 return -EIO;
759 return 0;
762 int qemu_savevm_state_complete(QEMUFile *f)
764 SaveStateEntry *se;
766 for (se = first_se; se != NULL; se = se->next) {
767 if (se->save_live_state == NULL)
768 continue;
770 /* Section type */
771 qemu_put_byte(f, QEMU_VM_SECTION_END);
772 qemu_put_be32(f, se->section_id);
774 se->save_live_state(f, QEMU_VM_SECTION_END, se->opaque);
777 for(se = first_se; se != NULL; se = se->next) {
778 int len;
780 if (se->save_state == NULL)
781 continue;
783 /* Section type */
784 qemu_put_byte(f, QEMU_VM_SECTION_FULL);
785 qemu_put_be32(f, se->section_id);
787 /* ID string */
788 len = strlen(se->idstr);
789 qemu_put_byte(f, len);
790 qemu_put_buffer(f, (uint8_t *)se->idstr, len);
792 qemu_put_be32(f, se->instance_id);
793 qemu_put_be32(f, se->version_id);
795 se->save_state(f, se->opaque);
798 qemu_put_byte(f, QEMU_VM_EOF);
800 if (qemu_file_has_error(f))
801 return -EIO;
803 return 0;
806 int qemu_savevm_state(QEMUFile *f)
808 int saved_vm_running;
809 int ret;
811 saved_vm_running = vm_running;
812 vm_stop(0);
814 bdrv_flush_all();
816 ret = qemu_savevm_state_begin(f);
817 if (ret < 0)
818 goto out;
820 do {
821 ret = qemu_savevm_state_iterate(f);
822 if (ret < 0)
823 goto out;
824 } while (ret == 0);
826 ret = qemu_savevm_state_complete(f);
828 out:
829 if (qemu_file_has_error(f))
830 ret = -EIO;
832 if (!ret && saved_vm_running)
833 vm_start();
835 return ret;
838 static SaveStateEntry *find_se(const char *idstr, int instance_id)
840 SaveStateEntry *se;
842 for(se = first_se; se != NULL; se = se->next) {
843 if (!strcmp(se->idstr, idstr) &&
844 instance_id == se->instance_id)
845 return se;
847 return NULL;
850 typedef struct LoadStateEntry {
851 SaveStateEntry *se;
852 int section_id;
853 int version_id;
854 struct LoadStateEntry *next;
855 } LoadStateEntry;
857 static int qemu_loadvm_state_v2(QEMUFile *f)
859 SaveStateEntry *se;
860 int len, ret, instance_id, record_len, version_id;
861 int64_t total_len, end_pos, cur_pos;
862 char idstr[256];
864 total_len = qemu_get_be64(f);
865 end_pos = total_len + qemu_ftell(f);
866 for(;;) {
867 if (qemu_ftell(f) >= end_pos)
868 break;
869 len = qemu_get_byte(f);
870 qemu_get_buffer(f, (uint8_t *)idstr, len);
871 idstr[len] = '\0';
872 instance_id = qemu_get_be32(f);
873 version_id = qemu_get_be32(f);
874 record_len = qemu_get_be32(f);
875 cur_pos = qemu_ftell(f);
876 se = find_se(idstr, instance_id);
877 if (!se) {
878 fprintf(stderr, "qemu: warning: instance 0x%x of device '%s' not present in current VM\n",
879 instance_id, idstr);
880 } else {
881 ret = se->load_state(f, se->opaque, version_id);
882 if (ret < 0) {
883 fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
884 instance_id, idstr);
885 return ret;
888 /* always seek to exact end of record */
889 qemu_fseek(f, cur_pos + record_len, SEEK_SET);
892 if (qemu_file_has_error(f))
893 return -EIO;
895 return 0;
898 int qemu_loadvm_state(QEMUFile *f)
900 LoadStateEntry *first_le = NULL;
901 uint8_t section_type;
902 unsigned int v;
903 int ret;
905 v = qemu_get_be32(f);
906 if (v != QEMU_VM_FILE_MAGIC)
907 return -EINVAL;
909 v = qemu_get_be32(f);
910 if (v == QEMU_VM_FILE_VERSION_COMPAT)
911 return qemu_loadvm_state_v2(f);
912 if (v != QEMU_VM_FILE_VERSION)
913 return -ENOTSUP;
915 while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
916 uint32_t instance_id, version_id, section_id;
917 LoadStateEntry *le;
918 SaveStateEntry *se;
919 char idstr[257];
920 int len;
922 switch (section_type) {
923 case QEMU_VM_SECTION_START:
924 case QEMU_VM_SECTION_FULL:
925 /* Read section start */
926 section_id = qemu_get_be32(f);
927 len = qemu_get_byte(f);
928 qemu_get_buffer(f, (uint8_t *)idstr, len);
929 idstr[len] = 0;
930 instance_id = qemu_get_be32(f);
931 version_id = qemu_get_be32(f);
933 /* Find savevm section */
934 se = find_se(idstr, instance_id);
935 if (se == NULL) {
936 fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
937 ret = -EINVAL;
938 goto out;
941 /* Validate version */
942 if (version_id > se->version_id) {
943 fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
944 version_id, idstr, se->version_id);
945 ret = -EINVAL;
946 goto out;
949 /* Add entry */
950 le = qemu_mallocz(sizeof(*le));
952 le->se = se;
953 le->section_id = section_id;
954 le->version_id = version_id;
955 le->next = first_le;
956 first_le = le;
958 le->se->load_state(f, le->se->opaque, le->version_id);
959 break;
960 case QEMU_VM_SECTION_PART:
961 case QEMU_VM_SECTION_END:
962 section_id = qemu_get_be32(f);
964 for (le = first_le; le && le->section_id != section_id; le = le->next);
965 if (le == NULL) {
966 fprintf(stderr, "Unknown savevm section %d\n", section_id);
967 ret = -EINVAL;
968 goto out;
971 le->se->load_state(f, le->se->opaque, le->version_id);
972 break;
973 default:
974 fprintf(stderr, "Unknown savevm section type %d\n", section_type);
975 ret = -EINVAL;
976 goto out;
980 ret = 0;
982 out:
983 while (first_le) {
984 LoadStateEntry *le = first_le;
985 first_le = first_le->next;
986 qemu_free(le);
989 if (qemu_file_has_error(f))
990 ret = -EIO;
992 return ret;
995 /* device can contain snapshots */
996 static int bdrv_can_snapshot(BlockDriverState *bs)
998 return (bs &&
999 !bdrv_is_removable(bs) &&
1000 !bdrv_is_read_only(bs));
1003 /* device must be snapshots in order to have a reliable snapshot */
1004 static int bdrv_has_snapshot(BlockDriverState *bs)
1006 return (bs &&
1007 !bdrv_is_removable(bs) &&
1008 !bdrv_is_read_only(bs));
1011 static BlockDriverState *get_bs_snapshots(void)
1013 BlockDriverState *bs;
1014 int i;
1016 if (bs_snapshots)
1017 return bs_snapshots;
1018 for(i = 0; i <= nb_drives; i++) {
1019 bs = drives_table[i].bdrv;
1020 if (bdrv_can_snapshot(bs))
1021 goto ok;
1023 return NULL;
1025 bs_snapshots = bs;
1026 return bs;
1029 static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
1030 const char *name)
1032 QEMUSnapshotInfo *sn_tab, *sn;
1033 int nb_sns, i, ret;
1035 ret = -ENOENT;
1036 nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1037 if (nb_sns < 0)
1038 return ret;
1039 for(i = 0; i < nb_sns; i++) {
1040 sn = &sn_tab[i];
1041 if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
1042 *sn_info = *sn;
1043 ret = 0;
1044 break;
1047 qemu_free(sn_tab);
1048 return ret;
1051 void do_savevm(Monitor *mon, const char *name)
1053 BlockDriverState *bs, *bs1;
1054 QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1;
1055 int must_delete, ret, i;
1056 QEMUFile *f;
1057 int saved_vm_running;
1058 uint32_t vm_state_size;
1059 #ifdef _WIN32
1060 struct _timeb tb;
1061 #else
1062 struct timeval tv;
1063 #endif
1065 bs = get_bs_snapshots();
1066 if (!bs) {
1067 monitor_printf(mon, "No block device can accept snapshots\n");
1068 return;
1071 /* ??? Should this occur after vm_stop? */
1072 qemu_aio_flush();
1074 saved_vm_running = vm_running;
1075 vm_stop(0);
1077 must_delete = 0;
1078 if (name) {
1079 ret = bdrv_snapshot_find(bs, old_sn, name);
1080 if (ret >= 0) {
1081 must_delete = 1;
1084 memset(sn, 0, sizeof(*sn));
1085 if (must_delete) {
1086 pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
1087 pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
1088 } else {
1089 if (name)
1090 pstrcpy(sn->name, sizeof(sn->name), name);
1093 /* fill auxiliary fields */
1094 #ifdef _WIN32
1095 _ftime(&tb);
1096 sn->date_sec = tb.time;
1097 sn->date_nsec = tb.millitm * 1000000;
1098 #else
1099 gettimeofday(&tv, NULL);
1100 sn->date_sec = tv.tv_sec;
1101 sn->date_nsec = tv.tv_usec * 1000;
1102 #endif
1103 sn->vm_clock_nsec = qemu_get_clock(vm_clock);
1105 /* save the VM state */
1106 f = qemu_fopen_bdrv(bs, 1);
1107 if (!f) {
1108 monitor_printf(mon, "Could not open VM state file\n");
1109 goto the_end;
1111 ret = qemu_savevm_state(f);
1112 vm_state_size = qemu_ftell(f);
1113 qemu_fclose(f);
1114 if (ret < 0) {
1115 monitor_printf(mon, "Error %d while writing VM\n", ret);
1116 goto the_end;
1119 /* create the snapshots */
1121 for(i = 0; i < nb_drives; i++) {
1122 bs1 = drives_table[i].bdrv;
1123 if (bdrv_has_snapshot(bs1)) {
1124 if (must_delete) {
1125 ret = bdrv_snapshot_delete(bs1, old_sn->id_str);
1126 if (ret < 0) {
1127 monitor_printf(mon,
1128 "Error while deleting snapshot on '%s'\n",
1129 bdrv_get_device_name(bs1));
1132 /* Write VM state size only to the image that contains the state */
1133 sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
1134 ret = bdrv_snapshot_create(bs1, sn);
1135 if (ret < 0) {
1136 monitor_printf(mon, "Error while creating snapshot on '%s'\n",
1137 bdrv_get_device_name(bs1));
1142 the_end:
1143 if (saved_vm_running)
1144 vm_start();
1147 void do_loadvm(Monitor *mon, const char *name)
1149 BlockDriverState *bs, *bs1;
1150 QEMUSnapshotInfo sn;
1151 QEMUFile *f;
1152 int i, ret;
1153 int saved_vm_running;
1155 bs = get_bs_snapshots();
1156 if (!bs) {
1157 monitor_printf(mon, "No block device supports snapshots\n");
1158 return;
1161 /* Flush all IO requests so they don't interfere with the new state. */
1162 qemu_aio_flush();
1164 saved_vm_running = vm_running;
1165 vm_stop(0);
1167 for(i = 0; i <= nb_drives; i++) {
1168 bs1 = drives_table[i].bdrv;
1169 if (bdrv_has_snapshot(bs1)) {
1170 ret = bdrv_snapshot_goto(bs1, name);
1171 if (ret < 0) {
1172 if (bs != bs1)
1173 monitor_printf(mon, "Warning: ");
1174 switch(ret) {
1175 case -ENOTSUP:
1176 monitor_printf(mon,
1177 "Snapshots not supported on device '%s'\n",
1178 bdrv_get_device_name(bs1));
1179 break;
1180 case -ENOENT:
1181 monitor_printf(mon, "Could not find snapshot '%s' on "
1182 "device '%s'\n",
1183 name, bdrv_get_device_name(bs1));
1184 break;
1185 default:
1186 monitor_printf(mon, "Error %d while activating snapshot on"
1187 " '%s'\n", ret, bdrv_get_device_name(bs1));
1188 break;
1190 /* fatal on snapshot block device */
1191 if (bs == bs1)
1192 goto the_end;
1197 /* Don't even try to load empty VM states */
1198 ret = bdrv_snapshot_find(bs, &sn, name);
1199 if ((ret >= 0) && (sn.vm_state_size == 0))
1200 goto the_end;
1202 /* restore the VM state */
1203 f = qemu_fopen_bdrv(bs, 0);
1204 if (!f) {
1205 monitor_printf(mon, "Could not open VM state file\n");
1206 goto the_end;
1208 ret = qemu_loadvm_state(f);
1209 qemu_fclose(f);
1210 if (ret < 0) {
1211 monitor_printf(mon, "Error %d while loading VM state\n", ret);
1213 the_end:
1214 if (saved_vm_running)
1215 vm_start();
1218 void do_delvm(Monitor *mon, const char *name)
1220 BlockDriverState *bs, *bs1;
1221 int i, ret;
1223 bs = get_bs_snapshots();
1224 if (!bs) {
1225 monitor_printf(mon, "No block device supports snapshots\n");
1226 return;
1229 for(i = 0; i <= nb_drives; i++) {
1230 bs1 = drives_table[i].bdrv;
1231 if (bdrv_has_snapshot(bs1)) {
1232 ret = bdrv_snapshot_delete(bs1, name);
1233 if (ret < 0) {
1234 if (ret == -ENOTSUP)
1235 monitor_printf(mon,
1236 "Snapshots not supported on device '%s'\n",
1237 bdrv_get_device_name(bs1));
1238 else
1239 monitor_printf(mon, "Error %d while deleting snapshot on "
1240 "'%s'\n", ret, bdrv_get_device_name(bs1));
1246 void do_info_snapshots(Monitor *mon)
1248 BlockDriverState *bs, *bs1;
1249 QEMUSnapshotInfo *sn_tab, *sn;
1250 int nb_sns, i;
1251 char buf[256];
1253 bs = get_bs_snapshots();
1254 if (!bs) {
1255 monitor_printf(mon, "No available block device supports snapshots\n");
1256 return;
1258 monitor_printf(mon, "Snapshot devices:");
1259 for(i = 0; i <= nb_drives; i++) {
1260 bs1 = drives_table[i].bdrv;
1261 if (bdrv_has_snapshot(bs1)) {
1262 if (bs == bs1)
1263 monitor_printf(mon, " %s", bdrv_get_device_name(bs1));
1266 monitor_printf(mon, "\n");
1268 nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1269 if (nb_sns < 0) {
1270 monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
1271 return;
1273 monitor_printf(mon, "Snapshot list (from %s):\n",
1274 bdrv_get_device_name(bs));
1275 monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
1276 for(i = 0; i < nb_sns; i++) {
1277 sn = &sn_tab[i];
1278 monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
1280 qemu_free(sn_tab);