xref: /openbmc/qemu/block.c (revision 1fdd69330872ef91d92482472eef79350d2f379b)
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  */
812f500a6d3SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *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) {
82645673671SKevin Wolf         filename = file->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;
1404f67503e5SMax Reitz     BlockDriverState *file = NULL, *bs;
1405ce343771SMax Reitz     BlockDriver *drv = NULL;
140674fe54f2SKevin Wolf     const char *drvname;
140734b5d2c6SMax Reitz     Error *local_err = NULL;
1408b1e6fc08SKevin Wolf     int snapshot_flags = 0;
140933e3963eSbellard 
1410f67503e5SMax Reitz     assert(pbs);
1411f3930ed0SKevin Wolf     assert(!child_role || !flags);
1412f3930ed0SKevin Wolf     assert(!child_role == !parent);
1413f67503e5SMax Reitz 
1414ddf5636dSMax Reitz     if (reference) {
1415ddf5636dSMax Reitz         bool options_non_empty = options ? qdict_size(options) : false;
1416ddf5636dSMax Reitz         QDECREF(options);
1417ddf5636dSMax Reitz 
1418ddf5636dSMax Reitz         if (*pbs) {
1419ddf5636dSMax Reitz             error_setg(errp, "Cannot reuse an existing BDS when referencing "
1420ddf5636dSMax Reitz                        "another block device");
1421ddf5636dSMax Reitz             return -EINVAL;
1422ddf5636dSMax Reitz         }
1423ddf5636dSMax Reitz 
1424ddf5636dSMax Reitz         if (filename || options_non_empty) {
1425ddf5636dSMax Reitz             error_setg(errp, "Cannot reference an existing block device with "
1426ddf5636dSMax Reitz                        "additional options or a new filename");
1427ddf5636dSMax Reitz             return -EINVAL;
1428ddf5636dSMax Reitz         }
1429ddf5636dSMax Reitz 
1430ddf5636dSMax Reitz         bs = bdrv_lookup_bs(reference, reference, errp);
1431ddf5636dSMax Reitz         if (!bs) {
1432ddf5636dSMax Reitz             return -ENODEV;
1433ddf5636dSMax Reitz         }
1434ddf5636dSMax Reitz         bdrv_ref(bs);
1435ddf5636dSMax Reitz         *pbs = bs;
1436ddf5636dSMax Reitz         return 0;
1437ddf5636dSMax Reitz     }
1438ddf5636dSMax Reitz 
1439f67503e5SMax Reitz     if (*pbs) {
1440f67503e5SMax Reitz         bs = *pbs;
1441f67503e5SMax Reitz     } else {
1442e4e9986bSMarkus Armbruster         bs = bdrv_new();
1443f67503e5SMax Reitz     }
1444f67503e5SMax Reitz 
1445de9c0cecSKevin Wolf     /* NULL means an empty set of options */
1446de9c0cecSKevin Wolf     if (options == NULL) {
1447de9c0cecSKevin Wolf         options = qdict_new();
1448de9c0cecSKevin Wolf     }
1449de9c0cecSKevin Wolf 
1450f3930ed0SKevin Wolf     if (child_role) {
1451bddcec37SKevin Wolf         bs->inherits_from = parent;
1452f3930ed0SKevin Wolf         flags = child_role->inherit_flags(parent->open_flags);
1453f3930ed0SKevin Wolf     }
1454f3930ed0SKevin Wolf 
1455053e1578SMax Reitz     ret = bdrv_fill_options(&options, &filename, &flags, &local_err);
1456462f5bcfSKevin Wolf     if (local_err) {
1457462f5bcfSKevin Wolf         goto fail;
1458462f5bcfSKevin Wolf     }
1459462f5bcfSKevin Wolf 
146076c591b0SKevin Wolf     /* Find the right image format driver */
146176c591b0SKevin Wolf     drvname = qdict_get_try_str(options, "driver");
146276c591b0SKevin Wolf     if (drvname) {
146376c591b0SKevin Wolf         drv = bdrv_find_format(drvname);
146476c591b0SKevin Wolf         qdict_del(options, "driver");
146576c591b0SKevin Wolf         if (!drv) {
146676c591b0SKevin Wolf             error_setg(errp, "Unknown driver: '%s'", drvname);
146776c591b0SKevin Wolf             ret = -EINVAL;
146876c591b0SKevin Wolf             goto fail;
146976c591b0SKevin Wolf         }
147076c591b0SKevin Wolf     }
147176c591b0SKevin Wolf 
147276c591b0SKevin Wolf     assert(drvname || !(flags & BDRV_O_PROTOCOL));
147376c591b0SKevin Wolf 
1474f3930ed0SKevin Wolf     bs->open_flags = flags;
1475de9c0cecSKevin Wolf     bs->options = options;
1476b6ad491aSKevin Wolf     options = qdict_clone_shallow(options);
1477de9c0cecSKevin Wolf 
1478f500a6d3SKevin Wolf     /* Open image file without format layer */
1479f4788adcSKevin Wolf     if ((flags & BDRV_O_PROTOCOL) == 0) {
1480be028adcSJeff Cody         if (flags & BDRV_O_RDWR) {
1481be028adcSJeff Cody             flags |= BDRV_O_ALLOW_RDWR;
1482be028adcSJeff Cody         }
1483b1e6fc08SKevin Wolf         if (flags & BDRV_O_SNAPSHOT) {
1484b1e6fc08SKevin Wolf             snapshot_flags = bdrv_temp_snapshot_flags(flags);
1485b1e6fc08SKevin Wolf             flags = bdrv_backing_flags(flags);
1486b1e6fc08SKevin Wolf         }
1487be028adcSJeff Cody 
1488f67503e5SMax Reitz         assert(file == NULL);
1489f3930ed0SKevin Wolf         bs->open_flags = flags;
1490*1fdd6933SKevin Wolf 
1491*1fdd6933SKevin Wolf         bs->file_child = bdrv_open_child(filename, options, "file", bs,
1492*1fdd6933SKevin Wolf                                          &child_file, true, &local_err);
1493*1fdd6933SKevin Wolf         if (local_err) {
1494*1fdd6933SKevin Wolf             ret = -EINVAL;
14958bfea15dSKevin Wolf             goto fail;
1496f500a6d3SKevin Wolf         }
1497*1fdd6933SKevin Wolf 
1498*1fdd6933SKevin Wolf         if (bs->file_child) {
1499*1fdd6933SKevin Wolf             file = bs->file_child->bs;
1500*1fdd6933SKevin Wolf         }
1501f4788adcSKevin Wolf     }
1502f500a6d3SKevin Wolf 
150376c591b0SKevin Wolf     /* Image format probing */
150438f3ef57SKevin Wolf     bs->probed = !drv;
150576c591b0SKevin Wolf     if (!drv && file) {
150634b5d2c6SMax Reitz         ret = find_image_format(file, filename, &drv, &local_err);
150717b005f1SKevin Wolf         if (ret < 0) {
150817b005f1SKevin Wolf             goto fail;
150917b005f1SKevin Wolf         }
151076c591b0SKevin Wolf     } else if (!drv) {
15112a05cbe4SMax Reitz         error_setg(errp, "Must specify either driver or file");
15122a05cbe4SMax Reitz         ret = -EINVAL;
15138bfea15dSKevin Wolf         goto fail;
15142a05cbe4SMax Reitz     }
1515f500a6d3SKevin Wolf 
151653a29513SMax Reitz     /* BDRV_O_PROTOCOL must be set iff a protocol BDS is about to be created */
151753a29513SMax Reitz     assert(!!(flags & BDRV_O_PROTOCOL) == !!drv->bdrv_file_open);
151853a29513SMax Reitz     /* file must be NULL if a protocol BDS is about to be created
151953a29513SMax Reitz      * (the inverse results in an error message from bdrv_open_common()) */
152053a29513SMax Reitz     assert(!(flags & BDRV_O_PROTOCOL) || !file);
152153a29513SMax Reitz 
1522b6ce07aaSKevin Wolf     /* Open the image */
152334b5d2c6SMax Reitz     ret = bdrv_open_common(bs, file, options, flags, drv, &local_err);
1524b6ce07aaSKevin Wolf     if (ret < 0) {
15258bfea15dSKevin Wolf         goto fail;
15266987307cSChristoph Hellwig     }
15276987307cSChristoph Hellwig 
15282a05cbe4SMax Reitz     if (file && (bs->file != file)) {
15294f6fd349SFam Zheng         bdrv_unref(file);
1530f500a6d3SKevin Wolf         file = NULL;
1531f500a6d3SKevin Wolf     }
1532f500a6d3SKevin Wolf 
1533b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
15349156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
153531ca6d07SKevin Wolf         QDict *backing_options;
153631ca6d07SKevin Wolf 
15375726d872SBenoît Canet         qdict_extract_subqdict(options, &backing_options, "backing.");
153834b5d2c6SMax Reitz         ret = bdrv_open_backing_file(bs, backing_options, &local_err);
1539b6ce07aaSKevin Wolf         if (ret < 0) {
1540b6ad491aSKevin Wolf             goto close_and_fail;
1541b6ce07aaSKevin Wolf         }
1542b6ce07aaSKevin Wolf     }
1543b6ce07aaSKevin Wolf 
154491af7014SMax Reitz     bdrv_refresh_filename(bs);
154591af7014SMax Reitz 
1546b998875dSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay. bs points to the
1547b998875dSKevin Wolf      * temporary snapshot afterwards. */
1548b1e6fc08SKevin Wolf     if (snapshot_flags) {
15496b8aeca5SChen Gang         ret = bdrv_append_temp_snapshot(bs, snapshot_flags, &local_err);
1550b998875dSKevin Wolf         if (local_err) {
1551b998875dSKevin Wolf             goto close_and_fail;
1552b998875dSKevin Wolf         }
1553b998875dSKevin Wolf     }
1554b998875dSKevin Wolf 
1555b6ad491aSKevin Wolf     /* Check if any unknown options were used */
15565acd9d81SMax Reitz     if (options && (qdict_size(options) != 0)) {
1557b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
15585acd9d81SMax Reitz         if (flags & BDRV_O_PROTOCOL) {
15595acd9d81SMax Reitz             error_setg(errp, "Block protocol '%s' doesn't support the option "
15605acd9d81SMax Reitz                        "'%s'", drv->format_name, entry->key);
15615acd9d81SMax Reitz         } else {
156234b5d2c6SMax Reitz             error_setg(errp, "Block format '%s' used by device '%s' doesn't "
15635acd9d81SMax Reitz                        "support the option '%s'", drv->format_name,
1564bfb197e0SMarkus Armbruster                        bdrv_get_device_name(bs), entry->key);
15655acd9d81SMax Reitz         }
1566b6ad491aSKevin Wolf 
1567b6ad491aSKevin Wolf         ret = -EINVAL;
1568b6ad491aSKevin Wolf         goto close_and_fail;
1569b6ad491aSKevin Wolf     }
1570b6ad491aSKevin Wolf 
1571b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
1572a7f53e26SMarkus Armbruster         if (bs->blk) {
1573a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
1574a7f53e26SMarkus Armbruster         }
1575c3adb58fSMarkus Armbruster     } else if (!runstate_check(RUN_STATE_PRELAUNCH)
1576c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_INMIGRATE)
1577c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_PAUSED)) { /* HACK */
1578c3adb58fSMarkus Armbruster         error_setg(errp,
1579c3adb58fSMarkus Armbruster                    "Guest must be stopped for opening of encrypted image");
1580c3adb58fSMarkus Armbruster         ret = -EBUSY;
1581c3adb58fSMarkus Armbruster         goto close_and_fail;
1582b6ce07aaSKevin Wolf     }
1583b6ce07aaSKevin Wolf 
1584c3adb58fSMarkus Armbruster     QDECREF(options);
1585f67503e5SMax Reitz     *pbs = bs;
1586b6ce07aaSKevin Wolf     return 0;
1587b6ce07aaSKevin Wolf 
15888bfea15dSKevin Wolf fail:
1589f500a6d3SKevin Wolf     if (file != NULL) {
15904f6fd349SFam Zheng         bdrv_unref(file);
1591f500a6d3SKevin Wolf     }
1592de9c0cecSKevin Wolf     QDECREF(bs->options);
1593b6ad491aSKevin Wolf     QDECREF(options);
1594de9c0cecSKevin Wolf     bs->options = NULL;
1595f67503e5SMax Reitz     if (!*pbs) {
1596f67503e5SMax Reitz         /* If *pbs is NULL, a new BDS has been created in this function and
1597f67503e5SMax Reitz            needs to be freed now. Otherwise, it does not need to be closed,
1598f67503e5SMax Reitz            since it has not really been opened yet. */
1599f67503e5SMax Reitz         bdrv_unref(bs);
1600f67503e5SMax Reitz     }
160184d18f06SMarkus Armbruster     if (local_err) {
160234b5d2c6SMax Reitz         error_propagate(errp, local_err);
160334b5d2c6SMax Reitz     }
1604b6ad491aSKevin Wolf     return ret;
1605de9c0cecSKevin Wolf 
1606b6ad491aSKevin Wolf close_and_fail:
1607f67503e5SMax Reitz     /* See fail path, but now the BDS has to be always closed */
1608f67503e5SMax Reitz     if (*pbs) {
1609b6ad491aSKevin Wolf         bdrv_close(bs);
1610f67503e5SMax Reitz     } else {
1611f67503e5SMax Reitz         bdrv_unref(bs);
1612f67503e5SMax Reitz     }
1613b6ad491aSKevin Wolf     QDECREF(options);
161484d18f06SMarkus Armbruster     if (local_err) {
161534b5d2c6SMax Reitz         error_propagate(errp, local_err);
161634b5d2c6SMax Reitz     }
1617b6ce07aaSKevin Wolf     return ret;
1618b6ce07aaSKevin Wolf }
1619b6ce07aaSKevin Wolf 
1620f3930ed0SKevin Wolf int bdrv_open(BlockDriverState **pbs, const char *filename,
16216ebf9aa2SMax Reitz               const char *reference, QDict *options, int flags, Error **errp)
1622f3930ed0SKevin Wolf {
1623f3930ed0SKevin Wolf     return bdrv_open_inherit(pbs, filename, reference, options, flags, NULL,
1624ce343771SMax Reitz                              NULL, errp);
1625f3930ed0SKevin Wolf }
1626f3930ed0SKevin Wolf 
1627e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1628e971aa12SJeff Cody      bool prepared;
1629e971aa12SJeff Cody      BDRVReopenState state;
1630e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1631e971aa12SJeff Cody } BlockReopenQueueEntry;
1632e971aa12SJeff Cody 
1633e971aa12SJeff Cody /*
1634e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1635e971aa12SJeff Cody  * reopen of multiple devices.
1636e971aa12SJeff Cody  *
1637e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1638e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1639e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1640e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1641e971aa12SJeff Cody  * atomic 'set'.
1642e971aa12SJeff Cody  *
1643e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1644e971aa12SJeff Cody  *
16454d2cb092SKevin Wolf  * options contains the changed options for the associated bs
16464d2cb092SKevin Wolf  * (the BlockReopenQueue takes ownership)
16474d2cb092SKevin Wolf  *
1648e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1649e971aa12SJeff Cody  *
1650e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1651e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1652e971aa12SJeff Cody  *
1653e971aa12SJeff Cody  */
1654e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
16554d2cb092SKevin Wolf                                     BlockDriverState *bs,
16564d2cb092SKevin Wolf                                     QDict *options, int flags)
1657e971aa12SJeff Cody {
1658e971aa12SJeff Cody     assert(bs != NULL);
1659e971aa12SJeff Cody 
1660e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
166167251a31SKevin Wolf     BdrvChild *child;
16624d2cb092SKevin Wolf     QDict *old_options;
166367251a31SKevin Wolf 
1664e971aa12SJeff Cody     if (bs_queue == NULL) {
1665e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1666e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1667e971aa12SJeff Cody     }
1668e971aa12SJeff Cody 
16694d2cb092SKevin Wolf     if (!options) {
16704d2cb092SKevin Wolf         options = qdict_new();
16714d2cb092SKevin Wolf     }
16724d2cb092SKevin Wolf 
16734d2cb092SKevin Wolf     old_options = qdict_clone_shallow(bs->options);
16744d2cb092SKevin Wolf     qdict_join(options, old_options, false);
16754d2cb092SKevin Wolf     QDECREF(old_options);
16764d2cb092SKevin Wolf 
1677f1f25a2eSKevin Wolf     /* bdrv_open() masks this flag out */
1678f1f25a2eSKevin Wolf     flags &= ~BDRV_O_PROTOCOL;
1679f1f25a2eSKevin Wolf 
168067251a31SKevin Wolf     QLIST_FOREACH(child, &bs->children, next) {
168167251a31SKevin Wolf         int child_flags;
168267251a31SKevin Wolf 
168367251a31SKevin Wolf         if (child->bs->inherits_from != bs) {
168467251a31SKevin Wolf             continue;
168567251a31SKevin Wolf         }
168667251a31SKevin Wolf 
168767251a31SKevin Wolf         child_flags = child->role->inherit_flags(flags);
16884d2cb092SKevin Wolf         /* TODO Pass down child flags (backing.*, extents.*, ...) */
16894d2cb092SKevin Wolf         bdrv_reopen_queue(bs_queue, child->bs, NULL, child_flags);
1690e971aa12SJeff Cody     }
1691e971aa12SJeff Cody 
1692e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1693e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1694e971aa12SJeff Cody 
1695e971aa12SJeff Cody     bs_entry->state.bs = bs;
16964d2cb092SKevin Wolf     bs_entry->state.options = options;
1697e971aa12SJeff Cody     bs_entry->state.flags = flags;
1698e971aa12SJeff Cody 
1699e971aa12SJeff Cody     return bs_queue;
1700e971aa12SJeff Cody }
1701e971aa12SJeff Cody 
1702e971aa12SJeff Cody /*
1703e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1704e971aa12SJeff Cody  *
1705e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1706e971aa12SJeff Cody  * via bdrv_reopen_queue().
1707e971aa12SJeff Cody  *
1708e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1709e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1710e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1711e971aa12SJeff Cody  * data cleaned up.
1712e971aa12SJeff Cody  *
1713e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1714e971aa12SJeff Cody  * to all devices.
1715e971aa12SJeff Cody  *
1716e971aa12SJeff Cody  */
1717e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1718e971aa12SJeff Cody {
1719e971aa12SJeff Cody     int ret = -1;
1720e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1721e971aa12SJeff Cody     Error *local_err = NULL;
1722e971aa12SJeff Cody 
1723e971aa12SJeff Cody     assert(bs_queue != NULL);
1724e971aa12SJeff Cody 
1725e971aa12SJeff Cody     bdrv_drain_all();
1726e971aa12SJeff Cody 
1727e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1728e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1729e971aa12SJeff Cody             error_propagate(errp, local_err);
1730e971aa12SJeff Cody             goto cleanup;
1731e971aa12SJeff Cody         }
1732e971aa12SJeff Cody         bs_entry->prepared = true;
1733e971aa12SJeff Cody     }
1734e971aa12SJeff Cody 
1735e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1736e971aa12SJeff Cody      * changes
1737e971aa12SJeff Cody      */
1738e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1739e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1740e971aa12SJeff Cody     }
1741e971aa12SJeff Cody 
1742e971aa12SJeff Cody     ret = 0;
1743e971aa12SJeff Cody 
1744e971aa12SJeff Cody cleanup:
1745e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1746e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1747e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1748e971aa12SJeff Cody         }
17494d2cb092SKevin Wolf         QDECREF(bs_entry->state.options);
1750e971aa12SJeff Cody         g_free(bs_entry);
1751e971aa12SJeff Cody     }
1752e971aa12SJeff Cody     g_free(bs_queue);
1753e971aa12SJeff Cody     return ret;
1754e971aa12SJeff Cody }
1755e971aa12SJeff Cody 
1756e971aa12SJeff Cody 
1757e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1758e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1759e971aa12SJeff Cody {
1760e971aa12SJeff Cody     int ret = -1;
1761e971aa12SJeff Cody     Error *local_err = NULL;
17624d2cb092SKevin Wolf     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, NULL, bdrv_flags);
1763e971aa12SJeff Cody 
1764e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1765e971aa12SJeff Cody     if (local_err != NULL) {
1766e971aa12SJeff Cody         error_propagate(errp, local_err);
1767e971aa12SJeff Cody     }
1768e971aa12SJeff Cody     return ret;
1769e971aa12SJeff Cody }
1770e971aa12SJeff Cody 
1771e971aa12SJeff Cody 
1772e971aa12SJeff Cody /*
1773e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1774e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1775e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1776e971aa12SJeff Cody  *
1777e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1778e971aa12SJeff Cody  * flags are the new open flags
1779e971aa12SJeff Cody  * queue is the reopen queue
1780e971aa12SJeff Cody  *
1781e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1782e971aa12SJeff Cody  * as well.
1783e971aa12SJeff Cody  *
1784e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1785e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1786e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1787e971aa12SJeff Cody  *
1788e971aa12SJeff Cody  */
1789e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1790e971aa12SJeff Cody                         Error **errp)
1791e971aa12SJeff Cody {
1792e971aa12SJeff Cody     int ret = -1;
1793e971aa12SJeff Cody     Error *local_err = NULL;
1794e971aa12SJeff Cody     BlockDriver *drv;
1795e971aa12SJeff Cody 
1796e971aa12SJeff Cody     assert(reopen_state != NULL);
1797e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1798e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1799e971aa12SJeff Cody 
1800e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1801e971aa12SJeff Cody      * to r/w */
1802e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1803e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
180481e5f78aSAlberto Garcia         error_setg(errp, "Node '%s' is read only",
180581e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1806e971aa12SJeff Cody         goto error;
1807e971aa12SJeff Cody     }
1808e971aa12SJeff Cody 
1809e971aa12SJeff Cody 
1810e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1811e971aa12SJeff Cody     if (ret) {
1812e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1813e971aa12SJeff Cody                   strerror(-ret));
1814e971aa12SJeff Cody         goto error;
1815e971aa12SJeff Cody     }
1816e971aa12SJeff Cody 
1817e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1818e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1819e971aa12SJeff Cody         if (ret) {
1820e971aa12SJeff Cody             if (local_err != NULL) {
1821e971aa12SJeff Cody                 error_propagate(errp, local_err);
1822e971aa12SJeff Cody             } else {
1823d8b6895fSLuiz Capitulino                 error_setg(errp, "failed while preparing to reopen image '%s'",
1824e971aa12SJeff Cody                            reopen_state->bs->filename);
1825e971aa12SJeff Cody             }
1826e971aa12SJeff Cody             goto error;
1827e971aa12SJeff Cody         }
1828e971aa12SJeff Cody     } else {
1829e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1830e971aa12SJeff Cody          * handler for each supported drv. */
183181e5f78aSAlberto Garcia         error_setg(errp, "Block format '%s' used by node '%s' "
183281e5f78aSAlberto Garcia                    "does not support reopening files", drv->format_name,
183381e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1834e971aa12SJeff Cody         ret = -1;
1835e971aa12SJeff Cody         goto error;
1836e971aa12SJeff Cody     }
1837e971aa12SJeff Cody 
18384d2cb092SKevin Wolf     /* Options that are not handled are only okay if they are unchanged
18394d2cb092SKevin Wolf      * compared to the old state. It is expected that some options are only
18404d2cb092SKevin Wolf      * used for the initial open, but not reopen (e.g. filename) */
18414d2cb092SKevin Wolf     if (qdict_size(reopen_state->options)) {
18424d2cb092SKevin Wolf         const QDictEntry *entry = qdict_first(reopen_state->options);
18434d2cb092SKevin Wolf 
18444d2cb092SKevin Wolf         do {
18454d2cb092SKevin Wolf             QString *new_obj = qobject_to_qstring(entry->value);
18464d2cb092SKevin Wolf             const char *new = qstring_get_str(new_obj);
18474d2cb092SKevin Wolf             const char *old = qdict_get_try_str(reopen_state->bs->options,
18484d2cb092SKevin Wolf                                                 entry->key);
18494d2cb092SKevin Wolf 
18504d2cb092SKevin Wolf             if (!old || strcmp(new, old)) {
18514d2cb092SKevin Wolf                 error_setg(errp, "Cannot change the option '%s'", entry->key);
18524d2cb092SKevin Wolf                 ret = -EINVAL;
18534d2cb092SKevin Wolf                 goto error;
18544d2cb092SKevin Wolf             }
18554d2cb092SKevin Wolf         } while ((entry = qdict_next(reopen_state->options, entry)));
18564d2cb092SKevin Wolf     }
18574d2cb092SKevin Wolf 
1858e971aa12SJeff Cody     ret = 0;
1859e971aa12SJeff Cody 
1860e971aa12SJeff Cody error:
1861e971aa12SJeff Cody     return ret;
1862e971aa12SJeff Cody }
1863e971aa12SJeff Cody 
1864e971aa12SJeff Cody /*
1865e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1866e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1867e971aa12SJeff Cody  * the active BlockDriverState contents.
1868e971aa12SJeff Cody  */
1869e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1870e971aa12SJeff Cody {
1871e971aa12SJeff Cody     BlockDriver *drv;
1872e971aa12SJeff Cody 
1873e971aa12SJeff Cody     assert(reopen_state != NULL);
1874e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1875e971aa12SJeff Cody     assert(drv != NULL);
1876e971aa12SJeff Cody 
1877e971aa12SJeff Cody     /* If there are any driver level actions to take */
1878e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1879e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1880e971aa12SJeff Cody     }
1881e971aa12SJeff Cody 
1882e971aa12SJeff Cody     /* set BDS specific flags now */
1883e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1884e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1885e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1886e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1887355ef4acSKevin Wolf 
18883baca891SKevin Wolf     bdrv_refresh_limits(reopen_state->bs, NULL);
1889e971aa12SJeff Cody }
1890e971aa12SJeff Cody 
1891e971aa12SJeff Cody /*
1892e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1893e971aa12SJeff Cody  * reopen_state
1894e971aa12SJeff Cody  */
1895e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1896e971aa12SJeff Cody {
1897e971aa12SJeff Cody     BlockDriver *drv;
1898e971aa12SJeff Cody 
1899e971aa12SJeff Cody     assert(reopen_state != NULL);
1900e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1901e971aa12SJeff Cody     assert(drv != NULL);
1902e971aa12SJeff Cody 
1903e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1904e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1905e971aa12SJeff Cody     }
1906e971aa12SJeff Cody }
1907e971aa12SJeff Cody 
1908e971aa12SJeff Cody 
1909fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1910fc01f7e7Sbellard {
191133384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
191233384421SMax Reitz 
19133e914655SPaolo Bonzini     if (bs->job) {
19143e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
19153e914655SPaolo Bonzini     }
191699b7e775SAlberto Garcia 
191799b7e775SAlberto Garcia     /* Disable I/O limits and drain all pending throttled requests */
191899b7e775SAlberto Garcia     if (bs->io_limits_enabled) {
191999b7e775SAlberto Garcia         bdrv_io_limits_disable(bs);
192099b7e775SAlberto Garcia     }
192199b7e775SAlberto Garcia 
192253ec73e2SFam Zheng     bdrv_drain(bs); /* complete I/O */
192358fda173SStefan Hajnoczi     bdrv_flush(bs);
192453ec73e2SFam Zheng     bdrv_drain(bs); /* in case flush left pending I/O */
1925d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
19267094f12fSKevin Wolf 
19273cbc002cSPaolo Bonzini     if (bs->drv) {
19286e93e7c4SKevin Wolf         BdrvChild *child, *next;
19296e93e7c4SKevin Wolf 
19309a7dedbcSKevin Wolf         bs->drv->bdrv_close(bs);
19319a7dedbcSKevin Wolf 
19329a7dedbcSKevin Wolf         if (bs->backing_hd) {
19339a7dedbcSKevin Wolf             BlockDriverState *backing_hd = bs->backing_hd;
19349a7dedbcSKevin Wolf             bdrv_set_backing_hd(bs, NULL);
19359a7dedbcSKevin Wolf             bdrv_unref(backing_hd);
19369a7dedbcSKevin Wolf         }
19379a7dedbcSKevin Wolf 
19386e93e7c4SKevin Wolf         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
193933a60407SKevin Wolf             /* TODO Remove bdrv_unref() from drivers' close function and use
194033a60407SKevin Wolf              * bdrv_unref_child() here */
1941bddcec37SKevin Wolf             if (child->bs->inherits_from == bs) {
1942bddcec37SKevin Wolf                 child->bs->inherits_from = NULL;
1943bddcec37SKevin Wolf             }
194433a60407SKevin Wolf             bdrv_detach_child(child);
19456e93e7c4SKevin Wolf         }
19466e93e7c4SKevin Wolf 
19477267c094SAnthony Liguori         g_free(bs->opaque);
1948ea2384d3Sbellard         bs->opaque = NULL;
1949ea2384d3Sbellard         bs->drv = NULL;
195053fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1951a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1952a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
19536405875cSPaolo Bonzini         bs->total_sectors = 0;
19546405875cSPaolo Bonzini         bs->encrypted = 0;
19556405875cSPaolo Bonzini         bs->valid_key = 0;
19566405875cSPaolo Bonzini         bs->sg = 0;
19570d51b4deSAsias He         bs->zero_beyond_eof = false;
1958de9c0cecSKevin Wolf         QDECREF(bs->options);
1959de9c0cecSKevin Wolf         bs->options = NULL;
196091af7014SMax Reitz         QDECREF(bs->full_open_options);
196191af7014SMax Reitz         bs->full_open_options = NULL;
1962b338082bSbellard 
196366f82ceeSKevin Wolf         if (bs->file != NULL) {
19644f6fd349SFam Zheng             bdrv_unref(bs->file);
19650ac9377dSPaolo Bonzini             bs->file = NULL;
196666f82ceeSKevin Wolf         }
19679ca11154SPavel Hrdina     }
196866f82ceeSKevin Wolf 
1969a7f53e26SMarkus Armbruster     if (bs->blk) {
1970a7f53e26SMarkus Armbruster         blk_dev_change_media_cb(bs->blk, false);
1971a7f53e26SMarkus Armbruster     }
197298f90dbaSZhi Yong Wu 
197333384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
197433384421SMax Reitz         g_free(ban);
197533384421SMax Reitz     }
197633384421SMax Reitz     QLIST_INIT(&bs->aio_notifiers);
1977b338082bSbellard }
1978b338082bSbellard 
19792bc93fedSMORITA Kazutaka void bdrv_close_all(void)
19802bc93fedSMORITA Kazutaka {
19812bc93fedSMORITA Kazutaka     BlockDriverState *bs;
19822bc93fedSMORITA Kazutaka 
1983dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
1984ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
1985ed78cda3SStefan Hajnoczi 
1986ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
19872bc93fedSMORITA Kazutaka         bdrv_close(bs);
1988ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
19892bc93fedSMORITA Kazutaka     }
19902bc93fedSMORITA Kazutaka }
19912bc93fedSMORITA Kazutaka 
1992dc364f4cSBenoît Canet /* make a BlockDriverState anonymous by removing from bdrv_state and
1993dc364f4cSBenoît Canet  * graph_bdrv_state list.
1994d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1995d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1996d22b2f41SRyan Harper {
1997bfb197e0SMarkus Armbruster     /*
1998bfb197e0SMarkus Armbruster      * Take care to remove bs from bdrv_states only when it's actually
1999bfb197e0SMarkus Armbruster      * in it.  Note that bs->device_list.tqe_prev is initially null,
2000bfb197e0SMarkus Armbruster      * and gets set to non-null by QTAILQ_INSERT_TAIL().  Establish
2001bfb197e0SMarkus Armbruster      * the useful invariant "bs in bdrv_states iff bs->tqe_prev" by
2002bfb197e0SMarkus Armbruster      * resetting it to null on remove.
2003bfb197e0SMarkus Armbruster      */
2004bfb197e0SMarkus Armbruster     if (bs->device_list.tqe_prev) {
2005dc364f4cSBenoît Canet         QTAILQ_REMOVE(&bdrv_states, bs, device_list);
2006bfb197e0SMarkus Armbruster         bs->device_list.tqe_prev = NULL;
2007d22b2f41SRyan Harper     }
2008dc364f4cSBenoît Canet     if (bs->node_name[0] != '\0') {
2009dc364f4cSBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs, node_list);
2010dc364f4cSBenoît Canet     }
2011dc364f4cSBenoît Canet     bs->node_name[0] = '\0';
2012d22b2f41SRyan Harper }
2013d22b2f41SRyan Harper 
2014e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
2015e023b2e2SPaolo Bonzini {
2016e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
2017e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
2018e023b2e2SPaolo Bonzini     }
2019e023b2e2SPaolo Bonzini }
2020e023b2e2SPaolo Bonzini 
20214ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
20224ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
20234ddc07caSPaolo Bonzini {
20244ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
20254ddc07caSPaolo Bonzini 
20264ddc07caSPaolo Bonzini     /* dev info */
20271b7fd729SPaolo Bonzini     bs_dest->guest_block_size   = bs_src->guest_block_size;
20284ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
20294ddc07caSPaolo Bonzini 
20304ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
20314ddc07caSPaolo Bonzini 
2032cc0681c4SBenoît Canet     /* i/o throttled req */
203376f4afb4SAlberto Garcia     bs_dest->throttle_state     = bs_src->throttle_state,
203476f4afb4SAlberto Garcia     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
203576f4afb4SAlberto Garcia     bs_dest->pending_reqs[0]    = bs_src->pending_reqs[0];
203676f4afb4SAlberto Garcia     bs_dest->pending_reqs[1]    = bs_src->pending_reqs[1];
203776f4afb4SAlberto Garcia     bs_dest->throttled_reqs[0]  = bs_src->throttled_reqs[0];
203876f4afb4SAlberto Garcia     bs_dest->throttled_reqs[1]  = bs_src->throttled_reqs[1];
203976f4afb4SAlberto Garcia     memcpy(&bs_dest->round_robin,
204076f4afb4SAlberto Garcia            &bs_src->round_robin,
204176f4afb4SAlberto Garcia            sizeof(bs_dest->round_robin));
20420e5b0a2dSBenoît Canet     memcpy(&bs_dest->throttle_timers,
20430e5b0a2dSBenoît Canet            &bs_src->throttle_timers,
20440e5b0a2dSBenoît Canet            sizeof(ThrottleTimers));
20454ddc07caSPaolo Bonzini 
20464ddc07caSPaolo Bonzini     /* r/w error */
20474ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
20484ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
20494ddc07caSPaolo Bonzini 
20504ddc07caSPaolo Bonzini     /* i/o status */
20514ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
20524ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
20534ddc07caSPaolo Bonzini 
20544ddc07caSPaolo Bonzini     /* dirty bitmap */
2055e4654d2dSFam Zheng     bs_dest->dirty_bitmaps      = bs_src->dirty_bitmaps;
20564ddc07caSPaolo Bonzini 
20579fcb0251SFam Zheng     /* reference count */
20589fcb0251SFam Zheng     bs_dest->refcnt             = bs_src->refcnt;
20599fcb0251SFam Zheng 
20604ddc07caSPaolo Bonzini     /* job */
20614ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
20624ddc07caSPaolo Bonzini 
20634ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
2064dc364f4cSBenoît Canet     bs_dest->device_list = bs_src->device_list;
20657e7d56d9SMarkus Armbruster     bs_dest->blk = bs_src->blk;
20667e7d56d9SMarkus Armbruster 
2067fbe40ff7SFam Zheng     memcpy(bs_dest->op_blockers, bs_src->op_blockers,
2068fbe40ff7SFam Zheng            sizeof(bs_dest->op_blockers));
20694ddc07caSPaolo Bonzini }
20704ddc07caSPaolo Bonzini 
20714ddc07caSPaolo Bonzini /*
20724ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
20734ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
20744ddc07caSPaolo Bonzini  * actually attached to a device.
20754ddc07caSPaolo Bonzini  *
20764ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
20774ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
20784ddc07caSPaolo Bonzini  *
2079bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
20804ddc07caSPaolo Bonzini  *
20814ddc07caSPaolo Bonzini  * This function does not create any image files.
20824ddc07caSPaolo Bonzini  */
20834ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
20844ddc07caSPaolo Bonzini {
20854ddc07caSPaolo Bonzini     BlockDriverState tmp;
2086bddcec37SKevin Wolf     BdrvChild *child;
20874ddc07caSPaolo Bonzini 
20886ee4ce1eSKevin Wolf     bdrv_drain(bs_new);
20896ee4ce1eSKevin Wolf     bdrv_drain(bs_old);
20904ddc07caSPaolo Bonzini 
209190ce8a06SBenoît Canet     /* The code needs to swap the node_name but simply swapping node_list won't
209290ce8a06SBenoît Canet      * work so first remove the nodes from the graph list, do the swap then
209390ce8a06SBenoît Canet      * insert them back if needed.
209490ce8a06SBenoît Canet      */
209590ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
209690ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_new, node_list);
209790ce8a06SBenoît Canet     }
209890ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
209990ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_old, node_list);
210090ce8a06SBenoît Canet     }
210190ce8a06SBenoît Canet 
2102db628338SAlberto Garcia     /* If the BlockDriverState is part of a throttling group acquire
2103db628338SAlberto Garcia      * its lock since we're going to mess with the protected fields.
2104db628338SAlberto Garcia      * Otherwise there's no need to worry since no one else can touch
2105db628338SAlberto Garcia      * them. */
2106db628338SAlberto Garcia     if (bs_old->throttle_state) {
2107db628338SAlberto Garcia         throttle_group_lock(bs_old);
2108db628338SAlberto Garcia     }
2109db628338SAlberto Garcia 
2110bfb197e0SMarkus Armbruster     /* bs_new must be unattached and shouldn't have anything fancy enabled */
21117e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
2112e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs_new->dirty_bitmaps));
21134ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21144ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2115db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21160e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21174ddc07caSPaolo Bonzini 
21184ddc07caSPaolo Bonzini     tmp = *bs_new;
21194ddc07caSPaolo Bonzini     *bs_new = *bs_old;
21204ddc07caSPaolo Bonzini     *bs_old = tmp;
21214ddc07caSPaolo Bonzini 
21224ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
21234ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
21244ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
21254ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
21264ddc07caSPaolo Bonzini 
2127bfb197e0SMarkus Armbruster     /* bs_new must remain unattached */
21287e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
21294ddc07caSPaolo Bonzini 
21304ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
21314ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21324ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2133db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21340e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21354ddc07caSPaolo Bonzini 
2136db628338SAlberto Garcia     /* Release the ThrottleGroup lock */
2137db628338SAlberto Garcia     if (bs_old->throttle_state) {
2138db628338SAlberto Garcia         throttle_group_unlock(bs_old);
2139db628338SAlberto Garcia     }
2140db628338SAlberto Garcia 
214190ce8a06SBenoît Canet     /* insert the nodes back into the graph node list if needed */
214290ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
214390ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_new, node_list);
214490ce8a06SBenoît Canet     }
214590ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
214690ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_old, node_list);
214790ce8a06SBenoît Canet     }
214890ce8a06SBenoît Canet 
21496e93e7c4SKevin Wolf     /*
21506e93e7c4SKevin Wolf      * Update lh_first.le_prev for non-empty lists.
21516e93e7c4SKevin Wolf      *
21526e93e7c4SKevin Wolf      * The head of the op blocker list doesn't change because it is moved back
21536e93e7c4SKevin Wolf      * in bdrv_move_feature_fields().
21546e93e7c4SKevin Wolf      */
21556ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_old->tracked_requests));
21566ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_new->tracked_requests));
21576ee4ce1eSKevin Wolf 
21586e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_new->children, next);
21596e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_old->children, next);
21606e93e7c4SKevin Wolf 
2161bddcec37SKevin Wolf     /* Update references in bs->opaque and children */
2162bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_old->children, next) {
2163bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_new) {
2164bddcec37SKevin Wolf             child->bs->inherits_from = bs_old;
2165bddcec37SKevin Wolf         }
2166bddcec37SKevin Wolf     }
2167bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_new->children, next) {
2168bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_old) {
2169bddcec37SKevin Wolf             child->bs->inherits_from = bs_new;
2170bddcec37SKevin Wolf         }
2171bddcec37SKevin Wolf     }
2172bddcec37SKevin Wolf 
21734ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
21744ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
21754ddc07caSPaolo Bonzini }
21764ddc07caSPaolo Bonzini 
21778802d1fdSJeff Cody /*
21788802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
21798802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
21808802d1fdSJeff Cody  *
21818802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
21828802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
21838802d1fdSJeff Cody  *
2184bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
2185f6801b83SJeff Cody  *
21868802d1fdSJeff Cody  * This function does not create any image files.
21878802d1fdSJeff Cody  */
21888802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
21898802d1fdSJeff Cody {
21904ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
21918802d1fdSJeff Cody 
21928802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
21938802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
21948d24cce1SFam Zheng     bdrv_set_backing_hd(bs_top, bs_new);
21958802d1fdSJeff Cody }
21968802d1fdSJeff Cody 
21974f6fd349SFam Zheng static void bdrv_delete(BlockDriverState *bs)
2198b338082bSbellard {
21993e914655SPaolo Bonzini     assert(!bs->job);
22003718d8abSFam Zheng     assert(bdrv_op_blocker_is_empty(bs));
22014f6fd349SFam Zheng     assert(!bs->refcnt);
2202e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
220318846deeSMarkus Armbruster 
2204e1b5c52eSStefan Hajnoczi     bdrv_close(bs);
2205e1b5c52eSStefan Hajnoczi 
22061b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
2207d22b2f41SRyan Harper     bdrv_make_anon(bs);
220834c6f050Saurel32 
22097267c094SAnthony Liguori     g_free(bs);
2210fc01f7e7Sbellard }
2211fc01f7e7Sbellard 
2212e97fc193Saliguori /*
2213e97fc193Saliguori  * Run consistency checks on an image
2214e97fc193Saliguori  *
2215e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
2216a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
2217e076f338SKevin Wolf  * check are stored in res.
2218e97fc193Saliguori  */
22194534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
2220e97fc193Saliguori {
2221908bcd54SMax Reitz     if (bs->drv == NULL) {
2222908bcd54SMax Reitz         return -ENOMEDIUM;
2223908bcd54SMax Reitz     }
2224e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
2225e97fc193Saliguori         return -ENOTSUP;
2226e97fc193Saliguori     }
2227e97fc193Saliguori 
2228e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
22294534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
2230e97fc193Saliguori }
2231e97fc193Saliguori 
22328a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
22338a426614SKevin Wolf 
223433e3963eSbellard /* commit COW file into the raw image */
223533e3963eSbellard int bdrv_commit(BlockDriverState *bs)
223633e3963eSbellard {
223719cb3738Sbellard     BlockDriver *drv = bs->drv;
223872706ea4SJeff Cody     int64_t sector, total_sectors, length, backing_length;
22398a426614SKevin Wolf     int n, ro, open_flags;
22400bce597dSJeff Cody     int ret = 0;
224172706ea4SJeff Cody     uint8_t *buf = NULL;
224233e3963eSbellard 
224319cb3738Sbellard     if (!drv)
224419cb3738Sbellard         return -ENOMEDIUM;
224533e3963eSbellard 
22464dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
22474dca4b63SNaphtali Sprei         return -ENOTSUP;
22484dca4b63SNaphtali Sprei     }
22494dca4b63SNaphtali Sprei 
2250bb00021dSFam Zheng     if (bdrv_op_is_blocked(bs, BLOCK_OP_TYPE_COMMIT_SOURCE, NULL) ||
2251bb00021dSFam Zheng         bdrv_op_is_blocked(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET, NULL)) {
22522d3735d3SStefan Hajnoczi         return -EBUSY;
22532d3735d3SStefan Hajnoczi     }
22542d3735d3SStefan Hajnoczi 
22554dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
22564dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
22574dca4b63SNaphtali Sprei 
22584dca4b63SNaphtali Sprei     if (ro) {
22590bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
22600bce597dSJeff Cody             return -EACCES;
22614dca4b63SNaphtali Sprei         }
2262ea2384d3Sbellard     }
2263ea2384d3Sbellard 
226472706ea4SJeff Cody     length = bdrv_getlength(bs);
226572706ea4SJeff Cody     if (length < 0) {
226672706ea4SJeff Cody         ret = length;
226772706ea4SJeff Cody         goto ro_cleanup;
226872706ea4SJeff Cody     }
226972706ea4SJeff Cody 
227072706ea4SJeff Cody     backing_length = bdrv_getlength(bs->backing_hd);
227172706ea4SJeff Cody     if (backing_length < 0) {
227272706ea4SJeff Cody         ret = backing_length;
227372706ea4SJeff Cody         goto ro_cleanup;
227472706ea4SJeff Cody     }
227572706ea4SJeff Cody 
227672706ea4SJeff Cody     /* If our top snapshot is larger than the backing file image,
227772706ea4SJeff Cody      * grow the backing file image if possible.  If not possible,
227872706ea4SJeff Cody      * we must return an error */
227972706ea4SJeff Cody     if (length > backing_length) {
228072706ea4SJeff Cody         ret = bdrv_truncate(bs->backing_hd, length);
228172706ea4SJeff Cody         if (ret < 0) {
228272706ea4SJeff Cody             goto ro_cleanup;
228372706ea4SJeff Cody         }
228472706ea4SJeff Cody     }
228572706ea4SJeff Cody 
228672706ea4SJeff Cody     total_sectors = length >> BDRV_SECTOR_BITS;
2287857d4f46SKevin Wolf 
2288857d4f46SKevin Wolf     /* qemu_try_blockalign() for bs will choose an alignment that works for
2289857d4f46SKevin Wolf      * bs->backing_hd as well, so no need to compare the alignment manually. */
2290857d4f46SKevin Wolf     buf = qemu_try_blockalign(bs, COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
2291857d4f46SKevin Wolf     if (buf == NULL) {
2292857d4f46SKevin Wolf         ret = -ENOMEM;
2293857d4f46SKevin Wolf         goto ro_cleanup;
2294857d4f46SKevin Wolf     }
22958a426614SKevin Wolf 
22968a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
2297d663640cSPaolo Bonzini         ret = bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n);
2298d663640cSPaolo Bonzini         if (ret < 0) {
2299d663640cSPaolo Bonzini             goto ro_cleanup;
2300d663640cSPaolo Bonzini         }
2301d663640cSPaolo Bonzini         if (ret) {
2302dabfa6ccSKevin Wolf             ret = bdrv_read(bs, sector, buf, n);
2303dabfa6ccSKevin Wolf             if (ret < 0) {
23044dca4b63SNaphtali Sprei                 goto ro_cleanup;
230533e3963eSbellard             }
230633e3963eSbellard 
2307dabfa6ccSKevin Wolf             ret = bdrv_write(bs->backing_hd, sector, buf, n);
2308dabfa6ccSKevin Wolf             if (ret < 0) {
23094dca4b63SNaphtali Sprei                 goto ro_cleanup;
231033e3963eSbellard             }
231133e3963eSbellard         }
231233e3963eSbellard     }
231395389c86Sbellard 
23141d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
23151d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
2316dabfa6ccSKevin Wolf         if (ret < 0) {
2317dabfa6ccSKevin Wolf             goto ro_cleanup;
2318dabfa6ccSKevin Wolf         }
23191d44952fSChristoph Hellwig         bdrv_flush(bs);
23201d44952fSChristoph Hellwig     }
232195389c86Sbellard 
23223f5075aeSChristoph Hellwig     /*
23233f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
23243f5075aeSChristoph Hellwig      * stable on disk.
23253f5075aeSChristoph Hellwig      */
2326dabfa6ccSKevin Wolf     if (bs->backing_hd) {
23273f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
2328dabfa6ccSKevin Wolf     }
23294dca4b63SNaphtali Sprei 
2330dabfa6ccSKevin Wolf     ret = 0;
23314dca4b63SNaphtali Sprei ro_cleanup:
2332857d4f46SKevin Wolf     qemu_vfree(buf);
23334dca4b63SNaphtali Sprei 
23344dca4b63SNaphtali Sprei     if (ro) {
23350bce597dSJeff Cody         /* ignoring error return here */
23360bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
23374dca4b63SNaphtali Sprei     }
23384dca4b63SNaphtali Sprei 
23391d44952fSChristoph Hellwig     return ret;
234033e3963eSbellard }
234133e3963eSbellard 
2342e8877497SStefan Hajnoczi int bdrv_commit_all(void)
23436ab4b5abSMarkus Armbruster {
23446ab4b5abSMarkus Armbruster     BlockDriverState *bs;
23456ab4b5abSMarkus Armbruster 
2346dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
2347ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
2348ed78cda3SStefan Hajnoczi 
2349ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
2350272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
2351e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
2352e8877497SStefan Hajnoczi             if (ret < 0) {
2353ed78cda3SStefan Hajnoczi                 aio_context_release(aio_context);
2354e8877497SStefan Hajnoczi                 return ret;
23556ab4b5abSMarkus Armbruster             }
23566ab4b5abSMarkus Armbruster         }
2357ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
2358272d2d8eSJeff Cody     }
2359e8877497SStefan Hajnoczi     return 0;
2360e8877497SStefan Hajnoczi }
23616ab4b5abSMarkus Armbruster 
2362756e6736SKevin Wolf /*
2363756e6736SKevin Wolf  * Return values:
2364756e6736SKevin Wolf  * 0        - success
2365756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
2366756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
2367756e6736SKevin Wolf  *            image file header
2368756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
2369756e6736SKevin Wolf  */
2370756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
2371756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
2372756e6736SKevin Wolf {
2373756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
2374469ef350SPaolo Bonzini     int ret;
2375756e6736SKevin Wolf 
23765f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
23775f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
23785f377794SPaolo Bonzini         return -EINVAL;
23795f377794SPaolo Bonzini     }
23805f377794SPaolo Bonzini 
2381756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
2382469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
2383756e6736SKevin Wolf     } else {
2384469ef350SPaolo Bonzini         ret = -ENOTSUP;
2385756e6736SKevin Wolf     }
2386469ef350SPaolo Bonzini 
2387469ef350SPaolo Bonzini     if (ret == 0) {
2388469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
2389469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
2390469ef350SPaolo Bonzini     }
2391469ef350SPaolo Bonzini     return ret;
2392756e6736SKevin Wolf }
2393756e6736SKevin Wolf 
23946ebdcee2SJeff Cody /*
23956ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
23966ebdcee2SJeff Cody  *
23976ebdcee2SJeff Cody  * active is the current topmost image.
23986ebdcee2SJeff Cody  *
23996ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
24006ebdcee2SJeff Cody  * or if active == bs.
24014caf0fcdSJeff Cody  *
24024caf0fcdSJeff Cody  * Returns the bottommost base image if bs == NULL.
24036ebdcee2SJeff Cody  */
24046ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
24056ebdcee2SJeff Cody                                     BlockDriverState *bs)
24066ebdcee2SJeff Cody {
24074caf0fcdSJeff Cody     while (active && bs != active->backing_hd) {
24084caf0fcdSJeff Cody         active = active->backing_hd;
24096ebdcee2SJeff Cody     }
24106ebdcee2SJeff Cody 
24114caf0fcdSJeff Cody     return active;
24126ebdcee2SJeff Cody }
24136ebdcee2SJeff Cody 
24144caf0fcdSJeff Cody /* Given a BDS, searches for the base layer. */
24154caf0fcdSJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
24164caf0fcdSJeff Cody {
24174caf0fcdSJeff Cody     return bdrv_find_overlay(bs, NULL);
24186ebdcee2SJeff Cody }
24196ebdcee2SJeff Cody 
24206ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
24216ebdcee2SJeff Cody     BlockDriverState *bs;
24226ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
24236ebdcee2SJeff Cody } BlkIntermediateStates;
24246ebdcee2SJeff Cody 
24256ebdcee2SJeff Cody 
24266ebdcee2SJeff Cody /*
24276ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
24286ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
24296ebdcee2SJeff Cody  *
24306ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
24316ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
24326ebdcee2SJeff Cody  *
24336ebdcee2SJeff Cody  * E.g., this will convert the following chain:
24346ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
24356ebdcee2SJeff Cody  *
24366ebdcee2SJeff Cody  * to
24376ebdcee2SJeff Cody  *
24386ebdcee2SJeff Cody  * bottom <- base <- active
24396ebdcee2SJeff Cody  *
24406ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
24416ebdcee2SJeff Cody  *
24426ebdcee2SJeff Cody  * base <- intermediate <- top <- active
24436ebdcee2SJeff Cody  *
24446ebdcee2SJeff Cody  * to
24456ebdcee2SJeff Cody  *
24466ebdcee2SJeff Cody  * base <- active
24476ebdcee2SJeff Cody  *
244854e26900SJeff Cody  * If backing_file_str is non-NULL, it will be used when modifying top's
244954e26900SJeff Cody  * overlay image metadata.
245054e26900SJeff Cody  *
24516ebdcee2SJeff Cody  * Error conditions:
24526ebdcee2SJeff Cody  *  if active == top, that is considered an error
24536ebdcee2SJeff Cody  *
24546ebdcee2SJeff Cody  */
24556ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
245654e26900SJeff Cody                            BlockDriverState *base, const char *backing_file_str)
24576ebdcee2SJeff Cody {
24586ebdcee2SJeff Cody     BlockDriverState *intermediate;
24596ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
24606ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
24616ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
24626ebdcee2SJeff Cody     int ret = -EIO;
24636ebdcee2SJeff Cody 
24646ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
24656ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
24666ebdcee2SJeff Cody 
24676ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
24686ebdcee2SJeff Cody         goto exit;
24696ebdcee2SJeff Cody     }
24706ebdcee2SJeff Cody 
24716ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
24726ebdcee2SJeff Cody 
24736ebdcee2SJeff Cody     if (new_top_bs == NULL) {
24746ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
24756ebdcee2SJeff Cody         goto exit;
24766ebdcee2SJeff Cody     }
24776ebdcee2SJeff Cody 
24786ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
24796ebdcee2SJeff Cody      * to do, no intermediate images */
24806ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
24816ebdcee2SJeff Cody         ret = 0;
24826ebdcee2SJeff Cody         goto exit;
24836ebdcee2SJeff Cody     }
24846ebdcee2SJeff Cody 
24856ebdcee2SJeff Cody     intermediate = top;
24866ebdcee2SJeff Cody 
24876ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
24886ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
24896ebdcee2SJeff Cody      */
24906ebdcee2SJeff Cody     while (intermediate) {
24915839e53bSMarkus Armbruster         intermediate_state = g_new0(BlkIntermediateStates, 1);
24926ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
24936ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
24946ebdcee2SJeff Cody 
24956ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
24966ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
24976ebdcee2SJeff Cody             break;
24986ebdcee2SJeff Cody         }
24996ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
25006ebdcee2SJeff Cody     }
25016ebdcee2SJeff Cody     if (base_bs == NULL) {
25026ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
25036ebdcee2SJeff Cody          * unravel everything, and exit with error */
25046ebdcee2SJeff Cody         goto exit;
25056ebdcee2SJeff Cody     }
25066ebdcee2SJeff Cody 
25076ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
250854e26900SJeff Cody     backing_file_str = backing_file_str ? backing_file_str : base_bs->filename;
250954e26900SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, backing_file_str,
25106ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
25116ebdcee2SJeff Cody     if (ret) {
25126ebdcee2SJeff Cody         goto exit;
25136ebdcee2SJeff Cody     }
2514920beae1SFam Zheng     bdrv_set_backing_hd(new_top_bs, base_bs);
25156ebdcee2SJeff Cody 
25166ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25176ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
2518920beae1SFam Zheng         bdrv_set_backing_hd(intermediate_state->bs, NULL);
25194f6fd349SFam Zheng         bdrv_unref(intermediate_state->bs);
25206ebdcee2SJeff Cody     }
25216ebdcee2SJeff Cody     ret = 0;
25226ebdcee2SJeff Cody 
25236ebdcee2SJeff Cody exit:
25246ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25256ebdcee2SJeff Cody         g_free(intermediate_state);
25266ebdcee2SJeff Cody     }
25276ebdcee2SJeff Cody     return ret;
25286ebdcee2SJeff Cody }
25296ebdcee2SJeff Cody 
253083f64091Sbellard /**
253183f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
253283f64091Sbellard  */
253383f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
253483f64091Sbellard {
253583f64091Sbellard     BlockDriver *drv = bs->drv;
253651762288SStefan Hajnoczi     int ret;
253783f64091Sbellard     if (!drv)
253819cb3738Sbellard         return -ENOMEDIUM;
253983f64091Sbellard     if (!drv->bdrv_truncate)
254083f64091Sbellard         return -ENOTSUP;
254159f2689dSNaphtali Sprei     if (bs->read_only)
254259f2689dSNaphtali Sprei         return -EACCES;
25439c75e168SJeff Cody 
254451762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
254551762288SStefan Hajnoczi     if (ret == 0) {
254651762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2547ce1ffea8SJohn Snow         bdrv_dirty_bitmap_truncate(bs);
2548a7f53e26SMarkus Armbruster         if (bs->blk) {
2549a7f53e26SMarkus Armbruster             blk_dev_resize_cb(bs->blk);
2550a7f53e26SMarkus Armbruster         }
255151762288SStefan Hajnoczi     }
255251762288SStefan Hajnoczi     return ret;
255383f64091Sbellard }
255483f64091Sbellard 
255583f64091Sbellard /**
25564a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
25574a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
25584a1d5e1fSFam Zheng  */
25594a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
25604a1d5e1fSFam Zheng {
25614a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
25624a1d5e1fSFam Zheng     if (!drv) {
25634a1d5e1fSFam Zheng         return -ENOMEDIUM;
25644a1d5e1fSFam Zheng     }
25654a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
25664a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
25674a1d5e1fSFam Zheng     }
25684a1d5e1fSFam Zheng     if (bs->file) {
25694a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
25704a1d5e1fSFam Zheng     }
25714a1d5e1fSFam Zheng     return -ENOTSUP;
25724a1d5e1fSFam Zheng }
25734a1d5e1fSFam Zheng 
25744a1d5e1fSFam Zheng /**
257565a9bb25SMarkus Armbruster  * Return number of sectors on success, -errno on error.
257683f64091Sbellard  */
257765a9bb25SMarkus Armbruster int64_t bdrv_nb_sectors(BlockDriverState *bs)
257883f64091Sbellard {
257983f64091Sbellard     BlockDriver *drv = bs->drv;
258065a9bb25SMarkus Armbruster 
258183f64091Sbellard     if (!drv)
258219cb3738Sbellard         return -ENOMEDIUM;
258351762288SStefan Hajnoczi 
2584b94a2610SKevin Wolf     if (drv->has_variable_length) {
2585b94a2610SKevin Wolf         int ret = refresh_total_sectors(bs, bs->total_sectors);
2586b94a2610SKevin Wolf         if (ret < 0) {
2587b94a2610SKevin Wolf             return ret;
2588fc01f7e7Sbellard         }
258946a4e4e6SStefan Hajnoczi     }
259065a9bb25SMarkus Armbruster     return bs->total_sectors;
259165a9bb25SMarkus Armbruster }
259265a9bb25SMarkus Armbruster 
259365a9bb25SMarkus Armbruster /**
259465a9bb25SMarkus Armbruster  * Return length in bytes on success, -errno on error.
259565a9bb25SMarkus Armbruster  * The length is always a multiple of BDRV_SECTOR_SIZE.
259665a9bb25SMarkus Armbruster  */
259765a9bb25SMarkus Armbruster int64_t bdrv_getlength(BlockDriverState *bs)
259865a9bb25SMarkus Armbruster {
259965a9bb25SMarkus Armbruster     int64_t ret = bdrv_nb_sectors(bs);
260065a9bb25SMarkus Armbruster 
26014a9c9ea0SFam Zheng     ret = ret > INT64_MAX / BDRV_SECTOR_SIZE ? -EFBIG : ret;
260265a9bb25SMarkus Armbruster     return ret < 0 ? ret : ret * BDRV_SECTOR_SIZE;
260346a4e4e6SStefan Hajnoczi }
2604fc01f7e7Sbellard 
260519cb3738Sbellard /* return 0 as number of sectors if no device present or error */
260696b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2607fc01f7e7Sbellard {
260865a9bb25SMarkus Armbruster     int64_t nb_sectors = bdrv_nb_sectors(bs);
260965a9bb25SMarkus Armbruster 
261065a9bb25SMarkus Armbruster     *nb_sectors_ptr = nb_sectors < 0 ? 0 : nb_sectors;
2611fc01f7e7Sbellard }
2612cf98951bSbellard 
2613ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2614ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
2615abd7f68dSMarkus Armbruster {
2616abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
2617abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
2618abd7f68dSMarkus Armbruster }
2619abd7f68dSMarkus Armbruster 
26201ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2621abd7f68dSMarkus Armbruster {
2622abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
2623abd7f68dSMarkus Armbruster }
2624abd7f68dSMarkus Armbruster 
26253e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
26263e1caa5fSPaolo Bonzini {
26273e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
26283e1caa5fSPaolo Bonzini 
26293e1caa5fSPaolo Bonzini     switch (on_err) {
26303e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
2631a589569fSWenchao Xia         return (error == ENOSPC) ?
2632a589569fSWenchao Xia                BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
26333e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
2634a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_STOP;
26353e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
2636a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_REPORT;
26373e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
2638a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_IGNORE;
26393e1caa5fSPaolo Bonzini     default:
26403e1caa5fSPaolo Bonzini         abort();
26413e1caa5fSPaolo Bonzini     }
26423e1caa5fSPaolo Bonzini }
26433e1caa5fSPaolo Bonzini 
2644c7c2ff0cSLuiz Capitulino static void send_qmp_error_event(BlockDriverState *bs,
2645c7c2ff0cSLuiz Capitulino                                  BlockErrorAction action,
2646c7c2ff0cSLuiz Capitulino                                  bool is_read, int error)
2647c7c2ff0cSLuiz Capitulino {
2648573742a5SPeter Maydell     IoOperationType optype;
2649c7c2ff0cSLuiz Capitulino 
2650573742a5SPeter Maydell     optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
2651573742a5SPeter Maydell     qapi_event_send_block_io_error(bdrv_get_device_name(bs), optype, action,
2652c7c2ff0cSLuiz Capitulino                                    bdrv_iostatus_is_enabled(bs),
2653624ff573SLuiz Capitulino                                    error == ENOSPC, strerror(error),
2654624ff573SLuiz Capitulino                                    &error_abort);
2655c7c2ff0cSLuiz Capitulino }
2656c7c2ff0cSLuiz Capitulino 
26573e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
26583e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
26593e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
26603e1caa5fSPaolo Bonzini  */
26613e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
26623e1caa5fSPaolo Bonzini                        bool is_read, int error)
26633e1caa5fSPaolo Bonzini {
26643e1caa5fSPaolo Bonzini     assert(error >= 0);
26652bd3bce8SPaolo Bonzini 
2666a589569fSWenchao Xia     if (action == BLOCK_ERROR_ACTION_STOP) {
26672bd3bce8SPaolo Bonzini         /* First set the iostatus, so that "info block" returns an iostatus
26682bd3bce8SPaolo Bonzini          * that matches the events raised so far (an additional error iostatus
26692bd3bce8SPaolo Bonzini          * is fine, but not a lost one).
26702bd3bce8SPaolo Bonzini          */
26713e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
26722bd3bce8SPaolo Bonzini 
26732bd3bce8SPaolo Bonzini         /* Then raise the request to stop the VM and the event.
26742bd3bce8SPaolo Bonzini          * qemu_system_vmstop_request_prepare has two effects.  First,
26752bd3bce8SPaolo Bonzini          * it ensures that the STOP event always comes after the
26762bd3bce8SPaolo Bonzini          * BLOCK_IO_ERROR event.  Second, it ensures that even if management
26772bd3bce8SPaolo Bonzini          * can observe the STOP event and do a "cont" before the STOP
26782bd3bce8SPaolo Bonzini          * event is issued, the VM will not stop.  In this case, vm_start()
26792bd3bce8SPaolo Bonzini          * also ensures that the STOP/RESUME pair of events is emitted.
26802bd3bce8SPaolo Bonzini          */
26812bd3bce8SPaolo Bonzini         qemu_system_vmstop_request_prepare();
2682c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26832bd3bce8SPaolo Bonzini         qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
26842bd3bce8SPaolo Bonzini     } else {
2685c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26863e1caa5fSPaolo Bonzini     }
26873e1caa5fSPaolo Bonzini }
26883e1caa5fSPaolo Bonzini 
2689b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
2690b338082bSbellard {
2691b338082bSbellard     return bs->read_only;
2692b338082bSbellard }
2693b338082bSbellard 
2694985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
2695985a03b0Sths {
2696985a03b0Sths     return bs->sg;
2697985a03b0Sths }
2698985a03b0Sths 
2699e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
2700e900a7b7SChristoph Hellwig {
2701e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
2702e900a7b7SChristoph Hellwig }
2703e900a7b7SChristoph Hellwig 
2704425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2705425b0148SPaolo Bonzini {
2706425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
270755b110f2SJeff Cody 
270855b110f2SJeff Cody     /* so a reopen() will preserve wce */
270955b110f2SJeff Cody     if (wce) {
271055b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
271155b110f2SJeff Cody     } else {
271255b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
271355b110f2SJeff Cody     }
2714425b0148SPaolo Bonzini }
2715425b0148SPaolo Bonzini 
2716ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
2717ea2384d3Sbellard {
2718ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
2719ea2384d3Sbellard         return 1;
2720ea2384d3Sbellard     return bs->encrypted;
2721ea2384d3Sbellard }
2722ea2384d3Sbellard 
2723c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
2724c0f4ce77Saliguori {
2725c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
2726c0f4ce77Saliguori 
2727c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2728c0f4ce77Saliguori         return 1;
2729c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
2730c0f4ce77Saliguori }
2731c0f4ce77Saliguori 
2732ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
2733ea2384d3Sbellard {
2734ea2384d3Sbellard     int ret;
2735ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
2736ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
2737ea2384d3Sbellard         if (ret < 0)
2738ea2384d3Sbellard             return ret;
2739ea2384d3Sbellard         if (!bs->encrypted)
2740ea2384d3Sbellard             return 0;
2741ea2384d3Sbellard     }
2742fd04a2aeSShahar Havivi     if (!bs->encrypted) {
2743fd04a2aeSShahar Havivi         return -EINVAL;
2744fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2745fd04a2aeSShahar Havivi         return -ENOMEDIUM;
2746fd04a2aeSShahar Havivi     }
2747c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
2748bb5fc20fSaliguori     if (ret < 0) {
2749bb5fc20fSaliguori         bs->valid_key = 0;
2750bb5fc20fSaliguori     } else if (!bs->valid_key) {
2751bb5fc20fSaliguori         bs->valid_key = 1;
2752a7f53e26SMarkus Armbruster         if (bs->blk) {
2753bb5fc20fSaliguori             /* call the change callback now, we skipped it on open */
2754a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
2755a7f53e26SMarkus Armbruster         }
2756bb5fc20fSaliguori     }
2757c0f4ce77Saliguori     return ret;
2758ea2384d3Sbellard }
2759ea2384d3Sbellard 
27604d2855a3SMarkus Armbruster /*
27614d2855a3SMarkus Armbruster  * Provide an encryption key for @bs.
27624d2855a3SMarkus Armbruster  * If @key is non-null:
27634d2855a3SMarkus Armbruster  *     If @bs is not encrypted, fail.
27644d2855a3SMarkus Armbruster  *     Else if the key is invalid, fail.
27654d2855a3SMarkus Armbruster  *     Else set @bs's key to @key, replacing the existing key, if any.
27664d2855a3SMarkus Armbruster  * If @key is null:
27674d2855a3SMarkus Armbruster  *     If @bs is encrypted and still lacks a key, fail.
27684d2855a3SMarkus Armbruster  *     Else do nothing.
27694d2855a3SMarkus Armbruster  * On failure, store an error object through @errp if non-null.
27704d2855a3SMarkus Armbruster  */
27714d2855a3SMarkus Armbruster void bdrv_add_key(BlockDriverState *bs, const char *key, Error **errp)
27724d2855a3SMarkus Armbruster {
27734d2855a3SMarkus Armbruster     if (key) {
27744d2855a3SMarkus Armbruster         if (!bdrv_is_encrypted(bs)) {
277581e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is not encrypted",
277681e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs));
27774d2855a3SMarkus Armbruster         } else if (bdrv_set_key(bs, key) < 0) {
2778c6bd8c70SMarkus Armbruster             error_setg(errp, QERR_INVALID_PASSWORD);
27794d2855a3SMarkus Armbruster         }
27804d2855a3SMarkus Armbruster     } else {
27814d2855a3SMarkus Armbruster         if (bdrv_key_required(bs)) {
2782b1ca6391SMarkus Armbruster             error_set(errp, ERROR_CLASS_DEVICE_ENCRYPTED,
2783b1ca6391SMarkus Armbruster                       "'%s' (%s) is encrypted",
278481e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs),
27854d2855a3SMarkus Armbruster                       bdrv_get_encrypted_filename(bs));
27864d2855a3SMarkus Armbruster         }
27874d2855a3SMarkus Armbruster     }
27884d2855a3SMarkus Armbruster }
27894d2855a3SMarkus Armbruster 
2790f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
2791ea2384d3Sbellard {
2792f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
2793ea2384d3Sbellard }
2794ea2384d3Sbellard 
2795ada42401SStefan Hajnoczi static int qsort_strcmp(const void *a, const void *b)
2796ada42401SStefan Hajnoczi {
2797ada42401SStefan Hajnoczi     return strcmp(a, b);
2798ada42401SStefan Hajnoczi }
2799ada42401SStefan Hajnoczi 
2800ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2801ea2384d3Sbellard                          void *opaque)
2802ea2384d3Sbellard {
2803ea2384d3Sbellard     BlockDriver *drv;
2804e855e4fbSJeff Cody     int count = 0;
2805ada42401SStefan Hajnoczi     int i;
2806e855e4fbSJeff Cody     const char **formats = NULL;
2807ea2384d3Sbellard 
28088a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2809e855e4fbSJeff Cody         if (drv->format_name) {
2810e855e4fbSJeff Cody             bool found = false;
2811e855e4fbSJeff Cody             int i = count;
2812e855e4fbSJeff Cody             while (formats && i && !found) {
2813e855e4fbSJeff Cody                 found = !strcmp(formats[--i], drv->format_name);
2814e855e4fbSJeff Cody             }
2815e855e4fbSJeff Cody 
2816e855e4fbSJeff Cody             if (!found) {
28175839e53bSMarkus Armbruster                 formats = g_renew(const char *, formats, count + 1);
2818e855e4fbSJeff Cody                 formats[count++] = drv->format_name;
2819ea2384d3Sbellard             }
2820ea2384d3Sbellard         }
2821e855e4fbSJeff Cody     }
2822ada42401SStefan Hajnoczi 
2823ada42401SStefan Hajnoczi     qsort(formats, count, sizeof(formats[0]), qsort_strcmp);
2824ada42401SStefan Hajnoczi 
2825ada42401SStefan Hajnoczi     for (i = 0; i < count; i++) {
2826ada42401SStefan Hajnoczi         it(opaque, formats[i]);
2827ada42401SStefan Hajnoczi     }
2828ada42401SStefan Hajnoczi 
2829e855e4fbSJeff Cody     g_free(formats);
2830e855e4fbSJeff Cody }
2831ea2384d3Sbellard 
2832dc364f4cSBenoît Canet /* This function is to find a node in the bs graph */
2833dc364f4cSBenoît Canet BlockDriverState *bdrv_find_node(const char *node_name)
2834dc364f4cSBenoît Canet {
2835dc364f4cSBenoît Canet     BlockDriverState *bs;
2836dc364f4cSBenoît Canet 
2837dc364f4cSBenoît Canet     assert(node_name);
2838dc364f4cSBenoît Canet 
2839dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2840dc364f4cSBenoît Canet         if (!strcmp(node_name, bs->node_name)) {
2841dc364f4cSBenoît Canet             return bs;
2842dc364f4cSBenoît Canet         }
2843dc364f4cSBenoît Canet     }
2844dc364f4cSBenoît Canet     return NULL;
2845dc364f4cSBenoît Canet }
2846dc364f4cSBenoît Canet 
2847c13163fbSBenoît Canet /* Put this QMP function here so it can access the static graph_bdrv_states. */
2848d5a8ee60SAlberto Garcia BlockDeviceInfoList *bdrv_named_nodes_list(Error **errp)
2849c13163fbSBenoît Canet {
2850c13163fbSBenoît Canet     BlockDeviceInfoList *list, *entry;
2851c13163fbSBenoît Canet     BlockDriverState *bs;
2852c13163fbSBenoît Canet 
2853c13163fbSBenoît Canet     list = NULL;
2854c13163fbSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2855d5a8ee60SAlberto Garcia         BlockDeviceInfo *info = bdrv_block_device_info(bs, errp);
2856d5a8ee60SAlberto Garcia         if (!info) {
2857d5a8ee60SAlberto Garcia             qapi_free_BlockDeviceInfoList(list);
2858d5a8ee60SAlberto Garcia             return NULL;
2859d5a8ee60SAlberto Garcia         }
2860c13163fbSBenoît Canet         entry = g_malloc0(sizeof(*entry));
2861d5a8ee60SAlberto Garcia         entry->value = info;
2862c13163fbSBenoît Canet         entry->next = list;
2863c13163fbSBenoît Canet         list = entry;
2864c13163fbSBenoît Canet     }
2865c13163fbSBenoît Canet 
2866c13163fbSBenoît Canet     return list;
2867c13163fbSBenoît Canet }
2868c13163fbSBenoît Canet 
286912d3ba82SBenoît Canet BlockDriverState *bdrv_lookup_bs(const char *device,
287012d3ba82SBenoît Canet                                  const char *node_name,
287112d3ba82SBenoît Canet                                  Error **errp)
287212d3ba82SBenoît Canet {
28737f06d47eSMarkus Armbruster     BlockBackend *blk;
28747f06d47eSMarkus Armbruster     BlockDriverState *bs;
287512d3ba82SBenoît Canet 
287612d3ba82SBenoît Canet     if (device) {
28777f06d47eSMarkus Armbruster         blk = blk_by_name(device);
287812d3ba82SBenoît Canet 
28797f06d47eSMarkus Armbruster         if (blk) {
28807f06d47eSMarkus Armbruster             return blk_bs(blk);
288112d3ba82SBenoît Canet         }
2882dd67fa50SBenoît Canet     }
288312d3ba82SBenoît Canet 
2884dd67fa50SBenoît Canet     if (node_name) {
288512d3ba82SBenoît Canet         bs = bdrv_find_node(node_name);
288612d3ba82SBenoît Canet 
2887dd67fa50SBenoît Canet         if (bs) {
2888dd67fa50SBenoît Canet             return bs;
2889dd67fa50SBenoît Canet         }
289012d3ba82SBenoît Canet     }
289112d3ba82SBenoît Canet 
2892dd67fa50SBenoît Canet     error_setg(errp, "Cannot find device=%s nor node_name=%s",
2893dd67fa50SBenoît Canet                      device ? device : "",
2894dd67fa50SBenoît Canet                      node_name ? node_name : "");
2895dd67fa50SBenoît Canet     return NULL;
289612d3ba82SBenoît Canet }
289712d3ba82SBenoît Canet 
28985a6684d2SJeff Cody /* If 'base' is in the same chain as 'top', return true. Otherwise,
28995a6684d2SJeff Cody  * return false.  If either argument is NULL, return false. */
29005a6684d2SJeff Cody bool bdrv_chain_contains(BlockDriverState *top, BlockDriverState *base)
29015a6684d2SJeff Cody {
29025a6684d2SJeff Cody     while (top && top != base) {
29035a6684d2SJeff Cody         top = top->backing_hd;
29045a6684d2SJeff Cody     }
29055a6684d2SJeff Cody 
29065a6684d2SJeff Cody     return top != NULL;
29075a6684d2SJeff Cody }
29085a6684d2SJeff Cody 
290904df765aSFam Zheng BlockDriverState *bdrv_next_node(BlockDriverState *bs)
291004df765aSFam Zheng {
291104df765aSFam Zheng     if (!bs) {
291204df765aSFam Zheng         return QTAILQ_FIRST(&graph_bdrv_states);
291304df765aSFam Zheng     }
291404df765aSFam Zheng     return QTAILQ_NEXT(bs, node_list);
291504df765aSFam Zheng }
291604df765aSFam Zheng 
29172f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
29182f399b0aSMarkus Armbruster {
29192f399b0aSMarkus Armbruster     if (!bs) {
29202f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
29212f399b0aSMarkus Armbruster     }
2922dc364f4cSBenoît Canet     return QTAILQ_NEXT(bs, device_list);
29232f399b0aSMarkus Armbruster }
29242f399b0aSMarkus Armbruster 
292520a9e77dSFam Zheng const char *bdrv_get_node_name(const BlockDriverState *bs)
292620a9e77dSFam Zheng {
292720a9e77dSFam Zheng     return bs->node_name;
292820a9e77dSFam Zheng }
292920a9e77dSFam Zheng 
29307f06d47eSMarkus Armbruster /* TODO check what callers really want: bs->node_name or blk_name() */
2931bfb197e0SMarkus Armbruster const char *bdrv_get_device_name(const BlockDriverState *bs)
2932ea2384d3Sbellard {
2933bfb197e0SMarkus Armbruster     return bs->blk ? blk_name(bs->blk) : "";
2934ea2384d3Sbellard }
2935ea2384d3Sbellard 
29369b2aa84fSAlberto Garcia /* This can be used to identify nodes that might not have a device
29379b2aa84fSAlberto Garcia  * name associated. Since node and device names live in the same
29389b2aa84fSAlberto Garcia  * namespace, the result is unambiguous. The exception is if both are
29399b2aa84fSAlberto Garcia  * absent, then this returns an empty (non-null) string. */
29409b2aa84fSAlberto Garcia const char *bdrv_get_device_or_node_name(const BlockDriverState *bs)
29419b2aa84fSAlberto Garcia {
29429b2aa84fSAlberto Garcia     return bs->blk ? blk_name(bs->blk) : bs->node_name;
29439b2aa84fSAlberto Garcia }
29449b2aa84fSAlberto Garcia 
2945c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
2946c8433287SMarkus Armbruster {
2947c8433287SMarkus Armbruster     return bs->open_flags;
2948c8433287SMarkus Armbruster }
2949c8433287SMarkus Armbruster 
29503ac21627SPeter Lieven int bdrv_has_zero_init_1(BlockDriverState *bs)
29513ac21627SPeter Lieven {
29523ac21627SPeter Lieven     return 1;
29533ac21627SPeter Lieven }
29543ac21627SPeter Lieven 
2955f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
2956f2feebbdSKevin Wolf {
2957f2feebbdSKevin Wolf     assert(bs->drv);
2958f2feebbdSKevin Wolf 
295911212d8fSPaolo Bonzini     /* If BS is a copy on write image, it is initialized to
296011212d8fSPaolo Bonzini        the contents of the base image, which may not be zeroes.  */
296111212d8fSPaolo Bonzini     if (bs->backing_hd) {
296211212d8fSPaolo Bonzini         return 0;
296311212d8fSPaolo Bonzini     }
2964336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
2965336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
2966f2feebbdSKevin Wolf     }
2967f2feebbdSKevin Wolf 
29683ac21627SPeter Lieven     /* safe default */
29693ac21627SPeter Lieven     return 0;
2970f2feebbdSKevin Wolf }
2971f2feebbdSKevin Wolf 
29724ce78691SPeter Lieven bool bdrv_unallocated_blocks_are_zero(BlockDriverState *bs)
29734ce78691SPeter Lieven {
29744ce78691SPeter Lieven     BlockDriverInfo bdi;
29754ce78691SPeter Lieven 
29764ce78691SPeter Lieven     if (bs->backing_hd) {
29774ce78691SPeter Lieven         return false;
29784ce78691SPeter Lieven     }
29794ce78691SPeter Lieven 
29804ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29814ce78691SPeter Lieven         return bdi.unallocated_blocks_are_zero;
29824ce78691SPeter Lieven     }
29834ce78691SPeter Lieven 
29844ce78691SPeter Lieven     return false;
29854ce78691SPeter Lieven }
29864ce78691SPeter Lieven 
29874ce78691SPeter Lieven bool bdrv_can_write_zeroes_with_unmap(BlockDriverState *bs)
29884ce78691SPeter Lieven {
29894ce78691SPeter Lieven     BlockDriverInfo bdi;
29904ce78691SPeter Lieven 
29914ce78691SPeter Lieven     if (bs->backing_hd || !(bs->open_flags & BDRV_O_UNMAP)) {
29924ce78691SPeter Lieven         return false;
29934ce78691SPeter Lieven     }
29944ce78691SPeter Lieven 
29954ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29964ce78691SPeter Lieven         return bdi.can_write_zeroes_with_unmap;
29974ce78691SPeter Lieven     }
29984ce78691SPeter Lieven 
29994ce78691SPeter Lieven     return false;
30004ce78691SPeter Lieven }
30014ce78691SPeter Lieven 
3002045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
3003045df330Saliguori {
3004045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
3005045df330Saliguori         return bs->backing_file;
3006045df330Saliguori     else if (bs->encrypted)
3007045df330Saliguori         return bs->filename;
3008045df330Saliguori     else
3009045df330Saliguori         return NULL;
3010045df330Saliguori }
3011045df330Saliguori 
301283f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
301383f64091Sbellard                                char *filename, int filename_size)
301483f64091Sbellard {
301583f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
301683f64091Sbellard }
301783f64091Sbellard 
3018faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3019faea38e7Sbellard {
3020faea38e7Sbellard     BlockDriver *drv = bs->drv;
3021faea38e7Sbellard     if (!drv)
302219cb3738Sbellard         return -ENOMEDIUM;
3023faea38e7Sbellard     if (!drv->bdrv_get_info)
3024faea38e7Sbellard         return -ENOTSUP;
3025faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3026faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3027faea38e7Sbellard }
3028faea38e7Sbellard 
3029eae041feSMax Reitz ImageInfoSpecific *bdrv_get_specific_info(BlockDriverState *bs)
3030eae041feSMax Reitz {
3031eae041feSMax Reitz     BlockDriver *drv = bs->drv;
3032eae041feSMax Reitz     if (drv && drv->bdrv_get_specific_info) {
3033eae041feSMax Reitz         return drv->bdrv_get_specific_info(bs);
3034eae041feSMax Reitz     }
3035eae041feSMax Reitz     return NULL;
3036eae041feSMax Reitz }
3037eae041feSMax Reitz 
30388b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
30398b9b0cc2SKevin Wolf {
3040bf736fe3SKevin Wolf     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
30418b9b0cc2SKevin Wolf         return;
30428b9b0cc2SKevin Wolf     }
30438b9b0cc2SKevin Wolf 
3044bf736fe3SKevin Wolf     bs->drv->bdrv_debug_event(bs, event);
304541c695c7SKevin Wolf }
30468b9b0cc2SKevin Wolf 
304741c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
304841c695c7SKevin Wolf                           const char *tag)
304941c695c7SKevin Wolf {
305041c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
305141c695c7SKevin Wolf         bs = bs->file;
305241c695c7SKevin Wolf     }
305341c695c7SKevin Wolf 
305441c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
305541c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
305641c695c7SKevin Wolf     }
305741c695c7SKevin Wolf 
305841c695c7SKevin Wolf     return -ENOTSUP;
305941c695c7SKevin Wolf }
306041c695c7SKevin Wolf 
30614cc70e93SFam Zheng int bdrv_debug_remove_breakpoint(BlockDriverState *bs, const char *tag)
30624cc70e93SFam Zheng {
30634cc70e93SFam Zheng     while (bs && bs->drv && !bs->drv->bdrv_debug_remove_breakpoint) {
30644cc70e93SFam Zheng         bs = bs->file;
30654cc70e93SFam Zheng     }
30664cc70e93SFam Zheng 
30674cc70e93SFam Zheng     if (bs && bs->drv && bs->drv->bdrv_debug_remove_breakpoint) {
30684cc70e93SFam Zheng         return bs->drv->bdrv_debug_remove_breakpoint(bs, tag);
30694cc70e93SFam Zheng     }
30704cc70e93SFam Zheng 
30714cc70e93SFam Zheng     return -ENOTSUP;
30724cc70e93SFam Zheng }
30734cc70e93SFam Zheng 
307441c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
307541c695c7SKevin Wolf {
3076938789eaSMax Reitz     while (bs && (!bs->drv || !bs->drv->bdrv_debug_resume)) {
307741c695c7SKevin Wolf         bs = bs->file;
307841c695c7SKevin Wolf     }
307941c695c7SKevin Wolf 
308041c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
308141c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
308241c695c7SKevin Wolf     }
308341c695c7SKevin Wolf 
308441c695c7SKevin Wolf     return -ENOTSUP;
308541c695c7SKevin Wolf }
308641c695c7SKevin Wolf 
308741c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
308841c695c7SKevin Wolf {
308941c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
309041c695c7SKevin Wolf         bs = bs->file;
309141c695c7SKevin Wolf     }
309241c695c7SKevin Wolf 
309341c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
309441c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
309541c695c7SKevin Wolf     }
309641c695c7SKevin Wolf 
309741c695c7SKevin Wolf     return false;
30988b9b0cc2SKevin Wolf }
30998b9b0cc2SKevin Wolf 
3100199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3101199630b6SBlue Swirl {
3102199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3103199630b6SBlue Swirl }
3104199630b6SBlue Swirl 
3105b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3106b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3107b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3108b1b1d783SJeff Cody  * the CWD rather than the chain. */
3109e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3110e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3111e8a6bb9cSMarcelo Tosatti {
3112b1b1d783SJeff Cody     char *filename_full = NULL;
3113b1b1d783SJeff Cody     char *backing_file_full = NULL;
3114b1b1d783SJeff Cody     char *filename_tmp = NULL;
3115b1b1d783SJeff Cody     int is_protocol = 0;
3116b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3117b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3118b1b1d783SJeff Cody 
3119b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3120e8a6bb9cSMarcelo Tosatti         return NULL;
3121e8a6bb9cSMarcelo Tosatti     }
3122e8a6bb9cSMarcelo Tosatti 
3123b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
3124b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
3125b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
3126b1b1d783SJeff Cody 
3127b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
3128b1b1d783SJeff Cody 
3129b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3130b1b1d783SJeff Cody 
3131b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
3132b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
3133b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3134b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3135b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3136b1b1d783SJeff Cody                 break;
3137b1b1d783SJeff Cody             }
3138e8a6bb9cSMarcelo Tosatti         } else {
3139b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
3140b1b1d783SJeff Cody              * image's filename path */
3141b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3142b1b1d783SJeff Cody                          backing_file);
3143b1b1d783SJeff Cody 
3144b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
3145b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
3146b1b1d783SJeff Cody                 continue;
3147b1b1d783SJeff Cody             }
3148b1b1d783SJeff Cody 
3149b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
3150b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
3151b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3152b1b1d783SJeff Cody                          curr_bs->backing_file);
3153b1b1d783SJeff Cody 
3154b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
3155b1b1d783SJeff Cody                 continue;
3156b1b1d783SJeff Cody             }
3157b1b1d783SJeff Cody 
3158b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
3159b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3160b1b1d783SJeff Cody                 break;
3161b1b1d783SJeff Cody             }
3162e8a6bb9cSMarcelo Tosatti         }
3163e8a6bb9cSMarcelo Tosatti     }
3164e8a6bb9cSMarcelo Tosatti 
3165b1b1d783SJeff Cody     g_free(filename_full);
3166b1b1d783SJeff Cody     g_free(backing_file_full);
3167b1b1d783SJeff Cody     g_free(filename_tmp);
3168b1b1d783SJeff Cody     return retval;
3169e8a6bb9cSMarcelo Tosatti }
3170e8a6bb9cSMarcelo Tosatti 
3171f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
3172f198fd1cSBenoît Canet {
3173f198fd1cSBenoît Canet     if (!bs->drv) {
3174f198fd1cSBenoît Canet         return 0;
3175f198fd1cSBenoît Canet     }
3176f198fd1cSBenoît Canet 
3177f198fd1cSBenoît Canet     if (!bs->backing_hd) {
3178f198fd1cSBenoît Canet         return 0;
3179f198fd1cSBenoît Canet     }
3180f198fd1cSBenoît Canet 
3181f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3182f198fd1cSBenoît Canet }
3183f198fd1cSBenoît Canet 
3184ea2384d3Sbellard void bdrv_init(void)
3185ea2384d3Sbellard {
31865efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
3187ea2384d3Sbellard }
3188ce1a14dcSpbrook 
3189eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
3190eb852011SMarkus Armbruster {
3191eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
3192eb852011SMarkus Armbruster     bdrv_init();
3193eb852011SMarkus Armbruster }
3194eb852011SMarkus Armbruster 
31955a8a30dbSKevin Wolf void bdrv_invalidate_cache(BlockDriverState *bs, Error **errp)
31960f15423cSAnthony Liguori {
31975a8a30dbSKevin Wolf     Error *local_err = NULL;
31985a8a30dbSKevin Wolf     int ret;
31995a8a30dbSKevin Wolf 
32003456a8d1SKevin Wolf     if (!bs->drv)  {
32013456a8d1SKevin Wolf         return;
32020f15423cSAnthony Liguori     }
32033456a8d1SKevin Wolf 
32047ea2d269SAlexey Kardashevskiy     if (!(bs->open_flags & BDRV_O_INCOMING)) {
32057ea2d269SAlexey Kardashevskiy         return;
32067ea2d269SAlexey Kardashevskiy     }
32077ea2d269SAlexey Kardashevskiy     bs->open_flags &= ~BDRV_O_INCOMING;
32087ea2d269SAlexey Kardashevskiy 
32093456a8d1SKevin Wolf     if (bs->drv->bdrv_invalidate_cache) {
32105a8a30dbSKevin Wolf         bs->drv->bdrv_invalidate_cache(bs, &local_err);
32113456a8d1SKevin Wolf     } else if (bs->file) {
32125a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs->file, &local_err);
32135a8a30dbSKevin Wolf     }
32145a8a30dbSKevin Wolf     if (local_err) {
32155a8a30dbSKevin Wolf         error_propagate(errp, local_err);
32165a8a30dbSKevin Wolf         return;
32173456a8d1SKevin Wolf     }
32183456a8d1SKevin Wolf 
32195a8a30dbSKevin Wolf     ret = refresh_total_sectors(bs, bs->total_sectors);
32205a8a30dbSKevin Wolf     if (ret < 0) {
32215a8a30dbSKevin Wolf         error_setg_errno(errp, -ret, "Could not refresh total sector count");
32225a8a30dbSKevin Wolf         return;
32235a8a30dbSKevin Wolf     }
32240f15423cSAnthony Liguori }
32250f15423cSAnthony Liguori 
32265a8a30dbSKevin Wolf void bdrv_invalidate_cache_all(Error **errp)
32270f15423cSAnthony Liguori {
32280f15423cSAnthony Liguori     BlockDriverState *bs;
32295a8a30dbSKevin Wolf     Error *local_err = NULL;
32300f15423cSAnthony Liguori 
3231dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
3232ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
3233ed78cda3SStefan Hajnoczi 
3234ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
32355a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs, &local_err);
3236ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
32375a8a30dbSKevin Wolf         if (local_err) {
32385a8a30dbSKevin Wolf             error_propagate(errp, local_err);
32395a8a30dbSKevin Wolf             return;
32405a8a30dbSKevin Wolf         }
32410f15423cSAnthony Liguori     }
32420f15423cSAnthony Liguori }
32430f15423cSAnthony Liguori 
3244f9f05dc5SKevin Wolf /**************************************************************/
324519cb3738Sbellard /* removable device support */
324619cb3738Sbellard 
324719cb3738Sbellard /**
324819cb3738Sbellard  * Return TRUE if the media is present
324919cb3738Sbellard  */
325019cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
325119cb3738Sbellard {
325219cb3738Sbellard     BlockDriver *drv = bs->drv;
3253a1aff5bfSMarkus Armbruster 
325419cb3738Sbellard     if (!drv)
325519cb3738Sbellard         return 0;
325619cb3738Sbellard     if (!drv->bdrv_is_inserted)
3257a1aff5bfSMarkus Armbruster         return 1;
3258a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
325919cb3738Sbellard }
326019cb3738Sbellard 
326119cb3738Sbellard /**
32628e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
32638e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
326419cb3738Sbellard  */
326519cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
326619cb3738Sbellard {
326719cb3738Sbellard     BlockDriver *drv = bs->drv;
326819cb3738Sbellard 
32698e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
32708e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
32718e49ca46SMarkus Armbruster     }
32728e49ca46SMarkus Armbruster     return -ENOTSUP;
327319cb3738Sbellard }
327419cb3738Sbellard 
327519cb3738Sbellard /**
327619cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
327719cb3738Sbellard  */
3278f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
327919cb3738Sbellard {
328019cb3738Sbellard     BlockDriver *drv = bs->drv;
3281bfb197e0SMarkus Armbruster     const char *device_name;
328219cb3738Sbellard 
3283822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
3284822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
328519cb3738Sbellard     }
32866f382ed2SLuiz Capitulino 
3287bfb197e0SMarkus Armbruster     device_name = bdrv_get_device_name(bs);
3288bfb197e0SMarkus Armbruster     if (device_name[0] != '\0') {
3289bfb197e0SMarkus Armbruster         qapi_event_send_device_tray_moved(device_name,
3290a5ee7bd4SWenchao Xia                                           eject_flag, &error_abort);
32916f382ed2SLuiz Capitulino     }
329219cb3738Sbellard }
329319cb3738Sbellard 
329419cb3738Sbellard /**
329519cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
329619cb3738Sbellard  * to eject it manually).
329719cb3738Sbellard  */
3298025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
329919cb3738Sbellard {
330019cb3738Sbellard     BlockDriver *drv = bs->drv;
330119cb3738Sbellard 
3302025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
3303b8c6d095SStefan Hajnoczi 
3304025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
3305025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
330619cb3738Sbellard     }
330719cb3738Sbellard }
3308985a03b0Sths 
33091b7fd729SPaolo Bonzini void bdrv_set_guest_block_size(BlockDriverState *bs, int align)
33107b6f9300SMarkus Armbruster {
33111b7fd729SPaolo Bonzini     bs->guest_block_size = align;
33127b6f9300SMarkus Armbruster }
33137cd1e32aSlirans@il.ibm.com 
33140db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_find_dirty_bitmap(BlockDriverState *bs, const char *name)
33150db6e54aSFam Zheng {
33160db6e54aSFam Zheng     BdrvDirtyBitmap *bm;
33170db6e54aSFam Zheng 
33180db6e54aSFam Zheng     assert(name);
33190db6e54aSFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
33200db6e54aSFam Zheng         if (bm->name && !strcmp(name, bm->name)) {
33210db6e54aSFam Zheng             return bm;
33220db6e54aSFam Zheng         }
33230db6e54aSFam Zheng     }
33240db6e54aSFam Zheng     return NULL;
33250db6e54aSFam Zheng }
33260db6e54aSFam Zheng 
332720dca810SJohn Snow void bdrv_dirty_bitmap_make_anon(BdrvDirtyBitmap *bitmap)
33280db6e54aSFam Zheng {
33299bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
33300db6e54aSFam Zheng     g_free(bitmap->name);
33310db6e54aSFam Zheng     bitmap->name = NULL;
33320db6e54aSFam Zheng }
33330db6e54aSFam Zheng 
33340db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_create_dirty_bitmap(BlockDriverState *bs,
33355fba6c0eSJohn Snow                                           uint32_t granularity,
33360db6e54aSFam Zheng                                           const char *name,
3337b8afb520SFam Zheng                                           Error **errp)
33387cd1e32aSlirans@il.ibm.com {
33397cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
3340e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
33415fba6c0eSJohn Snow     uint32_t sector_granularity;
3342a55eb92cSJan Kiszka 
334350717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
334450717e94SPaolo Bonzini 
33450db6e54aSFam Zheng     if (name && bdrv_find_dirty_bitmap(bs, name)) {
33460db6e54aSFam Zheng         error_setg(errp, "Bitmap already exists: %s", name);
33470db6e54aSFam Zheng         return NULL;
33480db6e54aSFam Zheng     }
33495fba6c0eSJohn Snow     sector_granularity = granularity >> BDRV_SECTOR_BITS;
33505fba6c0eSJohn Snow     assert(sector_granularity);
335157322b78SMarkus Armbruster     bitmap_size = bdrv_nb_sectors(bs);
3352b8afb520SFam Zheng     if (bitmap_size < 0) {
3353b8afb520SFam Zheng         error_setg_errno(errp, -bitmap_size, "could not get length of device");
3354b8afb520SFam Zheng         errno = -bitmap_size;
3355b8afb520SFam Zheng         return NULL;
3356b8afb520SFam Zheng     }
33575839e53bSMarkus Armbruster     bitmap = g_new0(BdrvDirtyBitmap, 1);
33585fba6c0eSJohn Snow     bitmap->bitmap = hbitmap_alloc(bitmap_size, ctz32(sector_granularity));
3359e74e6b78SJohn Snow     bitmap->size = bitmap_size;
33600db6e54aSFam Zheng     bitmap->name = g_strdup(name);
3361b8e6fb75SJohn Snow     bitmap->disabled = false;
3362e4654d2dSFam Zheng     QLIST_INSERT_HEAD(&bs->dirty_bitmaps, bitmap, list);
3363e4654d2dSFam Zheng     return bitmap;
3364e4654d2dSFam Zheng }
3365e4654d2dSFam Zheng 
33669bd2b08fSJohn Snow bool bdrv_dirty_bitmap_frozen(BdrvDirtyBitmap *bitmap)
33679bd2b08fSJohn Snow {
33689bd2b08fSJohn Snow     return bitmap->successor;
33699bd2b08fSJohn Snow }
33709bd2b08fSJohn Snow 
3371b8e6fb75SJohn Snow bool bdrv_dirty_bitmap_enabled(BdrvDirtyBitmap *bitmap)
3372b8e6fb75SJohn Snow {
33739bd2b08fSJohn Snow     return !(bitmap->disabled || bitmap->successor);
33749bd2b08fSJohn Snow }
33759bd2b08fSJohn Snow 
33769abe3bdcSJohn Snow DirtyBitmapStatus bdrv_dirty_bitmap_status(BdrvDirtyBitmap *bitmap)
33779abe3bdcSJohn Snow {
33789abe3bdcSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33799abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_FROZEN;
33809abe3bdcSJohn Snow     } else if (!bdrv_dirty_bitmap_enabled(bitmap)) {
33819abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_DISABLED;
33829abe3bdcSJohn Snow     } else {
33839abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_ACTIVE;
33849abe3bdcSJohn Snow     }
33859abe3bdcSJohn Snow }
33869abe3bdcSJohn Snow 
33879bd2b08fSJohn Snow /**
33889bd2b08fSJohn Snow  * Create a successor bitmap destined to replace this bitmap after an operation.
33899bd2b08fSJohn Snow  * Requires that the bitmap is not frozen and has no successor.
33909bd2b08fSJohn Snow  */
33919bd2b08fSJohn Snow int bdrv_dirty_bitmap_create_successor(BlockDriverState *bs,
33929bd2b08fSJohn Snow                                        BdrvDirtyBitmap *bitmap, Error **errp)
33939bd2b08fSJohn Snow {
33949bd2b08fSJohn Snow     uint64_t granularity;
33959bd2b08fSJohn Snow     BdrvDirtyBitmap *child;
33969bd2b08fSJohn Snow 
33979bd2b08fSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33989bd2b08fSJohn Snow         error_setg(errp, "Cannot create a successor for a bitmap that is "
33999bd2b08fSJohn Snow                    "currently frozen");
34009bd2b08fSJohn Snow         return -1;
34019bd2b08fSJohn Snow     }
34029bd2b08fSJohn Snow     assert(!bitmap->successor);
34039bd2b08fSJohn Snow 
34049bd2b08fSJohn Snow     /* Create an anonymous successor */
34059bd2b08fSJohn Snow     granularity = bdrv_dirty_bitmap_granularity(bitmap);
34069bd2b08fSJohn Snow     child = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp);
34079bd2b08fSJohn Snow     if (!child) {
34089bd2b08fSJohn Snow         return -1;
34099bd2b08fSJohn Snow     }
34109bd2b08fSJohn Snow 
34119bd2b08fSJohn Snow     /* Successor will be on or off based on our current state. */
34129bd2b08fSJohn Snow     child->disabled = bitmap->disabled;
34139bd2b08fSJohn Snow 
34149bd2b08fSJohn Snow     /* Install the successor and freeze the parent */
34159bd2b08fSJohn Snow     bitmap->successor = child;
34169bd2b08fSJohn Snow     return 0;
34179bd2b08fSJohn Snow }
34189bd2b08fSJohn Snow 
34199bd2b08fSJohn Snow /**
34209bd2b08fSJohn Snow  * For a bitmap with a successor, yield our name to the successor,
34219bd2b08fSJohn Snow  * delete the old bitmap, and return a handle to the new bitmap.
34229bd2b08fSJohn Snow  */
34239bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_dirty_bitmap_abdicate(BlockDriverState *bs,
34249bd2b08fSJohn Snow                                             BdrvDirtyBitmap *bitmap,
34259bd2b08fSJohn Snow                                             Error **errp)
34269bd2b08fSJohn Snow {
34279bd2b08fSJohn Snow     char *name;
34289bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = bitmap->successor;
34299bd2b08fSJohn Snow 
34309bd2b08fSJohn Snow     if (successor == NULL) {
34319bd2b08fSJohn Snow         error_setg(errp, "Cannot relinquish control if "
34329bd2b08fSJohn Snow                    "there's no successor present");
34339bd2b08fSJohn Snow         return NULL;
34349bd2b08fSJohn Snow     }
34359bd2b08fSJohn Snow 
34369bd2b08fSJohn Snow     name = bitmap->name;
34379bd2b08fSJohn Snow     bitmap->name = NULL;
34389bd2b08fSJohn Snow     successor->name = name;
34399bd2b08fSJohn Snow     bitmap->successor = NULL;
34409bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, bitmap);
34419bd2b08fSJohn Snow 
34429bd2b08fSJohn Snow     return successor;
34439bd2b08fSJohn Snow }
34449bd2b08fSJohn Snow 
34459bd2b08fSJohn Snow /**
34469bd2b08fSJohn Snow  * In cases of failure where we can no longer safely delete the parent,
34479bd2b08fSJohn Snow  * we may wish to re-join the parent and child/successor.
34489bd2b08fSJohn Snow  * The merged parent will be un-frozen, but not explicitly re-enabled.
34499bd2b08fSJohn Snow  */
34509bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_reclaim_dirty_bitmap(BlockDriverState *bs,
34519bd2b08fSJohn Snow                                            BdrvDirtyBitmap *parent,
34529bd2b08fSJohn Snow                                            Error **errp)
34539bd2b08fSJohn Snow {
34549bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = parent->successor;
34559bd2b08fSJohn Snow 
34569bd2b08fSJohn Snow     if (!successor) {
34579bd2b08fSJohn Snow         error_setg(errp, "Cannot reclaim a successor when none is present");
34589bd2b08fSJohn Snow         return NULL;
34599bd2b08fSJohn Snow     }
34609bd2b08fSJohn Snow 
34619bd2b08fSJohn Snow     if (!hbitmap_merge(parent->bitmap, successor->bitmap)) {
34629bd2b08fSJohn Snow         error_setg(errp, "Merging of parent and successor bitmap failed");
34639bd2b08fSJohn Snow         return NULL;
34649bd2b08fSJohn Snow     }
34659bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, successor);
34669bd2b08fSJohn Snow     parent->successor = NULL;
34679bd2b08fSJohn Snow 
34689bd2b08fSJohn Snow     return parent;
3469b8e6fb75SJohn Snow }
3470b8e6fb75SJohn Snow 
3471ce1ffea8SJohn Snow /**
3472ce1ffea8SJohn Snow  * Truncates _all_ bitmaps attached to a BDS.
3473ce1ffea8SJohn Snow  */
3474ce1ffea8SJohn Snow static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs)
3475ce1ffea8SJohn Snow {
3476ce1ffea8SJohn Snow     BdrvDirtyBitmap *bitmap;
3477ce1ffea8SJohn Snow     uint64_t size = bdrv_nb_sectors(bs);
3478ce1ffea8SJohn Snow 
3479ce1ffea8SJohn Snow     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
348006207b0fSJohn Snow         assert(!bdrv_dirty_bitmap_frozen(bitmap));
3481ce1ffea8SJohn Snow         hbitmap_truncate(bitmap->bitmap, size);
34825270b6a0SJohn Snow         bitmap->size = size;
3483ce1ffea8SJohn Snow     }
3484ce1ffea8SJohn Snow }
3485ce1ffea8SJohn Snow 
3486e4654d2dSFam Zheng void bdrv_release_dirty_bitmap(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
3487e4654d2dSFam Zheng {
3488e4654d2dSFam Zheng     BdrvDirtyBitmap *bm, *next;
3489e4654d2dSFam Zheng     QLIST_FOREACH_SAFE(bm, &bs->dirty_bitmaps, list, next) {
3490e4654d2dSFam Zheng         if (bm == bitmap) {
34919bd2b08fSJohn Snow             assert(!bdrv_dirty_bitmap_frozen(bm));
3492e4654d2dSFam Zheng             QLIST_REMOVE(bitmap, list);
3493e4654d2dSFam Zheng             hbitmap_free(bitmap->bitmap);
34940db6e54aSFam Zheng             g_free(bitmap->name);
3495e4654d2dSFam Zheng             g_free(bitmap);
3496e4654d2dSFam Zheng             return;
34977cd1e32aSlirans@il.ibm.com         }
34987cd1e32aSlirans@il.ibm.com     }
34997cd1e32aSlirans@il.ibm.com }
35007cd1e32aSlirans@il.ibm.com 
3501b8e6fb75SJohn Snow void bdrv_disable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3502b8e6fb75SJohn Snow {
35039bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3504b8e6fb75SJohn Snow     bitmap->disabled = true;
3505b8e6fb75SJohn Snow }
3506b8e6fb75SJohn Snow 
3507b8e6fb75SJohn Snow void bdrv_enable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3508b8e6fb75SJohn Snow {
35099bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3510b8e6fb75SJohn Snow     bitmap->disabled = false;
3511b8e6fb75SJohn Snow }
3512b8e6fb75SJohn Snow 
351321b56835SFam Zheng BlockDirtyInfoList *bdrv_query_dirty_bitmaps(BlockDriverState *bs)
351421b56835SFam Zheng {
351521b56835SFam Zheng     BdrvDirtyBitmap *bm;
351621b56835SFam Zheng     BlockDirtyInfoList *list = NULL;
351721b56835SFam Zheng     BlockDirtyInfoList **plist = &list;
351821b56835SFam Zheng 
351921b56835SFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
35205839e53bSMarkus Armbruster         BlockDirtyInfo *info = g_new0(BlockDirtyInfo, 1);
35215839e53bSMarkus Armbruster         BlockDirtyInfoList *entry = g_new0(BlockDirtyInfoList, 1);
352220dca810SJohn Snow         info->count = bdrv_get_dirty_count(bm);
3523592fdd02SJohn Snow         info->granularity = bdrv_dirty_bitmap_granularity(bm);
35240db6e54aSFam Zheng         info->has_name = !!bm->name;
35250db6e54aSFam Zheng         info->name = g_strdup(bm->name);
35269abe3bdcSJohn Snow         info->status = bdrv_dirty_bitmap_status(bm);
352721b56835SFam Zheng         entry->value = info;
352821b56835SFam Zheng         *plist = entry;
352921b56835SFam Zheng         plist = &entry->next;
353021b56835SFam Zheng     }
353121b56835SFam Zheng 
353221b56835SFam Zheng     return list;
353321b56835SFam Zheng }
353421b56835SFam Zheng 
3535e4654d2dSFam Zheng int bdrv_get_dirty(BlockDriverState *bs, BdrvDirtyBitmap *bitmap, int64_t sector)
35367cd1e32aSlirans@il.ibm.com {
3537e4654d2dSFam Zheng     if (bitmap) {
3538e4654d2dSFam Zheng         return hbitmap_get(bitmap->bitmap, sector);
35397cd1e32aSlirans@il.ibm.com     } else {
35407cd1e32aSlirans@il.ibm.com         return 0;
35417cd1e32aSlirans@il.ibm.com     }
35427cd1e32aSlirans@il.ibm.com }
35437cd1e32aSlirans@il.ibm.com 
3544341ebc2fSJohn Snow /**
3545341ebc2fSJohn Snow  * Chooses a default granularity based on the existing cluster size,
3546341ebc2fSJohn Snow  * but clamped between [4K, 64K]. Defaults to 64K in the case that there
3547341ebc2fSJohn Snow  * is no cluster size information available.
3548341ebc2fSJohn Snow  */
3549341ebc2fSJohn Snow uint32_t bdrv_get_default_bitmap_granularity(BlockDriverState *bs)
3550341ebc2fSJohn Snow {
3551341ebc2fSJohn Snow     BlockDriverInfo bdi;
3552341ebc2fSJohn Snow     uint32_t granularity;
3553341ebc2fSJohn Snow 
3554341ebc2fSJohn Snow     if (bdrv_get_info(bs, &bdi) >= 0 && bdi.cluster_size > 0) {
3555341ebc2fSJohn Snow         granularity = MAX(4096, bdi.cluster_size);
3556341ebc2fSJohn Snow         granularity = MIN(65536, granularity);
3557341ebc2fSJohn Snow     } else {
3558341ebc2fSJohn Snow         granularity = 65536;
3559341ebc2fSJohn Snow     }
3560341ebc2fSJohn Snow 
3561341ebc2fSJohn Snow     return granularity;
3562341ebc2fSJohn Snow }
3563341ebc2fSJohn Snow 
3564592fdd02SJohn Snow uint32_t bdrv_dirty_bitmap_granularity(BdrvDirtyBitmap *bitmap)
3565592fdd02SJohn Snow {
3566592fdd02SJohn Snow     return BDRV_SECTOR_SIZE << hbitmap_granularity(bitmap->bitmap);
3567592fdd02SJohn Snow }
3568592fdd02SJohn Snow 
356920dca810SJohn Snow void bdrv_dirty_iter_init(BdrvDirtyBitmap *bitmap, HBitmapIter *hbi)
35701755da16SPaolo Bonzini {
3571e4654d2dSFam Zheng     hbitmap_iter_init(hbi, bitmap->bitmap, 0);
35721755da16SPaolo Bonzini }
35731755da16SPaolo Bonzini 
357420dca810SJohn Snow void bdrv_set_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3575c4237dfaSVladimir Sementsov-Ogievskiy                            int64_t cur_sector, int nr_sectors)
3576c4237dfaSVladimir Sementsov-Ogievskiy {
3577b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3578c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3579c4237dfaSVladimir Sementsov-Ogievskiy }
3580c4237dfaSVladimir Sementsov-Ogievskiy 
358120dca810SJohn Snow void bdrv_reset_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3582c4237dfaSVladimir Sementsov-Ogievskiy                              int64_t cur_sector, int nr_sectors)
3583c4237dfaSVladimir Sementsov-Ogievskiy {
3584b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3585c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_reset(bitmap->bitmap, cur_sector, nr_sectors);
3586c4237dfaSVladimir Sementsov-Ogievskiy }
3587c4237dfaSVladimir Sementsov-Ogievskiy 
3588e74e6b78SJohn Snow void bdrv_clear_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3589e74e6b78SJohn Snow {
3590e74e6b78SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3591c6a8c328SWen Congyang     hbitmap_reset_all(bitmap->bitmap);
3592e74e6b78SJohn Snow }
3593e74e6b78SJohn Snow 
3594e0c47b6cSStefan Hajnoczi void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
35951755da16SPaolo Bonzini                     int nr_sectors)
35961755da16SPaolo Bonzini {
3597e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
3598e4654d2dSFam Zheng     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
3599b8e6fb75SJohn Snow         if (!bdrv_dirty_bitmap_enabled(bitmap)) {
3600b8e6fb75SJohn Snow             continue;
3601b8e6fb75SJohn Snow         }
3602e4654d2dSFam Zheng         hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3603e4654d2dSFam Zheng     }
36041755da16SPaolo Bonzini }
36051755da16SPaolo Bonzini 
3606d58d8453SJohn Snow /**
3607d58d8453SJohn Snow  * Advance an HBitmapIter to an arbitrary offset.
3608d58d8453SJohn Snow  */
3609d58d8453SJohn Snow void bdrv_set_dirty_iter(HBitmapIter *hbi, int64_t offset)
3610d58d8453SJohn Snow {
3611d58d8453SJohn Snow     assert(hbi->hb);
3612d58d8453SJohn Snow     hbitmap_iter_init(hbi, hbi->hb, offset);
3613d58d8453SJohn Snow }
3614d58d8453SJohn Snow 
361520dca810SJohn Snow int64_t bdrv_get_dirty_count(BdrvDirtyBitmap *bitmap)
3616aaa0eb75SLiran Schour {
3617e4654d2dSFam Zheng     return hbitmap_count(bitmap->bitmap);
3618aaa0eb75SLiran Schour }
3619f88e1a42SJes Sorensen 
36209fcb0251SFam Zheng /* Get a reference to bs */
36219fcb0251SFam Zheng void bdrv_ref(BlockDriverState *bs)
36229fcb0251SFam Zheng {
36239fcb0251SFam Zheng     bs->refcnt++;
36249fcb0251SFam Zheng }
36259fcb0251SFam Zheng 
36269fcb0251SFam Zheng /* Release a previously grabbed reference to bs.
36279fcb0251SFam Zheng  * If after releasing, reference count is zero, the BlockDriverState is
36289fcb0251SFam Zheng  * deleted. */
36299fcb0251SFam Zheng void bdrv_unref(BlockDriverState *bs)
36309fcb0251SFam Zheng {
36319a4d5ca6SJeff Cody     if (!bs) {
36329a4d5ca6SJeff Cody         return;
36339a4d5ca6SJeff Cody     }
36349fcb0251SFam Zheng     assert(bs->refcnt > 0);
36359fcb0251SFam Zheng     if (--bs->refcnt == 0) {
36369fcb0251SFam Zheng         bdrv_delete(bs);
36379fcb0251SFam Zheng     }
36389fcb0251SFam Zheng }
36399fcb0251SFam Zheng 
3640fbe40ff7SFam Zheng struct BdrvOpBlocker {
3641fbe40ff7SFam Zheng     Error *reason;
3642fbe40ff7SFam Zheng     QLIST_ENTRY(BdrvOpBlocker) list;
3643fbe40ff7SFam Zheng };
3644fbe40ff7SFam Zheng 
3645fbe40ff7SFam Zheng bool bdrv_op_is_blocked(BlockDriverState *bs, BlockOpType op, Error **errp)
3646fbe40ff7SFam Zheng {
3647fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3648fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3649fbe40ff7SFam Zheng     if (!QLIST_EMPTY(&bs->op_blockers[op])) {
3650fbe40ff7SFam Zheng         blocker = QLIST_FIRST(&bs->op_blockers[op]);
3651fbe40ff7SFam Zheng         if (errp) {
365281e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is busy: %s",
365381e5f78aSAlberto Garcia                        bdrv_get_device_or_node_name(bs),
3654bfb197e0SMarkus Armbruster                        error_get_pretty(blocker->reason));
3655fbe40ff7SFam Zheng         }
3656fbe40ff7SFam Zheng         return true;
3657fbe40ff7SFam Zheng     }
3658fbe40ff7SFam Zheng     return false;
3659fbe40ff7SFam Zheng }
3660fbe40ff7SFam Zheng 
3661fbe40ff7SFam Zheng void bdrv_op_block(BlockDriverState *bs, BlockOpType op, Error *reason)
3662fbe40ff7SFam Zheng {
3663fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3664fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3665fbe40ff7SFam Zheng 
36665839e53bSMarkus Armbruster     blocker = g_new0(BdrvOpBlocker, 1);
3667fbe40ff7SFam Zheng     blocker->reason = reason;
3668fbe40ff7SFam Zheng     QLIST_INSERT_HEAD(&bs->op_blockers[op], blocker, list);
3669fbe40ff7SFam Zheng }
3670fbe40ff7SFam Zheng 
3671fbe40ff7SFam Zheng void bdrv_op_unblock(BlockDriverState *bs, BlockOpType op, Error *reason)
3672fbe40ff7SFam Zheng {
3673fbe40ff7SFam Zheng     BdrvOpBlocker *blocker, *next;
3674fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3675fbe40ff7SFam Zheng     QLIST_FOREACH_SAFE(blocker, &bs->op_blockers[op], list, next) {
3676fbe40ff7SFam Zheng         if (blocker->reason == reason) {
3677fbe40ff7SFam Zheng             QLIST_REMOVE(blocker, list);
3678fbe40ff7SFam Zheng             g_free(blocker);
3679fbe40ff7SFam Zheng         }
3680fbe40ff7SFam Zheng     }
3681fbe40ff7SFam Zheng }
3682fbe40ff7SFam Zheng 
3683fbe40ff7SFam Zheng void bdrv_op_block_all(BlockDriverState *bs, Error *reason)
3684fbe40ff7SFam Zheng {
3685fbe40ff7SFam Zheng     int i;
3686fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3687fbe40ff7SFam Zheng         bdrv_op_block(bs, i, reason);
3688fbe40ff7SFam Zheng     }
3689fbe40ff7SFam Zheng }
3690fbe40ff7SFam Zheng 
3691fbe40ff7SFam Zheng void bdrv_op_unblock_all(BlockDriverState *bs, Error *reason)
3692fbe40ff7SFam Zheng {
3693fbe40ff7SFam Zheng     int i;
3694fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3695fbe40ff7SFam Zheng         bdrv_op_unblock(bs, i, reason);
3696fbe40ff7SFam Zheng     }
3697fbe40ff7SFam Zheng }
3698fbe40ff7SFam Zheng 
3699fbe40ff7SFam Zheng bool bdrv_op_blocker_is_empty(BlockDriverState *bs)
3700fbe40ff7SFam Zheng {
3701fbe40ff7SFam Zheng     int i;
3702fbe40ff7SFam Zheng 
3703fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3704fbe40ff7SFam Zheng         if (!QLIST_EMPTY(&bs->op_blockers[i])) {
3705fbe40ff7SFam Zheng             return false;
3706fbe40ff7SFam Zheng         }
3707fbe40ff7SFam Zheng     }
3708fbe40ff7SFam Zheng     return true;
3709fbe40ff7SFam Zheng }
3710fbe40ff7SFam Zheng 
371128a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
371228a7282aSLuiz Capitulino {
3713d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
371458e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
371528a7282aSLuiz Capitulino }
371628a7282aSLuiz Capitulino 
371728a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
371828a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
371928a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
372028a7282aSLuiz Capitulino {
3721d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
372292aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
372392aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
372492aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
372528a7282aSLuiz Capitulino }
372628a7282aSLuiz Capitulino 
372728a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
372828a7282aSLuiz Capitulino {
3729d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
373028a7282aSLuiz Capitulino }
373128a7282aSLuiz Capitulino 
373228a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
373328a7282aSLuiz Capitulino {
373428a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
373558e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
37363bd293c3SPaolo Bonzini         if (bs->job) {
37373bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
37383bd293c3SPaolo Bonzini         }
373928a7282aSLuiz Capitulino     }
374028a7282aSLuiz Capitulino }
374128a7282aSLuiz Capitulino 
374228a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
374328a7282aSLuiz Capitulino {
37443e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
37453e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
374658e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
374758e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
374828a7282aSLuiz Capitulino     }
374928a7282aSLuiz Capitulino }
375028a7282aSLuiz Capitulino 
3751d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
3752f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
3753f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
3754f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
3755f88e1a42SJes Sorensen {
375683d0521aSChunyan Liu     QemuOptsList *create_opts = NULL;
375783d0521aSChunyan Liu     QemuOpts *opts = NULL;
375883d0521aSChunyan Liu     const char *backing_fmt, *backing_file;
375983d0521aSChunyan Liu     int64_t size;
3760f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
3761cc84d90fSMax Reitz     Error *local_err = NULL;
3762f88e1a42SJes Sorensen     int ret = 0;
3763f88e1a42SJes Sorensen 
3764f88e1a42SJes Sorensen     /* Find driver and parse its options */
3765f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
3766f88e1a42SJes Sorensen     if (!drv) {
376771c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
3768d92ada22SLuiz Capitulino         return;
3769f88e1a42SJes Sorensen     }
3770f88e1a42SJes Sorensen 
3771b65a5e12SMax Reitz     proto_drv = bdrv_find_protocol(filename, true, errp);
3772f88e1a42SJes Sorensen     if (!proto_drv) {
3773d92ada22SLuiz Capitulino         return;
3774f88e1a42SJes Sorensen     }
3775f88e1a42SJes Sorensen 
3776c6149724SMax Reitz     if (!drv->create_opts) {
3777c6149724SMax Reitz         error_setg(errp, "Format driver '%s' does not support image creation",
3778c6149724SMax Reitz                    drv->format_name);
3779c6149724SMax Reitz         return;
3780c6149724SMax Reitz     }
3781c6149724SMax Reitz 
3782c6149724SMax Reitz     if (!proto_drv->create_opts) {
3783c6149724SMax Reitz         error_setg(errp, "Protocol driver '%s' does not support image creation",
3784c6149724SMax Reitz                    proto_drv->format_name);
3785c6149724SMax Reitz         return;
3786c6149724SMax Reitz     }
3787c6149724SMax Reitz 
3788c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, drv->create_opts);
3789c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, proto_drv->create_opts);
3790f88e1a42SJes Sorensen 
3791f88e1a42SJes Sorensen     /* Create parameter list with default values */
379283d0521aSChunyan Liu     opts = qemu_opts_create(create_opts, NULL, 0, &error_abort);
379339101f25SMarkus Armbruster     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, img_size, &error_abort);
3794f88e1a42SJes Sorensen 
3795f88e1a42SJes Sorensen     /* Parse -o options */
3796f88e1a42SJes Sorensen     if (options) {
3797dc523cd3SMarkus Armbruster         qemu_opts_do_parse(opts, options, NULL, &local_err);
3798dc523cd3SMarkus Armbruster         if (local_err) {
3799dc523cd3SMarkus Armbruster             error_report_err(local_err);
3800dc523cd3SMarkus Armbruster             local_err = NULL;
380183d0521aSChunyan Liu             error_setg(errp, "Invalid options for file format '%s'", fmt);
3802f88e1a42SJes Sorensen             goto out;
3803f88e1a42SJes Sorensen         }
3804f88e1a42SJes Sorensen     }
3805f88e1a42SJes Sorensen 
3806f88e1a42SJes Sorensen     if (base_filename) {
3807f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, base_filename, &local_err);
38086be4194bSMarkus Armbruster         if (local_err) {
380971c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
381071c79813SLuiz Capitulino                        fmt);
3811f88e1a42SJes Sorensen             goto out;
3812f88e1a42SJes Sorensen         }
3813f88e1a42SJes Sorensen     }
3814f88e1a42SJes Sorensen 
3815f88e1a42SJes Sorensen     if (base_fmt) {
3816f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FMT, base_fmt, &local_err);
38176be4194bSMarkus Armbruster         if (local_err) {
381871c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
381971c79813SLuiz Capitulino                              "format '%s'", fmt);
3820f88e1a42SJes Sorensen             goto out;
3821f88e1a42SJes Sorensen         }
3822f88e1a42SJes Sorensen     }
3823f88e1a42SJes Sorensen 
382483d0521aSChunyan Liu     backing_file = qemu_opt_get(opts, BLOCK_OPT_BACKING_FILE);
382583d0521aSChunyan Liu     if (backing_file) {
382683d0521aSChunyan Liu         if (!strcmp(filename, backing_file)) {
382771c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
382871c79813SLuiz Capitulino                              "same filename as the backing file");
3829792da93aSJes Sorensen             goto out;
3830792da93aSJes Sorensen         }
3831792da93aSJes Sorensen     }
3832792da93aSJes Sorensen 
383383d0521aSChunyan Liu     backing_fmt = qemu_opt_get(opts, BLOCK_OPT_BACKING_FMT);
3834f88e1a42SJes Sorensen 
3835f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
3836f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
383783d0521aSChunyan Liu     size = qemu_opt_get_size(opts, BLOCK_OPT_SIZE, 0);
383883d0521aSChunyan Liu     if (size == -1) {
383983d0521aSChunyan Liu         if (backing_file) {
384066f6b814SMax Reitz             BlockDriverState *bs;
384129168018SMax Reitz             char *full_backing = g_new0(char, PATH_MAX);
384252bf1e72SMarkus Armbruster             int64_t size;
384363090dacSPaolo Bonzini             int back_flags;
3844e6641719SMax Reitz             QDict *backing_options = NULL;
384563090dacSPaolo Bonzini 
384629168018SMax Reitz             bdrv_get_full_backing_filename_from_filename(filename, backing_file,
384729168018SMax Reitz                                                          full_backing, PATH_MAX,
384829168018SMax Reitz                                                          &local_err);
384929168018SMax Reitz             if (local_err) {
385029168018SMax Reitz                 g_free(full_backing);
385129168018SMax Reitz                 goto out;
385229168018SMax Reitz             }
385329168018SMax Reitz 
385463090dacSPaolo Bonzini             /* backing files always opened read-only */
385563090dacSPaolo Bonzini             back_flags =
385663090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
3857f88e1a42SJes Sorensen 
3858e6641719SMax Reitz             if (backing_fmt) {
3859e6641719SMax Reitz                 backing_options = qdict_new();
3860e6641719SMax Reitz                 qdict_put(backing_options, "driver",
3861e6641719SMax Reitz                           qstring_from_str(backing_fmt));
3862e6641719SMax Reitz             }
3863e6641719SMax Reitz 
3864f67503e5SMax Reitz             bs = NULL;
3865e6641719SMax Reitz             ret = bdrv_open(&bs, full_backing, NULL, backing_options,
38666ebf9aa2SMax Reitz                             back_flags, &local_err);
386729168018SMax Reitz             g_free(full_backing);
3868f88e1a42SJes Sorensen             if (ret < 0) {
3869f88e1a42SJes Sorensen                 goto out;
3870f88e1a42SJes Sorensen             }
387152bf1e72SMarkus Armbruster             size = bdrv_getlength(bs);
387252bf1e72SMarkus Armbruster             if (size < 0) {
387352bf1e72SMarkus Armbruster                 error_setg_errno(errp, -size, "Could not get size of '%s'",
387452bf1e72SMarkus Armbruster                                  backing_file);
387552bf1e72SMarkus Armbruster                 bdrv_unref(bs);
387652bf1e72SMarkus Armbruster                 goto out;
387752bf1e72SMarkus Armbruster             }
3878f88e1a42SJes Sorensen 
387939101f25SMarkus Armbruster             qemu_opt_set_number(opts, BLOCK_OPT_SIZE, size, &error_abort);
388066f6b814SMax Reitz 
388166f6b814SMax Reitz             bdrv_unref(bs);
3882f88e1a42SJes Sorensen         } else {
388371c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
3884f88e1a42SJes Sorensen             goto out;
3885f88e1a42SJes Sorensen         }
3886f88e1a42SJes Sorensen     }
3887f88e1a42SJes Sorensen 
3888f382d43aSMiroslav Rezanina     if (!quiet) {
3889f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
389043c5d8f8SFam Zheng         qemu_opts_print(opts, " ");
3891f88e1a42SJes Sorensen         puts("");
3892f382d43aSMiroslav Rezanina     }
389383d0521aSChunyan Liu 
3894c282e1fdSChunyan Liu     ret = bdrv_create(drv, filename, opts, &local_err);
389583d0521aSChunyan Liu 
3896cc84d90fSMax Reitz     if (ret == -EFBIG) {
3897cc84d90fSMax Reitz         /* This is generally a better message than whatever the driver would
3898cc84d90fSMax Reitz          * deliver (especially because of the cluster_size_hint), since that
3899cc84d90fSMax Reitz          * is most probably not much different from "image too large". */
3900f3f4d2c0SKevin Wolf         const char *cluster_size_hint = "";
390183d0521aSChunyan Liu         if (qemu_opt_get_size(opts, BLOCK_OPT_CLUSTER_SIZE, 0)) {
3902f3f4d2c0SKevin Wolf             cluster_size_hint = " (try using a larger cluster size)";
3903f3f4d2c0SKevin Wolf         }
3904cc84d90fSMax Reitz         error_setg(errp, "The image size is too large for file format '%s'"
3905cc84d90fSMax Reitz                    "%s", fmt, cluster_size_hint);
3906cc84d90fSMax Reitz         error_free(local_err);
3907cc84d90fSMax Reitz         local_err = NULL;
3908f88e1a42SJes Sorensen     }
3909f88e1a42SJes Sorensen 
3910f88e1a42SJes Sorensen out:
391183d0521aSChunyan Liu     qemu_opts_del(opts);
391283d0521aSChunyan Liu     qemu_opts_free(create_opts);
391384d18f06SMarkus Armbruster     if (local_err) {
3914cc84d90fSMax Reitz         error_propagate(errp, local_err);
3915cc84d90fSMax Reitz     }
3916f88e1a42SJes Sorensen }
391785d126f3SStefan Hajnoczi 
391885d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
391985d126f3SStefan Hajnoczi {
3920dcd04228SStefan Hajnoczi     return bs->aio_context;
3921dcd04228SStefan Hajnoczi }
3922dcd04228SStefan Hajnoczi 
3923dcd04228SStefan Hajnoczi void bdrv_detach_aio_context(BlockDriverState *bs)
3924dcd04228SStefan Hajnoczi {
392533384421SMax Reitz     BdrvAioNotifier *baf;
392633384421SMax Reitz 
3927dcd04228SStefan Hajnoczi     if (!bs->drv) {
3928dcd04228SStefan Hajnoczi         return;
3929dcd04228SStefan Hajnoczi     }
3930dcd04228SStefan Hajnoczi 
393133384421SMax Reitz     QLIST_FOREACH(baf, &bs->aio_notifiers, list) {
393233384421SMax Reitz         baf->detach_aio_context(baf->opaque);
393333384421SMax Reitz     }
393433384421SMax Reitz 
393513af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39360e5b0a2dSBenoît Canet         throttle_timers_detach_aio_context(&bs->throttle_timers);
393713af91ebSStefan Hajnoczi     }
3938dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_detach_aio_context) {
3939dcd04228SStefan Hajnoczi         bs->drv->bdrv_detach_aio_context(bs);
3940dcd04228SStefan Hajnoczi     }
3941dcd04228SStefan Hajnoczi     if (bs->file) {
3942dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->file);
3943dcd04228SStefan Hajnoczi     }
3944dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3945dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->backing_hd);
3946dcd04228SStefan Hajnoczi     }
3947dcd04228SStefan Hajnoczi 
3948dcd04228SStefan Hajnoczi     bs->aio_context = NULL;
3949dcd04228SStefan Hajnoczi }
3950dcd04228SStefan Hajnoczi 
3951dcd04228SStefan Hajnoczi void bdrv_attach_aio_context(BlockDriverState *bs,
3952dcd04228SStefan Hajnoczi                              AioContext *new_context)
3953dcd04228SStefan Hajnoczi {
395433384421SMax Reitz     BdrvAioNotifier *ban;
395533384421SMax Reitz 
3956dcd04228SStefan Hajnoczi     if (!bs->drv) {
3957dcd04228SStefan Hajnoczi         return;
3958dcd04228SStefan Hajnoczi     }
3959dcd04228SStefan Hajnoczi 
3960dcd04228SStefan Hajnoczi     bs->aio_context = new_context;
3961dcd04228SStefan Hajnoczi 
3962dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3963dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->backing_hd, new_context);
3964dcd04228SStefan Hajnoczi     }
3965dcd04228SStefan Hajnoczi     if (bs->file) {
3966dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->file, new_context);
3967dcd04228SStefan Hajnoczi     }
3968dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_attach_aio_context) {
3969dcd04228SStefan Hajnoczi         bs->drv->bdrv_attach_aio_context(bs, new_context);
3970dcd04228SStefan Hajnoczi     }
397113af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39720e5b0a2dSBenoît Canet         throttle_timers_attach_aio_context(&bs->throttle_timers, new_context);
397313af91ebSStefan Hajnoczi     }
397433384421SMax Reitz 
397533384421SMax Reitz     QLIST_FOREACH(ban, &bs->aio_notifiers, list) {
397633384421SMax Reitz         ban->attached_aio_context(new_context, ban->opaque);
397733384421SMax Reitz     }
3978dcd04228SStefan Hajnoczi }
3979dcd04228SStefan Hajnoczi 
3980dcd04228SStefan Hajnoczi void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
3981dcd04228SStefan Hajnoczi {
398253ec73e2SFam Zheng     bdrv_drain(bs); /* ensure there are no in-flight requests */
3983dcd04228SStefan Hajnoczi 
3984dcd04228SStefan Hajnoczi     bdrv_detach_aio_context(bs);
3985dcd04228SStefan Hajnoczi 
3986dcd04228SStefan Hajnoczi     /* This function executes in the old AioContext so acquire the new one in
3987dcd04228SStefan Hajnoczi      * case it runs in a different thread.
3988dcd04228SStefan Hajnoczi      */
3989dcd04228SStefan Hajnoczi     aio_context_acquire(new_context);
3990dcd04228SStefan Hajnoczi     bdrv_attach_aio_context(bs, new_context);
3991dcd04228SStefan Hajnoczi     aio_context_release(new_context);
399285d126f3SStefan Hajnoczi }
3993d616b224SStefan Hajnoczi 
399433384421SMax Reitz void bdrv_add_aio_context_notifier(BlockDriverState *bs,
399533384421SMax Reitz         void (*attached_aio_context)(AioContext *new_context, void *opaque),
399633384421SMax Reitz         void (*detach_aio_context)(void *opaque), void *opaque)
399733384421SMax Reitz {
399833384421SMax Reitz     BdrvAioNotifier *ban = g_new(BdrvAioNotifier, 1);
399933384421SMax Reitz     *ban = (BdrvAioNotifier){
400033384421SMax Reitz         .attached_aio_context = attached_aio_context,
400133384421SMax Reitz         .detach_aio_context   = detach_aio_context,
400233384421SMax Reitz         .opaque               = opaque
400333384421SMax Reitz     };
400433384421SMax Reitz 
400533384421SMax Reitz     QLIST_INSERT_HEAD(&bs->aio_notifiers, ban, list);
400633384421SMax Reitz }
400733384421SMax Reitz 
400833384421SMax Reitz void bdrv_remove_aio_context_notifier(BlockDriverState *bs,
400933384421SMax Reitz                                       void (*attached_aio_context)(AioContext *,
401033384421SMax Reitz                                                                    void *),
401133384421SMax Reitz                                       void (*detach_aio_context)(void *),
401233384421SMax Reitz                                       void *opaque)
401333384421SMax Reitz {
401433384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
401533384421SMax Reitz 
401633384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
401733384421SMax Reitz         if (ban->attached_aio_context == attached_aio_context &&
401833384421SMax Reitz             ban->detach_aio_context   == detach_aio_context   &&
401933384421SMax Reitz             ban->opaque               == opaque)
402033384421SMax Reitz         {
402133384421SMax Reitz             QLIST_REMOVE(ban, list);
402233384421SMax Reitz             g_free(ban);
402333384421SMax Reitz 
402433384421SMax Reitz             return;
402533384421SMax Reitz         }
402633384421SMax Reitz     }
402733384421SMax Reitz 
402833384421SMax Reitz     abort();
402933384421SMax Reitz }
403033384421SMax Reitz 
403177485434SMax Reitz int bdrv_amend_options(BlockDriverState *bs, QemuOpts *opts,
403277485434SMax Reitz                        BlockDriverAmendStatusCB *status_cb)
40336f176b48SMax Reitz {
4034c282e1fdSChunyan Liu     if (!bs->drv->bdrv_amend_options) {
40356f176b48SMax Reitz         return -ENOTSUP;
40366f176b48SMax Reitz     }
403777485434SMax Reitz     return bs->drv->bdrv_amend_options(bs, opts, status_cb);
40386f176b48SMax Reitz }
4039f6186f49SBenoît Canet 
4040b5042a36SBenoît Canet /* This function will be called by the bdrv_recurse_is_first_non_filter method
4041b5042a36SBenoît Canet  * of block filter and by bdrv_is_first_non_filter.
4042b5042a36SBenoît Canet  * It is used to test if the given bs is the candidate or recurse more in the
4043b5042a36SBenoît Canet  * node graph.
4044212a5a8fSBenoît Canet  */
4045212a5a8fSBenoît Canet bool bdrv_recurse_is_first_non_filter(BlockDriverState *bs,
4046212a5a8fSBenoît Canet                                       BlockDriverState *candidate)
4047f6186f49SBenoît Canet {
4048b5042a36SBenoît Canet     /* return false if basic checks fails */
4049b5042a36SBenoît Canet     if (!bs || !bs->drv) {
4050b5042a36SBenoît Canet         return false;
4051b5042a36SBenoît Canet     }
4052b5042a36SBenoît Canet 
4053b5042a36SBenoît Canet     /* the code reached a non block filter driver -> check if the bs is
4054b5042a36SBenoît Canet      * the same as the candidate. It's the recursion termination condition.
4055b5042a36SBenoît Canet      */
4056b5042a36SBenoît Canet     if (!bs->drv->is_filter) {
4057b5042a36SBenoît Canet         return bs == candidate;
4058b5042a36SBenoît Canet     }
4059b5042a36SBenoît Canet     /* Down this path the driver is a block filter driver */
4060b5042a36SBenoît Canet 
4061b5042a36SBenoît Canet     /* If the block filter recursion method is defined use it to recurse down
4062b5042a36SBenoît Canet      * the node graph.
4063b5042a36SBenoît Canet      */
4064b5042a36SBenoît Canet     if (bs->drv->bdrv_recurse_is_first_non_filter) {
4065212a5a8fSBenoît Canet         return bs->drv->bdrv_recurse_is_first_non_filter(bs, candidate);
4066212a5a8fSBenoît Canet     }
4067212a5a8fSBenoît Canet 
4068b5042a36SBenoît Canet     /* the driver is a block filter but don't allow to recurse -> return false
4069b5042a36SBenoît Canet      */
4070b5042a36SBenoît Canet     return false;
4071212a5a8fSBenoît Canet }
4072212a5a8fSBenoît Canet 
4073212a5a8fSBenoît Canet /* This function checks if the candidate is the first non filter bs down it's
4074212a5a8fSBenoît Canet  * bs chain. Since we don't have pointers to parents it explore all bs chains
4075212a5a8fSBenoît Canet  * from the top. Some filters can choose not to pass down the recursion.
4076212a5a8fSBenoît Canet  */
4077212a5a8fSBenoît Canet bool bdrv_is_first_non_filter(BlockDriverState *candidate)
4078212a5a8fSBenoît Canet {
4079212a5a8fSBenoît Canet     BlockDriverState *bs;
4080212a5a8fSBenoît Canet 
4081212a5a8fSBenoît Canet     /* walk down the bs forest recursively */
4082212a5a8fSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
4083212a5a8fSBenoît Canet         bool perm;
4084212a5a8fSBenoît Canet 
4085b5042a36SBenoît Canet         /* try to recurse in this top level bs */
4086e6dc8a1fSKevin Wolf         perm = bdrv_recurse_is_first_non_filter(bs, candidate);
4087212a5a8fSBenoît Canet 
4088212a5a8fSBenoît Canet         /* candidate is the first non filter */
4089212a5a8fSBenoît Canet         if (perm) {
4090212a5a8fSBenoît Canet             return true;
4091212a5a8fSBenoît Canet         }
4092212a5a8fSBenoît Canet     }
4093212a5a8fSBenoît Canet 
4094212a5a8fSBenoît Canet     return false;
4095f6186f49SBenoît Canet }
409609158f00SBenoît Canet 
4097e12f3784SWen Congyang BlockDriverState *check_to_replace_node(BlockDriverState *parent_bs,
4098e12f3784SWen Congyang                                         const char *node_name, Error **errp)
409909158f00SBenoît Canet {
410009158f00SBenoît Canet     BlockDriverState *to_replace_bs = bdrv_find_node(node_name);
41015a7e7a0bSStefan Hajnoczi     AioContext *aio_context;
41025a7e7a0bSStefan Hajnoczi 
410309158f00SBenoît Canet     if (!to_replace_bs) {
410409158f00SBenoît Canet         error_setg(errp, "Node name '%s' not found", node_name);
410509158f00SBenoît Canet         return NULL;
410609158f00SBenoît Canet     }
410709158f00SBenoît Canet 
41085a7e7a0bSStefan Hajnoczi     aio_context = bdrv_get_aio_context(to_replace_bs);
41095a7e7a0bSStefan Hajnoczi     aio_context_acquire(aio_context);
41105a7e7a0bSStefan Hajnoczi 
411109158f00SBenoît Canet     if (bdrv_op_is_blocked(to_replace_bs, BLOCK_OP_TYPE_REPLACE, errp)) {
41125a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41135a7e7a0bSStefan Hajnoczi         goto out;
411409158f00SBenoît Canet     }
411509158f00SBenoît Canet 
411609158f00SBenoît Canet     /* We don't want arbitrary node of the BDS chain to be replaced only the top
411709158f00SBenoît Canet      * most non filter in order to prevent data corruption.
411809158f00SBenoît Canet      * Another benefit is that this tests exclude backing files which are
411909158f00SBenoît Canet      * blocked by the backing blockers.
412009158f00SBenoît Canet      */
4121e12f3784SWen Congyang     if (!bdrv_recurse_is_first_non_filter(parent_bs, to_replace_bs)) {
412209158f00SBenoît Canet         error_setg(errp, "Only top most non filter can be replaced");
41235a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41245a7e7a0bSStefan Hajnoczi         goto out;
412509158f00SBenoît Canet     }
412609158f00SBenoît Canet 
41275a7e7a0bSStefan Hajnoczi out:
41285a7e7a0bSStefan Hajnoczi     aio_context_release(aio_context);
412909158f00SBenoît Canet     return to_replace_bs;
413009158f00SBenoît Canet }
4131448ad91dSMing Lei 
413291af7014SMax Reitz static bool append_open_options(QDict *d, BlockDriverState *bs)
413391af7014SMax Reitz {
413491af7014SMax Reitz     const QDictEntry *entry;
413591af7014SMax Reitz     bool found_any = false;
413691af7014SMax Reitz 
413791af7014SMax Reitz     for (entry = qdict_first(bs->options); entry;
413891af7014SMax Reitz          entry = qdict_next(bs->options, entry))
413991af7014SMax Reitz     {
414091af7014SMax Reitz         /* Only take options for this level and exclude all non-driver-specific
414191af7014SMax Reitz          * options */
414291af7014SMax Reitz         if (!strchr(qdict_entry_key(entry), '.') &&
414391af7014SMax Reitz             strcmp(qdict_entry_key(entry), "node-name"))
414491af7014SMax Reitz         {
414591af7014SMax Reitz             qobject_incref(qdict_entry_value(entry));
414691af7014SMax Reitz             qdict_put_obj(d, qdict_entry_key(entry), qdict_entry_value(entry));
414791af7014SMax Reitz             found_any = true;
414891af7014SMax Reitz         }
414991af7014SMax Reitz     }
415091af7014SMax Reitz 
415191af7014SMax Reitz     return found_any;
415291af7014SMax Reitz }
415391af7014SMax Reitz 
415491af7014SMax Reitz /* Updates the following BDS fields:
415591af7014SMax Reitz  *  - exact_filename: A filename which may be used for opening a block device
415691af7014SMax Reitz  *                    which (mostly) equals the given BDS (even without any
415791af7014SMax Reitz  *                    other options; so reading and writing must return the same
415891af7014SMax Reitz  *                    results, but caching etc. may be different)
415991af7014SMax Reitz  *  - full_open_options: Options which, when given when opening a block device
416091af7014SMax Reitz  *                       (without a filename), result in a BDS (mostly)
416191af7014SMax Reitz  *                       equalling the given one
416291af7014SMax Reitz  *  - filename: If exact_filename is set, it is copied here. Otherwise,
416391af7014SMax Reitz  *              full_open_options is converted to a JSON object, prefixed with
416491af7014SMax Reitz  *              "json:" (for use through the JSON pseudo protocol) and put here.
416591af7014SMax Reitz  */
416691af7014SMax Reitz void bdrv_refresh_filename(BlockDriverState *bs)
416791af7014SMax Reitz {
416891af7014SMax Reitz     BlockDriver *drv = bs->drv;
416991af7014SMax Reitz     QDict *opts;
417091af7014SMax Reitz 
417191af7014SMax Reitz     if (!drv) {
417291af7014SMax Reitz         return;
417391af7014SMax Reitz     }
417491af7014SMax Reitz 
417591af7014SMax Reitz     /* This BDS's file name will most probably depend on its file's name, so
417691af7014SMax Reitz      * refresh that first */
417791af7014SMax Reitz     if (bs->file) {
417891af7014SMax Reitz         bdrv_refresh_filename(bs->file);
417991af7014SMax Reitz     }
418091af7014SMax Reitz 
418191af7014SMax Reitz     if (drv->bdrv_refresh_filename) {
418291af7014SMax Reitz         /* Obsolete information is of no use here, so drop the old file name
418391af7014SMax Reitz          * information before refreshing it */
418491af7014SMax Reitz         bs->exact_filename[0] = '\0';
418591af7014SMax Reitz         if (bs->full_open_options) {
418691af7014SMax Reitz             QDECREF(bs->full_open_options);
418791af7014SMax Reitz             bs->full_open_options = NULL;
418891af7014SMax Reitz         }
418991af7014SMax Reitz 
419091af7014SMax Reitz         drv->bdrv_refresh_filename(bs);
419191af7014SMax Reitz     } else if (bs->file) {
419291af7014SMax Reitz         /* Try to reconstruct valid information from the underlying file */
419391af7014SMax Reitz         bool has_open_options;
419491af7014SMax Reitz 
419591af7014SMax Reitz         bs->exact_filename[0] = '\0';
419691af7014SMax Reitz         if (bs->full_open_options) {
419791af7014SMax Reitz             QDECREF(bs->full_open_options);
419891af7014SMax Reitz             bs->full_open_options = NULL;
419991af7014SMax Reitz         }
420091af7014SMax Reitz 
420191af7014SMax Reitz         opts = qdict_new();
420291af7014SMax Reitz         has_open_options = append_open_options(opts, bs);
420391af7014SMax Reitz 
420491af7014SMax Reitz         /* If no specific options have been given for this BDS, the filename of
420591af7014SMax Reitz          * the underlying file should suffice for this one as well */
420691af7014SMax Reitz         if (bs->file->exact_filename[0] && !has_open_options) {
420791af7014SMax Reitz             strcpy(bs->exact_filename, bs->file->exact_filename);
420891af7014SMax Reitz         }
420991af7014SMax Reitz         /* Reconstructing the full options QDict is simple for most format block
421091af7014SMax Reitz          * drivers, as long as the full options are known for the underlying
421191af7014SMax Reitz          * file BDS. The full options QDict of that file BDS should somehow
421291af7014SMax Reitz          * contain a representation of the filename, therefore the following
421391af7014SMax Reitz          * suffices without querying the (exact_)filename of this BDS. */
421491af7014SMax Reitz         if (bs->file->full_open_options) {
421591af7014SMax Reitz             qdict_put_obj(opts, "driver",
421691af7014SMax Reitz                           QOBJECT(qstring_from_str(drv->format_name)));
421791af7014SMax Reitz             QINCREF(bs->file->full_open_options);
421891af7014SMax Reitz             qdict_put_obj(opts, "file", QOBJECT(bs->file->full_open_options));
421991af7014SMax Reitz 
422091af7014SMax Reitz             bs->full_open_options = opts;
422191af7014SMax Reitz         } else {
422291af7014SMax Reitz             QDECREF(opts);
422391af7014SMax Reitz         }
422491af7014SMax Reitz     } else if (!bs->full_open_options && qdict_size(bs->options)) {
422591af7014SMax Reitz         /* There is no underlying file BDS (at least referenced by BDS.file),
422691af7014SMax Reitz          * so the full options QDict should be equal to the options given
422791af7014SMax Reitz          * specifically for this block device when it was opened (plus the
422891af7014SMax Reitz          * driver specification).
422991af7014SMax Reitz          * Because those options don't change, there is no need to update
423091af7014SMax Reitz          * full_open_options when it's already set. */
423191af7014SMax Reitz 
423291af7014SMax Reitz         opts = qdict_new();
423391af7014SMax Reitz         append_open_options(opts, bs);
423491af7014SMax Reitz         qdict_put_obj(opts, "driver",
423591af7014SMax Reitz                       QOBJECT(qstring_from_str(drv->format_name)));
423691af7014SMax Reitz 
423791af7014SMax Reitz         if (bs->exact_filename[0]) {
423891af7014SMax Reitz             /* This may not work for all block protocol drivers (some may
423991af7014SMax Reitz              * require this filename to be parsed), but we have to find some
424091af7014SMax Reitz              * default solution here, so just include it. If some block driver
424191af7014SMax Reitz              * does not support pure options without any filename at all or
424291af7014SMax Reitz              * needs some special format of the options QDict, it needs to
424391af7014SMax Reitz              * implement the driver-specific bdrv_refresh_filename() function.
424491af7014SMax Reitz              */
424591af7014SMax Reitz             qdict_put_obj(opts, "filename",
424691af7014SMax Reitz                           QOBJECT(qstring_from_str(bs->exact_filename)));
424791af7014SMax Reitz         }
424891af7014SMax Reitz 
424991af7014SMax Reitz         bs->full_open_options = opts;
425091af7014SMax Reitz     }
425191af7014SMax Reitz 
425291af7014SMax Reitz     if (bs->exact_filename[0]) {
425391af7014SMax Reitz         pstrcpy(bs->filename, sizeof(bs->filename), bs->exact_filename);
425491af7014SMax Reitz     } else if (bs->full_open_options) {
425591af7014SMax Reitz         QString *json = qobject_to_json(QOBJECT(bs->full_open_options));
425691af7014SMax Reitz         snprintf(bs->filename, sizeof(bs->filename), "json:%s",
425791af7014SMax Reitz                  qstring_get_str(json));
425891af7014SMax Reitz         QDECREF(json);
425991af7014SMax Reitz     }
426091af7014SMax Reitz }
42615366d0c8SBenoît Canet 
42625366d0c8SBenoît Canet /* This accessor function purpose is to allow the device models to access the
42635366d0c8SBenoît Canet  * BlockAcctStats structure embedded inside a BlockDriverState without being
42645366d0c8SBenoît Canet  * aware of the BlockDriverState structure layout.
42655366d0c8SBenoît Canet  * It will go away when the BlockAcctStats structure will be moved inside
42665366d0c8SBenoît Canet  * the device models.
42675366d0c8SBenoît Canet  */
42685366d0c8SBenoît Canet BlockAcctStats *bdrv_get_stats(BlockDriverState *bs)
42695366d0c8SBenoît Canet {
42705366d0c8SBenoît Canet     return &bs->stats;
42715366d0c8SBenoît Canet }
4272