Statistics
| Branch: | Revision:

root / savevm.c @ 7685ee6a

History | View | Annotate | Download (48.8 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 <signal.h>
27
#include <time.h>
28
#include <errno.h>
29
#include <sys/time.h>
30
#include <zlib.h>
31

    
32
/* Needed early for CONFIG_BSD etc. */
33
#include "config-host.h"
34

    
35
#ifndef _WIN32
36
#include <sys/times.h>
37
#include <sys/wait.h>
38
#include <termios.h>
39
#include <sys/mman.h>
40
#include <sys/ioctl.h>
41
#include <sys/resource.h>
42
#include <sys/socket.h>
43
#include <netinet/in.h>
44
#include <net/if.h>
45
#include <arpa/inet.h>
46
#include <dirent.h>
47
#include <netdb.h>
48
#include <sys/select.h>
49
#ifdef CONFIG_BSD
50
#include <sys/stat.h>
51
#if defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || defined(__DragonFly__)
52
#include <libutil.h>
53
#else
54
#include <util.h>
55
#endif
56
#ifdef __linux__
57
#include <pty.h>
58
#include <malloc.h>
59
#include <linux/rtc.h>
60
#endif
61
#endif
62
#endif
63

    
64
#ifdef _WIN32
65
#include <windows.h>
66
#include <malloc.h>
67
#include <sys/timeb.h>
68
#include <mmsystem.h>
69
#define getopt_long_only getopt_long
70
#define memalign(align, size) malloc(size)
71
#endif
72

    
73
#include "qemu-common.h"
74
#include "hw/hw.h"
75
#include "hw/qdev.h"
76
#include "net.h"
77
#include "monitor.h"
78
#include "sysemu.h"
79
#include "qemu-timer.h"
80
#include "qemu-char.h"
81
#include "blockdev.h"
82
#include "audio/audio.h"
83
#include "migration.h"
84
#include "qemu_socket.h"
85
#include "qemu-queue.h"
86

    
87
#define SELF_ANNOUNCE_ROUNDS 5
88

    
89
#ifndef ETH_P_RARP
90
#define ETH_P_RARP 0x8035
91
#endif
92
#define ARP_HTYPE_ETH 0x0001
93
#define ARP_PTYPE_IP 0x0800
94
#define ARP_OP_REQUEST_REV 0x3
95

    
96
static int announce_self_create(uint8_t *buf,
97
                                uint8_t *mac_addr)
98
{
99
    /* Ethernet header. */
100
    memset(buf, 0xff, 6);         /* destination MAC addr */
101
    memcpy(buf + 6, mac_addr, 6); /* source MAC addr */
102
    *(uint16_t *)(buf + 12) = htons(ETH_P_RARP); /* ethertype */
103

    
104
    /* RARP header. */
105
    *(uint16_t *)(buf + 14) = htons(ARP_HTYPE_ETH); /* hardware addr space */
106
    *(uint16_t *)(buf + 16) = htons(ARP_PTYPE_IP); /* protocol addr space */
107
    *(buf + 18) = 6; /* hardware addr length (ethernet) */
108
    *(buf + 19) = 4; /* protocol addr length (IPv4) */
109
    *(uint16_t *)(buf + 20) = htons(ARP_OP_REQUEST_REV); /* opcode */
110
    memcpy(buf + 22, mac_addr, 6); /* source hw addr */
111
    memset(buf + 28, 0x00, 4);     /* source protocol addr */
112
    memcpy(buf + 32, mac_addr, 6); /* target hw addr */
113
    memset(buf + 38, 0x00, 4);     /* target protocol addr */
114

    
115
    /* Padding to get up to 60 bytes (ethernet min packet size, minus FCS). */
116
    memset(buf + 42, 0x00, 18);
117

    
118
    return 60; /* len (FCS will be added by hardware) */
119
}
120

    
121
static void qemu_announce_self_iter(NICState *nic, void *opaque)
122
{
123
    uint8_t buf[60];
124
    int len;
125

    
126
    len = announce_self_create(buf, nic->conf->macaddr.a);
127

    
128
    qemu_send_packet_raw(&nic->nc, buf, len);
129
}
130

    
131

    
132
static void qemu_announce_self_once(void *opaque)
133
{
134
    static int count = SELF_ANNOUNCE_ROUNDS;
135
    QEMUTimer *timer = *(QEMUTimer **)opaque;
136

    
137
    qemu_foreach_nic(qemu_announce_self_iter, NULL);
138

    
139
    if (--count) {
140
        /* delay 50ms, 150ms, 250ms, ... */
141
        qemu_mod_timer(timer, qemu_get_clock(rt_clock) +
142
                       50 + (SELF_ANNOUNCE_ROUNDS - count - 1) * 100);
143
    } else {
144
            qemu_del_timer(timer);
145
            qemu_free_timer(timer);
146
    }
147
}
148

    
149
void qemu_announce_self(void)
150
{
151
        static QEMUTimer *timer;
152
        timer = qemu_new_timer(rt_clock, qemu_announce_self_once, &timer);
153
        qemu_announce_self_once(&timer);
154
}
155

    
156
/***********************************************************/
157
/* savevm/loadvm support */
158

    
159
#define IO_BUF_SIZE 32768
160

    
161
struct QEMUFile {
162
    QEMUFilePutBufferFunc *put_buffer;
163
    QEMUFileGetBufferFunc *get_buffer;
164
    QEMUFileCloseFunc *close;
165
    QEMUFileRateLimit *rate_limit;
166
    QEMUFileSetRateLimit *set_rate_limit;
167
    QEMUFileGetRateLimit *get_rate_limit;
168
    void *opaque;
169
    int is_write;
170

    
171
    int64_t buf_offset; /* start of buffer when writing, end of buffer
172
                           when reading */
173
    int buf_index;
174
    int buf_size; /* 0 when writing */
175
    uint8_t buf[IO_BUF_SIZE];
176

    
177
    int has_error;
178
};
179

    
180
typedef struct QEMUFileStdio
181
{
182
    FILE *stdio_file;
183
    QEMUFile *file;
184
} QEMUFileStdio;
185

    
186
typedef struct QEMUFileSocket
187
{
188
    int fd;
189
    QEMUFile *file;
190
} QEMUFileSocket;
191

    
192
static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
193
{
194
    QEMUFileSocket *s = opaque;
195
    ssize_t len;
196

    
197
    do {
198
        len = recv(s->fd, (void *)buf, size, 0);
199
    } while (len == -1 && socket_error() == EINTR);
200

    
201
    if (len == -1)
202
        len = -socket_error();
203

    
204
    return len;
205
}
206

    
207
static int socket_close(void *opaque)
208
{
209
    QEMUFileSocket *s = opaque;
210
    qemu_free(s);
211
    return 0;
212
}
213

    
214
static int stdio_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
215
{
216
    QEMUFileStdio *s = opaque;
217
    return fwrite(buf, 1, size, s->stdio_file);
218
}
219

    
220
static int stdio_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
221
{
222
    QEMUFileStdio *s = opaque;
223
    FILE *fp = s->stdio_file;
224
    int bytes;
225

    
226
    do {
227
        clearerr(fp);
228
        bytes = fread(buf, 1, size, fp);
229
    } while ((bytes == 0) && ferror(fp) && (errno == EINTR));
230
    return bytes;
231
}
232

    
233
static int stdio_pclose(void *opaque)
234
{
235
    QEMUFileStdio *s = opaque;
236
    int ret;
237
    ret = pclose(s->stdio_file);
238
    qemu_free(s);
239
    return ret;
240
}
241

    
242
static int stdio_fclose(void *opaque)
243
{
244
    QEMUFileStdio *s = opaque;
245
    fclose(s->stdio_file);
246
    qemu_free(s);
247
    return 0;
248
}
249

    
250
QEMUFile *qemu_popen(FILE *stdio_file, const char *mode)
251
{
252
    QEMUFileStdio *s;
253

    
254
    if (stdio_file == NULL || mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) {
255
        fprintf(stderr, "qemu_popen: Argument validity check failed\n");
256
        return NULL;
257
    }
258

    
259
    s = qemu_mallocz(sizeof(QEMUFileStdio));
260

    
261
    s->stdio_file = stdio_file;
262

    
263
    if(mode[0] == 'r') {
264
        s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_pclose, 
265
                                 NULL, NULL, NULL);
266
    } else {
267
        s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_pclose, 
268
                                 NULL, NULL, NULL);
269
    }
