xref: /openbmc/qemu/block.c (revision 9a4f4c31563b96a075f3deae83e72c726e0c84f8)
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,
88ce343771SMax Reitz                              const BdrvChildRole *child_role, Error **errp);
89f3930ed0SKevin Wolf 
90ce1ffea8SJohn Snow static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs);
91eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
92eb852011SMarkus Armbruster static int use_bdrv_whitelist;
93eb852011SMarkus Armbruster 
949e0b22f4SStefan Hajnoczi #ifdef _WIN32
959e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
969e0b22f4SStefan Hajnoczi {
979e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
989e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
999e0b22f4SStefan Hajnoczi             filename[1] == ':');
1009e0b22f4SStefan Hajnoczi }
1019e0b22f4SStefan Hajnoczi 
1029e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
1039e0b22f4SStefan Hajnoczi {
1049e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
1059e0b22f4SStefan Hajnoczi         filename[2] == '\0')
1069e0b22f4SStefan Hajnoczi         return 1;
1079e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
1089e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
1099e0b22f4SStefan Hajnoczi         return 1;
1109e0b22f4SStefan Hajnoczi     return 0;
1119e0b22f4SStefan Hajnoczi }
1129e0b22f4SStefan Hajnoczi #endif
1139e0b22f4SStefan Hajnoczi 
114339064d5SKevin Wolf size_t bdrv_opt_mem_align(BlockDriverState *bs)
115339064d5SKevin Wolf {
116339064d5SKevin Wolf     if (!bs || !bs->drv) {
117459b4e66SDenis V. Lunev         /* page size or 4k (hdd sector size) should be on the safe side */
118459b4e66SDenis V. Lunev         return MAX(4096, getpagesize());
119339064d5SKevin Wolf     }
120339064d5SKevin Wolf 
121339064d5SKevin Wolf     return bs->bl.opt_mem_alignment;
122339064d5SKevin Wolf }
123339064d5SKevin Wolf 
1244196d2f0SDenis V. Lunev size_t bdrv_min_mem_align(BlockDriverState *bs)
1254196d2f0SDenis V. Lunev {
1264196d2f0SDenis V. Lunev     if (!bs || !bs->drv) {
127459b4e66SDenis V. Lunev         /* page size or 4k (hdd sector size) should be on the safe side */
128459b4e66SDenis V. Lunev         return MAX(4096, getpagesize());
1294196d2f0SDenis V. Lunev     }
1304196d2f0SDenis V. Lunev 
1314196d2f0SDenis V. Lunev     return bs->bl.min_mem_alignment;
1324196d2f0SDenis V. Lunev }
1334196d2f0SDenis V. Lunev 
1349e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
1355c98415bSMax Reitz int path_has_protocol(const char *path)
1369e0b22f4SStefan Hajnoczi {
137947995c0SPaolo Bonzini     const char *p;
138947995c0SPaolo Bonzini 
1399e0b22f4SStefan Hajnoczi #ifdef _WIN32
1409e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
1419e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
1429e0b22f4SStefan Hajnoczi         return 0;
1439e0b22f4SStefan Hajnoczi     }
144947995c0SPaolo Bonzini     p = path + strcspn(path, ":/\\");
145947995c0SPaolo Bonzini #else
146947995c0SPaolo Bonzini     p = path + strcspn(path, ":/");
1479e0b22f4SStefan Hajnoczi #endif
1489e0b22f4SStefan Hajnoczi 
149947995c0SPaolo Bonzini     return *p == ':';
1509e0b22f4SStefan Hajnoczi }
1519e0b22f4SStefan Hajnoczi 
15283f64091Sbellard int path_is_absolute(const char *path)
15383f64091Sbellard {
15421664424Sbellard #ifdef _WIN32
15521664424Sbellard     /* specific case for names like: "\\.\d:" */
156f53f4da9SPaolo Bonzini     if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
15721664424Sbellard         return 1;
158f53f4da9SPaolo Bonzini     }
159f53f4da9SPaolo Bonzini     return (*path == '/' || *path == '\\');
1603b9f94e1Sbellard #else
161f53f4da9SPaolo Bonzini     return (*path == '/');
1623b9f94e1Sbellard #endif
16383f64091Sbellard }
16483f64091Sbellard 
16583f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
16683f64091Sbellard    path to it by considering it is relative to base_path. URL are
16783f64091Sbellard    supported. */
16883f64091Sbellard void path_combine(char *dest, int dest_size,
16983f64091Sbellard                   const char *base_path,
17083f64091Sbellard                   const char *filename)
17183f64091Sbellard {
17283f64091Sbellard     const char *p, *p1;
17383f64091Sbellard     int len;
17483f64091Sbellard 
17583f64091Sbellard     if (dest_size <= 0)
17683f64091Sbellard         return;
17783f64091Sbellard     if (path_is_absolute(filename)) {
17883f64091Sbellard         pstrcpy(dest, dest_size, filename);
17983f64091Sbellard     } else {
18083f64091Sbellard         p = strchr(base_path, ':');
18183f64091Sbellard         if (p)
18283f64091Sbellard             p++;
18383f64091Sbellard         else
18483f64091Sbellard             p = base_path;
1853b9f94e1Sbellard         p1 = strrchr(base_path, '/');
1863b9f94e1Sbellard #ifdef _WIN32
1873b9f94e1Sbellard         {
1883b9f94e1Sbellard             const char *p2;
1893b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
1903b9f94e1Sbellard             if (!p1 || p2 > p1)
1913b9f94e1Sbellard                 p1 = p2;
1923b9f94e1Sbellard         }
1933b9f94e1Sbellard #endif
19483f64091Sbellard         if (p1)
19583f64091Sbellard             p1++;
19683f64091Sbellard         else
19783f64091Sbellard             p1 = base_path;
19883f64091Sbellard         if (p1 > p)
19983f64091Sbellard             p = p1;
20083f64091Sbellard         len = p - base_path;
20183f64091Sbellard         if (len > dest_size - 1)
20283f64091Sbellard             len = dest_size - 1;
20383f64091Sbellard         memcpy(dest, base_path, len);
20483f64091Sbellard         dest[len] = '\0';
20583f64091Sbellard         pstrcat(dest, dest_size, filename);
20683f64091Sbellard     }
20783f64091Sbellard }
20883f64091Sbellard 
2090a82855aSMax Reitz void bdrv_get_full_backing_filename_from_filename(const char *backed,
2100a82855aSMax Reitz                                                   const char *backing,
2119f07429eSMax Reitz                                                   char *dest, size_t sz,
2129f07429eSMax Reitz                                                   Error **errp)
2130a82855aSMax Reitz {
2149f07429eSMax Reitz     if (backing[0] == '\0' || path_has_protocol(backing) ||
2159f07429eSMax Reitz         path_is_absolute(backing))
2169f07429eSMax Reitz     {
2170a82855aSMax Reitz         pstrcpy(dest, sz, backing);
2189f07429eSMax Reitz     } else if (backed[0] == '\0' || strstart(backed, "json:", NULL)) {
2199f07429eSMax Reitz         error_setg(errp, "Cannot use relative backing file names for '%s'",
2209f07429eSMax Reitz                    backed);
2210a82855aSMax Reitz     } else {
2220a82855aSMax Reitz         path_combine(dest, sz, backed, backing);
2230a82855aSMax Reitz     }
2240a82855aSMax Reitz }
2250a82855aSMax Reitz 
2269f07429eSMax Reitz void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz,
2279f07429eSMax Reitz                                     Error **errp)
228dc5a1371SPaolo Bonzini {
2299f07429eSMax Reitz     char *backed = bs->exact_filename[0] ? bs->exact_filename : bs->filename;
2309f07429eSMax Reitz 
2319f07429eSMax Reitz     bdrv_get_full_backing_filename_from_filename(backed, bs->backing_file,
2329f07429eSMax Reitz                                                  dest, sz, errp);
233dc5a1371SPaolo Bonzini }
234dc5a1371SPaolo Bonzini 
2350eb7217eSStefan Hajnoczi void bdrv_register(BlockDriver *bdrv)
2360eb7217eSStefan Hajnoczi {
2370eb7217eSStefan Hajnoczi     bdrv_setup_io_funcs(bdrv);
238b2e12bc6SChristoph Hellwig 
2398a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
240ea2384d3Sbellard }
241b338082bSbellard 
2427f06d47eSMarkus Armbruster BlockDriverState *bdrv_new_root(void)
243fc01f7e7Sbellard {
2447f06d47eSMarkus Armbruster     BlockDriverState *bs = bdrv_new();
245e4e9986bSMarkus Armbruster 
246e4e9986bSMarkus Armbruster     QTAILQ_INSERT_TAIL(&bdrv_states, bs, device_list);
247e4e9986bSMarkus Armbruster     return bs;
248e4e9986bSMarkus Armbruster }
249e4e9986bSMarkus Armbruster 
250e4e9986bSMarkus Armbruster BlockDriverState *bdrv_new(void)
251e4e9986bSMarkus Armbruster {
252e4e9986bSMarkus Armbruster     BlockDriverState *bs;
253e4e9986bSMarkus Armbruster     int i;
254e4e9986bSMarkus Armbruster 
2555839e53bSMarkus Armbruster     bs = g_new0(BlockDriverState, 1);
256e4654d2dSFam Zheng     QLIST_INIT(&bs->dirty_bitmaps);
257fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
258fbe40ff7SFam Zheng         QLIST_INIT(&bs->op_blockers[i]);
259fbe40ff7SFam Zheng     }
26028a7282aSLuiz Capitulino     bdrv_iostatus_disable(bs);
261d7d512f6SPaolo Bonzini     notifier_list_init(&bs->close_notifiers);
262d616b224SStefan Hajnoczi     notifier_with_return_list_init(&bs->before_write_notifiers);
263cc0681c4SBenoît Canet     qemu_co_queue_init(&bs->throttled_reqs[0]);
264cc0681c4SBenoît Canet     qemu_co_queue_init(&bs->throttled_reqs[1]);
2659fcb0251SFam Zheng     bs->refcnt = 1;
266dcd04228SStefan Hajnoczi     bs->aio_context = qemu_get_aio_context();
267d7d512f6SPaolo Bonzini 
268b338082bSbellard     return bs;
269b338082bSbellard }
270b338082bSbellard 
271d7d512f6SPaolo Bonzini void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
272d7d512f6SPaolo Bonzini {
273d7d512f6SPaolo Bonzini     notifier_list_add(&bs->close_notifiers, notify);
274d7d512f6SPaolo Bonzini }
275d7d512f6SPaolo Bonzini 
276ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
277ea2384d3Sbellard {
278ea2384d3Sbellard     BlockDriver *drv1;
2798a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
2808a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
281ea2384d3Sbellard             return drv1;
282ea2384d3Sbellard         }
2838a22f02aSStefan Hajnoczi     }
284ea2384d3Sbellard     return NULL;
285ea2384d3Sbellard }
286ea2384d3Sbellard 
287b64ec4e4SFam Zheng static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
288eb852011SMarkus Armbruster {
289b64ec4e4SFam Zheng     static const char *whitelist_rw[] = {
290b64ec4e4SFam Zheng         CONFIG_BDRV_RW_WHITELIST
291b64ec4e4SFam Zheng     };
292b64ec4e4SFam Zheng     static const char *whitelist_ro[] = {
293b64ec4e4SFam Zheng         CONFIG_BDRV_RO_WHITELIST
294eb852011SMarkus Armbruster     };
295eb852011SMarkus Armbruster     const char **p;
296eb852011SMarkus Armbruster 
297b64ec4e4SFam Zheng     if (!whitelist_rw[0] && !whitelist_ro[0]) {
298eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
299b64ec4e4SFam Zheng     }
300eb852011SMarkus Armbruster 
301b64ec4e4SFam Zheng     for (p = whitelist_rw; *p; p++) {
302eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
303eb852011SMarkus Armbruster             return 1;
304eb852011SMarkus Armbruster         }
305eb852011SMarkus Armbruster     }
306b64ec4e4SFam Zheng     if (read_only) {
307b64ec4e4SFam Zheng         for (p = whitelist_ro; *p; p++) {
308b64ec4e4SFam Zheng             if (!strcmp(drv->format_name, *p)) {
309b64ec4e4SFam Zheng                 return 1;
310b64ec4e4SFam Zheng             }
311b64ec4e4SFam Zheng         }
312b64ec4e4SFam Zheng     }
313eb852011SMarkus Armbruster     return 0;
314eb852011SMarkus Armbruster }
315eb852011SMarkus Armbruster 
3165b7e1542SZhi Yong Wu typedef struct CreateCo {
3175b7e1542SZhi Yong Wu     BlockDriver *drv;
3185b7e1542SZhi Yong Wu     char *filename;
31983d0521aSChunyan Liu     QemuOpts *opts;
3205b7e1542SZhi Yong Wu     int ret;
321cc84d90fSMax Reitz     Error *err;
3225b7e1542SZhi Yong Wu } CreateCo;
3235b7e1542SZhi Yong Wu 
3245b7e1542SZhi Yong Wu static void coroutine_fn bdrv_create_co_entry(void *opaque)
3255b7e1542SZhi Yong Wu {
326cc84d90fSMax Reitz     Error *local_err = NULL;
327cc84d90fSMax Reitz     int ret;
328cc84d90fSMax Reitz 
3295b7e1542SZhi Yong Wu     CreateCo *cco = opaque;
3305b7e1542SZhi Yong Wu     assert(cco->drv);
3315b7e1542SZhi Yong Wu 
332c282e1fdSChunyan Liu     ret = cco->drv->bdrv_create(cco->filename, cco->opts, &local_err);
33384d18f06SMarkus Armbruster     if (local_err) {
334cc84d90fSMax Reitz         error_propagate(&cco->err, local_err);
335cc84d90fSMax Reitz     }
336cc84d90fSMax Reitz     cco->ret = ret;
3375b7e1542SZhi Yong Wu }
3385b7e1542SZhi Yong Wu 
3390e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
34083d0521aSChunyan Liu                 QemuOpts *opts, Error **errp)
341ea2384d3Sbellard {
3425b7e1542SZhi Yong Wu     int ret;
3430e7e1989SKevin Wolf 
3445b7e1542SZhi Yong Wu     Coroutine *co;
3455b7e1542SZhi Yong Wu     CreateCo cco = {
3465b7e1542SZhi Yong Wu         .drv = drv,
3475b7e1542SZhi Yong Wu         .filename = g_strdup(filename),
34883d0521aSChunyan Liu         .opts = opts,
3495b7e1542SZhi Yong Wu         .ret = NOT_DONE,
350cc84d90fSMax Reitz         .err = NULL,
3515b7e1542SZhi Yong Wu     };
3525b7e1542SZhi Yong Wu 
353c282e1fdSChunyan Liu     if (!drv->bdrv_create) {
354cc84d90fSMax Reitz         error_setg(errp, "Driver '%s' does not support image creation", drv->format_name);
35580168bffSLuiz Capitulino         ret = -ENOTSUP;
35680168bffSLuiz Capitulino         goto out;
3575b7e1542SZhi Yong Wu     }
3585b7e1542SZhi Yong Wu 
3595b7e1542SZhi Yong Wu     if (qemu_in_coroutine()) {
3605b7e1542SZhi Yong Wu         /* Fast-path if already in coroutine context */
3615b7e1542SZhi Yong Wu         bdrv_create_co_entry(&cco);
3625b7e1542SZhi Yong Wu     } else {
3635b7e1542SZhi Yong Wu         co = qemu_coroutine_create(bdrv_create_co_entry);
3645b7e1542SZhi Yong Wu         qemu_coroutine_enter(co, &cco);
3655b7e1542SZhi Yong Wu         while (cco.ret == NOT_DONE) {
366b47ec2c4SPaolo Bonzini             aio_poll(qemu_get_aio_context(), true);
3675b7e1542SZhi Yong Wu         }
3685b7e1542SZhi Yong Wu     }
3695b7e1542SZhi Yong Wu 
3705b7e1542SZhi Yong Wu     ret = cco.ret;
371cc84d90fSMax Reitz     if (ret < 0) {
37284d18f06SMarkus Armbruster         if (cco.err) {
373cc84d90fSMax Reitz             error_propagate(errp, cco.err);
374cc84d90fSMax Reitz         } else {
375cc84d90fSMax Reitz             error_setg_errno(errp, -ret, "Could not create image");
376cc84d90fSMax Reitz         }
377cc84d90fSMax Reitz     }
3785b7e1542SZhi Yong Wu 
37980168bffSLuiz Capitulino out:
38080168bffSLuiz Capitulino     g_free(cco.filename);
3815b7e1542SZhi Yong Wu     return ret;
382ea2384d3Sbellard }
383ea2384d3Sbellard 
384c282e1fdSChunyan Liu int bdrv_create_file(const char *filename, QemuOpts *opts, Error **errp)
38584a12e66SChristoph Hellwig {
38684a12e66SChristoph Hellwig     BlockDriver *drv;
387cc84d90fSMax Reitz     Error *local_err = NULL;
388cc84d90fSMax Reitz     int ret;
38984a12e66SChristoph Hellwig 
390b65a5e12SMax Reitz     drv = bdrv_find_protocol(filename, true, errp);
39184a12e66SChristoph Hellwig     if (drv == NULL) {
39216905d71SStefan Hajnoczi         return -ENOENT;
39384a12e66SChristoph Hellwig     }
39484a12e66SChristoph Hellwig 
395c282e1fdSChunyan Liu     ret = bdrv_create(drv, filename, opts, &local_err);
39684d18f06SMarkus Armbruster     if (local_err) {
397cc84d90fSMax Reitz         error_propagate(errp, local_err);
398cc84d90fSMax Reitz     }
399cc84d90fSMax Reitz     return ret;
40084a12e66SChristoph Hellwig }
40184a12e66SChristoph Hellwig 
402892b7de8SEkaterina Tumanova /**
403892b7de8SEkaterina Tumanova  * Try to get @bs's logical and physical block size.
404892b7de8SEkaterina Tumanova  * On success, store them in @bsz struct and return 0.
405892b7de8SEkaterina Tumanova  * On failure return -errno.
406892b7de8SEkaterina Tumanova  * @bs must not be empty.
407892b7de8SEkaterina Tumanova  */
408892b7de8SEkaterina Tumanova int bdrv_probe_blocksizes(BlockDriverState *bs, BlockSizes *bsz)
409892b7de8SEkaterina Tumanova {
410892b7de8SEkaterina Tumanova     BlockDriver *drv = bs->drv;
411892b7de8SEkaterina Tumanova 
412892b7de8SEkaterina Tumanova     if (drv && drv->bdrv_probe_blocksizes) {
413892b7de8SEkaterina Tumanova         return drv->bdrv_probe_blocksizes(bs, bsz);
414892b7de8SEkaterina Tumanova     }
415892b7de8SEkaterina Tumanova 
416892b7de8SEkaterina Tumanova     return -ENOTSUP;
417892b7de8SEkaterina Tumanova }
418892b7de8SEkaterina Tumanova 
419892b7de8SEkaterina Tumanova /**
420892b7de8SEkaterina Tumanova  * Try to get @bs's geometry (cyls, heads, sectors).
421892b7de8SEkaterina Tumanova  * On success, store them in @geo struct and return 0.
422892b7de8SEkaterina Tumanova  * On failure return -errno.
423892b7de8SEkaterina Tumanova  * @bs must not be empty.
424892b7de8SEkaterina Tumanova  */
425892b7de8SEkaterina Tumanova int bdrv_probe_geometry(BlockDriverState *bs, HDGeometry *geo)
426892b7de8SEkaterina Tumanova {
427892b7de8SEkaterina Tumanova     BlockDriver *drv = bs->drv;
428892b7de8SEkaterina Tumanova 
429892b7de8SEkaterina Tumanova     if (drv && drv->bdrv_probe_geometry) {
430892b7de8SEkaterina Tumanova         return drv->bdrv_probe_geometry(bs, geo);
431892b7de8SEkaterina Tumanova     }
432892b7de8SEkaterina Tumanova 
433892b7de8SEkaterina Tumanova     return -ENOTSUP;
434892b7de8SEkaterina Tumanova }
435892b7de8SEkaterina Tumanova 
436eba25057SJim Meyering /*
437eba25057SJim Meyering  * Create a uniquely-named empty temporary file.
438eba25057SJim Meyering  * Return 0 upon success, otherwise a negative errno value.
439eba25057SJim Meyering  */
440eba25057SJim Meyering int get_tmp_filename(char *filename, int size)
441eba25057SJim Meyering {
442d5249393Sbellard #ifdef _WIN32
4433b9f94e1Sbellard     char temp_dir[MAX_PATH];
444eba25057SJim Meyering     /* GetTempFileName requires that its output buffer (4th param)
445eba25057SJim Meyering        have length MAX_PATH or greater.  */
446eba25057SJim Meyering     assert(size >= MAX_PATH);
447eba25057SJim Meyering     return (GetTempPath(MAX_PATH, temp_dir)
448eba25057SJim Meyering             && GetTempFileName(temp_dir, "qem", 0, filename)
449eba25057SJim Meyering             ? 0 : -GetLastError());
450d5249393Sbellard #else
451ea2384d3Sbellard     int fd;
4527ccfb2ebSblueswir1     const char *tmpdir;
4530badc1eeSaurel32     tmpdir = getenv("TMPDIR");
45469bef793SAmit Shah     if (!tmpdir) {
45569bef793SAmit Shah         tmpdir = "/var/tmp";
45669bef793SAmit Shah     }
457eba25057SJim Meyering     if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
458eba25057SJim Meyering         return -EOVERFLOW;
459ea2384d3Sbellard     }
460eba25057SJim Meyering     fd = mkstemp(filename);
461fe235a06SDunrong Huang     if (fd < 0) {
462fe235a06SDunrong Huang         return -errno;
463fe235a06SDunrong Huang     }
464fe235a06SDunrong Huang     if (close(fd) != 0) {
465fe235a06SDunrong Huang         unlink(filename);
466eba25057SJim Meyering         return -errno;
467eba25057SJim Meyering     }
468eba25057SJim Meyering     return 0;
469d5249393Sbellard #endif
470eba25057SJim Meyering }
471ea2384d3Sbellard 
472f3a5d3f8SChristoph Hellwig /*
473f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
474f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
475f3a5d3f8SChristoph Hellwig  */
476f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
477f3a5d3f8SChristoph Hellwig {
478508c7cb3SChristoph Hellwig     int score_max = 0, score;
479508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
480f3a5d3f8SChristoph Hellwig 
4818a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
482508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
483508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
484508c7cb3SChristoph Hellwig             if (score > score_max) {
485508c7cb3SChristoph Hellwig                 score_max = score;
486508c7cb3SChristoph Hellwig                 drv = d;
487f3a5d3f8SChristoph Hellwig             }
488508c7cb3SChristoph Hellwig         }
489f3a5d3f8SChristoph Hellwig     }
490f3a5d3f8SChristoph Hellwig 
491508c7cb3SChristoph Hellwig     return drv;
492f3a5d3f8SChristoph Hellwig }
493f3a5d3f8SChristoph Hellwig 
49498289620SKevin Wolf BlockDriver *bdrv_find_protocol(const char *filename,
495b65a5e12SMax Reitz                                 bool allow_protocol_prefix,
496b65a5e12SMax Reitz                                 Error **errp)
49784a12e66SChristoph Hellwig {
49884a12e66SChristoph Hellwig     BlockDriver *drv1;
49984a12e66SChristoph Hellwig     char protocol[128];
50084a12e66SChristoph Hellwig     int len;
50184a12e66SChristoph Hellwig     const char *p;
50284a12e66SChristoph Hellwig 
50366f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
50466f82ceeSKevin Wolf 
50539508e7aSChristoph Hellwig     /*
50639508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
50739508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
50839508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
50939508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
51039508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
51139508e7aSChristoph Hellwig      */
51284a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
51339508e7aSChristoph Hellwig     if (drv1) {
51484a12e66SChristoph Hellwig         return drv1;
51584a12e66SChristoph Hellwig     }
51639508e7aSChristoph Hellwig 
51798289620SKevin Wolf     if (!path_has_protocol(filename) || !allow_protocol_prefix) {
518ef810437SMax Reitz         return &bdrv_file;
51939508e7aSChristoph Hellwig     }
52098289620SKevin Wolf 
5219e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
5229e0b22f4SStefan Hajnoczi     assert(p != NULL);
52384a12e66SChristoph Hellwig     len = p - filename;
52484a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
52584a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
52684a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
52784a12e66SChristoph Hellwig     protocol[len] = '\0';
52884a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
52984a12e66SChristoph Hellwig         if (drv1->protocol_name &&
53084a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
53184a12e66SChristoph Hellwig             return drv1;
53284a12e66SChristoph Hellwig         }
53384a12e66SChristoph Hellwig     }
534b65a5e12SMax Reitz 
535b65a5e12SMax Reitz     error_setg(errp, "Unknown protocol '%s'", protocol);
53684a12e66SChristoph Hellwig     return NULL;
53784a12e66SChristoph Hellwig }
53884a12e66SChristoph Hellwig 
539c6684249SMarkus Armbruster /*
540c6684249SMarkus Armbruster  * Guess image format by probing its contents.
541c6684249SMarkus Armbruster  * This is not a good idea when your image is raw (CVE-2008-2004), but
542c6684249SMarkus Armbruster  * we do it anyway for backward compatibility.
543c6684249SMarkus Armbruster  *
544c6684249SMarkus Armbruster  * @buf         contains the image's first @buf_size bytes.
5457cddd372SKevin Wolf  * @buf_size    is the buffer size in bytes (generally BLOCK_PROBE_BUF_SIZE,
5467cddd372SKevin Wolf  *              but can be smaller if the image file is smaller)
547c6684249SMarkus Armbruster  * @filename    is its filename.
548c6684249SMarkus Armbruster  *
549c6684249SMarkus Armbruster  * For all block drivers, call the bdrv_probe() method to get its
550c6684249SMarkus Armbruster  * probing score.
551c6684249SMarkus Armbruster  * Return the first block driver with the highest probing score.
552c6684249SMarkus Armbruster  */
55338f3ef57SKevin Wolf BlockDriver *bdrv_probe_all(const uint8_t *buf, int buf_size,
554c6684249SMarkus Armbruster                             const char *filename)
555c6684249SMarkus Armbruster {
556c6684249SMarkus Armbruster     int score_max = 0, score;
557c6684249SMarkus Armbruster     BlockDriver *drv = NULL, *d;
558c6684249SMarkus Armbruster 
559c6684249SMarkus Armbruster     QLIST_FOREACH(d, &bdrv_drivers, list) {
560c6684249SMarkus Armbruster         if (d->bdrv_probe) {
561c6684249SMarkus Armbruster             score = d->bdrv_probe(buf, buf_size, filename);
562c6684249SMarkus Armbruster             if (score > score_max) {
563c6684249SMarkus Armbruster                 score_max = score;
564c6684249SMarkus Armbruster                 drv = d;
565c6684249SMarkus Armbruster             }
566c6684249SMarkus Armbruster         }
567c6684249SMarkus Armbruster     }
568c6684249SMarkus Armbruster 
569c6684249SMarkus Armbruster     return drv;
570c6684249SMarkus Armbruster }
571c6684249SMarkus Armbruster 
572f500a6d3SKevin Wolf static int find_image_format(BlockDriverState *bs, const char *filename,
57334b5d2c6SMax Reitz                              BlockDriver **pdrv, Error **errp)
574ea2384d3Sbellard {
575c6684249SMarkus Armbruster     BlockDriver *drv;
5767cddd372SKevin Wolf     uint8_t buf[BLOCK_PROBE_BUF_SIZE];
577f500a6d3SKevin Wolf     int ret = 0;
578f8ea0b00SNicholas Bellinger 
57908a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
580b192af8aSDimitris Aragiorgis     if (bdrv_is_sg(bs) || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
581ef810437SMax Reitz         *pdrv = &bdrv_raw;
582c98ac35dSStefan Weil         return ret;
5831a396859SNicholas A. Bellinger     }
584f8ea0b00SNicholas Bellinger 
58583f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
586ea2384d3Sbellard     if (ret < 0) {
58734b5d2c6SMax Reitz         error_setg_errno(errp, -ret, "Could not read image for determining its "
58834b5d2c6SMax Reitz                          "format");
589c98ac35dSStefan Weil         *pdrv = NULL;
590c98ac35dSStefan Weil         return ret;
591ea2384d3Sbellard     }
592ea2384d3Sbellard 
593c6684249SMarkus Armbruster     drv = bdrv_probe_all(buf, ret, filename);
594c98ac35dSStefan Weil     if (!drv) {
59534b5d2c6SMax Reitz         error_setg(errp, "Could not determine image format: No compatible "
59634b5d2c6SMax Reitz                    "driver found");
597c98ac35dSStefan Weil         ret = -ENOENT;
598c98ac35dSStefan Weil     }
599c98ac35dSStefan Weil     *pdrv = drv;
600c98ac35dSStefan Weil     return ret;
601ea2384d3Sbellard }
602ea2384d3Sbellard 
60351762288SStefan Hajnoczi /**
60451762288SStefan Hajnoczi  * Set the current 'total_sectors' value
60565a9bb25SMarkus Armbruster  * Return 0 on success, -errno on error.
60651762288SStefan Hajnoczi  */
60751762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
60851762288SStefan Hajnoczi {
60951762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
61051762288SStefan Hajnoczi 
611396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
612b192af8aSDimitris Aragiorgis     if (bdrv_is_sg(bs))
613396759adSNicholas Bellinger         return 0;
614396759adSNicholas Bellinger 
61551762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
61651762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
61751762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
61851762288SStefan Hajnoczi         if (length < 0) {
61951762288SStefan Hajnoczi             return length;
62051762288SStefan Hajnoczi         }
6217e382003SFam Zheng         hint = DIV_ROUND_UP(length, BDRV_SECTOR_SIZE);
62251762288SStefan Hajnoczi     }
62351762288SStefan Hajnoczi 
62451762288SStefan Hajnoczi     bs->total_sectors = hint;
62551762288SStefan Hajnoczi     return 0;
62651762288SStefan Hajnoczi }
62751762288SStefan Hajnoczi 
628c3993cdcSStefan Hajnoczi /**
6299e8f1835SPaolo Bonzini  * Set open flags for a given discard mode
6309e8f1835SPaolo Bonzini  *
6319e8f1835SPaolo Bonzini  * Return 0 on success, -1 if the discard mode was invalid.
6329e8f1835SPaolo Bonzini  */
6339e8f1835SPaolo Bonzini int bdrv_parse_discard_flags(const char *mode, int *flags)
6349e8f1835SPaolo Bonzini {
6359e8f1835SPaolo Bonzini     *flags &= ~BDRV_O_UNMAP;
6369e8f1835SPaolo Bonzini 
6379e8f1835SPaolo Bonzini     if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
6389e8f1835SPaolo Bonzini         /* do nothing */
6399e8f1835SPaolo Bonzini     } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
6409e8f1835SPaolo Bonzini         *flags |= BDRV_O_UNMAP;
6419e8f1835SPaolo Bonzini     } else {
6429e8f1835SPaolo Bonzini         return -1;
6439e8f1835SPaolo Bonzini     }
6449e8f1835SPaolo Bonzini 
6459e8f1835SPaolo Bonzini     return 0;
6469e8f1835SPaolo Bonzini }
6479e8f1835SPaolo Bonzini 
6489e8f1835SPaolo Bonzini /**
649c3993cdcSStefan Hajnoczi  * Set open flags for a given cache mode
650c3993cdcSStefan Hajnoczi  *
651c3993cdcSStefan Hajnoczi  * Return 0 on success, -1 if the cache mode was invalid.
652c3993cdcSStefan Hajnoczi  */
653c3993cdcSStefan Hajnoczi int bdrv_parse_cache_flags(const char *mode, int *flags)
654c3993cdcSStefan Hajnoczi {
655c3993cdcSStefan Hajnoczi     *flags &= ~BDRV_O_CACHE_MASK;
656c3993cdcSStefan Hajnoczi 
657c3993cdcSStefan Hajnoczi     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
658c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
65992196b2fSStefan Hajnoczi     } else if (!strcmp(mode, "directsync")) {
66092196b2fSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE;
661c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writeback")) {
662c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
663c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "unsafe")) {
664c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
665c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NO_FLUSH;
666c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writethrough")) {
667c3993cdcSStefan Hajnoczi         /* this is the default */
668c3993cdcSStefan Hajnoczi     } else {
669c3993cdcSStefan Hajnoczi         return -1;
670c3993cdcSStefan Hajnoczi     }
671c3993cdcSStefan Hajnoczi 
672c3993cdcSStefan Hajnoczi     return 0;
673c3993cdcSStefan Hajnoczi }
674c3993cdcSStefan Hajnoczi 
6750b50cc88SKevin Wolf /*
676b1e6fc08SKevin Wolf  * Returns the flags that a temporary snapshot should get, based on the
677b1e6fc08SKevin Wolf  * originally requested flags (the originally requested image will have flags
678b1e6fc08SKevin Wolf  * like a backing file)
679b1e6fc08SKevin Wolf  */
680b1e6fc08SKevin Wolf static int bdrv_temp_snapshot_flags(int flags)
681b1e6fc08SKevin Wolf {
682b1e6fc08SKevin Wolf     return (flags & ~BDRV_O_SNAPSHOT) | BDRV_O_TEMPORARY;
683b1e6fc08SKevin Wolf }
684b1e6fc08SKevin Wolf 
685b1e6fc08SKevin Wolf /*
686f3930ed0SKevin Wolf  * Returns the flags that bs->file should get if a protocol driver is expected,
687f3930ed0SKevin Wolf  * based on the given flags for the parent BDS
6880b50cc88SKevin Wolf  */
6890b50cc88SKevin Wolf static int bdrv_inherited_flags(int flags)
6900b50cc88SKevin Wolf {
6910b50cc88SKevin Wolf     /* Enable protocol handling, disable format probing for bs->file */
6920b50cc88SKevin Wolf     flags |= BDRV_O_PROTOCOL;
6930b50cc88SKevin Wolf 
6940b50cc88SKevin Wolf     /* Our block drivers take care to send flushes and respect unmap policy,
6950b50cc88SKevin Wolf      * so we can enable both unconditionally on lower layers. */
6960b50cc88SKevin Wolf     flags |= BDRV_O_CACHE_WB | BDRV_O_UNMAP;
6970b50cc88SKevin Wolf 
6980b50cc88SKevin Wolf     /* Clear flags that only apply to the top layer */
6995669b44dSKevin Wolf     flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING | BDRV_O_COPY_ON_READ);
7000b50cc88SKevin Wolf 
7010b50cc88SKevin Wolf     return flags;
7020b50cc88SKevin Wolf }
7030b50cc88SKevin Wolf 
704f3930ed0SKevin Wolf const BdrvChildRole child_file = {
705f3930ed0SKevin Wolf     .inherit_flags = bdrv_inherited_flags,
706f3930ed0SKevin Wolf };
707f3930ed0SKevin Wolf 
708f3930ed0SKevin Wolf /*
709f3930ed0SKevin Wolf  * Returns the flags that bs->file should get if the use of formats (and not
710f3930ed0SKevin Wolf  * only protocols) is permitted for it, based on the given flags for the parent
711f3930ed0SKevin Wolf  * BDS
712f3930ed0SKevin Wolf  */
713f3930ed0SKevin Wolf static int bdrv_inherited_fmt_flags(int parent_flags)
714f3930ed0SKevin Wolf {
715f3930ed0SKevin Wolf     int flags = child_file.inherit_flags(parent_flags);
716f3930ed0SKevin Wolf     return flags & ~BDRV_O_PROTOCOL;
717f3930ed0SKevin Wolf }
718f3930ed0SKevin Wolf 
719f3930ed0SKevin Wolf const BdrvChildRole child_format = {
720f3930ed0SKevin Wolf     .inherit_flags = bdrv_inherited_fmt_flags,
721f3930ed0SKevin Wolf };
722f3930ed0SKevin Wolf 
723317fc44eSKevin Wolf /*
724317fc44eSKevin Wolf  * Returns the flags that bs->backing_hd should get, based on the given flags
725317fc44eSKevin Wolf  * for the parent BDS
726317fc44eSKevin Wolf  */
727317fc44eSKevin Wolf static int bdrv_backing_flags(int flags)
728317fc44eSKevin Wolf {
729317fc44eSKevin Wolf     /* backing files always opened read-only */
730317fc44eSKevin Wolf     flags &= ~(BDRV_O_RDWR | BDRV_O_COPY_ON_READ);
731317fc44eSKevin Wolf 
732317fc44eSKevin Wolf     /* snapshot=on is handled on the top layer */
7338bfea15dSKevin Wolf     flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_TEMPORARY);
734317fc44eSKevin Wolf 
735317fc44eSKevin Wolf     return flags;
736317fc44eSKevin Wolf }
737317fc44eSKevin Wolf 
738f3930ed0SKevin Wolf static const BdrvChildRole child_backing = {
739f3930ed0SKevin Wolf     .inherit_flags = bdrv_backing_flags,
740f3930ed0SKevin Wolf };
741f3930ed0SKevin Wolf 
7427b272452SKevin Wolf static int bdrv_open_flags(BlockDriverState *bs, int flags)
7437b272452SKevin Wolf {
7447b272452SKevin Wolf     int open_flags = flags | BDRV_O_CACHE_WB;
7457b272452SKevin Wolf 
7467b272452SKevin Wolf     /*
7477b272452SKevin Wolf      * Clear flags that are internal to the block layer before opening the
7487b272452SKevin Wolf      * image.
7497b272452SKevin Wolf      */
75020cca275SKevin Wolf     open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING | BDRV_O_PROTOCOL);
7517b272452SKevin Wolf 
7527b272452SKevin Wolf     /*
7537b272452SKevin Wolf      * Snapshots should be writable.
7547b272452SKevin Wolf      */
7558bfea15dSKevin Wolf     if (flags & BDRV_O_TEMPORARY) {
7567b272452SKevin Wolf         open_flags |= BDRV_O_RDWR;
7577b272452SKevin Wolf     }
7587b272452SKevin Wolf 
7597b272452SKevin Wolf     return open_flags;
7607b272452SKevin Wolf }
7617b272452SKevin Wolf 
762636ea370SKevin Wolf static void bdrv_assign_node_name(BlockDriverState *bs,
7636913c0c2SBenoît Canet                                   const char *node_name,
7646913c0c2SBenoît Canet                                   Error **errp)
7656913c0c2SBenoît Canet {
7666913c0c2SBenoît Canet     if (!node_name) {
767636ea370SKevin Wolf         return;
7686913c0c2SBenoît Canet     }
7696913c0c2SBenoît Canet 
7709aebf3b8SKevin Wolf     /* Check for empty string or invalid characters */
771f5bebbbbSMarkus Armbruster     if (!id_wellformed(node_name)) {
7729aebf3b8SKevin Wolf         error_setg(errp, "Invalid node name");
773636ea370SKevin Wolf         return;
7746913c0c2SBenoît Canet     }
7756913c0c2SBenoît Canet 
7760c5e94eeSBenoît Canet     /* takes care of avoiding namespaces collisions */
7777f06d47eSMarkus Armbruster     if (blk_by_name(node_name)) {
7780c5e94eeSBenoît Canet         error_setg(errp, "node-name=%s is conflicting with a device id",
7790c5e94eeSBenoît Canet                    node_name);
780636ea370SKevin Wolf         return;
7810c5e94eeSBenoît Canet     }
7820c5e94eeSBenoît Canet 
7836913c0c2SBenoît Canet     /* takes care of avoiding duplicates node names */
7846913c0c2SBenoît Canet     if (bdrv_find_node(node_name)) {
7856913c0c2SBenoît Canet         error_setg(errp, "Duplicate node name");
786636ea370SKevin Wolf         return;
7876913c0c2SBenoît Canet     }
7886913c0c2SBenoît Canet 
7896913c0c2SBenoît Canet     /* copy node name into the bs and insert it into the graph list */
7906913c0c2SBenoît Canet     pstrcpy(bs->node_name, sizeof(bs->node_name), node_name);
7916913c0c2SBenoît Canet     QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs, node_list);
7926913c0c2SBenoît Canet }
7936913c0c2SBenoît Canet 
79418edf289SKevin Wolf static QemuOptsList bdrv_runtime_opts = {
79518edf289SKevin Wolf     .name = "bdrv_common",
79618edf289SKevin Wolf     .head = QTAILQ_HEAD_INITIALIZER(bdrv_runtime_opts.head),
79718edf289SKevin Wolf     .desc = {
79818edf289SKevin Wolf         {
79918edf289SKevin Wolf             .name = "node-name",
80018edf289SKevin Wolf             .type = QEMU_OPT_STRING,
80118edf289SKevin Wolf             .help = "Node name of the block device node",
80218edf289SKevin Wolf         },
80318edf289SKevin Wolf         { /* end of list */ }
80418edf289SKevin Wolf     },
80518edf289SKevin Wolf };
80618edf289SKevin Wolf 
807b6ce07aaSKevin Wolf /*
80857915332SKevin Wolf  * Common part for opening disk images and files
809b6ad491aSKevin Wolf  *
810b6ad491aSKevin Wolf  * Removes all processed options from *options.
81157915332SKevin Wolf  */
812*9a4f4c31SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, BdrvChild *file,
81334b5d2c6SMax Reitz     QDict *options, int flags, BlockDriver *drv, Error **errp)
81457915332SKevin Wolf {
81557915332SKevin Wolf     int ret, open_flags;
816035fccdfSKevin Wolf     const char *filename;
8176913c0c2SBenoît Canet     const char *node_name = NULL;
81818edf289SKevin Wolf     QemuOpts *opts;
81934b5d2c6SMax Reitz     Error *local_err = NULL;
82057915332SKevin Wolf 
82157915332SKevin Wolf     assert(drv != NULL);
8226405875cSPaolo Bonzini     assert(bs->file == NULL);
823707ff828SKevin Wolf     assert(options != NULL && bs->options != options);
82457915332SKevin Wolf 
82545673671SKevin Wolf     if (file != NULL) {
826*9a4f4c31SKevin Wolf         filename = file->bs->filename;
82745673671SKevin Wolf     } else {
82845673671SKevin Wolf         filename = qdict_get_try_str(options, "filename");
82945673671SKevin Wolf     }
83045673671SKevin Wolf 
831765003dbSKevin Wolf     if (drv->bdrv_needs_filename && !filename) {
832765003dbSKevin Wolf         error_setg(errp, "The '%s' block driver requires a file name",
833765003dbSKevin Wolf                    drv->format_name);
834765003dbSKevin Wolf         return -EINVAL;
835765003dbSKevin Wolf     }
836765003dbSKevin Wolf 
83745673671SKevin Wolf     trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
83828dcee10SStefan Hajnoczi 
83918edf289SKevin Wolf     opts = qemu_opts_create(&bdrv_runtime_opts, NULL, 0, &error_abort);
84018edf289SKevin Wolf     qemu_opts_absorb_qdict(opts, options, &local_err);
84118edf289SKevin Wolf     if (local_err) {
84218edf289SKevin Wolf         error_propagate(errp, local_err);
84318edf289SKevin Wolf         ret = -EINVAL;
84418edf289SKevin Wolf         goto fail_opts;
84518edf289SKevin Wolf     }
84618edf289SKevin Wolf 
84718edf289SKevin Wolf     node_name = qemu_opt_get(opts, "node-name");
848636ea370SKevin Wolf     bdrv_assign_node_name(bs, node_name, &local_err);
8490fb6395cSMarkus Armbruster     if (local_err) {
850636ea370SKevin Wolf         error_propagate(errp, local_err);
85118edf289SKevin Wolf         ret = -EINVAL;
85218edf289SKevin Wolf         goto fail_opts;
8535d186eb0SKevin Wolf     }
8545d186eb0SKevin Wolf 
8551b7fd729SPaolo Bonzini     bs->guest_block_size = 512;
856c25f53b0SPaolo Bonzini     bs->request_alignment = 512;
8570d51b4deSAsias He     bs->zero_beyond_eof = true;
858b64ec4e4SFam Zheng     open_flags = bdrv_open_flags(bs, flags);
859b64ec4e4SFam Zheng     bs->read_only = !(open_flags & BDRV_O_RDWR);
860b64ec4e4SFam Zheng 
861b64ec4e4SFam Zheng     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv, bs->read_only)) {
8628f94a6e4SKevin Wolf         error_setg(errp,
8638f94a6e4SKevin Wolf                    !bs->read_only && bdrv_is_whitelisted(drv, true)
8648f94a6e4SKevin Wolf                         ? "Driver '%s' can only be used for read-only devices"
8658f94a6e4SKevin Wolf                         : "Driver '%s' is not whitelisted",
8668f94a6e4SKevin Wolf                    drv->format_name);
86718edf289SKevin Wolf         ret = -ENOTSUP;
86818edf289SKevin Wolf         goto fail_opts;
869b64ec4e4SFam Zheng     }
87057915332SKevin Wolf 
87153fec9d3SStefan Hajnoczi     assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
8720ebd24e0SKevin Wolf     if (flags & BDRV_O_COPY_ON_READ) {
8730ebd24e0SKevin Wolf         if (!bs->read_only) {
87453fec9d3SStefan Hajnoczi             bdrv_enable_copy_on_read(bs);
8750ebd24e0SKevin Wolf         } else {
8760ebd24e0SKevin Wolf             error_setg(errp, "Can't use copy-on-read on read-only device");
87718edf289SKevin Wolf             ret = -EINVAL;
87818edf289SKevin Wolf             goto fail_opts;
8790ebd24e0SKevin Wolf         }
88053fec9d3SStefan Hajnoczi     }
88153fec9d3SStefan Hajnoczi 
882c2ad1b0cSKevin Wolf     if (filename != NULL) {
88357915332SKevin Wolf         pstrcpy(bs->filename, sizeof(bs->filename), filename);
884c2ad1b0cSKevin Wolf     } else {
885c2ad1b0cSKevin Wolf         bs->filename[0] = '\0';
886c2ad1b0cSKevin Wolf     }
88791af7014SMax Reitz     pstrcpy(bs->exact_filename, sizeof(bs->exact_filename), bs->filename);
88857915332SKevin Wolf 
88957915332SKevin Wolf     bs->drv = drv;
8907267c094SAnthony Liguori     bs->opaque = g_malloc0(drv->instance_size);
89157915332SKevin Wolf 
89203f541bdSStefan Hajnoczi     bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
893e7c63796SStefan Hajnoczi 
89466f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
89566f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
8965d186eb0SKevin Wolf         assert(file == NULL);
897030be321SBenoît Canet         assert(!drv->bdrv_needs_filename || filename != NULL);
89834b5d2c6SMax Reitz         ret = drv->bdrv_file_open(bs, options, open_flags, &local_err);
899f500a6d3SKevin Wolf     } else {
9002af5ef70SKevin Wolf         if (file == NULL) {
90134b5d2c6SMax Reitz             error_setg(errp, "Can't use '%s' as a block driver for the "
90234b5d2c6SMax Reitz                        "protocol level", drv->format_name);
9032af5ef70SKevin Wolf             ret = -EINVAL;
9042af5ef70SKevin Wolf             goto free_and_fail;
9052af5ef70SKevin Wolf         }
906f500a6d3SKevin Wolf         bs->file = file;
90734b5d2c6SMax Reitz         ret = drv->bdrv_open(bs, options, open_flags, &local_err);
90866f82ceeSKevin Wolf     }
90966f82ceeSKevin Wolf 
91057915332SKevin Wolf     if (ret < 0) {
91184d18f06SMarkus Armbruster         if (local_err) {
91234b5d2c6SMax Reitz             error_propagate(errp, local_err);
9132fa9aa59SDunrong Huang         } else if (bs->filename[0]) {
9142fa9aa59SDunrong Huang             error_setg_errno(errp, -ret, "Could not open '%s'", bs->filename);
91534b5d2c6SMax Reitz         } else {
91634b5d2c6SMax Reitz             error_setg_errno(errp, -ret, "Could not open image");
91734b5d2c6SMax Reitz         }
91857915332SKevin Wolf         goto free_and_fail;
91957915332SKevin Wolf     }
92057915332SKevin Wolf 
921a1f688f4SMarkus Armbruster     if (bs->encrypted) {
922a1f688f4SMarkus Armbruster         error_report("Encrypted images are deprecated");
923a1f688f4SMarkus Armbruster         error_printf("Support for them will be removed in a future release.\n"
924a1f688f4SMarkus Armbruster                      "You can use 'qemu-img convert' to convert your image"
925a1f688f4SMarkus Armbruster                      " to an unencrypted one.\n");
926a1f688f4SMarkus Armbruster     }
927a1f688f4SMarkus Armbruster 
92851762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
92951762288SStefan Hajnoczi     if (ret < 0) {
93034b5d2c6SMax Reitz         error_setg_errno(errp, -ret, "Could not refresh total sector count");
93151762288SStefan Hajnoczi         goto free_and_fail;
93257915332SKevin Wolf     }
93351762288SStefan Hajnoczi 
9343baca891SKevin Wolf     bdrv_refresh_limits(bs, &local_err);
9353baca891SKevin Wolf     if (local_err) {
9363baca891SKevin Wolf         error_propagate(errp, local_err);
9373baca891SKevin Wolf         ret = -EINVAL;
9383baca891SKevin Wolf         goto free_and_fail;
9393baca891SKevin Wolf     }
9403baca891SKevin Wolf 
941c25f53b0SPaolo Bonzini     assert(bdrv_opt_mem_align(bs) != 0);
9424196d2f0SDenis V. Lunev     assert(bdrv_min_mem_align(bs) != 0);
943b192af8aSDimitris Aragiorgis     assert((bs->request_alignment != 0) || bdrv_is_sg(bs));
94418edf289SKevin Wolf 
94518edf289SKevin Wolf     qemu_opts_del(opts);
94657915332SKevin Wolf     return 0;
94757915332SKevin Wolf 
94857915332SKevin Wolf free_and_fail:
94966f82ceeSKevin Wolf     bs->file = NULL;
9507267c094SAnthony Liguori     g_free(bs->opaque);
95157915332SKevin Wolf     bs->opaque = NULL;
95257915332SKevin Wolf     bs->drv = NULL;
95318edf289SKevin Wolf fail_opts:
95418edf289SKevin Wolf     qemu_opts_del(opts);
95557915332SKevin Wolf     return ret;
95657915332SKevin Wolf }
95757915332SKevin Wolf 
9585e5c4f63SKevin Wolf static QDict *parse_json_filename(const char *filename, Error **errp)
9595e5c4f63SKevin Wolf {
9605e5c4f63SKevin Wolf     QObject *options_obj;
9615e5c4f63SKevin Wolf     QDict *options;
9625e5c4f63SKevin Wolf     int ret;
9635e5c4f63SKevin Wolf 
9645e5c4f63SKevin Wolf     ret = strstart(filename, "json:", &filename);
9655e5c4f63SKevin Wolf     assert(ret);
9665e5c4f63SKevin Wolf 
9675e5c4f63SKevin Wolf     options_obj = qobject_from_json(filename);
9685e5c4f63SKevin Wolf     if (!options_obj) {
9695e5c4f63SKevin Wolf         error_setg(errp, "Could not parse the JSON options");
9705e5c4f63SKevin Wolf         return NULL;
9715e5c4f63SKevin Wolf     }
9725e5c4f63SKevin Wolf 
9735e5c4f63SKevin Wolf     if (qobject_type(options_obj) != QTYPE_QDICT) {
9745e5c4f63SKevin Wolf         qobject_decref(options_obj);
9755e5c4f63SKevin Wolf         error_setg(errp, "Invalid JSON object given");
9765e5c4f63SKevin Wolf         return NULL;
9775e5c4f63SKevin Wolf     }
9785e5c4f63SKevin Wolf 
9795e5c4f63SKevin Wolf     options = qobject_to_qdict(options_obj);
9805e5c4f63SKevin Wolf     qdict_flatten(options);
9815e5c4f63SKevin Wolf 
9825e5c4f63SKevin Wolf     return options;
9835e5c4f63SKevin Wolf }
9845e5c4f63SKevin Wolf 
98557915332SKevin Wolf /*
986f54120ffSKevin Wolf  * Fills in default options for opening images and converts the legacy
987f54120ffSKevin Wolf  * filename/flags pair to option QDict entries.
98853a29513SMax Reitz  * The BDRV_O_PROTOCOL flag in *flags will be set or cleared accordingly if a
98953a29513SMax Reitz  * block driver has been specified explicitly.
990f54120ffSKevin Wolf  */
99153a29513SMax Reitz static int bdrv_fill_options(QDict **options, const char **pfilename,
992053e1578SMax Reitz                              int *flags, Error **errp)
993f54120ffSKevin Wolf {
9945e5c4f63SKevin Wolf     const char *filename = *pfilename;
995f54120ffSKevin Wolf     const char *drvname;
99653a29513SMax Reitz     bool protocol = *flags & BDRV_O_PROTOCOL;
997f54120ffSKevin Wolf     bool parse_filename = false;
998053e1578SMax Reitz     BlockDriver *drv = NULL;
999f54120ffSKevin Wolf     Error *local_err = NULL;
1000f54120ffSKevin Wolf 
10015e5c4f63SKevin Wolf     /* Parse json: pseudo-protocol */
10025e5c4f63SKevin Wolf     if (filename && g_str_has_prefix(filename, "json:")) {
10035e5c4f63SKevin Wolf         QDict *json_options = parse_json_filename(filename, &local_err);
10045e5c4f63SKevin Wolf         if (local_err) {
10055e5c4f63SKevin Wolf             error_propagate(errp, local_err);
10065e5c4f63SKevin Wolf             return -EINVAL;
10075e5c4f63SKevin Wolf         }
10085e5c4f63SKevin Wolf 
10095e5c4f63SKevin Wolf         /* Options given in the filename have lower priority than options
10105e5c4f63SKevin Wolf          * specified directly */
10115e5c4f63SKevin Wolf         qdict_join(*options, json_options, false);
10125e5c4f63SKevin Wolf         QDECREF(json_options);
10135e5c4f63SKevin Wolf         *pfilename = filename = NULL;
10145e5c4f63SKevin Wolf     }
10155e5c4f63SKevin Wolf 
101653a29513SMax Reitz     drvname = qdict_get_try_str(*options, "driver");
1017053e1578SMax Reitz     if (drvname) {
1018053e1578SMax Reitz         drv = bdrv_find_format(drvname);
1019053e1578SMax Reitz         if (!drv) {
1020053e1578SMax Reitz             error_setg(errp, "Unknown driver '%s'", drvname);
1021053e1578SMax Reitz             return -ENOENT;
1022053e1578SMax Reitz         }
102353a29513SMax Reitz         /* If the user has explicitly specified the driver, this choice should
102453a29513SMax Reitz          * override the BDRV_O_PROTOCOL flag */
1025053e1578SMax Reitz         protocol = drv->bdrv_file_open;
102653a29513SMax Reitz     }
102753a29513SMax Reitz 
102853a29513SMax Reitz     if (protocol) {
102953a29513SMax Reitz         *flags |= BDRV_O_PROTOCOL;
103053a29513SMax Reitz     } else {
103153a29513SMax Reitz         *flags &= ~BDRV_O_PROTOCOL;
103253a29513SMax Reitz     }
103353a29513SMax Reitz 
1034f54120ffSKevin Wolf     /* Fetch the file name from the options QDict if necessary */
103517b005f1SKevin Wolf     if (protocol && filename) {
1036f54120ffSKevin Wolf         if (!qdict_haskey(*options, "filename")) {
1037f54120ffSKevin Wolf             qdict_put(*options, "filename", qstring_from_str(filename));
1038f54120ffSKevin Wolf             parse_filename = true;
1039f54120ffSKevin Wolf         } else {
1040f54120ffSKevin Wolf             error_setg(errp, "Can't specify 'file' and 'filename' options at "
1041f54120ffSKevin Wolf                              "the same time");
1042f54120ffSKevin Wolf             return -EINVAL;
1043f54120ffSKevin Wolf         }
1044f54120ffSKevin Wolf     }
1045f54120ffSKevin Wolf 
1046f54120ffSKevin Wolf     /* Find the right block driver */
1047f54120ffSKevin Wolf     filename = qdict_get_try_str(*options, "filename");
1048f54120ffSKevin Wolf 
104917b005f1SKevin Wolf     if (!drvname && protocol) {
1050f54120ffSKevin Wolf         if (filename) {
1051b65a5e12SMax Reitz             drv = bdrv_find_protocol(filename, parse_filename, errp);
1052f54120ffSKevin Wolf             if (!drv) {
1053f54120ffSKevin Wolf                 return -EINVAL;
1054f54120ffSKevin Wolf             }
1055f54120ffSKevin Wolf 
1056f54120ffSKevin Wolf             drvname = drv->format_name;
1057f54120ffSKevin Wolf             qdict_put(*options, "driver", qstring_from_str(drvname));
1058f54120ffSKevin Wolf         } else {
1059f54120ffSKevin Wolf             error_setg(errp, "Must specify either driver or file");
1060f54120ffSKevin Wolf             return -EINVAL;
1061f54120ffSKevin Wolf         }
106217b005f1SKevin Wolf     }
106317b005f1SKevin Wolf 
106417b005f1SKevin Wolf     assert(drv || !protocol);
1065f54120ffSKevin Wolf 
1066f54120ffSKevin Wolf     /* Driver-specific filename parsing */
106717b005f1SKevin Wolf     if (drv && drv->bdrv_parse_filename && parse_filename) {
1068f54120ffSKevin Wolf         drv->bdrv_parse_filename(filename, *options, &local_err);
1069f54120ffSKevin Wolf         if (local_err) {
1070f54120ffSKevin Wolf             error_propagate(errp, local_err);
1071f54120ffSKevin Wolf             return -EINVAL;
1072f54120ffSKevin Wolf         }
1073f54120ffSKevin Wolf 
1074f54120ffSKevin Wolf         if (!drv->bdrv_needs_filename) {
1075f54120ffSKevin Wolf             qdict_del(*options, "filename");
1076f54120ffSKevin Wolf         }
1077f54120ffSKevin Wolf     }
1078f54120ffSKevin Wolf 
1079f54120ffSKevin Wolf     return 0;
1080f54120ffSKevin Wolf }
1081f54120ffSKevin Wolf 
1082b4b059f6SKevin Wolf static BdrvChild *bdrv_attach_child(BlockDriverState *parent_bs,
1083df581792SKevin Wolf                                     BlockDriverState *child_bs,
1084df581792SKevin Wolf                                     const BdrvChildRole *child_role)
1085df581792SKevin Wolf {
1086df581792SKevin Wolf     BdrvChild *child = g_new(BdrvChild, 1);
1087df581792SKevin Wolf     *child = (BdrvChild) {
1088df581792SKevin Wolf         .bs     = child_bs,
1089df581792SKevin Wolf         .role   = child_role,
1090df581792SKevin Wolf     };
1091df581792SKevin Wolf 
1092df581792SKevin Wolf     QLIST_INSERT_HEAD(&parent_bs->children, child, next);
1093b4b059f6SKevin Wolf 
1094b4b059f6SKevin Wolf     return child;
1095df581792SKevin Wolf }
1096df581792SKevin Wolf 
109733a60407SKevin Wolf static void bdrv_detach_child(BdrvChild *child)
109833a60407SKevin Wolf {
109933a60407SKevin Wolf     QLIST_REMOVE(child, next);
110033a60407SKevin Wolf     g_free(child);
110133a60407SKevin Wolf }
110233a60407SKevin Wolf 
110333a60407SKevin Wolf void bdrv_unref_child(BlockDriverState *parent, BdrvChild *child)
110433a60407SKevin Wolf {
110533a60407SKevin Wolf     BlockDriverState *child_bs = child->bs;
110633a60407SKevin Wolf 
110733a60407SKevin Wolf     if (child->bs->inherits_from == parent) {
110833a60407SKevin Wolf         child->bs->inherits_from = NULL;
110933a60407SKevin Wolf     }
111033a60407SKevin Wolf 
111133a60407SKevin Wolf     bdrv_detach_child(child);
111233a60407SKevin Wolf     bdrv_unref(child_bs);
111333a60407SKevin Wolf }
111433a60407SKevin Wolf 
11158d24cce1SFam Zheng void bdrv_set_backing_hd(BlockDriverState *bs, BlockDriverState *backing_hd)
11168d24cce1SFam Zheng {
11178d24cce1SFam Zheng 
1118826b6ca0SFam Zheng     if (bs->backing_hd) {
1119826b6ca0SFam Zheng         assert(bs->backing_blocker);
1120826b6ca0SFam Zheng         bdrv_op_unblock_all(bs->backing_hd, bs->backing_blocker);
112180a1e130SKevin Wolf         bdrv_detach_child(bs->backing_child);
1122826b6ca0SFam Zheng     } else if (backing_hd) {
1123826b6ca0SFam Zheng         error_setg(&bs->backing_blocker,
112481e5f78aSAlberto Garcia                    "node is used as backing hd of '%s'",
112581e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(bs));
1126826b6ca0SFam Zheng     }
1127826b6ca0SFam Zheng 
11288d24cce1SFam Zheng     bs->backing_hd = backing_hd;
11298d24cce1SFam Zheng     if (!backing_hd) {
1130826b6ca0SFam Zheng         error_free(bs->backing_blocker);
1131826b6ca0SFam Zheng         bs->backing_blocker = NULL;
113280a1e130SKevin Wolf         bs->backing_child = NULL;
11338d24cce1SFam Zheng         goto out;
11348d24cce1SFam Zheng     }
113580a1e130SKevin Wolf     bs->backing_child = bdrv_attach_child(bs, backing_hd, &child_backing);
11368d24cce1SFam Zheng     bs->open_flags &= ~BDRV_O_NO_BACKING;
11378d24cce1SFam Zheng     pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_hd->filename);
11388d24cce1SFam Zheng     pstrcpy(bs->backing_format, sizeof(bs->backing_format),
11398d24cce1SFam Zheng             backing_hd->drv ? backing_hd->drv->format_name : "");
1140826b6ca0SFam Zheng 
1141826b6ca0SFam Zheng     bdrv_op_block_all(bs->backing_hd, bs->backing_blocker);
1142826b6ca0SFam Zheng     /* Otherwise we won't be able to commit due to check in bdrv_commit */
1143bb00021dSFam Zheng     bdrv_op_unblock(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET,
1144826b6ca0SFam Zheng                     bs->backing_blocker);
11458d24cce1SFam Zheng out:
11463baca891SKevin Wolf     bdrv_refresh_limits(bs, NULL);
11478d24cce1SFam Zheng }
11488d24cce1SFam Zheng 
114931ca6d07SKevin Wolf /*
115031ca6d07SKevin Wolf  * Opens the backing file for a BlockDriverState if not yet open
115131ca6d07SKevin Wolf  *
115231ca6d07SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
115331ca6d07SKevin Wolf  * empty set of options. The reference to the QDict is transferred to this
115431ca6d07SKevin Wolf  * function (even on failure), so if the caller intends to reuse the dictionary,
115531ca6d07SKevin Wolf  * it needs to use QINCREF() before calling bdrv_file_open.
115631ca6d07SKevin Wolf  */
115734b5d2c6SMax Reitz int bdrv_open_backing_file(BlockDriverState *bs, QDict *options, Error **errp)
11589156df12SPaolo Bonzini {
11591ba4b6a5SBenoît Canet     char *backing_filename = g_malloc0(PATH_MAX);
1160317fc44eSKevin Wolf     int ret = 0;
11618d24cce1SFam Zheng     BlockDriverState *backing_hd;
116234b5d2c6SMax Reitz     Error *local_err = NULL;
11639156df12SPaolo Bonzini 
11649156df12SPaolo Bonzini     if (bs->backing_hd != NULL) {
116531ca6d07SKevin Wolf         QDECREF(options);
11661ba4b6a5SBenoît Canet         goto free_exit;
11679156df12SPaolo Bonzini     }
11689156df12SPaolo Bonzini 
116931ca6d07SKevin Wolf     /* NULL means an empty set of options */
117031ca6d07SKevin Wolf     if (options == NULL) {
117131ca6d07SKevin Wolf         options = qdict_new();
117231ca6d07SKevin Wolf     }
117331ca6d07SKevin Wolf 
11749156df12SPaolo Bonzini     bs->open_flags &= ~BDRV_O_NO_BACKING;
11751cb6f506SKevin Wolf     if (qdict_haskey(options, "file.filename")) {
11761cb6f506SKevin Wolf         backing_filename[0] = '\0';
11771cb6f506SKevin Wolf     } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
117831ca6d07SKevin Wolf         QDECREF(options);
11791ba4b6a5SBenoît Canet         goto free_exit;
1180dbecebddSFam Zheng     } else {
11819f07429eSMax Reitz         bdrv_get_full_backing_filename(bs, backing_filename, PATH_MAX,
11829f07429eSMax Reitz                                        &local_err);
11839f07429eSMax Reitz         if (local_err) {
11849f07429eSMax Reitz             ret = -EINVAL;
11859f07429eSMax Reitz             error_propagate(errp, local_err);
11869f07429eSMax Reitz             QDECREF(options);
11879f07429eSMax Reitz             goto free_exit;
11889f07429eSMax Reitz         }
11899156df12SPaolo Bonzini     }
11909156df12SPaolo Bonzini 
11918ee79e70SKevin Wolf     if (!bs->drv || !bs->drv->supports_backing) {
11928ee79e70SKevin Wolf         ret = -EINVAL;
11938ee79e70SKevin Wolf         error_setg(errp, "Driver doesn't support backing files");
11948ee79e70SKevin Wolf         QDECREF(options);
11958ee79e70SKevin Wolf         goto free_exit;
11968ee79e70SKevin Wolf     }
11978ee79e70SKevin Wolf 
1198e4e9986bSMarkus Armbruster     backing_hd = bdrv_new();
11998d24cce1SFam Zheng 
1200c5f6e493SKevin Wolf     if (bs->backing_format[0] != '\0' && !qdict_haskey(options, "driver")) {
1201c5f6e493SKevin Wolf         qdict_put(options, "driver", qstring_from_str(bs->backing_format));
12029156df12SPaolo Bonzini     }
12039156df12SPaolo Bonzini 
1204f67503e5SMax Reitz     assert(bs->backing_hd == NULL);
1205f3930ed0SKevin Wolf     ret = bdrv_open_inherit(&backing_hd,
1206f3930ed0SKevin Wolf                             *backing_filename ? backing_filename : NULL,
1207ce343771SMax Reitz                             NULL, options, 0, bs, &child_backing, &local_err);
12089156df12SPaolo Bonzini     if (ret < 0) {
12098d24cce1SFam Zheng         bdrv_unref(backing_hd);
12108d24cce1SFam Zheng         backing_hd = NULL;
12119156df12SPaolo Bonzini         bs->open_flags |= BDRV_O_NO_BACKING;
1212b04b6b6eSFam Zheng         error_setg(errp, "Could not open backing file: %s",
1213b04b6b6eSFam Zheng                    error_get_pretty(local_err));
1214b04b6b6eSFam Zheng         error_free(local_err);
12151ba4b6a5SBenoît Canet         goto free_exit;
12169156df12SPaolo Bonzini     }
1217df581792SKevin Wolf 
12188d24cce1SFam Zheng     bdrv_set_backing_hd(bs, backing_hd);
1219d80ac658SPeter Feiner 
12201ba4b6a5SBenoît Canet free_exit:
12211ba4b6a5SBenoît Canet     g_free(backing_filename);
12221ba4b6a5SBenoît Canet     return ret;
12239156df12SPaolo Bonzini }
12249156df12SPaolo Bonzini 
1225b6ce07aaSKevin Wolf /*
1226da557aacSMax Reitz  * Opens a disk image whose options are given as BlockdevRef in another block
1227da557aacSMax Reitz  * device's options.
1228da557aacSMax Reitz  *
1229da557aacSMax Reitz  * If allow_none is true, no image will be opened if filename is false and no
1230b4b059f6SKevin Wolf  * BlockdevRef is given. NULL will be returned, but errp remains unset.
1231da557aacSMax Reitz  *
1232da557aacSMax Reitz  * bdrev_key specifies the key for the image's BlockdevRef in the options QDict.
1233da557aacSMax Reitz  * That QDict has to be flattened; therefore, if the BlockdevRef is a QDict
1234da557aacSMax Reitz  * itself, all options starting with "${bdref_key}." are considered part of the
1235da557aacSMax Reitz  * BlockdevRef.
1236da557aacSMax Reitz  *
1237da557aacSMax Reitz  * The BlockdevRef will be removed from the options QDict.
1238da557aacSMax Reitz  */
1239b4b059f6SKevin Wolf BdrvChild *bdrv_open_child(const char *filename,
1240f3930ed0SKevin Wolf                            QDict *options, const char *bdref_key,
1241b4b059f6SKevin Wolf                            BlockDriverState* parent,
1242b4b059f6SKevin Wolf                            const BdrvChildRole *child_role,
1243f7d9fd8cSMax Reitz                            bool allow_none, Error **errp)
1244da557aacSMax Reitz {
1245b4b059f6SKevin Wolf     BdrvChild *c = NULL;
1246b4b059f6SKevin Wolf     BlockDriverState *bs;
1247da557aacSMax Reitz     QDict *image_options;
1248da557aacSMax Reitz     int ret;
1249da557aacSMax Reitz     char *bdref_key_dot;
1250da557aacSMax Reitz     const char *reference;
1251da557aacSMax Reitz 
1252df581792SKevin Wolf     assert(child_role != NULL);
1253f67503e5SMax Reitz 
1254da557aacSMax Reitz     bdref_key_dot = g_strdup_printf("%s.", bdref_key);
1255da557aacSMax Reitz     qdict_extract_subqdict(options, &image_options, bdref_key_dot);
1256da557aacSMax Reitz     g_free(bdref_key_dot);
1257da557aacSMax Reitz 
1258da557aacSMax Reitz     reference = qdict_get_try_str(options, bdref_key);
1259da557aacSMax Reitz     if (!filename && !reference && !qdict_size(image_options)) {
1260b4b059f6SKevin Wolf         if (!allow_none) {
1261da557aacSMax Reitz             error_setg(errp, "A block device must be specified for \"%s\"",
1262da557aacSMax Reitz                        bdref_key);
1263da557aacSMax Reitz         }
1264b20e61e0SMarkus Armbruster         QDECREF(image_options);
1265da557aacSMax Reitz         goto done;
1266da557aacSMax Reitz     }
1267da557aacSMax Reitz 
1268b4b059f6SKevin Wolf     bs = NULL;
1269b4b059f6SKevin Wolf     ret = bdrv_open_inherit(&bs, filename, reference, image_options, 0,
1270ce343771SMax Reitz                             parent, child_role, errp);
1271df581792SKevin Wolf     if (ret < 0) {
1272df581792SKevin Wolf         goto done;
1273df581792SKevin Wolf     }
1274df581792SKevin Wolf 
1275b4b059f6SKevin Wolf     c = bdrv_attach_child(parent, bs, child_role);
1276da557aacSMax Reitz 
1277da557aacSMax Reitz done:
1278da557aacSMax Reitz     qdict_del(options, bdref_key);
1279b4b059f6SKevin Wolf     return c;
1280b4b059f6SKevin Wolf }
1281b4b059f6SKevin Wolf 
1282b4b059f6SKevin Wolf /*
1283b4b059f6SKevin Wolf  * This is a version of bdrv_open_child() that returns 0/-EINVAL instead of
1284b4b059f6SKevin Wolf  * a BdrvChild object.
1285b4b059f6SKevin Wolf  *
1286b4b059f6SKevin Wolf  * If allow_none is true, no image will be opened if filename is false and no
1287b4b059f6SKevin Wolf  * BlockdevRef is given. *pbs will remain unchanged and 0 will be returned.
1288b4b059f6SKevin Wolf  *
1289b4b059f6SKevin Wolf  * To conform with the behavior of bdrv_open(), *pbs has to be NULL.
1290b4b059f6SKevin Wolf  */
1291b4b059f6SKevin Wolf int bdrv_open_image(BlockDriverState **pbs, const char *filename,
1292b4b059f6SKevin Wolf                     QDict *options, const char *bdref_key,
1293b4b059f6SKevin Wolf                     BlockDriverState* parent, const BdrvChildRole *child_role,
1294b4b059f6SKevin Wolf                     bool allow_none, Error **errp)
1295b4b059f6SKevin Wolf {
1296b4b059f6SKevin Wolf     Error *local_err = NULL;
1297b4b059f6SKevin Wolf     BdrvChild *c;
1298b4b059f6SKevin Wolf 
1299b4b059f6SKevin Wolf     assert(pbs);
1300b4b059f6SKevin Wolf     assert(*pbs == NULL);
1301b4b059f6SKevin Wolf 
1302b4b059f6SKevin Wolf     c = bdrv_open_child(filename, options, bdref_key, parent, child_role,
1303b4b059f6SKevin Wolf                         allow_none, &local_err);
1304b4b059f6SKevin Wolf     if (local_err) {
1305b4b059f6SKevin Wolf         error_propagate(errp, local_err);
1306b4b059f6SKevin Wolf         return -EINVAL;
1307b4b059f6SKevin Wolf     }
1308b4b059f6SKevin Wolf 
1309b4b059f6SKevin Wolf     if (c != NULL) {
1310b4b059f6SKevin Wolf         *pbs = c->bs;
1311b4b059f6SKevin Wolf     }
1312b4b059f6SKevin Wolf 
1313b4b059f6SKevin Wolf     return 0;
1314da557aacSMax Reitz }
1315da557aacSMax Reitz 
13166b8aeca5SChen Gang int bdrv_append_temp_snapshot(BlockDriverState *bs, int flags, Error **errp)
1317b998875dSKevin Wolf {
1318b998875dSKevin Wolf     /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
13191ba4b6a5SBenoît Canet     char *tmp_filename = g_malloc0(PATH_MAX + 1);
1320b998875dSKevin Wolf     int64_t total_size;
132183d0521aSChunyan Liu     QemuOpts *opts = NULL;
1322b998875dSKevin Wolf     QDict *snapshot_options;
1323b998875dSKevin Wolf     BlockDriverState *bs_snapshot;
1324c2e0dbbfSFam Zheng     Error *local_err = NULL;
1325b998875dSKevin Wolf     int ret;
1326b998875dSKevin Wolf 
1327b998875dSKevin Wolf     /* if snapshot, we create a temporary backing file and open it
1328b998875dSKevin Wolf        instead of opening 'filename' directly */
1329b998875dSKevin Wolf 
1330b998875dSKevin Wolf     /* Get the required size from the image */
1331f187743aSKevin Wolf     total_size = bdrv_getlength(bs);
1332f187743aSKevin Wolf     if (total_size < 0) {
13336b8aeca5SChen Gang         ret = total_size;
1334f187743aSKevin Wolf         error_setg_errno(errp, -total_size, "Could not get image size");
13351ba4b6a5SBenoît Canet         goto out;
1336f187743aSKevin Wolf     }
1337b998875dSKevin Wolf 
1338b998875dSKevin Wolf     /* Create the temporary image */
13391ba4b6a5SBenoît Canet     ret = get_tmp_filename(tmp_filename, PATH_MAX + 1);
1340b998875dSKevin Wolf     if (ret < 0) {
1341b998875dSKevin Wolf         error_setg_errno(errp, -ret, "Could not get temporary filename");
13421ba4b6a5SBenoît Canet         goto out;
1343b998875dSKevin Wolf     }
1344b998875dSKevin Wolf 
1345ef810437SMax Reitz     opts = qemu_opts_create(bdrv_qcow2.create_opts, NULL, 0,
1346c282e1fdSChunyan Liu                             &error_abort);
134739101f25SMarkus Armbruster     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, total_size, &error_abort);
1348ef810437SMax Reitz     ret = bdrv_create(&bdrv_qcow2, tmp_filename, opts, &local_err);
134983d0521aSChunyan Liu     qemu_opts_del(opts);
1350b998875dSKevin Wolf     if (ret < 0) {
1351b998875dSKevin Wolf         error_setg_errno(errp, -ret, "Could not create temporary overlay "
1352b998875dSKevin Wolf                          "'%s': %s", tmp_filename,
1353b998875dSKevin Wolf                          error_get_pretty(local_err));
1354b998875dSKevin Wolf         error_free(local_err);
13551ba4b6a5SBenoît Canet         goto out;
1356b998875dSKevin Wolf     }
1357b998875dSKevin Wolf 
1358b998875dSKevin Wolf     /* Prepare a new options QDict for the temporary file */
1359b998875dSKevin Wolf     snapshot_options = qdict_new();
1360b998875dSKevin Wolf     qdict_put(snapshot_options, "file.driver",
1361b998875dSKevin Wolf               qstring_from_str("file"));
1362b998875dSKevin Wolf     qdict_put(snapshot_options, "file.filename",
1363b998875dSKevin Wolf               qstring_from_str(tmp_filename));
1364e6641719SMax Reitz     qdict_put(snapshot_options, "driver",
1365e6641719SMax Reitz               qstring_from_str("qcow2"));
1366b998875dSKevin Wolf 
1367e4e9986bSMarkus Armbruster     bs_snapshot = bdrv_new();
1368b998875dSKevin Wolf 
1369b998875dSKevin Wolf     ret = bdrv_open(&bs_snapshot, NULL, NULL, snapshot_options,
13706ebf9aa2SMax Reitz                     flags, &local_err);
1371b998875dSKevin Wolf     if (ret < 0) {
1372b998875dSKevin Wolf         error_propagate(errp, local_err);
13731ba4b6a5SBenoît Canet         goto out;
1374b998875dSKevin Wolf     }
1375b998875dSKevin Wolf 
1376b998875dSKevin Wolf     bdrv_append(bs_snapshot, bs);
13771ba4b6a5SBenoît Canet 
13781ba4b6a5SBenoît Canet out:
13791ba4b6a5SBenoît Canet     g_free(tmp_filename);
13806b8aeca5SChen Gang     return ret;
1381b998875dSKevin Wolf }
1382b998875dSKevin Wolf 
1383da557aacSMax Reitz /*
1384b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
1385de9c0cecSKevin Wolf  *
1386de9c0cecSKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
1387de9c0cecSKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
1388de9c0cecSKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
1389de9c0cecSKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_open.
1390f67503e5SMax Reitz  *
1391f67503e5SMax Reitz  * If *pbs is NULL, a new BDS will be created with a pointer to it stored there.
1392f67503e5SMax Reitz  * If it is not NULL, the referenced BDS will be reused.
1393ddf5636dSMax Reitz  *
1394ddf5636dSMax Reitz  * The reference parameter may be used to specify an existing block device which
1395ddf5636dSMax Reitz  * should be opened. If specified, neither options nor a filename may be given,
1396ddf5636dSMax Reitz  * nor can an existing BDS be reused (that is, *pbs has to be NULL).
1397b6ce07aaSKevin Wolf  */
1398f3930ed0SKevin Wolf static int bdrv_open_inherit(BlockDriverState **pbs, const char *filename,
1399ddf5636dSMax Reitz                              const char *reference, QDict *options, int flags,
1400f3930ed0SKevin Wolf                              BlockDriverState *parent,
1401ce343771SMax Reitz                              const BdrvChildRole *child_role, Error **errp)
1402ea2384d3Sbellard {
1403b6ce07aaSKevin Wolf     int ret;
1404*9a4f4c31SKevin Wolf     BdrvChild *file = NULL;
1405*9a4f4c31SKevin Wolf     BlockDriverState *bs;
1406ce343771SMax Reitz     BlockDriver *drv = NULL;
140774fe54f2SKevin Wolf     const char *drvname;
140834b5d2c6SMax Reitz     Error *local_err = NULL;
1409b1e6fc08SKevin Wolf     int snapshot_flags = 0;
141033e3963eSbellard 
1411f67503e5SMax Reitz     assert(pbs);
1412f3930ed0SKevin Wolf     assert(!child_role || !flags);
1413f3930ed0SKevin Wolf     assert(!child_role == !parent);
1414f67503e5SMax Reitz 
1415ddf5636dSMax Reitz     if (reference) {
1416ddf5636dSMax Reitz         bool options_non_empty = options ? qdict_size(options) : false;
1417ddf5636dSMax Reitz         QDECREF(options);
1418ddf5636dSMax Reitz 
1419ddf5636dSMax Reitz         if (*pbs) {
1420ddf5636dSMax Reitz             error_setg(errp, "Cannot reuse an existing BDS when referencing "
1421ddf5636dSMax Reitz                        "another block device");
1422ddf5636dSMax Reitz             return -EINVAL;
1423ddf5636dSMax Reitz         }
1424ddf5636dSMax Reitz 
1425ddf5636dSMax Reitz         if (filename || options_non_empty) {
1426ddf5636dSMax Reitz             error_setg(errp, "Cannot reference an existing block device with "
1427ddf5636dSMax Reitz                        "additional options or a new filename");
1428ddf5636dSMax Reitz             return -EINVAL;
1429ddf5636dSMax Reitz         }
1430ddf5636dSMax Reitz 
1431ddf5636dSMax Reitz         bs = bdrv_lookup_bs(reference, reference, errp);
1432ddf5636dSMax Reitz         if (!bs) {
1433ddf5636dSMax Reitz             return -ENODEV;
1434ddf5636dSMax Reitz         }
1435ddf5636dSMax Reitz         bdrv_ref(bs);
1436ddf5636dSMax Reitz         *pbs = bs;
1437ddf5636dSMax Reitz         return 0;
1438ddf5636dSMax Reitz     }
1439ddf5636dSMax Reitz 
1440f67503e5SMax Reitz     if (*pbs) {
1441f67503e5SMax Reitz         bs = *pbs;
1442f67503e5SMax Reitz     } else {
1443e4e9986bSMarkus Armbruster         bs = bdrv_new();
1444f67503e5SMax Reitz     }
1445f67503e5SMax Reitz 
1446de9c0cecSKevin Wolf     /* NULL means an empty set of options */
1447de9c0cecSKevin Wolf     if (options == NULL) {
1448de9c0cecSKevin Wolf         options = qdict_new();
1449de9c0cecSKevin Wolf     }
1450de9c0cecSKevin Wolf 
1451f3930ed0SKevin Wolf     if (child_role) {
1452bddcec37SKevin Wolf         bs->inherits_from = parent;
1453f3930ed0SKevin Wolf         flags = child_role->inherit_flags(parent->open_flags);
1454f3930ed0SKevin Wolf     }
1455f3930ed0SKevin Wolf 
1456053e1578SMax Reitz     ret = bdrv_fill_options(&options, &filename, &flags, &local_err);
1457462f5bcfSKevin Wolf     if (local_err) {
1458462f5bcfSKevin Wolf         goto fail;
1459462f5bcfSKevin Wolf     }
1460462f5bcfSKevin Wolf 
146176c591b0SKevin Wolf     /* Find the right image format driver */
146276c591b0SKevin Wolf     drvname = qdict_get_try_str(options, "driver");
146376c591b0SKevin Wolf     if (drvname) {
146476c591b0SKevin Wolf         drv = bdrv_find_format(drvname);
146576c591b0SKevin Wolf         qdict_del(options, "driver");
146676c591b0SKevin Wolf         if (!drv) {
146776c591b0SKevin Wolf             error_setg(errp, "Unknown driver: '%s'", drvname);
146876c591b0SKevin Wolf             ret = -EINVAL;
146976c591b0SKevin Wolf             goto fail;
147076c591b0SKevin Wolf         }
147176c591b0SKevin Wolf     }
147276c591b0SKevin Wolf 
147376c591b0SKevin Wolf     assert(drvname || !(flags & BDRV_O_PROTOCOL));
147476c591b0SKevin Wolf 
1475f3930ed0SKevin Wolf     bs->open_flags = flags;
1476de9c0cecSKevin Wolf     bs->options = options;
1477b6ad491aSKevin Wolf     options = qdict_clone_shallow(options);
1478de9c0cecSKevin Wolf 
1479f500a6d3SKevin Wolf     /* Open image file without format layer */
1480f4788adcSKevin Wolf     if ((flags & BDRV_O_PROTOCOL) == 0) {
1481be028adcSJeff Cody         if (flags & BDRV_O_RDWR) {
1482be028adcSJeff Cody             flags |= BDRV_O_ALLOW_RDWR;
1483be028adcSJeff Cody         }
1484b1e6fc08SKevin Wolf         if (flags & BDRV_O_SNAPSHOT) {
1485b1e6fc08SKevin Wolf             snapshot_flags = bdrv_temp_snapshot_flags(flags);
1486b1e6fc08SKevin Wolf             flags = bdrv_backing_flags(flags);
1487b1e6fc08SKevin Wolf         }
1488be028adcSJeff Cody 
1489f3930ed0SKevin Wolf         bs->open_flags = flags;
14901fdd6933SKevin Wolf 
1491*9a4f4c31SKevin Wolf         file = bdrv_open_child(filename, options, "file", bs,
14921fdd6933SKevin Wolf                                &child_file, true, &local_err);
14931fdd6933SKevin Wolf         if (local_err) {
14941fdd6933SKevin Wolf             ret = -EINVAL;
14958bfea15dSKevin Wolf             goto fail;
1496f500a6d3SKevin Wolf         }
1497f4788adcSKevin Wolf     }
1498f500a6d3SKevin Wolf 
149976c591b0SKevin Wolf     /* Image format probing */
150038f3ef57SKevin Wolf     bs->probed = !drv;
150176c591b0SKevin Wolf     if (!drv && file) {
1502*9a4f4c31SKevin Wolf         ret = find_image_format(file->bs, filename, &drv, &local_err);
150317b005f1SKevin Wolf         if (ret < 0) {
150417b005f1SKevin Wolf             goto fail;
150517b005f1SKevin Wolf         }
150676c591b0SKevin Wolf     } else if (!drv) {
15072a05cbe4SMax Reitz         error_setg(errp, "Must specify either driver or file");
15082a05cbe4SMax Reitz         ret = -EINVAL;
15098bfea15dSKevin Wolf         goto fail;
15102a05cbe4SMax Reitz     }
1511f500a6d3SKevin Wolf 
151253a29513SMax Reitz     /* BDRV_O_PROTOCOL must be set iff a protocol BDS is about to be created */
151353a29513SMax Reitz     assert(!!(flags & BDRV_O_PROTOCOL) == !!drv->bdrv_file_open);
151453a29513SMax Reitz     /* file must be NULL if a protocol BDS is about to be created
151553a29513SMax Reitz      * (the inverse results in an error message from bdrv_open_common()) */
151653a29513SMax Reitz     assert(!(flags & BDRV_O_PROTOCOL) || !file);
151753a29513SMax Reitz 
1518b6ce07aaSKevin Wolf     /* Open the image */
151934b5d2c6SMax Reitz     ret = bdrv_open_common(bs, file, options, flags, drv, &local_err);
1520b6ce07aaSKevin Wolf     if (ret < 0) {
15218bfea15dSKevin Wolf         goto fail;
15226987307cSChristoph Hellwig     }
15236987307cSChristoph Hellwig 
15242a05cbe4SMax Reitz     if (file && (bs->file != file)) {
1525*9a4f4c31SKevin Wolf         bdrv_unref_child(bs, file);
1526f500a6d3SKevin Wolf         file = NULL;
1527f500a6d3SKevin Wolf     }
1528f500a6d3SKevin Wolf 
1529b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
15309156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
153131ca6d07SKevin Wolf         QDict *backing_options;
153231ca6d07SKevin Wolf 
15335726d872SBenoît Canet         qdict_extract_subqdict(options, &backing_options, "backing.");
153434b5d2c6SMax Reitz         ret = bdrv_open_backing_file(bs, backing_options, &local_err);
1535b6ce07aaSKevin Wolf         if (ret < 0) {
1536b6ad491aSKevin Wolf             goto close_and_fail;
1537b6ce07aaSKevin Wolf         }
1538b6ce07aaSKevin Wolf     }
1539b6ce07aaSKevin Wolf 
154091af7014SMax Reitz     bdrv_refresh_filename(bs);
154191af7014SMax Reitz 
1542b998875dSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay. bs points to the
1543b998875dSKevin Wolf      * temporary snapshot afterwards. */
1544b1e6fc08SKevin Wolf     if (snapshot_flags) {
15456b8aeca5SChen Gang         ret = bdrv_append_temp_snapshot(bs, snapshot_flags, &local_err);
1546b998875dSKevin Wolf         if (local_err) {
1547b998875dSKevin Wolf             goto close_and_fail;
1548b998875dSKevin Wolf         }
1549b998875dSKevin Wolf     }
1550b998875dSKevin Wolf 
1551b6ad491aSKevin Wolf     /* Check if any unknown options were used */
15525acd9d81SMax Reitz     if (options && (qdict_size(options) != 0)) {
1553b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
15545acd9d81SMax Reitz         if (flags & BDRV_O_PROTOCOL) {
15555acd9d81SMax Reitz             error_setg(errp, "Block protocol '%s' doesn't support the option "
15565acd9d81SMax Reitz                        "'%s'", drv->format_name, entry->key);
15575acd9d81SMax Reitz         } else {
155834b5d2c6SMax Reitz             error_setg(errp, "Block format '%s' used by device '%s' doesn't "
15595acd9d81SMax Reitz                        "support the option '%s'", drv->format_name,
1560bfb197e0SMarkus Armbruster                        bdrv_get_device_name(bs), entry->key);
15615acd9d81SMax Reitz         }
1562b6ad491aSKevin Wolf 
1563b6ad491aSKevin Wolf         ret = -EINVAL;
1564b6ad491aSKevin Wolf         goto close_and_fail;
1565b6ad491aSKevin Wolf     }
1566b6ad491aSKevin Wolf 
1567b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
1568a7f53e26SMarkus Armbruster         if (bs->blk) {
1569a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
1570a7f53e26SMarkus Armbruster         }
1571c3adb58fSMarkus Armbruster     } else if (!runstate_check(RUN_STATE_PRELAUNCH)
1572c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_INMIGRATE)
1573c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_PAUSED)) { /* HACK */
1574c3adb58fSMarkus Armbruster         error_setg(errp,
1575c3adb58fSMarkus Armbruster                    "Guest must be stopped for opening of encrypted image");
1576c3adb58fSMarkus Armbruster         ret = -EBUSY;
1577c3adb58fSMarkus Armbruster         goto close_and_fail;
1578b6ce07aaSKevin Wolf     }
1579b6ce07aaSKevin Wolf 
1580c3adb58fSMarkus Armbruster     QDECREF(options);
1581f67503e5SMax Reitz     *pbs = bs;
1582b6ce07aaSKevin Wolf     return 0;
1583b6ce07aaSKevin Wolf 
15848bfea15dSKevin Wolf fail:
1585f500a6d3SKevin Wolf     if (file != NULL) {
1586*9a4f4c31SKevin Wolf         bdrv_unref_child(bs, file);
1587f500a6d3SKevin Wolf     }
1588de9c0cecSKevin Wolf     QDECREF(bs->options);
1589b6ad491aSKevin Wolf     QDECREF(options);
1590de9c0cecSKevin Wolf     bs->options = NULL;
1591f67503e5SMax Reitz     if (!*pbs) {
1592f67503e5SMax Reitz         /* If *pbs is NULL, a new BDS has been created in this function and
1593f67503e5SMax Reitz            needs to be freed now. Otherwise, it does not need to be closed,
1594f67503e5SMax Reitz            since it has not really been opened yet. */
1595f67503e5SMax Reitz         bdrv_unref(bs);
1596f67503e5SMax Reitz     }
159784d18f06SMarkus Armbruster     if (local_err) {
159834b5d2c6SMax Reitz         error_propagate(errp, local_err);
159934b5d2c6SMax Reitz     }
1600b6ad491aSKevin Wolf     return ret;
1601de9c0cecSKevin Wolf 
1602b6ad491aSKevin Wolf close_and_fail:
1603f67503e5SMax Reitz     /* See fail path, but now the BDS has to be always closed */
1604f67503e5SMax Reitz     if (*pbs) {
1605b6ad491aSKevin Wolf         bdrv_close(bs);
1606f67503e5SMax Reitz     } else {
1607f67503e5SMax Reitz         bdrv_unref(bs);
1608f67503e5SMax Reitz     }
1609b6ad491aSKevin Wolf     QDECREF(options);
161084d18f06SMarkus Armbruster     if (local_err) {
161134b5d2c6SMax Reitz         error_propagate(errp, local_err);
161234b5d2c6SMax Reitz     }
1613b6ce07aaSKevin Wolf     return ret;
1614b6ce07aaSKevin Wolf }
1615b6ce07aaSKevin Wolf 
1616f3930ed0SKevin Wolf int bdrv_open(BlockDriverState **pbs, const char *filename,
16176ebf9aa2SMax Reitz               const char *reference, QDict *options, int flags, Error **errp)
1618f3930ed0SKevin Wolf {
1619f3930ed0SKevin Wolf     return bdrv_open_inherit(pbs, filename, reference, options, flags, NULL,
1620ce343771SMax Reitz                              NULL, errp);
1621f3930ed0SKevin Wolf }
1622f3930ed0SKevin Wolf 
1623e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1624e971aa12SJeff Cody      bool prepared;
1625e971aa12SJeff Cody      BDRVReopenState state;
1626e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1627e971aa12SJeff Cody } BlockReopenQueueEntry;
1628e971aa12SJeff Cody 
1629e971aa12SJeff Cody /*
1630e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1631e971aa12SJeff Cody  * reopen of multiple devices.
1632e971aa12SJeff Cody  *
1633e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1634e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1635e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1636e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1637e971aa12SJeff Cody  * atomic 'set'.
1638e971aa12SJeff Cody  *
1639e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1640e971aa12SJeff Cody  *
16414d2cb092SKevin Wolf  * options contains the changed options for the associated bs
16424d2cb092SKevin Wolf  * (the BlockReopenQueue takes ownership)
16434d2cb092SKevin Wolf  *
1644e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1645e971aa12SJeff Cody  *
1646e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1647e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1648e971aa12SJeff Cody  *
1649e971aa12SJeff Cody  */
1650e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
16514d2cb092SKevin Wolf                                     BlockDriverState *bs,
16524d2cb092SKevin Wolf                                     QDict *options, int flags)
1653e971aa12SJeff Cody {
1654e971aa12SJeff Cody     assert(bs != NULL);
1655e971aa12SJeff Cody 
1656e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
165767251a31SKevin Wolf     BdrvChild *child;
16584d2cb092SKevin Wolf     QDict *old_options;
165967251a31SKevin Wolf 
1660e971aa12SJeff Cody     if (bs_queue == NULL) {
1661e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1662e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1663e971aa12SJeff Cody     }
1664e971aa12SJeff Cody 
16654d2cb092SKevin Wolf     if (!options) {
16664d2cb092SKevin Wolf         options = qdict_new();
16674d2cb092SKevin Wolf     }
16684d2cb092SKevin Wolf 
16694d2cb092SKevin Wolf     old_options = qdict_clone_shallow(bs->options);
16704d2cb092SKevin Wolf     qdict_join(options, old_options, false);
16714d2cb092SKevin Wolf     QDECREF(old_options);
16724d2cb092SKevin Wolf 
1673f1f25a2eSKevin Wolf     /* bdrv_open() masks this flag out */
1674f1f25a2eSKevin Wolf     flags &= ~BDRV_O_PROTOCOL;
1675f1f25a2eSKevin Wolf 
167667251a31SKevin Wolf     QLIST_FOREACH(child, &bs->children, next) {
167767251a31SKevin Wolf         int child_flags;
167867251a31SKevin Wolf 
167967251a31SKevin Wolf         if (child->bs->inherits_from != bs) {
168067251a31SKevin Wolf             continue;
168167251a31SKevin Wolf         }
168267251a31SKevin Wolf 
168367251a31SKevin Wolf         child_flags = child->role->inherit_flags(flags);
16844d2cb092SKevin Wolf         /* TODO Pass down child flags (backing.*, extents.*, ...) */
16854d2cb092SKevin Wolf         bdrv_reopen_queue(bs_queue, child->bs, NULL, child_flags);
1686e971aa12SJeff Cody     }
1687e971aa12SJeff Cody 
1688e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1689e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1690e971aa12SJeff Cody 
1691e971aa12SJeff Cody     bs_entry->state.bs = bs;
16924d2cb092SKevin Wolf     bs_entry->state.options = options;
1693e971aa12SJeff Cody     bs_entry->state.flags = flags;
1694e971aa12SJeff Cody 
1695e971aa12SJeff Cody     return bs_queue;
1696e971aa12SJeff Cody }
1697e971aa12SJeff Cody 
1698e971aa12SJeff Cody /*
1699e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1700e971aa12SJeff Cody  *
1701e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1702e971aa12SJeff Cody  * via bdrv_reopen_queue().
1703e971aa12SJeff Cody  *
1704e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1705e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1706e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1707e971aa12SJeff Cody  * data cleaned up.
1708e971aa12SJeff Cody  *
1709e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1710e971aa12SJeff Cody  * to all devices.
1711e971aa12SJeff Cody  *
1712e971aa12SJeff Cody  */
1713e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1714e971aa12SJeff Cody {
1715e971aa12SJeff Cody     int ret = -1;
1716e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1717e971aa12SJeff Cody     Error *local_err = NULL;
1718e971aa12SJeff Cody 
1719e971aa12SJeff Cody     assert(bs_queue != NULL);
1720e971aa12SJeff Cody 
1721e971aa12SJeff Cody     bdrv_drain_all();
1722e971aa12SJeff Cody 
1723e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1724e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1725e971aa12SJeff Cody             error_propagate(errp, local_err);
1726e971aa12SJeff Cody             goto cleanup;
1727e971aa12SJeff Cody         }
1728e971aa12SJeff Cody         bs_entry->prepared = true;
1729e971aa12SJeff Cody     }
1730e971aa12SJeff Cody 
1731e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1732e971aa12SJeff Cody      * changes
1733e971aa12SJeff Cody      */
1734e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1735e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1736e971aa12SJeff Cody     }
1737e971aa12SJeff Cody 
1738e971aa12SJeff Cody     ret = 0;
1739e971aa12SJeff Cody 
1740e971aa12SJeff Cody cleanup:
1741e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1742e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1743e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1744e971aa12SJeff Cody         }
17454d2cb092SKevin Wolf         QDECREF(bs_entry->state.options);
1746e971aa12SJeff Cody         g_free(bs_entry);
1747e971aa12SJeff Cody     }
1748e971aa12SJeff Cody     g_free(bs_queue);
1749e971aa12SJeff Cody     return ret;
1750e971aa12SJeff Cody }
1751e971aa12SJeff Cody 
1752e971aa12SJeff Cody 
1753e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1754e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1755e971aa12SJeff Cody {
1756e971aa12SJeff Cody     int ret = -1;
1757e971aa12SJeff Cody     Error *local_err = NULL;
17584d2cb092SKevin Wolf     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, NULL, bdrv_flags);
1759e971aa12SJeff Cody 
1760e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1761e971aa12SJeff Cody     if (local_err != NULL) {
1762e971aa12SJeff Cody         error_propagate(errp, local_err);
1763e971aa12SJeff Cody     }
1764e971aa12SJeff Cody     return ret;
1765e971aa12SJeff Cody }
1766e971aa12SJeff Cody 
1767e971aa12SJeff Cody 
1768e971aa12SJeff Cody /*
1769e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1770e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1771e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1772e971aa12SJeff Cody  *
1773e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1774e971aa12SJeff Cody  * flags are the new open flags
1775e971aa12SJeff Cody  * queue is the reopen queue
1776e971aa12SJeff Cody  *
1777e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1778e971aa12SJeff Cody  * as well.
1779e971aa12SJeff Cody  *
1780e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1781e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1782e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1783e971aa12SJeff Cody  *
1784e971aa12SJeff Cody  */
1785e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1786e971aa12SJeff Cody                         Error **errp)
1787e971aa12SJeff Cody {
1788e971aa12SJeff Cody     int ret = -1;
1789e971aa12SJeff Cody     Error *local_err = NULL;
1790e971aa12SJeff Cody     BlockDriver *drv;
1791e971aa12SJeff Cody 
1792e971aa12SJeff Cody     assert(reopen_state != NULL);
1793e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1794e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1795e971aa12SJeff Cody 
1796e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1797e971aa12SJeff Cody      * to r/w */
1798e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1799e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
180081e5f78aSAlberto Garcia         error_setg(errp, "Node '%s' is read only",
180181e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1802e971aa12SJeff Cody         goto error;
1803e971aa12SJeff Cody     }
1804e971aa12SJeff Cody 
1805e971aa12SJeff Cody 
1806e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1807e971aa12SJeff Cody     if (ret) {
1808e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1809e971aa12SJeff Cody                   strerror(-ret));
1810e971aa12SJeff Cody         goto error;
1811e971aa12SJeff Cody     }
1812e971aa12SJeff Cody 
1813e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1814e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1815e971aa12SJeff Cody         if (ret) {
1816e971aa12SJeff Cody             if (local_err != NULL) {
1817e971aa12SJeff Cody                 error_propagate(errp, local_err);
1818e971aa12SJeff Cody             } else {
1819d8b6895fSLuiz Capitulino                 error_setg(errp, "failed while preparing to reopen image '%s'",
1820e971aa12SJeff Cody                            reopen_state->bs->filename);
1821e971aa12SJeff Cody             }
1822e971aa12SJeff Cody             goto error;
1823e971aa12SJeff Cody         }
1824e971aa12SJeff Cody     } else {
1825e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1826e971aa12SJeff Cody          * handler for each supported drv. */
182781e5f78aSAlberto Garcia         error_setg(errp, "Block format '%s' used by node '%s' "
182881e5f78aSAlberto Garcia                    "does not support reopening files", drv->format_name,
182981e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1830e971aa12SJeff Cody         ret = -1;
1831e971aa12SJeff Cody         goto error;
1832e971aa12SJeff Cody     }
1833e971aa12SJeff Cody 
18344d2cb092SKevin Wolf     /* Options that are not handled are only okay if they are unchanged
18354d2cb092SKevin Wolf      * compared to the old state. It is expected that some options are only
18364d2cb092SKevin Wolf      * used for the initial open, but not reopen (e.g. filename) */
18374d2cb092SKevin Wolf     if (qdict_size(reopen_state->options)) {
18384d2cb092SKevin Wolf         const QDictEntry *entry = qdict_first(reopen_state->options);
18394d2cb092SKevin Wolf 
18404d2cb092SKevin Wolf         do {
18414d2cb092SKevin Wolf             QString *new_obj = qobject_to_qstring(entry->value);
18424d2cb092SKevin Wolf             const char *new = qstring_get_str(new_obj);
18434d2cb092SKevin Wolf             const char *old = qdict_get_try_str(reopen_state->bs->options,
18444d2cb092SKevin Wolf                                                 entry->key);
18454d2cb092SKevin Wolf 
18464d2cb092SKevin Wolf             if (!old || strcmp(new, old)) {
18474d2cb092SKevin Wolf                 error_setg(errp, "Cannot change the option '%s'", entry->key);
18484d2cb092SKevin Wolf                 ret = -EINVAL;
18494d2cb092SKevin Wolf                 goto error;
18504d2cb092SKevin Wolf             }
18514d2cb092SKevin Wolf         } while ((entry = qdict_next(reopen_state->options, entry)));
18524d2cb092SKevin Wolf     }
18534d2cb092SKevin Wolf 
1854e971aa12SJeff Cody     ret = 0;
1855e971aa12SJeff Cody 
1856e971aa12SJeff Cody error:
1857e971aa12SJeff Cody     return ret;
1858e971aa12SJeff Cody }
1859e971aa12SJeff Cody 
1860e971aa12SJeff Cody /*
1861e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1862e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1863e971aa12SJeff Cody  * the active BlockDriverState contents.
1864e971aa12SJeff Cody  */
1865e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1866e971aa12SJeff Cody {
1867e971aa12SJeff Cody     BlockDriver *drv;
1868e971aa12SJeff Cody 
1869e971aa12SJeff Cody     assert(reopen_state != NULL);
1870e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1871e971aa12SJeff Cody     assert(drv != NULL);
1872e971aa12SJeff Cody 
1873e971aa12SJeff Cody     /* If there are any driver level actions to take */
1874e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1875e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1876e971aa12SJeff Cody     }
1877e971aa12SJeff Cody 
1878e971aa12SJeff Cody     /* set BDS specific flags now */
1879e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1880e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1881e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1882e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1883355ef4acSKevin Wolf 
18843baca891SKevin Wolf     bdrv_refresh_limits(reopen_state->bs, NULL);
1885e971aa12SJeff Cody }
1886e971aa12SJeff Cody 
1887e971aa12SJeff Cody /*
1888e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1889e971aa12SJeff Cody  * reopen_state
1890e971aa12SJeff Cody  */
1891e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1892e971aa12SJeff Cody {
1893e971aa12SJeff Cody     BlockDriver *drv;
1894e971aa12SJeff Cody 
1895e971aa12SJeff Cody     assert(reopen_state != NULL);
1896e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1897e971aa12SJeff Cody     assert(drv != NULL);
1898e971aa12SJeff Cody 
1899e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1900e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1901e971aa12SJeff Cody     }
1902e971aa12SJeff Cody }
1903e971aa12SJeff Cody 
1904e971aa12SJeff Cody 
1905fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1906fc01f7e7Sbellard {
190733384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
190833384421SMax Reitz 
19093e914655SPaolo Bonzini     if (bs->job) {
19103e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
19113e914655SPaolo Bonzini     }
191299b7e775SAlberto Garcia 
191399b7e775SAlberto Garcia     /* Disable I/O limits and drain all pending throttled requests */
191499b7e775SAlberto Garcia     if (bs->io_limits_enabled) {
191599b7e775SAlberto Garcia         bdrv_io_limits_disable(bs);
191699b7e775SAlberto Garcia     }
191799b7e775SAlberto Garcia 
191853ec73e2SFam Zheng     bdrv_drain(bs); /* complete I/O */
191958fda173SStefan Hajnoczi     bdrv_flush(bs);
192053ec73e2SFam Zheng     bdrv_drain(bs); /* in case flush left pending I/O */
1921d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
19227094f12fSKevin Wolf 
19233cbc002cSPaolo Bonzini     if (bs->drv) {
19246e93e7c4SKevin Wolf         BdrvChild *child, *next;
19256e93e7c4SKevin Wolf 
19269a7dedbcSKevin Wolf         bs->drv->bdrv_close(bs);
1927*9a4f4c31SKevin Wolf         bs->drv = NULL;
19289a7dedbcSKevin Wolf 
19299a7dedbcSKevin Wolf         if (bs->backing_hd) {
19309a7dedbcSKevin Wolf             BlockDriverState *backing_hd = bs->backing_hd;
19319a7dedbcSKevin Wolf             bdrv_set_backing_hd(bs, NULL);
19329a7dedbcSKevin Wolf             bdrv_unref(backing_hd);
19339a7dedbcSKevin Wolf         }
19349a7dedbcSKevin Wolf 
1935*9a4f4c31SKevin Wolf         if (bs->file != NULL) {
1936*9a4f4c31SKevin Wolf             bdrv_unref_child(bs, bs->file);
1937*9a4f4c31SKevin Wolf             bs->file = NULL;
1938*9a4f4c31SKevin Wolf         }
1939*9a4f4c31SKevin Wolf 
19406e93e7c4SKevin Wolf         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
194133a60407SKevin Wolf             /* TODO Remove bdrv_unref() from drivers' close function and use
194233a60407SKevin Wolf              * bdrv_unref_child() here */
1943bddcec37SKevin Wolf             if (child->bs->inherits_from == bs) {
1944bddcec37SKevin Wolf                 child->bs->inherits_from = NULL;
1945bddcec37SKevin Wolf             }
194633a60407SKevin Wolf             bdrv_detach_child(child);
19476e93e7c4SKevin Wolf         }
19486e93e7c4SKevin Wolf 
19497267c094SAnthony Liguori         g_free(bs->opaque);
1950ea2384d3Sbellard         bs->opaque = NULL;
195153fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1952a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1953a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
19546405875cSPaolo Bonzini         bs->total_sectors = 0;
19556405875cSPaolo Bonzini         bs->encrypted = 0;
19566405875cSPaolo Bonzini         bs->valid_key = 0;
19576405875cSPaolo Bonzini         bs->sg = 0;
19580d51b4deSAsias He         bs->zero_beyond_eof = false;
1959de9c0cecSKevin Wolf         QDECREF(bs->options);
1960de9c0cecSKevin Wolf         bs->options = NULL;
196191af7014SMax Reitz         QDECREF(bs->full_open_options);
196291af7014SMax Reitz         bs->full_open_options = NULL;
19639ca11154SPavel Hrdina     }
196466f82ceeSKevin Wolf 
1965a7f53e26SMarkus Armbruster     if (bs->blk) {
1966a7f53e26SMarkus Armbruster         blk_dev_change_media_cb(bs->blk, false);
1967a7f53e26SMarkus Armbruster     }
196898f90dbaSZhi Yong Wu 
196933384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
197033384421SMax Reitz         g_free(ban);
197133384421SMax Reitz     }
197233384421SMax Reitz     QLIST_INIT(&bs->aio_notifiers);
1973b338082bSbellard }
1974b338082bSbellard 
19752bc93fedSMORITA Kazutaka void bdrv_close_all(void)
19762bc93fedSMORITA Kazutaka {
19772bc93fedSMORITA Kazutaka     BlockDriverState *bs;
19782bc93fedSMORITA Kazutaka 
1979dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
1980ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
1981ed78cda3SStefan Hajnoczi 
1982ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
19832bc93fedSMORITA Kazutaka         bdrv_close(bs);
1984ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
19852bc93fedSMORITA Kazutaka     }
19862bc93fedSMORITA Kazutaka }
19872bc93fedSMORITA Kazutaka 
1988dc364f4cSBenoît Canet /* make a BlockDriverState anonymous by removing from bdrv_state and
1989dc364f4cSBenoît Canet  * graph_bdrv_state list.
1990d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1991d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1992d22b2f41SRyan Harper {
1993bfb197e0SMarkus Armbruster     /*
1994bfb197e0SMarkus Armbruster      * Take care to remove bs from bdrv_states only when it's actually
1995bfb197e0SMarkus Armbruster      * in it.  Note that bs->device_list.tqe_prev is initially null,
1996bfb197e0SMarkus Armbruster      * and gets set to non-null by QTAILQ_INSERT_TAIL().  Establish
1997bfb197e0SMarkus Armbruster      * the useful invariant "bs in bdrv_states iff bs->tqe_prev" by
1998bfb197e0SMarkus Armbruster      * resetting it to null on remove.
1999bfb197e0SMarkus Armbruster      */
2000bfb197e0SMarkus Armbruster     if (bs->device_list.tqe_prev) {
2001dc364f4cSBenoît Canet         QTAILQ_REMOVE(&bdrv_states, bs, device_list);
2002bfb197e0SMarkus Armbruster         bs->device_list.tqe_prev = NULL;
2003d22b2f41SRyan Harper     }
2004dc364f4cSBenoît Canet     if (bs->node_name[0] != '\0') {
2005dc364f4cSBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs, node_list);
2006dc364f4cSBenoît Canet     }
2007dc364f4cSBenoît Canet     bs->node_name[0] = '\0';
2008d22b2f41SRyan Harper }
2009d22b2f41SRyan Harper 
2010e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
2011e023b2e2SPaolo Bonzini {
2012e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
2013e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
2014e023b2e2SPaolo Bonzini     }
2015e023b2e2SPaolo Bonzini }
2016e023b2e2SPaolo Bonzini 
20174ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
20184ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
20194ddc07caSPaolo Bonzini {
20204ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
20214ddc07caSPaolo Bonzini 
20224ddc07caSPaolo Bonzini     /* dev info */
20231b7fd729SPaolo Bonzini     bs_dest->guest_block_size   = bs_src->guest_block_size;
20244ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
20254ddc07caSPaolo Bonzini 
20264ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
20274ddc07caSPaolo Bonzini 
2028cc0681c4SBenoît Canet     /* i/o throttled req */
202976f4afb4SAlberto Garcia     bs_dest->throttle_state     = bs_src->throttle_state,
203076f4afb4SAlberto Garcia     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
203176f4afb4SAlberto Garcia     bs_dest->pending_reqs[0]    = bs_src->pending_reqs[0];
203276f4afb4SAlberto Garcia     bs_dest->pending_reqs[1]    = bs_src->pending_reqs[1];
203376f4afb4SAlberto Garcia     bs_dest->throttled_reqs[0]  = bs_src->throttled_reqs[0];
203476f4afb4SAlberto Garcia     bs_dest->throttled_reqs[1]  = bs_src->throttled_reqs[1];
203576f4afb4SAlberto Garcia     memcpy(&bs_dest->round_robin,
203676f4afb4SAlberto Garcia            &bs_src->round_robin,
203776f4afb4SAlberto Garcia            sizeof(bs_dest->round_robin));
20380e5b0a2dSBenoît Canet     memcpy(&bs_dest->throttle_timers,
20390e5b0a2dSBenoît Canet            &bs_src->throttle_timers,
20400e5b0a2dSBenoît Canet            sizeof(ThrottleTimers));
20414ddc07caSPaolo Bonzini 
20424ddc07caSPaolo Bonzini     /* r/w error */
20434ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
20444ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
20454ddc07caSPaolo Bonzini 
20464ddc07caSPaolo Bonzini     /* i/o status */
20474ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
20484ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
20494ddc07caSPaolo Bonzini 
20504ddc07caSPaolo Bonzini     /* dirty bitmap */
2051e4654d2dSFam Zheng     bs_dest->dirty_bitmaps      = bs_src->dirty_bitmaps;
20524ddc07caSPaolo Bonzini 
20539fcb0251SFam Zheng     /* reference count */
20549fcb0251SFam Zheng     bs_dest->refcnt             = bs_src->refcnt;
20559fcb0251SFam Zheng 
20564ddc07caSPaolo Bonzini     /* job */
20574ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
20584ddc07caSPaolo Bonzini 
20594ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
2060dc364f4cSBenoît Canet     bs_dest->device_list = bs_src->device_list;
20617e7d56d9SMarkus Armbruster     bs_dest->blk = bs_src->blk;
20627e7d56d9SMarkus Armbruster 
2063fbe40ff7SFam Zheng     memcpy(bs_dest->op_blockers, bs_src->op_blockers,
2064fbe40ff7SFam Zheng            sizeof(bs_dest->op_blockers));
20654ddc07caSPaolo Bonzini }
20664ddc07caSPaolo Bonzini 
20674ddc07caSPaolo Bonzini /*
20684ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
20694ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
20704ddc07caSPaolo Bonzini  * actually attached to a device.
20714ddc07caSPaolo Bonzini  *
20724ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
20734ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
20744ddc07caSPaolo Bonzini  *
2075bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
20764ddc07caSPaolo Bonzini  *
20774ddc07caSPaolo Bonzini  * This function does not create any image files.
20784ddc07caSPaolo Bonzini  */
20794ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
20804ddc07caSPaolo Bonzini {
20814ddc07caSPaolo Bonzini     BlockDriverState tmp;
2082bddcec37SKevin Wolf     BdrvChild *child;
20834ddc07caSPaolo Bonzini 
20846ee4ce1eSKevin Wolf     bdrv_drain(bs_new);
20856ee4ce1eSKevin Wolf     bdrv_drain(bs_old);
20864ddc07caSPaolo Bonzini 
208790ce8a06SBenoît Canet     /* The code needs to swap the node_name but simply swapping node_list won't
208890ce8a06SBenoît Canet      * work so first remove the nodes from the graph list, do the swap then
208990ce8a06SBenoît Canet      * insert them back if needed.
209090ce8a06SBenoît Canet      */
209190ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
209290ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_new, node_list);
209390ce8a06SBenoît Canet     }
209490ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
209590ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_old, node_list);
209690ce8a06SBenoît Canet     }
209790ce8a06SBenoît Canet 
2098db628338SAlberto Garcia     /* If the BlockDriverState is part of a throttling group acquire
2099db628338SAlberto Garcia      * its lock since we're going to mess with the protected fields.
2100db628338SAlberto Garcia      * Otherwise there's no need to worry since no one else can touch
2101db628338SAlberto Garcia      * them. */
2102db628338SAlberto Garcia     if (bs_old->throttle_state) {
2103db628338SAlberto Garcia         throttle_group_lock(bs_old);
2104db628338SAlberto Garcia     }
2105db628338SAlberto Garcia 
2106bfb197e0SMarkus Armbruster     /* bs_new must be unattached and shouldn't have anything fancy enabled */
21077e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
2108e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs_new->dirty_bitmaps));
21094ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21104ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2111db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21120e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21134ddc07caSPaolo Bonzini 
21144ddc07caSPaolo Bonzini     tmp = *bs_new;
21154ddc07caSPaolo Bonzini     *bs_new = *bs_old;
21164ddc07caSPaolo Bonzini     *bs_old = tmp;
21174ddc07caSPaolo Bonzini 
21184ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
21194ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
21204ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
21214ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
21224ddc07caSPaolo Bonzini 
2123bfb197e0SMarkus Armbruster     /* bs_new must remain unattached */
21247e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
21254ddc07caSPaolo Bonzini 
21264ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
21274ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21284ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2129db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21300e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21314ddc07caSPaolo Bonzini 
2132db628338SAlberto Garcia     /* Release the ThrottleGroup lock */
2133db628338SAlberto Garcia     if (bs_old->throttle_state) {
2134db628338SAlberto Garcia         throttle_group_unlock(bs_old);
2135db628338SAlberto Garcia     }
2136db628338SAlberto Garcia 
213790ce8a06SBenoît Canet     /* insert the nodes back into the graph node list if needed */
213890ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
213990ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_new, node_list);
214090ce8a06SBenoît Canet     }
214190ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
214290ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_old, node_list);
214390ce8a06SBenoît Canet     }
214490ce8a06SBenoît Canet 
21456e93e7c4SKevin Wolf     /*
21466e93e7c4SKevin Wolf      * Update lh_first.le_prev for non-empty lists.
21476e93e7c4SKevin Wolf      *
21486e93e7c4SKevin Wolf      * The head of the op blocker list doesn't change because it is moved back
21496e93e7c4SKevin Wolf      * in bdrv_move_feature_fields().
21506e93e7c4SKevin Wolf      */
21516ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_old->tracked_requests));
21526ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_new->tracked_requests));
21536ee4ce1eSKevin Wolf 
21546e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_new->children, next);
21556e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_old->children, next);
21566e93e7c4SKevin Wolf 
2157bddcec37SKevin Wolf     /* Update references in bs->opaque and children */
2158bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_old->children, next) {
2159bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_new) {
2160bddcec37SKevin Wolf             child->bs->inherits_from = bs_old;
2161bddcec37SKevin Wolf         }
2162bddcec37SKevin Wolf     }
2163bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_new->children, next) {
2164bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_old) {
2165bddcec37SKevin Wolf             child->bs->inherits_from = bs_new;
2166bddcec37SKevin Wolf         }
2167bddcec37SKevin Wolf     }
2168bddcec37SKevin Wolf 
21694ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
21704ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
21714ddc07caSPaolo Bonzini }
21724ddc07caSPaolo Bonzini 
21738802d1fdSJeff Cody /*
21748802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
21758802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
21768802d1fdSJeff Cody  *
21778802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
21788802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
21798802d1fdSJeff Cody  *
2180bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
2181f6801b83SJeff Cody  *
21828802d1fdSJeff Cody  * This function does not create any image files.
21838802d1fdSJeff Cody  */
21848802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
21858802d1fdSJeff Cody {
21864ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
21878802d1fdSJeff Cody 
21888802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
21898802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
21908d24cce1SFam Zheng     bdrv_set_backing_hd(bs_top, bs_new);
21918802d1fdSJeff Cody }
21928802d1fdSJeff Cody 
21934f6fd349SFam Zheng static void bdrv_delete(BlockDriverState *bs)
2194b338082bSbellard {
21953e914655SPaolo Bonzini     assert(!bs->job);
21963718d8abSFam Zheng     assert(bdrv_op_blocker_is_empty(bs));
21974f6fd349SFam Zheng     assert(!bs->refcnt);
2198e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
219918846deeSMarkus Armbruster 
2200e1b5c52eSStefan Hajnoczi     bdrv_close(bs);
2201e1b5c52eSStefan Hajnoczi 
22021b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
2203d22b2f41SRyan Harper     bdrv_make_anon(bs);
220434c6f050Saurel32 
22057267c094SAnthony Liguori     g_free(bs);
2206fc01f7e7Sbellard }
2207fc01f7e7Sbellard 
2208e97fc193Saliguori /*
2209e97fc193Saliguori  * Run consistency checks on an image
2210e97fc193Saliguori  *
2211e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
2212a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
2213e076f338SKevin Wolf  * check are stored in res.
2214e97fc193Saliguori  */
22154534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
2216e97fc193Saliguori {
2217908bcd54SMax Reitz     if (bs->drv == NULL) {
2218908bcd54SMax Reitz         return -ENOMEDIUM;
2219908bcd54SMax Reitz     }
2220e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
2221e97fc193Saliguori         return -ENOTSUP;
2222e97fc193Saliguori     }
2223e97fc193Saliguori 
2224e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
22254534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
2226e97fc193Saliguori }
2227e97fc193Saliguori 
22288a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
22298a426614SKevin Wolf 
223033e3963eSbellard /* commit COW file into the raw image */
223133e3963eSbellard int bdrv_commit(BlockDriverState *bs)
223233e3963eSbellard {
223319cb3738Sbellard     BlockDriver *drv = bs->drv;
223472706ea4SJeff Cody     int64_t sector, total_sectors, length, backing_length;
22358a426614SKevin Wolf     int n, ro, open_flags;
22360bce597dSJeff Cody     int ret = 0;
223772706ea4SJeff Cody     uint8_t *buf = NULL;
223833e3963eSbellard 
223919cb3738Sbellard     if (!drv)
224019cb3738Sbellard         return -ENOMEDIUM;
224133e3963eSbellard 
22424dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
22434dca4b63SNaphtali Sprei         return -ENOTSUP;
22444dca4b63SNaphtali Sprei     }
22454dca4b63SNaphtali Sprei 
2246bb00021dSFam Zheng     if (bdrv_op_is_blocked(bs, BLOCK_OP_TYPE_COMMIT_SOURCE, NULL) ||
2247bb00021dSFam Zheng         bdrv_op_is_blocked(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET, NULL)) {
22482d3735d3SStefan Hajnoczi         return -EBUSY;
22492d3735d3SStefan Hajnoczi     }
22502d3735d3SStefan Hajnoczi 
22514dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
22524dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
22534dca4b63SNaphtali Sprei 
22544dca4b63SNaphtali Sprei     if (ro) {
22550bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
22560bce597dSJeff Cody             return -EACCES;
22574dca4b63SNaphtali Sprei         }
2258ea2384d3Sbellard     }
2259ea2384d3Sbellard 
226072706ea4SJeff Cody     length = bdrv_getlength(bs);
226172706ea4SJeff Cody     if (length < 0) {
226272706ea4SJeff Cody         ret = length;
226372706ea4SJeff Cody         goto ro_cleanup;
226472706ea4SJeff Cody     }
226572706ea4SJeff Cody 
226672706ea4SJeff Cody     backing_length = bdrv_getlength(bs->backing_hd);
226772706ea4SJeff Cody     if (backing_length < 0) {
226872706ea4SJeff Cody         ret = backing_length;
226972706ea4SJeff Cody         goto ro_cleanup;
227072706ea4SJeff Cody     }
227172706ea4SJeff Cody 
227272706ea4SJeff Cody     /* If our top snapshot is larger than the backing file image,
227372706ea4SJeff Cody      * grow the backing file image if possible.  If not possible,
227472706ea4SJeff Cody      * we must return an error */
227572706ea4SJeff Cody     if (length > backing_length) {
227672706ea4SJeff Cody         ret = bdrv_truncate(bs->backing_hd, length);
227772706ea4SJeff Cody         if (ret < 0) {
227872706ea4SJeff Cody             goto ro_cleanup;
227972706ea4SJeff Cody         }
228072706ea4SJeff Cody     }
228172706ea4SJeff Cody 
228272706ea4SJeff Cody     total_sectors = length >> BDRV_SECTOR_BITS;
2283857d4f46SKevin Wolf 
2284857d4f46SKevin Wolf     /* qemu_try_blockalign() for bs will choose an alignment that works for
2285857d4f46SKevin Wolf      * bs->backing_hd as well, so no need to compare the alignment manually. */
2286857d4f46SKevin Wolf     buf = qemu_try_blockalign(bs, COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
2287857d4f46SKevin Wolf     if (buf == NULL) {
2288857d4f46SKevin Wolf         ret = -ENOMEM;
2289857d4f46SKevin Wolf         goto ro_cleanup;
2290857d4f46SKevin Wolf     }
22918a426614SKevin Wolf 
22928a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
2293d663640cSPaolo Bonzini         ret = bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n);
2294d663640cSPaolo Bonzini         if (ret < 0) {
2295d663640cSPaolo Bonzini             goto ro_cleanup;
2296d663640cSPaolo Bonzini         }
2297d663640cSPaolo Bonzini         if (ret) {
2298dabfa6ccSKevin Wolf             ret = bdrv_read(bs, sector, buf, n);
2299dabfa6ccSKevin Wolf             if (ret < 0) {
23004dca4b63SNaphtali Sprei                 goto ro_cleanup;
230133e3963eSbellard             }
230233e3963eSbellard 
2303dabfa6ccSKevin Wolf             ret = bdrv_write(bs->backing_hd, sector, buf, n);
2304dabfa6ccSKevin Wolf             if (ret < 0) {
23054dca4b63SNaphtali Sprei                 goto ro_cleanup;
230633e3963eSbellard             }
230733e3963eSbellard         }
230833e3963eSbellard     }
230995389c86Sbellard 
23101d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
23111d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
2312dabfa6ccSKevin Wolf         if (ret < 0) {
2313dabfa6ccSKevin Wolf             goto ro_cleanup;
2314dabfa6ccSKevin Wolf         }
23151d44952fSChristoph Hellwig         bdrv_flush(bs);
23161d44952fSChristoph Hellwig     }
231795389c86Sbellard 
23183f5075aeSChristoph Hellwig     /*
23193f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
23203f5075aeSChristoph Hellwig      * stable on disk.
23213f5075aeSChristoph Hellwig      */
2322dabfa6ccSKevin Wolf     if (bs->backing_hd) {
23233f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
2324dabfa6ccSKevin Wolf     }
23254dca4b63SNaphtali Sprei 
2326dabfa6ccSKevin Wolf     ret = 0;
23274dca4b63SNaphtali Sprei ro_cleanup:
2328857d4f46SKevin Wolf     qemu_vfree(buf);
23294dca4b63SNaphtali Sprei 
23304dca4b63SNaphtali Sprei     if (ro) {
23310bce597dSJeff Cody         /* ignoring error return here */
23320bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
23334dca4b63SNaphtali Sprei     }
23344dca4b63SNaphtali Sprei 
23351d44952fSChristoph Hellwig     return ret;
233633e3963eSbellard }
233733e3963eSbellard 
2338e8877497SStefan Hajnoczi int bdrv_commit_all(void)
23396ab4b5abSMarkus Armbruster {
23406ab4b5abSMarkus Armbruster     BlockDriverState *bs;
23416ab4b5abSMarkus Armbruster 
2342dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
2343ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
2344ed78cda3SStefan Hajnoczi 
2345ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
2346272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
2347e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
2348e8877497SStefan Hajnoczi             if (ret < 0) {
2349ed78cda3SStefan Hajnoczi                 aio_context_release(aio_context);
2350e8877497SStefan Hajnoczi                 return ret;
23516ab4b5abSMarkus Armbruster             }
23526ab4b5abSMarkus Armbruster         }
2353ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
2354272d2d8eSJeff Cody     }
2355e8877497SStefan Hajnoczi     return 0;
2356e8877497SStefan Hajnoczi }
23576ab4b5abSMarkus Armbruster 
2358756e6736SKevin Wolf /*
2359756e6736SKevin Wolf  * Return values:
2360756e6736SKevin Wolf  * 0        - success
2361756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
2362756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
2363756e6736SKevin Wolf  *            image file header
2364756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
2365756e6736SKevin Wolf  */
2366756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
2367756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
2368756e6736SKevin Wolf {
2369756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
2370469ef350SPaolo Bonzini     int ret;
2371756e6736SKevin Wolf 
23725f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
23735f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
23745f377794SPaolo Bonzini         return -EINVAL;
23755f377794SPaolo Bonzini     }
23765f377794SPaolo Bonzini 
2377756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
2378469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
2379756e6736SKevin Wolf     } else {
2380469ef350SPaolo Bonzini         ret = -ENOTSUP;
2381756e6736SKevin Wolf     }
2382469ef350SPaolo Bonzini 
2383469ef350SPaolo Bonzini     if (ret == 0) {
2384469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
2385469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
2386469ef350SPaolo Bonzini     }
2387469ef350SPaolo Bonzini     return ret;
2388756e6736SKevin Wolf }
2389756e6736SKevin Wolf 
23906ebdcee2SJeff Cody /*
23916ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
23926ebdcee2SJeff Cody  *
23936ebdcee2SJeff Cody  * active is the current topmost image.
23946ebdcee2SJeff Cody  *
23956ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
23966ebdcee2SJeff Cody  * or if active == bs.
23974caf0fcdSJeff Cody  *
23984caf0fcdSJeff Cody  * Returns the bottommost base image if bs == NULL.
23996ebdcee2SJeff Cody  */
24006ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
24016ebdcee2SJeff Cody                                     BlockDriverState *bs)
24026ebdcee2SJeff Cody {
24034caf0fcdSJeff Cody     while (active && bs != active->backing_hd) {
24044caf0fcdSJeff Cody         active = active->backing_hd;
24056ebdcee2SJeff Cody     }
24066ebdcee2SJeff Cody 
24074caf0fcdSJeff Cody     return active;
24086ebdcee2SJeff Cody }
24096ebdcee2SJeff Cody 
24104caf0fcdSJeff Cody /* Given a BDS, searches for the base layer. */
24114caf0fcdSJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
24124caf0fcdSJeff Cody {
24134caf0fcdSJeff Cody     return bdrv_find_overlay(bs, NULL);
24146ebdcee2SJeff Cody }
24156ebdcee2SJeff Cody 
24166ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
24176ebdcee2SJeff Cody     BlockDriverState *bs;
24186ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
24196ebdcee2SJeff Cody } BlkIntermediateStates;
24206ebdcee2SJeff Cody 
24216ebdcee2SJeff Cody 
24226ebdcee2SJeff Cody /*
24236ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
24246ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
24256ebdcee2SJeff Cody  *
24266ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
24276ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
24286ebdcee2SJeff Cody  *
24296ebdcee2SJeff Cody  * E.g., this will convert the following chain:
24306ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
24316ebdcee2SJeff Cody  *
24326ebdcee2SJeff Cody  * to
24336ebdcee2SJeff Cody  *
24346ebdcee2SJeff Cody  * bottom <- base <- active
24356ebdcee2SJeff Cody  *
24366ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
24376ebdcee2SJeff Cody  *
24386ebdcee2SJeff Cody  * base <- intermediate <- top <- active
24396ebdcee2SJeff Cody  *
24406ebdcee2SJeff Cody  * to
24416ebdcee2SJeff Cody  *
24426ebdcee2SJeff Cody  * base <- active
24436ebdcee2SJeff Cody  *
244454e26900SJeff Cody  * If backing_file_str is non-NULL, it will be used when modifying top's
244554e26900SJeff Cody  * overlay image metadata.
244654e26900SJeff Cody  *
24476ebdcee2SJeff Cody  * Error conditions:
24486ebdcee2SJeff Cody  *  if active == top, that is considered an error
24496ebdcee2SJeff Cody  *
24506ebdcee2SJeff Cody  */
24516ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
245254e26900SJeff Cody                            BlockDriverState *base, const char *backing_file_str)
24536ebdcee2SJeff Cody {
24546ebdcee2SJeff Cody     BlockDriverState *intermediate;
24556ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
24566ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
24576ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
24586ebdcee2SJeff Cody     int ret = -EIO;
24596ebdcee2SJeff Cody 
24606ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
24616ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
24626ebdcee2SJeff Cody 
24636ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
24646ebdcee2SJeff Cody         goto exit;
24656ebdcee2SJeff Cody     }
24666ebdcee2SJeff Cody 
24676ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
24686ebdcee2SJeff Cody 
24696ebdcee2SJeff Cody     if (new_top_bs == NULL) {
24706ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
24716ebdcee2SJeff Cody         goto exit;
24726ebdcee2SJeff Cody     }
24736ebdcee2SJeff Cody 
24746ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
24756ebdcee2SJeff Cody      * to do, no intermediate images */
24766ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
24776ebdcee2SJeff Cody         ret = 0;
24786ebdcee2SJeff Cody         goto exit;
24796ebdcee2SJeff Cody     }
24806ebdcee2SJeff Cody 
24816ebdcee2SJeff Cody     intermediate = top;
24826ebdcee2SJeff Cody 
24836ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
24846ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
24856ebdcee2SJeff Cody      */
24866ebdcee2SJeff Cody     while (intermediate) {
24875839e53bSMarkus Armbruster         intermediate_state = g_new0(BlkIntermediateStates, 1);
24886ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
24896ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
24906ebdcee2SJeff Cody 
24916ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
24926ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
24936ebdcee2SJeff Cody             break;
24946ebdcee2SJeff Cody         }
24956ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
24966ebdcee2SJeff Cody     }
24976ebdcee2SJeff Cody     if (base_bs == NULL) {
24986ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
24996ebdcee2SJeff Cody          * unravel everything, and exit with error */
25006ebdcee2SJeff Cody         goto exit;
25016ebdcee2SJeff Cody     }
25026ebdcee2SJeff Cody 
25036ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
250454e26900SJeff Cody     backing_file_str = backing_file_str ? backing_file_str : base_bs->filename;
250554e26900SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, backing_file_str,
25066ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
25076ebdcee2SJeff Cody     if (ret) {
25086ebdcee2SJeff Cody         goto exit;
25096ebdcee2SJeff Cody     }
2510920beae1SFam Zheng     bdrv_set_backing_hd(new_top_bs, base_bs);
25116ebdcee2SJeff Cody 
25126ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25136ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
2514920beae1SFam Zheng         bdrv_set_backing_hd(intermediate_state->bs, NULL);
25154f6fd349SFam Zheng         bdrv_unref(intermediate_state->bs);
25166ebdcee2SJeff Cody     }
25176ebdcee2SJeff Cody     ret = 0;
25186ebdcee2SJeff Cody 
25196ebdcee2SJeff Cody exit:
25206ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25216ebdcee2SJeff Cody         g_free(intermediate_state);
25226ebdcee2SJeff Cody     }
25236ebdcee2SJeff Cody     return ret;
25246ebdcee2SJeff Cody }
25256ebdcee2SJeff Cody 
252683f64091Sbellard /**
252783f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
252883f64091Sbellard  */
252983f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
253083f64091Sbellard {
253183f64091Sbellard     BlockDriver *drv = bs->drv;
253251762288SStefan Hajnoczi     int ret;
253383f64091Sbellard     if (!drv)
253419cb3738Sbellard         return -ENOMEDIUM;
253583f64091Sbellard     if (!drv->bdrv_truncate)
253683f64091Sbellard         return -ENOTSUP;
253759f2689dSNaphtali Sprei     if (bs->read_only)
253859f2689dSNaphtali Sprei         return -EACCES;
25399c75e168SJeff Cody 
254051762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
254151762288SStefan Hajnoczi     if (ret == 0) {
254251762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2543ce1ffea8SJohn Snow         bdrv_dirty_bitmap_truncate(bs);
2544a7f53e26SMarkus Armbruster         if (bs->blk) {
2545a7f53e26SMarkus Armbruster             blk_dev_resize_cb(bs->blk);
2546a7f53e26SMarkus Armbruster         }
254751762288SStefan Hajnoczi     }
254851762288SStefan Hajnoczi     return ret;
254983f64091Sbellard }
255083f64091Sbellard 
255183f64091Sbellard /**
25524a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
25534a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
25544a1d5e1fSFam Zheng  */
25554a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
25564a1d5e1fSFam Zheng {
25574a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
25584a1d5e1fSFam Zheng     if (!drv) {
25594a1d5e1fSFam Zheng         return -ENOMEDIUM;
25604a1d5e1fSFam Zheng     }
25614a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
25624a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
25634a1d5e1fSFam Zheng     }
25644a1d5e1fSFam Zheng     if (bs->file) {
2565*9a4f4c31SKevin Wolf         return bdrv_get_allocated_file_size(bs->file->bs);
25664a1d5e1fSFam Zheng     }
25674a1d5e1fSFam Zheng     return -ENOTSUP;
25684a1d5e1fSFam Zheng }
25694a1d5e1fSFam Zheng 
25704a1d5e1fSFam Zheng /**
257165a9bb25SMarkus Armbruster  * Return number of sectors on success, -errno on error.
257283f64091Sbellard  */
257365a9bb25SMarkus Armbruster int64_t bdrv_nb_sectors(BlockDriverState *bs)
257483f64091Sbellard {
257583f64091Sbellard     BlockDriver *drv = bs->drv;
257665a9bb25SMarkus Armbruster 
257783f64091Sbellard     if (!drv)
257819cb3738Sbellard         return -ENOMEDIUM;
257951762288SStefan Hajnoczi 
2580b94a2610SKevin Wolf     if (drv->has_variable_length) {
2581b94a2610SKevin Wolf         int ret = refresh_total_sectors(bs, bs->total_sectors);
2582b94a2610SKevin Wolf         if (ret < 0) {
2583b94a2610SKevin Wolf             return ret;
2584fc01f7e7Sbellard         }
258546a4e4e6SStefan Hajnoczi     }
258665a9bb25SMarkus Armbruster     return bs->total_sectors;
258765a9bb25SMarkus Armbruster }
258865a9bb25SMarkus Armbruster 
258965a9bb25SMarkus Armbruster /**
259065a9bb25SMarkus Armbruster  * Return length in bytes on success, -errno on error.
259165a9bb25SMarkus Armbruster  * The length is always a multiple of BDRV_SECTOR_SIZE.
259265a9bb25SMarkus Armbruster  */
259365a9bb25SMarkus Armbruster int64_t bdrv_getlength(BlockDriverState *bs)
259465a9bb25SMarkus Armbruster {
259565a9bb25SMarkus Armbruster     int64_t ret = bdrv_nb_sectors(bs);
259665a9bb25SMarkus Armbruster 
25974a9c9ea0SFam Zheng     ret = ret > INT64_MAX / BDRV_SECTOR_SIZE ? -EFBIG : ret;
259865a9bb25SMarkus Armbruster     return ret < 0 ? ret : ret * BDRV_SECTOR_SIZE;
259946a4e4e6SStefan Hajnoczi }
2600fc01f7e7Sbellard 
260119cb3738Sbellard /* return 0 as number of sectors if no device present or error */
260296b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2603fc01f7e7Sbellard {
260465a9bb25SMarkus Armbruster     int64_t nb_sectors = bdrv_nb_sectors(bs);
260565a9bb25SMarkus Armbruster 
260665a9bb25SMarkus Armbruster     *nb_sectors_ptr = nb_sectors < 0 ? 0 : nb_sectors;
2607fc01f7e7Sbellard }
2608cf98951bSbellard 
2609ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2610ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
2611abd7f68dSMarkus Armbruster {
2612abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
2613abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
2614abd7f68dSMarkus Armbruster }
2615abd7f68dSMarkus Armbruster 
26161ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2617abd7f68dSMarkus Armbruster {
2618abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
2619abd7f68dSMarkus Armbruster }
2620abd7f68dSMarkus Armbruster 
26213e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
26223e1caa5fSPaolo Bonzini {
26233e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
26243e1caa5fSPaolo Bonzini 
26253e1caa5fSPaolo Bonzini     switch (on_err) {
26263e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
2627a589569fSWenchao Xia         return (error == ENOSPC) ?
2628a589569fSWenchao Xia                BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
26293e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
2630a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_STOP;
26313e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
2632a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_REPORT;
26333e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
2634a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_IGNORE;
26353e1caa5fSPaolo Bonzini     default:
26363e1caa5fSPaolo Bonzini         abort();
26373e1caa5fSPaolo Bonzini     }
26383e1caa5fSPaolo Bonzini }
26393e1caa5fSPaolo Bonzini 
2640c7c2ff0cSLuiz Capitulino static void send_qmp_error_event(BlockDriverState *bs,
2641c7c2ff0cSLuiz Capitulino                                  BlockErrorAction action,
2642c7c2ff0cSLuiz Capitulino                                  bool is_read, int error)
2643c7c2ff0cSLuiz Capitulino {
2644573742a5SPeter Maydell     IoOperationType optype;
2645c7c2ff0cSLuiz Capitulino 
2646573742a5SPeter Maydell     optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
2647573742a5SPeter Maydell     qapi_event_send_block_io_error(bdrv_get_device_name(bs), optype, action,
2648c7c2ff0cSLuiz Capitulino                                    bdrv_iostatus_is_enabled(bs),
2649624ff573SLuiz Capitulino                                    error == ENOSPC, strerror(error),
2650624ff573SLuiz Capitulino                                    &error_abort);
2651c7c2ff0cSLuiz Capitulino }
2652c7c2ff0cSLuiz Capitulino 
26533e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
26543e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
26553e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
26563e1caa5fSPaolo Bonzini  */
26573e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
26583e1caa5fSPaolo Bonzini                        bool is_read, int error)
26593e1caa5fSPaolo Bonzini {
26603e1caa5fSPaolo Bonzini     assert(error >= 0);
26612bd3bce8SPaolo Bonzini 
2662a589569fSWenchao Xia     if (action == BLOCK_ERROR_ACTION_STOP) {
26632bd3bce8SPaolo Bonzini         /* First set the iostatus, so that "info block" returns an iostatus
26642bd3bce8SPaolo Bonzini          * that matches the events raised so far (an additional error iostatus
26652bd3bce8SPaolo Bonzini          * is fine, but not a lost one).
26662bd3bce8SPaolo Bonzini          */
26673e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
26682bd3bce8SPaolo Bonzini 
26692bd3bce8SPaolo Bonzini         /* Then raise the request to stop the VM and the event.
26702bd3bce8SPaolo Bonzini          * qemu_system_vmstop_request_prepare has two effects.  First,
26712bd3bce8SPaolo Bonzini          * it ensures that the STOP event always comes after the
26722bd3bce8SPaolo Bonzini          * BLOCK_IO_ERROR event.  Second, it ensures that even if management
26732bd3bce8SPaolo Bonzini          * can observe the STOP event and do a "cont" before the STOP
26742bd3bce8SPaolo Bonzini          * event is issued, the VM will not stop.  In this case, vm_start()
26752bd3bce8SPaolo Bonzini          * also ensures that the STOP/RESUME pair of events is emitted.
26762bd3bce8SPaolo Bonzini          */
26772bd3bce8SPaolo Bonzini         qemu_system_vmstop_request_prepare();
2678c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26792bd3bce8SPaolo Bonzini         qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
26802bd3bce8SPaolo Bonzini     } else {
2681c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26823e1caa5fSPaolo Bonzini     }
26833e1caa5fSPaolo Bonzini }
26843e1caa5fSPaolo Bonzini 
2685b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
2686b338082bSbellard {
2687b338082bSbellard     return bs->read_only;
2688b338082bSbellard }
2689b338082bSbellard 
2690985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
2691985a03b0Sths {
2692985a03b0Sths     return bs->sg;
2693985a03b0Sths }
2694985a03b0Sths 
2695e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
2696e900a7b7SChristoph Hellwig {
2697e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
2698e900a7b7SChristoph Hellwig }
2699e900a7b7SChristoph Hellwig 
2700425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2701425b0148SPaolo Bonzini {
2702425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
270355b110f2SJeff Cody 
270455b110f2SJeff Cody     /* so a reopen() will preserve wce */
270555b110f2SJeff Cody     if (wce) {
270655b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
270755b110f2SJeff Cody     } else {
270855b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
270955b110f2SJeff Cody     }
2710425b0148SPaolo Bonzini }
2711425b0148SPaolo Bonzini 
2712ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
2713ea2384d3Sbellard {
2714ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
2715ea2384d3Sbellard         return 1;
2716ea2384d3Sbellard     return bs->encrypted;
2717ea2384d3Sbellard }
2718ea2384d3Sbellard 
2719c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
2720c0f4ce77Saliguori {
2721c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
2722c0f4ce77Saliguori 
2723c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2724c0f4ce77Saliguori         return 1;
2725c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
2726c0f4ce77Saliguori }
2727c0f4ce77Saliguori 
2728ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
2729ea2384d3Sbellard {
2730ea2384d3Sbellard     int ret;
2731ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
2732ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
2733ea2384d3Sbellard         if (ret < 0)
2734ea2384d3Sbellard             return ret;
2735ea2384d3Sbellard         if (!bs->encrypted)
2736ea2384d3Sbellard             return 0;
2737ea2384d3Sbellard     }
2738fd04a2aeSShahar Havivi     if (!bs->encrypted) {
2739fd04a2aeSShahar Havivi         return -EINVAL;
2740fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2741fd04a2aeSShahar Havivi         return -ENOMEDIUM;
2742fd04a2aeSShahar Havivi     }
2743c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
2744bb5fc20fSaliguori     if (ret < 0) {
2745bb5fc20fSaliguori         bs->valid_key = 0;
2746bb5fc20fSaliguori     } else if (!bs->valid_key) {
2747bb5fc20fSaliguori         bs->valid_key = 1;
2748a7f53e26SMarkus Armbruster         if (bs->blk) {
2749bb5fc20fSaliguori             /* call the change callback now, we skipped it on open */
2750a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
2751a7f53e26SMarkus Armbruster         }
2752bb5fc20fSaliguori     }
2753c0f4ce77Saliguori     return ret;
2754ea2384d3Sbellard }
2755ea2384d3Sbellard 
27564d2855a3SMarkus Armbruster /*
27574d2855a3SMarkus Armbruster  * Provide an encryption key for @bs.
27584d2855a3SMarkus Armbruster  * If @key is non-null:
27594d2855a3SMarkus Armbruster  *     If @bs is not encrypted, fail.
27604d2855a3SMarkus Armbruster  *     Else if the key is invalid, fail.
27614d2855a3SMarkus Armbruster  *     Else set @bs's key to @key, replacing the existing key, if any.
27624d2855a3SMarkus Armbruster  * If @key is null:
27634d2855a3SMarkus Armbruster  *     If @bs is encrypted and still lacks a key, fail.
27644d2855a3SMarkus Armbruster  *     Else do nothing.
27654d2855a3SMarkus Armbruster  * On failure, store an error object through @errp if non-null.
27664d2855a3SMarkus Armbruster  */
27674d2855a3SMarkus Armbruster void bdrv_add_key(BlockDriverState *bs, const char *key, Error **errp)
27684d2855a3SMarkus Armbruster {
27694d2855a3SMarkus Armbruster     if (key) {
27704d2855a3SMarkus Armbruster         if (!bdrv_is_encrypted(bs)) {
277181e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is not encrypted",
277281e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs));
27734d2855a3SMarkus Armbruster         } else if (bdrv_set_key(bs, key) < 0) {
2774c6bd8c70SMarkus Armbruster             error_setg(errp, QERR_INVALID_PASSWORD);
27754d2855a3SMarkus Armbruster         }
27764d2855a3SMarkus Armbruster     } else {
27774d2855a3SMarkus Armbruster         if (bdrv_key_required(bs)) {
2778b1ca6391SMarkus Armbruster             error_set(errp, ERROR_CLASS_DEVICE_ENCRYPTED,
2779b1ca6391SMarkus Armbruster                       "'%s' (%s) is encrypted",
278081e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs),
27814d2855a3SMarkus Armbruster                       bdrv_get_encrypted_filename(bs));
27824d2855a3SMarkus Armbruster         }
27834d2855a3SMarkus Armbruster     }
27844d2855a3SMarkus Armbruster }
27854d2855a3SMarkus Armbruster 
2786f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
2787ea2384d3Sbellard {
2788f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
2789ea2384d3Sbellard }
2790ea2384d3Sbellard 
2791ada42401SStefan Hajnoczi static int qsort_strcmp(const void *a, const void *b)
2792ada42401SStefan Hajnoczi {
2793ada42401SStefan Hajnoczi     return strcmp(a, b);
2794ada42401SStefan Hajnoczi }
2795ada42401SStefan Hajnoczi 
2796ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2797ea2384d3Sbellard                          void *opaque)
2798ea2384d3Sbellard {
2799ea2384d3Sbellard     BlockDriver *drv;
2800e855e4fbSJeff Cody     int count = 0;
2801ada42401SStefan Hajnoczi     int i;
2802e855e4fbSJeff Cody     const char **formats = NULL;
2803ea2384d3Sbellard 
28048a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2805e855e4fbSJeff Cody         if (drv->format_name) {
2806e855e4fbSJeff Cody             bool found = false;
2807e855e4fbSJeff Cody             int i = count;
2808e855e4fbSJeff Cody             while (formats && i && !found) {
2809e855e4fbSJeff Cody                 found = !strcmp(formats[--i], drv->format_name);
2810e855e4fbSJeff Cody             }
2811e855e4fbSJeff Cody 
2812e855e4fbSJeff Cody             if (!found) {
28135839e53bSMarkus Armbruster                 formats = g_renew(const char *, formats, count + 1);
2814e855e4fbSJeff Cody                 formats[count++] = drv->format_name;
2815ea2384d3Sbellard             }
2816ea2384d3Sbellard         }
2817e855e4fbSJeff Cody     }
2818ada42401SStefan Hajnoczi 
2819ada42401SStefan Hajnoczi     qsort(formats, count, sizeof(formats[0]), qsort_strcmp);
2820ada42401SStefan Hajnoczi 
2821ada42401SStefan Hajnoczi     for (i = 0; i < count; i++) {
2822ada42401SStefan Hajnoczi         it(opaque, formats[i]);
2823ada42401SStefan Hajnoczi     }
2824ada42401SStefan Hajnoczi 
2825e855e4fbSJeff Cody     g_free(formats);
2826e855e4fbSJeff Cody }
2827ea2384d3Sbellard 
2828dc364f4cSBenoît Canet /* This function is to find a node in the bs graph */
2829dc364f4cSBenoît Canet BlockDriverState *bdrv_find_node(const char *node_name)
2830dc364f4cSBenoît Canet {
2831dc364f4cSBenoît Canet     BlockDriverState *bs;
2832dc364f4cSBenoît Canet 
2833dc364f4cSBenoît Canet     assert(node_name);
2834dc364f4cSBenoît Canet 
2835dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2836dc364f4cSBenoît Canet         if (!strcmp(node_name, bs->node_name)) {
2837dc364f4cSBenoît Canet             return bs;
2838dc364f4cSBenoît Canet         }
2839dc364f4cSBenoît Canet     }
2840dc364f4cSBenoît Canet     return NULL;
2841dc364f4cSBenoît Canet }
2842dc364f4cSBenoît Canet 
2843c13163fbSBenoît Canet /* Put this QMP function here so it can access the static graph_bdrv_states. */
2844d5a8ee60SAlberto Garcia BlockDeviceInfoList *bdrv_named_nodes_list(Error **errp)
2845c13163fbSBenoît Canet {
2846c13163fbSBenoît Canet     BlockDeviceInfoList *list, *entry;
2847c13163fbSBenoît Canet     BlockDriverState *bs;
2848c13163fbSBenoît Canet 
2849c13163fbSBenoît Canet     list = NULL;
2850c13163fbSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2851d5a8ee60SAlberto Garcia         BlockDeviceInfo *info = bdrv_block_device_info(bs, errp);
2852d5a8ee60SAlberto Garcia         if (!info) {
2853d5a8ee60SAlberto Garcia             qapi_free_BlockDeviceInfoList(list);
2854d5a8ee60SAlberto Garcia             return NULL;
2855d5a8ee60SAlberto Garcia         }
2856c13163fbSBenoît Canet         entry = g_malloc0(sizeof(*entry));
2857d5a8ee60SAlberto Garcia         entry->value = info;
2858c13163fbSBenoît Canet         entry->next = list;
2859c13163fbSBenoît Canet         list = entry;
2860c13163fbSBenoît Canet     }
2861c13163fbSBenoît Canet 
2862c13163fbSBenoît Canet     return list;
2863c13163fbSBenoît Canet }
2864c13163fbSBenoît Canet 
286512d3ba82SBenoît Canet BlockDriverState *bdrv_lookup_bs(const char *device,
286612d3ba82SBenoît Canet                                  const char *node_name,
286712d3ba82SBenoît Canet                                  Error **errp)
286812d3ba82SBenoît Canet {
28697f06d47eSMarkus Armbruster     BlockBackend *blk;
28707f06d47eSMarkus Armbruster     BlockDriverState *bs;
287112d3ba82SBenoît Canet 
287212d3ba82SBenoît Canet     if (device) {
28737f06d47eSMarkus Armbruster         blk = blk_by_name(device);
287412d3ba82SBenoît Canet 
28757f06d47eSMarkus Armbruster         if (blk) {
28767f06d47eSMarkus Armbruster             return blk_bs(blk);
287712d3ba82SBenoît Canet         }
2878dd67fa50SBenoît Canet     }
287912d3ba82SBenoît Canet 
2880dd67fa50SBenoît Canet     if (node_name) {
288112d3ba82SBenoît Canet         bs = bdrv_find_node(node_name);
288212d3ba82SBenoît Canet 
2883dd67fa50SBenoît Canet         if (bs) {
2884dd67fa50SBenoît Canet             return bs;
2885dd67fa50SBenoît Canet         }
288612d3ba82SBenoît Canet     }
288712d3ba82SBenoît Canet 
2888dd67fa50SBenoît Canet     error_setg(errp, "Cannot find device=%s nor node_name=%s",
2889dd67fa50SBenoît Canet                      device ? device : "",
2890dd67fa50SBenoît Canet                      node_name ? node_name : "");
2891dd67fa50SBenoît Canet     return NULL;
289212d3ba82SBenoît Canet }
289312d3ba82SBenoît Canet 
28945a6684d2SJeff Cody /* If 'base' is in the same chain as 'top', return true. Otherwise,
28955a6684d2SJeff Cody  * return false.  If either argument is NULL, return false. */
28965a6684d2SJeff Cody bool bdrv_chain_contains(BlockDriverState *top, BlockDriverState *base)
28975a6684d2SJeff Cody {
28985a6684d2SJeff Cody     while (top && top != base) {
28995a6684d2SJeff Cody         top = top->backing_hd;
29005a6684d2SJeff Cody     }
29015a6684d2SJeff Cody 
29025a6684d2SJeff Cody     return top != NULL;
29035a6684d2SJeff Cody }
29045a6684d2SJeff Cody 
290504df765aSFam Zheng BlockDriverState *bdrv_next_node(BlockDriverState *bs)
290604df765aSFam Zheng {
290704df765aSFam Zheng     if (!bs) {
290804df765aSFam Zheng         return QTAILQ_FIRST(&graph_bdrv_states);
290904df765aSFam Zheng     }
291004df765aSFam Zheng     return QTAILQ_NEXT(bs, node_list);
291104df765aSFam Zheng }
291204df765aSFam Zheng 
29132f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
29142f399b0aSMarkus Armbruster {
29152f399b0aSMarkus Armbruster     if (!bs) {
29162f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
29172f399b0aSMarkus Armbruster     }
2918dc364f4cSBenoît Canet     return QTAILQ_NEXT(bs, device_list);
29192f399b0aSMarkus Armbruster }
29202f399b0aSMarkus Armbruster 
292120a9e77dSFam Zheng const char *bdrv_get_node_name(const BlockDriverState *bs)
292220a9e77dSFam Zheng {
292320a9e77dSFam Zheng     return bs->node_name;
292420a9e77dSFam Zheng }
292520a9e77dSFam Zheng 
29267f06d47eSMarkus Armbruster /* TODO check what callers really want: bs->node_name or blk_name() */
2927bfb197e0SMarkus Armbruster const char *bdrv_get_device_name(const BlockDriverState *bs)
2928ea2384d3Sbellard {
2929bfb197e0SMarkus Armbruster     return bs->blk ? blk_name(bs->blk) : "";
2930ea2384d3Sbellard }
2931ea2384d3Sbellard 
29329b2aa84fSAlberto Garcia /* This can be used to identify nodes that might not have a device
29339b2aa84fSAlberto Garcia  * name associated. Since node and device names live in the same
29349b2aa84fSAlberto Garcia  * namespace, the result is unambiguous. The exception is if both are
29359b2aa84fSAlberto Garcia  * absent, then this returns an empty (non-null) string. */
29369b2aa84fSAlberto Garcia const char *bdrv_get_device_or_node_name(const BlockDriverState *bs)
29379b2aa84fSAlberto Garcia {
29389b2aa84fSAlberto Garcia     return bs->blk ? blk_name(bs->blk) : bs->node_name;
29399b2aa84fSAlberto Garcia }
29409b2aa84fSAlberto Garcia 
2941c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
2942c8433287SMarkus Armbruster {
2943c8433287SMarkus Armbruster     return bs->open_flags;
2944c8433287SMarkus Armbruster }
2945c8433287SMarkus Armbruster 
29463ac21627SPeter Lieven int bdrv_has_zero_init_1(BlockDriverState *bs)
29473ac21627SPeter Lieven {
29483ac21627SPeter Lieven     return 1;
29493ac21627SPeter Lieven }
29503ac21627SPeter Lieven 
2951f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
2952f2feebbdSKevin Wolf {
2953f2feebbdSKevin Wolf     assert(bs->drv);
2954f2feebbdSKevin Wolf 
295511212d8fSPaolo Bonzini     /* If BS is a copy on write image, it is initialized to
295611212d8fSPaolo Bonzini        the contents of the base image, which may not be zeroes.  */
295711212d8fSPaolo Bonzini     if (bs->backing_hd) {
295811212d8fSPaolo Bonzini         return 0;
295911212d8fSPaolo Bonzini     }
2960336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
2961336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
2962f2feebbdSKevin Wolf     }
2963f2feebbdSKevin Wolf 
29643ac21627SPeter Lieven     /* safe default */
29653ac21627SPeter Lieven     return 0;
2966f2feebbdSKevin Wolf }
2967f2feebbdSKevin Wolf 
29684ce78691SPeter Lieven bool bdrv_unallocated_blocks_are_zero(BlockDriverState *bs)
29694ce78691SPeter Lieven {
29704ce78691SPeter Lieven     BlockDriverInfo bdi;
29714ce78691SPeter Lieven 
29724ce78691SPeter Lieven     if (bs->backing_hd) {
29734ce78691SPeter Lieven         return false;
29744ce78691SPeter Lieven     }
29754ce78691SPeter Lieven 
29764ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29774ce78691SPeter Lieven         return bdi.unallocated_blocks_are_zero;
29784ce78691SPeter Lieven     }
29794ce78691SPeter Lieven 
29804ce78691SPeter Lieven     return false;
29814ce78691SPeter Lieven }
29824ce78691SPeter Lieven 
29834ce78691SPeter Lieven bool bdrv_can_write_zeroes_with_unmap(BlockDriverState *bs)
29844ce78691SPeter Lieven {
29854ce78691SPeter Lieven     BlockDriverInfo bdi;
29864ce78691SPeter Lieven 
29874ce78691SPeter Lieven     if (bs->backing_hd || !(bs->open_flags & BDRV_O_UNMAP)) {
29884ce78691SPeter Lieven         return false;
29894ce78691SPeter Lieven     }
29904ce78691SPeter Lieven 
29914ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29924ce78691SPeter Lieven         return bdi.can_write_zeroes_with_unmap;
29934ce78691SPeter Lieven     }
29944ce78691SPeter Lieven 
29954ce78691SPeter Lieven     return false;
29964ce78691SPeter Lieven }
29974ce78691SPeter Lieven 
2998045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
2999045df330Saliguori {
3000045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
3001045df330Saliguori         return bs->backing_file;
3002045df330Saliguori     else if (bs->encrypted)
3003045df330Saliguori         return bs->filename;
3004045df330Saliguori     else
3005045df330Saliguori         return NULL;
3006045df330Saliguori }
3007045df330Saliguori 
300883f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
300983f64091Sbellard                                char *filename, int filename_size)
301083f64091Sbellard {
301183f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
301283f64091Sbellard }
301383f64091Sbellard 
3014faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3015faea38e7Sbellard {
3016faea38e7Sbellard     BlockDriver *drv = bs->drv;
3017faea38e7Sbellard     if (!drv)
301819cb3738Sbellard         return -ENOMEDIUM;
3019faea38e7Sbellard     if (!drv->bdrv_get_info)
3020faea38e7Sbellard         return -ENOTSUP;
3021faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3022faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3023faea38e7Sbellard }
3024faea38e7Sbellard 
3025eae041feSMax Reitz ImageInfoSpecific *bdrv_get_specific_info(BlockDriverState *bs)
3026eae041feSMax Reitz {
3027eae041feSMax Reitz     BlockDriver *drv = bs->drv;
3028eae041feSMax Reitz     if (drv && drv->bdrv_get_specific_info) {
3029eae041feSMax Reitz         return drv->bdrv_get_specific_info(bs);
3030eae041feSMax Reitz     }
3031eae041feSMax Reitz     return NULL;
3032eae041feSMax Reitz }
3033eae041feSMax Reitz 
30348b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
30358b9b0cc2SKevin Wolf {
3036bf736fe3SKevin Wolf     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
30378b9b0cc2SKevin Wolf         return;
30388b9b0cc2SKevin Wolf     }
30398b9b0cc2SKevin Wolf 
3040bf736fe3SKevin Wolf     bs->drv->bdrv_debug_event(bs, event);
304141c695c7SKevin Wolf }
30428b9b0cc2SKevin Wolf 
304341c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
304441c695c7SKevin Wolf                           const char *tag)
304541c695c7SKevin Wolf {
304641c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
3047*9a4f4c31SKevin Wolf         bs = bs->file ? bs->file->bs : NULL;
304841c695c7SKevin Wolf     }
304941c695c7SKevin Wolf 
305041c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
305141c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
305241c695c7SKevin Wolf     }
305341c695c7SKevin Wolf 
305441c695c7SKevin Wolf     return -ENOTSUP;
305541c695c7SKevin Wolf }
305641c695c7SKevin Wolf 
30574cc70e93SFam Zheng int bdrv_debug_remove_breakpoint(BlockDriverState *bs, const char *tag)
30584cc70e93SFam Zheng {
30594cc70e93SFam Zheng     while (bs && bs->drv && !bs->drv->bdrv_debug_remove_breakpoint) {
3060*9a4f4c31SKevin Wolf         bs = bs->file ? bs->file->bs : NULL;
30614cc70e93SFam Zheng     }
30624cc70e93SFam Zheng 
30634cc70e93SFam Zheng     if (bs && bs->drv && bs->drv->bdrv_debug_remove_breakpoint) {
30644cc70e93SFam Zheng         return bs->drv->bdrv_debug_remove_breakpoint(bs, tag);
30654cc70e93SFam Zheng     }
30664cc70e93SFam Zheng 
30674cc70e93SFam Zheng     return -ENOTSUP;
30684cc70e93SFam Zheng }
30694cc70e93SFam Zheng 
307041c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
307141c695c7SKevin Wolf {
3072938789eaSMax Reitz     while (bs && (!bs->drv || !bs->drv->bdrv_debug_resume)) {
3073*9a4f4c31SKevin Wolf         bs = bs->file ? bs->file->bs : NULL;
307441c695c7SKevin Wolf     }
307541c695c7SKevin Wolf 
307641c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
307741c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
307841c695c7SKevin Wolf     }
307941c695c7SKevin Wolf 
308041c695c7SKevin Wolf     return -ENOTSUP;
308141c695c7SKevin Wolf }
308241c695c7SKevin Wolf 
308341c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
308441c695c7SKevin Wolf {
308541c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
3086*9a4f4c31SKevin Wolf         bs = bs->file ? bs->file->bs : NULL;
308741c695c7SKevin Wolf     }
308841c695c7SKevin Wolf 
308941c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
309041c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
309141c695c7SKevin Wolf     }
309241c695c7SKevin Wolf 
309341c695c7SKevin Wolf     return false;
30948b9b0cc2SKevin Wolf }
30958b9b0cc2SKevin Wolf 
3096199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3097199630b6SBlue Swirl {
3098199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3099199630b6SBlue Swirl }
3100199630b6SBlue Swirl 
3101b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3102b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3103b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3104b1b1d783SJeff Cody  * the CWD rather than the chain. */
3105e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3106e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3107e8a6bb9cSMarcelo Tosatti {
3108b1b1d783SJeff Cody     char *filename_full = NULL;
3109b1b1d783SJeff Cody     char *backing_file_full = NULL;
3110b1b1d783SJeff Cody     char *filename_tmp = NULL;
3111b1b1d783SJeff Cody     int is_protocol = 0;
3112b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3113b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3114b1b1d783SJeff Cody 
3115b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3116e8a6bb9cSMarcelo Tosatti         return NULL;
3117e8a6bb9cSMarcelo Tosatti     }
3118e8a6bb9cSMarcelo Tosatti 
3119b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
3120b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
3121b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
3122b1b1d783SJeff Cody 
3123b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
3124b1b1d783SJeff Cody 
3125b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3126b1b1d783SJeff Cody 
3127b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
3128b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
3129b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3130b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3131b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3132b1b1d783SJeff Cody                 break;
3133b1b1d783SJeff Cody             }
3134e8a6bb9cSMarcelo Tosatti         } else {
3135b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
3136b1b1d783SJeff Cody              * image's filename path */
3137b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3138b1b1d783SJeff Cody                          backing_file);
3139b1b1d783SJeff Cody 
3140b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
3141b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
3142b1b1d783SJeff Cody                 continue;
3143b1b1d783SJeff Cody             }
3144b1b1d783SJeff Cody 
3145b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
3146b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
3147b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3148b1b1d783SJeff Cody                          curr_bs->backing_file);
3149b1b1d783SJeff Cody 
3150b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
3151b1b1d783SJeff Cody                 continue;
3152b1b1d783SJeff Cody             }
3153b1b1d783SJeff Cody 
3154b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
3155b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3156b1b1d783SJeff Cody                 break;
3157b1b1d783SJeff Cody             }
3158e8a6bb9cSMarcelo Tosatti         }
3159e8a6bb9cSMarcelo Tosatti     }
3160e8a6bb9cSMarcelo Tosatti 
3161b1b1d783SJeff Cody     g_free(filename_full);
3162b1b1d783SJeff Cody     g_free(backing_file_full);
3163b1b1d783SJeff Cody     g_free(filename_tmp);
3164b1b1d783SJeff Cody     return retval;
3165e8a6bb9cSMarcelo Tosatti }
3166e8a6bb9cSMarcelo Tosatti 
3167f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
3168f198fd1cSBenoît Canet {
3169f198fd1cSBenoît Canet     if (!bs->drv) {
3170f198fd1cSBenoît Canet         return 0;
3171f198fd1cSBenoît Canet     }
3172f198fd1cSBenoît Canet 
3173f198fd1cSBenoît Canet     if (!bs->backing_hd) {
3174f198fd1cSBenoît Canet         return 0;
3175f198fd1cSBenoît Canet     }
3176f198fd1cSBenoît Canet 
3177f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3178f198fd1cSBenoît Canet }
3179f198fd1cSBenoît Canet 
3180ea2384d3Sbellard void bdrv_init(void)
3181ea2384d3Sbellard {
31825efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
3183ea2384d3Sbellard }
3184ce1a14dcSpbrook 
3185eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
3186eb852011SMarkus Armbruster {
3187eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
3188eb852011SMarkus Armbruster     bdrv_init();
3189eb852011SMarkus Armbruster }
3190eb852011SMarkus Armbruster 
31915a8a30dbSKevin Wolf void bdrv_invalidate_cache(BlockDriverState *bs, Error **errp)
31920f15423cSAnthony Liguori {
31935a8a30dbSKevin Wolf     Error *local_err = NULL;
31945a8a30dbSKevin Wolf     int ret;
31955a8a30dbSKevin Wolf 
31963456a8d1SKevin Wolf     if (!bs->drv)  {
31973456a8d1SKevin Wolf         return;
31980f15423cSAnthony Liguori     }
31993456a8d1SKevin Wolf 
32007ea2d269SAlexey Kardashevskiy     if (!(bs->open_flags & BDRV_O_INCOMING)) {
32017ea2d269SAlexey Kardashevskiy         return;
32027ea2d269SAlexey Kardashevskiy     }
32037ea2d269SAlexey Kardashevskiy     bs->open_flags &= ~BDRV_O_INCOMING;
32047ea2d269SAlexey Kardashevskiy 
32053456a8d1SKevin Wolf     if (bs->drv->bdrv_invalidate_cache) {
32065a8a30dbSKevin Wolf         bs->drv->bdrv_invalidate_cache(bs, &local_err);
32073456a8d1SKevin Wolf     } else if (bs->file) {
3208*9a4f4c31SKevin Wolf         bdrv_invalidate_cache(bs->file->bs, &local_err);
32095a8a30dbSKevin Wolf     }
32105a8a30dbSKevin Wolf     if (local_err) {
32115a8a30dbSKevin Wolf         error_propagate(errp, local_err);
32125a8a30dbSKevin Wolf         return;
32133456a8d1SKevin Wolf     }
32143456a8d1SKevin Wolf 
32155a8a30dbSKevin Wolf     ret = refresh_total_sectors(bs, bs->total_sectors);
32165a8a30dbSKevin Wolf     if (ret < 0) {
32175a8a30dbSKevin Wolf         error_setg_errno(errp, -ret, "Could not refresh total sector count");
32185a8a30dbSKevin Wolf         return;
32195a8a30dbSKevin Wolf     }
32200f15423cSAnthony Liguori }
32210f15423cSAnthony Liguori 
32225a8a30dbSKevin Wolf void bdrv_invalidate_cache_all(Error **errp)
32230f15423cSAnthony Liguori {
32240f15423cSAnthony Liguori     BlockDriverState *bs;
32255a8a30dbSKevin Wolf     Error *local_err = NULL;
32260f15423cSAnthony Liguori 
3227dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
3228ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
3229ed78cda3SStefan Hajnoczi 
3230ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
32315a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs, &local_err);
3232ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
32335a8a30dbSKevin Wolf         if (local_err) {
32345a8a30dbSKevin Wolf             error_propagate(errp, local_err);
32355a8a30dbSKevin Wolf             return;
32365a8a30dbSKevin Wolf         }
32370f15423cSAnthony Liguori     }
32380f15423cSAnthony Liguori }
32390f15423cSAnthony Liguori 
3240f9f05dc5SKevin Wolf /**************************************************************/
324119cb3738Sbellard /* removable device support */
324219cb3738Sbellard 
324319cb3738Sbellard /**
324419cb3738Sbellard  * Return TRUE if the media is present
324519cb3738Sbellard  */
324619cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
324719cb3738Sbellard {
324819cb3738Sbellard     BlockDriver *drv = bs->drv;
3249a1aff5bfSMarkus Armbruster 
325019cb3738Sbellard     if (!drv)
325119cb3738Sbellard         return 0;
325219cb3738Sbellard     if (!drv->bdrv_is_inserted)
3253a1aff5bfSMarkus Armbruster         return 1;
3254a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
325519cb3738Sbellard }
325619cb3738Sbellard 
325719cb3738Sbellard /**
32588e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
32598e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
326019cb3738Sbellard  */
326119cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
326219cb3738Sbellard {
326319cb3738Sbellard     BlockDriver *drv = bs->drv;
326419cb3738Sbellard 
32658e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
32668e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
32678e49ca46SMarkus Armbruster     }
32688e49ca46SMarkus Armbruster     return -ENOTSUP;
326919cb3738Sbellard }
327019cb3738Sbellard 
327119cb3738Sbellard /**
327219cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
327319cb3738Sbellard  */
3274f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
327519cb3738Sbellard {
327619cb3738Sbellard     BlockDriver *drv = bs->drv;
3277bfb197e0SMarkus Armbruster     const char *device_name;
327819cb3738Sbellard 
3279822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
3280822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
328119cb3738Sbellard     }
32826f382ed2SLuiz Capitulino 
3283bfb197e0SMarkus Armbruster     device_name = bdrv_get_device_name(bs);
3284bfb197e0SMarkus Armbruster     if (device_name[0] != '\0') {
3285bfb197e0SMarkus Armbruster         qapi_event_send_device_tray_moved(device_name,
3286a5ee7bd4SWenchao Xia                                           eject_flag, &error_abort);
32876f382ed2SLuiz Capitulino     }
328819cb3738Sbellard }
328919cb3738Sbellard 
329019cb3738Sbellard /**
329119cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
329219cb3738Sbellard  * to eject it manually).
329319cb3738Sbellard  */
3294025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
329519cb3738Sbellard {
329619cb3738Sbellard     BlockDriver *drv = bs->drv;
329719cb3738Sbellard 
3298025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
3299b8c6d095SStefan Hajnoczi 
3300025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
3301025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
330219cb3738Sbellard     }
330319cb3738Sbellard }
3304985a03b0Sths 
33051b7fd729SPaolo Bonzini void bdrv_set_guest_block_size(BlockDriverState *bs, int align)
33067b6f9300SMarkus Armbruster {
33071b7fd729SPaolo Bonzini     bs->guest_block_size = align;
33087b6f9300SMarkus Armbruster }
33097cd1e32aSlirans@il.ibm.com 
33100db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_find_dirty_bitmap(BlockDriverState *bs, const char *name)
33110db6e54aSFam Zheng {
33120db6e54aSFam Zheng     BdrvDirtyBitmap *bm;
33130db6e54aSFam Zheng 
33140db6e54aSFam Zheng     assert(name);
33150db6e54aSFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
33160db6e54aSFam Zheng         if (bm->name && !strcmp(name, bm->name)) {
33170db6e54aSFam Zheng             return bm;
33180db6e54aSFam Zheng         }
33190db6e54aSFam Zheng     }
33200db6e54aSFam Zheng     return NULL;
33210db6e54aSFam Zheng }
33220db6e54aSFam Zheng 
332320dca810SJohn Snow void bdrv_dirty_bitmap_make_anon(BdrvDirtyBitmap *bitmap)
33240db6e54aSFam Zheng {
33259bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
33260db6e54aSFam Zheng     g_free(bitmap->name);
33270db6e54aSFam Zheng     bitmap->name = NULL;
33280db6e54aSFam Zheng }
33290db6e54aSFam Zheng 
33300db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_create_dirty_bitmap(BlockDriverState *bs,
33315fba6c0eSJohn Snow                                           uint32_t granularity,
33320db6e54aSFam Zheng                                           const char *name,
3333b8afb520SFam Zheng                                           Error **errp)
33347cd1e32aSlirans@il.ibm.com {
33357cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
3336e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
33375fba6c0eSJohn Snow     uint32_t sector_granularity;
3338a55eb92cSJan Kiszka 
333950717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
334050717e94SPaolo Bonzini 
33410db6e54aSFam Zheng     if (name && bdrv_find_dirty_bitmap(bs, name)) {
33420db6e54aSFam Zheng         error_setg(errp, "Bitmap already exists: %s", name);
33430db6e54aSFam Zheng         return NULL;
33440db6e54aSFam Zheng     }
33455fba6c0eSJohn Snow     sector_granularity = granularity >> BDRV_SECTOR_BITS;
33465fba6c0eSJohn Snow     assert(sector_granularity);
334757322b78SMarkus Armbruster     bitmap_size = bdrv_nb_sectors(bs);
3348b8afb520SFam Zheng     if (bitmap_size < 0) {
3349b8afb520SFam Zheng         error_setg_errno(errp, -bitmap_size, "could not get length of device");
3350b8afb520SFam Zheng         errno = -bitmap_size;
3351b8afb520SFam Zheng         return NULL;
3352b8afb520SFam Zheng     }
33535839e53bSMarkus Armbruster     bitmap = g_new0(BdrvDirtyBitmap, 1);
33545fba6c0eSJohn Snow     bitmap->bitmap = hbitmap_alloc(bitmap_size, ctz32(sector_granularity));
3355e74e6b78SJohn Snow     bitmap->size = bitmap_size;
33560db6e54aSFam Zheng     bitmap->name = g_strdup(name);
3357b8e6fb75SJohn Snow     bitmap->disabled = false;
3358e4654d2dSFam Zheng     QLIST_INSERT_HEAD(&bs->dirty_bitmaps, bitmap, list);
3359e4654d2dSFam Zheng     return bitmap;
3360e4654d2dSFam Zheng }
3361e4654d2dSFam Zheng 
33629bd2b08fSJohn Snow bool bdrv_dirty_bitmap_frozen(BdrvDirtyBitmap *bitmap)
33639bd2b08fSJohn Snow {
33649bd2b08fSJohn Snow     return bitmap->successor;
33659bd2b08fSJohn Snow }
33669bd2b08fSJohn Snow 
3367b8e6fb75SJohn Snow bool bdrv_dirty_bitmap_enabled(BdrvDirtyBitmap *bitmap)
3368b8e6fb75SJohn Snow {
33699bd2b08fSJohn Snow     return !(bitmap->disabled || bitmap->successor);
33709bd2b08fSJohn Snow }
33719bd2b08fSJohn Snow 
33729abe3bdcSJohn Snow DirtyBitmapStatus bdrv_dirty_bitmap_status(BdrvDirtyBitmap *bitmap)
33739abe3bdcSJohn Snow {
33749abe3bdcSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33759abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_FROZEN;
33769abe3bdcSJohn Snow     } else if (!bdrv_dirty_bitmap_enabled(bitmap)) {
33779abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_DISABLED;
33789abe3bdcSJohn Snow     } else {
33799abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_ACTIVE;
33809abe3bdcSJohn Snow     }
33819abe3bdcSJohn Snow }
33829abe3bdcSJohn Snow 
33839bd2b08fSJohn Snow /**
33849bd2b08fSJohn Snow  * Create a successor bitmap destined to replace this bitmap after an operation.
33859bd2b08fSJohn Snow  * Requires that the bitmap is not frozen and has no successor.
33869bd2b08fSJohn Snow  */
33879bd2b08fSJohn Snow int bdrv_dirty_bitmap_create_successor(BlockDriverState *bs,
33889bd2b08fSJohn Snow                                        BdrvDirtyBitmap *bitmap, Error **errp)
33899bd2b08fSJohn Snow {
33909bd2b08fSJohn Snow     uint64_t granularity;
33919bd2b08fSJohn Snow     BdrvDirtyBitmap *child;
33929bd2b08fSJohn Snow 
33939bd2b08fSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33949bd2b08fSJohn Snow         error_setg(errp, "Cannot create a successor for a bitmap that is "
33959bd2b08fSJohn Snow                    "currently frozen");
33969bd2b08fSJohn Snow         return -1;
33979bd2b08fSJohn Snow     }
33989bd2b08fSJohn Snow     assert(!bitmap->successor);
33999bd2b08fSJohn Snow 
34009bd2b08fSJohn Snow     /* Create an anonymous successor */
34019bd2b08fSJohn Snow     granularity = bdrv_dirty_bitmap_granularity(bitmap);
34029bd2b08fSJohn Snow     child = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp);
34039bd2b08fSJohn Snow     if (!child) {
34049bd2b08fSJohn Snow         return -1;
34059bd2b08fSJohn Snow     }
34069bd2b08fSJohn Snow 
34079bd2b08fSJohn Snow     /* Successor will be on or off based on our current state. */
34089bd2b08fSJohn Snow     child->disabled = bitmap->disabled;
34099bd2b08fSJohn Snow 
34109bd2b08fSJohn Snow     /* Install the successor and freeze the parent */
34119bd2b08fSJohn Snow     bitmap->successor = child;
34129bd2b08fSJohn Snow     return 0;
34139bd2b08fSJohn Snow }
34149bd2b08fSJohn Snow 
34159bd2b08fSJohn Snow /**
34169bd2b08fSJohn Snow  * For a bitmap with a successor, yield our name to the successor,
34179bd2b08fSJohn Snow  * delete the old bitmap, and return a handle to the new bitmap.
34189bd2b08fSJohn Snow  */
34199bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_dirty_bitmap_abdicate(BlockDriverState *bs,
34209bd2b08fSJohn Snow                                             BdrvDirtyBitmap *bitmap,
34219bd2b08fSJohn Snow                                             Error **errp)
34229bd2b08fSJohn Snow {
34239bd2b08fSJohn Snow     char *name;
34249bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = bitmap->successor;
34259bd2b08fSJohn Snow 
34269bd2b08fSJohn Snow     if (successor == NULL) {
34279bd2b08fSJohn Snow         error_setg(errp, "Cannot relinquish control if "
34289bd2b08fSJohn Snow                    "there's no successor present");
34299bd2b08fSJohn Snow         return NULL;
34309bd2b08fSJohn Snow     }
34319bd2b08fSJohn Snow 
34329bd2b08fSJohn Snow     name = bitmap->name;
34339bd2b08fSJohn Snow     bitmap->name = NULL;
34349bd2b08fSJohn Snow     successor->name = name;
34359bd2b08fSJohn Snow     bitmap->successor = NULL;
34369bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, bitmap);
34379bd2b08fSJohn Snow 
34389bd2b08fSJohn Snow     return successor;
34399bd2b08fSJohn Snow }
34409bd2b08fSJohn Snow 
34419bd2b08fSJohn Snow /**
34429bd2b08fSJohn Snow  * In cases of failure where we can no longer safely delete the parent,
34439bd2b08fSJohn Snow  * we may wish to re-join the parent and child/successor.
34449bd2b08fSJohn Snow  * The merged parent will be un-frozen, but not explicitly re-enabled.
34459bd2b08fSJohn Snow  */
34469bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_reclaim_dirty_bitmap(BlockDriverState *bs,
34479bd2b08fSJohn Snow                                            BdrvDirtyBitmap *parent,
34489bd2b08fSJohn Snow                                            Error **errp)
34499bd2b08fSJohn Snow {
34509bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = parent->successor;
34519bd2b08fSJohn Snow 
34529bd2b08fSJohn Snow     if (!successor) {
34539bd2b08fSJohn Snow         error_setg(errp, "Cannot reclaim a successor when none is present");
34549bd2b08fSJohn Snow         return NULL;
34559bd2b08fSJohn Snow     }
34569bd2b08fSJohn Snow 
34579bd2b08fSJohn Snow     if (!hbitmap_merge(parent->bitmap, successor->bitmap)) {
34589bd2b08fSJohn Snow         error_setg(errp, "Merging of parent and successor bitmap failed");
34599bd2b08fSJohn Snow         return NULL;
34609bd2b08fSJohn Snow     }
34619bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, successor);
34629bd2b08fSJohn Snow     parent->successor = NULL;
34639bd2b08fSJohn Snow 
34649bd2b08fSJohn Snow     return parent;
3465b8e6fb75SJohn Snow }
3466b8e6fb75SJohn Snow 
3467ce1ffea8SJohn Snow /**
3468ce1ffea8SJohn Snow  * Truncates _all_ bitmaps attached to a BDS.
3469ce1ffea8SJohn Snow  */
3470ce1ffea8SJohn Snow static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs)
3471ce1ffea8SJohn Snow {
3472ce1ffea8SJohn Snow     BdrvDirtyBitmap *bitmap;
3473ce1ffea8SJohn Snow     uint64_t size = bdrv_nb_sectors(bs);
3474ce1ffea8SJohn Snow 
3475ce1ffea8SJohn Snow     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
347606207b0fSJohn Snow         assert(!bdrv_dirty_bitmap_frozen(bitmap));
3477ce1ffea8SJohn Snow         hbitmap_truncate(bitmap->bitmap, size);
34785270b6a0SJohn Snow         bitmap->size = size;
3479ce1ffea8SJohn Snow     }
3480ce1ffea8SJohn Snow }
3481ce1ffea8SJohn Snow 
3482e4654d2dSFam Zheng void bdrv_release_dirty_bitmap(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
3483e4654d2dSFam Zheng {
3484e4654d2dSFam Zheng     BdrvDirtyBitmap *bm, *next;
3485e4654d2dSFam Zheng     QLIST_FOREACH_SAFE(bm, &bs->dirty_bitmaps, list, next) {
3486e4654d2dSFam Zheng         if (bm == bitmap) {
34879bd2b08fSJohn Snow             assert(!bdrv_dirty_bitmap_frozen(bm));
3488e4654d2dSFam Zheng             QLIST_REMOVE(bitmap, list);
3489e4654d2dSFam Zheng             hbitmap_free(bitmap->bitmap);
34900db6e54aSFam Zheng             g_free(bitmap->name);
3491e4654d2dSFam Zheng             g_free(bitmap);
3492e4654d2dSFam Zheng             return;
34937cd1e32aSlirans@il.ibm.com         }
34947cd1e32aSlirans@il.ibm.com     }
34957cd1e32aSlirans@il.ibm.com }
34967cd1e32aSlirans@il.ibm.com 
3497b8e6fb75SJohn Snow void bdrv_disable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3498b8e6fb75SJohn Snow {
34999bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3500b8e6fb75SJohn Snow     bitmap->disabled = true;
3501b8e6fb75SJohn Snow }
3502b8e6fb75SJohn Snow 
3503b8e6fb75SJohn Snow void bdrv_enable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3504b8e6fb75SJohn Snow {
35059bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3506b8e6fb75SJohn Snow     bitmap->disabled = false;
3507b8e6fb75SJohn Snow }
3508b8e6fb75SJohn Snow 
350921b56835SFam Zheng BlockDirtyInfoList *bdrv_query_dirty_bitmaps(BlockDriverState *bs)
351021b56835SFam Zheng {
351121b56835SFam Zheng     BdrvDirtyBitmap *bm;
351221b56835SFam Zheng     BlockDirtyInfoList *list = NULL;
351321b56835SFam Zheng     BlockDirtyInfoList **plist = &list;
351421b56835SFam Zheng 
351521b56835SFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
35165839e53bSMarkus Armbruster         BlockDirtyInfo *info = g_new0(BlockDirtyInfo, 1);
35175839e53bSMarkus Armbruster         BlockDirtyInfoList *entry = g_new0(BlockDirtyInfoList, 1);
351820dca810SJohn Snow         info->count = bdrv_get_dirty_count(bm);
3519592fdd02SJohn Snow         info->granularity = bdrv_dirty_bitmap_granularity(bm);
35200db6e54aSFam Zheng         info->has_name = !!bm->name;
35210db6e54aSFam Zheng         info->name = g_strdup(bm->name);
35229abe3bdcSJohn Snow         info->status = bdrv_dirty_bitmap_status(bm);
352321b56835SFam Zheng         entry->value = info;
352421b56835SFam Zheng         *plist = entry;
352521b56835SFam Zheng         plist = &entry->next;
352621b56835SFam Zheng     }
352721b56835SFam Zheng 
352821b56835SFam Zheng     return list;
352921b56835SFam Zheng }
353021b56835SFam Zheng 
3531e4654d2dSFam Zheng int bdrv_get_dirty(BlockDriverState *bs, BdrvDirtyBitmap *bitmap, int64_t sector)
35327cd1e32aSlirans@il.ibm.com {
3533e4654d2dSFam Zheng     if (bitmap) {
3534e4654d2dSFam Zheng         return hbitmap_get(bitmap->bitmap, sector);
35357cd1e32aSlirans@il.ibm.com     } else {
35367cd1e32aSlirans@il.ibm.com         return 0;
35377cd1e32aSlirans@il.ibm.com     }
35387cd1e32aSlirans@il.ibm.com }
35397cd1e32aSlirans@il.ibm.com 
3540341ebc2fSJohn Snow /**
3541341ebc2fSJohn Snow  * Chooses a default granularity based on the existing cluster size,
3542341ebc2fSJohn Snow  * but clamped between [4K, 64K]. Defaults to 64K in the case that there
3543341ebc2fSJohn Snow  * is no cluster size information available.
3544341ebc2fSJohn Snow  */
3545341ebc2fSJohn Snow uint32_t bdrv_get_default_bitmap_granularity(BlockDriverState *bs)
3546341ebc2fSJohn Snow {
3547341ebc2fSJohn Snow     BlockDriverInfo bdi;
3548341ebc2fSJohn Snow     uint32_t granularity;
3549341ebc2fSJohn Snow 
3550341ebc2fSJohn Snow     if (bdrv_get_info(bs, &bdi) >= 0 && bdi.cluster_size > 0) {
3551341ebc2fSJohn Snow         granularity = MAX(4096, bdi.cluster_size);
3552341ebc2fSJohn Snow         granularity = MIN(65536, granularity);
3553341ebc2fSJohn Snow     } else {
3554341ebc2fSJohn Snow         granularity = 65536;
3555341ebc2fSJohn Snow     }
3556341ebc2fSJohn Snow 
3557341ebc2fSJohn Snow     return granularity;
3558341ebc2fSJohn Snow }
3559341ebc2fSJohn Snow 
3560592fdd02SJohn Snow uint32_t bdrv_dirty_bitmap_granularity(BdrvDirtyBitmap *bitmap)
3561592fdd02SJohn Snow {
3562592fdd02SJohn Snow     return BDRV_SECTOR_SIZE << hbitmap_granularity(bitmap->bitmap);
3563592fdd02SJohn Snow }
3564592fdd02SJohn Snow 
356520dca810SJohn Snow void bdrv_dirty_iter_init(BdrvDirtyBitmap *bitmap, HBitmapIter *hbi)
35661755da16SPaolo Bonzini {
3567e4654d2dSFam Zheng     hbitmap_iter_init(hbi, bitmap->bitmap, 0);
35681755da16SPaolo Bonzini }
35691755da16SPaolo Bonzini 
357020dca810SJohn Snow void bdrv_set_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3571c4237dfaSVladimir Sementsov-Ogievskiy                            int64_t cur_sector, int nr_sectors)
3572c4237dfaSVladimir Sementsov-Ogievskiy {
3573b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3574c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3575c4237dfaSVladimir Sementsov-Ogievskiy }
3576c4237dfaSVladimir Sementsov-Ogievskiy 
357720dca810SJohn Snow void bdrv_reset_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3578c4237dfaSVladimir Sementsov-Ogievskiy                              int64_t cur_sector, int nr_sectors)
3579c4237dfaSVladimir Sementsov-Ogievskiy {
3580b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3581c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_reset(bitmap->bitmap, cur_sector, nr_sectors);
3582c4237dfaSVladimir Sementsov-Ogievskiy }
3583c4237dfaSVladimir Sementsov-Ogievskiy 
3584e74e6b78SJohn Snow void bdrv_clear_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3585e74e6b78SJohn Snow {
3586e74e6b78SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3587c6a8c328SWen Congyang     hbitmap_reset_all(bitmap->bitmap);
3588e74e6b78SJohn Snow }
3589e74e6b78SJohn Snow 
3590e0c47b6cSStefan Hajnoczi void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
35911755da16SPaolo Bonzini                     int nr_sectors)
35921755da16SPaolo Bonzini {
3593e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
3594e4654d2dSFam Zheng     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
3595b8e6fb75SJohn Snow         if (!bdrv_dirty_bitmap_enabled(bitmap)) {
3596b8e6fb75SJohn Snow             continue;
3597b8e6fb75SJohn Snow         }
3598e4654d2dSFam Zheng         hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3599e4654d2dSFam Zheng     }
36001755da16SPaolo Bonzini }
36011755da16SPaolo Bonzini 
3602d58d8453SJohn Snow /**
3603d58d8453SJohn Snow  * Advance an HBitmapIter to an arbitrary offset.
3604d58d8453SJohn Snow  */
3605d58d8453SJohn Snow void bdrv_set_dirty_iter(HBitmapIter *hbi, int64_t offset)
3606d58d8453SJohn Snow {
3607d58d8453SJohn Snow     assert(hbi->hb);
3608d58d8453SJohn Snow     hbitmap_iter_init(hbi, hbi->hb, offset);
3609d58d8453SJohn Snow }
3610d58d8453SJohn Snow 
361120dca810SJohn Snow int64_t bdrv_get_dirty_count(BdrvDirtyBitmap *bitmap)
3612aaa0eb75SLiran Schour {
3613e4654d2dSFam Zheng     return hbitmap_count(bitmap->bitmap);
3614aaa0eb75SLiran Schour }
3615f88e1a42SJes Sorensen 
36169fcb0251SFam Zheng /* Get a reference to bs */
36179fcb0251SFam Zheng void bdrv_ref(BlockDriverState *bs)
36189fcb0251SFam Zheng {
36199fcb0251SFam Zheng     bs->refcnt++;
36209fcb0251SFam Zheng }
36219fcb0251SFam Zheng 
36229fcb0251SFam Zheng /* Release a previously grabbed reference to bs.
36239fcb0251SFam Zheng  * If after releasing, reference count is zero, the BlockDriverState is
36249fcb0251SFam Zheng  * deleted. */
36259fcb0251SFam Zheng void bdrv_unref(BlockDriverState *bs)
36269fcb0251SFam Zheng {
36279a4d5ca6SJeff Cody     if (!bs) {
36289a4d5ca6SJeff Cody         return;
36299a4d5ca6SJeff Cody     }
36309fcb0251SFam Zheng     assert(bs->refcnt > 0);
36319fcb0251SFam Zheng     if (--bs->refcnt == 0) {
36329fcb0251SFam Zheng         bdrv_delete(bs);
36339fcb0251SFam Zheng     }
36349fcb0251SFam Zheng }
36359fcb0251SFam Zheng 
3636fbe40ff7SFam Zheng struct BdrvOpBlocker {
3637fbe40ff7SFam Zheng     Error *reason;
3638fbe40ff7SFam Zheng     QLIST_ENTRY(BdrvOpBlocker) list;
3639fbe40ff7SFam Zheng };
3640fbe40ff7SFam Zheng 
3641fbe40ff7SFam Zheng bool bdrv_op_is_blocked(BlockDriverState *bs, BlockOpType op, Error **errp)
3642fbe40ff7SFam Zheng {
3643fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3644fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3645fbe40ff7SFam Zheng     if (!QLIST_EMPTY(&bs->op_blockers[op])) {
3646fbe40ff7SFam Zheng         blocker = QLIST_FIRST(&bs->op_blockers[op]);
3647fbe40ff7SFam Zheng         if (errp) {
364881e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is busy: %s",
364981e5f78aSAlberto Garcia                        bdrv_get_device_or_node_name(bs),
3650bfb197e0SMarkus Armbruster                        error_get_pretty(blocker->reason));
3651fbe40ff7SFam Zheng         }
3652fbe40ff7SFam Zheng         return true;
3653fbe40ff7SFam Zheng     }
3654fbe40ff7SFam Zheng     return false;
3655fbe40ff7SFam Zheng }
3656fbe40ff7SFam Zheng 
3657fbe40ff7SFam Zheng void bdrv_op_block(BlockDriverState *bs, BlockOpType op, Error *reason)
3658fbe40ff7SFam Zheng {
3659fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3660fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3661fbe40ff7SFam Zheng 
36625839e53bSMarkus Armbruster     blocker = g_new0(BdrvOpBlocker, 1);
3663fbe40ff7SFam Zheng     blocker->reason = reason;
3664fbe40ff7SFam Zheng     QLIST_INSERT_HEAD(&bs->op_blockers[op], blocker, list);
3665fbe40ff7SFam Zheng }
3666fbe40ff7SFam Zheng 
3667fbe40ff7SFam Zheng void bdrv_op_unblock(BlockDriverState *bs, BlockOpType op, Error *reason)
3668fbe40ff7SFam Zheng {
3669fbe40ff7SFam Zheng     BdrvOpBlocker *blocker, *next;
3670fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3671fbe40ff7SFam Zheng     QLIST_FOREACH_SAFE(blocker, &bs->op_blockers[op], list, next) {
3672fbe40ff7SFam Zheng         if (blocker->reason == reason) {
3673fbe40ff7SFam Zheng             QLIST_REMOVE(blocker, list);
3674fbe40ff7SFam Zheng             g_free(blocker);
3675fbe40ff7SFam Zheng         }
3676fbe40ff7SFam Zheng     }
3677fbe40ff7SFam Zheng }
3678fbe40ff7SFam Zheng 
3679fbe40ff7SFam Zheng void bdrv_op_block_all(BlockDriverState *bs, Error *reason)
3680fbe40ff7SFam Zheng {
3681fbe40ff7SFam Zheng     int i;
3682fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3683fbe40ff7SFam Zheng         bdrv_op_block(bs, i, reason);
3684fbe40ff7SFam Zheng     }
3685fbe40ff7SFam Zheng }
3686fbe40ff7SFam Zheng 
3687fbe40ff7SFam Zheng void bdrv_op_unblock_all(BlockDriverState *bs, Error *reason)
3688fbe40ff7SFam Zheng {
3689fbe40ff7SFam Zheng     int i;
3690fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3691fbe40ff7SFam Zheng         bdrv_op_unblock(bs, i, reason);
3692fbe40ff7SFam Zheng     }
3693fbe40ff7SFam Zheng }
3694fbe40ff7SFam Zheng 
3695fbe40ff7SFam Zheng bool bdrv_op_blocker_is_empty(BlockDriverState *bs)
3696fbe40ff7SFam Zheng {
3697fbe40ff7SFam Zheng     int i;
3698fbe40ff7SFam Zheng 
3699fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3700fbe40ff7SFam Zheng         if (!QLIST_EMPTY(&bs->op_blockers[i])) {
3701fbe40ff7SFam Zheng             return false;
3702fbe40ff7SFam Zheng         }
3703fbe40ff7SFam Zheng     }
3704fbe40ff7SFam Zheng     return true;
3705fbe40ff7SFam Zheng }
3706fbe40ff7SFam Zheng 
370728a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
370828a7282aSLuiz Capitulino {
3709d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
371058e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
371128a7282aSLuiz Capitulino }
371228a7282aSLuiz Capitulino 
371328a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
371428a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
371528a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
371628a7282aSLuiz Capitulino {
3717d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
371892aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
371992aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
372092aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
372128a7282aSLuiz Capitulino }
372228a7282aSLuiz Capitulino 
372328a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
372428a7282aSLuiz Capitulino {
3725d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
372628a7282aSLuiz Capitulino }
372728a7282aSLuiz Capitulino 
372828a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
372928a7282aSLuiz Capitulino {
373028a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
373158e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
37323bd293c3SPaolo Bonzini         if (bs->job) {
37333bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
37343bd293c3SPaolo Bonzini         }
373528a7282aSLuiz Capitulino     }
373628a7282aSLuiz Capitulino }
373728a7282aSLuiz Capitulino 
373828a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
373928a7282aSLuiz Capitulino {
37403e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
37413e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
374258e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
374358e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
374428a7282aSLuiz Capitulino     }
374528a7282aSLuiz Capitulino }
374628a7282aSLuiz Capitulino 
3747d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
3748f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
3749f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
3750f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
3751f88e1a42SJes Sorensen {
375283d0521aSChunyan Liu     QemuOptsList *create_opts = NULL;
375383d0521aSChunyan Liu     QemuOpts *opts = NULL;
375483d0521aSChunyan Liu     const char *backing_fmt, *backing_file;
375583d0521aSChunyan Liu     int64_t size;
3756f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
3757cc84d90fSMax Reitz     Error *local_err = NULL;
3758f88e1a42SJes Sorensen     int ret = 0;
3759f88e1a42SJes Sorensen 
3760f88e1a42SJes Sorensen     /* Find driver and parse its options */
3761f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
3762f88e1a42SJes Sorensen     if (!drv) {
376371c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
3764d92ada22SLuiz Capitulino         return;
3765f88e1a42SJes Sorensen     }
3766f88e1a42SJes Sorensen 
3767b65a5e12SMax Reitz     proto_drv = bdrv_find_protocol(filename, true, errp);
3768f88e1a42SJes Sorensen     if (!proto_drv) {
3769d92ada22SLuiz Capitulino         return;
3770f88e1a42SJes Sorensen     }
3771f88e1a42SJes Sorensen 
3772c6149724SMax Reitz     if (!drv->create_opts) {
3773c6149724SMax Reitz         error_setg(errp, "Format driver '%s' does not support image creation",
3774c6149724SMax Reitz                    drv->format_name);
3775c6149724SMax Reitz         return;
3776c6149724SMax Reitz     }
3777c6149724SMax Reitz 
3778c6149724SMax Reitz     if (!proto_drv->create_opts) {
3779c6149724SMax Reitz         error_setg(errp, "Protocol driver '%s' does not support image creation",
3780c6149724SMax Reitz                    proto_drv->format_name);
3781c6149724SMax Reitz         return;
3782c6149724SMax Reitz     }
3783c6149724SMax Reitz 
3784c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, drv->create_opts);
3785c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, proto_drv->create_opts);
3786f88e1a42SJes Sorensen 
3787f88e1a42SJes Sorensen     /* Create parameter list with default values */
378883d0521aSChunyan Liu     opts = qemu_opts_create(create_opts, NULL, 0, &error_abort);
378939101f25SMarkus Armbruster     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, img_size, &error_abort);
3790f88e1a42SJes Sorensen 
3791f88e1a42SJes Sorensen     /* Parse -o options */
3792f88e1a42SJes Sorensen     if (options) {
3793dc523cd3SMarkus Armbruster         qemu_opts_do_parse(opts, options, NULL, &local_err);
3794dc523cd3SMarkus Armbruster         if (local_err) {
3795dc523cd3SMarkus Armbruster             error_report_err(local_err);
3796dc523cd3SMarkus Armbruster             local_err = NULL;
379783d0521aSChunyan Liu             error_setg(errp, "Invalid options for file format '%s'", fmt);
3798f88e1a42SJes Sorensen             goto out;
3799f88e1a42SJes Sorensen         }
3800f88e1a42SJes Sorensen     }
3801f88e1a42SJes Sorensen 
3802f88e1a42SJes Sorensen     if (base_filename) {
3803f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, base_filename, &local_err);
38046be4194bSMarkus Armbruster         if (local_err) {
380571c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
380671c79813SLuiz Capitulino                        fmt);
3807f88e1a42SJes Sorensen             goto out;
3808f88e1a42SJes Sorensen         }
3809f88e1a42SJes Sorensen     }
3810f88e1a42SJes Sorensen 
3811f88e1a42SJes Sorensen     if (base_fmt) {
3812f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FMT, base_fmt, &local_err);
38136be4194bSMarkus Armbruster         if (local_err) {
381471c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
381571c79813SLuiz Capitulino                              "format '%s'", fmt);
3816f88e1a42SJes Sorensen             goto out;
3817f88e1a42SJes Sorensen         }
3818f88e1a42SJes Sorensen     }
3819f88e1a42SJes Sorensen 
382083d0521aSChunyan Liu     backing_file = qemu_opt_get(opts, BLOCK_OPT_BACKING_FILE);
382183d0521aSChunyan Liu     if (backing_file) {
382283d0521aSChunyan Liu         if (!strcmp(filename, backing_file)) {
382371c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
382471c79813SLuiz Capitulino                              "same filename as the backing file");
3825792da93aSJes Sorensen             goto out;
3826792da93aSJes Sorensen         }
3827792da93aSJes Sorensen     }
3828792da93aSJes Sorensen 
382983d0521aSChunyan Liu     backing_fmt = qemu_opt_get(opts, BLOCK_OPT_BACKING_FMT);
3830f88e1a42SJes Sorensen 
3831f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
3832f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
383383d0521aSChunyan Liu     size = qemu_opt_get_size(opts, BLOCK_OPT_SIZE, 0);
383483d0521aSChunyan Liu     if (size == -1) {
383583d0521aSChunyan Liu         if (backing_file) {
383666f6b814SMax Reitz             BlockDriverState *bs;
383729168018SMax Reitz             char *full_backing = g_new0(char, PATH_MAX);
383852bf1e72SMarkus Armbruster             int64_t size;
383963090dacSPaolo Bonzini             int back_flags;
3840e6641719SMax Reitz             QDict *backing_options = NULL;
384163090dacSPaolo Bonzini 
384229168018SMax Reitz             bdrv_get_full_backing_filename_from_filename(filename, backing_file,
384329168018SMax Reitz                                                          full_backing, PATH_MAX,
384429168018SMax Reitz                                                          &local_err);
384529168018SMax Reitz             if (local_err) {
384629168018SMax Reitz                 g_free(full_backing);
384729168018SMax Reitz                 goto out;
384829168018SMax Reitz             }
384929168018SMax Reitz 
385063090dacSPaolo Bonzini             /* backing files always opened read-only */
385163090dacSPaolo Bonzini             back_flags =
385263090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
3853f88e1a42SJes Sorensen 
3854e6641719SMax Reitz             if (backing_fmt) {
3855e6641719SMax Reitz                 backing_options = qdict_new();
3856e6641719SMax Reitz                 qdict_put(backing_options, "driver",
3857e6641719SMax Reitz                           qstring_from_str(backing_fmt));
3858e6641719SMax Reitz             }
3859e6641719SMax Reitz 
3860f67503e5SMax Reitz             bs = NULL;
3861e6641719SMax Reitz             ret = bdrv_open(&bs, full_backing, NULL, backing_options,
38626ebf9aa2SMax Reitz                             back_flags, &local_err);
386329168018SMax Reitz             g_free(full_backing);
3864f88e1a42SJes Sorensen             if (ret < 0) {
3865f88e1a42SJes Sorensen                 goto out;
3866f88e1a42SJes Sorensen             }
386752bf1e72SMarkus Armbruster             size = bdrv_getlength(bs);
386852bf1e72SMarkus Armbruster             if (size < 0) {
386952bf1e72SMarkus Armbruster                 error_setg_errno(errp, -size, "Could not get size of '%s'",
387052bf1e72SMarkus Armbruster                                  backing_file);
387152bf1e72SMarkus Armbruster                 bdrv_unref(bs);
387252bf1e72SMarkus Armbruster                 goto out;
387352bf1e72SMarkus Armbruster             }
3874f88e1a42SJes Sorensen 
387539101f25SMarkus Armbruster             qemu_opt_set_number(opts, BLOCK_OPT_SIZE, size, &error_abort);
387666f6b814SMax Reitz 
387766f6b814SMax Reitz             bdrv_unref(bs);
3878f88e1a42SJes Sorensen         } else {
387971c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
3880f88e1a42SJes Sorensen             goto out;
3881f88e1a42SJes Sorensen         }
3882f88e1a42SJes Sorensen     }
3883f88e1a42SJes Sorensen 
3884f382d43aSMiroslav Rezanina     if (!quiet) {
3885f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
388643c5d8f8SFam Zheng         qemu_opts_print(opts, " ");
3887f88e1a42SJes Sorensen         puts("");
3888f382d43aSMiroslav Rezanina     }
388983d0521aSChunyan Liu 
3890c282e1fdSChunyan Liu     ret = bdrv_create(drv, filename, opts, &local_err);
389183d0521aSChunyan Liu 
3892cc84d90fSMax Reitz     if (ret == -EFBIG) {
3893cc84d90fSMax Reitz         /* This is generally a better message than whatever the driver would
3894cc84d90fSMax Reitz          * deliver (especially because of the cluster_size_hint), since that
3895cc84d90fSMax Reitz          * is most probably not much different from "image too large". */
3896f3f4d2c0SKevin Wolf         const char *cluster_size_hint = "";
389783d0521aSChunyan Liu         if (qemu_opt_get_size(opts, BLOCK_OPT_CLUSTER_SIZE, 0)) {
3898f3f4d2c0SKevin Wolf             cluster_size_hint = " (try using a larger cluster size)";
3899f3f4d2c0SKevin Wolf         }
3900cc84d90fSMax Reitz         error_setg(errp, "The image size is too large for file format '%s'"
3901cc84d90fSMax Reitz                    "%s", fmt, cluster_size_hint);
3902cc84d90fSMax Reitz         error_free(local_err);
3903cc84d90fSMax Reitz         local_err = NULL;
3904f88e1a42SJes Sorensen     }
3905f88e1a42SJes Sorensen 
3906f88e1a42SJes Sorensen out:
390783d0521aSChunyan Liu     qemu_opts_del(opts);
390883d0521aSChunyan Liu     qemu_opts_free(create_opts);
390984d18f06SMarkus Armbruster     if (local_err) {
3910cc84d90fSMax Reitz         error_propagate(errp, local_err);
3911cc84d90fSMax Reitz     }
3912f88e1a42SJes Sorensen }
391385d126f3SStefan Hajnoczi 
391485d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
391585d126f3SStefan Hajnoczi {
3916dcd04228SStefan Hajnoczi     return bs->aio_context;
3917dcd04228SStefan Hajnoczi }
3918dcd04228SStefan Hajnoczi 
3919dcd04228SStefan Hajnoczi void bdrv_detach_aio_context(BlockDriverState *bs)
3920dcd04228SStefan Hajnoczi {
392133384421SMax Reitz     BdrvAioNotifier *baf;
392233384421SMax Reitz 
3923dcd04228SStefan Hajnoczi     if (!bs->drv) {
3924dcd04228SStefan Hajnoczi         return;
3925dcd04228SStefan Hajnoczi     }
3926dcd04228SStefan Hajnoczi 
392733384421SMax Reitz     QLIST_FOREACH(baf, &bs->aio_notifiers, list) {
392833384421SMax Reitz         baf->detach_aio_context(baf->opaque);
392933384421SMax Reitz     }
393033384421SMax Reitz 
393113af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39320e5b0a2dSBenoît Canet         throttle_timers_detach_aio_context(&bs->throttle_timers);
393313af91ebSStefan Hajnoczi     }
3934dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_detach_aio_context) {
3935dcd04228SStefan Hajnoczi         bs->drv->bdrv_detach_aio_context(bs);
3936dcd04228SStefan Hajnoczi     }
3937dcd04228SStefan Hajnoczi     if (bs->file) {
3938*9a4f4c31SKevin Wolf         bdrv_detach_aio_context(bs->file->bs);
3939dcd04228SStefan Hajnoczi     }
3940dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3941dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->backing_hd);
3942dcd04228SStefan Hajnoczi     }
3943dcd04228SStefan Hajnoczi 
3944dcd04228SStefan Hajnoczi     bs->aio_context = NULL;
3945dcd04228SStefan Hajnoczi }
3946dcd04228SStefan Hajnoczi 
3947dcd04228SStefan Hajnoczi void bdrv_attach_aio_context(BlockDriverState *bs,
3948dcd04228SStefan Hajnoczi                              AioContext *new_context)
3949dcd04228SStefan Hajnoczi {
395033384421SMax Reitz     BdrvAioNotifier *ban;
395133384421SMax Reitz 
3952dcd04228SStefan Hajnoczi     if (!bs->drv) {
3953dcd04228SStefan Hajnoczi         return;
3954dcd04228SStefan Hajnoczi     }
3955dcd04228SStefan Hajnoczi 
3956dcd04228SStefan Hajnoczi     bs->aio_context = new_context;
3957dcd04228SStefan Hajnoczi 
3958dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3959dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->backing_hd, new_context);
3960dcd04228SStefan Hajnoczi     }
3961dcd04228SStefan Hajnoczi     if (bs->file) {
3962*9a4f4c31SKevin Wolf         bdrv_attach_aio_context(bs->file->bs, new_context);
3963dcd04228SStefan Hajnoczi     }
3964dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_attach_aio_context) {
3965dcd04228SStefan Hajnoczi         bs->drv->bdrv_attach_aio_context(bs, new_context);
3966dcd04228SStefan Hajnoczi     }
396713af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39680e5b0a2dSBenoît Canet         throttle_timers_attach_aio_context(&bs->throttle_timers, new_context);
396913af91ebSStefan Hajnoczi     }
397033384421SMax Reitz 
397133384421SMax Reitz     QLIST_FOREACH(ban, &bs->aio_notifiers, list) {
397233384421SMax Reitz         ban->attached_aio_context(new_context, ban->opaque);
397333384421SMax Reitz     }
3974dcd04228SStefan Hajnoczi }
3975dcd04228SStefan Hajnoczi 
3976dcd04228SStefan Hajnoczi void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
3977dcd04228SStefan Hajnoczi {
397853ec73e2SFam Zheng     bdrv_drain(bs); /* ensure there are no in-flight requests */
3979dcd04228SStefan Hajnoczi 
3980dcd04228SStefan Hajnoczi     bdrv_detach_aio_context(bs);
3981dcd04228SStefan Hajnoczi 
3982dcd04228SStefan Hajnoczi     /* This function executes in the old AioContext so acquire the new one in
3983dcd04228SStefan Hajnoczi      * case it runs in a different thread.
3984dcd04228SStefan Hajnoczi      */
3985dcd04228SStefan Hajnoczi     aio_context_acquire(new_context);
3986dcd04228SStefan Hajnoczi     bdrv_attach_aio_context(bs, new_context);
3987dcd04228SStefan Hajnoczi     aio_context_release(new_context);
398885d126f3SStefan Hajnoczi }
3989d616b224SStefan Hajnoczi 
399033384421SMax Reitz void bdrv_add_aio_context_notifier(BlockDriverState *bs,
399133384421SMax Reitz         void (*attached_aio_context)(AioContext *new_context, void *opaque),
399233384421SMax Reitz         void (*detach_aio_context)(void *opaque), void *opaque)
399333384421SMax Reitz {
399433384421SMax Reitz     BdrvAioNotifier *ban = g_new(BdrvAioNotifier, 1);
399533384421SMax Reitz     *ban = (BdrvAioNotifier){
399633384421SMax Reitz         .attached_aio_context = attached_aio_context,
399733384421SMax Reitz         .detach_aio_context   = detach_aio_context,
399833384421SMax Reitz         .opaque               = opaque
399933384421SMax Reitz     };
400033384421SMax Reitz 
400133384421SMax Reitz     QLIST_INSERT_HEAD(&bs->aio_notifiers, ban, list);
400233384421SMax Reitz }
400333384421SMax Reitz 
400433384421SMax Reitz void bdrv_remove_aio_context_notifier(BlockDriverState *bs,
400533384421SMax Reitz                                       void (*attached_aio_context)(AioContext *,
400633384421SMax Reitz                                                                    void *),
400733384421SMax Reitz                                       void (*detach_aio_context)(void *),
400833384421SMax Reitz                                       void *opaque)
400933384421SMax Reitz {
401033384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
401133384421SMax Reitz 
401233384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
401333384421SMax Reitz         if (ban->attached_aio_context == attached_aio_context &&
401433384421SMax Reitz             ban->detach_aio_context   == detach_aio_context   &&
401533384421SMax Reitz             ban->opaque               == opaque)
401633384421SMax Reitz         {
401733384421SMax Reitz             QLIST_REMOVE(ban, list);
401833384421SMax Reitz             g_free(ban);
401933384421SMax Reitz 
402033384421SMax Reitz             return;
402133384421SMax Reitz         }
402233384421SMax Reitz     }
402333384421SMax Reitz 
402433384421SMax Reitz     abort();
402533384421SMax Reitz }
402633384421SMax Reitz 
402777485434SMax Reitz int bdrv_amend_options(BlockDriverState *bs, QemuOpts *opts,
402877485434SMax Reitz                        BlockDriverAmendStatusCB *status_cb)
40296f176b48SMax Reitz {
4030c282e1fdSChunyan Liu     if (!bs->drv->bdrv_amend_options) {
40316f176b48SMax Reitz         return -ENOTSUP;
40326f176b48SMax Reitz     }
403377485434SMax Reitz     return bs->drv->bdrv_amend_options(bs, opts, status_cb);
40346f176b48SMax Reitz }
4035f6186f49SBenoît Canet 
4036b5042a36SBenoît Canet /* This function will be called by the bdrv_recurse_is_first_non_filter method
4037b5042a36SBenoît Canet  * of block filter and by bdrv_is_first_non_filter.
4038b5042a36SBenoît Canet  * It is used to test if the given bs is the candidate or recurse more in the
4039b5042a36SBenoît Canet  * node graph.
4040212a5a8fSBenoît Canet  */
4041212a5a8fSBenoît Canet bool bdrv_recurse_is_first_non_filter(BlockDriverState *bs,
4042212a5a8fSBenoît Canet                                       BlockDriverState *candidate)
4043f6186f49SBenoît Canet {
4044b5042a36SBenoît Canet     /* return false if basic checks fails */
4045b5042a36SBenoît Canet     if (!bs || !bs->drv) {
4046b5042a36SBenoît Canet         return false;
4047b5042a36SBenoît Canet     }
4048b5042a36SBenoît Canet 
4049b5042a36SBenoît Canet     /* the code reached a non block filter driver -> check if the bs is
4050b5042a36SBenoît Canet      * the same as the candidate. It's the recursion termination condition.
4051b5042a36SBenoît Canet      */
4052b5042a36SBenoît Canet     if (!bs->drv->is_filter) {
4053b5042a36SBenoît Canet         return bs == candidate;
4054b5042a36SBenoît Canet     }
4055b5042a36SBenoît Canet     /* Down this path the driver is a block filter driver */
4056b5042a36SBenoît Canet 
4057b5042a36SBenoît Canet     /* If the block filter recursion method is defined use it to recurse down
4058b5042a36SBenoît Canet      * the node graph.
4059b5042a36SBenoît Canet      */
4060b5042a36SBenoît Canet     if (bs->drv->bdrv_recurse_is_first_non_filter) {
4061212a5a8fSBenoît Canet         return bs->drv->bdrv_recurse_is_first_non_filter(bs, candidate);
4062212a5a8fSBenoît Canet     }
4063212a5a8fSBenoît Canet 
4064b5042a36SBenoît Canet     /* the driver is a block filter but don't allow to recurse -> return false
4065b5042a36SBenoît Canet      */
4066b5042a36SBenoît Canet     return false;
4067212a5a8fSBenoît Canet }
4068212a5a8fSBenoît Canet 
4069212a5a8fSBenoît Canet /* This function checks if the candidate is the first non filter bs down it's
4070212a5a8fSBenoît Canet  * bs chain. Since we don't have pointers to parents it explore all bs chains
4071212a5a8fSBenoît Canet  * from the top. Some filters can choose not to pass down the recursion.
4072212a5a8fSBenoît Canet  */
4073212a5a8fSBenoît Canet bool bdrv_is_first_non_filter(BlockDriverState *candidate)
4074212a5a8fSBenoît Canet {
4075212a5a8fSBenoît Canet     BlockDriverState *bs;
4076212a5a8fSBenoît Canet 
4077212a5a8fSBenoît Canet     /* walk down the bs forest recursively */
4078212a5a8fSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
4079212a5a8fSBenoît Canet         bool perm;
4080212a5a8fSBenoît Canet 
4081b5042a36SBenoît Canet         /* try to recurse in this top level bs */
4082e6dc8a1fSKevin Wolf         perm = bdrv_recurse_is_first_non_filter(bs, candidate);
4083212a5a8fSBenoît Canet 
4084212a5a8fSBenoît Canet         /* candidate is the first non filter */
4085212a5a8fSBenoît Canet         if (perm) {
4086212a5a8fSBenoît Canet             return true;
4087212a5a8fSBenoît Canet         }
4088212a5a8fSBenoît Canet     }
4089212a5a8fSBenoît Canet 
4090212a5a8fSBenoît Canet     return false;
4091f6186f49SBenoît Canet }
409209158f00SBenoît Canet 
4093e12f3784SWen Congyang BlockDriverState *check_to_replace_node(BlockDriverState *parent_bs,
4094e12f3784SWen Congyang                                         const char *node_name, Error **errp)
409509158f00SBenoît Canet {
409609158f00SBenoît Canet     BlockDriverState *to_replace_bs = bdrv_find_node(node_name);
40975a7e7a0bSStefan Hajnoczi     AioContext *aio_context;
40985a7e7a0bSStefan Hajnoczi 
409909158f00SBenoît Canet     if (!to_replace_bs) {
410009158f00SBenoît Canet         error_setg(errp, "Node name '%s' not found", node_name);
410109158f00SBenoît Canet         return NULL;
410209158f00SBenoît Canet     }
410309158f00SBenoît Canet 
41045a7e7a0bSStefan Hajnoczi     aio_context = bdrv_get_aio_context(to_replace_bs);
41055a7e7a0bSStefan Hajnoczi     aio_context_acquire(aio_context);
41065a7e7a0bSStefan Hajnoczi 
410709158f00SBenoît Canet     if (bdrv_op_is_blocked(to_replace_bs, BLOCK_OP_TYPE_REPLACE, errp)) {
41085a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41095a7e7a0bSStefan Hajnoczi         goto out;
411009158f00SBenoît Canet     }
411109158f00SBenoît Canet 
411209158f00SBenoît Canet     /* We don't want arbitrary node of the BDS chain to be replaced only the top
411309158f00SBenoît Canet      * most non filter in order to prevent data corruption.
411409158f00SBenoît Canet      * Another benefit is that this tests exclude backing files which are
411509158f00SBenoît Canet      * blocked by the backing blockers.
411609158f00SBenoît Canet      */
4117e12f3784SWen Congyang     if (!bdrv_recurse_is_first_non_filter(parent_bs, to_replace_bs)) {
411809158f00SBenoît Canet         error_setg(errp, "Only top most non filter can be replaced");
41195a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41205a7e7a0bSStefan Hajnoczi         goto out;
412109158f00SBenoît Canet     }
412209158f00SBenoît Canet 
41235a7e7a0bSStefan Hajnoczi out:
41245a7e7a0bSStefan Hajnoczi     aio_context_release(aio_context);
412509158f00SBenoît Canet     return to_replace_bs;
412609158f00SBenoît Canet }
4127448ad91dSMing Lei 
412891af7014SMax Reitz static bool append_open_options(QDict *d, BlockDriverState *bs)
412991af7014SMax Reitz {
413091af7014SMax Reitz     const QDictEntry *entry;
413191af7014SMax Reitz     bool found_any = false;
413291af7014SMax Reitz 
413391af7014SMax Reitz     for (entry = qdict_first(bs->options); entry;
413491af7014SMax Reitz          entry = qdict_next(bs->options, entry))
413591af7014SMax Reitz     {
413691af7014SMax Reitz         /* Only take options for this level and exclude all non-driver-specific
413791af7014SMax Reitz          * options */
413891af7014SMax Reitz         if (!strchr(qdict_entry_key(entry), '.') &&
413991af7014SMax Reitz             strcmp(qdict_entry_key(entry), "node-name"))
414091af7014SMax Reitz         {
414191af7014SMax Reitz             qobject_incref(qdict_entry_value(entry));
414291af7014SMax Reitz             qdict_put_obj(d, qdict_entry_key(entry), qdict_entry_value(entry));
414391af7014SMax Reitz             found_any = true;
414491af7014SMax Reitz         }
414591af7014SMax Reitz     }
414691af7014SMax Reitz 
414791af7014SMax Reitz     return found_any;
414891af7014SMax Reitz }
414991af7014SMax Reitz 
415091af7014SMax Reitz /* Updates the following BDS fields:
415191af7014SMax Reitz  *  - exact_filename: A filename which may be used for opening a block device
415291af7014SMax Reitz  *                    which (mostly) equals the given BDS (even without any
415391af7014SMax Reitz  *                    other options; so reading and writing must return the same
415491af7014SMax Reitz  *                    results, but caching etc. may be different)
415591af7014SMax Reitz  *  - full_open_options: Options which, when given when opening a block device
415691af7014SMax Reitz  *                       (without a filename), result in a BDS (mostly)
415791af7014SMax Reitz  *                       equalling the given one
415891af7014SMax Reitz  *  - filename: If exact_filename is set, it is copied here. Otherwise,
415991af7014SMax Reitz  *              full_open_options is converted to a JSON object, prefixed with
416091af7014SMax Reitz  *              "json:" (for use through the JSON pseudo protocol) and put here.
416191af7014SMax Reitz  */
416291af7014SMax Reitz void bdrv_refresh_filename(BlockDriverState *bs)
416391af7014SMax Reitz {
416491af7014SMax Reitz     BlockDriver *drv = bs->drv;
416591af7014SMax Reitz     QDict *opts;
416691af7014SMax Reitz 
416791af7014SMax Reitz     if (!drv) {
416891af7014SMax Reitz         return;
416991af7014SMax Reitz     }
417091af7014SMax Reitz 
417191af7014SMax Reitz     /* This BDS's file name will most probably depend on its file's name, so
417291af7014SMax Reitz      * refresh that first */
417391af7014SMax Reitz     if (bs->file) {
4174*9a4f4c31SKevin Wolf         bdrv_refresh_filename(bs->file->bs);
417591af7014SMax Reitz     }
417691af7014SMax Reitz 
417791af7014SMax Reitz     if (drv->bdrv_refresh_filename) {
417891af7014SMax Reitz         /* Obsolete information is of no use here, so drop the old file name
417991af7014SMax Reitz          * information before refreshing it */
418091af7014SMax Reitz         bs->exact_filename[0] = '\0';
418191af7014SMax Reitz         if (bs->full_open_options) {
418291af7014SMax Reitz             QDECREF(bs->full_open_options);
418391af7014SMax Reitz             bs->full_open_options = NULL;
418491af7014SMax Reitz         }
418591af7014SMax Reitz 
418691af7014SMax Reitz         drv->bdrv_refresh_filename(bs);
418791af7014SMax Reitz     } else if (bs->file) {
418891af7014SMax Reitz         /* Try to reconstruct valid information from the underlying file */
418991af7014SMax Reitz         bool has_open_options;
419091af7014SMax Reitz 
419191af7014SMax Reitz         bs->exact_filename[0] = '\0';
419291af7014SMax Reitz         if (bs->full_open_options) {
419391af7014SMax Reitz             QDECREF(bs->full_open_options);
419491af7014SMax Reitz             bs->full_open_options = NULL;
419591af7014SMax Reitz         }
419691af7014SMax Reitz 
419791af7014SMax Reitz         opts = qdict_new();
419891af7014SMax Reitz         has_open_options = append_open_options(opts, bs);
419991af7014SMax Reitz 
420091af7014SMax Reitz         /* If no specific options have been given for this BDS, the filename of
420191af7014SMax Reitz          * the underlying file should suffice for this one as well */
4202*9a4f4c31SKevin Wolf         if (bs->file->bs->exact_filename[0] && !has_open_options) {
4203*9a4f4c31SKevin Wolf             strcpy(bs->exact_filename, bs->file->bs->exact_filename);
420491af7014SMax Reitz         }
420591af7014SMax Reitz         /* Reconstructing the full options QDict is simple for most format block
420691af7014SMax Reitz          * drivers, as long as the full options are known for the underlying
420791af7014SMax Reitz          * file BDS. The full options QDict of that file BDS should somehow
420891af7014SMax Reitz          * contain a representation of the filename, therefore the following
420991af7014SMax Reitz          * suffices without querying the (exact_)filename of this BDS. */
4210*9a4f4c31SKevin Wolf         if (bs->file->bs->full_open_options) {
421191af7014SMax Reitz             qdict_put_obj(opts, "driver",
421291af7014SMax Reitz                           QOBJECT(qstring_from_str(drv->format_name)));
4213*9a4f4c31SKevin Wolf             QINCREF(bs->file->bs->full_open_options);
4214*9a4f4c31SKevin Wolf             qdict_put_obj(opts, "file",
4215*9a4f4c31SKevin Wolf                           QOBJECT(bs->file->bs->full_open_options));
421691af7014SMax Reitz 
421791af7014SMax Reitz             bs->full_open_options = opts;
421891af7014SMax Reitz         } else {
421991af7014SMax Reitz             QDECREF(opts);
422091af7014SMax Reitz         }
422191af7014SMax Reitz     } else if (!bs->full_open_options && qdict_size(bs->options)) {
422291af7014SMax Reitz         /* There is no underlying file BDS (at least referenced by BDS.file),
422391af7014SMax Reitz          * so the full options QDict should be equal to the options given
422491af7014SMax Reitz          * specifically for this block device when it was opened (plus the
422591af7014SMax Reitz          * driver specification).
422691af7014SMax Reitz          * Because those options don't change, there is no need to update
422791af7014SMax Reitz          * full_open_options when it's already set. */
422891af7014SMax Reitz 
422991af7014SMax Reitz         opts = qdict_new();
423091af7014SMax Reitz         append_open_options(opts, bs);
423191af7014SMax Reitz         qdict_put_obj(opts, "driver",
423291af7014SMax Reitz                       QOBJECT(qstring_from_str(drv->format_name)));
423391af7014SMax Reitz 
423491af7014SMax Reitz         if (bs->exact_filename[0]) {
423591af7014SMax Reitz             /* This may not work for all block protocol drivers (some may
423691af7014SMax Reitz              * require this filename to be parsed), but we have to find some
423791af7014SMax Reitz              * default solution here, so just include it. If some block driver
423891af7014SMax Reitz              * does not support pure options without any filename at all or
423991af7014SMax Reitz              * needs some special format of the options QDict, it needs to
424091af7014SMax Reitz              * implement the driver-specific bdrv_refresh_filename() function.
424191af7014SMax Reitz              */
424291af7014SMax Reitz             qdict_put_obj(opts, "filename",
424391af7014SMax Reitz                           QOBJECT(qstring_from_str(bs->exact_filename)));
424491af7014SMax Reitz         }
424591af7014SMax Reitz 
424691af7014SMax Reitz         bs->full_open_options = opts;
424791af7014SMax Reitz     }
424891af7014SMax Reitz 
424991af7014SMax Reitz     if (bs->exact_filename[0]) {
425091af7014SMax Reitz         pstrcpy(bs->filename, sizeof(bs->filename), bs->exact_filename);
425191af7014SMax Reitz     } else if (bs->full_open_options) {
425291af7014SMax Reitz         QString *json = qobject_to_json(QOBJECT(bs->full_open_options));
425391af7014SMax Reitz         snprintf(bs->filename, sizeof(bs->filename), "json:%s",
425491af7014SMax Reitz                  qstring_get_str(json));
425591af7014SMax Reitz         QDECREF(json);
425691af7014SMax Reitz     }
425791af7014SMax Reitz }
42585366d0c8SBenoît Canet 
42595366d0c8SBenoît Canet /* This accessor function purpose is to allow the device models to access the
42605366d0c8SBenoît Canet  * BlockAcctStats structure embedded inside a BlockDriverState without being
42615366d0c8SBenoît Canet  * aware of the BlockDriverState structure layout.
42625366d0c8SBenoît Canet  * It will go away when the BlockAcctStats structure will be moved inside
42635366d0c8SBenoît Canet  * the device models.
42645366d0c8SBenoît Canet  */
42655366d0c8SBenoît Canet BlockAcctStats *bdrv_get_stats(BlockDriverState *bs)
42665366d0c8SBenoît Canet {
42675366d0c8SBenoît Canet     return &bs->stats;
42685366d0c8SBenoît Canet }
4269