xref: /openbmc/qemu/block.c (revision 6ebf9aa2ef7f3e094d91ea27140dc6e73774386a)
1fc01f7e7Sbellard /*
2fc01f7e7Sbellard  * QEMU System Emulator block driver
3fc01f7e7Sbellard  *
4fc01f7e7Sbellard  * Copyright (c) 2003 Fabrice Bellard
5fc01f7e7Sbellard  *
6fc01f7e7Sbellard  * Permission is hereby granted, free of charge, to any person obtaining a copy
7fc01f7e7Sbellard  * of this software and associated documentation files (the "Software"), to deal
8fc01f7e7Sbellard  * in the Software without restriction, including without limitation the rights
9fc01f7e7Sbellard  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10fc01f7e7Sbellard  * copies of the Software, and to permit persons to whom the Software is
11fc01f7e7Sbellard  * furnished to do so, subject to the following conditions:
12fc01f7e7Sbellard  *
13fc01f7e7Sbellard  * The above copyright notice and this permission notice shall be included in
14fc01f7e7Sbellard  * all copies or substantial portions of the Software.
15fc01f7e7Sbellard  *
16fc01f7e7Sbellard  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17fc01f7e7Sbellard  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18fc01f7e7Sbellard  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19fc01f7e7Sbellard  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20fc01f7e7Sbellard  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21fc01f7e7Sbellard  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22fc01f7e7Sbellard  * THE SOFTWARE.
23fc01f7e7Sbellard  */
243990d09aSblueswir1 #include "config-host.h"
25faf07963Spbrook #include "qemu-common.h"
266d519a5fSStefan Hajnoczi #include "trace.h"
27737e150eSPaolo Bonzini #include "block/block_int.h"
28737e150eSPaolo Bonzini #include "block/blockjob.h"
29d49b6836SMarkus Armbruster #include "qemu/error-report.h"
301de7afc9SPaolo Bonzini #include "qemu/module.h"
31cc7a8ea7SMarkus Armbruster #include "qapi/qmp/qerror.h"
327b1b5d19SPaolo Bonzini #include "qapi/qmp/qjson.h"
33bfb197e0SMarkus Armbruster #include "sysemu/block-backend.h"
349c17d615SPaolo Bonzini #include "sysemu/sysemu.h"
351de7afc9SPaolo Bonzini #include "qemu/notify.h"
36737e150eSPaolo Bonzini #include "block/coroutine.h"
37c13163fbSBenoît Canet #include "block/qapi.h"
38b2023818SLuiz Capitulino #include "qmp-commands.h"
391de7afc9SPaolo Bonzini #include "qemu/timer.h"
40a5ee7bd4SWenchao Xia #include "qapi-event.h"
41db628338SAlberto Garcia #include "block/throttle-groups.h"
42fc01f7e7Sbellard 
4371e72a19SJuan Quintela #ifdef CONFIG_BSD
447674e7bfSbellard #include <sys/types.h>
457674e7bfSbellard #include <sys/stat.h>
467674e7bfSbellard #include <sys/ioctl.h>
4772cf2d4fSBlue Swirl #include <sys/queue.h>
48c5e97233Sblueswir1 #ifndef __DragonFly__
497674e7bfSbellard #include <sys/disk.h>
507674e7bfSbellard #endif
51c5e97233Sblueswir1 #endif
527674e7bfSbellard 
5349dc768dSaliguori #ifdef _WIN32
5449dc768dSaliguori #include <windows.h>
5549dc768dSaliguori #endif
5649dc768dSaliguori 
579bd2b08fSJohn Snow /**
589bd2b08fSJohn Snow  * A BdrvDirtyBitmap can be in three possible states:
599bd2b08fSJohn Snow  * (1) successor is NULL and disabled is false: full r/w mode
609bd2b08fSJohn Snow  * (2) successor is NULL and disabled is true: read only mode ("disabled")
619bd2b08fSJohn Snow  * (3) successor is set: frozen mode.
629bd2b08fSJohn Snow  *     A frozen bitmap cannot be renamed, deleted, anonymized, cleared, set,
639bd2b08fSJohn Snow  *     or enabled. A frozen bitmap can only abdicate() or reclaim().
649bd2b08fSJohn Snow  */
65e4654d2dSFam Zheng struct BdrvDirtyBitmap {
66aa0c7ca5SJohn Snow     HBitmap *bitmap;            /* Dirty sector bitmap implementation */
67aa0c7ca5SJohn Snow     BdrvDirtyBitmap *successor; /* Anonymous child; implies frozen status */
68aa0c7ca5SJohn Snow     char *name;                 /* Optional non-empty unique ID */
69aa0c7ca5SJohn Snow     int64_t size;               /* Size of the bitmap (Number of sectors) */
70aa0c7ca5SJohn Snow     bool disabled;              /* Bitmap is read-only */
71e4654d2dSFam Zheng     QLIST_ENTRY(BdrvDirtyBitmap) list;
72e4654d2dSFam Zheng };
73e4654d2dSFam Zheng 
741c9805a3SStefan Hajnoczi #define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
751c9805a3SStefan Hajnoczi 
761b7bdbc1SStefan Hajnoczi static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
771b7bdbc1SStefan Hajnoczi     QTAILQ_HEAD_INITIALIZER(bdrv_states);
787ee930d0Sblueswir1 
79dc364f4cSBenoît Canet static QTAILQ_HEAD(, BlockDriverState) graph_bdrv_states =
80dc364f4cSBenoît Canet     QTAILQ_HEAD_INITIALIZER(graph_bdrv_states);
81dc364f4cSBenoît Canet 
828a22f02aSStefan Hajnoczi static QLIST_HEAD(, BlockDriver) bdrv_drivers =
838a22f02aSStefan Hajnoczi     QLIST_HEAD_INITIALIZER(bdrv_drivers);
84ea2384d3Sbellard 
85f3930ed0SKevin Wolf static int bdrv_open_inherit(BlockDriverState **pbs, const char *filename,
86f3930ed0SKevin Wolf                              const char *reference, QDict *options, int flags,
87f3930ed0SKevin Wolf                              BlockDriverState *parent,
88f3930ed0SKevin Wolf                              const BdrvChildRole *child_role,
89f3930ed0SKevin Wolf                              BlockDriver *drv, Error **errp);
90f3930ed0SKevin Wolf 
91ce1ffea8SJohn Snow static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs);
92eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
93eb852011SMarkus Armbruster static int use_bdrv_whitelist;
94eb852011SMarkus Armbruster 
959e0b22f4SStefan Hajnoczi #ifdef _WIN32
969e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
979e0b22f4SStefan Hajnoczi {
989e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
999e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
1009e0b22f4SStefan Hajnoczi             filename[1] == ':');
1019e0b22f4SStefan Hajnoczi }
1029e0b22f4SStefan Hajnoczi 
1039e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
1049e0b22f4SStefan Hajnoczi {
1059e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
1069e0b22f4SStefan Hajnoczi         filename[2] == '\0')
1079e0b22f4SStefan Hajnoczi         return 1;
1089e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
1099e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
1109e0b22f4SStefan Hajnoczi         return 1;
1119e0b22f4SStefan Hajnoczi     return 0;
1129e0b22f4SStefan Hajnoczi }
1139e0b22f4SStefan Hajnoczi #endif
1149e0b22f4SStefan Hajnoczi 
115339064d5SKevin Wolf size_t bdrv_opt_mem_align(BlockDriverState *bs)
116339064d5SKevin Wolf {
117339064d5SKevin Wolf     if (!bs || !bs->drv) {
118459b4e66SDenis V. Lunev         /* page size or 4k (hdd sector size) should be on the safe side */
119459b4e66SDenis V. Lunev         return MAX(4096, getpagesize());
120339064d5SKevin Wolf     }
121339064d5SKevin Wolf 
122339064d5SKevin Wolf     return bs->bl.opt_mem_alignment;
123339064d5SKevin Wolf }
124339064d5SKevin Wolf 
1254196d2f0SDenis V. Lunev size_t bdrv_min_mem_align(BlockDriverState *bs)
1264196d2f0SDenis V. Lunev {
1274196d2f0SDenis V. Lunev     if (!bs || !bs->drv) {
128459b4e66SDenis V. Lunev         /* page size or 4k (hdd sector size) should be on the safe side */
129459b4e66SDenis V. Lunev         return MAX(4096, getpagesize());
1304196d2f0SDenis V. Lunev     }
1314196d2f0SDenis V. Lunev 
1324196d2f0SDenis V. Lunev     return bs->bl.min_mem_alignment;
1334196d2f0SDenis V. Lunev }
1344196d2f0SDenis V. Lunev 
1359e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
1365c98415bSMax Reitz int path_has_protocol(const char *path)
1379e0b22f4SStefan Hajnoczi {
138947995c0SPaolo Bonzini     const char *p;
139947995c0SPaolo Bonzini 
1409e0b22f4SStefan Hajnoczi #ifdef _WIN32
1419e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
1429e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
1439e0b22f4SStefan Hajnoczi         return 0;
1449e0b22f4SStefan Hajnoczi     }
145947995c0SPaolo Bonzini     p = path + strcspn(path, ":/\\");
146947995c0SPaolo Bonzini #else
147947995c0SPaolo Bonzini     p = path + strcspn(path, ":/");
1489e0b22f4SStefan Hajnoczi #endif
1499e0b22f4SStefan Hajnoczi 
150947995c0SPaolo Bonzini     return *p == ':';
1519e0b22f4SStefan Hajnoczi }
1529e0b22f4SStefan Hajnoczi 
15383f64091Sbellard int path_is_absolute(const char *path)
15483f64091Sbellard {
15521664424Sbellard #ifdef _WIN32
15621664424Sbellard     /* specific case for names like: "\\.\d:" */
157f53f4da9SPaolo Bonzini     if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
15821664424Sbellard         return 1;
159f53f4da9SPaolo Bonzini     }
160f53f4da9SPaolo Bonzini     return (*path == '/' || *path == '\\');
1613b9f94e1Sbellard #else
162f53f4da9SPaolo Bonzini     return (*path == '/');
1633b9f94e1Sbellard #endif
16483f64091Sbellard }
16583f64091Sbellard 
16683f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
16783f64091Sbellard    path to it by considering it is relative to base_path. URL are
16883f64091Sbellard    supported. */
16983f64091Sbellard void path_combine(char *dest, int dest_size,
17083f64091Sbellard                   const char *base_path,
17183f64091Sbellard                   const char *filename)
17283f64091Sbellard {
17383f64091Sbellard     const char *p, *p1;
17483f64091Sbellard     int len;
17583f64091Sbellard 
17683f64091Sbellard     if (dest_size <= 0)
17783f64091Sbellard         return;
17883f64091Sbellard     if (path_is_absolute(filename)) {
17983f64091Sbellard         pstrcpy(dest, dest_size, filename);
18083f64091Sbellard     } else {
18183f64091Sbellard         p = strchr(base_path, ':');
18283f64091Sbellard         if (p)
18383f64091Sbellard             p++;
18483f64091Sbellard         else
18583f64091Sbellard             p = base_path;
1863b9f94e1Sbellard         p1 = strrchr(base_path, '/');
1873b9f94e1Sbellard #ifdef _WIN32
1883b9f94e1Sbellard         {
1893b9f94e1Sbellard             const char *p2;
1903b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
1913b9f94e1Sbellard             if (!p1 || p2 > p1)
1923b9f94e1Sbellard                 p1 = p2;
1933b9f94e1Sbellard         }
1943b9f94e1Sbellard #endif
19583f64091Sbellard         if (p1)
19683f64091Sbellard             p1++;
19783f64091Sbellard         else
19883f64091Sbellard             p1 = base_path;
19983f64091Sbellard         if (p1 > p)
20083f64091Sbellard             p = p1;
20183f64091Sbellard         len = p - base_path;
20283f64091Sbellard         if (len > dest_size - 1)
20383f64091Sbellard             len = dest_size - 1;
20483f64091Sbellard         memcpy(dest, base_path, len);
20583f64091Sbellard         dest[len] = '\0';
20683f64091Sbellard         pstrcat(dest, dest_size, filename);
20783f64091Sbellard     }
20883f64091Sbellard }
20983f64091Sbellard 
2100a82855aSMax Reitz void bdrv_get_full_backing_filename_from_filename(const char *backed,
2110a82855aSMax Reitz                                                   const char *backing,
2129f07429eSMax Reitz                                                   char *dest, size_t sz,
2139f07429eSMax Reitz                                                   Error **errp)
2140a82855aSMax Reitz {
2159f07429eSMax Reitz     if (backing[0] == '\0' || path_has_protocol(backing) ||
2169f07429eSMax Reitz         path_is_absolute(backing))
2179f07429eSMax Reitz     {
2180a82855aSMax Reitz         pstrcpy(dest, sz, backing);
2199f07429eSMax Reitz     } else if (backed[0] == '\0' || strstart(backed, "json:", NULL)) {
2209f07429eSMax Reitz         error_setg(errp, "Cannot use relative backing file names for '%s'",
2219f07429eSMax Reitz                    backed);
2220a82855aSMax Reitz     } else {
2230a82855aSMax Reitz         path_combine(dest, sz, backed, backing);
2240a82855aSMax Reitz     }
2250a82855aSMax Reitz }
2260a82855aSMax Reitz 
2279f07429eSMax Reitz void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz,
2289f07429eSMax Reitz                                     Error **errp)
229dc5a1371SPaolo Bonzini {
2309f07429eSMax Reitz     char *backed = bs->exact_filename[0] ? bs->exact_filename : bs->filename;
2319f07429eSMax Reitz 
2329f07429eSMax Reitz     bdrv_get_full_backing_filename_from_filename(backed, bs->backing_file,
2339f07429eSMax Reitz                                                  dest, sz, errp);
234dc5a1371SPaolo Bonzini }
235dc5a1371SPaolo Bonzini 
2360eb7217eSStefan Hajnoczi void bdrv_register(BlockDriver *bdrv)
2370eb7217eSStefan Hajnoczi {
2380eb7217eSStefan Hajnoczi     bdrv_setup_io_funcs(bdrv);
239b2e12bc6SChristoph Hellwig 
2408a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
241ea2384d3Sbellard }
242b338082bSbellard 
2437f06d47eSMarkus Armbruster BlockDriverState *bdrv_new_root(void)
244fc01f7e7Sbellard {
2457f06d47eSMarkus Armbruster     BlockDriverState *bs = bdrv_new();
246e4e9986bSMarkus Armbruster 
247e4e9986bSMarkus Armbruster     QTAILQ_INSERT_TAIL(&bdrv_states, bs, device_list);
248e4e9986bSMarkus Armbruster     return bs;
249e4e9986bSMarkus Armbruster }
250e4e9986bSMarkus Armbruster 
251e4e9986bSMarkus Armbruster BlockDriverState *bdrv_new(void)
252e4e9986bSMarkus Armbruster {
253e4e9986bSMarkus Armbruster     BlockDriverState *bs;
254e4e9986bSMarkus Armbruster     int i;
255e4e9986bSMarkus Armbruster 
2565839e53bSMarkus Armbruster     bs = g_new0(BlockDriverState, 1);
257e4654d2dSFam Zheng     QLIST_INIT(&bs->dirty_bitmaps);
258fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
259fbe40ff7SFam Zheng         QLIST_INIT(&bs->op_blockers[i]);
260fbe40ff7SFam Zheng     }
26128a7282aSLuiz Capitulino     bdrv_iostatus_disable(bs);
262d7d512f6SPaolo Bonzini     notifier_list_init(&bs->close_notifiers);
263d616b224SStefan Hajnoczi     notifier_with_return_list_init(&bs->before_write_notifiers);
264cc0681c4SBenoît Canet     qemu_co_queue_init(&bs->throttled_reqs[0]);
265cc0681c4SBenoît Canet     qemu_co_queue_init(&bs->throttled_reqs[1]);
2669fcb0251SFam Zheng     bs->refcnt = 1;
267dcd04228SStefan Hajnoczi     bs->aio_context = qemu_get_aio_context();
268d7d512f6SPaolo Bonzini 
269b338082bSbellard     return bs;
270b338082bSbellard }
271b338082bSbellard 
272d7d512f6SPaolo Bonzini void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
273d7d512f6SPaolo Bonzini {
274d7d512f6SPaolo Bonzini     notifier_list_add(&bs->close_notifiers, notify);
275d7d512f6SPaolo Bonzini }
276d7d512f6SPaolo Bonzini 
277ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
278ea2384d3Sbellard {
279ea2384d3Sbellard     BlockDriver *drv1;
2808a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
2818a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
282ea2384d3Sbellard             return drv1;
283ea2384d3Sbellard         }
2848a22f02aSStefan Hajnoczi     }
285ea2384d3Sbellard     return NULL;
286ea2384d3Sbellard }
287ea2384d3Sbellard 
288b64ec4e4SFam Zheng static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
289eb852011SMarkus Armbruster {
290b64ec4e4SFam Zheng     static const char *whitelist_rw[] = {
291b64ec4e4SFam Zheng         CONFIG_BDRV_RW_WHITELIST
292b64ec4e4SFam Zheng     };
293b64ec4e4SFam Zheng     static const char *whitelist_ro[] = {
294b64ec4e4SFam Zheng         CONFIG_BDRV_RO_WHITELIST
295eb852011SMarkus Armbruster     };
296eb852011SMarkus Armbruster     const char **p;
297eb852011SMarkus Armbruster 
298b64ec4e4SFam Zheng     if (!whitelist_rw[0] && !whitelist_ro[0]) {
299eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
300b64ec4e4SFam Zheng     }
301eb852011SMarkus Armbruster 
302b64ec4e4SFam Zheng     for (p = whitelist_rw; *p; p++) {
303eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
304eb852011SMarkus Armbruster             return 1;
305eb852011SMarkus Armbruster         }
306eb852011SMarkus Armbruster     }
307b64ec4e4SFam Zheng     if (read_only) {
308b64ec4e4SFam Zheng         for (p = whitelist_ro; *p; p++) {
309b64ec4e4SFam Zheng             if (!strcmp(drv->format_name, *p)) {
310b64ec4e4SFam Zheng                 return 1;
311b64ec4e4SFam Zheng             }
312b64ec4e4SFam Zheng         }
313b64ec4e4SFam Zheng     }
314eb852011SMarkus Armbruster     return 0;
315eb852011SMarkus Armbruster }
316eb852011SMarkus Armbruster 
317b64ec4e4SFam Zheng BlockDriver *bdrv_find_whitelisted_format(const char *format_name,
318b64ec4e4SFam Zheng                                           bool read_only)
319eb852011SMarkus Armbruster {
320eb852011SMarkus Armbruster     BlockDriver *drv = bdrv_find_format(format_name);
321b64ec4e4SFam Zheng     return drv && bdrv_is_whitelisted(drv, read_only) ? drv : NULL;
322eb852011SMarkus Armbruster }
323eb852011SMarkus Armbruster 
3245b7e1542SZhi Yong Wu typedef struct CreateCo {
3255b7e1542SZhi Yong Wu     BlockDriver *drv;
3265b7e1542SZhi Yong Wu     char *filename;
32783d0521aSChunyan Liu     QemuOpts *opts;
3285b7e1542SZhi Yong Wu     int ret;
329cc84d90fSMax Reitz     Error *err;
3305b7e1542SZhi Yong Wu } CreateCo;
3315b7e1542SZhi Yong Wu 
3325b7e1542SZhi Yong Wu static void coroutine_fn bdrv_create_co_entry(void *opaque)
3335b7e1542SZhi Yong Wu {
334cc84d90fSMax Reitz     Error *local_err = NULL;
335cc84d90fSMax Reitz     int ret;
336cc84d90fSMax Reitz 
3375b7e1542SZhi Yong Wu     CreateCo *cco = opaque;
3385b7e1542SZhi Yong Wu     assert(cco->drv);
3395b7e1542SZhi Yong Wu 
340c282e1fdSChunyan Liu     ret = cco->drv->bdrv_create(cco->filename, cco->opts, &local_err);
34184d18f06SMarkus Armbruster     if (local_err) {
342cc84d90fSMax Reitz         error_propagate(&cco->err, local_err);
343cc84d90fSMax Reitz     }
344cc84d90fSMax Reitz     cco->ret = ret;
3455b7e1542SZhi Yong Wu }
3465b7e1542SZhi Yong Wu 
3470e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
34883d0521aSChunyan Liu                 QemuOpts *opts, Error **errp)
349ea2384d3Sbellard {
3505b7e1542SZhi Yong Wu     int ret;
3510e7e1989SKevin Wolf 
3525b7e1542SZhi Yong Wu     Coroutine *co;
3535b7e1542SZhi Yong Wu     CreateCo cco = {
3545b7e1542SZhi Yong Wu         .drv = drv,
3555b7e1542SZhi Yong Wu         .filename = g_strdup(filename),
35683d0521aSChunyan Liu         .opts = opts,
3575b7e1542SZhi Yong Wu         .ret = NOT_DONE,
358cc84d90fSMax Reitz         .err = NULL,
3595b7e1542SZhi Yong Wu     };
3605b7e1542SZhi Yong Wu 
361c282e1fdSChunyan Liu     if (!drv->bdrv_create) {
362cc84d90fSMax Reitz         error_setg(errp, "Driver '%s' does not support image creation", drv->format_name);
36380168bffSLuiz Capitulino         ret = -ENOTSUP;
36480168bffSLuiz Capitulino         goto out;
3655b7e1542SZhi Yong Wu     }
3665b7e1542SZhi Yong Wu 
3675b7e1542SZhi Yong Wu     if (qemu_in_coroutine()) {
3685b7e1542SZhi Yong Wu         /* Fast-path if already in coroutine context */
3695b7e1542SZhi Yong Wu         bdrv_create_co_entry(&cco);
3705b7e1542SZhi Yong Wu     } else {
3715b7e1542SZhi Yong Wu         co = qemu_coroutine_create(bdrv_create_co_entry);
3725b7e1542SZhi Yong Wu         qemu_coroutine_enter(co, &cco);
3735b7e1542SZhi Yong Wu         while (cco.ret == NOT_DONE) {
374b47ec2c4SPaolo Bonzini             aio_poll(qemu_get_aio_context(), true);
3755b7e1542SZhi Yong Wu         }
3765b7e1542SZhi Yong Wu     }
3775b7e1542SZhi Yong Wu 
3785b7e1542SZhi Yong Wu     ret = cco.ret;
379cc84d90fSMax Reitz     if (ret < 0) {
38084d18f06SMarkus Armbruster         if (cco.err) {
381cc84d90fSMax Reitz             error_propagate(errp, cco.err);
382cc84d90fSMax Reitz         } else {
383cc84d90fSMax Reitz             error_setg_errno(errp, -ret, "Could not create image");
384cc84d90fSMax Reitz         }
385cc84d90fSMax Reitz     }
3865b7e1542SZhi Yong Wu 
38780168bffSLuiz Capitulino out:
38880168bffSLuiz Capitulino     g_free(cco.filename);
3895b7e1542SZhi Yong Wu     return ret;
390ea2384d3Sbellard }
391ea2384d3Sbellard 
392c282e1fdSChunyan Liu int bdrv_create_file(const char *filename, QemuOpts *opts, Error **errp)
39384a12e66SChristoph Hellwig {
39484a12e66SChristoph Hellwig     BlockDriver *drv;
395cc84d90fSMax Reitz     Error *local_err = NULL;
396cc84d90fSMax Reitz     int ret;
39784a12e66SChristoph Hellwig 
398b65a5e12SMax Reitz     drv = bdrv_find_protocol(filename, true, errp);
39984a12e66SChristoph Hellwig     if (drv == NULL) {
40016905d71SStefan Hajnoczi         return -ENOENT;
40184a12e66SChristoph Hellwig     }
40284a12e66SChristoph Hellwig 
403c282e1fdSChunyan Liu     ret = bdrv_create(drv, filename, opts, &local_err);
40484d18f06SMarkus Armbruster     if (local_err) {
405cc84d90fSMax Reitz         error_propagate(errp, local_err);
406cc84d90fSMax Reitz     }
407cc84d90fSMax Reitz     return ret;
40884a12e66SChristoph Hellwig }
40984a12e66SChristoph Hellwig 
410892b7de8SEkaterina Tumanova /**
411892b7de8SEkaterina Tumanova  * Try to get @bs's logical and physical block size.
412892b7de8SEkaterina Tumanova  * On success, store them in @bsz struct and return 0.
413892b7de8SEkaterina Tumanova  * On failure return -errno.
414892b7de8SEkaterina Tumanova  * @bs must not be empty.
415892b7de8SEkaterina Tumanova  */
416892b7de8SEkaterina Tumanova int bdrv_probe_blocksizes(BlockDriverState *bs, BlockSizes *bsz)
417892b7de8SEkaterina Tumanova {
418892b7de8SEkaterina Tumanova     BlockDriver *drv = bs->drv;
419892b7de8SEkaterina Tumanova 
420892b7de8SEkaterina Tumanova     if (drv && drv->bdrv_probe_blocksizes) {
421892b7de8SEkaterina Tumanova         return drv->bdrv_probe_blocksizes(bs, bsz);
422892b7de8SEkaterina Tumanova     }
423892b7de8SEkaterina Tumanova 
424892b7de8SEkaterina Tumanova     return -ENOTSUP;
425892b7de8SEkaterina Tumanova }
426892b7de8SEkaterina Tumanova 
427892b7de8SEkaterina Tumanova /**
428892b7de8SEkaterina Tumanova  * Try to get @bs's geometry (cyls, heads, sectors).
429892b7de8SEkaterina Tumanova  * On success, store them in @geo struct and return 0.
430892b7de8SEkaterina Tumanova  * On failure return -errno.
431892b7de8SEkaterina Tumanova  * @bs must not be empty.
432892b7de8SEkaterina Tumanova  */
433892b7de8SEkaterina Tumanova int bdrv_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
434892b7de8SEkaterina Tumanova {
435892b7de8SEkaterina Tumanova     BlockDriver *drv = bs->drv;
436892b7de8SEkaterina Tumanova 
437892b7de8SEkaterina Tumanova     if (drv && drv->bdrv_probe_geometry) {
438892b7de8SEkaterina Tumanova         return drv->bdrv_probe_geometry(bs, geo);
439892b7de8SEkaterina Tumanova     }
440892b7de8SEkaterina Tumanova 
441892b7de8SEkaterina Tumanova     return -ENOTSUP;
442892b7de8SEkaterina Tumanova }
443892b7de8SEkaterina Tumanova 
444eba25057SJim Meyering /*
445eba25057SJim Meyering  * Create a uniquely-named empty temporary file.
446eba25057SJim Meyering  * Return 0 upon success, otherwise a negative errno value.
447eba25057SJim Meyering  */
448eba25057SJim Meyering int get_tmp_filename(char *filename, int size)
449eba25057SJim Meyering {
450d5249393Sbellard #ifdef _WIN32
4513b9f94e1Sbellard     char temp_dir[MAX_PATH];
452eba25057SJim Meyering     /* GetTempFileName requires that its output buffer (4th param)
453eba25057SJim Meyering        have length MAX_PATH or greater.  */
454eba25057SJim Meyering     assert(size >= MAX_PATH);
455eba25057SJim Meyering     return (GetTempPath(MAX_PATH, temp_dir)
456eba25057SJim Meyering             && GetTempFileName(temp_dir, "qem", 0, filename)
457eba25057SJim Meyering             ? 0 : -GetLastError());
458d5249393Sbellard #else
459ea2384d3Sbellard     int fd;
4607ccfb2ebSblueswir1     const char *tmpdir;
4610badc1eeSaurel32     tmpdir = getenv("TMPDIR");
46269bef793SAmit Shah     if (!tmpdir) {
46369bef793SAmit Shah         tmpdir = "/var/tmp";
46469bef793SAmit Shah     }
465eba25057SJim Meyering     if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
466eba25057SJim Meyering         return -EOVERFLOW;
467ea2384d3Sbellard     }
468eba25057SJim Meyering     fd = mkstemp(filename);
469fe235a06SDunrong Huang     if (fd < 0) {
470fe235a06SDunrong Huang         return -errno;
471fe235a06SDunrong Huang     }
472fe235a06SDunrong Huang     if (close(fd) != 0) {
473fe235a06SDunrong Huang         unlink(filename);
474eba25057SJim Meyering         return -errno;
475eba25057SJim Meyering     }
476eba25057SJim Meyering     return 0;
477d5249393Sbellard #endif
478eba25057SJim Meyering }
479ea2384d3Sbellard 
480f3a5d3f8SChristoph Hellwig /*
481f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
482f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
483f3a5d3f8SChristoph Hellwig  */
484f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
485f3a5d3f8SChristoph Hellwig {
486508c7cb3SChristoph Hellwig     int score_max = 0, score;
487508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
488f3a5d3f8SChristoph Hellwig 
4898a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
490508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
491508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
492508c7cb3SChristoph Hellwig             if (score > score_max) {
493508c7cb3SChristoph Hellwig                 score_max = score;
494508c7cb3SChristoph Hellwig                 drv = d;
495f3a5d3f8SChristoph Hellwig             }
496508c7cb3SChristoph Hellwig         }
497f3a5d3f8SChristoph Hellwig     }
498f3a5d3f8SChristoph Hellwig 
499508c7cb3SChristoph Hellwig     return drv;
500f3a5d3f8SChristoph Hellwig }
501f3a5d3f8SChristoph Hellwig 
50298289620SKevin Wolf BlockDriver *bdrv_find_protocol(const char *filename,
503b65a5e12SMax Reitz                                 bool allow_protocol_prefix,
504b65a5e12SMax Reitz                                 Error **errp)
50584a12e66SChristoph Hellwig {
50684a12e66SChristoph Hellwig     BlockDriver *drv1;
50784a12e66SChristoph Hellwig     char protocol[128];
50884a12e66SChristoph Hellwig     int len;
50984a12e66SChristoph Hellwig     const char *p;
51084a12e66SChristoph Hellwig 
51166f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
51266f82ceeSKevin Wolf 
51339508e7aSChristoph Hellwig     /*
51439508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
51539508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
51639508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
51739508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
51839508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
51939508e7aSChristoph Hellwig      */
52084a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
52139508e7aSChristoph Hellwig     if (drv1) {
52284a12e66SChristoph Hellwig         return drv1;
52384a12e66SChristoph Hellwig     }
52439508e7aSChristoph Hellwig 
52598289620SKevin Wolf     if (!path_has_protocol(filename) || !allow_protocol_prefix) {
526ef810437SMax Reitz         return &bdrv_file;
52739508e7aSChristoph Hellwig     }
52898289620SKevin Wolf 
5299e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
5309e0b22f4SStefan Hajnoczi     assert(p != NULL);
53184a12e66SChristoph Hellwig     len = p - filename;
53284a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
53384a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
53484a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
53584a12e66SChristoph Hellwig     protocol[len] = '\0';
53684a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
53784a12e66SChristoph Hellwig         if (drv1->protocol_name &&
53884a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
53984a12e66SChristoph Hellwig             return drv1;
54084a12e66SChristoph Hellwig         }
54184a12e66SChristoph Hellwig     }
542b65a5e12SMax Reitz 
543b65a5e12SMax Reitz     error_setg(errp, "Unknown protocol '%s'", protocol);
54484a12e66SChristoph Hellwig     return NULL;
54584a12e66SChristoph Hellwig }
54684a12e66SChristoph Hellwig 
547c6684249SMarkus Armbruster /*
548c6684249SMarkus Armbruster  * Guess image format by probing its contents.
549c6684249SMarkus Armbruster  * This is not a good idea when your image is raw (CVE-2008-2004), but
550c6684249SMarkus Armbruster  * we do it anyway for backward compatibility.
551c6684249SMarkus Armbruster  *
552c6684249SMarkus Armbruster  * @buf         contains the image's first @buf_size bytes.
5537cddd372SKevin Wolf  * @buf_size    is the buffer size in bytes (generally BLOCK_PROBE_BUF_SIZE,
5547cddd372SKevin Wolf  *              but can be smaller if the image file is smaller)
555c6684249SMarkus Armbruster  * @filename    is its filename.
556c6684249SMarkus Armbruster  *
557c6684249SMarkus Armbruster  * For all block drivers, call the bdrv_probe() method to get its
558c6684249SMarkus Armbruster  * probing score.
559c6684249SMarkus Armbruster  * Return the first block driver with the highest probing score.
560c6684249SMarkus Armbruster  */
56138f3ef57SKevin Wolf BlockDriver *bdrv_probe_all(const uint8_t *buf, int buf_size,
562c6684249SMarkus Armbruster                             const char *filename)
563c6684249SMarkus Armbruster {
564c6684249SMarkus Armbruster     int score_max = 0, score;
565c6684249SMarkus Armbruster     BlockDriver *drv = NULL, *d;
566c6684249SMarkus Armbruster 
567c6684249SMarkus Armbruster     QLIST_FOREACH(d, &bdrv_drivers, list) {
568c6684249SMarkus Armbruster         if (d->bdrv_probe) {
569c6684249SMarkus Armbruster             score = d->bdrv_probe(buf, buf_size, filename);
570c6684249SMarkus Armbruster             if (score > score_max) {
571c6684249SMarkus Armbruster                 score_max = score;
572c6684249SMarkus Armbruster                 drv = d;
573c6684249SMarkus Armbruster             }
574c6684249SMarkus Armbruster         }
575c6684249SMarkus Armbruster     }
576c6684249SMarkus Armbruster 
577c6684249SMarkus Armbruster     return drv;
578c6684249SMarkus Armbruster }
579c6684249SMarkus Armbruster 
580f500a6d3SKevin Wolf static int find_image_format(BlockDriverState *bs, const char *filename,
58134b5d2c6SMax Reitz                              BlockDriver **pdrv, Error **errp)
582ea2384d3Sbellard {
583c6684249SMarkus Armbruster     BlockDriver *drv;
5847cddd372SKevin Wolf     uint8_t buf[BLOCK_PROBE_BUF_SIZE];
585f500a6d3SKevin Wolf     int ret = 0;
586f8ea0b00SNicholas Bellinger 
58708a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
588b192af8aSDimitris Aragiorgis     if (bdrv_is_sg(bs) || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
589ef810437SMax Reitz         *pdrv = &bdrv_raw;
590c98ac35dSStefan Weil         return ret;
5911a396859SNicholas A. Bellinger     }
592f8ea0b00SNicholas Bellinger 
59383f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
594ea2384d3Sbellard     if (ret < 0) {
59534b5d2c6SMax Reitz         error_setg_errno(errp, -ret, "Could not read image for determining its "
59634b5d2c6SMax Reitz                          "format");
597c98ac35dSStefan Weil         *pdrv = NULL;
598c98ac35dSStefan Weil         return ret;
599ea2384d3Sbellard     }
600ea2384d3Sbellard 
601c6684249SMarkus Armbruster     drv = bdrv_probe_all(buf, ret, filename);
602c98ac35dSStefan Weil     if (!drv) {
60334b5d2c6SMax Reitz         error_setg(errp, "Could not determine image format: No compatible "
60434b5d2c6SMax Reitz                    "driver found");
605c98ac35dSStefan Weil         ret = -ENOENT;
606c98ac35dSStefan Weil     }
607c98ac35dSStefan Weil     *pdrv = drv;
608c98ac35dSStefan Weil     return ret;
609ea2384d3Sbellard }
610ea2384d3Sbellard 
61151762288SStefan Hajnoczi /**
61251762288SStefan Hajnoczi  * Set the current 'total_sectors' value
61365a9bb25SMarkus Armbruster  * Return 0 on success, -errno on error.
61451762288SStefan Hajnoczi  */
61551762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
61651762288SStefan Hajnoczi {
61751762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
61851762288SStefan Hajnoczi 
619396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
620b192af8aSDimitris Aragiorgis     if (bdrv_is_sg(bs))
621396759adSNicholas Bellinger         return 0;
622396759adSNicholas Bellinger 
62351762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
62451762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
62551762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
62651762288SStefan Hajnoczi         if (length < 0) {
62751762288SStefan Hajnoczi             return length;
62851762288SStefan Hajnoczi         }
6297e382003SFam Zheng         hint = DIV_ROUND_UP(length, BDRV_SECTOR_SIZE);
63051762288SStefan Hajnoczi     }
63151762288SStefan Hajnoczi 
63251762288SStefan Hajnoczi     bs->total_sectors = hint;
63351762288SStefan Hajnoczi     return 0;
63451762288SStefan Hajnoczi }
63551762288SStefan Hajnoczi 
636c3993cdcSStefan Hajnoczi /**
6379e8f1835SPaolo Bonzini  * Set open flags for a given discard mode
6389e8f1835SPaolo Bonzini  *
6399e8f1835SPaolo Bonzini  * Return 0 on success, -1 if the discard mode was invalid.
6409e8f1835SPaolo Bonzini  */
6419e8f1835SPaolo Bonzini int bdrv_parse_discard_flags(const char *mode, int *flags)
6429e8f1835SPaolo Bonzini {
6439e8f1835SPaolo Bonzini     *flags &= ~BDRV_O_UNMAP;
6449e8f1835SPaolo Bonzini 
6459e8f1835SPaolo Bonzini     if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
6469e8f1835SPaolo Bonzini         /* do nothing */
6479e8f1835SPaolo Bonzini     } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
6489e8f1835SPaolo Bonzini         *flags |= BDRV_O_UNMAP;
6499e8f1835SPaolo Bonzini     } else {
6509e8f1835SPaolo Bonzini         return -1;
6519e8f1835SPaolo Bonzini     }
6529e8f1835SPaolo Bonzini 
6539e8f1835SPaolo Bonzini     return 0;
6549e8f1835SPaolo Bonzini }
6559e8f1835SPaolo Bonzini 
6569e8f1835SPaolo Bonzini /**
657c3993cdcSStefan Hajnoczi  * Set open flags for a given cache mode
658c3993cdcSStefan Hajnoczi  *
659c3993cdcSStefan Hajnoczi  * Return 0 on success, -1 if the cache mode was invalid.
660c3993cdcSStefan Hajnoczi  */
661c3993cdcSStefan Hajnoczi int bdrv_parse_cache_flags(const char *mode, int *flags)
662c3993cdcSStefan Hajnoczi {
663c3993cdcSStefan Hajnoczi     *flags &= ~BDRV_O_CACHE_MASK;
664c3993cdcSStefan Hajnoczi 
665c3993cdcSStefan Hajnoczi     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
666c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
66792196b2fSStefan Hajnoczi     } else if (!strcmp(mode, "directsync")) {
66892196b2fSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE;
669c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writeback")) {
670c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
671c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "unsafe")) {
672c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
673c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NO_FLUSH;
674c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writethrough")) {
675c3993cdcSStefan Hajnoczi         /* this is the default */
676c3993cdcSStefan Hajnoczi     } else {
677c3993cdcSStefan Hajnoczi         return -1;
678c3993cdcSStefan Hajnoczi     }
679c3993cdcSStefan Hajnoczi 
680c3993cdcSStefan Hajnoczi     return 0;
681c3993cdcSStefan Hajnoczi }
682c3993cdcSStefan Hajnoczi 
6830b50cc88SKevin Wolf /*
684b1e6fc08SKevin Wolf  * Returns the flags that a temporary snapshot should get, based on the
685b1e6fc08SKevin Wolf  * originally requested flags (the originally requested image will have flags
686b1e6fc08SKevin Wolf  * like a backing file)
687b1e6fc08SKevin Wolf  */
688b1e6fc08SKevin Wolf static int bdrv_temp_snapshot_flags(int flags)
689b1e6fc08SKevin Wolf {
690b1e6fc08SKevin Wolf     return (flags & ~BDRV_O_SNAPSHOT) | BDRV_O_TEMPORARY;
691b1e6fc08SKevin Wolf }
692b1e6fc08SKevin Wolf 
693b1e6fc08SKevin Wolf /*
694f3930ed0SKevin Wolf  * Returns the flags that bs->file should get if a protocol driver is expected,
695f3930ed0SKevin Wolf  * based on the given flags for the parent BDS
6960b50cc88SKevin Wolf  */
6970b50cc88SKevin Wolf static int bdrv_inherited_flags(int flags)
6980b50cc88SKevin Wolf {
6990b50cc88SKevin Wolf     /* Enable protocol handling, disable format probing for bs->file */
7000b50cc88SKevin Wolf     flags |= BDRV_O_PROTOCOL;
7010b50cc88SKevin Wolf 
7020b50cc88SKevin Wolf     /* Our block drivers take care to send flushes and respect unmap policy,
7030b50cc88SKevin Wolf      * so we can enable both unconditionally on lower layers. */
7040b50cc88SKevin Wolf     flags |= BDRV_O_CACHE_WB | BDRV_O_UNMAP;
7050b50cc88SKevin Wolf 
7060b50cc88SKevin Wolf     /* Clear flags that only apply to the top layer */
7075669b44dSKevin Wolf     flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING | BDRV_O_COPY_ON_READ);
7080b50cc88SKevin Wolf 
7090b50cc88SKevin Wolf     return flags;
7100b50cc88SKevin Wolf }
7110b50cc88SKevin Wolf 
712f3930ed0SKevin Wolf const BdrvChildRole child_file = {
713f3930ed0SKevin Wolf     .inherit_flags = bdrv_inherited_flags,
714f3930ed0SKevin Wolf };
715f3930ed0SKevin Wolf 
716f3930ed0SKevin Wolf /*
717f3930ed0SKevin Wolf  * Returns the flags that bs->file should get if the use of formats (and not
718f3930ed0SKevin Wolf  * only protocols) is permitted for it, based on the given flags for the parent
719f3930ed0SKevin Wolf  * BDS
720f3930ed0SKevin Wolf  */
721f3930ed0SKevin Wolf static int bdrv_inherited_fmt_flags(int parent_flags)
722f3930ed0SKevin Wolf {
723f3930ed0SKevin Wolf     int flags = child_file.inherit_flags(parent_flags);
724f3930ed0SKevin Wolf     return flags & ~BDRV_O_PROTOCOL;
725f3930ed0SKevin Wolf }
726f3930ed0SKevin Wolf 
727f3930ed0SKevin Wolf const BdrvChildRole child_format = {
728f3930ed0SKevin Wolf     .inherit_flags = bdrv_inherited_fmt_flags,
729f3930ed0SKevin Wolf };
730f3930ed0SKevin Wolf 
731317fc44eSKevin Wolf /*
732317fc44eSKevin Wolf  * Returns the flags that bs->backing_hd should get, based on the given flags
733317fc44eSKevin Wolf  * for the parent BDS
734317fc44eSKevin Wolf  */
735317fc44eSKevin Wolf static int bdrv_backing_flags(int flags)
736317fc44eSKevin Wolf {
737317fc44eSKevin Wolf     /* backing files always opened read-only */
738317fc44eSKevin Wolf     flags &= ~(BDRV_O_RDWR | BDRV_O_COPY_ON_READ);
739317fc44eSKevin Wolf 
740317fc44eSKevin Wolf     /* snapshot=on is handled on the top layer */
7418bfea15dSKevin Wolf     flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_TEMPORARY);
742317fc44eSKevin Wolf 
743317fc44eSKevin Wolf     return flags;
744317fc44eSKevin Wolf }
745317fc44eSKevin Wolf 
746f3930ed0SKevin Wolf static const BdrvChildRole child_backing = {
747f3930ed0SKevin Wolf     .inherit_flags = bdrv_backing_flags,
748f3930ed0SKevin Wolf };
749f3930ed0SKevin Wolf 
7507b272452SKevin Wolf static int bdrv_open_flags(BlockDriverState *bs, int flags)
7517b272452SKevin Wolf {
7527b272452SKevin Wolf     int open_flags = flags | BDRV_O_CACHE_WB;
7537b272452SKevin Wolf 
7547b272452SKevin Wolf     /*
7557b272452SKevin Wolf      * Clear flags that are internal to the block layer before opening the
7567b272452SKevin Wolf      * image.
7577b272452SKevin Wolf      */
75820cca275SKevin Wolf     open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING | BDRV_O_PROTOCOL);
7597b272452SKevin Wolf 
7607b272452SKevin Wolf     /*
7617b272452SKevin Wolf      * Snapshots should be writable.
7627b272452SKevin Wolf      */
7638bfea15dSKevin Wolf     if (flags & BDRV_O_TEMPORARY) {
7647b272452SKevin Wolf         open_flags |= BDRV_O_RDWR;
7657b272452SKevin Wolf     }
7667b272452SKevin Wolf 
7677b272452SKevin Wolf     return open_flags;
7687b272452SKevin Wolf }
7697b272452SKevin Wolf 
770636ea370SKevin Wolf static void bdrv_assign_node_name(BlockDriverState *bs,
7716913c0c2SBenoît Canet                                   const char *node_name,
7726913c0c2SBenoît Canet                                   Error **errp)
7736913c0c2SBenoît Canet {
7746913c0c2SBenoît Canet     if (!node_name) {
775636ea370SKevin Wolf         return;
7766913c0c2SBenoît Canet     }
7776913c0c2SBenoît Canet 
7789aebf3b8SKevin Wolf     /* Check for empty string or invalid characters */
779f5bebbbbSMarkus Armbruster     if (!id_wellformed(node_name)) {
7809aebf3b8SKevin Wolf         error_setg(errp, "Invalid node name");
781636ea370SKevin Wolf         return;
7826913c0c2SBenoît Canet     }
7836913c0c2SBenoît Canet 
7840c5e94eeSBenoît Canet     /* takes care of avoiding namespaces collisions */
7857f06d47eSMarkus Armbruster     if (blk_by_name(node_name)) {
7860c5e94eeSBenoît Canet         error_setg(errp, "node-name=%s is conflicting with a device id",
7870c5e94eeSBenoît Canet                    node_name);
788636ea370SKevin Wolf         return;
7890c5e94eeSBenoît Canet     }
7900c5e94eeSBenoît Canet 
7916913c0c2SBenoît Canet     /* takes care of avoiding duplicates node names */
7926913c0c2SBenoît Canet     if (bdrv_find_node(node_name)) {
7936913c0c2SBenoît Canet         error_setg(errp, "Duplicate node name");
794636ea370SKevin Wolf         return;
7956913c0c2SBenoît Canet     }
7966913c0c2SBenoît Canet 
7976913c0c2SBenoît Canet     /* copy node name into the bs and insert it into the graph list */
7986913c0c2SBenoît Canet     pstrcpy(bs->node_name, sizeof(bs->node_name), node_name);
7996913c0c2SBenoît Canet     QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs, node_list);
8006913c0c2SBenoît Canet }
8016913c0c2SBenoît Canet 
80218edf289SKevin Wolf static QemuOptsList bdrv_runtime_opts = {
80318edf289SKevin Wolf     .name = "bdrv_common",
80418edf289SKevin Wolf     .head = QTAILQ_HEAD_INITIALIZER(bdrv_runtime_opts.head),
80518edf289SKevin Wolf     .desc = {
80618edf289SKevin Wolf         {
80718edf289SKevin Wolf             .name = "node-name",
80818edf289SKevin Wolf             .type = QEMU_OPT_STRING,
80918edf289SKevin Wolf             .help = "Node name of the block device node",
81018edf289SKevin Wolf         },
81118edf289SKevin Wolf         { /* end of list */ }
81218edf289SKevin Wolf     },
81318edf289SKevin Wolf };
81418edf289SKevin Wolf 
815b6ce07aaSKevin Wolf /*
81657915332SKevin Wolf  * Common part for opening disk images and files
817b6ad491aSKevin Wolf  *
818b6ad491aSKevin Wolf  * Removes all processed options from *options.
81957915332SKevin Wolf  */
820f500a6d3SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
82134b5d2c6SMax Reitz     QDict *options, int flags, BlockDriver *drv, Error **errp)
82257915332SKevin Wolf {
82357915332SKevin Wolf     int ret, open_flags;
824035fccdfSKevin Wolf     const char *filename;
8256913c0c2SBenoît Canet     const char *node_name = NULL;
82618edf289SKevin Wolf     QemuOpts *opts;
82734b5d2c6SMax Reitz     Error *local_err = NULL;
82857915332SKevin Wolf 
82957915332SKevin Wolf     assert(drv != NULL);
8306405875cSPaolo Bonzini     assert(bs->file == NULL);
831707ff828SKevin Wolf     assert(options != NULL && bs->options != options);
83257915332SKevin Wolf 
83345673671SKevin Wolf     if (file != NULL) {
83445673671SKevin Wolf         filename = file->filename;
83545673671SKevin Wolf     } else {
83645673671SKevin Wolf         filename = qdict_get_try_str(options, "filename");
83745673671SKevin Wolf     }
83845673671SKevin Wolf 
839765003dbSKevin Wolf     if (drv->bdrv_needs_filename && !filename) {
840765003dbSKevin Wolf         error_setg(errp, "The '%s' block driver requires a file name",
841765003dbSKevin Wolf                    drv->format_name);
842765003dbSKevin Wolf         return -EINVAL;
843765003dbSKevin Wolf     }
844765003dbSKevin Wolf 
84545673671SKevin Wolf     trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
84628dcee10SStefan Hajnoczi 
84718edf289SKevin Wolf     opts = qemu_opts_create(&bdrv_runtime_opts, NULL, 0, &error_abort);
84818edf289SKevin Wolf     qemu_opts_absorb_qdict(opts, options, &local_err);
84918edf289SKevin Wolf     if (local_err) {
85018edf289SKevin Wolf         error_propagate(errp, local_err);
85118edf289SKevin Wolf         ret = -EINVAL;
85218edf289SKevin Wolf         goto fail_opts;
85318edf289SKevin Wolf     }
85418edf289SKevin Wolf 
85518edf289SKevin Wolf     node_name = qemu_opt_get(opts, "node-name");
856636ea370SKevin Wolf     bdrv_assign_node_name(bs, node_name, &local_err);
8570fb6395cSMarkus Armbruster     if (local_err) {
858636ea370SKevin Wolf         error_propagate(errp, local_err);
85918edf289SKevin Wolf         ret = -EINVAL;
86018edf289SKevin Wolf         goto fail_opts;
8615d186eb0SKevin Wolf     }
8625d186eb0SKevin Wolf 
8631b7fd729SPaolo Bonzini     bs->guest_block_size = 512;
864c25f53b0SPaolo Bonzini     bs->request_alignment = 512;
8650d51b4deSAsias He     bs->zero_beyond_eof = true;
866b64ec4e4SFam Zheng     open_flags = bdrv_open_flags(bs, flags);
867b64ec4e4SFam Zheng     bs->read_only = !(open_flags & BDRV_O_RDWR);
868b64ec4e4SFam Zheng 
869b64ec4e4SFam Zheng     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv, bs->read_only)) {
8708f94a6e4SKevin Wolf         error_setg(errp,
8718f94a6e4SKevin Wolf                    !bs->read_only && bdrv_is_whitelisted(drv, true)
8728f94a6e4SKevin Wolf                         ? "Driver '%s' can only be used for read-only devices"
8738f94a6e4SKevin Wolf                         : "Driver '%s' is not whitelisted",
8748f94a6e4SKevin Wolf                    drv->format_name);
87518edf289SKevin Wolf         ret = -ENOTSUP;
87618edf289SKevin Wolf         goto fail_opts;
877b64ec4e4SFam Zheng     }
87857915332SKevin Wolf 
87953fec9d3SStefan Hajnoczi     assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
8800ebd24e0SKevin Wolf     if (flags & BDRV_O_COPY_ON_READ) {
8810ebd24e0SKevin Wolf         if (!bs->read_only) {
88253fec9d3SStefan Hajnoczi             bdrv_enable_copy_on_read(bs);
8830ebd24e0SKevin Wolf         } else {
8840ebd24e0SKevin Wolf             error_setg(errp, "Can't use copy-on-read on read-only device");
88518edf289SKevin Wolf             ret = -EINVAL;
88618edf289SKevin Wolf             goto fail_opts;
8870ebd24e0SKevin Wolf         }
88853fec9d3SStefan Hajnoczi     }
88953fec9d3SStefan Hajnoczi 
890c2ad1b0cSKevin Wolf     if (filename != NULL) {
89157915332SKevin Wolf         pstrcpy(bs->filename, sizeof(bs->filename), filename);
892c2ad1b0cSKevin Wolf     } else {
893c2ad1b0cSKevin Wolf         bs->filename[0] = '\0';
894c2ad1b0cSKevin Wolf     }
89591af7014SMax Reitz     pstrcpy(bs->exact_filename, sizeof(bs->exact_filename), bs->filename);
89657915332SKevin Wolf 
89757915332SKevin Wolf     bs->drv = drv;
8987267c094SAnthony Liguori     bs->opaque = g_malloc0(drv->instance_size);
89957915332SKevin Wolf 
90003f541bdSStefan Hajnoczi     bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
901e7c63796SStefan Hajnoczi 
90266f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
90366f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
9045d186eb0SKevin Wolf         assert(file == NULL);
905030be321SBenoît Canet         assert(!drv->bdrv_needs_filename || filename != NULL);
90634b5d2c6SMax Reitz         ret = drv->bdrv_file_open(bs, options, open_flags, &local_err);
907f500a6d3SKevin Wolf     } else {
9082af5ef70SKevin Wolf         if (file == NULL) {
90934b5d2c6SMax Reitz             error_setg(errp, "Can't use '%s' as a block driver for the "
91034b5d2c6SMax Reitz                        "protocol level", drv->format_name);
9112af5ef70SKevin Wolf             ret = -EINVAL;
9122af5ef70SKevin Wolf             goto free_and_fail;
9132af5ef70SKevin Wolf         }
914f500a6d3SKevin Wolf         bs->file = file;
91534b5d2c6SMax Reitz         ret = drv->bdrv_open(bs, options, open_flags, &local_err);
91666f82ceeSKevin Wolf     }
91766f82ceeSKevin Wolf 
91857915332SKevin Wolf     if (ret < 0) {
91984d18f06SMarkus Armbruster         if (local_err) {
92034b5d2c6SMax Reitz             error_propagate(errp, local_err);
9212fa9aa59SDunrong Huang         } else if (bs->filename[0]) {
9222fa9aa59SDunrong Huang             error_setg_errno(errp, -ret, "Could not open '%s'", bs->filename);
92334b5d2c6SMax Reitz         } else {
92434b5d2c6SMax Reitz             error_setg_errno(errp, -ret, "Could not open image");
92534b5d2c6SMax Reitz         }
92657915332SKevin Wolf         goto free_and_fail;
92757915332SKevin Wolf     }
92857915332SKevin Wolf 
929a1f688f4SMarkus Armbruster     if (bs->encrypted) {
930a1f688f4SMarkus Armbruster         error_report("Encrypted images are deprecated");
931a1f688f4SMarkus Armbruster         error_printf("Support for them will be removed in a future release.\n"
932a1f688f4SMarkus Armbruster                      "You can use 'qemu-img convert' to convert your image"
933a1f688f4SMarkus Armbruster                      " to an unencrypted one.\n");
934a1f688f4SMarkus Armbruster     }
935a1f688f4SMarkus Armbruster 
93651762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
93751762288SStefan Hajnoczi     if (ret < 0) {
93834b5d2c6SMax Reitz         error_setg_errno(errp, -ret, "Could not refresh total sector count");
93951762288SStefan Hajnoczi         goto free_and_fail;
94057915332SKevin Wolf     }
94151762288SStefan Hajnoczi 
9423baca891SKevin Wolf     bdrv_refresh_limits(bs, &local_err);
9433baca891SKevin Wolf     if (local_err) {
9443baca891SKevin Wolf         error_propagate(errp, local_err);
9453baca891SKevin Wolf         ret = -EINVAL;
9463baca891SKevin Wolf         goto free_and_fail;
9473baca891SKevin Wolf     }
9483baca891SKevin Wolf 
949c25f53b0SPaolo Bonzini     assert(bdrv_opt_mem_align(bs) != 0);
9504196d2f0SDenis V. Lunev     assert(bdrv_min_mem_align(bs) != 0);
951b192af8aSDimitris Aragiorgis     assert((bs->request_alignment != 0) || bdrv_is_sg(bs));
95218edf289SKevin Wolf 
95318edf289SKevin Wolf     qemu_opts_del(opts);
95457915332SKevin Wolf     return 0;
95557915332SKevin Wolf 
95657915332SKevin Wolf free_and_fail:
95766f82ceeSKevin Wolf     bs->file = NULL;
9587267c094SAnthony Liguori     g_free(bs->opaque);
95957915332SKevin Wolf     bs->opaque = NULL;
96057915332SKevin Wolf     bs->drv = NULL;
96118edf289SKevin Wolf fail_opts:
96218edf289SKevin Wolf     qemu_opts_del(opts);
96357915332SKevin Wolf     return ret;
96457915332SKevin Wolf }
96557915332SKevin Wolf 
9665e5c4f63SKevin Wolf static QDict *parse_json_filename(const char *filename, Error **errp)
9675e5c4f63SKevin Wolf {
9685e5c4f63SKevin Wolf     QObject *options_obj;
9695e5c4f63SKevin Wolf     QDict *options;
9705e5c4f63SKevin Wolf     int ret;
9715e5c4f63SKevin Wolf 
9725e5c4f63SKevin Wolf     ret = strstart(filename, "json:", &filename);
9735e5c4f63SKevin Wolf     assert(ret);
9745e5c4f63SKevin Wolf 
9755e5c4f63SKevin Wolf     options_obj = qobject_from_json(filename);
9765e5c4f63SKevin Wolf     if (!options_obj) {
9775e5c4f63SKevin Wolf         error_setg(errp, "Could not parse the JSON options");
9785e5c4f63SKevin Wolf         return NULL;
9795e5c4f63SKevin Wolf     }
9805e5c4f63SKevin Wolf 
9815e5c4f63SKevin Wolf     if (qobject_type(options_obj) != QTYPE_QDICT) {
9825e5c4f63SKevin Wolf         qobject_decref(options_obj);
9835e5c4f63SKevin Wolf         error_setg(errp, "Invalid JSON object given");
9845e5c4f63SKevin Wolf         return NULL;
9855e5c4f63SKevin Wolf     }
9865e5c4f63SKevin Wolf 
9875e5c4f63SKevin Wolf     options = qobject_to_qdict(options_obj);
9885e5c4f63SKevin Wolf     qdict_flatten(options);
9895e5c4f63SKevin Wolf 
9905e5c4f63SKevin Wolf     return options;
9915e5c4f63SKevin Wolf }
9925e5c4f63SKevin Wolf 
99357915332SKevin Wolf /*
994f54120ffSKevin Wolf  * Fills in default options for opening images and converts the legacy
995f54120ffSKevin Wolf  * filename/flags pair to option QDict entries.
99653a29513SMax Reitz  * The BDRV_O_PROTOCOL flag in *flags will be set or cleared accordingly if a
99753a29513SMax Reitz  * block driver has been specified explicitly.
998f54120ffSKevin Wolf  */
99953a29513SMax Reitz static int bdrv_fill_options(QDict **options, const char **pfilename,
100053a29513SMax Reitz                              int *flags, BlockDriver *drv, Error **errp)
1001f54120ffSKevin Wolf {
10025e5c4f63SKevin Wolf     const char *filename = *pfilename;
1003f54120ffSKevin Wolf     const char *drvname;
100453a29513SMax Reitz     bool protocol = *flags & BDRV_O_PROTOCOL;
1005f54120ffSKevin Wolf     bool parse_filename = false;
100653a29513SMax Reitz     BlockDriver *tmp_drv;
1007f54120ffSKevin Wolf     Error *local_err = NULL;
1008f54120ffSKevin Wolf 
10095e5c4f63SKevin Wolf     /* Parse json: pseudo-protocol */
10105e5c4f63SKevin Wolf     if (filename && g_str_has_prefix(filename, "json:")) {
10115e5c4f63SKevin Wolf         QDict *json_options = parse_json_filename(filename, &local_err);
10125e5c4f63SKevin Wolf         if (local_err) {
10135e5c4f63SKevin Wolf             error_propagate(errp, local_err);
10145e5c4f63SKevin Wolf             return -EINVAL;
10155e5c4f63SKevin Wolf         }
10165e5c4f63SKevin Wolf 
10175e5c4f63SKevin Wolf         /* Options given in the filename have lower priority than options
10185e5c4f63SKevin Wolf          * specified directly */
10195e5c4f63SKevin Wolf         qdict_join(*options, json_options, false);
10205e5c4f63SKevin Wolf         QDECREF(json_options);
10215e5c4f63SKevin Wolf         *pfilename = filename = NULL;
10225e5c4f63SKevin Wolf     }
10235e5c4f63SKevin Wolf 
102453a29513SMax Reitz     drvname = qdict_get_try_str(*options, "driver");
102553a29513SMax Reitz 
102653a29513SMax Reitz     /* If the user has explicitly specified the driver, this choice should
102753a29513SMax Reitz      * override the BDRV_O_PROTOCOL flag */
102853a29513SMax Reitz     tmp_drv = drv;
102953a29513SMax Reitz     if (!tmp_drv && drvname) {
103053a29513SMax Reitz         tmp_drv = bdrv_find_format(drvname);
103153a29513SMax Reitz     }
103253a29513SMax Reitz     if (tmp_drv) {
103353a29513SMax Reitz         protocol = tmp_drv->bdrv_file_open;
103453a29513SMax Reitz     }
103553a29513SMax Reitz 
103653a29513SMax Reitz     if (protocol) {
103753a29513SMax Reitz         *flags |= BDRV_O_PROTOCOL;
103853a29513SMax Reitz     } else {
103953a29513SMax Reitz         *flags &= ~BDRV_O_PROTOCOL;
104053a29513SMax Reitz     }
104153a29513SMax Reitz 
1042f54120ffSKevin Wolf     /* Fetch the file name from the options QDict if necessary */
104317b005f1SKevin Wolf     if (protocol && filename) {
1044f54120ffSKevin Wolf         if (!qdict_haskey(*options, "filename")) {
1045f54120ffSKevin Wolf             qdict_put(*options, "filename", qstring_from_str(filename));
1046f54120ffSKevin Wolf             parse_filename = true;
1047f54120ffSKevin Wolf         } else {
1048f54120ffSKevin Wolf             error_setg(errp, "Can't specify 'file' and 'filename' options at "
1049f54120ffSKevin Wolf                              "the same time");
1050f54120ffSKevin Wolf             return -EINVAL;
1051f54120ffSKevin Wolf         }
1052f54120ffSKevin Wolf     }
1053f54120ffSKevin Wolf 
1054f54120ffSKevin Wolf     /* Find the right block driver */
1055f54120ffSKevin Wolf     filename = qdict_get_try_str(*options, "filename");
1056f54120ffSKevin Wolf 
105717b005f1SKevin Wolf     if (drv) {
105817b005f1SKevin Wolf         if (drvname) {
105917b005f1SKevin Wolf             error_setg(errp, "Driver specified twice");
106017b005f1SKevin Wolf             return -EINVAL;
106117b005f1SKevin Wolf         }
106217b005f1SKevin Wolf         drvname = drv->format_name;
106317b005f1SKevin Wolf         qdict_put(*options, "driver", qstring_from_str(drvname));
106417b005f1SKevin Wolf     } else {
106517b005f1SKevin Wolf         if (!drvname && protocol) {
1066f54120ffSKevin Wolf             if (filename) {
1067b65a5e12SMax Reitz                 drv = bdrv_find_protocol(filename, parse_filename, errp);
1068f54120ffSKevin Wolf                 if (!drv) {
1069f54120ffSKevin Wolf                     return -EINVAL;
1070f54120ffSKevin Wolf                 }
1071f54120ffSKevin Wolf 
1072f54120ffSKevin Wolf                 drvname = drv->format_name;
1073f54120ffSKevin Wolf                 qdict_put(*options, "driver", qstring_from_str(drvname));
1074f54120ffSKevin Wolf             } else {
1075f54120ffSKevin Wolf                 error_setg(errp, "Must specify either driver or file");
1076f54120ffSKevin Wolf                 return -EINVAL;
1077f54120ffSKevin Wolf             }
107817b005f1SKevin Wolf         } else if (drvname) {
1079f54120ffSKevin Wolf             drv = bdrv_find_format(drvname);
1080f54120ffSKevin Wolf             if (!drv) {
1081f54120ffSKevin Wolf                 error_setg(errp, "Unknown driver '%s'", drvname);
1082f54120ffSKevin Wolf                 return -ENOENT;
1083f54120ffSKevin Wolf             }
108417b005f1SKevin Wolf         }
108517b005f1SKevin Wolf     }
108617b005f1SKevin Wolf 
108717b005f1SKevin Wolf     assert(drv || !protocol);
1088f54120ffSKevin Wolf 
1089f54120ffSKevin Wolf     /* Driver-specific filename parsing */
109017b005f1SKevin Wolf     if (drv && drv->bdrv_parse_filename && parse_filename) {
1091f54120ffSKevin Wolf         drv->bdrv_parse_filename(filename, *options, &local_err);
1092f54120ffSKevin Wolf         if (local_err) {
1093f54120ffSKevin Wolf             error_propagate(errp, local_err);
1094f54120ffSKevin Wolf             return -EINVAL;
1095f54120ffSKevin Wolf         }
1096f54120ffSKevin Wolf 
1097f54120ffSKevin Wolf         if (!drv->bdrv_needs_filename) {
1098f54120ffSKevin Wolf             qdict_del(*options, "filename");
1099f54120ffSKevin Wolf         }
1100f54120ffSKevin Wolf     }
1101f54120ffSKevin Wolf 
1102f54120ffSKevin Wolf     return 0;
1103f54120ffSKevin Wolf }
1104f54120ffSKevin Wolf 
1105b4b059f6SKevin Wolf static BdrvChild *bdrv_attach_child(BlockDriverState *parent_bs,
1106df581792SKevin Wolf                                     BlockDriverState *child_bs,
1107df581792SKevin Wolf                                     const BdrvChildRole *child_role)
1108df581792SKevin Wolf {
1109df581792SKevin Wolf     BdrvChild *child = g_new(BdrvChild, 1);
1110df581792SKevin Wolf     *child = (BdrvChild) {
1111df581792SKevin Wolf         .bs     = child_bs,
1112df581792SKevin Wolf         .role   = child_role,
1113df581792SKevin Wolf     };
1114df581792SKevin Wolf 
1115df581792SKevin Wolf     QLIST_INSERT_HEAD(&parent_bs->children, child, next);
1116b4b059f6SKevin Wolf 
1117b4b059f6SKevin Wolf     return child;
1118df581792SKevin Wolf }
1119df581792SKevin Wolf 
112033a60407SKevin Wolf static void bdrv_detach_child(BdrvChild *child)
112133a60407SKevin Wolf {
112233a60407SKevin Wolf     QLIST_REMOVE(child, next);
112333a60407SKevin Wolf     g_free(child);
112433a60407SKevin Wolf }
112533a60407SKevin Wolf 
112633a60407SKevin Wolf void bdrv_unref_child(BlockDriverState *parent, BdrvChild *child)
112733a60407SKevin Wolf {
112833a60407SKevin Wolf     BlockDriverState *child_bs = child->bs;
112933a60407SKevin Wolf 
113033a60407SKevin Wolf     if (child->bs->inherits_from == parent) {
113133a60407SKevin Wolf         child->bs->inherits_from = NULL;
113233a60407SKevin Wolf     }
113333a60407SKevin Wolf 
113433a60407SKevin Wolf     bdrv_detach_child(child);
113533a60407SKevin Wolf     bdrv_unref(child_bs);
113633a60407SKevin Wolf }
113733a60407SKevin Wolf 
11388d24cce1SFam Zheng void bdrv_set_backing_hd(BlockDriverState *bs, BlockDriverState *backing_hd)
11398d24cce1SFam Zheng {
11408d24cce1SFam Zheng 
1141826b6ca0SFam Zheng     if (bs->backing_hd) {
1142826b6ca0SFam Zheng         assert(bs->backing_blocker);
1143826b6ca0SFam Zheng         bdrv_op_unblock_all(bs->backing_hd, bs->backing_blocker);
114480a1e130SKevin Wolf         bdrv_detach_child(bs->backing_child);
1145826b6ca0SFam Zheng     } else if (backing_hd) {
1146826b6ca0SFam Zheng         error_setg(&bs->backing_blocker,
114781e5f78aSAlberto Garcia                    "node is used as backing hd of '%s'",
114881e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(bs));
1149826b6ca0SFam Zheng     }
1150826b6ca0SFam Zheng 
11518d24cce1SFam Zheng     bs->backing_hd = backing_hd;
11528d24cce1SFam Zheng     if (!backing_hd) {
1153826b6ca0SFam Zheng         error_free(bs->backing_blocker);
1154826b6ca0SFam Zheng         bs->backing_blocker = NULL;
115580a1e130SKevin Wolf         bs->backing_child = NULL;
11568d24cce1SFam Zheng         goto out;
11578d24cce1SFam Zheng     }
115880a1e130SKevin Wolf     bs->backing_child = bdrv_attach_child(bs, backing_hd, &child_backing);
11598d24cce1SFam Zheng     bs->open_flags &= ~BDRV_O_NO_BACKING;
11608d24cce1SFam Zheng     pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_hd->filename);
11618d24cce1SFam Zheng     pstrcpy(bs->backing_format, sizeof(bs->backing_format),
11628d24cce1SFam Zheng             backing_hd->drv ? backing_hd->drv->format_name : "");
1163826b6ca0SFam Zheng 
1164826b6ca0SFam Zheng     bdrv_op_block_all(bs->backing_hd, bs->backing_blocker);
1165826b6ca0SFam Zheng     /* Otherwise we won't be able to commit due to check in bdrv_commit */
1166bb00021dSFam Zheng     bdrv_op_unblock(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET,
1167826b6ca0SFam Zheng                     bs->backing_blocker);
11688d24cce1SFam Zheng out:
11693baca891SKevin Wolf     bdrv_refresh_limits(bs, NULL);
11708d24cce1SFam Zheng }
11718d24cce1SFam Zheng 
117231ca6d07SKevin Wolf /*
117331ca6d07SKevin Wolf  * Opens the backing file for a BlockDriverState if not yet open
117431ca6d07SKevin Wolf  *
117531ca6d07SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
117631ca6d07SKevin Wolf  * empty set of options. The reference to the QDict is transferred to this
117731ca6d07SKevin Wolf  * function (even on failure), so if the caller intends to reuse the dictionary,
117831ca6d07SKevin Wolf  * it needs to use QINCREF() before calling bdrv_file_open.
117931ca6d07SKevin Wolf  */
118034b5d2c6SMax Reitz int bdrv_open_backing_file(BlockDriverState *bs, QDict *options, Error **errp)
11819156df12SPaolo Bonzini {
11821ba4b6a5SBenoît Canet     char *backing_filename = g_malloc0(PATH_MAX);
1183317fc44eSKevin Wolf     int ret = 0;
11848d24cce1SFam Zheng     BlockDriverState *backing_hd;
118534b5d2c6SMax Reitz     Error *local_err = NULL;
11869156df12SPaolo Bonzini 
11879156df12SPaolo Bonzini     if (bs->backing_hd != NULL) {
118831ca6d07SKevin Wolf         QDECREF(options);
11891ba4b6a5SBenoît Canet         goto free_exit;
11909156df12SPaolo Bonzini     }
11919156df12SPaolo Bonzini 
119231ca6d07SKevin Wolf     /* NULL means an empty set of options */
119331ca6d07SKevin Wolf     if (options == NULL) {
119431ca6d07SKevin Wolf         options = qdict_new();
119531ca6d07SKevin Wolf     }
119631ca6d07SKevin Wolf 
11979156df12SPaolo Bonzini     bs->open_flags &= ~BDRV_O_NO_BACKING;
11981cb6f506SKevin Wolf     if (qdict_haskey(options, "file.filename")) {
11991cb6f506SKevin Wolf         backing_filename[0] = '\0';
12001cb6f506SKevin Wolf     } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
120131ca6d07SKevin Wolf         QDECREF(options);
12021ba4b6a5SBenoît Canet         goto free_exit;
1203dbecebddSFam Zheng     } else {
12049f07429eSMax Reitz         bdrv_get_full_backing_filename(bs, backing_filename, PATH_MAX,
12059f07429eSMax Reitz                                        &local_err);
12069f07429eSMax Reitz         if (local_err) {
12079f07429eSMax Reitz             ret = -EINVAL;
12089f07429eSMax Reitz             error_propagate(errp, local_err);
12099f07429eSMax Reitz             QDECREF(options);
12109f07429eSMax Reitz             goto free_exit;
12119f07429eSMax Reitz         }
12129156df12SPaolo Bonzini     }
12139156df12SPaolo Bonzini 
12148ee79e70SKevin Wolf     if (!bs->drv || !bs->drv->supports_backing) {
12158ee79e70SKevin Wolf         ret = -EINVAL;
12168ee79e70SKevin Wolf         error_setg(errp, "Driver doesn't support backing files");
12178ee79e70SKevin Wolf         QDECREF(options);
12188ee79e70SKevin Wolf         goto free_exit;
12198ee79e70SKevin Wolf     }
12208ee79e70SKevin Wolf 
1221e4e9986bSMarkus Armbruster     backing_hd = bdrv_new();
12228d24cce1SFam Zheng 
1223c5f6e493SKevin Wolf     if (bs->backing_format[0] != '\0' && !qdict_haskey(options, "driver")) {
1224c5f6e493SKevin Wolf         qdict_put(options, "driver", qstring_from_str(bs->backing_format));
12259156df12SPaolo Bonzini     }
12269156df12SPaolo Bonzini 
1227f67503e5SMax Reitz     assert(bs->backing_hd == NULL);
1228f3930ed0SKevin Wolf     ret = bdrv_open_inherit(&backing_hd,
1229f3930ed0SKevin Wolf                             *backing_filename ? backing_filename : NULL,
1230f3930ed0SKevin Wolf                             NULL, options, 0, bs, &child_backing,
1231f3930ed0SKevin Wolf                             NULL, &local_err);
12329156df12SPaolo Bonzini     if (ret < 0) {
12338d24cce1SFam Zheng         bdrv_unref(backing_hd);
12348d24cce1SFam Zheng         backing_hd = NULL;
12359156df12SPaolo Bonzini         bs->open_flags |= BDRV_O_NO_BACKING;
1236b04b6b6eSFam Zheng         error_setg(errp, "Could not open backing file: %s",
1237b04b6b6eSFam Zheng                    error_get_pretty(local_err));
1238b04b6b6eSFam Zheng         error_free(local_err);
12391ba4b6a5SBenoît Canet         goto free_exit;
12409156df12SPaolo Bonzini     }
1241df581792SKevin Wolf 
12428d24cce1SFam Zheng     bdrv_set_backing_hd(bs, backing_hd);
1243d80ac658SPeter Feiner 
12441ba4b6a5SBenoît Canet free_exit:
12451ba4b6a5SBenoît Canet     g_free(backing_filename);
12461ba4b6a5SBenoît Canet     return ret;
12479156df12SPaolo Bonzini }
12489156df12SPaolo Bonzini 
1249b6ce07aaSKevin Wolf /*
1250da557aacSMax Reitz  * Opens a disk image whose options are given as BlockdevRef in another block
1251da557aacSMax Reitz  * device's options.
1252da557aacSMax Reitz  *
1253da557aacSMax Reitz  * If allow_none is true, no image will be opened if filename is false and no
1254b4b059f6SKevin Wolf  * BlockdevRef is given. NULL will be returned, but errp remains unset.
1255da557aacSMax Reitz  *
1256da557aacSMax Reitz  * bdrev_key specifies the key for the image's BlockdevRef in the options QDict.
1257da557aacSMax Reitz  * That QDict has to be flattened; therefore, if the BlockdevRef is a QDict
1258da557aacSMax Reitz  * itself, all options starting with "${bdref_key}." are considered part of the
1259da557aacSMax Reitz  * BlockdevRef.
1260da557aacSMax Reitz  *
1261da557aacSMax Reitz  * The BlockdevRef will be removed from the options QDict.
1262da557aacSMax Reitz  */
1263b4b059f6SKevin Wolf BdrvChild *bdrv_open_child(const char *filename,
1264f3930ed0SKevin Wolf                            QDict *options, const char *bdref_key,
1265b4b059f6SKevin Wolf                            BlockDriverState* parent,
1266b4b059f6SKevin Wolf                            const BdrvChildRole *child_role,
1267f7d9fd8cSMax Reitz                            bool allow_none, Error **errp)
1268da557aacSMax Reitz {
1269b4b059f6SKevin Wolf     BdrvChild *c = NULL;
1270b4b059f6SKevin Wolf     BlockDriverState *bs;
1271da557aacSMax Reitz     QDict *image_options;
1272da557aacSMax Reitz     int ret;
1273da557aacSMax Reitz     char *bdref_key_dot;
1274da557aacSMax Reitz     const char *reference;
1275da557aacSMax Reitz 
1276df581792SKevin Wolf     assert(child_role != NULL);
1277f67503e5SMax Reitz 
1278da557aacSMax Reitz     bdref_key_dot = g_strdup_printf("%s.", bdref_key);
1279da557aacSMax Reitz     qdict_extract_subqdict(options, &image_options, bdref_key_dot);
1280da557aacSMax Reitz     g_free(bdref_key_dot);
1281da557aacSMax Reitz 
1282da557aacSMax Reitz     reference = qdict_get_try_str(options, bdref_key);
1283da557aacSMax Reitz     if (!filename && !reference && !qdict_size(image_options)) {
1284b4b059f6SKevin Wolf         if (!allow_none) {
1285da557aacSMax Reitz             error_setg(errp, "A block device must be specified for \"%s\"",
1286da557aacSMax Reitz                        bdref_key);
1287da557aacSMax Reitz         }
1288b20e61e0SMarkus Armbruster         QDECREF(image_options);
1289da557aacSMax Reitz         goto done;
1290da557aacSMax Reitz     }
1291da557aacSMax Reitz 
1292b4b059f6SKevin Wolf     bs = NULL;
1293b4b059f6SKevin Wolf     ret = bdrv_open_inherit(&bs, filename, reference, image_options, 0,
1294f3930ed0SKevin Wolf                             parent, child_role, NULL, errp);
1295df581792SKevin Wolf     if (ret < 0) {
1296df581792SKevin Wolf         goto done;
1297df581792SKevin Wolf     }
1298df581792SKevin Wolf 
1299b4b059f6SKevin Wolf     c = bdrv_attach_child(parent, bs, child_role);
1300da557aacSMax Reitz 
1301da557aacSMax Reitz done:
1302da557aacSMax Reitz     qdict_del(options, bdref_key);
1303b4b059f6SKevin Wolf     return c;
1304b4b059f6SKevin Wolf }
1305b4b059f6SKevin Wolf 
1306b4b059f6SKevin Wolf /*
1307b4b059f6SKevin Wolf  * This is a version of bdrv_open_child() that returns 0/-EINVAL instead of
1308b4b059f6SKevin Wolf  * a BdrvChild object.
1309b4b059f6SKevin Wolf  *
1310b4b059f6SKevin Wolf  * If allow_none is true, no image will be opened if filename is false and no
1311b4b059f6SKevin Wolf  * BlockdevRef is given. *pbs will remain unchanged and 0 will be returned.
1312b4b059f6SKevin Wolf  *
1313b4b059f6SKevin Wolf  * To conform with the behavior of bdrv_open(), *pbs has to be NULL.
1314b4b059f6SKevin Wolf  */
1315b4b059f6SKevin Wolf int bdrv_open_image(BlockDriverState **pbs, const char *filename,
1316b4b059f6SKevin Wolf                     QDict *options, const char *bdref_key,
1317b4b059f6SKevin Wolf                     BlockDriverState* parent, const BdrvChildRole *child_role,
1318b4b059f6SKevin Wolf                     bool allow_none, Error **errp)
1319b4b059f6SKevin Wolf {
1320b4b059f6SKevin Wolf     Error *local_err = NULL;
1321b4b059f6SKevin Wolf     BdrvChild *c;
1322b4b059f6SKevin Wolf 
1323b4b059f6SKevin Wolf     assert(pbs);
1324b4b059f6SKevin Wolf     assert(*pbs == NULL);
1325b4b059f6SKevin Wolf 
1326b4b059f6SKevin Wolf     c = bdrv_open_child(filename, options, bdref_key, parent, child_role,
1327b4b059f6SKevin Wolf                         allow_none, &local_err);
1328b4b059f6SKevin Wolf     if (local_err) {
1329b4b059f6SKevin Wolf         error_propagate(errp, local_err);
1330b4b059f6SKevin Wolf         return -EINVAL;
1331b4b059f6SKevin Wolf     }
1332b4b059f6SKevin Wolf 
1333b4b059f6SKevin Wolf     if (c != NULL) {
1334b4b059f6SKevin Wolf         *pbs = c->bs;
1335b4b059f6SKevin Wolf     }
1336b4b059f6SKevin Wolf 
1337b4b059f6SKevin Wolf     return 0;
1338da557aacSMax Reitz }
1339da557aacSMax Reitz 
13406b8aeca5SChen Gang int bdrv_append_temp_snapshot(BlockDriverState *bs, int flags, Error **errp)
1341b998875dSKevin Wolf {
1342b998875dSKevin Wolf     /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
13431ba4b6a5SBenoît Canet     char *tmp_filename = g_malloc0(PATH_MAX + 1);
1344b998875dSKevin Wolf     int64_t total_size;
134583d0521aSChunyan Liu     QemuOpts *opts = NULL;
1346b998875dSKevin Wolf     QDict *snapshot_options;
1347b998875dSKevin Wolf     BlockDriverState *bs_snapshot;
1348c2e0dbbfSFam Zheng     Error *local_err = NULL;
1349b998875dSKevin Wolf     int ret;
1350b998875dSKevin Wolf 
1351b998875dSKevin Wolf     /* if snapshot, we create a temporary backing file and open it
1352b998875dSKevin Wolf        instead of opening 'filename' directly */
1353b998875dSKevin Wolf 
1354b998875dSKevin Wolf     /* Get the required size from the image */
1355f187743aSKevin Wolf     total_size = bdrv_getlength(bs);
1356f187743aSKevin Wolf     if (total_size < 0) {
13576b8aeca5SChen Gang         ret = total_size;
1358f187743aSKevin Wolf         error_setg_errno(errp, -total_size, "Could not get image size");
13591ba4b6a5SBenoît Canet         goto out;
1360f187743aSKevin Wolf     }
1361b998875dSKevin Wolf 
1362b998875dSKevin Wolf     /* Create the temporary image */
13631ba4b6a5SBenoît Canet     ret = get_tmp_filename(tmp_filename, PATH_MAX + 1);
1364b998875dSKevin Wolf     if (ret < 0) {
1365b998875dSKevin Wolf         error_setg_errno(errp, -ret, "Could not get temporary filename");
13661ba4b6a5SBenoît Canet         goto out;
1367b998875dSKevin Wolf     }
1368b998875dSKevin Wolf 
1369ef810437SMax Reitz     opts = qemu_opts_create(bdrv_qcow2.create_opts, NULL, 0,
1370c282e1fdSChunyan Liu                             &error_abort);
137139101f25SMarkus Armbruster     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, total_size, &error_abort);
1372ef810437SMax Reitz     ret = bdrv_create(&bdrv_qcow2, tmp_filename, opts, &local_err);
137383d0521aSChunyan Liu     qemu_opts_del(opts);
1374b998875dSKevin Wolf     if (ret < 0) {
1375b998875dSKevin Wolf         error_setg_errno(errp, -ret, "Could not create temporary overlay "
1376b998875dSKevin Wolf                          "'%s': %s", tmp_filename,
1377b998875dSKevin Wolf                          error_get_pretty(local_err));
1378b998875dSKevin Wolf         error_free(local_err);
13791ba4b6a5SBenoît Canet         goto out;
1380b998875dSKevin Wolf     }
1381b998875dSKevin Wolf 
1382b998875dSKevin Wolf     /* Prepare a new options QDict for the temporary file */
1383b998875dSKevin Wolf     snapshot_options = qdict_new();
1384b998875dSKevin Wolf     qdict_put(snapshot_options, "file.driver",
1385b998875dSKevin Wolf               qstring_from_str("file"));
1386b998875dSKevin Wolf     qdict_put(snapshot_options, "file.filename",
1387b998875dSKevin Wolf               qstring_from_str(tmp_filename));
1388e6641719SMax Reitz     qdict_put(snapshot_options, "driver",
1389e6641719SMax Reitz               qstring_from_str("qcow2"));
1390b998875dSKevin Wolf 
1391e4e9986bSMarkus Armbruster     bs_snapshot = bdrv_new();
1392b998875dSKevin Wolf 
1393b998875dSKevin Wolf     ret = bdrv_open(&bs_snapshot, NULL, NULL, snapshot_options,
1394*6ebf9aa2SMax Reitz                     flags, &local_err);
1395b998875dSKevin Wolf     if (ret < 0) {
1396b998875dSKevin Wolf         error_propagate(errp, local_err);
13971ba4b6a5SBenoît Canet         goto out;
1398b998875dSKevin Wolf     }
1399b998875dSKevin Wolf 
1400b998875dSKevin Wolf     bdrv_append(bs_snapshot, bs);
14011ba4b6a5SBenoît Canet 
14021ba4b6a5SBenoît Canet out:
14031ba4b6a5SBenoît Canet     g_free(tmp_filename);
14046b8aeca5SChen Gang     return ret;
1405b998875dSKevin Wolf }
1406b998875dSKevin Wolf 
1407da557aacSMax Reitz /*
1408b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
1409de9c0cecSKevin Wolf  *
1410de9c0cecSKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
1411de9c0cecSKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
1412de9c0cecSKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
1413de9c0cecSKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_open.
1414f67503e5SMax Reitz  *
1415f67503e5SMax Reitz  * If *pbs is NULL, a new BDS will be created with a pointer to it stored there.
1416f67503e5SMax Reitz  * If it is not NULL, the referenced BDS will be reused.
1417ddf5636dSMax Reitz  *
1418ddf5636dSMax Reitz  * The reference parameter may be used to specify an existing block device which
1419ddf5636dSMax Reitz  * should be opened. If specified, neither options nor a filename may be given,
1420ddf5636dSMax Reitz  * nor can an existing BDS be reused (that is, *pbs has to be NULL).
1421b6ce07aaSKevin Wolf  */
1422f3930ed0SKevin Wolf static int bdrv_open_inherit(BlockDriverState **pbs, const char *filename,
1423ddf5636dSMax Reitz                              const char *reference, QDict *options, int flags,
1424f3930ed0SKevin Wolf                              BlockDriverState *parent,
1425f3930ed0SKevin Wolf                              const BdrvChildRole *child_role,
1426ddf5636dSMax Reitz                              BlockDriver *drv, Error **errp)
1427ea2384d3Sbellard {
1428b6ce07aaSKevin Wolf     int ret;
1429f67503e5SMax Reitz     BlockDriverState *file = NULL, *bs;
143074fe54f2SKevin Wolf     const char *drvname;
143134b5d2c6SMax Reitz     Error *local_err = NULL;
1432b1e6fc08SKevin Wolf     int snapshot_flags = 0;
143333e3963eSbellard 
1434f67503e5SMax Reitz     assert(pbs);
1435f3930ed0SKevin Wolf     assert(!child_role || !flags);
1436f3930ed0SKevin Wolf     assert(!child_role == !parent);
1437f67503e5SMax Reitz 
1438ddf5636dSMax Reitz     if (reference) {
1439ddf5636dSMax Reitz         bool options_non_empty = options ? qdict_size(options) : false;
1440ddf5636dSMax Reitz         QDECREF(options);
1441ddf5636dSMax Reitz 
1442ddf5636dSMax Reitz         if (*pbs) {
1443ddf5636dSMax Reitz             error_setg(errp, "Cannot reuse an existing BDS when referencing "
1444ddf5636dSMax Reitz                        "another block device");
1445ddf5636dSMax Reitz             return -EINVAL;
1446ddf5636dSMax Reitz         }
1447ddf5636dSMax Reitz 
1448ddf5636dSMax Reitz         if (filename || options_non_empty) {
1449ddf5636dSMax Reitz             error_setg(errp, "Cannot reference an existing block device with "
1450ddf5636dSMax Reitz                        "additional options or a new filename");
1451ddf5636dSMax Reitz             return -EINVAL;
1452ddf5636dSMax Reitz         }
1453ddf5636dSMax Reitz 
1454ddf5636dSMax Reitz         bs = bdrv_lookup_bs(reference, reference, errp);
1455ddf5636dSMax Reitz         if (!bs) {
1456ddf5636dSMax Reitz             return -ENODEV;
1457ddf5636dSMax Reitz         }
1458ddf5636dSMax Reitz         bdrv_ref(bs);
1459ddf5636dSMax Reitz         *pbs = bs;
1460ddf5636dSMax Reitz         return 0;
1461ddf5636dSMax Reitz     }
1462ddf5636dSMax Reitz 
1463f67503e5SMax Reitz     if (*pbs) {
1464f67503e5SMax Reitz         bs = *pbs;
1465f67503e5SMax Reitz     } else {
1466e4e9986bSMarkus Armbruster         bs = bdrv_new();
1467f67503e5SMax Reitz     }
1468f67503e5SMax Reitz 
1469de9c0cecSKevin Wolf     /* NULL means an empty set of options */
1470de9c0cecSKevin Wolf     if (options == NULL) {
1471de9c0cecSKevin Wolf         options = qdict_new();
1472de9c0cecSKevin Wolf     }
1473de9c0cecSKevin Wolf 
1474f3930ed0SKevin Wolf     if (child_role) {
1475bddcec37SKevin Wolf         bs->inherits_from = parent;
1476f3930ed0SKevin Wolf         flags = child_role->inherit_flags(parent->open_flags);
1477f3930ed0SKevin Wolf     }
1478f3930ed0SKevin Wolf 
147953a29513SMax Reitz     ret = bdrv_fill_options(&options, &filename, &flags, drv, &local_err);
1480462f5bcfSKevin Wolf     if (local_err) {
1481462f5bcfSKevin Wolf         goto fail;
1482462f5bcfSKevin Wolf     }
1483462f5bcfSKevin Wolf 
148476c591b0SKevin Wolf     /* Find the right image format driver */
148576c591b0SKevin Wolf     drv = NULL;
148676c591b0SKevin Wolf     drvname = qdict_get_try_str(options, "driver");
148776c591b0SKevin Wolf     if (drvname) {
148876c591b0SKevin Wolf         drv = bdrv_find_format(drvname);
148976c591b0SKevin Wolf         qdict_del(options, "driver");
149076c591b0SKevin Wolf         if (!drv) {
149176c591b0SKevin Wolf             error_setg(errp, "Unknown driver: '%s'", drvname);
149276c591b0SKevin Wolf             ret = -EINVAL;
149376c591b0SKevin Wolf             goto fail;
149476c591b0SKevin Wolf         }
149576c591b0SKevin Wolf     }
149676c591b0SKevin Wolf 
149776c591b0SKevin Wolf     assert(drvname || !(flags & BDRV_O_PROTOCOL));
149876c591b0SKevin Wolf 
1499f3930ed0SKevin Wolf     bs->open_flags = flags;
1500de9c0cecSKevin Wolf     bs->options = options;
1501b6ad491aSKevin Wolf     options = qdict_clone_shallow(options);
1502de9c0cecSKevin Wolf 
1503f500a6d3SKevin Wolf     /* Open image file without format layer */
1504f4788adcSKevin Wolf     if ((flags & BDRV_O_PROTOCOL) == 0) {
1505be028adcSJeff Cody         if (flags & BDRV_O_RDWR) {
1506be028adcSJeff Cody             flags |= BDRV_O_ALLOW_RDWR;
1507be028adcSJeff Cody         }
1508b1e6fc08SKevin Wolf         if (flags & BDRV_O_SNAPSHOT) {
1509b1e6fc08SKevin Wolf             snapshot_flags = bdrv_temp_snapshot_flags(flags);
1510b1e6fc08SKevin Wolf             flags = bdrv_backing_flags(flags);
1511b1e6fc08SKevin Wolf         }
1512be028adcSJeff Cody 
1513f67503e5SMax Reitz         assert(file == NULL);
1514f3930ed0SKevin Wolf         bs->open_flags = flags;
1515054963f8SMax Reitz         ret = bdrv_open_image(&file, filename, options, "file",
1516f3930ed0SKevin Wolf                               bs, &child_file, true, &local_err);
1517f500a6d3SKevin Wolf         if (ret < 0) {
15188bfea15dSKevin Wolf             goto fail;
1519f500a6d3SKevin Wolf         }
1520f4788adcSKevin Wolf     }
1521f500a6d3SKevin Wolf 
152276c591b0SKevin Wolf     /* Image format probing */
152338f3ef57SKevin Wolf     bs->probed = !drv;
152476c591b0SKevin Wolf     if (!drv && file) {
152534b5d2c6SMax Reitz         ret = find_image_format(file, filename, &drv, &local_err);
152617b005f1SKevin Wolf         if (ret < 0) {
152717b005f1SKevin Wolf             goto fail;
152817b005f1SKevin Wolf         }
152976c591b0SKevin Wolf     } else if (!drv) {
15302a05cbe4SMax Reitz         error_setg(errp, "Must specify either driver or file");
15312a05cbe4SMax Reitz         ret = -EINVAL;
15328bfea15dSKevin Wolf         goto fail;
15332a05cbe4SMax Reitz     }
1534f500a6d3SKevin Wolf 
153553a29513SMax Reitz     /* BDRV_O_PROTOCOL must be set iff a protocol BDS is about to be created */
153653a29513SMax Reitz     assert(!!(flags & BDRV_O_PROTOCOL) == !!drv->bdrv_file_open);
153753a29513SMax Reitz     /* file must be NULL if a protocol BDS is about to be created
153853a29513SMax Reitz      * (the inverse results in an error message from bdrv_open_common()) */
153953a29513SMax Reitz     assert(!(flags & BDRV_O_PROTOCOL) || !file);
154053a29513SMax Reitz 
1541b6ce07aaSKevin Wolf     /* Open the image */
154234b5d2c6SMax Reitz     ret = bdrv_open_common(bs, file, options, flags, drv, &local_err);
1543b6ce07aaSKevin Wolf     if (ret < 0) {
15448bfea15dSKevin Wolf         goto fail;
15456987307cSChristoph Hellwig     }
15466987307cSChristoph Hellwig 
15472a05cbe4SMax Reitz     if (file && (bs->file != file)) {
15484f6fd349SFam Zheng         bdrv_unref(file);
1549f500a6d3SKevin Wolf         file = NULL;
1550f500a6d3SKevin Wolf     }
1551f500a6d3SKevin Wolf 
1552b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
15539156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
155431ca6d07SKevin Wolf         QDict *backing_options;
155531ca6d07SKevin Wolf 
15565726d872SBenoît Canet         qdict_extract_subqdict(options, &backing_options, "backing.");
155734b5d2c6SMax Reitz         ret = bdrv_open_backing_file(bs, backing_options, &local_err);
1558b6ce07aaSKevin Wolf         if (ret < 0) {
1559b6ad491aSKevin Wolf             goto close_and_fail;
1560b6ce07aaSKevin Wolf         }
1561b6ce07aaSKevin Wolf     }
1562b6ce07aaSKevin Wolf 
156391af7014SMax Reitz     bdrv_refresh_filename(bs);
156491af7014SMax Reitz 
1565b998875dSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay. bs points to the
1566b998875dSKevin Wolf      * temporary snapshot afterwards. */
1567b1e6fc08SKevin Wolf     if (snapshot_flags) {
15686b8aeca5SChen Gang         ret = bdrv_append_temp_snapshot(bs, snapshot_flags, &local_err);
1569b998875dSKevin Wolf         if (local_err) {
1570b998875dSKevin Wolf             goto close_and_fail;
1571b998875dSKevin Wolf         }
1572b998875dSKevin Wolf     }
1573b998875dSKevin Wolf 
1574b6ad491aSKevin Wolf     /* Check if any unknown options were used */
15755acd9d81SMax Reitz     if (options && (qdict_size(options) != 0)) {
1576b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
15775acd9d81SMax Reitz         if (flags & BDRV_O_PROTOCOL) {
15785acd9d81SMax Reitz             error_setg(errp, "Block protocol '%s' doesn't support the option "
15795acd9d81SMax Reitz                        "'%s'", drv->format_name, entry->key);
15805acd9d81SMax Reitz         } else {
158134b5d2c6SMax Reitz             error_setg(errp, "Block format '%s' used by device '%s' doesn't "
15825acd9d81SMax Reitz                        "support the option '%s'", drv->format_name,
1583bfb197e0SMarkus Armbruster                        bdrv_get_device_name(bs), entry->key);
15845acd9d81SMax Reitz         }
1585b6ad491aSKevin Wolf 
1586b6ad491aSKevin Wolf         ret = -EINVAL;
1587b6ad491aSKevin Wolf         goto close_and_fail;
1588b6ad491aSKevin Wolf     }
1589b6ad491aSKevin Wolf 
1590b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
1591a7f53e26SMarkus Armbruster         if (bs->blk) {
1592a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
1593a7f53e26SMarkus Armbruster         }
1594c3adb58fSMarkus Armbruster     } else if (!runstate_check(RUN_STATE_PRELAUNCH)
1595c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_INMIGRATE)
1596c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_PAUSED)) { /* HACK */
1597c3adb58fSMarkus Armbruster         error_setg(errp,
1598c3adb58fSMarkus Armbruster                    "Guest must be stopped for opening of encrypted image");
1599c3adb58fSMarkus Armbruster         ret = -EBUSY;
1600c3adb58fSMarkus Armbruster         goto close_and_fail;
1601b6ce07aaSKevin Wolf     }
1602b6ce07aaSKevin Wolf 
1603c3adb58fSMarkus Armbruster     QDECREF(options);
1604f67503e5SMax Reitz     *pbs = bs;
1605b6ce07aaSKevin Wolf     return 0;
1606b6ce07aaSKevin Wolf 
16078bfea15dSKevin Wolf fail:
1608f500a6d3SKevin Wolf     if (file != NULL) {
16094f6fd349SFam Zheng         bdrv_unref(file);
1610f500a6d3SKevin Wolf     }
1611de9c0cecSKevin Wolf     QDECREF(bs->options);
1612b6ad491aSKevin Wolf     QDECREF(options);
1613de9c0cecSKevin Wolf     bs->options = NULL;
1614f67503e5SMax Reitz     if (!*pbs) {
1615f67503e5SMax Reitz         /* If *pbs is NULL, a new BDS has been created in this function and
1616f67503e5SMax Reitz            needs to be freed now. Otherwise, it does not need to be closed,
1617f67503e5SMax Reitz            since it has not really been opened yet. */
1618f67503e5SMax Reitz         bdrv_unref(bs);
1619f67503e5SMax Reitz     }
162084d18f06SMarkus Armbruster     if (local_err) {
162134b5d2c6SMax Reitz         error_propagate(errp, local_err);
162234b5d2c6SMax Reitz     }
1623b6ad491aSKevin Wolf     return ret;
1624de9c0cecSKevin Wolf 
1625b6ad491aSKevin Wolf close_and_fail:
1626f67503e5SMax Reitz     /* See fail path, but now the BDS has to be always closed */
1627f67503e5SMax Reitz     if (*pbs) {
1628b6ad491aSKevin Wolf         bdrv_close(bs);
1629f67503e5SMax Reitz     } else {
1630f67503e5SMax Reitz         bdrv_unref(bs);
1631f67503e5SMax Reitz     }
1632b6ad491aSKevin Wolf     QDECREF(options);
163384d18f06SMarkus Armbruster     if (local_err) {
163434b5d2c6SMax Reitz         error_propagate(errp, local_err);
163534b5d2c6SMax Reitz     }
1636b6ce07aaSKevin Wolf     return ret;
1637b6ce07aaSKevin Wolf }
1638b6ce07aaSKevin Wolf 
1639f3930ed0SKevin Wolf int bdrv_open(BlockDriverState **pbs, const char *filename,
1640*6ebf9aa2SMax Reitz               const char *reference, QDict *options, int flags, Error **errp)
1641f3930ed0SKevin Wolf {
1642f3930ed0SKevin Wolf     return bdrv_open_inherit(pbs, filename, reference, options, flags, NULL,
1643*6ebf9aa2SMax Reitz                              NULL, NULL, errp);
1644f3930ed0SKevin Wolf }
1645f3930ed0SKevin Wolf 
1646e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1647e971aa12SJeff Cody      bool prepared;
1648e971aa12SJeff Cody      BDRVReopenState state;
1649e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1650e971aa12SJeff Cody } BlockReopenQueueEntry;
1651e971aa12SJeff Cody 
1652e971aa12SJeff Cody /*
1653e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1654e971aa12SJeff Cody  * reopen of multiple devices.
1655e971aa12SJeff Cody  *
1656e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1657e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1658e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1659e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1660e971aa12SJeff Cody  * atomic 'set'.
1661e971aa12SJeff Cody  *
1662e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1663e971aa12SJeff Cody  *
1664e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1665e971aa12SJeff Cody  *
1666e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1667e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1668e971aa12SJeff Cody  *
1669e971aa12SJeff Cody  */
1670e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1671e971aa12SJeff Cody                                     BlockDriverState *bs, int flags)
1672e971aa12SJeff Cody {
1673e971aa12SJeff Cody     assert(bs != NULL);
1674e971aa12SJeff Cody 
1675e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
167667251a31SKevin Wolf     BdrvChild *child;
167767251a31SKevin Wolf 
1678e971aa12SJeff Cody     if (bs_queue == NULL) {
1679e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1680e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1681e971aa12SJeff Cody     }
1682e971aa12SJeff Cody 
1683f1f25a2eSKevin Wolf     /* bdrv_open() masks this flag out */
1684f1f25a2eSKevin Wolf     flags &= ~BDRV_O_PROTOCOL;
1685f1f25a2eSKevin Wolf 
168667251a31SKevin Wolf     QLIST_FOREACH(child, &bs->children, next) {
168767251a31SKevin Wolf         int child_flags;
168867251a31SKevin Wolf 
168967251a31SKevin Wolf         if (child->bs->inherits_from != bs) {
169067251a31SKevin Wolf             continue;
169167251a31SKevin Wolf         }
169267251a31SKevin Wolf 
169367251a31SKevin Wolf         child_flags = child->role->inherit_flags(flags);
169467251a31SKevin Wolf         bdrv_reopen_queue(bs_queue, child->bs, child_flags);
1695e971aa12SJeff Cody     }
1696e971aa12SJeff Cody 
1697e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1698e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1699e971aa12SJeff Cody 
1700e971aa12SJeff Cody     bs_entry->state.bs = bs;
1701e971aa12SJeff Cody     bs_entry->state.flags = flags;
1702e971aa12SJeff Cody 
1703e971aa12SJeff Cody     return bs_queue;
1704e971aa12SJeff Cody }
1705e971aa12SJeff Cody 
1706e971aa12SJeff Cody /*
1707e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1708e971aa12SJeff Cody  *
1709e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1710e971aa12SJeff Cody  * via bdrv_reopen_queue().
1711e971aa12SJeff Cody  *
1712e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1713e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1714e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1715e971aa12SJeff Cody  * data cleaned up.
1716e971aa12SJeff Cody  *
1717e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1718e971aa12SJeff Cody  * to all devices.
1719e971aa12SJeff Cody  *
1720e971aa12SJeff Cody  */
1721e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1722e971aa12SJeff Cody {
1723e971aa12SJeff Cody     int ret = -1;
1724e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1725e971aa12SJeff Cody     Error *local_err = NULL;
1726e971aa12SJeff Cody 
1727e971aa12SJeff Cody     assert(bs_queue != NULL);
1728e971aa12SJeff Cody 
1729e971aa12SJeff Cody     bdrv_drain_all();
1730e971aa12SJeff Cody 
1731e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1732e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1733e971aa12SJeff Cody             error_propagate(errp, local_err);
1734e971aa12SJeff Cody             goto cleanup;
1735e971aa12SJeff Cody         }
1736e971aa12SJeff Cody         bs_entry->prepared = true;
1737e971aa12SJeff Cody     }
1738e971aa12SJeff Cody 
1739e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1740e971aa12SJeff Cody      * changes
1741e971aa12SJeff Cody      */
1742e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1743e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1744e971aa12SJeff Cody     }
1745e971aa12SJeff Cody 
1746e971aa12SJeff Cody     ret = 0;
1747e971aa12SJeff Cody 
1748e971aa12SJeff Cody cleanup:
1749e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1750e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1751e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1752e971aa12SJeff Cody         }
1753e971aa12SJeff Cody         g_free(bs_entry);
1754e971aa12SJeff Cody     }
1755e971aa12SJeff Cody     g_free(bs_queue);
1756e971aa12SJeff Cody     return ret;
1757e971aa12SJeff Cody }
1758e971aa12SJeff Cody 
1759e971aa12SJeff Cody 
1760e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1761e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1762e971aa12SJeff Cody {
1763e971aa12SJeff Cody     int ret = -1;
1764e971aa12SJeff Cody     Error *local_err = NULL;
1765e971aa12SJeff Cody     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, bdrv_flags);
1766e971aa12SJeff Cody 
1767e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1768e971aa12SJeff Cody     if (local_err != NULL) {
1769e971aa12SJeff Cody         error_propagate(errp, local_err);
1770e971aa12SJeff Cody     }
1771e971aa12SJeff Cody     return ret;
1772e971aa12SJeff Cody }
1773e971aa12SJeff Cody 
1774e971aa12SJeff Cody 
1775e971aa12SJeff Cody /*
1776e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1777e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1778e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1779e971aa12SJeff Cody  *
1780e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1781e971aa12SJeff Cody  * flags are the new open flags
1782e971aa12SJeff Cody  * queue is the reopen queue
1783e971aa12SJeff Cody  *
1784e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1785e971aa12SJeff Cody  * as well.
1786e971aa12SJeff Cody  *
1787e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1788e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1789e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1790e971aa12SJeff Cody  *
1791e971aa12SJeff Cody  */
1792e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1793e971aa12SJeff Cody                         Error **errp)
1794e971aa12SJeff Cody {
1795e971aa12SJeff Cody     int ret = -1;
1796e971aa12SJeff Cody     Error *local_err = NULL;
1797e971aa12SJeff Cody     BlockDriver *drv;
1798e971aa12SJeff Cody 
1799e971aa12SJeff Cody     assert(reopen_state != NULL);
1800e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1801e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1802e971aa12SJeff Cody 
1803e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1804e971aa12SJeff Cody      * to r/w */
1805e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1806e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
180781e5f78aSAlberto Garcia         error_setg(errp, "Node '%s' is read only",
180881e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1809e971aa12SJeff Cody         goto error;
1810e971aa12SJeff Cody     }
1811e971aa12SJeff Cody 
1812e971aa12SJeff Cody 
1813e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1814e971aa12SJeff Cody     if (ret) {
1815e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1816e971aa12SJeff Cody                   strerror(-ret));
1817e971aa12SJeff Cody         goto error;
1818e971aa12SJeff Cody     }
1819e971aa12SJeff Cody 
1820e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1821e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1822e971aa12SJeff Cody         if (ret) {
1823e971aa12SJeff Cody             if (local_err != NULL) {
1824e971aa12SJeff Cody                 error_propagate(errp, local_err);
1825e971aa12SJeff Cody             } else {
1826d8b6895fSLuiz Capitulino                 error_setg(errp, "failed while preparing to reopen image '%s'",
1827e971aa12SJeff Cody                            reopen_state->bs->filename);
1828e971aa12SJeff Cody             }
1829e971aa12SJeff Cody             goto error;
1830e971aa12SJeff Cody         }
1831e971aa12SJeff Cody     } else {
1832e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1833e971aa12SJeff Cody          * handler for each supported drv. */
183481e5f78aSAlberto Garcia         error_setg(errp, "Block format '%s' used by node '%s' "
183581e5f78aSAlberto Garcia                    "does not support reopening files", drv->format_name,
183681e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1837e971aa12SJeff Cody         ret = -1;
1838e971aa12SJeff Cody         goto error;
1839e971aa12SJeff Cody     }
1840e971aa12SJeff Cody 
1841e971aa12SJeff Cody     ret = 0;
1842e971aa12SJeff Cody 
1843e971aa12SJeff Cody error:
1844e971aa12SJeff Cody     return ret;
1845e971aa12SJeff Cody }
1846e971aa12SJeff Cody 
1847e971aa12SJeff Cody /*
1848e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1849e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1850e971aa12SJeff Cody  * the active BlockDriverState contents.
1851e971aa12SJeff Cody  */
1852e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1853e971aa12SJeff Cody {
1854e971aa12SJeff Cody     BlockDriver *drv;
1855e971aa12SJeff Cody 
1856e971aa12SJeff Cody     assert(reopen_state != NULL);
1857e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1858e971aa12SJeff Cody     assert(drv != NULL);
1859e971aa12SJeff Cody 
1860e971aa12SJeff Cody     /* If there are any driver level actions to take */
1861e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1862e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1863e971aa12SJeff Cody     }
1864e971aa12SJeff Cody 
1865e971aa12SJeff Cody     /* set BDS specific flags now */
1866e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1867e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1868e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1869e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1870355ef4acSKevin Wolf 
18713baca891SKevin Wolf     bdrv_refresh_limits(reopen_state->bs, NULL);
1872e971aa12SJeff Cody }
1873e971aa12SJeff Cody 
1874e971aa12SJeff Cody /*
1875e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1876e971aa12SJeff Cody  * reopen_state
1877e971aa12SJeff Cody  */
1878e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1879e971aa12SJeff Cody {
1880e971aa12SJeff Cody     BlockDriver *drv;
1881e971aa12SJeff Cody 
1882e971aa12SJeff Cody     assert(reopen_state != NULL);
1883e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1884e971aa12SJeff Cody     assert(drv != NULL);
1885e971aa12SJeff Cody 
1886e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1887e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1888e971aa12SJeff Cody     }
1889e971aa12SJeff Cody }
1890e971aa12SJeff Cody 
1891e971aa12SJeff Cody 
1892fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1893fc01f7e7Sbellard {
189433384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
189533384421SMax Reitz 
18963e914655SPaolo Bonzini     if (bs->job) {
18973e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
18983e914655SPaolo Bonzini     }
189953ec73e2SFam Zheng     bdrv_drain(bs); /* complete I/O */
190058fda173SStefan Hajnoczi     bdrv_flush(bs);
190153ec73e2SFam Zheng     bdrv_drain(bs); /* in case flush left pending I/O */
1902d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
19037094f12fSKevin Wolf 
19043cbc002cSPaolo Bonzini     if (bs->drv) {
19056e93e7c4SKevin Wolf         BdrvChild *child, *next;
19066e93e7c4SKevin Wolf 
19079a7dedbcSKevin Wolf         bs->drv->bdrv_close(bs);
19089a7dedbcSKevin Wolf 
19099a7dedbcSKevin Wolf         if (bs->backing_hd) {
19109a7dedbcSKevin Wolf             BlockDriverState *backing_hd = bs->backing_hd;
19119a7dedbcSKevin Wolf             bdrv_set_backing_hd(bs, NULL);
19129a7dedbcSKevin Wolf             bdrv_unref(backing_hd);
19139a7dedbcSKevin Wolf         }
19149a7dedbcSKevin Wolf 
19156e93e7c4SKevin Wolf         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
191633a60407SKevin Wolf             /* TODO Remove bdrv_unref() from drivers' close function and use
191733a60407SKevin Wolf              * bdrv_unref_child() here */
1918bddcec37SKevin Wolf             if (child->bs->inherits_from == bs) {
1919bddcec37SKevin Wolf                 child->bs->inherits_from = NULL;
1920bddcec37SKevin Wolf             }
192133a60407SKevin Wolf             bdrv_detach_child(child);
19226e93e7c4SKevin Wolf         }
19236e93e7c4SKevin Wolf 
19247267c094SAnthony Liguori         g_free(bs->opaque);
1925ea2384d3Sbellard         bs->opaque = NULL;
1926ea2384d3Sbellard         bs->drv = NULL;
192753fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1928a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1929a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
19306405875cSPaolo Bonzini         bs->total_sectors = 0;
19316405875cSPaolo Bonzini         bs->encrypted = 0;
19326405875cSPaolo Bonzini         bs->valid_key = 0;
19336405875cSPaolo Bonzini         bs->sg = 0;
19340d51b4deSAsias He         bs->zero_beyond_eof = false;
1935de9c0cecSKevin Wolf         QDECREF(bs->options);
1936de9c0cecSKevin Wolf         bs->options = NULL;
193791af7014SMax Reitz         QDECREF(bs->full_open_options);
193891af7014SMax Reitz         bs->full_open_options = NULL;
1939b338082bSbellard 
194066f82ceeSKevin Wolf         if (bs->file != NULL) {
19414f6fd349SFam Zheng             bdrv_unref(bs->file);
19420ac9377dSPaolo Bonzini             bs->file = NULL;
194366f82ceeSKevin Wolf         }
19449ca11154SPavel Hrdina     }
194566f82ceeSKevin Wolf 
1946a7f53e26SMarkus Armbruster     if (bs->blk) {
1947a7f53e26SMarkus Armbruster         blk_dev_change_media_cb(bs->blk, false);
1948a7f53e26SMarkus Armbruster     }
194998f90dbaSZhi Yong Wu 
195098f90dbaSZhi Yong Wu     /*throttling disk I/O limits*/
195198f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
195298f90dbaSZhi Yong Wu         bdrv_io_limits_disable(bs);
195398f90dbaSZhi Yong Wu     }
195433384421SMax Reitz 
195533384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
195633384421SMax Reitz         g_free(ban);
195733384421SMax Reitz     }
195833384421SMax Reitz     QLIST_INIT(&bs->aio_notifiers);
1959b338082bSbellard }
1960b338082bSbellard 
19612bc93fedSMORITA Kazutaka void bdrv_close_all(void)
19622bc93fedSMORITA Kazutaka {
19632bc93fedSMORITA Kazutaka     BlockDriverState *bs;
19642bc93fedSMORITA Kazutaka 
1965dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
1966ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
1967ed78cda3SStefan Hajnoczi 
1968ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
19692bc93fedSMORITA Kazutaka         bdrv_close(bs);
1970ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
19712bc93fedSMORITA Kazutaka     }
19722bc93fedSMORITA Kazutaka }
19732bc93fedSMORITA Kazutaka 
1974dc364f4cSBenoît Canet /* make a BlockDriverState anonymous by removing from bdrv_state and
1975dc364f4cSBenoît Canet  * graph_bdrv_state list.
1976d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1977d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1978d22b2f41SRyan Harper {
1979bfb197e0SMarkus Armbruster     /*
1980bfb197e0SMarkus Armbruster      * Take care to remove bs from bdrv_states only when it's actually
1981bfb197e0SMarkus Armbruster      * in it.  Note that bs->device_list.tqe_prev is initially null,
1982bfb197e0SMarkus Armbruster      * and gets set to non-null by QTAILQ_INSERT_TAIL().  Establish
1983bfb197e0SMarkus Armbruster      * the useful invariant "bs in bdrv_states iff bs->tqe_prev" by
1984bfb197e0SMarkus Armbruster      * resetting it to null on remove.
1985bfb197e0SMarkus Armbruster      */
1986bfb197e0SMarkus Armbruster     if (bs->device_list.tqe_prev) {
1987dc364f4cSBenoît Canet         QTAILQ_REMOVE(&bdrv_states, bs, device_list);
1988bfb197e0SMarkus Armbruster         bs->device_list.tqe_prev = NULL;
1989d22b2f41SRyan Harper     }
1990dc364f4cSBenoît Canet     if (bs->node_name[0] != '\0') {
1991dc364f4cSBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs, node_list);
1992dc364f4cSBenoît Canet     }
1993dc364f4cSBenoît Canet     bs->node_name[0] = '\0';
1994d22b2f41SRyan Harper }
1995d22b2f41SRyan Harper 
1996e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
1997e023b2e2SPaolo Bonzini {
1998e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
1999e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
2000e023b2e2SPaolo Bonzini     }
2001e023b2e2SPaolo Bonzini }
2002e023b2e2SPaolo Bonzini 
20034ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
20044ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
20054ddc07caSPaolo Bonzini {
20064ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
20074ddc07caSPaolo Bonzini 
20084ddc07caSPaolo Bonzini     /* dev info */
20091b7fd729SPaolo Bonzini     bs_dest->guest_block_size   = bs_src->guest_block_size;
20104ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
20114ddc07caSPaolo Bonzini 
20124ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
20134ddc07caSPaolo Bonzini 
2014cc0681c4SBenoît Canet     /* i/o throttled req */
201576f4afb4SAlberto Garcia     bs_dest->throttle_state     = bs_src->throttle_state,
201676f4afb4SAlberto Garcia     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
201776f4afb4SAlberto Garcia     bs_dest->pending_reqs[0]    = bs_src->pending_reqs[0];
201876f4afb4SAlberto Garcia     bs_dest->pending_reqs[1]    = bs_src->pending_reqs[1];
201976f4afb4SAlberto Garcia     bs_dest->throttled_reqs[0]  = bs_src->throttled_reqs[0];
202076f4afb4SAlberto Garcia     bs_dest->throttled_reqs[1]  = bs_src->throttled_reqs[1];
202176f4afb4SAlberto Garcia     memcpy(&bs_dest->round_robin,
202276f4afb4SAlberto Garcia            &bs_src->round_robin,
202376f4afb4SAlberto Garcia            sizeof(bs_dest->round_robin));
20240e5b0a2dSBenoît Canet     memcpy(&bs_dest->throttle_timers,
20250e5b0a2dSBenoît Canet            &bs_src->throttle_timers,
20260e5b0a2dSBenoît Canet            sizeof(ThrottleTimers));
20274ddc07caSPaolo Bonzini 
20284ddc07caSPaolo Bonzini     /* r/w error */
20294ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
20304ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
20314ddc07caSPaolo Bonzini 
20324ddc07caSPaolo Bonzini     /* i/o status */
20334ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
20344ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
20354ddc07caSPaolo Bonzini 
20364ddc07caSPaolo Bonzini     /* dirty bitmap */
2037e4654d2dSFam Zheng     bs_dest->dirty_bitmaps      = bs_src->dirty_bitmaps;
20384ddc07caSPaolo Bonzini 
20399fcb0251SFam Zheng     /* reference count */
20409fcb0251SFam Zheng     bs_dest->refcnt             = bs_src->refcnt;
20419fcb0251SFam Zheng 
20424ddc07caSPaolo Bonzini     /* job */
20434ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
20444ddc07caSPaolo Bonzini 
20454ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
2046dc364f4cSBenoît Canet     bs_dest->device_list = bs_src->device_list;
20477e7d56d9SMarkus Armbruster     bs_dest->blk = bs_src->blk;
20487e7d56d9SMarkus Armbruster 
2049fbe40ff7SFam Zheng     memcpy(bs_dest->op_blockers, bs_src->op_blockers,
2050fbe40ff7SFam Zheng            sizeof(bs_dest->op_blockers));
20514ddc07caSPaolo Bonzini }
20524ddc07caSPaolo Bonzini 
20534ddc07caSPaolo Bonzini /*
20544ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
20554ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
20564ddc07caSPaolo Bonzini  * actually attached to a device.
20574ddc07caSPaolo Bonzini  *
20584ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
20594ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
20604ddc07caSPaolo Bonzini  *
2061bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
20624ddc07caSPaolo Bonzini  *
20634ddc07caSPaolo Bonzini  * This function does not create any image files.
20644ddc07caSPaolo Bonzini  */
20654ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
20664ddc07caSPaolo Bonzini {
20674ddc07caSPaolo Bonzini     BlockDriverState tmp;
2068bddcec37SKevin Wolf     BdrvChild *child;
20694ddc07caSPaolo Bonzini 
20706ee4ce1eSKevin Wolf     bdrv_drain(bs_new);
20716ee4ce1eSKevin Wolf     bdrv_drain(bs_old);
20724ddc07caSPaolo Bonzini 
207390ce8a06SBenoît Canet     /* The code needs to swap the node_name but simply swapping node_list won't
207490ce8a06SBenoît Canet      * work so first remove the nodes from the graph list, do the swap then
207590ce8a06SBenoît Canet      * insert them back if needed.
207690ce8a06SBenoît Canet      */
207790ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
207890ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_new, node_list);
207990ce8a06SBenoît Canet     }
208090ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
208190ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_old, node_list);
208290ce8a06SBenoît Canet     }
208390ce8a06SBenoît Canet 
2084db628338SAlberto Garcia     /* If the BlockDriverState is part of a throttling group acquire
2085db628338SAlberto Garcia      * its lock since we're going to mess with the protected fields.
2086db628338SAlberto Garcia      * Otherwise there's no need to worry since no one else can touch
2087db628338SAlberto Garcia      * them. */
2088db628338SAlberto Garcia     if (bs_old->throttle_state) {
2089db628338SAlberto Garcia         throttle_group_lock(bs_old);
2090db628338SAlberto Garcia     }
2091db628338SAlberto Garcia 
2092bfb197e0SMarkus Armbruster     /* bs_new must be unattached and shouldn't have anything fancy enabled */
20937e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
2094e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs_new->dirty_bitmaps));
20954ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
20964ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2097db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
20980e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
20994ddc07caSPaolo Bonzini 
21004ddc07caSPaolo Bonzini     tmp = *bs_new;
21014ddc07caSPaolo Bonzini     *bs_new = *bs_old;
21024ddc07caSPaolo Bonzini     *bs_old = tmp;
21034ddc07caSPaolo Bonzini 
21044ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
21054ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
21064ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
21074ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
21084ddc07caSPaolo Bonzini 
2109bfb197e0SMarkus Armbruster     /* bs_new must remain unattached */
21107e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
21114ddc07caSPaolo Bonzini 
21124ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
21134ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21144ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2115db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21160e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21174ddc07caSPaolo Bonzini 
2118db628338SAlberto Garcia     /* Release the ThrottleGroup lock */
2119db628338SAlberto Garcia     if (bs_old->throttle_state) {
2120db628338SAlberto Garcia         throttle_group_unlock(bs_old);
2121db628338SAlberto Garcia     }
2122db628338SAlberto Garcia 
212390ce8a06SBenoît Canet     /* insert the nodes back into the graph node list if needed */
212490ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
212590ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_new, node_list);
212690ce8a06SBenoît Canet     }
212790ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
212890ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_old, node_list);
212990ce8a06SBenoît Canet     }
213090ce8a06SBenoît Canet 
21316e93e7c4SKevin Wolf     /*
21326e93e7c4SKevin Wolf      * Update lh_first.le_prev for non-empty lists.
21336e93e7c4SKevin Wolf      *
21346e93e7c4SKevin Wolf      * The head of the op blocker list doesn't change because it is moved back
21356e93e7c4SKevin Wolf      * in bdrv_move_feature_fields().
21366e93e7c4SKevin Wolf      */
21376ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_old->tracked_requests));
21386ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_new->tracked_requests));
21396ee4ce1eSKevin Wolf 
21406e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_new->children, next);
21416e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_old->children, next);
21426e93e7c4SKevin Wolf 
2143bddcec37SKevin Wolf     /* Update references in bs->opaque and children */
2144bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_old->children, next) {
2145bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_new) {
2146bddcec37SKevin Wolf             child->bs->inherits_from = bs_old;
2147bddcec37SKevin Wolf         }
2148bddcec37SKevin Wolf     }
2149bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_new->children, next) {
2150bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_old) {
2151bddcec37SKevin Wolf             child->bs->inherits_from = bs_new;
2152bddcec37SKevin Wolf         }
2153bddcec37SKevin Wolf     }
2154bddcec37SKevin Wolf 
21554ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
21564ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
21574ddc07caSPaolo Bonzini }
21584ddc07caSPaolo Bonzini 
21598802d1fdSJeff Cody /*
21608802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
21618802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
21628802d1fdSJeff Cody  *
21638802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
21648802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
21658802d1fdSJeff Cody  *
2166bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
2167f6801b83SJeff Cody  *
21688802d1fdSJeff Cody  * This function does not create any image files.
21698802d1fdSJeff Cody  */
21708802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
21718802d1fdSJeff Cody {
21724ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
21738802d1fdSJeff Cody 
21748802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
21758802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
21768d24cce1SFam Zheng     bdrv_set_backing_hd(bs_top, bs_new);
21778802d1fdSJeff Cody }
21788802d1fdSJeff Cody 
21794f6fd349SFam Zheng static void bdrv_delete(BlockDriverState *bs)
2180b338082bSbellard {
21813e914655SPaolo Bonzini     assert(!bs->job);
21823718d8abSFam Zheng     assert(bdrv_op_blocker_is_empty(bs));
21834f6fd349SFam Zheng     assert(!bs->refcnt);
2184e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
218518846deeSMarkus Armbruster 
2186e1b5c52eSStefan Hajnoczi     bdrv_close(bs);
2187e1b5c52eSStefan Hajnoczi 
21881b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
2189d22b2f41SRyan Harper     bdrv_make_anon(bs);
219034c6f050Saurel32 
21917267c094SAnthony Liguori     g_free(bs);
2192fc01f7e7Sbellard }
2193fc01f7e7Sbellard 
2194e97fc193Saliguori /*
2195e97fc193Saliguori  * Run consistency checks on an image
2196e97fc193Saliguori  *
2197e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
2198a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
2199e076f338SKevin Wolf  * check are stored in res.
2200e97fc193Saliguori  */
22014534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
2202e97fc193Saliguori {
2203908bcd54SMax Reitz     if (bs->drv == NULL) {
2204908bcd54SMax Reitz         return -ENOMEDIUM;
2205908bcd54SMax Reitz     }
2206e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
2207e97fc193Saliguori         return -ENOTSUP;
2208e97fc193Saliguori     }
2209e97fc193Saliguori 
2210e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
22114534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
2212e97fc193Saliguori }
2213e97fc193Saliguori 
22148a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
22158a426614SKevin Wolf 
221633e3963eSbellard /* commit COW file into the raw image */
221733e3963eSbellard int bdrv_commit(BlockDriverState *bs)
221833e3963eSbellard {
221919cb3738Sbellard     BlockDriver *drv = bs->drv;
222072706ea4SJeff Cody     int64_t sector, total_sectors, length, backing_length;
22218a426614SKevin Wolf     int n, ro, open_flags;
22220bce597dSJeff Cody     int ret = 0;
222372706ea4SJeff Cody     uint8_t *buf = NULL;
222433e3963eSbellard 
222519cb3738Sbellard     if (!drv)
222619cb3738Sbellard         return -ENOMEDIUM;
222733e3963eSbellard 
22284dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
22294dca4b63SNaphtali Sprei         return -ENOTSUP;
22304dca4b63SNaphtali Sprei     }
22314dca4b63SNaphtali Sprei 
2232bb00021dSFam Zheng     if (bdrv_op_is_blocked(bs, BLOCK_OP_TYPE_COMMIT_SOURCE, NULL) ||
2233bb00021dSFam Zheng         bdrv_op_is_blocked(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET, NULL)) {
22342d3735d3SStefan Hajnoczi         return -EBUSY;
22352d3735d3SStefan Hajnoczi     }
22362d3735d3SStefan Hajnoczi 
22374dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
22384dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
22394dca4b63SNaphtali Sprei 
22404dca4b63SNaphtali Sprei     if (ro) {
22410bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
22420bce597dSJeff Cody             return -EACCES;
22434dca4b63SNaphtali Sprei         }
2244ea2384d3Sbellard     }
2245ea2384d3Sbellard 
224672706ea4SJeff Cody     length = bdrv_getlength(bs);
224772706ea4SJeff Cody     if (length < 0) {
224872706ea4SJeff Cody         ret = length;
224972706ea4SJeff Cody         goto ro_cleanup;
225072706ea4SJeff Cody     }
225172706ea4SJeff Cody 
225272706ea4SJeff Cody     backing_length = bdrv_getlength(bs->backing_hd);
225372706ea4SJeff Cody     if (backing_length < 0) {
225472706ea4SJeff Cody         ret = backing_length;
225572706ea4SJeff Cody         goto ro_cleanup;
225672706ea4SJeff Cody     }
225772706ea4SJeff Cody 
225872706ea4SJeff Cody     /* If our top snapshot is larger than the backing file image,
225972706ea4SJeff Cody      * grow the backing file image if possible.  If not possible,
226072706ea4SJeff Cody      * we must return an error */
226172706ea4SJeff Cody     if (length > backing_length) {
226272706ea4SJeff Cody         ret = bdrv_truncate(bs->backing_hd, length);
226372706ea4SJeff Cody         if (ret < 0) {
226472706ea4SJeff Cody             goto ro_cleanup;
226572706ea4SJeff Cody         }
226672706ea4SJeff Cody     }
226772706ea4SJeff Cody 
226872706ea4SJeff Cody     total_sectors = length >> BDRV_SECTOR_BITS;
2269857d4f46SKevin Wolf 
2270857d4f46SKevin Wolf     /* qemu_try_blockalign() for bs will choose an alignment that works for
2271857d4f46SKevin Wolf      * bs->backing_hd as well, so no need to compare the alignment manually. */
2272857d4f46SKevin Wolf     buf = qemu_try_blockalign(bs, COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
2273857d4f46SKevin Wolf     if (buf == NULL) {
2274857d4f46SKevin Wolf         ret = -ENOMEM;
2275857d4f46SKevin Wolf         goto ro_cleanup;
2276857d4f46SKevin Wolf     }
22778a426614SKevin Wolf 
22788a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
2279d663640cSPaolo Bonzini         ret = bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n);
2280d663640cSPaolo Bonzini         if (ret < 0) {
2281d663640cSPaolo Bonzini             goto ro_cleanup;
2282d663640cSPaolo Bonzini         }
2283d663640cSPaolo Bonzini         if (ret) {
2284dabfa6ccSKevin Wolf             ret = bdrv_read(bs, sector, buf, n);
2285dabfa6ccSKevin Wolf             if (ret < 0) {
22864dca4b63SNaphtali Sprei                 goto ro_cleanup;
228733e3963eSbellard             }
228833e3963eSbellard 
2289dabfa6ccSKevin Wolf             ret = bdrv_write(bs->backing_hd, sector, buf, n);
2290dabfa6ccSKevin Wolf             if (ret < 0) {
22914dca4b63SNaphtali Sprei                 goto ro_cleanup;
229233e3963eSbellard             }
229333e3963eSbellard         }
229433e3963eSbellard     }
229595389c86Sbellard 
22961d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
22971d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
2298dabfa6ccSKevin Wolf         if (ret < 0) {
2299dabfa6ccSKevin Wolf             goto ro_cleanup;
2300dabfa6ccSKevin Wolf         }
23011d44952fSChristoph Hellwig         bdrv_flush(bs);
23021d44952fSChristoph Hellwig     }
230395389c86Sbellard 
23043f5075aeSChristoph Hellwig     /*
23053f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
23063f5075aeSChristoph Hellwig      * stable on disk.
23073f5075aeSChristoph Hellwig      */
2308dabfa6ccSKevin Wolf     if (bs->backing_hd) {
23093f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
2310dabfa6ccSKevin Wolf     }
23114dca4b63SNaphtali Sprei 
2312dabfa6ccSKevin Wolf     ret = 0;
23134dca4b63SNaphtali Sprei ro_cleanup:
2314857d4f46SKevin Wolf     qemu_vfree(buf);
23154dca4b63SNaphtali Sprei 
23164dca4b63SNaphtali Sprei     if (ro) {
23170bce597dSJeff Cody         /* ignoring error return here */
23180bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
23194dca4b63SNaphtali Sprei     }
23204dca4b63SNaphtali Sprei 
23211d44952fSChristoph Hellwig     return ret;
232233e3963eSbellard }
232333e3963eSbellard 
2324e8877497SStefan Hajnoczi int bdrv_commit_all(void)
23256ab4b5abSMarkus Armbruster {
23266ab4b5abSMarkus Armbruster     BlockDriverState *bs;
23276ab4b5abSMarkus Armbruster 
2328dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
2329ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
2330ed78cda3SStefan Hajnoczi 
2331ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
2332272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
2333e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
2334e8877497SStefan Hajnoczi             if (ret < 0) {
2335ed78cda3SStefan Hajnoczi                 aio_context_release(aio_context);
2336e8877497SStefan Hajnoczi                 return ret;
23376ab4b5abSMarkus Armbruster             }
23386ab4b5abSMarkus Armbruster         }
2339ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
2340272d2d8eSJeff Cody     }
2341e8877497SStefan Hajnoczi     return 0;
2342e8877497SStefan Hajnoczi }
23436ab4b5abSMarkus Armbruster 
2344756e6736SKevin Wolf /*
2345756e6736SKevin Wolf  * Return values:
2346756e6736SKevin Wolf  * 0        - success
2347756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
2348756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
2349756e6736SKevin Wolf  *            image file header
2350756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
2351756e6736SKevin Wolf  */
2352756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
2353756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
2354756e6736SKevin Wolf {
2355756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
2356469ef350SPaolo Bonzini     int ret;
2357756e6736SKevin Wolf 
23585f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
23595f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
23605f377794SPaolo Bonzini         return -EINVAL;
23615f377794SPaolo Bonzini     }
23625f377794SPaolo Bonzini 
2363756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
2364469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
2365756e6736SKevin Wolf     } else {
2366469ef350SPaolo Bonzini         ret = -ENOTSUP;
2367756e6736SKevin Wolf     }
2368469ef350SPaolo Bonzini 
2369469ef350SPaolo Bonzini     if (ret == 0) {
2370469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
2371469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
2372469ef350SPaolo Bonzini     }
2373469ef350SPaolo Bonzini     return ret;
2374756e6736SKevin Wolf }
2375756e6736SKevin Wolf 
23766ebdcee2SJeff Cody /*
23776ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
23786ebdcee2SJeff Cody  *
23796ebdcee2SJeff Cody  * active is the current topmost image.
23806ebdcee2SJeff Cody  *
23816ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
23826ebdcee2SJeff Cody  * or if active == bs.
23834caf0fcdSJeff Cody  *
23844caf0fcdSJeff Cody  * Returns the bottommost base image if bs == NULL.
23856ebdcee2SJeff Cody  */
23866ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
23876ebdcee2SJeff Cody                                     BlockDriverState *bs)
23886ebdcee2SJeff Cody {
23894caf0fcdSJeff Cody     while (active && bs != active->backing_hd) {
23904caf0fcdSJeff Cody         active = active->backing_hd;
23916ebdcee2SJeff Cody     }
23926ebdcee2SJeff Cody 
23934caf0fcdSJeff Cody     return active;
23946ebdcee2SJeff Cody }
23956ebdcee2SJeff Cody 
23964caf0fcdSJeff Cody /* Given a BDS, searches for the base layer. */
23974caf0fcdSJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
23984caf0fcdSJeff Cody {
23994caf0fcdSJeff Cody     return bdrv_find_overlay(bs, NULL);
24006ebdcee2SJeff Cody }
24016ebdcee2SJeff Cody 
24026ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
24036ebdcee2SJeff Cody     BlockDriverState *bs;
24046ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
24056ebdcee2SJeff Cody } BlkIntermediateStates;
24066ebdcee2SJeff Cody 
24076ebdcee2SJeff Cody 
24086ebdcee2SJeff Cody /*
24096ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
24106ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
24116ebdcee2SJeff Cody  *
24126ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
24136ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
24146ebdcee2SJeff Cody  *
24156ebdcee2SJeff Cody  * E.g., this will convert the following chain:
24166ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
24176ebdcee2SJeff Cody  *
24186ebdcee2SJeff Cody  * to
24196ebdcee2SJeff Cody  *
24206ebdcee2SJeff Cody  * bottom <- base <- active
24216ebdcee2SJeff Cody  *
24226ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
24236ebdcee2SJeff Cody  *
24246ebdcee2SJeff Cody  * base <- intermediate <- top <- active
24256ebdcee2SJeff Cody  *
24266ebdcee2SJeff Cody  * to
24276ebdcee2SJeff Cody  *
24286ebdcee2SJeff Cody  * base <- active
24296ebdcee2SJeff Cody  *
243054e26900SJeff Cody  * If backing_file_str is non-NULL, it will be used when modifying top's
243154e26900SJeff Cody  * overlay image metadata.
243254e26900SJeff Cody  *
24336ebdcee2SJeff Cody  * Error conditions:
24346ebdcee2SJeff Cody  *  if active == top, that is considered an error
24356ebdcee2SJeff Cody  *
24366ebdcee2SJeff Cody  */
24376ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
243854e26900SJeff Cody                            BlockDriverState *base, const char *backing_file_str)
24396ebdcee2SJeff Cody {
24406ebdcee2SJeff Cody     BlockDriverState *intermediate;
24416ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
24426ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
24436ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
24446ebdcee2SJeff Cody     int ret = -EIO;
24456ebdcee2SJeff Cody 
24466ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
24476ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
24486ebdcee2SJeff Cody 
24496ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
24506ebdcee2SJeff Cody         goto exit;
24516ebdcee2SJeff Cody     }
24526ebdcee2SJeff Cody 
24536ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
24546ebdcee2SJeff Cody 
24556ebdcee2SJeff Cody     if (new_top_bs == NULL) {
24566ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
24576ebdcee2SJeff Cody         goto exit;
24586ebdcee2SJeff Cody     }
24596ebdcee2SJeff Cody 
24606ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
24616ebdcee2SJeff Cody      * to do, no intermediate images */
24626ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
24636ebdcee2SJeff Cody         ret = 0;
24646ebdcee2SJeff Cody         goto exit;
24656ebdcee2SJeff Cody     }
24666ebdcee2SJeff Cody 
24676ebdcee2SJeff Cody     intermediate = top;
24686ebdcee2SJeff Cody 
24696ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
24706ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
24716ebdcee2SJeff Cody      */
24726ebdcee2SJeff Cody     while (intermediate) {
24735839e53bSMarkus Armbruster         intermediate_state = g_new0(BlkIntermediateStates, 1);
24746ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
24756ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
24766ebdcee2SJeff Cody 
24776ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
24786ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
24796ebdcee2SJeff Cody             break;
24806ebdcee2SJeff Cody         }
24816ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
24826ebdcee2SJeff Cody     }
24836ebdcee2SJeff Cody     if (base_bs == NULL) {
24846ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
24856ebdcee2SJeff Cody          * unravel everything, and exit with error */
24866ebdcee2SJeff Cody         goto exit;
24876ebdcee2SJeff Cody     }
24886ebdcee2SJeff Cody 
24896ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
249054e26900SJeff Cody     backing_file_str = backing_file_str ? backing_file_str : base_bs->filename;
249154e26900SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, backing_file_str,
24926ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
24936ebdcee2SJeff Cody     if (ret) {
24946ebdcee2SJeff Cody         goto exit;
24956ebdcee2SJeff Cody     }
2496920beae1SFam Zheng     bdrv_set_backing_hd(new_top_bs, base_bs);
24976ebdcee2SJeff Cody 
24986ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
24996ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
2500920beae1SFam Zheng         bdrv_set_backing_hd(intermediate_state->bs, NULL);
25014f6fd349SFam Zheng         bdrv_unref(intermediate_state->bs);
25026ebdcee2SJeff Cody     }
25036ebdcee2SJeff Cody     ret = 0;
25046ebdcee2SJeff Cody 
25056ebdcee2SJeff Cody exit:
25066ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25076ebdcee2SJeff Cody         g_free(intermediate_state);
25086ebdcee2SJeff Cody     }
25096ebdcee2SJeff Cody     return ret;
25106ebdcee2SJeff Cody }
25116ebdcee2SJeff Cody 
251283f64091Sbellard /**
251383f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
251483f64091Sbellard  */
251583f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
251683f64091Sbellard {
251783f64091Sbellard     BlockDriver *drv = bs->drv;
251851762288SStefan Hajnoczi     int ret;
251983f64091Sbellard     if (!drv)
252019cb3738Sbellard         return -ENOMEDIUM;
252183f64091Sbellard     if (!drv->bdrv_truncate)
252283f64091Sbellard         return -ENOTSUP;
252359f2689dSNaphtali Sprei     if (bs->read_only)
252459f2689dSNaphtali Sprei         return -EACCES;
25259c75e168SJeff Cody 
252651762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
252751762288SStefan Hajnoczi     if (ret == 0) {
252851762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2529ce1ffea8SJohn Snow         bdrv_dirty_bitmap_truncate(bs);
2530a7f53e26SMarkus Armbruster         if (bs->blk) {
2531a7f53e26SMarkus Armbruster             blk_dev_resize_cb(bs->blk);
2532a7f53e26SMarkus Armbruster         }
253351762288SStefan Hajnoczi     }
253451762288SStefan Hajnoczi     return ret;
253583f64091Sbellard }
253683f64091Sbellard 
253783f64091Sbellard /**
25384a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
25394a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
25404a1d5e1fSFam Zheng  */
25414a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
25424a1d5e1fSFam Zheng {
25434a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
25444a1d5e1fSFam Zheng     if (!drv) {
25454a1d5e1fSFam Zheng         return -ENOMEDIUM;
25464a1d5e1fSFam Zheng     }
25474a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
25484a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
25494a1d5e1fSFam Zheng     }
25504a1d5e1fSFam Zheng     if (bs->file) {
25514a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
25524a1d5e1fSFam Zheng     }
25534a1d5e1fSFam Zheng     return -ENOTSUP;
25544a1d5e1fSFam Zheng }
25554a1d5e1fSFam Zheng 
25564a1d5e1fSFam Zheng /**
255765a9bb25SMarkus Armbruster  * Return number of sectors on success, -errno on error.
255883f64091Sbellard  */
255965a9bb25SMarkus Armbruster int64_t bdrv_nb_sectors(BlockDriverState *bs)
256083f64091Sbellard {
256183f64091Sbellard     BlockDriver *drv = bs->drv;
256265a9bb25SMarkus Armbruster 
256383f64091Sbellard     if (!drv)
256419cb3738Sbellard         return -ENOMEDIUM;
256551762288SStefan Hajnoczi 
2566b94a2610SKevin Wolf     if (drv->has_variable_length) {
2567b94a2610SKevin Wolf         int ret = refresh_total_sectors(bs, bs->total_sectors);
2568b94a2610SKevin Wolf         if (ret < 0) {
2569b94a2610SKevin Wolf             return ret;
2570fc01f7e7Sbellard         }
257146a4e4e6SStefan Hajnoczi     }
257265a9bb25SMarkus Armbruster     return bs->total_sectors;
257365a9bb25SMarkus Armbruster }
257465a9bb25SMarkus Armbruster 
257565a9bb25SMarkus Armbruster /**
257665a9bb25SMarkus Armbruster  * Return length in bytes on success, -errno on error.
257765a9bb25SMarkus Armbruster  * The length is always a multiple of BDRV_SECTOR_SIZE.
257865a9bb25SMarkus Armbruster  */
257965a9bb25SMarkus Armbruster int64_t bdrv_getlength(BlockDriverState *bs)
258065a9bb25SMarkus Armbruster {
258165a9bb25SMarkus Armbruster     int64_t ret = bdrv_nb_sectors(bs);
258265a9bb25SMarkus Armbruster 
25834a9c9ea0SFam Zheng     ret = ret > INT64_MAX / BDRV_SECTOR_SIZE ? -EFBIG : ret;
258465a9bb25SMarkus Armbruster     return ret < 0 ? ret : ret * BDRV_SECTOR_SIZE;
258546a4e4e6SStefan Hajnoczi }
2586fc01f7e7Sbellard 
258719cb3738Sbellard /* return 0 as number of sectors if no device present or error */
258896b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2589fc01f7e7Sbellard {
259065a9bb25SMarkus Armbruster     int64_t nb_sectors = bdrv_nb_sectors(bs);
259165a9bb25SMarkus Armbruster 
259265a9bb25SMarkus Armbruster     *nb_sectors_ptr = nb_sectors < 0 ? 0 : nb_sectors;
2593fc01f7e7Sbellard }
2594cf98951bSbellard 
2595ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2596ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
2597abd7f68dSMarkus Armbruster {
2598abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
2599abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
2600abd7f68dSMarkus Armbruster }
2601abd7f68dSMarkus Armbruster 
26021ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2603abd7f68dSMarkus Armbruster {
2604abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
2605abd7f68dSMarkus Armbruster }
2606abd7f68dSMarkus Armbruster 
26073e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
26083e1caa5fSPaolo Bonzini {
26093e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
26103e1caa5fSPaolo Bonzini 
26113e1caa5fSPaolo Bonzini     switch (on_err) {
26123e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
2613a589569fSWenchao Xia         return (error == ENOSPC) ?
2614a589569fSWenchao Xia                BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
26153e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
2616a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_STOP;
26173e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
2618a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_REPORT;
26193e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
2620a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_IGNORE;
26213e1caa5fSPaolo Bonzini     default:
26223e1caa5fSPaolo Bonzini         abort();
26233e1caa5fSPaolo Bonzini     }
26243e1caa5fSPaolo Bonzini }
26253e1caa5fSPaolo Bonzini 
2626c7c2ff0cSLuiz Capitulino static void send_qmp_error_event(BlockDriverState *bs,
2627c7c2ff0cSLuiz Capitulino                                  BlockErrorAction action,
2628c7c2ff0cSLuiz Capitulino                                  bool is_read, int error)
2629c7c2ff0cSLuiz Capitulino {
2630573742a5SPeter Maydell     IoOperationType optype;
2631c7c2ff0cSLuiz Capitulino 
2632573742a5SPeter Maydell     optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
2633573742a5SPeter Maydell     qapi_event_send_block_io_error(bdrv_get_device_name(bs), optype, action,
2634c7c2ff0cSLuiz Capitulino                                    bdrv_iostatus_is_enabled(bs),
2635624ff573SLuiz Capitulino                                    error == ENOSPC, strerror(error),
2636624ff573SLuiz Capitulino                                    &error_abort);
2637c7c2ff0cSLuiz Capitulino }
2638c7c2ff0cSLuiz Capitulino 
26393e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
26403e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
26413e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
26423e1caa5fSPaolo Bonzini  */
26433e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
26443e1caa5fSPaolo Bonzini                        bool is_read, int error)
26453e1caa5fSPaolo Bonzini {
26463e1caa5fSPaolo Bonzini     assert(error >= 0);
26472bd3bce8SPaolo Bonzini 
2648a589569fSWenchao Xia     if (action == BLOCK_ERROR_ACTION_STOP) {
26492bd3bce8SPaolo Bonzini         /* First set the iostatus, so that "info block" returns an iostatus
26502bd3bce8SPaolo Bonzini          * that matches the events raised so far (an additional error iostatus
26512bd3bce8SPaolo Bonzini          * is fine, but not a lost one).
26522bd3bce8SPaolo Bonzini          */
26533e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
26542bd3bce8SPaolo Bonzini 
26552bd3bce8SPaolo Bonzini         /* Then raise the request to stop the VM and the event.
26562bd3bce8SPaolo Bonzini          * qemu_system_vmstop_request_prepare has two effects.  First,
26572bd3bce8SPaolo Bonzini          * it ensures that the STOP event always comes after the
26582bd3bce8SPaolo Bonzini          * BLOCK_IO_ERROR event.  Second, it ensures that even if management
26592bd3bce8SPaolo Bonzini          * can observe the STOP event and do a "cont" before the STOP
26602bd3bce8SPaolo Bonzini          * event is issued, the VM will not stop.  In this case, vm_start()
26612bd3bce8SPaolo Bonzini          * also ensures that the STOP/RESUME pair of events is emitted.
26622bd3bce8SPaolo Bonzini          */
26632bd3bce8SPaolo Bonzini         qemu_system_vmstop_request_prepare();
2664c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26652bd3bce8SPaolo Bonzini         qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
26662bd3bce8SPaolo Bonzini     } else {
2667c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26683e1caa5fSPaolo Bonzini     }
26693e1caa5fSPaolo Bonzini }
26703e1caa5fSPaolo Bonzini 
2671b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
2672b338082bSbellard {
2673b338082bSbellard     return bs->read_only;
2674b338082bSbellard }
2675b338082bSbellard 
2676985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
2677985a03b0Sths {
2678985a03b0Sths     return bs->sg;
2679985a03b0Sths }
2680985a03b0Sths 
2681e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
2682e900a7b7SChristoph Hellwig {
2683e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
2684e900a7b7SChristoph Hellwig }
2685e900a7b7SChristoph Hellwig 
2686425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2687425b0148SPaolo Bonzini {
2688425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
268955b110f2SJeff Cody 
269055b110f2SJeff Cody     /* so a reopen() will preserve wce */
269155b110f2SJeff Cody     if (wce) {
269255b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
269355b110f2SJeff Cody     } else {
269455b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
269555b110f2SJeff Cody     }
2696425b0148SPaolo Bonzini }
2697425b0148SPaolo Bonzini 
2698ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
2699ea2384d3Sbellard {
2700ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
2701ea2384d3Sbellard         return 1;
2702ea2384d3Sbellard     return bs->encrypted;
2703ea2384d3Sbellard }
2704ea2384d3Sbellard 
2705c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
2706c0f4ce77Saliguori {
2707c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
2708c0f4ce77Saliguori 
2709c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2710c0f4ce77Saliguori         return 1;
2711c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
2712c0f4ce77Saliguori }
2713c0f4ce77Saliguori 
2714ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
2715ea2384d3Sbellard {
2716ea2384d3Sbellard     int ret;
2717ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
2718ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
2719ea2384d3Sbellard         if (ret < 0)
2720ea2384d3Sbellard             return ret;
2721ea2384d3Sbellard         if (!bs->encrypted)
2722ea2384d3Sbellard             return 0;
2723ea2384d3Sbellard     }
2724fd04a2aeSShahar Havivi     if (!bs->encrypted) {
2725fd04a2aeSShahar Havivi         return -EINVAL;
2726fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2727fd04a2aeSShahar Havivi         return -ENOMEDIUM;
2728fd04a2aeSShahar Havivi     }
2729c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
2730bb5fc20fSaliguori     if (ret < 0) {
2731bb5fc20fSaliguori         bs->valid_key = 0;
2732bb5fc20fSaliguori     } else if (!bs->valid_key) {
2733bb5fc20fSaliguori         bs->valid_key = 1;
2734a7f53e26SMarkus Armbruster         if (bs->blk) {
2735bb5fc20fSaliguori             /* call the change callback now, we skipped it on open */
2736a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
2737a7f53e26SMarkus Armbruster         }
2738bb5fc20fSaliguori     }
2739c0f4ce77Saliguori     return ret;
2740ea2384d3Sbellard }
2741ea2384d3Sbellard 
27424d2855a3SMarkus Armbruster /*
27434d2855a3SMarkus Armbruster  * Provide an encryption key for @bs.
27444d2855a3SMarkus Armbruster  * If @key is non-null:
27454d2855a3SMarkus Armbruster  *     If @bs is not encrypted, fail.
27464d2855a3SMarkus Armbruster  *     Else if the key is invalid, fail.
27474d2855a3SMarkus Armbruster  *     Else set @bs's key to @key, replacing the existing key, if any.
27484d2855a3SMarkus Armbruster  * If @key is null:
27494d2855a3SMarkus Armbruster  *     If @bs is encrypted and still lacks a key, fail.
27504d2855a3SMarkus Armbruster  *     Else do nothing.
27514d2855a3SMarkus Armbruster  * On failure, store an error object through @errp if non-null.
27524d2855a3SMarkus Armbruster  */
27534d2855a3SMarkus Armbruster void bdrv_add_key(BlockDriverState *bs, const char *key, Error **errp)
27544d2855a3SMarkus Armbruster {
27554d2855a3SMarkus Armbruster     if (key) {
27564d2855a3SMarkus Armbruster         if (!bdrv_is_encrypted(bs)) {
275781e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is not encrypted",
275881e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs));
27594d2855a3SMarkus Armbruster         } else if (bdrv_set_key(bs, key) < 0) {
2760c6bd8c70SMarkus Armbruster             error_setg(errp, QERR_INVALID_PASSWORD);
27614d2855a3SMarkus Armbruster         }
27624d2855a3SMarkus Armbruster     } else {
27634d2855a3SMarkus Armbruster         if (bdrv_key_required(bs)) {
2764b1ca6391SMarkus Armbruster             error_set(errp, ERROR_CLASS_DEVICE_ENCRYPTED,
2765b1ca6391SMarkus Armbruster                       "'%s' (%s) is encrypted",
276681e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs),
27674d2855a3SMarkus Armbruster                       bdrv_get_encrypted_filename(bs));
27684d2855a3SMarkus Armbruster         }
27694d2855a3SMarkus Armbruster     }
27704d2855a3SMarkus Armbruster }
27714d2855a3SMarkus Armbruster 
2772f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
2773ea2384d3Sbellard {
2774f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
2775ea2384d3Sbellard }
2776ea2384d3Sbellard 
2777ada42401SStefan Hajnoczi static int qsort_strcmp(const void *a, const void *b)
2778ada42401SStefan Hajnoczi {
2779ada42401SStefan Hajnoczi     return strcmp(a, b);
2780ada42401SStefan Hajnoczi }
2781ada42401SStefan Hajnoczi 
2782ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2783ea2384d3Sbellard                          void *opaque)
2784ea2384d3Sbellard {
2785ea2384d3Sbellard     BlockDriver *drv;
2786e855e4fbSJeff Cody     int count = 0;
2787ada42401SStefan Hajnoczi     int i;
2788e855e4fbSJeff Cody     const char **formats = NULL;
2789ea2384d3Sbellard 
27908a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2791e855e4fbSJeff Cody         if (drv->format_name) {
2792e855e4fbSJeff Cody             bool found = false;
2793e855e4fbSJeff Cody             int i = count;
2794e855e4fbSJeff Cody             while (formats && i && !found) {
2795e855e4fbSJeff Cody                 found = !strcmp(formats[--i], drv->format_name);
2796e855e4fbSJeff Cody             }
2797e855e4fbSJeff Cody 
2798e855e4fbSJeff Cody             if (!found) {
27995839e53bSMarkus Armbruster                 formats = g_renew(const char *, formats, count + 1);
2800e855e4fbSJeff Cody                 formats[count++] = drv->format_name;
2801ea2384d3Sbellard             }
2802ea2384d3Sbellard         }
2803e855e4fbSJeff Cody     }
2804ada42401SStefan Hajnoczi 
2805ada42401SStefan Hajnoczi     qsort(formats, count, sizeof(formats[0]), qsort_strcmp);
2806ada42401SStefan Hajnoczi 
2807ada42401SStefan Hajnoczi     for (i = 0; i < count; i++) {
2808ada42401SStefan Hajnoczi         it(opaque, formats[i]);
2809ada42401SStefan Hajnoczi     }
2810ada42401SStefan Hajnoczi 
2811e855e4fbSJeff Cody     g_free(formats);
2812e855e4fbSJeff Cody }
2813ea2384d3Sbellard 
2814dc364f4cSBenoît Canet /* This function is to find a node in the bs graph */
2815dc364f4cSBenoît Canet BlockDriverState *bdrv_find_node(const char *node_name)
2816dc364f4cSBenoît Canet {
2817dc364f4cSBenoît Canet     BlockDriverState *bs;
2818dc364f4cSBenoît Canet 
2819dc364f4cSBenoît Canet     assert(node_name);
2820dc364f4cSBenoît Canet 
2821dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2822dc364f4cSBenoît Canet         if (!strcmp(node_name, bs->node_name)) {
2823dc364f4cSBenoît Canet             return bs;
2824dc364f4cSBenoît Canet         }
2825dc364f4cSBenoît Canet     }
2826dc364f4cSBenoît Canet     return NULL;
2827dc364f4cSBenoît Canet }
2828dc364f4cSBenoît Canet 
2829c13163fbSBenoît Canet /* Put this QMP function here so it can access the static graph_bdrv_states. */
2830d5a8ee60SAlberto Garcia BlockDeviceInfoList *bdrv_named_nodes_list(Error **errp)
2831c13163fbSBenoît Canet {
2832c13163fbSBenoît Canet     BlockDeviceInfoList *list, *entry;
2833c13163fbSBenoît Canet     BlockDriverState *bs;
2834c13163fbSBenoît Canet 
2835c13163fbSBenoît Canet     list = NULL;
2836c13163fbSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2837d5a8ee60SAlberto Garcia         BlockDeviceInfo *info = bdrv_block_device_info(bs, errp);
2838d5a8ee60SAlberto Garcia         if (!info) {
2839d5a8ee60SAlberto Garcia             qapi_free_BlockDeviceInfoList(list);
2840d5a8ee60SAlberto Garcia             return NULL;
2841d5a8ee60SAlberto Garcia         }
2842c13163fbSBenoît Canet         entry = g_malloc0(sizeof(*entry));
2843d5a8ee60SAlberto Garcia         entry->value = info;
2844c13163fbSBenoît Canet         entry->next = list;
2845c13163fbSBenoît Canet         list = entry;
2846c13163fbSBenoît Canet     }
2847c13163fbSBenoît Canet 
2848c13163fbSBenoît Canet     return list;
2849c13163fbSBenoît Canet }
2850c13163fbSBenoît Canet 
285112d3ba82SBenoît Canet BlockDriverState *bdrv_lookup_bs(const char *device,
285212d3ba82SBenoît Canet                                  const char *node_name,
285312d3ba82SBenoît Canet                                  Error **errp)
285412d3ba82SBenoît Canet {
28557f06d47eSMarkus Armbruster     BlockBackend *blk;
28567f06d47eSMarkus Armbruster     BlockDriverState *bs;
285712d3ba82SBenoît Canet 
285812d3ba82SBenoît Canet     if (device) {
28597f06d47eSMarkus Armbruster         blk = blk_by_name(device);
286012d3ba82SBenoît Canet 
28617f06d47eSMarkus Armbruster         if (blk) {
28627f06d47eSMarkus Armbruster             return blk_bs(blk);
286312d3ba82SBenoît Canet         }
2864dd67fa50SBenoît Canet     }
286512d3ba82SBenoît Canet 
2866dd67fa50SBenoît Canet     if (node_name) {
286712d3ba82SBenoît Canet         bs = bdrv_find_node(node_name);
286812d3ba82SBenoît Canet 
2869dd67fa50SBenoît Canet         if (bs) {
2870dd67fa50SBenoît Canet             return bs;
2871dd67fa50SBenoît Canet         }
287212d3ba82SBenoît Canet     }
287312d3ba82SBenoît Canet 
2874dd67fa50SBenoît Canet     error_setg(errp, "Cannot find device=%s nor node_name=%s",
2875dd67fa50SBenoît Canet                      device ? device : "",
2876dd67fa50SBenoît Canet                      node_name ? node_name : "");
2877dd67fa50SBenoît Canet     return NULL;
287812d3ba82SBenoît Canet }
287912d3ba82SBenoît Canet 
28805a6684d2SJeff Cody /* If 'base' is in the same chain as 'top', return true. Otherwise,
28815a6684d2SJeff Cody  * return false.  If either argument is NULL, return false. */
28825a6684d2SJeff Cody bool bdrv_chain_contains(BlockDriverState *top, BlockDriverState *base)
28835a6684d2SJeff Cody {
28845a6684d2SJeff Cody     while (top && top != base) {
28855a6684d2SJeff Cody         top = top->backing_hd;
28865a6684d2SJeff Cody     }
28875a6684d2SJeff Cody 
28885a6684d2SJeff Cody     return top != NULL;
28895a6684d2SJeff Cody }
28905a6684d2SJeff Cody 
289104df765aSFam Zheng BlockDriverState *bdrv_next_node(BlockDriverState *bs)
289204df765aSFam Zheng {
289304df765aSFam Zheng     if (!bs) {
289404df765aSFam Zheng         return QTAILQ_FIRST(&graph_bdrv_states);
289504df765aSFam Zheng     }
289604df765aSFam Zheng     return QTAILQ_NEXT(bs, node_list);
289704df765aSFam Zheng }
289804df765aSFam Zheng 
28992f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
29002f399b0aSMarkus Armbruster {
29012f399b0aSMarkus Armbruster     if (!bs) {
29022f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
29032f399b0aSMarkus Armbruster     }
2904dc364f4cSBenoît Canet     return QTAILQ_NEXT(bs, device_list);
29052f399b0aSMarkus Armbruster }
29062f399b0aSMarkus Armbruster 
290720a9e77dSFam Zheng const char *bdrv_get_node_name(const BlockDriverState *bs)
290820a9e77dSFam Zheng {
290920a9e77dSFam Zheng     return bs->node_name;
291020a9e77dSFam Zheng }
291120a9e77dSFam Zheng 
29127f06d47eSMarkus Armbruster /* TODO check what callers really want: bs->node_name or blk_name() */
2913bfb197e0SMarkus Armbruster const char *bdrv_get_device_name(const BlockDriverState *bs)
2914ea2384d3Sbellard {
2915bfb197e0SMarkus Armbruster     return bs->blk ? blk_name(bs->blk) : "";
2916ea2384d3Sbellard }
2917ea2384d3Sbellard 
29189b2aa84fSAlberto Garcia /* This can be used to identify nodes that might not have a device
29199b2aa84fSAlberto Garcia  * name associated. Since node and device names live in the same
29209b2aa84fSAlberto Garcia  * namespace, the result is unambiguous. The exception is if both are
29219b2aa84fSAlberto Garcia  * absent, then this returns an empty (non-null) string. */
29229b2aa84fSAlberto Garcia const char *bdrv_get_device_or_node_name(const BlockDriverState *bs)
29239b2aa84fSAlberto Garcia {
29249b2aa84fSAlberto Garcia     return bs->blk ? blk_name(bs->blk) : bs->node_name;
29259b2aa84fSAlberto Garcia }
29269b2aa84fSAlberto Garcia 
2927c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
2928c8433287SMarkus Armbruster {
2929c8433287SMarkus Armbruster     return bs->open_flags;
2930c8433287SMarkus Armbruster }
2931c8433287SMarkus Armbruster 
29323ac21627SPeter Lieven int bdrv_has_zero_init_1(BlockDriverState *bs)
29333ac21627SPeter Lieven {
29343ac21627SPeter Lieven     return 1;
29353ac21627SPeter Lieven }
29363ac21627SPeter Lieven 
2937f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
2938f2feebbdSKevin Wolf {
2939f2feebbdSKevin Wolf     assert(bs->drv);
2940f2feebbdSKevin Wolf 
294111212d8fSPaolo Bonzini     /* If BS is a copy on write image, it is initialized to
294211212d8fSPaolo Bonzini        the contents of the base image, which may not be zeroes.  */
294311212d8fSPaolo Bonzini     if (bs->backing_hd) {
294411212d8fSPaolo Bonzini         return 0;
294511212d8fSPaolo Bonzini     }
2946336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
2947336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
2948f2feebbdSKevin Wolf     }
2949f2feebbdSKevin Wolf 
29503ac21627SPeter Lieven     /* safe default */
29513ac21627SPeter Lieven     return 0;
2952f2feebbdSKevin Wolf }
2953f2feebbdSKevin Wolf 
29544ce78691SPeter Lieven bool bdrv_unallocated_blocks_are_zero(BlockDriverState *bs)
29554ce78691SPeter Lieven {
29564ce78691SPeter Lieven     BlockDriverInfo bdi;
29574ce78691SPeter Lieven 
29584ce78691SPeter Lieven     if (bs->backing_hd) {
29594ce78691SPeter Lieven         return false;
29604ce78691SPeter Lieven     }
29614ce78691SPeter Lieven 
29624ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29634ce78691SPeter Lieven         return bdi.unallocated_blocks_are_zero;
29644ce78691SPeter Lieven     }
29654ce78691SPeter Lieven 
29664ce78691SPeter Lieven     return false;
29674ce78691SPeter Lieven }
29684ce78691SPeter Lieven 
29694ce78691SPeter Lieven bool bdrv_can_write_zeroes_with_unmap(BlockDriverState *bs)
29704ce78691SPeter Lieven {
29714ce78691SPeter Lieven     BlockDriverInfo bdi;
29724ce78691SPeter Lieven 
29734ce78691SPeter Lieven     if (bs->backing_hd || !(bs->open_flags & BDRV_O_UNMAP)) {
29744ce78691SPeter Lieven         return false;
29754ce78691SPeter Lieven     }
29764ce78691SPeter Lieven 
29774ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29784ce78691SPeter Lieven         return bdi.can_write_zeroes_with_unmap;
29794ce78691SPeter Lieven     }
29804ce78691SPeter Lieven 
29814ce78691SPeter Lieven     return false;
29824ce78691SPeter Lieven }
29834ce78691SPeter Lieven 
2984045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
2985045df330Saliguori {
2986045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
2987045df330Saliguori         return bs->backing_file;
2988045df330Saliguori     else if (bs->encrypted)
2989045df330Saliguori         return bs->filename;
2990045df330Saliguori     else
2991045df330Saliguori         return NULL;
2992045df330Saliguori }
2993045df330Saliguori 
299483f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
299583f64091Sbellard                                char *filename, int filename_size)
299683f64091Sbellard {
299783f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
299883f64091Sbellard }
299983f64091Sbellard 
3000faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3001faea38e7Sbellard {
3002faea38e7Sbellard     BlockDriver *drv = bs->drv;
3003faea38e7Sbellard     if (!drv)
300419cb3738Sbellard         return -ENOMEDIUM;
3005faea38e7Sbellard     if (!drv->bdrv_get_info)
3006faea38e7Sbellard         return -ENOTSUP;
3007faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3008faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3009faea38e7Sbellard }
3010faea38e7Sbellard 
3011eae041feSMax Reitz ImageInfoSpecific *bdrv_get_specific_info(BlockDriverState *bs)
3012eae041feSMax Reitz {
3013eae041feSMax Reitz     BlockDriver *drv = bs->drv;
3014eae041feSMax Reitz     if (drv && drv->bdrv_get_specific_info) {
3015eae041feSMax Reitz         return drv->bdrv_get_specific_info(bs);
3016eae041feSMax Reitz     }
3017eae041feSMax Reitz     return NULL;
3018eae041feSMax Reitz }
3019eae041feSMax Reitz 
30208b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
30218b9b0cc2SKevin Wolf {
3022bf736fe3SKevin Wolf     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
30238b9b0cc2SKevin Wolf         return;
30248b9b0cc2SKevin Wolf     }
30258b9b0cc2SKevin Wolf 
3026bf736fe3SKevin Wolf     bs->drv->bdrv_debug_event(bs, event);
302741c695c7SKevin Wolf }
30288b9b0cc2SKevin Wolf 
302941c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
303041c695c7SKevin Wolf                           const char *tag)
303141c695c7SKevin Wolf {
303241c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
303341c695c7SKevin Wolf         bs = bs->file;
303441c695c7SKevin Wolf     }
303541c695c7SKevin Wolf 
303641c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
303741c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
303841c695c7SKevin Wolf     }
303941c695c7SKevin Wolf 
304041c695c7SKevin Wolf     return -ENOTSUP;
304141c695c7SKevin Wolf }
304241c695c7SKevin Wolf 
30434cc70e93SFam Zheng int bdrv_debug_remove_breakpoint(BlockDriverState *bs, const char *tag)
30444cc70e93SFam Zheng {
30454cc70e93SFam Zheng     while (bs && bs->drv && !bs->drv->bdrv_debug_remove_breakpoint) {
30464cc70e93SFam Zheng         bs = bs->file;
30474cc70e93SFam Zheng     }
30484cc70e93SFam Zheng 
30494cc70e93SFam Zheng     if (bs && bs->drv && bs->drv->bdrv_debug_remove_breakpoint) {
30504cc70e93SFam Zheng         return bs->drv->bdrv_debug_remove_breakpoint(bs, tag);
30514cc70e93SFam Zheng     }
30524cc70e93SFam Zheng 
30534cc70e93SFam Zheng     return -ENOTSUP;
30544cc70e93SFam Zheng }
30554cc70e93SFam Zheng 
305641c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
305741c695c7SKevin Wolf {
3058938789eaSMax Reitz     while (bs && (!bs->drv || !bs->drv->bdrv_debug_resume)) {
305941c695c7SKevin Wolf         bs = bs->file;
306041c695c7SKevin Wolf     }
306141c695c7SKevin Wolf 
306241c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
306341c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
306441c695c7SKevin Wolf     }
306541c695c7SKevin Wolf 
306641c695c7SKevin Wolf     return -ENOTSUP;
306741c695c7SKevin Wolf }
306841c695c7SKevin Wolf 
306941c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
307041c695c7SKevin Wolf {
307141c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
307241c695c7SKevin Wolf         bs = bs->file;
307341c695c7SKevin Wolf     }
307441c695c7SKevin Wolf 
307541c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
307641c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
307741c695c7SKevin Wolf     }
307841c695c7SKevin Wolf 
307941c695c7SKevin Wolf     return false;
30808b9b0cc2SKevin Wolf }
30818b9b0cc2SKevin Wolf 
3082199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3083199630b6SBlue Swirl {
3084199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3085199630b6SBlue Swirl }
3086199630b6SBlue Swirl 
3087b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3088b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3089b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3090b1b1d783SJeff Cody  * the CWD rather than the chain. */
3091e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3092e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3093e8a6bb9cSMarcelo Tosatti {
3094b1b1d783SJeff Cody     char *filename_full = NULL;
3095b1b1d783SJeff Cody     char *backing_file_full = NULL;
3096b1b1d783SJeff Cody     char *filename_tmp = NULL;
3097b1b1d783SJeff Cody     int is_protocol = 0;
3098b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3099b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3100b1b1d783SJeff Cody 
3101b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3102e8a6bb9cSMarcelo Tosatti         return NULL;
3103e8a6bb9cSMarcelo Tosatti     }
3104e8a6bb9cSMarcelo Tosatti 
3105b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
3106b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
3107b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
3108b1b1d783SJeff Cody 
3109b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
3110b1b1d783SJeff Cody 
3111b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3112b1b1d783SJeff Cody 
3113b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
3114b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
3115b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3116b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3117b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3118b1b1d783SJeff Cody                 break;
3119b1b1d783SJeff Cody             }
3120e8a6bb9cSMarcelo Tosatti         } else {
3121b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
3122b1b1d783SJeff Cody              * image's filename path */
3123b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3124b1b1d783SJeff Cody                          backing_file);
3125b1b1d783SJeff Cody 
3126b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
3127b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
3128b1b1d783SJeff Cody                 continue;
3129b1b1d783SJeff Cody             }
3130b1b1d783SJeff Cody 
3131b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
3132b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
3133b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3134b1b1d783SJeff Cody                          curr_bs->backing_file);
3135b1b1d783SJeff Cody 
3136b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
3137b1b1d783SJeff Cody                 continue;
3138b1b1d783SJeff Cody             }
3139b1b1d783SJeff Cody 
3140b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
3141b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3142b1b1d783SJeff Cody                 break;
3143b1b1d783SJeff Cody             }
3144e8a6bb9cSMarcelo Tosatti         }
3145e8a6bb9cSMarcelo Tosatti     }
3146e8a6bb9cSMarcelo Tosatti 
3147b1b1d783SJeff Cody     g_free(filename_full);
3148b1b1d783SJeff Cody     g_free(backing_file_full);
3149b1b1d783SJeff Cody     g_free(filename_tmp);
3150b1b1d783SJeff Cody     return retval;
3151e8a6bb9cSMarcelo Tosatti }
3152e8a6bb9cSMarcelo Tosatti 
3153f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
3154f198fd1cSBenoît Canet {
3155f198fd1cSBenoît Canet     if (!bs->drv) {
3156f198fd1cSBenoît Canet         return 0;
3157f198fd1cSBenoît Canet     }
3158f198fd1cSBenoît Canet 
3159f198fd1cSBenoît Canet     if (!bs->backing_hd) {
3160f198fd1cSBenoît Canet         return 0;
3161f198fd1cSBenoît Canet     }
3162f198fd1cSBenoît Canet 
3163f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3164f198fd1cSBenoît Canet }
3165f198fd1cSBenoît Canet 
3166ea2384d3Sbellard void bdrv_init(void)
3167ea2384d3Sbellard {
31685efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
3169ea2384d3Sbellard }
3170ce1a14dcSpbrook 
3171eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
3172eb852011SMarkus Armbruster {
3173eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
3174eb852011SMarkus Armbruster     bdrv_init();
3175eb852011SMarkus Armbruster }
3176eb852011SMarkus Armbruster 
31775a8a30dbSKevin Wolf void bdrv_invalidate_cache(BlockDriverState *bs, Error **errp)
31780f15423cSAnthony Liguori {
31795a8a30dbSKevin Wolf     Error *local_err = NULL;
31805a8a30dbSKevin Wolf     int ret;
31815a8a30dbSKevin Wolf 
31823456a8d1SKevin Wolf     if (!bs->drv)  {
31833456a8d1SKevin Wolf         return;
31840f15423cSAnthony Liguori     }
31853456a8d1SKevin Wolf 
31867ea2d269SAlexey Kardashevskiy     if (!(bs->open_flags & BDRV_O_INCOMING)) {
31877ea2d269SAlexey Kardashevskiy         return;
31887ea2d269SAlexey Kardashevskiy     }
31897ea2d269SAlexey Kardashevskiy     bs->open_flags &= ~BDRV_O_INCOMING;
31907ea2d269SAlexey Kardashevskiy 
31913456a8d1SKevin Wolf     if (bs->drv->bdrv_invalidate_cache) {
31925a8a30dbSKevin Wolf         bs->drv->bdrv_invalidate_cache(bs, &local_err);
31933456a8d1SKevin Wolf     } else if (bs->file) {
31945a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs->file, &local_err);
31955a8a30dbSKevin Wolf     }
31965a8a30dbSKevin Wolf     if (local_err) {
31975a8a30dbSKevin Wolf         error_propagate(errp, local_err);
31985a8a30dbSKevin Wolf         return;
31993456a8d1SKevin Wolf     }
32003456a8d1SKevin Wolf 
32015a8a30dbSKevin Wolf     ret = refresh_total_sectors(bs, bs->total_sectors);
32025a8a30dbSKevin Wolf     if (ret < 0) {
32035a8a30dbSKevin Wolf         error_setg_errno(errp, -ret, "Could not refresh total sector count");
32045a8a30dbSKevin Wolf         return;
32055a8a30dbSKevin Wolf     }
32060f15423cSAnthony Liguori }
32070f15423cSAnthony Liguori 
32085a8a30dbSKevin Wolf void bdrv_invalidate_cache_all(Error **errp)
32090f15423cSAnthony Liguori {
32100f15423cSAnthony Liguori     BlockDriverState *bs;
32115a8a30dbSKevin Wolf     Error *local_err = NULL;
32120f15423cSAnthony Liguori 
3213dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
3214ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
3215ed78cda3SStefan Hajnoczi 
3216ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
32175a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs, &local_err);
3218ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
32195a8a30dbSKevin Wolf         if (local_err) {
32205a8a30dbSKevin Wolf             error_propagate(errp, local_err);
32215a8a30dbSKevin Wolf             return;
32225a8a30dbSKevin Wolf         }
32230f15423cSAnthony Liguori     }
32240f15423cSAnthony Liguori }
32250f15423cSAnthony Liguori 
3226f9f05dc5SKevin Wolf /**************************************************************/
322719cb3738Sbellard /* removable device support */
322819cb3738Sbellard 
322919cb3738Sbellard /**
323019cb3738Sbellard  * Return TRUE if the media is present
323119cb3738Sbellard  */
323219cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
323319cb3738Sbellard {
323419cb3738Sbellard     BlockDriver *drv = bs->drv;
3235a1aff5bfSMarkus Armbruster 
323619cb3738Sbellard     if (!drv)
323719cb3738Sbellard         return 0;
323819cb3738Sbellard     if (!drv->bdrv_is_inserted)
3239a1aff5bfSMarkus Armbruster         return 1;
3240a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
324119cb3738Sbellard }
324219cb3738Sbellard 
324319cb3738Sbellard /**
32448e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
32458e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
324619cb3738Sbellard  */
324719cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
324819cb3738Sbellard {
324919cb3738Sbellard     BlockDriver *drv = bs->drv;
325019cb3738Sbellard 
32518e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
32528e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
32538e49ca46SMarkus Armbruster     }
32548e49ca46SMarkus Armbruster     return -ENOTSUP;
325519cb3738Sbellard }
325619cb3738Sbellard 
325719cb3738Sbellard /**
325819cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
325919cb3738Sbellard  */
3260f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
326119cb3738Sbellard {
326219cb3738Sbellard     BlockDriver *drv = bs->drv;
3263bfb197e0SMarkus Armbruster     const char *device_name;
326419cb3738Sbellard 
3265822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
3266822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
326719cb3738Sbellard     }
32686f382ed2SLuiz Capitulino 
3269bfb197e0SMarkus Armbruster     device_name = bdrv_get_device_name(bs);
3270bfb197e0SMarkus Armbruster     if (device_name[0] != '\0') {
3271bfb197e0SMarkus Armbruster         qapi_event_send_device_tray_moved(device_name,
3272a5ee7bd4SWenchao Xia                                           eject_flag, &error_abort);
32736f382ed2SLuiz Capitulino     }
327419cb3738Sbellard }
327519cb3738Sbellard 
327619cb3738Sbellard /**
327719cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
327819cb3738Sbellard  * to eject it manually).
327919cb3738Sbellard  */
3280025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
328119cb3738Sbellard {
328219cb3738Sbellard     BlockDriver *drv = bs->drv;
328319cb3738Sbellard 
3284025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
3285b8c6d095SStefan Hajnoczi 
3286025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
3287025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
328819cb3738Sbellard     }
328919cb3738Sbellard }
3290985a03b0Sths 
32911b7fd729SPaolo Bonzini void bdrv_set_guest_block_size(BlockDriverState *bs, int align)
32927b6f9300SMarkus Armbruster {
32931b7fd729SPaolo Bonzini     bs->guest_block_size = align;
32947b6f9300SMarkus Armbruster }
32957cd1e32aSlirans@il.ibm.com 
32960db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_find_dirty_bitmap(BlockDriverState *bs, const char *name)
32970db6e54aSFam Zheng {
32980db6e54aSFam Zheng     BdrvDirtyBitmap *bm;
32990db6e54aSFam Zheng 
33000db6e54aSFam Zheng     assert(name);
33010db6e54aSFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
33020db6e54aSFam Zheng         if (bm->name && !strcmp(name, bm->name)) {
33030db6e54aSFam Zheng             return bm;
33040db6e54aSFam Zheng         }
33050db6e54aSFam Zheng     }
33060db6e54aSFam Zheng     return NULL;
33070db6e54aSFam Zheng }
33080db6e54aSFam Zheng 
330920dca810SJohn Snow void bdrv_dirty_bitmap_make_anon(BdrvDirtyBitmap *bitmap)
33100db6e54aSFam Zheng {
33119bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
33120db6e54aSFam Zheng     g_free(bitmap->name);
33130db6e54aSFam Zheng     bitmap->name = NULL;
33140db6e54aSFam Zheng }
33150db6e54aSFam Zheng 
33160db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_create_dirty_bitmap(BlockDriverState *bs,
33175fba6c0eSJohn Snow                                           uint32_t granularity,
33180db6e54aSFam Zheng                                           const char *name,
3319b8afb520SFam Zheng                                           Error **errp)
33207cd1e32aSlirans@il.ibm.com {
33217cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
3322e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
33235fba6c0eSJohn Snow     uint32_t sector_granularity;
3324a55eb92cSJan Kiszka 
332550717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
332650717e94SPaolo Bonzini 
33270db6e54aSFam Zheng     if (name && bdrv_find_dirty_bitmap(bs, name)) {
33280db6e54aSFam Zheng         error_setg(errp, "Bitmap already exists: %s", name);
33290db6e54aSFam Zheng         return NULL;
33300db6e54aSFam Zheng     }
33315fba6c0eSJohn Snow     sector_granularity = granularity >> BDRV_SECTOR_BITS;
33325fba6c0eSJohn Snow     assert(sector_granularity);
333357322b78SMarkus Armbruster     bitmap_size = bdrv_nb_sectors(bs);
3334b8afb520SFam Zheng     if (bitmap_size < 0) {
3335b8afb520SFam Zheng         error_setg_errno(errp, -bitmap_size, "could not get length of device");
3336b8afb520SFam Zheng         errno = -bitmap_size;
3337b8afb520SFam Zheng         return NULL;
3338b8afb520SFam Zheng     }
33395839e53bSMarkus Armbruster     bitmap = g_new0(BdrvDirtyBitmap, 1);
33405fba6c0eSJohn Snow     bitmap->bitmap = hbitmap_alloc(bitmap_size, ctz32(sector_granularity));
3341e74e6b78SJohn Snow     bitmap->size = bitmap_size;
33420db6e54aSFam Zheng     bitmap->name = g_strdup(name);
3343b8e6fb75SJohn Snow     bitmap->disabled = false;
3344e4654d2dSFam Zheng     QLIST_INSERT_HEAD(&bs->dirty_bitmaps, bitmap, list);
3345e4654d2dSFam Zheng     return bitmap;
3346e4654d2dSFam Zheng }
3347e4654d2dSFam Zheng 
33489bd2b08fSJohn Snow bool bdrv_dirty_bitmap_frozen(BdrvDirtyBitmap *bitmap)
33499bd2b08fSJohn Snow {
33509bd2b08fSJohn Snow     return bitmap->successor;
33519bd2b08fSJohn Snow }
33529bd2b08fSJohn Snow 
3353b8e6fb75SJohn Snow bool bdrv_dirty_bitmap_enabled(BdrvDirtyBitmap *bitmap)
3354b8e6fb75SJohn Snow {
33559bd2b08fSJohn Snow     return !(bitmap->disabled || bitmap->successor);
33569bd2b08fSJohn Snow }
33579bd2b08fSJohn Snow 
33589abe3bdcSJohn Snow DirtyBitmapStatus bdrv_dirty_bitmap_status(BdrvDirtyBitmap *bitmap)
33599abe3bdcSJohn Snow {
33609abe3bdcSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33619abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_FROZEN;
33629abe3bdcSJohn Snow     } else if (!bdrv_dirty_bitmap_enabled(bitmap)) {
33639abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_DISABLED;
33649abe3bdcSJohn Snow     } else {
33659abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_ACTIVE;
33669abe3bdcSJohn Snow     }
33679abe3bdcSJohn Snow }
33689abe3bdcSJohn Snow 
33699bd2b08fSJohn Snow /**
33709bd2b08fSJohn Snow  * Create a successor bitmap destined to replace this bitmap after an operation.
33719bd2b08fSJohn Snow  * Requires that the bitmap is not frozen and has no successor.
33729bd2b08fSJohn Snow  */
33739bd2b08fSJohn Snow int bdrv_dirty_bitmap_create_successor(BlockDriverState *bs,
33749bd2b08fSJohn Snow                                        BdrvDirtyBitmap *bitmap, Error **errp)
33759bd2b08fSJohn Snow {
33769bd2b08fSJohn Snow     uint64_t granularity;
33779bd2b08fSJohn Snow     BdrvDirtyBitmap *child;
33789bd2b08fSJohn Snow 
33799bd2b08fSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33809bd2b08fSJohn Snow         error_setg(errp, "Cannot create a successor for a bitmap that is "
33819bd2b08fSJohn Snow                    "currently frozen");
33829bd2b08fSJohn Snow         return -1;
33839bd2b08fSJohn Snow     }
33849bd2b08fSJohn Snow     assert(!bitmap->successor);
33859bd2b08fSJohn Snow 
33869bd2b08fSJohn Snow     /* Create an anonymous successor */
33879bd2b08fSJohn Snow     granularity = bdrv_dirty_bitmap_granularity(bitmap);
33889bd2b08fSJohn Snow     child = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp);
33899bd2b08fSJohn Snow     if (!child) {
33909bd2b08fSJohn Snow         return -1;
33919bd2b08fSJohn Snow     }
33929bd2b08fSJohn Snow 
33939bd2b08fSJohn Snow     /* Successor will be on or off based on our current state. */
33949bd2b08fSJohn Snow     child->disabled = bitmap->disabled;
33959bd2b08fSJohn Snow 
33969bd2b08fSJohn Snow     /* Install the successor and freeze the parent */
33979bd2b08fSJohn Snow     bitmap->successor = child;
33989bd2b08fSJohn Snow     return 0;
33999bd2b08fSJohn Snow }
34009bd2b08fSJohn Snow 
34019bd2b08fSJohn Snow /**
34029bd2b08fSJohn Snow  * For a bitmap with a successor, yield our name to the successor,
34039bd2b08fSJohn Snow  * delete the old bitmap, and return a handle to the new bitmap.
34049bd2b08fSJohn Snow  */
34059bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_dirty_bitmap_abdicate(BlockDriverState *bs,
34069bd2b08fSJohn Snow                                             BdrvDirtyBitmap *bitmap,
34079bd2b08fSJohn Snow                                             Error **errp)
34089bd2b08fSJohn Snow {
34099bd2b08fSJohn Snow     char *name;
34109bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = bitmap->successor;
34119bd2b08fSJohn Snow 
34129bd2b08fSJohn Snow     if (successor == NULL) {
34139bd2b08fSJohn Snow         error_setg(errp, "Cannot relinquish control if "
34149bd2b08fSJohn Snow                    "there's no successor present");
34159bd2b08fSJohn Snow         return NULL;
34169bd2b08fSJohn Snow     }
34179bd2b08fSJohn Snow 
34189bd2b08fSJohn Snow     name = bitmap->name;
34199bd2b08fSJohn Snow     bitmap->name = NULL;
34209bd2b08fSJohn Snow     successor->name = name;
34219bd2b08fSJohn Snow     bitmap->successor = NULL;
34229bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, bitmap);
34239bd2b08fSJohn Snow 
34249bd2b08fSJohn Snow     return successor;
34259bd2b08fSJohn Snow }
34269bd2b08fSJohn Snow 
34279bd2b08fSJohn Snow /**
34289bd2b08fSJohn Snow  * In cases of failure where we can no longer safely delete the parent,
34299bd2b08fSJohn Snow  * we may wish to re-join the parent and child/successor.
34309bd2b08fSJohn Snow  * The merged parent will be un-frozen, but not explicitly re-enabled.
34319bd2b08fSJohn Snow  */
34329bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_reclaim_dirty_bitmap(BlockDriverState *bs,
34339bd2b08fSJohn Snow                                            BdrvDirtyBitmap *parent,
34349bd2b08fSJohn Snow                                            Error **errp)
34359bd2b08fSJohn Snow {
34369bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = parent->successor;
34379bd2b08fSJohn Snow 
34389bd2b08fSJohn Snow     if (!successor) {
34399bd2b08fSJohn Snow         error_setg(errp, "Cannot reclaim a successor when none is present");
34409bd2b08fSJohn Snow         return NULL;
34419bd2b08fSJohn Snow     }
34429bd2b08fSJohn Snow 
34439bd2b08fSJohn Snow     if (!hbitmap_merge(parent->bitmap, successor->bitmap)) {
34449bd2b08fSJohn Snow         error_setg(errp, "Merging of parent and successor bitmap failed");
34459bd2b08fSJohn Snow         return NULL;
34469bd2b08fSJohn Snow     }
34479bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, successor);
34489bd2b08fSJohn Snow     parent->successor = NULL;
34499bd2b08fSJohn Snow 
34509bd2b08fSJohn Snow     return parent;
3451b8e6fb75SJohn Snow }
3452b8e6fb75SJohn Snow 
3453ce1ffea8SJohn Snow /**
3454ce1ffea8SJohn Snow  * Truncates _all_ bitmaps attached to a BDS.
3455ce1ffea8SJohn Snow  */
3456ce1ffea8SJohn Snow static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs)
3457ce1ffea8SJohn Snow {
3458ce1ffea8SJohn Snow     BdrvDirtyBitmap *bitmap;
3459ce1ffea8SJohn Snow     uint64_t size = bdrv_nb_sectors(bs);
3460ce1ffea8SJohn Snow 
3461ce1ffea8SJohn Snow     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
346206207b0fSJohn Snow         assert(!bdrv_dirty_bitmap_frozen(bitmap));
3463ce1ffea8SJohn Snow         hbitmap_truncate(bitmap->bitmap, size);
34645270b6a0SJohn Snow         bitmap->size = size;
3465ce1ffea8SJohn Snow     }
3466ce1ffea8SJohn Snow }
3467ce1ffea8SJohn Snow 
3468e4654d2dSFam Zheng void bdrv_release_dirty_bitmap(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
3469e4654d2dSFam Zheng {
3470e4654d2dSFam Zheng     BdrvDirtyBitmap *bm, *next;
3471e4654d2dSFam Zheng     QLIST_FOREACH_SAFE(bm, &bs->dirty_bitmaps, list, next) {
3472e4654d2dSFam Zheng         if (bm == bitmap) {
34739bd2b08fSJohn Snow             assert(!bdrv_dirty_bitmap_frozen(bm));
3474e4654d2dSFam Zheng             QLIST_REMOVE(bitmap, list);
3475e4654d2dSFam Zheng             hbitmap_free(bitmap->bitmap);
34760db6e54aSFam Zheng             g_free(bitmap->name);
3477e4654d2dSFam Zheng             g_free(bitmap);
3478e4654d2dSFam Zheng             return;
34797cd1e32aSlirans@il.ibm.com         }
34807cd1e32aSlirans@il.ibm.com     }
34817cd1e32aSlirans@il.ibm.com }
34827cd1e32aSlirans@il.ibm.com 
3483b8e6fb75SJohn Snow void bdrv_disable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3484b8e6fb75SJohn Snow {
34859bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3486b8e6fb75SJohn Snow     bitmap->disabled = true;
3487b8e6fb75SJohn Snow }
3488b8e6fb75SJohn Snow 
3489b8e6fb75SJohn Snow void bdrv_enable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3490b8e6fb75SJohn Snow {
34919bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3492b8e6fb75SJohn Snow     bitmap->disabled = false;
3493b8e6fb75SJohn Snow }
3494b8e6fb75SJohn Snow 
349521b56835SFam Zheng BlockDirtyInfoList *bdrv_query_dirty_bitmaps(BlockDriverState *bs)
349621b56835SFam Zheng {
349721b56835SFam Zheng     BdrvDirtyBitmap *bm;
349821b56835SFam Zheng     BlockDirtyInfoList *list = NULL;
349921b56835SFam Zheng     BlockDirtyInfoList **plist = &list;
350021b56835SFam Zheng 
350121b56835SFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
35025839e53bSMarkus Armbruster         BlockDirtyInfo *info = g_new0(BlockDirtyInfo, 1);
35035839e53bSMarkus Armbruster         BlockDirtyInfoList *entry = g_new0(BlockDirtyInfoList, 1);
350420dca810SJohn Snow         info->count = bdrv_get_dirty_count(bm);
3505592fdd02SJohn Snow         info->granularity = bdrv_dirty_bitmap_granularity(bm);
35060db6e54aSFam Zheng         info->has_name = !!bm->name;
35070db6e54aSFam Zheng         info->name = g_strdup(bm->name);
35089abe3bdcSJohn Snow         info->status = bdrv_dirty_bitmap_status(bm);
350921b56835SFam Zheng         entry->value = info;
351021b56835SFam Zheng         *plist = entry;
351121b56835SFam Zheng         plist = &entry->next;
351221b56835SFam Zheng     }
351321b56835SFam Zheng 
351421b56835SFam Zheng     return list;
351521b56835SFam Zheng }
351621b56835SFam Zheng 
3517e4654d2dSFam Zheng int bdrv_get_dirty(BlockDriverState *bs, BdrvDirtyBitmap *bitmap, int64_t sector)
35187cd1e32aSlirans@il.ibm.com {
3519e4654d2dSFam Zheng     if (bitmap) {
3520e4654d2dSFam Zheng         return hbitmap_get(bitmap->bitmap, sector);
35217cd1e32aSlirans@il.ibm.com     } else {
35227cd1e32aSlirans@il.ibm.com         return 0;
35237cd1e32aSlirans@il.ibm.com     }
35247cd1e32aSlirans@il.ibm.com }
35257cd1e32aSlirans@il.ibm.com 
3526341ebc2fSJohn Snow /**
3527341ebc2fSJohn Snow  * Chooses a default granularity based on the existing cluster size,
3528341ebc2fSJohn Snow  * but clamped between [4K, 64K]. Defaults to 64K in the case that there
3529341ebc2fSJohn Snow  * is no cluster size information available.
3530341ebc2fSJohn Snow  */
3531341ebc2fSJohn Snow uint32_t bdrv_get_default_bitmap_granularity(BlockDriverState *bs)
3532341ebc2fSJohn Snow {
3533341ebc2fSJohn Snow     BlockDriverInfo bdi;
3534341ebc2fSJohn Snow     uint32_t granularity;
3535341ebc2fSJohn Snow 
3536341ebc2fSJohn Snow     if (bdrv_get_info(bs, &bdi) >= 0 && bdi.cluster_size > 0) {
3537341ebc2fSJohn Snow         granularity = MAX(4096, bdi.cluster_size);
3538341ebc2fSJohn Snow         granularity = MIN(65536, granularity);
3539341ebc2fSJohn Snow     } else {
3540341ebc2fSJohn Snow         granularity = 65536;
3541341ebc2fSJohn Snow     }
3542341ebc2fSJohn Snow 
3543341ebc2fSJohn Snow     return granularity;
3544341ebc2fSJohn Snow }
3545341ebc2fSJohn Snow 
3546592fdd02SJohn Snow uint32_t bdrv_dirty_bitmap_granularity(BdrvDirtyBitmap *bitmap)
3547592fdd02SJohn Snow {
3548592fdd02SJohn Snow     return BDRV_SECTOR_SIZE << hbitmap_granularity(bitmap->bitmap);
3549592fdd02SJohn Snow }
3550592fdd02SJohn Snow 
355120dca810SJohn Snow void bdrv_dirty_iter_init(BdrvDirtyBitmap *bitmap, HBitmapIter *hbi)
35521755da16SPaolo Bonzini {
3553e4654d2dSFam Zheng     hbitmap_iter_init(hbi, bitmap->bitmap, 0);
35541755da16SPaolo Bonzini }
35551755da16SPaolo Bonzini 
355620dca810SJohn Snow void bdrv_set_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3557c4237dfaSVladimir Sementsov-Ogievskiy                            int64_t cur_sector, int nr_sectors)
3558c4237dfaSVladimir Sementsov-Ogievskiy {
3559b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3560c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3561c4237dfaSVladimir Sementsov-Ogievskiy }
3562c4237dfaSVladimir Sementsov-Ogievskiy 
356320dca810SJohn Snow void bdrv_reset_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3564c4237dfaSVladimir Sementsov-Ogievskiy                              int64_t cur_sector, int nr_sectors)
3565c4237dfaSVladimir Sementsov-Ogievskiy {
3566b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3567c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_reset(bitmap->bitmap, cur_sector, nr_sectors);
3568c4237dfaSVladimir Sementsov-Ogievskiy }
3569c4237dfaSVladimir Sementsov-Ogievskiy 
3570e74e6b78SJohn Snow void bdrv_clear_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3571e74e6b78SJohn Snow {
3572e74e6b78SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3573c6a8c328SWen Congyang     hbitmap_reset_all(bitmap->bitmap);
3574e74e6b78SJohn Snow }
3575e74e6b78SJohn Snow 
3576e0c47b6cSStefan Hajnoczi void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
35771755da16SPaolo Bonzini                     int nr_sectors)
35781755da16SPaolo Bonzini {
3579e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
3580e4654d2dSFam Zheng     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
3581b8e6fb75SJohn Snow         if (!bdrv_dirty_bitmap_enabled(bitmap)) {
3582b8e6fb75SJohn Snow             continue;
3583b8e6fb75SJohn Snow         }
3584e4654d2dSFam Zheng         hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3585e4654d2dSFam Zheng     }
35861755da16SPaolo Bonzini }
35871755da16SPaolo Bonzini 
3588d58d8453SJohn Snow /**
3589d58d8453SJohn Snow  * Advance an HBitmapIter to an arbitrary offset.
3590d58d8453SJohn Snow  */
3591d58d8453SJohn Snow void bdrv_set_dirty_iter(HBitmapIter *hbi, int64_t offset)
3592d58d8453SJohn Snow {
3593d58d8453SJohn Snow     assert(hbi->hb);
3594d58d8453SJohn Snow     hbitmap_iter_init(hbi, hbi->hb, offset);
3595d58d8453SJohn Snow }
3596d58d8453SJohn Snow 
359720dca810SJohn Snow int64_t bdrv_get_dirty_count(BdrvDirtyBitmap *bitmap)
3598aaa0eb75SLiran Schour {
3599e4654d2dSFam Zheng     return hbitmap_count(bitmap->bitmap);
3600aaa0eb75SLiran Schour }
3601f88e1a42SJes Sorensen 
36029fcb0251SFam Zheng /* Get a reference to bs */
36039fcb0251SFam Zheng void bdrv_ref(BlockDriverState *bs)
36049fcb0251SFam Zheng {
36059fcb0251SFam Zheng     bs->refcnt++;
36069fcb0251SFam Zheng }
36079fcb0251SFam Zheng 
36089fcb0251SFam Zheng /* Release a previously grabbed reference to bs.
36099fcb0251SFam Zheng  * If after releasing, reference count is zero, the BlockDriverState is
36109fcb0251SFam Zheng  * deleted. */
36119fcb0251SFam Zheng void bdrv_unref(BlockDriverState *bs)
36129fcb0251SFam Zheng {
36139a4d5ca6SJeff Cody     if (!bs) {
36149a4d5ca6SJeff Cody         return;
36159a4d5ca6SJeff Cody     }
36169fcb0251SFam Zheng     assert(bs->refcnt > 0);
36179fcb0251SFam Zheng     if (--bs->refcnt == 0) {
36189fcb0251SFam Zheng         bdrv_delete(bs);
36199fcb0251SFam Zheng     }
36209fcb0251SFam Zheng }
36219fcb0251SFam Zheng 
3622fbe40ff7SFam Zheng struct BdrvOpBlocker {
3623fbe40ff7SFam Zheng     Error *reason;
3624fbe40ff7SFam Zheng     QLIST_ENTRY(BdrvOpBlocker) list;
3625fbe40ff7SFam Zheng };
3626fbe40ff7SFam Zheng 
3627fbe40ff7SFam Zheng bool bdrv_op_is_blocked(BlockDriverState *bs, BlockOpType op, Error **errp)
3628fbe40ff7SFam Zheng {
3629fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3630fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3631fbe40ff7SFam Zheng     if (!QLIST_EMPTY(&bs->op_blockers[op])) {
3632fbe40ff7SFam Zheng         blocker = QLIST_FIRST(&bs->op_blockers[op]);
3633fbe40ff7SFam Zheng         if (errp) {
363481e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is busy: %s",
363581e5f78aSAlberto Garcia                        bdrv_get_device_or_node_name(bs),
3636bfb197e0SMarkus Armbruster                        error_get_pretty(blocker->reason));
3637fbe40ff7SFam Zheng         }
3638fbe40ff7SFam Zheng         return true;
3639fbe40ff7SFam Zheng     }
3640fbe40ff7SFam Zheng     return false;
3641fbe40ff7SFam Zheng }
3642fbe40ff7SFam Zheng 
3643fbe40ff7SFam Zheng void bdrv_op_block(BlockDriverState *bs, BlockOpType op, Error *reason)
3644fbe40ff7SFam Zheng {
3645fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3646fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3647fbe40ff7SFam Zheng 
36485839e53bSMarkus Armbruster     blocker = g_new0(BdrvOpBlocker, 1);
3649fbe40ff7SFam Zheng     blocker->reason = reason;
3650fbe40ff7SFam Zheng     QLIST_INSERT_HEAD(&bs->op_blockers[op], blocker, list);
3651fbe40ff7SFam Zheng }
3652fbe40ff7SFam Zheng 
3653fbe40ff7SFam Zheng void bdrv_op_unblock(BlockDriverState *bs, BlockOpType op, Error *reason)
3654fbe40ff7SFam Zheng {
3655fbe40ff7SFam Zheng     BdrvOpBlocker *blocker, *next;
3656fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3657fbe40ff7SFam Zheng     QLIST_FOREACH_SAFE(blocker, &bs->op_blockers[op], list, next) {
3658fbe40ff7SFam Zheng         if (blocker->reason == reason) {
3659fbe40ff7SFam Zheng             QLIST_REMOVE(blocker, list);
3660fbe40ff7SFam Zheng             g_free(blocker);
3661fbe40ff7SFam Zheng         }
3662fbe40ff7SFam Zheng     }
3663fbe40ff7SFam Zheng }
3664fbe40ff7SFam Zheng 
3665fbe40ff7SFam Zheng void bdrv_op_block_all(BlockDriverState *bs, Error *reason)
3666fbe40ff7SFam Zheng {
3667fbe40ff7SFam Zheng     int i;
3668fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3669fbe40ff7SFam Zheng         bdrv_op_block(bs, i, reason);
3670fbe40ff7SFam Zheng     }
3671fbe40ff7SFam Zheng }
3672fbe40ff7SFam Zheng 
3673fbe40ff7SFam Zheng void bdrv_op_unblock_all(BlockDriverState *bs, Error *reason)
3674fbe40ff7SFam Zheng {
3675fbe40ff7SFam Zheng     int i;
3676fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3677fbe40ff7SFam Zheng         bdrv_op_unblock(bs, i, reason);
3678fbe40ff7SFam Zheng     }
3679fbe40ff7SFam Zheng }
3680fbe40ff7SFam Zheng 
3681fbe40ff7SFam Zheng bool bdrv_op_blocker_is_empty(BlockDriverState *bs)
3682fbe40ff7SFam Zheng {
3683fbe40ff7SFam Zheng     int i;
3684fbe40ff7SFam Zheng 
3685fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3686fbe40ff7SFam Zheng         if (!QLIST_EMPTY(&bs->op_blockers[i])) {
3687fbe40ff7SFam Zheng             return false;
3688fbe40ff7SFam Zheng         }
3689fbe40ff7SFam Zheng     }
3690fbe40ff7SFam Zheng     return true;
3691fbe40ff7SFam Zheng }
3692fbe40ff7SFam Zheng 
369328a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
369428a7282aSLuiz Capitulino {
3695d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
369658e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
369728a7282aSLuiz Capitulino }
369828a7282aSLuiz Capitulino 
369928a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
370028a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
370128a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
370228a7282aSLuiz Capitulino {
3703d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
370492aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
370592aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
370692aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
370728a7282aSLuiz Capitulino }
370828a7282aSLuiz Capitulino 
370928a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
371028a7282aSLuiz Capitulino {
3711d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
371228a7282aSLuiz Capitulino }
371328a7282aSLuiz Capitulino 
371428a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
371528a7282aSLuiz Capitulino {
371628a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
371758e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
37183bd293c3SPaolo Bonzini         if (bs->job) {
37193bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
37203bd293c3SPaolo Bonzini         }
372128a7282aSLuiz Capitulino     }
372228a7282aSLuiz Capitulino }
372328a7282aSLuiz Capitulino 
372428a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
372528a7282aSLuiz Capitulino {
37263e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
37273e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
372858e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
372958e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
373028a7282aSLuiz Capitulino     }
373128a7282aSLuiz Capitulino }
373228a7282aSLuiz Capitulino 
3733d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
3734f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
3735f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
3736f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
3737f88e1a42SJes Sorensen {
373883d0521aSChunyan Liu     QemuOptsList *create_opts = NULL;
373983d0521aSChunyan Liu     QemuOpts *opts = NULL;
374083d0521aSChunyan Liu     const char *backing_fmt, *backing_file;
374183d0521aSChunyan Liu     int64_t size;
3742f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
3743cc84d90fSMax Reitz     Error *local_err = NULL;
3744f88e1a42SJes Sorensen     int ret = 0;
3745f88e1a42SJes Sorensen 
3746f88e1a42SJes Sorensen     /* Find driver and parse its options */
3747f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
3748f88e1a42SJes Sorensen     if (!drv) {
374971c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
3750d92ada22SLuiz Capitulino         return;
3751f88e1a42SJes Sorensen     }
3752f88e1a42SJes Sorensen 
3753b65a5e12SMax Reitz     proto_drv = bdrv_find_protocol(filename, true, errp);
3754f88e1a42SJes Sorensen     if (!proto_drv) {
3755d92ada22SLuiz Capitulino         return;
3756f88e1a42SJes Sorensen     }
3757f88e1a42SJes Sorensen 
3758c6149724SMax Reitz     if (!drv->create_opts) {
3759c6149724SMax Reitz         error_setg(errp, "Format driver '%s' does not support image creation",
3760c6149724SMax Reitz                    drv->format_name);
3761c6149724SMax Reitz         return;
3762c6149724SMax Reitz     }
3763c6149724SMax Reitz 
3764c6149724SMax Reitz     if (!proto_drv->create_opts) {
3765c6149724SMax Reitz         error_setg(errp, "Protocol driver '%s' does not support image creation",
3766c6149724SMax Reitz                    proto_drv->format_name);
3767c6149724SMax Reitz         return;
3768c6149724SMax Reitz     }
3769c6149724SMax Reitz 
3770c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, drv->create_opts);
3771c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, proto_drv->create_opts);
3772f88e1a42SJes Sorensen 
3773f88e1a42SJes Sorensen     /* Create parameter list with default values */
377483d0521aSChunyan Liu     opts = qemu_opts_create(create_opts, NULL, 0, &error_abort);
377539101f25SMarkus Armbruster     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, img_size, &error_abort);
3776f88e1a42SJes Sorensen 
3777f88e1a42SJes Sorensen     /* Parse -o options */
3778f88e1a42SJes Sorensen     if (options) {
3779dc523cd3SMarkus Armbruster         qemu_opts_do_parse(opts, options, NULL, &local_err);
3780dc523cd3SMarkus Armbruster         if (local_err) {
3781dc523cd3SMarkus Armbruster             error_report_err(local_err);
3782dc523cd3SMarkus Armbruster             local_err = NULL;
378383d0521aSChunyan Liu             error_setg(errp, "Invalid options for file format '%s'", fmt);
3784f88e1a42SJes Sorensen             goto out;
3785f88e1a42SJes Sorensen         }
3786f88e1a42SJes Sorensen     }
3787f88e1a42SJes Sorensen 
3788f88e1a42SJes Sorensen     if (base_filename) {
3789f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, base_filename, &local_err);
37906be4194bSMarkus Armbruster         if (local_err) {
379171c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
379271c79813SLuiz Capitulino                        fmt);
3793f88e1a42SJes Sorensen             goto out;
3794f88e1a42SJes Sorensen         }
3795f88e1a42SJes Sorensen     }
3796f88e1a42SJes Sorensen 
3797f88e1a42SJes Sorensen     if (base_fmt) {
3798f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FMT, base_fmt, &local_err);
37996be4194bSMarkus Armbruster         if (local_err) {
380071c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
380171c79813SLuiz Capitulino                              "format '%s'", fmt);
3802f88e1a42SJes Sorensen             goto out;
3803f88e1a42SJes Sorensen         }
3804f88e1a42SJes Sorensen     }
3805f88e1a42SJes Sorensen 
380683d0521aSChunyan Liu     backing_file = qemu_opt_get(opts, BLOCK_OPT_BACKING_FILE);
380783d0521aSChunyan Liu     if (backing_file) {
380883d0521aSChunyan Liu         if (!strcmp(filename, backing_file)) {
380971c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
381071c79813SLuiz Capitulino                              "same filename as the backing file");
3811792da93aSJes Sorensen             goto out;
3812792da93aSJes Sorensen         }
3813792da93aSJes Sorensen     }
3814792da93aSJes Sorensen 
381583d0521aSChunyan Liu     backing_fmt = qemu_opt_get(opts, BLOCK_OPT_BACKING_FMT);
3816f88e1a42SJes Sorensen 
3817f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
3818f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
381983d0521aSChunyan Liu     size = qemu_opt_get_size(opts, BLOCK_OPT_SIZE, 0);
382083d0521aSChunyan Liu     if (size == -1) {
382183d0521aSChunyan Liu         if (backing_file) {
382266f6b814SMax Reitz             BlockDriverState *bs;
382329168018SMax Reitz             char *full_backing = g_new0(char, PATH_MAX);
382452bf1e72SMarkus Armbruster             int64_t size;
382563090dacSPaolo Bonzini             int back_flags;
3826e6641719SMax Reitz             QDict *backing_options = NULL;
382763090dacSPaolo Bonzini 
382829168018SMax Reitz             bdrv_get_full_backing_filename_from_filename(filename, backing_file,
382929168018SMax Reitz                                                          full_backing, PATH_MAX,
383029168018SMax Reitz                                                          &local_err);
383129168018SMax Reitz             if (local_err) {
383229168018SMax Reitz                 g_free(full_backing);
383329168018SMax Reitz                 goto out;
383429168018SMax Reitz             }
383529168018SMax Reitz 
383663090dacSPaolo Bonzini             /* backing files always opened read-only */
383763090dacSPaolo Bonzini             back_flags =
383863090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
3839f88e1a42SJes Sorensen 
3840e6641719SMax Reitz             if (backing_fmt) {
3841e6641719SMax Reitz                 backing_options = qdict_new();
3842e6641719SMax Reitz                 qdict_put(backing_options, "driver",
3843e6641719SMax Reitz                           qstring_from_str(backing_fmt));
3844e6641719SMax Reitz             }
3845e6641719SMax Reitz 
3846f67503e5SMax Reitz             bs = NULL;
3847e6641719SMax Reitz             ret = bdrv_open(&bs, full_backing, NULL, backing_options,
3848*6ebf9aa2SMax Reitz                             back_flags, &local_err);
384929168018SMax Reitz             g_free(full_backing);
3850f88e1a42SJes Sorensen             if (ret < 0) {
3851f88e1a42SJes Sorensen                 goto out;
3852f88e1a42SJes Sorensen             }
385352bf1e72SMarkus Armbruster             size = bdrv_getlength(bs);
385452bf1e72SMarkus Armbruster             if (size < 0) {
385552bf1e72SMarkus Armbruster                 error_setg_errno(errp, -size, "Could not get size of '%s'",
385652bf1e72SMarkus Armbruster                                  backing_file);
385752bf1e72SMarkus Armbruster                 bdrv_unref(bs);
385852bf1e72SMarkus Armbruster                 goto out;
385952bf1e72SMarkus Armbruster             }
3860f88e1a42SJes Sorensen 
386139101f25SMarkus Armbruster             qemu_opt_set_number(opts, BLOCK_OPT_SIZE, size, &error_abort);
386266f6b814SMax Reitz 
386366f6b814SMax Reitz             bdrv_unref(bs);
3864f88e1a42SJes Sorensen         } else {
386571c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
3866f88e1a42SJes Sorensen             goto out;
3867f88e1a42SJes Sorensen         }
3868f88e1a42SJes Sorensen     }
3869f88e1a42SJes Sorensen 
3870f382d43aSMiroslav Rezanina     if (!quiet) {
3871f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
387243c5d8f8SFam Zheng         qemu_opts_print(opts, " ");
3873f88e1a42SJes Sorensen         puts("");
3874f382d43aSMiroslav Rezanina     }
387583d0521aSChunyan Liu 
3876c282e1fdSChunyan Liu     ret = bdrv_create(drv, filename, opts, &local_err);
387783d0521aSChunyan Liu 
3878cc84d90fSMax Reitz     if (ret == -EFBIG) {
3879cc84d90fSMax Reitz         /* This is generally a better message than whatever the driver would
3880cc84d90fSMax Reitz          * deliver (especially because of the cluster_size_hint), since that
3881cc84d90fSMax Reitz          * is most probably not much different from "image too large". */
3882f3f4d2c0SKevin Wolf         const char *cluster_size_hint = "";
388383d0521aSChunyan Liu         if (qemu_opt_get_size(opts, BLOCK_OPT_CLUSTER_SIZE, 0)) {
3884f3f4d2c0SKevin Wolf             cluster_size_hint = " (try using a larger cluster size)";
3885f3f4d2c0SKevin Wolf         }
3886cc84d90fSMax Reitz         error_setg(errp, "The image size is too large for file format '%s'"
3887cc84d90fSMax Reitz                    "%s", fmt, cluster_size_hint);
3888cc84d90fSMax Reitz         error_free(local_err);
3889cc84d90fSMax Reitz         local_err = NULL;
3890f88e1a42SJes Sorensen     }
3891f88e1a42SJes Sorensen 
3892f88e1a42SJes Sorensen out:
389383d0521aSChunyan Liu     qemu_opts_del(opts);
389483d0521aSChunyan Liu     qemu_opts_free(create_opts);
389584d18f06SMarkus Armbruster     if (local_err) {
3896cc84d90fSMax Reitz         error_propagate(errp, local_err);
3897cc84d90fSMax Reitz     }
3898f88e1a42SJes Sorensen }
389985d126f3SStefan Hajnoczi 
390085d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
390185d126f3SStefan Hajnoczi {
3902dcd04228SStefan Hajnoczi     return bs->aio_context;
3903dcd04228SStefan Hajnoczi }
3904dcd04228SStefan Hajnoczi 
3905dcd04228SStefan Hajnoczi void bdrv_detach_aio_context(BlockDriverState *bs)
3906dcd04228SStefan Hajnoczi {
390733384421SMax Reitz     BdrvAioNotifier *baf;
390833384421SMax Reitz 
3909dcd04228SStefan Hajnoczi     if (!bs->drv) {
3910dcd04228SStefan Hajnoczi         return;
3911dcd04228SStefan Hajnoczi     }
3912dcd04228SStefan Hajnoczi 
391333384421SMax Reitz     QLIST_FOREACH(baf, &bs->aio_notifiers, list) {
391433384421SMax Reitz         baf->detach_aio_context(baf->opaque);
391533384421SMax Reitz     }
391633384421SMax Reitz 
391713af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39180e5b0a2dSBenoît Canet         throttle_timers_detach_aio_context(&bs->throttle_timers);
391913af91ebSStefan Hajnoczi     }
3920dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_detach_aio_context) {
3921dcd04228SStefan Hajnoczi         bs->drv->bdrv_detach_aio_context(bs);
3922dcd04228SStefan Hajnoczi     }
3923dcd04228SStefan Hajnoczi     if (bs->file) {
3924dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->file);
3925dcd04228SStefan Hajnoczi     }
3926dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3927dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->backing_hd);
3928dcd04228SStefan Hajnoczi     }
3929dcd04228SStefan Hajnoczi 
3930dcd04228SStefan Hajnoczi     bs->aio_context = NULL;
3931dcd04228SStefan Hajnoczi }
3932dcd04228SStefan Hajnoczi 
3933dcd04228SStefan Hajnoczi void bdrv_attach_aio_context(BlockDriverState *bs,
3934dcd04228SStefan Hajnoczi                              AioContext *new_context)
3935dcd04228SStefan Hajnoczi {
393633384421SMax Reitz     BdrvAioNotifier *ban;
393733384421SMax Reitz 
3938dcd04228SStefan Hajnoczi     if (!bs->drv) {
3939dcd04228SStefan Hajnoczi         return;
3940dcd04228SStefan Hajnoczi     }
3941dcd04228SStefan Hajnoczi 
3942dcd04228SStefan Hajnoczi     bs->aio_context = new_context;
3943dcd04228SStefan Hajnoczi 
3944dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3945dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->backing_hd, new_context);
3946dcd04228SStefan Hajnoczi     }
3947dcd04228SStefan Hajnoczi     if (bs->file) {
3948dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->file, new_context);
3949dcd04228SStefan Hajnoczi     }
3950dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_attach_aio_context) {
3951dcd04228SStefan Hajnoczi         bs->drv->bdrv_attach_aio_context(bs, new_context);
3952dcd04228SStefan Hajnoczi     }
395313af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39540e5b0a2dSBenoît Canet         throttle_timers_attach_aio_context(&bs->throttle_timers, new_context);
395513af91ebSStefan Hajnoczi     }
395633384421SMax Reitz 
395733384421SMax Reitz     QLIST_FOREACH(ban, &bs->aio_notifiers, list) {
395833384421SMax Reitz         ban->attached_aio_context(new_context, ban->opaque);
395933384421SMax Reitz     }
3960dcd04228SStefan Hajnoczi }
3961dcd04228SStefan Hajnoczi 
3962dcd04228SStefan Hajnoczi void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
3963dcd04228SStefan Hajnoczi {
396453ec73e2SFam Zheng     bdrv_drain(bs); /* ensure there are no in-flight requests */
3965dcd04228SStefan Hajnoczi 
3966dcd04228SStefan Hajnoczi     bdrv_detach_aio_context(bs);
3967dcd04228SStefan Hajnoczi 
3968dcd04228SStefan Hajnoczi     /* This function executes in the old AioContext so acquire the new one in
3969dcd04228SStefan Hajnoczi      * case it runs in a different thread.
3970dcd04228SStefan Hajnoczi      */
3971dcd04228SStefan Hajnoczi     aio_context_acquire(new_context);
3972dcd04228SStefan Hajnoczi     bdrv_attach_aio_context(bs, new_context);
3973dcd04228SStefan Hajnoczi     aio_context_release(new_context);
397485d126f3SStefan Hajnoczi }
3975d616b224SStefan Hajnoczi 
397633384421SMax Reitz void bdrv_add_aio_context_notifier(BlockDriverState *bs,
397733384421SMax Reitz         void (*attached_aio_context)(AioContext *new_context, void *opaque),
397833384421SMax Reitz         void (*detach_aio_context)(void *opaque), void *opaque)
397933384421SMax Reitz {
398033384421SMax Reitz     BdrvAioNotifier *ban = g_new(BdrvAioNotifier, 1);
398133384421SMax Reitz     *ban = (BdrvAioNotifier){
398233384421SMax Reitz         .attached_aio_context = attached_aio_context,
398333384421SMax Reitz         .detach_aio_context   = detach_aio_context,
398433384421SMax Reitz         .opaque               = opaque
398533384421SMax Reitz     };
398633384421SMax Reitz 
398733384421SMax Reitz     QLIST_INSERT_HEAD(&bs->aio_notifiers, ban, list);
398833384421SMax Reitz }
398933384421SMax Reitz 
399033384421SMax Reitz void bdrv_remove_aio_context_notifier(BlockDriverState *bs,
399133384421SMax Reitz                                       void (*attached_aio_context)(AioContext *,
399233384421SMax Reitz                                                                    void *),
399333384421SMax Reitz                                       void (*detach_aio_context)(void *),
399433384421SMax Reitz                                       void *opaque)
399533384421SMax Reitz {
399633384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
399733384421SMax Reitz 
399833384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
399933384421SMax Reitz         if (ban->attached_aio_context == attached_aio_context &&
400033384421SMax Reitz             ban->detach_aio_context   == detach_aio_context   &&
400133384421SMax Reitz             ban->opaque               == opaque)
400233384421SMax Reitz         {
400333384421SMax Reitz             QLIST_REMOVE(ban, list);
400433384421SMax Reitz             g_free(ban);
400533384421SMax Reitz 
400633384421SMax Reitz             return;
400733384421SMax Reitz         }
400833384421SMax Reitz     }
400933384421SMax Reitz 
401033384421SMax Reitz     abort();
401133384421SMax Reitz }
401233384421SMax Reitz 
401377485434SMax Reitz int bdrv_amend_options(BlockDriverState *bs, QemuOpts *opts,
401477485434SMax Reitz                        BlockDriverAmendStatusCB *status_cb)
40156f176b48SMax Reitz {
4016c282e1fdSChunyan Liu     if (!bs->drv->bdrv_amend_options) {
40176f176b48SMax Reitz         return -ENOTSUP;
40186f176b48SMax Reitz     }
401977485434SMax Reitz     return bs->drv->bdrv_amend_options(bs, opts, status_cb);
40206f176b48SMax Reitz }
4021f6186f49SBenoît Canet 
4022b5042a36SBenoît Canet /* This function will be called by the bdrv_recurse_is_first_non_filter method
4023b5042a36SBenoît Canet  * of block filter and by bdrv_is_first_non_filter.
4024b5042a36SBenoît Canet  * It is used to test if the given bs is the candidate or recurse more in the
4025b5042a36SBenoît Canet  * node graph.
4026212a5a8fSBenoît Canet  */
4027212a5a8fSBenoît Canet bool bdrv_recurse_is_first_non_filter(BlockDriverState *bs,
4028212a5a8fSBenoît Canet                                       BlockDriverState *candidate)
4029f6186f49SBenoît Canet {
4030b5042a36SBenoît Canet     /* return false if basic checks fails */
4031b5042a36SBenoît Canet     if (!bs || !bs->drv) {
4032b5042a36SBenoît Canet         return false;
4033b5042a36SBenoît Canet     }
4034b5042a36SBenoît Canet 
4035b5042a36SBenoît Canet     /* the code reached a non block filter driver -> check if the bs is
4036b5042a36SBenoît Canet      * the same as the candidate. It's the recursion termination condition.
4037b5042a36SBenoît Canet      */
4038b5042a36SBenoît Canet     if (!bs->drv->is_filter) {
4039b5042a36SBenoît Canet         return bs == candidate;
4040b5042a36SBenoît Canet     }
4041b5042a36SBenoît Canet     /* Down this path the driver is a block filter driver */
4042b5042a36SBenoît Canet 
4043b5042a36SBenoît Canet     /* If the block filter recursion method is defined use it to recurse down
4044b5042a36SBenoît Canet      * the node graph.
4045b5042a36SBenoît Canet      */
4046b5042a36SBenoît Canet     if (bs->drv->bdrv_recurse_is_first_non_filter) {
4047212a5a8fSBenoît Canet         return bs->drv->bdrv_recurse_is_first_non_filter(bs, candidate);
4048212a5a8fSBenoît Canet     }
4049212a5a8fSBenoît Canet 
4050b5042a36SBenoît Canet     /* the driver is a block filter but don't allow to recurse -> return false
4051b5042a36SBenoît Canet      */
4052b5042a36SBenoît Canet     return false;
4053212a5a8fSBenoît Canet }
4054212a5a8fSBenoît Canet 
4055212a5a8fSBenoît Canet /* This function checks if the candidate is the first non filter bs down it's
4056212a5a8fSBenoît Canet  * bs chain. Since we don't have pointers to parents it explore all bs chains
4057212a5a8fSBenoît Canet  * from the top. Some filters can choose not to pass down the recursion.
4058212a5a8fSBenoît Canet  */
4059212a5a8fSBenoît Canet bool bdrv_is_first_non_filter(BlockDriverState *candidate)
4060212a5a8fSBenoît Canet {
4061212a5a8fSBenoît Canet     BlockDriverState *bs;
4062212a5a8fSBenoît Canet 
4063212a5a8fSBenoît Canet     /* walk down the bs forest recursively */
4064212a5a8fSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
4065212a5a8fSBenoît Canet         bool perm;
4066212a5a8fSBenoît Canet 
4067b5042a36SBenoît Canet         /* try to recurse in this top level bs */
4068e6dc8a1fSKevin Wolf         perm = bdrv_recurse_is_first_non_filter(bs, candidate);
4069212a5a8fSBenoît Canet 
4070212a5a8fSBenoît Canet         /* candidate is the first non filter */
4071212a5a8fSBenoît Canet         if (perm) {
4072212a5a8fSBenoît Canet             return true;
4073212a5a8fSBenoît Canet         }
4074212a5a8fSBenoît Canet     }
4075212a5a8fSBenoît Canet 
4076212a5a8fSBenoît Canet     return false;
4077f6186f49SBenoît Canet }
407809158f00SBenoît Canet 
4079e12f3784SWen Congyang BlockDriverState *check_to_replace_node(BlockDriverState *parent_bs,
4080e12f3784SWen Congyang                                         const char *node_name, Error **errp)
408109158f00SBenoît Canet {
408209158f00SBenoît Canet     BlockDriverState *to_replace_bs = bdrv_find_node(node_name);
40835a7e7a0bSStefan Hajnoczi     AioContext *aio_context;
40845a7e7a0bSStefan Hajnoczi 
408509158f00SBenoît Canet     if (!to_replace_bs) {
408609158f00SBenoît Canet         error_setg(errp, "Node name '%s' not found", node_name);
408709158f00SBenoît Canet         return NULL;
408809158f00SBenoît Canet     }
408909158f00SBenoît Canet 
40905a7e7a0bSStefan Hajnoczi     aio_context = bdrv_get_aio_context(to_replace_bs);
40915a7e7a0bSStefan Hajnoczi     aio_context_acquire(aio_context);
40925a7e7a0bSStefan Hajnoczi 
409309158f00SBenoît Canet     if (bdrv_op_is_blocked(to_replace_bs, BLOCK_OP_TYPE_REPLACE, errp)) {
40945a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
40955a7e7a0bSStefan Hajnoczi         goto out;
409609158f00SBenoît Canet     }
409709158f00SBenoît Canet 
409809158f00SBenoît Canet     /* We don't want arbitrary node of the BDS chain to be replaced only the top
409909158f00SBenoît Canet      * most non filter in order to prevent data corruption.
410009158f00SBenoît Canet      * Another benefit is that this tests exclude backing files which are
410109158f00SBenoît Canet      * blocked by the backing blockers.
410209158f00SBenoît Canet      */
4103e12f3784SWen Congyang     if (!bdrv_recurse_is_first_non_filter(parent_bs, to_replace_bs)) {
410409158f00SBenoît Canet         error_setg(errp, "Only top most non filter can be replaced");
41055a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41065a7e7a0bSStefan Hajnoczi         goto out;
410709158f00SBenoît Canet     }
410809158f00SBenoît Canet 
41095a7e7a0bSStefan Hajnoczi out:
41105a7e7a0bSStefan Hajnoczi     aio_context_release(aio_context);
411109158f00SBenoît Canet     return to_replace_bs;
411209158f00SBenoît Canet }
4113448ad91dSMing Lei 
411491af7014SMax Reitz static bool append_open_options(QDict *d, BlockDriverState *bs)
411591af7014SMax Reitz {
411691af7014SMax Reitz     const QDictEntry *entry;
411791af7014SMax Reitz     bool found_any = false;
411891af7014SMax Reitz 
411991af7014SMax Reitz     for (entry = qdict_first(bs->options); entry;
412091af7014SMax Reitz          entry = qdict_next(bs->options, entry))
412191af7014SMax Reitz     {
412291af7014SMax Reitz         /* Only take options for this level and exclude all non-driver-specific
412391af7014SMax Reitz          * options */
412491af7014SMax Reitz         if (!strchr(qdict_entry_key(entry), '.') &&
412591af7014SMax Reitz             strcmp(qdict_entry_key(entry), "node-name"))
412691af7014SMax Reitz         {
412791af7014SMax Reitz             qobject_incref(qdict_entry_value(entry));
412891af7014SMax Reitz             qdict_put_obj(d, qdict_entry_key(entry), qdict_entry_value(entry));
412991af7014SMax Reitz             found_any = true;
413091af7014SMax Reitz         }
413191af7014SMax Reitz     }
413291af7014SMax Reitz 
413391af7014SMax Reitz     return found_any;
413491af7014SMax Reitz }
413591af7014SMax Reitz 
413691af7014SMax Reitz /* Updates the following BDS fields:
413791af7014SMax Reitz  *  - exact_filename: A filename which may be used for opening a block device
413891af7014SMax Reitz  *                    which (mostly) equals the given BDS (even without any
413991af7014SMax Reitz  *                    other options; so reading and writing must return the same
414091af7014SMax Reitz  *                    results, but caching etc. may be different)
414191af7014SMax Reitz  *  - full_open_options: Options which, when given when opening a block device
414291af7014SMax Reitz  *                       (without a filename), result in a BDS (mostly)
414391af7014SMax Reitz  *                       equalling the given one
414491af7014SMax Reitz  *  - filename: If exact_filename is set, it is copied here. Otherwise,
414591af7014SMax Reitz  *              full_open_options is converted to a JSON object, prefixed with
414691af7014SMax Reitz  *              "json:" (for use through the JSON pseudo protocol) and put here.
414791af7014SMax Reitz  */
414891af7014SMax Reitz void bdrv_refresh_filename(BlockDriverState *bs)
414991af7014SMax Reitz {
415091af7014SMax Reitz     BlockDriver *drv = bs->drv;
415191af7014SMax Reitz     QDict *opts;
415291af7014SMax Reitz 
415391af7014SMax Reitz     if (!drv) {
415491af7014SMax Reitz         return;
415591af7014SMax Reitz     }
415691af7014SMax Reitz 
415791af7014SMax Reitz     /* This BDS's file name will most probably depend on its file's name, so
415891af7014SMax Reitz      * refresh that first */
415991af7014SMax Reitz     if (bs->file) {
416091af7014SMax Reitz         bdrv_refresh_filename(bs->file);
416191af7014SMax Reitz     }
416291af7014SMax Reitz 
416391af7014SMax Reitz     if (drv->bdrv_refresh_filename) {
416491af7014SMax Reitz         /* Obsolete information is of no use here, so drop the old file name
416591af7014SMax Reitz          * information before refreshing it */
416691af7014SMax Reitz         bs->exact_filename[0] = '\0';
416791af7014SMax Reitz         if (bs->full_open_options) {
416891af7014SMax Reitz             QDECREF(bs->full_open_options);
416991af7014SMax Reitz             bs->full_open_options = NULL;
417091af7014SMax Reitz         }
417191af7014SMax Reitz 
417291af7014SMax Reitz         drv->bdrv_refresh_filename(bs);
417391af7014SMax Reitz     } else if (bs->file) {
417491af7014SMax Reitz         /* Try to reconstruct valid information from the underlying file */
417591af7014SMax Reitz         bool has_open_options;
417691af7014SMax Reitz 
417791af7014SMax Reitz         bs->exact_filename[0] = '\0';
417891af7014SMax Reitz         if (bs->full_open_options) {
417991af7014SMax Reitz             QDECREF(bs->full_open_options);
418091af7014SMax Reitz             bs->full_open_options = NULL;
418191af7014SMax Reitz         }
418291af7014SMax Reitz 
418391af7014SMax Reitz         opts = qdict_new();
418491af7014SMax Reitz         has_open_options = append_open_options(opts, bs);
418591af7014SMax Reitz 
418691af7014SMax Reitz         /* If no specific options have been given for this BDS, the filename of
418791af7014SMax Reitz          * the underlying file should suffice for this one as well */
418891af7014SMax Reitz         if (bs->file->exact_filename[0] && !has_open_options) {
418991af7014SMax Reitz             strcpy(bs->exact_filename, bs->file->exact_filename);
419091af7014SMax Reitz         }
419191af7014SMax Reitz         /* Reconstructing the full options QDict is simple for most format block
419291af7014SMax Reitz          * drivers, as long as the full options are known for the underlying
419391af7014SMax Reitz          * file BDS. The full options QDict of that file BDS should somehow
419491af7014SMax Reitz          * contain a representation of the filename, therefore the following
419591af7014SMax Reitz          * suffices without querying the (exact_)filename of this BDS. */
419691af7014SMax Reitz         if (bs->file->full_open_options) {
419791af7014SMax Reitz             qdict_put_obj(opts, "driver",
419891af7014SMax Reitz                           QOBJECT(qstring_from_str(drv->format_name)));
419991af7014SMax Reitz             QINCREF(bs->file->full_open_options);
420091af7014SMax Reitz             qdict_put_obj(opts, "file", QOBJECT(bs->file->full_open_options));
420191af7014SMax Reitz 
420291af7014SMax Reitz             bs->full_open_options = opts;
420391af7014SMax Reitz         } else {
420491af7014SMax Reitz             QDECREF(opts);
420591af7014SMax Reitz         }
420691af7014SMax Reitz     } else if (!bs->full_open_options && qdict_size(bs->options)) {
420791af7014SMax Reitz         /* There is no underlying file BDS (at least referenced by BDS.file),
420891af7014SMax Reitz          * so the full options QDict should be equal to the options given
420991af7014SMax Reitz          * specifically for this block device when it was opened (plus the
421091af7014SMax Reitz          * driver specification).
421191af7014SMax Reitz          * Because those options don't change, there is no need to update
421291af7014SMax Reitz          * full_open_options when it's already set. */
421391af7014SMax Reitz 
421491af7014SMax Reitz         opts = qdict_new();
421591af7014SMax Reitz         append_open_options(opts, bs);
421691af7014SMax Reitz         qdict_put_obj(opts, "driver",
421791af7014SMax Reitz                       QOBJECT(qstring_from_str(drv->format_name)));
421891af7014SMax Reitz 
421991af7014SMax Reitz         if (bs->exact_filename[0]) {
422091af7014SMax Reitz             /* This may not work for all block protocol drivers (some may
422191af7014SMax Reitz              * require this filename to be parsed), but we have to find some
422291af7014SMax Reitz              * default solution here, so just include it. If some block driver
422391af7014SMax Reitz              * does not support pure options without any filename at all or
422491af7014SMax Reitz              * needs some special format of the options QDict, it needs to
422591af7014SMax Reitz              * implement the driver-specific bdrv_refresh_filename() function.
422691af7014SMax Reitz              */
422791af7014SMax Reitz             qdict_put_obj(opts, "filename",
422891af7014SMax Reitz                           QOBJECT(qstring_from_str(bs->exact_filename)));
422991af7014SMax Reitz         }
423091af7014SMax Reitz 
423191af7014SMax Reitz         bs->full_open_options = opts;
423291af7014SMax Reitz     }
423391af7014SMax Reitz 
423491af7014SMax Reitz     if (bs->exact_filename[0]) {
423591af7014SMax Reitz         pstrcpy(bs->filename, sizeof(bs->filename), bs->exact_filename);
423691af7014SMax Reitz     } else if (bs->full_open_options) {
423791af7014SMax Reitz         QString *json = qobject_to_json(QOBJECT(bs->full_open_options));
423891af7014SMax Reitz         snprintf(bs->filename, sizeof(bs->filename), "json:%s",
423991af7014SMax Reitz                  qstring_get_str(json));
424091af7014SMax Reitz         QDECREF(json);
424191af7014SMax Reitz     }
424291af7014SMax Reitz }
42435366d0c8SBenoît Canet 
42445366d0c8SBenoît Canet /* This accessor function purpose is to allow the device models to access the
42455366d0c8SBenoît Canet  * BlockAcctStats structure embedded inside a BlockDriverState without being
42465366d0c8SBenoît Canet  * aware of the BlockDriverState structure layout.
42475366d0c8SBenoît Canet  * It will go away when the BlockAcctStats structure will be moved inside
42485366d0c8SBenoît Canet  * the device models.
42495366d0c8SBenoît Canet  */
42505366d0c8SBenoît Canet BlockAcctStats *bdrv_get_stats(BlockDriverState *bs)
42515366d0c8SBenoît Canet {
42525366d0c8SBenoît Canet     return &bs->stats;
42535366d0c8SBenoît Canet }
4254