270
    return s->file;
271
}
272

    
273
QEMUFile *qemu_popen_cmd(const char *command, const char *mode)
274
{
275
    FILE *popen_file;
276

    
277
    popen_file = popen(command, mode);
278
    if(popen_file == NULL) {
279
        return NULL;
280
    }
281

    
282
    return qemu_popen(popen_file, mode);
283
}
284

    
285
int qemu_stdio_fd(QEMUFile *f)
286
{
287
    QEMUFileStdio *p;
288
    int fd;
289

    
290
    p = (QEMUFileStdio *)f->opaque;
291
    fd = fileno(p->stdio_file);
292

    
293
    return fd;
294
}
295

    
296
QEMUFile *qemu_fdopen(int fd, const char *mode)
297
{
298
    QEMUFileStdio *s;
299

    
300
    if (mode == NULL ||
301
        (mode[0] != 'r' && mode[0] != 'w') ||
302
        mode[1] != 'b' || mode[2] != 0) {
303
        fprintf(stderr, "qemu_fdopen: Argument validity check failed\n");
304
        return NULL;
305
    }
306

    
307
    s = qemu_mallocz(sizeof(QEMUFileStdio));
308
    s->stdio_file = fdopen(fd, mode);
309
    if (!s->stdio_file)
310
        goto fail;
311

    
312
    if(mode[0] == 'r') {
313
        s->file = qemu_fopen_ops(s, NULL, stdio_get_buffer, stdio_fclose, 
314
                                 NULL, NULL, NULL);
315
    } else {
316
        s->file = qemu_fopen_ops(s, stdio_put_buffer, NULL, stdio_fclose, 
317
                                 NULL, NULL, NULL);
318
    }
319
    return s->file;
320

    
321
fail:
322
    qemu_free(s);
323
    return NULL;
324
}
325

    
326
QEMUFile *qemu_fopen_socket(int fd)
327
{
328
    QEMUFileSocket *s = qemu_mallocz(sizeof(QEMUFileSocket));
329

    
330
    s->fd = fd;
331
    s->file = qemu_fopen_ops(s, NULL, socket_get_buffer, socket_close, 
332
                             NULL, NULL, NULL);
333
    return s->file;
334
}
335

    
336
static int file_put_buffer(void *opaque, const uint8_t *buf,
337
                            int64_t pos, int size)
338
{
339
    QEMUFileStdio *s = opaque;
340
    fseek(s->stdio_file, pos, SEEK_SET);
341
    return fwrite(buf, 1, size, s->stdio_file);
342
}
343

    
344
static int file_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
345
{
346
    QEMUFileStdio *s = opaque;
347
    fseek(s->stdio_file, pos, SEEK_SET);
348
    return fread(buf, 1, size, s->stdio_file);
349
}
350

    
351
QEMUFile *qemu_fopen(const char *filename, const char *mode)
352
{
353
    QEMUFileStdio *s;
354

    
355
    if (mode == NULL ||
356
        (mode[0] != 'r' && mode[0] != 'w') ||
357
        mode[1] != 'b' || mode[2] != 0) {
358
        fprintf(stderr, "qemu_fopen: Argument validity check failed\n");
359
        return NULL;
360
    }
361

    
362
    s = qemu_mallocz(sizeof(QEMUFileStdio));
363

    
364
    s->stdio_file = fopen(filename, mode);
365
    if (!s->stdio_file)
366
        goto fail;
367
    
368
    if(mode[0] == 'w') {
369
        s->file = qemu_fopen_ops(s, file_put_buffer, NULL, stdio_fclose, 
370
                                 NULL, NULL, NULL);
371
    } else {
372
        s->file = qemu_fopen_ops(s, NULL, file_get_buffer, stdio_fclose, 
373
                               NULL, NULL, NULL);
374
    }
375
    return s->file;
376
fail:
377
    qemu_free(s);
378
    return NULL;
379
}
380

    
381
static int block_put_buffer(void *opaque, const uint8_t *buf,
382
                           int64_t pos, int size)
383
{
384
    bdrv_save_vmstate(opaque, buf, pos, size);
385
    return size;
386
}
387

    
388
static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
389
{
390
    return bdrv_load_vmstate(opaque, buf, pos, size);
391
}
392

    
393
static int bdrv_fclose(void *opaque)
394
{
395
    return 0;
396
}
397

    
398
static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int is_writable)
399
{
400
    if (is_writable)
401
        return qemu_fopen_ops(bs, block_put_buffer, NULL, bdrv_fclose, 
402
                              NULL, NULL, NULL);
403
    return qemu_fopen_ops(bs, NULL, block_get_buffer, bdrv_fclose, NULL, NULL, NULL);
404
}
405

    
406
QEMUFile *qemu_fopen_ops(void *opaque, QEMUFilePutBufferFunc *put_buffer,
407
                         QEMUFileGetBufferFunc *get_buffer,
408
                         QEMUFileCloseFunc *close,
409
                         QEMUFileRateLimit *rate_limit,
410
                         QEMUFileSetRateLimit *set_rate_limit,
411
                         QEMUFileGetRateLimit *get_rate_limit)
412
{
413
    QEMUFile *f;
414

    
415
    f = qemu_mallocz(sizeof(QEMUFile));
416

    
417
    f->opaque = opaque;
418
    f->put_buffer = put_buffer;
419
    f->get_buffer = get_buffer;
420
    f->close = close;
421
    f->rate_limit = rate_limit;
422
    f->set_rate_limit = set_rate_limit;
423
    f->get_rate_limit = get_rate_limit;
424
    f->is_write = 0;
425

    
426
    return f;
427
}
428

    
429
int qemu_file_has_error(QEMUFile *f)
430
{
431
    return f->has_error;
432
}
433

    
434
void qemu_file_set_error(QEMUFile *f)
435
{
436
    f->has_error = 1;
437
}
438

    
439
void qemu_fflush(QEMUFile *f)
440
{
441
    if (!f->put_buffer)
442
        return;
443

    
444
    if (f->is_write && f->buf_index > 0) {
445
        int len;
446

    
447
        len = f->put_buffer(f->opaque, f->buf, f->buf_offset, f->buf_index);
448
        if (len > 0)
449
            f->buf_offset += f->buf_index;
450
        else
451
            f->has_error = 1;
452
        f->buf_index = 0;
453
    }
454
}
455

    
456
static void qemu_fill_buffer(QEMUFile *f)
457
{
458
    int len;
459

    
460
    if (!f->get_buffer)
461
        return;
462

    
463
    if (f->is_write)
464
        abort();
465

    
466
    len = f->get_buffer(f->opaque, f->buf, f->buf_offset, IO_BUF_SIZE);
467
    if (len > 0) {
468
        f->buf_index = 0;
469
        f->buf_size = len;
470
        f->buf_offset += len;
471
    } else if (len != -EAGAIN)
472
        f->has_error = 1;
473
}
474

    
475
int qemu_fclose(QEMUFile *f)
476
{
477
    int ret = 0;
478
    qemu_fflush(f);
479
    if (f->close)
480
        ret = f->close(f->opaque);
481
    qemu_free(f);
482
    return ret;
483
}
484

    
485
void qemu_file_put_notify(QEMUFile *f)
486
{
487
    f->put_buffer(f->opaque, NULL, 0, 0);
488
}
489

    
490
void qemu_put_buffer(QEMUFile *f, const uint8_t *buf, int size)
491
{
492
    int l;
493

    
494
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
495
        fprintf(stderr,
496
                "Attempted to write to buffer while read buffer is not empty\n");
497
        abort();
498
    }
499

    
500
    while (!f->has_error && size > 0) {
501
        l = IO_BUF_SIZE - f->buf_index;
502
        if (l > size)
503
            l = size;
504
        memcpy(f->buf + f->buf_index, buf, l);
505
        f->is_write = 1;
506
        f->buf_index += l;
507
        buf += l;
508
        size -= l;
509
        if (f->buf_index >= IO_BUF_SIZE)
510
            qemu_fflush(f);
511
    }
