block.c 88.4 KB
Newer Older
B
bellard 已提交
1 2
/*
 * QEMU System Emulator block driver
3
 *
B
bellard 已提交
4
 * Copyright (c) 2003 Fabrice Bellard
5
 *
B
bellard 已提交
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 * 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.
 */
24
#include "config-host.h"
P
pbrook 已提交
25
#include "qemu-common.h"
26
#include "trace.h"
A
aliguori 已提交
27
#include "monitor.h"
B
bellard 已提交
28
#include "block_int.h"
29
#include "module.h"
30
#include "qemu-objects.h"
31
#include "qemu-coroutine.h"
B
bellard 已提交
32

J
Juan Quintela 已提交
33
#ifdef CONFIG_BSD
B
bellard 已提交
34 35 36
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
B
Blue Swirl 已提交
37
#include <sys/queue.h>
38
#ifndef __DragonFly__
B
bellard 已提交
39 40
#include <sys/disk.h>
#endif
41
#endif
B
bellard 已提交
42

43 44 45 46
#ifdef _WIN32
#include <windows.h>
#endif

47 48
static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
49
        BlockDriverCompletionFunc *cb, void *opaque);
50 51
static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
52
        BlockDriverCompletionFunc *cb, void *opaque);
53 54
static BlockDriverAIOCB *bdrv_aio_flush_em(BlockDriverState *bs,
        BlockDriverCompletionFunc *cb, void *opaque);
55 56
static BlockDriverAIOCB *bdrv_aio_noop_em(BlockDriverState *bs,
        BlockDriverCompletionFunc *cb, void *opaque);
57
static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
B
bellard 已提交
58 59 60
                        uint8_t *buf, int nb_sectors);
static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
                         const uint8_t *buf, int nb_sectors);
61 62 63 64 65 66
static BlockDriverAIOCB *bdrv_co_aio_readv_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
        BlockDriverCompletionFunc *cb, void *opaque);
static BlockDriverAIOCB *bdrv_co_aio_writev_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
        BlockDriverCompletionFunc *cb, void *opaque);
67 68 69 70 71 72
static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
                                         int64_t sector_num, int nb_sectors,
                                         QEMUIOVector *iov);
static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
                                         int64_t sector_num, int nb_sectors,
                                         QEMUIOVector *iov);
73
static int coroutine_fn bdrv_co_flush_em(BlockDriverState *bs);
B
bellard 已提交
74

75 76
static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
    QTAILQ_HEAD_INITIALIZER(bdrv_states);
77

78 79
static QLIST_HEAD(, BlockDriver) bdrv_drivers =
    QLIST_HEAD_INITIALIZER(bdrv_drivers);
B
bellard 已提交
80

81 82 83
/* The device to use for VM snapshots */
static BlockDriverState *bs_snapshots;

84 85 86
/* If non-zero, use only whitelisted block drivers */
static int use_bdrv_whitelist;

87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
#ifdef _WIN32
static int is_windows_drive_prefix(const char *filename)
{
    return (((filename[0] >= 'a' && filename[0] <= 'z') ||
             (filename[0] >= 'A' && filename[0] <= 'Z')) &&
            filename[1] == ':');
}

int is_windows_drive(const char *filename)
{
    if (is_windows_drive_prefix(filename) &&
        filename[2] == '\0')
        return 1;
    if (strstart(filename, "\\\\.\\", NULL) ||
        strstart(filename, "//./", NULL))
        return 1;
    return 0;
}
#endif

/* check if the path starts with "<protocol>:" */
static int path_has_protocol(const char *path)
{
#ifdef _WIN32
    if (is_windows_drive(path) ||
        is_windows_drive_prefix(path)) {
        return 0;
    }
#endif

    return strchr(path, ':') != NULL;
}

B
bellard 已提交
120
int path_is_absolute(const char *path)
121
{
B
bellard 已提交
122
    const char *p;
B
bellard 已提交
123 124 125 126 127
#ifdef _WIN32
    /* specific case for names like: "\\.\d:" */
    if (*path == '/' || *path == '\\')
        return 1;
#endif
B
bellard 已提交
128 129 130 131 132
    p = strchr(path, ':');
    if (p)
        p++;
    else
        p = path;
133 134 135 136 137
#ifdef _WIN32
    return (*p == '/' || *p == '\\');
#else
    return (*p == '/');
#endif
138 139
}

B
bellard 已提交
140 141 142 143 144 145
/* if filename is absolute, just copy it to dest. Otherwise, build a
   path to it by considering it is relative to base_path. URL are
   supported. */
void path_combine(char *dest, int dest_size,
                  const char *base_path,
                  const char *filename)
146
{
B
bellard 已提交
147 148 149 150 151 152 153 154 155 156 157 158 159
    const char *p, *p1;
    int len;

    if (dest_size <= 0)
        return;
    if (path_is_absolute(filename)) {
        pstrcpy(dest, dest_size, filename);
    } else {
        p = strchr(base_path, ':');
        if (p)
            p++;
        else
            p = base_path;
160 161 162 163 164 165 166 167 168
        p1 = strrchr(base_path, '/');
#ifdef _WIN32
        {
            const char *p2;
            p2 = strrchr(base_path, '\\');
            if (!p1 || p2 > p1)
                p1 = p2;
        }
#endif
B
bellard 已提交
169 170 171 172 173 174 175 176 177 178 179 180
        if (p1)
            p1++;
        else
            p1 = base_path;
        if (p1 > p)
            p = p1;
        len = p - base_path;
        if (len > dest_size - 1)
            len = dest_size - 1;
        memcpy(dest, base_path, len);
        dest[len] = '\0';
        pstrcat(dest, dest_size, filename);
181 182 183
    }
}

184
void bdrv_register(BlockDriver *bdrv)
B
bellard 已提交
185
{
186 187 188 189 190 191
    if (bdrv->bdrv_co_readv) {
        /* Emulate AIO by coroutines, and sync by AIO */
        bdrv->bdrv_aio_readv = bdrv_co_aio_readv_em;
        bdrv->bdrv_aio_writev = bdrv_co_aio_writev_em;
        bdrv->bdrv_read = bdrv_read_em;
        bdrv->bdrv_write = bdrv_write_em;
192 193 194 195 196 197 198 199 200 201 202 203 204
     } else {
        bdrv->bdrv_co_readv = bdrv_co_readv_em;
        bdrv->bdrv_co_writev = bdrv_co_writev_em;

        if (!bdrv->bdrv_aio_readv) {
            /* add AIO emulation layer */
            bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
            bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
        } else if (!bdrv->bdrv_read) {
            /* add synchronous IO emulation layer */
            bdrv->bdrv_read = bdrv_read_em;
            bdrv->bdrv_write = bdrv_write_em;
        }
B
bellard 已提交
205
    }
206 207 208 209

    if (!bdrv->bdrv_aio_flush)
        bdrv->bdrv_aio_flush = bdrv_aio_flush_em;

210
    QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
B
bellard 已提交
211
}
B
bellard 已提交
212 213 214 215

/* create a new block device (by default it is empty) */
BlockDriverState *bdrv_new(const char *device_name)
{
216
    BlockDriverState *bs;
B
bellard 已提交
217

218
    bs = g_malloc0(sizeof(BlockDriverState));
B
bellard 已提交
219
    pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
B
bellard 已提交
220
    if (device_name[0] != '\0') {
221
        QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
B
bellard 已提交
222
    }
B
bellard 已提交
223 224 225
    return bs;
}

B
bellard 已提交
226 227 228
BlockDriver *bdrv_find_format(const char *format_name)
{
    BlockDriver *drv1;
229 230
    QLIST_FOREACH(drv1, &bdrv_drivers, list) {
        if (!strcmp(drv1->format_name, format_name)) {
B
bellard 已提交
231
            return drv1;
232
        }
B
bellard 已提交
233 234 235 236
    }
    return NULL;
}

237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260
static int bdrv_is_whitelisted(BlockDriver *drv)
{
    static const char *whitelist[] = {
        CONFIG_BDRV_WHITELIST
    };
    const char **p;

    if (!whitelist[0])
        return 1;               /* no whitelist, anything goes */

    for (p = whitelist; *p; p++) {
        if (!strcmp(drv->format_name, *p)) {
            return 1;
        }
    }
    return 0;
}

BlockDriver *bdrv_find_whitelisted_format(const char *format_name)
{
    BlockDriver *drv = bdrv_find_format(format_name);
    return drv && bdrv_is_whitelisted(drv) ? drv : NULL;
}

261 262
int bdrv_create(BlockDriver *drv, const char* filename,
    QEMUOptionParameter *options)
B
bellard 已提交
263 264 265
{
    if (!drv->bdrv_create)
        return -ENOTSUP;
266 267

    return drv->bdrv_create(filename, options);
B
bellard 已提交
268 269
}

270 271 272 273
int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
{
    BlockDriver *drv;

274
    drv = bdrv_find_protocol(filename);
275
    if (drv == NULL) {
276
        return -ENOENT;
277 278 279 280 281
    }

    return bdrv_create(drv, filename, options);
}

B
bellard 已提交
282
#ifdef _WIN32
283
void get_tmp_filename(char *filename, int size)
B
bellard 已提交
284
{
285
    char temp_dir[MAX_PATH];
286

287 288
    GetTempPath(MAX_PATH, temp_dir);
    GetTempFileName(temp_dir, "qem", 0, filename);
B
bellard 已提交
289 290
}
#else
291
void get_tmp_filename(char *filename, int size)
B
bellard 已提交
292
{
B
bellard 已提交
293
    int fd;
294
    const char *tmpdir;
B
bellard 已提交
295
    /* XXX: race condition possible */
A
aurel32 已提交
296 297 298 299
    tmpdir = getenv("TMPDIR");
    if (!tmpdir)
        tmpdir = "/tmp";
    snprintf(filename, size, "%s/vl.XXXXXX", tmpdir);
B
bellard 已提交
300 301 302
    fd = mkstemp(filename);
    close(fd);
}
B
bellard 已提交
303
#endif
B
bellard 已提交
304

305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326
/*
 * Detect host devices. By convention, /dev/cdrom[N] is always
 * recognized as a host CDROM.
 */
static BlockDriver *find_hdev_driver(const char *filename)
{
    int score_max = 0, score;
    BlockDriver *drv = NULL, *d;

    QLIST_FOREACH(d, &bdrv_drivers, list) {
        if (d->bdrv_probe_device) {
            score = d->bdrv_probe_device(filename);
            if (score > score_max) {
                score_max = score;
                drv = d;
            }
        }
    }

    return drv;
}

327
BlockDriver *bdrv_find_protocol(const char *filename)
B
bellard 已提交
328 329 330
{
    BlockDriver *drv1;
    char protocol[128];
331
    int len;
B
bellard 已提交
332
    const char *p;
B
bellard 已提交
333

334 335
    /* TODO Drivers without bdrv_file_open must be specified explicitly */

336 337 338 339 340 341 342 343 344 345 346 347
    /*
     * XXX(hch): we really should not let host device detection
     * override an explicit protocol specification, but moving this
     * later breaks access to device names with colons in them.
     * Thanks to the brain-dead persistent naming schemes on udev-
     * based Linux systems those actually are quite common.
     */
    drv1 = find_hdev_driver(filename);
    if (drv1) {
        return drv1;
    }

348
    if (!path_has_protocol(filename)) {
349
        return bdrv_find_format("file");
350
    }
351 352
    p = strchr(filename, ':');
    assert(p != NULL);
353 354 355 356 357
    len = p - filename;
    if (len > sizeof(protocol) - 1)
        len = sizeof(protocol) - 1;
    memcpy(protocol, filename, len);
    protocol[len] = '\0';
358
    QLIST_FOREACH(drv1, &bdrv_drivers, list) {
359
        if (drv1->protocol_name &&
360
            !strcmp(drv1->protocol_name, protocol)) {
B
bellard 已提交
361
            return drv1;
362
        }
B
bellard 已提交
363 364 365 366
    }
    return NULL;
}

367
static int find_image_format(const char *filename, BlockDriver **pdrv)
368 369 370 371 372 373
{
    int ret, score, score_max;
    BlockDriver *drv1, *drv;
    uint8_t buf[2048];
    BlockDriverState *bs;

374
    ret = bdrv_file_open(&bs, filename, 0);
375 376 377 378
    if (ret < 0) {
        *pdrv = NULL;
        return ret;
    }
379

380 381
    /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
    if (bs->sg || !bdrv_is_inserted(bs)) {
382
        bdrv_delete(bs);
383 384 385 386 387 388
        drv = bdrv_find_format("raw");
        if (!drv) {
            ret = -ENOENT;
        }
        *pdrv = drv;
        return ret;
389
    }
390

B
bellard 已提交
391 392 393
    ret = bdrv_pread(bs, 0, buf, sizeof(buf));
    bdrv_delete(bs);
    if (ret < 0) {
394 395
        *pdrv = NULL;
        return ret;
B
bellard 已提交
396 397
    }

B
bellard 已提交
398
    score_max = 0;
399
    drv = NULL;
400
    QLIST_FOREACH(drv1, &bdrv_drivers, list) {
B
bellard 已提交
401 402 403 404 405 406
        if (drv1->bdrv_probe) {
            score = drv1->bdrv_probe(buf, ret, filename);
            if (score > score_max) {
                score_max = score;
                drv = drv1;
            }
B
bellard 已提交
407
        }
B
bellard 已提交
408
    }
409 410 411 412 413
    if (!drv) {
        ret = -ENOENT;
    }
    *pdrv = drv;
    return ret;
B
bellard 已提交
414 415
}

416 417 418 419 420 421 422
/**
 * Set the current 'total_sectors' value
 */
