savevm.c 60.2 KB
Newer Older
aliguori's avatar
aliguori committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
/*
 * QEMU System Emulator
 *
 * Copyright (c) 2003-2008 Fabrice Bellard
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

blueswir1's avatar
blueswir1 committed
25
#include "config-host.h"
26 27
#include "qemu-common.h"
#include "hw/hw.h"
28
#include "hw/qdev.h"
Paolo Bonzini's avatar
Paolo Bonzini committed
29
#include "net/net.h"
30
#include "monitor/monitor.h"
31
#include "sysemu/sysemu.h"
32
#include "qemu/timer.h"
33
#include "audio/audio.h"
34
#include "migration/migration.h"
35 36
#include "qemu/sockets.h"
#include "qemu/queue.h"
37
#include "sysemu/cpus.h"
38
#include "exec/memory.h"
39
#include "qmp-commands.h"
40
#include "trace.h"
41
#include "qemu/bitops.h"
42

aliguori's avatar
aliguori committed
43 44
#define SELF_ANNOUNCE_ROUNDS 5

Nolan's avatar
Nolan committed
45
#ifndef ETH_P_RARP
Stefan Berger's avatar
Stefan Berger committed
46
#define ETH_P_RARP 0x8035
Nolan's avatar
Nolan committed
47 48 49 50 51 52
#endif
#define ARP_HTYPE_ETH 0x0001
#define ARP_PTYPE_IP 0x0800
#define ARP_OP_REQUEST_REV 0x3

static int announce_self_create(uint8_t *buf,
aliguori's avatar
aliguori committed
53 54
				uint8_t *mac_addr)
{
Nolan's avatar
Nolan committed
55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
    /* Ethernet header. */
    memset(buf, 0xff, 6);         /* destination MAC addr */
    memcpy(buf + 6, mac_addr, 6); /* source MAC addr */
    *(uint16_t *)(buf + 12) = htons(ETH_P_RARP); /* ethertype */

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

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

    return 60; /* len (FCS will be added by hardware) */
aliguori's avatar
aliguori committed
75 76
}

77
static void qemu_announce_self_iter(NICState *nic, void *opaque)
aliguori's avatar
aliguori committed
78
{
Nolan's avatar
Nolan committed
79
    uint8_t buf[60];
80 81 82 83
    int len;

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

Jason Wang's avatar
Jason Wang committed
84
    qemu_send_packet_raw(qemu_get_queue(nic), buf, len);
85 86 87 88 89
}


static void qemu_announce_self_once(void *opaque)
{
90 91
    static int count = SELF_ANNOUNCE_ROUNDS;
    QEMUTimer *timer = *(QEMUTimer **)opaque;
aliguori's avatar
aliguori committed
92

93 94
    qemu_foreach_nic(qemu_announce_self_iter, NULL);

Nolan's avatar
Nolan committed
95 96
    if (--count) {
        /* delay 50ms, 150ms, 250ms, ... */
97
        qemu_mod_timer(timer, qemu_get_clock_ms(rt_clock) +
Nolan's avatar
Nolan committed
98
                       50 + (SELF_ANNOUNCE_ROUNDS - count - 1) * 100);
99 100 101 102 103 104 105 106 107
    } else {
	    qemu_del_timer(timer);
	    qemu_free_timer(timer);
    }
}

void qemu_announce_self(void)
{
	static QEMUTimer *timer;
108
	timer = qemu_new_timer_ms(rt_clock, qemu_announce_self_once, &timer);
109
	qemu_announce_self_once(&timer);
aliguori's avatar
aliguori committed
110 111 112 113 114 115 116 117
}

/***********************************************************/
/* savevm/loadvm support */

#define IO_BUF_SIZE 32768

struct QEMUFile {
118
    const QEMUFileOps *ops;
aliguori's avatar
aliguori committed
119 120 121
    void *opaque;
    int is_write;

122 123 124
    int64_t bytes_xfer;
    int64_t xfer_limit;

125 126
    int64_t pos; /* start of buffer when writing, end of buffer
                    when reading */
aliguori's avatar
aliguori committed
127 128 129 130
    int buf_index;
    int buf_size; /* 0 when writing */
    uint8_t buf[IO_BUF_SIZE];

131
    int last_error;
aliguori's avatar
aliguori committed
132 133
};