512
}
513

    
514
void qemu_put_byte(QEMUFile *f, int v)
515
{
516
    if (!f->has_error && f->is_write == 0 && f->buf_index > 0) {
517
        fprintf(stderr,
518
                "Attempted to write to buffer while read buffer is not empty\n");
519
        abort();
520
    }
521

    
522
    f->buf[f->buf_index++] = v;
523
    f->is_write = 1;
524
    if (f->buf_index >= IO_BUF_SIZE)
525
        qemu_fflush(f);
526
}
527

    
528
int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size1)
529
{
530
    int size, l;
531

    
532
    if (f->is_write)
533
        abort();
534

    
535
    size = size1;
536
    while (size > 0) {
537
        l = f->buf_size - f->buf_index;
538
        if (l == 0) {
539
            qemu_fill_buffer(f);
540
            l = f->buf_size - f->buf_index;
541
            if (l == 0)
542
                break;
543
        }
544
        if (l > size)
545
            l = size;
546
        memcpy(buf, f->buf + f->buf_index, l);
547
        f->buf_index += l;
548
        buf += l;
549
        size -= l;
550
    }
551
    return size1 - size;
552
}
553

    
554
int qemu_get_byte(QEMUFile *f)
555
{
556
    if (f->is_write)
557
        abort();
558

    
559
    if (f->buf_index >= f->buf_size) {
560
        qemu_fill_buffer(f);
561
        if (f->buf_index >= f->buf_size)
562
            return 0;
563
    }
564
    return f->buf[f->buf_index++];
565
}
566

    
567
int64_t qemu_ftell(QEMUFile *f)
568
{
569
    return f->buf_offset - f->buf_size + f->buf_index;
570
}
571

    
572
int64_t qemu_fseek(QEMUFile *f, int64_t pos, int whence)
573
{
574
    if (whence == SEEK_SET) {
575
        /* nothing to do */
576
    } else if (whence == SEEK_CUR) {
577
        pos += qemu_ftell(f);
578
    } else {
579
        /* SEEK_END not supported */
580
        return -1;
581
    }
582
    if (f->put_buffer) {
583
        qemu_fflush(f);
584
        f->buf_offset = pos;
585
    } else {
586
        f->buf_offset = pos;
587
        f->buf_index = 0;
588
        f->buf_size = 0;
589
    }
590
    return pos;
591
}
592

    
593
int qemu_file_rate_limit(QEMUFile *f)
594
{
595
    if (f->rate_limit)
596
        return f->rate_limit(f->opaque);
597

    
598
    return 0;
599
}
600

    
601
size_t qemu_file_get_rate_limit(QEMUFile *f)
602
{
603
    if (f->get_rate_limit)
604
        return f->get_rate_limit(f->opaque);
605

    
606
    return 0;
607
}
608

    
609
size_t qemu_file_set_rate_limit(QEMUFile *f, size_t new_rate)
610
{
611
    /* any failed or completed migration keeps its state to allow probing of
612
     * migration data, but has no associated file anymore */
613
    if (f && f->set_rate_limit)
614
        return f->set_rate_limit(f->opaque, new_rate);
615

    
616
    return 0;
617
}
618

    
619
void qemu_put_be16(QEMUFile *f, unsigned int v)
620
{
621
    qemu_put_byte(f, v >> 8);
622
    qemu_put_byte(f, v);
623
}
624

    
625
void qemu_put_be32(QEMUFile *f, unsigned int v)
626
{
627
    qemu_put_byte(f, v >> 24);
628
    qemu_put_byte(f, v >> 16);
629
    qemu_put_byte(f, v >> 8);
630
    qemu_put_byte(f, v);
631
}
632

    
633
void qemu_put_be64(QEMUFile *f, uint64_t v)
634
{
635
    qemu_put_be32(f, v >> 32);
636
    qemu_put_be32(f, v);
637
}
638

    
639
unsigned int qemu_get_be16(QEMUFile *f)
640
{
641
    unsigned int v;
642
    v = qemu_get_byte(f) << 8;
643
    v |= qemu_get_byte(f);
644
    return v;
645
}
646

    
647
unsigned int qemu_get_be32(QEMUFile *f)
648
{
649
    unsigned int v;
650
    v = qemu_get_byte(f) << 24;
651
    v |= qemu_get_byte(f) << 16;
652
    v |= qemu_get_byte(f) << 8;
653
    v |= qemu_get_byte(f);
654
    return v;
655
}
656

    
657
uint64_t qemu_get_be64(QEMUFile *f)
658
{
659
    uint64_t v;
660
    v = (uint64_t)qemu_get_be32(f) << 32;
661
    v |= qemu_get_be32(f);
662
    return v;
663
}
664

    
665
/* 8 bit int */
666

    
667
static int get_int8(QEMUFile *f, void *pv, size_t size)
668
{
669
    int8_t *v = pv;
670
    qemu_get_s8s(f, v);
671
    return 0;
672
}
673

    
674
static void put_int8(QEMUFile *f, void *pv, size_t size)
675
{
676
    int8_t *v = pv;
677
    qemu_put_s8s(f, v);
678
}
679

    
680
const VMStateInfo vmstate_info_int8 = {
681
    .name = "int8",
682
    .get  = get_int8,
683
    .put  = put_int8,
684
};
685

    
686
/* 16 bit int */
687

    
688
static int get_int16(QEMUFile *f, void *pv, size_t size)
689
{
690
    int16_t *v = pv;
691
    qemu_get_sbe16s(f, v);
692
    return 0;
693
}
694

    
695
static void put_int16(QEMUFile *f, void *pv, size_t size)
696
{
697
    int16_t *v = pv;
698
    qemu_put_sbe16s(f, v);
699
}
700

    
701
const VMStateInfo vmstate_info_int16 = {
702
    .name = "int16",
703
    .get  = get_int16,
704
    .put  = put_int16,
705
};
706

    
707
/* 32 bit int */
708

    
709
static int get_int32(QEMUFile *f, void *pv, size_t size)
710
{
711
    int32_t *v = pv;
712
    qemu_get_sbe32s(f, v);
713
    return 0;
714
}
715

    
716
static void put_int32(QEMUFile *f, void *pv, size_t size)
717
{
718
    int32_t *v = pv;
719
    qemu_put_sbe32s(f, v);
720
}
721

    
722
const VMStateInfo vmstate_info_int32 = {
723
    .name = "int32",
724
    .get  = get_int32,
725
    .put  = put_int32,
726
};
727

    
728
/* 32 bit int. See that the received value is the same than the one
729
   in the field */
730

    
731
static int get_int32_equal(QEMUFile *f, void *pv, size_t size)
732
{
733
    int32_t *v = pv;
734
    int32_t v2;
735
    qemu_get_sbe32s(f, &v2);
736

    
737
    if (*v == v2)
738
        return 0;
739
    return -EINVAL;
740
}
741

    
742
const VMStateInfo vmstate_info_int32_equal = {
743
    .name = "int32 equal",
744
    .get  = get_int32_equal,
745
    .put  = put_int32,
746
};
747

    
748
/* 32 bit int. See that the received value is the less or the same
749
   than the one in the field */
