Statistics
| Branch: | Revision:

root / savevm.c @ f7897430

History | View | Annotate | Download (31.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 179a2c19 blueswir1
/* Needed early for HOST_BSD etc. */
33 d40cdb10 blueswir1
#include "config-host.h"
34 d40cdb10 blueswir1
35 a672b469 aliguori
#ifndef _WIN32
36 a672b469 aliguori
#include <sys/times.h>
37 a672b469 aliguori
#include <sys/wait.h>
38 a672b469 aliguori
#include <termios.h>
39 a672b469 aliguori
#include <sys/mman.h>
40 a672b469 aliguori
#include <sys/ioctl.h>
41 a672b469 aliguori
#include <sys/resource.h>
42 a672b469 aliguori
#include <sys/socket.h>
43 a672b469 aliguori
#include <netinet/in.h>
44 a672b469 aliguori
#include <net/if.h>
45 a672b469 aliguori
#if defined(__NetBSD__)
46 a672b469 aliguori
#include <net/if_tap.h>
47 a672b469 aliguori
#endif
48 a672b469 aliguori
#ifdef __linux__
49 a672b469 aliguori
#include <linux/if_tun.h>
50 a672b469 aliguori
#endif
51 a672b469 aliguori
#include <arpa/inet.h>
52 a672b469 aliguori
#include <dirent.h>
53 a672b469 aliguori
#include <netdb.h>
54 a672b469 aliguori
#include <sys/select.h>
55 179a2c19 blueswir1
#ifdef HOST_BSD
56 a672b469 aliguori
#include <sys/stat.h>
57 c5e97233 blueswir1
#if defined(__FreeBSD__) || defined(__DragonFly__)
58 a672b469 aliguori
#include <libutil.h>
59 a672b469 aliguori
#else
60 a672b469 aliguori
#include <util.h>
61 a672b469 aliguori
#endif
62 a672b469 aliguori
#elif defined (__GLIBC__) && defined (__FreeBSD_kernel__)
63 a672b469 aliguori
#include <freebsd/stdlib.h>
64 a672b469 aliguori
#else
65 a672b469 aliguori
#ifdef __linux__
66 a672b469 aliguori
#include <pty.h>
67 a672b469 aliguori
#include <malloc.h>
68 a672b469 aliguori
#include <linux/rtc.h>
69 a672b469 aliguori
#endif
70 a672b469 aliguori
#endif
71 a672b469 aliguori
#endif
72 a672b469 aliguori
73 a672b469 aliguori
#ifdef _WIN32
74 49dc768d aliguori
#include <windows.h>
75 a672b469 aliguori
#include <malloc.h>
76 a672b469 aliguori
#include <sys/timeb.h>
77 a672b469 aliguori
#include <mmsystem.h>
78 a672b469 aliguori
#define getopt_long_only getopt_long
79 a672b469 aliguori
#define memalign(align, size) malloc(size)
80 a672b469 aliguori
#endif
81 a672b469 aliguori
82 511d2b14 blueswir1
#include "qemu-common.h"
83 511d2b14 blueswir1
#include "hw/hw.h"
84 511d2b14 blueswir1
#include "net.h"
85 511d2b14 blueswir1
#include "monitor.h"
86 511d2b14 blueswir1
#include "sysemu.h"
87 511d2b14 blueswir1
#include "qemu-timer.h"
88 511d2b14 blueswir1
#include "qemu-char.h"
89 511d2b14 blueswir1
#include "block.h"
90 511d2b14 blueswir1
#include "audio/audio.h"
91 511d2b14 blueswir1
#include "migration.h"
92 511d2b14 blueswir1
#include "qemu_socket.h"
93 511d2b14 blueswir1
94 a672b469 aliguori
/* point to the block driver where the snapshots are managed */
95 a672b469 aliguori
static BlockDriverState *bs_snapshots;
96 a672b469 aliguori
97 a672b469 aliguori
#define SELF_ANNOUNCE_ROUNDS 5
98 a672b469 aliguori
#define ETH_P_EXPERIMENTAL 0x01F1 /* just a number */
99 a672b469 aliguori
//#define ETH_P_EXPERIMENTAL 0x0012 /* make it the size of the packet */
100 a672b469 aliguori
#define EXPERIMENTAL_MAGIC 0xf1f23f4f
101 a672b469 aliguori
102 a672b469 aliguori
static int announce_self_create(uint8_t *buf, 
103 a672b469 aliguori
                                uint8_t *mac_addr)
104 a672b469 aliguori
{
105 a672b469 aliguori
    uint32_t magic = EXPERIMENTAL_MAGIC;
106 a672b469 aliguori
    uint16_t proto = htons(ETH_P_EXPERIMENTAL);
107 a672b469 aliguori
108 a672b469 aliguori
    /* FIXME: should we send a different packet (arp/rarp/ping)? */
109 a672b469 aliguori
110 a672b469 aliguori
    memset(buf, 0xff, 6);         /* h_dst */
111 a672b469 aliguori
    memcpy(buf + 6, mac_addr, 6); /* h_src */
112 a672b469 aliguori
    memcpy(buf + 12, &proto, 2);  /* h_proto */
113 a672b469 aliguori
    memcpy(buf + 14, &magic, 4);  /* magic */
114 a672b469 aliguori
115 a672b469 aliguori
    return 18; /* len */
116 a672b469 aliguori
}
117 a672b469 aliguori
118 a672b469 aliguori
void qemu_announce_self(void)
119 a672b469 aliguori
{
120 a672b469 aliguori
    int i, j, len;
121 a672b469 aliguori
    VLANState *vlan;
122 a672b469 aliguori
    VLANClientState *vc;
123 a672b469 aliguori
    uint8_t buf[256];
124 a672b469 aliguori
125 3450df30 aliguori
    for (i = 0; i < MAX_NICS; i++) {
126 3450df30 aliguori
        if (!nd_table[i].used)
127 3450df30 aliguori
            continue;
128 a672b469 aliguori
        len = announce_self_create(buf, nd_table[i].macaddr);
129 a672b469 aliguori
        vlan = nd_table[i].vlan;
130 a672b469 aliguori
        for(vc = vlan->first_client; vc != NULL; vc = vc->next) {
131 a672b469 aliguori
            for (j=0; j < SELF_ANNOUNCE_ROUNDS; j++)
132 a672b469 aliguori
                vc->fd_read(vc->opaque, buf, len);
133 a672b469 aliguori
        }
134 a672b469 aliguori
    }
135 a672b469 aliguori
}
136 a672b469 aliguori
137 a672b469 aliguori
/***********************************************************/
138 a672b469 aliguori
/* savevm/loadvm support */
139 a672b469 aliguori
140 a672b469 aliguori
#define IO_BUF_SIZE 32768
141 a672b469 aliguori
142 a672b469 aliguori
struct QEMUFile {
143 a672b469 aliguori
    QEMUFilePutBufferFunc *put_buffer;
144 a672b469 aliguori
    QEMUFileGetBufferFunc *get_buffer;
145 a672b469 aliguori
    QEMUFileCloseFunc *close;
146 a672b469 aliguori
    QEMUFileRateLimit *rate_limit;
147 a672b469 aliguori
    void *opaque;
148 a672b469 aliguori
    int is_write;
149 a672b469 aliguori
150 a672b469 aliguori
    int64_t buf_offset; /* start of buffer when writing, end of buffer
151 a672b469 aliguori
                           when reading */
152 a672b469 aliguori
    int buf_index;
153 a672b469 aliguori
    int buf_size; /* 0 when writing */
154 a672b469 aliguori
    uint8_t buf[IO_BUF_SIZE];
155 a672b469 aliguori
156 a672b469 aliguori
    int has_error;
157 a672b469 aliguori
};
158 a672b469 aliguori
159 a672b469 aliguori
typedef struct QEMUFilePopen
160 a672b469 aliguori
{
161 a672b469 aliguori
    FILE *popen_file;
162 a672b469 aliguori
    QEMUFile *file;
163 a672b469 aliguori
} QEMUFilePopen;
164 a672b469 aliguori
165 a672b469 aliguori
typedef struct QEMUFileSocket
166 a672b469 aliguori
{
167 a672b469 aliguori
    int fd;
168 a672b469 aliguori
    QEMUFile *file;
169 a672b469 aliguori
} QEMUFileSocket;
170 a672b469 aliguori
171 a672b469 aliguori
static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
172 a672b469 aliguori
{
173 a672b469 aliguori
    QEMUFileSocket *s = opaque;
174 a672b469 aliguori
    ssize_t len;
175 a672b469 aliguori
176 a672b469 aliguori
    do {
177 a672b469 aliguori
        len = recv(s->fd, buf, size, 0);
178 a672b469 aliguori
    } while (len == -1 && socket_error() == EINTR);
179 a672b469 aliguori
180 a672b469 aliguori
    if (len == -1)
181 a672b469 aliguori
        len = -socket_error();
182 a672b469 aliguori
183 a672b469 aliguori
    return len;
184 a672b469 aliguori
}
185 a672b469 aliguori
186 a672b469 aliguori
static int socket_close(void *opaque)
187 a672b469 aliguori
{
188 a672b469 aliguori
    QEMUFileSocket *s = opaque;
189 a672b469 aliguori
    qemu_free(s);
190 a672b469 aliguori
    return 0;
191 a672b469 aliguori
}
192 a672b469 aliguori
193 a672b469 aliguori
static int popen_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
194 a672b469 aliguori
{
195 a672b469 aliguori
    QEMUFilePopen *s = opaque;
196 a672b469 aliguori
    return fwrite(buf, 1, size, s->popen_file);
197 a672b469 aliguori
}
198 a672b469 aliguori
199 a672b469 aliguori
static int popen_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
200 a672b469 aliguori
{
201 a672b469 aliguori
    QEMUFilePopen *s = opaque;
202 a672b469 aliguori
    return fread(buf, 1, size, s->popen_file);
203 a672b469 aliguori
}
204 a672b469 aliguori
205 a672b469 aliguori
static int popen_close(void *opaque)
206 a672b469 aliguori
{
207 a672b469 aliguori
    QEMUFilePopen *s = opaque;
208 a672b469 aliguori
    pclose(s->popen_file);
209 a672b469 aliguori
    qemu_free(s);
210 a672b469 aliguori
    return 0;
211 a672b469 aliguori
}
212 a672b469 aliguori
213 a672b469 aliguori
QEMUFile *qemu_popen(FILE *popen_file, const char *mode)
214 a672b469 aliguori
{
215 a672b469 aliguori
    QEMUFilePopen *s;
216 a672b469 aliguori
217 a672b469 aliguori
    if (popen_file == NULL || mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) {
218 a672b469 aliguori
        fprintf(stderr, "qemu_popen: Argument validity check failed\n");
219 a672b469 aliguori
        return NULL;
220 a672b469 aliguori
    }
221 a672b469 aliguori
222 a672b469 aliguori
    s = qemu_mallocz(sizeof(QEMUFilePopen));
223 a672b469 aliguori
224 a672b469 aliguori
    s->popen_file = popen_file;
225 a672b469 aliguori
226 a672b469 aliguori
    if(mode[0] == 'r') {
227 a672b469 aliguori
        s->file = qemu_fopen_ops(s, NULL, popen_get_buffer, popen_close, NULL);
228 a672b469 aliguori
    } else {
229 a672b469 aliguori
        s->file = qemu_fopen_ops(s, popen_put_buffer, NULL, popen_close, NULL);
230 a672b469 aliguori
    }
231 a672b469 aliguori
    fprintf(stderr, "qemu_popen: returning result of qemu_fopen_ops\n");
232 a672b469 aliguori
    return s->file;
233 a672b469 aliguori
}
234 a672b469 aliguori
235 a672b469 aliguori
QEMUFile *qemu_popen_cmd(const char *command, const char *mode)
236 a672b469 aliguori
{
237 a672b469 aliguori
    FILE *popen_file;
238 a672b469 aliguori
239 a672b469 aliguori
    popen_file = popen(command, mode);
240 a672b469 aliguori
    if(popen_file == NULL) {
241 a672b469 aliguori
        return NULL;
242 a672b469 aliguori
    }
243 a672b469 aliguori
244 a672b469 aliguori
    return qemu_popen(popen_file, mode);
245 a672b469 aliguori
}
246 a672b469 aliguori
247 a672b469 aliguori
QEMUFile *qemu_fopen_socket(int fd)
248 a672b469 aliguori
{
249 a672b469 aliguori
    QEMUFileSocket *s = qemu_mallocz(sizeof(QEMUFileSocket));
250 a672b469 aliguori
251 a672b469 aliguori
    s->fd = fd;
252 a672b469 aliguori
    s->file = qemu_fopen_ops(s, NULL, socket_get_buffer, socket_close, NULL);
253 a672b469 aliguori
    return s->file;
254 a672b469 aliguori
}
255 a672b469 aliguori
256 a672b469 aliguori
typedef struct QEMUFileStdio
257 a672b469 aliguori
{
258 a672b469 aliguori
    FILE *outfile;
259 a672b469 aliguori
} QEMUFileStdio;
260 a672b469 aliguori
261 a672b469 aliguori
static int file_put_buffer(void *opaque, const uint8_t *buf,
262 a672b469 aliguori
                            int64_t pos, int size)
263 a672b469 aliguori
{
264 a672b469 aliguori
    QEMUFileStdio *s = opaque;
265 a672b469 aliguori
    fseek(s->outfile, pos, SEEK_SET);
266 a672b469 aliguori
    fwrite(buf, 1, size, s->outfile);
267 a672b469 aliguori
    return size;
268 a672b469 aliguori
}
269 a672b469 aliguori
270 a672b469 aliguori
static int file_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
271 a672b469 aliguori
{
272 a672b469 aliguori
    QEMUFileStdio *s = opaque;
273 a672b469 aliguori
    fseek(s->outfile, pos, SEEK_SET);
274 a672b469 aliguori
    return fread(buf, 1, size, s->outfile);
275 a672b469 aliguori
}
276 a672b469 aliguori
277 a672b469 aliguori
static int file_close(void *opaque)
278 a672b469 aliguori
{
279 a672b469 aliguori
    QEMUFileStdio *s = opaque;
280 a672b469 aliguori
    fclose(s->outfile);
281 a672b469 aliguori
    qemu_free(s);
282 a672b469 aliguori
    return 0;
283 a672b469 aliguori
}
284 a672b469 aliguori
285 a672b469 aliguori
QEMUFile *qemu_fopen(const char *filename, const char *mode)
286 a672b469 aliguori
{
287 a672b469 aliguori
    QEMUFileStdio *s;
288 a672b469 aliguori
289 a672b469 aliguori
    s = qemu_mallocz(sizeof(QEMUFileStdio));
290 a672b469 aliguori
291 a672b469 aliguori
    s->outfile = fopen(filename, mode);
292 a672b469 aliguori
    if (!s->outfile)
293 a672b469 aliguori
        goto fail;
294 a672b469 aliguori
295 a672b469 aliguori
    if (!strcmp(mode, "wb"))
296 a672b469 aliguori
        return qemu_fopen_ops(s, file_put_buffer, NULL, file_close, NULL);
297 a672b469 aliguori
    else if (!strcmp(mode, "rb"))
298 a672b469 aliguori
        return qemu_fopen_ops(s, NULL, file_get_buffer, file_close, NULL);
299 a672b469 aliguori
300 a672b469 aliguori
fail:
301 a672b469 aliguori
    if (s->outfile)
302 a672b469 aliguori
        fclose(s->outfile);
303 a672b469 aliguori
    qemu_free(s);
304 a672b469 aliguori
    return NULL;
305 a672b469 aliguori
}
306 a672b469 aliguori
307 a672b469 aliguori
typedef struct QEMUFileBdrv
308 a672b469 aliguori
{
309 a672b469 aliguori
    BlockDriverState *bs;
310 a672b469 aliguori
    int64_t base_offset;
311 a672b469 aliguori
} QEMUFileBdrv;
312 a672b469 aliguori
313 178e08a5 aliguori
static int block_put_buffer(void *opaque, const uint8_t *buf,
314 a672b469 aliguori
                           int64_t pos, int size)
315 a672b469 aliguori
{
316 a672b469 aliguori
    QEMUFileBdrv *s = opaque;
317 178e08a5 aliguori
    bdrv_put_buffer(s->bs, buf, s->base_offset + pos, size);
318 a672b469 aliguori
    return size;
319 a672b469 aliguori
}
320 a672b469 aliguori
321 178e08a5 aliguori
static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
322 a672b469 aliguori
{
323 a672b469 aliguori
    QEMUFileBdrv *s = opaque;
324 178e08a5 aliguori
    return bdrv_get_buffer(s->bs, buf, s->base_offset + pos, size);
325 a672b469 aliguori
}
326 a672b469 aliguori
327 a672b469 aliguori
static int bdrv_fclose(void *opaque)
328 a672b469 aliguori
{
329 a672b469 aliguori
    QEMUFileBdrv *s = opaque;
330 a672b469 aliguori
    qemu_free(s);
331 a672b469 aliguori
    return 0;
332 a672b469 aliguori
}
333 a672b469 aliguori
334 a672b469 aliguori
static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int64_t offset, int is_writable)
335 a672b469 aliguori
{
336 a672b469 aliguori
    QEMUFileBdrv *s;
337 a672b469 aliguori
338 a672b469 aliguori
    s = qemu_mallocz(sizeof(QEMUFileBdrv));
339 a672b469 aliguori
340 a672b469 aliguori
    s->bs = bs;
341 a672b469 aliguori
    s->base_offset = offset;
342 a672b469 aliguori
343 a672b469 aliguori
    if (is_writable)
344 178e08a5 aliguori
        return qemu_fopen_ops(s, block_put_buffer, NULL, bdrv_fclose, NULL);
345 a672b469 aliguori
346 178e08a5 aliguori
    return qemu_fopen_ops(s, NULL, block_get_buffer, bdrv_fclose, NULL);
347 a672b469 aliguori
}
348 a672b469 aliguori
349 a672b469 aliguori
QEMUFile *qemu_fopen_ops(void *opaque, QEMUFilePutBufferFunc *put_buffer,
350 a672b469 aliguori
                         QEMUFileGetBufferFunc *get_buffer,
351 a672b469 aliguori
                         QEMUFileCloseFunc *close,
352 a672b469 aliguori
                         QEMUFileRateLimit *rate_limit)
