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