static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
{
    BlockDriver *drv = bs->drv;

423 424 425 426
    /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
    if (bs->sg)
        return 0;

427 428 429 430 431 432 433 434 435 436 437 438 439
    /* query actual device if possible, otherwise just trust the hint */
    if (drv->bdrv_getlength) {
        int64_t length = drv->bdrv_getlength(bs);
        if (length < 0) {
            return length;
        }
        hint = length >> BDRV_SECTOR_BITS;
    }

    bs->total_sectors = hint;
    return 0;
}

440 441 442 443 444 445 446 447 448 449
/*
 * Common part for opening disk images and files
 */
static int bdrv_open_common(BlockDriverState *bs, const char *filename,
    int flags, BlockDriver *drv)
{
    int ret, open_flags;

    assert(drv != NULL);

450
    bs->file = NULL;
451
    bs->total_sectors = 0;
452 453 454 455 456 457 458 459 460 461 462 463 464
    bs->encrypted = 0;
    bs->valid_key = 0;
    bs->open_flags = flags;
    /* buffer_alignment defaulted to 512, drivers can change this value */
    bs->buffer_alignment = 512;

    pstrcpy(bs->filename, sizeof(bs->filename), filename);

    if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv)) {
        return -ENOTSUP;
    }

    bs->drv = drv;
465
    bs->opaque = g_malloc0(drv->instance_size);
466

467
    if (flags & BDRV_O_CACHE_WB)
468 469 470 471 472 473 474 475 476
        bs->enable_write_cache = 1;

    /*
     * Clear flags that are internal to the block layer before opening the
     * image.
     */
    open_flags = flags & ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);

    /*
S
Stefan Weil 已提交
477
     * Snapshots should be writable.
478 479 480 481 482
     */
    if (bs->is_temporary) {
        open_flags |= BDRV_O_RDWR;
    }

483 484 485 486 487 488 489 490 491 492
    /* Open the image, either directly or using a protocol */
    if (drv->bdrv_file_open) {
        ret = drv->bdrv_file_open(bs, filename, open_flags);
    } else {
        ret = bdrv_file_open(&bs->file, filename, open_flags);
        if (ret >= 0) {
            ret = drv->bdrv_open(bs, open_flags);
        }
    }

493 494 495 496 497
    if (ret < 0) {
        goto free_and_fail;
    }

    bs->keep_read_only = bs->read_only = !(open_flags & BDRV_O_RDWR);
498 499 500 501

    ret = refresh_total_sectors(bs, bs->total_sectors);
    if (ret < 0) {
        goto free_and_fail;
502
    }
503

504 505 506 507 508 509 510 511
#ifndef _WIN32
    if (bs->is_temporary) {
        unlink(filename);
    }
#endif
    return 0;

free_and_fail:
512 513 514 515
    if (bs->file) {
        bdrv_delete(bs->file);
        bs->file = NULL;
    }
516
    g_free(bs->opaque);
517 518 519 520 521
    bs->opaque = NULL;
    bs->drv = NULL;
    return ret;
}

K
Kevin Wolf 已提交
522 523 524
/*
 * Opens a file using a protocol (file, host_device, nbd, ...)
 */
B
bellard 已提交
525
int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
B
bellard 已提交
526
{
B
bellard 已提交
527
    BlockDriverState *bs;
528
    BlockDriver *drv;
B
bellard 已提交
529 530
    int ret;

531
    drv = bdrv_find_protocol(filename);
532 533 534 535
    if (!drv) {
        return -ENOENT;
    }

B
bellard 已提交
536
    bs = bdrv_new("");
K
Kevin Wolf 已提交
537
    ret = bdrv_open_common(bs, filename, flags, drv);
B
bellard 已提交
538 539 540
    if (ret < 0) {
        bdrv_delete(bs);
        return ret;
541
    }
542
    bs->growable = 1;
B
bellard 已提交
543 544 545 546
    *pbs = bs;
    return 0;
}

K
Kevin Wolf 已提交
547 548 549
/*
 * Opens a disk image (raw, qcow2, vmdk, ...)
 */
K
Kevin Wolf 已提交
550 551
int bdrv_open(BlockDriverState *bs, const char *filename, int flags,
              BlockDriver *drv)
B
bellard 已提交
552
{
K
Kevin Wolf 已提交
553
    int ret;
B
bellard 已提交
554

B
bellard 已提交
555
    if (flags & BDRV_O_SNAPSHOT) {
B
bellard 已提交
556 557
        BlockDriverState *bs1;
        int64_t total_size;
A
aliguori 已提交
558
        int is_protocol = 0;
K
Kevin Wolf 已提交
559 560
        BlockDriver *bdrv_qcow2;
        QEMUOptionParameter *options;
K
Kevin Wolf 已提交
561 562
        char tmp_filename[PATH_MAX];
        char backing_filename[PATH_MAX];
563

B
bellard 已提交
564 565
        /* if snapshot, we create a temporary backing file and open it
           instead of opening 'filename' directly */
566

B
bellard 已提交
567 568
        /* if there is a backing file, use it */
        bs1 = bdrv_new("");
K
Kevin Wolf 已提交
569
        ret = bdrv_open(bs1, filename, 0, drv);
570
        if (ret < 0) {
B
bellard 已提交
571
            bdrv_delete(bs1);
572
            return ret;
B
bellard 已提交
573
        }
574
        total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
A
aliguori 已提交
575 576 577 578

        if (bs1->drv && bs1->drv->protocol_name)
            is_protocol = 1;

B
bellard 已提交
579
        bdrv_delete(bs1);
580

B
bellard 已提交
581
        get_tmp_filename(tmp_filename, sizeof(tmp_filename));
A
aliguori 已提交
582 583 584 585 586

        /* Real path is meaningless for protocols */
        if (is_protocol)
            snprintf(backing_filename, sizeof(backing_filename),
                     "%s", filename);
587 588
        else if (!realpath(filename, backing_filename))
            return -errno;
A
aliguori 已提交
589

K
Kevin Wolf 已提交
590 591 592
        bdrv_qcow2 = bdrv_find_format("qcow2");
        options = parse_option_parameters("", bdrv_qcow2->create_options, NULL);

593
        set_option_parameter_int(options, BLOCK_OPT_SIZE, total_size);
K
Kevin Wolf 已提交
594 595 596 597 598 599 600
        set_option_parameter(options, BLOCK_OPT_BACKING_FILE, backing_filename);
        if (drv) {
            set_option_parameter(options, BLOCK_OPT_BACKING_FMT,
                drv->format_name);
        }

        ret = bdrv_create(bdrv_qcow2, tmp_filename, options);
601
        free_option_parameters(options);
602 603
        if (ret < 0) {
            return ret;
B
bellard 已提交
604
        }
K
Kevin Wolf 已提交
605

B
bellard 已提交
606
        filename = tmp_filename;
K
Kevin Wolf 已提交
607
        drv = bdrv_qcow2;
B
bellard 已提交
608 609
        bs->is_temporary = 1;
    }
B
bellard 已提交
610

K
Kevin Wolf 已提交
611
    /* Find the right image format driver */
612
    if (!drv) {
613
        ret = find_image_format(filename, &drv);
614
    }
615

616 617
    if (!drv) {
        goto unlink_and_fail;
B
bellard 已提交
618
    }
K
Kevin Wolf 已提交
619 620 621 622

    /* Open the image */
    ret = bdrv_open_common(bs, filename, flags, drv);
    if (ret < 0) {
623 624 625
        goto unlink_and_fail;
    }

K
Kevin Wolf 已提交
626 627 628 629 630 631 632
    /* If there is a backing file, use it */
    if ((flags & BDRV_O_NO_BACKING) == 0 && bs->backing_file[0] != '\0') {
        char backing_filename[PATH_MAX];
        int back_flags;
        BlockDriver *back_drv = NULL;

        bs->backing_hd = bdrv_new("");
633 634 635 636 637 638 639 640 641 642

        if (path_has_protocol(bs->backing_file)) {
            pstrcpy(backing_filename, sizeof(backing_filename),
                    bs->backing_file);
        } else {
            path_combine(backing_filename, sizeof(backing_filename),
                         filename, bs->backing_file);
        }

        if (bs->backing_format[0] != '\0') {
K
Kevin Wolf 已提交
643
            back_drv = bdrv_find_format(bs->backing_format);
644
        }
K
Kevin Wolf 已提交
645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666

        /* backing files always opened read-only */
        back_flags =
            flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);

        ret = bdrv_open(bs->backing_hd, backing_filename, back_flags, back_drv);
        if (ret < 0) {
            bdrv_close(bs);
            return ret;
        }
        if (bs->is_temporary) {
            bs->backing_hd->keep_read_only = !(flags & BDRV_O_RDWR);
        } else {
            /* base image inherits from "parent" */
            bs->backing_hd->keep_read_only = bs->keep_read_only;
        }
    }

    if (!bdrv_key_required(bs)) {
        /* call the change callback */
        bs->media_changed = 1;
        if (bs->change_cb)
667
            bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
K
Kevin Wolf 已提交
668 669 670 671 672 673 674 675 676 677 678
    }

    return 0;

unlink_and_fail:
    if (bs->is_temporary) {
        unlink(filename);
    }
    return ret;
}

B
bellard 已提交
679 680
void bdrv_close(BlockDriverState *bs)
{
B
bellard 已提交
681
    if (bs->drv) {
682 683 684
        if (bs == bs_snapshots) {
            bs_snapshots = NULL;
        }
685
        if (bs->backing_hd) {
B
bellard 已提交
686
            bdrv_delete(bs->backing_hd);
687 688
            bs->backing_hd = NULL;
        }
B
bellard 已提交
689
        bs->drv->bdrv_close(bs);
690
        g_free(bs->opaque);
B
bellard 已提交
691 692 693 694
#ifdef _WIN32
        if (bs->is_temporary) {
            unlink(bs->filename);
        }
B
bellard 已提交
695
#endif
B
bellard 已提交
696 697
        bs->opaque = NULL;
        bs->drv = NULL;
B
bellard 已提交
698

699 700 701 702
        if (bs->file != NULL) {
            bdrv_close(bs->file);
        }

B
bellard 已提交
703
        /* call the change callback */
B
bellard 已提交
704
        bs->media_changed = 1;
B
bellard 已提交
705
        if (bs->change_cb)
706
            bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
B
bellard 已提交
707 708 709
    }
}

710 711 712 713 714 715 716 717 718
void bdrv_close_all(void)
{
    BlockDriverState *bs;

    QTAILQ_FOREACH(bs, &bdrv_states, list) {
        bdrv_close(bs);
    }
}

719 720 721 722 723 724 725 726 727 728
/* make a BlockDriverState anonymous by removing from bdrv_state list.
   Also, NULL terminate the device_name to prevent double remove */
void bdrv_make_anon(BlockDriverState *bs)
{
    if (bs->device_name[0] != '\0') {
        QTAILQ_REMOVE(&bdrv_states, bs, list);
    }
    bs->device_name[0] = '\0';
}

B
bellard 已提交
729 730
void bdrv_delete(BlockDriverState *bs)
{
731 732
    assert(!bs->peer);

733
    /* remove from list, if necessary */
734
    bdrv_make_anon(bs);
735

B
bellard 已提交
736
    bdrv_close(bs);
737 738 739 740
    if (bs->file != NULL) {
        bdrv_delete(bs->file);
    }

741
    assert(bs != bs_snapshots);
742
    g_free(bs);
B
bellard 已提交
743 744
}

745 746 747 748 749 750 751 752 753 754 755 756 757
int bdrv_attach(BlockDriverState *bs, DeviceState *qdev)
{
    if (bs->peer) {
        return -EBUSY;
    }
    bs->peer = qdev;
    return 0;
}

void bdrv_detach(BlockDriverState *bs, DeviceState *qdev)
{
    assert(bs->peer == qdev);
    bs->peer = NULL;
758 759
    bs->change_cb = NULL;
    bs->change_opaque = NULL;
760 761 762 763 764 765 766
}

DeviceState *bdrv_get_attached(BlockDriverState *bs)
{
    return bs->peer;
}

A
aliguori 已提交
767 768 769
/*
 * Run consistency checks on an image
 *
770
 * Returns 0 if the check could be completed (it doesn't mean that the image is
771
 * free of errors) or -errno when an internal error occurred. The results of the
772
 * check are stored in res.
A
aliguori 已提交
773
 */
774
int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res)
A
aliguori 已提交
775 776 777 778 779
{
    if (bs->drv->bdrv_check == NULL) {
        return -ENOTSUP;
    }

780
    memset(res, 0, sizeof(*res));
781
    return bs->drv->bdrv_check(bs, res);
A
aliguori 已提交
782 783
}

784 785
#define COMMIT_BUF_SECTORS 2048

786 787 788
/* commit COW file into the raw image */
int bdrv_commit(BlockDriverState *bs)
{
B
bellard 已提交
789
    BlockDriver *drv = bs->drv;
790
    BlockDriver *backing_drv;
791 792
    int64_t sector, total_sectors;
    int n, ro, open_flags;
793
    int ret = 0, rw_ret = 0;
794
    uint8_t *buf;
795 796
    char filename[1024];
    BlockDriverState *bs_rw, *bs_ro;
797

B
bellard 已提交
798 799
    if (!drv)
        return -ENOMEDIUM;
800 801 802
    
    if (!bs->backing_hd) {
        return -ENOTSUP;
803 804
    }

805 806 807
    if (bs->backing_hd->keep_read_only) {
        return -EACCES;
    }
808 809

    backing_drv = bs->backing_hd->drv;
810 811 812 813 814 815 816 817 818
    ro = bs->backing_hd->read_only;
    strncpy(filename, bs->backing_hd->filename, sizeof(filename));
    open_flags =  bs->backing_hd->open_flags;

    if (ro) {
        /* re-open as RW */
        bdrv_delete(bs->backing_hd);
        bs->backing_hd = NULL;
        bs_rw = bdrv_new("");
819 820
        rw_ret = bdrv_open(bs_rw, filename, open_flags | BDRV_O_RDWR,
            backing_drv);
821 822 823 824
        if (rw_ret < 0) {
            bdrv_delete(bs_rw);
            /* try to re-open read-only */
            bs_ro = bdrv_new("");
825 826
            ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
                backing_drv);
827 828 829 830 831 832 833 834 835 836
            if (ret < 0) {
                bdrv_delete(bs_ro);
                /* drive not functional anymore */
                bs->drv = NULL;
                return ret;
            }
            bs->backing_hd = bs_ro;
            return rw_ret;
        }
        bs->backing_hd = bs_rw;
B
bellard 已提交
837
    }