353 a672b469 aliguori
{
354 a672b469 aliguori
    QEMUFile *f;
355 a672b469 aliguori
356 a672b469 aliguori
    f = qemu_mallocz(sizeof(QEMUFile));
357 a672b469 aliguori
358 a672b469 aliguori
    f->opaque = opaque;
359 a672b469 aliguori
    f->put_buffer = put_buffer;
360 a672b469 aliguori
    f->get_buffer = get_buffer;
361 a672b469 aliguori
    f->close = close;
362 a672b469 aliguori
    f->rate_limit = rate_limit;
363 a672b469 aliguori
    f->is_write = 0;
364 a672b469 aliguori
365 a672b469 aliguori
    return f;
366 a672b469 aliguori
}
367 a672b469 aliguori
368 a672b469 aliguori
int qemu_file_has_error(QEMUFile *f)
369 a672b469 aliguori
{
370 a672b469 aliguori
    return f->has_error;
371 a672b469 aliguori
}
372 a672b469 aliguori
373 4dabe248 aliguori
void qemu_file_set_error(QEMUFile *f)
374 4dabe248 aliguori
{
375 4dabe248 aliguori
    f->has_error = 1;
376 4dabe248 aliguori
}
377 4dabe248 aliguori
378 a672b469 aliguori
void qemu_fflush(QEMUFile *f)
379 a672b469 aliguori
{
380 a672b469 aliguori
    if (!f->put_buffer)
381 a672b469 aliguori
        return;
382 a672b469 aliguori
383 a672b469 aliguori
    if (f->is_write && f->buf_index > 0) {
384 a672b469 aliguori
        int len;
385 a672b469 aliguori
386 a672b469 aliguori
        len = f->put_buffer(f->opaque, f->buf, f->buf_offset, f->buf_index);
387 a672b469 aliguori
        if (len > 0)
388 a672b469 aliguori
            f->buf_offset += f->buf_index;
389 a672b469 aliguori
        else
390 a672b469 aliguori
            f->has_error = 1;
391 a672b469 aliguori
        f->buf_index = 0;
392 a672b469 aliguori
    }
393 a672b469 aliguori
}
394 a672b469 aliguori
395 a672b469 aliguori
static void qemu_fill_buffer(QEMUFile *f)
396 a672b469 aliguori
{
397 a672b469 aliguori
    int len;
398 a672b469 aliguori
399 a672b469 aliguori
    if (!f->get_buffer)
400 a672b469 aliguori
        return;
401 a672b469 aliguori
402 a672b469 aliguori
    if (f->is_write)
403 a672b469 aliguori
        abort();
404 a672b469 aliguori
405 a672b469 aliguori
    len = f->get_buffer(f->opaque, f->buf, f->buf_offset, IO_BUF_SIZE);
406 a672b469 aliguori
    if (len > 0) {
407 a672b469 aliguori
        f->buf_index = 0;
408 a672b469 aliguori
        f->buf_size = len;
409 a672b469 aliguori
        f->buf_offset += len;
410 a672b469 aliguori
    } else if (len != -EAGAIN)
411 a672b469 aliguori
        f->has_error = 1;
412 a672b469 aliguori
}
413 a672b469 aliguori
414 a672b469 aliguori
int qemu_fclose(QEMUFile *f)
415 a672b469 aliguori
{
416 a672b469 aliguori
    int ret = 0;
417 a672b469 aliguori
    qemu_fflush(f);
418 a672b469 aliguori
    if (f->close)
419 a672b469 aliguori
        ret = f->close(f->opaque);
420 a672b469 aliguori
    qemu_free(f);
421 a672b469 aliguori
    return ret;
422 a672b469 aliguori
}
423 a672b469 aliguori
424 a672b469 aliguori
void qemu_file_put_notify(QEMUFile *f)
425 a672b469 aliguori
{
426 a672b469 aliguori
    f->put_buffer(f->opaque, NULL, 0, 0);
427 a672b469 aliguori
}
428 a672b469 aliguori
429 a672b469 aliguori
void qemu_put_buffer(QEMUFile *f, const uint8_t *buf, int size)
430 a672b469 aliguori
{
431 a672b469 aliguori
    int l;
432 a672b469 aliguori
433 a672b469 aliguori
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
434 a672b469 aliguori
        fprintf(stderr,
435 a672b469 aliguori
                "Attempted to write to buffer while read buffer is not empty\n");
436 a672b469 aliguori
        abort();
437 a672b469 aliguori
    }