134
typedef struct QEMUFileStdio
aliguori's avatar
aliguori committed
135
{
136
    FILE *stdio_file;
aliguori's avatar
aliguori committed
137
    QEMUFile *file;
138
} QEMUFileStdio;
aliguori's avatar
aliguori committed
139 140 141 142 143 144 145

typedef struct QEMUFileSocket
{
    int fd;
    QEMUFile *file;
} QEMUFileSocket;

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173
typedef struct {
    Coroutine *co;
    int fd;
} FDYieldUntilData;

static void fd_coroutine_enter(void *opaque)
{
    FDYieldUntilData *data = opaque;
    qemu_set_fd_handler(data->fd, NULL, NULL, NULL);
    qemu_coroutine_enter(data->co, NULL);
}

/**
 * Yield until a file descriptor becomes readable
 *
 * Note that this function clobbers the handlers for the file descriptor.
 */
static void coroutine_fn yield_until_fd_readable(int fd)
{
    FDYieldUntilData data;

    assert(qemu_in_coroutine());
    data.co = qemu_coroutine_self();
    data.fd = fd;
    qemu_set_fd_handler(fd, fd_coroutine_enter, NULL, &data);
    qemu_coroutine_yield();
}

Paolo Bonzini's avatar
Paolo Bonzini committed
174 175 176 177 178 179 180
static int socket_get_fd(void *opaque)
{
    QEMUFileSocket *s = opaque;

    return s->fd;
}

aliguori's avatar
aliguori committed
181 182 183 184 185
static int socket_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
{
    QEMUFileSocket *s = opaque;
    ssize_t len;

186
    for (;;) {
Blue Swirl's avatar
Blue Swirl committed
187
        len = qemu_recv(s->fd, buf, size, 0);
188 189 190 191
        if (len != -1) {
            break;
        }
        if (socket_error() == EAGAIN) {
192
            yield_until_fd_readable(s->fd);
193 194 195 196
        } else if (socket_error() != EINTR) {
            break;
        }
    }
aliguori's avatar
aliguori committed
197

198
    if (len == -1) {
aliguori's avatar
aliguori committed
199
        len = -socket_error();
200
    }
aliguori's avatar
aliguori committed
201 202 203
    return len;
}

204 205 206 207 208 209 210 211 212 213 214 215
static int socket_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
{
    QEMUFileSocket *s = opaque;
    ssize_t len;

    len = qemu_send_full(s->fd, buf, size, 0);
    if (len < size) {
        len = -socket_error();
    }
    return len;
}

aliguori's avatar
aliguori committed
216 217 218
static int socket_close(void *opaque)
{
    QEMUFileSocket *s = opaque;
219
    closesocket(s->fd);
220
    g_free(s);
aliguori's avatar
aliguori committed
221 222 223
    return 0;
}

Paolo Bonzini's avatar
Paolo Bonzini committed
224 225 226 227 228 229 230
static int stdio_get_fd(void *opaque)
{
    QEMUFileStdio *s = opaque;

    return fileno(s->stdio_file);
}

231
static int stdio_put_buffer(void *opaque, const uint8_t *buf, int64_t pos, int size)
aliguori's avatar
aliguori committed
232
{
233 234
    QEMUFileStdio *s = opaque;
    return fwrite(buf, 1, size, s->stdio_file);
aliguori's avatar
aliguori committed
235 236
}

237
static int stdio_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
aliguori's avatar
aliguori committed
238
{
239 240
    QEMUFileStdio *s = opaque;
    FILE *fp = s->stdio_file;
241 242
    int bytes;

243
    for (;;) {
244 245
        clearerr(fp);
        bytes = fread(buf, 1, size, fp);
246 247 248 249
        if (bytes != 0 || !ferror(fp)) {
            break;
        }
        if (errno == EAGAIN) {
250
            yield_until_fd_readable(fileno(fp));
251 252 253 254
        } else if (errno != EINTR) {
            break;
        }
    }
255
    return bytes;
aliguori's avatar
aliguori committed
256 257
}

258 259 260
static int stdio_pclose(void *opaque)
{
    QEMUFileStdio *s = opaque;
261 262
    int ret;
    ret = pclose(s->stdio_file);
263 264
    if (ret == -1) {
        ret = -errno;
265 266 267
    } else if (!WIFEXITED(ret) || WEXITSTATUS(ret) != 0) {
        /* close succeeded, but non-zero exit code: */
        ret = -EIO; /* fake errno value */
268
    }
269
    g_free(s);
270
    return ret;
271 272 273
}

