Statistics
| Branch: | Revision:

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, &params);
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
}