Statistics
| Branch: | Revision:

root / savevm.c @ f9092b10

History | View | Annotate | Download (46.2 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 "net.h"
76
#include "monitor.h"
77
#include "sysemu.h"
78
#include "qemu-timer.h"
79
#include "qemu-char.h"
80
#include "blockdev.h"
81
#include "audio/audio.h"
82
#include "migration.h"
83
#include "qemu_socket.h"
84
#include "qemu-queue.h"
85

    
86
#define SELF_ANNOUNCE_ROUNDS 5
87

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

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

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

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

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

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

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

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

    
130

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

    
136
    qemu_foreach_nic(qemu_announce_self_iter, NULL);
137

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

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

    
155
/***********************************************************/
156
/* savevm/loadvm support */
157

    
158
#define IO_BUF_SIZE 32768
159

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

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

    
176
    int has_error;
177
};
178

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

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

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

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

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

    
203
    return len;
204
}
205

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

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

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

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

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

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

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

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

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

    
260
    s->stdio_file = stdio_file;
261

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

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

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

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

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

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

    
292
    return fd;
293
}
294

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    
425
    return f;
426
}
427

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    
597
    return 0;
598
}
599

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

    
605
    return 0;
606
}
607

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

    
615
    return 0;
616
}
617

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

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

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

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

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

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

    
664
/* 8 bit int */
665

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

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

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

    
685
/* 16 bit int */
686

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

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

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

    
706
/* 32 bit int */
707

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

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

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

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

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

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

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

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

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

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

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

    
767
/* 64 bit int */
768

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

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

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

    
788
/* 8 bit unsigned int */
789

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

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

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

    
809
/* 16 bit unsigned int */
810

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

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

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

    
830
/* 32 bit unsigned int */
831

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

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

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

    
851
/* 64 bit unsigned int */
852

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

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

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

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

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

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

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

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

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

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

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

    
912
/* timers  */
913

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

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

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

    
933
/* uint8_t buffers */
934

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

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

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

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

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

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

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

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

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

    
988
typedef struct SaveStateEntry {
989
    QTAILQ_ENTRY(SaveStateEntry) entry;
990
    char idstr[256];
991
    int instance_id;
992
    int alias_id;
993
    int version_id;
994
    int section_id;
995
    SaveSetParamsHandler *set_params;
996
    SaveLiveStateHandler *save_live_state;
997
    SaveStateHandler *save_state;
998
    LoadStateHandler *load_state;
999
    const VMStateDescription *vmsd;
1000
    void *opaque;
1001
} SaveStateEntry;
1002

    
1003

    
1004
static QTAILQ_HEAD(savevm_handlers, SaveStateEntry) savevm_handlers =
1005
    QTAILQ_HEAD_INITIALIZER(savevm_handlers);
1006
static int global_section_id;
1007

    
1008
static int calculate_new_instance_id(const char *idstr)
1009
{
1010
    SaveStateEntry *se;
1011
    int instance_id = 0;
1012

    
1013
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1014
        if (strcmp(idstr, se->idstr) == 0
1015
            && instance_id <= se->instance_id) {
1016
            instance_id = se->instance_id + 1;
1017
        }
1018
    }
1019
    return instance_id;
1020
}
1021

    
1022
/* TODO: Individual devices generally have very little idea about the rest
1023
   of the system, so instance_id should be removed/replaced.
1024
   Meanwhile pass -1 as instance_id if you do not already have a clearly
1025
   distinguishing id for all instances of your device class. */
1026
int register_savevm_live(const char *idstr,
1027
                         int instance_id,
1028
                         int version_id,
1029
                         SaveSetParamsHandler *set_params,
1030
                         SaveLiveStateHandler *save_live_state,
1031
                         SaveStateHandler *save_state,
1032
                         LoadStateHandler *load_state,
1033
                         void *opaque)
1034
{
1035
    SaveStateEntry *se;
1036

    
1037
    se = qemu_mallocz(sizeof(SaveStateEntry));
1038
    pstrcpy(se->idstr, sizeof(se->idstr), idstr);
1039
    se->version_id = version_id;
1040
    se->section_id = global_section_id++;
1041
    se->set_params = set_params;
1042
    se->save_live_state = save_live_state;
1043
    se->save_state = save_state;
1044
    se->load_state = load_state;
1045
    se->opaque = opaque;
1046
    se->vmsd = NULL;
1047

    
1048
    if (instance_id == -1) {
1049
        se->instance_id = calculate_new_instance_id(idstr);
1050
    } else {
1051
        se->instance_id = instance_id;
1052
    }
1053
    /* add at the end of list */
1054
    QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry);