static int stdio_fclose(void *opaque)
aliguori's avatar
aliguori committed
274
{
275
    QEMUFileStdio *s = opaque;
276
    int ret = 0;
277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294

    if (s->file->ops->put_buffer) {
        int fd = fileno(s->stdio_file);
        struct stat st;

        ret = fstat(fd, &st);
        if (ret == 0 && S_ISREG(st.st_mode)) {
            /*
             * If the file handle is a regular file make sure the
             * data is flushed to disk before signaling success.
             */
            ret = fsync(fd);
            if (ret != 0) {
                ret = -errno;
                return ret;
            }
        }
    }
295 296 297
    if (fclose(s->stdio_file) == EOF) {
        ret = -errno;
    }
298
    g_free(s);
299
    return ret;
aliguori's avatar
aliguori committed
300 301
}

302
static const QEMUFileOps stdio_pipe_read_ops = {
Paolo Bonzini's avatar
Paolo Bonzini committed
303
    .get_fd =     stdio_get_fd,
304 305 306 307 308
    .get_buffer = stdio_get_buffer,
    .close =      stdio_pclose
};

static const QEMUFileOps stdio_pipe_write_ops = {
Paolo Bonzini's avatar
Paolo Bonzini committed
309
    .get_fd =     stdio_get_fd,
310 311 312 313
    .put_buffer = stdio_put_buffer,
    .close =      stdio_pclose
};

314
QEMUFile *qemu_popen_cmd(const char *command, const char *mode)
aliguori's avatar
aliguori committed
315
{
316
    FILE *stdio_file;
317
    QEMUFileStdio *s;
aliguori's avatar
aliguori committed
318

319 320 321 322 323 324
    stdio_file = popen(command, mode);
    if (stdio_file == NULL) {
        return NULL;
    }

    if (mode == NULL || (mode[0] != 'r' && mode[0] != 'w') || mode[1] != 0) {
aliguori's avatar
aliguori committed
325 326 327 328
        fprintf(stderr, "qemu_popen: Argument validity check failed\n");
        return NULL;
    }

329
    s = g_malloc0(sizeof(QEMUFileStdio));
aliguori's avatar
aliguori committed
330

331
    s->stdio_file = stdio_file;
aliguori's avatar
aliguori committed
332 333

    if(mode[0] == 'r') {
334
        s->file = qemu_fopen_ops(s, &stdio_pipe_read_ops);
aliguori's avatar
aliguori committed
335
    } else {
336
        s->file = qemu_fopen_ops(s, &stdio_pipe_write_ops);
aliguori's avatar
aliguori committed
337 338 339 340
    }
    return s->file;
}

341
static const QEMUFileOps stdio_file_read_ops = {
Paolo Bonzini's avatar
Paolo Bonzini committed
342
    .get_fd =     stdio_get_fd,
343 344 345 346 347
    .get_buffer = stdio_get_buffer,
    .close =      stdio_fclose
};

static const QEMUFileOps stdio_file_write_ops = {
Paolo Bonzini's avatar
Paolo Bonzini committed
348
    .get_fd =     stdio_get_fd,
349 350 351 352
    .put_buffer = stdio_put_buffer,
    .close =      stdio_fclose
};

Paolo Bonzini's avatar
Paolo Bonzini committed
353 354 355 356 357 358 359 360 361 362 363
QEMUFile *qemu_fdopen(int fd, const char *mode)
{
    QEMUFileStdio *s;

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

364
    s = g_malloc0(sizeof(QEMUFileStdio));
Paolo Bonzini's avatar
Paolo Bonzini committed
365 366 367 368 369
    s->stdio_file = fdopen(fd, mode);
    if (!s->stdio_file)
        goto fail;

    if(mode[0] == 'r') {
370
        s->file = qemu_fopen_ops(s, &stdio_file_read_ops);
Paolo Bonzini's avatar
Paolo Bonzini committed
371
    } else {
372
        s->file = qemu_fopen_ops(s, &stdio_file_write_ops);
Paolo Bonzini's avatar
Paolo Bonzini committed
373 374 375 376
    }
    return s->file;

fail:
377
    g_free(s);
Paolo Bonzini's avatar
Paolo Bonzini committed
378 379 380
    return NULL;
}

