Statistics
| Branch: | Revision:

root / savevm.c @ 8595387e

History | View | Annotate | Download (44.4 kB)

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