1055
    return 0;
1056
}
1057

    
1058
int register_savevm(const char *idstr,
1059
                    int instance_id,
1060
                    int version_id,
1061
                    SaveStateHandler *save_state,
1062
                    LoadStateHandler *load_state,
1063
                    void *opaque)
1064
{
1065
    return register_savevm_live(idstr, instance_id, version_id,
1066
                                NULL, NULL, save_state, load_state, opaque);
1067
}
1068

    
1069
void unregister_savevm(const char *idstr, void *opaque)
1070
{
1071
    SaveStateEntry *se, *new_se;
1072

    
1073
    QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) {
1074
        if (strcmp(se->idstr, idstr) == 0 && se->opaque == opaque) {
1075
            QTAILQ_REMOVE(&savevm_handlers, se, entry);
1076
            qemu_free(se);
1077
        }
1078
    }
1079
}
1080

    
1081
int vmstate_register_with_alias_id(int instance_id,
1082
                                   const VMStateDescription *vmsd,
1083
                                   void *opaque, int alias_id,
1084
                                   int required_for_version)
1085
{
1086
    SaveStateEntry *se;
1087

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

    
1091
    se = qemu_mallocz(sizeof(SaveStateEntry));
1092
    pstrcpy(se->idstr, sizeof(se->idstr), vmsd->name);
1093
    se->version_id = vmsd->version_id;
1094
    se->section_id = global_section_id++;
1095
    se->save_live_state = NULL;
1096
    se->save_state = NULL;
1097
    se->load_state = NULL;
1098
    se->opaque = opaque;
1099
    se->vmsd = vmsd;
1100
    se->alias_id = alias_id;
1101

    
1102
    if (instance_id == -1) {
1103
        se->instance_id = calculate_new_instance_id(vmsd->name);
1104
    } else {
1105
        se->instance_id = instance_id;
1106
    }
1107
    /* add at the end of list */
1108
    QTAILQ_INSERT_TAIL(&savevm_handlers, se, entry);
1109
    return 0;
1110
}
1111

    
1112
int vmstate_register(int instance_id, const VMStateDescription *vmsd,
1113
                     void *opaque)
1114
{
1115
    return vmstate_register_with_alias_id(instance_id, vmsd, opaque, -1, 0);
1116
}
1117

    
1118
void vmstate_unregister(const VMStateDescription *vmsd, void *opaque)
1119
{
1120
    SaveStateEntry *se, *new_se;
1121

    
1122
    QTAILQ_FOREACH_SAFE(se, &savevm_handlers, entry, new_se) {
1123
        if (se->vmsd == vmsd && se->opaque == opaque) {
1124
            QTAILQ_REMOVE(&savevm_handlers, se, entry);
1125
            qemu_free(se);
1126
        }
1127
    }
1128
}
1129

    
1130
int vmstate_load_state(QEMUFile *f, const VMStateDescription *vmsd,
1131
                       void *opaque, int version_id)