381
static const QEMUFileOps socket_read_ops = {
Paolo Bonzini's avatar
Paolo Bonzini committed
382
    .get_fd =     socket_get_fd,
383 384 385 386
    .get_buffer = socket_get_buffer,
    .close =      socket_close
};

387 388 389 390 391 392 393
static const QEMUFileOps socket_write_ops = {
    .get_fd =     socket_get_fd,
    .put_buffer = socket_put_buffer,
    .close =      socket_close
};

QEMUFile *qemu_fopen_socket(int fd, const char *mode)
aliguori's avatar
aliguori committed
394
{
395
    QEMUFileSocket *s = g_malloc0(sizeof(QEMUFileSocket));
aliguori's avatar
aliguori committed
396

397 398 399 400 401 402 403
    if (mode == NULL ||
        (mode[0] != 'r' && mode[0] != 'w') ||
        mode[1] != 'b' || mode[2] != 0) {
        fprintf(stderr, "qemu_fopen: Argument validity check failed\n");
        return NULL;
    }

aliguori's avatar
aliguori committed
404
    s->fd = fd;
405
    if (mode[0] == 'w') {
406
        socket_set_block(s->fd);
407 408 409 410
        s->file = qemu_fopen_ops(s, &socket_write_ops);
    } else {
        s->file = qemu_fopen_ops(s, &socket_read_ops);
    }
aliguori's avatar
aliguori committed
411 412 413 414 415 416 417
    return s->file;
}

QEMUFile *qemu_fopen(const char *filename, const char *mode)
{
    QEMUFileStdio *s;

418 419 420
    if (mode == NULL ||
	(mode[0] != 'r' && mode[0] != 'w') ||
	mode[1] != 'b' || mode[2] != 0) {
Blue Swirl's avatar
Blue Swirl committed
421
        fprintf(stderr, "qemu_fopen: Argument validity check failed\n");
422 423 424
        return NULL;
    }

425
    s = g_malloc0(sizeof(QEMUFileStdio));
aliguori's avatar
aliguori committed
426

427 428
    s->stdio_file = fopen(filename, mode);
    if (!s->stdio_file)
aliguori's avatar
aliguori committed
429
        goto fail;
lirans@il.ibm.com's avatar
lirans@il.ibm.com committed
430
    
431
    if(mode[0] == 'w') {
432
        s->file = qemu_fopen_ops(s, &stdio_file_write_ops);
433
    } else {
434
        s->file = qemu_fopen_ops(s, &stdio_file_read_ops);
435 436
    }
    return s->file;
aliguori's avatar
aliguori committed
437
fail:
438
    g_free(s);
aliguori's avatar
aliguori committed
439 440 441
    return NULL;
}

442
static int block_put_buffer(void *opaque, const uint8_t *buf,
aliguori's avatar
aliguori committed
443 444
                           int64_t pos, int size)
{
445
    bdrv_save_vmstate(opaque, buf, pos, size);
aliguori's avatar
aliguori committed
446 447 448
    return size;
}

449
static int block_get_buffer(void *opaque, uint8_t *buf, int64_t pos, int size)
aliguori's avatar
aliguori committed
450
{
451
    return bdrv_load_vmstate(opaque, buf, pos, size);
aliguori's avatar
aliguori committed
452 453 454 455
}

static int bdrv_fclose(void *opaque)
{
456
    return bdrv_flush(opaque);
aliguori's avatar
aliguori committed
457 458
}

459 460 461 462 463 464 465 466 467 468
static const QEMUFileOps bdrv_read_ops = {
    .get_buffer = block_get_buffer,
    .close =      bdrv_fclose
};

static const QEMUFileOps bdrv_write_ops = {
    .put_buffer = block_put_buffer,
    .close =      bdrv_fclose
};

469
static QEMUFile *qemu_fopen_bdrv(BlockDriverState *bs, int is_writable)
aliguori's avatar
aliguori committed
470 471
{
    if (is_writable)
472 473
        return qemu_fopen_ops(bs, &bdrv_write_ops);
    return qemu_fopen_ops(bs, &bdrv_read_ops);
aliguori's avatar
aliguori committed
474 475
}