838

J
Jan Kiszka 已提交
839
    total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
840
    buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
841 842 843 844 845 846 847 848 849 850 851 852 853

    for (sector = 0; sector < total_sectors; sector += n) {
        if (drv->bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {

            if (bdrv_read(bs, sector, buf, n) != 0) {
                ret = -EIO;
                goto ro_cleanup;
            }

            if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
                ret = -EIO;
                goto ro_cleanup;
            }
B
bellard 已提交
854
        }
855
    }
856

857 858 859 860
    if (drv->bdrv_make_empty) {
        ret = drv->bdrv_make_empty(bs);
        bdrv_flush(bs);
    }
861

862 863 864 865 866 867
    /*
     * Make sure all data we wrote to the backing device is actually
     * stable on disk.
     */
    if (bs->backing_hd)
        bdrv_flush(bs->backing_hd);
868 869

ro_cleanup:
870
    g_free(buf);
871 872 873 874 875 876

    if (ro) {
        /* re-open as RO */
        bdrv_delete(bs->backing_hd);
        bs->backing_hd = NULL;
        bs_ro = bdrv_new("");
877 878
        ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
            backing_drv);
879 880 881 882 883 884 885 886 887 888
        if (ret < 0) {
            bdrv_delete(bs_ro);
            /* drive not functional anymore */
            bs->drv = NULL;
            return ret;
        }
        bs->backing_hd = bs_ro;
        bs->backing_hd->keep_read_only = 0;
    }

889
    return ret;
890 891
}

892 893 894 895 896 897 898 899 900
void bdrv_commit_all(void)
{
    BlockDriverState *bs;

    QTAILQ_FOREACH(bs, &bdrv_states, list) {
        bdrv_commit(bs);
    }
}

K
Kevin Wolf 已提交
901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920
/*
 * Return values:
 * 0        - success
 * -EINVAL  - backing format specified, but no file
 * -ENOSPC  - can't update the backing file because no space is left in the
 *            image file header
 * -ENOTSUP - format driver doesn't support changing the backing file
 */
int bdrv_change_backing_file(BlockDriverState *bs,
    const char *backing_file, const char *backing_fmt)
{
    BlockDriver *drv = bs->drv;

    if (drv->bdrv_change_backing_file != NULL) {
        return drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
    } else {
        return -ENOTSUP;
    }
}

921 922 923 924 925 926 927 928 929 930 931 932 933
static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
                                   size_t size)
{
    int64_t len;

    if (!bdrv_is_inserted(bs))
        return -ENOMEDIUM;

    if (bs->growable)
        return 0;

    len = bdrv_getlength(bs);

K
Kevin Wolf 已提交
934 935 936 937
    if (offset < 0)
        return -EIO;

    if ((offset > len) || (len - offset < size))
938 939 940 941 942 943 944 945
        return -EIO;

    return 0;
}

static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
                              int nb_sectors)
{
946 947
    return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
                                   nb_sectors * BDRV_SECTOR_SIZE);
948 949
}

950 951 952 953 954 955 956 957 958 959 960
static inline bool bdrv_has_async_rw(BlockDriver *drv)
{
    return drv->bdrv_co_readv != bdrv_co_readv_em
        || drv->bdrv_aio_readv != bdrv_aio_readv_em;
}

static inline bool bdrv_has_async_flush(BlockDriver *drv)
{
    return drv->bdrv_aio_flush != bdrv_aio_flush_em;
}

B
bellard 已提交
961
/* return < 0 if error. See bdrv_write() for the return codes */
962
int bdrv_read(BlockDriverState *bs, int64_t sector_num,
B
bellard 已提交
963 964
              uint8_t *buf, int nb_sectors)
{
B
bellard 已提交
965 966
    BlockDriver *drv = bs->drv;

B
bellard 已提交
967 968
    if (!drv)
        return -ENOMEDIUM;
969 970 971 972 973 974 975 976 977 978 979 980

    if (bdrv_has_async_rw(drv) && qemu_in_coroutine()) {
        QEMUIOVector qiov;
        struct iovec iov = {
            .iov_base = (void *)buf,
            .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
        };

        qemu_iovec_init_external(&qiov, &iov, 1);
        return bdrv_co_readv(bs, sector_num, nb_sectors, &qiov);
    }

981 982
    if (bdrv_check_request(bs, sector_num, nb_sectors))
        return -EIO;
B
bellard 已提交
983

984
    return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
B
bellard 已提交
985 986
}

987
static void set_dirty_bitmap(BlockDriverState *bs, int64_t sector_num,
988
                             int nb_sectors, int dirty)
989 990
{
    int64_t start, end;
991
    unsigned long val, idx, bit;
992

J
Jan Kiszka 已提交
993
    start = sector_num / BDRV_SECTORS_PER_DIRTY_CHUNK;
994
    end = (sector_num + nb_sectors - 1) / BDRV_SECTORS_PER_DIRTY_CHUNK;
995 996

    for (; start <= end; start++) {
997 998 999 1000
        idx = start / (sizeof(unsigned long) * 8);
        bit = start % (sizeof(unsigned long) * 8);
        val = bs->dirty_bitmap[idx];
        if (dirty) {
1001
            if (!(val & (1UL << bit))) {
1002
                bs->dirty_count++;
1003
                val |= 1UL << bit;
1004
            }
1005
        } else {
1006
            if (val & (1UL << bit)) {
1007
                bs->dirty_count--;
1008
                val &= ~(1UL << bit);
1009
            }
1010 1011
        }
        bs->dirty_bitmap[idx] = val;
1012 1013 1014
    }
}

1015
/* Return < 0 if error. Important errors are:
B
bellard 已提交
1016 1017 1018 1019 1020
  -EIO         generic I/O error (may happen for all errors)
  -ENOMEDIUM   No media inserted.
  -EINVAL      Invalid sector number or nb_sectors
  -EACCES      Trying to write a read-only device
*/
1021
int bdrv_write(BlockDriverState *bs, int64_t sector_num,
B
bellard 已提交
1022 1023
               const uint8_t *buf, int nb_sectors)
{
B
bellard 已提交
1024
    BlockDriver *drv = bs->drv;
1025

B
bellard 已提交
1026 1027
    if (!bs->drv)
        return -ENOMEDIUM;
1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039

    if (bdrv_has_async_rw(drv) && qemu_in_coroutine()) {
        QEMUIOVector qiov;
        struct iovec iov = {
            .iov_base = (void *)buf,
            .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
        };

        qemu_iovec_init_external(&qiov, &iov, 1);
        return bdrv_co_writev(bs, sector_num, nb_sectors, &qiov);
    }

B
bellard 已提交
1040
    if (bs->read_only)
B
bellard 已提交
1041
        return -EACCES;
1042 1043
    if (bdrv_check_request(bs, sector_num, nb_sectors))
        return -EIO;
1044

1045
    if (bs->dirty_bitmap) {
1046 1047
        set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
    }
1048

1049 1050 1051 1052
    if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
        bs->wr_highest_sector = sector_num + nb_sectors - 1;
    }

1053
    return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
B
bellard 已提交
1054 1055
}

1056 1057
int bdrv_pread(BlockDriverState *bs, int64_t offset,
               void *buf, int count1)
B
bellard 已提交
1058
{
J
Jan Kiszka 已提交
1059
    uint8_t tmp_buf[BDRV_SECTOR_SIZE];
B
bellard 已提交
1060 1061
    int len, nb_sectors, count;
    int64_t sector_num;
1062
    int ret;
B
bellard 已提交
1063 1064 1065

    count = count1;
    /* first read to align to sector start */
J
Jan Kiszka 已提交
1066
    len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
B
bellard 已提交
1067 1068
    if (len > count)
        len = count;
J
Jan Kiszka 已提交
1069
    sector_num = offset >> BDRV_SECTOR_BITS;
B
bellard 已提交
1070
    if (len > 0) {
1071 1072
        if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
            return ret;
J
Jan Kiszka 已提交
1073
        memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
B
bellard 已提交
1074 1075 1076 1077 1078 1079 1080 1081
        count -= len;
        if (count == 0)
            return count1;
        sector_num++;
        buf += len;
    }

    /* read the sectors "in place" */
J
Jan Kiszka 已提交
1082
    nb_sectors = count >> BDRV_SECTOR_BITS;
B
bellard 已提交
1083
    if (nb_sectors > 0) {
1084 1085
        if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
            return ret;
B
bellard 已提交
1086
        sector_num += nb_sectors;
J
Jan Kiszka 已提交
1087
        len = nb_sectors << BDRV_SECTOR_BITS;
B
bellard 已提交
1088 1089 1090 1091 1092 1093
        buf += len;
        count -= len;
    }

    /* add data from the last sector */
    if (count > 0) {
1094 1095
        if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
            return ret;
B
bellard 已提交
1096 1097 1098 1099 1100
        memcpy(buf, tmp_buf, count);
    }
    return count1;
}

1101 1102
int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
                const void *buf, int count1)
B
bellard 已提交
1103
{
J
Jan Kiszka 已提交
1104
    uint8_t tmp_buf[BDRV_SECTOR_SIZE];
B
bellard 已提交
1105 1106
    int len, nb_sectors, count;
    int64_t sector_num;
1107
    int ret;
B
bellard 已提交
1108 1109 1110

    count = count1;
    /* first write to align to sector start */
J
Jan Kiszka 已提交
1111
    len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
B
bellard 已提交
1112 1113
    if (len > count)
        len = count;
J
Jan Kiszka 已提交
1114
    sector_num = offset >> BDRV_SECTOR_BITS;
B
bellard 已提交
1115
    if (len > 0) {
1116 1117
        if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
            return ret;
J
Jan Kiszka 已提交
1118
        memcpy(tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), buf, len);
1119 1120
        if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
            return ret;
B
bellard 已提交
1121 1122 1123 1124 1125 1126 1127 1128
        count -= len;
        if (count == 0)
            return count1;
        sector_num++;
        buf += len;
    }

    /* write the sectors "in place" */
J
Jan Kiszka 已提交
1129
    nb_sectors = count >> BDRV_SECTOR_BITS;
B
bellard 已提交
1130
    if (nb_sectors > 0) {
1131 1132
        if ((ret = bdrv_write(bs, sector_num, buf, nb_sectors)) < 0)
            return ret;
B
bellard 已提交
1133
        sector_num += nb_sectors;
J
Jan Kiszka 已提交
1134
        len = nb_sectors << BDRV_SECTOR_BITS;
B
bellard 已提交
1135 1136 1137 1138 1139 1140
        buf += len;
        count -= len;
    }

    /* add data from the last sector */
    if (count > 0) {
1141 1142
        if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
            return ret;
B
bellard 已提交
1143
        memcpy(tmp_buf, buf, count);
1144 1145
        if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
            return ret;
B
bellard 已提交
1146 1147 1148 1149
    }
    return count1;
}

K
Kevin Wolf 已提交
1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173
/*
 * Writes to the file and ensures that no writes are reordered across this
 * request (acts as a barrier)
 *
 * Returns 0 on success, -errno in error cases.
 */
int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
    const void *buf, int count)
{
    int ret;

    ret = bdrv_pwrite(bs, offset, buf, count);
    if (ret < 0) {
        return ret;
    }

    /* No flush needed for cache=writethrough, it uses O_DSYNC */
    if ((bs->open_flags & BDRV_O_CACHE_MASK) != 0) {
        bdrv_flush(bs);
    }

    return 0;
}

K
Kevin Wolf 已提交
1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218
int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
    int nb_sectors, QEMUIOVector *qiov)
{
    BlockDriver *drv = bs->drv;

    trace_bdrv_co_readv(bs, sector_num, nb_sectors);

    if (!drv) {
        return -ENOMEDIUM;
    }
    if (bdrv_check_request(bs, sector_num, nb_sectors)) {
        return -EIO;
    }

    return drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
}

int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
    int nb_sectors, QEMUIOVector *qiov)
{
    BlockDriver *drv = bs->drv;

    trace_bdrv_co_writev(bs, sector_num, nb_sectors);

    if (!bs->drv) {
        return -ENOMEDIUM;
    }
    if (bs->read_only) {
        return -EACCES;
    }
    if (bdrv_check_request(bs, sector_num, nb_sectors)) {
        return -EIO;
    }

    if (bs->dirty_bitmap) {
        set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
    }

    if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
        bs->wr_highest_sector = sector_num + nb_sectors - 1;
    }

    return drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
}

B
bellard 已提交
1219 1220 1221 1222 1223 1224
/**
 * Truncate file to 'offset' bytes (needed only for file protocols)
 */
int bdrv_truncate(BlockDriverState *bs, int64_t offset)
{
    BlockDriver *drv = bs->drv;
1225
    int ret;
B
bellard 已提交
1226
    if (!drv)
B
bellard 已提交
1227
        return -ENOMEDIUM;
B
bellard 已提交
1228 1229
    if (!drv->bdrv_truncate)
        return -ENOTSUP;
1230 1231
    if (bs->read_only)
        return -EACCES;
M
Marcelo Tosatti 已提交
1232 1233
    if (bdrv_in_use(bs))
        return -EBUSY;
1234 1235 1236
    ret = drv->bdrv_truncate(bs, offset);
    if (ret == 0) {
        ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
1237 1238 1239
        if (bs->change_cb) {
            bs->change_cb(bs->change_opaque, CHANGE_SIZE);
        }
1240 1241
    }
    return ret;
B
bellard 已提交
1242 1243
}