1132
{
1133
    VMStateField *field = vmsd->fields;
1134

    
1135
    if (version_id > vmsd->version_id) {
1136
        return -EINVAL;
1137
    }
1138
    if (version_id < vmsd->minimum_version_id_old) {
1139
        return -EINVAL;
1140
    }
1141
    if  (version_id < vmsd->minimum_version_id) {
1142
        return vmsd->load_state_old(f, opaque, version_id);
1143
    }
1144
    if (vmsd->pre_load) {
1145
        int ret = vmsd->pre_load(opaque);
1146
        if (ret)
1147
            return ret;
1148
    }
1149
    while(field->name) {
1150
        if ((field->field_exists &&
1151
             field->field_exists(opaque, version_id)) ||
1152
            (!field->field_exists &&
1153
             field->version_id <= version_id)) {
1154
            void *base_addr = opaque + field->offset;
1155
            int ret, i, n_elems = 1;
1156
            int size = field->size;
1157

    
1158
            if (field->flags & VMS_VBUFFER) {
1159
                size = *(int32_t *)(opaque+field->size_offset);
1160
                if (field->flags & VMS_MULTIPLY) {
1161
                    size *= field->size;
1162
                }
1163
            }
1164
            if (field->flags & VMS_ARRAY) {
1165
                n_elems = field->num;
1166
            } else if (field->flags & VMS_VARRAY_INT32) {
1167
                n_elems = *(int32_t *)(opaque+field->num_offset);
1168
            } else if (field->flags & VMS_VARRAY_UINT16) {
1169
                n_elems = *(uint16_t *)(opaque+field->num_offset);
1170
            }
1171
            if (field->flags & VMS_POINTER) {
1172
                base_addr = *(void **)base_addr + field->start;
1173
            }
1174
            for (i = 0; i < n_elems; i++) {
1175
                void *addr = base_addr + size * i;
1176

    
1177
                if (field->flags & VMS_ARRAY_OF_POINTER) {
1178
                    addr = *(void **)addr;
1179
                }
1180
                if (field->flags & VMS_STRUCT) {
1181
                    ret = vmstate_load_state(f, field->vmsd, addr, field->vmsd->version_id);
1182
                } else {
1183
                    ret = field->info->get(f, addr, size);
1184

    
1185
                }
1186
                if (ret < 0) {
1187
                    return ret;
1188
                }
1189
            }
1190
        }
1191
        field++;
1192
    }
1193
    if (vmsd->post_load) {
1194
        return vmsd->post_load(opaque, version_id);
1195
    }
1196
    return 0;
1197
}
1198

    
1199
void vmstate_save_state(QEMUFile *f, const VMStateDescription *vmsd,
1200
                        void *opaque)
1201
{
1202
    VMStateField *field = vmsd->fields;
1203

    
1204
    if (vmsd->pre_save) {
1205
        vmsd->pre_save(opaque);
1206
    }
1207
    while(field->name) {
1208
        if (!field->field_exists ||
1209
            field->field_exists(opaque, vmsd->version_id)) {
1210
            void *base_addr = opaque + field->offset;
1211
            int i, n_elems = 1;
1212
            int size = field->size;
1213

    
1214
            if (field->flags & VMS_VBUFFER) {
1215
                size = *(int32_t *)(opaque+field->size_offset);
1216
                if (field->flags & VMS_MULTIPLY) {
1217
                    size *= field->size;
1218
                }
1219
            }
1220
            if (field->flags & VMS_ARRAY) {
1221
                n_elems = field->num;
1222
            } else if (field->flags & VMS_VARRAY_INT32) {
1223
                n_elems = *(int32_t *)(opaque+field->num_offset);
1224
            } else if (field->flags & VMS_VARRAY_UINT16) {
1225
                n_elems = *(uint16_t *)(opaque+field->num_offset);
1226
            }
1227
            if (field->flags & VMS_POINTER) {
1228
                base_addr = *(void **)base_addr + field->start;
1229
            }
1230
            for (i = 0; i < n_elems; i++) {
1231
                void *addr = base_addr + size * i;
1232

    
1233
                if (field->flags & VMS_ARRAY_OF_POINTER) {
1234
                    addr = *(void **)addr;
1235
                }
1236
                if (field->flags & VMS_STRUCT) {
1237
                    vmstate_save_state(f, field->vmsd, addr);
1238
                } else {
1239
                    field->info->put(f, addr, size);
1240
                }
1241
            }
1242
        }
1243
        field++;
1244
    }
1245
}
1246

    
1247
static int vmstate_load(QEMUFile *f, SaveStateEntry *se, int version_id)
1248
{
1249
    if (!se->vmsd) {         /* Old style */
1250
        return se->load_state(f, se->opaque, version_id);
1251
    }
1252
    return vmstate_load_state(f, se->vmsd, se->opaque, version_id);
1253
}
1254

    
1255
static void vmstate_save(QEMUFile *f, SaveStateEntry *se)
1256
{
1257
    if (!se->vmsd) {         /* Old style */
1258
        se->save_state(f, se->opaque);
1259
        return;
1260
    }
1261
    vmstate_save_state(f,se->vmsd, se->opaque);
1262
}
1263

    
1264
#define QEMU_VM_FILE_MAGIC           0x5145564d
1265
#define QEMU_VM_FILE_VERSION_COMPAT  0x00000002
1266
#define QEMU_VM_FILE_VERSION         0x00000003
1267

    
1268
#define QEMU_VM_EOF                  0x00
1269
#define QEMU_VM_SECTION_START        0x01
1270
#define QEMU_VM_SECTION_PART         0x02
1271
#define QEMU_VM_SECTION_END          0x03
1272
#define QEMU_VM_SECTION_FULL         0x04
1273

    
1274
int qemu_savevm_state_begin(Monitor *mon, QEMUFile *f, int blk_enable,
1275
                            int shared)