476
QEMUFile *qemu_fopen_ops(void *opaque, const QEMUFileOps *ops)
aliguori's avatar
aliguori committed
477 478 479
{
    QEMUFile *f;

480
    f = g_malloc0(sizeof(QEMUFile));
aliguori's avatar
aliguori committed
481 482

    f->opaque = opaque;
483
    f->ops = ops;
aliguori's avatar
aliguori committed
484 485 486 487
    f->is_write = 0;
    return f;
}

488
int qemu_file_get_error(QEMUFile *f)
aliguori's avatar
aliguori committed
489
{
490
    return f->last_error;
aliguori's avatar
aliguori committed
491 492
}

493
static void qemu_file_set_error(QEMUFile *f, int ret)
494
{
495 496 497
    if (f->last_error == 0) {
        f->last_error = ret;
    }
498 499
}

500 501 502
/** Flushes QEMUFile buffer
 *
 */
503
static void qemu_fflush(QEMUFile *f)
aliguori's avatar
aliguori committed
504
{
505 506
    int ret = 0;

507 508 509
    if (!f->ops->put_buffer) {
        return;
    }
aliguori's avatar
aliguori committed
510
    if (f->is_write && f->buf_index > 0) {
511
        ret = f->ops->put_buffer(f->opaque, f->buf, f->pos, f->buf_index);
512
        if (ret >= 0) {
513
            f->pos += f->buf_index;
514
        }
aliguori's avatar
aliguori committed
515 516
        f->buf_index = 0;
    }
517 518 519
    if (ret < 0) {
        qemu_file_set_error(f, ret);
    }
aliguori's avatar
aliguori committed
520 521 522 523 524
}

static void qemu_fill_buffer(QEMUFile *f)
{
    int len;
525
    int pending;
aliguori's avatar
aliguori committed
526

527
    if (!f->ops->get_buffer)
aliguori's avatar
aliguori committed
528 529 530 531 532
        return;

    if (f->is_write)
        abort();

533 534 535 536 537 538 539
    pending = f->buf_size - f->buf_index;
    if (pending > 0) {
        memmove(f->buf, f->buf + f->buf_index, pending);
    }
    f->buf_index = 0;
    f->buf_size = pending;

540
    len = f->ops->get_buffer(f->opaque, f->buf + pending, f->pos,
541
                        IO_BUF_SIZE - pending);
aliguori's avatar
aliguori committed
542
    if (len > 0) {
543
        f->buf_size += len;
544
        f->pos += len;
545
    } else if (len == 0) {
546
        qemu_file_set_error(f, -EIO);
aliguori's avatar
aliguori committed
547
    } else if (len != -EAGAIN)
548
        qemu_file_set_error(f, len);
aliguori's avatar
aliguori committed
549 550
}

Paolo Bonzini's avatar
Paolo Bonzini committed
551 552 553 554 555 556 557 558
int qemu_get_fd(QEMUFile *f)
{
    if (f->ops->get_fd) {
        return f->ops->get_fd(f->opaque);
    }
    return -1;
}

559 560 561 562 563 564 565 566 567 568
/** Closes the file
 *
 * Returns negative error value if any error happened on previous operations or
 * while closing the file. Returns 0 or positive number on success.
 *
 * The meaning of return value on success depends on the specific backend
 * being used.
 */
int qemu_fclose(QEMUFile *f)
{
569
    int ret;
570 571
    qemu_fflush(f);
    ret = qemu_file_get_error(f);
572

573 574
    if (f->ops->close) {
        int ret2 = f->ops->close(f->opaque);
575 576 577
        if (ret >= 0) {
            ret = ret2;
        }
578
    }
579 580 581 582 583 584
    /* If any error was spotted before closing, we should report it
     * instead of the close() return value.
     */
    if (f->last_error) {
        ret = f->last_error;
    }
585
    g_free(f);
aliguori's avatar
aliguori committed
586 587 588 589 590 591 592
    return ret;
}

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

593 594 595 596 597
    if (f->last_error) {
        return;
    }

    if (f->is_write == 0 && f->buf_index > 0) {
aliguori's avatar
aliguori committed
598 599 600 601 602
        fprintf(stderr,
                "Attempted to write to buffer while read buffer is not empty\n");
        abort();
    }