1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262
/**
 * Length of a allocated file in bytes. Sparse files are counted by actual
 * allocated space. Return < 0 if error or unknown.
 */
int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
{
    BlockDriver *drv = bs->drv;
    if (!drv) {
        return -ENOMEDIUM;
    }
    if (drv->bdrv_get_allocated_file_size) {
        return drv->bdrv_get_allocated_file_size(bs);
    }
    if (bs->file) {
        return bdrv_get_allocated_file_size(bs->file);
    }
    return -ENOTSUP;
}

B
bellard 已提交
1263 1264 1265 1266 1267 1268 1269
/**
 * Length of a file in bytes. Return < 0 if error or unknown.
 */
int64_t bdrv_getlength(BlockDriverState *bs)
{
    BlockDriver *drv = bs->drv;
    if (!drv)
B
bellard 已提交
1270
        return -ENOMEDIUM;
1271

1272 1273 1274 1275
    if (bs->growable || bs->removable) {
        if (drv->bdrv_getlength) {
            return drv->bdrv_getlength(bs);
        }
B
bellard 已提交
1276
    }
1277
    return bs->total_sectors * BDRV_SECTOR_SIZE;
B
bellard 已提交
1278 1279
}

B
bellard 已提交
1280
/* return 0 as number of sectors if no device present or error */
1281
void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
B
bellard 已提交
1282
{
B
bellard 已提交
1283 1284 1285 1286 1287
    int64_t length;
    length = bdrv_getlength(bs);
    if (length < 0)
        length = 0;
    else
J
Jan Kiszka 已提交
1288
        length = length >> BDRV_SECTOR_BITS;
B
bellard 已提交
1289
    *nb_sectors_ptr = length;
B
bellard 已提交
1290
}
B
bellard 已提交
1291

1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308
struct partition {
        uint8_t boot_ind;           /* 0x80 - active */
        uint8_t head;               /* starting head */
        uint8_t sector;             /* starting sector */
        uint8_t cyl;                /* starting cylinder */
        uint8_t sys_ind;            /* What partition type */
        uint8_t end_head;           /* end head */
        uint8_t end_sector;         /* end sector */
        uint8_t end_cyl;            /* end cylinder */
        uint32_t start_sect;        /* starting sector counting from 0 */
        uint32_t nr_sects;          /* nr of sectors in partition */
} __attribute__((packed));

/* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */
static int guess_disk_lchs(BlockDriverState *bs,
                           int *pcylinders, int *pheads, int *psectors)
{
1309
    uint8_t buf[BDRV_SECTOR_SIZE];
1310 1311 1312
    int ret, i, heads, sectors, cylinders;
    struct partition *p;
    uint32_t nr_sects;
B
blueswir1 已提交
1313
    uint64_t nb_sectors;
1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352

    bdrv_get_geometry(bs, &nb_sectors);

    ret = bdrv_read(bs, 0, buf, 1);
    if (ret < 0)
        return -1;
    /* test msdos magic */
    if (buf[510] != 0x55 || buf[511] != 0xaa)
        return -1;
    for(i = 0; i < 4; i++) {
        p = ((struct partition *)(buf + 0x1be)) + i;
        nr_sects = le32_to_cpu(p->nr_sects);
        if (nr_sects && p->end_head) {
            /* We make the assumption that the partition terminates on
               a cylinder boundary */
            heads = p->end_head + 1;
            sectors = p->end_sector & 63;
            if (sectors == 0)
                continue;
            cylinders = nb_sectors / (heads * sectors);
            if (cylinders < 1 || cylinders > 16383)
                continue;
            *pheads = heads;
            *psectors = sectors;
            *pcylinders = cylinders;
#if 0
            printf("guessed geometry: LCHS=%d %d %d\n",
                   cylinders, heads, sectors);
#endif
            return 0;
        }
    }
    return -1;
}

void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs)
{
    int translation, lba_detected = 0;
    int cylinders, heads, secs;
B
blueswir1 已提交
1353
    uint64_t nb_sectors;
1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407

    /* if a geometry hint is available, use it */
    bdrv_get_geometry(bs, &nb_sectors);
    bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs);
    translation = bdrv_get_translation_hint(bs);
    if (cylinders != 0) {
        *pcyls = cylinders;
        *pheads = heads;
        *psecs = secs;
    } else {
        if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) {
            if (heads > 16) {
                /* if heads > 16, it means that a BIOS LBA
                   translation was active, so the default
                   hardware geometry is OK */
                lba_detected = 1;
                goto default_geometry;
            } else {
                *pcyls = cylinders;
                *pheads = heads;
                *psecs = secs;
                /* disable any translation to be in sync with
                   the logical geometry */
                if (translation == BIOS_ATA_TRANSLATION_AUTO) {
                    bdrv_set_translation_hint(bs,
                                              BIOS_ATA_TRANSLATION_NONE);
                }
            }
        } else {
        default_geometry:
            /* if no geometry, use a standard physical disk geometry */
            cylinders = nb_sectors / (16 * 63);

            if (cylinders > 16383)
                cylinders = 16383;
            else if (cylinders < 2)
                cylinders = 2;
            *pcyls = cylinders;
            *pheads = 16;
            *psecs = 63;
            if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) {
                if ((*pcyls * *pheads) <= 131072) {
                    bdrv_set_translation_hint(bs,
                                              BIOS_ATA_TRANSLATION_LARGE);
                } else {
                    bdrv_set_translation_hint(bs,
                                              BIOS_ATA_TRANSLATION_LBA);
                }
            }
        }
        bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs);
    }
}

1408
void bdrv_set_geometry_hint(BlockDriverState *bs,
B
bellard 已提交
1409 1410 1411 1412 1413 1414 1415
                            int cyls, int heads, int secs)
{
    bs->cyls = cyls;
    bs->heads = heads;
    bs->secs = secs;
}

B
bellard 已提交
1416 1417 1418 1419 1420
void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
{
    bs->translation = translation;
}

1421
void bdrv_get_geometry_hint(BlockDriverState *bs,
B
bellard 已提交
1422 1423 1424 1425 1426 1427 1428
                            int *pcyls, int *pheads, int *psecs)
{
    *pcyls = bs->cyls;
    *pheads = bs->heads;
    *psecs = bs->secs;
}

1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531
/* Recognize floppy formats */
typedef struct FDFormat {
    FDriveType drive;
    uint8_t last_sect;
    uint8_t max_track;
    uint8_t max_head;
} FDFormat;

static const FDFormat fd_formats[] = {
    /* First entry is default format */
    /* 1.44 MB 3"1/2 floppy disks */
    { FDRIVE_DRV_144, 18, 80, 1, },
    { FDRIVE_DRV_144, 20, 80, 1, },
    { FDRIVE_DRV_144, 21, 80, 1, },
    { FDRIVE_DRV_144, 21, 82, 1, },
    { FDRIVE_DRV_144, 21, 83, 1, },
    { FDRIVE_DRV_144, 22, 80, 1, },
    { FDRIVE_DRV_144, 23, 80, 1, },
    { FDRIVE_DRV_144, 24, 80, 1, },
    /* 2.88 MB 3"1/2 floppy disks */
    { FDRIVE_DRV_288, 36, 80, 1, },
    { FDRIVE_DRV_288, 39, 80, 1, },
    { FDRIVE_DRV_288, 40, 80, 1, },
    { FDRIVE_DRV_288, 44, 80, 1, },
    { FDRIVE_DRV_288, 48, 80, 1, },
    /* 720 kB 3"1/2 floppy disks */
    { FDRIVE_DRV_144,  9, 80, 1, },
    { FDRIVE_DRV_144, 10, 80, 1, },
    { FDRIVE_DRV_144, 10, 82, 1, },
    { FDRIVE_DRV_144, 10, 83, 1, },
    { FDRIVE_DRV_144, 13, 80, 1, },
    { FDRIVE_DRV_144, 14, 80, 1, },
    /* 1.2 MB 5"1/4 floppy disks */
    { FDRIVE_DRV_120, 15, 80, 1, },
    { FDRIVE_DRV_120, 18, 80, 1, },
    { FDRIVE_DRV_120, 18, 82, 1, },
    { FDRIVE_DRV_120, 18, 83, 1, },
    { FDRIVE_DRV_120, 20, 80, 1, },
    /* 720 kB 5"1/4 floppy disks */
    { FDRIVE_DRV_120,  9, 80, 1, },
    { FDRIVE_DRV_120, 11, 80, 1, },
    /* 360 kB 5"1/4 floppy disks */
    { FDRIVE_DRV_120,  9, 40, 1, },
    { FDRIVE_DRV_120,  9, 40, 0, },
    { FDRIVE_DRV_120, 10, 41, 1, },
    { FDRIVE_DRV_120, 10, 42, 1, },
    /* 320 kB 5"1/4 floppy disks */
    { FDRIVE_DRV_120,  8, 40, 1, },
    { FDRIVE_DRV_120,  8, 40, 0, },
    /* 360 kB must match 5"1/4 better than 3"1/2... */
    { FDRIVE_DRV_144,  9, 80, 0, },
    /* end */
    { FDRIVE_DRV_NONE, -1, -1, 0, },
};

void bdrv_get_floppy_geometry_hint(BlockDriverState *bs, int *nb_heads,
                                   int *max_track, int *last_sect,
                                   FDriveType drive_in, FDriveType *drive)
{
    const FDFormat *parse;
    uint64_t nb_sectors, size;
    int i, first_match, match;

    bdrv_get_geometry_hint(bs, nb_heads, max_track, last_sect);
    if (*nb_heads != 0 && *max_track != 0 && *last_sect != 0) {
        /* User defined disk */
    } else {
        bdrv_get_geometry(bs, &nb_sectors);
        match = -1;
        first_match = -1;
        for (i = 0; ; i++) {
            parse = &fd_formats[i];
            if (parse->drive == FDRIVE_DRV_NONE) {
                break;
            }
            if (drive_in == parse->drive ||
                drive_in == FDRIVE_DRV_NONE) {
                size = (parse->max_head + 1) * parse->max_track *
                    parse->last_sect;
                if (nb_sectors == size) {
                    match = i;
                    break;
                }
                if (first_match == -1) {
                    first_match = i;
                }
            }
        }
        if (match == -1) {
            if (first_match == -1) {
                match = 1;
            } else {
                match = first_match;
            }
            parse = &fd_formats[match];
        }
        *nb_heads = parse->max_head + 1;
        *max_track = parse->max_track;
        *last_sect = parse->last_sect;
        *drive = parse->drive;
    }
}

B
bellard 已提交
1532 1533 1534 1535 1536
int bdrv_get_translation_hint(BlockDriverState *bs)
{
    return bs->translation;
}

1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548
void bdrv_set_on_error(BlockDriverState *bs, BlockErrorAction on_read_error,
                       BlockErrorAction on_write_error)
{
    bs->on_read_error = on_read_error;
    bs->on_write_error = on_write_error;
}

BlockErrorAction bdrv_get_on_error(BlockDriverState *bs, int is_read)
{
    return is_read ? bs->on_read_error : bs->on_write_error;
}

1549 1550 1551 1552 1553 1554 1555 1556
void bdrv_set_removable(BlockDriverState *bs, int removable)
{
    bs->removable = removable;
    if (removable && bs == bs_snapshots) {
        bs_snapshots = NULL;
    }
}

B
bellard 已提交
1557 1558 1559 1560 1561 1562 1563 1564 1565 1566
int bdrv_is_removable(BlockDriverState *bs)
{
    return bs->removable;
}

int bdrv_is_read_only(BlockDriverState *bs)
{
    return bs->read_only;
}

1567 1568 1569 1570 1571
int bdrv_is_sg(BlockDriverState *bs)
{
    return bs->sg;
}

1572 1573 1574 1575 1576
int bdrv_enable_write_cache(BlockDriverState *bs)
{
    return bs->enable_write_cache;
}

B
bellard 已提交
1577
/* XXX: no longer used */
1578
void bdrv_set_change_cb(BlockDriverState *bs,
1579 1580
                        void (*change_cb)(void *opaque, int reason),
                        void *opaque)
B
bellard 已提交
1581 1582 1583 1584 1585
{
    bs->change_cb = change_cb;
    bs->change_opaque = opaque;
}

B
bellard 已提交
1586 1587 1588 1589 1590 1591 1592
int bdrv_is_encrypted(BlockDriverState *bs)
{
    if (bs->backing_hd && bs->backing_hd->encrypted)
        return 1;
    return bs->encrypted;
}

1593 1594 1595 1596 1597 1598 1599 1600 1601
int bdrv_key_required(BlockDriverState *bs)
{
    BlockDriverState *backing_hd = bs->backing_hd;

    if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
        return 1;
    return (bs->encrypted && !bs->valid_key);
}

B
bellard 已提交
1602 1603 1604 1605 1606 1607 1608 1609 1610 1611
int bdrv_set_key(BlockDriverState *bs, const char *key)
{
    int ret;
    if (bs->backing_hd && bs->backing_hd->encrypted) {
        ret = bdrv_set_key(bs->backing_hd, key);
        if (ret < 0)
            return ret;
        if (!bs->encrypted)
            return 0;
    }
1612 1613 1614 1615 1616
    if (!bs->encrypted) {
        return -EINVAL;
    } else if (!bs->drv || !bs->drv->bdrv_set_key) {
        return -ENOMEDIUM;
    }
1617
    ret = bs->drv->bdrv_set_key(bs, key);
1618 1619 1620 1621 1622 1623 1624
    if (ret < 0) {
        bs->valid_key = 0;
    } else if (!bs->valid_key) {
        bs->valid_key = 1;
        /* call the change callback now, we skipped it on open */
        bs->media_changed = 1;
        if (bs->change_cb)
1625
            bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
1626
    }
1627
    return ret;
B
bellard 已提交
1628 1629 1630 1631
}