438 a672b469 aliguori
439 a672b469 aliguori
    while (!f->has_error && size > 0) {
440 a672b469 aliguori
        l = IO_BUF_SIZE - f->buf_index;
441 a672b469 aliguori
        if (l > size)
442 a672b469 aliguori
            l = size;
443 a672b469 aliguori
        memcpy(f->buf + f->buf_index, buf, l);
444 a672b469 aliguori
        f->is_write = 1;
445 a672b469 aliguori
        f->buf_index += l;
446 a672b469 aliguori
        buf += l;
447 a672b469 aliguori
        size -= l;
448 a672b469 aliguori
        if (f->buf_index >= IO_BUF_SIZE)
449 a672b469 aliguori
            qemu_fflush(f);
450 a672b469 aliguori
    }
451 a672b469 aliguori
}
452 a672b469 aliguori
453 a672b469 aliguori
void qemu_put_byte(QEMUFile *f, int v)
454 a672b469 aliguori
{
455 a672b469 aliguori
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
456 a672b469 aliguori
        fprintf(stderr,
457 a672b469 aliguori
                "Attempted to write to buffer while read buffer is not empty\n");
458 a672b469 aliguori
        abort();
459 a672b469 aliguori
    }
460 a672b469 aliguori
461 a672b469 aliguori
    f->buf[f->buf_index++] = v;
462 a672b469 aliguori
    f->is_write = 1;
463 a672b469 aliguori
    if (f->buf_index >= IO_BUF_SIZE)
464 a672b469 aliguori
        qemu_fflush(f);