603
    while (size > 0) {
aliguori's avatar
aliguori committed
604 605 606 607 608 609
        l = IO_BUF_SIZE - f->buf_index;
        if (l > size)
            l = size;
        memcpy(f->buf + f->buf_index, buf, l);
        f->is_write = 1;
        f->buf_index += l;
610
        f->bytes_xfer += l;
aliguori's avatar
aliguori committed
611 612
        buf += l;
        size -= l;
613
        if (f->buf_index >= IO_BUF_SIZE) {
614 615
            qemu_fflush(f);
            if (qemu_file_get_error(f)) {
616 617
                break;
            }
618
        }
aliguori's avatar
aliguori committed
619 620 621 622 623
    }
}

void qemu_put_byte(QEMUFile *f, int v)
{
624 625 626 627 628
    if (f->last_error) {
        return;
    }

    if (f->is_write == 0 && f->buf_index > 0) {
aliguori's avatar
aliguori committed
629 630 631 632 633 634 635
        fprintf(stderr,
                "Attempted to write to buffer while read buffer is not empty\n");
        abort();
    }

    f->buf[f->buf_index++] = v;
    f->is_write = 1;
636
    if (f->buf_index >= IO_BUF_SIZE) {
637
        qemu_fflush(f);
638
    }
aliguori's avatar
aliguori committed
639 640
}

641
static void qemu_file_skip(QEMUFile *f, int size)
aliguori's avatar
aliguori committed
642
{
643 644 645 646 647 648
    if (f->buf_index + size <= f->buf_size) {
        f->buf_index += size;
    }
}

static int qemu_peek_buffer(QEMUFile *f, uint8_t *buf, int size, size_t offset)
aliguori's avatar
aliguori committed
649
{
650 651
    int pending;
    int index;
aliguori's avatar
aliguori committed
652

653
    if (f->is_write) {
aliguori's avatar
aliguori committed
654
        abort();
655
    }
aliguori's avatar
aliguori committed
656

657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686
    index = f->buf_index + offset;
    pending = f->buf_size - index;
    if (pending < size) {
        qemu_fill_buffer(f);
        index = f->buf_index + offset;
        pending = f->buf_size - index;
    }

    if (pending <= 0) {
        return 0;
    }
    if (size > pending) {
        size = pending;
    }

    memcpy(buf, f->buf + index, size);
    return size;
}

int qemu_get_buffer(QEMUFile *f, uint8_t *buf, int size)
{
    int pending = size;
    int done = 0;

    while (pending > 0) {
        int res;

        res = qemu_peek_buffer(f, buf, pending, 0);
        if (res == 0) {
            return done;
aliguori's avatar
aliguori committed
687
        }
688 689 690 691
        qemu_file_skip(f, res);
        buf += res;
        pending -= res;
        done += res;
aliguori's avatar
aliguori committed
692
    }
693
    return done;
aliguori's avatar
aliguori committed
694 695
}

696
static int qemu_peek_byte(QEMUFile *f, int offset)
Juan Quintela's avatar
Juan Quintela committed
697
{
698 699
    int index = f->buf_index + offset;

700
    if (f->is_write) {
Juan Quintela's avatar
Juan Quintela committed
701
        abort();
702
    }
Juan Quintela's avatar
Juan Quintela committed
703

704
    if (index >= f->buf_size) {
Juan Quintela's avatar
Juan Quintela committed
705
        qemu_fill_buffer(f);
706 707
        index = f->buf_index + offset;
        if (index >= f->buf_size) {
Juan Quintela's avatar
Juan Quintela committed
708
            return 0;
709
        }
Juan Quintela's avatar
Juan Quintela committed
710
    }
711
    return f->buf[index];
Juan Quintela's avatar
Juan Quintela committed
712 713
}

aliguori's avatar
aliguori committed
714 715
int qemu_get_byte(QEMUFile *f)
{
716
    int result;
aliguori's avatar
aliguori committed
717

718 719
    result = qemu_peek_byte(f, 0);
    qemu_file_skip(f, 1);
720
    return result;
aliguori's avatar
aliguori committed
721 722
}

723
int64_t qemu_ftell(QEMUFile *f)
aliguori's avatar
aliguori committed
724
{
725 726
    qemu_fflush(f);
    return f->pos;
aliguori's avatar
aliguori committed
727 728 729 730
}

int qemu_file_rate_limit(QEMUFile *f)
{
731 732 733 734 735 736
    if (qemu_file_get_error(f)) {
        return 1;
    }
    if (f->xfer_limit > 0 && f->bytes_xfer > f->xfer_limit) {
        return 1;
    }
aliguori's avatar
aliguori committed
737 738 739
    return 0;
}

