block: disable I/O limits at the beginning of bdrv_close()
[sdk/emulator/qemu.git] / block.c
1 /*
2  * QEMU System Emulator block driver
3  *
4  * Copyright (c) 2003 Fabrice Bellard
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in
14  * all copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22  * THE SOFTWARE.
23  */
24 #include "config-host.h"
25 #include "qemu-common.h"
26 #include "trace.h"
27 #include "block/block_int.h"
28 #include "block/blockjob.h"
29 #include "qemu/error-report.h"
30 #include "qemu/module.h"
31 #include "qapi/qmp/qerror.h"
32 #include "qapi/qmp/qjson.h"
33 #include "sysemu/block-backend.h"
34 #include "sysemu/sysemu.h"
35 #include "qemu/notify.h"
36 #include "block/coroutine.h"
37 #include "block/qapi.h"
38 #include "qmp-commands.h"
39 #include "qemu/timer.h"
40 #include "qapi-event.h"
41 #include "block/throttle-groups.h"
42
43 #ifdef CONFIG_BSD
44 #include <sys/types.h>
45 #include <sys/stat.h>
46 #include <sys/ioctl.h>
47 #include <sys/queue.h>
48 #ifndef __DragonFly__
49 #include <sys/disk.h>
50 #endif
51 #endif
52
53 #ifdef _WIN32
54 #include <windows.h>
55 #endif
56
57 /**
58  * A BdrvDirtyBitmap can be in three possible states:
59  * (1) successor is NULL and disabled is false: full r/w mode
60  * (2) successor is NULL and disabled is true: read only mode ("disabled")
61  * (3) successor is set: frozen mode.
62  *     A frozen bitmap cannot be renamed, deleted, anonymized, cleared, set,
63  *     or enabled. A frozen bitmap can only abdicate() or reclaim().
64  */
65 struct BdrvDirtyBitmap {
66     HBitmap *bitmap;            /* Dirty sector bitmap implementation */
67     BdrvDirtyBitmap *successor; /* Anonymous child; implies frozen status */
68     char *name;                 /* Optional non-empty unique ID */
69     int64_t size;               /* Size of the bitmap (Number of sectors) */
70     bool disabled;              /* Bitmap is read-only */
71     QLIST_ENTRY(BdrvDirtyBitmap) list;
72 };
73
74 #define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
75
76 static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
77     QTAILQ_HEAD_INITIALIZER(bdrv_states);
78
79 static QTAILQ_HEAD(, BlockDriverState) graph_bdrv_states =
80     QTAILQ_HEAD_INITIALIZER(graph_bdrv_states);
81
82 static QLIST_HEAD(, BlockDriver) bdrv_drivers =
83     QLIST_HEAD_INITIALIZER(bdrv_drivers);
84
85 static int bdrv_open_inherit(BlockDriverState **pbs, const char *filename,
86                              const char *reference, QDict *options, int flags,
87                              BlockDriverState *parent,
88                              const BdrvChildRole *child_role, Error **errp);
89
90 static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs);
91 /* If non-zero, use only whitelisted block drivers */
92 static int use_bdrv_whitelist;
93
94 #ifdef _WIN32
95 static int is_windows_drive_prefix(const char *filename)
96 {
97     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
98              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
99             filename[1] == ':');
100 }
101
102 int is_windows_drive(const char *filename)
103 {
104     if (is_windows_drive_prefix(filename) &&
105         filename[2] == '\0')
106         return 1;
107     if (strstart(filename, "\\\\.\\", NULL) ||
108         strstart(filename, "//./", NULL))
109         return 1;
110     return 0;
111 }
112 #endif
113
114 size_t bdrv_opt_mem_align(BlockDriverState *bs)
115 {
116     if (!bs || !bs->drv) {
117         /* page size or 4k (hdd sector size) should be on the safe side */
118         return MAX(4096, getpagesize());
119     }
120
121     return bs->bl.opt_mem_alignment;
122 }
123
124 size_t bdrv_min_mem_align(BlockDriverState *bs)
125 {
126     if (!bs || !bs->drv) {
127         /* page size or 4k (hdd sector size) should be on the safe side */
128         return MAX(4096, getpagesize());
129     }
130
131     return bs->bl.min_mem_alignment;
132 }
133
134 /* check if the path starts with "<protocol>:" */
135 int path_has_protocol(const char *path)
136 {
137     const char *p;
138
139 #ifdef _WIN32
140     if (is_windows_drive(path) ||
141         is_windows_drive_prefix(path)) {
142         return 0;
143     }
144     p = path + strcspn(path, ":/\\");
145 #else
146     p = path + strcspn(path, ":/");
147 #endif
148
149     return *p == ':';
150 }
151
152 int path_is_absolute(const char *path)
153 {
154 #ifdef _WIN32
155     /* specific case for names like: "\\.\d:" */
156     if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
157         return 1;
158     }
159     return (*path == '/' || *path == '\\');
160 #else
161     return (*path == '/');
162 #endif
163 }
164
165 /* if filename is absolute, just copy it to dest. Otherwise, build a
166    path to it by considering it is relative to base_path. URL are
167    supported. */
168 void path_combine(char *dest, int dest_size,
169                   const char *base_path,
170                   const char *filename)
171 {
172     const char *p, *p1;
173     int len;
174
175     if (dest_size <= 0)
176         return;
177     if (path_is_absolute(filename)) {
178         pstrcpy(dest, dest_size, filename);
179     } else {
180         p = strchr(base_path, ':');
181         if (p)
182             p++;
183         else
184             p = base_path;
185         p1 = strrchr(base_path, '/');
186 #ifdef _WIN32
187         {
188             const char *p2;
189             p2 = strrchr(base_path, '\\');
190             if (!p1 || p2 > p1)
191                 p1 = p2;
192         }
193 #endif
194         if (p1)
195             p1++;
196         else
197             p1 = base_path;
198         if (p1 > p)
199             p = p1;
200         len = p - base_path;
201         if (len > dest_size - 1)
202             len = dest_size - 1;
203         memcpy(dest, base_path, len);
204         dest[len] = '\0';
205         pstrcat(dest, dest_size, filename);
206     }
207 }
208
209 void bdrv_get_full_backing_filename_from_filename(const char *backed,
210                                                   const char *backing,
211                                                   char *dest, size_t sz,
212                                                   Error **errp)
213 {
214     if (backing[0] == '\0' || path_has_protocol(backing) ||
215         path_is_absolute(backing))
216     {
217         pstrcpy(dest, sz, backing);
218     } else if (backed[0] == '\0' || strstart(backed, "json:", NULL)) {
219         error_setg(errp, "Cannot use relative backing file names for '%s'",
220                    backed);
221     } else {
222         path_combine(dest, sz, backed, backing);
223     }
224 }
225
226 void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz,
227                                     Error **errp)
228 {
229     char *backed = bs->exact_filename[0] ? bs->exact_filename : bs->filename;
230
231     bdrv_get_full_backing_filename_from_filename(backed, bs->backing_file,
232                                                  dest, sz, errp);
233 }
234
235 void bdrv_register(BlockDriver *bdrv)
236 {
237     bdrv_setup_io_funcs(bdrv);
238
239     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
240 }
241
242 BlockDriverState *bdrv_new_root(void)
243 {
244     BlockDriverState *bs = bdrv_new();
245
246     QTAILQ_INSERT_TAIL(&bdrv_states, bs, device_list);
247     return bs;
248 }
249
250 BlockDriverState *bdrv_new(void)
251 {
252     BlockDriverState *bs;
253     int i;
254
255     bs = g_new0(BlockDriverState, 1);
256     QLIST_INIT(&bs->dirty_bitmaps);
257     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
258         QLIST_INIT(&bs->op_blockers[i]);
259     }
260     bdrv_iostatus_disable(bs);
261     notifier_list_init(&bs->close_notifiers);
262     notifier_with_return_list_init(&bs->before_write_notifiers);
263     qemu_co_queue_init(&bs->throttled_reqs[0]);
264     qemu_co_queue_init(&bs->throttled_reqs[1]);
265     bs->refcnt = 1;
266     bs->aio_context = qemu_get_aio_context();
267
268     return bs;
269 }
270
271 void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
272 {
273     notifier_list_add(&bs->close_notifiers, notify);
274 }
275
276 BlockDriver *bdrv_find_format(const char *format_name)
277 {
278     BlockDriver *drv1;
279     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
280         if (!strcmp(drv1->format_name, format_name)) {
281             return drv1;
282         }
283     }
284     return NULL;
285 }
286
287 static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
288 {
289     static const char *whitelist_rw[] = {
290         CONFIG_BDRV_RW_WHITELIST
291     };
292     static const char *whitelist_ro[] = {
293         CONFIG_BDRV_RO_WHITELIST
294     };
295     const char **p;
296
297     if (!whitelist_rw[0] && !whitelist_ro[0]) {
298         return 1;               /* no whitelist, anything goes */
299     }
300
301     for (p = whitelist_rw; *p; p++) {
302         if (!strcmp(drv->format_name, *p)) {
303             return 1;
304         }
305     }
306     if (read_only) {
307         for (p = whitelist_ro; *p; p++) {
308             if (!strcmp(drv->format_name, *p)) {
309                 return 1;
310             }
311         }
312     }
313     return 0;
314 }
315
316 typedef struct CreateCo {
317     BlockDriver *drv;
318     char *filename;
319     QemuOpts *opts;
320     int ret;
321     Error *err;
322 } CreateCo;
323
324 static void coroutine_fn bdrv_create_co_entry(void *opaque)
325 {
326     Error *local_err = NULL;
327     int ret;
328
329     CreateCo *cco = opaque;
330     assert(cco->drv);
331
332     ret = cco->drv->bdrv_create(cco->filename, cco->opts, &local_err);
333     if (local_err) {
334         error_propagate(&cco->err, local_err);
335     }
336     cco->ret = ret;
337 }
338
339 int bdrv_create(BlockDriver *drv, const char* filename,
340                 QemuOpts *opts, Error **errp)
341 {
342     int ret;
343
344     Coroutine *co;
345     CreateCo cco = {
346         .drv = drv,
347         .filename = g_strdup(filename),
348         .opts = opts,
349         .ret = NOT_DONE,
350         .err = NULL,
351     };
352
353     if (!drv->bdrv_create) {
354         error_setg(errp, "Driver '%s' does not support image creation", drv->format_name);
355         ret = -ENOTSUP;
356         goto out;
357     }
358
359     if (qemu_in_coroutine()) {
360         /* Fast-path if already in coroutine context */
361         bdrv_create_co_entry(&cco);
362     } else {
363         co = qemu_coroutine_create(bdrv_create_co_entry);
364         qemu_coroutine_enter(co, &cco);
365         while (cco.ret == NOT_DONE) {
366             aio_poll(qemu_get_aio_context(), true);
367         }
368     }
369
370     ret = cco.ret;
371     if (ret < 0) {
372         if (cco.err) {
373             error_propagate(errp, cco.err);
374         } else {
375             error_setg_errno(errp, -ret, "Could not create image");
376         }
377     }
378
379 out:
380     g_free(cco.filename);
381     return ret;
382 }
383
384 int bdrv_create_file(const char *filename, QemuOpts *opts, Error **errp)
385 {
386     BlockDriver *drv;
387     Error *local_err = NULL;
388     int ret;
389
390     drv = bdrv_find_protocol(filename, true, errp);
391     if (drv == NULL) {
392         return -ENOENT;
393     }
394
395     ret = bdrv_create(drv, filename, opts, &local_err);
396     if (local_err) {
397         error_propagate(errp, local_err);
398     }
399     return ret;
400 }
401
402 /**
403  * Try to get @bs's logical and physical block size.
404  * On success, store them in @bsz struct and return 0.
405  * On failure return -errno.
406  * @bs must not be empty.
407  */
408 int bdrv_probe_blocksizes(BlockDriverState *bs, BlockSizes *bsz)
409 {
410     BlockDriver *drv = bs->drv;
411
412     if (drv && drv->bdrv_probe_blocksizes) {
413         return drv->bdrv_probe_blocksizes(bs, bsz);
414     }
415
416     return -ENOTSUP;
417 }
418
419 /**
420  * Try to get @bs's geometry (cyls, heads, sectors).
421  * On success, store them in @geo struct and return 0.
422  * On failure return -errno.
423  * @bs must not be empty.
424  */
425 int bdrv_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
426 {
427     BlockDriver *drv = bs->drv;
428
429     if (drv && drv->bdrv_probe_geometry) {
430         return drv->bdrv_probe_geometry(bs, geo);
431     }
432
433     return -ENOTSUP;
434 }
435
436 /*
437  * Create a uniquely-named empty temporary file.
438  * Return 0 upon success, otherwise a negative errno value.
439  */
440 int get_tmp_filename(char *filename, int size)
441 {
442 #ifdef _WIN32
443     char temp_dir[MAX_PATH];
444     /* GetTempFileName requires that its output buffer (4th param)
445        have length MAX_PATH or greater.  */
446     assert(size >= MAX_PATH);
447     return (GetTempPath(MAX_PATH, temp_dir)
448             && GetTempFileName(temp_dir, "qem", 0, filename)
449             ? 0 : -GetLastError());
450 #else
451     int fd;
452     const char *tmpdir;
453     tmpdir = getenv("TMPDIR");
454     if (!tmpdir) {
455         tmpdir = "/var/tmp";
456     }
457     if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
458         return -EOVERFLOW;
459     }
460     fd = mkstemp(filename);
461     if (fd < 0) {
462         return -errno;
463     }
464     if (close(fd) != 0) {
465         unlink(filename);
466         return -errno;
467     }
468     return 0;
469 #endif
470 }
471
472 /*
473  * Detect host devices. By convention, /dev/cdrom[N] is always
474  * recognized as a host CDROM.
475  */
476 static BlockDriver *find_hdev_driver(const char *filename)
477 {
478     int score_max = 0, score;
479     BlockDriver *drv = NULL, *d;
480
481     QLIST_FOREACH(d, &bdrv_drivers, list) {
482         if (d->bdrv_probe_device) {
483             score = d->bdrv_probe_device(filename);
484             if (score > score_max) {
485                 score_max = score;
486                 drv = d;
487             }
488         }
489     }
490
491     return drv;
492 }
493
494 BlockDriver *bdrv_find_protocol(const char *filename,
495                                 bool allow_protocol_prefix,
496                                 Error **errp)
497 {
498     BlockDriver *drv1;
499     char protocol[128];
500     int len;
501     const char *p;
502
503     /* TODO Drivers without bdrv_file_open must be specified explicitly */
504
505     /*
506      * XXX(hch): we really should not let host device detection
507      * override an explicit protocol specification, but moving this
508      * later breaks access to device names with colons in them.
509      * Thanks to the brain-dead persistent naming schemes on udev-
510      * based Linux systems those actually are quite common.
511      */
512     drv1 = find_hdev_driver(filename);
513     if (drv1) {
514         return drv1;
515     }
516
517     if (!path_has_protocol(filename) || !allow_protocol_prefix) {
518         return &bdrv_file;
519     }
520
521     p = strchr(filename, ':');
522     assert(p != NULL);
523     len = p - filename;
524     if (len > sizeof(protocol) - 1)
525         len = sizeof(protocol) - 1;
526     memcpy(protocol, filename, len);
527     protocol[len] = '\0';
528     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
529         if (drv1->protocol_name &&
530             !strcmp(drv1->protocol_name, protocol)) {
531             return drv1;
532         }
533     }
534
535     error_setg(errp, "Unknown protocol '%s'", protocol);
536     return NULL;
537 }
538
539 /*
540  * Guess image format by probing its contents.
541  * This is not a good idea when your image is raw (CVE-2008-2004), but
542  * we do it anyway for backward compatibility.
543  *
544  * @buf         contains the image's first @buf_size bytes.
545  * @buf_size    is the buffer size in bytes (generally BLOCK_PROBE_BUF_SIZE,
546  *              but can be smaller if the image file is smaller)
547  * @filename    is its filename.
548  *
549  * For all block drivers, call the bdrv_probe() method to get its
550  * probing score.
551  * Return the first block driver with the highest probing score.
552  */
553 BlockDriver *bdrv_probe_all(const uint8_t *buf, int buf_size,
554                             const char *filename)
555 {
556     int score_max = 0, score;
557     BlockDriver *drv = NULL, *d;
558
559     QLIST_FOREACH(d, &bdrv_drivers, list) {
560         if (d->bdrv_probe) {
561             score = d->bdrv_probe(buf, buf_size, filename);
562             if (score > score_max) {
563                 score_max = score;
564                 drv = d;
565             }
566         }
567     }
568
569     return drv;
570 }
571
572 static int find_image_format(BlockDriverState *bs, const char *filename,
573                              BlockDriver **pdrv, Error **errp)
574 {
575     BlockDriver *drv;
576     uint8_t buf[BLOCK_PROBE_BUF_SIZE];
577     int ret = 0;
578
579     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
580     if (bdrv_is_sg(bs) || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
581         *pdrv = &bdrv_raw;
582         return ret;
583     }
584
585     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
586     if (ret < 0) {
587         error_setg_errno(errp, -ret, "Could not read image for determining its "
588                          "format");
589         *pdrv = NULL;
590         return ret;
591     }
592
593     drv = bdrv_probe_all(buf, ret, filename);
594     if (!drv) {
595         error_setg(errp, "Could not determine image format: No compatible "
596                    "driver found");
597         ret = -ENOENT;
598     }
599     *pdrv = drv;
600     return ret;
601 }
602
603 /**
604  * Set the current 'total_sectors' value
605  * Return 0 on success, -errno on error.
606  */
607 static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
608 {
609     BlockDriver *drv = bs->drv;
610
611     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
612     if (bdrv_is_sg(bs))
613         return 0;
614
615     /* query actual device if possible, otherwise just trust the hint */
616     if (drv->bdrv_getlength) {
617         int64_t length = drv->bdrv_getlength(bs);
618         if (length < 0) {
619             return length;
620         }
621         hint = DIV_ROUND_UP(length, BDRV_SECTOR_SIZE);
622     }
623
624     bs->total_sectors = hint;
625     return 0;
626 }
627
628 /**
629  * Set open flags for a given discard mode
630  *
631  * Return 0 on success, -1 if the discard mode was invalid.
632  */
633 int bdrv_parse_discard_flags(const char *mode, int *flags)
634 {
635     *flags &= ~BDRV_O_UNMAP;
636
637     if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
638         /* do nothing */
639     } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
640         *flags |= BDRV_O_UNMAP;
641     } else {
642         return -1;
643     }
644
645     return 0;
646 }
647
648 /**
649  * Set open flags for a given cache mode
650  *
651  * Return 0 on success, -1 if the cache mode was invalid.
652  */
653 int bdrv_parse_cache_flags(const char *mode, int *flags)
654 {
655     *flags &= ~BDRV_O_CACHE_MASK;
656
657     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
658         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
659     } else if (!strcmp(mode, "directsync")) {
660         *flags |= BDRV_O_NOCACHE;
661     } else if (!strcmp(mode, "writeback")) {
662         *flags |= BDRV_O_CACHE_WB;
663     } else if (!strcmp(mode, "unsafe")) {
664         *flags |= BDRV_O_CACHE_WB;
665         *flags |= BDRV_O_NO_FLUSH;
666     } else if (!strcmp(mode, "writethrough")) {
667         /* this is the default */
668     } else {
669         return -1;
670     }
671
672     return 0;
673 }
674
675 /*
676  * Returns the flags that a temporary snapshot should get, based on the
677  * originally requested flags (the originally requested image will have flags
678  * like a backing file)
679  */
680 static int bdrv_temp_snapshot_flags(int flags)
681 {
682     return (flags & ~BDRV_O_SNAPSHOT) | BDRV_O_TEMPORARY;
683 }
684
685 /*
686  * Returns the flags that bs->file should get if a protocol driver is expected,
687  * based on the given flags for the parent BDS
688  */
689 static int bdrv_inherited_flags(int flags)
690 {
691     /* Enable protocol handling, disable format probing for bs->file */
692     flags |= BDRV_O_PROTOCOL;
693
694     /* Our block drivers take care to send flushes and respect unmap policy,
695      * so we can enable both unconditionally on lower layers. */
696     flags |= BDRV_O_CACHE_WB | BDRV_O_UNMAP;
697
698     /* Clear flags that only apply to the top layer */
699     flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING | BDRV_O_COPY_ON_READ);
700
701     return flags;
702 }
703
704 const BdrvChildRole child_file = {
705     .inherit_flags = bdrv_inherited_flags,
706 };
707
708 /*
709  * Returns the flags that bs->file should get if the use of formats (and not
710  * only protocols) is permitted for it, based on the given flags for the parent
711  * BDS
712  */
713 static int bdrv_inherited_fmt_flags(int parent_flags)
714 {
715     int flags = child_file.inherit_flags(parent_flags);
716     return flags & ~BDRV_O_PROTOCOL;
717 }
718
719 const BdrvChildRole child_format = {
720     .inherit_flags = bdrv_inherited_fmt_flags,
721 };
722
723 /*
724  * Returns the flags that bs->backing_hd should get, based on the given flags
725  * for the parent BDS
726  */
727 static int bdrv_backing_flags(int flags)
728 {
729     /* backing files always opened read-only */
730     flags &= ~(BDRV_O_RDWR | BDRV_O_COPY_ON_READ);
731
732     /* snapshot=on is handled on the top layer */
733     flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_TEMPORARY);
734
735     return flags;
736 }
737
738 static const BdrvChildRole child_backing = {
739     .inherit_flags = bdrv_backing_flags,
740 };
741
742 static int bdrv_open_flags(BlockDriverState *bs, int flags)
743 {
744     int open_flags = flags | BDRV_O_CACHE_WB;
745
746     /*
747      * Clear flags that are internal to the block layer before opening the
748      * image.
749      */
750     open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING | BDRV_O_PROTOCOL);
751
752     /*
753      * Snapshots should be writable.
754      */
755     if (flags & BDRV_O_TEMPORARY) {
756         open_flags |= BDRV_O_RDWR;
757     }
758
759     return open_flags;
760 }
761
762 static void bdrv_assign_node_name(BlockDriverState *bs,
763                                   const char *node_name,
764                                   Error **errp)
765 {
766     if (!node_name) {
767         return;
768     }
769
770     /* Check for empty string or invalid characters */
771     if (!id_wellformed(node_name)) {
772         error_setg(errp, "Invalid node name");
773         return;
774     }
775
776     /* takes care of avoiding namespaces collisions */
777     if (blk_by_name(node_name)) {
778         error_setg(errp, "node-name=%s is conflicting with a device id",
779                    node_name);
780         return;
781     }
782
783     /* takes care of avoiding duplicates node names */
784     if (bdrv_find_node(node_name)) {
785         error_setg(errp, "Duplicate node name");
786         return;
787     }
788
789     /* copy node name into the bs and insert it into the graph list */
790     pstrcpy(bs->node_name, sizeof(bs->node_name), node_name);
791     QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs, node_list);
792 }
793
794 static QemuOptsList bdrv_runtime_opts = {
795     .name = "bdrv_common",
796     .head = QTAILQ_HEAD_INITIALIZER(bdrv_runtime_opts.head),
797     .desc = {
798         {
799             .name = "node-name",
800             .type = QEMU_OPT_STRING,
801             .help = "Node name of the block device node",
802         },
803         { /* end of list */ }
804     },
805 };
806
807 /*
808  * Common part for opening disk images and files
809  *
810  * Removes all processed options from *options.
811  */
812 static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
813     QDict *options, int flags, BlockDriver *drv, Error **errp)
814 {
815     int ret, open_flags;
816     const char *filename;
817     const char *node_name = NULL;
818     QemuOpts *opts;
819     Error *local_err = NULL;
820
821     assert(drv != NULL);
822     assert(bs->file == NULL);
823     assert(options != NULL && bs->options != options);
824
825     if (file != NULL) {
826         filename = file->filename;
827     } else {
828         filename = qdict_get_try_str(options, "filename");
829     }
830
831     if (drv->bdrv_needs_filename && !filename) {
832         error_setg(errp, "The '%s' block driver requires a file name",
833                    drv->format_name);
834         return -EINVAL;
835     }
836
837     trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
838
839     opts = qemu_opts_create(&bdrv_runtime_opts, NULL, 0, &error_abort);
840     qemu_opts_absorb_qdict(opts, options, &local_err);
841     if (local_err) {
842         error_propagate(errp, local_err);
843         ret = -EINVAL;
844         goto fail_opts;
845     }
846
847     node_name = qemu_opt_get(opts, "node-name");
848     bdrv_assign_node_name(bs, node_name, &local_err);
849     if (local_err) {
850         error_propagate(errp, local_err);
851         ret = -EINVAL;
852         goto fail_opts;
853     }
854
855     bs->guest_block_size = 512;
856     bs->request_alignment = 512;
857     bs->zero_beyond_eof = true;
858     open_flags = bdrv_open_flags(bs, flags);
859     bs->read_only = !(open_flags & BDRV_O_RDWR);
860
861     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv, bs->read_only)) {
862         error_setg(errp,
863                    !bs->read_only && bdrv_is_whitelisted(drv, true)
864                         ? "Driver '%s' can only be used for read-only devices"
865                         : "Driver '%s' is not whitelisted",
866                    drv->format_name);
867         ret = -ENOTSUP;
868         goto fail_opts;
869     }
870
871     assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
872     if (flags & BDRV_O_COPY_ON_READ) {
873         if (!bs->read_only) {
874             bdrv_enable_copy_on_read(bs);
875         } else {
876             error_setg(errp, "Can't use copy-on-read on read-only device");
877             ret = -EINVAL;
878             goto fail_opts;
879         }
880     }
881
882     if (filename != NULL) {
883         pstrcpy(bs->filename, sizeof(bs->filename), filename);
884     } else {
885         bs->filename[0] = '\0';
886     }
887     pstrcpy(bs->exact_filename, sizeof(bs->exact_filename), bs->filename);
888
889     bs->drv = drv;
890     bs->opaque = g_malloc0(drv->instance_size);
891
892     bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
893
894     /* Open the image, either directly or using a protocol */
895     if (drv->bdrv_file_open) {
896         assert(file == NULL);
897         assert(!drv->bdrv_needs_filename || filename != NULL);
898         ret = drv->bdrv_file_open(bs, options, open_flags, &local_err);
899     } else {
900         if (file == NULL) {
901             error_setg(errp, "Can't use '%s' as a block driver for the "
902                        "protocol level", drv->format_name);
903             ret = -EINVAL;
904             goto free_and_fail;
905         }
906         bs->file = file;
907         ret = drv->bdrv_open(bs, options, open_flags, &local_err);
908     }
909
910     if (ret < 0) {
911         if (local_err) {
912             error_propagate(errp, local_err);
913         } else if (bs->filename[0]) {
914             error_setg_errno(errp, -ret, "Could not open '%s'", bs->filename);
915         } else {
916             error_setg_errno(errp, -ret, "Could not open image");
917         }
918         goto free_and_fail;
919     }
920
921     if (bs->encrypted) {
922         error_report("Encrypted images are deprecated");
923         error_printf("Support for them will be removed in a future release.\n"
924                      "You can use 'qemu-img convert' to convert your image"
925                      " to an unencrypted one.\n");
926     }
927
928     ret = refresh_total_sectors(bs, bs->total_sectors);
929     if (ret < 0) {
930         error_setg_errno(errp, -ret, "Could not refresh total sector count");
931         goto free_and_fail;
932     }
933
934     bdrv_refresh_limits(bs, &local_err);
935     if (local_err) {
936         error_propagate(errp, local_err);
937         ret = -EINVAL;
938         goto free_and_fail;
939     }
940
941     assert(bdrv_opt_mem_align(bs) != 0);
942     assert(bdrv_min_mem_align(bs) != 0);
943     assert((bs->request_alignment != 0) || bdrv_is_sg(bs));
944
945     qemu_opts_del(opts);
946     return 0;
947
948 free_and_fail:
949     bs->file = NULL;
950     g_free(bs->opaque);
951     bs->opaque = NULL;
952     bs->drv = NULL;
953 fail_opts:
954     qemu_opts_del(opts);
955     return ret;
956 }
957
958 static QDict *parse_json_filename(const char *filename, Error **errp)
959 {
960     QObject *options_obj;
961     QDict *options;
962     int ret;
963
964     ret = strstart(filename, "json:", &filename);
965     assert(ret);
966
967     options_obj = qobject_from_json(filename);
968     if (!options_obj) {
969         error_setg(errp, "Could not parse the JSON options");
970         return NULL;
971     }
972
973     if (qobject_type(options_obj) != QTYPE_QDICT) {
974         qobject_decref(options_obj);
975         error_setg(errp, "Invalid JSON object given");
976         return NULL;
977     }
978
979     options = qobject_to_qdict(options_obj);
980     qdict_flatten(options);
981
982     return options;
983 }
984
985 /*
986  * Fills in default options for opening images and converts the legacy
987  * filename/flags pair to option QDict entries.
988  * The BDRV_O_PROTOCOL flag in *flags will be set or cleared accordingly if a
989  * block driver has been specified explicitly.
990  */
991 static int bdrv_fill_options(QDict **options, const char **pfilename,
992                              int *flags, Error **errp)
993 {
994     const char *filename = *pfilename;
995     const char *drvname;
996     bool protocol = *flags & BDRV_O_PROTOCOL;
997     bool parse_filename = false;
998     BlockDriver *drv = NULL;
999     Error *local_err = NULL;
1000
1001     /* Parse json: pseudo-protocol */
1002     if (filename && g_str_has_prefix(filename, "json:")) {
1003         QDict *json_options = parse_json_filename(filename, &local_err);
1004         if (local_err) {
1005             error_propagate(errp, local_err);
1006             return -EINVAL;
1007         }
1008
1009         /* Options given in the filename have lower priority than options
1010          * specified directly */
1011         qdict_join(*options, json_options, false);
1012         QDECREF(json_options);
1013         *pfilename = filename = NULL;
1014     }
1015
1016     drvname = qdict_get_try_str(*options, "driver");
1017     if (drvname) {
1018         drv = bdrv_find_format(drvname);
1019         if (!drv) {
1020             error_setg(errp, "Unknown driver '%s'", drvname);
1021             return -ENOENT;
1022         }
1023         /* If the user has explicitly specified the driver, this choice should
1024          * override the BDRV_O_PROTOCOL flag */
1025         protocol = drv->bdrv_file_open;
1026     }
1027
1028     if (protocol) {
1029         *flags |= BDRV_O_PROTOCOL;
1030     } else {
1031         *flags &= ~BDRV_O_PROTOCOL;
1032     }
1033
1034     /* Fetch the file name from the options QDict if necessary */
1035     if (protocol && filename) {
1036         if (!qdict_haskey(*options, "filename")) {
1037             qdict_put(*options, "filename", qstring_from_str(filename));
1038             parse_filename = true;
1039         } else {
1040             error_setg(errp, "Can't specify 'file' and 'filename' options at "
1041                              "the same time");
1042             return -EINVAL;
1043         }
1044     }
1045
1046     /* Find the right block driver */
1047     filename = qdict_get_try_str(*options, "filename");
1048
1049     if (!drvname && protocol) {
1050         if (filename) {
1051             drv = bdrv_find_protocol(filename, parse_filename, errp);
1052             if (!drv) {
1053                 return -EINVAL;
1054             }
1055
1056             drvname = drv->format_name;
1057             qdict_put(*options, "driver", qstring_from_str(drvname));
1058         } else {
1059             error_setg(errp, "Must specify either driver or file");
1060             return -EINVAL;
1061         }
1062     }
1063
1064     assert(drv || !protocol);
1065
1066     /* Driver-specific filename parsing */
1067     if (drv && drv->bdrv_parse_filename && parse_filename) {
1068         drv->bdrv_parse_filename(filename, *options, &local_err);
1069         if (local_err) {
1070             error_propagate(errp, local_err);
1071             return -EINVAL;
1072         }
1073
1074         if (!drv->bdrv_needs_filename) {
1075             qdict_del(*options, "filename");
1076         }
1077     }
1078
1079     return 0;
1080 }
1081
1082 static BdrvChild *bdrv_attach_child(BlockDriverState *parent_bs,
1083                                     BlockDriverState *child_bs,
1084                                     const BdrvChildRole *child_role)
1085 {
1086     BdrvChild *child = g_new(BdrvChild, 1);
1087     *child = (BdrvChild) {
1088         .bs     = child_bs,
1089         .role   = child_role,
1090     };
1091
1092     QLIST_INSERT_HEAD(&parent_bs->children, child, next);
1093
1094     return child;
1095 }
1096
1097 static void bdrv_detach_child(BdrvChild *child)
1098 {
1099     QLIST_REMOVE(child, next);
1100     g_free(child);
1101 }
1102
1103 void bdrv_unref_child(BlockDriverState *parent, BdrvChild *child)
1104 {
1105     BlockDriverState *child_bs = child->bs;
1106
1107     if (child->bs->inherits_from == parent) {
1108         child->bs->inherits_from = NULL;
1109     }
1110
1111     bdrv_detach_child(child);
1112     bdrv_unref(child_bs);
1113 }
1114
1115 void bdrv_set_backing_hd(BlockDriverState *bs, BlockDriverState *backing_hd)
1116 {
1117
1118     if (bs->backing_hd) {
1119         assert(bs->backing_blocker);
1120         bdrv_op_unblock_all(bs->backing_hd, bs->backing_blocker);
1121         bdrv_detach_child(bs->backing_child);
1122     } else if (backing_hd) {
1123         error_setg(&bs->backing_blocker,
1124                    "node is used as backing hd of '%s'",
1125                    bdrv_get_device_or_node_name(bs));
1126     }
1127
1128     bs->backing_hd = backing_hd;
1129     if (!backing_hd) {
1130         error_free(bs->backing_blocker);
1131         bs->backing_blocker = NULL;
1132         bs->backing_child = NULL;
1133         goto out;
1134     }
1135     bs->backing_child = bdrv_attach_child(bs, backing_hd, &child_backing);
1136     bs->open_flags &= ~BDRV_O_NO_BACKING;
1137     pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_hd->filename);
1138     pstrcpy(bs->backing_format, sizeof(bs->backing_format),
1139             backing_hd->drv ? backing_hd->drv->format_name : "");
1140
1141     bdrv_op_block_all(bs->backing_hd, bs->backing_blocker);
1142     /* Otherwise we won't be able to commit due to check in bdrv_commit */
1143     bdrv_op_unblock(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET,
1144                     bs->backing_blocker);
1145 out:
1146     bdrv_refresh_limits(bs, NULL);
1147 }
1148
1149 /*
1150  * Opens the backing file for a BlockDriverState if not yet open
1151  *
1152  * options is a QDict of options to pass to the block drivers, or NULL for an
1153  * empty set of options. The reference to the QDict is transferred to this
1154  * function (even on failure), so if the caller intends to reuse the dictionary,
1155  * it needs to use QINCREF() before calling bdrv_file_open.
1156  */
1157 int bdrv_open_backing_file(BlockDriverState *bs, QDict *options, Error **errp)
1158 {
1159     char *backing_filename = g_malloc0(PATH_MAX);
1160     int ret = 0;
1161     BlockDriverState *backing_hd;
1162     Error *local_err = NULL;
1163
1164     if (bs->backing_hd != NULL) {
1165         QDECREF(options);
1166         goto free_exit;
1167     }
1168
1169     /* NULL means an empty set of options */
1170     if (options == NULL) {
1171         options = qdict_new();
1172     }
1173
1174     bs->open_flags &= ~BDRV_O_NO_BACKING;
1175     if (qdict_haskey(options, "file.filename")) {
1176         backing_filename[0] = '\0';
1177     } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
1178         QDECREF(options);
1179         goto free_exit;
1180     } else {
1181         bdrv_get_full_backing_filename(bs, backing_filename, PATH_MAX,
1182                                        &local_err);
1183         if (local_err) {
1184             ret = -EINVAL;
1185             error_propagate(errp, local_err);
1186             QDECREF(options);
1187             goto free_exit;
1188         }
1189     }
1190
1191     if (!bs->drv || !bs->drv->supports_backing) {
1192         ret = -EINVAL;
1193         error_setg(errp, "Driver doesn't support backing files");
1194         QDECREF(options);
1195         goto free_exit;
1196     }
1197
1198     backing_hd = bdrv_new();
1199
1200     if (bs->backing_format[0] != '\0' && !qdict_haskey(options, "driver")) {
1201         qdict_put(options, "driver", qstring_from_str(bs->backing_format));
1202     }
1203
1204     assert(bs->backing_hd == NULL);
1205     ret = bdrv_open_inherit(&backing_hd,
1206                             *backing_filename ? backing_filename : NULL,
1207                             NULL, options, 0, bs, &child_backing, &local_err);
1208     if (ret < 0) {
1209         bdrv_unref(backing_hd);
1210         backing_hd = NULL;
1211         bs->open_flags |= BDRV_O_NO_BACKING;
1212         error_setg(errp, "Could not open backing file: %s",
1213                    error_get_pretty(local_err));
1214         error_free(local_err);
1215         goto free_exit;
1216     }
1217
1218     bdrv_set_backing_hd(bs, backing_hd);
1219
1220 free_exit:
1221     g_free(backing_filename);
1222     return ret;
1223 }
1224
1225 /*
1226  * Opens a disk image whose options are given as BlockdevRef in another block
1227  * device's options.
1228  *
1229  * If allow_none is true, no image will be opened if filename is false and no
1230  * BlockdevRef is given. NULL will be returned, but errp remains unset.
1231  *
1232  * bdrev_key specifies the key for the image's BlockdevRef in the options QDict.
1233  * That QDict has to be flattened; therefore, if the BlockdevRef is a QDict
1234  * itself, all options starting with "${bdref_key}." are considered part of the
1235  * BlockdevRef.
1236  *
1237  * The BlockdevRef will be removed from the options QDict.
1238  */
1239 BdrvChild *bdrv_open_child(const char *filename,
1240                            QDict *options, const char *bdref_key,
1241                            BlockDriverState* parent,
1242                            const BdrvChildRole *child_role,
1243                            bool allow_none, Error **errp)
1244 {
1245     BdrvChild *c = NULL;
1246     BlockDriverState *bs;
1247     QDict *image_options;
1248     int ret;
1249     char *bdref_key_dot;
1250     const char *reference;
1251
1252     assert(child_role != NULL);
1253
1254     bdref_key_dot = g_strdup_printf("%s.", bdref_key);
1255     qdict_extract_subqdict(options, &image_options, bdref_key_dot);
1256     g_free(bdref_key_dot);
1257
1258     reference = qdict_get_try_str(options, bdref_key);
1259     if (!filename && !reference && !qdict_size(image_options)) {
1260         if (!allow_none) {
1261             error_setg(errp, "A block device must be specified for \"%s\"",
1262                        bdref_key);
1263         }
1264         QDECREF(image_options);
1265         goto done;
1266     }
1267
1268     bs = NULL;
1269     ret = bdrv_open_inherit(&bs, filename, reference, image_options, 0,
1270                             parent, child_role, errp);
1271     if (ret < 0) {
1272         goto done;
1273     }
1274
1275     c = bdrv_attach_child(parent, bs, child_role);
1276
1277 done:
1278     qdict_del(options, bdref_key);
1279     return c;
1280 }
1281
1282 /*
1283  * This is a version of bdrv_open_child() that returns 0/-EINVAL instead of
1284  * a BdrvChild object.
1285  *
1286  * If allow_none is true, no image will be opened if filename is false and no
1287  * BlockdevRef is given. *pbs will remain unchanged and 0 will be returned.
1288  *
1289  * To conform with the behavior of bdrv_open(), *pbs has to be NULL.
1290  */
1291 int bdrv_open_image(BlockDriverState **pbs, const char *filename,
1292                     QDict *options, const char *bdref_key,
1293                     BlockDriverState* parent, const BdrvChildRole *child_role,
1294                     bool allow_none, Error **errp)
1295 {
1296     Error *local_err = NULL;
1297     BdrvChild *c;
1298
1299     assert(pbs);
1300     assert(*pbs == NULL);
1301
1302     c = bdrv_open_child(filename, options, bdref_key, parent, child_role,
1303                         allow_none, &local_err);
1304     if (local_err) {
1305         error_propagate(errp, local_err);
1306         return -EINVAL;
1307     }
1308
1309     if (c != NULL) {
1310         *pbs = c->bs;
1311     }
1312
1313     return 0;
1314 }
1315
1316 int bdrv_append_temp_snapshot(BlockDriverState *bs, int flags, Error **errp)
1317 {
1318     /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
1319     char *tmp_filename = g_malloc0(PATH_MAX + 1);
1320     int64_t total_size;
1321     QemuOpts *opts = NULL;
1322     QDict *snapshot_options;
1323     BlockDriverState *bs_snapshot;
1324     Error *local_err = NULL;
1325     int ret;
1326
1327     /* if snapshot, we create a temporary backing file and open it
1328        instead of opening 'filename' directly */
1329
1330     /* Get the required size from the image */
1331     total_size = bdrv_getlength(bs);
1332     if (total_size < 0) {
1333         ret = total_size;
1334         error_setg_errno(errp, -total_size, "Could not get image size");
1335         goto out;
1336     }
1337
1338     /* Create the temporary image */
1339     ret = get_tmp_filename(tmp_filename, PATH_MAX + 1);
1340     if (ret < 0) {
1341         error_setg_errno(errp, -ret, "Could not get temporary filename");
1342         goto out;
1343     }
1344
1345     opts = qemu_opts_create(bdrv_qcow2.create_opts, NULL, 0,
1346                             &error_abort);
1347     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, total_size, &error_abort);
1348     ret = bdrv_create(&bdrv_qcow2, tmp_filename, opts, &local_err);
1349     qemu_opts_del(opts);
1350     if (ret < 0) {
1351         error_setg_errno(errp, -ret, "Could not create temporary overlay "
1352                          "'%s': %s", tmp_filename,
1353                          error_get_pretty(local_err));
1354         error_free(local_err);
1355         goto out;
1356     }
1357
1358     /* Prepare a new options QDict for the temporary file */
1359     snapshot_options = qdict_new();
1360     qdict_put(snapshot_options, "file.driver",
1361               qstring_from_str("file"));
1362     qdict_put(snapshot_options, "file.filename",
1363               qstring_from_str(tmp_filename));
1364     qdict_put(snapshot_options, "driver",
1365               qstring_from_str("qcow2"));
1366
1367     bs_snapshot = bdrv_new();
1368
1369     ret = bdrv_open(&bs_snapshot, NULL, NULL, snapshot_options,
1370                     flags, &local_err);
1371     if (ret < 0) {
1372         error_propagate(errp, local_err);
1373         goto out;
1374     }
1375
1376     bdrv_append(bs_snapshot, bs);
1377
1378 out:
1379     g_free(tmp_filename);
1380     return ret;
1381 }
1382
1383 /*
1384  * Opens a disk image (raw, qcow2, vmdk, ...)
1385  *
1386  * options is a QDict of options to pass to the block drivers, or NULL for an
1387  * empty set of options. The reference to the QDict belongs to the block layer
1388  * after the call (even on failure), so if the caller intends to reuse the
1389  * dictionary, it needs to use QINCREF() before calling bdrv_open.
1390  *
1391  * If *pbs is NULL, a new BDS will be created with a pointer to it stored there.
1392  * If it is not NULL, the referenced BDS will be reused.
1393  *
1394  * The reference parameter may be used to specify an existing block device which
1395  * should be opened. If specified, neither options nor a filename may be given,
1396  * nor can an existing BDS be reused (that is, *pbs has to be NULL).
1397  */
1398 static int bdrv_open_inherit(BlockDriverState **pbs, const char *filename,
1399                              const char *reference, QDict *options, int flags,
1400                              BlockDriverState *parent,
1401                              const BdrvChildRole *child_role, Error **errp)
1402 {
1403     int ret;
1404     BlockDriverState *file = NULL, *bs;
1405     BlockDriver *drv = NULL;
1406     const char *drvname;
1407     Error *local_err = NULL;
1408     int snapshot_flags = 0;
1409
1410     assert(pbs);
1411     assert(!child_role || !flags);
1412     assert(!child_role == !parent);
1413
1414     if (reference) {
1415         bool options_non_empty = options ? qdict_size(options) : false;
1416         QDECREF(options);
1417
1418         if (*pbs) {
1419             error_setg(errp, "Cannot reuse an existing BDS when referencing "
1420                        "another block device");
1421             return -EINVAL;
1422         }
1423
1424         if (filename || options_non_empty) {
1425             error_setg(errp, "Cannot reference an existing block device with "
1426                        "additional options or a new filename");
1427             return -EINVAL;
1428         }
1429
1430         bs = bdrv_lookup_bs(reference, reference, errp);
1431         if (!bs) {
1432             return -ENODEV;
1433         }
1434         bdrv_ref(bs);
1435         *pbs = bs;
1436         return 0;
1437     }
1438
1439     if (*pbs) {
1440         bs = *pbs;
1441     } else {
1442         bs = bdrv_new();
1443     }
1444
1445     /* NULL means an empty set of options */
1446     if (options == NULL) {
1447         options = qdict_new();
1448     }
1449
1450     if (child_role) {
1451         bs->inherits_from = parent;
1452         flags = child_role->inherit_flags(parent->open_flags);
1453     }
1454
1455     ret = bdrv_fill_options(&options, &filename, &flags, &local_err);
1456     if (local_err) {
1457         goto fail;
1458     }
1459
1460     /* Find the right image format driver */
1461     drvname = qdict_get_try_str(options, "driver");
1462     if (drvname) {
1463         drv = bdrv_find_format(drvname);
1464         qdict_del(options, "driver");
1465         if (!drv) {
1466             error_setg(errp, "Unknown driver: '%s'", drvname);
1467             ret = -EINVAL;
1468             goto fail;
1469         }
1470     }
1471
1472     assert(drvname || !(flags & BDRV_O_PROTOCOL));
1473
1474     bs->open_flags = flags;
1475     bs->options = options;
1476     options = qdict_clone_shallow(options);
1477
1478     /* Open image file without format layer */
1479     if ((flags & BDRV_O_PROTOCOL) == 0) {
1480         if (flags & BDRV_O_RDWR) {
1481             flags |= BDRV_O_ALLOW_RDWR;
1482         }
1483         if (flags & BDRV_O_SNAPSHOT) {
1484             snapshot_flags = bdrv_temp_snapshot_flags(flags);
1485             flags = bdrv_backing_flags(flags);
1486         }
1487
1488         assert(file == NULL);
1489         bs->open_flags = flags;
1490         ret = bdrv_open_image(&file, filename, options, "file",
1491                               bs, &child_file, true, &local_err);
1492         if (ret < 0) {
1493             goto fail;
1494         }
1495     }
1496
1497     /* Image format probing */
1498     bs->probed = !drv;
1499     if (!drv && file) {
1500         ret = find_image_format(file, filename, &drv, &local_err);
1501         if (ret < 0) {
1502             goto fail;
1503         }
1504     } else if (!drv) {
1505         error_setg(errp, "Must specify either driver or file");
1506         ret = -EINVAL;
1507         goto fail;
1508     }
1509
1510     /* BDRV_O_PROTOCOL must be set iff a protocol BDS is about to be created */
1511     assert(!!(flags & BDRV_O_PROTOCOL) == !!drv->bdrv_file_open);
1512     /* file must be NULL if a protocol BDS is about to be created
1513      * (the inverse results in an error message from bdrv_open_common()) */
1514     assert(!(flags & BDRV_O_PROTOCOL) || !file);
1515
1516     /* Open the image */
1517     ret = bdrv_open_common(bs, file, options, flags, drv, &local_err);
1518     if (ret < 0) {
1519         goto fail;
1520     }
1521
1522     if (file && (bs->file != file)) {
1523         bdrv_unref(file);
1524         file = NULL;
1525     }
1526
1527     /* If there is a backing file, use it */
1528     if ((flags & BDRV_O_NO_BACKING) == 0) {
1529         QDict *backing_options;
1530
1531         qdict_extract_subqdict(options, &backing_options, "backing.");
1532         ret = bdrv_open_backing_file(bs, backing_options, &local_err);
1533         if (ret < 0) {
1534             goto close_and_fail;
1535         }
1536     }
1537
1538     bdrv_refresh_filename(bs);
1539
1540     /* For snapshot=on, create a temporary qcow2 overlay. bs points to the
1541      * temporary snapshot afterwards. */
1542     if (snapshot_flags) {
1543         ret = bdrv_append_temp_snapshot(bs, snapshot_flags, &local_err);
1544         if (local_err) {
1545             goto close_and_fail;
1546         }
1547     }
1548
1549     /* Check if any unknown options were used */
1550     if (options && (qdict_size(options) != 0)) {
1551         const QDictEntry *entry = qdict_first(options);
1552         if (flags & BDRV_O_PROTOCOL) {
1553             error_setg(errp, "Block protocol '%s' doesn't support the option "
1554                        "'%s'", drv->format_name, entry->key);
1555         } else {
1556             error_setg(errp, "Block format '%s' used by device '%s' doesn't "
1557                        "support the option '%s'", drv->format_name,
1558                        bdrv_get_device_name(bs), entry->key);
1559         }
1560
1561         ret = -EINVAL;
1562         goto close_and_fail;
1563     }
1564
1565     if (!bdrv_key_required(bs)) {
1566         if (bs->blk) {
1567             blk_dev_change_media_cb(bs->blk, true);
1568         }
1569     } else if (!runstate_check(RUN_STATE_PRELAUNCH)
1570                && !runstate_check(RUN_STATE_INMIGRATE)
1571                && !runstate_check(RUN_STATE_PAUSED)) { /* HACK */
1572         error_setg(errp,
1573                    "Guest must be stopped for opening of encrypted image");
1574         ret = -EBUSY;
1575         goto close_and_fail;
1576     }
1577
1578     QDECREF(options);
1579     *pbs = bs;
1580     return 0;
1581
1582 fail:
1583     if (file != NULL) {
1584         bdrv_unref(file);
1585     }
1586     QDECREF(bs->options);
1587     QDECREF(options);
1588     bs->options = NULL;
1589     if (!*pbs) {
1590         /* If *pbs is NULL, a new BDS has been created in this function and
1591            needs to be freed now. Otherwise, it does not need to be closed,
1592            since it has not really been opened yet. */
1593         bdrv_unref(bs);
1594     }
1595     if (local_err) {
1596         error_propagate(errp, local_err);
1597     }
1598     return ret;
1599
1600 close_and_fail:
1601     /* See fail path, but now the BDS has to be always closed */
1602     if (*pbs) {
1603         bdrv_close(bs);
1604     } else {
1605         bdrv_unref(bs);
1606     }
1607     QDECREF(options);
1608     if (local_err) {
1609         error_propagate(errp, local_err);
1610     }
1611     return ret;
1612 }
1613
1614 int bdrv_open(BlockDriverState **pbs, const char *filename,
1615               const char *reference, QDict *options, int flags, Error **errp)
1616 {
1617     return bdrv_open_inherit(pbs, filename, reference, options, flags, NULL,
1618                              NULL, errp);
1619 }
1620
1621 typedef struct BlockReopenQueueEntry {
1622      bool prepared;
1623      BDRVReopenState state;
1624      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1625 } BlockReopenQueueEntry;
1626
1627 /*
1628  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1629  * reopen of multiple devices.
1630  *
1631  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1632  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1633  * be created and initialized. This newly created BlockReopenQueue should be
1634  * passed back in for subsequent calls that are intended to be of the same
1635  * atomic 'set'.
1636  *
1637  * bs is the BlockDriverState to add to the reopen queue.
1638  *
1639  * options contains the changed options for the associated bs
1640  * (the BlockReopenQueue takes ownership)
1641  *
1642  * flags contains the open flags for the associated bs
1643  *
1644  * returns a pointer to bs_queue, which is either the newly allocated
1645  * bs_queue, or the existing bs_queue being used.
1646  *
1647  */
1648 BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1649                                     BlockDriverState *bs,
1650                                     QDict *options, int flags)
1651 {
1652     assert(bs != NULL);
1653
1654     BlockReopenQueueEntry *bs_entry;
1655     BdrvChild *child;
1656     QDict *old_options;
1657
1658     if (bs_queue == NULL) {
1659         bs_queue = g_new0(BlockReopenQueue, 1);
1660         QSIMPLEQ_INIT(bs_queue);
1661     }
1662
1663     if (!options) {
1664         options = qdict_new();
1665     }
1666
1667     old_options = qdict_clone_shallow(bs->options);
1668     qdict_join(options, old_options, false);
1669     QDECREF(old_options);
1670
1671     /* bdrv_open() masks this flag out */
1672     flags &= ~BDRV_O_PROTOCOL;
1673
1674     QLIST_FOREACH(child, &bs->children, next) {
1675         int child_flags;
1676
1677         if (child->bs->inherits_from != bs) {
1678             continue;
1679         }
1680
1681         child_flags = child->role->inherit_flags(flags);
1682         /* TODO Pass down child flags (backing.*, extents.*, ...) */
1683         bdrv_reopen_queue(bs_queue, child->bs, NULL, child_flags);
1684     }
1685
1686     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1687     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1688
1689     bs_entry->state.bs = bs;
1690     bs_entry->state.options = options;
1691     bs_entry->state.flags = flags;
1692
1693     return bs_queue;
1694 }
1695
1696 /*
1697  * Reopen multiple BlockDriverStates atomically & transactionally.
1698  *
1699  * The queue passed in (bs_queue) must have been built up previous
1700  * via bdrv_reopen_queue().
1701  *
1702  * Reopens all BDS specified in the queue, with the appropriate
1703  * flags.  All devices are prepared for reopen, and failure of any
1704  * device will cause all device changes to be abandonded, and intermediate
1705  * data cleaned up.
1706  *
1707  * If all devices prepare successfully, then the changes are committed
1708  * to all devices.
1709  *
1710  */
1711 int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1712 {
1713     int ret = -1;
1714     BlockReopenQueueEntry *bs_entry, *next;
1715     Error *local_err = NULL;
1716
1717     assert(bs_queue != NULL);
1718
1719     bdrv_drain_all();
1720
1721     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1722         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1723             error_propagate(errp, local_err);
1724             goto cleanup;
1725         }
1726         bs_entry->prepared = true;
1727     }
1728
1729     /* If we reach this point, we have success and just need to apply the
1730      * changes
1731      */
1732     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1733         bdrv_reopen_commit(&bs_entry->state);
1734     }
1735
1736     ret = 0;
1737
1738 cleanup:
1739     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1740         if (ret && bs_entry->prepared) {
1741             bdrv_reopen_abort(&bs_entry->state);
1742         }
1743         QDECREF(bs_entry->state.options);
1744         g_free(bs_entry);
1745     }
1746     g_free(bs_queue);
1747     return ret;
1748 }
1749
1750
1751 /* Reopen a single BlockDriverState with the specified flags. */
1752 int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1753 {
1754     int ret = -1;
1755     Error *local_err = NULL;
1756     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, NULL, bdrv_flags);
1757
1758     ret = bdrv_reopen_multiple(queue, &local_err);
1759     if (local_err != NULL) {
1760         error_propagate(errp, local_err);
1761     }
1762     return ret;
1763 }
1764
1765
1766 /*
1767  * Prepares a BlockDriverState for reopen. All changes are staged in the
1768  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1769  * the block driver layer .bdrv_reopen_prepare()
1770  *
1771  * bs is the BlockDriverState to reopen
1772  * flags are the new open flags
1773  * queue is the reopen queue
1774  *
1775  * Returns 0 on success, non-zero on error.  On error errp will be set
1776  * as well.
1777  *
1778  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1779  * It is the responsibility of the caller to then call the abort() or
1780  * commit() for any other BDS that have been left in a prepare() state
1781  *
1782  */
1783 int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1784                         Error **errp)
1785 {
1786     int ret = -1;
1787     Error *local_err = NULL;
1788     BlockDriver *drv;
1789
1790     assert(reopen_state != NULL);
1791     assert(reopen_state->bs->drv != NULL);
1792     drv = reopen_state->bs->drv;
1793
1794     /* if we are to stay read-only, do not allow permission change
1795      * to r/w */
1796     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1797         reopen_state->flags & BDRV_O_RDWR) {
1798         error_setg(errp, "Node '%s' is read only",
1799                    bdrv_get_device_or_node_name(reopen_state->bs));
1800         goto error;
1801     }
1802
1803
1804     ret = bdrv_flush(reopen_state->bs);
1805     if (ret) {
1806         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1807                   strerror(-ret));
1808         goto error;
1809     }
1810
1811     if (drv->bdrv_reopen_prepare) {
1812         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1813         if (ret) {
1814             if (local_err != NULL) {
1815                 error_propagate(errp, local_err);
1816             } else {
1817                 error_setg(errp, "failed while preparing to reopen image '%s'",
1818                            reopen_state->bs->filename);
1819             }
1820             goto error;
1821         }
1822     } else {
1823         /* It is currently mandatory to have a bdrv_reopen_prepare()
1824          * handler for each supported drv. */
1825         error_setg(errp, "Block format '%s' used by node '%s' "
1826                    "does not support reopening files", drv->format_name,
1827                    bdrv_get_device_or_node_name(reopen_state->bs));
1828         ret = -1;
1829         goto error;
1830     }
1831
1832     /* Options that are not handled are only okay if they are unchanged
1833      * compared to the old state. It is expected that some options are only
1834      * used for the initial open, but not reopen (e.g. filename) */
1835     if (qdict_size(reopen_state->options)) {
1836         const QDictEntry *entry = qdict_first(reopen_state->options);
1837
1838         do {
1839             QString *new_obj = qobject_to_qstring(entry->value);
1840             const char *new = qstring_get_str(new_obj);
1841             const char *old = qdict_get_try_str(reopen_state->bs->options,
1842                                                 entry->key);
1843
1844             if (!old || strcmp(new, old)) {
1845                 error_setg(errp, "Cannot change the option '%s'", entry->key);
1846                 ret = -EINVAL;
1847                 goto error;
1848             }
1849         } while ((entry = qdict_next(reopen_state->options, entry)));
1850     }
1851
1852     ret = 0;
1853
1854 error:
1855     return ret;
1856 }
1857
1858 /*
1859  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1860  * makes them final by swapping the staging BlockDriverState contents into
1861  * the active BlockDriverState contents.
1862  */
1863 void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1864 {
1865     BlockDriver *drv;
1866
1867     assert(reopen_state != NULL);
1868     drv = reopen_state->bs->drv;
1869     assert(drv != NULL);
1870
1871     /* If there are any driver level actions to take */
1872     if (drv->bdrv_reopen_commit) {
1873         drv->bdrv_reopen_commit(reopen_state);
1874     }
1875
1876     /* set BDS specific flags now */
1877     reopen_state->bs->open_flags         = reopen_state->flags;
1878     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1879                                               BDRV_O_CACHE_WB);
1880     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1881
1882     bdrv_refresh_limits(reopen_state->bs, NULL);
1883 }
1884
1885 /*
1886  * Abort the reopen, and delete and free the staged changes in
1887  * reopen_state
1888  */
1889 void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1890 {
1891     BlockDriver *drv;
1892
1893     assert(reopen_state != NULL);
1894     drv = reopen_state->bs->drv;
1895     assert(drv != NULL);
1896
1897     if (drv->bdrv_reopen_abort) {
1898         drv->bdrv_reopen_abort(reopen_state);
1899     }
1900 }
1901
1902
1903 void bdrv_close(BlockDriverState *bs)
1904 {
1905     BdrvAioNotifier *ban, *ban_next;
1906
1907     if (bs->job) {
1908         block_job_cancel_sync(bs->job);
1909     }
1910
1911     /* Disable I/O limits and drain all pending throttled requests */
1912     if (bs->io_limits_enabled) {
1913         bdrv_io_limits_disable(bs);
1914     }
1915
1916     bdrv_drain(bs); /* complete I/O */
1917     bdrv_flush(bs);
1918     bdrv_drain(bs); /* in case flush left pending I/O */
1919     notifier_list_notify(&bs->close_notifiers, bs);
1920
1921     if (bs->drv) {
1922         BdrvChild *child, *next;
1923
1924         bs->drv->bdrv_close(bs);
1925
1926         if (bs->backing_hd) {
1927             BlockDriverState *backing_hd = bs->backing_hd;
1928             bdrv_set_backing_hd(bs, NULL);
1929             bdrv_unref(backing_hd);
1930         }
1931
1932         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
1933             /* TODO Remove bdrv_unref() from drivers' close function and use
1934              * bdrv_unref_child() here */
1935             if (child->bs->inherits_from == bs) {
1936                 child->bs->inherits_from = NULL;
1937             }
1938             bdrv_detach_child(child);
1939         }
1940
1941         g_free(bs->opaque);
1942         bs->opaque = NULL;
1943         bs->drv = NULL;
1944         bs->copy_on_read = 0;
1945         bs->backing_file[0] = '\0';
1946         bs->backing_format[0] = '\0';
1947         bs->total_sectors = 0;
1948         bs->encrypted = 0;
1949         bs->valid_key = 0;
1950         bs->sg = 0;
1951         bs->zero_beyond_eof = false;
1952         QDECREF(bs->options);
1953         bs->options = NULL;
1954         QDECREF(bs->full_open_options);
1955         bs->full_open_options = NULL;
1956
1957         if (bs->file != NULL) {
1958             bdrv_unref(bs->file);
1959             bs->file = NULL;
1960         }
1961     }
1962
1963     if (bs->blk) {
1964         blk_dev_change_media_cb(bs->blk, false);
1965     }
1966
1967     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
1968         g_free(ban);
1969     }
1970     QLIST_INIT(&bs->aio_notifiers);
1971 }
1972
1973 void bdrv_close_all(void)
1974 {
1975     BlockDriverState *bs;
1976
1977     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
1978         AioContext *aio_context = bdrv_get_aio_context(bs);
1979
1980         aio_context_acquire(aio_context);
1981         bdrv_close(bs);
1982         aio_context_release(aio_context);
1983     }
1984 }
1985
1986 /* make a BlockDriverState anonymous by removing from bdrv_state and
1987  * graph_bdrv_state list.
1988    Also, NULL terminate the device_name to prevent double remove */
1989 void bdrv_make_anon(BlockDriverState *bs)
1990 {
1991     /*
1992      * Take care to remove bs from bdrv_states only when it's actually
1993      * in it.  Note that bs->device_list.tqe_prev is initially null,
1994      * and gets set to non-null by QTAILQ_INSERT_TAIL().  Establish
1995      * the useful invariant "bs in bdrv_states iff bs->tqe_prev" by
1996      * resetting it to null on remove.
1997      */
1998     if (bs->device_list.tqe_prev) {
1999         QTAILQ_REMOVE(&bdrv_states, bs, device_list);
2000         bs->device_list.tqe_prev = NULL;
2001     }
2002     if (bs->node_name[0] != '\0') {
2003         QTAILQ_REMOVE(&graph_bdrv_states, bs, node_list);
2004     }
2005     bs->node_name[0] = '\0';
2006 }
2007
2008 static void bdrv_rebind(BlockDriverState *bs)
2009 {
2010     if (bs->drv && bs->drv->bdrv_rebind) {
2011         bs->drv->bdrv_rebind(bs);
2012     }
2013 }
2014
2015 static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
2016                                      BlockDriverState *bs_src)
2017 {
2018     /* move some fields that need to stay attached to the device */
2019
2020     /* dev info */
2021     bs_dest->guest_block_size   = bs_src->guest_block_size;
2022     bs_dest->copy_on_read       = bs_src->copy_on_read;
2023
2024     bs_dest->enable_write_cache = bs_src->enable_write_cache;
2025
2026     /* i/o throttled req */
2027     bs_dest->throttle_state     = bs_src->throttle_state,
2028     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
2029     bs_dest->pending_reqs[0]    = bs_src->pending_reqs[0];
2030     bs_dest->pending_reqs[1]    = bs_src->pending_reqs[1];
2031     bs_dest->throttled_reqs[0]  = bs_src->throttled_reqs[0];
2032     bs_dest->throttled_reqs[1]  = bs_src->throttled_reqs[1];
2033     memcpy(&bs_dest->round_robin,
2034            &bs_src->round_robin,
2035            sizeof(bs_dest->round_robin));
2036     memcpy(&bs_dest->throttle_timers,
2037            &bs_src->throttle_timers,
2038            sizeof(ThrottleTimers));
2039
2040     /* r/w error */
2041     bs_dest->on_read_error      = bs_src->on_read_error;
2042     bs_dest->on_write_error     = bs_src->on_write_error;
2043
2044     /* i/o status */
2045     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
2046     bs_dest->iostatus           = bs_src->iostatus;
2047
2048     /* dirty bitmap */
2049     bs_dest->dirty_bitmaps      = bs_src->dirty_bitmaps;
2050
2051     /* reference count */
2052     bs_dest->refcnt             = bs_src->refcnt;
2053
2054     /* job */
2055     bs_dest->job                = bs_src->job;
2056
2057     /* keep the same entry in bdrv_states */
2058     bs_dest->device_list = bs_src->device_list;
2059     bs_dest->blk = bs_src->blk;
2060
2061     memcpy(bs_dest->op_blockers, bs_src->op_blockers,
2062            sizeof(bs_dest->op_blockers));
2063 }
2064
2065 /*
2066  * Swap bs contents for two image chains while they are live,
2067  * while keeping required fields on the BlockDriverState that is
2068  * actually attached to a device.
2069  *
2070  * This will modify the BlockDriverState fields, and swap contents
2071  * between bs_new and bs_old. Both bs_new and bs_old are modified.
2072  *
2073  * bs_new must not be attached to a BlockBackend.
2074  *
2075  * This function does not create any image files.
2076  */
2077 void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
2078 {
2079     BlockDriverState tmp;
2080     BdrvChild *child;
2081
2082     bdrv_drain(bs_new);
2083     bdrv_drain(bs_old);
2084
2085     /* The code needs to swap the node_name but simply swapping node_list won't
2086      * work so first remove the nodes from the graph list, do the swap then
2087      * insert them back if needed.
2088      */
2089     if (bs_new->node_name[0] != '\0') {
2090         QTAILQ_REMOVE(&graph_bdrv_states, bs_new, node_list);
2091     }
2092     if (bs_old->node_name[0] != '\0') {
2093         QTAILQ_REMOVE(&graph_bdrv_states, bs_old, node_list);
2094     }
2095
2096     /* If the BlockDriverState is part of a throttling group acquire
2097      * its lock since we're going to mess with the protected fields.
2098      * Otherwise there's no need to worry since no one else can touch
2099      * them. */
2100     if (bs_old->throttle_state) {
2101         throttle_group_lock(bs_old);
2102     }
2103
2104     /* bs_new must be unattached and shouldn't have anything fancy enabled */
2105     assert(!bs_new->blk);
2106     assert(QLIST_EMPTY(&bs_new->dirty_bitmaps));
2107     assert(bs_new->job == NULL);
2108     assert(bs_new->io_limits_enabled == false);
2109     assert(bs_new->throttle_state == NULL);
2110     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
2111
2112     tmp = *bs_new;
2113     *bs_new = *bs_old;
2114     *bs_old = tmp;
2115
2116     /* there are some fields that should not be swapped, move them back */
2117     bdrv_move_feature_fields(&tmp, bs_old);
2118     bdrv_move_feature_fields(bs_old, bs_new);
2119     bdrv_move_feature_fields(bs_new, &tmp);
2120
2121     /* bs_new must remain unattached */
2122     assert(!bs_new->blk);
2123
2124     /* Check a few fields that should remain attached to the device */
2125     assert(bs_new->job == NULL);
2126     assert(bs_new->io_limits_enabled == false);
2127     assert(bs_new->throttle_state == NULL);
2128     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
2129
2130     /* Release the ThrottleGroup lock */
2131     if (bs_old->throttle_state) {
2132         throttle_group_unlock(bs_old);
2133     }
2134
2135     /* insert the nodes back into the graph node list if needed */
2136     if (bs_new->node_name[0] != '\0') {
2137         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_new, node_list);
2138     }
2139     if (bs_old->node_name[0] != '\0') {
2140         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_old, node_list);
2141     }
2142
2143     /*
2144      * Update lh_first.le_prev for non-empty lists.
2145      *
2146      * The head of the op blocker list doesn't change because it is moved back
2147      * in bdrv_move_feature_fields().
2148      */
2149     assert(QLIST_EMPTY(&bs_old->tracked_requests));
2150     assert(QLIST_EMPTY(&bs_new->tracked_requests));
2151
2152     QLIST_FIX_HEAD_PTR(&bs_new->children, next);
2153     QLIST_FIX_HEAD_PTR(&bs_old->children, next);
2154
2155     /* Update references in bs->opaque and children */
2156     QLIST_FOREACH(child, &bs_old->children, next) {
2157         if (child->bs->inherits_from == bs_new) {
2158             child->bs->inherits_from = bs_old;
2159         }
2160     }
2161     QLIST_FOREACH(child, &bs_new->children, next) {
2162         if (child->bs->inherits_from == bs_old) {
2163             child->bs->inherits_from = bs_new;
2164         }
2165     }
2166
2167     bdrv_rebind(bs_new);
2168     bdrv_rebind(bs_old);
2169 }
2170
2171 /*
2172  * Add new bs contents at the top of an image chain while the chain is
2173  * live, while keeping required fields on the top layer.
2174  *
2175  * This will modify the BlockDriverState fields, and swap contents
2176  * between bs_new and bs_top. Both bs_new and bs_top are modified.
2177  *
2178  * bs_new must not be attached to a BlockBackend.
2179  *
2180  * This function does not create any image files.
2181  */
2182 void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
2183 {
2184     bdrv_swap(bs_new, bs_top);
2185
2186     /* The contents of 'tmp' will become bs_top, as we are
2187      * swapping bs_new and bs_top contents. */
2188     bdrv_set_backing_hd(bs_top, bs_new);
2189 }
2190
2191 static void bdrv_delete(BlockDriverState *bs)
2192 {
2193     assert(!bs->job);
2194     assert(bdrv_op_blocker_is_empty(bs));
2195     assert(!bs->refcnt);
2196     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
2197
2198     bdrv_close(bs);
2199
2200     /* remove from list, if necessary */
2201     bdrv_make_anon(bs);
2202
2203     g_free(bs);
2204 }
2205
2206 /*
2207  * Run consistency checks on an image
2208  *
2209  * Returns 0 if the check could be completed (it doesn't mean that the image is
2210  * free of errors) or -errno when an internal error occurred. The results of the
2211  * check are stored in res.
2212  */
2213 int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
2214 {
2215     if (bs->drv == NULL) {
2216         return -ENOMEDIUM;
2217     }
2218     if (bs->drv->bdrv_check == NULL) {
2219         return -ENOTSUP;
2220     }
2221
2222     memset(res, 0, sizeof(*res));
2223     return bs->drv->bdrv_check(bs, res, fix);
2224 }
2225
2226 #define COMMIT_BUF_SECTORS 2048
2227
2228 /* commit COW file into the raw image */
2229 int bdrv_commit(BlockDriverState *bs)
2230 {
2231     BlockDriver *drv = bs->drv;
2232     int64_t sector, total_sectors, length, backing_length;
2233     int n, ro, open_flags;
2234     int ret = 0;
2235     uint8_t *buf = NULL;
2236
2237     if (!drv)
2238         return -ENOMEDIUM;
2239
2240     if (!bs->backing_hd) {
2241         return -ENOTSUP;
2242     }
2243
2244     if (bdrv_op_is_blocked(bs, BLOCK_OP_TYPE_COMMIT_SOURCE, NULL) ||
2245         bdrv_op_is_blocked(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET, NULL)) {
2246         return -EBUSY;
2247     }
2248
2249     ro = bs->backing_hd->read_only;
2250     open_flags =  bs->backing_hd->open_flags;
2251
2252     if (ro) {
2253         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
2254             return -EACCES;
2255         }
2256     }
2257
2258     length = bdrv_getlength(bs);
2259     if (length < 0) {
2260         ret = length;
2261         goto ro_cleanup;
2262     }
2263
2264     backing_length = bdrv_getlength(bs->backing_hd);
2265     if (backing_length < 0) {
2266         ret = backing_length;
2267         goto ro_cleanup;
2268     }
2269
2270     /* If our top snapshot is larger than the backing file image,
2271      * grow the backing file image if possible.  If not possible,
2272      * we must return an error */
2273     if (length > backing_length) {
2274         ret = bdrv_truncate(bs->backing_hd, length);
2275         if (ret < 0) {
2276             goto ro_cleanup;
2277         }
2278     }
2279
2280     total_sectors = length >> BDRV_SECTOR_BITS;
2281
2282     /* qemu_try_blockalign() for bs will choose an alignment that works for
2283      * bs->backing_hd as well, so no need to compare the alignment manually. */
2284     buf = qemu_try_blockalign(bs, COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
2285     if (buf == NULL) {
2286         ret = -ENOMEM;
2287         goto ro_cleanup;
2288     }
2289
2290     for (sector = 0; sector < total_sectors; sector += n) {
2291         ret = bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n);
2292         if (ret < 0) {
2293             goto ro_cleanup;
2294         }
2295         if (ret) {
2296             ret = bdrv_read(bs, sector, buf, n);
2297             if (ret < 0) {
2298                 goto ro_cleanup;
2299             }
2300
2301             ret = bdrv_write(bs->backing_hd, sector, buf, n);
2302             if (ret < 0) {
2303                 goto ro_cleanup;
2304             }
2305         }
2306     }
2307
2308     if (drv->bdrv_make_empty) {
2309         ret = drv->bdrv_make_empty(bs);
2310         if (ret < 0) {
2311             goto ro_cleanup;
2312         }
2313         bdrv_flush(bs);
2314     }
2315
2316     /*
2317      * Make sure all data we wrote to the backing device is actually
2318      * stable on disk.
2319      */
2320     if (bs->backing_hd) {
2321         bdrv_flush(bs->backing_hd);
2322     }
2323
2324     ret = 0;
2325 ro_cleanup:
2326     qemu_vfree(buf);
2327
2328     if (ro) {
2329         /* ignoring error return here */
2330         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
2331     }
2332
2333     return ret;
2334 }
2335
2336 int bdrv_commit_all(void)
2337 {
2338     BlockDriverState *bs;
2339
2340     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
2341         AioContext *aio_context = bdrv_get_aio_context(bs);
2342
2343         aio_context_acquire(aio_context);
2344         if (bs->drv && bs->backing_hd) {
2345             int ret = bdrv_commit(bs);
2346             if (ret < 0) {
2347                 aio_context_release(aio_context);
2348                 return ret;
2349             }
2350         }
2351         aio_context_release(aio_context);
2352     }
2353     return 0;
2354 }
2355
2356 /*
2357  * Return values:
2358  * 0        - success
2359  * -EINVAL  - backing format specified, but no file
2360  * -ENOSPC  - can't update the backing file because no space is left in the
2361  *            image file header
2362  * -ENOTSUP - format driver doesn't support changing the backing file
2363  */
2364 int bdrv_change_backing_file(BlockDriverState *bs,
2365     const char *backing_file, const char *backing_fmt)
2366 {
2367     BlockDriver *drv = bs->drv;
2368     int ret;
2369
2370     /* Backing file format doesn't make sense without a backing file */
2371     if (backing_fmt && !backing_file) {
2372         return -EINVAL;
2373     }
2374
2375     if (drv->bdrv_change_backing_file != NULL) {
2376         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
2377     } else {
2378         ret = -ENOTSUP;
2379     }
2380
2381     if (ret == 0) {
2382         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
2383         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
2384     }
2385     return ret;
2386 }
2387
2388 /*
2389  * Finds the image layer in the chain that has 'bs' as its backing file.
2390  *
2391  * active is the current topmost image.
2392  *
2393  * Returns NULL if bs is not found in active's image chain,
2394  * or if active == bs.
2395  *
2396  * Returns the bottommost base image if bs == NULL.
2397  */
2398 BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
2399                                     BlockDriverState *bs)
2400 {
2401     while (active && bs != active->backing_hd) {
2402         active = active->backing_hd;
2403     }
2404
2405     return active;
2406 }
2407
2408 /* Given a BDS, searches for the base layer. */
2409 BlockDriverState *bdrv_find_base(BlockDriverState *bs)
2410 {
2411     return bdrv_find_overlay(bs, NULL);
2412 }
2413
2414 typedef struct BlkIntermediateStates {
2415     BlockDriverState *bs;
2416     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
2417 } BlkIntermediateStates;
2418
2419
2420 /*
2421  * Drops images above 'base' up to and including 'top', and sets the image
2422  * above 'top' to have base as its backing file.
2423  *
2424  * Requires that the overlay to 'top' is opened r/w, so that the backing file
2425  * information in 'bs' can be properly updated.
2426  *
2427  * E.g., this will convert the following chain:
2428  * bottom <- base <- intermediate <- top <- active
2429  *
2430  * to
2431  *
2432  * bottom <- base <- active
2433  *
2434  * It is allowed for bottom==base, in which case it converts:
2435  *
2436  * base <- intermediate <- top <- active
2437  *
2438  * to
2439  *
2440  * base <- active
2441  *
2442  * If backing_file_str is non-NULL, it will be used when modifying top's
2443  * overlay image metadata.
2444  *
2445  * Error conditions:
2446  *  if active == top, that is considered an error
2447  *
2448  */
2449 int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
2450                            BlockDriverState *base, const char *backing_file_str)
2451 {
2452     BlockDriverState *intermediate;
2453     BlockDriverState *base_bs = NULL;
2454     BlockDriverState *new_top_bs = NULL;
2455     BlkIntermediateStates *intermediate_state, *next;
2456     int ret = -EIO;
2457
2458     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
2459     QSIMPLEQ_INIT(&states_to_delete);
2460
2461     if (!top->drv || !base->drv) {
2462         goto exit;
2463     }
2464
2465     new_top_bs = bdrv_find_overlay(active, top);
2466
2467     if (new_top_bs == NULL) {
2468         /* we could not find the image above 'top', this is an error */
2469         goto exit;
2470     }
2471
2472     /* special case of new_top_bs->backing_hd already pointing to base - nothing
2473      * to do, no intermediate images */
2474     if (new_top_bs->backing_hd == base) {
2475         ret = 0;
2476         goto exit;
2477     }
2478
2479     intermediate = top;
2480
2481     /* now we will go down through the list, and add each BDS we find
2482      * into our deletion queue, until we hit the 'base'
2483      */
2484     while (intermediate) {
2485         intermediate_state = g_new0(BlkIntermediateStates, 1);
2486         intermediate_state->bs = intermediate;
2487         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
2488
2489         if (intermediate->backing_hd == base) {
2490             base_bs = intermediate->backing_hd;
2491             break;
2492         }
2493         intermediate = intermediate->backing_hd;
2494     }
2495     if (base_bs == NULL) {
2496         /* something went wrong, we did not end at the base. safely
2497          * unravel everything, and exit with error */
2498         goto exit;
2499     }
2500
2501     /* success - we can delete the intermediate states, and link top->base */
2502     backing_file_str = backing_file_str ? backing_file_str : base_bs->filename;
2503     ret = bdrv_change_backing_file(new_top_bs, backing_file_str,
2504                                    base_bs->drv ? base_bs->drv->format_name : "");
2505     if (ret) {
2506         goto exit;
2507     }
2508     bdrv_set_backing_hd(new_top_bs, base_bs);
2509
2510     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
2511         /* so that bdrv_close() does not recursively close the chain */
2512         bdrv_set_backing_hd(intermediate_state->bs, NULL);
2513         bdrv_unref(intermediate_state->bs);
2514     }
2515     ret = 0;
2516
2517 exit:
2518     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
2519         g_free(intermediate_state);
2520     }
2521     return ret;
2522 }
2523
2524 /**
2525  * Truncate file to 'offset' bytes (needed only for file protocols)
2526  */
2527 int bdrv_truncate(BlockDriverState *bs, int64_t offset)
2528 {
2529     BlockDriver *drv = bs->drv;
2530     int ret;
2531     if (!drv)
2532         return -ENOMEDIUM;
2533     if (!drv->bdrv_truncate)
2534         return -ENOTSUP;
2535     if (bs->read_only)
2536         return -EACCES;
2537
2538     ret = drv->bdrv_truncate(bs, offset);
2539     if (ret == 0) {
2540         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2541         bdrv_dirty_bitmap_truncate(bs);
2542         if (bs->blk) {
2543             blk_dev_resize_cb(bs->blk);
2544         }
2545     }
2546     return ret;
2547 }
2548
2549 /**
2550  * Length of a allocated file in bytes. Sparse files are counted by actual
2551  * allocated space. Return < 0 if error or unknown.
2552  */
2553 int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
2554 {
2555     BlockDriver *drv = bs->drv;
2556     if (!drv) {
2557         return -ENOMEDIUM;
2558     }
2559     if (drv->bdrv_get_allocated_file_size) {
2560         return drv->bdrv_get_allocated_file_size(bs);
2561     }
2562     if (bs->file) {
2563         return bdrv_get_allocated_file_size(bs->file);
2564     }
2565     return -ENOTSUP;
2566 }
2567
2568 /**
2569  * Return number of sectors on success, -errno on error.
2570  */
2571 int64_t bdrv_nb_sectors(BlockDriverState *bs)
2572 {
2573     BlockDriver *drv = bs->drv;
2574
2575     if (!drv)
2576         return -ENOMEDIUM;
2577
2578     if (drv->has_variable_length) {
2579         int ret = refresh_total_sectors(bs, bs->total_sectors);
2580         if (ret < 0) {
2581             return ret;
2582         }
2583     }
2584     return bs->total_sectors;
2585 }
2586
2587 /**
2588  * Return length in bytes on success, -errno on error.
2589  * The length is always a multiple of BDRV_SECTOR_SIZE.
2590  */
2591 int64_t bdrv_getlength(BlockDriverState *bs)
2592 {
2593     int64_t ret = bdrv_nb_sectors(bs);
2594
2595     ret = ret > INT64_MAX / BDRV_SECTOR_SIZE ? -EFBIG : ret;
2596     return ret < 0 ? ret : ret * BDRV_SECTOR_SIZE;
2597 }
2598
2599 /* return 0 as number of sectors if no device present or error */
2600 void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2601 {
2602     int64_t nb_sectors = bdrv_nb_sectors(bs);
2603
2604     *nb_sectors_ptr = nb_sectors < 0 ? 0 : nb_sectors;
2605 }
2606
2607 void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2608                        BlockdevOnError on_write_error)
2609 {
2610     bs->on_read_error = on_read_error;
2611     bs->on_write_error = on_write_error;
2612 }
2613
2614 BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2615 {
2616     return is_read ? bs->on_read_error : bs->on_write_error;
2617 }
2618
2619 BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
2620 {
2621     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
2622
2623     switch (on_err) {
2624     case BLOCKDEV_ON_ERROR_ENOSPC:
2625         return (error == ENOSPC) ?
2626                BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
2627     case BLOCKDEV_ON_ERROR_STOP:
2628         return BLOCK_ERROR_ACTION_STOP;
2629     case BLOCKDEV_ON_ERROR_REPORT:
2630         return BLOCK_ERROR_ACTION_REPORT;
2631     case BLOCKDEV_ON_ERROR_IGNORE:
2632         return BLOCK_ERROR_ACTION_IGNORE;
2633     default:
2634         abort();
2635     }
2636 }
2637
2638 static void send_qmp_error_event(BlockDriverState *bs,
2639                                  BlockErrorAction action,
2640                                  bool is_read, int error)
2641 {
2642     IoOperationType optype;
2643
2644     optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
2645     qapi_event_send_block_io_error(bdrv_get_device_name(bs), optype, action,
2646                                    bdrv_iostatus_is_enabled(bs),
2647                                    error == ENOSPC, strerror(error),
2648                                    &error_abort);
2649 }
2650
2651 /* This is done by device models because, while the block layer knows
2652  * about the error, it does not know whether an operation comes from
2653  * the device or the block layer (from a job, for example).
2654  */
2655 void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
2656                        bool is_read, int error)
2657 {
2658     assert(error >= 0);
2659
2660     if (action == BLOCK_ERROR_ACTION_STOP) {
2661         /* First set the iostatus, so that "info block" returns an iostatus
2662          * that matches the events raised so far (an additional error iostatus
2663          * is fine, but not a lost one).
2664          */
2665         bdrv_iostatus_set_err(bs, error);
2666
2667         /* Then raise the request to stop the VM and the event.
2668          * qemu_system_vmstop_request_prepare has two effects.  First,
2669          * it ensures that the STOP event always comes after the
2670          * BLOCK_IO_ERROR event.  Second, it ensures that even if management
2671          * can observe the STOP event and do a "cont" before the STOP
2672          * event is issued, the VM will not stop.  In this case, vm_start()
2673          * also ensures that the STOP/RESUME pair of events is emitted.
2674          */
2675         qemu_system_vmstop_request_prepare();
2676         send_qmp_error_event(bs, action, is_read, error);
2677         qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
2678     } else {
2679         send_qmp_error_event(bs, action, is_read, error);
2680     }
2681 }
2682
2683 int bdrv_is_read_only(BlockDriverState *bs)
2684 {
2685     return bs->read_only;
2686 }
2687
2688 int bdrv_is_sg(BlockDriverState *bs)
2689 {
2690     return bs->sg;
2691 }
2692
2693 int bdrv_enable_write_cache(BlockDriverState *bs)
2694 {
2695     return bs->enable_write_cache;
2696 }
2697
2698 void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2699 {
2700     bs->enable_write_cache = wce;
2701
2702     /* so a reopen() will preserve wce */
2703     if (wce) {
2704         bs->open_flags |= BDRV_O_CACHE_WB;
2705     } else {
2706         bs->open_flags &= ~BDRV_O_CACHE_WB;
2707     }
2708 }
2709
2710 int bdrv_is_encrypted(BlockDriverState *bs)
2711 {
2712     if (bs->backing_hd && bs->backing_hd->encrypted)
2713         return 1;
2714     return bs->encrypted;
2715 }
2716
2717 int bdrv_key_required(BlockDriverState *bs)
2718 {
2719     BlockDriverState *backing_hd = bs->backing_hd;
2720
2721     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2722         return 1;
2723     return (bs->encrypted && !bs->valid_key);
2724 }
2725
2726 int bdrv_set_key(BlockDriverState *bs, const char *key)
2727 {
2728     int ret;
2729     if (bs->backing_hd && bs->backing_hd->encrypted) {
2730         ret = bdrv_set_key(bs->backing_hd, key);
2731         if (ret < 0)
2732             return ret;
2733         if (!bs->encrypted)
2734             return 0;
2735     }
2736     if (!bs->encrypted) {
2737         return -EINVAL;
2738     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2739         return -ENOMEDIUM;
2740     }
2741     ret = bs->drv->bdrv_set_key(bs, key);
2742     if (ret < 0) {
2743         bs->valid_key = 0;
2744     } else if (!bs->valid_key) {
2745         bs->valid_key = 1;
2746         if (bs->blk) {
2747             /* call the change callback now, we skipped it on open */
2748             blk_dev_change_media_cb(bs->blk, true);
2749         }
2750     }
2751     return ret;
2752 }
2753
2754 /*
2755  * Provide an encryption key for @bs.
2756  * If @key is non-null:
2757  *     If @bs is not encrypted, fail.
2758  *     Else if the key is invalid, fail.
2759  *     Else set @bs's key to @key, replacing the existing key, if any.
2760  * If @key is null:
2761  *     If @bs is encrypted and still lacks a key, fail.
2762  *     Else do nothing.
2763  * On failure, store an error object through @errp if non-null.
2764  */
2765 void bdrv_add_key(BlockDriverState *bs, const char *key, Error **errp)
2766 {
2767     if (key) {
2768         if (!bdrv_is_encrypted(bs)) {
2769             error_setg(errp, "Node '%s' is not encrypted",
2770                       bdrv_get_device_or_node_name(bs));
2771         } else if (bdrv_set_key(bs, key) < 0) {
2772             error_setg(errp, QERR_INVALID_PASSWORD);
2773         }
2774     } else {
2775         if (bdrv_key_required(bs)) {
2776             error_set(errp, ERROR_CLASS_DEVICE_ENCRYPTED,
2777                       "'%s' (%s) is encrypted",
2778                       bdrv_get_device_or_node_name(bs),
2779                       bdrv_get_encrypted_filename(bs));
2780         }
2781     }
2782 }
2783
2784 const char *bdrv_get_format_name(BlockDriverState *bs)
2785 {
2786     return bs->drv ? bs->drv->format_name : NULL;
2787 }
2788
2789 static int qsort_strcmp(const void *a, const void *b)
2790 {
2791     return strcmp(a, b);
2792 }
2793
2794 void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2795                          void *opaque)
2796 {
2797     BlockDriver *drv;
2798     int count = 0;
2799     int i;
2800     const char **formats = NULL;
2801
2802     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2803         if (drv->format_name) {
2804             bool found = false;
2805             int i = count;
2806             while (formats && i && !found) {
2807                 found = !strcmp(formats[--i], drv->format_name);
2808             }
2809
2810             if (!found) {
2811                 formats = g_renew(const char *, formats, count + 1);
2812                 formats[count++] = drv->format_name;
2813             }
2814         }
2815     }
2816
2817     qsort(formats, count, sizeof(formats[0]), qsort_strcmp);
2818
2819     for (i = 0; i < count; i++) {
2820         it(opaque, formats[i]);
2821     }
2822
2823     g_free(formats);
2824 }
2825
2826 /* This function is to find a node in the bs graph */
2827 BlockDriverState *bdrv_find_node(const char *node_name)
2828 {
2829     BlockDriverState *bs;
2830
2831     assert(node_name);
2832
2833     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2834         if (!strcmp(node_name, bs->node_name)) {
2835             return bs;
2836         }
2837     }
2838     return NULL;
2839 }
2840
2841 /* Put this QMP function here so it can access the static graph_bdrv_states. */
2842 BlockDeviceInfoList *bdrv_named_nodes_list(Error **errp)
2843 {
2844     BlockDeviceInfoList *list, *entry;
2845     BlockDriverState *bs;
2846
2847     list = NULL;
2848     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2849         BlockDeviceInfo *info = bdrv_block_device_info(bs, errp);
2850         if (!info) {
2851             qapi_free_BlockDeviceInfoList(list);
2852             return NULL;
2853         }
2854         entry = g_malloc0(sizeof(*entry));
2855         entry->value = info;
2856         entry->next = list;
2857         list = entry;
2858     }
2859
2860     return list;
2861 }
2862
2863 BlockDriverState *bdrv_lookup_bs(const char *device,
2864                                  const char *node_name,
2865                                  Error **errp)
2866 {
2867     BlockBackend *blk;
2868     BlockDriverState *bs;
2869
2870     if (device) {
2871         blk = blk_by_name(device);
2872
2873         if (blk) {
2874             return blk_bs(blk);
2875         }
2876     }
2877
2878     if (node_name) {
2879         bs = bdrv_find_node(node_name);
2880
2881         if (bs) {
2882             return bs;
2883         }
2884     }
2885
2886     error_setg(errp, "Cannot find device=%s nor node_name=%s",
2887                      device ? device : "",
2888                      node_name ? node_name : "");
2889     return NULL;
2890 }
2891
2892 /* If 'base' is in the same chain as 'top', return true. Otherwise,
2893  * return false.  If either argument is NULL, return false. */
2894 bool bdrv_chain_contains(BlockDriverState *top, BlockDriverState *base)
2895 {
2896     while (top && top != base) {
2897         top = top->backing_hd;
2898     }
2899
2900     return top != NULL;
2901 }
2902
2903 BlockDriverState *bdrv_next_node(BlockDriverState *bs)
2904 {
2905     if (!bs) {
2906         return QTAILQ_FIRST(&graph_bdrv_states);
2907     }
2908     return QTAILQ_NEXT(bs, node_list);
2909 }
2910
2911 BlockDriverState *bdrv_next(BlockDriverState *bs)
2912 {
2913     if (!bs) {
2914         return QTAILQ_FIRST(&bdrv_states);
2915     }
2916     return QTAILQ_NEXT(bs, device_list);
2917 }
2918
2919 const char *bdrv_get_node_name(const BlockDriverState *bs)
2920 {
2921     return bs->node_name;
2922 }
2923
2924 /* TODO check what callers really want: bs->node_name or blk_name() */
2925 const char *bdrv_get_device_name(const BlockDriverState *bs)
2926 {
2927     return bs->blk ? blk_name(bs->blk) : "";
2928 }
2929
2930 /* This can be used to identify nodes that might not have a device
2931  * name associated. Since node and device names live in the same
2932  * namespace, the result is unambiguous. The exception is if both are
2933  * absent, then this returns an empty (non-null) string. */
2934 const char *bdrv_get_device_or_node_name(const BlockDriverState *bs)
2935 {
2936     return bs->blk ? blk_name(bs->blk) : bs->node_name;
2937 }
2938
2939 int bdrv_get_flags(BlockDriverState *bs)
2940 {
2941     return bs->open_flags;
2942 }
2943
2944 int bdrv_has_zero_init_1(BlockDriverState *bs)
2945 {
2946     return 1;
2947 }
2948
2949 int bdrv_has_zero_init(BlockDriverState *bs)
2950 {
2951     assert(bs->drv);
2952
2953     /* If BS is a copy on write image, it is initialized to
2954        the contents of the base image, which may not be zeroes.  */
2955     if (bs->backing_hd) {
2956         return 0;
2957     }
2958     if (bs->drv->bdrv_has_zero_init) {
2959         return bs->drv->bdrv_has_zero_init(bs);
2960     }
2961
2962     /* safe default */
2963     return 0;
2964 }
2965
2966 bool bdrv_unallocated_blocks_are_zero(BlockDriverState *bs)
2967 {
2968     BlockDriverInfo bdi;
2969
2970     if (bs->backing_hd) {
2971         return false;
2972     }
2973
2974     if (bdrv_get_info(bs, &bdi) == 0) {
2975         return bdi.unallocated_blocks_are_zero;
2976     }
2977
2978     return false;
2979 }
2980
2981 bool bdrv_can_write_zeroes_with_unmap(BlockDriverState *bs)
2982 {
2983     BlockDriverInfo bdi;
2984
2985     if (bs->backing_hd || !(bs->open_flags & BDRV_O_UNMAP)) {
2986         return false;
2987     }
2988
2989     if (bdrv_get_info(bs, &bdi) == 0) {
2990         return bdi.can_write_zeroes_with_unmap;
2991     }
2992
2993     return false;
2994 }
2995
2996 const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
2997 {
2998     if (bs->backing_hd && bs->backing_hd->encrypted)
2999         return bs->backing_file;
3000     else if (bs->encrypted)
3001         return bs->filename;
3002     else
3003         return NULL;
3004 }
3005
3006 void bdrv_get_backing_filename(BlockDriverState *bs,
3007                                char *filename, int filename_size)
3008 {
3009     pstrcpy(filename, filename_size, bs->backing_file);
3010 }
3011
3012 int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3013 {
3014     BlockDriver *drv = bs->drv;
3015     if (!drv)
3016         return -ENOMEDIUM;
3017     if (!drv->bdrv_get_info)
3018         return -ENOTSUP;
3019     memset(bdi, 0, sizeof(*bdi));
3020     return drv->bdrv_get_info(bs, bdi);
3021 }
3022
3023 ImageInfoSpecific *bdrv_get_specific_info(BlockDriverState *bs)
3024 {
3025     BlockDriver *drv = bs->drv;
3026     if (drv && drv->bdrv_get_specific_info) {
3027         return drv->bdrv_get_specific_info(bs);
3028     }
3029     return NULL;
3030 }
3031
3032 void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
3033 {
3034     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
3035         return;
3036     }
3037
3038     bs->drv->bdrv_debug_event(bs, event);
3039 }
3040
3041 int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
3042                           const char *tag)
3043 {
3044     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
3045         bs = bs->file;
3046     }
3047
3048     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
3049         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
3050     }
3051
3052     return -ENOTSUP;
3053 }
3054
3055 int bdrv_debug_remove_breakpoint(BlockDriverState *bs, const char *tag)
3056 {
3057     while (bs && bs->drv && !bs->drv->bdrv_debug_remove_breakpoint) {
3058         bs = bs->file;
3059     }
3060
3061     if (bs && bs->drv && bs->drv->bdrv_debug_remove_breakpoint) {
3062         return bs->drv->bdrv_debug_remove_breakpoint(bs, tag);
3063     }
3064
3065     return -ENOTSUP;
3066 }
3067
3068 int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
3069 {
3070     while (bs && (!bs->drv || !bs->drv->bdrv_debug_resume)) {
3071         bs = bs->file;
3072     }
3073
3074     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
3075         return bs->drv->bdrv_debug_resume(bs, tag);
3076     }
3077
3078     return -ENOTSUP;
3079 }
3080
3081 bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
3082 {
3083     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
3084         bs = bs->file;
3085     }
3086
3087     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
3088         return bs->drv->bdrv_debug_is_suspended(bs, tag);
3089     }
3090
3091     return false;
3092 }
3093
3094 int bdrv_is_snapshot(BlockDriverState *bs)
3095 {
3096     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3097 }
3098
3099 /* backing_file can either be relative, or absolute, or a protocol.  If it is
3100  * relative, it must be relative to the chain.  So, passing in bs->filename
3101  * from a BDS as backing_file should not be done, as that may be relative to
3102  * the CWD rather than the chain. */
3103 BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3104         const char *backing_file)
3105 {
3106     char *filename_full = NULL;
3107     char *backing_file_full = NULL;
3108     char *filename_tmp = NULL;
3109     int is_protocol = 0;
3110     BlockDriverState *curr_bs = NULL;
3111     BlockDriverState *retval = NULL;
3112
3113     if (!bs || !bs->drv || !backing_file) {
3114         return NULL;
3115     }
3116
3117     filename_full     = g_malloc(PATH_MAX);
3118     backing_file_full = g_malloc(PATH_MAX);
3119     filename_tmp      = g_malloc(PATH_MAX);
3120
3121     is_protocol = path_has_protocol(backing_file);
3122
3123     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3124
3125         /* If either of the filename paths is actually a protocol, then
3126          * compare unmodified paths; otherwise make paths relative */
3127         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3128             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3129                 retval = curr_bs->backing_hd;
3130                 break;
3131             }
3132         } else {
3133             /* If not an absolute filename path, make it relative to the current
3134              * image's filename path */
3135             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3136                          backing_file);
3137
3138             /* We are going to compare absolute pathnames */
3139             if (!realpath(filename_tmp, filename_full)) {
3140                 continue;
3141             }
3142
3143             /* We need to make sure the backing filename we are comparing against
3144              * is relative to the current image filename (or absolute) */
3145             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3146                          curr_bs->backing_file);
3147
3148             if (!realpath(filename_tmp, backing_file_full)) {
3149                 continue;
3150             }
3151
3152             if (strcmp(backing_file_full, filename_full) == 0) {
3153                 retval = curr_bs->backing_hd;
3154                 break;
3155             }
3156         }
3157     }
3158
3159     g_free(filename_full);
3160     g_free(backing_file_full);
3161     g_free(filename_tmp);
3162     return retval;
3163 }
3164
3165 int bdrv_get_backing_file_depth(BlockDriverState *bs)
3166 {
3167     if (!bs->drv) {
3168         return 0;
3169     }
3170
3171     if (!bs->backing_hd) {
3172         return 0;
3173     }
3174
3175     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3176 }
3177
3178 void bdrv_init(void)
3179 {
3180     module_call_init(MODULE_INIT_BLOCK);
3181 }
3182
3183 void bdrv_init_with_whitelist(void)
3184 {
3185     use_bdrv_whitelist = 1;
3186     bdrv_init();
3187 }
3188
3189 void bdrv_invalidate_cache(BlockDriverState *bs, Error **errp)
3190 {
3191     Error *local_err = NULL;
3192     int ret;
3193
3194     if (!bs->drv)  {
3195         return;
3196     }
3197
3198     if (!(bs->open_flags & BDRV_O_INCOMING)) {
3199         return;
3200     }
3201     bs->open_flags &= ~BDRV_O_INCOMING;
3202
3203     if (bs->drv->bdrv_invalidate_cache) {
3204         bs->drv->bdrv_invalidate_cache(bs, &local_err);
3205     } else if (bs->file) {
3206         bdrv_invalidate_cache(bs->file, &local_err);
3207     }
3208     if (local_err) {
3209         error_propagate(errp, local_err);
3210         return;
3211     }
3212
3213     ret = refresh_total_sectors(bs, bs->total_sectors);
3214     if (ret < 0) {
3215         error_setg_errno(errp, -ret, "Could not refresh total sector count");
3216         return;
3217     }
3218 }
3219
3220 void bdrv_invalidate_cache_all(Error **errp)
3221 {
3222     BlockDriverState *bs;
3223     Error *local_err = NULL;
3224
3225     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
3226         AioContext *aio_context = bdrv_get_aio_context(bs);
3227
3228         aio_context_acquire(aio_context);
3229         bdrv_invalidate_cache(bs, &local_err);
3230         aio_context_release(aio_context);
3231         if (local_err) {
3232             error_propagate(errp, local_err);
3233             return;
3234         }
3235     }
3236 }
3237
3238 /**************************************************************/
3239 /* removable device support */
3240
3241 /**
3242  * Return TRUE if the media is present
3243  */
3244 int bdrv_is_inserted(BlockDriverState *bs)
3245 {
3246     BlockDriver *drv = bs->drv;
3247
3248     if (!drv)
3249         return 0;
3250     if (!drv->bdrv_is_inserted)
3251         return 1;
3252     return drv->bdrv_is_inserted(bs);
3253 }
3254
3255 /**
3256  * Return whether the media changed since the last call to this
3257  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
3258  */
3259 int bdrv_media_changed(BlockDriverState *bs)
3260 {
3261     BlockDriver *drv = bs->drv;
3262
3263     if (drv && drv->bdrv_media_changed) {
3264         return drv->bdrv_media_changed(bs);
3265     }
3266     return -ENOTSUP;
3267 }
3268
3269 /**
3270  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
3271  */
3272 void bdrv_eject(BlockDriverState *bs, bool eject_flag)
3273 {
3274     BlockDriver *drv = bs->drv;
3275     const char *device_name;
3276
3277     if (drv && drv->bdrv_eject) {
3278         drv->bdrv_eject(bs, eject_flag);
3279     }
3280
3281     device_name = bdrv_get_device_name(bs);
3282     if (device_name[0] != '\0') {
3283         qapi_event_send_device_tray_moved(device_name,
3284                                           eject_flag, &error_abort);
3285     }
3286 }
3287
3288 /**
3289  * Lock or unlock the media (if it is locked, the user won't be able
3290  * to eject it manually).
3291  */
3292 void bdrv_lock_medium(BlockDriverState *bs, bool locked)
3293 {
3294     BlockDriver *drv = bs->drv;
3295
3296     trace_bdrv_lock_medium(bs, locked);
3297
3298     if (drv && drv->bdrv_lock_medium) {
3299         drv->bdrv_lock_medium(bs, locked);
3300     }
3301 }
3302
3303 void bdrv_set_guest_block_size(BlockDriverState *bs, int align)
3304 {
3305     bs->guest_block_size = align;
3306 }
3307
3308 BdrvDirtyBitmap *bdrv_find_dirty_bitmap(BlockDriverState *bs, const char *name)
3309 {
3310     BdrvDirtyBitmap *bm;
3311
3312     assert(name);
3313     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
3314         if (bm->name && !strcmp(name, bm->name)) {
3315             return bm;
3316         }
3317     }
3318     return NULL;
3319 }
3320
3321 void bdrv_dirty_bitmap_make_anon(BdrvDirtyBitmap *bitmap)
3322 {
3323     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3324     g_free(bitmap->name);
3325     bitmap->name = NULL;
3326 }
3327
3328 BdrvDirtyBitmap *bdrv_create_dirty_bitmap(BlockDriverState *bs,
3329                                           uint32_t granularity,
3330                                           const char *name,
3331                                           Error **errp)
3332 {
3333     int64_t bitmap_size;
3334     BdrvDirtyBitmap *bitmap;
3335     uint32_t sector_granularity;
3336
3337     assert((granularity & (granularity - 1)) == 0);
3338
3339     if (name && bdrv_find_dirty_bitmap(bs, name)) {
3340         error_setg(errp, "Bitmap already exists: %s", name);
3341         return NULL;
3342     }
3343     sector_granularity = granularity >> BDRV_SECTOR_BITS;
3344     assert(sector_granularity);
3345     bitmap_size = bdrv_nb_sectors(bs);
3346     if (bitmap_size < 0) {
3347         error_setg_errno(errp, -bitmap_size, "could not get length of device");
3348         errno = -bitmap_size;
3349         return NULL;
3350     }
3351     bitmap = g_new0(BdrvDirtyBitmap, 1);
3352     bitmap->bitmap = hbitmap_alloc(bitmap_size, ctz32(sector_granularity));
3353     bitmap->size = bitmap_size;
3354     bitmap->name = g_strdup(name);
3355     bitmap->disabled = false;
3356     QLIST_INSERT_HEAD(&bs->dirty_bitmaps, bitmap, list);
3357     return bitmap;
3358 }
3359
3360 bool bdrv_dirty_bitmap_frozen(BdrvDirtyBitmap *bitmap)
3361 {
3362     return bitmap->successor;
3363 }
3364
3365 bool bdrv_dirty_bitmap_enabled(BdrvDirtyBitmap *bitmap)
3366 {
3367     return !(bitmap->disabled || bitmap->successor);
3368 }
3369
3370 DirtyBitmapStatus bdrv_dirty_bitmap_status(BdrvDirtyBitmap *bitmap)
3371 {
3372     if (bdrv_dirty_bitmap_frozen(bitmap)) {
3373         return DIRTY_BITMAP_STATUS_FROZEN;
3374     } else if (!bdrv_dirty_bitmap_enabled(bitmap)) {
3375         return DIRTY_BITMAP_STATUS_DISABLED;
3376     } else {
3377         return DIRTY_BITMAP_STATUS_ACTIVE;
3378     }
3379 }
3380
3381 /**
3382  * Create a successor bitmap destined to replace this bitmap after an operation.
3383  * Requires that the bitmap is not frozen and has no successor.
3384  */
3385 int bdrv_dirty_bitmap_create_successor(BlockDriverState *bs,
3386                                        BdrvDirtyBitmap *bitmap, Error **errp)
3387 {
3388     uint64_t granularity;
3389     BdrvDirtyBitmap *child;
3390
3391     if (bdrv_dirty_bitmap_frozen(bitmap)) {
3392         error_setg(errp, "Cannot create a successor for a bitmap that is "
3393                    "currently frozen");
3394         return -1;
3395     }
3396     assert(!bitmap->successor);
3397
3398     /* Create an anonymous successor */
3399     granularity = bdrv_dirty_bitmap_granularity(bitmap);
3400     child = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp);
3401     if (!child) {
3402         return -1;
3403     }
3404
3405     /* Successor will be on or off based on our current state. */
3406     child->disabled = bitmap->disabled;
3407
3408     /* Install the successor and freeze the parent */
3409     bitmap->successor = child;
3410     return 0;
3411 }
3412
3413 /**
3414  * For a bitmap with a successor, yield our name to the successor,
3415  * delete the old bitmap, and return a handle to the new bitmap.
3416  */
3417 BdrvDirtyBitmap *bdrv_dirty_bitmap_abdicate(BlockDriverState *bs,
3418                                             BdrvDirtyBitmap *bitmap,
3419                                             Error **errp)
3420 {
3421     char *name;
3422     BdrvDirtyBitmap *successor = bitmap->successor;
3423
3424     if (successor == NULL) {
3425         error_setg(errp, "Cannot relinquish control if "
3426                    "there's no successor present");
3427         return NULL;
3428     }
3429
3430     name = bitmap->name;
3431     bitmap->name = NULL;
3432     successor->name = name;
3433     bitmap->successor = NULL;
3434     bdrv_release_dirty_bitmap(bs, bitmap);
3435
3436     return successor;
3437 }
3438
3439 /**
3440  * In cases of failure where we can no longer safely delete the parent,
3441  * we may wish to re-join the parent and child/successor.
3442  * The merged parent will be un-frozen, but not explicitly re-enabled.
3443  */
3444 BdrvDirtyBitmap *bdrv_reclaim_dirty_bitmap(BlockDriverState *bs,
3445                                            BdrvDirtyBitmap *parent,
3446                                            Error **errp)
3447 {
3448     BdrvDirtyBitmap *successor = parent->successor;
3449
3450     if (!successor) {
3451         error_setg(errp, "Cannot reclaim a successor when none is present");
3452         return NULL;
3453     }
3454
3455     if (!hbitmap_merge(parent->bitmap, successor->bitmap)) {
3456         error_setg(errp, "Merging of parent and successor bitmap failed");
3457         return NULL;
3458     }
3459     bdrv_release_dirty_bitmap(bs, successor);
3460     parent->successor = NULL;
3461
3462     return parent;
3463 }
3464
3465 /**
3466  * Truncates _all_ bitmaps attached to a BDS.
3467  */
3468 static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs)
3469 {
3470     BdrvDirtyBitmap *bitmap;
3471     uint64_t size = bdrv_nb_sectors(bs);
3472
3473     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
3474         assert(!bdrv_dirty_bitmap_frozen(bitmap));
3475         hbitmap_truncate(bitmap->bitmap, size);
3476         bitmap->size = size;
3477     }
3478 }
3479
3480 void bdrv_release_dirty_bitmap(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
3481 {
3482     BdrvDirtyBitmap *bm, *next;
3483     QLIST_FOREACH_SAFE(bm, &bs->dirty_bitmaps, list, next) {
3484         if (bm == bitmap) {
3485             assert(!bdrv_dirty_bitmap_frozen(bm));
3486             QLIST_REMOVE(bitmap, list);
3487             hbitmap_free(bitmap->bitmap);
3488             g_free(bitmap->name);
3489             g_free(bitmap);
3490             return;
3491         }
3492     }
3493 }
3494
3495 void bdrv_disable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3496 {
3497     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3498     bitmap->disabled = true;
3499 }
3500
3501 void bdrv_enable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3502 {
3503     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3504     bitmap->disabled = false;
3505 }
3506
3507 BlockDirtyInfoList *bdrv_query_dirty_bitmaps(BlockDriverState *bs)
3508 {
3509     BdrvDirtyBitmap *bm;
3510     BlockDirtyInfoList *list = NULL;
3511     BlockDirtyInfoList **plist = &list;
3512
3513     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
3514         BlockDirtyInfo *info = g_new0(BlockDirtyInfo, 1);
3515         BlockDirtyInfoList *entry = g_new0(BlockDirtyInfoList, 1);
3516         info->count = bdrv_get_dirty_count(bm);
3517         info->granularity = bdrv_dirty_bitmap_granularity(bm);
3518         info->has_name = !!bm->name;
3519         info->name = g_strdup(bm->name);
3520         info->status = bdrv_dirty_bitmap_status(bm);
3521         entry->value = info;
3522         *plist = entry;
3523         plist = &entry->next;
3524     }
3525
3526     return list;
3527 }
3528
3529 int bdrv_get_dirty(BlockDriverState *bs, BdrvDirtyBitmap *bitmap, int64_t sector)
3530 {
3531     if (bitmap) {
3532         return hbitmap_get(bitmap->bitmap, sector);
3533     } else {
3534         return 0;
3535     }
3536 }
3537
3538 /**
3539  * Chooses a default granularity based on the existing cluster size,
3540  * but clamped between [4K, 64K]. Defaults to 64K in the case that there
3541  * is no cluster size information available.
3542  */
3543 uint32_t bdrv_get_default_bitmap_granularity(BlockDriverState *bs)
3544 {
3545     BlockDriverInfo bdi;
3546     uint32_t granularity;
3547
3548     if (bdrv_get_info(bs, &bdi) >= 0 && bdi.cluster_size > 0) {
3549         granularity = MAX(4096, bdi.cluster_size);
3550         granularity = MIN(65536, granularity);
3551     } else {
3552         granularity = 65536;
3553     }
3554
3555     return granularity;
3556 }
3557
3558 uint32_t bdrv_dirty_bitmap_granularity(BdrvDirtyBitmap *bitmap)
3559 {
3560     return BDRV_SECTOR_SIZE << hbitmap_granularity(bitmap->bitmap);
3561 }
3562
3563 void bdrv_dirty_iter_init(BdrvDirtyBitmap *bitmap, HBitmapIter *hbi)
3564 {
3565     hbitmap_iter_init(hbi, bitmap->bitmap, 0);
3566 }
3567
3568 void bdrv_set_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3569                            int64_t cur_sector, int nr_sectors)
3570 {
3571     assert(bdrv_dirty_bitmap_enabled(bitmap));
3572     hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3573 }
3574
3575 void bdrv_reset_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3576                              int64_t cur_sector, int nr_sectors)
3577 {
3578     assert(bdrv_dirty_bitmap_enabled(bitmap));
3579     hbitmap_reset(bitmap->bitmap, cur_sector, nr_sectors);
3580 }
3581
3582 void bdrv_clear_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3583 {
3584     assert(bdrv_dirty_bitmap_enabled(bitmap));
3585     hbitmap_reset_all(bitmap->bitmap);
3586 }
3587
3588 void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
3589                     int nr_sectors)
3590 {
3591     BdrvDirtyBitmap *bitmap;
3592     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
3593         if (!bdrv_dirty_bitmap_enabled(bitmap)) {
3594             continue;
3595         }
3596         hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3597     }
3598 }
3599
3600 /**
3601  * Advance an HBitmapIter to an arbitrary offset.
3602  */
3603 void bdrv_set_dirty_iter(HBitmapIter *hbi, int64_t offset)
3604 {
3605     assert(hbi->hb);
3606     hbitmap_iter_init(hbi, hbi->hb, offset);
3607 }
3608
3609 int64_t bdrv_get_dirty_count(BdrvDirtyBitmap *bitmap)
3610 {
3611     return hbitmap_count(bitmap->bitmap);
3612 }
3613
3614 /* Get a reference to bs */
3615 void bdrv_ref(BlockDriverState *bs)
3616 {
3617     bs->refcnt++;
3618 }
3619
3620 /* Release a previously grabbed reference to bs.
3621  * If after releasing, reference count is zero, the BlockDriverState is
3622  * deleted. */
3623 void bdrv_unref(BlockDriverState *bs)
3624 {
3625     if (!bs) {
3626         return;
3627     }
3628     assert(bs->refcnt > 0);
3629     if (--bs->refcnt == 0) {
3630         bdrv_delete(bs);
3631     }
3632 }
3633
3634 struct BdrvOpBlocker {
3635     Error *reason;
3636     QLIST_ENTRY(BdrvOpBlocker) list;
3637 };
3638
3639 bool bdrv_op_is_blocked(BlockDriverState *bs, BlockOpType op, Error **errp)
3640 {
3641     BdrvOpBlocker *blocker;
3642     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3643     if (!QLIST_EMPTY(&bs->op_blockers[op])) {
3644         blocker = QLIST_FIRST(&bs->op_blockers[op]);
3645         if (errp) {
3646             error_setg(errp, "Node '%s' is busy: %s",
3647                        bdrv_get_device_or_node_name(bs),
3648                        error_get_pretty(blocker->reason));
3649         }
3650         return true;
3651     }
3652     return false;
3653 }
3654
3655 void bdrv_op_block(BlockDriverState *bs, BlockOpType op, Error *reason)
3656 {
3657     BdrvOpBlocker *blocker;
3658     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3659
3660     blocker = g_new0(BdrvOpBlocker, 1);
3661     blocker->reason = reason;
3662     QLIST_INSERT_HEAD(&bs->op_blockers[op], blocker, list);
3663 }
3664
3665 void bdrv_op_unblock(BlockDriverState *bs, BlockOpType op, Error *reason)
3666 {
3667     BdrvOpBlocker *blocker, *next;
3668     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3669     QLIST_FOREACH_SAFE(blocker, &bs->op_blockers[op], list, next) {
3670         if (blocker->reason == reason) {
3671             QLIST_REMOVE(blocker, list);
3672             g_free(blocker);
3673         }
3674     }
3675 }
3676
3677 void bdrv_op_block_all(BlockDriverState *bs, Error *reason)
3678 {
3679     int i;
3680     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3681         bdrv_op_block(bs, i, reason);
3682     }
3683 }
3684
3685 void bdrv_op_unblock_all(BlockDriverState *bs, Error *reason)
3686 {
3687     int i;
3688     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3689         bdrv_op_unblock(bs, i, reason);
3690     }
3691 }
3692
3693 bool bdrv_op_blocker_is_empty(BlockDriverState *bs)
3694 {
3695     int i;
3696
3697     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3698         if (!QLIST_EMPTY(&bs->op_blockers[i])) {
3699             return false;
3700         }
3701     }
3702     return true;
3703 }
3704
3705 void bdrv_iostatus_enable(BlockDriverState *bs)
3706 {
3707     bs->iostatus_enabled = true;
3708     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
3709 }
3710
3711 /* The I/O status is only enabled if the drive explicitly
3712  * enables it _and_ the VM is configured to stop on errors */
3713 bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
3714 {
3715     return (bs->iostatus_enabled &&
3716            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
3717             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
3718             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
3719 }
3720
3721 void bdrv_iostatus_disable(BlockDriverState *bs)
3722 {
3723     bs->iostatus_enabled = false;
3724 }
3725
3726 void bdrv_iostatus_reset(BlockDriverState *bs)
3727 {
3728     if (bdrv_iostatus_is_enabled(bs)) {
3729         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
3730         if (bs->job) {
3731             block_job_iostatus_reset(bs->job);
3732         }
3733     }
3734 }
3735
3736 void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
3737 {
3738     assert(bdrv_iostatus_is_enabled(bs));
3739     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
3740         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
3741                                          BLOCK_DEVICE_IO_STATUS_FAILED;
3742     }
3743 }
3744
3745 void bdrv_img_create(const char *filename, const char *fmt,
3746                      const char *base_filename, const char *base_fmt,
3747                      char *options, uint64_t img_size, int flags,
3748                      Error **errp, bool quiet)
3749 {
3750     QemuOptsList *create_opts = NULL;
3751     QemuOpts *opts = NULL;
3752     const char *backing_fmt, *backing_file;
3753     int64_t size;
3754     BlockDriver *drv, *proto_drv;
3755     Error *local_err = NULL;
3756     int ret = 0;
3757
3758     /* Find driver and parse its options */
3759     drv = bdrv_find_format(fmt);
3760     if (!drv) {
3761         error_setg(errp, "Unknown file format '%s'", fmt);
3762         return;
3763     }
3764
3765     proto_drv = bdrv_find_protocol(filename, true, errp);
3766     if (!proto_drv) {
3767         return;
3768     }
3769
3770     if (!drv->create_opts) {
3771         error_setg(errp, "Format driver '%s' does not support image creation",
3772                    drv->format_name);
3773         return;
3774     }
3775
3776     if (!proto_drv->create_opts) {
3777         error_setg(errp, "Protocol driver '%s' does not support image creation",
3778                    proto_drv->format_name);
3779         return;
3780     }
3781
3782     create_opts = qemu_opts_append(create_opts, drv->create_opts);
3783     create_opts = qemu_opts_append(create_opts, proto_drv->create_opts);
3784
3785     /* Create parameter list with default values */
3786     opts = qemu_opts_create(create_opts, NULL, 0, &error_abort);
3787     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, img_size, &error_abort);
3788
3789     /* Parse -o options */
3790     if (options) {
3791         qemu_opts_do_parse(opts, options, NULL, &local_err);
3792         if (local_err) {
3793             error_report_err(local_err);
3794             local_err = NULL;
3795             error_setg(errp, "Invalid options for file format '%s'", fmt);
3796             goto out;
3797         }
3798     }
3799
3800     if (base_filename) {
3801         qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, base_filename, &local_err);
3802         if (local_err) {
3803             error_setg(errp, "Backing file not supported for file format '%s'",
3804                        fmt);
3805             goto out;
3806         }
3807     }
3808
3809     if (base_fmt) {
3810         qemu_opt_set(opts, BLOCK_OPT_BACKING_FMT, base_fmt, &local_err);
3811         if (local_err) {
3812             error_setg(errp, "Backing file format not supported for file "
3813                              "format '%s'", fmt);
3814             goto out;
3815         }
3816     }
3817
3818     backing_file = qemu_opt_get(opts, BLOCK_OPT_BACKING_FILE);
3819     if (backing_file) {
3820         if (!strcmp(filename, backing_file)) {
3821             error_setg(errp, "Error: Trying to create an image with the "
3822                              "same filename as the backing file");
3823             goto out;
3824         }
3825     }
3826
3827     backing_fmt = qemu_opt_get(opts, BLOCK_OPT_BACKING_FMT);
3828
3829     // The size for the image must always be specified, with one exception:
3830     // If we are using a backing file, we can obtain the size from there
3831     size = qemu_opt_get_size(opts, BLOCK_OPT_SIZE, 0);
3832     if (size == -1) {
3833         if (backing_file) {
3834             BlockDriverState *bs;
3835             char *full_backing = g_new0(char, PATH_MAX);
3836             int64_t size;
3837             int back_flags;
3838             QDict *backing_options = NULL;
3839
3840             bdrv_get_full_backing_filename_from_filename(filename, backing_file,
3841                                                          full_backing, PATH_MAX,
3842                                                          &local_err);
3843             if (local_err) {
3844                 g_free(full_backing);
3845                 goto out;
3846             }
3847
3848             /* backing files always opened read-only */
3849             back_flags =
3850                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
3851
3852             if (backing_fmt) {
3853                 backing_options = qdict_new();
3854                 qdict_put(backing_options, "driver",
3855                           qstring_from_str(backing_fmt));
3856             }
3857
3858             bs = NULL;
3859             ret = bdrv_open(&bs, full_backing, NULL, backing_options,
3860                             back_flags, &local_err);
3861             g_free(full_backing);
3862             if (ret < 0) {
3863                 goto out;
3864             }
3865             size = bdrv_getlength(bs);
3866             if (size < 0) {
3867                 error_setg_errno(errp, -size, "Could not get size of '%s'",
3868                                  backing_file);
3869                 bdrv_unref(bs);
3870                 goto out;
3871             }
3872
3873             qemu_opt_set_number(opts, BLOCK_OPT_SIZE, size, &error_abort);
3874
3875             bdrv_unref(bs);
3876         } else {
3877             error_setg(errp, "Image creation needs a size parameter");
3878             goto out;
3879         }
3880     }
3881
3882     if (!quiet) {
3883         printf("Formatting '%s', fmt=%s ", filename, fmt);
3884         qemu_opts_print(opts, " ");
3885         puts("");
3886     }
3887
3888     ret = bdrv_create(drv, filename, opts, &local_err);
3889
3890     if (ret == -EFBIG) {
3891         /* This is generally a better message than whatever the driver would
3892          * deliver (especially because of the cluster_size_hint), since that
3893          * is most probably not much different from "image too large". */
3894         const char *cluster_size_hint = "";
3895         if (qemu_opt_get_size(opts, BLOCK_OPT_CLUSTER_SIZE, 0)) {
3896             cluster_size_hint = " (try using a larger cluster size)";
3897         }
3898         error_setg(errp, "The image size is too large for file format '%s'"
3899                    "%s", fmt, cluster_size_hint);
3900         error_free(local_err);
3901         local_err = NULL;
3902     }
3903
3904 out:
3905     qemu_opts_del(opts);
3906     qemu_opts_free(create_opts);
3907     if (local_err) {
3908         error_propagate(errp, local_err);
3909     }
3910 }
3911
3912 AioContext *bdrv_get_aio_context(BlockDriverState *bs)
3913 {
3914     return bs->aio_context;
3915 }
3916
3917 void bdrv_detach_aio_context(BlockDriverState *bs)
3918 {
3919     BdrvAioNotifier *baf;
3920
3921     if (!bs->drv) {
3922         return;
3923     }
3924
3925     QLIST_FOREACH(baf, &bs->aio_notifiers, list) {
3926         baf->detach_aio_context(baf->opaque);
3927     }
3928
3929     if (bs->io_limits_enabled) {
3930         throttle_timers_detach_aio_context(&bs->throttle_timers);
3931     }
3932     if (bs->drv->bdrv_detach_aio_context) {
3933         bs->drv->bdrv_detach_aio_context(bs);
3934     }
3935     if (bs->file) {
3936         bdrv_detach_aio_context(bs->file);
3937     }
3938     if (bs->backing_hd) {
3939         bdrv_detach_aio_context(bs->backing_hd);
3940     }
3941
3942     bs->aio_context = NULL;
3943 }
3944
3945 void bdrv_attach_aio_context(BlockDriverState *bs,
3946                              AioContext *new_context)
3947 {
3948     BdrvAioNotifier *ban;
3949
3950     if (!bs->drv) {
3951         return;
3952     }
3953
3954     bs->aio_context = new_context;
3955
3956     if (bs->backing_hd) {
3957         bdrv_attach_aio_context(bs->backing_hd, new_context);
3958     }
3959     if (bs->file) {
3960         bdrv_attach_aio_context(bs->file, new_context);
3961     }
3962     if (bs->drv->bdrv_attach_aio_context) {
3963         bs->drv->bdrv_attach_aio_context(bs, new_context);
3964     }
3965     if (bs->io_limits_enabled) {
3966         throttle_timers_attach_aio_context(&bs->throttle_timers, new_context);
3967     }
3968
3969     QLIST_FOREACH(ban, &bs->aio_notifiers, list) {
3970         ban->attached_aio_context(new_context, ban->opaque);
3971     }
3972 }
3973
3974 void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
3975 {
3976     bdrv_drain(bs); /* ensure there are no in-flight requests */
3977
3978     bdrv_detach_aio_context(bs);
3979
3980     /* This function executes in the old AioContext so acquire the new one in
3981      * case it runs in a different thread.
3982      */
3983     aio_context_acquire(new_context);
3984     bdrv_attach_aio_context(bs, new_context);
3985     aio_context_release(new_context);
3986 }
3987
3988 void bdrv_add_aio_context_notifier(BlockDriverState *bs,
3989         void (*attached_aio_context)(AioContext *new_context, void *opaque),
3990         void (*detach_aio_context)(void *opaque), void *opaque)
3991 {
3992     BdrvAioNotifier *ban = g_new(BdrvAioNotifier, 1);
3993     *ban = (BdrvAioNotifier){
3994         .attached_aio_context = attached_aio_context,
3995         .detach_aio_context   = detach_aio_context,
3996         .opaque               = opaque
3997     };
3998
3999     QLIST_INSERT_HEAD(&bs->aio_notifiers, ban, list);
4000 }
4001
4002 void bdrv_remove_aio_context_notifier(BlockDriverState *bs,
4003                                       void (*attached_aio_context)(AioContext *,
4004                                                                    void *),
4005                                       void (*detach_aio_context)(void *),
4006                                       void *opaque)
4007 {
4008     BdrvAioNotifier *ban, *ban_next;
4009
4010     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
4011         if (ban->attached_aio_context == attached_aio_context &&
4012             ban->detach_aio_context   == detach_aio_context   &&
4013             ban->opaque               == opaque)
4014         {
4015             QLIST_REMOVE(ban, list);
4016             g_free(ban);
4017
4018             return;
4019         }
4020     }
4021
4022     abort();
4023 }
4024
4025 int bdrv_amend_options(BlockDriverState *bs, QemuOpts *opts,
4026                        BlockDriverAmendStatusCB *status_cb)
4027 {
4028     if (!bs->drv->bdrv_amend_options) {
4029         return -ENOTSUP;
4030     }
4031     return bs->drv->bdrv_amend_options(bs, opts, status_cb);
4032 }
4033
4034 /* This function will be called by the bdrv_recurse_is_first_non_filter method
4035  * of block filter and by bdrv_is_first_non_filter.
4036  * It is used to test if the given bs is the candidate or recurse more in the
4037  * node graph.
4038  */
4039 bool bdrv_recurse_is_first_non_filter(BlockDriverState *bs,
4040                                       BlockDriverState *candidate)
4041 {
4042     /* return false if basic checks fails */
4043     if (!bs || !bs->drv) {
4044         return false;
4045     }
4046
4047     /* the code reached a non block filter driver -> check if the bs is
4048      * the same as the candidate. It's the recursion termination condition.
4049      */
4050     if (!bs->drv->is_filter) {
4051         return bs == candidate;
4052     }
4053     /* Down this path the driver is a block filter driver */
4054
4055     /* If the block filter recursion method is defined use it to recurse down
4056      * the node graph.
4057      */
4058     if (bs->drv->bdrv_recurse_is_first_non_filter) {
4059         return bs->drv->bdrv_recurse_is_first_non_filter(bs, candidate);
4060     }
4061
4062     /* the driver is a block filter but don't allow to recurse -> return false
4063      */
4064     return false;
4065 }
4066
4067 /* This function checks if the candidate is the first non filter bs down it's
4068  * bs chain. Since we don't have pointers to parents it explore all bs chains
4069  * from the top. Some filters can choose not to pass down the recursion.
4070  */
4071 bool bdrv_is_first_non_filter(BlockDriverState *candidate)
4072 {
4073     BlockDriverState *bs;
4074
4075     /* walk down the bs forest recursively */
4076     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
4077         bool perm;
4078
4079         /* try to recurse in this top level bs */
4080         perm = bdrv_recurse_is_first_non_filter(bs, candidate);
4081
4082         /* candidate is the first non filter */
4083         if (perm) {
4084             return true;
4085         }
4086     }
4087
4088     return false;
4089 }
4090
4091 BlockDriverState *check_to_replace_node(BlockDriverState *parent_bs,
4092                                         const char *node_name, Error **errp)
4093 {
4094     BlockDriverState *to_replace_bs = bdrv_find_node(node_name);
4095     AioContext *aio_context;
4096
4097     if (!to_replace_bs) {
4098         error_setg(errp, "Node name '%s' not found", node_name);
4099         return NULL;
4100     }
4101
4102     aio_context = bdrv_get_aio_context(to_replace_bs);
4103     aio_context_acquire(aio_context);
4104
4105     if (bdrv_op_is_blocked(to_replace_bs, BLOCK_OP_TYPE_REPLACE, errp)) {
4106         to_replace_bs = NULL;
4107         goto out;
4108     }
4109
4110     /* We don't want arbitrary node of the BDS chain to be replaced only the top
4111      * most non filter in order to prevent data corruption.
4112      * Another benefit is that this tests exclude backing files which are
4113      * blocked by the backing blockers.
4114      */
4115     if (!bdrv_recurse_is_first_non_filter(parent_bs, to_replace_bs)) {
4116         error_setg(errp, "Only top most non filter can be replaced");
4117         to_replace_bs = NULL;
4118         goto out;
4119     }
4120
4121 out:
4122     aio_context_release(aio_context);
4123     return to_replace_bs;
4124 }
4125
4126 static bool append_open_options(QDict *d, BlockDriverState *bs)
4127 {
4128     const QDictEntry *entry;
4129     bool found_any = false;
4130
4131     for (entry = qdict_first(bs->options); entry;
4132          entry = qdict_next(bs->options, entry))
4133     {
4134         /* Only take options for this level and exclude all non-driver-specific
4135          * options */
4136         if (!strchr(qdict_entry_key(entry), '.') &&
4137             strcmp(qdict_entry_key(entry), "node-name"))
4138         {
4139             qobject_incref(qdict_entry_value(entry));
4140             qdict_put_obj(d, qdict_entry_key(entry), qdict_entry_value(entry));
4141             found_any = true;
4142         }
4143     }
4144
4145     return found_any;
4146 }
4147
4148 /* Updates the following BDS fields:
4149  *  - exact_filename: A filename which may be used for opening a block device
4150  *                    which (mostly) equals the given BDS (even without any
4151  *                    other options; so reading and writing must return the same
4152  *                    results, but caching etc. may be different)
4153  *  - full_open_options: Options which, when given when opening a block device
4154  *                       (without a filename), result in a BDS (mostly)
4155  *                       equalling the given one
4156  *  - filename: If exact_filename is set, it is copied here. Otherwise,
4157  *              full_open_options is converted to a JSON object, prefixed with
4158  *              "json:" (for use through the JSON pseudo protocol) and put here.
4159  */
4160 void bdrv_refresh_filename(BlockDriverState *bs)
4161 {
4162     BlockDriver *drv = bs->drv;
4163     QDict *opts;
4164
4165     if (!drv) {
4166         return;
4167     }
4168
4169     /* This BDS's file name will most probably depend on its file's name, so
4170      * refresh that first */
4171     if (bs->file) {
4172         bdrv_refresh_filename(bs->file);
4173     }
4174
4175     if (drv->bdrv_refresh_filename) {
4176         /* Obsolete information is of no use here, so drop the old file name
4177          * information before refreshing it */
4178         bs->exact_filename[0] = '\0';
4179         if (bs->full_open_options) {
4180             QDECREF(bs->full_open_options);
4181             bs->full_open_options = NULL;
4182         }
4183
4184         drv->bdrv_refresh_filename(bs);
4185     } else if (bs->file) {
4186         /* Try to reconstruct valid information from the underlying file */
4187         bool has_open_options;
4188
4189         bs->exact_filename[0] = '\0';
4190         if (bs->full_open_options) {
4191             QDECREF(bs->full_open_options);
4192             bs->full_open_options = NULL;
4193         }
4194
4195         opts = qdict_new();
4196         has_open_options = append_open_options(opts, bs);
4197
4198         /* If no specific options have been given for this BDS, the filename of
4199          * the underlying file should suffice for this one as well */
4200         if (bs->file->exact_filename[0] && !has_open_options) {
4201             strcpy(bs->exact_filename, bs->file->exact_filename);
4202         }
4203         /* Reconstructing the full options QDict is simple for most format block
4204          * drivers, as long as the full options are known for the underlying
4205          * file BDS. The full options QDict of that file BDS should somehow
4206          * contain a representation of the filename, therefore the following
4207          * suffices without querying the (exact_)filename of this BDS. */
4208         if (bs->file->full_open_options) {
4209             qdict_put_obj(opts, "driver",
4210                           QOBJECT(qstring_from_str(drv->format_name)));
4211             QINCREF(bs->file->full_open_options);
4212             qdict_put_obj(opts, "file", QOBJECT(bs->file->full_open_options));
4213
4214             bs->full_open_options = opts;
4215         } else {
4216             QDECREF(opts);
4217         }
4218     } else if (!bs->full_open_options && qdict_size(bs->options)) {
4219         /* There is no underlying file BDS (at least referenced by BDS.file),
4220          * so the full options QDict should be equal to the options given
4221          * specifically for this block device when it was opened (plus the
4222          * driver specification).
4223          * Because those options don't change, there is no need to update
4224          * full_open_options when it's already set. */
4225
4226         opts = qdict_new();
4227         append_open_options(opts, bs);
4228         qdict_put_obj(opts, "driver",
4229                       QOBJECT(qstring_from_str(drv->format_name)));
4230
4231         if (bs->exact_filename[0]) {
4232             /* This may not work for all block protocol drivers (some may
4233              * require this filename to be parsed), but we have to find some
4234              * default solution here, so just include it. If some block driver
4235              * does not support pure options without any filename at all or
4236              * needs some special format of the options QDict, it needs to
4237              * implement the driver-specific bdrv_refresh_filename() function.
4238              */
4239             qdict_put_obj(opts, "filename",
4240                           QOBJECT(qstring_from_str(bs->exact_filename)));
4241         }
4242
4243         bs->full_open_options = opts;
4244     }
4245
4246     if (bs->exact_filename[0]) {
4247         pstrcpy(bs->filename, sizeof(bs->filename), bs->exact_filename);
4248     } else if (bs->full_open_options) {
4249         QString *json = qobject_to_json(QOBJECT(bs->full_open_options));
4250         snprintf(bs->filename, sizeof(bs->filename), "json:%s",
4251                  qstring_get_str(json));
4252         QDECREF(json);
4253     }
4254 }
4255
4256 /* This accessor function purpose is to allow the device models to access the
4257  * BlockAcctStats structure embedded inside a BlockDriverState without being
4258  * aware of the BlockDriverState structure layout.
4259  * It will go away when the BlockAcctStats structure will be moved inside
4260  * the device models.
4261  */
4262 BlockAcctStats *bdrv_get_stats(BlockDriverState *bs)
4263 {
4264     return &bs->stats;
4265 }