xref: /openbmc/qemu/block.c (revision 4d2cb0925176f3eb75ef8e5f9c02cc84d7930de2)
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;
1490054963f8SMax Reitz         ret = bdrv_open_image(&file, filename, options, "file",
1491f3930ed0SKevin Wolf                               bs, &child_file, true, &local_err);
1492f500a6d3SKevin Wolf         if (ret < 0) {
14938bfea15dSKevin Wolf             goto fail;
1494f500a6d3SKevin Wolf         }
1495f4788adcSKevin Wolf     }
1496f500a6d3SKevin Wolf 
149776c591b0SKevin Wolf     /* Image format probing */
149838f3ef57SKevin Wolf     bs->probed = !drv;
149976c591b0SKevin Wolf     if (!drv && file) {
150034b5d2c6SMax Reitz         ret = find_image_format(file, filename, &drv, &local_err);
150117b005f1SKevin Wolf         if (ret < 0) {
150217b005f1SKevin Wolf             goto fail;
150317b005f1SKevin Wolf         }
150476c591b0SKevin Wolf     } else if (!drv) {
15052a05cbe4SMax Reitz         error_setg(errp, "Must specify either driver or file");
15062a05cbe4SMax Reitz         ret = -EINVAL;
15078bfea15dSKevin Wolf         goto fail;
15082a05cbe4SMax Reitz     }
1509f500a6d3SKevin Wolf 
151053a29513SMax Reitz     /* BDRV_O_PROTOCOL must be set iff a protocol BDS is about to be created */
151153a29513SMax Reitz     assert(!!(flags & BDRV_O_PROTOCOL) == !!drv->bdrv_file_open);
151253a29513SMax Reitz     /* file must be NULL if a protocol BDS is about to be created
151353a29513SMax Reitz      * (the inverse results in an error message from bdrv_open_common()) */
151453a29513SMax Reitz     assert(!(flags & BDRV_O_PROTOCOL) || !file);
151553a29513SMax Reitz 
1516b6ce07aaSKevin Wolf     /* Open the image */
151734b5d2c6SMax Reitz     ret = bdrv_open_common(bs, file, options, flags, drv, &local_err);
1518b6ce07aaSKevin Wolf     if (ret < 0) {
15198bfea15dSKevin Wolf         goto fail;
15206987307cSChristoph Hellwig     }
15216987307cSChristoph Hellwig 
15222a05cbe4SMax Reitz     if (file && (bs->file != file)) {
15234f6fd349SFam Zheng         bdrv_unref(file);
1524f500a6d3SKevin Wolf         file = NULL;
1525f500a6d3SKevin Wolf     }
1526f500a6d3SKevin Wolf 
1527b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
15289156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
152931ca6d07SKevin Wolf         QDict *backing_options;
153031ca6d07SKevin Wolf 
15315726d872SBenoît Canet         qdict_extract_subqdict(options, &backing_options, "backing.");
153234b5d2c6SMax Reitz         ret = bdrv_open_backing_file(bs, backing_options, &local_err);
1533b6ce07aaSKevin Wolf         if (ret < 0) {
1534b6ad491aSKevin Wolf             goto close_and_fail;
1535b6ce07aaSKevin Wolf         }
1536b6ce07aaSKevin Wolf     }
1537b6ce07aaSKevin Wolf 
153891af7014SMax Reitz     bdrv_refresh_filename(bs);
153991af7014SMax Reitz 
1540b998875dSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay. bs points to the
1541b998875dSKevin Wolf      * temporary snapshot afterwards. */
1542b1e6fc08SKevin Wolf     if (snapshot_flags) {
15436b8aeca5SChen Gang         ret = bdrv_append_temp_snapshot(bs, snapshot_flags, &local_err);
1544b998875dSKevin Wolf         if (local_err) {
1545b998875dSKevin Wolf             goto close_and_fail;
1546b998875dSKevin Wolf         }
1547b998875dSKevin Wolf     }
1548b998875dSKevin Wolf 
1549b6ad491aSKevin Wolf     /* Check if any unknown options were used */
15505acd9d81SMax Reitz     if (options && (qdict_size(options) != 0)) {
1551b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
15525acd9d81SMax Reitz         if (flags & BDRV_O_PROTOCOL) {
15535acd9d81SMax Reitz             error_setg(errp, "Block protocol '%s' doesn't support the option "
15545acd9d81SMax Reitz                        "'%s'", drv->format_name, entry->key);
15555acd9d81SMax Reitz         } else {
155634b5d2c6SMax Reitz             error_setg(errp, "Block format '%s' used by device '%s' doesn't "
15575acd9d81SMax Reitz                        "support the option '%s'", drv->format_name,
1558bfb197e0SMarkus Armbruster                        bdrv_get_device_name(bs), entry->key);
15595acd9d81SMax Reitz         }
1560b6ad491aSKevin Wolf 
1561b6ad491aSKevin Wolf         ret = -EINVAL;
1562b6ad491aSKevin Wolf         goto close_and_fail;
1563b6ad491aSKevin Wolf     }
1564b6ad491aSKevin Wolf 
1565b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
1566a7f53e26SMarkus Armbruster         if (bs->blk) {
1567a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
1568a7f53e26SMarkus Armbruster         }
1569c3adb58fSMarkus Armbruster     } else if (!runstate_check(RUN_STATE_PRELAUNCH)
1570c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_INMIGRATE)
1571c3adb58fSMarkus Armbruster                && !runstate_check(RUN_STATE_PAUSED)) { /* HACK */
1572c3adb58fSMarkus Armbruster         error_setg(errp,
1573c3adb58fSMarkus Armbruster                    "Guest must be stopped for opening of encrypted image");
1574c3adb58fSMarkus Armbruster         ret = -EBUSY;
1575c3adb58fSMarkus Armbruster         goto close_and_fail;
1576b6ce07aaSKevin Wolf     }
1577b6ce07aaSKevin Wolf 
1578c3adb58fSMarkus Armbruster     QDECREF(options);
1579f67503e5SMax Reitz     *pbs = bs;
1580b6ce07aaSKevin Wolf     return 0;
1581b6ce07aaSKevin Wolf 
15828bfea15dSKevin Wolf fail:
1583f500a6d3SKevin Wolf     if (file != NULL) {
15844f6fd349SFam Zheng         bdrv_unref(file);
1585f500a6d3SKevin Wolf     }
1586de9c0cecSKevin Wolf     QDECREF(bs->options);
1587b6ad491aSKevin Wolf     QDECREF(options);
1588de9c0cecSKevin Wolf     bs->options = NULL;
1589f67503e5SMax Reitz     if (!*pbs) {
1590f67503e5SMax Reitz         /* If *pbs is NULL, a new BDS has been created in this function and
1591f67503e5SMax Reitz            needs to be freed now. Otherwise, it does not need to be closed,
1592f67503e5SMax Reitz            since it has not really been opened yet. */
1593f67503e5SMax Reitz         bdrv_unref(bs);
1594f67503e5SMax Reitz     }
159584d18f06SMarkus Armbruster     if (local_err) {
159634b5d2c6SMax Reitz         error_propagate(errp, local_err);
159734b5d2c6SMax Reitz     }
1598b6ad491aSKevin Wolf     return ret;
1599de9c0cecSKevin Wolf 
1600b6ad491aSKevin Wolf close_and_fail:
1601f67503e5SMax Reitz     /* See fail path, but now the BDS has to be always closed */
1602f67503e5SMax Reitz     if (*pbs) {
1603b6ad491aSKevin Wolf         bdrv_close(bs);
1604f67503e5SMax Reitz     } else {
1605f67503e5SMax Reitz         bdrv_unref(bs);
1606f67503e5SMax Reitz     }
1607b6ad491aSKevin Wolf     QDECREF(options);
160884d18f06SMarkus Armbruster     if (local_err) {
160934b5d2c6SMax Reitz         error_propagate(errp, local_err);
161034b5d2c6SMax Reitz     }
1611b6ce07aaSKevin Wolf     return ret;
1612b6ce07aaSKevin Wolf }
1613b6ce07aaSKevin Wolf 
1614f3930ed0SKevin Wolf int bdrv_open(BlockDriverState **pbs, const char *filename,
16156ebf9aa2SMax Reitz               const char *reference, QDict *options, int flags, Error **errp)
1616f3930ed0SKevin Wolf {
1617f3930ed0SKevin Wolf     return bdrv_open_inherit(pbs, filename, reference, options, flags, NULL,
1618ce343771SMax Reitz                              NULL, errp);
1619f3930ed0SKevin Wolf }
1620f3930ed0SKevin Wolf 
1621e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1622e971aa12SJeff Cody      bool prepared;
1623e971aa12SJeff Cody      BDRVReopenState state;
1624e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1625e971aa12SJeff Cody } BlockReopenQueueEntry;
1626e971aa12SJeff Cody 
1627e971aa12SJeff Cody /*
1628e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1629e971aa12SJeff Cody  * reopen of multiple devices.
1630e971aa12SJeff Cody  *
1631e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1632e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1633e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1634e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1635e971aa12SJeff Cody  * atomic 'set'.
1636e971aa12SJeff Cody  *
1637e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1638e971aa12SJeff Cody  *
1639*4d2cb092SKevin Wolf  * options contains the changed options for the associated bs
1640*4d2cb092SKevin Wolf  * (the BlockReopenQueue takes ownership)
1641*4d2cb092SKevin Wolf  *
1642e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1643e971aa12SJeff Cody  *
1644e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1645e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1646e971aa12SJeff Cody  *
1647e971aa12SJeff Cody  */
1648e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1649*4d2cb092SKevin Wolf                                     BlockDriverState *bs,
1650*4d2cb092SKevin Wolf                                     QDict *options, int flags)
1651e971aa12SJeff Cody {
1652e971aa12SJeff Cody     assert(bs != NULL);
1653e971aa12SJeff Cody 
1654e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
165567251a31SKevin Wolf     BdrvChild *child;
1656*4d2cb092SKevin Wolf     QDict *old_options;
165767251a31SKevin Wolf 
1658e971aa12SJeff Cody     if (bs_queue == NULL) {
1659e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1660e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1661e971aa12SJeff Cody     }
1662e971aa12SJeff Cody 
1663*4d2cb092SKevin Wolf     if (!options) {
1664*4d2cb092SKevin Wolf         options = qdict_new();
1665*4d2cb092SKevin Wolf     }
1666*4d2cb092SKevin Wolf 
1667*4d2cb092SKevin Wolf     old_options = qdict_clone_shallow(bs->options);
1668*4d2cb092SKevin Wolf     qdict_join(options, old_options, false);
1669*4d2cb092SKevin Wolf     QDECREF(old_options);
1670*4d2cb092SKevin Wolf 
1671f1f25a2eSKevin Wolf     /* bdrv_open() masks this flag out */
1672f1f25a2eSKevin Wolf     flags &= ~BDRV_O_PROTOCOL;
1673f1f25a2eSKevin Wolf 
167467251a31SKevin Wolf     QLIST_FOREACH(child, &bs->children, next) {
167567251a31SKevin Wolf         int child_flags;
167667251a31SKevin Wolf 
167767251a31SKevin Wolf         if (child->bs->inherits_from != bs) {
167867251a31SKevin Wolf             continue;
167967251a31SKevin Wolf         }
168067251a31SKevin Wolf 
168167251a31SKevin Wolf         child_flags = child->role->inherit_flags(flags);
1682*4d2cb092SKevin Wolf         /* TODO Pass down child flags (backing.*, extents.*, ...) */
1683*4d2cb092SKevin Wolf         bdrv_reopen_queue(bs_queue, child->bs, NULL, child_flags);
1684e971aa12SJeff Cody     }
1685e971aa12SJeff Cody 
1686e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1687e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1688e971aa12SJeff Cody 
1689e971aa12SJeff Cody     bs_entry->state.bs = bs;
1690*4d2cb092SKevin Wolf     bs_entry->state.options = options;
1691e971aa12SJeff Cody     bs_entry->state.flags = flags;
1692e971aa12SJeff Cody 
1693e971aa12SJeff Cody     return bs_queue;
1694e971aa12SJeff Cody }
1695e971aa12SJeff Cody 
1696e971aa12SJeff Cody /*
1697e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1698e971aa12SJeff Cody  *
1699e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1700e971aa12SJeff Cody  * via bdrv_reopen_queue().
1701e971aa12SJeff Cody  *
1702e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1703e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1704e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1705e971aa12SJeff Cody  * data cleaned up.
1706e971aa12SJeff Cody  *
1707e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1708e971aa12SJeff Cody  * to all devices.
1709e971aa12SJeff Cody  *
1710e971aa12SJeff Cody  */
1711e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1712e971aa12SJeff Cody {
1713e971aa12SJeff Cody     int ret = -1;
1714e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1715e971aa12SJeff Cody     Error *local_err = NULL;
1716e971aa12SJeff Cody 
1717e971aa12SJeff Cody     assert(bs_queue != NULL);
1718e971aa12SJeff Cody 
1719e971aa12SJeff Cody     bdrv_drain_all();
1720e971aa12SJeff Cody 
1721e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1722e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1723e971aa12SJeff Cody             error_propagate(errp, local_err);
1724e971aa12SJeff Cody             goto cleanup;
1725e971aa12SJeff Cody         }
1726e971aa12SJeff Cody         bs_entry->prepared = true;
1727e971aa12SJeff Cody     }
1728e971aa12SJeff Cody 
1729e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1730e971aa12SJeff Cody      * changes
1731e971aa12SJeff Cody      */
1732e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1733e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1734e971aa12SJeff Cody     }
1735e971aa12SJeff Cody 
1736e971aa12SJeff Cody     ret = 0;
1737e971aa12SJeff Cody 
1738e971aa12SJeff Cody cleanup:
1739e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1740e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1741e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1742e971aa12SJeff Cody         }
1743*4d2cb092SKevin Wolf         QDECREF(bs_entry->state.options);
1744e971aa12SJeff Cody         g_free(bs_entry);
1745e971aa12SJeff Cody     }
1746e971aa12SJeff Cody     g_free(bs_queue);
1747e971aa12SJeff Cody     return ret;
1748e971aa12SJeff Cody }
1749e971aa12SJeff Cody 
1750e971aa12SJeff Cody 
1751e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1752e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1753e971aa12SJeff Cody {
1754e971aa12SJeff Cody     int ret = -1;
1755e971aa12SJeff Cody     Error *local_err = NULL;
1756*4d2cb092SKevin Wolf     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, NULL, bdrv_flags);
1757e971aa12SJeff Cody 
1758e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1759e971aa12SJeff Cody     if (local_err != NULL) {
1760e971aa12SJeff Cody         error_propagate(errp, local_err);
1761e971aa12SJeff Cody     }
1762e971aa12SJeff Cody     return ret;
1763e971aa12SJeff Cody }
1764e971aa12SJeff Cody 
1765e971aa12SJeff Cody 
1766e971aa12SJeff Cody /*
1767e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1768e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1769e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1770e971aa12SJeff Cody  *
1771e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1772e971aa12SJeff Cody  * flags are the new open flags
1773e971aa12SJeff Cody  * queue is the reopen queue
1774e971aa12SJeff Cody  *
1775e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1776e971aa12SJeff Cody  * as well.
1777e971aa12SJeff Cody  *
1778e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1779e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1780e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1781e971aa12SJeff Cody  *
1782e971aa12SJeff Cody  */
1783e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1784e971aa12SJeff Cody                         Error **errp)
1785e971aa12SJeff Cody {
1786e971aa12SJeff Cody     int ret = -1;
1787e971aa12SJeff Cody     Error *local_err = NULL;
1788e971aa12SJeff Cody     BlockDriver *drv;
1789e971aa12SJeff Cody 
1790e971aa12SJeff Cody     assert(reopen_state != NULL);
1791e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1792e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1793e971aa12SJeff Cody 
1794e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1795e971aa12SJeff Cody      * to r/w */
1796e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1797e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
179881e5f78aSAlberto Garcia         error_setg(errp, "Node '%s' is read only",
179981e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1800e971aa12SJeff Cody         goto error;
1801e971aa12SJeff Cody     }
1802e971aa12SJeff Cody 
1803e971aa12SJeff Cody 
1804e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1805e971aa12SJeff Cody     if (ret) {
1806e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1807e971aa12SJeff Cody                   strerror(-ret));
1808e971aa12SJeff Cody         goto error;
1809e971aa12SJeff Cody     }
1810e971aa12SJeff Cody 
1811e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1812e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1813e971aa12SJeff Cody         if (ret) {
1814e971aa12SJeff Cody             if (local_err != NULL) {
1815e971aa12SJeff Cody                 error_propagate(errp, local_err);
1816e971aa12SJeff Cody             } else {
1817d8b6895fSLuiz Capitulino                 error_setg(errp, "failed while preparing to reopen image '%s'",
1818e971aa12SJeff Cody                            reopen_state->bs->filename);
1819e971aa12SJeff Cody             }
1820e971aa12SJeff Cody             goto error;
1821e971aa12SJeff Cody         }
1822e971aa12SJeff Cody     } else {
1823e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1824e971aa12SJeff Cody          * handler for each supported drv. */
182581e5f78aSAlberto Garcia         error_setg(errp, "Block format '%s' used by node '%s' "
182681e5f78aSAlberto Garcia                    "does not support reopening files", drv->format_name,
182781e5f78aSAlberto Garcia                    bdrv_get_device_or_node_name(reopen_state->bs));
1828e971aa12SJeff Cody         ret = -1;
1829e971aa12SJeff Cody         goto error;
1830e971aa12SJeff Cody     }
1831e971aa12SJeff Cody 
1832*4d2cb092SKevin Wolf     /* Options that are not handled are only okay if they are unchanged
1833*4d2cb092SKevin Wolf      * compared to the old state. It is expected that some options are only
1834*4d2cb092SKevin Wolf      * used for the initial open, but not reopen (e.g. filename) */
1835*4d2cb092SKevin Wolf     if (qdict_size(reopen_state->options)) {
1836*4d2cb092SKevin Wolf         const QDictEntry *entry = qdict_first(reopen_state->options);
1837*4d2cb092SKevin Wolf 
1838*4d2cb092SKevin Wolf         do {
1839*4d2cb092SKevin Wolf             QString *new_obj = qobject_to_qstring(entry->value);
1840*4d2cb092SKevin Wolf             const char *new = qstring_get_str(new_obj);
1841*4d2cb092SKevin Wolf             const char *old = qdict_get_try_str(reopen_state->bs->options,
1842*4d2cb092SKevin Wolf                                                 entry->key);
1843*4d2cb092SKevin Wolf 
1844*4d2cb092SKevin Wolf             if (!old || strcmp(new, old)) {
1845*4d2cb092SKevin Wolf                 error_setg(errp, "Cannot change the option '%s'", entry->key);
1846*4d2cb092SKevin Wolf                 ret = -EINVAL;
1847*4d2cb092SKevin Wolf                 goto error;
1848*4d2cb092SKevin Wolf             }
1849*4d2cb092SKevin Wolf         } while ((entry = qdict_next(reopen_state->options, entry)));
1850*4d2cb092SKevin Wolf     }
1851*4d2cb092SKevin Wolf 
1852e971aa12SJeff Cody     ret = 0;
1853e971aa12SJeff Cody 
1854e971aa12SJeff Cody error:
1855e971aa12SJeff Cody     return ret;
1856e971aa12SJeff Cody }
1857e971aa12SJeff Cody 
1858e971aa12SJeff Cody /*
1859e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1860e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1861e971aa12SJeff Cody  * the active BlockDriverState contents.
1862e971aa12SJeff Cody  */
1863e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1864e971aa12SJeff Cody {
1865e971aa12SJeff Cody     BlockDriver *drv;
1866e971aa12SJeff Cody 
1867e971aa12SJeff Cody     assert(reopen_state != NULL);
1868e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1869e971aa12SJeff Cody     assert(drv != NULL);
1870e971aa12SJeff Cody 
1871e971aa12SJeff Cody     /* If there are any driver level actions to take */
1872e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1873e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1874e971aa12SJeff Cody     }
1875e971aa12SJeff Cody 
1876e971aa12SJeff Cody     /* set BDS specific flags now */
1877e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1878e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1879e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1880e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1881355ef4acSKevin Wolf 
18823baca891SKevin Wolf     bdrv_refresh_limits(reopen_state->bs, NULL);
1883e971aa12SJeff Cody }
1884e971aa12SJeff Cody 
1885e971aa12SJeff Cody /*
1886e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1887e971aa12SJeff Cody  * reopen_state
1888e971aa12SJeff Cody  */
1889e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1890e971aa12SJeff Cody {
1891e971aa12SJeff Cody     BlockDriver *drv;
1892e971aa12SJeff Cody 
1893e971aa12SJeff Cody     assert(reopen_state != NULL);
1894e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1895e971aa12SJeff Cody     assert(drv != NULL);
1896e971aa12SJeff Cody 
1897e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1898e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1899e971aa12SJeff Cody     }
1900e971aa12SJeff Cody }
1901e971aa12SJeff Cody 
1902e971aa12SJeff Cody 
1903fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1904fc01f7e7Sbellard {
190533384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
190633384421SMax Reitz 
19073e914655SPaolo Bonzini     if (bs->job) {
19083e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
19093e914655SPaolo Bonzini     }
191053ec73e2SFam Zheng     bdrv_drain(bs); /* complete I/O */
191158fda173SStefan Hajnoczi     bdrv_flush(bs);
191253ec73e2SFam Zheng     bdrv_drain(bs); /* in case flush left pending I/O */
1913d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
19147094f12fSKevin Wolf 
19153cbc002cSPaolo Bonzini     if (bs->drv) {
19166e93e7c4SKevin Wolf         BdrvChild *child, *next;
19176e93e7c4SKevin Wolf 
19189a7dedbcSKevin Wolf         bs->drv->bdrv_close(bs);
19199a7dedbcSKevin Wolf 
19209a7dedbcSKevin Wolf         if (bs->backing_hd) {
19219a7dedbcSKevin Wolf             BlockDriverState *backing_hd = bs->backing_hd;
19229a7dedbcSKevin Wolf             bdrv_set_backing_hd(bs, NULL);
19239a7dedbcSKevin Wolf             bdrv_unref(backing_hd);
19249a7dedbcSKevin Wolf         }
19259a7dedbcSKevin Wolf 
19266e93e7c4SKevin Wolf         QLIST_FOREACH_SAFE(child, &bs->children, next, next) {
192733a60407SKevin Wolf             /* TODO Remove bdrv_unref() from drivers' close function and use
192833a60407SKevin Wolf              * bdrv_unref_child() here */
1929bddcec37SKevin Wolf             if (child->bs->inherits_from == bs) {
1930bddcec37SKevin Wolf                 child->bs->inherits_from = NULL;
1931bddcec37SKevin Wolf             }
193233a60407SKevin Wolf             bdrv_detach_child(child);
19336e93e7c4SKevin Wolf         }
19346e93e7c4SKevin Wolf 
19357267c094SAnthony Liguori         g_free(bs->opaque);
1936ea2384d3Sbellard         bs->opaque = NULL;
1937ea2384d3Sbellard         bs->drv = NULL;
193853fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1939a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1940a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
19416405875cSPaolo Bonzini         bs->total_sectors = 0;
19426405875cSPaolo Bonzini         bs->encrypted = 0;
19436405875cSPaolo Bonzini         bs->valid_key = 0;
19446405875cSPaolo Bonzini         bs->sg = 0;
19450d51b4deSAsias He         bs->zero_beyond_eof = false;
1946de9c0cecSKevin Wolf         QDECREF(bs->options);
1947de9c0cecSKevin Wolf         bs->options = NULL;
194891af7014SMax Reitz         QDECREF(bs->full_open_options);
194991af7014SMax Reitz         bs->full_open_options = NULL;
1950b338082bSbellard 
195166f82ceeSKevin Wolf         if (bs->file != NULL) {
19524f6fd349SFam Zheng             bdrv_unref(bs->file);
19530ac9377dSPaolo Bonzini             bs->file = NULL;
195466f82ceeSKevin Wolf         }
19559ca11154SPavel Hrdina     }
195666f82ceeSKevin Wolf 
1957a7f53e26SMarkus Armbruster     if (bs->blk) {
1958a7f53e26SMarkus Armbruster         blk_dev_change_media_cb(bs->blk, false);
1959a7f53e26SMarkus Armbruster     }
196098f90dbaSZhi Yong Wu 
196198f90dbaSZhi Yong Wu     /*throttling disk I/O limits*/
196298f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
196398f90dbaSZhi Yong Wu         bdrv_io_limits_disable(bs);
196498f90dbaSZhi Yong Wu     }
196533384421SMax Reitz 
196633384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
196733384421SMax Reitz         g_free(ban);
196833384421SMax Reitz     }
196933384421SMax Reitz     QLIST_INIT(&bs->aio_notifiers);
1970b338082bSbellard }
1971b338082bSbellard 
19722bc93fedSMORITA Kazutaka void bdrv_close_all(void)
19732bc93fedSMORITA Kazutaka {
19742bc93fedSMORITA Kazutaka     BlockDriverState *bs;
19752bc93fedSMORITA Kazutaka 
1976dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
1977ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
1978ed78cda3SStefan Hajnoczi 
1979ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
19802bc93fedSMORITA Kazutaka         bdrv_close(bs);
1981ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
19822bc93fedSMORITA Kazutaka     }
19832bc93fedSMORITA Kazutaka }
19842bc93fedSMORITA Kazutaka 
1985dc364f4cSBenoît Canet /* make a BlockDriverState anonymous by removing from bdrv_state and
1986dc364f4cSBenoît Canet  * graph_bdrv_state list.
1987d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1988d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1989d22b2f41SRyan Harper {
1990bfb197e0SMarkus Armbruster     /*
1991bfb197e0SMarkus Armbruster      * Take care to remove bs from bdrv_states only when it's actually
1992bfb197e0SMarkus Armbruster      * in it.  Note that bs->device_list.tqe_prev is initially null,
1993bfb197e0SMarkus Armbruster      * and gets set to non-null by QTAILQ_INSERT_TAIL().  Establish
1994bfb197e0SMarkus Armbruster      * the useful invariant "bs in bdrv_states iff bs->tqe_prev" by
1995bfb197e0SMarkus Armbruster      * resetting it to null on remove.
1996bfb197e0SMarkus Armbruster      */
1997bfb197e0SMarkus Armbruster     if (bs->device_list.tqe_prev) {
1998dc364f4cSBenoît Canet         QTAILQ_REMOVE(&bdrv_states, bs, device_list);
1999bfb197e0SMarkus Armbruster         bs->device_list.tqe_prev = NULL;
2000d22b2f41SRyan Harper     }
2001dc364f4cSBenoît Canet     if (bs->node_name[0] != '\0') {
2002dc364f4cSBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs, node_list);
2003dc364f4cSBenoît Canet     }
2004dc364f4cSBenoît Canet     bs->node_name[0] = '\0';
2005d22b2f41SRyan Harper }
2006d22b2f41SRyan Harper 
2007e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
2008e023b2e2SPaolo Bonzini {
2009e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
2010e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
2011e023b2e2SPaolo Bonzini     }
2012e023b2e2SPaolo Bonzini }
2013e023b2e2SPaolo Bonzini 
20144ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
20154ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
20164ddc07caSPaolo Bonzini {
20174ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
20184ddc07caSPaolo Bonzini 
20194ddc07caSPaolo Bonzini     /* dev info */
20201b7fd729SPaolo Bonzini     bs_dest->guest_block_size   = bs_src->guest_block_size;
20214ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
20224ddc07caSPaolo Bonzini 
20234ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
20244ddc07caSPaolo Bonzini 
2025cc0681c4SBenoît Canet     /* i/o throttled req */
202676f4afb4SAlberto Garcia     bs_dest->throttle_state     = bs_src->throttle_state,
202776f4afb4SAlberto Garcia     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
202876f4afb4SAlberto Garcia     bs_dest->pending_reqs[0]    = bs_src->pending_reqs[0];
202976f4afb4SAlberto Garcia     bs_dest->pending_reqs[1]    = bs_src->pending_reqs[1];
203076f4afb4SAlberto Garcia     bs_dest->throttled_reqs[0]  = bs_src->throttled_reqs[0];
203176f4afb4SAlberto Garcia     bs_dest->throttled_reqs[1]  = bs_src->throttled_reqs[1];
203276f4afb4SAlberto Garcia     memcpy(&bs_dest->round_robin,
203376f4afb4SAlberto Garcia            &bs_src->round_robin,
203476f4afb4SAlberto Garcia            sizeof(bs_dest->round_robin));
20350e5b0a2dSBenoît Canet     memcpy(&bs_dest->throttle_timers,
20360e5b0a2dSBenoît Canet            &bs_src->throttle_timers,
20370e5b0a2dSBenoît Canet            sizeof(ThrottleTimers));
20384ddc07caSPaolo Bonzini 
20394ddc07caSPaolo Bonzini     /* r/w error */
20404ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
20414ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
20424ddc07caSPaolo Bonzini 
20434ddc07caSPaolo Bonzini     /* i/o status */
20444ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
20454ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
20464ddc07caSPaolo Bonzini 
20474ddc07caSPaolo Bonzini     /* dirty bitmap */
2048e4654d2dSFam Zheng     bs_dest->dirty_bitmaps      = bs_src->dirty_bitmaps;
20494ddc07caSPaolo Bonzini 
20509fcb0251SFam Zheng     /* reference count */
20519fcb0251SFam Zheng     bs_dest->refcnt             = bs_src->refcnt;
20529fcb0251SFam Zheng 
20534ddc07caSPaolo Bonzini     /* job */
20544ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
20554ddc07caSPaolo Bonzini 
20564ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
2057dc364f4cSBenoît Canet     bs_dest->device_list = bs_src->device_list;
20587e7d56d9SMarkus Armbruster     bs_dest->blk = bs_src->blk;
20597e7d56d9SMarkus Armbruster 
2060fbe40ff7SFam Zheng     memcpy(bs_dest->op_blockers, bs_src->op_blockers,
2061fbe40ff7SFam Zheng            sizeof(bs_dest->op_blockers));
20624ddc07caSPaolo Bonzini }
20634ddc07caSPaolo Bonzini 
20644ddc07caSPaolo Bonzini /*
20654ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
20664ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
20674ddc07caSPaolo Bonzini  * actually attached to a device.
20684ddc07caSPaolo Bonzini  *
20694ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
20704ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
20714ddc07caSPaolo Bonzini  *
2072bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
20734ddc07caSPaolo Bonzini  *
20744ddc07caSPaolo Bonzini  * This function does not create any image files.
20754ddc07caSPaolo Bonzini  */
20764ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
20774ddc07caSPaolo Bonzini {
20784ddc07caSPaolo Bonzini     BlockDriverState tmp;
2079bddcec37SKevin Wolf     BdrvChild *child;
20804ddc07caSPaolo Bonzini 
20816ee4ce1eSKevin Wolf     bdrv_drain(bs_new);
20826ee4ce1eSKevin Wolf     bdrv_drain(bs_old);
20834ddc07caSPaolo Bonzini 
208490ce8a06SBenoît Canet     /* The code needs to swap the node_name but simply swapping node_list won't
208590ce8a06SBenoît Canet      * work so first remove the nodes from the graph list, do the swap then
208690ce8a06SBenoît Canet      * insert them back if needed.
208790ce8a06SBenoît Canet      */
208890ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
208990ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_new, node_list);
209090ce8a06SBenoît Canet     }
209190ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
209290ce8a06SBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs_old, node_list);
209390ce8a06SBenoît Canet     }
209490ce8a06SBenoît Canet 
2095db628338SAlberto Garcia     /* If the BlockDriverState is part of a throttling group acquire
2096db628338SAlberto Garcia      * its lock since we're going to mess with the protected fields.
2097db628338SAlberto Garcia      * Otherwise there's no need to worry since no one else can touch
2098db628338SAlberto Garcia      * them. */
2099db628338SAlberto Garcia     if (bs_old->throttle_state) {
2100db628338SAlberto Garcia         throttle_group_lock(bs_old);
2101db628338SAlberto Garcia     }
2102db628338SAlberto Garcia 
2103bfb197e0SMarkus Armbruster     /* bs_new must be unattached and shouldn't have anything fancy enabled */
21047e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
2105e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs_new->dirty_bitmaps));
21064ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21074ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2108db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21090e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21104ddc07caSPaolo Bonzini 
21114ddc07caSPaolo Bonzini     tmp = *bs_new;
21124ddc07caSPaolo Bonzini     *bs_new = *bs_old;
21134ddc07caSPaolo Bonzini     *bs_old = tmp;
21144ddc07caSPaolo Bonzini 
21154ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
21164ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
21174ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
21184ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
21194ddc07caSPaolo Bonzini 
2120bfb197e0SMarkus Armbruster     /* bs_new must remain unattached */
21217e7d56d9SMarkus Armbruster     assert(!bs_new->blk);
21224ddc07caSPaolo Bonzini 
21234ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
21244ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
21254ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
2126db628338SAlberto Garcia     assert(bs_new->throttle_state == NULL);
21270e5b0a2dSBenoît Canet     assert(!throttle_timers_are_initialized(&bs_new->throttle_timers));
21284ddc07caSPaolo Bonzini 
2129db628338SAlberto Garcia     /* Release the ThrottleGroup lock */
2130db628338SAlberto Garcia     if (bs_old->throttle_state) {
2131db628338SAlberto Garcia         throttle_group_unlock(bs_old);
2132db628338SAlberto Garcia     }
2133db628338SAlberto Garcia 
213490ce8a06SBenoît Canet     /* insert the nodes back into the graph node list if needed */
213590ce8a06SBenoît Canet     if (bs_new->node_name[0] != '\0') {
213690ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_new, node_list);
213790ce8a06SBenoît Canet     }
213890ce8a06SBenoît Canet     if (bs_old->node_name[0] != '\0') {
213990ce8a06SBenoît Canet         QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs_old, node_list);
214090ce8a06SBenoît Canet     }
214190ce8a06SBenoît Canet 
21426e93e7c4SKevin Wolf     /*
21436e93e7c4SKevin Wolf      * Update lh_first.le_prev for non-empty lists.
21446e93e7c4SKevin Wolf      *
21456e93e7c4SKevin Wolf      * The head of the op blocker list doesn't change because it is moved back
21466e93e7c4SKevin Wolf      * in bdrv_move_feature_fields().
21476e93e7c4SKevin Wolf      */
21486ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_old->tracked_requests));
21496ee4ce1eSKevin Wolf     assert(QLIST_EMPTY(&bs_new->tracked_requests));
21506ee4ce1eSKevin Wolf 
21516e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_new->children, next);
21526e93e7c4SKevin Wolf     QLIST_FIX_HEAD_PTR(&bs_old->children, next);
21536e93e7c4SKevin Wolf 
2154bddcec37SKevin Wolf     /* Update references in bs->opaque and children */
2155bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_old->children, next) {
2156bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_new) {
2157bddcec37SKevin Wolf             child->bs->inherits_from = bs_old;
2158bddcec37SKevin Wolf         }
2159bddcec37SKevin Wolf     }
2160bddcec37SKevin Wolf     QLIST_FOREACH(child, &bs_new->children, next) {
2161bddcec37SKevin Wolf         if (child->bs->inherits_from == bs_old) {
2162bddcec37SKevin Wolf             child->bs->inherits_from = bs_new;
2163bddcec37SKevin Wolf         }
2164bddcec37SKevin Wolf     }
2165bddcec37SKevin Wolf 
21664ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
21674ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
21684ddc07caSPaolo Bonzini }
21694ddc07caSPaolo Bonzini 
21708802d1fdSJeff Cody /*
21718802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
21728802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
21738802d1fdSJeff Cody  *
21748802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
21758802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
21768802d1fdSJeff Cody  *
2177bfb197e0SMarkus Armbruster  * bs_new must not be attached to a BlockBackend.
2178f6801b83SJeff Cody  *
21798802d1fdSJeff Cody  * This function does not create any image files.
21808802d1fdSJeff Cody  */
21818802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
21828802d1fdSJeff Cody {
21834ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
21848802d1fdSJeff Cody 
21858802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
21868802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
21878d24cce1SFam Zheng     bdrv_set_backing_hd(bs_top, bs_new);
21888802d1fdSJeff Cody }
21898802d1fdSJeff Cody 
21904f6fd349SFam Zheng static void bdrv_delete(BlockDriverState *bs)
2191b338082bSbellard {
21923e914655SPaolo Bonzini     assert(!bs->job);
21933718d8abSFam Zheng     assert(bdrv_op_blocker_is_empty(bs));
21944f6fd349SFam Zheng     assert(!bs->refcnt);
2195e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
219618846deeSMarkus Armbruster 
2197e1b5c52eSStefan Hajnoczi     bdrv_close(bs);
2198e1b5c52eSStefan Hajnoczi 
21991b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
2200d22b2f41SRyan Harper     bdrv_make_anon(bs);
220134c6f050Saurel32 
22027267c094SAnthony Liguori     g_free(bs);
2203fc01f7e7Sbellard }
2204fc01f7e7Sbellard 
2205e97fc193Saliguori /*
2206e97fc193Saliguori  * Run consistency checks on an image
2207e97fc193Saliguori  *
2208e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
2209a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
2210e076f338SKevin Wolf  * check are stored in res.
2211e97fc193Saliguori  */
22124534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
2213e97fc193Saliguori {
2214908bcd54SMax Reitz     if (bs->drv == NULL) {
2215908bcd54SMax Reitz         return -ENOMEDIUM;
2216908bcd54SMax Reitz     }
2217e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
2218e97fc193Saliguori         return -ENOTSUP;
2219e97fc193Saliguori     }
2220e97fc193Saliguori 
2221e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
22224534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
2223e97fc193Saliguori }
2224e97fc193Saliguori 
22258a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
22268a426614SKevin Wolf 
222733e3963eSbellard /* commit COW file into the raw image */
222833e3963eSbellard int bdrv_commit(BlockDriverState *bs)
222933e3963eSbellard {
223019cb3738Sbellard     BlockDriver *drv = bs->drv;
223172706ea4SJeff Cody     int64_t sector, total_sectors, length, backing_length;
22328a426614SKevin Wolf     int n, ro, open_flags;
22330bce597dSJeff Cody     int ret = 0;
223472706ea4SJeff Cody     uint8_t *buf = NULL;
223533e3963eSbellard 
223619cb3738Sbellard     if (!drv)
223719cb3738Sbellard         return -ENOMEDIUM;
223833e3963eSbellard 
22394dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
22404dca4b63SNaphtali Sprei         return -ENOTSUP;
22414dca4b63SNaphtali Sprei     }
22424dca4b63SNaphtali Sprei 
2243bb00021dSFam Zheng     if (bdrv_op_is_blocked(bs, BLOCK_OP_TYPE_COMMIT_SOURCE, NULL) ||
2244bb00021dSFam Zheng         bdrv_op_is_blocked(bs->backing_hd, BLOCK_OP_TYPE_COMMIT_TARGET, NULL)) {
22452d3735d3SStefan Hajnoczi         return -EBUSY;
22462d3735d3SStefan Hajnoczi     }
22472d3735d3SStefan Hajnoczi 
22484dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
22494dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
22504dca4b63SNaphtali Sprei 
22514dca4b63SNaphtali Sprei     if (ro) {
22520bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
22530bce597dSJeff Cody             return -EACCES;
22544dca4b63SNaphtali Sprei         }
2255ea2384d3Sbellard     }
2256ea2384d3Sbellard 
225772706ea4SJeff Cody     length = bdrv_getlength(bs);
225872706ea4SJeff Cody     if (length < 0) {
225972706ea4SJeff Cody         ret = length;
226072706ea4SJeff Cody         goto ro_cleanup;
226172706ea4SJeff Cody     }
226272706ea4SJeff Cody 
226372706ea4SJeff Cody     backing_length = bdrv_getlength(bs->backing_hd);
226472706ea4SJeff Cody     if (backing_length < 0) {
226572706ea4SJeff Cody         ret = backing_length;
226672706ea4SJeff Cody         goto ro_cleanup;
226772706ea4SJeff Cody     }
226872706ea4SJeff Cody 
226972706ea4SJeff Cody     /* If our top snapshot is larger than the backing file image,
227072706ea4SJeff Cody      * grow the backing file image if possible.  If not possible,
227172706ea4SJeff Cody      * we must return an error */
227272706ea4SJeff Cody     if (length > backing_length) {
227372706ea4SJeff Cody         ret = bdrv_truncate(bs->backing_hd, length);
227472706ea4SJeff Cody         if (ret < 0) {
227572706ea4SJeff Cody             goto ro_cleanup;
227672706ea4SJeff Cody         }
227772706ea4SJeff Cody     }
227872706ea4SJeff Cody 
227972706ea4SJeff Cody     total_sectors = length >> BDRV_SECTOR_BITS;
2280857d4f46SKevin Wolf 
2281857d4f46SKevin Wolf     /* qemu_try_blockalign() for bs will choose an alignment that works for
2282857d4f46SKevin Wolf      * bs->backing_hd as well, so no need to compare the alignment manually. */
2283857d4f46SKevin Wolf     buf = qemu_try_blockalign(bs, COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
2284857d4f46SKevin Wolf     if (buf == NULL) {
2285857d4f46SKevin Wolf         ret = -ENOMEM;
2286857d4f46SKevin Wolf         goto ro_cleanup;
2287857d4f46SKevin Wolf     }
22888a426614SKevin Wolf 
22898a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
2290d663640cSPaolo Bonzini         ret = bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n);
2291d663640cSPaolo Bonzini         if (ret < 0) {
2292d663640cSPaolo Bonzini             goto ro_cleanup;
2293d663640cSPaolo Bonzini         }
2294d663640cSPaolo Bonzini         if (ret) {
2295dabfa6ccSKevin Wolf             ret = bdrv_read(bs, sector, buf, n);
2296dabfa6ccSKevin Wolf             if (ret < 0) {
22974dca4b63SNaphtali Sprei                 goto ro_cleanup;
229833e3963eSbellard             }
229933e3963eSbellard 
2300dabfa6ccSKevin Wolf             ret = bdrv_write(bs->backing_hd, sector, buf, n);
2301dabfa6ccSKevin Wolf             if (ret < 0) {
23024dca4b63SNaphtali Sprei                 goto ro_cleanup;
230333e3963eSbellard             }
230433e3963eSbellard         }
230533e3963eSbellard     }
230695389c86Sbellard 
23071d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
23081d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
2309dabfa6ccSKevin Wolf         if (ret < 0) {
2310dabfa6ccSKevin Wolf             goto ro_cleanup;
2311dabfa6ccSKevin Wolf         }
23121d44952fSChristoph Hellwig         bdrv_flush(bs);
23131d44952fSChristoph Hellwig     }
231495389c86Sbellard 
23153f5075aeSChristoph Hellwig     /*
23163f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
23173f5075aeSChristoph Hellwig      * stable on disk.
23183f5075aeSChristoph Hellwig      */
2319dabfa6ccSKevin Wolf     if (bs->backing_hd) {
23203f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
2321dabfa6ccSKevin Wolf     }
23224dca4b63SNaphtali Sprei 
2323dabfa6ccSKevin Wolf     ret = 0;
23244dca4b63SNaphtali Sprei ro_cleanup:
2325857d4f46SKevin Wolf     qemu_vfree(buf);
23264dca4b63SNaphtali Sprei 
23274dca4b63SNaphtali Sprei     if (ro) {
23280bce597dSJeff Cody         /* ignoring error return here */
23290bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
23304dca4b63SNaphtali Sprei     }
23314dca4b63SNaphtali Sprei 
23321d44952fSChristoph Hellwig     return ret;
233333e3963eSbellard }
233433e3963eSbellard 
2335e8877497SStefan Hajnoczi int bdrv_commit_all(void)
23366ab4b5abSMarkus Armbruster {
23376ab4b5abSMarkus Armbruster     BlockDriverState *bs;
23386ab4b5abSMarkus Armbruster 
2339dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
2340ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
2341ed78cda3SStefan Hajnoczi 
2342ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
2343272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
2344e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
2345e8877497SStefan Hajnoczi             if (ret < 0) {
2346ed78cda3SStefan Hajnoczi                 aio_context_release(aio_context);
2347e8877497SStefan Hajnoczi                 return ret;
23486ab4b5abSMarkus Armbruster             }
23496ab4b5abSMarkus Armbruster         }
2350ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
2351272d2d8eSJeff Cody     }
2352e8877497SStefan Hajnoczi     return 0;
2353e8877497SStefan Hajnoczi }
23546ab4b5abSMarkus Armbruster 
2355756e6736SKevin Wolf /*
2356756e6736SKevin Wolf  * Return values:
2357756e6736SKevin Wolf  * 0        - success
2358756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
2359756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
2360756e6736SKevin Wolf  *            image file header
2361756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
2362756e6736SKevin Wolf  */
2363756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
2364756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
2365756e6736SKevin Wolf {
2366756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
2367469ef350SPaolo Bonzini     int ret;
2368756e6736SKevin Wolf 
23695f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
23705f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
23715f377794SPaolo Bonzini         return -EINVAL;
23725f377794SPaolo Bonzini     }
23735f377794SPaolo Bonzini 
2374756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
2375469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
2376756e6736SKevin Wolf     } else {
2377469ef350SPaolo Bonzini         ret = -ENOTSUP;
2378756e6736SKevin Wolf     }
2379469ef350SPaolo Bonzini 
2380469ef350SPaolo Bonzini     if (ret == 0) {
2381469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
2382469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
2383469ef350SPaolo Bonzini     }
2384469ef350SPaolo Bonzini     return ret;
2385756e6736SKevin Wolf }
2386756e6736SKevin Wolf 
23876ebdcee2SJeff Cody /*
23886ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
23896ebdcee2SJeff Cody  *
23906ebdcee2SJeff Cody  * active is the current topmost image.
23916ebdcee2SJeff Cody  *
23926ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
23936ebdcee2SJeff Cody  * or if active == bs.
23944caf0fcdSJeff Cody  *
23954caf0fcdSJeff Cody  * Returns the bottommost base image if bs == NULL.
23966ebdcee2SJeff Cody  */
23976ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
23986ebdcee2SJeff Cody                                     BlockDriverState *bs)
23996ebdcee2SJeff Cody {
24004caf0fcdSJeff Cody     while (active && bs != active->backing_hd) {
24014caf0fcdSJeff Cody         active = active->backing_hd;
24026ebdcee2SJeff Cody     }
24036ebdcee2SJeff Cody 
24044caf0fcdSJeff Cody     return active;
24056ebdcee2SJeff Cody }
24066ebdcee2SJeff Cody 
24074caf0fcdSJeff Cody /* Given a BDS, searches for the base layer. */
24084caf0fcdSJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
24094caf0fcdSJeff Cody {
24104caf0fcdSJeff Cody     return bdrv_find_overlay(bs, NULL);
24116ebdcee2SJeff Cody }
24126ebdcee2SJeff Cody 
24136ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
24146ebdcee2SJeff Cody     BlockDriverState *bs;
24156ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
24166ebdcee2SJeff Cody } BlkIntermediateStates;
24176ebdcee2SJeff Cody 
24186ebdcee2SJeff Cody 
24196ebdcee2SJeff Cody /*
24206ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
24216ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
24226ebdcee2SJeff Cody  *
24236ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
24246ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
24256ebdcee2SJeff Cody  *
24266ebdcee2SJeff Cody  * E.g., this will convert the following chain:
24276ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
24286ebdcee2SJeff Cody  *
24296ebdcee2SJeff Cody  * to
24306ebdcee2SJeff Cody  *
24316ebdcee2SJeff Cody  * bottom <- base <- active
24326ebdcee2SJeff Cody  *
24336ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
24346ebdcee2SJeff Cody  *
24356ebdcee2SJeff Cody  * base <- intermediate <- top <- active
24366ebdcee2SJeff Cody  *
24376ebdcee2SJeff Cody  * to
24386ebdcee2SJeff Cody  *
24396ebdcee2SJeff Cody  * base <- active
24406ebdcee2SJeff Cody  *
244154e26900SJeff Cody  * If backing_file_str is non-NULL, it will be used when modifying top's
244254e26900SJeff Cody  * overlay image metadata.
244354e26900SJeff Cody  *
24446ebdcee2SJeff Cody  * Error conditions:
24456ebdcee2SJeff Cody  *  if active == top, that is considered an error
24466ebdcee2SJeff Cody  *
24476ebdcee2SJeff Cody  */
24486ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
244954e26900SJeff Cody                            BlockDriverState *base, const char *backing_file_str)
24506ebdcee2SJeff Cody {
24516ebdcee2SJeff Cody     BlockDriverState *intermediate;
24526ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
24536ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
24546ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
24556ebdcee2SJeff Cody     int ret = -EIO;
24566ebdcee2SJeff Cody 
24576ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
24586ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
24596ebdcee2SJeff Cody 
24606ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
24616ebdcee2SJeff Cody         goto exit;
24626ebdcee2SJeff Cody     }
24636ebdcee2SJeff Cody 
24646ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
24656ebdcee2SJeff Cody 
24666ebdcee2SJeff Cody     if (new_top_bs == NULL) {
24676ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
24686ebdcee2SJeff Cody         goto exit;
24696ebdcee2SJeff Cody     }
24706ebdcee2SJeff Cody 
24716ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
24726ebdcee2SJeff Cody      * to do, no intermediate images */
24736ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
24746ebdcee2SJeff Cody         ret = 0;
24756ebdcee2SJeff Cody         goto exit;
24766ebdcee2SJeff Cody     }
24776ebdcee2SJeff Cody 
24786ebdcee2SJeff Cody     intermediate = top;
24796ebdcee2SJeff Cody 
24806ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
24816ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
24826ebdcee2SJeff Cody      */
24836ebdcee2SJeff Cody     while (intermediate) {
24845839e53bSMarkus Armbruster         intermediate_state = g_new0(BlkIntermediateStates, 1);
24856ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
24866ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
24876ebdcee2SJeff Cody 
24886ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
24896ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
24906ebdcee2SJeff Cody             break;
24916ebdcee2SJeff Cody         }
24926ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
24936ebdcee2SJeff Cody     }
24946ebdcee2SJeff Cody     if (base_bs == NULL) {
24956ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
24966ebdcee2SJeff Cody          * unravel everything, and exit with error */
24976ebdcee2SJeff Cody         goto exit;
24986ebdcee2SJeff Cody     }
24996ebdcee2SJeff Cody 
25006ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
250154e26900SJeff Cody     backing_file_str = backing_file_str ? backing_file_str : base_bs->filename;
250254e26900SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, backing_file_str,
25036ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
25046ebdcee2SJeff Cody     if (ret) {
25056ebdcee2SJeff Cody         goto exit;
25066ebdcee2SJeff Cody     }
2507920beae1SFam Zheng     bdrv_set_backing_hd(new_top_bs, base_bs);
25086ebdcee2SJeff Cody 
25096ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25106ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
2511920beae1SFam Zheng         bdrv_set_backing_hd(intermediate_state->bs, NULL);
25124f6fd349SFam Zheng         bdrv_unref(intermediate_state->bs);
25136ebdcee2SJeff Cody     }
25146ebdcee2SJeff Cody     ret = 0;
25156ebdcee2SJeff Cody 
25166ebdcee2SJeff Cody exit:
25176ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
25186ebdcee2SJeff Cody         g_free(intermediate_state);
25196ebdcee2SJeff Cody     }
25206ebdcee2SJeff Cody     return ret;
25216ebdcee2SJeff Cody }
25226ebdcee2SJeff Cody 
252383f64091Sbellard /**
252483f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
252583f64091Sbellard  */
252683f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
252783f64091Sbellard {
252883f64091Sbellard     BlockDriver *drv = bs->drv;
252951762288SStefan Hajnoczi     int ret;
253083f64091Sbellard     if (!drv)
253119cb3738Sbellard         return -ENOMEDIUM;
253283f64091Sbellard     if (!drv->bdrv_truncate)
253383f64091Sbellard         return -ENOTSUP;
253459f2689dSNaphtali Sprei     if (bs->read_only)
253559f2689dSNaphtali Sprei         return -EACCES;
25369c75e168SJeff Cody 
253751762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
253851762288SStefan Hajnoczi     if (ret == 0) {
253951762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2540ce1ffea8SJohn Snow         bdrv_dirty_bitmap_truncate(bs);
2541a7f53e26SMarkus Armbruster         if (bs->blk) {
2542a7f53e26SMarkus Armbruster             blk_dev_resize_cb(bs->blk);
2543a7f53e26SMarkus Armbruster         }
254451762288SStefan Hajnoczi     }
254551762288SStefan Hajnoczi     return ret;
254683f64091Sbellard }
254783f64091Sbellard 
254883f64091Sbellard /**
25494a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
25504a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
25514a1d5e1fSFam Zheng  */
25524a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
25534a1d5e1fSFam Zheng {
25544a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
25554a1d5e1fSFam Zheng     if (!drv) {
25564a1d5e1fSFam Zheng         return -ENOMEDIUM;
25574a1d5e1fSFam Zheng     }
25584a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
25594a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
25604a1d5e1fSFam Zheng     }
25614a1d5e1fSFam Zheng     if (bs->file) {
25624a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
25634a1d5e1fSFam Zheng     }
25644a1d5e1fSFam Zheng     return -ENOTSUP;
25654a1d5e1fSFam Zheng }
25664a1d5e1fSFam Zheng 
25674a1d5e1fSFam Zheng /**
256865a9bb25SMarkus Armbruster  * Return number of sectors on success, -errno on error.
256983f64091Sbellard  */
257065a9bb25SMarkus Armbruster int64_t bdrv_nb_sectors(BlockDriverState *bs)
257183f64091Sbellard {
257283f64091Sbellard     BlockDriver *drv = bs->drv;
257365a9bb25SMarkus Armbruster 
257483f64091Sbellard     if (!drv)
257519cb3738Sbellard         return -ENOMEDIUM;
257651762288SStefan Hajnoczi 
2577b94a2610SKevin Wolf     if (drv->has_variable_length) {
2578b94a2610SKevin Wolf         int ret = refresh_total_sectors(bs, bs->total_sectors);
2579b94a2610SKevin Wolf         if (ret < 0) {
2580b94a2610SKevin Wolf             return ret;
2581fc01f7e7Sbellard         }
258246a4e4e6SStefan Hajnoczi     }
258365a9bb25SMarkus Armbruster     return bs->total_sectors;
258465a9bb25SMarkus Armbruster }
258565a9bb25SMarkus Armbruster 
258665a9bb25SMarkus Armbruster /**
258765a9bb25SMarkus Armbruster  * Return length in bytes on success, -errno on error.
258865a9bb25SMarkus Armbruster  * The length is always a multiple of BDRV_SECTOR_SIZE.
258965a9bb25SMarkus Armbruster  */
259065a9bb25SMarkus Armbruster int64_t bdrv_getlength(BlockDriverState *bs)
259165a9bb25SMarkus Armbruster {
259265a9bb25SMarkus Armbruster     int64_t ret = bdrv_nb_sectors(bs);
259365a9bb25SMarkus Armbruster 
25944a9c9ea0SFam Zheng     ret = ret > INT64_MAX / BDRV_SECTOR_SIZE ? -EFBIG : ret;
259565a9bb25SMarkus Armbruster     return ret < 0 ? ret : ret * BDRV_SECTOR_SIZE;
259646a4e4e6SStefan Hajnoczi }
2597fc01f7e7Sbellard 
259819cb3738Sbellard /* return 0 as number of sectors if no device present or error */
259996b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2600fc01f7e7Sbellard {
260165a9bb25SMarkus Armbruster     int64_t nb_sectors = bdrv_nb_sectors(bs);
260265a9bb25SMarkus Armbruster 
260365a9bb25SMarkus Armbruster     *nb_sectors_ptr = nb_sectors < 0 ? 0 : nb_sectors;
2604fc01f7e7Sbellard }
2605cf98951bSbellard 
2606ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2607ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
2608abd7f68dSMarkus Armbruster {
2609abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
2610abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
2611abd7f68dSMarkus Armbruster }
2612abd7f68dSMarkus Armbruster 
26131ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2614abd7f68dSMarkus Armbruster {
2615abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
2616abd7f68dSMarkus Armbruster }
2617abd7f68dSMarkus Armbruster 
26183e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
26193e1caa5fSPaolo Bonzini {
26203e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
26213e1caa5fSPaolo Bonzini 
26223e1caa5fSPaolo Bonzini     switch (on_err) {
26233e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
2624a589569fSWenchao Xia         return (error == ENOSPC) ?
2625a589569fSWenchao Xia                BLOCK_ERROR_ACTION_STOP : BLOCK_ERROR_ACTION_REPORT;
26263e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
2627a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_STOP;
26283e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
2629a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_REPORT;
26303e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
2631a589569fSWenchao Xia         return BLOCK_ERROR_ACTION_IGNORE;
26323e1caa5fSPaolo Bonzini     default:
26333e1caa5fSPaolo Bonzini         abort();
26343e1caa5fSPaolo Bonzini     }
26353e1caa5fSPaolo Bonzini }
26363e1caa5fSPaolo Bonzini 
2637c7c2ff0cSLuiz Capitulino static void send_qmp_error_event(BlockDriverState *bs,
2638c7c2ff0cSLuiz Capitulino                                  BlockErrorAction action,
2639c7c2ff0cSLuiz Capitulino                                  bool is_read, int error)
2640c7c2ff0cSLuiz Capitulino {
2641573742a5SPeter Maydell     IoOperationType optype;
2642c7c2ff0cSLuiz Capitulino 
2643573742a5SPeter Maydell     optype = is_read ? IO_OPERATION_TYPE_READ : IO_OPERATION_TYPE_WRITE;
2644573742a5SPeter Maydell     qapi_event_send_block_io_error(bdrv_get_device_name(bs), optype, action,
2645c7c2ff0cSLuiz Capitulino                                    bdrv_iostatus_is_enabled(bs),
2646624ff573SLuiz Capitulino                                    error == ENOSPC, strerror(error),
2647624ff573SLuiz Capitulino                                    &error_abort);
2648c7c2ff0cSLuiz Capitulino }
2649c7c2ff0cSLuiz Capitulino 
26503e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
26513e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
26523e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
26533e1caa5fSPaolo Bonzini  */
26543e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
26553e1caa5fSPaolo Bonzini                        bool is_read, int error)
26563e1caa5fSPaolo Bonzini {
26573e1caa5fSPaolo Bonzini     assert(error >= 0);
26582bd3bce8SPaolo Bonzini 
2659a589569fSWenchao Xia     if (action == BLOCK_ERROR_ACTION_STOP) {
26602bd3bce8SPaolo Bonzini         /* First set the iostatus, so that "info block" returns an iostatus
26612bd3bce8SPaolo Bonzini          * that matches the events raised so far (an additional error iostatus
26622bd3bce8SPaolo Bonzini          * is fine, but not a lost one).
26632bd3bce8SPaolo Bonzini          */
26643e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
26652bd3bce8SPaolo Bonzini 
26662bd3bce8SPaolo Bonzini         /* Then raise the request to stop the VM and the event.
26672bd3bce8SPaolo Bonzini          * qemu_system_vmstop_request_prepare has two effects.  First,
26682bd3bce8SPaolo Bonzini          * it ensures that the STOP event always comes after the
26692bd3bce8SPaolo Bonzini          * BLOCK_IO_ERROR event.  Second, it ensures that even if management
26702bd3bce8SPaolo Bonzini          * can observe the STOP event and do a "cont" before the STOP
26712bd3bce8SPaolo Bonzini          * event is issued, the VM will not stop.  In this case, vm_start()
26722bd3bce8SPaolo Bonzini          * also ensures that the STOP/RESUME pair of events is emitted.
26732bd3bce8SPaolo Bonzini          */
26742bd3bce8SPaolo Bonzini         qemu_system_vmstop_request_prepare();
2675c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26762bd3bce8SPaolo Bonzini         qemu_system_vmstop_request(RUN_STATE_IO_ERROR);
26772bd3bce8SPaolo Bonzini     } else {
2678c7c2ff0cSLuiz Capitulino         send_qmp_error_event(bs, action, is_read, error);
26793e1caa5fSPaolo Bonzini     }
26803e1caa5fSPaolo Bonzini }
26813e1caa5fSPaolo Bonzini 
2682b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
2683b338082bSbellard {
2684b338082bSbellard     return bs->read_only;
2685b338082bSbellard }
2686b338082bSbellard 
2687985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
2688985a03b0Sths {
2689985a03b0Sths     return bs->sg;
2690985a03b0Sths }
2691985a03b0Sths 
2692e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
2693e900a7b7SChristoph Hellwig {
2694e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
2695e900a7b7SChristoph Hellwig }
2696e900a7b7SChristoph Hellwig 
2697425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2698425b0148SPaolo Bonzini {
2699425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
270055b110f2SJeff Cody 
270155b110f2SJeff Cody     /* so a reopen() will preserve wce */
270255b110f2SJeff Cody     if (wce) {
270355b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
270455b110f2SJeff Cody     } else {
270555b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
270655b110f2SJeff Cody     }
2707425b0148SPaolo Bonzini }
2708425b0148SPaolo Bonzini 
2709ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
2710ea2384d3Sbellard {
2711ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
2712ea2384d3Sbellard         return 1;
2713ea2384d3Sbellard     return bs->encrypted;
2714ea2384d3Sbellard }
2715ea2384d3Sbellard 
2716c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
2717c0f4ce77Saliguori {
2718c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
2719c0f4ce77Saliguori 
2720c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2721c0f4ce77Saliguori         return 1;
2722c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
2723c0f4ce77Saliguori }
2724c0f4ce77Saliguori 
2725ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
2726ea2384d3Sbellard {
2727ea2384d3Sbellard     int ret;
2728ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
2729ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
2730ea2384d3Sbellard         if (ret < 0)
2731ea2384d3Sbellard             return ret;
2732ea2384d3Sbellard         if (!bs->encrypted)
2733ea2384d3Sbellard             return 0;
2734ea2384d3Sbellard     }
2735fd04a2aeSShahar Havivi     if (!bs->encrypted) {
2736fd04a2aeSShahar Havivi         return -EINVAL;
2737fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2738fd04a2aeSShahar Havivi         return -ENOMEDIUM;
2739fd04a2aeSShahar Havivi     }
2740c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
2741bb5fc20fSaliguori     if (ret < 0) {
2742bb5fc20fSaliguori         bs->valid_key = 0;
2743bb5fc20fSaliguori     } else if (!bs->valid_key) {
2744bb5fc20fSaliguori         bs->valid_key = 1;
2745a7f53e26SMarkus Armbruster         if (bs->blk) {
2746bb5fc20fSaliguori             /* call the change callback now, we skipped it on open */
2747a7f53e26SMarkus Armbruster             blk_dev_change_media_cb(bs->blk, true);
2748a7f53e26SMarkus Armbruster         }
2749bb5fc20fSaliguori     }
2750c0f4ce77Saliguori     return ret;
2751ea2384d3Sbellard }
2752ea2384d3Sbellard 
27534d2855a3SMarkus Armbruster /*
27544d2855a3SMarkus Armbruster  * Provide an encryption key for @bs.
27554d2855a3SMarkus Armbruster  * If @key is non-null:
27564d2855a3SMarkus Armbruster  *     If @bs is not encrypted, fail.
27574d2855a3SMarkus Armbruster  *     Else if the key is invalid, fail.
27584d2855a3SMarkus Armbruster  *     Else set @bs's key to @key, replacing the existing key, if any.
27594d2855a3SMarkus Armbruster  * If @key is null:
27604d2855a3SMarkus Armbruster  *     If @bs is encrypted and still lacks a key, fail.
27614d2855a3SMarkus Armbruster  *     Else do nothing.
27624d2855a3SMarkus Armbruster  * On failure, store an error object through @errp if non-null.
27634d2855a3SMarkus Armbruster  */
27644d2855a3SMarkus Armbruster void bdrv_add_key(BlockDriverState *bs, const char *key, Error **errp)
27654d2855a3SMarkus Armbruster {
27664d2855a3SMarkus Armbruster     if (key) {
27674d2855a3SMarkus Armbruster         if (!bdrv_is_encrypted(bs)) {
276881e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is not encrypted",
276981e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs));
27704d2855a3SMarkus Armbruster         } else if (bdrv_set_key(bs, key) < 0) {
2771c6bd8c70SMarkus Armbruster             error_setg(errp, QERR_INVALID_PASSWORD);
27724d2855a3SMarkus Armbruster         }
27734d2855a3SMarkus Armbruster     } else {
27744d2855a3SMarkus Armbruster         if (bdrv_key_required(bs)) {
2775b1ca6391SMarkus Armbruster             error_set(errp, ERROR_CLASS_DEVICE_ENCRYPTED,
2776b1ca6391SMarkus Armbruster                       "'%s' (%s) is encrypted",
277781e5f78aSAlberto Garcia                       bdrv_get_device_or_node_name(bs),
27784d2855a3SMarkus Armbruster                       bdrv_get_encrypted_filename(bs));
27794d2855a3SMarkus Armbruster         }
27804d2855a3SMarkus Armbruster     }
27814d2855a3SMarkus Armbruster }
27824d2855a3SMarkus Armbruster 
2783f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
2784ea2384d3Sbellard {
2785f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
2786ea2384d3Sbellard }
2787ea2384d3Sbellard 
2788ada42401SStefan Hajnoczi static int qsort_strcmp(const void *a, const void *b)
2789ada42401SStefan Hajnoczi {
2790ada42401SStefan Hajnoczi     return strcmp(a, b);
2791ada42401SStefan Hajnoczi }
2792ada42401SStefan Hajnoczi 
2793ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2794ea2384d3Sbellard                          void *opaque)
2795ea2384d3Sbellard {
2796ea2384d3Sbellard     BlockDriver *drv;
2797e855e4fbSJeff Cody     int count = 0;
2798ada42401SStefan Hajnoczi     int i;
2799e855e4fbSJeff Cody     const char **formats = NULL;
2800ea2384d3Sbellard 
28018a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2802e855e4fbSJeff Cody         if (drv->format_name) {
2803e855e4fbSJeff Cody             bool found = false;
2804e855e4fbSJeff Cody             int i = count;
2805e855e4fbSJeff Cody             while (formats && i && !found) {
2806e855e4fbSJeff Cody                 found = !strcmp(formats[--i], drv->format_name);
2807e855e4fbSJeff Cody             }
2808e855e4fbSJeff Cody 
2809e855e4fbSJeff Cody             if (!found) {
28105839e53bSMarkus Armbruster                 formats = g_renew(const char *, formats, count + 1);
2811e855e4fbSJeff Cody                 formats[count++] = drv->format_name;
2812ea2384d3Sbellard             }
2813ea2384d3Sbellard         }
2814e855e4fbSJeff Cody     }
2815ada42401SStefan Hajnoczi 
2816ada42401SStefan Hajnoczi     qsort(formats, count, sizeof(formats[0]), qsort_strcmp);
2817ada42401SStefan Hajnoczi 
2818ada42401SStefan Hajnoczi     for (i = 0; i < count; i++) {
2819ada42401SStefan Hajnoczi         it(opaque, formats[i]);
2820ada42401SStefan Hajnoczi     }
2821ada42401SStefan Hajnoczi 
2822e855e4fbSJeff Cody     g_free(formats);
2823e855e4fbSJeff Cody }
2824ea2384d3Sbellard 
2825dc364f4cSBenoît Canet /* This function is to find a node in the bs graph */
2826dc364f4cSBenoît Canet BlockDriverState *bdrv_find_node(const char *node_name)
2827dc364f4cSBenoît Canet {
2828dc364f4cSBenoît Canet     BlockDriverState *bs;
2829dc364f4cSBenoît Canet 
2830dc364f4cSBenoît Canet     assert(node_name);
2831dc364f4cSBenoît Canet 
2832dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2833dc364f4cSBenoît Canet         if (!strcmp(node_name, bs->node_name)) {
2834dc364f4cSBenoît Canet             return bs;
2835dc364f4cSBenoît Canet         }
2836dc364f4cSBenoît Canet     }
2837dc364f4cSBenoît Canet     return NULL;
2838dc364f4cSBenoît Canet }
2839dc364f4cSBenoît Canet 
2840c13163fbSBenoît Canet /* Put this QMP function here so it can access the static graph_bdrv_states. */
2841d5a8ee60SAlberto Garcia BlockDeviceInfoList *bdrv_named_nodes_list(Error **errp)
2842c13163fbSBenoît Canet {
2843c13163fbSBenoît Canet     BlockDeviceInfoList *list, *entry;
2844c13163fbSBenoît Canet     BlockDriverState *bs;
2845c13163fbSBenoît Canet 
2846c13163fbSBenoît Canet     list = NULL;
2847c13163fbSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
2848d5a8ee60SAlberto Garcia         BlockDeviceInfo *info = bdrv_block_device_info(bs, errp);
2849d5a8ee60SAlberto Garcia         if (!info) {
2850d5a8ee60SAlberto Garcia             qapi_free_BlockDeviceInfoList(list);
2851d5a8ee60SAlberto Garcia             return NULL;
2852d5a8ee60SAlberto Garcia         }
2853c13163fbSBenoît Canet         entry = g_malloc0(sizeof(*entry));
2854d5a8ee60SAlberto Garcia         entry->value = info;
2855c13163fbSBenoît Canet         entry->next = list;
2856c13163fbSBenoît Canet         list = entry;
2857c13163fbSBenoît Canet     }
2858c13163fbSBenoît Canet 
2859c13163fbSBenoît Canet     return list;
2860c13163fbSBenoît Canet }
2861c13163fbSBenoît Canet 
286212d3ba82SBenoît Canet BlockDriverState *bdrv_lookup_bs(const char *device,
286312d3ba82SBenoît Canet                                  const char *node_name,
286412d3ba82SBenoît Canet                                  Error **errp)
286512d3ba82SBenoît Canet {
28667f06d47eSMarkus Armbruster     BlockBackend *blk;
28677f06d47eSMarkus Armbruster     BlockDriverState *bs;
286812d3ba82SBenoît Canet 
286912d3ba82SBenoît Canet     if (device) {
28707f06d47eSMarkus Armbruster         blk = blk_by_name(device);
287112d3ba82SBenoît Canet 
28727f06d47eSMarkus Armbruster         if (blk) {
28737f06d47eSMarkus Armbruster             return blk_bs(blk);
287412d3ba82SBenoît Canet         }
2875dd67fa50SBenoît Canet     }
287612d3ba82SBenoît Canet 
2877dd67fa50SBenoît Canet     if (node_name) {
287812d3ba82SBenoît Canet         bs = bdrv_find_node(node_name);
287912d3ba82SBenoît Canet 
2880dd67fa50SBenoît Canet         if (bs) {
2881dd67fa50SBenoît Canet             return bs;
2882dd67fa50SBenoît Canet         }
288312d3ba82SBenoît Canet     }
288412d3ba82SBenoît Canet 
2885dd67fa50SBenoît Canet     error_setg(errp, "Cannot find device=%s nor node_name=%s",
2886dd67fa50SBenoît Canet                      device ? device : "",
2887dd67fa50SBenoît Canet                      node_name ? node_name : "");
2888dd67fa50SBenoît Canet     return NULL;
288912d3ba82SBenoît Canet }
289012d3ba82SBenoît Canet 
28915a6684d2SJeff Cody /* If 'base' is in the same chain as 'top', return true. Otherwise,
28925a6684d2SJeff Cody  * return false.  If either argument is NULL, return false. */
28935a6684d2SJeff Cody bool bdrv_chain_contains(BlockDriverState *top, BlockDriverState *base)
28945a6684d2SJeff Cody {
28955a6684d2SJeff Cody     while (top && top != base) {
28965a6684d2SJeff Cody         top = top->backing_hd;
28975a6684d2SJeff Cody     }
28985a6684d2SJeff Cody 
28995a6684d2SJeff Cody     return top != NULL;
29005a6684d2SJeff Cody }
29015a6684d2SJeff Cody 
290204df765aSFam Zheng BlockDriverState *bdrv_next_node(BlockDriverState *bs)
290304df765aSFam Zheng {
290404df765aSFam Zheng     if (!bs) {
290504df765aSFam Zheng         return QTAILQ_FIRST(&graph_bdrv_states);
290604df765aSFam Zheng     }
290704df765aSFam Zheng     return QTAILQ_NEXT(bs, node_list);
290804df765aSFam Zheng }
290904df765aSFam Zheng 
29102f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
29112f399b0aSMarkus Armbruster {
29122f399b0aSMarkus Armbruster     if (!bs) {
29132f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
29142f399b0aSMarkus Armbruster     }
2915dc364f4cSBenoît Canet     return QTAILQ_NEXT(bs, device_list);
29162f399b0aSMarkus Armbruster }
29172f399b0aSMarkus Armbruster 
291820a9e77dSFam Zheng const char *bdrv_get_node_name(const BlockDriverState *bs)
291920a9e77dSFam Zheng {
292020a9e77dSFam Zheng     return bs->node_name;
292120a9e77dSFam Zheng }
292220a9e77dSFam Zheng 
29237f06d47eSMarkus Armbruster /* TODO check what callers really want: bs->node_name or blk_name() */
2924bfb197e0SMarkus Armbruster const char *bdrv_get_device_name(const BlockDriverState *bs)
2925ea2384d3Sbellard {
2926bfb197e0SMarkus Armbruster     return bs->blk ? blk_name(bs->blk) : "";
2927ea2384d3Sbellard }
2928ea2384d3Sbellard 
29299b2aa84fSAlberto Garcia /* This can be used to identify nodes that might not have a device
29309b2aa84fSAlberto Garcia  * name associated. Since node and device names live in the same
29319b2aa84fSAlberto Garcia  * namespace, the result is unambiguous. The exception is if both are
29329b2aa84fSAlberto Garcia  * absent, then this returns an empty (non-null) string. */
29339b2aa84fSAlberto Garcia const char *bdrv_get_device_or_node_name(const BlockDriverState *bs)
29349b2aa84fSAlberto Garcia {
29359b2aa84fSAlberto Garcia     return bs->blk ? blk_name(bs->blk) : bs->node_name;
29369b2aa84fSAlberto Garcia }
29379b2aa84fSAlberto Garcia 
2938c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
2939c8433287SMarkus Armbruster {
2940c8433287SMarkus Armbruster     return bs->open_flags;
2941c8433287SMarkus Armbruster }
2942c8433287SMarkus Armbruster 
29433ac21627SPeter Lieven int bdrv_has_zero_init_1(BlockDriverState *bs)
29443ac21627SPeter Lieven {
29453ac21627SPeter Lieven     return 1;
29463ac21627SPeter Lieven }
29473ac21627SPeter Lieven 
2948f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
2949f2feebbdSKevin Wolf {
2950f2feebbdSKevin Wolf     assert(bs->drv);
2951f2feebbdSKevin Wolf 
295211212d8fSPaolo Bonzini     /* If BS is a copy on write image, it is initialized to
295311212d8fSPaolo Bonzini        the contents of the base image, which may not be zeroes.  */
295411212d8fSPaolo Bonzini     if (bs->backing_hd) {
295511212d8fSPaolo Bonzini         return 0;
295611212d8fSPaolo Bonzini     }
2957336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
2958336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
2959f2feebbdSKevin Wolf     }
2960f2feebbdSKevin Wolf 
29613ac21627SPeter Lieven     /* safe default */
29623ac21627SPeter Lieven     return 0;
2963f2feebbdSKevin Wolf }
2964f2feebbdSKevin Wolf 
29654ce78691SPeter Lieven bool bdrv_unallocated_blocks_are_zero(BlockDriverState *bs)
29664ce78691SPeter Lieven {
29674ce78691SPeter Lieven     BlockDriverInfo bdi;
29684ce78691SPeter Lieven 
29694ce78691SPeter Lieven     if (bs->backing_hd) {
29704ce78691SPeter Lieven         return false;
29714ce78691SPeter Lieven     }
29724ce78691SPeter Lieven 
29734ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29744ce78691SPeter Lieven         return bdi.unallocated_blocks_are_zero;
29754ce78691SPeter Lieven     }
29764ce78691SPeter Lieven 
29774ce78691SPeter Lieven     return false;
29784ce78691SPeter Lieven }
29794ce78691SPeter Lieven 
29804ce78691SPeter Lieven bool bdrv_can_write_zeroes_with_unmap(BlockDriverState *bs)
29814ce78691SPeter Lieven {
29824ce78691SPeter Lieven     BlockDriverInfo bdi;
29834ce78691SPeter Lieven 
29844ce78691SPeter Lieven     if (bs->backing_hd || !(bs->open_flags & BDRV_O_UNMAP)) {
29854ce78691SPeter Lieven         return false;
29864ce78691SPeter Lieven     }
29874ce78691SPeter Lieven 
29884ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
29894ce78691SPeter Lieven         return bdi.can_write_zeroes_with_unmap;
29904ce78691SPeter Lieven     }
29914ce78691SPeter Lieven 
29924ce78691SPeter Lieven     return false;
29934ce78691SPeter Lieven }
29944ce78691SPeter Lieven 
2995045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
2996045df330Saliguori {
2997045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
2998045df330Saliguori         return bs->backing_file;
2999045df330Saliguori     else if (bs->encrypted)
3000045df330Saliguori         return bs->filename;
3001045df330Saliguori     else
3002045df330Saliguori         return NULL;
3003045df330Saliguori }
3004045df330Saliguori 
300583f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
300683f64091Sbellard                                char *filename, int filename_size)
300783f64091Sbellard {
300883f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
300983f64091Sbellard }
301083f64091Sbellard 
3011faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3012faea38e7Sbellard {
3013faea38e7Sbellard     BlockDriver *drv = bs->drv;
3014faea38e7Sbellard     if (!drv)
301519cb3738Sbellard         return -ENOMEDIUM;
3016faea38e7Sbellard     if (!drv->bdrv_get_info)
3017faea38e7Sbellard         return -ENOTSUP;
3018faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3019faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3020faea38e7Sbellard }
3021faea38e7Sbellard 
3022eae041feSMax Reitz ImageInfoSpecific *bdrv_get_specific_info(BlockDriverState *bs)
3023eae041feSMax Reitz {
3024eae041feSMax Reitz     BlockDriver *drv = bs->drv;
3025eae041feSMax Reitz     if (drv && drv->bdrv_get_specific_info) {
3026eae041feSMax Reitz         return drv->bdrv_get_specific_info(bs);
3027eae041feSMax Reitz     }
3028eae041feSMax Reitz     return NULL;
3029eae041feSMax Reitz }
3030eae041feSMax Reitz 
30318b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
30328b9b0cc2SKevin Wolf {
3033bf736fe3SKevin Wolf     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
30348b9b0cc2SKevin Wolf         return;
30358b9b0cc2SKevin Wolf     }
30368b9b0cc2SKevin Wolf 
3037bf736fe3SKevin Wolf     bs->drv->bdrv_debug_event(bs, event);
303841c695c7SKevin Wolf }
30398b9b0cc2SKevin Wolf 
304041c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
304141c695c7SKevin Wolf                           const char *tag)
304241c695c7SKevin Wolf {
304341c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
304441c695c7SKevin Wolf         bs = bs->file;
304541c695c7SKevin Wolf     }
304641c695c7SKevin Wolf 
304741c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
304841c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
304941c695c7SKevin Wolf     }
305041c695c7SKevin Wolf 
305141c695c7SKevin Wolf     return -ENOTSUP;
305241c695c7SKevin Wolf }
305341c695c7SKevin Wolf 
30544cc70e93SFam Zheng int bdrv_debug_remove_breakpoint(BlockDriverState *bs, const char *tag)
30554cc70e93SFam Zheng {
30564cc70e93SFam Zheng     while (bs && bs->drv && !bs->drv->bdrv_debug_remove_breakpoint) {
30574cc70e93SFam Zheng         bs = bs->file;
30584cc70e93SFam Zheng     }
30594cc70e93SFam Zheng 
30604cc70e93SFam Zheng     if (bs && bs->drv && bs->drv->bdrv_debug_remove_breakpoint) {
30614cc70e93SFam Zheng         return bs->drv->bdrv_debug_remove_breakpoint(bs, tag);
30624cc70e93SFam Zheng     }
30634cc70e93SFam Zheng 
30644cc70e93SFam Zheng     return -ENOTSUP;
30654cc70e93SFam Zheng }
30664cc70e93SFam Zheng 
306741c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
306841c695c7SKevin Wolf {
3069938789eaSMax Reitz     while (bs && (!bs->drv || !bs->drv->bdrv_debug_resume)) {
307041c695c7SKevin Wolf         bs = bs->file;
307141c695c7SKevin Wolf     }
307241c695c7SKevin Wolf 
307341c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
307441c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
307541c695c7SKevin Wolf     }
307641c695c7SKevin Wolf 
307741c695c7SKevin Wolf     return -ENOTSUP;
307841c695c7SKevin Wolf }
307941c695c7SKevin Wolf 
308041c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
308141c695c7SKevin Wolf {
308241c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
308341c695c7SKevin Wolf         bs = bs->file;
308441c695c7SKevin Wolf     }
308541c695c7SKevin Wolf 
308641c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
308741c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
308841c695c7SKevin Wolf     }
308941c695c7SKevin Wolf 
309041c695c7SKevin Wolf     return false;
30918b9b0cc2SKevin Wolf }
30928b9b0cc2SKevin Wolf 
3093199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3094199630b6SBlue Swirl {
3095199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3096199630b6SBlue Swirl }
3097199630b6SBlue Swirl 
3098b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3099b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3100b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3101b1b1d783SJeff Cody  * the CWD rather than the chain. */
3102e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3103e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3104e8a6bb9cSMarcelo Tosatti {
3105b1b1d783SJeff Cody     char *filename_full = NULL;
3106b1b1d783SJeff Cody     char *backing_file_full = NULL;
3107b1b1d783SJeff Cody     char *filename_tmp = NULL;
3108b1b1d783SJeff Cody     int is_protocol = 0;
3109b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3110b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3111b1b1d783SJeff Cody 
3112b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3113e8a6bb9cSMarcelo Tosatti         return NULL;
3114e8a6bb9cSMarcelo Tosatti     }
3115e8a6bb9cSMarcelo Tosatti 
3116b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
3117b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
3118b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
3119b1b1d783SJeff Cody 
3120b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
3121b1b1d783SJeff Cody 
3122b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3123b1b1d783SJeff Cody 
3124b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
3125b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
3126b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3127b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3128b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3129b1b1d783SJeff Cody                 break;
3130b1b1d783SJeff Cody             }
3131e8a6bb9cSMarcelo Tosatti         } else {
3132b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
3133b1b1d783SJeff Cody              * image's filename path */
3134b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3135b1b1d783SJeff Cody                          backing_file);
3136b1b1d783SJeff Cody 
3137b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
3138b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
3139b1b1d783SJeff Cody                 continue;
3140b1b1d783SJeff Cody             }
3141b1b1d783SJeff Cody 
3142b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
3143b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
3144b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3145b1b1d783SJeff Cody                          curr_bs->backing_file);
3146b1b1d783SJeff Cody 
3147b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
3148b1b1d783SJeff Cody                 continue;
3149b1b1d783SJeff Cody             }
3150b1b1d783SJeff Cody 
3151b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
3152b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3153b1b1d783SJeff Cody                 break;
3154b1b1d783SJeff Cody             }
3155e8a6bb9cSMarcelo Tosatti         }
3156e8a6bb9cSMarcelo Tosatti     }
3157e8a6bb9cSMarcelo Tosatti 
3158b1b1d783SJeff Cody     g_free(filename_full);
3159b1b1d783SJeff Cody     g_free(backing_file_full);
3160b1b1d783SJeff Cody     g_free(filename_tmp);
3161b1b1d783SJeff Cody     return retval;
3162e8a6bb9cSMarcelo Tosatti }
3163e8a6bb9cSMarcelo Tosatti 
3164f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
3165f198fd1cSBenoît Canet {
3166f198fd1cSBenoît Canet     if (!bs->drv) {
3167f198fd1cSBenoît Canet         return 0;
3168f198fd1cSBenoît Canet     }
3169f198fd1cSBenoît Canet 
3170f198fd1cSBenoît Canet     if (!bs->backing_hd) {
3171f198fd1cSBenoît Canet         return 0;
3172f198fd1cSBenoît Canet     }
3173f198fd1cSBenoît Canet 
3174f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3175f198fd1cSBenoît Canet }
3176f198fd1cSBenoît Canet 
3177ea2384d3Sbellard void bdrv_init(void)
3178ea2384d3Sbellard {
31795efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
3180ea2384d3Sbellard }
3181ce1a14dcSpbrook 
3182eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
3183eb852011SMarkus Armbruster {
3184eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
3185eb852011SMarkus Armbruster     bdrv_init();
3186eb852011SMarkus Armbruster }
3187eb852011SMarkus Armbruster 
31885a8a30dbSKevin Wolf void bdrv_invalidate_cache(BlockDriverState *bs, Error **errp)
31890f15423cSAnthony Liguori {
31905a8a30dbSKevin Wolf     Error *local_err = NULL;
31915a8a30dbSKevin Wolf     int ret;
31925a8a30dbSKevin Wolf 
31933456a8d1SKevin Wolf     if (!bs->drv)  {
31943456a8d1SKevin Wolf         return;
31950f15423cSAnthony Liguori     }
31963456a8d1SKevin Wolf 
31977ea2d269SAlexey Kardashevskiy     if (!(bs->open_flags & BDRV_O_INCOMING)) {
31987ea2d269SAlexey Kardashevskiy         return;
31997ea2d269SAlexey Kardashevskiy     }
32007ea2d269SAlexey Kardashevskiy     bs->open_flags &= ~BDRV_O_INCOMING;
32017ea2d269SAlexey Kardashevskiy 
32023456a8d1SKevin Wolf     if (bs->drv->bdrv_invalidate_cache) {
32035a8a30dbSKevin Wolf         bs->drv->bdrv_invalidate_cache(bs, &local_err);
32043456a8d1SKevin Wolf     } else if (bs->file) {
32055a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs->file, &local_err);
32065a8a30dbSKevin Wolf     }
32075a8a30dbSKevin Wolf     if (local_err) {
32085a8a30dbSKevin Wolf         error_propagate(errp, local_err);
32095a8a30dbSKevin Wolf         return;
32103456a8d1SKevin Wolf     }
32113456a8d1SKevin Wolf 
32125a8a30dbSKevin Wolf     ret = refresh_total_sectors(bs, bs->total_sectors);
32135a8a30dbSKevin Wolf     if (ret < 0) {
32145a8a30dbSKevin Wolf         error_setg_errno(errp, -ret, "Could not refresh total sector count");
32155a8a30dbSKevin Wolf         return;
32165a8a30dbSKevin Wolf     }
32170f15423cSAnthony Liguori }
32180f15423cSAnthony Liguori 
32195a8a30dbSKevin Wolf void bdrv_invalidate_cache_all(Error **errp)
32200f15423cSAnthony Liguori {
32210f15423cSAnthony Liguori     BlockDriverState *bs;
32225a8a30dbSKevin Wolf     Error *local_err = NULL;
32230f15423cSAnthony Liguori 
3224dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
3225ed78cda3SStefan Hajnoczi         AioContext *aio_context = bdrv_get_aio_context(bs);
3226ed78cda3SStefan Hajnoczi 
3227ed78cda3SStefan Hajnoczi         aio_context_acquire(aio_context);
32285a8a30dbSKevin Wolf         bdrv_invalidate_cache(bs, &local_err);
3229ed78cda3SStefan Hajnoczi         aio_context_release(aio_context);
32305a8a30dbSKevin Wolf         if (local_err) {
32315a8a30dbSKevin Wolf             error_propagate(errp, local_err);
32325a8a30dbSKevin Wolf             return;
32335a8a30dbSKevin Wolf         }
32340f15423cSAnthony Liguori     }
32350f15423cSAnthony Liguori }
32360f15423cSAnthony Liguori 
3237f9f05dc5SKevin Wolf /**************************************************************/
323819cb3738Sbellard /* removable device support */
323919cb3738Sbellard 
324019cb3738Sbellard /**
324119cb3738Sbellard  * Return TRUE if the media is present
324219cb3738Sbellard  */
324319cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
324419cb3738Sbellard {
324519cb3738Sbellard     BlockDriver *drv = bs->drv;
3246a1aff5bfSMarkus Armbruster 
324719cb3738Sbellard     if (!drv)
324819cb3738Sbellard         return 0;
324919cb3738Sbellard     if (!drv->bdrv_is_inserted)
3250a1aff5bfSMarkus Armbruster         return 1;
3251a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
325219cb3738Sbellard }
325319cb3738Sbellard 
325419cb3738Sbellard /**
32558e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
32568e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
325719cb3738Sbellard  */
325819cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
325919cb3738Sbellard {
326019cb3738Sbellard     BlockDriver *drv = bs->drv;
326119cb3738Sbellard 
32628e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
32638e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
32648e49ca46SMarkus Armbruster     }
32658e49ca46SMarkus Armbruster     return -ENOTSUP;
326619cb3738Sbellard }
326719cb3738Sbellard 
326819cb3738Sbellard /**
326919cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
327019cb3738Sbellard  */
3271f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
327219cb3738Sbellard {
327319cb3738Sbellard     BlockDriver *drv = bs->drv;
3274bfb197e0SMarkus Armbruster     const char *device_name;
327519cb3738Sbellard 
3276822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
3277822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
327819cb3738Sbellard     }
32796f382ed2SLuiz Capitulino 
3280bfb197e0SMarkus Armbruster     device_name = bdrv_get_device_name(bs);
3281bfb197e0SMarkus Armbruster     if (device_name[0] != '\0') {
3282bfb197e0SMarkus Armbruster         qapi_event_send_device_tray_moved(device_name,
3283a5ee7bd4SWenchao Xia                                           eject_flag, &error_abort);
32846f382ed2SLuiz Capitulino     }
328519cb3738Sbellard }
328619cb3738Sbellard 
328719cb3738Sbellard /**
328819cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
328919cb3738Sbellard  * to eject it manually).
329019cb3738Sbellard  */
3291025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
329219cb3738Sbellard {
329319cb3738Sbellard     BlockDriver *drv = bs->drv;
329419cb3738Sbellard 
3295025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
3296b8c6d095SStefan Hajnoczi 
3297025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
3298025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
329919cb3738Sbellard     }
330019cb3738Sbellard }
3301985a03b0Sths 
33021b7fd729SPaolo Bonzini void bdrv_set_guest_block_size(BlockDriverState *bs, int align)
33037b6f9300SMarkus Armbruster {
33041b7fd729SPaolo Bonzini     bs->guest_block_size = align;
33057b6f9300SMarkus Armbruster }
33067cd1e32aSlirans@il.ibm.com 
33070db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_find_dirty_bitmap(BlockDriverState *bs, const char *name)
33080db6e54aSFam Zheng {
33090db6e54aSFam Zheng     BdrvDirtyBitmap *bm;
33100db6e54aSFam Zheng 
33110db6e54aSFam Zheng     assert(name);
33120db6e54aSFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
33130db6e54aSFam Zheng         if (bm->name && !strcmp(name, bm->name)) {
33140db6e54aSFam Zheng             return bm;
33150db6e54aSFam Zheng         }
33160db6e54aSFam Zheng     }
33170db6e54aSFam Zheng     return NULL;
33180db6e54aSFam Zheng }
33190db6e54aSFam Zheng 
332020dca810SJohn Snow void bdrv_dirty_bitmap_make_anon(BdrvDirtyBitmap *bitmap)
33210db6e54aSFam Zheng {
33229bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
33230db6e54aSFam Zheng     g_free(bitmap->name);
33240db6e54aSFam Zheng     bitmap->name = NULL;
33250db6e54aSFam Zheng }
33260db6e54aSFam Zheng 
33270db6e54aSFam Zheng BdrvDirtyBitmap *bdrv_create_dirty_bitmap(BlockDriverState *bs,
33285fba6c0eSJohn Snow                                           uint32_t granularity,
33290db6e54aSFam Zheng                                           const char *name,
3330b8afb520SFam Zheng                                           Error **errp)
33317cd1e32aSlirans@il.ibm.com {
33327cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
3333e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
33345fba6c0eSJohn Snow     uint32_t sector_granularity;
3335a55eb92cSJan Kiszka 
333650717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
333750717e94SPaolo Bonzini 
33380db6e54aSFam Zheng     if (name && bdrv_find_dirty_bitmap(bs, name)) {
33390db6e54aSFam Zheng         error_setg(errp, "Bitmap already exists: %s", name);
33400db6e54aSFam Zheng         return NULL;
33410db6e54aSFam Zheng     }
33425fba6c0eSJohn Snow     sector_granularity = granularity >> BDRV_SECTOR_BITS;
33435fba6c0eSJohn Snow     assert(sector_granularity);
334457322b78SMarkus Armbruster     bitmap_size = bdrv_nb_sectors(bs);
3345b8afb520SFam Zheng     if (bitmap_size < 0) {
3346b8afb520SFam Zheng         error_setg_errno(errp, -bitmap_size, "could not get length of device");
3347b8afb520SFam Zheng         errno = -bitmap_size;
3348b8afb520SFam Zheng         return NULL;
3349b8afb520SFam Zheng     }
33505839e53bSMarkus Armbruster     bitmap = g_new0(BdrvDirtyBitmap, 1);
33515fba6c0eSJohn Snow     bitmap->bitmap = hbitmap_alloc(bitmap_size, ctz32(sector_granularity));
3352e74e6b78SJohn Snow     bitmap->size = bitmap_size;
33530db6e54aSFam Zheng     bitmap->name = g_strdup(name);
3354b8e6fb75SJohn Snow     bitmap->disabled = false;
3355e4654d2dSFam Zheng     QLIST_INSERT_HEAD(&bs->dirty_bitmaps, bitmap, list);
3356e4654d2dSFam Zheng     return bitmap;
3357e4654d2dSFam Zheng }
3358e4654d2dSFam Zheng 
33599bd2b08fSJohn Snow bool bdrv_dirty_bitmap_frozen(BdrvDirtyBitmap *bitmap)
33609bd2b08fSJohn Snow {
33619bd2b08fSJohn Snow     return bitmap->successor;
33629bd2b08fSJohn Snow }
33639bd2b08fSJohn Snow 
3364b8e6fb75SJohn Snow bool bdrv_dirty_bitmap_enabled(BdrvDirtyBitmap *bitmap)
3365b8e6fb75SJohn Snow {
33669bd2b08fSJohn Snow     return !(bitmap->disabled || bitmap->successor);
33679bd2b08fSJohn Snow }
33689bd2b08fSJohn Snow 
33699abe3bdcSJohn Snow DirtyBitmapStatus bdrv_dirty_bitmap_status(BdrvDirtyBitmap *bitmap)
33709abe3bdcSJohn Snow {
33719abe3bdcSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33729abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_FROZEN;
33739abe3bdcSJohn Snow     } else if (!bdrv_dirty_bitmap_enabled(bitmap)) {
33749abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_DISABLED;
33759abe3bdcSJohn Snow     } else {
33769abe3bdcSJohn Snow         return DIRTY_BITMAP_STATUS_ACTIVE;
33779abe3bdcSJohn Snow     }
33789abe3bdcSJohn Snow }
33799abe3bdcSJohn Snow 
33809bd2b08fSJohn Snow /**
33819bd2b08fSJohn Snow  * Create a successor bitmap destined to replace this bitmap after an operation.
33829bd2b08fSJohn Snow  * Requires that the bitmap is not frozen and has no successor.
33839bd2b08fSJohn Snow  */
33849bd2b08fSJohn Snow int bdrv_dirty_bitmap_create_successor(BlockDriverState *bs,
33859bd2b08fSJohn Snow                                        BdrvDirtyBitmap *bitmap, Error **errp)
33869bd2b08fSJohn Snow {
33879bd2b08fSJohn Snow     uint64_t granularity;
33889bd2b08fSJohn Snow     BdrvDirtyBitmap *child;
33899bd2b08fSJohn Snow 
33909bd2b08fSJohn Snow     if (bdrv_dirty_bitmap_frozen(bitmap)) {
33919bd2b08fSJohn Snow         error_setg(errp, "Cannot create a successor for a bitmap that is "
33929bd2b08fSJohn Snow                    "currently frozen");
33939bd2b08fSJohn Snow         return -1;
33949bd2b08fSJohn Snow     }
33959bd2b08fSJohn Snow     assert(!bitmap->successor);
33969bd2b08fSJohn Snow 
33979bd2b08fSJohn Snow     /* Create an anonymous successor */
33989bd2b08fSJohn Snow     granularity = bdrv_dirty_bitmap_granularity(bitmap);
33999bd2b08fSJohn Snow     child = bdrv_create_dirty_bitmap(bs, granularity, NULL, errp);
34009bd2b08fSJohn Snow     if (!child) {
34019bd2b08fSJohn Snow         return -1;
34029bd2b08fSJohn Snow     }
34039bd2b08fSJohn Snow 
34049bd2b08fSJohn Snow     /* Successor will be on or off based on our current state. */
34059bd2b08fSJohn Snow     child->disabled = bitmap->disabled;
34069bd2b08fSJohn Snow 
34079bd2b08fSJohn Snow     /* Install the successor and freeze the parent */
34089bd2b08fSJohn Snow     bitmap->successor = child;
34099bd2b08fSJohn Snow     return 0;
34109bd2b08fSJohn Snow }
34119bd2b08fSJohn Snow 
34129bd2b08fSJohn Snow /**
34139bd2b08fSJohn Snow  * For a bitmap with a successor, yield our name to the successor,
34149bd2b08fSJohn Snow  * delete the old bitmap, and return a handle to the new bitmap.
34159bd2b08fSJohn Snow  */
34169bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_dirty_bitmap_abdicate(BlockDriverState *bs,
34179bd2b08fSJohn Snow                                             BdrvDirtyBitmap *bitmap,
34189bd2b08fSJohn Snow                                             Error **errp)
34199bd2b08fSJohn Snow {
34209bd2b08fSJohn Snow     char *name;
34219bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = bitmap->successor;
34229bd2b08fSJohn Snow 
34239bd2b08fSJohn Snow     if (successor == NULL) {
34249bd2b08fSJohn Snow         error_setg(errp, "Cannot relinquish control if "
34259bd2b08fSJohn Snow                    "there's no successor present");
34269bd2b08fSJohn Snow         return NULL;
34279bd2b08fSJohn Snow     }
34289bd2b08fSJohn Snow 
34299bd2b08fSJohn Snow     name = bitmap->name;
34309bd2b08fSJohn Snow     bitmap->name = NULL;
34319bd2b08fSJohn Snow     successor->name = name;
34329bd2b08fSJohn Snow     bitmap->successor = NULL;
34339bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, bitmap);
34349bd2b08fSJohn Snow 
34359bd2b08fSJohn Snow     return successor;
34369bd2b08fSJohn Snow }
34379bd2b08fSJohn Snow 
34389bd2b08fSJohn Snow /**
34399bd2b08fSJohn Snow  * In cases of failure where we can no longer safely delete the parent,
34409bd2b08fSJohn Snow  * we may wish to re-join the parent and child/successor.
34419bd2b08fSJohn Snow  * The merged parent will be un-frozen, but not explicitly re-enabled.
34429bd2b08fSJohn Snow  */
34439bd2b08fSJohn Snow BdrvDirtyBitmap *bdrv_reclaim_dirty_bitmap(BlockDriverState *bs,
34449bd2b08fSJohn Snow                                            BdrvDirtyBitmap *parent,
34459bd2b08fSJohn Snow                                            Error **errp)
34469bd2b08fSJohn Snow {
34479bd2b08fSJohn Snow     BdrvDirtyBitmap *successor = parent->successor;
34489bd2b08fSJohn Snow 
34499bd2b08fSJohn Snow     if (!successor) {
34509bd2b08fSJohn Snow         error_setg(errp, "Cannot reclaim a successor when none is present");
34519bd2b08fSJohn Snow         return NULL;
34529bd2b08fSJohn Snow     }
34539bd2b08fSJohn Snow 
34549bd2b08fSJohn Snow     if (!hbitmap_merge(parent->bitmap, successor->bitmap)) {
34559bd2b08fSJohn Snow         error_setg(errp, "Merging of parent and successor bitmap failed");
34569bd2b08fSJohn Snow         return NULL;
34579bd2b08fSJohn Snow     }
34589bd2b08fSJohn Snow     bdrv_release_dirty_bitmap(bs, successor);
34599bd2b08fSJohn Snow     parent->successor = NULL;
34609bd2b08fSJohn Snow 
34619bd2b08fSJohn Snow     return parent;
3462b8e6fb75SJohn Snow }
3463b8e6fb75SJohn Snow 
3464ce1ffea8SJohn Snow /**
3465ce1ffea8SJohn Snow  * Truncates _all_ bitmaps attached to a BDS.
3466ce1ffea8SJohn Snow  */
3467ce1ffea8SJohn Snow static void bdrv_dirty_bitmap_truncate(BlockDriverState *bs)
3468ce1ffea8SJohn Snow {
3469ce1ffea8SJohn Snow     BdrvDirtyBitmap *bitmap;
3470ce1ffea8SJohn Snow     uint64_t size = bdrv_nb_sectors(bs);
3471ce1ffea8SJohn Snow 
3472ce1ffea8SJohn Snow     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
347306207b0fSJohn Snow         assert(!bdrv_dirty_bitmap_frozen(bitmap));
3474ce1ffea8SJohn Snow         hbitmap_truncate(bitmap->bitmap, size);
34755270b6a0SJohn Snow         bitmap->size = size;
3476ce1ffea8SJohn Snow     }
3477ce1ffea8SJohn Snow }
3478ce1ffea8SJohn Snow 
3479e4654d2dSFam Zheng void bdrv_release_dirty_bitmap(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
3480e4654d2dSFam Zheng {
3481e4654d2dSFam Zheng     BdrvDirtyBitmap *bm, *next;
3482e4654d2dSFam Zheng     QLIST_FOREACH_SAFE(bm, &bs->dirty_bitmaps, list, next) {
3483e4654d2dSFam Zheng         if (bm == bitmap) {
34849bd2b08fSJohn Snow             assert(!bdrv_dirty_bitmap_frozen(bm));
3485e4654d2dSFam Zheng             QLIST_REMOVE(bitmap, list);
3486e4654d2dSFam Zheng             hbitmap_free(bitmap->bitmap);
34870db6e54aSFam Zheng             g_free(bitmap->name);
3488e4654d2dSFam Zheng             g_free(bitmap);
3489e4654d2dSFam Zheng             return;
34907cd1e32aSlirans@il.ibm.com         }
34917cd1e32aSlirans@il.ibm.com     }
34927cd1e32aSlirans@il.ibm.com }
34937cd1e32aSlirans@il.ibm.com 
3494b8e6fb75SJohn Snow void bdrv_disable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3495b8e6fb75SJohn Snow {
34969bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3497b8e6fb75SJohn Snow     bitmap->disabled = true;
3498b8e6fb75SJohn Snow }
3499b8e6fb75SJohn Snow 
3500b8e6fb75SJohn Snow void bdrv_enable_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3501b8e6fb75SJohn Snow {
35029bd2b08fSJohn Snow     assert(!bdrv_dirty_bitmap_frozen(bitmap));
3503b8e6fb75SJohn Snow     bitmap->disabled = false;
3504b8e6fb75SJohn Snow }
3505b8e6fb75SJohn Snow 
350621b56835SFam Zheng BlockDirtyInfoList *bdrv_query_dirty_bitmaps(BlockDriverState *bs)
350721b56835SFam Zheng {
350821b56835SFam Zheng     BdrvDirtyBitmap *bm;
350921b56835SFam Zheng     BlockDirtyInfoList *list = NULL;
351021b56835SFam Zheng     BlockDirtyInfoList **plist = &list;
351121b56835SFam Zheng 
351221b56835SFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
35135839e53bSMarkus Armbruster         BlockDirtyInfo *info = g_new0(BlockDirtyInfo, 1);
35145839e53bSMarkus Armbruster         BlockDirtyInfoList *entry = g_new0(BlockDirtyInfoList, 1);
351520dca810SJohn Snow         info->count = bdrv_get_dirty_count(bm);
3516592fdd02SJohn Snow         info->granularity = bdrv_dirty_bitmap_granularity(bm);
35170db6e54aSFam Zheng         info->has_name = !!bm->name;
35180db6e54aSFam Zheng         info->name = g_strdup(bm->name);
35199abe3bdcSJohn Snow         info->status = bdrv_dirty_bitmap_status(bm);
352021b56835SFam Zheng         entry->value = info;
352121b56835SFam Zheng         *plist = entry;
352221b56835SFam Zheng         plist = &entry->next;
352321b56835SFam Zheng     }
352421b56835SFam Zheng 
352521b56835SFam Zheng     return list;
352621b56835SFam Zheng }
352721b56835SFam Zheng 
3528e4654d2dSFam Zheng int bdrv_get_dirty(BlockDriverState *bs, BdrvDirtyBitmap *bitmap, int64_t sector)
35297cd1e32aSlirans@il.ibm.com {
3530e4654d2dSFam Zheng     if (bitmap) {
3531e4654d2dSFam Zheng         return hbitmap_get(bitmap->bitmap, sector);
35327cd1e32aSlirans@il.ibm.com     } else {
35337cd1e32aSlirans@il.ibm.com         return 0;
35347cd1e32aSlirans@il.ibm.com     }
35357cd1e32aSlirans@il.ibm.com }
35367cd1e32aSlirans@il.ibm.com 
3537341ebc2fSJohn Snow /**
3538341ebc2fSJohn Snow  * Chooses a default granularity based on the existing cluster size,
3539341ebc2fSJohn Snow  * but clamped between [4K, 64K]. Defaults to 64K in the case that there
3540341ebc2fSJohn Snow  * is no cluster size information available.
3541341ebc2fSJohn Snow  */
3542341ebc2fSJohn Snow uint32_t bdrv_get_default_bitmap_granularity(BlockDriverState *bs)
3543341ebc2fSJohn Snow {
3544341ebc2fSJohn Snow     BlockDriverInfo bdi;
3545341ebc2fSJohn Snow     uint32_t granularity;
3546341ebc2fSJohn Snow 
3547341ebc2fSJohn Snow     if (bdrv_get_info(bs, &bdi) >= 0 && bdi.cluster_size > 0) {
3548341ebc2fSJohn Snow         granularity = MAX(4096, bdi.cluster_size);
3549341ebc2fSJohn Snow         granularity = MIN(65536, granularity);
3550341ebc2fSJohn Snow     } else {
3551341ebc2fSJohn Snow         granularity = 65536;
3552341ebc2fSJohn Snow     }
3553341ebc2fSJohn Snow 
3554341ebc2fSJohn Snow     return granularity;
3555341ebc2fSJohn Snow }
3556341ebc2fSJohn Snow 
3557592fdd02SJohn Snow uint32_t bdrv_dirty_bitmap_granularity(BdrvDirtyBitmap *bitmap)
3558592fdd02SJohn Snow {
3559592fdd02SJohn Snow     return BDRV_SECTOR_SIZE << hbitmap_granularity(bitmap->bitmap);
3560592fdd02SJohn Snow }
3561592fdd02SJohn Snow 
356220dca810SJohn Snow void bdrv_dirty_iter_init(BdrvDirtyBitmap *bitmap, HBitmapIter *hbi)
35631755da16SPaolo Bonzini {
3564e4654d2dSFam Zheng     hbitmap_iter_init(hbi, bitmap->bitmap, 0);
35651755da16SPaolo Bonzini }
35661755da16SPaolo Bonzini 
356720dca810SJohn Snow void bdrv_set_dirty_bitmap(BdrvDirtyBitmap *bitmap,
3568c4237dfaSVladimir Sementsov-Ogievskiy                            int64_t cur_sector, int nr_sectors)
3569c4237dfaSVladimir Sementsov-Ogievskiy {
3570b8e6fb75SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3571c4237dfaSVladimir Sementsov-Ogievskiy     hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3572c4237dfaSVladimir Sementsov-Ogievskiy }
3573c4237dfaSVladimir Sementsov-Ogievskiy 
357420dca810SJohn Snow void bdrv_reset_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_reset(bitmap->bitmap, cur_sector, nr_sectors);
3579c4237dfaSVladimir Sementsov-Ogievskiy }
3580c4237dfaSVladimir Sementsov-Ogievskiy 
3581e74e6b78SJohn Snow void bdrv_clear_dirty_bitmap(BdrvDirtyBitmap *bitmap)
3582e74e6b78SJohn Snow {
3583e74e6b78SJohn Snow     assert(bdrv_dirty_bitmap_enabled(bitmap));
3584c6a8c328SWen Congyang     hbitmap_reset_all(bitmap->bitmap);
3585e74e6b78SJohn Snow }
3586e74e6b78SJohn Snow 
3587e0c47b6cSStefan Hajnoczi void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
35881755da16SPaolo Bonzini                     int nr_sectors)
35891755da16SPaolo Bonzini {
3590e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
3591e4654d2dSFam Zheng     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
3592b8e6fb75SJohn Snow         if (!bdrv_dirty_bitmap_enabled(bitmap)) {
3593b8e6fb75SJohn Snow             continue;
3594b8e6fb75SJohn Snow         }
3595e4654d2dSFam Zheng         hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
3596e4654d2dSFam Zheng     }
35971755da16SPaolo Bonzini }
35981755da16SPaolo Bonzini 
3599d58d8453SJohn Snow /**
3600d58d8453SJohn Snow  * Advance an HBitmapIter to an arbitrary offset.
3601d58d8453SJohn Snow  */
3602d58d8453SJohn Snow void bdrv_set_dirty_iter(HBitmapIter *hbi, int64_t offset)
3603d58d8453SJohn Snow {
3604d58d8453SJohn Snow     assert(hbi->hb);
3605d58d8453SJohn Snow     hbitmap_iter_init(hbi, hbi->hb, offset);
3606d58d8453SJohn Snow }
3607d58d8453SJohn Snow 
360820dca810SJohn Snow int64_t bdrv_get_dirty_count(BdrvDirtyBitmap *bitmap)
3609aaa0eb75SLiran Schour {
3610e4654d2dSFam Zheng     return hbitmap_count(bitmap->bitmap);
3611aaa0eb75SLiran Schour }
3612f88e1a42SJes Sorensen 
36139fcb0251SFam Zheng /* Get a reference to bs */
36149fcb0251SFam Zheng void bdrv_ref(BlockDriverState *bs)
36159fcb0251SFam Zheng {
36169fcb0251SFam Zheng     bs->refcnt++;
36179fcb0251SFam Zheng }
36189fcb0251SFam Zheng 
36199fcb0251SFam Zheng /* Release a previously grabbed reference to bs.
36209fcb0251SFam Zheng  * If after releasing, reference count is zero, the BlockDriverState is
36219fcb0251SFam Zheng  * deleted. */
36229fcb0251SFam Zheng void bdrv_unref(BlockDriverState *bs)
36239fcb0251SFam Zheng {
36249a4d5ca6SJeff Cody     if (!bs) {
36259a4d5ca6SJeff Cody         return;
36269a4d5ca6SJeff Cody     }
36279fcb0251SFam Zheng     assert(bs->refcnt > 0);
36289fcb0251SFam Zheng     if (--bs->refcnt == 0) {
36299fcb0251SFam Zheng         bdrv_delete(bs);
36309fcb0251SFam Zheng     }
36319fcb0251SFam Zheng }
36329fcb0251SFam Zheng 
3633fbe40ff7SFam Zheng struct BdrvOpBlocker {
3634fbe40ff7SFam Zheng     Error *reason;
3635fbe40ff7SFam Zheng     QLIST_ENTRY(BdrvOpBlocker) list;
3636fbe40ff7SFam Zheng };
3637fbe40ff7SFam Zheng 
3638fbe40ff7SFam Zheng bool bdrv_op_is_blocked(BlockDriverState *bs, BlockOpType op, Error **errp)
3639fbe40ff7SFam Zheng {
3640fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3641fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3642fbe40ff7SFam Zheng     if (!QLIST_EMPTY(&bs->op_blockers[op])) {
3643fbe40ff7SFam Zheng         blocker = QLIST_FIRST(&bs->op_blockers[op]);
3644fbe40ff7SFam Zheng         if (errp) {
364581e5f78aSAlberto Garcia             error_setg(errp, "Node '%s' is busy: %s",
364681e5f78aSAlberto Garcia                        bdrv_get_device_or_node_name(bs),
3647bfb197e0SMarkus Armbruster                        error_get_pretty(blocker->reason));
3648fbe40ff7SFam Zheng         }
3649fbe40ff7SFam Zheng         return true;
3650fbe40ff7SFam Zheng     }
3651fbe40ff7SFam Zheng     return false;
3652fbe40ff7SFam Zheng }
3653fbe40ff7SFam Zheng 
3654fbe40ff7SFam Zheng void bdrv_op_block(BlockDriverState *bs, BlockOpType op, Error *reason)
3655fbe40ff7SFam Zheng {
3656fbe40ff7SFam Zheng     BdrvOpBlocker *blocker;
3657fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3658fbe40ff7SFam Zheng 
36595839e53bSMarkus Armbruster     blocker = g_new0(BdrvOpBlocker, 1);
3660fbe40ff7SFam Zheng     blocker->reason = reason;
3661fbe40ff7SFam Zheng     QLIST_INSERT_HEAD(&bs->op_blockers[op], blocker, list);
3662fbe40ff7SFam Zheng }
3663fbe40ff7SFam Zheng 
3664fbe40ff7SFam Zheng void bdrv_op_unblock(BlockDriverState *bs, BlockOpType op, Error *reason)
3665fbe40ff7SFam Zheng {
3666fbe40ff7SFam Zheng     BdrvOpBlocker *blocker, *next;
3667fbe40ff7SFam Zheng     assert((int) op >= 0 && op < BLOCK_OP_TYPE_MAX);
3668fbe40ff7SFam Zheng     QLIST_FOREACH_SAFE(blocker, &bs->op_blockers[op], list, next) {
3669fbe40ff7SFam Zheng         if (blocker->reason == reason) {
3670fbe40ff7SFam Zheng             QLIST_REMOVE(blocker, list);
3671fbe40ff7SFam Zheng             g_free(blocker);
3672fbe40ff7SFam Zheng         }
3673fbe40ff7SFam Zheng     }
3674fbe40ff7SFam Zheng }
3675fbe40ff7SFam Zheng 
3676fbe40ff7SFam Zheng void bdrv_op_block_all(BlockDriverState *bs, Error *reason)
3677fbe40ff7SFam Zheng {
3678fbe40ff7SFam Zheng     int i;
3679fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3680fbe40ff7SFam Zheng         bdrv_op_block(bs, i, reason);
3681fbe40ff7SFam Zheng     }
3682fbe40ff7SFam Zheng }
3683fbe40ff7SFam Zheng 
3684fbe40ff7SFam Zheng void bdrv_op_unblock_all(BlockDriverState *bs, Error *reason)
3685fbe40ff7SFam Zheng {
3686fbe40ff7SFam Zheng     int i;
3687fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3688fbe40ff7SFam Zheng         bdrv_op_unblock(bs, i, reason);
3689fbe40ff7SFam Zheng     }
3690fbe40ff7SFam Zheng }
3691fbe40ff7SFam Zheng 
3692fbe40ff7SFam Zheng bool bdrv_op_blocker_is_empty(BlockDriverState *bs)
3693fbe40ff7SFam Zheng {
3694fbe40ff7SFam Zheng     int i;
3695fbe40ff7SFam Zheng 
3696fbe40ff7SFam Zheng     for (i = 0; i < BLOCK_OP_TYPE_MAX; i++) {
3697fbe40ff7SFam Zheng         if (!QLIST_EMPTY(&bs->op_blockers[i])) {
3698fbe40ff7SFam Zheng             return false;
3699fbe40ff7SFam Zheng         }
3700fbe40ff7SFam Zheng     }
3701fbe40ff7SFam Zheng     return true;
3702fbe40ff7SFam Zheng }
3703fbe40ff7SFam Zheng 
370428a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
370528a7282aSLuiz Capitulino {
3706d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
370758e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
370828a7282aSLuiz Capitulino }
370928a7282aSLuiz Capitulino 
371028a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
371128a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
371228a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
371328a7282aSLuiz Capitulino {
3714d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
371592aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
371692aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
371792aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
371828a7282aSLuiz Capitulino }
371928a7282aSLuiz Capitulino 
372028a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
372128a7282aSLuiz Capitulino {
3722d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
372328a7282aSLuiz Capitulino }
372428a7282aSLuiz Capitulino 
372528a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
372628a7282aSLuiz Capitulino {
372728a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
372858e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
37293bd293c3SPaolo Bonzini         if (bs->job) {
37303bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
37313bd293c3SPaolo Bonzini         }
373228a7282aSLuiz Capitulino     }
373328a7282aSLuiz Capitulino }
373428a7282aSLuiz Capitulino 
373528a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
373628a7282aSLuiz Capitulino {
37373e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
37383e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
373958e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
374058e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
374128a7282aSLuiz Capitulino     }
374228a7282aSLuiz Capitulino }
374328a7282aSLuiz Capitulino 
3744d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
3745f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
3746f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
3747f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
3748f88e1a42SJes Sorensen {
374983d0521aSChunyan Liu     QemuOptsList *create_opts = NULL;
375083d0521aSChunyan Liu     QemuOpts *opts = NULL;
375183d0521aSChunyan Liu     const char *backing_fmt, *backing_file;
375283d0521aSChunyan Liu     int64_t size;
3753f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
3754cc84d90fSMax Reitz     Error *local_err = NULL;
3755f88e1a42SJes Sorensen     int ret = 0;
3756f88e1a42SJes Sorensen 
3757f88e1a42SJes Sorensen     /* Find driver and parse its options */
3758f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
3759f88e1a42SJes Sorensen     if (!drv) {
376071c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
3761d92ada22SLuiz Capitulino         return;
3762f88e1a42SJes Sorensen     }
3763f88e1a42SJes Sorensen 
3764b65a5e12SMax Reitz     proto_drv = bdrv_find_protocol(filename, true, errp);
3765f88e1a42SJes Sorensen     if (!proto_drv) {
3766d92ada22SLuiz Capitulino         return;
3767f88e1a42SJes Sorensen     }
3768f88e1a42SJes Sorensen 
3769c6149724SMax Reitz     if (!drv->create_opts) {
3770c6149724SMax Reitz         error_setg(errp, "Format driver '%s' does not support image creation",
3771c6149724SMax Reitz                    drv->format_name);
3772c6149724SMax Reitz         return;
3773c6149724SMax Reitz     }
3774c6149724SMax Reitz 
3775c6149724SMax Reitz     if (!proto_drv->create_opts) {
3776c6149724SMax Reitz         error_setg(errp, "Protocol driver '%s' does not support image creation",
3777c6149724SMax Reitz                    proto_drv->format_name);
3778c6149724SMax Reitz         return;
3779c6149724SMax Reitz     }
3780c6149724SMax Reitz 
3781c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, drv->create_opts);
3782c282e1fdSChunyan Liu     create_opts = qemu_opts_append(create_opts, proto_drv->create_opts);
3783f88e1a42SJes Sorensen 
3784f88e1a42SJes Sorensen     /* Create parameter list with default values */
378583d0521aSChunyan Liu     opts = qemu_opts_create(create_opts, NULL, 0, &error_abort);
378639101f25SMarkus Armbruster     qemu_opt_set_number(opts, BLOCK_OPT_SIZE, img_size, &error_abort);
3787f88e1a42SJes Sorensen 
3788f88e1a42SJes Sorensen     /* Parse -o options */
3789f88e1a42SJes Sorensen     if (options) {
3790dc523cd3SMarkus Armbruster         qemu_opts_do_parse(opts, options, NULL, &local_err);
3791dc523cd3SMarkus Armbruster         if (local_err) {
3792dc523cd3SMarkus Armbruster             error_report_err(local_err);
3793dc523cd3SMarkus Armbruster             local_err = NULL;
379483d0521aSChunyan Liu             error_setg(errp, "Invalid options for file format '%s'", fmt);
3795f88e1a42SJes Sorensen             goto out;
3796f88e1a42SJes Sorensen         }
3797f88e1a42SJes Sorensen     }
3798f88e1a42SJes Sorensen 
3799f88e1a42SJes Sorensen     if (base_filename) {
3800f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FILE, base_filename, &local_err);
38016be4194bSMarkus Armbruster         if (local_err) {
380271c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
380371c79813SLuiz Capitulino                        fmt);
3804f88e1a42SJes Sorensen             goto out;
3805f88e1a42SJes Sorensen         }
3806f88e1a42SJes Sorensen     }
3807f88e1a42SJes Sorensen 
3808f88e1a42SJes Sorensen     if (base_fmt) {
3809f43e47dbSMarkus Armbruster         qemu_opt_set(opts, BLOCK_OPT_BACKING_FMT, base_fmt, &local_err);
38106be4194bSMarkus Armbruster         if (local_err) {
381171c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
381271c79813SLuiz Capitulino                              "format '%s'", fmt);
3813f88e1a42SJes Sorensen             goto out;
3814f88e1a42SJes Sorensen         }
3815f88e1a42SJes Sorensen     }
3816f88e1a42SJes Sorensen 
381783d0521aSChunyan Liu     backing_file = qemu_opt_get(opts, BLOCK_OPT_BACKING_FILE);
381883d0521aSChunyan Liu     if (backing_file) {
381983d0521aSChunyan Liu         if (!strcmp(filename, backing_file)) {
382071c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
382171c79813SLuiz Capitulino                              "same filename as the backing file");
3822792da93aSJes Sorensen             goto out;
3823792da93aSJes Sorensen         }
3824792da93aSJes Sorensen     }
3825792da93aSJes Sorensen 
382683d0521aSChunyan Liu     backing_fmt = qemu_opt_get(opts, BLOCK_OPT_BACKING_FMT);
3827f88e1a42SJes Sorensen 
3828f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
3829f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
383083d0521aSChunyan Liu     size = qemu_opt_get_size(opts, BLOCK_OPT_SIZE, 0);
383183d0521aSChunyan Liu     if (size == -1) {
383283d0521aSChunyan Liu         if (backing_file) {
383366f6b814SMax Reitz             BlockDriverState *bs;
383429168018SMax Reitz             char *full_backing = g_new0(char, PATH_MAX);
383552bf1e72SMarkus Armbruster             int64_t size;
383663090dacSPaolo Bonzini             int back_flags;
3837e6641719SMax Reitz             QDict *backing_options = NULL;
383863090dacSPaolo Bonzini 
383929168018SMax Reitz             bdrv_get_full_backing_filename_from_filename(filename, backing_file,
384029168018SMax Reitz                                                          full_backing, PATH_MAX,
384129168018SMax Reitz                                                          &local_err);
384229168018SMax Reitz             if (local_err) {
384329168018SMax Reitz                 g_free(full_backing);
384429168018SMax Reitz                 goto out;
384529168018SMax Reitz             }
384629168018SMax Reitz 
384763090dacSPaolo Bonzini             /* backing files always opened read-only */
384863090dacSPaolo Bonzini             back_flags =
384963090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
3850f88e1a42SJes Sorensen 
3851e6641719SMax Reitz             if (backing_fmt) {
3852e6641719SMax Reitz                 backing_options = qdict_new();
3853e6641719SMax Reitz                 qdict_put(backing_options, "driver",
3854e6641719SMax Reitz                           qstring_from_str(backing_fmt));
3855e6641719SMax Reitz             }
3856e6641719SMax Reitz 
3857f67503e5SMax Reitz             bs = NULL;
3858e6641719SMax Reitz             ret = bdrv_open(&bs, full_backing, NULL, backing_options,
38596ebf9aa2SMax Reitz                             back_flags, &local_err);
386029168018SMax Reitz             g_free(full_backing);
3861f88e1a42SJes Sorensen             if (ret < 0) {
3862f88e1a42SJes Sorensen                 goto out;
3863f88e1a42SJes Sorensen             }
386452bf1e72SMarkus Armbruster             size = bdrv_getlength(bs);
386552bf1e72SMarkus Armbruster             if (size < 0) {
386652bf1e72SMarkus Armbruster                 error_setg_errno(errp, -size, "Could not get size of '%s'",
386752bf1e72SMarkus Armbruster                                  backing_file);
386852bf1e72SMarkus Armbruster                 bdrv_unref(bs);
386952bf1e72SMarkus Armbruster                 goto out;
387052bf1e72SMarkus Armbruster             }
3871f88e1a42SJes Sorensen 
387239101f25SMarkus Armbruster             qemu_opt_set_number(opts, BLOCK_OPT_SIZE, size, &error_abort);
387366f6b814SMax Reitz 
387466f6b814SMax Reitz             bdrv_unref(bs);
3875f88e1a42SJes Sorensen         } else {
387671c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
3877f88e1a42SJes Sorensen             goto out;
3878f88e1a42SJes Sorensen         }
3879f88e1a42SJes Sorensen     }
3880f88e1a42SJes Sorensen 
3881f382d43aSMiroslav Rezanina     if (!quiet) {
3882f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
388343c5d8f8SFam Zheng         qemu_opts_print(opts, " ");
3884f88e1a42SJes Sorensen         puts("");
3885f382d43aSMiroslav Rezanina     }
388683d0521aSChunyan Liu 
3887c282e1fdSChunyan Liu     ret = bdrv_create(drv, filename, opts, &local_err);
388883d0521aSChunyan Liu 
3889cc84d90fSMax Reitz     if (ret == -EFBIG) {
3890cc84d90fSMax Reitz         /* This is generally a better message than whatever the driver would
3891cc84d90fSMax Reitz          * deliver (especially because of the cluster_size_hint), since that
3892cc84d90fSMax Reitz          * is most probably not much different from "image too large". */
3893f3f4d2c0SKevin Wolf         const char *cluster_size_hint = "";
389483d0521aSChunyan Liu         if (qemu_opt_get_size(opts, BLOCK_OPT_CLUSTER_SIZE, 0)) {
3895f3f4d2c0SKevin Wolf             cluster_size_hint = " (try using a larger cluster size)";
3896f3f4d2c0SKevin Wolf         }
3897cc84d90fSMax Reitz         error_setg(errp, "The image size is too large for file format '%s'"
3898cc84d90fSMax Reitz                    "%s", fmt, cluster_size_hint);
3899cc84d90fSMax Reitz         error_free(local_err);
3900cc84d90fSMax Reitz         local_err = NULL;
3901f88e1a42SJes Sorensen     }
3902f88e1a42SJes Sorensen 
3903f88e1a42SJes Sorensen out:
390483d0521aSChunyan Liu     qemu_opts_del(opts);
390583d0521aSChunyan Liu     qemu_opts_free(create_opts);
390684d18f06SMarkus Armbruster     if (local_err) {
3907cc84d90fSMax Reitz         error_propagate(errp, local_err);
3908cc84d90fSMax Reitz     }
3909f88e1a42SJes Sorensen }
391085d126f3SStefan Hajnoczi 
391185d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
391285d126f3SStefan Hajnoczi {
3913dcd04228SStefan Hajnoczi     return bs->aio_context;
3914dcd04228SStefan Hajnoczi }
3915dcd04228SStefan Hajnoczi 
3916dcd04228SStefan Hajnoczi void bdrv_detach_aio_context(BlockDriverState *bs)
3917dcd04228SStefan Hajnoczi {
391833384421SMax Reitz     BdrvAioNotifier *baf;
391933384421SMax Reitz 
3920dcd04228SStefan Hajnoczi     if (!bs->drv) {
3921dcd04228SStefan Hajnoczi         return;
3922dcd04228SStefan Hajnoczi     }
3923dcd04228SStefan Hajnoczi 
392433384421SMax Reitz     QLIST_FOREACH(baf, &bs->aio_notifiers, list) {
392533384421SMax Reitz         baf->detach_aio_context(baf->opaque);
392633384421SMax Reitz     }
392733384421SMax Reitz 
392813af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39290e5b0a2dSBenoît Canet         throttle_timers_detach_aio_context(&bs->throttle_timers);
393013af91ebSStefan Hajnoczi     }
3931dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_detach_aio_context) {
3932dcd04228SStefan Hajnoczi         bs->drv->bdrv_detach_aio_context(bs);
3933dcd04228SStefan Hajnoczi     }
3934dcd04228SStefan Hajnoczi     if (bs->file) {
3935dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->file);
3936dcd04228SStefan Hajnoczi     }
3937dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3938dcd04228SStefan Hajnoczi         bdrv_detach_aio_context(bs->backing_hd);
3939dcd04228SStefan Hajnoczi     }
3940dcd04228SStefan Hajnoczi 
3941dcd04228SStefan Hajnoczi     bs->aio_context = NULL;
3942dcd04228SStefan Hajnoczi }
3943dcd04228SStefan Hajnoczi 
3944dcd04228SStefan Hajnoczi void bdrv_attach_aio_context(BlockDriverState *bs,
3945dcd04228SStefan Hajnoczi                              AioContext *new_context)
3946dcd04228SStefan Hajnoczi {
394733384421SMax Reitz     BdrvAioNotifier *ban;
394833384421SMax Reitz 
3949dcd04228SStefan Hajnoczi     if (!bs->drv) {
3950dcd04228SStefan Hajnoczi         return;
3951dcd04228SStefan Hajnoczi     }
3952dcd04228SStefan Hajnoczi 
3953dcd04228SStefan Hajnoczi     bs->aio_context = new_context;
3954dcd04228SStefan Hajnoczi 
3955dcd04228SStefan Hajnoczi     if (bs->backing_hd) {
3956dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->backing_hd, new_context);
3957dcd04228SStefan Hajnoczi     }
3958dcd04228SStefan Hajnoczi     if (bs->file) {
3959dcd04228SStefan Hajnoczi         bdrv_attach_aio_context(bs->file, new_context);
3960dcd04228SStefan Hajnoczi     }
3961dcd04228SStefan Hajnoczi     if (bs->drv->bdrv_attach_aio_context) {
3962dcd04228SStefan Hajnoczi         bs->drv->bdrv_attach_aio_context(bs, new_context);
3963dcd04228SStefan Hajnoczi     }
396413af91ebSStefan Hajnoczi     if (bs->io_limits_enabled) {
39650e5b0a2dSBenoît Canet         throttle_timers_attach_aio_context(&bs->throttle_timers, new_context);
396613af91ebSStefan Hajnoczi     }
396733384421SMax Reitz 
396833384421SMax Reitz     QLIST_FOREACH(ban, &bs->aio_notifiers, list) {
396933384421SMax Reitz         ban->attached_aio_context(new_context, ban->opaque);
397033384421SMax Reitz     }
3971dcd04228SStefan Hajnoczi }
3972dcd04228SStefan Hajnoczi 
3973dcd04228SStefan Hajnoczi void bdrv_set_aio_context(BlockDriverState *bs, AioContext *new_context)
3974dcd04228SStefan Hajnoczi {
397553ec73e2SFam Zheng     bdrv_drain(bs); /* ensure there are no in-flight requests */
3976dcd04228SStefan Hajnoczi 
3977dcd04228SStefan Hajnoczi     bdrv_detach_aio_context(bs);
3978dcd04228SStefan Hajnoczi 
3979dcd04228SStefan Hajnoczi     /* This function executes in the old AioContext so acquire the new one in
3980dcd04228SStefan Hajnoczi      * case it runs in a different thread.
3981dcd04228SStefan Hajnoczi      */
3982dcd04228SStefan Hajnoczi     aio_context_acquire(new_context);
3983dcd04228SStefan Hajnoczi     bdrv_attach_aio_context(bs, new_context);
3984dcd04228SStefan Hajnoczi     aio_context_release(new_context);
398585d126f3SStefan Hajnoczi }
3986d616b224SStefan Hajnoczi 
398733384421SMax Reitz void bdrv_add_aio_context_notifier(BlockDriverState *bs,
398833384421SMax Reitz         void (*attached_aio_context)(AioContext *new_context, void *opaque),
398933384421SMax Reitz         void (*detach_aio_context)(void *opaque), void *opaque)
399033384421SMax Reitz {
399133384421SMax Reitz     BdrvAioNotifier *ban = g_new(BdrvAioNotifier, 1);
399233384421SMax Reitz     *ban = (BdrvAioNotifier){
399333384421SMax Reitz         .attached_aio_context = attached_aio_context,
399433384421SMax Reitz         .detach_aio_context   = detach_aio_context,
399533384421SMax Reitz         .opaque               = opaque
399633384421SMax Reitz     };
399733384421SMax Reitz 
399833384421SMax Reitz     QLIST_INSERT_HEAD(&bs->aio_notifiers, ban, list);
399933384421SMax Reitz }
400033384421SMax Reitz 
400133384421SMax Reitz void bdrv_remove_aio_context_notifier(BlockDriverState *bs,
400233384421SMax Reitz                                       void (*attached_aio_context)(AioContext *,
400333384421SMax Reitz                                                                    void *),
400433384421SMax Reitz                                       void (*detach_aio_context)(void *),
400533384421SMax Reitz                                       void *opaque)
400633384421SMax Reitz {
400733384421SMax Reitz     BdrvAioNotifier *ban, *ban_next;
400833384421SMax Reitz 
400933384421SMax Reitz     QLIST_FOREACH_SAFE(ban, &bs->aio_notifiers, list, ban_next) {
401033384421SMax Reitz         if (ban->attached_aio_context == attached_aio_context &&
401133384421SMax Reitz             ban->detach_aio_context   == detach_aio_context   &&
401233384421SMax Reitz             ban->opaque               == opaque)
401333384421SMax Reitz         {
401433384421SMax Reitz             QLIST_REMOVE(ban, list);
401533384421SMax Reitz             g_free(ban);
401633384421SMax Reitz 
401733384421SMax Reitz             return;
401833384421SMax Reitz         }
401933384421SMax Reitz     }
402033384421SMax Reitz 
402133384421SMax Reitz     abort();
402233384421SMax Reitz }
402333384421SMax Reitz 
402477485434SMax Reitz int bdrv_amend_options(BlockDriverState *bs, QemuOpts *opts,
402577485434SMax Reitz                        BlockDriverAmendStatusCB *status_cb)
40266f176b48SMax Reitz {
4027c282e1fdSChunyan Liu     if (!bs->drv->bdrv_amend_options) {
40286f176b48SMax Reitz         return -ENOTSUP;
40296f176b48SMax Reitz     }
403077485434SMax Reitz     return bs->drv->bdrv_amend_options(bs, opts, status_cb);
40316f176b48SMax Reitz }
4032f6186f49SBenoît Canet 
4033b5042a36SBenoît Canet /* This function will be called by the bdrv_recurse_is_first_non_filter method
4034b5042a36SBenoît Canet  * of block filter and by bdrv_is_first_non_filter.
4035b5042a36SBenoît Canet  * It is used to test if the given bs is the candidate or recurse more in the
4036b5042a36SBenoît Canet  * node graph.
4037212a5a8fSBenoît Canet  */
4038212a5a8fSBenoît Canet bool bdrv_recurse_is_first_non_filter(BlockDriverState *bs,
4039212a5a8fSBenoît Canet                                       BlockDriverState *candidate)
4040f6186f49SBenoît Canet {
4041b5042a36SBenoît Canet     /* return false if basic checks fails */
4042b5042a36SBenoît Canet     if (!bs || !bs->drv) {
4043b5042a36SBenoît Canet         return false;
4044b5042a36SBenoît Canet     }
4045b5042a36SBenoît Canet 
4046b5042a36SBenoît Canet     /* the code reached a non block filter driver -> check if the bs is
4047b5042a36SBenoît Canet      * the same as the candidate. It's the recursion termination condition.
4048b5042a36SBenoît Canet      */
4049b5042a36SBenoît Canet     if (!bs->drv->is_filter) {
4050b5042a36SBenoît Canet         return bs == candidate;
4051b5042a36SBenoît Canet     }
4052b5042a36SBenoît Canet     /* Down this path the driver is a block filter driver */
4053b5042a36SBenoît Canet 
4054b5042a36SBenoît Canet     /* If the block filter recursion method is defined use it to recurse down
4055b5042a36SBenoît Canet      * the node graph.
4056b5042a36SBenoît Canet      */
4057b5042a36SBenoît Canet     if (bs->drv->bdrv_recurse_is_first_non_filter) {
4058212a5a8fSBenoît Canet         return bs->drv->bdrv_recurse_is_first_non_filter(bs, candidate);
4059212a5a8fSBenoît Canet     }
4060212a5a8fSBenoît Canet 
4061b5042a36SBenoît Canet     /* the driver is a block filter but don't allow to recurse -> return false
4062b5042a36SBenoît Canet      */
4063b5042a36SBenoît Canet     return false;
4064212a5a8fSBenoît Canet }
4065212a5a8fSBenoît Canet 
4066212a5a8fSBenoît Canet /* This function checks if the candidate is the first non filter bs down it's
4067212a5a8fSBenoît Canet  * bs chain. Since we don't have pointers to parents it explore all bs chains
4068212a5a8fSBenoît Canet  * from the top. Some filters can choose not to pass down the recursion.
4069212a5a8fSBenoît Canet  */
4070212a5a8fSBenoît Canet bool bdrv_is_first_non_filter(BlockDriverState *candidate)
4071212a5a8fSBenoît Canet {
4072212a5a8fSBenoît Canet     BlockDriverState *bs;
4073212a5a8fSBenoît Canet 
4074212a5a8fSBenoît Canet     /* walk down the bs forest recursively */
4075212a5a8fSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
4076212a5a8fSBenoît Canet         bool perm;
4077212a5a8fSBenoît Canet 
4078b5042a36SBenoît Canet         /* try to recurse in this top level bs */
4079e6dc8a1fSKevin Wolf         perm = bdrv_recurse_is_first_non_filter(bs, candidate);
4080212a5a8fSBenoît Canet 
4081212a5a8fSBenoît Canet         /* candidate is the first non filter */
4082212a5a8fSBenoît Canet         if (perm) {
4083212a5a8fSBenoît Canet             return true;
4084212a5a8fSBenoît Canet         }
4085212a5a8fSBenoît Canet     }
4086212a5a8fSBenoît Canet 
4087212a5a8fSBenoît Canet     return false;
4088f6186f49SBenoît Canet }
408909158f00SBenoît Canet 
4090e12f3784SWen Congyang BlockDriverState *check_to_replace_node(BlockDriverState *parent_bs,
4091e12f3784SWen Congyang                                         const char *node_name, Error **errp)
409209158f00SBenoît Canet {
409309158f00SBenoît Canet     BlockDriverState *to_replace_bs = bdrv_find_node(node_name);
40945a7e7a0bSStefan Hajnoczi     AioContext *aio_context;
40955a7e7a0bSStefan Hajnoczi 
409609158f00SBenoît Canet     if (!to_replace_bs) {
409709158f00SBenoît Canet         error_setg(errp, "Node name '%s' not found", node_name);
409809158f00SBenoît Canet         return NULL;
409909158f00SBenoît Canet     }
410009158f00SBenoît Canet 
41015a7e7a0bSStefan Hajnoczi     aio_context = bdrv_get_aio_context(to_replace_bs);
41025a7e7a0bSStefan Hajnoczi     aio_context_acquire(aio_context);
41035a7e7a0bSStefan Hajnoczi 
410409158f00SBenoît Canet     if (bdrv_op_is_blocked(to_replace_bs, BLOCK_OP_TYPE_REPLACE, errp)) {
41055a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41065a7e7a0bSStefan Hajnoczi         goto out;
410709158f00SBenoît Canet     }
410809158f00SBenoît Canet 
410909158f00SBenoît Canet     /* We don't want arbitrary node of the BDS chain to be replaced only the top
411009158f00SBenoît Canet      * most non filter in order to prevent data corruption.
411109158f00SBenoît Canet      * Another benefit is that this tests exclude backing files which are
411209158f00SBenoît Canet      * blocked by the backing blockers.
411309158f00SBenoît Canet      */
4114e12f3784SWen Congyang     if (!bdrv_recurse_is_first_non_filter(parent_bs, to_replace_bs)) {
411509158f00SBenoît Canet         error_setg(errp, "Only top most non filter can be replaced");
41165a7e7a0bSStefan Hajnoczi         to_replace_bs = NULL;
41175a7e7a0bSStefan Hajnoczi         goto out;
411809158f00SBenoît Canet     }
411909158f00SBenoît Canet 
41205a7e7a0bSStefan Hajnoczi out:
41215a7e7a0bSStefan Hajnoczi     aio_context_release(aio_context);
412209158f00SBenoît Canet     return to_replace_bs;
412309158f00SBenoît Canet }
4124448ad91dSMing Lei 
412591af7014SMax Reitz static bool append_open_options(QDict *d, BlockDriverState *bs)
412691af7014SMax Reitz {
412791af7014SMax Reitz     const QDictEntry *entry;
412891af7014SMax Reitz     bool found_any = false;
412991af7014SMax Reitz 
413091af7014SMax Reitz     for (entry = qdict_first(bs->options); entry;
413191af7014SMax Reitz          entry = qdict_next(bs->options, entry))
413291af7014SMax Reitz     {
413391af7014SMax Reitz         /* Only take options for this level and exclude all non-driver-specific
413491af7014SMax Reitz          * options */
413591af7014SMax Reitz         if (!strchr(qdict_entry_key(entry), '.') &&
413691af7014SMax Reitz             strcmp(qdict_entry_key(entry), "node-name"))
413791af7014SMax Reitz         {
413891af7014SMax Reitz             qobject_incref(qdict_entry_value(entry));
413991af7014SMax Reitz             qdict_put_obj(d, qdict_entry_key(entry), qdict_entry_value(entry));
414091af7014SMax Reitz             found_any = true;
414191af7014SMax Reitz         }
414291af7014SMax Reitz     }
414391af7014SMax Reitz 
414491af7014SMax Reitz     return found_any;
414591af7014SMax Reitz }
414691af7014SMax Reitz 
414791af7014SMax Reitz /* Updates the following BDS fields:
414891af7014SMax Reitz  *  - exact_filename: A filename which may be used for opening a block device
414991af7014SMax Reitz  *                    which (mostly) equals the given BDS (even without any
415091af7014SMax Reitz  *                    other options; so reading and writing must return the same
415191af7014SMax Reitz  *                    results, but caching etc. may be different)
415291af7014SMax Reitz  *  - full_open_options: Options which, when given when opening a block device
415391af7014SMax Reitz  *                       (without a filename), result in a BDS (mostly)
415491af7014SMax Reitz  *                       equalling the given one
415591af7014SMax Reitz  *  - filename: If exact_filename is set, it is copied here. Otherwise,
415691af7014SMax Reitz  *              full_open_options is converted to a JSON object, prefixed with
415791af7014SMax Reitz  *              "json:" (for use through the JSON pseudo protocol) and put here.
415891af7014SMax Reitz  */
415991af7014SMax Reitz void bdrv_refresh_filename(BlockDriverState *bs)
416091af7014SMax Reitz {
416191af7014SMax Reitz     BlockDriver *drv = bs->drv;
416291af7014SMax Reitz     QDict *opts;
416391af7014SMax Reitz 
416491af7014SMax Reitz     if (!drv) {
416591af7014SMax Reitz         return;
416691af7014SMax Reitz     }
416791af7014SMax Reitz 
416891af7014SMax Reitz     /* This BDS's file name will most probably depend on its file's name, so
416991af7014SMax Reitz      * refresh that first */
417091af7014SMax Reitz     if (bs->file) {
417191af7014SMax Reitz         bdrv_refresh_filename(bs->file);
417291af7014SMax Reitz     }
417391af7014SMax Reitz 
417491af7014SMax Reitz     if (drv->bdrv_refresh_filename) {
417591af7014SMax Reitz         /* Obsolete information is of no use here, so drop the old file name
417691af7014SMax Reitz          * information before refreshing it */
417791af7014SMax Reitz         bs->exact_filename[0] = '\0';
417891af7014SMax Reitz         if (bs->full_open_options) {
417991af7014SMax Reitz             QDECREF(bs->full_open_options);
418091af7014SMax Reitz             bs->full_open_options = NULL;
418191af7014SMax Reitz         }
418291af7014SMax Reitz 
418391af7014SMax Reitz         drv->bdrv_refresh_filename(bs);
418491af7014SMax Reitz     } else if (bs->file) {
418591af7014SMax Reitz         /* Try to reconstruct valid information from the underlying file */
418691af7014SMax Reitz         bool has_open_options;
418791af7014SMax Reitz 
418891af7014SMax Reitz         bs->exact_filename[0] = '\0';
418991af7014SMax Reitz         if (bs->full_open_options) {
419091af7014SMax Reitz             QDECREF(bs->full_open_options);
419191af7014SMax Reitz             bs->full_open_options = NULL;
419291af7014SMax Reitz         }
419391af7014SMax Reitz 
419491af7014SMax Reitz         opts = qdict_new();
419591af7014SMax Reitz         has_open_options = append_open_options(opts, bs);
419691af7014SMax Reitz 
419791af7014SMax Reitz         /* If no specific options have been given for this BDS, the filename of
419891af7014SMax Reitz          * the underlying file should suffice for this one as well */
419991af7014SMax Reitz         if (bs->file->exact_filename[0] && !has_open_options) {
420091af7014SMax Reitz             strcpy(bs->exact_filename, bs->file->exact_filename);
420191af7014SMax Reitz         }
420291af7014SMax Reitz         /* Reconstructing the full options QDict is simple for most format block
420391af7014SMax Reitz          * drivers, as long as the full options are known for the underlying
420491af7014SMax Reitz          * file BDS. The full options QDict of that file BDS should somehow
420591af7014SMax Reitz          * contain a representation of the filename, therefore the following
420691af7014SMax Reitz          * suffices without querying the (exact_)filename of this BDS. */
420791af7014SMax Reitz         if (bs->file->full_open_options) {
420891af7014SMax Reitz             qdict_put_obj(opts, "driver",
420991af7014SMax Reitz                           QOBJECT(qstring_from_str(drv->format_name)));
421091af7014SMax Reitz             QINCREF(bs->file->full_open_options);
421191af7014SMax Reitz             qdict_put_obj(opts, "file", QOBJECT(bs->file->full_open_options));
421291af7014SMax Reitz 
421391af7014SMax Reitz             bs->full_open_options = opts;
421491af7014SMax Reitz         } else {
421591af7014SMax Reitz             QDECREF(opts);
421691af7014SMax Reitz         }
421791af7014SMax Reitz     } else if (!bs->full_open_options && qdict_size(bs->options)) {
421891af7014SMax Reitz         /* There is no underlying file BDS (at least referenced by BDS.file),
421991af7014SMax Reitz          * so the full options QDict should be equal to the options given
422091af7014SMax Reitz          * specifically for this block device when it was opened (plus the
422191af7014SMax Reitz          * driver specification).
422291af7014SMax Reitz          * Because those options don't change, there is no need to update
422391af7014SMax Reitz          * full_open_options when it's already set. */
422491af7014SMax Reitz 
422591af7014SMax Reitz         opts = qdict_new();
422691af7014SMax Reitz         append_open_options(opts, bs);
422791af7014SMax Reitz         qdict_put_obj(opts, "driver",
422891af7014SMax Reitz                       QOBJECT(qstring_from_str(drv->format_name)));
422991af7014SMax Reitz 
423091af7014SMax Reitz         if (bs->exact_filename[0]) {
423191af7014SMax Reitz             /* This may not work for all block protocol drivers (some may
423291af7014SMax Reitz              * require this filename to be parsed), but we have to find some
423391af7014SMax Reitz              * default solution here, so just include it. If some block driver
423491af7014SMax Reitz              * does not support pure options without any filename at all or
423591af7014SMax Reitz              * needs some special format of the options QDict, it needs to
423691af7014SMax Reitz              * implement the driver-specific bdrv_refresh_filename() function.
423791af7014SMax Reitz              */
423891af7014SMax Reitz             qdict_put_obj(opts, "filename",
423991af7014SMax Reitz                           QOBJECT(qstring_from_str(bs->exact_filename)));
424091af7014SMax Reitz         }
424191af7014SMax Reitz 
424291af7014SMax Reitz         bs->full_open_options = opts;
424391af7014SMax Reitz     }
424491af7014SMax Reitz 
424591af7014SMax Reitz     if (bs->exact_filename[0]) {
424691af7014SMax Reitz         pstrcpy(bs->filename, sizeof(bs->filename), bs->exact_filename);
424791af7014SMax Reitz     } else if (bs->full_open_options) {
424891af7014SMax Reitz         QString *json = qobject_to_json(QOBJECT(bs->full_open_options));
424991af7014SMax Reitz         snprintf(bs->filename, sizeof(bs->filename), "json:%s",
425091af7014SMax Reitz                  qstring_get_str(json));
425191af7014SMax Reitz         QDECREF(json);
425291af7014SMax Reitz     }
425391af7014SMax Reitz }
42545366d0c8SBenoît Canet 
42555366d0c8SBenoît Canet /* This accessor function purpose is to allow the device models to access the
42565366d0c8SBenoît Canet  * BlockAcctStats structure embedded inside a BlockDriverState without being
42575366d0c8SBenoît Canet  * aware of the BlockDriverState structure layout.
42585366d0c8SBenoît Canet  * It will go away when the BlockAcctStats structure will be moved inside
42595366d0c8SBenoît Canet  * the device models.
42605366d0c8SBenoît Canet  */
42615366d0c8SBenoît Canet BlockAcctStats *bdrv_get_stats(BlockDriverState *bs)
42625366d0c8SBenoît Canet {
42635366d0c8SBenoît Canet     return &bs->stats;
42645366d0c8SBenoît Canet }
4265