465 a672b469 aliguori
}
466 a672b469 aliguori
467 a672b469 aliguori
int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size1)
468 a672b469 aliguori
{
469 a672b469 aliguori
    int size, l;
470 a672b469 aliguori
471 a672b469 aliguori
    if (f->is_write)
472 a672b469 aliguori
        abort();
473 a672b469 aliguori
474 a672b469 aliguori
    size = size1;
475 a672b469 aliguori
    while (size > 0) {
476 a672b469 aliguori
        l = f->buf_size - f->buf_index;
477 a672b469 aliguori
        if (l == 0) {
478 a672b469 aliguori
            qemu_fill_buffer(f);
479 a672b469 aliguori
            l = f->buf_size - f->buf_index;
480 a672b469 aliguori
            if (l == 0)
481 a672b469 aliguori
                break;
482 a672b469 aliguori
        }
483 a672b469 aliguori
        if (l > size)
484 a672b469 aliguori
            l = size;
485 a672b469 aliguori
        memcpy(buf, f->buf + f->buf_index, l);
486 a672b469 aliguori
        f->buf_index += l;
487 a672b469 aliguori
        buf += l;
488 a672b469 aliguori
        size -= l;
489 a672b469 aliguori
    }
490 a672b469 aliguori
    return size1 - size;
491 a672b469 aliguori
}
492 a672b469 aliguori
493 a672b469 aliguori
int qemu_get_byte(QEMUFile *f)
494 a672b469 aliguori
{
495 a672b469 aliguori
    if (f->is_write)
496 a672b469 aliguori
        abort();
497 a672b469 aliguori
498 a672b469 aliguori
    if (f->buf_index >= f->buf_size) {
499 a672b469 aliguori
        qemu_fill_buffer(f);
500 a672b469 aliguori
        if (f->buf_index >= f->buf_size)
501 a672b469 aliguori
            return 0;
502 a672b469 aliguori
    }
503 a672b469 aliguori
    return f->buf[f->buf_index++];
504 a672b469 aliguori
}
505 a672b469 aliguori
506 a672b469 aliguori
int64_t qemu_ftell(QEMUFile *f)
507 a672b469 aliguori
{
508 a672b469 aliguori
    return f->buf_offset - f->buf_size + f->buf_index;
509 a672b469 aliguori
}
510 a672b469 aliguori
511 a672b469 aliguori
int64_t qemu_fseek(QEMUFile *f, int64_t pos, int whence)
512 a672b469 aliguori
{
513 a672b469 aliguori
    if (whence == SEEK_SET) {
514 a672b469 aliguori
        /* nothing to do */
515 a672b469 aliguori
    } else if (whence == SEEK_CUR) {
516 a672b469 aliguori
        pos += qemu_ftell(f);
517 a672b469 aliguori
    } else {
518 a672b469 aliguori
        /* SEEK_END not supported */
519 a672b469 aliguori
        return -1;
520 a672b469 aliguori
    }
521 a672b469 aliguori
    if (f->put_buffer) {
522 a672b469 aliguori
        qemu_fflush(f);
523 a672b469 aliguori
        f->buf_offset = pos;
524 a672b469 aliguori
    } else {
525 a672b469 aliguori
        f->buf_offset = pos;
526 a672b469 aliguori
        f->buf_index = 0;
527 a672b469 aliguori
        f->buf_size = 0;
528 a672b469 aliguori
    }
529 a672b469 aliguori
    return pos;
530 a672b469 aliguori
}
531 a672b469 aliguori
532 a672b469 aliguori
int qemu_file_rate_limit(QEMUFile *f)
533 a672b469 aliguori
{
534 a672b469 aliguori
    if (f->rate_limit)
535 a672b469 aliguori
        return f->rate_limit(f->opaque);
536 a672b469 aliguori
537 a672b469 aliguori
    return 0;
538 a672b469 aliguori
}
539 a672b469 aliguori
540 a672b469 aliguori
void qemu_put_be16(QEMUFile *f, unsigned int v)
541 a672b469 aliguori
{
542 a672b469 aliguori
    qemu_put_byte(f, v >> 8);
543 a672b469 aliguori
    qemu_put_byte(f, v);
544 a672b469 aliguori
}
545 a672b469 aliguori
546 a672b469 aliguori
void qemu_put_be32(QEMUFile *f, unsigned int v)
547 a672b469 aliguori
{
548 a672b469 aliguori
    qemu_put_byte(f, v >> 24);
549 a672b469 aliguori
    qemu_put_byte(f, v >> 16);
550 a672b469 aliguori
    qemu_put_byte(f, v >> 8);
551 a672b469 aliguori
    qemu_put_byte(f, v);
552 a672b469 aliguori
}
553 a672b469 aliguori
554 a672b469 aliguori
void qemu_put_be64(QEMUFile *f, uint64_t v)
555 a672b469 aliguori
{
556 a672b469 aliguori
    qemu_put_be32(f, v >> 32);
557 a672b469 aliguori
    qemu_put_be32(f, v);
558 a672b469 aliguori
}
559 a672b469 aliguori
560 a672b469 aliguori
unsigned int qemu_get_be16(QEMUFile *f)
561 a672b469 aliguori
{
562 a672b469 aliguori
    unsigned int v;
563 a672b469 aliguori
    v = qemu_get_byte(f) << 8;
564 a672b469 aliguori
    v |= qemu_get_byte(f);
565 a672b469 aliguori
    return v;
566 a672b469 aliguori
}
567 a672b469 aliguori
568 a672b469 aliguori
unsigned int qemu_get_be32(QEMUFile *f)
569 a672b469 aliguori
{
570 a672b469 aliguori
    unsigned int v;
571 a672b469 aliguori
    v = qemu_get_byte(f) << 24;
572 a672b469 aliguori
    v |= qemu_get_byte(f) << 16;
573 a672b469 aliguori
    v |= qemu_get_byte(f) << 8;
574 a672b469 aliguori
    v |= qemu_get_byte(f);
575 a672b469 aliguori
    return v;
576 a672b469 aliguori
}
577 a672b469 aliguori
578 a672b469 aliguori
uint64_t qemu_get_be64(QEMUFile *f)
579 a672b469 aliguori
{
580 a672b469 aliguori
    uint64_t v;
581 a672b469 aliguori
    v = (uint64_t)qemu_get_be32(f) << 32;
582 a672b469 aliguori
    v |= qemu_get_be32(f);
583 a672b469 aliguori
    return v;
584 a672b469 aliguori
}
585 a672b469 aliguori
586 a672b469 aliguori
typedef struct SaveStateEntry {
587 a672b469 aliguori
    char idstr[256];
588 a672b469 aliguori
    int instance_id;
589 a672b469 aliguori
    int version_id;
590 a672b469 aliguori
    int section_id;
591 a672b469 aliguori
    SaveLiveStateHandler *save_live_state;
592 a672b469 aliguori
    SaveStateHandler *save_state;
593 a672b469 aliguori
    LoadStateHandler *load_state;
594 a672b469 aliguori
    void *opaque;
595 a672b469 aliguori
    struct SaveStateEntry *next;
596 a672b469 aliguori
} SaveStateEntry;
597 a672b469 aliguori
598 a672b469 aliguori
static SaveStateEntry *first_se;
599 a672b469 aliguori
600 a672b469 aliguori
/* TODO: Individual devices generally have very little idea about the rest
601 a672b469 aliguori
   of the system, so instance_id should be removed/replaced.
602 a672b469 aliguori
   Meanwhile pass -1 as instance_id if you do not already have a clearly
603 a672b469 aliguori
   distinguishing id for all instances of your device class. */
604 a672b469 aliguori
int register_savevm_live(const char *idstr,
605 a672b469 aliguori
                         int instance_id,
606 a672b469 aliguori
                         int version_id,
607 a672b469 aliguori
                         SaveLiveStateHandler *save_live_state,
608 a672b469 aliguori
                         SaveStateHandler *save_state,
609 a672b469 aliguori
                         LoadStateHandler *load_state,
610 a672b469 aliguori
                         void *opaque)
611 a672b469 aliguori
{
612 a672b469 aliguori
    SaveStateEntry *se, **pse;
613 a672b469 aliguori
    static int global_section_id;
614 a672b469 aliguori
615 a672b469 aliguori
    se = qemu_malloc(sizeof(SaveStateEntry));
616 a672b469 aliguori
    pstrcpy(se->idstr, sizeof(se->idstr), idstr);
617 a672b469 aliguori
    se->instance_id = (instance_id == -1) ? 0 : instance_id;
618 a672b469 aliguori
    se->version_id = version_id;
619 a672b469 aliguori
    se->section_id = global_section_id++;
620 a672b469 aliguori
    se->save_live_state = save_live_state;
621 a672b469 aliguori
    se->save_state = save_state;
622 a672b469 aliguori
    se->load_state = load_state;
623 a672b469 aliguori
    se->opaque = opaque;
624 a672b469 aliguori
    se->next = NULL;
625 a672b469 aliguori
626 a672b469 aliguori
    /* add at the end of list */
627 a672b469 aliguori
    pse = &first_se;
628 a672b469 aliguori
    while (*pse != NULL) {
629 a672b469 aliguori
        if (instance_id == -1
630 a672b469 aliguori
                && strcmp(se->idstr, (*pse)->idstr) == 0
631 a672b469 aliguori
                && se->instance_id <= (*pse)->instance_id)
632 a672b469 aliguori
            se->instance_id = (*pse)->instance_id + 1;
633 a672b469 aliguori
        pse = &(*pse)->next;
634 a672b469 aliguori
    }
635 a672b469 aliguori
    *pse = se;
636 a672b469 aliguori
    return 0;
637 a672b469 aliguori
}
638 a672b469 aliguori
639 a672b469 aliguori
int register_savevm(const char *idstr,
640 a672b469 aliguori
                    int instance_id,
641 a672b469 aliguori
                    int version_id,
642 a672b469 aliguori
                    SaveStateHandler *save_state,
643 a672b469 aliguori
                    LoadStateHandler *load_state,
644 a672b469 aliguori
                    void *opaque)