void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
{
B
bellard 已提交
1632
    if (!bs->drv) {
B
bellard 已提交
1633 1634 1635 1636 1637 1638
        buf[0] = '\0';
    } else {
        pstrcpy(buf, buf_size, bs->drv->format_name);
    }
}

1639
void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
B
bellard 已提交
1640 1641 1642 1643
                         void *opaque)
{
    BlockDriver *drv;

1644
    QLIST_FOREACH(drv, &bdrv_drivers, list) {
B
bellard 已提交
1645 1646 1647 1648
        it(opaque, drv->format_name);
    }
}

B
bellard 已提交
1649 1650 1651 1652
BlockDriverState *bdrv_find(const char *name)
{
    BlockDriverState *bs;

1653 1654
    QTAILQ_FOREACH(bs, &bdrv_states, list) {
        if (!strcmp(name, bs->device_name)) {
B
bellard 已提交
1655
            return bs;
1656
        }
B
bellard 已提交
1657 1658 1659 1660
    }
    return NULL;
}

M
Markus Armbruster 已提交
1661 1662 1663 1664 1665 1666 1667 1668
BlockDriverState *bdrv_next(BlockDriverState *bs)
{
    if (!bs) {
        return QTAILQ_FIRST(&bdrv_states);
    }
    return QTAILQ_NEXT(bs, list);
}

1669
void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
B
bellard 已提交
1670 1671 1672
{
    BlockDriverState *bs;

1673
    QTAILQ_FOREACH(bs, &bdrv_states, list) {
1674
        it(opaque, bs);
B
bellard 已提交
1675 1676 1677
    }
}

B
bellard 已提交
1678 1679 1680 1681 1682
const char *bdrv_get_device_name(BlockDriverState *bs)
{
    return bs->device_name;
}

1683
int bdrv_flush(BlockDriverState *bs)
P
pbrook 已提交
1684
{
1685
    if (bs->open_flags & BDRV_O_NO_FLUSH) {
1686 1687 1688
        return 0;
    }

1689 1690 1691 1692
    if (bs->drv && bdrv_has_async_flush(bs->drv) && qemu_in_coroutine()) {
        return bdrv_co_flush_em(bs);
    }

1693 1694
    if (bs->drv && bs->drv->bdrv_flush) {
        return bs->drv->bdrv_flush(bs);
1695 1696
    }

1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707
    /*
     * Some block drivers always operate in either writethrough or unsafe mode
     * and don't support bdrv_flush therefore. Usually qemu doesn't know how
     * the server works (because the behaviour is hardcoded or depends on
     * server-side configuration), so we can't ensure that everything is safe
     * on disk. Returning an error doesn't work because that would break guests
     * even if the server operates in writethrough mode.
     *
     * Let's hope the user knows what he's doing.
     */
    return 0;
P
pbrook 已提交
1708 1709
}

A
aliguori 已提交
1710 1711 1712 1713
void bdrv_flush_all(void)
{
    BlockDriverState *bs;

1714 1715 1716
    QTAILQ_FOREACH(bs, &bdrv_states, list) {
        if (bs->drv && !bdrv_is_read_only(bs) &&
            (!bdrv_is_removable(bs) || bdrv_is_inserted(bs))) {
A
aliguori 已提交
1717
            bdrv_flush(bs);
1718 1719
        }
    }
A
aliguori 已提交
1720 1721
}

K
Kevin Wolf 已提交
1722 1723 1724 1725
int bdrv_has_zero_init(BlockDriverState *bs)
{
    assert(bs->drv);

K
Kevin Wolf 已提交
1726 1727
    if (bs->drv->bdrv_has_zero_init) {
        return bs->drv->bdrv_has_zero_init(bs);
K
Kevin Wolf 已提交
1728 1729 1730 1731 1732
    }

    return 1;
}

C
Christoph Hellwig 已提交
1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743
int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
{
    if (!bs->drv) {
        return -ENOMEDIUM;
    }
    if (!bs->drv->bdrv_discard) {
        return 0;
    }
    return bs->drv->bdrv_discard(bs, sector_num, nb_sectors);
}

1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770
/*
 * Returns true iff the specified sector is present in the disk image. Drivers
 * not implementing the functionality are assumed to not support backing files,
 * hence all their sectors are reported as allocated.
 *
 * 'pnum' is set to the number of sectors (including and immediately following
 * the specified sector) that are known to be in the same
 * allocated/unallocated state.
 *
 * 'nb_sectors' is the max value 'pnum' should be set to.
 */
int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
	int *pnum)
{
    int64_t n;
    if (!bs->drv->bdrv_is_allocated) {
        if (sector_num >= bs->total_sectors) {
            *pnum = 0;
            return 0;
        }
        n = bs->total_sectors - sector_num;
        *pnum = (n < nb_sectors) ? (n) : (nb_sectors);
        return 1;
    }
    return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum);
}

1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799
void bdrv_mon_event(const BlockDriverState *bdrv,
                    BlockMonEventAction action, int is_read)
{
    QObject *data;
    const char *action_str;

    switch (action) {
    case BDRV_ACTION_REPORT:
        action_str = "report";
        break;
    case BDRV_ACTION_IGNORE:
        action_str = "ignore";
        break;
    case BDRV_ACTION_STOP:
        action_str = "stop";
        break;
    default:
        abort();
    }

    data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
                              bdrv->device_name,
                              action_str,
                              is_read ? "read" : "write");
    monitor_protocol_event(QEVENT_BLOCK_IO_ERROR, data);

    qobject_decref(data);
}

1800
static void bdrv_print_dict(QObject *obj, void *opaque)
B
bellard 已提交
1801
{
1802 1803 1804 1805 1806
    QDict *bs_dict;
    Monitor *mon = opaque;

    bs_dict = qobject_to_qdict(obj);

1807
    monitor_printf(mon, "%s: removable=%d",
1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
                        qdict_get_str(bs_dict, "device"),
                        qdict_get_bool(bs_dict, "removable"));

    if (qdict_get_bool(bs_dict, "removable")) {
        monitor_printf(mon, " locked=%d", qdict_get_bool(bs_dict, "locked"));
    }

    if (qdict_haskey(bs_dict, "inserted")) {
        QDict *qdict = qobject_to_qdict(qdict_get(bs_dict, "inserted"));

        monitor_printf(mon, " file=");
        monitor_print_filename(mon, qdict_get_str(qdict, "file"));
        if (qdict_haskey(qdict, "backing_file")) {
            monitor_printf(mon, " backing_file=");
            monitor_print_filename(mon, qdict_get_str(qdict, "backing_file"));
        }
        monitor_printf(mon, " ro=%d drv=%s encrypted=%d",
                            qdict_get_bool(qdict, "ro"),
                            qdict_get_str(qdict, "drv"),
                            qdict_get_bool(qdict, "encrypted"));
    } else {
        monitor_printf(mon, " [not inserted]");
    }

    monitor_printf(mon, "\n");
}

void bdrv_info_print(Monitor *mon, const QObject *data)
{
    qlist_iter(qobject_to_qlist(data), bdrv_print_dict, mon);
}

void bdrv_info(Monitor *mon, QObject **ret_data)
{
    QList *bs_list;
B
bellard 已提交
1843 1844
    BlockDriverState *bs;

1845 1846
    bs_list = qlist_new();

1847
    QTAILQ_FOREACH(bs, &bdrv_states, list) {
1848 1849
        QObject *bs_obj;

1850
        bs_obj = qobject_from_jsonf("{ 'device': %s, 'type': 'unknown', "
1851
                                    "'removable': %i, 'locked': %i }",
1852
                                    bs->device_name, bs->removable,
1853 1854
                                    bs->locked);

B
bellard 已提交
1855
        if (bs->drv) {
1856 1857 1858 1859 1860 1861 1862 1863
            QObject *obj;
            QDict *bs_dict = qobject_to_qdict(bs_obj);

            obj = qobject_from_jsonf("{ 'file': %s, 'ro': %i, 'drv': %s, "
                                     "'encrypted': %i }",
                                     bs->filename, bs->read_only,
                                     bs->drv->format_name,
                                     bdrv_is_encrypted(bs));
1864
            if (bs->backing_file[0] != '\0') {
1865 1866 1867
                QDict *qdict = qobject_to_qdict(obj);
                qdict_put(qdict, "backing_file",
                          qstring_from_str(bs->backing_file));
A
aliguori 已提交
1868
            }
1869 1870

            qdict_put_obj(bs_dict, "inserted", obj);
B
bellard 已提交
1871
        }
1872
        qlist_append_obj(bs_list, bs_obj);
B
bellard 已提交
1873
    }
1874 1875

    *ret_data = QOBJECT(bs_list);
B
bellard 已提交
1876
}
1877

1878
static void bdrv_stats_iter(QObject *data, void *opaque)
1879
{
1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902
    QDict *qdict;
    Monitor *mon = opaque;

    qdict = qobject_to_qdict(data);
    monitor_printf(mon, "%s:", qdict_get_str(qdict, "device"));

    qdict = qobject_to_qdict(qdict_get(qdict, "stats"));
    monitor_printf(mon, " rd_bytes=%" PRId64
                        " wr_bytes=%" PRId64
                        " rd_operations=%" PRId64
                        " wr_operations=%" PRId64
                        "\n",
                        qdict_get_int(qdict, "rd_bytes"),
                        qdict_get_int(qdict, "wr_bytes"),
                        qdict_get_int(qdict, "rd_operations"),
                        qdict_get_int(qdict, "wr_operations"));
}

void bdrv_stats_print(Monitor *mon, const QObject *data)
{
    qlist_iter(qobject_to_qlist(data), bdrv_stats_iter, mon);
}

1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916
static QObject* bdrv_info_stats_bs(BlockDriverState *bs)
{
    QObject *res;
    QDict *dict;

    res = qobject_from_jsonf("{ 'stats': {"
                             "'rd_bytes': %" PRId64 ","
                             "'wr_bytes': %" PRId64 ","
                             "'rd_operations': %" PRId64 ","
                             "'wr_operations': %" PRId64 ","
                             "'wr_highest_offset': %" PRId64
                             "} }",
                             bs->rd_bytes, bs->wr_bytes,
                             bs->rd_ops, bs->wr_ops,
1917 1918
                             bs->wr_highest_sector *
                             (uint64_t)BDRV_SECTOR_SIZE);
1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932
    dict  = qobject_to_qdict(res);

    if (*bs->device_name) {
        qdict_put(dict, "device", qstring_from_str(bs->device_name));
    }

    if (bs->file) {
        QObject *parent = bdrv_info_stats_bs(bs->file);
        qdict_put_obj(dict, "parent", parent);
    }

    return res;
}

1933 1934 1935 1936
void bdrv_info_stats(Monitor *mon, QObject **ret_data)
{
    QObject *obj;
    QList *devices;
1937 1938
    BlockDriverState *bs;

1939 1940
    devices = qlist_new();

1941
    QTAILQ_FOREACH(bs, &bdrv_states, list) {
1942
        obj = bdrv_info_stats_bs(bs);
1943
        qlist_append_obj(devices, obj);
1944
    }
1945 1946

    *ret_data = QOBJECT(devices);
1947
}
B
bellard 已提交
1948

1949 1950 1951 1952 1953 1954 1955 1956 1957 1958
const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
{
    if (bs->backing_hd && bs->backing_hd->encrypted)
        return bs->backing_file;
    else if (bs->encrypted)
        return bs->filename;
    else
        return NULL;
}

1959
void bdrv_get_backing_filename(BlockDriverState *bs,
B
bellard 已提交
1960 1961
                               char *filename, int filename_size)
{
K
Kevin Wolf 已提交
1962
    if (!bs->backing_file) {
B
bellard 已提交
1963 1964 1965 1966 1967 1968
        pstrcpy(filename, filename_size, "");
    } else {
        pstrcpy(filename, filename_size, bs->backing_file);
    }
}

1969
int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
B
bellard 已提交
1970 1971 1972 1973
                          const uint8_t *buf, int nb_sectors)
{
    BlockDriver *drv = bs->drv;
    if (!drv)
B
bellard 已提交
1974
        return -ENOMEDIUM;
B
bellard 已提交
1975 1976
    if (!drv->bdrv_write_compressed)
        return -ENOTSUP;
K
Kevin Wolf 已提交
1977 1978
    if (bdrv_check_request(bs, sector_num, nb_sectors))
        return -EIO;
1979

1980
    if (bs->dirty_bitmap) {
1981 1982
        set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
    }
1983

B
bellard 已提交
1984 1985
    return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
}
1986

B
bellard 已提交
1987 1988 1989 1990
int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
{
    BlockDriver *drv = bs->drv;
    if (!drv)
B
bellard 已提交
1991
        return -ENOMEDIUM;
B
bellard 已提交
1992 1993 1994 1995 1996 1997
    if (!drv->bdrv_get_info)
        return -ENOTSUP;
    memset(bdi, 0, sizeof(*bdi));
    return drv->bdrv_get_info(bs, bdi);
}

1998 1999
int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
                      int64_t pos, int size)
2000 2001 2002 2003
{
    BlockDriver *drv = bs->drv;
    if (!drv)
        return -ENOMEDIUM;
2004 2005 2006 2007 2008
    if (drv->bdrv_save_vmstate)
        return drv->bdrv_save_vmstate(bs, buf, pos, size);
    if (bs->file)
        return bdrv_save_vmstate(bs->file, buf, pos, size);
    return -ENOTSUP;
2009 2010
}

2011 2012
int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
                      int64_t pos, int size)
2013 2014 2015 2016
{
    BlockDriver *drv = bs->drv;
    if (!drv)
        return -ENOMEDIUM;
2017 2018 2019 2020 2021
    if (drv->bdrv_load_vmstate)
        return drv->bdrv_load_vmstate(bs, buf, pos, size);
    if (bs->file)
        return bdrv_load_vmstate(bs->file, buf, pos, size);
    return -ENOTSUP;
2022 2023
}