1276
{
1277
    SaveStateEntry *se;
1278

    
1279
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1280
        if(se->set_params == NULL) {
1281
            continue;
1282
        }
1283
        se->set_params(blk_enable, shared, se->opaque);
1284
    }
1285
    
1286
    qemu_put_be32(f, QEMU_VM_FILE_MAGIC);
1287
    qemu_put_be32(f, QEMU_VM_FILE_VERSION);
1288

    
1289
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1290
        int len;
1291

    
1292
        if (se->save_live_state == NULL)
1293
            continue;
1294

    
1295
        /* Section type */
1296
        qemu_put_byte(f, QEMU_VM_SECTION_START);
1297
        qemu_put_be32(f, se->section_id);
1298

    
1299
        /* ID string */
1300
        len = strlen(se->idstr);
1301
        qemu_put_byte(f, len);
1302
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1303

    
1304
        qemu_put_be32(f, se->instance_id);
1305
        qemu_put_be32(f, se->version_id);
1306

    
1307
        se->save_live_state(mon, f, QEMU_VM_SECTION_START, se->opaque);
1308
    }
1309

    
1310
    if (qemu_file_has_error(f)) {
1311
        qemu_savevm_state_cancel(mon, f);
1312
        return -EIO;
1313
    }
1314

    
1315
    return 0;
1316
}
1317

    
1318
int qemu_savevm_state_iterate(Monitor *mon, QEMUFile *f)
1319
{
1320
    SaveStateEntry *se;
1321
    int ret = 1;
1322

    
1323
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1324
        if (se->save_live_state == NULL)
1325
            continue;
1326

    
1327
        /* Section type */
1328
        qemu_put_byte(f, QEMU_VM_SECTION_PART);
1329
        qemu_put_be32(f, se->section_id);
1330

    
1331
        ret = se->save_live_state(mon, f, QEMU_VM_SECTION_PART, se->opaque);
1332
        if (!ret) {
1333
            /* Do not proceed to the next vmstate before this one reported
1334
               completion of the current stage. This serializes the migration
1335
               and reduces the probability that a faster changing state is
1336
               synchronized over and over again. */
1337
            break;
1338
        }
1339
    }
1340

    
1341
    if (ret)
1342
        return 1;
1343

    
1344
    if (qemu_file_has_error(f)) {
1345
        qemu_savevm_state_cancel(mon, f);
1346
        return -EIO;
1347
    }
1348

    
1349
    return 0;
1350
}
1351

    
1352
int qemu_savevm_state_complete(Monitor *mon, QEMUFile *f)
1353
{
1354
    SaveStateEntry *se;
1355

    
1356
    cpu_synchronize_all_states();
1357

    
1358
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1359
        if (se->save_live_state == NULL)
1360
            continue;
1361

    
1362
        /* Section type */
1363
        qemu_put_byte(f, QEMU_VM_SECTION_END);
1364
        qemu_put_be32(f, se->section_id);
1365

    
1366
        se->save_live_state(mon, f, QEMU_VM_SECTION_END, se->opaque);
1367
    }
1368

    
1369
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1370
        int len;
1371

    
1372
        if (se->save_state == NULL && se->vmsd == NULL)
1373
            continue;
1374

    
1375
        /* Section type */
1376
        qemu_put_byte(f, QEMU_VM_SECTION_FULL);
1377
        qemu_put_be32(f, se->section_id);
1378

    
1379
        /* ID string */
1380
        len = strlen(se->idstr);
1381
        qemu_put_byte(f, len);
1382
        qemu_put_buffer(f, (uint8_t *)se->idstr, len);
1383

    
1384
        qemu_put_be32(f, se->instance_id);
1385
        qemu_put_be32(f, se->version_id);
1386

    
1387
        vmstate_save(f, se);
1388
    }
1389

    
1390
    qemu_put_byte(f, QEMU_VM_EOF);
1391

    
1392
    if (qemu_file_has_error(f))
1393
        return -EIO;
1394

    
1395
    return 0;
1396
}
1397

    
1398
void qemu_savevm_state_cancel(Monitor *mon, QEMUFile *f)
1399
{
1400
    SaveStateEntry *se;
1401

    
1402
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1403
        if (se->save_live_state) {
1404
            se->save_live_state(mon, f, -1, se->opaque);
1405
        }
1406
    }