750

    
751
static int get_int32_le(QEMUFile *f, void *pv, size_t size)
752
{
753
    int32_t *old = pv;
754
    int32_t new;
755
    qemu_get_sbe32s(f, &new);
756

    
757
    if (*old <= new)
758
        return 0;
759
    return -EINVAL;
760
}
761

    
762
const VMStateInfo vmstate_info_int32_le = {
763
    .name = "int32 equal",
764
    .get  = get_int32_le,
765
    .put  = put_int32,
766
};
767

    
768
/* 64 bit int */
769

    
770
static int get_int64(QEMUFile *f, void *pv, size_t size)
771
{
772
    int64_t *v = pv;
773
    qemu_get_sbe64s(f, v);
774
    return 0;
775
}
776

    
777
static void put_int64(QEMUFile *f, void *pv, size_t size)
778
{
779
    int64_t *v = pv;
780
    qemu_put_sbe64s(f, v);
781
}
782

    
783
const VMStateInfo vmstate_info_int64 = {
784
    .name = "int64",
785
    .get  = get_int64,
786
    .put  = put_int64,
787
};
788

    
789
/* 8 bit unsigned int */
790

    
791
static int get_uint8(QEMUFile *f, void *pv, size_t size)
792
{
793
    uint8_t *v = pv;
794
    qemu_get_8s(f, v);
795
    return 0;
796
}
797

    
798
static void put_uint8(QEMUFile *f, void *pv, size_t size)
799
{
800
    uint8_t *v = pv;
801
    qemu_put_8s(f, v);
802
}
803

    
804
const VMStateInfo vmstate_info_uint8 = {
805
    .name = "uint8",
806
    .get  = get_uint8,
807
    .put  = put_uint8,
808
};
809

    
810
/* 16 bit unsigned int */
811

    
812
static int get_uint16(QEMUFile *f, void *pv, size_t size)
813
{
814
    uint16_t *v = pv;
815
    qemu_get_be16s(f, v);
816
    return 0;
817
}
818

    
819
static void put_uint16(QEMUFile *f, void *pv, size_t size)
820
{
821
    uint16_t *v = pv;
822
    qemu_put_be16s(f, v);
823
}
824

    
825
const VMStateInfo vmstate_info_uint16 = {
826
    .name = "uint16",
827
    .get  = get_uint16,
828
    .put  = put_uint16,
829
};
830

    
831
/* 32 bit unsigned int */
832

    
833
static int get_uint32(QEMUFile *f, void *pv, size_t size)
834
{
835
    uint32_t *v = pv;
836
    qemu_get_be32s(f, v);
837
    return 0;
838
}
839

    
840
static void put_uint32(QEMUFile *f, void *pv, size_t size)
841
{
842
    uint32_t *v = pv;
843
    qemu_put_be32s(f, v);
844
}
845

    
846
const VMStateInfo vmstate_info_uint32 = {
847
    .name = "uint32",
848
    .get  = get_uint32,
849
    .put  = put_uint32,
850
};
851

    
852
/* 64 bit unsigned int */
853

    
854
static int get_uint64(QEMUFile *f, void *pv, size_t size)
855
{
856
    uint64_t *v = pv;
857
    qemu_get_be64s(f, v);
858
    return 0;
859
}
860

    
861
static void put_uint64(QEMUFile *f, void *pv, size_t size)
862
{
863
    uint64_t *v = pv;
864
    qemu_put_be64s(f, v);
865
}
866

    
867
const VMStateInfo vmstate_info_uint64 = {
868
    .name = "uint64",
869
    .get  = get_uint64,
870
    .put  = put_uint64,
871
};
872

    
873
/* 8 bit int. See that the received value is the same than the one
874
   in the field */
875

    
876
static int get_uint8_equal(QEMUFile *f, void *pv, size_t size)
877
{
878
    uint8_t *v = pv;
879
    uint8_t v2;
880
    qemu_get_8s(f, &v2);
881

    
882
    if (*v == v2)
883
        return 0;
884
    return -EINVAL;
885
}
886

    
887
const VMStateInfo vmstate_info_uint8_equal = {
888
    .name = "uint8 equal",
889
    .get  = get_uint8_equal,
890
    .put  = put_uint8,
891
};
892

    
893
/* 16 bit unsigned int int. See that the received value is the same than the one
894
   in the field */
895

    
896
static int get_uint16_equal(QEMUFile *f, void *pv, size_t size)
897
{
898
    uint16_t *v = pv;
899
    uint16_t v2;
900
    qemu_get_be16s(f, &v2);
901

    
902
    if (*v == v2)
903
        return 0;
904
    return -EINVAL;
905
}
906

    
907
const VMStateInfo vmstate_info_uint16_equal = {
908
    .name = "uint16 equal",
909
    .get  = get_uint16_equal,
910
    .put  = put_uint16,
911
};
912

    
913
/* timers  */
914

    
915
static int get_timer(QEMUFile *f, void *pv, size_t size)
916
{
917
    QEMUTimer *v = pv;
918
    qemu_get_timer(f, v);
919
    return 0;
920
}
921

    
922
static void put_timer(QEMUFile *f, void *pv, size_t size)
923
{
924
    QEMUTimer *v = pv;
925
    qemu_put_timer(f, v);
926
}
927

    
928
const VMStateInfo vmstate_info_timer = {
929
    .name = "timer",
930
    .get  = get_timer,
931
    .put  = put_timer,
932
};
933

    
934
/* uint8_t buffers */
935

    
936
static int get_buffer(QEMUFile *f, void *pv, size_t size)
937
{
938
    uint8_t *v = pv;
939
    qemu_get_buffer(f, v, size);
940
    return 0;
941
}
942

    
943
static void put_buffer(QEMUFile *f, void *pv, size_t size)
944
{
945
    uint8_t *v = pv;
946
    qemu_put_buffer(f, v, size);
947
}
948

    
949
const VMStateInfo vmstate_info_buffer = {
950
    .name = "buffer",
951
    .get  = get_buffer,
952
    .put  = put_buffer,
953
};
954

    
955
/* unused buffers: space that was used for some fields that are
956
   not usefull anymore */
957

    
958
static int get_unused_buffer(QEMUFile *f, void *pv, size_t size)
959
{
960
    uint8_t buf[1024];
961
    int block_len;
962

    
963
    while (size > 0) {
964
        block_len = MIN(sizeof(buf), size);
965
        size -= block_len;
966
        qemu_get_buffer(f, buf, block_len);
967
    }
968
   return 0;
969
}
970

    
971
static void put_unused_buffer(QEMUFile *f, void *pv, size_t size)
972
{
973
    static const uint8_t buf[1024];
974
    int block_len;
975

    
976
    while (size > 0) {
977
        block_len = MIN(sizeof(buf), size);
978
        size -= block_len;
979
        qemu_put_buffer(f, buf, block_len);
980
    }
981
}
982

    
983
const VMStateInfo vmstate_info_unused_buffer = {
984
    .name = "unused_buffer",
985
    .get  = get_unused_buffer,
986
    .put  = put_unused_buffer,
987
};
988

    
989
typedef struct CompatEntry {
990
    char idstr[256];
991
    int instance_id;
992
} CompatEntry;
993

    
994
typedef struct SaveStateEntry {
995
    QTAILQ_ENTRY(SaveStateEntry) entry;
996
    char idstr[256];
997
    int instance_id;
998
    int alias_id;
999
    int version_id;
1000
    int section_id;
1001
    SaveSetParamsHandler *set_params;
1002
    SaveLiveStateHandler *save_live_state;
1003
    SaveStateHandler *save_state;
1004
    LoadStateHandler *load_state;
1005
    const VMStateDescription *vmsd;
1006
    void *opaque;
1007
    CompatEntry *compat;
1008
} SaveStateEntry;
1009

    
1010

    
1011
static QTAILQ_HEAD(savevm_handlers, SaveStateEntry) savevm_handlers =
1012
    QTAILQ_HEAD_INITIALIZER(savevm_handlers);
1013
static int global_section_id;
1014

    
1015
static int calculate_new_instance_id(const char *idstr)
1016
{
1017
    SaveStateEntry *se;
1018
    int instance_id = 0;
1019

    
1020
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1021
        if (strcmp(idstr, se->idstr) == 0
1022
            && instance_id <= se->instance_id) {
1023
            instance_id = se->instance_id + 1;
1024
        }
1025
    }
1026
    return instance_id;
1027
}
1028

    
1029
static int calculate_compat_instance_id(const char *idstr)
1030
{
1031
    SaveStateEntry *se;
1032
    int instance_id = 0;
1033

    
1034
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1035
        if (!se->compat)
1036
            continue;
1037

    
1038
        if (strcmp(idstr, se->compat->idstr) == 0
1039
            && instance_id <= se->compat->instance_id) {
1040
            instance_id = se->compat->instance_id + 1;
1041
        }
1042
    }
1043
    return instance_id;
1044
}
1045

    
1046
/* TODO: Individual devices generally have very little idea about the rest
1047
   of the system, so instance_id should be removed/replaced.
1048
   Meanwhile pass -1 as instance_id if you do not already have a clearly
1049
   distinguishing id for all instances of your device class. */