740
int64_t qemu_file_get_rate_limit(QEMUFile *f)
lirans@il.ibm.com's avatar
lirans@il.ibm.com committed
741
{
742
    return f->xfer_limit;
lirans@il.ibm.com's avatar
lirans@il.ibm.com committed
743 744
}

745
void qemu_file_set_rate_limit(QEMUFile *f, int64_t limit)
746
{
747 748
    f->xfer_limit = limit;
}
749

750 751 752
void qemu_file_reset_rate_limit(QEMUFile *f)
{
    f->bytes_xfer = 0;
753 754
}

aliguori's avatar
aliguori committed
755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800
void qemu_put_be16(QEMUFile *f, unsigned int v)
{
    qemu_put_byte(f, v >> 8);
    qemu_put_byte(f, v);
}

void qemu_put_be32(QEMUFile *f, unsigned int v)
{
    qemu_put_byte(f, v >> 24);
    qemu_put_byte(f, v >> 16);
    qemu_put_byte(f, v >> 8);
    qemu_put_byte(f, v);
}

void qemu_put_be64(QEMUFile *f, uint64_t v)
{
    qemu_put_be32(f, v >> 32);
    qemu_put_be32(f, v);
}

unsigned int qemu_get_be16(QEMUFile *f)
{
    unsigned int v;
    v = qemu_get_byte(f) << 8;
    v |= qemu_get_byte(f);
    return v;
}

unsigned int qemu_get_be32(QEMUFile *f)
{
    unsigned int v;
    v = qemu_get_byte(f) << 24;
    v |= qemu_get_byte(f) << 16;
    v |= qemu_get_byte(f) << 8;
    v |= qemu_get_byte(f);
    return v;
}

uint64_t qemu_get_be64(QEMUFile *f)
{
    uint64_t v;
    v = (uint64_t)qemu_get_be32(f) << 32;
    v |= qemu_get_be32(f);
    return v;
}

801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824

/* timer */

void qemu_put_timer(QEMUFile *f, QEMUTimer *ts)
{
    uint64_t expire_time;

    expire_time = qemu_timer_expire_time_ns(ts);
    qemu_put_be64(f, expire_time);
}

void qemu_get_timer(QEMUFile *f, QEMUTimer *ts)
{
    uint64_t expire_time;

    expire_time = qemu_get_be64(f);
    if (expire_time != -1) {
        qemu_mod_timer_ns(ts, expire_time);
    } else {
        qemu_del_timer(ts);
    }
}


Gerd Hoffmann's avatar
Gerd Hoffmann committed
825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845
/* bool */

static int get_bool(QEMUFile *f, void *pv, size_t size)
{
    bool *v = pv;
    *v = qemu_get_byte(f);
    return 0;
}

static void put_bool(QEMUFile *f, void *pv, size_t size)
{
    bool *v = pv;
    qemu_put_byte(f, *v);
}

const VMStateInfo vmstate_info_bool = {
    .name = "bool",
    .get  = get_bool,
    .put  = put_bool,
};

846 847 848 849 850 851 852 853 854
/* 8 bit int */

static int get_int8(QEMUFile *f, void *pv, size_t size)
{
    int8_t *v = pv;
    qemu_get_s8s(f, v);
    return 0;
}

855
static void put_int8(QEMUFile *f, void *pv, size_t size)
856
{
857
    int8_t *v = pv;
858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875
    qemu_put_s8s(f, v);
}

const VMStateInfo vmstate_info_int8 = {
    .name = "int8",
    .get  = get_int8,
    .put  = put_int8,
};

/* 16 bit int */

static int get_int16(QEMUFile *f, void *pv, size_t size)
{
    int16_t *v = pv;
    qemu_get_sbe16s(f, v);
    return 0;
}

876
static void put_int16(QEMUFile *f, void *pv, size_t size)
877
{
878
    int16_t *v = pv;
879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896
    qemu_put_sbe16s(f, v);
}

const VMStateInfo vmstate_info_int16 = {
    .name = "int16",
    .get  = get_int16,
    .put  = put_int16,
};

/* 32 bit int */

static int get_int32(QEMUFile *f, void *pv, size_t size)
{
    int32_t *v = pv;
    qemu_get_sbe32s(f, v);
    return 0;
}