1407
}
1408

    
1409
static int qemu_savevm_state(Monitor *mon, QEMUFile *f)
1410
{
1411
    int saved_vm_running;
1412
    int ret;
1413

    
1414
    saved_vm_running = vm_running;
1415
    vm_stop(0);
1416

    
1417
    bdrv_flush_all();
1418

    
1419
    ret = qemu_savevm_state_begin(mon, f, 0, 0);
1420
    if (ret < 0)
1421
        goto out;
1422

    
1423
    do {
1424
        ret = qemu_savevm_state_iterate(mon, f);
1425
        if (ret < 0)
1426
            goto out;
1427
    } while (ret == 0);
1428

    
1429
    ret = qemu_savevm_state_complete(mon, f);
1430

    
1431
out:
1432
    if (qemu_file_has_error(f))
1433
        ret = -EIO;
1434

    
1435
    if (!ret && saved_vm_running)
1436
        vm_start();
1437

    
1438
    return ret;
1439
}
1440

    
1441
static SaveStateEntry *find_se(const char *idstr, int instance_id)
1442
{
1443
    SaveStateEntry *se;
1444

    
1445
    QTAILQ_FOREACH(se, &savevm_handlers, entry) {
1446
        if (!strcmp(se->idstr, idstr) &&
1447
            (instance_id == se->instance_id ||
1448
             instance_id == se->alias_id))
1449
            return se;
1450
    }
1451
    return NULL;
1452
}
1453

    
1454
typedef struct LoadStateEntry {
1455
    QLIST_ENTRY(LoadStateEntry) entry;
1456
    SaveStateEntry *se;
1457
    int section_id;
1458
    int version_id;
1459
} LoadStateEntry;
1460

    
1461
int qemu_loadvm_state(QEMUFile *f)
1462
{
1463
    QLIST_HEAD(, LoadStateEntry) loadvm_handlers =
1464
        QLIST_HEAD_INITIALIZER(loadvm_handlers);
1465
    LoadStateEntry *le, *new_le;
1466
    uint8_t section_type;
1467
    unsigned int v;
1468
    int ret;
1469

    
1470
    v = qemu_get_be32(f);
1471
    if (v != QEMU_VM_FILE_MAGIC)
1472
        return -EINVAL;
1473

    
1474
    v = qemu_get_be32(f);
1475
    if (v == QEMU_VM_FILE_VERSION_COMPAT) {
1476
        fprintf(stderr, "SaveVM v2 format is obsolete and don't work anymore\n");
1477
        return -ENOTSUP;
1478
    }
1479
    if (v != QEMU_VM_FILE_VERSION)
1480
        return -ENOTSUP;
1481

    
1482
    while ((section_type = qemu_get_byte(f)) != QEMU_VM_EOF) {
1483
        uint32_t instance_id, version_id, section_id;
1484
        SaveStateEntry *se;
1485
        char idstr[257];
1486
        int len;
1487

    
1488
        switch (section_type) {
1489
        case QEMU_VM_SECTION_START:
1490
        case QEMU_VM_SECTION_FULL:
1491
            /* Read section start */
1492
            section_id = qemu_get_be32(f);
1493
            len = qemu_get_byte(f);
1494
            qemu_get_buffer(f, (uint8_t *)idstr, len);
1495
            idstr[len] = 0;
1496
            instance_id = qemu_get_be32(f);
1497
            version_id = qemu_get_be32(f);
1498

    
1499
            /* Find savevm section */
1500
            se = find_se(idstr, instance_id);
1501
            if (se == NULL) {
1502
                fprintf(stderr, "Unknown savevm section or instance '%s' %d\n", idstr, instance_id);
1503
                ret = -EINVAL;
1504
                goto out;
1505
            }
1506

    
1507
            /* Validate version */
1508
            if (version_id > se->version_id) {
1509
                fprintf(stderr, "savevm: unsupported version %d for '%s' v%d\n",
1510
                        version_id, idstr, se->version_id);
1511
                ret = -EINVAL;
1512
                goto out;
1513
            }
1514

    
1515
            /* Add entry */
1516
            le = qemu_mallocz(sizeof(*le));
1517

    
1518
            le->se = se;
1519
            le->section_id = section_id;
1520
            le->version_id = version_id;
1521
            QLIST_INSERT_HEAD(&loadvm_handlers, le, entry);
1522

    
1523
            ret = vmstate_load(f, le->se, le->version_id);
1524
            if (ret < 0) {
1525
                fprintf(stderr, "qemu: warning: error while loading state for instance 0x%x of device '%s'\n",
1526
                        instance_id, idstr);
1527
                goto out;
1528
            }
1529
            break;
1530
        case QEMU_VM_SECTION_PART:
1531
        case QEMU_VM_SECTION_END:
1532
            section_id = qemu_get_be32(f);
1533

    
1534
            QLIST_FOREACH(le, &loadvm_handlers, entry) {
1535
                if (le->section_id == section_id) {
1536
                    break;
1537
                }
1538
            }
1539
            if (le == NULL) {
1540
                fprintf(stderr, "Unknown savevm section %d\n", section_id);
1541
                ret = -EINVAL;
1542
                goto out;
1543
            }
1544

    
1545
            ret = vmstate_load(f, le->se, le->version_id);
1546
            if (ret < 0) {
1547
                fprintf(stderr, "qemu: warning: error while loading state section id %d\n",
1548
                        section_id);
1549
                goto out;
1550
            }
1551
            break;
1552
        default:
1553
            fprintf(stderr, "Unknown savevm section type %d\n", section_type);
1554
            ret = -EINVAL;
1555
            goto out;
1556
        }
1557
    }
1558

    
1559
    cpu_synchronize_all_post_init();
1560

    
1561
    ret = 0;
1562

    
1563
out:
1564
    QLIST_FOREACH_SAFE(le, &loadvm_handlers, entry, new_le) {
1565
        QLIST_REMOVE(le, entry);
1566
        qemu_free(le);
1567
    }
1568

    
1569
    if (qemu_file_has_error(f))
1570
        ret = -EIO;
1571

    
1572
    return ret;
1573
}
1574

    
1575
static int bdrv_snapshot_find(BlockDriverState *bs, QEMUSnapshotInfo *sn_info,
1576
                              const char *name)