1050
int register_savevm_live(DeviceState *dev,
1051
                         const char *idstr,
1052
                         int instance_id,
1053
                         int version_id,
1054
                         SaveSetParamsHandler *set_params,
1055
                         SaveLiveStateHandler *save_live_state,
1056
                         SaveStateHandler *save_state,
1057
                         LoadStateHandler *load_state,
1058
                         void *opaque)
1059
{
1060
    SaveStateEntry *se;
1061

    
1062
    se = qemu_mallocz(sizeof(SaveStateEntry));
1063
    se->version_id = version_id;
1064
    se->section_id = global_section_id++;
1065
    se->set_params = set_params;
1066
    se->save_live_state = save_live_state;
1067
    se->save_state = save_state;
1068
    se->load_state = load_state;
1069
    se->opaque = opaque;
1070
    se->vmsd = NULL;
1071

    
1072
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1073
        char *id = dev->parent_bus->info->get_dev_path(dev);
1074
        if (id) {
1075
            pstrcpy(se->idstr, sizeof(se->idstr), id);
1076
            pstrcat(se->idstr, sizeof(se->idstr), "/");
1077
            qemu_free(id);
1078

    
1079
            se->compat = qemu_mallocz(sizeof(CompatEntry));
1080
            pstrcpy(se->compat->idstr, sizeof(se->compat->idstr), idstr);
1081
            se->compat->instance_id = instance_id == -1 ?
1082
                         calculate_compat_instance_id(idstr) : instance_id;
1083
            instance_id = -1;
1084
        }
1085
    }
1086
    pstrcat(se->idstr, sizeof(se->idstr), idstr);
1087

    
1088
    if (instance_id == -1) {
1089
        se->instance_id = calculate_new_instance_id(se->idstr);
1090
    } else {
1091
        se->instance_id = instance_id;
1092
    }
1093
    assert(!se->compat || se->instance_id == 0);
1094
    /* add at the end of list */
1095
    QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry);
1096
    return 0;
1097
}
1098

    
1099
int register_savevm(DeviceState *dev,
1100
                    const char *idstr,
1101
                    int instance_id,
1102
                    int version_id,
1103
                    SaveStateHandler *save_state,
1104
                    LoadStateHandler *load_state,
1105
                    void *opaque)
1106
{
1107
    return register_savevm_live(dev, idstr, instance_id, version_id,
1108
                                NULL, NULL, save_state, load_state, opaque);
1109
}
1110

    
1111
void unregister_savevm(DeviceState *dev, const char *idstr, void *opaque)
1112
{
1113
    SaveStateEntry *se, *new_se;
1114
    char id[256] = "";
1115

    
1116
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1117
        char *path = dev->parent_bus->info->get_dev_path(dev);
1118
        if (path) {
1119
            pstrcpy(id, sizeof(id), path);
1120
            pstrcat(id, sizeof(id), "/");
1121
            qemu_free(path);
1122
        }
1123
    }
1124
    pstrcat(id, sizeof(id), idstr);
1125

    
1126
    QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) {
1127
        if (strcmp(se->idstr, id) == 0 && se->opaque == opaque) {
1128
            QTAILQ_REMOVE(&savevm_handlers, se, entry);
1129
            qemu_free(se);
1130
        }
1131
    }
1132
}
1133

    
1134
int vmstate_register_with_alias_id(DeviceState *dev, int instance_id,
1135
                                   const VMStateDescription *vmsd,
1136
                                   void *opaque, int alias_id,
1137
                                   int required_for_version)
1138
{
1139
    SaveStateEntry *se;
1140

    
1141
    /* If this triggers, alias support can be dropped for the vmsd. */
1142
    assert(alias_id == -1 || required_for_version >= vmsd->minimum_version_id);
1143

    
1144
    se = qemu_mallocz(sizeof(SaveStateEntry));
1145
    se->version_id = vmsd->version_id;
1146
    se->section_id = global_section_id++;
1147
    se->save_live_state = NULL;
1148
    se->save_state = NULL;
1149
    se->load_state = NULL;
1150
    se->opaque = opaque;
1151
    se->vmsd = vmsd;
1152
    se->alias_id = alias_id;
1153

    
1154
    if (dev && dev->parent_bus && dev->parent_bus->info->get_dev_path) {
1155
        char *id = dev->parent_bus->info->get_dev_path(dev);
1156
        if (id) {
1157
            pstrcpy(se->idstr, sizeof(se->idstr), id);
1158
            pstrcat(se->idstr, sizeof(se->idstr), "/");
1159
            qemu_free(id);
1160

    
1161
            se->compat = qemu_mallocz(sizeof(CompatEntry));
1162
            pstrcpy(se->compat->idstr, sizeof(se->compat->idstr), vmsd->name);
1163
            se->compat->instance_id = instance_id == -1 ?
1164
                         calculate_compat_instance_id(vmsd->name) : instance_id;
1165
            instance_id = -1;
1166
        }
1167
    }
1168
    pstrcat(se->idstr, sizeof(se->idstr), vmsd->name);
1169

    
1170
    if (instance_id == -1) {
1171
        se->instance_id = calculate_new_instance_id(se->idstr);
1172
    } else {
1173
        se->instance_id = instance_id;
1174
    }
1175
    assert(!se->compat || se->instance_id == 0);
1176
    /* add at the end of list */
1177
    QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry);
1178
    return 0;
1179
}
1180

    
1181
int vmstate_register(DeviceState *dev, int instance_id,
1182
                     const VMStateDescription *vmsd, void *opaque)
1183
{
1184
    return vmstate_register_with_alias_id(dev, instance_id, vmsd,
1185
                                          opaque, -1, 0);
1186
}
1187

    
1188
void vmstate_unregister(DeviceState *dev, const VMStateDescription *vmsd,
1189
                        void *opaque)
1190
{
1191
    SaveStateEntry *se, *new_se;
1192

    
1193
    QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) {
1194
        if (se->vmsd == vmsd && se->opaque == opaque) {
1195
            QTAILQ_REMOVE(&savevm_handlers, se, entry);
1196
            qemu_free(se);
1197
        }
1198
    }
1199
}
1200

    
1201
int vmstate_load_state(QEMUFile *f, const VMStateDescription *vmsd,
1202
                       void *opaque, int version_id)
1203
{
1204
    VMStateField *field = vmsd->fields;
1205

    
1206
    if (version_id > vmsd->version_id) {
1207
        return -EINVAL;
1208
    }
1209
    if (version_id < vmsd->minimum_version_id_old) {
1210
        return -EINVAL;
1211
    }
1212
    if  (version_id < vmsd->minimum_version_id) {
1213
        return vmsd->load_state_old(f, opaque, version_id);
1214
    }
1215
    if (vmsd->pre_load) {
1216
        int ret = vmsd->pre_load(opaque);
1217
        if (ret)
1218
            return ret;
1219
    }
1220
    while(field->name) {
1221
        if ((field->field_exists &&
1222
             field->field_exists(opaque, version_id)) ||
1223
            (!field->field_exists &&
1224
             field->version_id <= version_id)) {
1225
            void *base_addr = opaque + field->offset;
1226
            int ret, i, n_elems = 1;
1227
            int size = field->size;
1228

    
1229
            if (field->flags & VMS_VBUFFER) {
1230
                size = *(int32_t *)(opaque+field->size_offset);
1231
                if (field->flags & VMS_MULTIPLY) {
1232
                    size *= field->size;
1233
                }
1234
            }
1235
            if (field->flags & VMS_ARRAY) {
1236
                n_elems = field->num;
1237
            } else if (field->flags & VMS_VARRAY_INT32) {
1238
                n_elems = *(int32_t *)(opaque+field->num_offset);
1239
            } else if (field->flags & VMS_VARRAY_UINT16) {
1240
                n_elems = *(uint16_t *)(opaque+field->num_offset);
1241
            }
1242
            if (field->flags & VMS_POINTER) {
1243
                base_addr = *(void **)base_addr + field->start;
1244
            }
1245
            for (i = 0; i < n_elems; i++) {
1246
                void *addr = base_addr + size * i;
1247

    
1248
                if (field->flags & VMS_ARRAY_OF_POINTER) {
1249
                    addr = *(void **)addr;
1250
                }
1251
                if (field->flags & VMS_STRUCT) {
1252
                    ret = vmstate_load_state(f, field->vmsd, addr, field->vmsd->version_id);
1253
                } else {
1254
                    ret = field->info->get(f, addr, size);
1255

    
1256
                }
1257
                if (ret < 0) {
1258
                    return ret;
1259
                }
1260
            }
1261
        }
1262
        field++;
1263
    }
1264
    if (vmsd->post_load) {
1265
        return vmsd->post_load(opaque, version_id);
1266
    }
1267
    return 0;
1268
}
1269

    
1270
void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd,
1271
                        void *opaque)
