root / savevm.c @ 29eee86f
History | View | Annotate | Download (62 kB)
1 |
/*
|
---|---|
2 |
* QEMU System Emulator
|
3 |
*
|
4 |
* Copyright (c) 2003-2008 Fabrice Bellard
|
5 |
*
|
6 |
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
7 |
* of this software and associated documentation files (the "Software"), to deal
|
8 |
* in the Software without restriction, including without limitation the rights
|
9 |
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
10 |
* copies of the Software, and to permit persons to whom the Software is
|
11 |
* furnished to do so, subject to the following conditions:
|
12 |
*
|
13 |
* The above copyright notice and this permission notice shall be included in
|
14 |
* all copies or substantial portions of the Software.
|
15 |
*
|
16 |
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
17 |
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
18 |
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
19 |
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
20 |
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
21 |
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
22 |
* THE SOFTWARE.
|
23 |
*/
|
24 |
#include <unistd.h> |
25 |
#include <fcntl.h> |
26 |
#include <time.h> |
27 |
#include <errno.h> |
28 |
#include <sys/time.h> |
29 |
#include <zlib.h> |
30 |
|
31 |
/* Needed early for CONFIG_BSD etc. */
|
32 |
#include "config-host.h" |
33 |
|
34 |
#ifndef _WIN32
|
35 |
#include <sys/times.h> |
36 |
#include <sys/wait.h> |
37 |
#include <termios.h> |
38 |
#include <sys/mman.h> |
39 |
#include <sys/ioctl.h> |
40 |
#include <sys/resource.h> |
41 |
#include <sys/socket.h> |
42 |
#include <netinet/in.h> |
43 |
#include <net/if.h> |
44 |
#include <arpa/inet.h> |
45 |
#include <dirent.h> |
46 |
#include <netdb.h> |
47 |
#include <sys/select.h> |
48 |
#ifdef CONFIG_BSD
|
49 |
#include <sys/stat.h> |
50 |
#if defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || defined(__DragonFly__)
|
51 |
#include <libutil.h> |
52 |
#else
|
53 |
#include <util.h> |
54 |
#endif
|
55 |
#ifdef __linux__
|
56 |
#include <pty.h> |
57 |
#include <malloc.h> |
58 |
#include <linux/rtc.h> |
59 |
#endif
|
60 |
#endif
|
61 |
#endif
|
62 |
|
63 |
#ifdef _WIN32
|
64 |
#include <windows.h> |
65 |
#include <malloc.h> |
66 |
#include <sys/timeb.h> |
67 |
#include <mmsystem.h> |
68 |
#define getopt_long_only getopt_long
|
69 |
#define memalign(align, size) malloc(size)
|
70 |
#endif
|
71 |
|
72 |
#include "qemu-common.h" |
73 |
#include "hw/hw.h" |
74 |
#include "hw/qdev.h" |
75 |
#include "net.h" |
76 |
#include "monitor.h" |
77 |
#include "sysemu.h" |
78 |
#include "qemu-timer.h" |
79 |
#include "qemu-char.h" |
80 |
#include "audio/audio.h" |
81 |
#include "migration.h" |
82 |
#include "qemu_socket.h" |
83 |
#include "qemu-queue.h" |
84 |
#include "qemu-timer.h" |
85 |
#include "cpus.h" |
86 |
#include "memory.h" |
87 |
#include "qmp-commands.h" |
88 |
#include "trace.h" |
89 |
|
90 |
#define SELF_ANNOUNCE_ROUNDS 5 |
91 |
|
92 |
#ifndef ETH_P_RARP
|
93 |
#define ETH_P_RARP 0x8035 |
94 |
#endif
|
95 |
#define ARP_HTYPE_ETH 0x0001 |
96 |
#define ARP_PTYPE_IP 0x0800 |
97 |
#define ARP_OP_REQUEST_REV 0x3 |
98 |
|
99 |
static int announce_self_create(uint8_t *buf, |
100 |
uint8_t *mac_addr) |
101 |
{ |
102 |
/* Ethernet header. */
|
103 |
memset(buf, 0xff, 6); /* destination MAC addr */ |
104 |
memcpy(buf + 6, mac_addr, 6); /* source MAC addr */ |
105 |
*(uint16_t *)(buf + 12) = htons(ETH_P_RARP); /* ethertype */ |
106 |
|
107 |
/* RARP header. */
|
108 |
*(uint16_t *)(buf + 14) = htons(ARP_HTYPE_ETH); /* hardware addr space */ |
109 |
*(uint16_t *)(buf + 16) = htons(ARP_PTYPE_IP); /* protocol addr space */ |
110 |
*(buf + 18) = 6; /* hardware addr length (ethernet) */ |
111 |
*(buf + 19) = 4; /* protocol addr length (IPv4) */ |
112 |
*(uint16_t *)(buf + 20) = htons(ARP_OP_REQUEST_REV); /* opcode */ |
113 |
memcpy(buf + 22, mac_addr, 6); /* source hw addr */ |
114 |
memset(buf + 28, 0x00, 4); /* source protocol addr */ |
115 |
memcpy(buf + 32, mac_addr, 6); /* target hw addr */ |
116 |
memset(buf + 38, 0x00, 4); /* target protocol addr */ |
117 |
|
118 |
/* Padding to get up to 60 bytes (ethernet min packet size, minus FCS). */
|
119 |
memset(buf + 42, 0x00, 18); |
120 |
|
121 |
return 60; /* len (FCS will be added by hardware) */ |
122 |
} |
123 |
|
124 |
static void qemu_announce_self_iter(NICState *nic, void *opaque) |
125 |
{ |
126 |
uint8_t buf[60];
|
127 |
int len;
|
128 |
|
129 |
len = announce_self_create(buf, nic->conf->macaddr.a); |
130 |
|
131 |
qemu_send_packet_raw(&nic->nc, buf, len); |
132 |
} |
133 |
|
134 |
|
135 |
static void qemu_announce_self_once(void *opaque) |
136 |
{ |
137 |
static int count = SELF_ANNOUNCE_ROUNDS; |
138 |
QEMUTimer *timer = *(QEMUTimer **)opaque; |
139 |
|
140 |
qemu_foreach_nic(qemu_announce_self_iter, NULL);
|
141 |
|
142 |
if (--count) {
|
143 |
/* delay 50ms, 150ms, 250ms, ... */
|
144 |
qemu_mod_timer(timer, qemu_get_clock_ms(rt_clock) + |
145 |
50 + (SELF_ANNOUNCE_ROUNDS - count - 1) * 100); |
146 |
} else {
|
147 |
qemu_del_timer(timer); |
148 |
qemu_free_timer(timer); |
149 |
} |
150 |
} |
151 |
|
152 |
void qemu_announce_self(void) |
153 |
{ |
154 |
static QEMUTimer *timer;
|
155 |
timer = qemu_new_timer_ms(rt_clock, qemu_announce_self_once, &timer); |
156 |
qemu_announce_self_once(&timer); |
157 |
} |
158 |
|
159 |
/***********************************************************/
|
160 |
/* savevm/loadvm support */
|
161 |
|
162 |
#define IO_BUF_SIZE 32768 |
163 |
|
164 |
struct QEMUFile {
|
165 |
QEMUFilePutBufferFunc *put_buffer; |
166 |
QEMUFileGetBufferFunc *get_buffer; |
167 |
QEMUFileCloseFunc *close; |
168 |
QEMUFileRateLimit *rate_limit; |
169 |
QEMUFileSetRateLimit *set_rate_limit; |
170 |
QEMUFileGetRateLimit *get_rate_limit; |
171 |
void *opaque;
|
172 |
int is_write;
|
173 |
|
174 |
int64_t buf_offset; /* start of buffer when writing, end of buffer
|
175 |
when reading */
|
176 |
int buf_index;
|
177 |
int buf_size; /* 0 when writing */ |
178 |
uint8_t buf[IO_BUF_SIZE]; |
179 |
|
180 |
int last_error;
|
181 |
}; |
182 |
|
183 |
typedef struct QEMUFileStdio |
184 |
{ |
185 |
FILE *stdio_file; |
186 |
QEMUFile *file; |
187 |
} QEMUFileStdio; |
188 |
|
189 |
typedef struct QEMUFileSocket |
190 |
{ |
191 |
int fd;
|
192 |
QEMUFile *file; |
193 |
} QEMUFileSocket; |
194 |
|
195 |
static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size) |
196 |
{ |
197 |
QEMUFileSocket *s = opaque; |
198 |
ssize_t len; |
199 |
|
200 |
do {
|
201 |
len = qemu_recv(s->fd, buf, size, 0);
|
202 |
} while (len == -1 && socket_error() == EINTR); |
203 |
|
204 |
if (len == -1) |
205 |
len = -socket_error(); |
206 |
|
207 |
return len;
|
208 |
} |
209 |
|
210 |
static int socket_close(void *opaque) |
211 |
{ |
212 |
QEMUFileSocket *s = opaque; |
213 |
g_free(s); |
214 |
return 0; |
215 |
} |
216 |
|
217 |
static int stdio_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size) |
218 |
{ |
219 |
QEMUFileStdio *s = opaque; |
220 |
return fwrite(buf, 1, size, s->stdio_file); |
221 |
} |
222 |
|
223 |
static int stdio_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size) |
224 |
{ |
225 |
QEMUFileStdio *s = opaque; |
226 |
FILE *fp = s->stdio_file; |
227 |
int bytes;
|
228 |
|
229 |
do {
|
230 |
clearerr(fp); |
231 |
bytes = fread(buf, 1, size, fp);
|
232 |
} while ((bytes == 0) && ferror(fp) && (errno == EINTR)); |
233 |
return bytes;
|
234 |
} |
235 |
|
236 |
static int stdio_pclose(void *opaque) |
237 |
{ |
238 |
QEMUFileStdio *s = opaque; |
239 |
int ret;
|
240 |
ret = pclose(s->stdio_file); |
241 |
if (ret == -1) { |
242 |
ret = -errno; |
243 |
} |
244 |
g_free(s); |
245 |
return ret;
|
246 |
} |
247 |
|
248 |
static int stdio_fclose(void *opaque) |
249 |
{ |
250 |
QEMUFileStdio *s = opaque; |
251 |
int ret = 0; |
252 |
if (fclose(s->stdio_file) == EOF) { |
253 |
ret = -errno; |
254 |
} |
255 |
g_free(s); |
256 |
return ret;
|
257 |
} |
258 |
|
259 |
QEMUFile *qemu_popen(FILE *stdio_file, const char *mode) |
260 |
{ |
261 |
QEMUFileStdio *s; |
262 |
|
263 |
if (stdio_file == NULL || mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) { |
264 |
fprintf(stderr, "qemu_popen: Argument validity check failed\n");
|
265 |
return NULL; |
266 |
} |
267 |
|
268 |
s = g_malloc0(sizeof(QEMUFileStdio));
|
269 |
|
270 |
s->stdio_file = stdio_file; |
271 |
|
272 |
if(mode[0] == 'r') { |
273 |
s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_pclose,
|
274 |
NULL, NULL, NULL); |
275 |
} else {
|
276 |
s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_pclose,
|
277 |
NULL, NULL, NULL); |
278 |
} |
279 |
return s->file;
|
280 |
} |
281 |
|
282 |
QEMUFile *qemu_popen_cmd(const char *command, const char *mode) |
283 |
{ |
284 |
FILE *popen_file; |
285 |
|
286 |
popen_file = popen(command, mode); |
287 |
if(popen_file == NULL) { |
288 |
return NULL; |
289 |
} |
290 |
|
291 |
return qemu_popen(popen_file, mode);
|
292 |
} |
293 |
|
294 |
int qemu_stdio_fd(QEMUFile *f)
|
295 |
{ |
296 |
QEMUFileStdio *p; |
297 |
int fd;
|
298 |
|
299 |
p = (QEMUFileStdio *)f->opaque; |
300 |
fd = fileno(p->stdio_file); |
301 |
|
302 |
return fd;
|
303 |
} |
304 |
|
305 |
QEMUFile *qemu_fdopen(int fd, const char *mode) |
306 |
{ |
307 |
QEMUFileStdio *s; |
308 |
|
309 |
if (mode == NULL || |
310 |
(mode[0] != 'r' && mode[0] != 'w') || |
311 |
mode[1] != 'b' || mode[2] != 0) { |
312 |
fprintf(stderr, "qemu_fdopen: Argument validity check failed\n");
|
313 |
return NULL; |
314 |
} |
315 |
|
316 |
s = g_malloc0(sizeof(QEMUFileStdio));
|
317 |
s->stdio_file = fdopen(fd, mode); |
318 |
if (!s->stdio_file)
|
319 |
goto fail;
|
320 |
|
321 |
if(mode[0] == 'r') { |
322 |
s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_fclose,
|
323 |
NULL, NULL, NULL); |
324 |
} else {
|
325 |
s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_fclose,
|
326 |
NULL, NULL, NULL); |
327 |
} |
328 |
return s->file;
|
329 |
|
330 |
fail:
|
331 |
g_free(s); |
332 |
return NULL; |
333 |
} |
334 |
|
335 |
QEMUFile *qemu_fopen_socket(int fd)
|
336 |
{ |
337 |
QEMUFileSocket *s = g_malloc0(sizeof(QEMUFileSocket));
|
338 |
|
339 |
s->fd = fd; |
340 |
s->file = qemu_fopen_ops(s, NULL, socket_get_buffer, socket_close,
|
341 |
NULL, NULL, NULL); |
342 |
return s->file;
|
343 |
} |
344 |
|
345 |
static int file_put_buffer(void *opaque, const uint8_t *buf, |
346 |
int64_t pos, int size)
|
347 |
{ |
348 |
QEMUFileStdio *s = opaque; |
349 |
fseek(s->stdio_file, pos, SEEK_SET); |
350 |
return fwrite(buf, 1, size, s->stdio_file); |
351 |
} |
352 |
|
353 |
static int file_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size) |
354 |
{ |
355 |
QEMUFileStdio *s = opaque; |
356 |
fseek(s->stdio_file, pos, SEEK_SET); |
357 |
return fread(buf, 1, size, s->stdio_file); |
358 |
} |
359 |
|
360 |
QEMUFile *qemu_fopen(const char *filename, const char *mode) |
361 |
{ |
362 |
QEMUFileStdio *s; |
363 |
|
364 |
if (mode == NULL || |
365 |
(mode[0] != 'r' && mode[0] != 'w') || |
366 |
mode[1] != 'b' || mode[2] != 0) { |
367 |
fprintf(stderr, "qemu_fopen: Argument validity check failed\n");
|
368 |
return NULL; |
369 |
} |
370 |
|
371 |
s = g_malloc0(sizeof(QEMUFileStdio));
|
372 |
|
373 |
s->stdio_file = fopen(filename, mode); |
374 |
if (!s->stdio_file)
|
375 |
goto fail;
|
376 |
|
377 |
if(mode[0] == 'w') { |
378 |
s->file = qemu_fopen_ops(s, file_put_buffer, NULL, stdio_fclose,
|
379 |
NULL, NULL, NULL); |
380 |
} else {
|
381 |
s->file = qemu_fopen_ops(s, NULL, file_get_buffer, stdio_fclose,
|
382 |
NULL, NULL, NULL); |
383 |
} |
384 |
return s->file;
|
385 |
fail:
|
386 |
g_free(s); |
387 |
return NULL; |
388 |
} |
389 |
|
390 |
static int block_put_buffer(void *opaque, const uint8_t *buf, |
391 |
int64_t pos, int size)
|
392 |
{ |
393 |
bdrv_save_vmstate(opaque, buf, pos, size); |
394 |
return size;
|
395 |
} |
396 |
|
397 |
static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size) |
398 |
{ |
399 |
return bdrv_load_vmstate(opaque, buf, pos, size);
|
400 |
} |
401 |
|
402 |
static int bdrv_fclose(void *opaque) |
403 |
{ |
404 |
return bdrv_flush(opaque);
|
405 |
} |
406 |
|
407 |
static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int is_writable) |
408 |
{ |
409 |
if (is_writable)
|
410 |
return qemu_fopen_ops(bs, block_put_buffer, NULL, bdrv_fclose, |
411 |
NULL, NULL, NULL); |
412 |
return qemu_fopen_ops(bs, NULL, block_get_buffer, bdrv_fclose, NULL, NULL, NULL); |
413 |
} |
414 |
|
415 |
QEMUFile *qemu_fopen_ops(void *opaque, QEMUFilePutBufferFunc *put_buffer,
|
416 |
QEMUFileGetBufferFunc *get_buffer, |
417 |
QEMUFileCloseFunc *close, |
418 |
QEMUFileRateLimit *rate_limit, |
419 |
QEMUFileSetRateLimit *set_rate_limit, |
420 |
QEMUFileGetRateLimit *get_rate_limit) |
421 |
{ |
422 |
QEMUFile *f; |
423 |
|
424 |
f = g_malloc0(sizeof(QEMUFile));
|
425 |
|
426 |
f->opaque = opaque; |
427 |
f->put_buffer = put_buffer; |
428 |
f->get_buffer = get_buffer; |
429 |
f->close = close; |
430 |
f->rate_limit = rate_limit; |
431 |
f->set_rate_limit = set_rate_limit; |
432 |
f->get_rate_limit = get_rate_limit; |
433 |
f->is_write = 0;
|
434 |
|
435 |
return f;
|
436 |
} |
437 |
|
438 |
int qemu_file_get_error(QEMUFile *f)
|
439 |
{ |
440 |
return f->last_error;
|
441 |
} |
442 |
|
443 |
void qemu_file_set_error(QEMUFile *f, int ret) |
444 |
{ |
445 |
f->last_error = ret; |
446 |
} |
447 |
|
448 |
/** Sets last_error conditionally
|
449 |
*
|
450 |
* Sets last_error only if ret is negative _and_ no error
|
451 |
* was set before.
|
452 |
*/
|
453 |
static void qemu_file_set_if_error(QEMUFile *f, int ret) |
454 |
{ |
455 |
if (ret < 0 && !f->last_error) { |
456 |
qemu_file_set_error(f, ret); |
457 |
} |
458 |
} |
459 |
|
460 |
/** Flushes QEMUFile buffer
|
461 |
*
|
462 |
*/
|
463 |
static int qemu_fflush(QEMUFile *f) |
464 |
{ |
465 |
int ret = 0; |
466 |
|
467 |
if (!f->put_buffer)
|
468 |
return 0; |
469 |
|
470 |
if (f->is_write && f->buf_index > 0) { |
471 |
ret = f->put_buffer(f->opaque, f->buf, f->buf_offset, f->buf_index); |
472 |
if (ret >= 0) { |
473 |
f->buf_offset += f->buf_index; |
474 |
} |
475 |
f->buf_index = 0;
|
476 |
} |
477 |
return ret;
|
478 |
} |
479 |
|
480 |
static void qemu_fill_buffer(QEMUFile *f) |
481 |
{ |
482 |
int len;
|
483 |
int pending;
|
484 |
|
485 |
if (!f->get_buffer)
|
486 |
return;
|
487 |
|
488 |
if (f->is_write)
|
489 |
abort(); |
490 |
|
491 |
pending = f->buf_size - f->buf_index; |
492 |
if (pending > 0) { |
493 |
memmove(f->buf, f->buf + f->buf_index, pending); |
494 |
} |
495 |
f->buf_index = 0;
|
496 |
f->buf_size = pending; |
497 |
|
498 |
len = f->get_buffer(f->opaque, f->buf + pending, f->buf_offset, |
499 |
IO_BUF_SIZE - pending); |
500 |
if (len > 0) { |
501 |
f->buf_size += len; |
502 |
f->buf_offset += len; |
503 |
} else if (len == 0) { |
504 |
f->last_error = -EIO; |
505 |
} else if (len != -EAGAIN) |
506 |
qemu_file_set_error(f, len); |
507 |
} |
508 |
|
509 |
/** Closes the file
|
510 |
*
|
511 |
* Returns negative error value if any error happened on previous operations or
|
512 |
* while closing the file. Returns 0 or positive number on success.
|
513 |
*
|
514 |
* The meaning of return value on success depends on the specific backend
|
515 |
* being used.
|
516 |
*/
|
517 |
int qemu_fclose(QEMUFile *f)
|
518 |
{ |
519 |
int ret;
|
520 |
ret = qemu_fflush(f); |
521 |
|
522 |
if (f->close) {
|
523 |
int ret2 = f->close(f->opaque);
|
524 |
if (ret >= 0) { |
525 |
ret = ret2; |
526 |
} |
527 |
} |
528 |
/* If any error was spotted before closing, we should report it
|
529 |
* instead of the close() return value.
|
530 |
*/
|
531 |
if (f->last_error) {
|
532 |
ret = f->last_error; |
533 |
} |
534 |
g_free(f); |
535 |
return ret;
|
536 |
} |
537 |
|
538 |
void qemu_file_put_notify(QEMUFile *f)
|
539 |
{ |
540 |
f->put_buffer(f->opaque, NULL, 0, 0); |
541 |
} |
542 |
|
543 |
void qemu_put_buffer(QEMUFile *f, const uint8_t *buf, int size) |
544 |
{ |
545 |
int l;
|
546 |
|
547 |
if (!f->last_error && f->is_write == 0 && f->buf_index > 0) { |
548 |
fprintf(stderr, |
549 |
"Attempted to write to buffer while read buffer is not empty\n");
|
550 |
abort(); |
551 |
} |
552 |
|
553 |
while (!f->last_error && size > 0) { |
554 |
l = IO_BUF_SIZE - f->buf_index; |
555 |
if (l > size)
|
556 |
l = size; |
557 |
memcpy(f->buf + f->buf_index, buf, l); |
558 |
f->is_write = 1;
|
559 |
f->buf_index += l; |
560 |
buf += l; |
561 |
size -= l; |
562 |
if (f->buf_index >= IO_BUF_SIZE) {
|
563 |
int ret = qemu_fflush(f);
|
564 |
qemu_file_set_if_error(f, ret); |
565 |
} |
566 |
} |
567 |
} |
568 |
|
569 |
void qemu_put_byte(QEMUFile *f, int v) |
570 |
{ |
571 |
if (!f->last_error && f->is_write == 0 && f->buf_index > 0) { |
572 |
fprintf(stderr, |
573 |
"Attempted to write to buffer while read buffer is not empty\n");
|
574 |
abort(); |
575 |
} |
576 |
|
577 |
f->buf[f->buf_index++] = v; |
578 |
f->is_write = 1;
|
579 |
if (f->buf_index >= IO_BUF_SIZE) {
|
580 |
int ret = qemu_fflush(f);
|
581 |
qemu_file_set_if_error(f, ret); |
582 |
} |
583 |
} |
584 |
|
585 |
static void qemu_file_skip(QEMUFile *f, int size) |
586 |
{ |
587 |
if (f->buf_index + size <= f->buf_size) {
|
588 |
f->buf_index += size; |
589 |
} |
590 |
} |
591 |
|
592 |
static int qemu_peek_buffer(QEMUFile *f, uint8_t *buf, int size, size_t offset) |
593 |
{ |
594 |
int pending;
|
595 |
int index;
|
596 |
|
597 |
if (f->is_write) {
|
598 |
abort(); |
599 |
} |
600 |
|
601 |
index = f->buf_index + offset; |
602 |
pending = f->buf_size - index; |
603 |
if (pending < size) {
|
604 |
qemu_fill_buffer(f); |
605 |
index = f->buf_index + offset; |
606 |
pending = f->buf_size - index; |
607 |
} |
608 |
|
609 |
if (pending <= 0) { |
610 |
return 0; |
611 |
} |
612 |
if (size > pending) {
|
613 |
size = pending; |
614 |
} |
615 |
|
616 |
memcpy(buf, f->buf + index, size); |
617 |
return size;
|
618 |
} |
619 |
|
620 |
int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size) |
621 |
{ |
622 |
int pending = size;
|
623 |
int done = 0; |
624 |
|
625 |
while (pending > 0) { |
626 |
int res;
|
627 |
|
628 |
res = qemu_peek_buffer(f, buf, pending, 0);
|
629 |
if (res == 0) { |
630 |
return done;
|
631 |
} |
632 |
qemu_file_skip(f, res); |
633 |
buf += res; |
634 |
pending -= res; |
635 |
done += res; |
636 |
} |
637 |
return done;
|
638 |
} |
639 |
|
640 |
static int qemu_peek_byte(QEMUFile *f, int offset) |
641 |
{ |
642 |
int index = f->buf_index + offset;
|
643 |
|
644 |
if (f->is_write) {
|
645 |
abort(); |
646 |
} |
647 |
|
648 |
if (index >= f->buf_size) {
|
649 |
qemu_fill_buffer(f); |
650 |
index = f->buf_index + offset; |
651 |
if (index >= f->buf_size) {
|
652 |
return 0; |
653 |
} |
654 |
} |
655 |
return f->buf[index];
|
656 |
} |
657 |
|
658 |
int qemu_get_byte(QEMUFile *f)
|
659 |
{ |
660 |
int result;
|
661 |
|
662 |
result = qemu_peek_byte(f, 0);
|
663 |
qemu_file_skip(f, 1);
|
664 |
return result;
|
665 |
} |
666 |
|
667 |
int64_t qemu_ftell(QEMUFile *f) |
668 |
{ |
669 |
return f->buf_offset - f->buf_size + f->buf_index;
|
670 |
} |
671 |
|
672 |
int qemu_file_rate_limit(QEMUFile *f)
|
673 |
{ |
674 |
if (f->rate_limit)
|
675 |
return f->rate_limit(f->opaque);
|
676 |
|
677 |
return 0; |
678 |
} |
679 |
|
680 |
int64_t qemu_file_get_rate_limit(QEMUFile *f) |
681 |
{ |
682 |
if (f->get_rate_limit)
|
683 |
return f->get_rate_limit(f->opaque);
|
684 |
|
685 |
return 0; |
686 |
} |
687 |
|
688 |
int64_t qemu_file_set_rate_limit(QEMUFile *f, int64_t new_rate) |
689 |
{ |
690 |
/* any failed or completed migration keeps its state to allow probing of
|
691 |
* migration data, but has no associated file anymore */
|
692 |
if (f && f->set_rate_limit)
|
693 |
return f->set_rate_limit(f->opaque, new_rate);
|
694 |
|
695 |
return 0; |
696 |
} |
697 |
|
698 |
void qemu_put_be16(QEMUFile *f, unsigned int v) |
699 |
{ |
700 |
qemu_put_byte(f, v >> 8);
|
701 |
qemu_put_byte(f, v); |
702 |
} |
703 |
|
704 |
void qemu_put_be32(QEMUFile *f, unsigned int v) |
705 |
{ |
706 |
qemu_put_byte(f, v >> 24);
|
707 |
qemu_put_byte(f, v >> 16);
|
708 |
qemu_put_byte(f, v >> 8);
|
709 |
qemu_put_byte(f, v); |
710 |
} |
711 |
|
712 |
void qemu_put_be64(QEMUFile *f, uint64_t v)
|
713 |
{ |
714 |
qemu_put_be32(f, v >> 32);
|
715 |
qemu_put_be32(f, v); |
716 |
} |
717 |
|
718 |
unsigned int qemu_get_be16(QEMUFile *f) |
719 |
{ |
720 |
unsigned int v; |
721 |
v = qemu_get_byte(f) << 8;
|
722 |
v |= qemu_get_byte(f); |
723 |
return v;
|
724 |
} |
725 |
|
726 |
unsigned int qemu_get_be32(QEMUFile *f) |
727 |
{ |
728 |
unsigned int v; |
729 |
v = qemu_get_byte(f) << 24;
|
730 |
v |= qemu_get_byte(f) << 16;
|
731 |
v |= qemu_get_byte(f) << 8;
|
732 |
v |= qemu_get_byte(f); |
733 |
return v;
|
734 |
} |
735 |
|
736 |
uint64_t qemu_get_be64(QEMUFile *f) |
737 |
{ |
738 |
uint64_t v; |
739 |
v = (uint64_t)qemu_get_be32(f) << 32;
|
740 |
v |= qemu_get_be32(f); |
741 |
return v;
|
742 |
} |
743 |
|
744 |
|
745 |
/* timer */
|
746 |
|
747 |
void qemu_put_timer(QEMUFile *f, QEMUTimer *ts)
|
748 |
{ |
749 |
uint64_t expire_time; |
750 |
|
751 |
expire_time = qemu_timer_expire_time_ns(ts); |
752 |
qemu_put_be64(f, expire_time); |
753 |
} |
754 |
|
755 |
void qemu_get_timer(QEMUFile *f, QEMUTimer *ts)
|
756 |
{ |
757 |
uint64_t expire_time; |
758 |
|
759 |
expire_time = qemu_get_be64(f); |
760 |
if (expire_time != -1) { |
761 |
qemu_mod_timer_ns(ts, expire_time); |
762 |
} else {
|
763 |
qemu_del_timer(ts); |
764 |
} |
765 |
} |
766 |
|
767 |
|
768 |
/* bool */
|
769 |
|
770 |
static int get_bool(QEMUFile *f, void *pv, size_t size) |
771 |
{ |
772 |
bool *v = pv;
|
773 |
*v = qemu_get_byte(f); |
774 |
return 0; |
775 |
} |
776 |
|
777 |
static void put_bool(QEMUFile *f, void *pv, size_t size) |
778 |
{ |
779 |
bool *v = pv;
|
780 |
qemu_put_byte(f, *v); |
781 |
} |
782 |
|
783 |
const VMStateInfo vmstate_info_bool = {
|
784 |
.name = "bool",
|
785 |
.get = get_bool, |
786 |
.put = put_bool, |
787 |
}; |
788 |
|
789 |
/* 8 bit int */
|
790 |
|
791 |
static int get_int8(QEMUFile *f, void *pv, size_t size) |
792 |
{ |
793 |
int8_t *v = pv; |
794 |
qemu_get_s8s(f, v); |
795 |
return 0; |
796 |
} |
797 |
|
798 |
static void put_int8(QEMUFile *f, void *pv, size_t size) |
799 |
{ |
800 |
int8_t *v = pv; |
801 |
qemu_put_s8s(f, v); |
802 |
} |
803 |
|
804 |
const VMStateInfo vmstate_info_int8 = {
|
805 |
.name = "int8",
|
806 |
.get = get_int8, |
807 |
.put = put_int8, |
808 |
}; |
809 |
|
810 |
/* 16 bit int */
|
811 |
|
812 |
static int get_int16(QEMUFile *f, void *pv, size_t size) |
813 |
{ |
814 |
int16_t *v = pv; |
815 |
qemu_get_sbe16s(f, v); |
816 |
return 0; |
817 |
} |
818 |
|
819 |
static void put_int16(QEMUFile *f, void *pv, size_t size) |
820 |
{ |
821 |
int16_t *v = pv; |
822 |
qemu_put_sbe16s(f, v); |
823 |
} |
824 |
|
825 |
const VMStateInfo vmstate_info_int16 = {
|
826 |
.name = "int16",
|
827 |
.get = get_int16, |
828 |
.put = put_int16, |
829 |
}; |
830 |
|
831 |
/* 32 bit int */
|
832 |
|
833 |
static int get_int32(QEMUFile *f, void *pv, size_t size) |
834 |
{ |
835 |
int32_t *v = pv; |
836 |
qemu_get_sbe32s(f, v); |
837 |
return 0; |
838 |
} |
839 |
|
840 |
static void put_int32(QEMUFile *f, void *pv, size_t size) |
841 |
{ |
842 |
int32_t *v = pv; |
843 |
qemu_put_sbe32s(f, v); |
844 |
} |
845 |
|
846 |
const VMStateInfo vmstate_info_int32 = {
|
847 |
.name = "int32",
|
848 |
.get = get_int32, |
849 |
.put = put_int32, |
850 |
}; |
851 |
|
852 |
/* 32 bit int. See that the received value is the same than the one
|
853 |
in the field */
|
854 |
|
855 |
static int get_int32_equal(QEMUFile *f, void *pv, size_t size) |
856 |
{ |
857 |
int32_t *v = pv; |
858 |
int32_t v2; |
859 |
qemu_get_sbe32s(f, &v2); |
860 |
|
861 |
if (*v == v2)
|
862 |
return 0; |
863 |
return -EINVAL;
|
864 |
} |
865 |
|
866 |
const VMStateInfo vmstate_info_int32_equal = {
|
867 |
.name = "int32 equal",
|
868 |
.get = get_int32_equal, |
869 |
.put = put_int32, |
870 |
}; |
871 |
|
872 |
/* 32 bit int. See that the received value is the less or the same
|
873 |
than the one in the field */
|
874 |
|
875 |
static int get_int32_le(QEMUFile *f, void *pv, size_t size) |
876 |
{ |
877 |
int32_t *old = pv; |
878 |
int32_t new; |
879 |
qemu_get_sbe32s(f, &new); |
880 |
|
881 |
if (*old <= new)
|
882 |
return 0; |
883 |
return -EINVAL;
|
884 |
} |
885 |
|
886 |
const VMStateInfo vmstate_info_int32_le = {
|
887 |
.name = "int32 equal",
|
888 |
.get = get_int32_le, |
889 |
.put = put_int32, |
890 |
}; |
891 |
|
892 |
/* 64 bit int */
|
893 |
|
894 |
static int get_int64(QEMUFile *f, void *pv, size_t size) |
895 |
{ |
896 |
int64_t *v = pv; |
897 |
qemu_get_sbe64s(f, v); |
898 |
return 0; |
899 |
} |
900 |
|
901 |
static void put_int64(QEMUFile *f, void *pv, size_t size) |
902 |
{ |
903 |
int64_t *v = pv; |
904 |
qemu_put_sbe64s(f, v); |
905 |
} |
906 |
|
907 |
const VMStateInfo vmstate_info_int64 = {
|
908 |
.name = "int64",
|
909 |
.get = get_int64, |
910 |
.put = put_int64, |
911 |
}; |
912 |
|
913 |
/* 8 bit unsigned int */
|
914 |
|
915 |
static int get_uint8(QEMUFile *f, void *pv, size_t size) |
916 |
{ |
917 |
uint8_t *v = pv; |
918 |
qemu_get_8s(f, v); |
919 |
return 0; |
920 |
} |
921 |
|
922 |
static void put_uint8(QEMUFile *f, void *pv, size_t size) |
923 |
{ |
924 |
uint8_t *v = pv; |
925 |
qemu_put_8s(f, v); |
926 |
} |
927 |
|
928 |
const VMStateInfo vmstate_info_uint8 = {
|
929 |
.name = "uint8",
|
930 |
.get = get_uint8, |
931 |
.put = put_uint8, |
932 |
}; |
933 |
|
934 |
/* 16 bit unsigned int */
|
935 |
|
936 |
static int get_uint16(QEMUFile *f, void *pv, size_t size) |
937 |
{ |
938 |
uint16_t *v = pv; |
939 |
qemu_get_be16s(f, v); |
940 |
return 0; |
941 |
} |
942 |
|
943 |
static void put_uint16(QEMUFile *f, void *pv, size_t size) |
944 |
{ |
945 |
uint16_t *v = pv; |
946 |
qemu_put_be16s(f, v); |
947 |
} |
948 |
|
949 |
const VMStateInfo vmstate_info_uint16 = {
|
950 |
.name = "uint16",
|
951 |
.get = get_uint16, |
952 |
.put = put_uint16, |
953 |
}; |
954 |
|
955 |
/* 32 bit unsigned int */
|
956 |
|
957 |
static int get_uint32(QEMUFile *f, void *pv, size_t size) |
958 |
{ |
959 |
uint32_t *v = pv; |
960 |
qemu_get_be32s(f, v); |
961 |
return 0; |
962 |
} |
963 |
|
964 |
static void put_uint32(QEMUFile *f, void *pv, size_t size) |
965 |
{ |
966 |
uint32_t *v = pv; |
967 |
qemu_put_be32s(f, v); |
968 |
} |
969 |
|
970 |
const VMStateInfo vmstate_info_uint32 = {
|
971 |
.name = "uint32",
|
972 |
.get = get_uint32, |
973 |
.put = put_uint32, |
974 |
}; |
975 |
|
976 |
/* 32 bit uint. See that the received value is the same than the one
|
977 |
in the field */
|
978 |
|
979 |
static int get_uint32_equal(QEMUFile *f, void *pv, size_t size) |
980 |
{ |
981 |
uint32_t *v = pv; |
982 |
uint32_t v2; |
983 |
qemu_get_be32s(f, &v2); |
984 |
|
985 |
if (*v == v2) {
|
986 |
return 0; |
987 |
} |
988 |
return -EINVAL;
|
989 |
} |
990 |
|
991 |
const VMStateInfo vmstate_info_uint32_equal = {
|
992 |
.name = "uint32 equal",
|
993 |
.get = get_uint32_equal, |
994 |
.put = put_uint32, |
995 |
}; |
996 |
|
997 |
/* 64 bit unsigned int */
|
998 |
|
999 |
static int get_uint64(QEMUFile *f, void *pv, size_t size) |
1000 |
{ |
1001 |
uint64_t *v = pv; |
1002 |
qemu_get_be64s(f, v); |
1003 |
return 0; |
1004 |
} |
1005 |
|
1006 |
static void put_uint64(QEMUFile *f, void *pv, size_t size) |
1007 |
{ |
1008 |
uint64_t *v = pv; |
1009 |
qemu_put_be64s(f, v); |
1010 |
} |
1011 |
|
1012 |
const VMStateInfo vmstate_info_uint64 = {
|
1013 |
.name = "uint64",
|
1014 |
.get = get_uint64, |
1015 |
.put = put_uint64, |
1016 |
}; |
1017 |
|
1018 |
/* 8 bit int. See that the received value is the same than the one
|
1019 |
in the field */
|
1020 |
|
1021 |
static int get_uint8_equal(QEMUFile *f, void *pv, size_t size) |
1022 |
{ |
1023 |
uint8_t *v = pv; |
1024 |
uint8_t v2; |
1025 |
qemu_get_8s(f, &v2); |
1026 |
|
1027 |
if (*v == v2)
|
1028 |
return 0; |
1029 |
return -EINVAL;
|
1030 |
} |
1031 |
|
1032 |
const VMStateInfo vmstate_info_uint8_equal = {
|
1033 |
.name = "uint8 equal",
|
1034 |
.get = get_uint8_equal, |
1035 |
.put = put_uint8, |
1036 |
}; |
1037 |
|
1038 |
/* 16 bit unsigned int int. See that the received value is the same than the one
|
1039 |
in the field */
|
1040 |
|
1041 |
static int get_uint16_equal(QEMUFile *f, void *pv, size_t size) |
1042 |
{ |
1043 |
uint16_t *v = pv; |
1044 |
uint16_t v2; |
1045 |
qemu_get_be16s(f, &v2); |
1046 |
|
1047 |
if (*v == v2)
|
1048 |
return 0; |
1049 |
return -EINVAL;
|
1050 |
} |
1051 |
|
1052 |
const VMStateInfo vmstate_info_uint16_equal = {
|
1053 |
.name = "uint16 equal",
|
1054 |
.get = get_uint16_equal, |
1055 |
.put = put_uint16, |
1056 |
}; |
1057 |
|
1058 |
/* timers */
|
1059 |
|
1060 |
static int get_timer(QEMUFile *f, void *pv, size_t size) |
1061 |
{ |
1062 |
QEMUTimer *v = pv; |
1063 |
qemu_get_timer(f, v); |
1064 |
return 0; |
1065 |
} |
1066 |
|
1067 |
static void put_timer(QEMUFile *f, void *pv, size_t size) |
1068 |
{ |
1069 |
QEMUTimer *v = pv; |
1070 |
qemu_put_timer(f, v); |
1071 |
} |
1072 |
|
1073 |
const VMStateInfo vmstate_info_timer = {
|
1074 |
.name = "timer",
|
1075 |
.get = get_timer, |
1076 |
.put = put_timer, |
1077 |
}; |
1078 |
|
1079 |
/* uint8_t buffers */
|
1080 |
|
1081 |
static int get_buffer(QEMUFile *f, void *pv, size_t size) |
1082 |
{ |
1083 |
uint8_t *v = pv; |
1084 |
qemu_get_buffer(f, v, size); |
1085 |
return 0; |
1086 |
} |
1087 |
|
1088 |
static void put_buffer(QEMUFile *f, void *pv, size_t size) |
1089 |
{ |
1090 |
uint8_t *v = pv; |
1091 |
qemu_put_buffer(f, v, size); |
1092 |
} |
1093 |
|
1094 |
const VMStateInfo vmstate_info_buffer = {
|
1095 |
.name = "buffer",
|
1096 |
.get = get_buffer, |
1097 |
.put = put_buffer, |
1098 |
}; |
1099 |
|
1100 |
/* unused buffers: space that was used for some fields that are
|
1101 |
not useful anymore */
|
1102 |
|
1103 |
static int get_unused_buffer(QEMUFile *f, void *pv, size_t size) |
1104 |
{ |
1105 |
uint8_t buf[1024];
|
1106 |
int block_len;
|
1107 |
|
1108 |
while (size > 0) { |
1109 |
block_len = MIN(sizeof(buf), size);
|
1110 |
size -= block_len; |
1111 |
qemu_get_buffer(f, buf, block_len); |
1112 |
} |
1113 |
return 0; |
1114 |
} |
1115 |
|
1116 |
static void put_unused_buffer(QEMUFile *f, void *pv, size_t size) |
1117 |
{ |
1118 |
static const uint8_t buf[1024]; |
1119 |
int block_len;
|
1120 |
|
1121 |
while (size > 0) { |
1122 |
block_len = MIN(sizeof(buf), size);
|
1123 |
size -= block_len; |
1124 |
qemu_put_buffer(f, buf, block_len); |
1125 |
} |
1126 |
} |
1127 |
|
1128 |
const VMStateInfo vmstate_info_unused_buffer = {
|
1129 |
.name = "unused_buffer",
|
1130 |
.get = get_unused_buffer, |
1131 |
.put = put_unused_buffer, |
1132 |
}; |
1133 |
|
1134 |
typedef struct CompatEntry { |
1135 |
char idstr[256]; |
1136 |
int instance_id;
|
1137 |
} CompatEntry; |
1138 |
|
1139 |
typedef struct SaveStateEntry { |
1140 |
QTAILQ_ENTRY(SaveStateEntry) entry; |
1141 |
char idstr[256]; |
1142 |
int instance_id;
|
1143 |
int alias_id;
|
1144 |
int version_id;
|
1145 |
int section_id;
|
1146 |
SaveVMHandlers *ops; |
1147 |
const VMStateDescription *vmsd;
|
1148 |
void *opaque;
|
1149 |
CompatEntry *compat; |
1150 |
int no_migrate;
|
1151 |
int is_ram;
|
1152 |
} SaveStateEntry; |
1153 |
|
1154 |
|
1155 |
static QTAILQ_HEAD(savevm_handlers, SaveStateEntry) savevm_handlers =
|
1156 |
QTAILQ_HEAD_INITIALIZER(savevm_handlers); |
1157 |
static int global_section_id; |
1158 |
|
1159 |
static int calculate_new_instance_id(const char *idstr) |
1160 |
{ |
1161 |
SaveStateEntry *se; |
1162 |
int instance_id = 0; |
1163 |
|
1164 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1165 |
if (strcmp(idstr, se->idstr) == 0 |
1166 |
&& instance_id <= se->instance_id) { |
1167 |
instance_id = se->instance_id + 1;
|
1168 |
} |
1169 |
} |
1170 |
return instance_id;
|
1171 |
} |
1172 |
|
1173 |
static int calculate_compat_instance_id(const char *idstr) |
1174 |
{ |
1175 |
SaveStateEntry *se; |
1176 |
int instance_id = 0; |
1177 |
|
1178 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1179 |
if (!se->compat)
|
1180 |
continue;
|
1181 |
|
1182 |
if (strcmp(idstr, se->compat->idstr) == 0 |
1183 |
&& instance_id <= se->compat->instance_id) { |
1184 |
instance_id = se->compat->instance_id + 1;
|
1185 |
} |
1186 |
} |
1187 |
return instance_id;
|
1188 |
} |
1189 |
|
1190 |
/* TODO: Individual devices generally have very little idea about the rest
|
1191 |
of the system, so instance_id should be removed/replaced.
|
1192 |
Meanwhile pass -1 as instance_id if you do not already have a clearly
|
1193 |
distinguishing id for all instances of your device class. */
|
1194 |
int register_savevm_live(DeviceState *dev,
|
1195 |
const char *idstr, |
1196 |
int instance_id,
|
1197 |
int version_id,
|
1198 |
SaveVMHandlers *ops, |
1199 |
void *opaque)
|
1200 |
{ |
1201 |
SaveStateEntry *se; |
1202 |
|
1203 |
se = g_malloc0(sizeof(SaveStateEntry));
|
1204 |
se->version_id = version_id; |
1205 |
se->section_id = global_section_id++; |
1206 |
se->ops = ops; |
1207 |
se->opaque = opaque; |
1208 |
se->vmsd = NULL;
|
1209 |
se->no_migrate = 0;
|
1210 |
/* if this is a live_savem then set is_ram */
|
1211 |
if (ops->save_live_setup != NULL) { |
1212 |
se->is_ram = 1;
|
1213 |
} |
1214 |
|
1215 |
if (dev) {
|
1216 |
char *id = qdev_get_dev_path(dev);
|
1217 |
if (id) {
|
1218 |
pstrcpy(se->idstr, sizeof(se->idstr), id);
|
1219 |
pstrcat(se->idstr, sizeof(se->idstr), "/"); |
1220 |
g_free(id); |
1221 |
|
1222 |
se->compat = g_malloc0(sizeof(CompatEntry));
|
1223 |
pstrcpy(se->compat->idstr, sizeof(se->compat->idstr), idstr);
|
1224 |
se->compat->instance_id = instance_id == -1 ?
|
1225 |
calculate_compat_instance_id(idstr) : instance_id; |
1226 |
instance_id = -1;
|
1227 |
} |
1228 |
} |
1229 |
pstrcat(se->idstr, sizeof(se->idstr), idstr);
|
1230 |
|
1231 |
if (instance_id == -1) { |
1232 |
se->instance_id = calculate_new_instance_id(se->idstr); |
1233 |
} else {
|
1234 |
se->instance_id = instance_id; |
1235 |
} |
1236 |
assert(!se->compat || se->instance_id == 0);
|
1237 |
/* add at the end of list */
|
1238 |
QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry); |
1239 |
return 0; |
1240 |
} |
1241 |
|
1242 |
int register_savevm(DeviceState *dev,
|
1243 |
const char *idstr, |
1244 |
int instance_id,
|
1245 |
int version_id,
|
1246 |
SaveStateHandler *save_state, |
1247 |
LoadStateHandler *load_state, |
1248 |
void *opaque)
|
1249 |
{ |
1250 |
SaveVMHandlers *ops = g_malloc0(sizeof(SaveVMHandlers));
|
1251 |
ops->save_state = save_state; |
1252 |
ops->load_state = load_state; |
1253 |
return register_savevm_live(dev, idstr, instance_id, version_id,
|
1254 |
ops, opaque); |
1255 |
} |
1256 |
|
1257 |
void unregister_savevm(DeviceState *dev, const char *idstr, void *opaque) |
1258 |
{ |
1259 |
SaveStateEntry *se, *new_se; |
1260 |
char id[256] = ""; |
1261 |
|
1262 |
if (dev) {
|
1263 |
char *path = qdev_get_dev_path(dev);
|
1264 |
if (path) {
|
1265 |
pstrcpy(id, sizeof(id), path);
|
1266 |
pstrcat(id, sizeof(id), "/"); |
1267 |
g_free(path); |
1268 |
} |
1269 |
} |
1270 |
pstrcat(id, sizeof(id), idstr);
|
1271 |
|
1272 |
QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) { |
1273 |
if (strcmp(se->idstr, id) == 0 && se->opaque == opaque) { |
1274 |
QTAILQ_REMOVE(&savevm_handlers, se, entry); |
1275 |
if (se->compat) {
|
1276 |
g_free(se->compat); |
1277 |
} |
1278 |
g_free(se->ops); |
1279 |
g_free(se); |
1280 |
} |
1281 |
} |
1282 |
} |
1283 |
|
1284 |
int vmstate_register_with_alias_id(DeviceState *dev, int instance_id, |
1285 |
const VMStateDescription *vmsd,
|
1286 |
void *opaque, int alias_id, |
1287 |
int required_for_version)
|
1288 |
{ |
1289 |
SaveStateEntry *se; |
1290 |
|
1291 |
/* If this triggers, alias support can be dropped for the vmsd. */
|
1292 |
assert(alias_id == -1 || required_for_version >= vmsd->minimum_version_id);
|
1293 |
|
1294 |
se = g_malloc0(sizeof(SaveStateEntry));
|
1295 |
se->version_id = vmsd->version_id; |
1296 |
se->section_id = global_section_id++; |
1297 |
se->opaque = opaque; |
1298 |
se->vmsd = vmsd; |
1299 |
se->alias_id = alias_id; |
1300 |
se->no_migrate = vmsd->unmigratable; |
1301 |
|
1302 |
if (dev) {
|
1303 |
char *id = qdev_get_dev_path(dev);
|
1304 |
if (id) {
|
1305 |
pstrcpy(se->idstr, sizeof(se->idstr), id);
|
1306 |
pstrcat(se->idstr, sizeof(se->idstr), "/"); |
1307 |
g_free(id); |
1308 |
|
1309 |
se->compat = g_malloc0(sizeof(CompatEntry));
|
1310 |
pstrcpy(se->compat->idstr, sizeof(se->compat->idstr), vmsd->name);
|
1311 |
se->compat->instance_id = instance_id == -1 ?
|
1312 |
calculate_compat_instance_id(vmsd->name) : instance_id; |
1313 |
instance_id = -1;
|
1314 |
} |
1315 |
} |
1316 |
pstrcat(se->idstr, sizeof(se->idstr), vmsd->name);
|
1317 |
|
1318 |
if (instance_id == -1) { |
1319 |
se->instance_id = calculate_new_instance_id(se->idstr); |
1320 |
} else {
|
1321 |
se->instance_id = instance_id; |
1322 |
} |
1323 |
assert(!se->compat || se->instance_id == 0);
|
1324 |
/* add at the end of list */
|
1325 |
QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry); |
1326 |
return 0; |
1327 |
} |
1328 |
|
1329 |
int vmstate_register(DeviceState *dev, int instance_id, |
1330 |
const VMStateDescription *vmsd, void *opaque) |
1331 |
{ |
1332 |
return vmstate_register_with_alias_id(dev, instance_id, vmsd,
|
1333 |
opaque, -1, 0); |
1334 |
} |
1335 |
|
1336 |
void vmstate_unregister(DeviceState *dev, const VMStateDescription *vmsd, |
1337 |
void *opaque)
|
1338 |
{ |
1339 |
SaveStateEntry *se, *new_se; |
1340 |
|
1341 |
QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) { |
1342 |
if (se->vmsd == vmsd && se->opaque == opaque) {
|
1343 |
QTAILQ_REMOVE(&savevm_handlers, se, entry); |
1344 |
if (se->compat) {
|
1345 |
g_free(se->compat); |
1346 |
} |
1347 |
g_free(se); |
1348 |
} |
1349 |
} |
1350 |
} |
1351 |
|
1352 |
static void vmstate_subsection_save(QEMUFile *f, const VMStateDescription *vmsd, |
1353 |
void *opaque);
|
1354 |
static int vmstate_subsection_load(QEMUFile *f, const VMStateDescription *vmsd, |
1355 |
void *opaque);
|
1356 |
|
1357 |
int vmstate_load_state(QEMUFile *f, const VMStateDescription *vmsd, |
1358 |
void *opaque, int version_id) |
1359 |
{ |
1360 |
VMStateField *field = vmsd->fields; |
1361 |
int ret;
|
1362 |
|
1363 |
if (version_id > vmsd->version_id) {
|
1364 |
return -EINVAL;
|
1365 |
} |
1366 |
if (version_id < vmsd->minimum_version_id_old) {
|
1367 |
return -EINVAL;
|
1368 |
} |
1369 |
if (version_id < vmsd->minimum_version_id) {
|
1370 |
return vmsd->load_state_old(f, opaque, version_id);
|
1371 |
} |
1372 |
if (vmsd->pre_load) {
|
1373 |
int ret = vmsd->pre_load(opaque);
|
1374 |
if (ret)
|
1375 |
return ret;
|
1376 |
} |
1377 |
while(field->name) {
|
1378 |
if ((field->field_exists &&
|
1379 |
field->field_exists(opaque, version_id)) || |
1380 |
(!field->field_exists && |
1381 |
field->version_id <= version_id)) { |
1382 |
void *base_addr = opaque + field->offset;
|
1383 |
int i, n_elems = 1; |
1384 |
int size = field->size;
|
1385 |
|
1386 |
if (field->flags & VMS_VBUFFER) {
|
1387 |
size = *(int32_t *)(opaque+field->size_offset); |
1388 |
if (field->flags & VMS_MULTIPLY) {
|
1389 |
size *= field->size; |
1390 |
} |
1391 |
} |
1392 |
if (field->flags & VMS_ARRAY) {
|
1393 |
n_elems = field->num; |
1394 |
} else if (field->flags & VMS_VARRAY_INT32) { |
1395 |
n_elems = *(int32_t *)(opaque+field->num_offset); |
1396 |
} else if (field->flags & VMS_VARRAY_UINT32) { |
1397 |
n_elems = *(uint32_t *)(opaque+field->num_offset); |
1398 |
} else if (field->flags & VMS_VARRAY_UINT16) { |
1399 |
n_elems = *(uint16_t *)(opaque+field->num_offset); |
1400 |
} else if (field->flags & VMS_VARRAY_UINT8) { |
1401 |
n_elems = *(uint8_t *)(opaque+field->num_offset); |
1402 |
} |
1403 |
if (field->flags & VMS_POINTER) {
|
1404 |
base_addr = *(void **)base_addr + field->start;
|
1405 |
} |
1406 |
for (i = 0; i < n_elems; i++) { |
1407 |
void *addr = base_addr + size * i;
|
1408 |
|
1409 |
if (field->flags & VMS_ARRAY_OF_POINTER) {
|
1410 |
addr = *(void **)addr;
|
1411 |
} |
1412 |
if (field->flags & VMS_STRUCT) {
|
1413 |
ret = vmstate_load_state(f, field->vmsd, addr, field->vmsd->version_id); |
1414 |
} else {
|
1415 |
ret = field->info->get(f, addr, size); |
1416 |
|
1417 |
} |
1418 |
if (ret < 0) { |
1419 |
return ret;
|
1420 |
} |
1421 |
} |
1422 |
} |
1423 |
field++; |
1424 |
} |
1425 |
ret = vmstate_subsection_load(f, vmsd, opaque); |
1426 |
if (ret != 0) { |
1427 |
return ret;
|
1428 |
} |
1429 |
if (vmsd->post_load) {
|
1430 |
return vmsd->post_load(opaque, version_id);
|
1431 |
} |
1432 |
return 0; |
1433 |
} |
1434 |
|
1435 |
void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd, |
1436 |
void *opaque)
|
1437 |
{ |
1438 |
VMStateField *field = vmsd->fields; |
1439 |
|
1440 |
if (vmsd->pre_save) {
|
1441 |
vmsd->pre_save(opaque); |
1442 |
} |
1443 |
while(field->name) {
|
1444 |
if (!field->field_exists ||
|
1445 |
field->field_exists(opaque, vmsd->version_id)) { |
1446 |
void *base_addr = opaque + field->offset;
|
1447 |
int i, n_elems = 1; |
1448 |
int size = field->size;
|
1449 |
|
1450 |
if (field->flags & VMS_VBUFFER) {
|
1451 |
size = *(int32_t *)(opaque+field->size_offset); |
1452 |
if (field->flags & VMS_MULTIPLY) {
|
1453 |
size *= field->size; |
1454 |
} |
1455 |
} |
1456 |
if (field->flags & VMS_ARRAY) {
|
1457 |
n_elems = field->num; |
1458 |
} else if (field->flags & VMS_VARRAY_INT32) { |
1459 |
n_elems = *(int32_t *)(opaque+field->num_offset); |
1460 |
} else if (field->flags & VMS_VARRAY_UINT32) { |
1461 |
n_elems = *(uint32_t *)(opaque+field->num_offset); |
1462 |
} else if (field->flags & VMS_VARRAY_UINT16) { |
1463 |
n_elems = *(uint16_t *)(opaque+field->num_offset); |
1464 |
} else if (field->flags & VMS_VARRAY_UINT8) { |
1465 |
n_elems = *(uint8_t *)(opaque+field->num_offset); |
1466 |
} |
1467 |
if (field->flags & VMS_POINTER) {
|
1468 |
base_addr = *(void **)base_addr + field->start;
|
1469 |
} |
1470 |
for (i = 0; i < n_elems; i++) { |
1471 |
void *addr = base_addr + size * i;
|
1472 |
|
1473 |
if (field->flags & VMS_ARRAY_OF_POINTER) {
|
1474 |
addr = *(void **)addr;
|
1475 |
} |
1476 |
if (field->flags & VMS_STRUCT) {
|
1477 |
vmstate_save_state(f, field->vmsd, addr); |
1478 |
} else {
|
1479 |
field->info->put(f, addr, size); |
1480 |
} |
1481 |
} |
1482 |
} |
1483 |
field++; |
1484 |
} |
1485 |
vmstate_subsection_save(f, vmsd, opaque); |
1486 |
} |
1487 |
|
1488 |
static int vmstate_load(QEMUFile *f, SaveStateEntry *se, int version_id) |
1489 |
{ |
1490 |
if (!se->vmsd) { /* Old style */ |
1491 |
return se->ops->load_state(f, se->opaque, version_id);
|
1492 |
} |
1493 |
return vmstate_load_state(f, se->vmsd, se->opaque, version_id);
|
1494 |
} |
1495 |
|
1496 |
static void vmstate_save(QEMUFile *f, SaveStateEntry *se) |
1497 |
{ |
1498 |
if (!se->vmsd) { /* Old style */ |
1499 |
se->ops->save_state(f, se->opaque); |
1500 |
return;
|
1501 |
} |
1502 |
vmstate_save_state(f,se->vmsd, se->opaque); |
1503 |
} |
1504 |
|
1505 |
#define QEMU_VM_FILE_MAGIC 0x5145564d |
1506 |
#define QEMU_VM_FILE_VERSION_COMPAT 0x00000002 |
1507 |
#define QEMU_VM_FILE_VERSION 0x00000003 |
1508 |
|
1509 |
#define QEMU_VM_EOF 0x00 |
1510 |
#define QEMU_VM_SECTION_START 0x01 |
1511 |
#define QEMU_VM_SECTION_PART 0x02 |
1512 |
#define QEMU_VM_SECTION_END 0x03 |
1513 |
#define QEMU_VM_SECTION_FULL 0x04 |
1514 |
#define QEMU_VM_SUBSECTION 0x05 |
1515 |
|
1516 |
bool qemu_savevm_state_blocked(Error **errp)
|
1517 |
{ |
1518 |
SaveStateEntry *se; |
1519 |
|
1520 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1521 |
if (se->no_migrate) {
|
1522 |
error_set(errp, QERR_MIGRATION_NOT_SUPPORTED, se->idstr); |
1523 |
return true; |
1524 |
} |
1525 |
} |
1526 |
return false; |
1527 |
} |
1528 |
|
1529 |
int qemu_savevm_state_begin(QEMUFile *f,
|
1530 |
const MigrationParams *params)
|
1531 |
{ |
1532 |
SaveStateEntry *se; |
1533 |
int ret;
|
1534 |
|
1535 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1536 |
if (!se->ops || !se->ops->set_params) {
|
1537 |
continue;
|
1538 |
} |
1539 |
se->ops->set_params(params, se->opaque); |
1540 |
} |
1541 |
|
1542 |
qemu_put_be32(f, QEMU_VM_FILE_MAGIC); |
1543 |
qemu_put_be32(f, QEMU_VM_FILE_VERSION); |
1544 |
|
1545 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1546 |
int len;
|
1547 |
|
1548 |
if (!se->ops || !se->ops->save_live_setup) {
|
1549 |
continue;
|
1550 |
} |
1551 |
if (se->ops && se->ops->is_active) {
|
1552 |
if (!se->ops->is_active(se->opaque)) {
|
1553 |
continue;
|
1554 |
} |
1555 |
} |
1556 |
/* Section type */
|
1557 |
qemu_put_byte(f, QEMU_VM_SECTION_START); |
1558 |
qemu_put_be32(f, se->section_id); |
1559 |
|
1560 |
/* ID string */
|
1561 |
len = strlen(se->idstr); |
1562 |
qemu_put_byte(f, len); |
1563 |
qemu_put_buffer(f, (uint8_t *)se->idstr, len); |
1564 |
|
1565 |
qemu_put_be32(f, se->instance_id); |
1566 |
qemu_put_be32(f, se->version_id); |
1567 |
|
1568 |
ret = se->ops->save_live_setup(f, se->opaque); |
1569 |
if (ret < 0) { |
1570 |
qemu_savevm_state_cancel(f); |
1571 |
return ret;
|
1572 |
} |
1573 |
} |
1574 |
ret = qemu_file_get_error(f); |
1575 |
if (ret != 0) { |
1576 |
qemu_savevm_state_cancel(f); |
1577 |
} |
1578 |
|
1579 |
return ret;
|
1580 |
|
1581 |
} |
1582 |
|
1583 |
/*
|
1584 |
* this function has three return values:
|
1585 |
* negative: there was one error, and we have -errno.
|
1586 |
* 0 : We haven't finished, caller have to go again
|
1587 |
* 1 : We have finished, we can go to complete phase
|
1588 |
*/
|
1589 |
int qemu_savevm_state_iterate(QEMUFile *f)
|
1590 |
{ |
1591 |
SaveStateEntry *se; |
1592 |
int ret = 1; |
1593 |
|
1594 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1595 |
if (!se->ops || !se->ops->save_live_iterate) {
|
1596 |
continue;
|
1597 |
} |
1598 |
if (se->ops && se->ops->is_active) {
|
1599 |
if (!se->ops->is_active(se->opaque)) {
|
1600 |
continue;
|
1601 |
} |
1602 |
} |
1603 |
if (qemu_file_rate_limit(f)) {
|
1604 |
return 0; |
1605 |
} |
1606 |
trace_savevm_section_start(); |
1607 |
/* Section type */
|
1608 |
qemu_put_byte(f, QEMU_VM_SECTION_PART); |
1609 |
qemu_put_be32(f, se->section_id); |
1610 |
|
1611 |
ret = se->ops->save_live_iterate(f, se->opaque); |
1612 |
trace_savevm_section_end(se->section_id); |
1613 |
|
1614 |
if (ret <= 0) { |
1615 |
/* Do not proceed to the next vmstate before this one reported
|
1616 |
completion of the current stage. This serializes the migration
|
1617 |
and reduces the probability that a faster changing state is
|
1618 |
synchronized over and over again. */
|
1619 |
break;
|
1620 |
} |
1621 |
} |
1622 |
if (ret != 0) { |
1623 |
return ret;
|
1624 |
} |
1625 |
ret = qemu_file_get_error(f); |
1626 |
if (ret != 0) { |
1627 |
qemu_savevm_state_cancel(f); |
1628 |
} |
1629 |
return ret;
|
1630 |
} |
1631 |
|
1632 |
int qemu_savevm_state_complete(QEMUFile *f)
|
1633 |
{ |
1634 |
SaveStateEntry *se; |
1635 |
int ret;
|
1636 |
|
1637 |
cpu_synchronize_all_states(); |
1638 |
|
1639 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1640 |
if (!se->ops || !se->ops->save_live_complete) {
|
1641 |
continue;
|
1642 |
} |
1643 |
if (se->ops && se->ops->is_active) {
|
1644 |
if (!se->ops->is_active(se->opaque)) {
|
1645 |
continue;
|
1646 |
} |
1647 |
} |
1648 |
trace_savevm_section_start(); |
1649 |
/* Section type */
|
1650 |
qemu_put_byte(f, QEMU_VM_SECTION_END); |
1651 |
qemu_put_be32(f, se->section_id); |
1652 |
|
1653 |
ret = se->ops->save_live_complete(f, se->opaque); |
1654 |
trace_savevm_section_end(se->section_id); |
1655 |
if (ret < 0) { |
1656 |
return ret;
|
1657 |
} |
1658 |
} |
1659 |
|
1660 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1661 |
int len;
|
1662 |
|
1663 |
if ((!se->ops || !se->ops->save_state) && !se->vmsd) {
|
1664 |
continue;
|
1665 |
} |
1666 |
trace_savevm_section_start(); |
1667 |
/* Section type */
|
1668 |
qemu_put_byte(f, QEMU_VM_SECTION_FULL); |
1669 |
qemu_put_be32(f, se->section_id); |
1670 |
|
1671 |
/* ID string */
|
1672 |
len = strlen(se->idstr); |
1673 |
qemu_put_byte(f, len); |
1674 |
qemu_put_buffer(f, (uint8_t *)se->idstr, len); |
1675 |
|
1676 |
qemu_put_be32(f, se->instance_id); |
1677 |
qemu_put_be32(f, se->version_id); |
1678 |
|
1679 |
vmstate_save(f, se); |
1680 |
trace_savevm_section_end(se->section_id); |
1681 |
} |
1682 |
|
1683 |
qemu_put_byte(f, QEMU_VM_EOF); |
1684 |
|
1685 |
return qemu_file_get_error(f);
|
1686 |
} |
1687 |
|
1688 |
void qemu_savevm_state_cancel(QEMUFile *f)
|
1689 |
{ |
1690 |
SaveStateEntry *se; |
1691 |
|
1692 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1693 |
if (se->ops && se->ops->cancel) {
|
1694 |
se->ops->cancel(se->opaque); |
1695 |
} |
1696 |
} |
1697 |
} |
1698 |
|
1699 |
static int qemu_savevm_state(QEMUFile *f) |
1700 |
{ |
1701 |
int ret;
|
1702 |
MigrationParams params = { |
1703 |
.blk = 0,
|
1704 |
.shared = 0
|
1705 |
}; |
1706 |
|
1707 |
if (qemu_savevm_state_blocked(NULL)) { |
1708 |
ret = -EINVAL; |
1709 |
goto out;
|
1710 |
} |
1711 |
|
1712 |
ret = qemu_savevm_state_begin(f, ¶ms); |
1713 |
if (ret < 0) |
1714 |
goto out;
|
1715 |
|
1716 |
do {
|
1717 |
ret = qemu_savevm_state_iterate(f); |
1718 |
if (ret < 0) |
1719 |
goto out;
|
1720 |
} while (ret == 0); |
1721 |
|
1722 |
ret = qemu_savevm_state_complete(f); |
1723 |
|
1724 |
out:
|
1725 |
if (ret == 0) { |
1726 |
ret = qemu_file_get_error(f); |
1727 |
} |
1728 |
|
1729 |
return ret;
|
1730 |
} |
1731 |
|
1732 |
static int qemu_save_device_state(QEMUFile *f) |
1733 |
{ |
1734 |
SaveStateEntry *se; |
1735 |
|
1736 |
qemu_put_be32(f, QEMU_VM_FILE_MAGIC); |
1737 |
qemu_put_be32(f, QEMU_VM_FILE_VERSION); |
1738 |
|
1739 |
cpu_synchronize_all_states(); |
1740 |
|
1741 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1742 |
int len;
|
1743 |
|
1744 |
if (se->is_ram) {
|
1745 |
continue;
|
1746 |
} |
1747 |
if ((!se->ops || !se->ops->save_state) && !se->vmsd) {
|
1748 |
continue;
|
1749 |
} |
1750 |
|
1751 |
/* Section type */
|
1752 |
qemu_put_byte(f, QEMU_VM_SECTION_FULL); |
1753 |
qemu_put_be32(f, se->section_id); |
1754 |
|
1755 |
/* ID string */
|
1756 |
len = strlen(se->idstr); |
1757 |
qemu_put_byte(f, len); |
1758 |
qemu_put_buffer(f, (uint8_t *)se->idstr, len); |
1759 |
|
1760 |
qemu_put_be32(f, se->instance_id); |
1761 |
qemu_put_be32(f, se->version_id); |
1762 |
|
1763 |
vmstate_save(f, se); |
1764 |
} |
1765 |
|
1766 |
qemu_put_byte(f, QEMU_VM_EOF); |
1767 |
|
1768 |
return qemu_file_get_error(f);
|
1769 |
} |
1770 |
|
1771 |
static SaveStateEntry *find_se(const char *idstr, int instance_id) |
1772 |
{ |
1773 |
SaveStateEntry *se; |
1774 |
|
1775 |
QTAILQ_FOREACH(se, &savevm_handlers, entry) { |
1776 |
if (!strcmp(se->idstr, idstr) &&
|
1777 |
(instance_id == se->instance_id || |
1778 |
instance_id == se->alias_id)) |
1779 |
return se;
|
1780 |
/* Migrating from an older version? */
|
1781 |
if (strstr(se->idstr, idstr) && se->compat) {
|
1782 |
if (!strcmp(se->compat->idstr, idstr) &&
|
1783 |
(instance_id == se->compat->instance_id || |
1784 |
instance_id == se->alias_id)) |
1785 |
return se;
|
1786 |
} |
1787 |
} |
1788 |
return NULL; |
1789 |
} |
1790 |
|
1791 |
static const VMStateDescription *vmstate_get_subsection(const VMStateSubsection *sub, char *idstr) |
1792 |
{ |
1793 |
while(sub && sub->needed) {
|
1794 |
if (strcmp(idstr, sub->vmsd->name) == 0) { |
1795 |
return sub->vmsd;
|
1796 |
} |
1797 |
sub++; |
1798 |
} |
1799 |
return NULL; |
1800 |
} |
1801 |
|
1802 |
static int vmstate_subsection_load(QEMUFile *f, const VMStateDescription *vmsd, |
1803 |
void *opaque)
|
1804 |
{ |
1805 |
while (qemu_peek_byte(f, 0) == QEMU_VM_SUBSECTION) { |
1806 |
char idstr[256]; |
1807 |
int ret;
|
1808 |
uint8_t version_id, len, size; |
1809 |
const VMStateDescription *sub_vmsd;
|
1810 |
|
1811 |
len = qemu_peek_byte(f, 1);
|
1812 |
if (len < strlen(vmsd->name) + 1) { |
1813 |
/* subsection name has be be "section_name/a" */
|
1814 |
return 0; |
1815 |
} |
1816 |
size = qemu_peek_buffer(f, (uint8_t *)idstr, len, 2);
|
1817 |
if (size != len) {
|
1818 |
return 0; |
1819 |
} |
1820 |
idstr[size] = 0;
|
1821 |
|
1822 |
if (strncmp(vmsd->name, idstr, strlen(vmsd->name)) != 0) { |
1823 |
/* it don't have a valid subsection name */
|
1824 |
return 0; |
1825 |
} |
1826 |
sub_vmsd = vmstate_get_subsection(vmsd->subsections, idstr); |
1827 |
if (sub_vmsd == NULL) { |
1828 |
return -ENOENT;
|
1829 |
} |
1830 |
qemu_file_skip(f, 1); /* subsection */ |
1831 |
qemu_file_skip(f, 1); /* len */ |
1832 |
qemu_file_skip(f, len); /* idstr */
|
1833 |
version_id = qemu_get_be32(f); |
1834 |
|
1835 |
ret = vmstate_load_state(f, sub_vmsd, opaque, version_id); |
1836 |
if (ret) {
|
1837 |
return ret;
|
1838 |
} |
1839 |
} |
1840 |
return 0; |
1841 |
} |
1842 |
|
1843 |
static void vmstate_subsection_save(QEMUFile *f, const VMStateDescription *vmsd, |
1844 |
void *opaque)
|
1845 |
{ |
1846 |
const VMStateSubsection *sub = vmsd->subsections;
|
1847 |
|
1848 |
while (sub && sub->needed) {
|
1849 |
if (sub->needed(opaque)) {
|
1850 |
const VMStateDescription *vmsd = sub->vmsd;
|
1851 |
uint8_t len; |
1852 |
|
1853 |
qemu_put_byte(f, QEMU_VM_SUBSECTION); |
1854 |
len = strlen(vmsd->name); |
1855 |
qemu_put_byte(f, len); |
1856 |
qemu_put_buffer(f, (uint8_t *)vmsd->name, len); |
1857 |
qemu_put_be32(f, vmsd->version_id); |
1858 |
vmstate_save_state(f, vmsd, opaque); |
1859 |
} |
1860 |
sub++; |
1861 |
} |
1862 |
} |
1863 |
|
1864 |
typedef struct LoadStateEntry { |
1865 |
QLIST_ENTRY(LoadStateEntry) entry; |
1866 |
SaveStateEntry *se; |
1867 |
int section_id;
|
1868 |
int version_id;
|
1869 |
} LoadStateEntry; |
1870 |
|
1871 |
int qemu_loadvm_state(QEMUFile *f)
|
1872 |
{ |
1873 |
QLIST_HEAD(, LoadStateEntry) loadvm_handlers = |
1874 |
QLIST_HEAD_INITIALIZER(loadvm_handlers); |
1875 |
LoadStateEntry *le, *new_le; |
1876 |
uint8_t section_type; |
1877 |
unsigned int v; |
1878 |
int ret;
|
1879 |
|
1880 |
if (qemu_savevm_state_blocked(NULL)) { |
1881 |
return -EINVAL;
|
1882 |
} |
1883 |
|
1884 |
v = qemu_get_be32(f); |
1885 |
if (v != QEMU_VM_FILE_MAGIC)
|
1886 |
return -EINVAL;
|
1887 |
|
1888 |
v = qemu_get_be32(f); |
1889 |
if (v == QEMU_VM_FILE_VERSION_COMPAT) {
|
1890 |
fprintf(stderr, "SaveVM v2 format is obsolete and don't work anymore\n");
|
1891 |
return -ENOTSUP;
|
1892 |
} |
1893 |
if (v != QEMU_VM_FILE_VERSION)
|
1894 |
return -ENOTSUP;
|
1895 |
|
1896 |
while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
|
1897 |
uint32_t instance_id, version_id, section_id; |
1898 |
SaveStateEntry *se; |
1899 |
char idstr[257]; |
1900 |
int len;
|
1901 |
|
1902 |
switch (section_type) {
|
1903 |
case QEMU_VM_SECTION_START:
|
1904 |
case QEMU_VM_SECTION_FULL:
|
1905 |
/* Read section start */
|
1906 |
section_id = qemu_get_be32(f); |
1907 |
len = qemu_get_byte(f); |
1908 |
qemu_get_buffer(f, (uint8_t *)idstr, len); |
1909 |
idstr[len] = 0;
|
1910 |
instance_id = qemu_get_be32(f); |
1911 |
version_id = qemu_get_be32(f); |
1912 |
|
1913 |
/* Find savevm section */
|
1914 |
se = find_se(idstr, instance_id); |
1915 |
if (se == NULL) { |
1916 |
fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
|
1917 |
ret = -EINVAL; |
1918 |
goto out;
|
1919 |
} |
1920 |
|
1921 |
/* Validate version */
|
1922 |
if (version_id > se->version_id) {
|
1923 |
fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
|
1924 |
version_id, idstr, se->version_id); |
1925 |
ret = -EINVAL; |
1926 |
goto out;
|
1927 |
} |
1928 |
|
1929 |
/* Add entry */
|
1930 |
le = g_malloc0(sizeof(*le));
|
1931 |
|
1932 |
le->se = se; |
1933 |
le->section_id = section_id; |
1934 |
le->version_id = version_id; |
1935 |
QLIST_INSERT_HEAD(&loadvm_handlers, le, entry); |
1936 |
|
1937 |
ret = vmstate_load(f, le->se, le->version_id); |
1938 |
if (ret < 0) { |
1939 |
fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
|
1940 |
instance_id, idstr); |
1941 |
goto out;
|
1942 |
} |
1943 |
break;
|
1944 |
case QEMU_VM_SECTION_PART:
|
1945 |
case QEMU_VM_SECTION_END:
|
1946 |
section_id = qemu_get_be32(f); |
1947 |
|
1948 |
QLIST_FOREACH(le, &loadvm_handlers, entry) { |
1949 |
if (le->section_id == section_id) {
|
1950 |
break;
|
1951 |
} |
1952 |
} |
1953 |
if (le == NULL) { |
1954 |
fprintf(stderr, "Unknown savevm section %d\n", section_id);
|
1955 |
ret = -EINVAL; |
1956 |
goto out;
|
1957 |
} |
1958 |
|
1959 |
ret = vmstate_load(f, le->se, le->version_id); |
1960 |
if (ret < 0) { |
1961 |
fprintf(stderr, "qemu: warning: error while loading state section id %d\n",
|
1962 |
section_id); |
1963 |
goto out;
|
1964 |
} |
1965 |
break;
|
1966 |
default:
|
1967 |
fprintf(stderr, "Unknown savevm section type %d\n", section_type);
|
1968 |
ret = -EINVAL; |
1969 |
goto out;
|
1970 |
} |
1971 |
} |
1972 |
|
1973 |
cpu_synchronize_all_post_init(); |
1974 |
|
1975 |
ret = 0;
|
1976 |
|
1977 |
out:
|
1978 |
QLIST_FOREACH_SAFE(le, &loadvm_handlers, entry, new_le) { |
1979 |
QLIST_REMOVE(le, entry); |
1980 |
g_free(le); |
1981 |
} |
1982 |
|
1983 |
if (ret == 0) { |
1984 |
ret = qemu_file_get_error(f); |
1985 |
} |
1986 |
|
1987 |
return ret;
|
1988 |
} |
1989 |
|
1990 |
static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info, |
1991 |
const char *name) |
1992 |
{ |
1993 |
QEMUSnapshotInfo *sn_tab, *sn; |
1994 |
int nb_sns, i, ret;
|
1995 |
|
1996 |
ret = -ENOENT; |
1997 |
nb_sns = bdrv_snapshot_list(bs, &sn_tab); |
1998 |
if (nb_sns < 0) |
1999 |
return ret;
|
2000 |
for(i = 0; i < nb_sns; i++) { |
2001 |
sn = &sn_tab[i]; |
2002 |
if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
|
2003 |
*sn_info = *sn; |
2004 |
ret = 0;
|
2005 |
break;
|
2006 |
} |
2007 |
} |
2008 |
g_free(sn_tab); |
2009 |
return ret;
|
2010 |
} |
2011 |
|
2012 |
/*
|
2013 |
* Deletes snapshots of a given name in all opened images.
|
2014 |
*/
|
2015 |
static int del_existing_snapshots(Monitor *mon, const char *name) |
2016 |
{ |
2017 |
BlockDriverState *bs; |
2018 |
QEMUSnapshotInfo sn1, *snapshot = &sn1; |
2019 |
int ret;
|
2020 |
|
2021 |
bs = NULL;
|
2022 |
while ((bs = bdrv_next(bs))) {
|
2023 |
if (bdrv_can_snapshot(bs) &&
|
2024 |
bdrv_snapshot_find(bs, snapshot, name) >= 0)
|
2025 |
{ |
2026 |
ret = bdrv_snapshot_delete(bs, name); |
2027 |
if (ret < 0) { |
2028 |
monitor_printf(mon, |
2029 |
"Error while deleting snapshot on '%s'\n",
|
2030 |
bdrv_get_device_name(bs)); |
2031 |
return -1; |
2032 |
} |
2033 |
} |
2034 |
} |
2035 |
|
2036 |
return 0; |
2037 |
} |
2038 |
|
2039 |
void do_savevm(Monitor *mon, const QDict *qdict) |
2040 |
{ |
2041 |
BlockDriverState *bs, *bs1; |
2042 |
QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1; |
2043 |
int ret;
|
2044 |
QEMUFile *f; |
2045 |
int saved_vm_running;
|
2046 |
uint64_t vm_state_size; |
2047 |
#ifdef _WIN32
|
2048 |
struct _timeb tb;
|
2049 |
struct tm *ptm;
|
2050 |
#else
|
2051 |
struct timeval tv;
|
2052 |
struct tm tm;
|
2053 |
#endif
|
2054 |
const char *name = qdict_get_try_str(qdict, "name"); |
2055 |
|
2056 |
/* Verify if there is a device that doesn't support snapshots and is writable */
|
2057 |
bs = NULL;
|
2058 |
while ((bs = bdrv_next(bs))) {
|
2059 |
|
2060 |
if (!bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
|
2061 |
continue;
|
2062 |
} |
2063 |
|
2064 |
if (!bdrv_can_snapshot(bs)) {
|
2065 |
monitor_printf(mon, "Device '%s' is writable but does not support snapshots.\n",
|
2066 |
bdrv_get_device_name(bs)); |
2067 |
return;
|
2068 |
} |
2069 |
} |
2070 |
|
2071 |
bs = bdrv_snapshots(); |
2072 |
if (!bs) {
|
2073 |
monitor_printf(mon, "No block device can accept snapshots\n");
|
2074 |
return;
|
2075 |
} |
2076 |
|
2077 |
saved_vm_running = runstate_is_running(); |
2078 |
vm_stop(RUN_STATE_SAVE_VM); |
2079 |
|
2080 |
memset(sn, 0, sizeof(*sn)); |
2081 |
|
2082 |
/* fill auxiliary fields */
|
2083 |
#ifdef _WIN32
|
2084 |
_ftime(&tb); |
2085 |
sn->date_sec = tb.time; |
2086 |
sn->date_nsec = tb.millitm * 1000000;
|
2087 |
#else
|
2088 |
gettimeofday(&tv, NULL);
|
2089 |
sn->date_sec = tv.tv_sec; |
2090 |
sn->date_nsec = tv.tv_usec * 1000;
|
2091 |
#endif
|
2092 |
sn->vm_clock_nsec = qemu_get_clock_ns(vm_clock); |
2093 |
|
2094 |
if (name) {
|
2095 |
ret = bdrv_snapshot_find(bs, old_sn, name); |
2096 |
if (ret >= 0) { |
2097 |
pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
|
2098 |
pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
|
2099 |
} else {
|
2100 |
pstrcpy(sn->name, sizeof(sn->name), name);
|
2101 |
} |
2102 |
} else {
|
2103 |
#ifdef _WIN32
|
2104 |
time_t t = tb.time; |
2105 |
ptm = localtime(&t); |
2106 |
strftime(sn->name, sizeof(sn->name), "vm-%Y%m%d%H%M%S", ptm); |
2107 |
#else
|
2108 |
/* cast below needed for OpenBSD where tv_sec is still 'long' */
|
2109 |
localtime_r((const time_t *)&tv.tv_sec, &tm);
|
2110 |
strftime(sn->name, sizeof(sn->name), "vm-%Y%m%d%H%M%S", &tm); |
2111 |
#endif
|
2112 |
} |
2113 |
|
2114 |
/* Delete old snapshots of the same name */
|
2115 |
if (name && del_existing_snapshots(mon, name) < 0) { |
2116 |
goto the_end;
|
2117 |
} |
2118 |
|
2119 |
/* save the VM state */
|
2120 |
f = qemu_fopen_bdrv(bs, 1);
|
2121 |
if (!f) {
|
2122 |
monitor_printf(mon, "Could not open VM state file\n");
|
2123 |
goto the_end;
|
2124 |
} |
2125 |
ret = qemu_savevm_state(f); |
2126 |
vm_state_size = qemu_ftell(f); |
2127 |
qemu_fclose(f); |
2128 |
if (ret < 0) { |
2129 |
monitor_printf(mon, "Error %d while writing VM\n", ret);
|
2130 |
goto the_end;
|
2131 |
} |
2132 |
|
2133 |
/* create the snapshots */
|
2134 |
|
2135 |
bs1 = NULL;
|
2136 |
while ((bs1 = bdrv_next(bs1))) {
|
2137 |
if (bdrv_can_snapshot(bs1)) {
|
2138 |
/* Write VM state size only to the image that contains the state */
|
2139 |
sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
|
2140 |
ret = bdrv_snapshot_create(bs1, sn); |
2141 |
if (ret < 0) { |
2142 |
monitor_printf(mon, "Error while creating snapshot on '%s'\n",
|
2143 |
bdrv_get_device_name(bs1)); |
2144 |
} |
2145 |
} |
2146 |
} |
2147 |
|
2148 |
the_end:
|
2149 |
if (saved_vm_running)
|
2150 |
vm_start(); |
2151 |
} |
2152 |
|
2153 |
void qmp_xen_save_devices_state(const char *filename, Error **errp) |
2154 |
{ |
2155 |
QEMUFile *f; |
2156 |
int saved_vm_running;
|
2157 |
int ret;
|
2158 |
|
2159 |
saved_vm_running = runstate_is_running(); |
2160 |
vm_stop(RUN_STATE_SAVE_VM); |
2161 |
|
2162 |
f = qemu_fopen(filename, "wb");
|
2163 |
if (!f) {
|
2164 |
error_set(errp, QERR_OPEN_FILE_FAILED, filename); |
2165 |
goto the_end;
|
2166 |
} |
2167 |
ret = qemu_save_device_state(f); |
2168 |
qemu_fclose(f); |
2169 |
if (ret < 0) { |
2170 |
error_set(errp, QERR_IO_ERROR); |
2171 |
} |
2172 |
|
2173 |
the_end:
|
2174 |
if (saved_vm_running)
|
2175 |
vm_start(); |
2176 |
} |
2177 |
|
2178 |
int load_vmstate(const char *name) |
2179 |
{ |
2180 |
BlockDriverState *bs, *bs_vm_state; |
2181 |
QEMUSnapshotInfo sn; |
2182 |
QEMUFile *f; |
2183 |
int ret;
|
2184 |
|
2185 |
bs_vm_state = bdrv_snapshots(); |
2186 |
if (!bs_vm_state) {
|
2187 |
error_report("No block device supports snapshots");
|
2188 |
return -ENOTSUP;
|
2189 |
} |
2190 |
|
2191 |
/* Don't even try to load empty VM states */
|
2192 |
ret = bdrv_snapshot_find(bs_vm_state, &sn, name); |
2193 |
if (ret < 0) { |
2194 |
return ret;
|
2195 |
} else if (sn.vm_state_size == 0) { |
2196 |
error_report("This is a disk-only snapshot. Revert to it offline "
|
2197 |
"using qemu-img.");
|
2198 |
return -EINVAL;
|
2199 |
} |
2200 |
|
2201 |
/* Verify if there is any device that doesn't support snapshots and is
|
2202 |
writable and check if the requested snapshot is available too. */
|
2203 |
bs = NULL;
|
2204 |
while ((bs = bdrv_next(bs))) {
|
2205 |
|
2206 |
if (!bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
|
2207 |
continue;
|
2208 |
} |
2209 |
|
2210 |
if (!bdrv_can_snapshot(bs)) {
|
2211 |
error_report("Device '%s' is writable but does not support snapshots.",
|
2212 |
bdrv_get_device_name(bs)); |
2213 |
return -ENOTSUP;
|
2214 |
} |
2215 |
|
2216 |
ret = bdrv_snapshot_find(bs, &sn, name); |
2217 |
if (ret < 0) { |
2218 |
error_report("Device '%s' does not have the requested snapshot '%s'",
|
2219 |
bdrv_get_device_name(bs), name); |
2220 |
return ret;
|
2221 |
} |
2222 |
} |
2223 |
|
2224 |
/* Flush all IO requests so they don't interfere with the new state. */
|
2225 |
bdrv_drain_all(); |
2226 |
|
2227 |
bs = NULL;
|
2228 |
while ((bs = bdrv_next(bs))) {
|
2229 |
if (bdrv_can_snapshot(bs)) {
|
2230 |
ret = bdrv_snapshot_goto(bs, name); |
2231 |
if (ret < 0) { |
2232 |
error_report("Error %d while activating snapshot '%s' on '%s'",
|
2233 |
ret, name, bdrv_get_device_name(bs)); |
2234 |
return ret;
|
2235 |
} |
2236 |
} |
2237 |
} |
2238 |
|
2239 |
/* restore the VM state */
|
2240 |
f = qemu_fopen_bdrv(bs_vm_state, 0);
|
2241 |
if (!f) {
|
2242 |
error_report("Could not open VM state file");
|
2243 |
return -EINVAL;
|
2244 |
} |
2245 |
|
2246 |
qemu_system_reset(VMRESET_SILENT); |
2247 |
ret = qemu_loadvm_state(f); |
2248 |
|
2249 |
qemu_fclose(f); |
2250 |
if (ret < 0) { |
2251 |
error_report("Error %d while loading VM state", ret);
|
2252 |
return ret;
|
2253 |
} |
2254 |
|
2255 |
return 0; |
2256 |
} |
2257 |
|
2258 |
void do_delvm(Monitor *mon, const QDict *qdict) |
2259 |
{ |
2260 |
BlockDriverState *bs, *bs1; |
2261 |
int ret;
|
2262 |
const char *name = qdict_get_str(qdict, "name"); |
2263 |
|
2264 |
bs = bdrv_snapshots(); |
2265 |
if (!bs) {
|
2266 |
monitor_printf(mon, "No block device supports snapshots\n");
|
2267 |
return;
|
2268 |
} |
2269 |
|
2270 |
bs1 = NULL;
|
2271 |
while ((bs1 = bdrv_next(bs1))) {
|
2272 |
if (bdrv_can_snapshot(bs1)) {
|
2273 |
ret = bdrv_snapshot_delete(bs1, name); |
2274 |
if (ret < 0) { |
2275 |
if (ret == -ENOTSUP)
|
2276 |
monitor_printf(mon, |
2277 |
"Snapshots not supported on device '%s'\n",
|
2278 |
bdrv_get_device_name(bs1)); |
2279 |
else
|
2280 |
monitor_printf(mon, "Error %d while deleting snapshot on "
|
2281 |
"'%s'\n", ret, bdrv_get_device_name(bs1));
|
2282 |
} |
2283 |
} |
2284 |
} |
2285 |
} |
2286 |
|
2287 |
void do_info_snapshots(Monitor *mon)
|
2288 |
{ |
2289 |
BlockDriverState *bs, *bs1; |
2290 |
QEMUSnapshotInfo *sn_tab, *sn, s, *sn_info = &s; |
2291 |
int nb_sns, i, ret, available;
|
2292 |
int total;
|
2293 |
int *available_snapshots;
|
2294 |
char buf[256]; |
2295 |
|
2296 |
bs = bdrv_snapshots(); |
2297 |
if (!bs) {
|
2298 |
monitor_printf(mon, "No available block device supports snapshots\n");
|
2299 |
return;
|
2300 |
} |
2301 |
|
2302 |
nb_sns = bdrv_snapshot_list(bs, &sn_tab); |
2303 |
if (nb_sns < 0) { |
2304 |
monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
|
2305 |
return;
|
2306 |
} |
2307 |
|
2308 |
if (nb_sns == 0) { |
2309 |
monitor_printf(mon, "There is no snapshot available.\n");
|
2310 |
return;
|
2311 |
} |
2312 |
|
2313 |
available_snapshots = g_malloc0(sizeof(int) * nb_sns); |
2314 |
total = 0;
|
2315 |
for (i = 0; i < nb_sns; i++) { |
2316 |
sn = &sn_tab[i]; |
2317 |
available = 1;
|
2318 |
bs1 = NULL;
|
2319 |
|
2320 |
while ((bs1 = bdrv_next(bs1))) {
|
2321 |
if (bdrv_can_snapshot(bs1) && bs1 != bs) {
|
2322 |
ret = bdrv_snapshot_find(bs1, sn_info, sn->id_str); |
2323 |
if (ret < 0) { |
2324 |
available = 0;
|
2325 |
break;
|
2326 |
} |
2327 |
} |
2328 |
} |
2329 |
|
2330 |
if (available) {
|
2331 |
available_snapshots[total] = i; |
2332 |
total++; |
2333 |
} |
2334 |
} |
2335 |
|
2336 |
if (total > 0) { |
2337 |
monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL)); |
2338 |
for (i = 0; i < total; i++) { |
2339 |
sn = &sn_tab[available_snapshots[i]]; |
2340 |
monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn)); |
2341 |
} |
2342 |
} else {
|
2343 |
monitor_printf(mon, "There is no suitable snapshot available\n");
|
2344 |
} |
2345 |
|
2346 |
g_free(sn_tab); |
2347 |
g_free(available_snapshots); |
2348 |
|
2349 |
} |
2350 |
|
2351 |
void vmstate_register_ram(MemoryRegion *mr, DeviceState *dev)
|
2352 |
{ |
2353 |
qemu_ram_set_idstr(memory_region_get_ram_addr(mr) & TARGET_PAGE_MASK, |
2354 |
memory_region_name(mr), dev); |
2355 |
} |
2356 |
|
2357 |
void vmstate_unregister_ram(MemoryRegion *mr, DeviceState *dev)
|
2358 |
{ |
2359 |
/* Nothing do to while the implementation is in RAMBlock */
|
2360 |
} |
2361 |
|
2362 |
void vmstate_register_ram_global(MemoryRegion *mr)
|
2363 |
{ |
2364 |
vmstate_register_ram(mr, NULL);
|
2365 |
} |
2366 |
|
2367 |
/*
|
2368 |
page = zrun nzrun
|
2369 |
| zrun nzrun page
|
2370 |
|
2371 |
zrun = length
|
2372 |
|
2373 |
nzrun = length byte...
|
2374 |
|
2375 |
length = uleb128 encoded integer
|
2376 |
*/
|
2377 |
int xbzrle_encode_buffer(uint8_t *old_buf, uint8_t *new_buf, int slen, |
2378 |
uint8_t *dst, int dlen)
|
2379 |
{ |
2380 |
uint32_t zrun_len = 0, nzrun_len = 0; |
2381 |
int d = 0, i = 0; |
2382 |
long res, xor;
|
2383 |
uint8_t *nzrun_start = NULL;
|
2384 |
|
2385 |
g_assert(!(((uintptr_t)old_buf | (uintptr_t)new_buf | slen) % |
2386 |
sizeof(long))); |
2387 |
|
2388 |
while (i < slen) {
|
2389 |
/* overflow */
|
2390 |
if (d + 2 > dlen) { |
2391 |
return -1; |
2392 |
} |
2393 |
|
2394 |
/* not aligned to sizeof(long) */
|
2395 |
res = (slen - i) % sizeof(long); |
2396 |
while (res && old_buf[i] == new_buf[i]) {
|
2397 |
zrun_len++; |
2398 |
i++; |
2399 |
res--; |
2400 |
} |
2401 |
|
2402 |
/* word at a time for speed */
|
2403 |
if (!res) {
|
2404 |
while (i < slen &&
|
2405 |
(*(long *)(old_buf + i)) == (*(long *)(new_buf + i))) { |
2406 |
i += sizeof(long); |
2407 |
zrun_len += sizeof(long); |
2408 |
} |
2409 |
|
2410 |
/* go over the rest */
|
2411 |
while (i < slen && old_buf[i] == new_buf[i]) {
|
2412 |
zrun_len++; |
2413 |
i++; |
2414 |
} |
2415 |
} |
2416 |
|
2417 |
/* buffer unchanged */
|
2418 |
if (zrun_len == slen) {
|
2419 |
return 0; |
2420 |
} |
2421 |
|
2422 |
/* skip last zero run */
|
2423 |
if (i == slen) {
|
2424 |
return d;
|
2425 |
} |
2426 |
|
2427 |
d += uleb128_encode_small(dst + d, zrun_len); |
2428 |
|
2429 |
zrun_len = 0;
|
2430 |
nzrun_start = new_buf + i; |
2431 |
|
2432 |
/* overflow */
|
2433 |
if (d + 2 > dlen) { |
2434 |
return -1; |
2435 |
} |
2436 |
/* not aligned to sizeof(long) */
|
2437 |
res = (slen - i) % sizeof(long); |
2438 |
while (res && old_buf[i] != new_buf[i]) {
|
2439 |
i++; |
2440 |
nzrun_len++; |
2441 |
res--; |
2442 |
} |
2443 |
|
2444 |
/* word at a time for speed, use of 32-bit long okay */
|
2445 |
if (!res) {
|
2446 |
/* truncation to 32-bit long okay */
|
2447 |
long mask = (long)0x0101010101010101ULL; |
2448 |
while (i < slen) {
|
2449 |
xor = *(long *)(old_buf + i) ^ *(long *)(new_buf + i); |
2450 |
if ((xor - mask) & ~xor & (mask << 7)) { |
2451 |
/* found the end of an nzrun within the current long */
|
2452 |
while (old_buf[i] != new_buf[i]) {
|
2453 |
nzrun_len++; |
2454 |
i++; |
2455 |
} |
2456 |
break;
|
2457 |
} else {
|
2458 |
i += sizeof(long); |
2459 |
nzrun_len += sizeof(long); |
2460 |
} |
2461 |
} |
2462 |
} |
2463 |
|
2464 |
d += uleb128_encode_small(dst + d, nzrun_len); |
2465 |
/* overflow */
|
2466 |
if (d + nzrun_len > dlen) {
|
2467 |
return -1; |
2468 |
} |
2469 |
memcpy(dst + d, nzrun_start, nzrun_len); |
2470 |
d += nzrun_len; |
2471 |
nzrun_len = 0;
|
2472 |
} |
2473 |
|
2474 |
return d;
|
2475 |
} |
2476 |
|
2477 |
int xbzrle_decode_buffer(uint8_t *src, int slen, uint8_t *dst, int dlen) |
2478 |
{ |
2479 |
int i = 0, d = 0; |
2480 |
int ret;
|
2481 |
uint32_t count = 0;
|
2482 |
|
2483 |
while (i < slen) {
|
2484 |
|
2485 |
/* zrun */
|
2486 |
if ((slen - i) < 2) { |
2487 |
return -1; |
2488 |
} |
2489 |
|
2490 |
ret = uleb128_decode_small(src + i, &count); |
2491 |
if (ret < 0 || (i && !count)) { |
2492 |
return -1; |
2493 |
} |
2494 |
i += ret; |
2495 |
d += count; |
2496 |
|
2497 |
/* overflow */
|
2498 |
if (d > dlen) {
|
2499 |
return -1; |
2500 |
} |
2501 |
|
2502 |
/* nzrun */
|
2503 |
if ((slen - i) < 2) { |
2504 |
return -1; |
2505 |
} |
2506 |
|
2507 |
ret = uleb128_decode_small(src + i, &count); |
2508 |
if (ret < 0 || !count) { |
2509 |
return -1; |
2510 |
} |
2511 |
i += ret; |
2512 |
|
2513 |
/* overflow */
|
2514 |
if (d + count > dlen || i + count > slen) {
|
2515 |
return -1; |
2516 |
} |
2517 |
|
2518 |
memcpy(dst + d, src + i, count); |
2519 |
d += count; |
2520 |
i += count; |
2521 |
} |
2522 |
|
2523 |
return d;
|
2524 |
} |