K
Kevin Wolf 已提交
2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
{
    BlockDriver *drv = bs->drv;

    if (!drv || !drv->bdrv_debug_event) {
        return;
    }

    return drv->bdrv_debug_event(bs, event);

}

B
bellard 已提交
2036 2037 2038
/**************************************************************/
/* handling of snapshots */

2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055
int bdrv_can_snapshot(BlockDriverState *bs)
{
    BlockDriver *drv = bs->drv;
    if (!drv || bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
        return 0;
    }

    if (!drv->bdrv_snapshot_create) {
        if (bs->file != NULL) {
            return bdrv_can_snapshot(bs->file);
        }
        return 0;
    }

    return 1;
}

2056 2057 2058 2059 2060
int bdrv_is_snapshot(BlockDriverState *bs)
{
    return !!(bs->open_flags & BDRV_O_SNAPSHOT);
}

2061 2062 2063 2064
BlockDriverState *bdrv_snapshots(void)
{
    BlockDriverState *bs;

2065
    if (bs_snapshots) {
2066
        return bs_snapshots;
2067
    }
2068 2069 2070 2071

    bs = NULL;
    while ((bs = bdrv_next(bs))) {
        if (bdrv_can_snapshot(bs)) {
2072 2073
            bs_snapshots = bs;
            return bs;
2074 2075 2076 2077 2078
        }
    }
    return NULL;
}

2079
int bdrv_snapshot_create(BlockDriverState *bs,
B
bellard 已提交
2080 2081 2082 2083
                         QEMUSnapshotInfo *sn_info)
{
    BlockDriver *drv = bs->drv;
    if (!drv)
B
bellard 已提交
2084
        return -ENOMEDIUM;
2085 2086 2087 2088 2089
    if (drv->bdrv_snapshot_create)
        return drv->bdrv_snapshot_create(bs, sn_info);
    if (bs->file)
        return bdrv_snapshot_create(bs->file, sn_info);
    return -ENOTSUP;
B
bellard 已提交
2090 2091
}

2092
int bdrv_snapshot_goto(BlockDriverState *bs,
B
bellard 已提交
2093 2094 2095
                       const char *snapshot_id)
{
    BlockDriver *drv = bs->drv;
2096 2097
    int ret, open_ret;

B
bellard 已提交
2098
    if (!drv)
B
bellard 已提交
2099
        return -ENOMEDIUM;
2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115
    if (drv->bdrv_snapshot_goto)
        return drv->bdrv_snapshot_goto(bs, snapshot_id);

    if (bs->file) {
        drv->bdrv_close(bs);
        ret = bdrv_snapshot_goto(bs->file, snapshot_id);
        open_ret = drv->bdrv_open(bs, bs->open_flags);
        if (open_ret < 0) {
            bdrv_delete(bs->file);
            bs->drv = NULL;
            return open_ret;
        }
        return ret;
    }

    return -ENOTSUP;
B
bellard 已提交
2116 2117 2118 2119 2120 2121
}

int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
{
    BlockDriver *drv = bs->drv;
    if (!drv)
B
bellard 已提交
2122
        return -ENOMEDIUM;
2123 2124 2125 2126 2127
    if (drv->bdrv_snapshot_delete)
        return drv->bdrv_snapshot_delete(bs, snapshot_id);
    if (bs->file)
        return bdrv_snapshot_delete(bs->file, snapshot_id);
    return -ENOTSUP;
B
bellard 已提交
2128 2129
}

2130
int bdrv_snapshot_list(BlockDriverState *bs,
B
bellard 已提交
2131 2132 2133 2134
                       QEMUSnapshotInfo **psn_info)
{
    BlockDriver *drv = bs->drv;
    if (!drv)
B
bellard 已提交
2135
        return -ENOMEDIUM;
2136 2137 2138 2139 2140
    if (drv->bdrv_snapshot_list)
        return drv->bdrv_snapshot_list(bs, psn_info);
    if (bs->file)
        return bdrv_snapshot_list(bs->file, psn_info);
    return -ENOTSUP;
B
bellard 已提交
2141 2142
}

E
edison 已提交
2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158
int bdrv_snapshot_load_tmp(BlockDriverState *bs,
        const char *snapshot_name)
{
    BlockDriver *drv = bs->drv;
    if (!drv) {
        return -ENOMEDIUM;
    }
    if (!bs->read_only) {
        return -EINVAL;
    }
    if (drv->bdrv_snapshot_load_tmp) {
        return drv->bdrv_snapshot_load_tmp(bs, snapshot_name);
    }
    return -ENOTSUP;
}

B
bellard 已提交
2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172
#define NB_SUFFIXES 4

char *get_human_readable_size(char *buf, int buf_size, int64_t size)
{
    static const char suffixes[NB_SUFFIXES] = "KMGT";
    int64_t base;
    int i;

    if (size <= 999) {
        snprintf(buf, buf_size, "%" PRId64, size);
    } else {
        base = 1024;
        for(i = 0; i < NB_SUFFIXES; i++) {
            if (size < (10 * base)) {
2173
                snprintf(buf, buf_size, "%0.1f%c",
B
bellard 已提交
2174 2175 2176 2177
                         (double)size / base,
                         suffixes[i]);
                break;
            } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
2178
                snprintf(buf, buf_size, "%" PRId64 "%c",
B
bellard 已提交
2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191
                         ((size + (base >> 1)) / base),
                         suffixes[i]);
                break;
            }
            base = base * 1024;
        }
    }
    return buf;
}

char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
{
    char buf1[128], date_buf[128], clock_buf[128];
2192 2193 2194
#ifdef _WIN32
    struct tm *ptm;
#else
B
bellard 已提交
2195
    struct tm tm;
2196
#endif
B
bellard 已提交
2197 2198 2199 2200
    time_t ti;
    int64_t secs;

    if (!sn) {
2201 2202
        snprintf(buf, buf_size,
                 "%-10s%-20s%7s%20s%15s",
B
bellard 已提交
2203 2204 2205
                 "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
    } else {
        ti = sn->date_sec;
2206 2207 2208 2209 2210
#ifdef _WIN32
        ptm = localtime(&ti);
        strftime(date_buf, sizeof(date_buf),
                 "%Y-%m-%d %H:%M:%S", ptm);
#else
B
bellard 已提交
2211 2212 2213
        localtime_r(&ti, &tm);
        strftime(date_buf, sizeof(date_buf),
                 "%Y-%m-%d %H:%M:%S", &tm);
2214
#endif
B
bellard 已提交
2215 2216 2217 2218 2219
        secs = sn->vm_clock_nsec / 1000000000;
        snprintf(clock_buf, sizeof(clock_buf),
                 "%02d:%02d:%02d.%03d",
                 (int)(secs / 3600),
                 (int)((secs / 60) % 60),
2220
                 (int)(secs % 60),
B
bellard 已提交
2221 2222
                 (int)((sn->vm_clock_nsec / 1000000) % 1000));
        snprintf(buf, buf_size,
2223
                 "%-10s%-20s%7s%20s%15s",
B
bellard 已提交
2224 2225 2226 2227 2228 2229 2230 2231
                 sn->id_str, sn->name,
                 get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
                 date_buf,
                 clock_buf);
    }
    return buf;
}

B
bellard 已提交
2232

B
bellard 已提交
2233
/**************************************************************/
B
bellard 已提交
2234
/* async I/Os */
B
bellard 已提交
2235

2236
BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
2237
                                 QEMUIOVector *qiov, int nb_sectors,
2238
                                 BlockDriverCompletionFunc *cb, void *opaque)
B
bellard 已提交
2239 2240
{
    BlockDriver *drv = bs->drv;
2241
    BlockDriverAIOCB *ret;
B
bellard 已提交
2242

2243 2244
    trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);

B
bellard 已提交
2245
    if (!drv)
2246
        return NULL;
2247 2248
    if (bdrv_check_request(bs, sector_num, nb_sectors))
        return NULL;
2249

2250 2251
    ret = drv->bdrv_aio_readv(bs, sector_num, qiov, nb_sectors,
                              cb, opaque);
2252 2253

    if (ret) {
R
Robert Wang 已提交
2254 2255 2256
        /* Update stats even though technically transfer has not happened. */
        bs->rd_bytes += (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
        bs->rd_ops++;
2257 2258 2259
    }

    return ret;
B
bellard 已提交
2260 2261
}

2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277
typedef struct BlockCompleteData {
    BlockDriverCompletionFunc *cb;
    void *opaque;
    BlockDriverState *bs;
    int64_t sector_num;
    int nb_sectors;
} BlockCompleteData;

static void block_complete_cb(void *opaque, int ret)
{
    BlockCompleteData *b = opaque;

    if (b->bs->dirty_bitmap) {
        set_dirty_bitmap(b->bs, b->sector_num, b->nb_sectors, 1);
    }
    b->cb(b->opaque, ret);
2278
    g_free(b);
2279 2280 2281 2282 2283 2284 2285 2286
}

static BlockCompleteData *blk_dirty_cb_alloc(BlockDriverState *bs,
                                             int64_t sector_num,
                                             int nb_sectors,
                                             BlockDriverCompletionFunc *cb,
                                             void *opaque)
{
2287
    BlockCompleteData *blkdata = g_malloc0(sizeof(BlockCompleteData));
2288 2289 2290 2291 2292 2293 2294 2295 2296 2297

    blkdata->bs = bs;
    blkdata->cb = cb;
    blkdata->opaque = opaque;
    blkdata->sector_num = sector_num;
    blkdata->nb_sectors = nb_sectors;

    return blkdata;
}

2298 2299 2300
BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
                                  QEMUIOVector *qiov, int nb_sectors,
                                  BlockDriverCompletionFunc *cb, void *opaque)
B
bellard 已提交
2301
{
B
bellard 已提交
2302
    BlockDriver *drv = bs->drv;
2303
    BlockDriverAIOCB *ret;
2304
    BlockCompleteData *blk_cb_data;
B
bellard 已提交
2305

2306 2307
    trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);

B
bellard 已提交
2308
    if (!drv)
2309
        return NULL;
B
bellard 已提交
2310
    if (bs->read_only)
2311
        return NULL;
2312 2313
    if (bdrv_check_request(bs, sector_num, nb_sectors))
        return NULL;
B
bellard 已提交
2314

2315
    if (bs->dirty_bitmap) {
2316 2317 2318 2319
        blk_cb_data = blk_dirty_cb_alloc(bs, sector_num, nb_sectors, cb,
                                         opaque);
        cb = &block_complete_cb;
        opaque = blk_cb_data;
2320
    }
2321

2322 2323
    ret = drv->bdrv_aio_writev(bs, sector_num, qiov, nb_sectors,
                               cb, opaque);
2324 2325

    if (ret) {
2326 2327 2328 2329 2330 2331
        /* Update stats even though technically transfer has not happened. */
        bs->wr_bytes += (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
        bs->wr_ops ++;
        if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
            bs->wr_highest_sector = sector_num + nb_sectors - 1;
        }
2332 2333 2334
    }

    return ret;
B
bellard 已提交
2335 2336
}

K
Kevin Wolf 已提交
2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355

typedef struct MultiwriteCB {
    int error;
    int num_requests;
    int num_callbacks;
    struct {
        BlockDriverCompletionFunc *cb;
        void *opaque;
        QEMUIOVector *free_qiov;
        void *free_buf;
    } callbacks[];
} MultiwriteCB;

static void multiwrite_user_cb(MultiwriteCB *mcb)
{
    int i;

    for (i = 0; i < mcb->num_callbacks; i++) {
        mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
2356 2357 2358
        if (mcb->callbacks[i].free_qiov) {
            qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
        }
2359
        g_free(mcb->callbacks[i].free_qiov);
2360
        qemu_vfree(mcb->callbacks[i].free_buf);
K
Kevin Wolf 已提交
2361 2362 2363 2364 2365 2366 2367
    }
}

static void multiwrite_cb(void *opaque, int ret)
{
    MultiwriteCB *mcb = opaque;

2368 2369
    trace_multiwrite_cb(mcb, ret);

2370
    if (ret < 0 && !mcb->error) {
K
Kevin Wolf 已提交
2371 2372 2373 2374 2375
        mcb->error = ret;
    }

    mcb->num_requests--;
    if (mcb->num_requests == 0) {
2376
        multiwrite_user_cb(mcb);
2377
        g_free(mcb);
K
Kevin Wolf 已提交
2378 2379 2380 2381 2382
    }
}

static int multiwrite_req_compare(const void *a, const void *b)
{
2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395
    const BlockRequest *req1 = a, *req2 = b;

    /*
     * Note that we can't simply subtract req2->sector from req1->sector
     * here as that could overflow the return value.
     */
    if (req1->sector > req2->sector) {
        return 1;
    } else if (req1->sector < req2->sector) {
        return -1;
    } else {
        return 0;
    }
K
Kevin Wolf 已提交
2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430
}

/*
 * Takes a bunch of requests and tries to merge them. Returns the number of
 * requests that remain after merging.
 */
static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
    int num_reqs, MultiwriteCB *mcb)
{
    int i, outidx;

    // Sort requests by start sector
    qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);

    // Check if adjacent requests touch the same clusters. If so, combine them,
    // filling up gaps with zero sectors.
    outidx = 0;
    for (i = 1; i < num_reqs; i++) {
        int merge = 0;
        int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;

        // This handles the cases that are valid for all block drivers, namely
        // exactly sequential writes and overlapping writes.
        if (reqs[i].sector <= oldreq_last) {
            merge = 1;
        }

        // The block driver may decide that it makes sense to combine requests
        // even if there is a gap of some sectors between them. In this case,
        // the gap is filled with zeros (therefore only applicable for yet
        // unused space in format like qcow2).
        if (!merge && bs->drv->bdrv_merge_requests) {
            merge = bs->drv->bdrv_merge_requests(bs, &reqs[outidx], &reqs[i]);
        }

2431 2432 2433 2434
        if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
            merge = 0;
        }