1272
{
1273
    VMStateField *field = vmsd->fields;
1274

    
1275
    if (vmsd->pre_save) {
1276
        vmsd->pre_save(opaque);
1277
    }
1278
    while(field->name) {
1279
        if (!field->field_exists ||
1280
            field->field_exists(opaque, vmsd->version_id)) {
1281
            void *base_addr = opaque + field->offset;
1282
            int i, n_elems = 1;
1283
            int size = field->size;
1284

    
1285
            if (field->flags & VMS_VBUFFER) {
1286
                size = *(int32_t *)(opaque+field->size_offset);
1287
                if (field->flags & VMS_MULTIPLY) {
1288
                    size *= field->size;
1289
                }
1290
            }
1291
            if (field->flags & VMS_ARRAY) {
1292
                n_elems = field->num;
1293
            } else if (field->flags & VMS_VARRAY_INT32) {
1294
                n_elems = *(int32_t *)(opaque+field->num_offset);
1295
            } else if (field->flags & VMS_VARRAY_UINT16) {
1296
                n_elems = *(uint16_t *)(opaque+field->num_offset);
1297
            }
1298
            if (field->flags & VMS_POINTER) {
1299
                base_addr = *(void **)base_addr + field->start;
1300
            }
1301
            for (i = 0; i < n_elems; i++) {
1302
                void *addr = base_addr + size * i;
1303

    
1304
                if (field->flags & VMS_ARRAY_OF_POINTER) {
1305
                    addr = *(void **)addr;
1306
                }
1307
                if (field->flags & VMS_STRUCT) {
1308
                    vmstate_save_state(f, field->vmsd, addr);
1309
                } else {
1310
                    field->info->put(f, addr, size);
1311
                }
1312
            }
1313
        }
1314
        field++;
1315
    }
1316
}
1317

    
1318
static int vmstate_load(QEMUFile *f, SaveStateEntry *se, int version_id)
1319
{
1320
    if (!se->vmsd) {         /* Old style */
1321
        return se->load_state(f, se->opaque, version_id);
1322
    }
1323
    return vmstate_load_state(f, se->vmsd, se->opaque, version_id);
1324
}
1325

    
1326
static void vmstate_save(QEMUFile *f, SaveStateEntry *se)
1327
{
1328
    if (!se->vmsd) {         /* Old style */
1329
        se->save_state(f, se->opaque);
1330
        return;
1331
    }
1332
    vmstate_save_state(f,se->vmsd, se->opaque);
1333
}
1334

    
1335
#define QEMU_VM_FILE_MAGIC           0x5145564d
1336
#define QEMU_VM_FILE_VERSION_COMPAT  0x00000002
1337
#define QEMU_VM_FILE_VERSION         0x00000003
1338

    
1339
#define QEMU_VM_EOF                  0x00
1340
#define QEMU_VM_SECTION_START        0x01
1341
#define QEMU_VM_SECTION_PART         0x02
1342
#define QEMU_VM_SECTION_END          0x03
1343
#define QEMU_VM_SECTION_FULL         0x04
1344

    
1345
int qemu_savevm_state_begin(Monitor *mon, QEMUFile *f, int blk_enable,
1346
                            int shared)
1347
{
1348
    SaveStateEntry *se;
1349

    
1350
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1351
        if(se->set_params == NULL) {
1352
            continue;
1353
        }
1354
        se->set_params(blk_enable, shared, se->opaque);
1355
    }
1356
    
1357
    qemu_put_be32(f, QEMU_VM_FILE_MAGIC);
1358
    qemu_put_be32(f, QEMU_VM_FILE_VERSION);
1359

    
1360
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1361
        int len;
1362

    
1363
        if (se->save_live_state == NULL)
1364
            continue;
1365

    
1366
        /* Section type */
1367
        qemu_put_byte(f, QEMU_VM_SECTION_START);
1368
        qemu_put_be32(f, se->section_id);
1369

    
1370
        /* ID string */
1371
        len = strlen(se->idstr);
1372
        qemu_put_byte(f, len);
1373
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1374

    
1375
        qemu_put_be32(f, se->instance_id);
1376
        qemu_put_be32(f, se->version_id);
1377

    
1378
        se->save_live_state(mon, f, QEMU_VM_SECTION_START, se->opaque);
1379
    }
1380

    
1381
    if (qemu_file_has_error(f)) {
1382
        qemu_savevm_state_cancel(mon, f);
1383
        return -EIO;
1384
    }
1385

    
1386
    return 0;
1387
}
1388

    
1389
int qemu_savevm_state_iterate(Monitor *mon, QEMUFile *f)
1390
{
1391
    SaveStateEntry *se;
1392
    int ret = 1;
1393

    
1394
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1395
        if (se->save_live_state == NULL)
1396
            continue;
1397

    
1398
        /* Section type */
1399
        qemu_put_byte(f, QEMU_VM_SECTION_PART);
1400
        qemu_put_be32(f, se->section_id);
1401

    
1402
        ret = se->save_live_state(mon, f, QEMU_VM_SECTION_PART, se->opaque);
1403
        if (!ret) {
1404
            /* Do not proceed to the next vmstate before this one reported
1405
               completion of the current stage. This serializes the migration
1406
               and reduces the probability that a faster changing state is
1407
               synchronized over and over again. */
1408
            break;
1409
        }
1410
    }
1411

    
1412
    if (ret)
1413
        return 1;
1414

    
1415
    if (qemu_file_has_error(f)) {
1416
        qemu_savevm_state_cancel(mon, f);
1417
        return -EIO;
1418
    }
1419

    
1420
    return 0;
1421
}
1422

    
1423
int qemu_savevm_state_complete(Monitor *mon, QEMUFile *f)
1424
{
1425
    SaveStateEntry *se;
1426

    
1427
    cpu_synchronize_all_states();
1428

    
1429
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1430
        if (se->save_live_state == NULL)
1431
            continue;
1432

    
1433
        /* Section type */
1434
        qemu_put_byte(f, QEMU_VM_SECTION_END);
1435
        qemu_put_be32(f, se->section_id);
1436

    
1437
        se->save_live_state(mon, f, QEMU_VM_SECTION_END, se->opaque);
1438
    }
1439

    
1440
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1441
        int len;
1442

    
1443
        if (se->save_state == NULL && se->vmsd == NULL)
1444
            continue;
1445

    
1446
        /* Section type */
1447
        qemu_put_byte(f, QEMU_VM_SECTION_FULL);
1448
        qemu_put_be32(f, se->section_id);
1449

    
1450
        /* ID string */
1451
        len = strlen(se->idstr);
1452
        qemu_put_byte(f, len);
1453
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1454

    
1455
        qemu_put_be32(f, se->instance_id);
1456
        qemu_put_be32(f, se->version_id);
1457

    
1458
        vmstate_save(f, se);
1459
    }
1460

    
1461
    qemu_put_byte(f, QEMU_VM_EOF);
1462

    
1463
    if (qemu_file_has_error(f))
1464
        return -EIO;
1465

    
1466
    return 0;
1467
}
1468

    
1469
void qemu_savevm_state_cancel(Monitor *mon, QEMUFile *f)
1470
{
1471
    SaveStateEntry *se;
1472

    
1473
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1474
        if (se->save_live_state) {
1475
            se->save_live_state(mon, f, -1, se->opaque);
1476
        }
1477
    }