1577
{
1578
    QEMUSnapshotInfo *sn_tab, *sn;
1579
    int nb_sns, i, ret;
1580

    
1581
    ret = -ENOENT;
1582
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1583
    if (nb_sns < 0)
1584
        return ret;
1585
    for(i = 0; i < nb_sns; i++) {
1586
        sn = &sn_tab[i];
1587
        if (!strcmp(sn->id_str, name) || !strcmp(sn->name, name)) {
1588
            *sn_info = *sn;
1589
            ret = 0;
1590
            break;
1591
        }
1592
    }
1593
    qemu_free(sn_tab);
1594
    return ret;
1595
}
1596

    
1597
/*
1598
 * Deletes snapshots of a given name in all opened images.
1599
 */
1600
static int del_existing_snapshots(Monitor *mon, const char *name)
1601
{
1602
    BlockDriverState *bs;
1603
    QEMUSnapshotInfo sn1, *snapshot = &sn1;
1604
    int ret;
1605

    
1606
    bs = NULL;
1607
    while ((bs = bdrv_next(bs))) {
1608
        if (bdrv_can_snapshot(bs) &&
1609
            bdrv_snapshot_find(bs, snapshot, name) >= 0)
1610
        {
1611
            ret = bdrv_snapshot_delete(bs, name);
1612
            if (ret < 0) {
1613
                monitor_printf(mon,
1614
                               "Error while deleting snapshot on '%s'\n",
1615
                               bdrv_get_device_name(bs));
1616
                return -1;
1617
            }
1618
        }
1619
    }
1620

    
1621
    return 0;
1622
}
1623

    
1624
void do_savevm(Monitor *mon, const QDict *qdict)
1625
{
1626
    BlockDriverState *bs, *bs1;
1627
    QEMUSnapshotInfo sn1, *sn = &sn1, old_sn1, *old_sn = &old_sn1;
1628
    int ret;
1629
    QEMUFile *f;
1630
    int saved_vm_running;
1631
    uint32_t vm_state_size;
1632
#ifdef _WIN32
1633
    struct _timeb tb;
1634
#else
1635
    struct timeval tv;
1636
#endif
1637
    const char *name = qdict_get_try_str(qdict, "name");
1638

    
1639
    /* Verify if there is a device that doesn't support snapshots and is writable */
1640
    bs = NULL;
1641
    while ((bs = bdrv_next(bs))) {
1642

    
1643
        if (bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
1644
            continue;
1645
        }
1646

    
1647
        if (!bdrv_can_snapshot(bs)) {
1648
            monitor_printf(mon, "Device '%s' is writable but does not support snapshots.\n",
1649
                               bdrv_get_device_name(bs));
1650
            return;
1651
        }
1652
    }
1653

    
1654
    bs = bdrv_snapshots();
1655
    if (!bs) {
1656
        monitor_printf(mon, "No block device can accept snapshots\n");
1657
        return;
1658
    }
1659
    /* ??? Should this occur after vm_stop?  */
1660
    qemu_aio_flush();
1661

    
1662
    saved_vm_running = vm_running;
1663
    vm_stop(0);
1664

    
1665
    memset(sn, 0, sizeof(*sn));
1666
    if (name) {
1667
        ret = bdrv_snapshot_find(bs, old_sn, name);
1668
        if (ret >= 0) {
1669
            pstrcpy(sn->name, sizeof(sn->name), old_sn->name);
1670
            pstrcpy(sn->id_str, sizeof(sn->id_str), old_sn->id_str);
1671
        } else {
1672
            pstrcpy(sn->name, sizeof(sn->name), name);
1673
        }
1674
    }
1675

    
1676
    /* fill auxiliary fields */
1677
#ifdef _WIN32
1678
    _ftime(&tb);
1679
    sn->date_sec = tb.time;
1680
    sn->date_nsec = tb.millitm * 1000000;
1681
#else
1682
    gettimeofday(&tv, NULL);
1683
    sn->date_sec = tv.tv_sec;
1684
    sn->date_nsec = tv.tv_usec * 1000;
1685
#endif
1686
    sn->vm_clock_nsec = qemu_get_clock(vm_clock);
1687

    
1688
    /* Delete old snapshots of the same name */
1689
    if (name && del_existing_snapshots(mon, name) < 0) {
1690
        goto the_end;
1691
    }
1692

    
1693
    /* save the VM state */
1694
    f = qemu_fopen_bdrv(bs, 1);
1695
    if (!f) {
1696
        monitor_printf(mon, "Could not open VM state file\n");
1697
        goto the_end;
1698
    }
1699
    ret = qemu_savevm_state(mon, f);
1700
    vm_state_size = qemu_ftell(f);
1701
    qemu_fclose(f);
1702
    if (ret < 0) {
1703
        monitor_printf(mon, "Error %d while writing VM\n", ret);
1704
        goto the_end;
1705
    }
1706

    
1707
    /* create the snapshots */
1708

    
1709
    bs1 = NULL;
1710
    while ((bs1 = bdrv_next(bs1))) {
1711
        if (bdrv_can_snapshot(bs1)) {
1712
            /* Write VM state size only to the image that contains the state */
1713
            sn->vm_state_size = (bs == bs1 ? vm_state_size : 0);
1714
            ret = bdrv_snapshot_create(bs1, sn);
1715
            if (ret < 0) {
1716
                monitor_printf(mon, "Error while creating snapshot on '%s'\n",
1717
                               bdrv_get_device_name(bs1));
1718
            }
1719
        }
1720
    }
1721

    
1722
 the_end:
1723
    if (saved_vm_running)
1724
        vm_start();
1725
}
1726

    
1727
int load_vmstate(const char *name)
1728
{
1729
    BlockDriverState *bs, *bs1;
1730
    QEMUSnapshotInfo sn;
1731
    QEMUFile *f;
1732
    int ret;
1733

    
1734
    /* Verify if there is a device that doesn't support snapshots and is writable */
1735
    bs = NULL;
1736
    while ((bs = bdrv_next(bs))) {
1737

    
1738
        if (bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
1739
            continue;
1740
        }
1741

    
1742
        if (!bdrv_can_snapshot(bs)) {
1743
            error_report("Device '%s' is writable but does not support snapshots.",
1744
                               bdrv_get_device_name(bs));
1745
            return -ENOTSUP;
1746
        }
1747
    }
1748

    
1749
    bs = bdrv_snapshots();
1750
    if (!bs) {
1751
        error_report("No block device supports snapshots");
1752
        return -EINVAL;
1753
    }
1754

    
1755
    /* Flush all IO requests so they don't interfere with the new state.  */
1756
    qemu_aio_flush();
1757

    
1758
    bs1 = NULL;
1759
    while ((bs1 = bdrv_next(bs1))) {
1760
        if (bdrv_can_snapshot(bs1)) {
1761
            ret = bdrv_snapshot_goto(bs1, name);
1762
            if (ret < 0) {
1763
                switch(ret) {
1764
                case -ENOTSUP:
1765
                    error_report("%sSnapshots not supported on device '%s'",
1766
                                 bs != bs1 ? "Warning: " : "",
1767
                                 bdrv_get_device_name(bs1));
1768
                    break;
1769
                case -ENOENT:
1770
                    error_report("%sCould not find snapshot '%s' on device '%s'",
1771
                                 bs != bs1 ? "Warning: " : "",
1772
                                 name, bdrv_get_device_name(bs1));
1773
                    break;
1774
                default:
1775
                    error_report("%sError %d while activating snapshot on '%s'",
1776
                                 bs != bs1 ? "Warning: " : "",
1777
                                 ret, bdrv_get_device_name(bs1));
1778
                    break;
1779
                }
1780
                /* fatal on snapshot block device */
1781
                if (bs == bs1)
1782
                    return 0;
1783
            }
1784
        }
1785
    }
1786

    
1787
    /* Don't even try to load empty VM states */
1788
    ret = bdrv_snapshot_find(bs, &sn, name);
1789
    if ((ret >= 0) && (sn.vm_state_size == 0))
1790
        return -EINVAL;
1791

    
1792
    /* restore the VM state */
1793
    f = qemu_fopen_bdrv(bs, 0);
1794
    if (!f) {
1795
        error_report("Could not open VM state file");
1796
        return -EINVAL;
1797
    }
1798
    ret = qemu_loadvm_state(f);
1799
    qemu_fclose(f);
1800
    if (ret < 0) {
1801
        error_report("Error %d while loading VM state", ret);
1802
        return ret;
1803
    }
1804
    return 0;
1805
}
1806

    
1807
void do_delvm(Monitor *mon, const QDict *qdict)
1808
{
1809
    BlockDriverState *bs, *bs1;
1810
    int ret;
1811
    const char *name = qdict_get_str(qdict, "name");
1812

    
1813
    bs = bdrv_snapshots();
1814
    if (!bs) {
1815
        monitor_printf(mon, "No block device supports snapshots\n");
1816
        return;
1817
    }
1818

    
1819
    bs1 = NULL;
1820
    while ((bs1 = bdrv_next(bs1))) {
1821
        if (bdrv_can_snapshot(bs1)) {
1822
            ret = bdrv_snapshot_delete(bs1, name);
1823
            if (ret < 0) {
1824
                if (ret == -ENOTSUP)
1825
                    monitor_printf(mon,
1826
                                   "Snapshots not supported on device '%s'\n",
1827
                                   bdrv_get_device_name(bs1));
1828
                else
1829
                    monitor_printf(mon, "Error %d while deleting snapshot on "
1830
                                   "'%s'\n", ret, bdrv_get_device_name(bs1));
1831
            }
1832
        }
1833
    }
1834
}
1835

    
1836
void do_info_snapshots(Monitor *mon)
1837
{
1838
    BlockDriverState *bs, *bs1;
1839
    QEMUSnapshotInfo *sn_tab, *sn;
1840
    int nb_sns, i;
1841
    char buf[256];
1842

    
1843
    bs = bdrv_snapshots();
1844
    if (!bs) {
1845
        monitor_printf(mon, "No available block device supports snapshots\n");
1846
        return;
1847
    }
1848
    monitor_printf(mon, "Snapshot devices:");
1849
    bs1 = NULL;
1850
    while ((bs1 = bdrv_next(bs1))) {
1851
        if (bdrv_can_snapshot(bs1)) {
1852
            if (bs == bs1)
1853
                monitor_printf(mon, " %s", bdrv_get_device_name(bs1));
1854
        }
1855
    }
1856
    monitor_printf(mon, "\n");
1857

    
1858
    nb_sns = bdrv_snapshot_list(bs, &sn_tab);
1859
    if (nb_sns < 0) {
1860
        monitor_printf(mon, "bdrv_snapshot_list: error %d\n", nb_sns);
1861
        return;
1862
    }
1863
    monitor_printf(mon, "Snapshot list (from %s):\n",
1864
                   bdrv_get_device_name(bs));
1865
    monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), NULL));
1866
    for(i = 0; i < nb_sns; i++) {
1867
        sn = &sn_tab[i];
1868
        monitor_printf(mon, "%s\n", bdrv_snapshot_dump(buf, sizeof(buf), sn));
1869
    }
1870
    qemu_free(sn_tab);
1871
}