K
Kevin Wolf 已提交
2435 2436
        if (merge) {
            size_t size;
2437
            QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
K
Kevin Wolf 已提交
2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457
            qemu_iovec_init(qiov,
                reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);

            // Add the first request to the merged one. If the requests are
            // overlapping, drop the last sectors of the first request.
            size = (reqs[i].sector - reqs[outidx].sector) << 9;
            qemu_iovec_concat(qiov, reqs[outidx].qiov, size);

            // We might need to add some zeros between the two requests
            if (reqs[i].sector > oldreq_last) {
                size_t zero_bytes = (reqs[i].sector - oldreq_last) << 9;
                uint8_t *buf = qemu_blockalign(bs, zero_bytes);
                memset(buf, 0, zero_bytes);
                qemu_iovec_add(qiov, buf, zero_bytes);
                mcb->callbacks[i].free_buf = buf;
            }

            // Add the second request
            qemu_iovec_concat(qiov, reqs[i].qiov, reqs[i].qiov->size);

2458
            reqs[outidx].nb_sectors = qiov->size >> 9;
K
Kevin Wolf 已提交
2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492
            reqs[outidx].qiov = qiov;

            mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
        } else {
            outidx++;
            reqs[outidx].sector     = reqs[i].sector;
            reqs[outidx].nb_sectors = reqs[i].nb_sectors;
            reqs[outidx].qiov       = reqs[i].qiov;
        }
    }

    return outidx + 1;
}

/*
 * Submit multiple AIO write requests at once.
 *
 * On success, the function returns 0 and all requests in the reqs array have
 * been submitted. In error case this function returns -1, and any of the
 * requests may or may not be submitted yet. In particular, this means that the
 * callback will be called for some of the requests, for others it won't. The
 * caller must check the error field of the BlockRequest to wait for the right
 * callbacks (if error != 0, no callback will be called).
 *
 * The implementation may modify the contents of the reqs array, e.g. to merge
 * requests. However, the fields opaque and error are left unmodified as they
 * are used to signal failure for a single request to the caller.
 */
int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
{
    BlockDriverAIOCB *acb;
    MultiwriteCB *mcb;
    int i;

2493 2494 2495 2496 2497 2498 2499 2500
    /* don't submit writes if we don't have a medium */
    if (bs->drv == NULL) {
        for (i = 0; i < num_reqs; i++) {
            reqs[i].error = -ENOMEDIUM;
        }
        return -1;
    }

K
Kevin Wolf 已提交
2501 2502 2503 2504 2505
    if (num_reqs == 0) {
        return 0;
    }

    // Create MultiwriteCB structure
2506
    mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
K
Kevin Wolf 已提交
2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517
    mcb->num_requests = 0;
    mcb->num_callbacks = num_reqs;

    for (i = 0; i < num_reqs; i++) {
        mcb->callbacks[i].cb = reqs[i].cb;
        mcb->callbacks[i].opaque = reqs[i].opaque;
    }

    // Check for mergable requests
    num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);

2518 2519
    trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);

2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540
    /*
     * Run the aio requests. As soon as one request can't be submitted
     * successfully, fail all requests that are not yet submitted (we must
     * return failure for all requests anyway)
     *
     * num_requests cannot be set to the right value immediately: If
     * bdrv_aio_writev fails for some request, num_requests would be too high
     * and therefore multiwrite_cb() would never recognize the multiwrite
     * request as completed. We also cannot use the loop variable i to set it
     * when the first request fails because the callback may already have been
     * called for previously submitted requests. Thus, num_requests must be
     * incremented for each request that is submitted.
     *
     * The problem that callbacks may be called early also means that we need
     * to take care that num_requests doesn't become 0 before all requests are
     * submitted - multiwrite_cb() would consider the multiwrite request
     * completed. A dummy request that is "completed" by a manual call to
     * multiwrite_cb() takes care of this.
     */
    mcb->num_requests = 1;

2541
    // Run the aio requests
K
Kevin Wolf 已提交
2542
    for (i = 0; i < num_reqs; i++) {
2543
        mcb->num_requests++;
K
Kevin Wolf 已提交
2544 2545 2546 2547 2548 2549 2550
        acb = bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
            reqs[i].nb_sectors, multiwrite_cb, mcb);

        if (acb == NULL) {
            // We can only fail the whole thing if no request has been
            // submitted yet. Otherwise we'll wait for the submitted AIOs to
            // complete and report the error in the callback.
2551
            if (i == 0) {
2552
                trace_bdrv_aio_multiwrite_earlyfail(mcb);
K
Kevin Wolf 已提交
2553 2554
                goto fail;
            } else {
2555
                trace_bdrv_aio_multiwrite_latefail(mcb, i);
2556
                multiwrite_cb(mcb, -EIO);
K
Kevin Wolf 已提交
2557 2558 2559 2560 2561
                break;
            }
        }
    }

2562 2563 2564
    /* Complete the dummy request */
    multiwrite_cb(mcb, 0);

K
Kevin Wolf 已提交
2565 2566 2567
    return 0;

fail:
2568 2569 2570
    for (i = 0; i < mcb->num_callbacks; i++) {
        reqs[i].error = -EIO;
    }
2571
    g_free(mcb);
K
Kevin Wolf 已提交
2572 2573 2574
    return -1;
}

2575 2576 2577 2578 2579
BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
        BlockDriverCompletionFunc *cb, void *opaque)
{
    BlockDriver *drv = bs->drv;

S
Stefan Hajnoczi 已提交
2580 2581
    trace_bdrv_aio_flush(bs, opaque);

2582 2583 2584 2585
    if (bs->open_flags & BDRV_O_NO_FLUSH) {
        return bdrv_aio_noop_em(bs, cb, opaque);
    }

2586 2587 2588 2589 2590
    if (!drv)
        return NULL;
    return drv->bdrv_aio_flush(bs, cb, opaque);
}

B
bellard 已提交
2591 2592
void bdrv_aio_cancel(BlockDriverAIOCB *acb)
{
2593
    acb->pool->cancel(acb);
B
bellard 已提交
2594 2595
}

2596

B
bellard 已提交
2597 2598 2599
/**************************************************************/
/* async block device emulation */

2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611
typedef struct BlockDriverAIOCBSync {
    BlockDriverAIOCB common;
    QEMUBH *bh;
    int ret;
    /* vector translation state */
    QEMUIOVector *qiov;
    uint8_t *bounce;
    int is_write;
} BlockDriverAIOCBSync;

static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
{
2612 2613
    BlockDriverAIOCBSync *acb =
        container_of(blockacb, BlockDriverAIOCBSync, common);
2614
    qemu_bh_delete(acb->bh);
A
Avi Kivity 已提交
2615
    acb->bh = NULL;
2616 2617 2618 2619 2620 2621 2622 2623
    qemu_aio_release(acb);
}

static AIOPool bdrv_em_aio_pool = {
    .aiocb_size         = sizeof(BlockDriverAIOCBSync),
    .cancel             = bdrv_aio_cancel_em,
};

2624
static void bdrv_aio_bh_cb(void *opaque)
B
bellard 已提交
2625
{
2626
    BlockDriverAIOCBSync *acb = opaque;
2627 2628 2629

    if (!acb->is_write)
        qemu_iovec_from_buffer(acb->qiov, acb->bounce, acb->qiov->size);
2630
    qemu_vfree(acb->bounce);
2631
    acb->common.cb(acb->common.opaque, acb->ret);
2632
    qemu_bh_delete(acb->bh);
A
Avi Kivity 已提交
2633
    acb->bh = NULL;
2634
    qemu_aio_release(acb);
B
bellard 已提交
2635
}
2636

2637 2638 2639 2640 2641 2642 2643 2644
static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
                                            int64_t sector_num,
                                            QEMUIOVector *qiov,
                                            int nb_sectors,
                                            BlockDriverCompletionFunc *cb,
                                            void *opaque,
                                            int is_write)

B
bellard 已提交
2645
{
2646 2647
    BlockDriverAIOCBSync *acb;

2648
    acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2649 2650
    acb->is_write = is_write;
    acb->qiov = qiov;
2651
    acb->bounce = qemu_blockalign(bs, qiov->size);
2652

2653 2654
    if (!acb->bh)
        acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2655 2656 2657 2658 2659 2660 2661 2662

    if (is_write) {
        qemu_iovec_to_buffer(acb->qiov, acb->bounce);
        acb->ret = bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
    } else {
        acb->ret = bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
    }

2663
    qemu_bh_schedule(acb->bh);
2664

2665
    return &acb->common;
2666 2667
}

2668 2669
static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
2670
        BlockDriverCompletionFunc *cb, void *opaque)
2671
{
2672 2673
    return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
}
B
bellard 已提交
2674

2675 2676 2677 2678 2679
static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
        BlockDriverCompletionFunc *cb, void *opaque)
{
    return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
2680 2681
}

2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764

typedef struct BlockDriverAIOCBCoroutine {
    BlockDriverAIOCB common;
    BlockRequest req;
    bool is_write;
    QEMUBH* bh;
} BlockDriverAIOCBCoroutine;

static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
{
    qemu_aio_flush();
}

static AIOPool bdrv_em_co_aio_pool = {
    .aiocb_size         = sizeof(BlockDriverAIOCBCoroutine),
    .cancel             = bdrv_aio_co_cancel_em,
};

static void bdrv_co_rw_bh(void *opaque)
{
    BlockDriverAIOCBCoroutine *acb = opaque;

    acb->common.cb(acb->common.opaque, acb->req.error);
    qemu_bh_delete(acb->bh);
    qemu_aio_release(acb);
}

static void coroutine_fn bdrv_co_rw(void *opaque)
{
    BlockDriverAIOCBCoroutine *acb = opaque;
    BlockDriverState *bs = acb->common.bs;

    if (!acb->is_write) {
        acb->req.error = bs->drv->bdrv_co_readv(bs, acb->req.sector,
            acb->req.nb_sectors, acb->req.qiov);
    } else {
        acb->req.error = bs->drv->bdrv_co_writev(bs, acb->req.sector,
            acb->req.nb_sectors, acb->req.qiov);
    }

    acb->bh = qemu_bh_new(bdrv_co_rw_bh, acb);
    qemu_bh_schedule(acb->bh);
}

static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
                                               int64_t sector_num,
                                               QEMUIOVector *qiov,
                                               int nb_sectors,
                                               BlockDriverCompletionFunc *cb,
                                               void *opaque,
                                               bool is_write)
{
    Coroutine *co;
    BlockDriverAIOCBCoroutine *acb;

    acb = qemu_aio_get(&bdrv_em_co_aio_pool, bs, cb, opaque);
    acb->req.sector = sector_num;
    acb->req.nb_sectors = nb_sectors;
    acb->req.qiov = qiov;
    acb->is_write = is_write;

    co = qemu_coroutine_create(bdrv_co_rw);
    qemu_coroutine_enter(co, acb);

    return &acb->common;
}

static BlockDriverAIOCB *bdrv_co_aio_readv_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
        BlockDriverCompletionFunc *cb, void *opaque)
{
    return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque,
                                 false);
}

static BlockDriverAIOCB *bdrv_co_aio_writev_em(BlockDriverState *bs,
        int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
        BlockDriverCompletionFunc *cb, void *opaque)
{
    return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque,
                                 true);
}

2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783
static BlockDriverAIOCB *bdrv_aio_flush_em(BlockDriverState *bs,
        BlockDriverCompletionFunc *cb, void *opaque)
{
    BlockDriverAIOCBSync *acb;

    acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
    acb->is_write = 1; /* don't bounce in the completion hadler */
    acb->qiov = NULL;
    acb->bounce = NULL;
    acb->ret = 0;

    if (!acb->bh)
        acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);

    bdrv_flush(bs);
    qemu_bh_schedule(acb->bh);
    return &acb->common;
}

2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802
static BlockDriverAIOCB *bdrv_aio_noop_em(BlockDriverState *bs,
        BlockDriverCompletionFunc *cb, void *opaque)
{
    BlockDriverAIOCBSync *acb;

    acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
    acb->is_write = 1; /* don't bounce in the completion handler */
    acb->qiov = NULL;
    acb->bounce = NULL;
    acb->ret = 0;

    if (!acb->bh) {
        acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
    }

    qemu_bh_schedule(acb->bh);
    return &acb->common;
}

B
bellard 已提交
2803 2804
/**************************************************************/
/* sync block device emulation */
B
bellard 已提交
2805

B
bellard 已提交
2806 2807 2808
static void bdrv_rw_em_cb(void *opaque, int ret)
{
    *(int *)opaque = ret;
B
bellard 已提交
2809 2810
}

B
bellard 已提交
2811 2812
#define NOT_DONE 0x7fffffff

2813
static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
B
bellard 已提交
2814
                        uint8_t *buf, int nb_sectors)
P
pbrook 已提交
2815
{
2816 2817
    int async_ret;
    BlockDriverAIOCB *acb;
2818 2819
    struct iovec iov;
    QEMUIOVector qiov;
B
bellard 已提交
2820 2821

    async_ret = NOT_DONE;
2822
    iov.iov_base = (void *)buf;
2823
    iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2824 2825 2826
    qemu_iovec_init_external(&qiov, &iov, 1);
    acb = bdrv_aio_readv(bs, sector_num, &qiov, nb_sectors,
        bdrv_rw_em_cb, &async_ret);
2827 2828 2829 2830
    if (acb == NULL) {
        async_ret = -1;
        goto fail;
    }
2831

B
bellard 已提交
2832 2833 2834
    while (async_ret == NOT_DONE) {
        qemu_aio_wait();
    }
2835

2836 2837

fail:
B
bellard 已提交
2838
    return async_ret;
P
pbrook 已提交
2839 2840
}