1478
}
1479

    
1480
static int qemu_savevm_state(Monitor *mon, QEMUFile *f)
1481
{
1482
    int saved_vm_running;
1483
    int ret;
1484

    
1485
    saved_vm_running = vm_running;
1486
    vm_stop(0);
1487

    
1488
    bdrv_flush_all();
1489

    
1490
    ret = qemu_savevm_state_begin(mon, f, 0, 0);
1491
    if (ret < 0)
1492
        goto out;
1493

    
1494
    do {
1495
        ret = qemu_savevm_state_iterate(mon, f);
1496
        if (ret < 0)
1497
            goto out;
1498
    } while (ret == 0);
1499

    
1500
    ret = qemu_savevm_state_complete(mon, f);
1501

    
1502
out:
1503
    if (qemu_file_has_error(f))
1504
        ret = -EIO;
1505

    
1506
    if (!ret && saved_vm_running)
1507
        vm_start();
1508

    
1509
    return ret;
1510
}
1511

    
1512
static SaveStateEntry *find_se(const char *idstr, int instance_id)
1513
{
1514
    SaveStateEntry *se;
1515

    
1516
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1517
        if (!strcmp(se->idstr, idstr) &&
1518
            (instance_id == se->instance_id ||
1519
             instance_id == se->alias_id))
1520
            return se;
1521
        /* Migrating from an older version? */
1522
        if (strstr(se->idstr, idstr) && se->compat) {
1523
            if (!strcmp(se->compat->idstr, idstr) &&
1524
                (instance_id == se->compat->instance_id ||
1525
                 instance_id == se->alias_id))
1526
                return se;
1527
        }
1528
    }
1529
    return NULL;
1530
}
1531

    
1532
typedef struct LoadStateEntry {
1533
    QLIST_ENTRY(LoadStateEntry) entry;
1534
    SaveStateEntry *se;
1535
    int section_id;
1536
    int version_id;
1537
} LoadStateEntry;
1538

    
1539
int qemu_loadvm_state(QEMUFile *f)
1540
{
1541
    QLIST_HEAD(, LoadStateEntry) loadvm_handlers =
1542
        QLIST_HEAD_INITIALIZER(loadvm_handlers);
1543
    LoadStateEntry *le, *new_le;
1544
    uint8_t section_type;
1545
    unsigned int v;
1546
    int ret;
1547

    
1548
    v = qemu_get_be32(f);
1549
    if (v != QEMU_VM_FILE_MAGIC)
1550
        return -EINVAL;
1551

    
1552
    v = qemu_get_be32(f);
1553
    if (v == QEMU_VM_FILE_VERSION_COMPAT) {
1554
        fprintf(stderr, "SaveVM v2 format is obsolete and don't work anymore\n");
1555
        return -ENOTSUP;
1556
    }
1557
    if (v != QEMU_VM_FILE_VERSION)
1558
        return -ENOTSUP;
1559

    
1560
    while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
1561
        uint32_t instance_id, version_id, section_id;
1562
        SaveStateEntry *se;
1563
        char idstr[257];
1564
        int len;
1565

    
1566
        switch (section_type) {
1567
        case QEMU_VM_SECTION_START:
1568
        case QEMU_VM_SECTION_FULL:
1569
            /* Read section start */
1570
            section_id = qemu_get_be32(f);
1571
            len = qemu_get_byte(f);
1572
            qemu_get_buffer(f, (uint8_t *)idstr, len);
1573
            idstr[len] = 0;
1574
            instance_id = qemu_get_be32(f);
1575
            version_id = qemu_get_be32(f);
1576

    
1577
            /* Find savevm section */
1578
            se = find_se(idstr, instance_id);
1579
            if (se == NULL) {
1580
                fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
1581
                ret = -EINVAL;
1582
                goto out;
1583
            }
1584

    
1585
            /* Validate version */
1586
            if (version_id > se->version_id) {
1587
                fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
1588
                        version_id, idstr, se->version_id);
1589
                ret = -EINVAL;
1590
                goto out;
1591
            }
1592

    
1593
            /* Add entry */
1594
            le = qemu_mallocz(sizeof(*le));
1595

    
1596
            le->se = se;
1597
            le->section_id = section_id;
1598
            le->version_id = version_id;
1599
            QLIST_INSERT_HEAD(&loadvm_handlers, le, entry);
1600

    
1601
            ret = vmstate_load(f, le->se, le->version_id);
1602
            if (ret < 0) {
1603
                fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
1604
                        instance_id, idstr);
1605
                goto out;
1606
            }
1607
            break;
1608
        case QEMU_VM_SECTION_PART:
1609
        case QEMU_VM_SECTION_END:
1610
            section_id = qemu_get_be32(f);
1611

    
1612
            QLIST_FOREACH(le, &loadvm_handlers, entry) {
1613
                if (le->section_id == section_id) {
1614
                    break;
1615
                }
1616
            }
1617
            if (le == NULL) {
1618
                fprintf(stderr, "Unknown savevm section %d\n", section_id);
1619
                ret = -EINVAL;
1620
                goto out;
1621
            }
1622

    
1623
            ret = vmstate_load(f, le->se, le->version_id);
1624
            if (ret < 0) {
1625
                fprintf(stderr, "qemu: warning: error while loading state section id %d\n",
1626
                        section_id);
1627
                goto out;
1628
            }
1629
            break;
1630
        default:
1631
            fprintf(stderr, "Unknown savevm section type %d\n", section_type);
1632
            ret = -EINVAL;
1633
            goto out;
1634
        }
1635
    }
1636

    
1637
    cpu_synchronize_all_post_init();
1638

    
1639
    ret = 0;
1640

    
1641
out:
1642
    QLIST_FOREACH_SAFE(le, &loadvm_handlers, entry, new_le) {
1643
        QLIST_REMOVE(le, entry);
1644
        qemu_free(le);
1645
    }
1646

    
1647
    if (qemu_file_has_error(f))
1648
        ret = -EIO;
1649

    
1650
    return ret;
1651
}
1652

    
1653
static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
1654
                              const char *name)
1655
{
1656
    QEMUSnapshotInfo *sn_tab, *sn;
1657
    int nb_sns, i, ret;
1658

    
1659
    ret = -ENOENT;
1660
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1661
    if (nb_sns < 0)
1662
        return ret;
1663
    for(i = 0; i < nb_sns; i++) {
1664
        sn = &sn_tab[i];
1665
        if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
1666
            *sn_info = *sn;
1667
            ret = 0;
1668
            break;
1669
        }
1670
    }
1671
    qemu_free(sn_tab);
1672
    return ret;
1673
}
1674

    
1675
/*
1676
 * Deletes snapshots of a given name in all opened images.
1677
 */
