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