645 a672b469 aliguori
{
646 a672b469 aliguori
    return register_savevm_live(idstr, instance_id, version_id,
647 a672b469 aliguori
                                NULL, save_state, load_state, opaque);
648 a672b469 aliguori
}
649 a672b469 aliguori
650 41bd13af aliguori
void unregister_savevm(const char *idstr, void *opaque)
651 41bd13af aliguori
{
652 41bd13af aliguori
    SaveStateEntry **pse;
653 41bd13af aliguori
654 41bd13af aliguori
    pse = &first_se;
655 41bd13af aliguori
    while (*pse != NULL) {
656 41bd13af aliguori
        if (strcmp((*pse)->idstr, idstr) == 0 && (*pse)->opaque == opaque) {
657 41bd13af aliguori
            SaveStateEntry *next = (*pse)->next;
658 41bd13af aliguori
            qemu_free(*pse);
659 41bd13af aliguori
            *pse = next;
660 41bd13af aliguori
            continue;
661 41bd13af aliguori
        }
662 41bd13af aliguori
        pse = &(*pse)->next;
663 41bd13af aliguori
    }
664 41bd13af aliguori
}
665 41bd13af aliguori
666 a672b469 aliguori
#define QEMU_VM_FILE_MAGIC           0x5145564d
667 a672b469 aliguori
#define QEMU_VM_FILE_VERSION_COMPAT  0x00000002
668 a672b469 aliguori
#define QEMU_VM_FILE_VERSION         0x00000003
669 a672b469 aliguori
670 a672b469 aliguori
#define QEMU_VM_EOF                  0x00
671 a672b469 aliguori
#define QEMU_VM_SECTION_START        0x01
672 a672b469 aliguori
#define QEMU_VM_SECTION_PART         0x02
673 a672b469 aliguori
#define QEMU_VM_SECTION_END          0x03
674 a672b469 aliguori
#define QEMU_VM_SECTION_FULL         0x04
675 a672b469 aliguori
676 a672b469 aliguori
int qemu_savevm_state_begin(QEMUFile *f)
677 a672b469 aliguori
{
678 a672b469 aliguori
    SaveStateEntry *se;
679 a672b469 aliguori
680 a672b469 aliguori
    qemu_put_be32(f, QEMU_VM_FILE_MAGIC);
681 a672b469 aliguori
    qemu_put_be32(f, QEMU_VM_FILE_VERSION);
682 a672b469 aliguori
683 a672b469 aliguori
    for (se = first_se; se != NULL; se = se->next) {
684 a672b469 aliguori
        int len;
685 a672b469 aliguori
686 a672b469 aliguori
        if (se->save_live_state == NULL)
687 a672b469 aliguori
            continue;
688 a672b469 aliguori
689 a672b469 aliguori
        /* Section type */
690 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_START);
691 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
692 a672b469 aliguori
693 a672b469 aliguori
        /* ID string */
694 a672b469 aliguori
        len = strlen(se->idstr);
695 a672b469 aliguori
        qemu_put_byte(f, len);
696 a672b469 aliguori
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
697 a672b469 aliguori
698 a672b469 aliguori
        qemu_put_be32(f, se->instance_id);
699 a672b469 aliguori
        qemu_put_be32(f, se->version_id);
700 a672b469 aliguori
701 a672b469 aliguori
        se->save_live_state(f, QEMU_VM_SECTION_START, se->opaque);
702 a672b469 aliguori
    }
703 a672b469 aliguori
704 a672b469 aliguori
    if (qemu_file_has_error(f))
705 a672b469 aliguori
        return -EIO;
706 a672b469 aliguori
707 a672b469 aliguori
    return 0;
708 a672b469 aliguori
}
709 a672b469 aliguori
710 a672b469 aliguori
int qemu_savevm_state_iterate(QEMUFile *f)
711 a672b469 aliguori
{
712 a672b469 aliguori
    SaveStateEntry *se;
713 a672b469 aliguori
    int ret = 1;
714 a672b469 aliguori
715 a672b469 aliguori
    for (se = first_se; se != NULL; se = se->next) {
716 a672b469 aliguori
        if (se->save_live_state == NULL)
717 a672b469 aliguori
            continue;
718 a672b469 aliguori
719 a672b469 aliguori
        /* Section type */
720 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_PART);
721 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
722 a672b469 aliguori
723 a672b469 aliguori
        ret &= !!se->save_live_state(f, QEMU_VM_SECTION_PART, se->opaque);
724 a672b469 aliguori
    }
725 a672b469 aliguori
726 a672b469 aliguori
    if (ret)
727 a672b469 aliguori
        return 1;
728 a672b469 aliguori
729 a672b469 aliguori
    if (qemu_file_has_error(f))
730 a672b469 aliguori
        return -EIO;
731 a672b469 aliguori
732 a672b469 aliguori
    return 0;
733 a672b469 aliguori
}
734 a672b469 aliguori
735 a672b469 aliguori
int qemu_savevm_state_complete(QEMUFile *f)
736 a672b469 aliguori
{
737 a672b469 aliguori
    SaveStateEntry *se;
738 a672b469 aliguori
739 a672b469 aliguori
    for (se = first_se; se != NULL; se = se->next) {
740 a672b469 aliguori
        if (se->save_live_state == NULL)
741 a672b469 aliguori
            continue;
742 a672b469 aliguori
743 a672b469 aliguori
        /* Section type */
744 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_END);
745 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
746 a672b469 aliguori
747 a672b469 aliguori
        se->save_live_state(f, QEMU_VM_SECTION_END, se->opaque);
748 a672b469 aliguori
    }
749 a672b469 aliguori
750 a672b469 aliguori
    for(se = first_se; se != NULL; se = se->next) {
751 a672b469 aliguori
        int len;
752 a672b469 aliguori
753 a672b469 aliguori
        if (se->save_state == NULL)
754 a672b469 aliguori
            continue;
755 a672b469 aliguori
756 a672b469 aliguori
        /* Section type */
757 a672b469 aliguori
        qemu_put_byte(f, QEMU_VM_SECTION_FULL);
758 a672b469 aliguori
        qemu_put_be32(f, se->section_id);
759 a672b469 aliguori
760 a672b469 aliguori
        /* ID string */
761 a672b469 aliguori
        len = strlen(se->idstr);
762 a672b469 aliguori
        qemu_put_byte(f, len);
763 a672b469 aliguori
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
764 a672b469 aliguori
765 a672b469 aliguori
        qemu_put_be32(f, se->instance_id);
766 a672b469 aliguori
        qemu_put_be32(f, se->version_id);
767 a672b469 aliguori
768 a672b469 aliguori
        se->save_state(f, se->opaque);
769 a672b469 aliguori
    }
770 a672b469 aliguori
771 a672b469 aliguori
    qemu_put_byte(f, QEMU_VM_EOF);
772 a672b469 aliguori
773 a672b469 aliguori
    if (qemu_file_has_error(f))
774 a672b469 aliguori
        return -EIO;
775 a672b469 aliguori
776 a672b469 aliguori
    return 0;