1678
static int del_existing_snapshots(Monitor *mon, const char *name)
1679
{
1680
    BlockDriverState *bs;
1681
    QEMUSnapshotInfo sn1, *snapshot = &sn1;
1682
    int ret;
1683

    
1684
    bs = NULL;
1685
    while ((bs = bdrv_next(bs))) {
1686
        if (bdrv_can_snapshot(bs) &&
1687
            bdrv_snapshot_find(bs, snapshot, name) >= 0)
1688
        {
1689
            ret = bdrv_snapshot_delete(bs, name);
1690
            if (ret < 0) {
1691
                monitor_printf(mon,
1692
                               "Error while deleting snapshot on '%s'\n",
1693
                               bdrv_get_device_name(bs));
1694
                return -1;
1695
            }
1696
        }
1697
    }
1698

    
1699
    return 0;
1700
}
1701

    
1702
void do_savevm(Monitor *mon, const QDict *qdict)
1703
{
1704
    BlockDriverState *bs, *bs1;
1705
    QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1;
1706
    int ret;
1707
    QEMUFile *f;
1708
    int saved_vm_running;
1709
    uint32_t vm_state_size;
1710
#ifdef _WIN32
1711
    struct _timeb tb;
1712
#else
1713
    struct timeval tv;
1714
#endif
1715
    const char *name = qdict_get_try_str(qdict, "name");
1716

    
1717
    /* Verify if there is a device that doesn't support snapshots and is writable */
1718
    bs = NULL;
1719
    while ((bs = bdrv_next(bs))) {
1720

    
1721
        if (bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
1722
            continue;
1723
        }
1724

    
1725
        if (!bdrv_can_snapshot(bs)) {
1726
            monitor_printf(mon, "Device '%s' is writable but does not support snapshots.\n",
1727
                               bdrv_get_device_name(bs));
1728
            return;
1729
        }
1730
    }
1731

    
1732
    bs = bdrv_snapshots();
1733
    if (!bs) {
1734
        monitor_printf(mon, "No block device can accept snapshots\n");
1735
        return;
1736
    }
1737
    /* ??? Should this occur after vm_stop?  */
1738
    qemu_aio_flush();
1739

    
1740
    saved_vm_running = vm_running;
1741
    vm_stop(0);
1742

    
1743
    memset(sn, 0, sizeof(*sn));
1744
    if (name) {
1745
        ret = bdrv_snapshot_find(bs, old_sn, name);
1746
        if (ret >= 0) {
1747
            pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
1748
            pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
1749
        } else {
1750
            pstrcpy(sn->name, sizeof(sn->name), name);
1751
        }
1752
    }
1753

    
1754
    /* fill auxiliary fields */
1755
#ifdef _WIN32
1756
    _ftime(&tb);
1757
    sn->date_sec = tb.time;
1758
    sn->date_nsec = tb.millitm * 1000000;
1759
#else
1760
    gettimeofday(&tv, NULL);
1761
    sn->date_sec = tv.tv_sec;
1762
    sn->date_nsec = tv.tv_usec * 1000;
1763
#endif
1764
    sn->vm_clock_nsec = qemu_get_clock(vm_clock);
1765

    
1766
    /* Delete old snapshots of the same name */
1767
    if (name && del_existing_snapshots(mon, name) < 0) {
1768
        goto the_end;
1769
    }
1770

    
1771
    /* save the VM state */
1772
    f = qemu_fopen_bdrv(bs, 1);
1773
    if (!f) {
1774
        monitor_printf(mon, "Could not open VM state file\n");
1775
        goto the_end;
1776
    }
1777
    ret = qemu_savevm_state(mon, f);
1778
    vm_state_size = qemu_ftell(f);
1779
    qemu_fclose(f);
1780
    if (ret < 0) {
1781
        monitor_printf(mon, "Error %d while writing VM\n", ret);
1782
        goto the_end;
1783
    }
1784

    
1785
    /* create the snapshots */
1786

    
1787
    bs1 = NULL;
1788
    while ((bs1 = bdrv_next(bs1))) {
1789
        if (bdrv_can_snapshot(bs1)) {
1790
            /* Write VM state size only to the image that contains the state */
1791
            sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
1792
            ret = bdrv_snapshot_create(bs1, sn);
1793
            if (ret < 0) {
1794
                monitor_printf(mon, "Error while creating snapshot on '%s'\n",
1795
                               bdrv_get_device_name(bs1));
1796
            }
1797
        }
1798
    }
1799

    
1800
 the_end:
1801
    if (saved_vm_running)
1802
        vm_start();
1803
}
1804

    
1805
int load_vmstate(const char *name)
1806
{
1807
    BlockDriverState *bs, *bs1;
1808
    QEMUSnapshotInfo sn;
1809
    QEMUFile *f;
1810
    int ret;
1811

    
1812
    /* Verify if there is a device that doesn't support snapshots and is writable */
1813
    bs = NULL;
1814
    while ((bs = bdrv_next(bs))) {
1815

    
1816
        if (bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
1817
            continue;
1818
        }
1819

    
1820
        if (!bdrv_can_snapshot(bs)) {
1821
            error_report("Device '%s' is writable but does not support snapshots.",
1822
                               bdrv_get_device_name(bs));
1823
            return -ENOTSUP;
1824
        }
1825
    }
1826

    
1827
    bs = bdrv_snapshots();
1828
    if (!bs) {
1829
        error_report("No block device supports snapshots");
1830
        return -EINVAL;
1831
    }
1832

    
1833
    /* Flush all IO requests so they don't interfere with the new state.  */
1834
    qemu_aio_flush();
1835

    
1836
    bs1 = NULL;
1837
    while ((bs1 = bdrv_next(bs1))) {
1838
        if (bdrv_can_snapshot(bs1)) {
1839
            ret = bdrv_snapshot_goto(bs1, name);
1840
            if (ret < 0) {
1841
                switch(ret) {
1842
                case -ENOTSUP:
1843
                    error_report("%sSnapshots not supported on device '%s'",
1844
                                 bs != bs1 ? "Warning: " : "",
1845
                                 bdrv_get_device_name(bs1));
1846
                    break;
1847
                case -ENOENT:
1848
                    error_report("%sCould not find snapshot '%s' on device '%s'",
1849
                                 bs != bs1 ? "Warning: " : "",
1850
                                 name, bdrv_get_device_name(bs1));
1851
                    break;
1852
                default:
1853
                    error_report("%sError %d while activating snapshot on '%s'",
1854
                                 bs != bs1 ? "Warning: " : "",
1855
                                 ret, bdrv_get_device_name(bs1));
1856
                    break;
1857
                }
1858
                /* fatal on snapshot block device */
1859
                if (bs == bs1)
1860
                    return 0;
1861
            }
1862
        }
1863
    }
1864

    
1865
    /* Don't even try to load empty VM states */
1866
    ret = bdrv_snapshot_find(bs, &sn, name);
1867
    if ((ret >= 0) && (sn.vm_state_size == 0))
1868
        return -EINVAL;
1869

    
1870
    /* restore the VM state */
1871
    f = qemu_fopen_bdrv(bs, 0);
1872
    if (!f) {
1873
        error_report("Could not open VM state file");
1874
        return -EINVAL;
1875
    }
1876
    ret = qemu_loadvm_state(f);
1877
    qemu_fclose(f);
1878
    if (ret < 0) {
1879
        error_report("Error %d while loading VM state", ret);
1880
        return ret;
1881
    }
1882
    return 0;
1883
}
1884

    
1885
void do_delvm(Monitor *mon, const QDict *qdict)
1886
{
1887
    BlockDriverState *bs, *bs1;
1888
    int ret;
1889
    const char *name = qdict_get_str(qdict, "name");
1890

    
1891
    bs = bdrv_snapshots();
1892
    if (!bs) {
1893
        monitor_printf(mon, "No block device supports snapshots\n");
1894
        return;
1895
    }
1896

    
1897
    bs1 = NULL;
1898
    while ((bs1 = bdrv_next(bs1))) {
1899
        if (bdrv_can_snapshot(bs1)) {
1900
            ret = bdrv_snapshot_delete(bs1, name);
1901
            if (ret < 0) {
1902
                if (ret == -ENOTSUP)
1903
                    monitor_printf(mon,
1904
                                   "Snapshots not supported on device '%s'\n",
1905
                                   bdrv_get_device_name(bs1));
1906
                else
1907
                    monitor_printf(mon, "Error %d while deleting snapshot on "
1908
                                   "'%s'\n", ret, bdrv_get_device_name(bs1));
1909
            }
1910
        }
1911
    }
1912
}
1913

    
1914
void do_info_snapshots(Monitor *mon)
1915
{
1916
    BlockDriverState *bs, *bs1;
1917
    QEMUSnapshotInfo *sn_tab, *sn;
1918
    int nb_sns, i;
1919
    char buf[256];
1920

    
1921
    bs = bdrv_snapshots();
1922
    if (!bs) {
1923
        monitor_printf(mon, "No available block device supports snapshots\n");
1924
        return;
1925
    }
1926
    monitor_printf(mon, "Snapshot devices:");
1927
    bs1 = NULL;
1928
    while ((bs1 = bdrv_next(bs1))) {
1929
        if (bdrv_can_snapshot(bs1)) {
1930
            if (bs == bs1)
1931
                monitor_printf(mon, " %s", bdrv_get_device_name(bs1));
1932
        }
1933
    }
1934
    monitor_printf(mon, "\n");
1935

    
1936
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1937
    if (nb_sns < 0) {
1938
        monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
1939
        return;
1940
    }
1941
    monitor_printf(mon, "Snapshot list (from %s):\n",
1942
                   bdrv_get_device_name(bs));
1943
    monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
1944
    for(i = 0; i < nb_sns; i++) {
1945
        sn = &sn_tab[i];
1946
        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
1947
    }
1948
    qemu_free(sn_tab);
1949
}