B
bellard 已提交
2841 2842 2843
static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
                         const uint8_t *buf, int nb_sectors)
{
2844 2845
    int async_ret;
    BlockDriverAIOCB *acb;
2846 2847
    struct iovec iov;
    QEMUIOVector qiov;
B
bellard 已提交
2848 2849

    async_ret = NOT_DONE;
2850
    iov.iov_base = (void *)buf;
2851
    iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2852 2853 2854
    qemu_iovec_init_external(&qiov, &iov, 1);
    acb = bdrv_aio_writev(bs, sector_num, &qiov, nb_sectors,
        bdrv_rw_em_cb, &async_ret);
2855 2856 2857 2858
    if (acb == NULL) {
        async_ret = -1;
        goto fail;
    }
B
bellard 已提交
2859 2860 2861
    while (async_ret == NOT_DONE) {
        qemu_aio_wait();
    }
2862 2863

fail:
B
bellard 已提交
2864 2865
    return async_ret;
}
B
bellard 已提交
2866 2867 2868

void bdrv_init(void)
{
2869
    module_call_init(MODULE_INIT_BLOCK);
B
bellard 已提交
2870
}
2871

2872 2873 2874 2875 2876 2877
void bdrv_init_with_whitelist(void)
{
    use_bdrv_whitelist = 1;
    bdrv_init();
}

2878 2879
void *qemu_aio_get(AIOPool *pool, BlockDriverState *bs,
                   BlockDriverCompletionFunc *cb, void *opaque)
2880 2881 2882
{
    BlockDriverAIOCB *acb;

2883 2884 2885
    if (pool->free_aiocb) {
        acb = pool->free_aiocb;
        pool->free_aiocb = acb->next;
2886
    } else {
2887
        acb = g_malloc0(pool->aiocb_size);
2888
        acb->pool = pool;
2889 2890 2891 2892 2893 2894 2895 2896 2897
    }
    acb->bs = bs;
    acb->cb = cb;
    acb->opaque = opaque;
    return acb;
}

void qemu_aio_release(void *p)
{
2898 2899 2900 2901
    BlockDriverAIOCB *acb = (BlockDriverAIOCB *)p;
    AIOPool *pool = acb->pool;
    acb->next = pool->free_aiocb;
    pool->free_aiocb = acb;
2902
}
B
bellard 已提交
2903

2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959
/**************************************************************/
/* Coroutine block device emulation */

typedef struct CoroutineIOCompletion {
    Coroutine *coroutine;
    int ret;
} CoroutineIOCompletion;

static void bdrv_co_io_em_complete(void *opaque, int ret)
{
    CoroutineIOCompletion *co = opaque;

    co->ret = ret;
    qemu_coroutine_enter(co->coroutine, NULL);
}

static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
                                      int nb_sectors, QEMUIOVector *iov,
                                      bool is_write)
{
    CoroutineIOCompletion co = {
        .coroutine = qemu_coroutine_self(),
    };
    BlockDriverAIOCB *acb;

    if (is_write) {
        acb = bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
                              bdrv_co_io_em_complete, &co);
    } else {
        acb = bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
                             bdrv_co_io_em_complete, &co);
    }

    trace_bdrv_co_io(is_write, acb);
    if (!acb) {
        return -EIO;
    }
    qemu_coroutine_yield();

    return co.ret;
}

static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
                                         int64_t sector_num, int nb_sectors,
                                         QEMUIOVector *iov)
{
    return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
}

static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
                                         int64_t sector_num, int nb_sectors,
                                         QEMUIOVector *iov)
{
    return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
}

2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974
static int coroutine_fn bdrv_co_flush_em(BlockDriverState *bs)
{
    CoroutineIOCompletion co = {
        .coroutine = qemu_coroutine_self(),
    };
    BlockDriverAIOCB *acb;

    acb = bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
    if (!acb) {
        return -EIO;
    }
    qemu_coroutine_yield();
    return co.ret;
}

B
bellard 已提交
2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987
/**************************************************************/
/* removable device support */

/**
 * Return TRUE if the media is present
 */
int bdrv_is_inserted(BlockDriverState *bs)
{
    BlockDriver *drv = bs->drv;
    int ret;
    if (!drv)
        return 0;
    if (!drv->bdrv_is_inserted)
2988
        return !bs->tray_open;
B
bellard 已提交
2989 2990 2991 2992 2993 2994
    ret = drv->bdrv_is_inserted(bs);
    return ret;
}

/**
 * Return TRUE if the media changed since the last call to this
2995
 * function. It is currently only used for floppy disks
B
bellard 已提交
2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014
 */
int bdrv_media_changed(BlockDriverState *bs)
{
    BlockDriver *drv = bs->drv;
    int ret;

    if (!drv || !drv->bdrv_media_changed)
        ret = -ENOTSUP;
    else
        ret = drv->bdrv_media_changed(bs);
    if (ret == -ENOTSUP)
        ret = bs->media_changed;
    bs->media_changed = 0;
    return ret;
}

/**
 * If eject_flag is TRUE, eject the media. Otherwise, close the tray
 */
3015
int bdrv_eject(BlockDriverState *bs, int eject_flag)
B
bellard 已提交
3016 3017 3018
{
    BlockDriver *drv = bs->drv;

3019
    if (eject_flag && bs->locked) {
3020 3021 3022
        return -EBUSY;
    }

3023 3024
    if (drv && drv->bdrv_eject) {
        drv->bdrv_eject(bs, eject_flag);
B
bellard 已提交
3025
    }
3026 3027
    bs->tray_open = eject_flag;
    return 0;
B
bellard 已提交
3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042
}

int bdrv_is_locked(BlockDriverState *bs)
{
    return bs->locked;
}

/**
 * Lock or unlock the media (if it is locked, the user won't be able
 * to eject it manually).
 */
void bdrv_set_locked(BlockDriverState *bs, int locked)
{
    BlockDriver *drv = bs->drv;

S
Stefan Hajnoczi 已提交
3043 3044
    trace_bdrv_set_locked(bs, locked);

B
bellard 已提交
3045 3046 3047 3048 3049
    bs->locked = locked;
    if (drv && drv->bdrv_set_locked) {
        drv->bdrv_set_locked(bs, locked);
    }
}
3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060

/* needed for generic scsi interface */

int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
{
    BlockDriver *drv = bs->drv;

    if (drv && drv->bdrv_ioctl)
        return drv->bdrv_ioctl(bs, req, buf);
    return -ENOTSUP;
}
3061

3062 3063 3064
BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
        unsigned long int req, void *buf,
        BlockDriverCompletionFunc *cb, void *opaque)
3065
{
3066
    BlockDriver *drv = bs->drv;
3067

3068 3069 3070
    if (drv && drv->bdrv_aio_ioctl)
        return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
    return NULL;
3071
}
3072

3073 3074


3075 3076 3077 3078
void *qemu_blockalign(BlockDriverState *bs, size_t size)
{
    return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
}
3079 3080 3081 3082

void bdrv_set_dirty_tracking(BlockDriverState *bs, int enable)
{
    int64_t bitmap_size;
3083

3084
    bs->dirty_count = 0;
3085
    if (enable) {
3086 3087 3088 3089
        if (!bs->dirty_bitmap) {
            bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS) +
                    BDRV_SECTORS_PER_DIRTY_CHUNK * 8 - 1;
            bitmap_size /= BDRV_SECTORS_PER_DIRTY_CHUNK * 8;
3090

3091
            bs->dirty_bitmap = g_malloc0(bitmap_size);
3092
        }
3093
    } else {
3094
        if (bs->dirty_bitmap) {
3095
            g_free(bs->dirty_bitmap);
3096
            bs->dirty_bitmap = NULL;
3097
        }
3098 3099 3100 3101 3102
    }
}

int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
{
J
Jan Kiszka 已提交
3103
    int64_t chunk = sector / (int64_t)BDRV_SECTORS_PER_DIRTY_CHUNK;
3104

3105 3106
    if (bs->dirty_bitmap &&
        (sector << BDRV_SECTOR_BITS) < bdrv_getlength(bs)) {
3107 3108
        return !!(bs->dirty_bitmap[chunk / (sizeof(unsigned long) * 8)] &
            (1UL << (chunk % (sizeof(unsigned long) * 8))));
3109 3110 3111 3112 3113
    } else {
        return 0;
    }
}

3114 3115
void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
                      int nr_sectors)
3116 3117 3118
{
    set_dirty_bitmap(bs, cur_sector, nr_sectors, 0);
}
3119 3120 3121 3122 3123

int64_t bdrv_get_dirty_count(BlockDriverState *bs)
{
    return bs->dirty_count;
}
J
Jes Sorensen 已提交
3124

3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135
void bdrv_set_in_use(BlockDriverState *bs, int in_use)
{
    assert(bs->in_use != in_use);
    bs->in_use = in_use;
}

int bdrv_in_use(BlockDriverState *bs)
{
    return bs->in_use;
}

J
Jes Sorensen 已提交
3136 3137 3138 3139 3140
int bdrv_img_create(const char *filename, const char *fmt,
                    const char *base_filename, const char *base_fmt,
                    char *options, uint64_t img_size, int flags)
{
    QEMUOptionParameter *param = NULL, *create_options = NULL;
K
Kevin Wolf 已提交
3141
    QEMUOptionParameter *backing_fmt, *backing_file, *size;
J
Jes Sorensen 已提交
3142 3143
    BlockDriverState *bs = NULL;
    BlockDriver *drv, *proto_drv;
3144
    BlockDriver *backing_drv = NULL;
J
Jes Sorensen 已提交
3145 3146 3147 3148 3149 3150
    int ret = 0;

    /* Find driver and parse its options */
    drv = bdrv_find_format(fmt);
    if (!drv) {
        error_report("Unknown file format '%s'", fmt);
3151
        ret = -EINVAL;
J
Jes Sorensen 已提交
3152 3153 3154 3155 3156 3157
        goto out;
    }

    proto_drv = bdrv_find_protocol(filename);
    if (!proto_drv) {
        error_report("Unknown protocol '%s'", filename);
3158
        ret = -EINVAL;
J
Jes Sorensen 已提交
3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176
        goto out;
    }

    create_options = append_option_parameters(create_options,
                                              drv->create_options);
    create_options = append_option_parameters(create_options,
                                              proto_drv->create_options);

    /* Create parameter list with default values */
    param = parse_option_parameters("", create_options, param);

    set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);

    /* Parse -o options */
    if (options) {
        param = parse_option_parameters(options, create_options, param);
        if (param == NULL) {
            error_report("Invalid options for file format '%s'.", fmt);
3177
            ret = -EINVAL;
J
Jes Sorensen 已提交
3178 3179 3180 3181 3182 3183 3184 3185 3186
            goto out;
        }
    }

    if (base_filename) {
        if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
                                 base_filename)) {
            error_report("Backing file not supported for file format '%s'",
                         fmt);
3187
            ret = -EINVAL;
J
Jes Sorensen 已提交
3188 3189 3190 3191 3192 3193 3194 3195
            goto out;
        }
    }

    if (base_fmt) {
        if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
            error_report("Backing file format not supported for file "
                         "format '%s'", fmt);
3196
            ret = -EINVAL;
J
Jes Sorensen 已提交
3197 3198 3199 3200
            goto out;
        }
    }

3201 3202 3203 3204 3205
    backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
    if (backing_file && backing_file->value.s) {
        if (!strcmp(filename, backing_file->value.s)) {
            error_report("Error: Trying to create an image with the "
                         "same filename as the backing file");
3206
            ret = -EINVAL;
3207 3208 3209 3210
            goto out;
        }
    }

J
Jes Sorensen 已提交
3211 3212
    backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
    if (backing_fmt && backing_fmt->value.s) {
3213 3214
        backing_drv = bdrv_find_format(backing_fmt->value.s);
        if (!backing_drv) {
J
Jes Sorensen 已提交
3215 3216
            error_report("Unknown backing file format '%s'",
                         backing_fmt->value.s);
3217
            ret = -EINVAL;
J
Jes Sorensen 已提交
3218 3219 3220 3221 3222 3223
            goto out;
        }
    }

    // The size for the image must always be specified, with one exception:
    // If we are using a backing file, we can obtain the size from there
K
Kevin Wolf 已提交
3224 3225
    size = get_option_parameter(param, BLOCK_OPT_SIZE);
    if (size && size->value.n == -1) {
J
Jes Sorensen 已提交
3226 3227 3228 3229 3230 3231
        if (backing_file && backing_file->value.s) {
            uint64_t size;
            char buf[32];

            bs = bdrv_new("");

3232
            ret = bdrv_open(bs, backing_file->value.s, flags, backing_drv);
J
Jes Sorensen 已提交
3233
            if (ret < 0) {
3234
                error_report("Could not open '%s'", backing_file->value.s);
J
Jes Sorensen 已提交
3235 3236 3237 3238 3239 3240 3241 3242 3243
                goto out;
            }
            bdrv_get_geometry(bs, &size);
            size *= 512;

            snprintf(buf, sizeof(buf), "%" PRId64, size);
            set_option_parameter(param, BLOCK_OPT_SIZE, buf);
        } else {
            error_report("Image creation needs a size parameter");
3244
            ret = -EINVAL;
J
Jes Sorensen 已提交
3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274
            goto out;
        }
    }

    printf("Formatting '%s', fmt=%s ", filename, fmt);
    print_option_parameters(param);
    puts("");

    ret = bdrv_create(drv, filename, param);

    if (ret < 0) {
        if (ret == -ENOTSUP) {
            error_report("Formatting or formatting option not supported for "
                         "file format '%s'", fmt);
        } else if (ret == -EFBIG) {
            error_report("The image size is too large for file format '%s'",
                         fmt);
        } else {
            error_report("%s: error while creating %s: %s", filename, fmt,
                         strerror(-ret));
        }
    }

out:
    free_option_parameters(create_options);
    free_option_parameters(param);

    if (bs) {
        bdrv_delete(bs);
    }
3275 3276

    return ret;
J
Jes Sorensen 已提交
3277
}