777 a672b469 aliguori
}
778 a672b469 aliguori
779 a672b469 aliguori
int qemu_savevm_state(QEMUFile *f)
780 a672b469 aliguori
{
781 a672b469 aliguori
    int saved_vm_running;
782 a672b469 aliguori
    int ret;
783 a672b469 aliguori
784 a672b469 aliguori
    saved_vm_running = vm_running;
785 a672b469 aliguori
    vm_stop(0);
786 a672b469 aliguori
787 a672b469 aliguori
    bdrv_flush_all();
788 a672b469 aliguori
789 a672b469 aliguori
    ret = qemu_savevm_state_begin(f);
790 a672b469 aliguori
    if (ret < 0)
791 a672b469 aliguori
        goto out;
792 a672b469 aliguori
793 a672b469 aliguori
    do {
794 a672b469 aliguori
        ret = qemu_savevm_state_iterate(f);
795 a672b469 aliguori
        if (ret < 0)
796 a672b469 aliguori
            goto out;
797 a672b469 aliguori
    } while (ret == 0);
798 a672b469 aliguori
799 a672b469 aliguori
    ret = qemu_savevm_state_complete(f);
800 a672b469 aliguori
801 a672b469 aliguori
out:
802 a672b469 aliguori
    if (qemu_file_has_error(f))
803 a672b469 aliguori
        ret = -EIO;
804 a672b469 aliguori
805 a672b469 aliguori
    if (!ret && saved_vm_running)
806 a672b469 aliguori
        vm_start();
807 a672b469 aliguori
808 a672b469 aliguori
    return ret;
809 a672b469 aliguori
}
810 a672b469 aliguori
811 a672b469 aliguori
static SaveStateEntry *find_se(const char *idstr, int instance_id)
812 a672b469 aliguori
{
813 a672b469 aliguori
    SaveStateEntry *se;
814 a672b469 aliguori
815 a672b469 aliguori
    for(se = first_se; se != NULL; se = se->next) {
816 a672b469 aliguori
        if (!strcmp(se->idstr, idstr) &&
817 a672b469 aliguori
            instance_id == se->instance_id)
818 a672b469 aliguori
            return se;
819 a672b469 aliguori
    }
820 a672b469 aliguori
    return NULL;
821 a672b469 aliguori
}
822 a672b469 aliguori
823 a672b469 aliguori
typedef struct LoadStateEntry {
824 a672b469 aliguori
    SaveStateEntry *se;
825 a672b469 aliguori
    int section_id;
826 a672b469 aliguori
    int version_id;
827 a672b469 aliguori
    struct LoadStateEntry *next;
828 a672b469 aliguori
} LoadStateEntry;
829 a672b469 aliguori
830 a672b469 aliguori
static int qemu_loadvm_state_v2(QEMUFile *f)
831 a672b469 aliguori
{
832 a672b469 aliguori
    SaveStateEntry *se;
833 a672b469 aliguori
    int len, ret, instance_id, record_len, version_id;
834 a672b469 aliguori
    int64_t total_len, end_pos, cur_pos;
835 a672b469 aliguori
    char idstr[256];
836 a672b469 aliguori
837 a672b469 aliguori
    total_len = qemu_get_be64(f);
838 a672b469 aliguori
    end_pos = total_len + qemu_ftell(f);
839 a672b469 aliguori
    for(;;) {
840 a672b469 aliguori
        if (qemu_ftell(f) >= end_pos)
841 a672b469 aliguori
            break;
842 a672b469 aliguori
        len = qemu_get_byte(f);
843 a672b469 aliguori
        qemu_get_buffer(f, (uint8_t *)idstr, len);
844 a672b469 aliguori
        idstr[len] = '\0';
845 a672b469 aliguori
        instance_id = qemu_get_be32(f);
846 a672b469 aliguori
        version_id = qemu_get_be32(f);
847 a672b469 aliguori
        record_len = qemu_get_be32(f);
848 a672b469 aliguori
        cur_pos = qemu_ftell(f);
849 a672b469 aliguori
        se = find_se(idstr, instance_id);
850 a672b469 aliguori
        if (!se) {
851 a672b469 aliguori
            fprintf(stderr, "qemu: warning: instance 0x%x of device '%s' not present in current VM\n",
852 a672b469 aliguori
                    instance_id, idstr);
853 a672b469 aliguori
        } else {
854 a672b469 aliguori
            ret = se->load_state(f, se->opaque, version_id);
855 a672b469 aliguori
            if (ret < 0) {
856 a672b469 aliguori
                fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
857 a672b469 aliguori
                        instance_id, idstr);
858 d02f7094 Anthony Liguori
                return ret;
859 a672b469 aliguori
            }
860 a672b469 aliguori
        }
861 a672b469 aliguori
        /* always seek to exact end of record */
862 a672b469 aliguori
        qemu_fseek(f, cur_pos + record_len, SEEK_SET);
863 a672b469 aliguori
    }
864 a672b469 aliguori
865 a672b469 aliguori
    if (qemu_file_has_error(f))
866 a672b469 aliguori
        return -EIO;
867 a672b469 aliguori
868 a672b469 aliguori
    return 0;
869 a672b469 aliguori
}
870 a672b469 aliguori
871 a672b469 aliguori
int qemu_loadvm_state(QEMUFile *f)
872 a672b469 aliguori
{
873 a672b469 aliguori
    LoadStateEntry *first_le = NULL;
874 a672b469 aliguori
    uint8_t section_type;
875 a672b469 aliguori
    unsigned int v;
876 a672b469 aliguori
    int ret;
877 a672b469 aliguori
878 a672b469 aliguori
    v = qemu_get_be32(f);
879 a672b469 aliguori
    if (v != QEMU_VM_FILE_MAGIC)
880 a672b469 aliguori
        return -EINVAL;
881 a672b469 aliguori
882 a672b469 aliguori
    v = qemu_get_be32(f);
883 a672b469 aliguori
    if (v == QEMU_VM_FILE_VERSION_COMPAT)
884 a672b469 aliguori
        return qemu_loadvm_state_v2(f);
885 a672b469 aliguori
    if (v != QEMU_VM_FILE_VERSION)
886 a672b469 aliguori
        return -ENOTSUP;
887 a672b469 aliguori
888 a672b469 aliguori
    while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
889 a672b469 aliguori
        uint32_t instance_id, version_id, section_id;
890 a672b469 aliguori
        LoadStateEntry *le;
891 a672b469 aliguori
        SaveStateEntry *se;
892 a672b469 aliguori
        char idstr[257];
893 a672b469 aliguori
        int len;
894 a672b469 aliguori
895 a672b469 aliguori
        switch (section_type) {
896 a672b469 aliguori
        case QEMU_VM_SECTION_START:
897 a672b469 aliguori
        case QEMU_VM_SECTION_FULL:
898 a672b469 aliguori
            /* Read section start */
899 a672b469 aliguori
            section_id = qemu_get_be32(f);
900 a672b469 aliguori
            len = qemu_get_byte(f);
901 a672b469 aliguori
            qemu_get_buffer(f, (uint8_t *)idstr, len);
902 a672b469 aliguori
            idstr[len] = 0;
903 a672b469 aliguori
            instance_id = qemu_get_be32(f);
904 a672b469 aliguori
            version_id = qemu_get_be32(f);
905 a672b469 aliguori
906 a672b469 aliguori
            /* Find savevm section */
907 a672b469 aliguori
            se = find_se(idstr, instance_id);
908 a672b469 aliguori
            if (se == NULL) {
909 a672b469 aliguori
                fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
910 a672b469 aliguori
                ret = -EINVAL;
911 a672b469 aliguori
                goto out;
912 a672b469 aliguori
            }
913 a672b469 aliguori
914 a672b469 aliguori
            /* Validate version */
915 a672b469 aliguori
            if (version_id > se->version_id) {
916 a672b469 aliguori
                fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
917 a672b469 aliguori
                        version_id, idstr, se->version_id);
918 a672b469 aliguori
                ret = -EINVAL;
919 a672b469 aliguori
                goto out;
920 a672b469 aliguori
            }
921 a672b469 aliguori
922 a672b469 aliguori
            /* Add entry */
923 a672b469 aliguori
            le = qemu_mallocz(sizeof(*le));
924 a672b469 aliguori
925 a672b469 aliguori
            le->se = se;
926 a672b469 aliguori
            le->section_id = section_id;
927 a672b469 aliguori
            le->version_id = version_id;
928 a672b469 aliguori
            le->next = first_le;
929 a672b469 aliguori
            first_le = le;
930 a672b469 aliguori
931 a672b469 aliguori
            le->se->load_state(f, le->se->opaque, le->version_id);
932 a672b469 aliguori
            break;
933 a672b469 aliguori
        case QEMU_VM_SECTION_PART:
934 a672b469 aliguori
        case QEMU_VM_SECTION_END:
935 a672b469 aliguori
            section_id = qemu_get_be32(f);
936 a672b469 aliguori
937 a672b469 aliguori
            for (le = first_le; le && le->section_id != section_id; le = le->next);
938 a672b469 aliguori
            if (le == NULL) {
939 a672b469 aliguori
                fprintf(stderr, "Unknown savevm section %d\n", section_id);
940 a672b469 aliguori
                ret = -EINVAL;
941 a672b469 aliguori
                goto out;
942 a672b469 aliguori
            }
943 a672b469 aliguori
944 a672b469 aliguori
            le->se->load_state(f, le->se->opaque, le->version_id);
945 a672b469 aliguori
            break;
946 a672b469 aliguori
        default:
947 a672b469 aliguori
            fprintf(stderr, "Unknown savevm section type %d\n", section_type);
948 a672b469 aliguori
            ret = -EINVAL;
949 a672b469 aliguori
            goto out;
950 a672b469 aliguori
        }
951 a672b469 aliguori
    }
952 a672b469 aliguori
953 a672b469 aliguori
    ret = 0;
954 a672b469 aliguori
955 a672b469 aliguori
out:
956 a672b469 aliguori
    while (first_le) {
957 a672b469 aliguori
        LoadStateEntry *le = first_le;
958 a672b469 aliguori
        first_le = first_le->next;
959 a672b469 aliguori
        qemu_free(le);
960 a672b469 aliguori
    }
961 a672b469 aliguori
962 a672b469 aliguori
    if (qemu_file_has_error(f))
963 a672b469 aliguori
        ret = -EIO;
964 a672b469 aliguori
965 a672b469 aliguori
    return ret;