897
static void put_int32(QEMUFile *f, void *pv, size_t size)
898
{
899
    int32_t *v = pv;
900 901 902 903 904 905 906 907 908
    qemu_put_sbe32s(f, v);
}

const VMStateInfo vmstate_info_int32 = {
    .name = "int32",
    .get  = get_int32,
    .put  = put_int32,
};

909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928
/* 32 bit int. See that the received value is the same than the one
   in the field */

static int get_int32_equal(QEMUFile *f, void *pv, size_t size)
{
    int32_t *v = pv;
    int32_t v2;
    qemu_get_sbe32s(f, &v2);

    if (*v == v2)
        return 0;
    return -EINVAL;
}

const VMStateInfo vmstate_info_int32_equal = {
    .name = "int32 equal",
    .get  = get_int32_equal,
    .put  = put_int32,
};

929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948
/* 32 bit int. See that the received value is the less or the same
   than the one in the field */

static int get_int32_le(QEMUFile *f, void *pv, size_t size)
{
    int32_t *old = pv;
    int32_t new;
    qemu_get_sbe32s(f, &new);

    if (*old <= new)
        return 0;
    return -EINVAL;
}

const VMStateInfo vmstate_info_int32_le = {
    .name = "int32 equal",
    .get  = get_int32_le,
    .put  = put_int32,
};

949 950 951 952 953 954 955 956 957
/* 64 bit int */

static int get_int64(QEMUFile *f, void *pv, size_t size)
{
    int64_t *v = pv;
    qemu_get_sbe64s(f, v);
    return 0;
}

958
static void put_int64(QEMUFile *f, void *pv, size_t size)
959
{
960
    int64_t *v = pv;
961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978
    qemu_put_sbe64s(f, v);
}

const VMStateInfo vmstate_info_int64 = {
    .name = "int64",
    .get  = get_int64,
    .put  = put_int64,
};

/* 8 bit unsigned int */

static int get_uint8(QEMUFile *f, void *pv, size_t size)
{
    uint8_t *v = pv;
    qemu_get_8s(f, v);
    return 0;
}

979
static void put_uint8(QEMUFile *f, void *pv, size_t size)
980
{
981
    uint8_t *v = pv;
982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999
    qemu_put_8s(f, v);
}

const VMStateInfo vmstate_info_uint8 = {
    .name = "uint8",
    .get  = get_uint8,
    .put  = put_uint8,
};

/* 16 bit unsigned int */

static int get_uint16(QEMUFile *f, void *pv, size_t size)
{
    uint16_t *v = pv;
    qemu_get_be16s(f, v);
    return 0;
}

1000
static void put_uint16(QEMUFile *f, void *pv, size_t size)
1001
{
1002
    uint16_t *v = pv;
1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020
    qemu_put_be16s(f, v);
}

const VMStateInfo vmstate_info_uint16 = {
    .name = "uint16",
    .get  = get_uint16,
    .put  = put_uint16,
};

/* 32 bit unsigned int */

static int get_uint32(QEMUFile *f, void *pv, size_t size)
{
    uint32_t *v = pv;
    qemu_get_be32s(f, v);
    return 0;
}

1021
static void put_uint32(QEMUFile *f, void *pv, size_t size)
1022
{
1023
    uint32_t *v = pv;
1024 1025 1026 1027 1028 1029 1030 1031 1032
    qemu_put_be32s(f, v);
}

const VMStateInfo vmstate_info_uint32 = {
    .name = "uint32",
    .get  = get_uint32,
    .put  = put_uint32,
};

1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053
/* 32 bit uint. See that the received value is the same than the one
   in the field */

static int get_uint32_equal(QEMUFile *f, void *pv, size_t size)
{
    uint32_t *v = pv;
    uint32_t v2;
    qemu_get_be32s(f, &v2);

    if (*v == v2) {
        return 0;
    }
    return -EINVAL;
}

const VMStateInfo vmstate_info_uint32_equal = {
    .name = "uint32 equal",
    .get  = get_uint32_equal,
    .put  = put_uint32,
};

1054 1055 1056 1057 1058 1059 1060 1061 1062
/* 64 bit unsigned int */

static int get_uint64(QEMUFile *f, void *pv, size_t size)
{
    uint64_t *v = pv;
    qemu_get_be64s(f, v);
    return 0;
}

1063
static void put_uint64(QEMUFile *f, void *pv, size_t size)
1064
{