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