966 a672b469 aliguori
}
967 a672b469 aliguori
968 a672b469 aliguori
/* device can contain snapshots */
969 a672b469 aliguori
static int bdrv_can_snapshot(BlockDriverState *bs)
970 a672b469 aliguori
{
971 a672b469 aliguori
    return (bs &&
972 a672b469 aliguori
            !bdrv_is_removable(bs) &&
973 a672b469 aliguori
            !bdrv_is_read_only(bs));
974 a672b469 aliguori
}
975 a672b469 aliguori
976 a672b469 aliguori
/* device must be snapshots in order to have a reliable snapshot */
977 a672b469 aliguori
static int bdrv_has_snapshot(BlockDriverState *bs)
978 a672b469 aliguori
{
979 a672b469 aliguori
    return (bs &&
980 a672b469 aliguori
            !bdrv_is_removable(bs) &&
981 a672b469 aliguori
            !bdrv_is_read_only(bs));
982 a672b469 aliguori
}
983 a672b469 aliguori
984 a672b469 aliguori
static BlockDriverState *get_bs_snapshots(void)
985 a672b469 aliguori
{
986 a672b469 aliguori
    BlockDriverState *bs;
987 a672b469 aliguori
    int i;
988 a672b469 aliguori
989 a672b469 aliguori
    if (bs_snapshots)
990 a672b469 aliguori
        return bs_snapshots;
991 a672b469 aliguori
    for(i = 0; i <= nb_drives; i++) {
992 a672b469 aliguori
        bs = drives_table[i].bdrv;
993 a672b469 aliguori
        if (bdrv_can_snapshot(bs))
994 a672b469 aliguori
            goto ok;
995 a672b469 aliguori
    }
996 a672b469 aliguori
    return NULL;
997 a672b469 aliguori
 ok:
998 a672b469 aliguori
    bs_snapshots = bs;
999 a672b469 aliguori
    return bs;
1000 a672b469 aliguori
}
1001 a672b469 aliguori
1002 a672b469 aliguori
static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
1003 a672b469 aliguori
                              const char *name)
1004 a672b469 aliguori
{
1005 a672b469 aliguori
    QEMUSnapshotInfo *sn_tab, *sn;
1006 a672b469 aliguori
    int nb_sns, i, ret;
1007 a672b469 aliguori
1008 a672b469 aliguori
    ret = -ENOENT;
1009 a672b469 aliguori
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1010 a672b469 aliguori
    if (nb_sns < 0)
1011 a672b469 aliguori
        return ret;
1012 a672b469 aliguori
    for(i = 0; i < nb_sns; i++) {
1013 a672b469 aliguori
        sn = &sn_tab[i];
1014 a672b469 aliguori
        if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
1015 a672b469 aliguori
            *sn_info = *sn;
1016 a672b469 aliguori
            ret = 0;
1017 a672b469 aliguori
            break;
1018 a672b469 aliguori
        }
1019 a672b469 aliguori
    }
1020 a672b469 aliguori
    qemu_free(sn_tab);
1021 a672b469 aliguori
    return ret;
1022 a672b469 aliguori
}
1023 a672b469 aliguori
1024 376253ec aliguori
void do_savevm(Monitor *mon, const char *name)
1025 a672b469 aliguori
{
1026 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1027 a672b469 aliguori
    QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1;
1028 a672b469 aliguori
    int must_delete, ret, i;
1029 a672b469 aliguori
    BlockDriverInfo bdi1, *bdi = &bdi1;
1030 a672b469 aliguori
    QEMUFile *f;
1031 a672b469 aliguori
    int saved_vm_running;
1032 2d22b18f aliguori
    uint32_t vm_state_size;
1033 a672b469 aliguori
#ifdef _WIN32
1034 a672b469 aliguori
    struct _timeb tb;
1035 a672b469 aliguori
#else
1036 a672b469 aliguori
    struct timeval tv;
1037 a672b469 aliguori
#endif
1038 a672b469 aliguori
1039 a672b469 aliguori
    bs = get_bs_snapshots();
1040 a672b469 aliguori
    if (!bs) {
1041 376253ec aliguori
        monitor_printf(mon, "No block device can accept snapshots\n");
1042 a672b469 aliguori
        return;
1043 a672b469 aliguori
    }
1044 a672b469 aliguori
1045 a672b469 aliguori
    /* ??? Should this occur after vm_stop?  */
1046 a672b469 aliguori
    qemu_aio_flush();
1047 a672b469 aliguori
1048 a672b469 aliguori
    saved_vm_running = vm_running;
1049 a672b469 aliguori
    vm_stop(0);
1050 a672b469 aliguori
1051 a672b469 aliguori
    must_delete = 0;
1052 a672b469 aliguori
    if (name) {
1053 a672b469 aliguori
        ret = bdrv_snapshot_find(bs, old_sn, name);
1054 a672b469 aliguori
        if (ret >= 0) {
1055 a672b469 aliguori
            must_delete = 1;
1056 a672b469 aliguori
        }
1057 a672b469 aliguori
    }
1058 a672b469 aliguori
    memset(sn, 0, sizeof(*sn));
1059 a672b469 aliguori
    if (must_delete) {
1060 a672b469 aliguori
        pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
1061 a672b469 aliguori
        pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
1062 a672b469 aliguori
    } else {
1063 a672b469 aliguori
        if (name)
1064 a672b469 aliguori
            pstrcpy(sn->name, sizeof(sn->name), name);
1065 a672b469 aliguori
    }
1066 a672b469 aliguori
1067 a672b469 aliguori
    /* fill auxiliary fields */
1068 a672b469 aliguori
#ifdef _WIN32
1069 a672b469 aliguori
    _ftime(&tb);
1070 a672b469 aliguori
    sn->date_sec = tb.time;
1071 a672b469 aliguori
    sn->date_nsec = tb.millitm * 1000000;
1072 a672b469 aliguori
#else
1073 a672b469 aliguori
    gettimeofday(&tv, NULL);
1074 a672b469 aliguori
    sn->date_sec = tv.tv_sec;
1075 a672b469 aliguori
    sn->date_nsec = tv.tv_usec * 1000;
1076 a672b469 aliguori
#endif
1077 a672b469 aliguori
    sn->vm_clock_nsec = qemu_get_clock(vm_clock);
1078 a672b469 aliguori
1079 a672b469 aliguori
    if (bdrv_get_info(bs, bdi) < 0 || bdi->vm_state_offset <= 0) {
1080 376253ec aliguori
        monitor_printf(mon, "Device %s does not support VM state snapshots\n",
1081 376253ec aliguori
                       bdrv_get_device_name(bs));
1082 a672b469 aliguori
        goto the_end;
1083 a672b469 aliguori
    }
1084 a672b469 aliguori
1085 a672b469 aliguori
    /* save the VM state */
1086 a672b469 aliguori
    f = qemu_fopen_bdrv(bs, bdi->vm_state_offset, 1);
1087 a672b469 aliguori
    if (!f) {
1088 376253ec aliguori
        monitor_printf(mon, "Could not open VM state file\n");
1089 a672b469 aliguori
        goto the_end;
1090 a672b469 aliguori
    }
1091 a672b469 aliguori
    ret = qemu_savevm_state(f);
1092 2d22b18f aliguori
    vm_state_size = qemu_ftell(f);
1093 a672b469 aliguori
    qemu_fclose(f);
1094 a672b469 aliguori
    if (ret < 0) {
1095 376253ec aliguori
        monitor_printf(mon, "Error %d while writing VM\n", ret);
1096 a672b469 aliguori
        goto the_end;
1097 a672b469 aliguori
    }
1098 a672b469 aliguori
1099 a672b469 aliguori
    /* create the snapshots */
1100 a672b469 aliguori
1101 a672b469 aliguori
    for(i = 0; i < nb_drives; i++) {
1102 a672b469 aliguori
        bs1 = drives_table[i].bdrv;
1103 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1104 a672b469 aliguori
            if (must_delete) {
1105 a672b469 aliguori
                ret = bdrv_snapshot_delete(bs1, old_sn->id_str);
1106 a672b469 aliguori
                if (ret < 0) {
1107 376253ec aliguori
                    monitor_printf(mon,
1108 376253ec aliguori
                                   "Error while deleting snapshot on '%s'\n",
1109 376253ec aliguori
                                   bdrv_get_device_name(bs1));
1110 a672b469 aliguori
                }
1111 a672b469 aliguori
            }
1112 2d22b18f aliguori
            /* Write VM state size only to the image that contains the state */
1113 2d22b18f aliguori
            sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
1114 a672b469 aliguori
            ret = bdrv_snapshot_create(bs1, sn);
1115 a672b469 aliguori
            if (ret < 0) {
1116 376253ec aliguori
                monitor_printf(mon, "Error while creating snapshot on '%s'\n",
1117 376253ec aliguori
                               bdrv_get_device_name(bs1));
1118 a672b469 aliguori
            }
1119 a672b469 aliguori
        }
1120 a672b469 aliguori
    }
1121 a672b469 aliguori
1122 a672b469 aliguori
 the_end:
