blockdev.c

   1 /*
   2  * QEMU host block devices
   3  *
   4  * Copyright (c) 2003-2008 Fabrice Bellard
   5  *
   6  * This work is licensed under the terms of the GNU GPL, version 2 or
   7  * later.  See the COPYING file in the top-level directory.
   8  */
   9
  10 #include "block.h"
  11 #include "blockdev.h"
  12 #include "monitor.h"
  13 #include "qerror.h"
  14 #include "qemu-option.h"
  15 #include "qemu-config.h"
  16 #include "qemu-objects.h"
  17 #include "sysemu.h"
  18 #include "block_int.h"
  19 #include "qmp-commands.h"
  20 #include "trace.h"
  21
  22 static QTAILQ_HEAD(drivelist, DriveInfo) drives = QTAILQ_HEAD_INITIALIZER(drives);
  23
  24 static const char *const if_name[IF_COUNT] = {
  25     [IF_NONE] = "none",
  26     [IF_IDE] = "ide",
  27     [IF_SCSI] = "scsi",
  28     [IF_FLOPPY] = "floppy",
  29     [IF_PFLASH] = "pflash",
  30     [IF_MTD] = "mtd",
  31     [IF_SD] = "sd",
  32     [IF_VIRTIO] = "virtio",
  33     [IF_XEN] = "xen",
  34 };
  35
  36 static const int if_max_devs[IF_COUNT] = {
  37     /*
  38      * Do not change these numbers!  They govern how drive option
  39      * index maps to unit and bus.  That mapping is ABI.
  40      *
  41      * All controllers used to imlement if=T drives need to support
  42      * if_max_devs[T] units, for any T with if_max_devs[T] != 0.
  43      * Otherwise, some index values map to "impossible" bus, unit
  44      * values.
  45      *
  46      * For instance, if you change [IF_SCSI] to 255, -drive
  47      * if=scsi,index=12 no longer means bus=1,unit=5, but
  48      * bus=0,unit=12.  With an lsi53c895a controller (7 units max),
  49      * the drive can't be set up.  Regression.
  50      */
  51     [IF_IDE] = 2,
  52     [IF_SCSI] = 7,
  53 };
  54
  55 /*
  56  * We automatically delete the drive when a device using it gets
  57  * unplugged.  Questionable feature, but we can't just drop it.
  58  * Device models call blockdev_mark_auto_del() to schedule the
  59  * automatic deletion, and generic qdev code calls blockdev_auto_del()
  60  * when deletion is actually safe.
  61  */
  62 void blockdev_mark_auto_del(BlockDriverState *bs)
  63 {
  64     DriveInfo *dinfo = drive_get_by_blockdev(bs);
  65
  66     if (dinfo) {
  67         dinfo->auto_del = 1;
  68     }
  69 }
  70
  71 void blockdev_auto_del(BlockDriverState *bs)
  72 {
  73     DriveInfo *dinfo = drive_get_by_blockdev(bs);
  74
  75     if (dinfo && dinfo->auto_del) {
  76         drive_put_ref(dinfo);
  77     }
  78 }
  79
  80 static int drive_index_to_bus_id(BlockInterfaceType type, int index)
  81 {
  82     int max_devs = if_max_devs[type];
  83     return max_devs ? index / max_devs : 0;
  84 }
  85
  86 static int drive_index_to_unit_id(BlockInterfaceType type, int index)
  87 {
  88     int max_devs = if_max_devs[type];
  89     return max_devs ? index % max_devs : index;
  90 }
  91
  92 QemuOpts *drive_def(const char *optstr)
  93 {
  94     return qemu_opts_parse(qemu_find_opts("drive"), optstr, 0);
  95 }
  96
  97 QemuOpts *drive_add(BlockInterfaceType type, int index, const char *file,
  98                     const char *optstr)
  99 {
 100     QemuOpts *opts;
 101     char buf[32];
 102
 103     opts = drive_def(optstr);
 104     if (!opts) {
 105         return NULL;
 106     }
 107     if (type != IF_DEFAULT) {
 108         qemu_opt_set(opts, "if", if_name[type]);
 109     }
 110     if (index >= 0) {
 111         snprintf(buf, sizeof(buf), "%d", index);
 112         qemu_opt_set(opts, "index", buf);
 113     }
 114     if (file)
 115         qemu_opt_set(opts, "file", file);
 116     return opts;
 117 }
 118
 119 DriveInfo *drive_get(BlockInterfaceType type, int bus, int unit)
 120 {
 121     DriveInfo *dinfo;
 122
 123     /* seek interface, bus and unit */
 124
 125     QTAILQ_FOREACH(dinfo, &drives, next) {
 126         if (dinfo->type == type &&
 127             dinfo->bus == bus &&
 128             dinfo->unit == unit)
 129             return dinfo;
 130     }
 131
 132     return NULL;
 133 }
 134
 135 DriveInfo *drive_get_by_index(BlockInterfaceType type, int index)
 136 {
 137     return drive_get(type,
 138                      drive_index_to_bus_id(type, index),
 139                      drive_index_to_unit_id(type, index));
 140 }
 141
 142 int drive_get_max_bus(BlockInterfaceType type)
 143 {
 144     int max_bus;
 145     DriveInfo *dinfo;
 146
 147     max_bus = -1;
 148     QTAILQ_FOREACH(dinfo, &drives, next) {
 149         if(dinfo->type == type &&
 150            dinfo->bus > max_bus)
 151             max_bus = dinfo->bus;
 152     }
 153     return max_bus;
 154 }
 155
 156 /* Get a block device.  This should only be used for single-drive devices
 157    (e.g. SD/Floppy/MTD).  Multi-disk devices (scsi/ide) should use the
 158    appropriate bus.  */
 159 DriveInfo *drive_get_next(BlockInterfaceType type)
 160 {
 161     static int next_block_unit[IF_COUNT];
 162
 163     return drive_get(type, 0, next_block_unit[type]++);
 164 }
 165
 166 DriveInfo *drive_get_by_blockdev(BlockDriverState *bs)
 167 {
 168     DriveInfo *dinfo;
 169
 170     QTAILQ_FOREACH(dinfo, &drives, next) {
 171         if (dinfo->bdrv == bs) {
 172             return dinfo;
 173         }
 174     }
 175     return NULL;
 176 }
 177
 178 static void bdrv_format_print(void *opaque, const char *name)
 179 {
 180     error_printf(" %s", name);
 181 }
 182
 183 static void drive_uninit(DriveInfo *dinfo)
 184 {
 185     qemu_opts_del(dinfo->opts);
 186     bdrv_delete(dinfo->bdrv);
 187     g_free(dinfo->id);
 188     QTAILQ_REMOVE(&drives, dinfo, next);
 189     g_free(dinfo);
 190 }
 191
 192 void drive_put_ref(DriveInfo *dinfo)
 193 {
 194     assert(dinfo->refcount);
 195     if (--dinfo->refcount == 0) {
 196         drive_uninit(dinfo);
 197     }
 198 }
 199
 200 void drive_get_ref(DriveInfo *dinfo)
 201 {
 202     dinfo->refcount++;
 203 }
 204
 205 typedef struct {
 206     QEMUBH *bh;
 207     DriveInfo *dinfo;
 208 } DrivePutRefBH;
 209
 210 static void drive_put_ref_bh(void *opaque)
 211 {
 212     DrivePutRefBH *s = opaque;
 213
 214     drive_put_ref(s->dinfo);
 215     qemu_bh_delete(s->bh);
 216     g_free(s);
 217 }
 218
 219 /*
 220  * Release a drive reference in a BH
 221  *
 222  * It is not possible to use drive_put_ref() from a callback function when the
 223  * callers still need the drive.  In such cases we schedule a BH to release the
 224  * reference.
 225  */
 226 static void drive_put_ref_bh_schedule(DriveInfo *dinfo)
 227 {
 228     DrivePutRefBH *s;
 229
 230     s = g_new(DrivePutRefBH, 1);
 231     s->bh = qemu_bh_new(drive_put_ref_bh, s);
 232     s->dinfo = dinfo;
 233     qemu_bh_schedule(s->bh);
 234 }
 235
 236 static int parse_block_error_action(const char *buf, int is_read)
 237 {
 238     if (!strcmp(buf, "ignore")) {
 239         return BLOCK_ERR_IGNORE;
 240     } else if (!is_read && !strcmp(buf, "enospc")) {
 241         return BLOCK_ERR_STOP_ENOSPC;
 242     } else if (!strcmp(buf, "stop")) {
 243         return BLOCK_ERR_STOP_ANY;
 244     } else if (!strcmp(buf, "report")) {
 245         return BLOCK_ERR_REPORT;
 246     } else {
 247         error_report("'%s' invalid %s error action",
 248                      buf, is_read ? "read" : "write");
 249         return -1;
 250     }
 251 }
 252
 253 static bool do_check_io_limits(BlockIOLimit *io_limits)
 254 {
 255     bool bps_flag;
 256     bool iops_flag;
 257
 258     assert(io_limits);
 259
 260     bps_flag  = (io_limits->bps[BLOCK_IO_LIMIT_TOTAL] != 0)
 261                  && ((io_limits->bps[BLOCK_IO_LIMIT_READ] != 0)
 262                  || (io_limits->bps[BLOCK_IO_LIMIT_WRITE] != 0));
 263     iops_flag = (io_limits->iops[BLOCK_IO_LIMIT_TOTAL] != 0)
 264                  && ((io_limits->iops[BLOCK_IO_LIMIT_READ] != 0)
 265                  || (io_limits->iops[BLOCK_IO_LIMIT_WRITE] != 0));
 266     if (bps_flag || iops_flag) {
 267         return false;
 268     }
 269
 270     return true;
 271 }
 272
 273 DriveInfo *drive_init(QemuOpts *opts, int default_to_scsi)
 274 {
 275     const char *buf;
 276     const char *file = NULL;
 277     char devname[128];
 278     const char *serial;
 279     const char *mediastr = "";
 280     BlockInterfaceType type;
 281     enum { MEDIA_DISK, MEDIA_CDROM } media;
 282     int bus_id, unit_id;
 283     int cyls, heads, secs, translation;
 284     BlockDriver *drv = NULL;
 285     int max_devs;
 286     int index;
 287     int ro = 0;
 288     int bdrv_flags = 0;
 289     int on_read_error, on_write_error;
 290     const char *devaddr;
 291     DriveInfo *dinfo;
 292     BlockIOLimit io_limits;
 293     int snapshot = 0;
 294     bool copy_on_read;
 295     int ret;
 296
 297     translation = BIOS_ATA_TRANSLATION_AUTO;
 298     media = MEDIA_DISK;
 299
 300     /* extract parameters */
 301     bus_id  = qemu_opt_get_number(opts, "bus", 0);
 302     unit_id = qemu_opt_get_number(opts, "unit", -1);
 303     index   = qemu_opt_get_number(opts, "index", -1);
 304
 305     cyls  = qemu_opt_get_number(opts, "cyls", 0);
 306     heads = qemu_opt_get_number(opts, "heads", 0);
 307     secs  = qemu_opt_get_number(opts, "secs", 0);
 308
 309     snapshot = qemu_opt_get_bool(opts, "snapshot", 0);
 310     ro = qemu_opt_get_bool(opts, "readonly", 0);
 311     copy_on_read = qemu_opt_get_bool(opts, "copy-on-read", false);
 312
 313     file = qemu_opt_get(opts, "file");
 314     serial = qemu_opt_get(opts, "serial");
 315
 316     if ((buf = qemu_opt_get(opts, "if")) != NULL) {
 317         pstrcpy(devname, sizeof(devname), buf);
 318         for (type = 0; type < IF_COUNT && strcmp(buf, if_name[type]); type++)
 319             ;
 320         if (type == IF_COUNT) {
 321             error_report("unsupported bus type '%s'", buf);
 322             return NULL;
 323         }
 324     } else {
 325         type = default_to_scsi ? IF_SCSI : IF_IDE;
 326         pstrcpy(devname, sizeof(devname), if_name[type]);
 327     }
 328
 329     max_devs = if_max_devs[type];
 330
 331     if (cyls || heads || secs) {
 332         if (cyls < 1 || (type == IF_IDE && cyls > 16383)) {
 333             error_report("invalid physical cyls number");
 334             return NULL;
 335         }
 336         if (heads < 1 || (type == IF_IDE && heads > 16)) {
 337             error_report("invalid physical heads number");
 338             return NULL;
 339         }
 340         if (secs < 1 || (type == IF_IDE && secs > 63)) {
 341             error_report("invalid physical secs number");
 342             return NULL;
 343         }
 344     }
 345
 346     if ((buf = qemu_opt_get(opts, "trans")) != NULL) {
 347         if (!cyls) {
 348             error_report("'%s' trans must be used with cyls, heads and secs",
 349                          buf);
 350             return NULL;
 351         }
 352         if (!strcmp(buf, "none"))
 353             translation = BIOS_ATA_TRANSLATION_NONE;
 354         else if (!strcmp(buf, "lba"))
 355             translation = BIOS_ATA_TRANSLATION_LBA;
 356         else if (!strcmp(buf, "auto"))
 357             translation = BIOS_ATA_TRANSLATION_AUTO;
 358         else {
 359             error_report("'%s' invalid translation type", buf);
 360             return NULL;
 361         }
 362     }
 363
 364     if ((buf = qemu_opt_get(opts, "media")) != NULL) {
 365         if (!strcmp(buf, "disk")) {
 366             media = MEDIA_DISK;
 367         } else if (!strcmp(buf, "cdrom")) {
 368             if (cyls || secs || heads) {
 369                 error_report("CHS can't be set with media=%s", buf);
 370                 return NULL;
 371             }
 372             media = MEDIA_CDROM;
 373         } else {
 374             error_report("'%s' invalid media", buf);
 375             return NULL;
 376         }
 377     }
 378
 379     if ((buf = qemu_opt_get(opts, "cache")) != NULL) {
 380         if (bdrv_parse_cache_flags(buf, &bdrv_flags) != 0) {
 381             error_report("invalid cache option");
 382             return NULL;
 383         }
 384     }
 385
 386 #ifdef CONFIG_LINUX_AIO
 387     if ((buf = qemu_opt_get(opts, "aio")) != NULL) {
 388         if (!strcmp(buf, "native")) {
 389             bdrv_flags |= BDRV_O_NATIVE_AIO;
 390         } else if (!strcmp(buf, "threads")) {
 391             /* this is the default */
 392         } else {
 393            error_report("invalid aio option");
 394            return NULL;
 395         }
 396     }
 397 #endif
 398
 399     if ((buf = qemu_opt_get(opts, "format")) != NULL) {
 400        if (strcmp(buf, "?") == 0) {
 401            error_printf("Supported formats:");
 402            bdrv_iterate_format(bdrv_format_print, NULL);
 403            error_printf("\n");
 404            return NULL;
 405         }
 406         drv = bdrv_find_whitelisted_format(buf);
 407         if (!drv) {
 408             error_report("'%s' invalid format", buf);
 409             return NULL;
 410         }
 411     }
 412
 413     /* disk I/O throttling */
 414     io_limits.bps[BLOCK_IO_LIMIT_TOTAL]  =
 415                            qemu_opt_get_number(opts, "bps", 0);
 416     io_limits.bps[BLOCK_IO_LIMIT_READ]   =
 417                            qemu_opt_get_number(opts, "bps_rd", 0);
 418     io_limits.bps[BLOCK_IO_LIMIT_WRITE]  =
 419                            qemu_opt_get_number(opts, "bps_wr", 0);
 420     io_limits.iops[BLOCK_IO_LIMIT_TOTAL] =
 421                            qemu_opt_get_number(opts, "iops", 0);
 422     io_limits.iops[BLOCK_IO_LIMIT_READ]  =
 423                            qemu_opt_get_number(opts, "iops_rd", 0);
 424     io_limits.iops[BLOCK_IO_LIMIT_WRITE] =
 425                            qemu_opt_get_number(opts, "iops_wr", 0);
 426
 427     if (!do_check_io_limits(&io_limits)) {
 428         error_report("bps(iops) and bps_rd/bps_wr(iops_rd/iops_wr) "
 429                      "cannot be used at the same time");
 430         return NULL;
 431     }
 432
 433     on_write_error = BLOCK_ERR_STOP_ENOSPC;
 434     if ((buf = qemu_opt_get(opts, "werror")) != NULL) {
 435         if (type != IF_IDE && type != IF_SCSI && type != IF_VIRTIO && type != IF_NONE) {
 436             error_report("werror is not supported by this bus type");
 437             return NULL;
 438         }
 439
 440         on_write_error = parse_block_error_action(buf, 0);
 441         if (on_write_error < 0) {
 442             return NULL;
 443         }
 444     }
 445
 446     on_read_error = BLOCK_ERR_REPORT;
 447     if ((buf = qemu_opt_get(opts, "rerror")) != NULL) {
 448         if (type != IF_IDE && type != IF_VIRTIO && type != IF_SCSI && type != IF_NONE) {
 449             error_report("rerror is not supported by this bus type");
 450             return NULL;
 451         }
 452
 453         on_read_error = parse_block_error_action(buf, 1);
 454         if (on_read_error < 0) {
 455             return NULL;
 456         }
 457     }
 458
 459     if ((devaddr = qemu_opt_get(opts, "addr")) != NULL) {
 460         if (type != IF_VIRTIO) {
 461             error_report("addr is not supported by this bus type");
 462             return NULL;
 463         }
 464     }
 465
 466     /* compute bus and unit according index */
 467
 468     if (index != -1) {
 469         if (bus_id != 0 || unit_id != -1) {
 470             error_report("index cannot be used with bus and unit");
 471             return NULL;
 472         }
 473         bus_id = drive_index_to_bus_id(type, index);
 474         unit_id = drive_index_to_unit_id(type, index);
 475     }
 476
 477     /* if user doesn't specify a unit_id,
 478      * try to find the first free
 479      */
 480
 481     if (unit_id == -1) {
 482        unit_id = 0;
 483        while (drive_get(type, bus_id, unit_id) != NULL) {
 484            unit_id++;
 485            if (max_devs && unit_id >= max_devs) {
 486                unit_id -= max_devs;
 487                bus_id++;
 488            }
 489        }
 490     }
 491
 492     /* check unit id */
 493
 494     if (max_devs && unit_id >= max_devs) {
 495         error_report("unit %d too big (max is %d)",
 496                      unit_id, max_devs - 1);
 497         return NULL;
 498     }
 499
 500     /*
 501      * catch multiple definitions
 502      */
 503
 504     if (drive_get(type, bus_id, unit_id) != NULL) {
 505         error_report("drive with bus=%d, unit=%d (index=%d) exists",
 506                      bus_id, unit_id, index);
 507         return NULL;
 508     }
 509
 510     /* init */
 511
 512     dinfo = g_malloc0(sizeof(*dinfo));
 513     if ((buf = qemu_opts_id(opts)) != NULL) {
 514         dinfo->id = g_strdup(buf);
 515     } else {
 516         /* no id supplied -> create one */
 517         dinfo->id = g_malloc0(32);
 518         if (type == IF_IDE || type == IF_SCSI)
 519             mediastr = (media == MEDIA_CDROM) ? "-cd" : "-hd";
 520         if (max_devs)
 521             snprintf(dinfo->id, 32, "%s%i%s%i",
 522                      devname, bus_id, mediastr, unit_id);
 523         else
 524             snprintf(dinfo->id, 32, "%s%s%i",
 525                      devname, mediastr, unit_id);
 526     }
 527     dinfo->bdrv = bdrv_new(dinfo->id);
 528     dinfo->devaddr = devaddr;
 529     dinfo->type = type;
 530     dinfo->bus = bus_id;
 531     dinfo->unit = unit_id;
 532     dinfo->opts = opts;
 533     dinfo->refcount = 1;
 534     if (serial)
 535         strncpy(dinfo->serial, serial, sizeof(dinfo->serial) - 1);
 536     QTAILQ_INSERT_TAIL(&drives, dinfo, next);
 537
 538     bdrv_set_on_error(dinfo->bdrv, on_read_error, on_write_error);
 539
 540     /* disk I/O throttling */
 541     bdrv_set_io_limits(dinfo->bdrv, &io_limits);
 542
 543     switch(type) {
 544     case IF_IDE:
 545     case IF_SCSI:
 546     case IF_XEN:
 547     case IF_NONE:
 548         switch(media) {
 549         case MEDIA_DISK:
 550             if (cyls != 0) {
 551                 bdrv_set_geometry_hint(dinfo->bdrv, cyls, heads, secs);
 552                 bdrv_set_translation_hint(dinfo->bdrv, translation);
 553             }
 554             break;
 555         case MEDIA_CDROM:
 556             dinfo->media_cd = 1;
 557             break;
 558         }
 559         break;
 560     case IF_SD:
 561     case IF_FLOPPY:
 562     case IF_PFLASH:
 563     case IF_MTD:
 564         break;
 565     case IF_VIRTIO:
 566         /* add virtio block device */
 567         opts = qemu_opts_create(qemu_find_opts("device"), NULL, 0);
 568         qemu_opt_set(opts, "driver", "virtio-blk");
 569         qemu_opt_set(opts, "drive", dinfo->id);
 570         if (devaddr)
 571             qemu_opt_set(opts, "addr", devaddr);
 572         break;
 573     default:
 574         abort();
 575     }
 576     if (!file || !*file) {
 577         return dinfo;
 578     }
 579     if (snapshot) {
 580         /* always use cache=unsafe with snapshot */
 581         bdrv_flags &= ~BDRV_O_CACHE_MASK;
 582         bdrv_flags |= (BDRV_O_SNAPSHOT|BDRV_O_CACHE_WB|BDRV_O_NO_FLUSH);
 583     }
 584
 585     if (copy_on_read) {
 586         bdrv_flags |= BDRV_O_COPY_ON_READ;
 587     }
 588
 589     if (media == MEDIA_CDROM) {
 590         /* CDROM is fine for any interface, don't check.  */
 591         ro = 1;
 592     } else if (ro == 1) {
 593         if (type != IF_SCSI && type != IF_VIRTIO && type != IF_FLOPPY && type != IF_NONE) {
 594             error_report("readonly not supported by this bus type");
 595             goto err;
 596         }
 597     }
 598
 599     bdrv_flags |= ro ? 0 : BDRV_O_RDWR;
 600
 601     ret = bdrv_open(dinfo->bdrv, file, bdrv_flags, drv);
 602     if (ret < 0) {
 603         error_report("could not open disk image %s: %s",
 604                      file, strerror(-ret));
 605         goto err;
 606     }
 607
 608     if (bdrv_key_required(dinfo->bdrv))
 609         autostart = 0;
 610     return dinfo;
 611
 612 err:
 613     bdrv_delete(dinfo->bdrv);
 614     g_free(dinfo->id);
 615     QTAILQ_REMOVE(&drives, dinfo, next);
 616     g_free(dinfo);
 617     return NULL;
 618 }
 619
 620 void do_commit(Monitor *mon, const QDict *qdict)
 621 {
 622     const char *device = qdict_get_str(qdict, "device");
 623     BlockDriverState *bs;
 624
 625     if (!strcmp(device, "all")) {
 626         bdrv_commit_all();
 627     } else {
 628         int ret;
 629
 630         bs = bdrv_find(device);
 631         if (!bs) {
 632             qerror_report(QERR_DEVICE_NOT_FOUND, device);
 633             return;
 634         }
 635         ret = bdrv_commit(bs);
 636         if (ret == -EBUSY) {
 637             qerror_report(QERR_DEVICE_IN_USE, device);
 638             return;
 639         }
 640     }
 641 }
 642
 643 void qmp_blockdev_snapshot_sync(const char *device, const char *snapshot_file,
 644                                 bool has_format, const char *format,
 645                                 Error **errp)
 646 {
 647     BlockDriverState *bs;
 648     BlockDriver *drv, *old_drv, *proto_drv;
 649     int ret = 0;
 650     int flags;
 651     char old_filename[1024];
 652
 653     bs = bdrv_find(device);
 654     if (!bs) {
 655         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 656         return;
 657     }
 658     if (bdrv_in_use(bs)) {
 659         error_set(errp, QERR_DEVICE_IN_USE, device);
 660         return;
 661     }
 662
 663     pstrcpy(old_filename, sizeof(old_filename), bs->filename);
 664
 665     old_drv = bs->drv;
 666     flags = bs->open_flags;
 667
 668     if (!has_format) {
 669         format = "qcow2";
 670     }
 671
 672     drv = bdrv_find_format(format);
 673     if (!drv) {
 674         error_set(errp, QERR_INVALID_BLOCK_FORMAT, format);
 675         return;
 676     }
 677
 678     proto_drv = bdrv_find_protocol(snapshot_file);
 679     if (!proto_drv) {
 680         error_set(errp, QERR_INVALID_BLOCK_FORMAT, format);
 681         return;
 682     }
 683
 684     ret = bdrv_img_create(snapshot_file, format, bs->filename,
 685                           bs->drv->format_name, NULL, -1, flags);
 686     if (ret) {
 687         error_set(errp, QERR_UNDEFINED_ERROR);
 688         return;
 689     }
 690
 691     bdrv_drain_all();
 692     bdrv_flush(bs);
 693
 694     bdrv_close(bs);
 695     ret = bdrv_open(bs, snapshot_file, flags, drv);
 696     /*
 697      * If reopening the image file we just created fails, fall back
 698      * and try to re-open the original image. If that fails too, we
 699      * are in serious trouble.
 700      */
 701     if (ret != 0) {
 702         ret = bdrv_open(bs, old_filename, flags, old_drv);
 703         if (ret != 0) {
 704             error_set(errp, QERR_OPEN_FILE_FAILED, old_filename);
 705         } else {
 706             error_set(errp, QERR_OPEN_FILE_FAILED, snapshot_file);
 707         }
 708     }
 709 }
 710
 711 static void eject_device(BlockDriverState *bs, int force, Error **errp)
 712 {
 713     if (bdrv_in_use(bs)) {
 714         error_set(errp, QERR_DEVICE_IN_USE, bdrv_get_device_name(bs));
 715         return;
 716     }
 717     if (!bdrv_dev_has_removable_media(bs)) {
 718         error_set(errp, QERR_DEVICE_NOT_REMOVABLE, bdrv_get_device_name(bs));
 719         return;
 720     }
 721
 722     if (bdrv_dev_is_medium_locked(bs) && !bdrv_dev_is_tray_open(bs)) {
 723         bdrv_dev_eject_request(bs, force);
 724         if (!force) {
 725             error_set(errp, QERR_DEVICE_LOCKED, bdrv_get_device_name(bs));
 726             return;
 727         }
 728     }
 729
 730     bdrv_close(bs);
 731 }
 732
 733 void qmp_eject(const char *device, bool has_force, bool force, Error **errp)
 734 {
 735     BlockDriverState *bs;
 736
 737     bs = bdrv_find(device);
 738     if (!bs) {
 739         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 740         return;
 741     }
 742
 743     eject_device(bs, force, errp);
 744 }
 745
 746 void qmp_block_passwd(const char *device, const char *password, Error **errp)
 747 {
 748     BlockDriverState *bs;
 749     int err;
 750
 751     bs = bdrv_find(device);
 752     if (!bs) {
 753         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 754         return;
 755     }
 756
 757     err = bdrv_set_key(bs, password);
 758     if (err == -EINVAL) {
 759         error_set(errp, QERR_DEVICE_NOT_ENCRYPTED, bdrv_get_device_name(bs));
 760         return;
 761     } else if (err < 0) {
 762         error_set(errp, QERR_INVALID_PASSWORD);
 763         return;
 764     }
 765 }
 766
 767 static void qmp_bdrv_open_encrypted(BlockDriverState *bs, const char *filename,
 768                                     int bdrv_flags, BlockDriver *drv,
 769                                     const char *password, Error **errp)
 770 {
 771     if (bdrv_open(bs, filename, bdrv_flags, drv) < 0) {
 772         error_set(errp, QERR_OPEN_FILE_FAILED, filename);
 773         return;
 774     }
 775
 776     if (bdrv_key_required(bs)) {
 777         if (password) {
 778             if (bdrv_set_key(bs, password) < 0) {
 779                 error_set(errp, QERR_INVALID_PASSWORD);
 780             }
 781         } else {
 782             error_set(errp, QERR_DEVICE_ENCRYPTED, bdrv_get_device_name(bs),
 783                       bdrv_get_encrypted_filename(bs));
 784         }
 785     } else if (password) {
 786         error_set(errp, QERR_DEVICE_NOT_ENCRYPTED, bdrv_get_device_name(bs));
 787     }
 788 }
 789
 790 void qmp_change_blockdev(const char *device, const char *filename,
 791                          bool has_format, const char *format, Error **errp)
 792 {
 793     BlockDriverState *bs;
 794     BlockDriver *drv = NULL;
 795     int bdrv_flags;
 796     Error *err = NULL;
 797
 798     bs = bdrv_find(device);
 799     if (!bs) {
 800         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 801         return;
 802     }
 803
 804     if (format) {
 805         drv = bdrv_find_whitelisted_format(format);
 806         if (!drv) {
 807             error_set(errp, QERR_INVALID_BLOCK_FORMAT, format);
 808             return;
 809         }
 810     }
 811
 812     eject_device(bs, 0, &err);
 813     if (error_is_set(&err)) {
 814         error_propagate(errp, err);
 815         return;
 816     }
 817
 818     bdrv_flags = bdrv_is_read_only(bs) ? 0 : BDRV_O_RDWR;
 819     bdrv_flags |= bdrv_is_snapshot(bs) ? BDRV_O_SNAPSHOT : 0;
 820
 821     qmp_bdrv_open_encrypted(bs, filename, bdrv_flags, drv, NULL, errp);
 822 }
 823
 824 /* throttling disk I/O limits */
 825 void qmp_block_set_io_throttle(const char *device, int64_t bps, int64_t bps_rd,
 826                                int64_t bps_wr, int64_t iops, int64_t iops_rd,
 827                                int64_t iops_wr, Error **errp)
 828 {
 829     BlockIOLimit io_limits;
 830     BlockDriverState *bs;
 831
 832     bs = bdrv_find(device);
 833     if (!bs) {
 834         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 835         return;
 836     }
 837
 838     io_limits.bps[BLOCK_IO_LIMIT_TOTAL] = bps;
 839     io_limits.bps[BLOCK_IO_LIMIT_READ]  = bps_rd;
 840     io_limits.bps[BLOCK_IO_LIMIT_WRITE] = bps_wr;
 841     io_limits.iops[BLOCK_IO_LIMIT_TOTAL]= iops;
 842     io_limits.iops[BLOCK_IO_LIMIT_READ] = iops_rd;
 843     io_limits.iops[BLOCK_IO_LIMIT_WRITE]= iops_wr;
 844
 845     if (!do_check_io_limits(&io_limits)) {
 846         error_set(errp, QERR_INVALID_PARAMETER_COMBINATION);
 847         return;
 848     }
 849
 850     bs->io_limits = io_limits;
 851     bs->slice_time = BLOCK_IO_SLICE_TIME;
 852
 853     if (!bs->io_limits_enabled && bdrv_io_limits_enabled(bs)) {
 854         bdrv_io_limits_enable(bs);
 855     } else if (bs->io_limits_enabled && !bdrv_io_limits_enabled(bs)) {
 856         bdrv_io_limits_disable(bs);
 857     } else {
 858         if (bs->block_timer) {
 859             qemu_mod_timer(bs->block_timer, qemu_get_clock_ns(vm_clock));
 860         }
 861     }
 862 }
 863
 864 int do_drive_del(Monitor *mon, const QDict *qdict, QObject **ret_data)
 865 {
 866     const char *id = qdict_get_str(qdict, "id");
 867     BlockDriverState *bs;
 868
 869     bs = bdrv_find(id);
 870     if (!bs) {
 871         qerror_report(QERR_DEVICE_NOT_FOUND, id);
 872         return -1;
 873     }
 874     if (bdrv_in_use(bs)) {
 875         qerror_report(QERR_DEVICE_IN_USE, id);
 876         return -1;
 877     }
 878
 879     /* quiesce block driver; prevent further io */
 880     bdrv_drain_all();
 881     bdrv_flush(bs);
 882     bdrv_close(bs);
 883
 884     /* if we have a device attached to this BlockDriverState
 885      * then we need to make the drive anonymous until the device
 886      * can be removed.  If this is a drive with no device backing
 887      * then we can just get rid of the block driver state right here.
 888      */
 889     if (bdrv_get_attached_dev(bs)) {
 890         bdrv_make_anon(bs);
 891     } else {
 892         drive_uninit(drive_get_by_blockdev(bs));
 893     }
 894
 895     return 0;
 896 }
 897
 898 void qmp_block_resize(const char *device, int64_t size, Error **errp)
 899 {
 900     BlockDriverState *bs;
 901
 902     bs = bdrv_find(device);
 903     if (!bs) {
 904         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 905         return;
 906     }
 907
 908     if (size < 0) {
 909         error_set(errp, QERR_INVALID_PARAMETER_VALUE, "size", "a >0 size");
 910         return;
 911     }
 912
 913     switch (bdrv_truncate(bs, size)) {
 914     case 0:
 915         break;
 916     case -ENOMEDIUM:
 917         error_set(errp, QERR_DEVICE_HAS_NO_MEDIUM, device);
 918         break;
 919     case -ENOTSUP:
 920         error_set(errp, QERR_UNSUPPORTED);
 921         break;
 922     case -EACCES:
 923         error_set(errp, QERR_DEVICE_IS_READ_ONLY, device);
 924         break;
 925     case -EBUSY:
 926         error_set(errp, QERR_DEVICE_IN_USE, device);
 927         break;
 928     default:
 929         error_set(errp, QERR_UNDEFINED_ERROR);
 930         break;
 931     }
 932 }
 933
 934 static QObject *qobject_from_block_job(BlockJob *job)
 935 {
 936     return qobject_from_jsonf("{ 'type': %s,"
 937                               "'device': %s,"
 938                               "'len': %" PRId64 ","
 939                               "'offset': %" PRId64 ","
 940                               "'speed': %" PRId64 " }",
 941                               job->job_type->job_type,
 942                               bdrv_get_device_name(job->bs),
 943                               job->len,
 944                               job->offset,
 945                               job->speed);
 946 }
 947
 948 static void block_stream_cb(void *opaque, int ret)
 949 {
 950     BlockDriverState *bs = opaque;
 951     QObject *obj;
 952
 953     trace_block_stream_cb(bs, bs->job, ret);
 954
 955     assert(bs->job);
 956     obj = qobject_from_block_job(bs->job);
 957     if (ret < 0) {
 958         QDict *dict = qobject_to_qdict(obj);
 959         qdict_put(dict, "error", qstring_from_str(strerror(-ret)));
 960     }
 961
 962     if (block_job_is_cancelled(bs->job)) {
 963         monitor_protocol_event(QEVENT_BLOCK_JOB_CANCELLED, obj);
 964     } else {
 965         monitor_protocol_event(QEVENT_BLOCK_JOB_COMPLETED, obj);
 966     }
 967     qobject_decref(obj);
 968
 969     drive_put_ref_bh_schedule(drive_get_by_blockdev(bs));
 970 }
 971
 972 void qmp_block_stream(const char *device, bool has_base,
 973                       const char *base, Error **errp)
 974 {
 975     BlockDriverState *bs;
 976     BlockDriverState *base_bs = NULL;
 977     int ret;
 978
 979     bs = bdrv_find(device);
 980     if (!bs) {
 981         error_set(errp, QERR_DEVICE_NOT_FOUND, device);
 982         return;
 983     }
 984
 985     if (base) {
 986         base_bs = bdrv_find_backing_image(bs, base);
 987         if (base_bs == NULL) {
 988             error_set(errp, QERR_BASE_NOT_FOUND, base);
 989             return;
 990         }
 991     }
 992
 993     ret = stream_start(bs, base_bs, base, block_stream_cb, bs);
 994     if (ret < 0) {
 995         switch (ret) {
 996         case -EBUSY:
 997             error_set(errp, QERR_DEVICE_IN_USE, device);
 998             return;
 999         default:
1000             error_set(errp, QERR_NOT_SUPPORTED);
1001             return;
1002         }
1003     }
1004
1005     /* Grab a reference so hotplug does not delete the BlockDriverState from
1006      * underneath us.
1007      */
1008     drive_get_ref(drive_get_by_blockdev(bs));
1009
1010     trace_qmp_block_stream(bs, bs->job);
1011 }
1012
1013 static BlockJob *find_block_job(const char *device)
1014 {
1015     BlockDriverState *bs;
1016
1017     bs = bdrv_find(device);
1018     if (!bs || !bs->job) {
1019         return NULL;
1020     }
1021     return bs->job;
1022 }
1023
1024 void qmp_block_job_set_speed(const char *device, int64_t value, Error **errp)
1025 {
1026     BlockJob *job = find_block_job(device);
1027
1028     if (!job) {
1029         error_set(errp, QERR_DEVICE_NOT_ACTIVE, device);
1030         return;
1031     }
1032
1033     if (block_job_set_speed(job, value) < 0) {
1034         error_set(errp, QERR_NOT_SUPPORTED);
1035     }
1036 }
1037
1038 void qmp_block_job_cancel(const char *device, Error **errp)
1039 {
1040     BlockJob *job = find_block_job(device);
1041
1042     if (!job) {
1043         error_set(errp, QERR_DEVICE_NOT_ACTIVE, device);
1044         return;
1045     }
1046
1047     trace_qmp_block_job_cancel(job);
1048     block_job_cancel(job);
1049 }
1050
1051 static void do_qmp_query_block_jobs_one(void *opaque, BlockDriverState *bs)
1052 {
1053     BlockJobInfoList **prev = opaque;
1054     BlockJob *job = bs->job;
1055
1056     if (job) {
1057         BlockJobInfoList *elem;
1058         BlockJobInfo *info = g_new(BlockJobInfo, 1);
1059         *info = (BlockJobInfo){
1060             .type   = g_strdup(job->job_type->job_type),
1061             .device = g_strdup(bdrv_get_device_name(bs)),
1062             .len    = job->len,
1063             .offset = job->offset,
1064             .speed  = job->speed,
1065         };
1066
1067         elem = g_new0(BlockJobInfoList, 1);
1068         elem->value = info;
1069
1070         (*prev)->next = elem;
1071         *prev = elem;
1072     }
1073 }
1074
1075 BlockJobInfoList *qmp_query_block_jobs(Error **errp)
1076 {
1077     /* Dummy is a fake list element for holding the head pointer */
1078     BlockJobInfoList dummy = {};
1079     BlockJobInfoList *prev = &dummy;
1080     bdrv_iterate(do_qmp_query_block_jobs_one, &prev);
1081     return dummy.next;
1082 }