1123 a672b469 aliguori
    if (saved_vm_running)
1124 a672b469 aliguori
        vm_start();
1125 a672b469 aliguori
}
1126 a672b469 aliguori
1127 376253ec aliguori
void do_loadvm(Monitor *mon, const char *name)
1128 a672b469 aliguori
{
1129 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1130 a672b469 aliguori
    BlockDriverInfo bdi1, *bdi = &bdi1;
1131 2d22b18f aliguori
    QEMUSnapshotInfo sn;
1132 a672b469 aliguori
    QEMUFile *f;
1133 a672b469 aliguori
    int i, ret;
1134 a672b469 aliguori
    int saved_vm_running;
1135 a672b469 aliguori
1136 a672b469 aliguori
    bs = get_bs_snapshots();
1137 a672b469 aliguori
    if (!bs) {
1138 376253ec aliguori
        monitor_printf(mon, "No block device supports snapshots\n");
1139 a672b469 aliguori
        return;
1140 a672b469 aliguori
    }
1141 a672b469 aliguori
1142 a672b469 aliguori
    /* Flush all IO requests so they don't interfere with the new state.  */
1143 a672b469 aliguori
    qemu_aio_flush();
1144 a672b469 aliguori
1145 a672b469 aliguori
    saved_vm_running = vm_running;
1146 a672b469 aliguori
    vm_stop(0);
1147 a672b469 aliguori
1148 a672b469 aliguori
    for(i = 0; i <= nb_drives; i++) {
1149 a672b469 aliguori
        bs1 = drives_table[i].bdrv;
1150 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1151 a672b469 aliguori
            ret = bdrv_snapshot_goto(bs1, name);
1152 a672b469 aliguori
            if (ret < 0) {
1153 a672b469 aliguori
                if (bs != bs1)
1154 376253ec aliguori
                    monitor_printf(mon, "Warning: ");
1155 a672b469 aliguori
                switch(ret) {
1156 a672b469 aliguori
                case -ENOTSUP:
1157 376253ec aliguori
                    monitor_printf(mon,
1158 376253ec aliguori
                                   "Snapshots not supported on device '%s'\n",
1159 376253ec aliguori
                                   bdrv_get_device_name(bs1));
1160 a672b469 aliguori
                    break;
1161 a672b469 aliguori
                case -ENOENT:
1162 376253ec aliguori
                    monitor_printf(mon, "Could not find snapshot '%s' on "
1163 376253ec aliguori
                                   "device '%s'\n",
1164 376253ec aliguori
                                   name, bdrv_get_device_name(bs1));
1165 a672b469 aliguori
                    break;
1166 a672b469 aliguori
                default:
1167 376253ec aliguori
                    monitor_printf(mon, "Error %d while activating snapshot on"
1168 376253ec aliguori
                                   " '%s'\n", ret, bdrv_get_device_name(bs1));
1169 a672b469 aliguori
                    break;
1170 a672b469 aliguori
                }
1171 a672b469 aliguori
                /* fatal on snapshot block device */
1172 a672b469 aliguori
                if (bs == bs1)
1173 a672b469 aliguori
                    goto the_end;
1174 a672b469 aliguori
            }
1175 a672b469 aliguori
        }
1176 a672b469 aliguori
    }
1177 a672b469 aliguori
1178 a672b469 aliguori
    if (bdrv_get_info(bs, bdi) < 0 || bdi->vm_state_offset <= 0) {
1179 376253ec aliguori
        monitor_printf(mon, "Device %s does not support VM state snapshots\n",
1180 376253ec aliguori
                       bdrv_get_device_name(bs));
1181 a672b469 aliguori
        return;
1182 a672b469 aliguori
    }
1183 a672b469 aliguori
1184 2d22b18f aliguori
    /* Don't even try to load empty VM states */
1185 2d22b18f aliguori
    ret = bdrv_snapshot_find(bs, &sn, name);
1186 2d22b18f aliguori
    if ((ret >= 0) && (sn.vm_state_size == 0))
1187 2d22b18f aliguori
        goto the_end;
1188 2d22b18f aliguori
1189 a672b469 aliguori
    /* restore the VM state */
1190 a672b469 aliguori
    f = qemu_fopen_bdrv(bs, bdi->vm_state_offset, 0);
1191 a672b469 aliguori
    if (!f) {
1192 376253ec aliguori
        monitor_printf(mon, "Could not open VM state file\n");
1193 a672b469 aliguori
        goto the_end;
1194 a672b469 aliguori
    }
1195 a672b469 aliguori
    ret = qemu_loadvm_state(f);
1196 a672b469 aliguori
    qemu_fclose(f);
1197 a672b469 aliguori
    if (ret < 0) {
1198 376253ec aliguori
        monitor_printf(mon, "Error %d while loading VM state\n", ret);
1199 a672b469 aliguori
    }
1200 a672b469 aliguori
 the_end:
1201 a672b469 aliguori
    if (saved_vm_running)
1202 a672b469 aliguori
        vm_start();
1203 a672b469 aliguori
}
1204 a672b469 aliguori
1205 376253ec aliguori
void do_delvm(Monitor *mon, const char *name)
1206 a672b469 aliguori
{
1207 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1208 a672b469 aliguori
    int i, ret;
1209 a672b469 aliguori
1210 a672b469 aliguori
    bs = get_bs_snapshots();
1211 a672b469 aliguori
    if (!bs) {
1212 376253ec aliguori
        monitor_printf(mon, "No block device supports snapshots\n");
1213 a672b469 aliguori
        return;
1214 a672b469 aliguori
    }
1215 a672b469 aliguori
1216 a672b469 aliguori
    for(i = 0; i <= nb_drives; i++) {
1217 a672b469 aliguori
        bs1 = drives_table[i].bdrv;
1218 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1219 a672b469 aliguori
            ret = bdrv_snapshot_delete(bs1, name);
1220 a672b469 aliguori
            if (ret < 0) {
1221 a672b469 aliguori
                if (ret == -ENOTSUP)
1222 376253ec aliguori
                    monitor_printf(mon,
1223 376253ec aliguori
                                   "Snapshots not supported on device '%s'\n",
1224 376253ec aliguori
                                   bdrv_get_device_name(bs1));
1225 a672b469 aliguori
                else
1226 376253ec aliguori
                    monitor_printf(mon, "Error %d while deleting snapshot on "
1227 376253ec aliguori
                                   "'%s'\n", ret, bdrv_get_device_name(bs1));
1228 a672b469 aliguori
            }
1229 a672b469 aliguori
        }
1230 a672b469 aliguori
    }
1231 a672b469 aliguori
}
1232 a672b469 aliguori
1233 376253ec aliguori
void do_info_snapshots(Monitor *mon)
1234 a672b469 aliguori
{
1235 a672b469 aliguori
    BlockDriverState *bs, *bs1;
1236 a672b469 aliguori
    QEMUSnapshotInfo *sn_tab, *sn;
1237 a672b469 aliguori
    int nb_sns, i;
1238 a672b469 aliguori
    char buf[256];
1239 a672b469 aliguori
1240 a672b469 aliguori
    bs = get_bs_snapshots();
1241 a672b469 aliguori
    if (!bs) {
1242 376253ec aliguori
        monitor_printf(mon, "No available block device supports snapshots\n");
1243 a672b469 aliguori
        return;
1244 a672b469 aliguori
    }
1245 376253ec aliguori
    monitor_printf(mon, "Snapshot devices:");
1246 a672b469 aliguori
    for(i = 0; i <= nb_drives; i++) {
1247 a672b469 aliguori
        bs1 = drives_table[i].bdrv;
1248 a672b469 aliguori
        if (bdrv_has_snapshot(bs1)) {
1249 a672b469 aliguori
            if (bs == bs1)
1250 376253ec aliguori
                monitor_printf(mon, " %s", bdrv_get_device_name(bs1));
1251 a672b469 aliguori
        }
1252 a672b469 aliguori
    }
1253 376253ec aliguori
    monitor_printf(mon, "\n");
1254 a672b469 aliguori
1255 a672b469 aliguori
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1256 a672b469 aliguori
    if (nb_sns < 0) {
1257 376253ec aliguori
        monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
1258 a672b469 aliguori
        return;
1259 a672b469 aliguori
    }
1260 376253ec aliguori
    monitor_printf(mon, "Snapshot list (from %s):\n",
1261 376253ec aliguori
                   bdrv_get_device_name(bs));
1262 376253ec aliguori
    monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
1263 a672b469 aliguori
    for(i = 0; i < nb_sns; i++) {
1264 a672b469 aliguori
        sn = &sn_tab[i];
1265 376253ec aliguori
        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
1266 a672b469 aliguori
    }
1267 a672b469 aliguori
    qemu_free(sn_tab);
1268 a672b469 aliguori
}