xref: /openbmc/qemu/block.c (revision e1b5c52e04d04bb93546c6e37e8884889d047cb1)
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"
2783c9089eSPaolo Bonzini #include "monitor/monitor.h"
28737e150eSPaolo Bonzini #include "block/block_int.h"
29737e150eSPaolo Bonzini #include "block/blockjob.h"
301de7afc9SPaolo Bonzini #include "qemu/module.h"
317b1b5d19SPaolo Bonzini #include "qapi/qmp/qjson.h"
329c17d615SPaolo Bonzini #include "sysemu/sysemu.h"
331de7afc9SPaolo Bonzini #include "qemu/notify.h"
34737e150eSPaolo Bonzini #include "block/coroutine.h"
35b2023818SLuiz Capitulino #include "qmp-commands.h"
361de7afc9SPaolo Bonzini #include "qemu/timer.h"
37fc01f7e7Sbellard 
3871e72a19SJuan Quintela #ifdef CONFIG_BSD
397674e7bfSbellard #include <sys/types.h>
407674e7bfSbellard #include <sys/stat.h>
417674e7bfSbellard #include <sys/ioctl.h>
4272cf2d4fSBlue Swirl #include <sys/queue.h>
43c5e97233Sblueswir1 #ifndef __DragonFly__
447674e7bfSbellard #include <sys/disk.h>
457674e7bfSbellard #endif
46c5e97233Sblueswir1 #endif
477674e7bfSbellard 
4849dc768dSaliguori #ifdef _WIN32
4949dc768dSaliguori #include <windows.h>
5049dc768dSaliguori #endif
5149dc768dSaliguori 
521c9805a3SStefan Hajnoczi #define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
531c9805a3SStefan Hajnoczi 
54470c0504SStefan Hajnoczi typedef enum {
55470c0504SStefan Hajnoczi     BDRV_REQ_COPY_ON_READ = 0x1,
56f08f2ddaSStefan Hajnoczi     BDRV_REQ_ZERO_WRITE   = 0x2,
57470c0504SStefan Hajnoczi } BdrvRequestFlags;
58470c0504SStefan Hajnoczi 
597d4b4ba5SMarkus Armbruster static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load);
60f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
61f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
62c87c0672Saliguori         BlockDriverCompletionFunc *cb, void *opaque);
63f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
64f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
65ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque);
66f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
67f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
68f9f05dc5SKevin Wolf                                          QEMUIOVector *iov);
69f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
70f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
71f9f05dc5SKevin Wolf                                          QEMUIOVector *iov);
72c5fbe571SStefan Hajnoczi static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
73470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
74470c0504SStefan Hajnoczi     BdrvRequestFlags flags);
751c9805a3SStefan Hajnoczi static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
76f08f2ddaSStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
77f08f2ddaSStefan Hajnoczi     BdrvRequestFlags flags);
78b2a61371SStefan Hajnoczi static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
79b2a61371SStefan Hajnoczi                                                int64_t sector_num,
80b2a61371SStefan Hajnoczi                                                QEMUIOVector *qiov,
81b2a61371SStefan Hajnoczi                                                int nb_sectors,
82b2a61371SStefan Hajnoczi                                                BlockDriverCompletionFunc *cb,
83b2a61371SStefan Hajnoczi                                                void *opaque,
848c5873d6SStefan Hajnoczi                                                bool is_write);
85b2a61371SStefan Hajnoczi static void coroutine_fn bdrv_co_do_rw(void *opaque);
86621f0589SKevin Wolf static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
87621f0589SKevin Wolf     int64_t sector_num, int nb_sectors);
88ec530c81Sbellard 
8998f90dbaSZhi Yong Wu static bool bdrv_exceed_bps_limits(BlockDriverState *bs, int nb_sectors,
9098f90dbaSZhi Yong Wu         bool is_write, double elapsed_time, uint64_t *wait);
9198f90dbaSZhi Yong Wu static bool bdrv_exceed_iops_limits(BlockDriverState *bs, bool is_write,
9298f90dbaSZhi Yong Wu         double elapsed_time, uint64_t *wait);
9398f90dbaSZhi Yong Wu static bool bdrv_exceed_io_limits(BlockDriverState *bs, int nb_sectors,
9498f90dbaSZhi Yong Wu         bool is_write, int64_t *wait);
9598f90dbaSZhi Yong Wu 
961b7bdbc1SStefan Hajnoczi static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
971b7bdbc1SStefan Hajnoczi     QTAILQ_HEAD_INITIALIZER(bdrv_states);
987ee930d0Sblueswir1 
998a22f02aSStefan Hajnoczi static QLIST_HEAD(, BlockDriver) bdrv_drivers =
1008a22f02aSStefan Hajnoczi     QLIST_HEAD_INITIALIZER(bdrv_drivers);
101ea2384d3Sbellard 
102eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
103eb852011SMarkus Armbruster static int use_bdrv_whitelist;
104eb852011SMarkus Armbruster 
1059e0b22f4SStefan Hajnoczi #ifdef _WIN32
1069e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
1079e0b22f4SStefan Hajnoczi {
1089e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
1099e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
1109e0b22f4SStefan Hajnoczi             filename[1] == ':');
1119e0b22f4SStefan Hajnoczi }
1129e0b22f4SStefan Hajnoczi 
1139e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
1149e0b22f4SStefan Hajnoczi {
1159e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
1169e0b22f4SStefan Hajnoczi         filename[2] == '\0')
1179e0b22f4SStefan Hajnoczi         return 1;
1189e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
1199e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
1209e0b22f4SStefan Hajnoczi         return 1;
1219e0b22f4SStefan Hajnoczi     return 0;
1229e0b22f4SStefan Hajnoczi }
1239e0b22f4SStefan Hajnoczi #endif
1249e0b22f4SStefan Hajnoczi 
1250563e191SZhi Yong Wu /* throttling disk I/O limits */
12698f90dbaSZhi Yong Wu void bdrv_io_limits_disable(BlockDriverState *bs)
12798f90dbaSZhi Yong Wu {
12898f90dbaSZhi Yong Wu     bs->io_limits_enabled = false;
12998f90dbaSZhi Yong Wu 
130b681a1c7SBenoît Canet     do {} while (qemu_co_enter_next(&bs->throttled_reqs));
13198f90dbaSZhi Yong Wu 
13298f90dbaSZhi Yong Wu     if (bs->block_timer) {
13398f90dbaSZhi Yong Wu         qemu_del_timer(bs->block_timer);
13498f90dbaSZhi Yong Wu         qemu_free_timer(bs->block_timer);
13598f90dbaSZhi Yong Wu         bs->block_timer = NULL;
13698f90dbaSZhi Yong Wu     }
13798f90dbaSZhi Yong Wu 
13898f90dbaSZhi Yong Wu     bs->slice_start = 0;
13998f90dbaSZhi Yong Wu     bs->slice_end   = 0;
14098f90dbaSZhi Yong Wu }
14198f90dbaSZhi Yong Wu 
1420563e191SZhi Yong Wu static void bdrv_block_timer(void *opaque)
1430563e191SZhi Yong Wu {
1440563e191SZhi Yong Wu     BlockDriverState *bs = opaque;
1450563e191SZhi Yong Wu 
146b681a1c7SBenoît Canet     qemu_co_enter_next(&bs->throttled_reqs);
1470563e191SZhi Yong Wu }
1480563e191SZhi Yong Wu 
1490563e191SZhi Yong Wu void bdrv_io_limits_enable(BlockDriverState *bs)
1500563e191SZhi Yong Wu {
1510563e191SZhi Yong Wu     qemu_co_queue_init(&bs->throttled_reqs);
1520563e191SZhi Yong Wu     bs->block_timer = qemu_new_timer_ns(vm_clock, bdrv_block_timer, bs);
1530563e191SZhi Yong Wu     bs->io_limits_enabled = true;
1540563e191SZhi Yong Wu }
1550563e191SZhi Yong Wu 
1560563e191SZhi Yong Wu bool bdrv_io_limits_enabled(BlockDriverState *bs)
1570563e191SZhi Yong Wu {
1580563e191SZhi Yong Wu     BlockIOLimit *io_limits = &bs->io_limits;
1590563e191SZhi Yong Wu     return io_limits->bps[BLOCK_IO_LIMIT_READ]
1600563e191SZhi Yong Wu          || io_limits->bps[BLOCK_IO_LIMIT_WRITE]
1610563e191SZhi Yong Wu          || io_limits->bps[BLOCK_IO_LIMIT_TOTAL]
1620563e191SZhi Yong Wu          || io_limits->iops[BLOCK_IO_LIMIT_READ]
1630563e191SZhi Yong Wu          || io_limits->iops[BLOCK_IO_LIMIT_WRITE]
1640563e191SZhi Yong Wu          || io_limits->iops[BLOCK_IO_LIMIT_TOTAL];
1650563e191SZhi Yong Wu }
1660563e191SZhi Yong Wu 
16798f90dbaSZhi Yong Wu static void bdrv_io_limits_intercept(BlockDriverState *bs,
16898f90dbaSZhi Yong Wu                                      bool is_write, int nb_sectors)
16998f90dbaSZhi Yong Wu {
17098f90dbaSZhi Yong Wu     int64_t wait_time = -1;
17198f90dbaSZhi Yong Wu 
17298f90dbaSZhi Yong Wu     if (!qemu_co_queue_empty(&bs->throttled_reqs)) {
17398f90dbaSZhi Yong Wu         qemu_co_queue_wait(&bs->throttled_reqs);
17498f90dbaSZhi Yong Wu     }
17598f90dbaSZhi Yong Wu 
17698f90dbaSZhi Yong Wu     /* In fact, we hope to keep each request's timing, in FIFO mode. The next
17798f90dbaSZhi Yong Wu      * throttled requests will not be dequeued until the current request is
17898f90dbaSZhi Yong Wu      * allowed to be serviced. So if the current request still exceeds the
17998f90dbaSZhi Yong Wu      * limits, it will be inserted to the head. All requests followed it will
18098f90dbaSZhi Yong Wu      * be still in throttled_reqs queue.
18198f90dbaSZhi Yong Wu      */
18298f90dbaSZhi Yong Wu 
18398f90dbaSZhi Yong Wu     while (bdrv_exceed_io_limits(bs, nb_sectors, is_write, &wait_time)) {
18498f90dbaSZhi Yong Wu         qemu_mod_timer(bs->block_timer,
18598f90dbaSZhi Yong Wu                        wait_time + qemu_get_clock_ns(vm_clock));
18698f90dbaSZhi Yong Wu         qemu_co_queue_wait_insert_head(&bs->throttled_reqs);
18798f90dbaSZhi Yong Wu     }
18898f90dbaSZhi Yong Wu 
18998f90dbaSZhi Yong Wu     qemu_co_queue_next(&bs->throttled_reqs);
19098f90dbaSZhi Yong Wu }
19198f90dbaSZhi Yong Wu 
1929e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
1939e0b22f4SStefan Hajnoczi static int path_has_protocol(const char *path)
1949e0b22f4SStefan Hajnoczi {
195947995c0SPaolo Bonzini     const char *p;
196947995c0SPaolo Bonzini 
1979e0b22f4SStefan Hajnoczi #ifdef _WIN32
1989e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
1999e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
2009e0b22f4SStefan Hajnoczi         return 0;
2019e0b22f4SStefan Hajnoczi     }
202947995c0SPaolo Bonzini     p = path + strcspn(path, ":/\\");
203947995c0SPaolo Bonzini #else
204947995c0SPaolo Bonzini     p = path + strcspn(path, ":/");
2059e0b22f4SStefan Hajnoczi #endif
2069e0b22f4SStefan Hajnoczi 
207947995c0SPaolo Bonzini     return *p == ':';
2089e0b22f4SStefan Hajnoczi }
2099e0b22f4SStefan Hajnoczi 
21083f64091Sbellard int path_is_absolute(const char *path)
21183f64091Sbellard {
21221664424Sbellard #ifdef _WIN32
21321664424Sbellard     /* specific case for names like: "\\.\d:" */
214f53f4da9SPaolo Bonzini     if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
21521664424Sbellard         return 1;
216f53f4da9SPaolo Bonzini     }
217f53f4da9SPaolo Bonzini     return (*path == '/' || *path == '\\');
2183b9f94e1Sbellard #else
219f53f4da9SPaolo Bonzini     return (*path == '/');
2203b9f94e1Sbellard #endif
22183f64091Sbellard }
22283f64091Sbellard 
22383f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
22483f64091Sbellard    path to it by considering it is relative to base_path. URL are
22583f64091Sbellard    supported. */
22683f64091Sbellard void path_combine(char *dest, int dest_size,
22783f64091Sbellard                   const char *base_path,
22883f64091Sbellard                   const char *filename)
22983f64091Sbellard {
23083f64091Sbellard     const char *p, *p1;
23183f64091Sbellard     int len;
23283f64091Sbellard 
23383f64091Sbellard     if (dest_size <= 0)
23483f64091Sbellard         return;
23583f64091Sbellard     if (path_is_absolute(filename)) {
23683f64091Sbellard         pstrcpy(dest, dest_size, filename);
23783f64091Sbellard     } else {
23883f64091Sbellard         p = strchr(base_path, ':');
23983f64091Sbellard         if (p)
24083f64091Sbellard             p++;
24183f64091Sbellard         else
24283f64091Sbellard             p = base_path;
2433b9f94e1Sbellard         p1 = strrchr(base_path, '/');
2443b9f94e1Sbellard #ifdef _WIN32
2453b9f94e1Sbellard         {
2463b9f94e1Sbellard             const char *p2;
2473b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
2483b9f94e1Sbellard             if (!p1 || p2 > p1)
2493b9f94e1Sbellard                 p1 = p2;
2503b9f94e1Sbellard         }
2513b9f94e1Sbellard #endif
25283f64091Sbellard         if (p1)
25383f64091Sbellard             p1++;
25483f64091Sbellard         else
25583f64091Sbellard             p1 = base_path;
25683f64091Sbellard         if (p1 > p)
25783f64091Sbellard             p = p1;
25883f64091Sbellard         len = p - base_path;
25983f64091Sbellard         if (len > dest_size - 1)
26083f64091Sbellard             len = dest_size - 1;
26183f64091Sbellard         memcpy(dest, base_path, len);
26283f64091Sbellard         dest[len] = '\0';
26383f64091Sbellard         pstrcat(dest, dest_size, filename);
26483f64091Sbellard     }
26583f64091Sbellard }
26683f64091Sbellard 
267dc5a1371SPaolo Bonzini void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz)
268dc5a1371SPaolo Bonzini {
269dc5a1371SPaolo Bonzini     if (bs->backing_file[0] == '\0' || path_has_protocol(bs->backing_file)) {
270dc5a1371SPaolo Bonzini         pstrcpy(dest, sz, bs->backing_file);
271dc5a1371SPaolo Bonzini     } else {
272dc5a1371SPaolo Bonzini         path_combine(dest, sz, bs->filename, bs->backing_file);
273dc5a1371SPaolo Bonzini     }
274dc5a1371SPaolo Bonzini }
275dc5a1371SPaolo Bonzini 
2765efa9d5aSAnthony Liguori void bdrv_register(BlockDriver *bdrv)
277ea2384d3Sbellard {
2788c5873d6SStefan Hajnoczi     /* Block drivers without coroutine functions need emulation */
2798c5873d6SStefan Hajnoczi     if (!bdrv->bdrv_co_readv) {
280f9f05dc5SKevin Wolf         bdrv->bdrv_co_readv = bdrv_co_readv_em;
281f9f05dc5SKevin Wolf         bdrv->bdrv_co_writev = bdrv_co_writev_em;
282f9f05dc5SKevin Wolf 
283f8c35c1dSStefan Hajnoczi         /* bdrv_co_readv_em()/brdv_co_writev_em() work in terms of aio, so if
284f8c35c1dSStefan Hajnoczi          * the block driver lacks aio we need to emulate that too.
285f8c35c1dSStefan Hajnoczi          */
286f9f05dc5SKevin Wolf         if (!bdrv->bdrv_aio_readv) {
28783f64091Sbellard             /* add AIO emulation layer */
288f141eafeSaliguori             bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
289f141eafeSaliguori             bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
29083f64091Sbellard         }
291f9f05dc5SKevin Wolf     }
292b2e12bc6SChristoph Hellwig 
2938a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
294ea2384d3Sbellard }
295b338082bSbellard 
296b338082bSbellard /* create a new block device (by default it is empty) */
297b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
298fc01f7e7Sbellard {
2991b7bdbc1SStefan Hajnoczi     BlockDriverState *bs;
300b338082bSbellard 
3017267c094SAnthony Liguori     bs = g_malloc0(sizeof(BlockDriverState));
302b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
303ea2384d3Sbellard     if (device_name[0] != '\0') {
3041b7bdbc1SStefan Hajnoczi         QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
305ea2384d3Sbellard     }
30628a7282aSLuiz Capitulino     bdrv_iostatus_disable(bs);
307d7d512f6SPaolo Bonzini     notifier_list_init(&bs->close_notifiers);
308d616b224SStefan Hajnoczi     notifier_with_return_list_init(&bs->before_write_notifiers);
309d7d512f6SPaolo Bonzini 
310b338082bSbellard     return bs;
311b338082bSbellard }
312b338082bSbellard 
313d7d512f6SPaolo Bonzini void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
314d7d512f6SPaolo Bonzini {
315d7d512f6SPaolo Bonzini     notifier_list_add(&bs->close_notifiers, notify);
316d7d512f6SPaolo Bonzini }
317d7d512f6SPaolo Bonzini 
318ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
319ea2384d3Sbellard {
320ea2384d3Sbellard     BlockDriver *drv1;
3218a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
3228a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
323ea2384d3Sbellard             return drv1;
324ea2384d3Sbellard         }
3258a22f02aSStefan Hajnoczi     }
326ea2384d3Sbellard     return NULL;
327ea2384d3Sbellard }
328ea2384d3Sbellard 
329b64ec4e4SFam Zheng static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
330eb852011SMarkus Armbruster {
331b64ec4e4SFam Zheng     static const char *whitelist_rw[] = {
332b64ec4e4SFam Zheng         CONFIG_BDRV_RW_WHITELIST
333b64ec4e4SFam Zheng     };
334b64ec4e4SFam Zheng     static const char *whitelist_ro[] = {
335b64ec4e4SFam Zheng         CONFIG_BDRV_RO_WHITELIST
336eb852011SMarkus Armbruster     };
337eb852011SMarkus Armbruster     const char **p;
338eb852011SMarkus Armbruster 
339b64ec4e4SFam Zheng     if (!whitelist_rw[0] && !whitelist_ro[0]) {
340eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
341b64ec4e4SFam Zheng     }
342eb852011SMarkus Armbruster 
343b64ec4e4SFam Zheng     for (p = whitelist_rw; *p; p++) {
344eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
345eb852011SMarkus Armbruster             return 1;
346eb852011SMarkus Armbruster         }
347eb852011SMarkus Armbruster     }
348b64ec4e4SFam Zheng     if (read_only) {
349b64ec4e4SFam Zheng         for (p = whitelist_ro; *p; p++) {
350b64ec4e4SFam Zheng             if (!strcmp(drv->format_name, *p)) {
351b64ec4e4SFam Zheng                 return 1;
352b64ec4e4SFam Zheng             }
353b64ec4e4SFam Zheng         }
354b64ec4e4SFam Zheng     }
355eb852011SMarkus Armbruster     return 0;
356eb852011SMarkus Armbruster }
357eb852011SMarkus Armbruster 
358b64ec4e4SFam Zheng BlockDriver *bdrv_find_whitelisted_format(const char *format_name,
359b64ec4e4SFam Zheng                                           bool read_only)
360eb852011SMarkus Armbruster {
361eb852011SMarkus Armbruster     BlockDriver *drv = bdrv_find_format(format_name);
362b64ec4e4SFam Zheng     return drv && bdrv_is_whitelisted(drv, read_only) ? drv : NULL;
363eb852011SMarkus Armbruster }
364eb852011SMarkus Armbruster 
3655b7e1542SZhi Yong Wu typedef struct CreateCo {
3665b7e1542SZhi Yong Wu     BlockDriver *drv;
3675b7e1542SZhi Yong Wu     char *filename;
3685b7e1542SZhi Yong Wu     QEMUOptionParameter *options;
3695b7e1542SZhi Yong Wu     int ret;
3705b7e1542SZhi Yong Wu } CreateCo;
3715b7e1542SZhi Yong Wu 
3725b7e1542SZhi Yong Wu static void coroutine_fn bdrv_create_co_entry(void *opaque)
3735b7e1542SZhi Yong Wu {
3745b7e1542SZhi Yong Wu     CreateCo *cco = opaque;
3755b7e1542SZhi Yong Wu     assert(cco->drv);
3765b7e1542SZhi Yong Wu 
3775b7e1542SZhi Yong Wu     cco->ret = cco->drv->bdrv_create(cco->filename, cco->options);
3785b7e1542SZhi Yong Wu }
3795b7e1542SZhi Yong Wu 
3800e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
3810e7e1989SKevin Wolf     QEMUOptionParameter *options)
382ea2384d3Sbellard {
3835b7e1542SZhi Yong Wu     int ret;
3840e7e1989SKevin Wolf 
3855b7e1542SZhi Yong Wu     Coroutine *co;
3865b7e1542SZhi Yong Wu     CreateCo cco = {
3875b7e1542SZhi Yong Wu         .drv = drv,
3885b7e1542SZhi Yong Wu         .filename = g_strdup(filename),
3895b7e1542SZhi Yong Wu         .options = options,
3905b7e1542SZhi Yong Wu         .ret = NOT_DONE,
3915b7e1542SZhi Yong Wu     };
3925b7e1542SZhi Yong Wu 
3935b7e1542SZhi Yong Wu     if (!drv->bdrv_create) {
39480168bffSLuiz Capitulino         ret = -ENOTSUP;
39580168bffSLuiz Capitulino         goto out;
3965b7e1542SZhi Yong Wu     }
3975b7e1542SZhi Yong Wu 
3985b7e1542SZhi Yong Wu     if (qemu_in_coroutine()) {
3995b7e1542SZhi Yong Wu         /* Fast-path if already in coroutine context */
4005b7e1542SZhi Yong Wu         bdrv_create_co_entry(&cco);
4015b7e1542SZhi Yong Wu     } else {
4025b7e1542SZhi Yong Wu         co = qemu_coroutine_create(bdrv_create_co_entry);
4035b7e1542SZhi Yong Wu         qemu_coroutine_enter(co, &cco);
4045b7e1542SZhi Yong Wu         while (cco.ret == NOT_DONE) {
4055b7e1542SZhi Yong Wu             qemu_aio_wait();
4065b7e1542SZhi Yong Wu         }
4075b7e1542SZhi Yong Wu     }
4085b7e1542SZhi Yong Wu 
4095b7e1542SZhi Yong Wu     ret = cco.ret;
4105b7e1542SZhi Yong Wu 
41180168bffSLuiz Capitulino out:
41280168bffSLuiz Capitulino     g_free(cco.filename);
4135b7e1542SZhi Yong Wu     return ret;
414ea2384d3Sbellard }
415ea2384d3Sbellard 
41684a12e66SChristoph Hellwig int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
41784a12e66SChristoph Hellwig {
41884a12e66SChristoph Hellwig     BlockDriver *drv;
41984a12e66SChristoph Hellwig 
42098289620SKevin Wolf     drv = bdrv_find_protocol(filename, true);
42184a12e66SChristoph Hellwig     if (drv == NULL) {
42216905d71SStefan Hajnoczi         return -ENOENT;
42384a12e66SChristoph Hellwig     }
42484a12e66SChristoph Hellwig 
42584a12e66SChristoph Hellwig     return bdrv_create(drv, filename, options);
42684a12e66SChristoph Hellwig }
42784a12e66SChristoph Hellwig 
428eba25057SJim Meyering /*
429eba25057SJim Meyering  * Create a uniquely-named empty temporary file.
430eba25057SJim Meyering  * Return 0 upon success, otherwise a negative errno value.
431eba25057SJim Meyering  */
432eba25057SJim Meyering int get_tmp_filename(char *filename, int size)
433eba25057SJim Meyering {
434d5249393Sbellard #ifdef _WIN32
4353b9f94e1Sbellard     char temp_dir[MAX_PATH];
436eba25057SJim Meyering     /* GetTempFileName requires that its output buffer (4th param)
437eba25057SJim Meyering        have length MAX_PATH or greater.  */
438eba25057SJim Meyering     assert(size >= MAX_PATH);
439eba25057SJim Meyering     return (GetTempPath(MAX_PATH, temp_dir)
440eba25057SJim Meyering             && GetTempFileName(temp_dir, "qem", 0, filename)
441eba25057SJim Meyering             ? 0 : -GetLastError());
442d5249393Sbellard #else
443ea2384d3Sbellard     int fd;
4447ccfb2ebSblueswir1     const char *tmpdir;
4450badc1eeSaurel32     tmpdir = getenv("TMPDIR");
4460badc1eeSaurel32     if (!tmpdir)
4470badc1eeSaurel32         tmpdir = "/tmp";
448eba25057SJim Meyering     if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
449eba25057SJim Meyering         return -EOVERFLOW;
450ea2384d3Sbellard     }
451eba25057SJim Meyering     fd = mkstemp(filename);
452fe235a06SDunrong Huang     if (fd < 0) {
453fe235a06SDunrong Huang         return -errno;
454fe235a06SDunrong Huang     }
455fe235a06SDunrong Huang     if (close(fd) != 0) {
456fe235a06SDunrong Huang         unlink(filename);
457eba25057SJim Meyering         return -errno;
458eba25057SJim Meyering     }
459eba25057SJim Meyering     return 0;
460d5249393Sbellard #endif
461eba25057SJim Meyering }
462ea2384d3Sbellard 
463f3a5d3f8SChristoph Hellwig /*
464f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
465f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
466f3a5d3f8SChristoph Hellwig  */
467f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
468f3a5d3f8SChristoph Hellwig {
469508c7cb3SChristoph Hellwig     int score_max = 0, score;
470508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
471f3a5d3f8SChristoph Hellwig 
4728a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
473508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
474508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
475508c7cb3SChristoph Hellwig             if (score > score_max) {
476508c7cb3SChristoph Hellwig                 score_max = score;
477508c7cb3SChristoph Hellwig                 drv = d;
478f3a5d3f8SChristoph Hellwig             }
479508c7cb3SChristoph Hellwig         }
480f3a5d3f8SChristoph Hellwig     }
481f3a5d3f8SChristoph Hellwig 
482508c7cb3SChristoph Hellwig     return drv;
483f3a5d3f8SChristoph Hellwig }
484f3a5d3f8SChristoph Hellwig 
48598289620SKevin Wolf BlockDriver *bdrv_find_protocol(const char *filename,
48698289620SKevin Wolf                                 bool allow_protocol_prefix)
48784a12e66SChristoph Hellwig {
48884a12e66SChristoph Hellwig     BlockDriver *drv1;
48984a12e66SChristoph Hellwig     char protocol[128];
49084a12e66SChristoph Hellwig     int len;
49184a12e66SChristoph Hellwig     const char *p;
49284a12e66SChristoph Hellwig 
49366f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
49466f82ceeSKevin Wolf 
49539508e7aSChristoph Hellwig     /*
49639508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
49739508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
49839508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
49939508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
50039508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
50139508e7aSChristoph Hellwig      */
50284a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
50339508e7aSChristoph Hellwig     if (drv1) {
50484a12e66SChristoph Hellwig         return drv1;
50584a12e66SChristoph Hellwig     }
50639508e7aSChristoph Hellwig 
50798289620SKevin Wolf     if (!path_has_protocol(filename) || !allow_protocol_prefix) {
50839508e7aSChristoph Hellwig         return bdrv_find_format("file");
50939508e7aSChristoph Hellwig     }
51098289620SKevin Wolf 
5119e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
5129e0b22f4SStefan Hajnoczi     assert(p != NULL);
51384a12e66SChristoph Hellwig     len = p - filename;
51484a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
51584a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
51684a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
51784a12e66SChristoph Hellwig     protocol[len] = '\0';
51884a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
51984a12e66SChristoph Hellwig         if (drv1->protocol_name &&
52084a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
52184a12e66SChristoph Hellwig             return drv1;
52284a12e66SChristoph Hellwig         }
52384a12e66SChristoph Hellwig     }
52484a12e66SChristoph Hellwig     return NULL;
52584a12e66SChristoph Hellwig }
52684a12e66SChristoph Hellwig 
527f500a6d3SKevin Wolf static int find_image_format(BlockDriverState *bs, const char *filename,
528f500a6d3SKevin Wolf                              BlockDriver **pdrv)
529ea2384d3Sbellard {
530f500a6d3SKevin Wolf     int score, score_max;
531ea2384d3Sbellard     BlockDriver *drv1, *drv;
53283f64091Sbellard     uint8_t buf[2048];
533f500a6d3SKevin Wolf     int ret = 0;
534f8ea0b00SNicholas Bellinger 
53508a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
5368e895599SPaolo Bonzini     if (bs->sg || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
537c98ac35dSStefan Weil         drv = bdrv_find_format("raw");
538c98ac35dSStefan Weil         if (!drv) {
539c98ac35dSStefan Weil             ret = -ENOENT;
540c98ac35dSStefan Weil         }
541c98ac35dSStefan Weil         *pdrv = drv;
542c98ac35dSStefan Weil         return ret;
5431a396859SNicholas A. Bellinger     }
544f8ea0b00SNicholas Bellinger 
54583f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
546ea2384d3Sbellard     if (ret < 0) {
547c98ac35dSStefan Weil         *pdrv = NULL;
548c98ac35dSStefan Weil         return ret;
549ea2384d3Sbellard     }
550ea2384d3Sbellard 
551ea2384d3Sbellard     score_max = 0;
55284a12e66SChristoph Hellwig     drv = NULL;
5538a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
55483f64091Sbellard         if (drv1->bdrv_probe) {
555ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
556ea2384d3Sbellard             if (score > score_max) {
557ea2384d3Sbellard                 score_max = score;
558ea2384d3Sbellard                 drv = drv1;
559ea2384d3Sbellard             }
560ea2384d3Sbellard         }
56183f64091Sbellard     }
562c98ac35dSStefan Weil     if (!drv) {
563c98ac35dSStefan Weil         ret = -ENOENT;
564c98ac35dSStefan Weil     }
565c98ac35dSStefan Weil     *pdrv = drv;
566c98ac35dSStefan Weil     return ret;
567ea2384d3Sbellard }
568ea2384d3Sbellard 
56951762288SStefan Hajnoczi /**
57051762288SStefan Hajnoczi  * Set the current 'total_sectors' value
57151762288SStefan Hajnoczi  */
57251762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
57351762288SStefan Hajnoczi {
57451762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
57551762288SStefan Hajnoczi 
576396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
577396759adSNicholas Bellinger     if (bs->sg)
578396759adSNicholas Bellinger         return 0;
579396759adSNicholas Bellinger 
58051762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
58151762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
58251762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
58351762288SStefan Hajnoczi         if (length < 0) {
58451762288SStefan Hajnoczi             return length;
58551762288SStefan Hajnoczi         }
58651762288SStefan Hajnoczi         hint = length >> BDRV_SECTOR_BITS;
58751762288SStefan Hajnoczi     }
58851762288SStefan Hajnoczi 
58951762288SStefan Hajnoczi     bs->total_sectors = hint;
59051762288SStefan Hajnoczi     return 0;
59151762288SStefan Hajnoczi }
59251762288SStefan Hajnoczi 
593c3993cdcSStefan Hajnoczi /**
5949e8f1835SPaolo Bonzini  * Set open flags for a given discard mode
5959e8f1835SPaolo Bonzini  *
5969e8f1835SPaolo Bonzini  * Return 0 on success, -1 if the discard mode was invalid.
5979e8f1835SPaolo Bonzini  */
5989e8f1835SPaolo Bonzini int bdrv_parse_discard_flags(const char *mode, int *flags)
5999e8f1835SPaolo Bonzini {
6009e8f1835SPaolo Bonzini     *flags &= ~BDRV_O_UNMAP;
6019e8f1835SPaolo Bonzini 
6029e8f1835SPaolo Bonzini     if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
6039e8f1835SPaolo Bonzini         /* do nothing */
6049e8f1835SPaolo Bonzini     } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
6059e8f1835SPaolo Bonzini         *flags |= BDRV_O_UNMAP;
6069e8f1835SPaolo Bonzini     } else {
6079e8f1835SPaolo Bonzini         return -1;
6089e8f1835SPaolo Bonzini     }
6099e8f1835SPaolo Bonzini 
6109e8f1835SPaolo Bonzini     return 0;
6119e8f1835SPaolo Bonzini }
6129e8f1835SPaolo Bonzini 
6139e8f1835SPaolo Bonzini /**
614c3993cdcSStefan Hajnoczi  * Set open flags for a given cache mode
615c3993cdcSStefan Hajnoczi  *
616c3993cdcSStefan Hajnoczi  * Return 0 on success, -1 if the cache mode was invalid.
617c3993cdcSStefan Hajnoczi  */
618c3993cdcSStefan Hajnoczi int bdrv_parse_cache_flags(const char *mode, int *flags)
619c3993cdcSStefan Hajnoczi {
620c3993cdcSStefan Hajnoczi     *flags &= ~BDRV_O_CACHE_MASK;
621c3993cdcSStefan Hajnoczi 
622c3993cdcSStefan Hajnoczi     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
623c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
62492196b2fSStefan Hajnoczi     } else if (!strcmp(mode, "directsync")) {
62592196b2fSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE;
626c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writeback")) {
627c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
628c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "unsafe")) {
629c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
630c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NO_FLUSH;
631c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writethrough")) {
632c3993cdcSStefan Hajnoczi         /* this is the default */
633c3993cdcSStefan Hajnoczi     } else {
634c3993cdcSStefan Hajnoczi         return -1;
635c3993cdcSStefan Hajnoczi     }
636c3993cdcSStefan Hajnoczi 
637c3993cdcSStefan Hajnoczi     return 0;
638c3993cdcSStefan Hajnoczi }
639c3993cdcSStefan Hajnoczi 
64053fec9d3SStefan Hajnoczi /**
64153fec9d3SStefan Hajnoczi  * The copy-on-read flag is actually a reference count so multiple users may
64253fec9d3SStefan Hajnoczi  * use the feature without worrying about clobbering its previous state.
64353fec9d3SStefan Hajnoczi  * Copy-on-read stays enabled until all users have called to disable it.
64453fec9d3SStefan Hajnoczi  */
64553fec9d3SStefan Hajnoczi void bdrv_enable_copy_on_read(BlockDriverState *bs)
64653fec9d3SStefan Hajnoczi {
64753fec9d3SStefan Hajnoczi     bs->copy_on_read++;
64853fec9d3SStefan Hajnoczi }
64953fec9d3SStefan Hajnoczi 
65053fec9d3SStefan Hajnoczi void bdrv_disable_copy_on_read(BlockDriverState *bs)
65153fec9d3SStefan Hajnoczi {
65253fec9d3SStefan Hajnoczi     assert(bs->copy_on_read > 0);
65353fec9d3SStefan Hajnoczi     bs->copy_on_read--;
65453fec9d3SStefan Hajnoczi }
65553fec9d3SStefan Hajnoczi 
6567b272452SKevin Wolf static int bdrv_open_flags(BlockDriverState *bs, int flags)
6577b272452SKevin Wolf {
6587b272452SKevin Wolf     int open_flags = flags | BDRV_O_CACHE_WB;
6597b272452SKevin Wolf 
6607b272452SKevin Wolf     /*
6617b272452SKevin Wolf      * Clear flags that are internal to the block layer before opening the
6627b272452SKevin Wolf      * image.
6637b272452SKevin Wolf      */
6647b272452SKevin Wolf     open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
6657b272452SKevin Wolf 
6667b272452SKevin Wolf     /*
6677b272452SKevin Wolf      * Snapshots should be writable.
6687b272452SKevin Wolf      */
6697b272452SKevin Wolf     if (bs->is_temporary) {
6707b272452SKevin Wolf         open_flags |= BDRV_O_RDWR;
6717b272452SKevin Wolf     }
6727b272452SKevin Wolf 
6737b272452SKevin Wolf     return open_flags;
6747b272452SKevin Wolf }
6757b272452SKevin Wolf 
676b6ce07aaSKevin Wolf /*
67757915332SKevin Wolf  * Common part for opening disk images and files
678b6ad491aSKevin Wolf  *
679b6ad491aSKevin Wolf  * Removes all processed options from *options.
68057915332SKevin Wolf  */
681f500a6d3SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
682035fccdfSKevin Wolf     QDict *options, int flags, BlockDriver *drv)
68357915332SKevin Wolf {
68457915332SKevin Wolf     int ret, open_flags;
685035fccdfSKevin Wolf     const char *filename;
68657915332SKevin Wolf 
68757915332SKevin Wolf     assert(drv != NULL);
6886405875cSPaolo Bonzini     assert(bs->file == NULL);
689707ff828SKevin Wolf     assert(options != NULL && bs->options != options);
69057915332SKevin Wolf 
69145673671SKevin Wolf     if (file != NULL) {
69245673671SKevin Wolf         filename = file->filename;
69345673671SKevin Wolf     } else {
69445673671SKevin Wolf         filename = qdict_get_try_str(options, "filename");
69545673671SKevin Wolf     }
69645673671SKevin Wolf 
69745673671SKevin Wolf     trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
69828dcee10SStefan Hajnoczi 
6995d186eb0SKevin Wolf     /* bdrv_open() with directly using a protocol as drv. This layer is already
7005d186eb0SKevin Wolf      * opened, so assign it to bs (while file becomes a closed BlockDriverState)
7015d186eb0SKevin Wolf      * and return immediately. */
7025d186eb0SKevin Wolf     if (file != NULL && drv->bdrv_file_open) {
7035d186eb0SKevin Wolf         bdrv_swap(file, bs);
7045d186eb0SKevin Wolf         return 0;
7055d186eb0SKevin Wolf     }
7065d186eb0SKevin Wolf 
70757915332SKevin Wolf     bs->open_flags = flags;
70857915332SKevin Wolf     bs->buffer_alignment = 512;
709b64ec4e4SFam Zheng     open_flags = bdrv_open_flags(bs, flags);
710b64ec4e4SFam Zheng     bs->read_only = !(open_flags & BDRV_O_RDWR);
711b64ec4e4SFam Zheng 
712b64ec4e4SFam Zheng     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv, bs->read_only)) {
713b64ec4e4SFam Zheng         return -ENOTSUP;
714b64ec4e4SFam Zheng     }
71557915332SKevin Wolf 
71653fec9d3SStefan Hajnoczi     assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
717b64ec4e4SFam Zheng     if (!bs->read_only && (flags & BDRV_O_COPY_ON_READ)) {
71853fec9d3SStefan Hajnoczi         bdrv_enable_copy_on_read(bs);
71953fec9d3SStefan Hajnoczi     }
72053fec9d3SStefan Hajnoczi 
721c2ad1b0cSKevin Wolf     if (filename != NULL) {
72257915332SKevin Wolf         pstrcpy(bs->filename, sizeof(bs->filename), filename);
723c2ad1b0cSKevin Wolf     } else {
724c2ad1b0cSKevin Wolf         bs->filename[0] = '\0';
725c2ad1b0cSKevin Wolf     }
72657915332SKevin Wolf 
72757915332SKevin Wolf     bs->drv = drv;
7287267c094SAnthony Liguori     bs->opaque = g_malloc0(drv->instance_size);
72957915332SKevin Wolf 
73003f541bdSStefan Hajnoczi     bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
731e7c63796SStefan Hajnoczi 
73266f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
73366f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
7345d186eb0SKevin Wolf         assert(file == NULL);
735c2ad1b0cSKevin Wolf         assert(drv->bdrv_parse_filename || filename != NULL);
73656d1b4d2SKevin Wolf         ret = drv->bdrv_file_open(bs, options, open_flags);
737f500a6d3SKevin Wolf     } else {
7382af5ef70SKevin Wolf         if (file == NULL) {
7392af5ef70SKevin Wolf             qerror_report(ERROR_CLASS_GENERIC_ERROR, "Can't use '%s' as a "
7402af5ef70SKevin Wolf                           "block driver for the protocol level",
7412af5ef70SKevin Wolf                           drv->format_name);
7422af5ef70SKevin Wolf             ret = -EINVAL;
7432af5ef70SKevin Wolf             goto free_and_fail;
7442af5ef70SKevin Wolf         }
745f500a6d3SKevin Wolf         assert(file != NULL);
746f500a6d3SKevin Wolf         bs->file = file;
747b6ad491aSKevin Wolf         ret = drv->bdrv_open(bs, options, open_flags);
74866f82ceeSKevin Wolf     }
74966f82ceeSKevin Wolf 
75057915332SKevin Wolf     if (ret < 0) {
75157915332SKevin Wolf         goto free_and_fail;
75257915332SKevin Wolf     }
75357915332SKevin Wolf 
75451762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
75551762288SStefan Hajnoczi     if (ret < 0) {
75651762288SStefan Hajnoczi         goto free_and_fail;
75757915332SKevin Wolf     }
75851762288SStefan Hajnoczi 
75957915332SKevin Wolf #ifndef _WIN32
76057915332SKevin Wolf     if (bs->is_temporary) {
761c2ad1b0cSKevin Wolf         assert(filename != NULL);
76257915332SKevin Wolf         unlink(filename);
76357915332SKevin Wolf     }
76457915332SKevin Wolf #endif
76557915332SKevin Wolf     return 0;
76657915332SKevin Wolf 
76757915332SKevin Wolf free_and_fail:
76866f82ceeSKevin Wolf     bs->file = NULL;
7697267c094SAnthony Liguori     g_free(bs->opaque);
77057915332SKevin Wolf     bs->opaque = NULL;
77157915332SKevin Wolf     bs->drv = NULL;
77257915332SKevin Wolf     return ret;
77357915332SKevin Wolf }
77457915332SKevin Wolf 
77557915332SKevin Wolf /*
776b6ce07aaSKevin Wolf  * Opens a file using a protocol (file, host_device, nbd, ...)
777787e4a85SKevin Wolf  *
778787e4a85SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
779787e4a85SKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
780787e4a85SKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
781787e4a85SKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_file_open.
782b6ce07aaSKevin Wolf  */
783787e4a85SKevin Wolf int bdrv_file_open(BlockDriverState **pbs, const char *filename,
784787e4a85SKevin Wolf                    QDict *options, int flags)
785b338082bSbellard {
78683f64091Sbellard     BlockDriverState *bs;
7876db95603SChristoph Hellwig     BlockDriver *drv;
788c2ad1b0cSKevin Wolf     const char *drvname;
78998289620SKevin Wolf     bool allow_protocol_prefix = false;
79083f64091Sbellard     int ret;
7913b0d4f61Sbellard 
792707ff828SKevin Wolf     /* NULL means an empty set of options */
793707ff828SKevin Wolf     if (options == NULL) {
794707ff828SKevin Wolf         options = qdict_new();
7953b0d4f61Sbellard     }
796707ff828SKevin Wolf 
797707ff828SKevin Wolf     bs = bdrv_new("");
798707ff828SKevin Wolf     bs->options = options;
799707ff828SKevin Wolf     options = qdict_clone_shallow(options);
800707ff828SKevin Wolf 
801035fccdfSKevin Wolf     /* Fetch the file name from the options QDict if necessary */
802035fccdfSKevin Wolf     if (!filename) {
803035fccdfSKevin Wolf         filename = qdict_get_try_str(options, "filename");
804035fccdfSKevin Wolf     } else if (filename && !qdict_haskey(options, "filename")) {
805035fccdfSKevin Wolf         qdict_put(options, "filename", qstring_from_str(filename));
80698289620SKevin Wolf         allow_protocol_prefix = true;
807035fccdfSKevin Wolf     } else {
808035fccdfSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR, "Can't specify 'file' and "
809035fccdfSKevin Wolf                       "'filename' options at the same time");
810035fccdfSKevin Wolf         ret = -EINVAL;
811035fccdfSKevin Wolf         goto fail;
812035fccdfSKevin Wolf     }
813035fccdfSKevin Wolf 
814c2ad1b0cSKevin Wolf     /* Find the right block driver */
815c2ad1b0cSKevin Wolf     drvname = qdict_get_try_str(options, "driver");
816c2ad1b0cSKevin Wolf     if (drvname) {
817b64ec4e4SFam Zheng         drv = bdrv_find_whitelisted_format(drvname, !(flags & BDRV_O_RDWR));
818c2ad1b0cSKevin Wolf         qdict_del(options, "driver");
819c2ad1b0cSKevin Wolf     } else if (filename) {
82098289620SKevin Wolf         drv = bdrv_find_protocol(filename, allow_protocol_prefix);
82198289620SKevin Wolf         if (!drv) {
82298289620SKevin Wolf             qerror_report(ERROR_CLASS_GENERIC_ERROR, "Unknown protocol");
82398289620SKevin Wolf         }
824c2ad1b0cSKevin Wolf     } else {
825c2ad1b0cSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR,
826c2ad1b0cSKevin Wolf                       "Must specify either driver or file");
827c2ad1b0cSKevin Wolf         drv = NULL;
828c2ad1b0cSKevin Wolf     }
829c2ad1b0cSKevin Wolf 
830c2ad1b0cSKevin Wolf     if (!drv) {
831c2ad1b0cSKevin Wolf         ret = -ENOENT;
832c2ad1b0cSKevin Wolf         goto fail;
833c2ad1b0cSKevin Wolf     }
834c2ad1b0cSKevin Wolf 
835c2ad1b0cSKevin Wolf     /* Parse the filename and open it */
836c2ad1b0cSKevin Wolf     if (drv->bdrv_parse_filename && filename) {
8376963a30dSKevin Wolf         Error *local_err = NULL;
8386963a30dSKevin Wolf         drv->bdrv_parse_filename(filename, options, &local_err);
8396963a30dSKevin Wolf         if (error_is_set(&local_err)) {
8406963a30dSKevin Wolf             qerror_report_err(local_err);
8416963a30dSKevin Wolf             error_free(local_err);
8426963a30dSKevin Wolf             ret = -EINVAL;
8436963a30dSKevin Wolf             goto fail;
8446963a30dSKevin Wolf         }
84556d1b4d2SKevin Wolf         qdict_del(options, "filename");
846c2ad1b0cSKevin Wolf     } else if (!drv->bdrv_parse_filename && !filename) {
847c2ad1b0cSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR,
848c2ad1b0cSKevin Wolf                       "The '%s' block driver requires a file name",
849c2ad1b0cSKevin Wolf                       drv->format_name);
850c2ad1b0cSKevin Wolf         ret = -EINVAL;
851c2ad1b0cSKevin Wolf         goto fail;
8526963a30dSKevin Wolf     }
8536963a30dSKevin Wolf 
854035fccdfSKevin Wolf     ret = bdrv_open_common(bs, NULL, options, flags, drv);
855707ff828SKevin Wolf     if (ret < 0) {
856707ff828SKevin Wolf         goto fail;
857707ff828SKevin Wolf     }
858707ff828SKevin Wolf 
859707ff828SKevin Wolf     /* Check if any unknown options were used */
860707ff828SKevin Wolf     if (qdict_size(options) != 0) {
861707ff828SKevin Wolf         const QDictEntry *entry = qdict_first(options);
862707ff828SKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR, "Block protocol '%s' doesn't "
863707ff828SKevin Wolf                       "support the option '%s'",
864707ff828SKevin Wolf                       drv->format_name, entry->key);
865707ff828SKevin Wolf         ret = -EINVAL;
866707ff828SKevin Wolf         goto fail;
867707ff828SKevin Wolf     }
868707ff828SKevin Wolf     QDECREF(options);
869707ff828SKevin Wolf 
87071d0770cSaliguori     bs->growable = 1;
87183f64091Sbellard     *pbs = bs;
87283f64091Sbellard     return 0;
873707ff828SKevin Wolf 
874707ff828SKevin Wolf fail:
875707ff828SKevin Wolf     QDECREF(options);
876707ff828SKevin Wolf     if (!bs->drv) {
877707ff828SKevin Wolf         QDECREF(bs->options);
878707ff828SKevin Wolf     }
879707ff828SKevin Wolf     bdrv_delete(bs);
880707ff828SKevin Wolf     return ret;
8813b0d4f61Sbellard }
8823b0d4f61Sbellard 
88331ca6d07SKevin Wolf /*
88431ca6d07SKevin Wolf  * Opens the backing file for a BlockDriverState if not yet open
88531ca6d07SKevin Wolf  *
88631ca6d07SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
88731ca6d07SKevin Wolf  * empty set of options. The reference to the QDict is transferred to this
88831ca6d07SKevin Wolf  * function (even on failure), so if the caller intends to reuse the dictionary,
88931ca6d07SKevin Wolf  * it needs to use QINCREF() before calling bdrv_file_open.
89031ca6d07SKevin Wolf  */
89131ca6d07SKevin Wolf int bdrv_open_backing_file(BlockDriverState *bs, QDict *options)
8929156df12SPaolo Bonzini {
8939156df12SPaolo Bonzini     char backing_filename[PATH_MAX];
8949156df12SPaolo Bonzini     int back_flags, ret;
8959156df12SPaolo Bonzini     BlockDriver *back_drv = NULL;
8969156df12SPaolo Bonzini 
8979156df12SPaolo Bonzini     if (bs->backing_hd != NULL) {
89831ca6d07SKevin Wolf         QDECREF(options);
8999156df12SPaolo Bonzini         return 0;
9009156df12SPaolo Bonzini     }
9019156df12SPaolo Bonzini 
90231ca6d07SKevin Wolf     /* NULL means an empty set of options */
90331ca6d07SKevin Wolf     if (options == NULL) {
90431ca6d07SKevin Wolf         options = qdict_new();
90531ca6d07SKevin Wolf     }
90631ca6d07SKevin Wolf 
9079156df12SPaolo Bonzini     bs->open_flags &= ~BDRV_O_NO_BACKING;
9081cb6f506SKevin Wolf     if (qdict_haskey(options, "file.filename")) {
9091cb6f506SKevin Wolf         backing_filename[0] = '\0';
9101cb6f506SKevin Wolf     } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
91131ca6d07SKevin Wolf         QDECREF(options);
9129156df12SPaolo Bonzini         return 0;
9139156df12SPaolo Bonzini     }
9149156df12SPaolo Bonzini 
9159156df12SPaolo Bonzini     bs->backing_hd = bdrv_new("");
9169156df12SPaolo Bonzini     bdrv_get_full_backing_filename(bs, backing_filename,
9179156df12SPaolo Bonzini                                    sizeof(backing_filename));
9189156df12SPaolo Bonzini 
9199156df12SPaolo Bonzini     if (bs->backing_format[0] != '\0') {
9209156df12SPaolo Bonzini         back_drv = bdrv_find_format(bs->backing_format);
9219156df12SPaolo Bonzini     }
9229156df12SPaolo Bonzini 
9239156df12SPaolo Bonzini     /* backing files always opened read-only */
9249156df12SPaolo Bonzini     back_flags = bs->open_flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT);
9259156df12SPaolo Bonzini 
92631ca6d07SKevin Wolf     ret = bdrv_open(bs->backing_hd,
92731ca6d07SKevin Wolf                     *backing_filename ? backing_filename : NULL, options,
928de9c0cecSKevin Wolf                     back_flags, back_drv);
9299156df12SPaolo Bonzini     if (ret < 0) {
9309156df12SPaolo Bonzini         bdrv_delete(bs->backing_hd);
9319156df12SPaolo Bonzini         bs->backing_hd = NULL;
9329156df12SPaolo Bonzini         bs->open_flags |= BDRV_O_NO_BACKING;
9339156df12SPaolo Bonzini         return ret;
9349156df12SPaolo Bonzini     }
9359156df12SPaolo Bonzini     return 0;
9369156df12SPaolo Bonzini }
9379156df12SPaolo Bonzini 
938707ff828SKevin Wolf static void extract_subqdict(QDict *src, QDict **dst, const char *start)
939707ff828SKevin Wolf {
940707ff828SKevin Wolf     const QDictEntry *entry, *next;
941707ff828SKevin Wolf     const char *p;
942707ff828SKevin Wolf 
943707ff828SKevin Wolf     *dst = qdict_new();
944707ff828SKevin Wolf     entry = qdict_first(src);
945707ff828SKevin Wolf 
946707ff828SKevin Wolf     while (entry != NULL) {
947707ff828SKevin Wolf         next = qdict_next(src, entry);
948707ff828SKevin Wolf         if (strstart(entry->key, start, &p)) {
949707ff828SKevin Wolf             qobject_incref(entry->value);
950707ff828SKevin Wolf             qdict_put_obj(*dst, p, entry->value);
951707ff828SKevin Wolf             qdict_del(src, entry->key);
952707ff828SKevin Wolf         }
953707ff828SKevin Wolf         entry = next;
954707ff828SKevin Wolf     }
955707ff828SKevin Wolf }
956707ff828SKevin Wolf 
957b6ce07aaSKevin Wolf /*
958b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
959de9c0cecSKevin Wolf  *
960de9c0cecSKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
961de9c0cecSKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
962de9c0cecSKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
963de9c0cecSKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_open.
964b6ce07aaSKevin Wolf  */
965de9c0cecSKevin Wolf int bdrv_open(BlockDriverState *bs, const char *filename, QDict *options,
966de9c0cecSKevin Wolf               int flags, BlockDriver *drv)
967ea2384d3Sbellard {
968b6ce07aaSKevin Wolf     int ret;
96989c9bc3dSStefan Weil     /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
97089c9bc3dSStefan Weil     char tmp_filename[PATH_MAX + 1];
971f500a6d3SKevin Wolf     BlockDriverState *file = NULL;
972707ff828SKevin Wolf     QDict *file_options = NULL;
97374fe54f2SKevin Wolf     const char *drvname;
97433e3963eSbellard 
975de9c0cecSKevin Wolf     /* NULL means an empty set of options */
976de9c0cecSKevin Wolf     if (options == NULL) {
977de9c0cecSKevin Wolf         options = qdict_new();
978de9c0cecSKevin Wolf     }
979de9c0cecSKevin Wolf 
980de9c0cecSKevin Wolf     bs->options = options;
981b6ad491aSKevin Wolf     options = qdict_clone_shallow(options);
982de9c0cecSKevin Wolf 
983de9c0cecSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay */
98483f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
985ea2384d3Sbellard         BlockDriverState *bs1;
986ea2384d3Sbellard         int64_t total_size;
98791a073a9SKevin Wolf         BlockDriver *bdrv_qcow2;
98808b392e1SKevin Wolf         QEMUOptionParameter *create_options;
989b6ce07aaSKevin Wolf         char backing_filename[PATH_MAX];
99033e3963eSbellard 
991c2ad1b0cSKevin Wolf         if (qdict_size(options) != 0) {
992c2ad1b0cSKevin Wolf             error_report("Can't use snapshot=on with driver-specific options");
993c2ad1b0cSKevin Wolf             ret = -EINVAL;
994c2ad1b0cSKevin Wolf             goto fail;
995c2ad1b0cSKevin Wolf         }
996c2ad1b0cSKevin Wolf         assert(filename != NULL);
997c2ad1b0cSKevin Wolf 
998ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
999ea2384d3Sbellard            instead of opening 'filename' directly */
1000ea2384d3Sbellard 
1001ea2384d3Sbellard         /* if there is a backing file, use it */
1002ea2384d3Sbellard         bs1 = bdrv_new("");
1003de9c0cecSKevin Wolf         ret = bdrv_open(bs1, filename, NULL, 0, drv);
100451d7c00cSaliguori         if (ret < 0) {
1005ea2384d3Sbellard             bdrv_delete(bs1);
1006de9c0cecSKevin Wolf             goto fail;
1007ea2384d3Sbellard         }
10083e82990bSJes Sorensen         total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
10097c96d46eSaliguori 
1010ea2384d3Sbellard         bdrv_delete(bs1);
1011ea2384d3Sbellard 
1012eba25057SJim Meyering         ret = get_tmp_filename(tmp_filename, sizeof(tmp_filename));
1013eba25057SJim Meyering         if (ret < 0) {
1014de9c0cecSKevin Wolf             goto fail;
1015eba25057SJim Meyering         }
10167c96d46eSaliguori 
10177c96d46eSaliguori         /* Real path is meaningless for protocols */
10184d70655bSStefan Hajnoczi         if (path_has_protocol(filename)) {
10197c96d46eSaliguori             snprintf(backing_filename, sizeof(backing_filename),
10207c96d46eSaliguori                      "%s", filename);
1021de9c0cecSKevin Wolf         } else if (!realpath(filename, backing_filename)) {
1022de9c0cecSKevin Wolf             ret = -errno;
1023de9c0cecSKevin Wolf             goto fail;
1024de9c0cecSKevin Wolf         }
10257c96d46eSaliguori 
102691a073a9SKevin Wolf         bdrv_qcow2 = bdrv_find_format("qcow2");
102708b392e1SKevin Wolf         create_options = parse_option_parameters("", bdrv_qcow2->create_options,
102808b392e1SKevin Wolf                                                  NULL);
102991a073a9SKevin Wolf 
103008b392e1SKevin Wolf         set_option_parameter_int(create_options, BLOCK_OPT_SIZE, total_size);
103108b392e1SKevin Wolf         set_option_parameter(create_options, BLOCK_OPT_BACKING_FILE,
103208b392e1SKevin Wolf                              backing_filename);
103391a073a9SKevin Wolf         if (drv) {
103408b392e1SKevin Wolf             set_option_parameter(create_options, BLOCK_OPT_BACKING_FMT,
103591a073a9SKevin Wolf                 drv->format_name);
103691a073a9SKevin Wolf         }
103791a073a9SKevin Wolf 
103808b392e1SKevin Wolf         ret = bdrv_create(bdrv_qcow2, tmp_filename, create_options);
103908b392e1SKevin Wolf         free_option_parameters(create_options);
104051d7c00cSaliguori         if (ret < 0) {
1041de9c0cecSKevin Wolf             goto fail;
1042ea2384d3Sbellard         }
104391a073a9SKevin Wolf 
1044ea2384d3Sbellard         filename = tmp_filename;
104591a073a9SKevin Wolf         drv = bdrv_qcow2;
1046ea2384d3Sbellard         bs->is_temporary = 1;
1047ea2384d3Sbellard     }
1048ea2384d3Sbellard 
1049f500a6d3SKevin Wolf     /* Open image file without format layer */
1050be028adcSJeff Cody     if (flags & BDRV_O_RDWR) {
1051be028adcSJeff Cody         flags |= BDRV_O_ALLOW_RDWR;
1052be028adcSJeff Cody     }
1053be028adcSJeff Cody 
1054707ff828SKevin Wolf     extract_subqdict(options, &file_options, "file.");
1055707ff828SKevin Wolf 
1056707ff828SKevin Wolf     ret = bdrv_file_open(&file, filename, file_options,
105750b05b6fSKevin Wolf                          bdrv_open_flags(bs, flags | BDRV_O_UNMAP));
1058f500a6d3SKevin Wolf     if (ret < 0) {
1059de9c0cecSKevin Wolf         goto fail;
1060f500a6d3SKevin Wolf     }
1061f500a6d3SKevin Wolf 
1062f500a6d3SKevin Wolf     /* Find the right image format driver */
106374fe54f2SKevin Wolf     drvname = qdict_get_try_str(options, "driver");
106474fe54f2SKevin Wolf     if (drvname) {
106574fe54f2SKevin Wolf         drv = bdrv_find_whitelisted_format(drvname, !(flags & BDRV_O_RDWR));
106674fe54f2SKevin Wolf         qdict_del(options, "driver");
106774fe54f2SKevin Wolf     }
106874fe54f2SKevin Wolf 
1069f500a6d3SKevin Wolf     if (!drv) {
1070f500a6d3SKevin Wolf         ret = find_image_format(file, filename, &drv);
1071f500a6d3SKevin Wolf     }
1072f500a6d3SKevin Wolf 
1073f500a6d3SKevin Wolf     if (!drv) {
1074f500a6d3SKevin Wolf         goto unlink_and_fail;
1075f500a6d3SKevin Wolf     }
1076f500a6d3SKevin Wolf 
1077b6ce07aaSKevin Wolf     /* Open the image */
1078035fccdfSKevin Wolf     ret = bdrv_open_common(bs, file, options, flags, drv);
1079b6ce07aaSKevin Wolf     if (ret < 0) {
10806987307cSChristoph Hellwig         goto unlink_and_fail;
10816987307cSChristoph Hellwig     }
10826987307cSChristoph Hellwig 
1083f500a6d3SKevin Wolf     if (bs->file != file) {
1084f500a6d3SKevin Wolf         bdrv_delete(file);
1085f500a6d3SKevin Wolf         file = NULL;
1086f500a6d3SKevin Wolf     }
1087f500a6d3SKevin Wolf 
1088b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
10899156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
109031ca6d07SKevin Wolf         QDict *backing_options;
109131ca6d07SKevin Wolf 
109231ca6d07SKevin Wolf         extract_subqdict(options, &backing_options, "backing.");
109331ca6d07SKevin Wolf         ret = bdrv_open_backing_file(bs, backing_options);
1094b6ce07aaSKevin Wolf         if (ret < 0) {
1095b6ad491aSKevin Wolf             goto close_and_fail;
1096b6ce07aaSKevin Wolf         }
1097b6ce07aaSKevin Wolf     }
1098b6ce07aaSKevin Wolf 
1099b6ad491aSKevin Wolf     /* Check if any unknown options were used */
1100b6ad491aSKevin Wolf     if (qdict_size(options) != 0) {
1101b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
1102b6ad491aSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR, "Block format '%s' used by "
1103b6ad491aSKevin Wolf             "device '%s' doesn't support the option '%s'",
1104b6ad491aSKevin Wolf             drv->format_name, bs->device_name, entry->key);
1105b6ad491aSKevin Wolf 
1106b6ad491aSKevin Wolf         ret = -EINVAL;
1107b6ad491aSKevin Wolf         goto close_and_fail;
1108b6ad491aSKevin Wolf     }
1109b6ad491aSKevin Wolf     QDECREF(options);
1110b6ad491aSKevin Wolf 
1111b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
11127d4b4ba5SMarkus Armbruster         bdrv_dev_change_media_cb(bs, true);
1113b6ce07aaSKevin Wolf     }
1114b6ce07aaSKevin Wolf 
111598f90dbaSZhi Yong Wu     /* throttling disk I/O limits */
111698f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
111798f90dbaSZhi Yong Wu         bdrv_io_limits_enable(bs);
111898f90dbaSZhi Yong Wu     }
111998f90dbaSZhi Yong Wu 
1120b6ce07aaSKevin Wolf     return 0;
1121b6ce07aaSKevin Wolf 
1122b6ce07aaSKevin Wolf unlink_and_fail:
1123f500a6d3SKevin Wolf     if (file != NULL) {
1124f500a6d3SKevin Wolf         bdrv_delete(file);
1125f500a6d3SKevin Wolf     }
1126b6ce07aaSKevin Wolf     if (bs->is_temporary) {
1127b6ce07aaSKevin Wolf         unlink(filename);
1128b6ce07aaSKevin Wolf     }
1129de9c0cecSKevin Wolf fail:
1130de9c0cecSKevin Wolf     QDECREF(bs->options);
1131b6ad491aSKevin Wolf     QDECREF(options);
1132de9c0cecSKevin Wolf     bs->options = NULL;
1133b6ad491aSKevin Wolf     return ret;
1134de9c0cecSKevin Wolf 
1135b6ad491aSKevin Wolf close_and_fail:
1136b6ad491aSKevin Wolf     bdrv_close(bs);
1137b6ad491aSKevin Wolf     QDECREF(options);
1138b6ce07aaSKevin Wolf     return ret;
1139b6ce07aaSKevin Wolf }
1140b6ce07aaSKevin Wolf 
1141e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1142e971aa12SJeff Cody      bool prepared;
1143e971aa12SJeff Cody      BDRVReopenState state;
1144e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1145e971aa12SJeff Cody } BlockReopenQueueEntry;
1146e971aa12SJeff Cody 
1147e971aa12SJeff Cody /*
1148e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1149e971aa12SJeff Cody  * reopen of multiple devices.
1150e971aa12SJeff Cody  *
1151e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1152e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1153e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1154e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1155e971aa12SJeff Cody  * atomic 'set'.
1156e971aa12SJeff Cody  *
1157e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1158e971aa12SJeff Cody  *
1159e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1160e971aa12SJeff Cody  *
1161e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1162e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1163e971aa12SJeff Cody  *
1164e971aa12SJeff Cody  */
1165e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1166e971aa12SJeff Cody                                     BlockDriverState *bs, int flags)
1167e971aa12SJeff Cody {
1168e971aa12SJeff Cody     assert(bs != NULL);
1169e971aa12SJeff Cody 
1170e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
1171e971aa12SJeff Cody     if (bs_queue == NULL) {
1172e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1173e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1174e971aa12SJeff Cody     }
1175e971aa12SJeff Cody 
1176e971aa12SJeff Cody     if (bs->file) {
1177e971aa12SJeff Cody         bdrv_reopen_queue(bs_queue, bs->file, flags);
1178e971aa12SJeff Cody     }
1179e971aa12SJeff Cody 
1180e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1181e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1182e971aa12SJeff Cody 
1183e971aa12SJeff Cody     bs_entry->state.bs = bs;
1184e971aa12SJeff Cody     bs_entry->state.flags = flags;
1185e971aa12SJeff Cody 
1186e971aa12SJeff Cody     return bs_queue;
1187e971aa12SJeff Cody }
1188e971aa12SJeff Cody 
1189e971aa12SJeff Cody /*
1190e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1191e971aa12SJeff Cody  *
1192e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1193e971aa12SJeff Cody  * via bdrv_reopen_queue().
1194e971aa12SJeff Cody  *
1195e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1196e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1197e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1198e971aa12SJeff Cody  * data cleaned up.
1199e971aa12SJeff Cody  *
1200e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1201e971aa12SJeff Cody  * to all devices.
1202e971aa12SJeff Cody  *
1203e971aa12SJeff Cody  */
1204e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1205e971aa12SJeff Cody {
1206e971aa12SJeff Cody     int ret = -1;
1207e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1208e971aa12SJeff Cody     Error *local_err = NULL;
1209e971aa12SJeff Cody 
1210e971aa12SJeff Cody     assert(bs_queue != NULL);
1211e971aa12SJeff Cody 
1212e971aa12SJeff Cody     bdrv_drain_all();
1213e971aa12SJeff Cody 
1214e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1215e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1216e971aa12SJeff Cody             error_propagate(errp, local_err);
1217e971aa12SJeff Cody             goto cleanup;
1218e971aa12SJeff Cody         }
1219e971aa12SJeff Cody         bs_entry->prepared = true;
1220e971aa12SJeff Cody     }
1221e971aa12SJeff Cody 
1222e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1223e971aa12SJeff Cody      * changes
1224e971aa12SJeff Cody      */
1225e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1226e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1227e971aa12SJeff Cody     }
1228e971aa12SJeff Cody 
1229e971aa12SJeff Cody     ret = 0;
1230e971aa12SJeff Cody 
1231e971aa12SJeff Cody cleanup:
1232e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1233e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1234e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1235e971aa12SJeff Cody         }
1236e971aa12SJeff Cody         g_free(bs_entry);
1237e971aa12SJeff Cody     }
1238e971aa12SJeff Cody     g_free(bs_queue);
1239e971aa12SJeff Cody     return ret;
1240e971aa12SJeff Cody }
1241e971aa12SJeff Cody 
1242e971aa12SJeff Cody 
1243e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1244e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1245e971aa12SJeff Cody {
1246e971aa12SJeff Cody     int ret = -1;
1247e971aa12SJeff Cody     Error *local_err = NULL;
1248e971aa12SJeff Cody     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, bdrv_flags);
1249e971aa12SJeff Cody 
1250e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1251e971aa12SJeff Cody     if (local_err != NULL) {
1252e971aa12SJeff Cody         error_propagate(errp, local_err);
1253e971aa12SJeff Cody     }
1254e971aa12SJeff Cody     return ret;
1255e971aa12SJeff Cody }
1256e971aa12SJeff Cody 
1257e971aa12SJeff Cody 
1258e971aa12SJeff Cody /*
1259e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1260e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1261e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1262e971aa12SJeff Cody  *
1263e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1264e971aa12SJeff Cody  * flags are the new open flags
1265e971aa12SJeff Cody  * queue is the reopen queue
1266e971aa12SJeff Cody  *
1267e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1268e971aa12SJeff Cody  * as well.
1269e971aa12SJeff Cody  *
1270e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1271e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1272e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1273e971aa12SJeff Cody  *
1274e971aa12SJeff Cody  */
1275e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1276e971aa12SJeff Cody                         Error **errp)
1277e971aa12SJeff Cody {
1278e971aa12SJeff Cody     int ret = -1;
1279e971aa12SJeff Cody     Error *local_err = NULL;
1280e971aa12SJeff Cody     BlockDriver *drv;
1281e971aa12SJeff Cody 
1282e971aa12SJeff Cody     assert(reopen_state != NULL);
1283e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1284e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1285e971aa12SJeff Cody 
1286e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1287e971aa12SJeff Cody      * to r/w */
1288e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1289e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
1290e971aa12SJeff Cody         error_set(errp, QERR_DEVICE_IS_READ_ONLY,
1291e971aa12SJeff Cody                   reopen_state->bs->device_name);
1292e971aa12SJeff Cody         goto error;
1293e971aa12SJeff Cody     }
1294e971aa12SJeff Cody 
1295e971aa12SJeff Cody 
1296e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1297e971aa12SJeff Cody     if (ret) {
1298e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1299e971aa12SJeff Cody                   strerror(-ret));
1300e971aa12SJeff Cody         goto error;
1301e971aa12SJeff Cody     }
1302e971aa12SJeff Cody 
1303e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1304e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1305e971aa12SJeff Cody         if (ret) {
1306e971aa12SJeff Cody             if (local_err != NULL) {
1307e971aa12SJeff Cody                 error_propagate(errp, local_err);
1308e971aa12SJeff Cody             } else {
1309d8b6895fSLuiz Capitulino                 error_setg(errp, "failed while preparing to reopen image '%s'",
1310e971aa12SJeff Cody                            reopen_state->bs->filename);
1311e971aa12SJeff Cody             }
1312e971aa12SJeff Cody             goto error;
1313e971aa12SJeff Cody         }
1314e971aa12SJeff Cody     } else {
1315e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1316e971aa12SJeff Cody          * handler for each supported drv. */
1317e971aa12SJeff Cody         error_set(errp, QERR_BLOCK_FORMAT_FEATURE_NOT_SUPPORTED,
1318e971aa12SJeff Cody                   drv->format_name, reopen_state->bs->device_name,
1319e971aa12SJeff Cody                  "reopening of file");
1320e971aa12SJeff Cody         ret = -1;
1321e971aa12SJeff Cody         goto error;
1322e971aa12SJeff Cody     }
1323e971aa12SJeff Cody 
1324e971aa12SJeff Cody     ret = 0;
1325e971aa12SJeff Cody 
1326e971aa12SJeff Cody error:
1327e971aa12SJeff Cody     return ret;
1328e971aa12SJeff Cody }
1329e971aa12SJeff Cody 
1330e971aa12SJeff Cody /*
1331e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1332e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1333e971aa12SJeff Cody  * the active BlockDriverState contents.
1334e971aa12SJeff Cody  */
1335e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1336e971aa12SJeff Cody {
1337e971aa12SJeff Cody     BlockDriver *drv;
1338e971aa12SJeff Cody 
1339e971aa12SJeff Cody     assert(reopen_state != NULL);
1340e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1341e971aa12SJeff Cody     assert(drv != NULL);
1342e971aa12SJeff Cody 
1343e971aa12SJeff Cody     /* If there are any driver level actions to take */
1344e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1345e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1346e971aa12SJeff Cody     }
1347e971aa12SJeff Cody 
1348e971aa12SJeff Cody     /* set BDS specific flags now */
1349e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1350e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1351e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1352e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1353e971aa12SJeff Cody }
1354e971aa12SJeff Cody 
1355e971aa12SJeff Cody /*
1356e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1357e971aa12SJeff Cody  * reopen_state
1358e971aa12SJeff Cody  */
1359e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1360e971aa12SJeff Cody {
1361e971aa12SJeff Cody     BlockDriver *drv;
1362e971aa12SJeff Cody 
1363e971aa12SJeff Cody     assert(reopen_state != NULL);
1364e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1365e971aa12SJeff Cody     assert(drv != NULL);
1366e971aa12SJeff Cody 
1367e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1368e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1369e971aa12SJeff Cody     }
1370e971aa12SJeff Cody }
1371e971aa12SJeff Cody 
1372e971aa12SJeff Cody 
1373fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1374fc01f7e7Sbellard {
13753e914655SPaolo Bonzini     if (bs->job) {
13763e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
13773e914655SPaolo Bonzini     }
137858fda173SStefan Hajnoczi     bdrv_drain_all(); /* complete I/O */
137958fda173SStefan Hajnoczi     bdrv_flush(bs);
138058fda173SStefan Hajnoczi     bdrv_drain_all(); /* in case flush left pending I/O */
1381d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
13827094f12fSKevin Wolf 
13833cbc002cSPaolo Bonzini     if (bs->drv) {
1384557df6acSStefan Hajnoczi         if (bs->backing_hd) {
1385ea2384d3Sbellard             bdrv_delete(bs->backing_hd);
1386557df6acSStefan Hajnoczi             bs->backing_hd = NULL;
1387557df6acSStefan Hajnoczi         }
1388ea2384d3Sbellard         bs->drv->bdrv_close(bs);
13897267c094SAnthony Liguori         g_free(bs->opaque);
1390ea2384d3Sbellard #ifdef _WIN32
1391ea2384d3Sbellard         if (bs->is_temporary) {
1392ea2384d3Sbellard             unlink(bs->filename);
1393ea2384d3Sbellard         }
139467b915a5Sbellard #endif
1395ea2384d3Sbellard         bs->opaque = NULL;
1396ea2384d3Sbellard         bs->drv = NULL;
139753fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1398a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1399a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
14006405875cSPaolo Bonzini         bs->total_sectors = 0;
14016405875cSPaolo Bonzini         bs->encrypted = 0;
14026405875cSPaolo Bonzini         bs->valid_key = 0;
14036405875cSPaolo Bonzini         bs->sg = 0;
14046405875cSPaolo Bonzini         bs->growable = 0;
1405de9c0cecSKevin Wolf         QDECREF(bs->options);
1406de9c0cecSKevin Wolf         bs->options = NULL;
1407b338082bSbellard 
140866f82ceeSKevin Wolf         if (bs->file != NULL) {
14090ac9377dSPaolo Bonzini             bdrv_delete(bs->file);
14100ac9377dSPaolo Bonzini             bs->file = NULL;
141166f82ceeSKevin Wolf         }
14129ca11154SPavel Hrdina     }
141366f82ceeSKevin Wolf 
14147d4b4ba5SMarkus Armbruster     bdrv_dev_change_media_cb(bs, false);
141598f90dbaSZhi Yong Wu 
141698f90dbaSZhi Yong Wu     /*throttling disk I/O limits*/
141798f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
141898f90dbaSZhi Yong Wu         bdrv_io_limits_disable(bs);
141998f90dbaSZhi Yong Wu     }
1420b338082bSbellard }
1421b338082bSbellard 
14222bc93fedSMORITA Kazutaka void bdrv_close_all(void)
14232bc93fedSMORITA Kazutaka {
14242bc93fedSMORITA Kazutaka     BlockDriverState *bs;
14252bc93fedSMORITA Kazutaka 
14262bc93fedSMORITA Kazutaka     QTAILQ_FOREACH(bs, &bdrv_states, list) {
14272bc93fedSMORITA Kazutaka         bdrv_close(bs);
14282bc93fedSMORITA Kazutaka     }
14292bc93fedSMORITA Kazutaka }
14302bc93fedSMORITA Kazutaka 
1431922453bcSStefan Hajnoczi /*
1432922453bcSStefan Hajnoczi  * Wait for pending requests to complete across all BlockDriverStates
1433922453bcSStefan Hajnoczi  *
1434922453bcSStefan Hajnoczi  * This function does not flush data to disk, use bdrv_flush_all() for that
1435922453bcSStefan Hajnoczi  * after calling this function.
14364c355d53SZhi Yong Wu  *
14374c355d53SZhi Yong Wu  * Note that completion of an asynchronous I/O operation can trigger any
14384c355d53SZhi Yong Wu  * number of other I/O operations on other devices---for example a coroutine
14394c355d53SZhi Yong Wu  * can be arbitrarily complex and a constant flow of I/O can come until the
14404c355d53SZhi Yong Wu  * coroutine is complete.  Because of this, it is not possible to have a
14414c355d53SZhi Yong Wu  * function to drain a single device's I/O queue.
1442922453bcSStefan Hajnoczi  */
1443922453bcSStefan Hajnoczi void bdrv_drain_all(void)
1444922453bcSStefan Hajnoczi {
1445922453bcSStefan Hajnoczi     BlockDriverState *bs;
14464c355d53SZhi Yong Wu     bool busy;
1447922453bcSStefan Hajnoczi 
14484c355d53SZhi Yong Wu     do {
14494c355d53SZhi Yong Wu         busy = qemu_aio_wait();
14504c355d53SZhi Yong Wu 
14514c355d53SZhi Yong Wu         /* FIXME: We do not have timer support here, so this is effectively
14524c355d53SZhi Yong Wu          * a busy wait.
14534c355d53SZhi Yong Wu          */
14544c355d53SZhi Yong Wu         QTAILQ_FOREACH(bs, &bdrv_states, list) {
1455b681a1c7SBenoît Canet             while (qemu_co_enter_next(&bs->throttled_reqs)) {
14564c355d53SZhi Yong Wu                 busy = true;
14574c355d53SZhi Yong Wu             }
14584c355d53SZhi Yong Wu         }
14594c355d53SZhi Yong Wu     } while (busy);
1460922453bcSStefan Hajnoczi 
1461922453bcSStefan Hajnoczi     /* If requests are still pending there is a bug somewhere */
1462922453bcSStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1463922453bcSStefan Hajnoczi         assert(QLIST_EMPTY(&bs->tracked_requests));
1464922453bcSStefan Hajnoczi         assert(qemu_co_queue_empty(&bs->throttled_reqs));
1465922453bcSStefan Hajnoczi     }
1466922453bcSStefan Hajnoczi }
1467922453bcSStefan Hajnoczi 
1468d22b2f41SRyan Harper /* make a BlockDriverState anonymous by removing from bdrv_state list.
1469d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1470d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1471d22b2f41SRyan Harper {
1472d22b2f41SRyan Harper     if (bs->device_name[0] != '\0') {
1473d22b2f41SRyan Harper         QTAILQ_REMOVE(&bdrv_states, bs, list);
1474d22b2f41SRyan Harper     }
1475d22b2f41SRyan Harper     bs->device_name[0] = '\0';
1476d22b2f41SRyan Harper }
1477d22b2f41SRyan Harper 
1478e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
1479e023b2e2SPaolo Bonzini {
1480e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
1481e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
1482e023b2e2SPaolo Bonzini     }
1483e023b2e2SPaolo Bonzini }
1484e023b2e2SPaolo Bonzini 
14854ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
14864ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
14874ddc07caSPaolo Bonzini {
14884ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
14894ddc07caSPaolo Bonzini     bs_dest->open_flags         = bs_src->open_flags;
14904ddc07caSPaolo Bonzini 
14914ddc07caSPaolo Bonzini     /* dev info */
14924ddc07caSPaolo Bonzini     bs_dest->dev_ops            = bs_src->dev_ops;
14934ddc07caSPaolo Bonzini     bs_dest->dev_opaque         = bs_src->dev_opaque;
14944ddc07caSPaolo Bonzini     bs_dest->dev                = bs_src->dev;
14954ddc07caSPaolo Bonzini     bs_dest->buffer_alignment   = bs_src->buffer_alignment;
14964ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
14974ddc07caSPaolo Bonzini 
14984ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
14994ddc07caSPaolo Bonzini 
15004ddc07caSPaolo Bonzini     /* i/o timing parameters */
15014ddc07caSPaolo Bonzini     bs_dest->slice_start        = bs_src->slice_start;
15024ddc07caSPaolo Bonzini     bs_dest->slice_end          = bs_src->slice_end;
15035905fbc9SStefan Hajnoczi     bs_dest->slice_submitted    = bs_src->slice_submitted;
15044ddc07caSPaolo Bonzini     bs_dest->io_limits          = bs_src->io_limits;
15054ddc07caSPaolo Bonzini     bs_dest->throttled_reqs     = bs_src->throttled_reqs;
15064ddc07caSPaolo Bonzini     bs_dest->block_timer        = bs_src->block_timer;
15074ddc07caSPaolo Bonzini     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
15084ddc07caSPaolo Bonzini 
15094ddc07caSPaolo Bonzini     /* r/w error */
15104ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
15114ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
15124ddc07caSPaolo Bonzini 
15134ddc07caSPaolo Bonzini     /* i/o status */
15144ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
15154ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
15164ddc07caSPaolo Bonzini 
15174ddc07caSPaolo Bonzini     /* dirty bitmap */
15184ddc07caSPaolo Bonzini     bs_dest->dirty_bitmap       = bs_src->dirty_bitmap;
15194ddc07caSPaolo Bonzini 
15204ddc07caSPaolo Bonzini     /* job */
15214ddc07caSPaolo Bonzini     bs_dest->in_use             = bs_src->in_use;
15224ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
15234ddc07caSPaolo Bonzini 
15244ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
15254ddc07caSPaolo Bonzini     pstrcpy(bs_dest->device_name, sizeof(bs_dest->device_name),
15264ddc07caSPaolo Bonzini             bs_src->device_name);
15274ddc07caSPaolo Bonzini     bs_dest->list = bs_src->list;
15284ddc07caSPaolo Bonzini }
15294ddc07caSPaolo Bonzini 
15304ddc07caSPaolo Bonzini /*
15314ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
15324ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
15334ddc07caSPaolo Bonzini  * actually attached to a device.
15344ddc07caSPaolo Bonzini  *
15354ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
15364ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
15374ddc07caSPaolo Bonzini  *
15384ddc07caSPaolo Bonzini  * bs_new is required to be anonymous.
15394ddc07caSPaolo Bonzini  *
15404ddc07caSPaolo Bonzini  * This function does not create any image files.
15414ddc07caSPaolo Bonzini  */
15424ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
15434ddc07caSPaolo Bonzini {
15444ddc07caSPaolo Bonzini     BlockDriverState tmp;
15454ddc07caSPaolo Bonzini 
15464ddc07caSPaolo Bonzini     /* bs_new must be anonymous and shouldn't have anything fancy enabled */
15474ddc07caSPaolo Bonzini     assert(bs_new->device_name[0] == '\0');
15484ddc07caSPaolo Bonzini     assert(bs_new->dirty_bitmap == NULL);
15494ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
15504ddc07caSPaolo Bonzini     assert(bs_new->dev == NULL);
15514ddc07caSPaolo Bonzini     assert(bs_new->in_use == 0);
15524ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
15534ddc07caSPaolo Bonzini     assert(bs_new->block_timer == NULL);
15544ddc07caSPaolo Bonzini 
15554ddc07caSPaolo Bonzini     tmp = *bs_new;
15564ddc07caSPaolo Bonzini     *bs_new = *bs_old;
15574ddc07caSPaolo Bonzini     *bs_old = tmp;
15584ddc07caSPaolo Bonzini 
15594ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
15604ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
15614ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
15624ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
15634ddc07caSPaolo Bonzini 
15644ddc07caSPaolo Bonzini     /* bs_new shouldn't be in bdrv_states even after the swap!  */
15654ddc07caSPaolo Bonzini     assert(bs_new->device_name[0] == '\0');
15664ddc07caSPaolo Bonzini 
15674ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
15684ddc07caSPaolo Bonzini     assert(bs_new->dev == NULL);
15694ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
15704ddc07caSPaolo Bonzini     assert(bs_new->in_use == 0);
15714ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
15724ddc07caSPaolo Bonzini     assert(bs_new->block_timer == NULL);
15734ddc07caSPaolo Bonzini 
15744ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
15754ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
15764ddc07caSPaolo Bonzini }
15774ddc07caSPaolo Bonzini 
15788802d1fdSJeff Cody /*
15798802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
15808802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
15818802d1fdSJeff Cody  *
15828802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
15838802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
15848802d1fdSJeff Cody  *
1585f6801b83SJeff Cody  * bs_new is required to be anonymous.
1586f6801b83SJeff Cody  *
15878802d1fdSJeff Cody  * This function does not create any image files.
15888802d1fdSJeff Cody  */
15898802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
15908802d1fdSJeff Cody {
15914ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
15928802d1fdSJeff Cody 
15938802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
15948802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
15954ddc07caSPaolo Bonzini     bs_top->backing_hd = bs_new;
15964ddc07caSPaolo Bonzini     bs_top->open_flags &= ~BDRV_O_NO_BACKING;
15974ddc07caSPaolo Bonzini     pstrcpy(bs_top->backing_file, sizeof(bs_top->backing_file),
15984ddc07caSPaolo Bonzini             bs_new->filename);
15994ddc07caSPaolo Bonzini     pstrcpy(bs_top->backing_format, sizeof(bs_top->backing_format),
16004ddc07caSPaolo Bonzini             bs_new->drv ? bs_new->drv->format_name : "");
16018802d1fdSJeff Cody }
16028802d1fdSJeff Cody 
1603b338082bSbellard void bdrv_delete(BlockDriverState *bs)
1604b338082bSbellard {
1605fa879d62SMarkus Armbruster     assert(!bs->dev);
16063e914655SPaolo Bonzini     assert(!bs->job);
16073e914655SPaolo Bonzini     assert(!bs->in_use);
160818846deeSMarkus Armbruster 
1609*e1b5c52eSStefan Hajnoczi     bdrv_close(bs);
1610*e1b5c52eSStefan Hajnoczi 
16111b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
1612d22b2f41SRyan Harper     bdrv_make_anon(bs);
161334c6f050Saurel32 
16147267c094SAnthony Liguori     g_free(bs);
1615fc01f7e7Sbellard }
1616fc01f7e7Sbellard 
1617fa879d62SMarkus Armbruster int bdrv_attach_dev(BlockDriverState *bs, void *dev)
1618fa879d62SMarkus Armbruster /* TODO change to DeviceState *dev when all users are qdevified */
161918846deeSMarkus Armbruster {
1620fa879d62SMarkus Armbruster     if (bs->dev) {
162118846deeSMarkus Armbruster         return -EBUSY;
162218846deeSMarkus Armbruster     }
1623fa879d62SMarkus Armbruster     bs->dev = dev;
162428a7282aSLuiz Capitulino     bdrv_iostatus_reset(bs);
162518846deeSMarkus Armbruster     return 0;
162618846deeSMarkus Armbruster }
162718846deeSMarkus Armbruster 
1628fa879d62SMarkus Armbruster /* TODO qdevified devices don't use this, remove when devices are qdevified */
1629fa879d62SMarkus Armbruster void bdrv_attach_dev_nofail(BlockDriverState *bs, void *dev)
163018846deeSMarkus Armbruster {
1631fa879d62SMarkus Armbruster     if (bdrv_attach_dev(bs, dev) < 0) {
1632fa879d62SMarkus Armbruster         abort();
1633fa879d62SMarkus Armbruster     }
1634fa879d62SMarkus Armbruster }
1635fa879d62SMarkus Armbruster 
1636fa879d62SMarkus Armbruster void bdrv_detach_dev(BlockDriverState *bs, void *dev)
1637fa879d62SMarkus Armbruster /* TODO change to DeviceState *dev when all users are qdevified */
1638fa879d62SMarkus Armbruster {
1639fa879d62SMarkus Armbruster     assert(bs->dev == dev);
1640fa879d62SMarkus Armbruster     bs->dev = NULL;
16410e49de52SMarkus Armbruster     bs->dev_ops = NULL;
16420e49de52SMarkus Armbruster     bs->dev_opaque = NULL;
164329e05f20SMarkus Armbruster     bs->buffer_alignment = 512;
164418846deeSMarkus Armbruster }
164518846deeSMarkus Armbruster 
1646fa879d62SMarkus Armbruster /* TODO change to return DeviceState * when all users are qdevified */
1647fa879d62SMarkus Armbruster void *bdrv_get_attached_dev(BlockDriverState *bs)
164818846deeSMarkus Armbruster {
1649fa879d62SMarkus Armbruster     return bs->dev;
165018846deeSMarkus Armbruster }
165118846deeSMarkus Armbruster 
16520e49de52SMarkus Armbruster void bdrv_set_dev_ops(BlockDriverState *bs, const BlockDevOps *ops,
16530e49de52SMarkus Armbruster                       void *opaque)
16540e49de52SMarkus Armbruster {
16550e49de52SMarkus Armbruster     bs->dev_ops = ops;
16560e49de52SMarkus Armbruster     bs->dev_opaque = opaque;
16570e49de52SMarkus Armbruster }
16580e49de52SMarkus Armbruster 
165932c81a4aSPaolo Bonzini void bdrv_emit_qmp_error_event(const BlockDriverState *bdrv,
166032c81a4aSPaolo Bonzini                                enum MonitorEvent ev,
16611ceee0d5SPaolo Bonzini                                BlockErrorAction action, bool is_read)
1662329c0a48SLuiz Capitulino {
1663329c0a48SLuiz Capitulino     QObject *data;
1664329c0a48SLuiz Capitulino     const char *action_str;
1665329c0a48SLuiz Capitulino 
1666329c0a48SLuiz Capitulino     switch (action) {
1667329c0a48SLuiz Capitulino     case BDRV_ACTION_REPORT:
1668329c0a48SLuiz Capitulino         action_str = "report";
1669329c0a48SLuiz Capitulino         break;
1670329c0a48SLuiz Capitulino     case BDRV_ACTION_IGNORE:
1671329c0a48SLuiz Capitulino         action_str = "ignore";
1672329c0a48SLuiz Capitulino         break;
1673329c0a48SLuiz Capitulino     case BDRV_ACTION_STOP:
1674329c0a48SLuiz Capitulino         action_str = "stop";
1675329c0a48SLuiz Capitulino         break;
1676329c0a48SLuiz Capitulino     default:
1677329c0a48SLuiz Capitulino         abort();
1678329c0a48SLuiz Capitulino     }
1679329c0a48SLuiz Capitulino 
1680329c0a48SLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
1681329c0a48SLuiz Capitulino                               bdrv->device_name,
1682329c0a48SLuiz Capitulino                               action_str,
1683329c0a48SLuiz Capitulino                               is_read ? "read" : "write");
168432c81a4aSPaolo Bonzini     monitor_protocol_event(ev, data);
1685329c0a48SLuiz Capitulino 
1686329c0a48SLuiz Capitulino     qobject_decref(data);
1687329c0a48SLuiz Capitulino }
1688329c0a48SLuiz Capitulino 
16896f382ed2SLuiz Capitulino static void bdrv_emit_qmp_eject_event(BlockDriverState *bs, bool ejected)
16906f382ed2SLuiz Capitulino {
16916f382ed2SLuiz Capitulino     QObject *data;
16926f382ed2SLuiz Capitulino 
16936f382ed2SLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'tray-open': %i }",
16946f382ed2SLuiz Capitulino                               bdrv_get_device_name(bs), ejected);
16956f382ed2SLuiz Capitulino     monitor_protocol_event(QEVENT_DEVICE_TRAY_MOVED, data);
16966f382ed2SLuiz Capitulino 
16976f382ed2SLuiz Capitulino     qobject_decref(data);
16986f382ed2SLuiz Capitulino }
16996f382ed2SLuiz Capitulino 
17007d4b4ba5SMarkus Armbruster static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load)
17010e49de52SMarkus Armbruster {
1702145feb17SMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->change_media_cb) {
17036f382ed2SLuiz Capitulino         bool tray_was_closed = !bdrv_dev_is_tray_open(bs);
17047d4b4ba5SMarkus Armbruster         bs->dev_ops->change_media_cb(bs->dev_opaque, load);
17056f382ed2SLuiz Capitulino         if (tray_was_closed) {
17066f382ed2SLuiz Capitulino             /* tray open */
17076f382ed2SLuiz Capitulino             bdrv_emit_qmp_eject_event(bs, true);
17086f382ed2SLuiz Capitulino         }
17096f382ed2SLuiz Capitulino         if (load) {
17106f382ed2SLuiz Capitulino             /* tray close */
17116f382ed2SLuiz Capitulino             bdrv_emit_qmp_eject_event(bs, false);
17126f382ed2SLuiz Capitulino         }
1713145feb17SMarkus Armbruster     }
1714145feb17SMarkus Armbruster }
1715145feb17SMarkus Armbruster 
17162c6942faSMarkus Armbruster bool bdrv_dev_has_removable_media(BlockDriverState *bs)
17172c6942faSMarkus Armbruster {
17182c6942faSMarkus Armbruster     return !bs->dev || (bs->dev_ops && bs->dev_ops->change_media_cb);
17192c6942faSMarkus Armbruster }
17202c6942faSMarkus Armbruster 
1721025ccaa7SPaolo Bonzini void bdrv_dev_eject_request(BlockDriverState *bs, bool force)
1722025ccaa7SPaolo Bonzini {
1723025ccaa7SPaolo Bonzini     if (bs->dev_ops && bs->dev_ops->eject_request_cb) {
1724025ccaa7SPaolo Bonzini         bs->dev_ops->eject_request_cb(bs->dev_opaque, force);
1725025ccaa7SPaolo Bonzini     }
1726025ccaa7SPaolo Bonzini }
1727025ccaa7SPaolo Bonzini 
1728e4def80bSMarkus Armbruster bool bdrv_dev_is_tray_open(BlockDriverState *bs)
1729e4def80bSMarkus Armbruster {
1730e4def80bSMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->is_tray_open) {
1731e4def80bSMarkus Armbruster         return bs->dev_ops->is_tray_open(bs->dev_opaque);
1732e4def80bSMarkus Armbruster     }
1733e4def80bSMarkus Armbruster     return false;
1734e4def80bSMarkus Armbruster }
1735e4def80bSMarkus Armbruster 
1736145feb17SMarkus Armbruster static void bdrv_dev_resize_cb(BlockDriverState *bs)
1737145feb17SMarkus Armbruster {
1738145feb17SMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->resize_cb) {
1739145feb17SMarkus Armbruster         bs->dev_ops->resize_cb(bs->dev_opaque);
17400e49de52SMarkus Armbruster     }
17410e49de52SMarkus Armbruster }
17420e49de52SMarkus Armbruster 
1743f107639aSMarkus Armbruster bool bdrv_dev_is_medium_locked(BlockDriverState *bs)
1744f107639aSMarkus Armbruster {
1745f107639aSMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->is_medium_locked) {
1746f107639aSMarkus Armbruster         return bs->dev_ops->is_medium_locked(bs->dev_opaque);
1747f107639aSMarkus Armbruster     }
1748f107639aSMarkus Armbruster     return false;
1749f107639aSMarkus Armbruster }
1750f107639aSMarkus Armbruster 
1751e97fc193Saliguori /*
1752e97fc193Saliguori  * Run consistency checks on an image
1753e97fc193Saliguori  *
1754e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
1755a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
1756e076f338SKevin Wolf  * check are stored in res.
1757e97fc193Saliguori  */
17584534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
1759e97fc193Saliguori {
1760e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
1761e97fc193Saliguori         return -ENOTSUP;
1762e97fc193Saliguori     }
1763e97fc193Saliguori 
1764e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
17654534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
1766e97fc193Saliguori }
1767e97fc193Saliguori 
17688a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
17698a426614SKevin Wolf 
177033e3963eSbellard /* commit COW file into the raw image */
177133e3963eSbellard int bdrv_commit(BlockDriverState *bs)
177233e3963eSbellard {
177319cb3738Sbellard     BlockDriver *drv = bs->drv;
17748a426614SKevin Wolf     int64_t sector, total_sectors;
17758a426614SKevin Wolf     int n, ro, open_flags;
17760bce597dSJeff Cody     int ret = 0;
17778a426614SKevin Wolf     uint8_t *buf;
1778c2cba3d9SJim Meyering     char filename[PATH_MAX];
177933e3963eSbellard 
178019cb3738Sbellard     if (!drv)
178119cb3738Sbellard         return -ENOMEDIUM;
178233e3963eSbellard 
17834dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
17844dca4b63SNaphtali Sprei         return -ENOTSUP;
17854dca4b63SNaphtali Sprei     }
17864dca4b63SNaphtali Sprei 
17872d3735d3SStefan Hajnoczi     if (bdrv_in_use(bs) || bdrv_in_use(bs->backing_hd)) {
17882d3735d3SStefan Hajnoczi         return -EBUSY;
17892d3735d3SStefan Hajnoczi     }
17902d3735d3SStefan Hajnoczi 
17914dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
1792c2cba3d9SJim Meyering     /* Use pstrcpy (not strncpy): filename must be NUL-terminated. */
1793c2cba3d9SJim Meyering     pstrcpy(filename, sizeof(filename), bs->backing_hd->filename);
17944dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
17954dca4b63SNaphtali Sprei 
17964dca4b63SNaphtali Sprei     if (ro) {
17970bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
17980bce597dSJeff Cody             return -EACCES;
17994dca4b63SNaphtali Sprei         }
1800ea2384d3Sbellard     }
1801ea2384d3Sbellard 
18026ea44308SJan Kiszka     total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
18037267c094SAnthony Liguori     buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
18048a426614SKevin Wolf 
18058a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
180605c4af54SStefan Hajnoczi         if (bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
18078a426614SKevin Wolf 
18088a426614SKevin Wolf             if (bdrv_read(bs, sector, buf, n) != 0) {
18094dca4b63SNaphtali Sprei                 ret = -EIO;
18104dca4b63SNaphtali Sprei                 goto ro_cleanup;
181133e3963eSbellard             }
181233e3963eSbellard 
18138a426614SKevin Wolf             if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
18144dca4b63SNaphtali Sprei                 ret = -EIO;
18154dca4b63SNaphtali Sprei                 goto ro_cleanup;
181633e3963eSbellard             }
181733e3963eSbellard         }
181833e3963eSbellard     }
181995389c86Sbellard 
18201d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
18211d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
18221d44952fSChristoph Hellwig         bdrv_flush(bs);
18231d44952fSChristoph Hellwig     }
182495389c86Sbellard 
18253f5075aeSChristoph Hellwig     /*
18263f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
18273f5075aeSChristoph Hellwig      * stable on disk.
18283f5075aeSChristoph Hellwig      */
18293f5075aeSChristoph Hellwig     if (bs->backing_hd)
18303f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
18314dca4b63SNaphtali Sprei 
18324dca4b63SNaphtali Sprei ro_cleanup:
18337267c094SAnthony Liguori     g_free(buf);
18344dca4b63SNaphtali Sprei 
18354dca4b63SNaphtali Sprei     if (ro) {
18360bce597dSJeff Cody         /* ignoring error return here */
18370bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
18384dca4b63SNaphtali Sprei     }
18394dca4b63SNaphtali Sprei 
18401d44952fSChristoph Hellwig     return ret;
184133e3963eSbellard }
184233e3963eSbellard 
1843e8877497SStefan Hajnoczi int bdrv_commit_all(void)
18446ab4b5abSMarkus Armbruster {
18456ab4b5abSMarkus Armbruster     BlockDriverState *bs;
18466ab4b5abSMarkus Armbruster 
18476ab4b5abSMarkus Armbruster     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1848272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
1849e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
1850e8877497SStefan Hajnoczi             if (ret < 0) {
1851e8877497SStefan Hajnoczi                 return ret;
18526ab4b5abSMarkus Armbruster             }
18536ab4b5abSMarkus Armbruster         }
1854272d2d8eSJeff Cody     }
1855e8877497SStefan Hajnoczi     return 0;
1856e8877497SStefan Hajnoczi }
18576ab4b5abSMarkus Armbruster 
1858dbffbdcfSStefan Hajnoczi /**
1859dbffbdcfSStefan Hajnoczi  * Remove an active request from the tracked requests list
1860dbffbdcfSStefan Hajnoczi  *
1861dbffbdcfSStefan Hajnoczi  * This function should be called when a tracked request is completing.
1862dbffbdcfSStefan Hajnoczi  */
1863dbffbdcfSStefan Hajnoczi static void tracked_request_end(BdrvTrackedRequest *req)
1864dbffbdcfSStefan Hajnoczi {
1865dbffbdcfSStefan Hajnoczi     QLIST_REMOVE(req, list);
1866f4658285SStefan Hajnoczi     qemu_co_queue_restart_all(&req->wait_queue);
1867dbffbdcfSStefan Hajnoczi }
1868dbffbdcfSStefan Hajnoczi 
1869dbffbdcfSStefan Hajnoczi /**
1870dbffbdcfSStefan Hajnoczi  * Add an active request to the tracked requests list
1871dbffbdcfSStefan Hajnoczi  */
1872dbffbdcfSStefan Hajnoczi static void tracked_request_begin(BdrvTrackedRequest *req,
1873dbffbdcfSStefan Hajnoczi                                   BlockDriverState *bs,
1874dbffbdcfSStefan Hajnoczi                                   int64_t sector_num,
1875dbffbdcfSStefan Hajnoczi                                   int nb_sectors, bool is_write)
1876dbffbdcfSStefan Hajnoczi {
1877dbffbdcfSStefan Hajnoczi     *req = (BdrvTrackedRequest){
1878dbffbdcfSStefan Hajnoczi         .bs = bs,
1879dbffbdcfSStefan Hajnoczi         .sector_num = sector_num,
1880dbffbdcfSStefan Hajnoczi         .nb_sectors = nb_sectors,
1881dbffbdcfSStefan Hajnoczi         .is_write = is_write,
18825f8b6491SStefan Hajnoczi         .co = qemu_coroutine_self(),
1883dbffbdcfSStefan Hajnoczi     };
1884dbffbdcfSStefan Hajnoczi 
1885f4658285SStefan Hajnoczi     qemu_co_queue_init(&req->wait_queue);
1886f4658285SStefan Hajnoczi 
1887dbffbdcfSStefan Hajnoczi     QLIST_INSERT_HEAD(&bs->tracked_requests, req, list);
1888dbffbdcfSStefan Hajnoczi }
1889dbffbdcfSStefan Hajnoczi 
1890d83947acSStefan Hajnoczi /**
1891d83947acSStefan Hajnoczi  * Round a region to cluster boundaries
1892d83947acSStefan Hajnoczi  */
1893343bded4SPaolo Bonzini void bdrv_round_to_clusters(BlockDriverState *bs,
1894d83947acSStefan Hajnoczi                             int64_t sector_num, int nb_sectors,
1895d83947acSStefan Hajnoczi                             int64_t *cluster_sector_num,
1896d83947acSStefan Hajnoczi                             int *cluster_nb_sectors)
1897d83947acSStefan Hajnoczi {
1898d83947acSStefan Hajnoczi     BlockDriverInfo bdi;
1899d83947acSStefan Hajnoczi 
1900d83947acSStefan Hajnoczi     if (bdrv_get_info(bs, &bdi) < 0 || bdi.cluster_size == 0) {
1901d83947acSStefan Hajnoczi         *cluster_sector_num = sector_num;
1902d83947acSStefan Hajnoczi         *cluster_nb_sectors = nb_sectors;
1903d83947acSStefan Hajnoczi     } else {
1904d83947acSStefan Hajnoczi         int64_t c = bdi.cluster_size / BDRV_SECTOR_SIZE;
1905d83947acSStefan Hajnoczi         *cluster_sector_num = QEMU_ALIGN_DOWN(sector_num, c);
1906d83947acSStefan Hajnoczi         *cluster_nb_sectors = QEMU_ALIGN_UP(sector_num - *cluster_sector_num +
1907d83947acSStefan Hajnoczi                                             nb_sectors, c);
1908d83947acSStefan Hajnoczi     }
1909d83947acSStefan Hajnoczi }
1910d83947acSStefan Hajnoczi 
1911f4658285SStefan Hajnoczi static bool tracked_request_overlaps(BdrvTrackedRequest *req,
1912f4658285SStefan Hajnoczi                                      int64_t sector_num, int nb_sectors) {
1913d83947acSStefan Hajnoczi     /*        aaaa   bbbb */
1914d83947acSStefan Hajnoczi     if (sector_num >= req->sector_num + req->nb_sectors) {
1915d83947acSStefan Hajnoczi         return false;
1916d83947acSStefan Hajnoczi     }
1917d83947acSStefan Hajnoczi     /* bbbb   aaaa        */
1918d83947acSStefan Hajnoczi     if (req->sector_num >= sector_num + nb_sectors) {
1919d83947acSStefan Hajnoczi         return false;
1920d83947acSStefan Hajnoczi     }
1921d83947acSStefan Hajnoczi     return true;
1922f4658285SStefan Hajnoczi }
1923f4658285SStefan Hajnoczi 
1924f4658285SStefan Hajnoczi static void coroutine_fn wait_for_overlapping_requests(BlockDriverState *bs,
1925f4658285SStefan Hajnoczi         int64_t sector_num, int nb_sectors)
1926f4658285SStefan Hajnoczi {
1927f4658285SStefan Hajnoczi     BdrvTrackedRequest *req;
1928d83947acSStefan Hajnoczi     int64_t cluster_sector_num;
1929d83947acSStefan Hajnoczi     int cluster_nb_sectors;
1930f4658285SStefan Hajnoczi     bool retry;
1931f4658285SStefan Hajnoczi 
1932d83947acSStefan Hajnoczi     /* If we touch the same cluster it counts as an overlap.  This guarantees
1933d83947acSStefan Hajnoczi      * that allocating writes will be serialized and not race with each other
1934d83947acSStefan Hajnoczi      * for the same cluster.  For example, in copy-on-read it ensures that the
1935d83947acSStefan Hajnoczi      * CoR read and write operations are atomic and guest writes cannot
1936d83947acSStefan Hajnoczi      * interleave between them.
1937d83947acSStefan Hajnoczi      */
1938343bded4SPaolo Bonzini     bdrv_round_to_clusters(bs, sector_num, nb_sectors,
1939d83947acSStefan Hajnoczi                            &cluster_sector_num, &cluster_nb_sectors);
1940d83947acSStefan Hajnoczi 
1941f4658285SStefan Hajnoczi     do {
1942f4658285SStefan Hajnoczi         retry = false;
1943f4658285SStefan Hajnoczi         QLIST_FOREACH(req, &bs->tracked_requests, list) {
1944d83947acSStefan Hajnoczi             if (tracked_request_overlaps(req, cluster_sector_num,
1945d83947acSStefan Hajnoczi                                          cluster_nb_sectors)) {
19465f8b6491SStefan Hajnoczi                 /* Hitting this means there was a reentrant request, for
19475f8b6491SStefan Hajnoczi                  * example, a block driver issuing nested requests.  This must
19485f8b6491SStefan Hajnoczi                  * never happen since it means deadlock.
19495f8b6491SStefan Hajnoczi                  */
19505f8b6491SStefan Hajnoczi                 assert(qemu_coroutine_self() != req->co);
19515f8b6491SStefan Hajnoczi 
1952f4658285SStefan Hajnoczi                 qemu_co_queue_wait(&req->wait_queue);
1953f4658285SStefan Hajnoczi                 retry = true;
1954f4658285SStefan Hajnoczi                 break;
1955f4658285SStefan Hajnoczi             }
1956f4658285SStefan Hajnoczi         }
1957f4658285SStefan Hajnoczi     } while (retry);
1958f4658285SStefan Hajnoczi }
1959f4658285SStefan Hajnoczi 
1960756e6736SKevin Wolf /*
1961756e6736SKevin Wolf  * Return values:
1962756e6736SKevin Wolf  * 0        - success
1963756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
1964756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
1965756e6736SKevin Wolf  *            image file header
1966756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
1967756e6736SKevin Wolf  */
1968756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
1969756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
1970756e6736SKevin Wolf {
1971756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
1972469ef350SPaolo Bonzini     int ret;
1973756e6736SKevin Wolf 
19745f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
19755f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
19765f377794SPaolo Bonzini         return -EINVAL;
19775f377794SPaolo Bonzini     }
19785f377794SPaolo Bonzini 
1979756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
1980469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
1981756e6736SKevin Wolf     } else {
1982469ef350SPaolo Bonzini         ret = -ENOTSUP;
1983756e6736SKevin Wolf     }
1984469ef350SPaolo Bonzini 
1985469ef350SPaolo Bonzini     if (ret == 0) {
1986469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
1987469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
1988469ef350SPaolo Bonzini     }
1989469ef350SPaolo Bonzini     return ret;
1990756e6736SKevin Wolf }
1991756e6736SKevin Wolf 
19926ebdcee2SJeff Cody /*
19936ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
19946ebdcee2SJeff Cody  *
19956ebdcee2SJeff Cody  * active is the current topmost image.
19966ebdcee2SJeff Cody  *
19976ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
19986ebdcee2SJeff Cody  * or if active == bs.
19996ebdcee2SJeff Cody  */
20006ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
20016ebdcee2SJeff Cody                                     BlockDriverState *bs)
20026ebdcee2SJeff Cody {
20036ebdcee2SJeff Cody     BlockDriverState *overlay = NULL;
20046ebdcee2SJeff Cody     BlockDriverState *intermediate;
20056ebdcee2SJeff Cody 
20066ebdcee2SJeff Cody     assert(active != NULL);
20076ebdcee2SJeff Cody     assert(bs != NULL);
20086ebdcee2SJeff Cody 
20096ebdcee2SJeff Cody     /* if bs is the same as active, then by definition it has no overlay
20106ebdcee2SJeff Cody      */
20116ebdcee2SJeff Cody     if (active == bs) {
20126ebdcee2SJeff Cody         return NULL;
20136ebdcee2SJeff Cody     }
20146ebdcee2SJeff Cody 
20156ebdcee2SJeff Cody     intermediate = active;
20166ebdcee2SJeff Cody     while (intermediate->backing_hd) {
20176ebdcee2SJeff Cody         if (intermediate->backing_hd == bs) {
20186ebdcee2SJeff Cody             overlay = intermediate;
20196ebdcee2SJeff Cody             break;
20206ebdcee2SJeff Cody         }
20216ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
20226ebdcee2SJeff Cody     }
20236ebdcee2SJeff Cody 
20246ebdcee2SJeff Cody     return overlay;
20256ebdcee2SJeff Cody }
20266ebdcee2SJeff Cody 
20276ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
20286ebdcee2SJeff Cody     BlockDriverState *bs;
20296ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
20306ebdcee2SJeff Cody } BlkIntermediateStates;
20316ebdcee2SJeff Cody 
20326ebdcee2SJeff Cody 
20336ebdcee2SJeff Cody /*
20346ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
20356ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
20366ebdcee2SJeff Cody  *
20376ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
20386ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
20396ebdcee2SJeff Cody  *
20406ebdcee2SJeff Cody  * E.g., this will convert the following chain:
20416ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
20426ebdcee2SJeff Cody  *
20436ebdcee2SJeff Cody  * to
20446ebdcee2SJeff Cody  *
20456ebdcee2SJeff Cody  * bottom <- base <- active
20466ebdcee2SJeff Cody  *
20476ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
20486ebdcee2SJeff Cody  *
20496ebdcee2SJeff Cody  * base <- intermediate <- top <- active
20506ebdcee2SJeff Cody  *
20516ebdcee2SJeff Cody  * to
20526ebdcee2SJeff Cody  *
20536ebdcee2SJeff Cody  * base <- active
20546ebdcee2SJeff Cody  *
20556ebdcee2SJeff Cody  * Error conditions:
20566ebdcee2SJeff Cody  *  if active == top, that is considered an error
20576ebdcee2SJeff Cody  *
20586ebdcee2SJeff Cody  */
20596ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
20606ebdcee2SJeff Cody                            BlockDriverState *base)
20616ebdcee2SJeff Cody {
20626ebdcee2SJeff Cody     BlockDriverState *intermediate;
20636ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
20646ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
20656ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
20666ebdcee2SJeff Cody     int ret = -EIO;
20676ebdcee2SJeff Cody 
20686ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
20696ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
20706ebdcee2SJeff Cody 
20716ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
20726ebdcee2SJeff Cody         goto exit;
20736ebdcee2SJeff Cody     }
20746ebdcee2SJeff Cody 
20756ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
20766ebdcee2SJeff Cody 
20776ebdcee2SJeff Cody     if (new_top_bs == NULL) {
20786ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
20796ebdcee2SJeff Cody         goto exit;
20806ebdcee2SJeff Cody     }
20816ebdcee2SJeff Cody 
20826ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
20836ebdcee2SJeff Cody      * to do, no intermediate images */
20846ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
20856ebdcee2SJeff Cody         ret = 0;
20866ebdcee2SJeff Cody         goto exit;
20876ebdcee2SJeff Cody     }
20886ebdcee2SJeff Cody 
20896ebdcee2SJeff Cody     intermediate = top;
20906ebdcee2SJeff Cody 
20916ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
20926ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
20936ebdcee2SJeff Cody      */
20946ebdcee2SJeff Cody     while (intermediate) {
20956ebdcee2SJeff Cody         intermediate_state = g_malloc0(sizeof(BlkIntermediateStates));
20966ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
20976ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
20986ebdcee2SJeff Cody 
20996ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
21006ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
21016ebdcee2SJeff Cody             break;
21026ebdcee2SJeff Cody         }
21036ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
21046ebdcee2SJeff Cody     }
21056ebdcee2SJeff Cody     if (base_bs == NULL) {
21066ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
21076ebdcee2SJeff Cody          * unravel everything, and exit with error */
21086ebdcee2SJeff Cody         goto exit;
21096ebdcee2SJeff Cody     }
21106ebdcee2SJeff Cody 
21116ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
21126ebdcee2SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, base_bs->filename,
21136ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
21146ebdcee2SJeff Cody     if (ret) {
21156ebdcee2SJeff Cody         goto exit;
21166ebdcee2SJeff Cody     }
21176ebdcee2SJeff Cody     new_top_bs->backing_hd = base_bs;
21186ebdcee2SJeff Cody 
21196ebdcee2SJeff Cody 
21206ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
21216ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
21226ebdcee2SJeff Cody         intermediate_state->bs->backing_hd = NULL;
21236ebdcee2SJeff Cody         bdrv_delete(intermediate_state->bs);
21246ebdcee2SJeff Cody     }
21256ebdcee2SJeff Cody     ret = 0;
21266ebdcee2SJeff Cody 
21276ebdcee2SJeff Cody exit:
21286ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
21296ebdcee2SJeff Cody         g_free(intermediate_state);
21306ebdcee2SJeff Cody     }
21316ebdcee2SJeff Cody     return ret;
21326ebdcee2SJeff Cody }
21336ebdcee2SJeff Cody 
21346ebdcee2SJeff Cody 
213571d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
213671d0770cSaliguori                                    size_t size)
213771d0770cSaliguori {
213871d0770cSaliguori     int64_t len;
213971d0770cSaliguori 
214071d0770cSaliguori     if (!bdrv_is_inserted(bs))
214171d0770cSaliguori         return -ENOMEDIUM;
214271d0770cSaliguori 
214371d0770cSaliguori     if (bs->growable)
214471d0770cSaliguori         return 0;
214571d0770cSaliguori 
214671d0770cSaliguori     len = bdrv_getlength(bs);
214771d0770cSaliguori 
2148fbb7b4e0SKevin Wolf     if (offset < 0)
2149fbb7b4e0SKevin Wolf         return -EIO;
2150fbb7b4e0SKevin Wolf 
2151fbb7b4e0SKevin Wolf     if ((offset > len) || (len - offset < size))
215271d0770cSaliguori         return -EIO;
215371d0770cSaliguori 
215471d0770cSaliguori     return 0;
215571d0770cSaliguori }
215671d0770cSaliguori 
215771d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
215871d0770cSaliguori                               int nb_sectors)
215971d0770cSaliguori {
2160eb5a3165SJes Sorensen     return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
2161eb5a3165SJes Sorensen                                    nb_sectors * BDRV_SECTOR_SIZE);
216271d0770cSaliguori }
216371d0770cSaliguori 
21641c9805a3SStefan Hajnoczi typedef struct RwCo {
21651c9805a3SStefan Hajnoczi     BlockDriverState *bs;
21661c9805a3SStefan Hajnoczi     int64_t sector_num;
21671c9805a3SStefan Hajnoczi     int nb_sectors;
21681c9805a3SStefan Hajnoczi     QEMUIOVector *qiov;
21691c9805a3SStefan Hajnoczi     bool is_write;
21701c9805a3SStefan Hajnoczi     int ret;
21714105eaaaSPeter Lieven     BdrvRequestFlags flags;
21721c9805a3SStefan Hajnoczi } RwCo;
21731c9805a3SStefan Hajnoczi 
21741c9805a3SStefan Hajnoczi static void coroutine_fn bdrv_rw_co_entry(void *opaque)
2175fc01f7e7Sbellard {
21761c9805a3SStefan Hajnoczi     RwCo *rwco = opaque;
2177fc01f7e7Sbellard 
21781c9805a3SStefan Hajnoczi     if (!rwco->is_write) {
21791c9805a3SStefan Hajnoczi         rwco->ret = bdrv_co_do_readv(rwco->bs, rwco->sector_num,
21804105eaaaSPeter Lieven                                      rwco->nb_sectors, rwco->qiov,
21814105eaaaSPeter Lieven                                      rwco->flags);
21821c9805a3SStefan Hajnoczi     } else {
21831c9805a3SStefan Hajnoczi         rwco->ret = bdrv_co_do_writev(rwco->bs, rwco->sector_num,
21844105eaaaSPeter Lieven                                       rwco->nb_sectors, rwco->qiov,
21854105eaaaSPeter Lieven                                       rwco->flags);
21861c9805a3SStefan Hajnoczi     }
21871c9805a3SStefan Hajnoczi }
2188e7a8a783SKevin Wolf 
21891c9805a3SStefan Hajnoczi /*
21908d3b1a2dSKevin Wolf  * Process a vectored synchronous request using coroutines
21911c9805a3SStefan Hajnoczi  */
21928d3b1a2dSKevin Wolf static int bdrv_rwv_co(BlockDriverState *bs, int64_t sector_num,
21934105eaaaSPeter Lieven                        QEMUIOVector *qiov, bool is_write,
21944105eaaaSPeter Lieven                        BdrvRequestFlags flags)
21951c9805a3SStefan Hajnoczi {
21961c9805a3SStefan Hajnoczi     Coroutine *co;
21971c9805a3SStefan Hajnoczi     RwCo rwco = {
21981c9805a3SStefan Hajnoczi         .bs = bs,
21991c9805a3SStefan Hajnoczi         .sector_num = sector_num,
22008d3b1a2dSKevin Wolf         .nb_sectors = qiov->size >> BDRV_SECTOR_BITS,
22018d3b1a2dSKevin Wolf         .qiov = qiov,
22021c9805a3SStefan Hajnoczi         .is_write = is_write,
22031c9805a3SStefan Hajnoczi         .ret = NOT_DONE,
22044105eaaaSPeter Lieven         .flags = flags,
22051c9805a3SStefan Hajnoczi     };
22068d3b1a2dSKevin Wolf     assert((qiov->size & (BDRV_SECTOR_SIZE - 1)) == 0);
22071c9805a3SStefan Hajnoczi 
2208498e386cSZhi Yong Wu     /**
2209498e386cSZhi Yong Wu      * In sync call context, when the vcpu is blocked, this throttling timer
2210498e386cSZhi Yong Wu      * will not fire; so the I/O throttling function has to be disabled here
2211498e386cSZhi Yong Wu      * if it has been enabled.
2212498e386cSZhi Yong Wu      */
2213498e386cSZhi Yong Wu     if (bs->io_limits_enabled) {
2214498e386cSZhi Yong Wu         fprintf(stderr, "Disabling I/O throttling on '%s' due "
2215498e386cSZhi Yong Wu                         "to synchronous I/O.\n", bdrv_get_device_name(bs));
2216498e386cSZhi Yong Wu         bdrv_io_limits_disable(bs);
2217498e386cSZhi Yong Wu     }
2218498e386cSZhi Yong Wu 
22191c9805a3SStefan Hajnoczi     if (qemu_in_coroutine()) {
22201c9805a3SStefan Hajnoczi         /* Fast-path if already in coroutine context */
22211c9805a3SStefan Hajnoczi         bdrv_rw_co_entry(&rwco);
22221c9805a3SStefan Hajnoczi     } else {
22231c9805a3SStefan Hajnoczi         co = qemu_coroutine_create(bdrv_rw_co_entry);
22241c9805a3SStefan Hajnoczi         qemu_coroutine_enter(co, &rwco);
22251c9805a3SStefan Hajnoczi         while (rwco.ret == NOT_DONE) {
22261c9805a3SStefan Hajnoczi             qemu_aio_wait();
22271c9805a3SStefan Hajnoczi         }
22281c9805a3SStefan Hajnoczi     }
22291c9805a3SStefan Hajnoczi     return rwco.ret;
2230e7a8a783SKevin Wolf }
2231e7a8a783SKevin Wolf 
22328d3b1a2dSKevin Wolf /*
22338d3b1a2dSKevin Wolf  * Process a synchronous request using coroutines
22348d3b1a2dSKevin Wolf  */
22358d3b1a2dSKevin Wolf static int bdrv_rw_co(BlockDriverState *bs, int64_t sector_num, uint8_t *buf,
22364105eaaaSPeter Lieven                       int nb_sectors, bool is_write, BdrvRequestFlags flags)
22378d3b1a2dSKevin Wolf {
22388d3b1a2dSKevin Wolf     QEMUIOVector qiov;
22398d3b1a2dSKevin Wolf     struct iovec iov = {
22408d3b1a2dSKevin Wolf         .iov_base = (void *)buf,
22418d3b1a2dSKevin Wolf         .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
22428d3b1a2dSKevin Wolf     };
22438d3b1a2dSKevin Wolf 
22448d3b1a2dSKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
22454105eaaaSPeter Lieven     return bdrv_rwv_co(bs, sector_num, &qiov, is_write, flags);
22468d3b1a2dSKevin Wolf }
22478d3b1a2dSKevin Wolf 
22481c9805a3SStefan Hajnoczi /* return < 0 if error. See bdrv_write() for the return codes */
22491c9805a3SStefan Hajnoczi int bdrv_read(BlockDriverState *bs, int64_t sector_num,
22501c9805a3SStefan Hajnoczi               uint8_t *buf, int nb_sectors)
22511c9805a3SStefan Hajnoczi {
22524105eaaaSPeter Lieven     return bdrv_rw_co(bs, sector_num, buf, nb_sectors, false, 0);
225383f64091Sbellard }
2254fc01f7e7Sbellard 
225507d27a44SMarkus Armbruster /* Just like bdrv_read(), but with I/O throttling temporarily disabled */
225607d27a44SMarkus Armbruster int bdrv_read_unthrottled(BlockDriverState *bs, int64_t sector_num,
225707d27a44SMarkus Armbruster                           uint8_t *buf, int nb_sectors)
225807d27a44SMarkus Armbruster {
225907d27a44SMarkus Armbruster     bool enabled;
226007d27a44SMarkus Armbruster     int ret;
226107d27a44SMarkus Armbruster 
226207d27a44SMarkus Armbruster     enabled = bs->io_limits_enabled;
226307d27a44SMarkus Armbruster     bs->io_limits_enabled = false;
22644e7395e8SPeter Lieven     ret = bdrv_read(bs, sector_num, buf, nb_sectors);
226507d27a44SMarkus Armbruster     bs->io_limits_enabled = enabled;
226607d27a44SMarkus Armbruster     return ret;
226707d27a44SMarkus Armbruster }
226807d27a44SMarkus Armbruster 
226919cb3738Sbellard /* Return < 0 if error. Important errors are:
227019cb3738Sbellard   -EIO         generic I/O error (may happen for all errors)
227119cb3738Sbellard   -ENOMEDIUM   No media inserted.
227219cb3738Sbellard   -EINVAL      Invalid sector number or nb_sectors
227319cb3738Sbellard   -EACCES      Trying to write a read-only device
227419cb3738Sbellard */
2275fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
2276fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
2277fc01f7e7Sbellard {
22784105eaaaSPeter Lieven     return bdrv_rw_co(bs, sector_num, (uint8_t *)buf, nb_sectors, true, 0);
227983f64091Sbellard }
228083f64091Sbellard 
22818d3b1a2dSKevin Wolf int bdrv_writev(BlockDriverState *bs, int64_t sector_num, QEMUIOVector *qiov)
22828d3b1a2dSKevin Wolf {
22834105eaaaSPeter Lieven     return bdrv_rwv_co(bs, sector_num, qiov, true, 0);
22844105eaaaSPeter Lieven }
22854105eaaaSPeter Lieven 
22864105eaaaSPeter Lieven int bdrv_write_zeroes(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
22874105eaaaSPeter Lieven {
22884105eaaaSPeter Lieven     return bdrv_rw_co(bs, sector_num, NULL, nb_sectors, true,
22894105eaaaSPeter Lieven                       BDRV_REQ_ZERO_WRITE);
22908d3b1a2dSKevin Wolf }
22918d3b1a2dSKevin Wolf 
2292eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset,
2293eda578e5Saliguori                void *buf, int count1)
229483f64091Sbellard {
22956ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
229683f64091Sbellard     int len, nb_sectors, count;
229783f64091Sbellard     int64_t sector_num;
22989a8c4cceSKevin Wolf     int ret;
229983f64091Sbellard 
230083f64091Sbellard     count = count1;
230183f64091Sbellard     /* first read to align to sector start */
23026ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
230383f64091Sbellard     if (len > count)
230483f64091Sbellard         len = count;
23056ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
230683f64091Sbellard     if (len > 0) {
23079a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23089a8c4cceSKevin Wolf             return ret;
23096ea44308SJan Kiszka         memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
231083f64091Sbellard         count -= len;
231183f64091Sbellard         if (count == 0)
231283f64091Sbellard             return count1;
231383f64091Sbellard         sector_num++;
231483f64091Sbellard         buf += len;
231583f64091Sbellard     }
231683f64091Sbellard 
231783f64091Sbellard     /* read the sectors "in place" */
23186ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
231983f64091Sbellard     if (nb_sectors > 0) {
23209a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
23219a8c4cceSKevin Wolf             return ret;
232283f64091Sbellard         sector_num += nb_sectors;
23236ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
232483f64091Sbellard         buf += len;
232583f64091Sbellard         count -= len;
232683f64091Sbellard     }
232783f64091Sbellard 
232883f64091Sbellard     /* add data from the last sector */
232983f64091Sbellard     if (count > 0) {
23309a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23319a8c4cceSKevin Wolf             return ret;
233283f64091Sbellard         memcpy(buf, tmp_buf, count);
233383f64091Sbellard     }
233483f64091Sbellard     return count1;
233583f64091Sbellard }
233683f64091Sbellard 
23378d3b1a2dSKevin Wolf int bdrv_pwritev(BlockDriverState *bs, int64_t offset, QEMUIOVector *qiov)
233883f64091Sbellard {
23396ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
234083f64091Sbellard     int len, nb_sectors, count;
234183f64091Sbellard     int64_t sector_num;
23429a8c4cceSKevin Wolf     int ret;
234383f64091Sbellard 
23448d3b1a2dSKevin Wolf     count = qiov->size;
23458d3b1a2dSKevin Wolf 
234683f64091Sbellard     /* first write to align to sector start */
23476ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
234883f64091Sbellard     if (len > count)
234983f64091Sbellard         len = count;
23506ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
235183f64091Sbellard     if (len > 0) {
23529a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23539a8c4cceSKevin Wolf             return ret;
23548d3b1a2dSKevin Wolf         qemu_iovec_to_buf(qiov, 0, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)),
23558d3b1a2dSKevin Wolf                           len);
23569a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
23579a8c4cceSKevin Wolf             return ret;
235883f64091Sbellard         count -= len;
235983f64091Sbellard         if (count == 0)
23608d3b1a2dSKevin Wolf             return qiov->size;
236183f64091Sbellard         sector_num++;
236283f64091Sbellard     }
236383f64091Sbellard 
236483f64091Sbellard     /* write the sectors "in place" */
23656ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
236683f64091Sbellard     if (nb_sectors > 0) {
23678d3b1a2dSKevin Wolf         QEMUIOVector qiov_inplace;
23688d3b1a2dSKevin Wolf 
23698d3b1a2dSKevin Wolf         qemu_iovec_init(&qiov_inplace, qiov->niov);
23708d3b1a2dSKevin Wolf         qemu_iovec_concat(&qiov_inplace, qiov, len,
23718d3b1a2dSKevin Wolf                           nb_sectors << BDRV_SECTOR_BITS);
23728d3b1a2dSKevin Wolf         ret = bdrv_writev(bs, sector_num, &qiov_inplace);
23738d3b1a2dSKevin Wolf         qemu_iovec_destroy(&qiov_inplace);
23748d3b1a2dSKevin Wolf         if (ret < 0) {
23759a8c4cceSKevin Wolf             return ret;
23768d3b1a2dSKevin Wolf         }
23778d3b1a2dSKevin Wolf 
237883f64091Sbellard         sector_num += nb_sectors;
23796ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
238083f64091Sbellard         count -= len;
238183f64091Sbellard     }
238283f64091Sbellard 
238383f64091Sbellard     /* add data from the last sector */
238483f64091Sbellard     if (count > 0) {
23859a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23869a8c4cceSKevin Wolf             return ret;
23878d3b1a2dSKevin Wolf         qemu_iovec_to_buf(qiov, qiov->size - count, tmp_buf, count);
23889a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
23899a8c4cceSKevin Wolf             return ret;
239083f64091Sbellard     }
23918d3b1a2dSKevin Wolf     return qiov->size;
23928d3b1a2dSKevin Wolf }
23938d3b1a2dSKevin Wolf 
23948d3b1a2dSKevin Wolf int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
23958d3b1a2dSKevin Wolf                 const void *buf, int count1)
23968d3b1a2dSKevin Wolf {
23978d3b1a2dSKevin Wolf     QEMUIOVector qiov;
23988d3b1a2dSKevin Wolf     struct iovec iov = {
23998d3b1a2dSKevin Wolf         .iov_base   = (void *) buf,
24008d3b1a2dSKevin Wolf         .iov_len    = count1,
24018d3b1a2dSKevin Wolf     };
24028d3b1a2dSKevin Wolf 
24038d3b1a2dSKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
24048d3b1a2dSKevin Wolf     return bdrv_pwritev(bs, offset, &qiov);
240583f64091Sbellard }
240683f64091Sbellard 
2407f08145feSKevin Wolf /*
2408f08145feSKevin Wolf  * Writes to the file and ensures that no writes are reordered across this
2409f08145feSKevin Wolf  * request (acts as a barrier)
2410f08145feSKevin Wolf  *
2411f08145feSKevin Wolf  * Returns 0 on success, -errno in error cases.
2412f08145feSKevin Wolf  */
2413f08145feSKevin Wolf int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
2414f08145feSKevin Wolf     const void *buf, int count)
2415f08145feSKevin Wolf {
2416f08145feSKevin Wolf     int ret;
2417f08145feSKevin Wolf 
2418f08145feSKevin Wolf     ret = bdrv_pwrite(bs, offset, buf, count);
2419f08145feSKevin Wolf     if (ret < 0) {
2420f08145feSKevin Wolf         return ret;
2421f08145feSKevin Wolf     }
2422f08145feSKevin Wolf 
2423f05fa4adSPaolo Bonzini     /* No flush needed for cache modes that already do it */
2424f05fa4adSPaolo Bonzini     if (bs->enable_write_cache) {
2425f08145feSKevin Wolf         bdrv_flush(bs);
2426f08145feSKevin Wolf     }
2427f08145feSKevin Wolf 
2428f08145feSKevin Wolf     return 0;
2429f08145feSKevin Wolf }
2430f08145feSKevin Wolf 
2431470c0504SStefan Hajnoczi static int coroutine_fn bdrv_co_do_copy_on_readv(BlockDriverState *bs,
2432ab185921SStefan Hajnoczi         int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2433ab185921SStefan Hajnoczi {
2434ab185921SStefan Hajnoczi     /* Perform I/O through a temporary buffer so that users who scribble over
2435ab185921SStefan Hajnoczi      * their read buffer while the operation is in progress do not end up
2436ab185921SStefan Hajnoczi      * modifying the image file.  This is critical for zero-copy guest I/O
2437ab185921SStefan Hajnoczi      * where anything might happen inside guest memory.
2438ab185921SStefan Hajnoczi      */
2439ab185921SStefan Hajnoczi     void *bounce_buffer;
2440ab185921SStefan Hajnoczi 
244179c053bdSStefan Hajnoczi     BlockDriver *drv = bs->drv;
2442ab185921SStefan Hajnoczi     struct iovec iov;
2443ab185921SStefan Hajnoczi     QEMUIOVector bounce_qiov;
2444ab185921SStefan Hajnoczi     int64_t cluster_sector_num;
2445ab185921SStefan Hajnoczi     int cluster_nb_sectors;
2446ab185921SStefan Hajnoczi     size_t skip_bytes;
2447ab185921SStefan Hajnoczi     int ret;
2448ab185921SStefan Hajnoczi 
2449ab185921SStefan Hajnoczi     /* Cover entire cluster so no additional backing file I/O is required when
2450ab185921SStefan Hajnoczi      * allocating cluster in the image file.
2451ab185921SStefan Hajnoczi      */
2452343bded4SPaolo Bonzini     bdrv_round_to_clusters(bs, sector_num, nb_sectors,
2453ab185921SStefan Hajnoczi                            &cluster_sector_num, &cluster_nb_sectors);
2454ab185921SStefan Hajnoczi 
2455470c0504SStefan Hajnoczi     trace_bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors,
2456ab185921SStefan Hajnoczi                                    cluster_sector_num, cluster_nb_sectors);
2457ab185921SStefan Hajnoczi 
2458ab185921SStefan Hajnoczi     iov.iov_len = cluster_nb_sectors * BDRV_SECTOR_SIZE;
2459ab185921SStefan Hajnoczi     iov.iov_base = bounce_buffer = qemu_blockalign(bs, iov.iov_len);
2460ab185921SStefan Hajnoczi     qemu_iovec_init_external(&bounce_qiov, &iov, 1);
2461ab185921SStefan Hajnoczi 
246279c053bdSStefan Hajnoczi     ret = drv->bdrv_co_readv(bs, cluster_sector_num, cluster_nb_sectors,
2463ab185921SStefan Hajnoczi                              &bounce_qiov);
2464ab185921SStefan Hajnoczi     if (ret < 0) {
2465ab185921SStefan Hajnoczi         goto err;
2466ab185921SStefan Hajnoczi     }
2467ab185921SStefan Hajnoczi 
246879c053bdSStefan Hajnoczi     if (drv->bdrv_co_write_zeroes &&
246979c053bdSStefan Hajnoczi         buffer_is_zero(bounce_buffer, iov.iov_len)) {
2470621f0589SKevin Wolf         ret = bdrv_co_do_write_zeroes(bs, cluster_sector_num,
247179c053bdSStefan Hajnoczi                                       cluster_nb_sectors);
247279c053bdSStefan Hajnoczi     } else {
2473f05fa4adSPaolo Bonzini         /* This does not change the data on the disk, it is not necessary
2474f05fa4adSPaolo Bonzini          * to flush even in cache=writethrough mode.
2475f05fa4adSPaolo Bonzini          */
247679c053bdSStefan Hajnoczi         ret = drv->bdrv_co_writev(bs, cluster_sector_num, cluster_nb_sectors,
2477ab185921SStefan Hajnoczi                                   &bounce_qiov);
247879c053bdSStefan Hajnoczi     }
247979c053bdSStefan Hajnoczi 
2480ab185921SStefan Hajnoczi     if (ret < 0) {
2481ab185921SStefan Hajnoczi         /* It might be okay to ignore write errors for guest requests.  If this
2482ab185921SStefan Hajnoczi          * is a deliberate copy-on-read then we don't want to ignore the error.
2483ab185921SStefan Hajnoczi          * Simply report it in all cases.
2484ab185921SStefan Hajnoczi          */
2485ab185921SStefan Hajnoczi         goto err;
2486ab185921SStefan Hajnoczi     }
2487ab185921SStefan Hajnoczi 
2488ab185921SStefan Hajnoczi     skip_bytes = (sector_num - cluster_sector_num) * BDRV_SECTOR_SIZE;
248903396148SMichael Tokarev     qemu_iovec_from_buf(qiov, 0, bounce_buffer + skip_bytes,
2490ab185921SStefan Hajnoczi                         nb_sectors * BDRV_SECTOR_SIZE);
2491ab185921SStefan Hajnoczi 
2492ab185921SStefan Hajnoczi err:
2493ab185921SStefan Hajnoczi     qemu_vfree(bounce_buffer);
2494ab185921SStefan Hajnoczi     return ret;
2495ab185921SStefan Hajnoczi }
2496ab185921SStefan Hajnoczi 
2497c5fbe571SStefan Hajnoczi /*
2498c5fbe571SStefan Hajnoczi  * Handle a read request in coroutine context
2499c5fbe571SStefan Hajnoczi  */
2500c5fbe571SStefan Hajnoczi static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
2501470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2502470c0504SStefan Hajnoczi     BdrvRequestFlags flags)
2503da1fa91dSKevin Wolf {
2504da1fa91dSKevin Wolf     BlockDriver *drv = bs->drv;
2505dbffbdcfSStefan Hajnoczi     BdrvTrackedRequest req;
2506dbffbdcfSStefan Hajnoczi     int ret;
2507da1fa91dSKevin Wolf 
2508da1fa91dSKevin Wolf     if (!drv) {
2509da1fa91dSKevin Wolf         return -ENOMEDIUM;
2510da1fa91dSKevin Wolf     }
2511da1fa91dSKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2512da1fa91dSKevin Wolf         return -EIO;
2513da1fa91dSKevin Wolf     }
2514da1fa91dSKevin Wolf 
251598f90dbaSZhi Yong Wu     /* throttling disk read I/O */
251698f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
251798f90dbaSZhi Yong Wu         bdrv_io_limits_intercept(bs, false, nb_sectors);
251898f90dbaSZhi Yong Wu     }
251998f90dbaSZhi Yong Wu 
2520f4658285SStefan Hajnoczi     if (bs->copy_on_read) {
2521470c0504SStefan Hajnoczi         flags |= BDRV_REQ_COPY_ON_READ;
2522470c0504SStefan Hajnoczi     }
2523470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2524470c0504SStefan Hajnoczi         bs->copy_on_read_in_flight++;
2525470c0504SStefan Hajnoczi     }
2526470c0504SStefan Hajnoczi 
2527470c0504SStefan Hajnoczi     if (bs->copy_on_read_in_flight) {
2528f4658285SStefan Hajnoczi         wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2529f4658285SStefan Hajnoczi     }
2530f4658285SStefan Hajnoczi 
2531dbffbdcfSStefan Hajnoczi     tracked_request_begin(&req, bs, sector_num, nb_sectors, false);
2532ab185921SStefan Hajnoczi 
2533470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2534ab185921SStefan Hajnoczi         int pnum;
2535ab185921SStefan Hajnoczi 
2536ab185921SStefan Hajnoczi         ret = bdrv_co_is_allocated(bs, sector_num, nb_sectors, &pnum);
2537ab185921SStefan Hajnoczi         if (ret < 0) {
2538ab185921SStefan Hajnoczi             goto out;
2539ab185921SStefan Hajnoczi         }
2540ab185921SStefan Hajnoczi 
2541ab185921SStefan Hajnoczi         if (!ret || pnum != nb_sectors) {
2542470c0504SStefan Hajnoczi             ret = bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors, qiov);
2543ab185921SStefan Hajnoczi             goto out;
2544ab185921SStefan Hajnoczi         }
2545ab185921SStefan Hajnoczi     }
2546ab185921SStefan Hajnoczi 
2547dbffbdcfSStefan Hajnoczi     ret = drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
2548ab185921SStefan Hajnoczi 
2549ab185921SStefan Hajnoczi out:
2550dbffbdcfSStefan Hajnoczi     tracked_request_end(&req);
2551470c0504SStefan Hajnoczi 
2552470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2553470c0504SStefan Hajnoczi         bs->copy_on_read_in_flight--;
2554470c0504SStefan Hajnoczi     }
2555470c0504SStefan Hajnoczi 
2556dbffbdcfSStefan Hajnoczi     return ret;
2557da1fa91dSKevin Wolf }
2558da1fa91dSKevin Wolf 
2559c5fbe571SStefan Hajnoczi int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
2560da1fa91dSKevin Wolf     int nb_sectors, QEMUIOVector *qiov)
2561da1fa91dSKevin Wolf {
2562c5fbe571SStefan Hajnoczi     trace_bdrv_co_readv(bs, sector_num, nb_sectors);
2563da1fa91dSKevin Wolf 
2564470c0504SStefan Hajnoczi     return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov, 0);
2565470c0504SStefan Hajnoczi }
2566470c0504SStefan Hajnoczi 
2567470c0504SStefan Hajnoczi int coroutine_fn bdrv_co_copy_on_readv(BlockDriverState *bs,
2568470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2569470c0504SStefan Hajnoczi {
2570470c0504SStefan Hajnoczi     trace_bdrv_co_copy_on_readv(bs, sector_num, nb_sectors);
2571470c0504SStefan Hajnoczi 
2572470c0504SStefan Hajnoczi     return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov,
2573470c0504SStefan Hajnoczi                             BDRV_REQ_COPY_ON_READ);
2574c5fbe571SStefan Hajnoczi }
2575c5fbe571SStefan Hajnoczi 
2576f08f2ddaSStefan Hajnoczi static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
2577f08f2ddaSStefan Hajnoczi     int64_t sector_num, int nb_sectors)
2578f08f2ddaSStefan Hajnoczi {
2579f08f2ddaSStefan Hajnoczi     BlockDriver *drv = bs->drv;
2580f08f2ddaSStefan Hajnoczi     QEMUIOVector qiov;
2581f08f2ddaSStefan Hajnoczi     struct iovec iov;
2582f08f2ddaSStefan Hajnoczi     int ret;
2583f08f2ddaSStefan Hajnoczi 
2584621f0589SKevin Wolf     /* TODO Emulate only part of misaligned requests instead of letting block
2585621f0589SKevin Wolf      * drivers return -ENOTSUP and emulate everything */
2586621f0589SKevin Wolf 
2587f08f2ddaSStefan Hajnoczi     /* First try the efficient write zeroes operation */
2588f08f2ddaSStefan Hajnoczi     if (drv->bdrv_co_write_zeroes) {
2589621f0589SKevin Wolf         ret = drv->bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2590621f0589SKevin Wolf         if (ret != -ENOTSUP) {
2591621f0589SKevin Wolf             return ret;
2592621f0589SKevin Wolf         }
2593f08f2ddaSStefan Hajnoczi     }
2594f08f2ddaSStefan Hajnoczi 
2595f08f2ddaSStefan Hajnoczi     /* Fall back to bounce buffer if write zeroes is unsupported */
2596f08f2ddaSStefan Hajnoczi     iov.iov_len  = nb_sectors * BDRV_SECTOR_SIZE;
2597f08f2ddaSStefan Hajnoczi     iov.iov_base = qemu_blockalign(bs, iov.iov_len);
2598f08f2ddaSStefan Hajnoczi     memset(iov.iov_base, 0, iov.iov_len);
2599f08f2ddaSStefan Hajnoczi     qemu_iovec_init_external(&qiov, &iov, 1);
2600f08f2ddaSStefan Hajnoczi 
2601f08f2ddaSStefan Hajnoczi     ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, &qiov);
2602f08f2ddaSStefan Hajnoczi 
2603f08f2ddaSStefan Hajnoczi     qemu_vfree(iov.iov_base);
2604f08f2ddaSStefan Hajnoczi     return ret;
2605f08f2ddaSStefan Hajnoczi }
2606f08f2ddaSStefan Hajnoczi 
2607c5fbe571SStefan Hajnoczi /*
2608c5fbe571SStefan Hajnoczi  * Handle a write request in coroutine context
2609c5fbe571SStefan Hajnoczi  */
2610c5fbe571SStefan Hajnoczi static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
2611f08f2ddaSStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2612f08f2ddaSStefan Hajnoczi     BdrvRequestFlags flags)
2613c5fbe571SStefan Hajnoczi {
2614c5fbe571SStefan Hajnoczi     BlockDriver *drv = bs->drv;
2615dbffbdcfSStefan Hajnoczi     BdrvTrackedRequest req;
26166b7cb247SStefan Hajnoczi     int ret;
2617da1fa91dSKevin Wolf 
2618da1fa91dSKevin Wolf     if (!bs->drv) {
2619da1fa91dSKevin Wolf         return -ENOMEDIUM;
2620da1fa91dSKevin Wolf     }
2621da1fa91dSKevin Wolf     if (bs->read_only) {
2622da1fa91dSKevin Wolf         return -EACCES;
2623da1fa91dSKevin Wolf     }
2624da1fa91dSKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2625da1fa91dSKevin Wolf         return -EIO;
2626da1fa91dSKevin Wolf     }
2627da1fa91dSKevin Wolf 
262898f90dbaSZhi Yong Wu     /* throttling disk write I/O */
262998f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
263098f90dbaSZhi Yong Wu         bdrv_io_limits_intercept(bs, true, nb_sectors);
263198f90dbaSZhi Yong Wu     }
263298f90dbaSZhi Yong Wu 
2633470c0504SStefan Hajnoczi     if (bs->copy_on_read_in_flight) {
2634f4658285SStefan Hajnoczi         wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2635f4658285SStefan Hajnoczi     }
2636f4658285SStefan Hajnoczi 
2637dbffbdcfSStefan Hajnoczi     tracked_request_begin(&req, bs, sector_num, nb_sectors, true);
2638dbffbdcfSStefan Hajnoczi 
2639d616b224SStefan Hajnoczi     ret = notifier_with_return_list_notify(&bs->before_write_notifiers, &req);
2640d616b224SStefan Hajnoczi 
2641d616b224SStefan Hajnoczi     if (ret < 0) {
2642d616b224SStefan Hajnoczi         /* Do nothing, write notifier decided to fail this request */
2643d616b224SStefan Hajnoczi     } else if (flags & BDRV_REQ_ZERO_WRITE) {
2644f08f2ddaSStefan Hajnoczi         ret = bdrv_co_do_write_zeroes(bs, sector_num, nb_sectors);
2645f08f2ddaSStefan Hajnoczi     } else {
26466b7cb247SStefan Hajnoczi         ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
2647f08f2ddaSStefan Hajnoczi     }
26486b7cb247SStefan Hajnoczi 
2649f05fa4adSPaolo Bonzini     if (ret == 0 && !bs->enable_write_cache) {
2650f05fa4adSPaolo Bonzini         ret = bdrv_co_flush(bs);
2651f05fa4adSPaolo Bonzini     }
2652f05fa4adSPaolo Bonzini 
2653da1fa91dSKevin Wolf     if (bs->dirty_bitmap) {
26541755da16SPaolo Bonzini         bdrv_set_dirty(bs, sector_num, nb_sectors);
2655da1fa91dSKevin Wolf     }
2656da1fa91dSKevin Wolf 
2657da1fa91dSKevin Wolf     if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
2658da1fa91dSKevin Wolf         bs->wr_highest_sector = sector_num + nb_sectors - 1;
2659da1fa91dSKevin Wolf     }
2660da1fa91dSKevin Wolf 
2661dbffbdcfSStefan Hajnoczi     tracked_request_end(&req);
2662dbffbdcfSStefan Hajnoczi 
26636b7cb247SStefan Hajnoczi     return ret;
2664da1fa91dSKevin Wolf }
2665da1fa91dSKevin Wolf 
2666c5fbe571SStefan Hajnoczi int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
2667c5fbe571SStefan Hajnoczi     int nb_sectors, QEMUIOVector *qiov)
2668c5fbe571SStefan Hajnoczi {
2669c5fbe571SStefan Hajnoczi     trace_bdrv_co_writev(bs, sector_num, nb_sectors);
2670c5fbe571SStefan Hajnoczi 
2671f08f2ddaSStefan Hajnoczi     return bdrv_co_do_writev(bs, sector_num, nb_sectors, qiov, 0);
2672f08f2ddaSStefan Hajnoczi }
2673f08f2ddaSStefan Hajnoczi 
2674f08f2ddaSStefan Hajnoczi int coroutine_fn bdrv_co_write_zeroes(BlockDriverState *bs,
2675f08f2ddaSStefan Hajnoczi                                       int64_t sector_num, int nb_sectors)
2676f08f2ddaSStefan Hajnoczi {
2677f08f2ddaSStefan Hajnoczi     trace_bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2678f08f2ddaSStefan Hajnoczi 
2679f08f2ddaSStefan Hajnoczi     return bdrv_co_do_writev(bs, sector_num, nb_sectors, NULL,
2680f08f2ddaSStefan Hajnoczi                              BDRV_REQ_ZERO_WRITE);
2681c5fbe571SStefan Hajnoczi }
2682c5fbe571SStefan Hajnoczi 
268383f64091Sbellard /**
268483f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
268583f64091Sbellard  */
268683f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
268783f64091Sbellard {
268883f64091Sbellard     BlockDriver *drv = bs->drv;
268951762288SStefan Hajnoczi     int ret;
269083f64091Sbellard     if (!drv)
269119cb3738Sbellard         return -ENOMEDIUM;
269283f64091Sbellard     if (!drv->bdrv_truncate)
269383f64091Sbellard         return -ENOTSUP;
269459f2689dSNaphtali Sprei     if (bs->read_only)
269559f2689dSNaphtali Sprei         return -EACCES;
26968591675fSMarcelo Tosatti     if (bdrv_in_use(bs))
26978591675fSMarcelo Tosatti         return -EBUSY;
269851762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
269951762288SStefan Hajnoczi     if (ret == 0) {
270051762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2701145feb17SMarkus Armbruster         bdrv_dev_resize_cb(bs);
270251762288SStefan Hajnoczi     }
270351762288SStefan Hajnoczi     return ret;
270483f64091Sbellard }
270583f64091Sbellard 
270683f64091Sbellard /**
27074a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
27084a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
27094a1d5e1fSFam Zheng  */
27104a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
27114a1d5e1fSFam Zheng {
27124a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
27134a1d5e1fSFam Zheng     if (!drv) {
27144a1d5e1fSFam Zheng         return -ENOMEDIUM;
27154a1d5e1fSFam Zheng     }
27164a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
27174a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
27184a1d5e1fSFam Zheng     }
27194a1d5e1fSFam Zheng     if (bs->file) {
27204a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
27214a1d5e1fSFam Zheng     }
27224a1d5e1fSFam Zheng     return -ENOTSUP;
27234a1d5e1fSFam Zheng }
27244a1d5e1fSFam Zheng 
27254a1d5e1fSFam Zheng /**
272683f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
272783f64091Sbellard  */
272883f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
272983f64091Sbellard {
273083f64091Sbellard     BlockDriver *drv = bs->drv;
273183f64091Sbellard     if (!drv)
273219cb3738Sbellard         return -ENOMEDIUM;
273351762288SStefan Hajnoczi 
27342c6942faSMarkus Armbruster     if (bs->growable || bdrv_dev_has_removable_media(bs)) {
273546a4e4e6SStefan Hajnoczi         if (drv->bdrv_getlength) {
273683f64091Sbellard             return drv->bdrv_getlength(bs);
2737fc01f7e7Sbellard         }
273846a4e4e6SStefan Hajnoczi     }
273946a4e4e6SStefan Hajnoczi     return bs->total_sectors * BDRV_SECTOR_SIZE;
274046a4e4e6SStefan Hajnoczi }
2741fc01f7e7Sbellard 
274219cb3738Sbellard /* return 0 as number of sectors if no device present or error */
274396b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2744fc01f7e7Sbellard {
274519cb3738Sbellard     int64_t length;
274619cb3738Sbellard     length = bdrv_getlength(bs);
274719cb3738Sbellard     if (length < 0)
274819cb3738Sbellard         length = 0;
274919cb3738Sbellard     else
27506ea44308SJan Kiszka         length = length >> BDRV_SECTOR_BITS;
275119cb3738Sbellard     *nb_sectors_ptr = length;
2752fc01f7e7Sbellard }
2753cf98951bSbellard 
27540563e191SZhi Yong Wu /* throttling disk io limits */
27550563e191SZhi Yong Wu void bdrv_set_io_limits(BlockDriverState *bs,
27560563e191SZhi Yong Wu                         BlockIOLimit *io_limits)
27570563e191SZhi Yong Wu {
27580563e191SZhi Yong Wu     bs->io_limits = *io_limits;
27590563e191SZhi Yong Wu     bs->io_limits_enabled = bdrv_io_limits_enabled(bs);
27600563e191SZhi Yong Wu }
27610563e191SZhi Yong Wu 
2762ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2763ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
2764abd7f68dSMarkus Armbruster {
2765abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
2766abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
2767abd7f68dSMarkus Armbruster }
2768abd7f68dSMarkus Armbruster 
27691ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2770abd7f68dSMarkus Armbruster {
2771abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
2772abd7f68dSMarkus Armbruster }
2773abd7f68dSMarkus Armbruster 
27743e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
27753e1caa5fSPaolo Bonzini {
27763e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
27773e1caa5fSPaolo Bonzini 
27783e1caa5fSPaolo Bonzini     switch (on_err) {
27793e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
27803e1caa5fSPaolo Bonzini         return (error == ENOSPC) ? BDRV_ACTION_STOP : BDRV_ACTION_REPORT;
27813e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
27823e1caa5fSPaolo Bonzini         return BDRV_ACTION_STOP;
27833e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
27843e1caa5fSPaolo Bonzini         return BDRV_ACTION_REPORT;
27853e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
27863e1caa5fSPaolo Bonzini         return BDRV_ACTION_IGNORE;
27873e1caa5fSPaolo Bonzini     default:
27883e1caa5fSPaolo Bonzini         abort();
27893e1caa5fSPaolo Bonzini     }
27903e1caa5fSPaolo Bonzini }
27913e1caa5fSPaolo Bonzini 
27923e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
27933e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
27943e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
27953e1caa5fSPaolo Bonzini  */
27963e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
27973e1caa5fSPaolo Bonzini                        bool is_read, int error)
27983e1caa5fSPaolo Bonzini {
27993e1caa5fSPaolo Bonzini     assert(error >= 0);
280032c81a4aSPaolo Bonzini     bdrv_emit_qmp_error_event(bs, QEVENT_BLOCK_IO_ERROR, action, is_read);
28013e1caa5fSPaolo Bonzini     if (action == BDRV_ACTION_STOP) {
28023e1caa5fSPaolo Bonzini         vm_stop(RUN_STATE_IO_ERROR);
28033e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
28043e1caa5fSPaolo Bonzini     }
28053e1caa5fSPaolo Bonzini }
28063e1caa5fSPaolo Bonzini 
2807b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
2808b338082bSbellard {
2809b338082bSbellard     return bs->read_only;
2810b338082bSbellard }
2811b338082bSbellard 
2812985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
2813985a03b0Sths {
2814985a03b0Sths     return bs->sg;
2815985a03b0Sths }
2816985a03b0Sths 
2817e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
2818e900a7b7SChristoph Hellwig {
2819e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
2820e900a7b7SChristoph Hellwig }
2821e900a7b7SChristoph Hellwig 
2822425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2823425b0148SPaolo Bonzini {
2824425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
282555b110f2SJeff Cody 
282655b110f2SJeff Cody     /* so a reopen() will preserve wce */
282755b110f2SJeff Cody     if (wce) {
282855b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
282955b110f2SJeff Cody     } else {
283055b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
283155b110f2SJeff Cody     }
2832425b0148SPaolo Bonzini }
2833425b0148SPaolo Bonzini 
2834ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
2835ea2384d3Sbellard {
2836ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
2837ea2384d3Sbellard         return 1;
2838ea2384d3Sbellard     return bs->encrypted;
2839ea2384d3Sbellard }
2840ea2384d3Sbellard 
2841c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
2842c0f4ce77Saliguori {
2843c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
2844c0f4ce77Saliguori 
2845c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2846c0f4ce77Saliguori         return 1;
2847c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
2848c0f4ce77Saliguori }
2849c0f4ce77Saliguori 
2850ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
2851ea2384d3Sbellard {
2852ea2384d3Sbellard     int ret;
2853ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
2854ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
2855ea2384d3Sbellard         if (ret < 0)
2856ea2384d3Sbellard             return ret;
2857ea2384d3Sbellard         if (!bs->encrypted)
2858ea2384d3Sbellard             return 0;
2859ea2384d3Sbellard     }
2860fd04a2aeSShahar Havivi     if (!bs->encrypted) {
2861fd04a2aeSShahar Havivi         return -EINVAL;
2862fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2863fd04a2aeSShahar Havivi         return -ENOMEDIUM;
2864fd04a2aeSShahar Havivi     }
2865c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
2866bb5fc20fSaliguori     if (ret < 0) {
2867bb5fc20fSaliguori         bs->valid_key = 0;
2868bb5fc20fSaliguori     } else if (!bs->valid_key) {
2869bb5fc20fSaliguori         bs->valid_key = 1;
2870bb5fc20fSaliguori         /* call the change callback now, we skipped it on open */
28717d4b4ba5SMarkus Armbruster         bdrv_dev_change_media_cb(bs, true);
2872bb5fc20fSaliguori     }
2873c0f4ce77Saliguori     return ret;
2874ea2384d3Sbellard }
2875ea2384d3Sbellard 
2876f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
2877ea2384d3Sbellard {
2878f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
2879ea2384d3Sbellard }
2880ea2384d3Sbellard 
2881ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2882ea2384d3Sbellard                          void *opaque)
2883ea2384d3Sbellard {
2884ea2384d3Sbellard     BlockDriver *drv;
2885ea2384d3Sbellard 
28868a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2887ea2384d3Sbellard         it(opaque, drv->format_name);
2888ea2384d3Sbellard     }
2889ea2384d3Sbellard }
2890ea2384d3Sbellard 
2891b338082bSbellard BlockDriverState *bdrv_find(const char *name)
2892b338082bSbellard {
2893b338082bSbellard     BlockDriverState *bs;
2894b338082bSbellard 
28951b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
28961b7bdbc1SStefan Hajnoczi         if (!strcmp(name, bs->device_name)) {
2897b338082bSbellard             return bs;
2898b338082bSbellard         }
28991b7bdbc1SStefan Hajnoczi     }
2900b338082bSbellard     return NULL;
2901b338082bSbellard }
2902b338082bSbellard 
29032f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
29042f399b0aSMarkus Armbruster {
29052f399b0aSMarkus Armbruster     if (!bs) {
29062f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
29072f399b0aSMarkus Armbruster     }
29082f399b0aSMarkus Armbruster     return QTAILQ_NEXT(bs, list);
29092f399b0aSMarkus Armbruster }
29102f399b0aSMarkus Armbruster 
291151de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
291281d0912dSbellard {
291381d0912dSbellard     BlockDriverState *bs;
291481d0912dSbellard 
29151b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
291651de9760Saliguori         it(opaque, bs);
291781d0912dSbellard     }
291881d0912dSbellard }
291981d0912dSbellard 
2920ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
2921ea2384d3Sbellard {
2922ea2384d3Sbellard     return bs->device_name;
2923ea2384d3Sbellard }
2924ea2384d3Sbellard 
2925c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
2926c8433287SMarkus Armbruster {
2927c8433287SMarkus Armbruster     return bs->open_flags;
2928c8433287SMarkus Armbruster }
2929c8433287SMarkus Armbruster 
2930f0f0fdfeSKevin Wolf int bdrv_flush_all(void)
2931c6ca28d6Saliguori {
2932c6ca28d6Saliguori     BlockDriverState *bs;
2933f0f0fdfeSKevin Wolf     int result = 0;
2934c6ca28d6Saliguori 
29351b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
2936f0f0fdfeSKevin Wolf         int ret = bdrv_flush(bs);
2937f0f0fdfeSKevin Wolf         if (ret < 0 && !result) {
2938f0f0fdfeSKevin Wolf             result = ret;
2939c6ca28d6Saliguori         }
29401b7bdbc1SStefan Hajnoczi     }
2941c6ca28d6Saliguori 
2942f0f0fdfeSKevin Wolf     return result;
2943f0f0fdfeSKevin Wolf }
2944f0f0fdfeSKevin Wolf 
29453ac21627SPeter Lieven int bdrv_has_zero_init_1(BlockDriverState *bs)
29463ac21627SPeter Lieven {
29473ac21627SPeter Lieven     return 1;
29483ac21627SPeter Lieven }
29493ac21627SPeter Lieven 
2950f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
2951f2feebbdSKevin Wolf {
2952f2feebbdSKevin Wolf     assert(bs->drv);
2953f2feebbdSKevin Wolf 
2954336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
2955336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
2956f2feebbdSKevin Wolf     }
2957f2feebbdSKevin Wolf 
29583ac21627SPeter Lieven     /* safe default */
29593ac21627SPeter Lieven     return 0;
2960f2feebbdSKevin Wolf }
2961f2feebbdSKevin Wolf 
2962376ae3f1SStefan Hajnoczi typedef struct BdrvCoIsAllocatedData {
2963376ae3f1SStefan Hajnoczi     BlockDriverState *bs;
2964b35b2bbaSMiroslav Rezanina     BlockDriverState *base;
2965376ae3f1SStefan Hajnoczi     int64_t sector_num;
2966376ae3f1SStefan Hajnoczi     int nb_sectors;
2967376ae3f1SStefan Hajnoczi     int *pnum;
2968376ae3f1SStefan Hajnoczi     int ret;
2969376ae3f1SStefan Hajnoczi     bool done;
2970376ae3f1SStefan Hajnoczi } BdrvCoIsAllocatedData;
2971376ae3f1SStefan Hajnoczi 
2972f58c7b35Sths /*
2973f58c7b35Sths  * Returns true iff the specified sector is present in the disk image. Drivers
2974f58c7b35Sths  * not implementing the functionality are assumed to not support backing files,
2975f58c7b35Sths  * hence all their sectors are reported as allocated.
2976f58c7b35Sths  *
2977bd9533e3SStefan Hajnoczi  * If 'sector_num' is beyond the end of the disk image the return value is 0
2978bd9533e3SStefan Hajnoczi  * and 'pnum' is set to 0.
2979bd9533e3SStefan Hajnoczi  *
2980f58c7b35Sths  * 'pnum' is set to the number of sectors (including and immediately following
2981f58c7b35Sths  * the specified sector) that are known to be in the same
2982f58c7b35Sths  * allocated/unallocated state.
2983f58c7b35Sths  *
2984bd9533e3SStefan Hajnoczi  * 'nb_sectors' is the max value 'pnum' should be set to.  If nb_sectors goes
2985bd9533e3SStefan Hajnoczi  * beyond the end of the disk image it will be clamped.
2986f58c7b35Sths  */
2987060f51c9SStefan Hajnoczi int coroutine_fn bdrv_co_is_allocated(BlockDriverState *bs, int64_t sector_num,
2988060f51c9SStefan Hajnoczi                                       int nb_sectors, int *pnum)
2989f58c7b35Sths {
2990f58c7b35Sths     int64_t n;
2991bd9533e3SStefan Hajnoczi 
29926aebab14SStefan Hajnoczi     if (sector_num >= bs->total_sectors) {
29936aebab14SStefan Hajnoczi         *pnum = 0;
29946aebab14SStefan Hajnoczi         return 0;
29956aebab14SStefan Hajnoczi     }
2996bd9533e3SStefan Hajnoczi 
29976aebab14SStefan Hajnoczi     n = bs->total_sectors - sector_num;
2998bd9533e3SStefan Hajnoczi     if (n < nb_sectors) {
2999bd9533e3SStefan Hajnoczi         nb_sectors = n;
3000bd9533e3SStefan Hajnoczi     }
3001bd9533e3SStefan Hajnoczi 
3002bd9533e3SStefan Hajnoczi     if (!bs->drv->bdrv_co_is_allocated) {
3003bd9533e3SStefan Hajnoczi         *pnum = nb_sectors;
30046aebab14SStefan Hajnoczi         return 1;
30056aebab14SStefan Hajnoczi     }
30066aebab14SStefan Hajnoczi 
3007060f51c9SStefan Hajnoczi     return bs->drv->bdrv_co_is_allocated(bs, sector_num, nb_sectors, pnum);
3008060f51c9SStefan Hajnoczi }
3009060f51c9SStefan Hajnoczi 
3010060f51c9SStefan Hajnoczi /* Coroutine wrapper for bdrv_is_allocated() */
3011060f51c9SStefan Hajnoczi static void coroutine_fn bdrv_is_allocated_co_entry(void *opaque)
3012060f51c9SStefan Hajnoczi {
3013060f51c9SStefan Hajnoczi     BdrvCoIsAllocatedData *data = opaque;
3014060f51c9SStefan Hajnoczi     BlockDriverState *bs = data->bs;
3015060f51c9SStefan Hajnoczi 
3016060f51c9SStefan Hajnoczi     data->ret = bdrv_co_is_allocated(bs, data->sector_num, data->nb_sectors,
3017060f51c9SStefan Hajnoczi                                      data->pnum);
3018060f51c9SStefan Hajnoczi     data->done = true;
3019060f51c9SStefan Hajnoczi }
3020060f51c9SStefan Hajnoczi 
3021060f51c9SStefan Hajnoczi /*
3022060f51c9SStefan Hajnoczi  * Synchronous wrapper around bdrv_co_is_allocated().
3023060f51c9SStefan Hajnoczi  *
3024060f51c9SStefan Hajnoczi  * See bdrv_co_is_allocated() for details.
3025060f51c9SStefan Hajnoczi  */
3026060f51c9SStefan Hajnoczi int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
3027060f51c9SStefan Hajnoczi                       int *pnum)
3028060f51c9SStefan Hajnoczi {
3029376ae3f1SStefan Hajnoczi     Coroutine *co;
3030376ae3f1SStefan Hajnoczi     BdrvCoIsAllocatedData data = {
3031376ae3f1SStefan Hajnoczi         .bs = bs,
3032376ae3f1SStefan Hajnoczi         .sector_num = sector_num,
3033376ae3f1SStefan Hajnoczi         .nb_sectors = nb_sectors,
3034376ae3f1SStefan Hajnoczi         .pnum = pnum,
3035376ae3f1SStefan Hajnoczi         .done = false,
3036376ae3f1SStefan Hajnoczi     };
3037376ae3f1SStefan Hajnoczi 
3038376ae3f1SStefan Hajnoczi     co = qemu_coroutine_create(bdrv_is_allocated_co_entry);
3039376ae3f1SStefan Hajnoczi     qemu_coroutine_enter(co, &data);
3040376ae3f1SStefan Hajnoczi     while (!data.done) {
3041376ae3f1SStefan Hajnoczi         qemu_aio_wait();
3042376ae3f1SStefan Hajnoczi     }
3043376ae3f1SStefan Hajnoczi     return data.ret;
3044376ae3f1SStefan Hajnoczi }
3045f58c7b35Sths 
3046188a7bbfSPaolo Bonzini /*
3047188a7bbfSPaolo Bonzini  * Given an image chain: ... -> [BASE] -> [INTER1] -> [INTER2] -> [TOP]
3048188a7bbfSPaolo Bonzini  *
3049188a7bbfSPaolo Bonzini  * Return true if the given sector is allocated in any image between
3050188a7bbfSPaolo Bonzini  * BASE and TOP (inclusive).  BASE can be NULL to check if the given
3051188a7bbfSPaolo Bonzini  * sector is allocated in any image of the chain.  Return false otherwise.
3052188a7bbfSPaolo Bonzini  *
3053188a7bbfSPaolo Bonzini  * 'pnum' is set to the number of sectors (including and immediately following
3054188a7bbfSPaolo Bonzini  *  the specified sector) that are known to be in the same
3055188a7bbfSPaolo Bonzini  *  allocated/unallocated state.
3056188a7bbfSPaolo Bonzini  *
3057188a7bbfSPaolo Bonzini  */
3058188a7bbfSPaolo Bonzini int coroutine_fn bdrv_co_is_allocated_above(BlockDriverState *top,
3059188a7bbfSPaolo Bonzini                                             BlockDriverState *base,
3060188a7bbfSPaolo Bonzini                                             int64_t sector_num,
3061188a7bbfSPaolo Bonzini                                             int nb_sectors, int *pnum)
3062188a7bbfSPaolo Bonzini {
3063188a7bbfSPaolo Bonzini     BlockDriverState *intermediate;
3064188a7bbfSPaolo Bonzini     int ret, n = nb_sectors;
3065188a7bbfSPaolo Bonzini 
3066188a7bbfSPaolo Bonzini     intermediate = top;
3067188a7bbfSPaolo Bonzini     while (intermediate && intermediate != base) {
3068188a7bbfSPaolo Bonzini         int pnum_inter;
3069188a7bbfSPaolo Bonzini         ret = bdrv_co_is_allocated(intermediate, sector_num, nb_sectors,
3070188a7bbfSPaolo Bonzini                                    &pnum_inter);
3071188a7bbfSPaolo Bonzini         if (ret < 0) {
3072188a7bbfSPaolo Bonzini             return ret;
3073188a7bbfSPaolo Bonzini         } else if (ret) {
3074188a7bbfSPaolo Bonzini             *pnum = pnum_inter;
3075188a7bbfSPaolo Bonzini             return 1;
3076188a7bbfSPaolo Bonzini         }
3077188a7bbfSPaolo Bonzini 
3078188a7bbfSPaolo Bonzini         /*
3079188a7bbfSPaolo Bonzini          * [sector_num, nb_sectors] is unallocated on top but intermediate
3080188a7bbfSPaolo Bonzini          * might have
3081188a7bbfSPaolo Bonzini          *
3082188a7bbfSPaolo Bonzini          * [sector_num+x, nr_sectors] allocated.
3083188a7bbfSPaolo Bonzini          */
308463ba17d3SVishvananda Ishaya         if (n > pnum_inter &&
308563ba17d3SVishvananda Ishaya             (intermediate == top ||
308663ba17d3SVishvananda Ishaya              sector_num + pnum_inter < intermediate->total_sectors)) {
3087188a7bbfSPaolo Bonzini             n = pnum_inter;
3088188a7bbfSPaolo Bonzini         }
3089188a7bbfSPaolo Bonzini 
3090188a7bbfSPaolo Bonzini         intermediate = intermediate->backing_hd;
3091188a7bbfSPaolo Bonzini     }
3092188a7bbfSPaolo Bonzini 
3093188a7bbfSPaolo Bonzini     *pnum = n;
3094188a7bbfSPaolo Bonzini     return 0;
3095188a7bbfSPaolo Bonzini }
3096188a7bbfSPaolo Bonzini 
3097b35b2bbaSMiroslav Rezanina /* Coroutine wrapper for bdrv_is_allocated_above() */
3098b35b2bbaSMiroslav Rezanina static void coroutine_fn bdrv_is_allocated_above_co_entry(void *opaque)
3099b35b2bbaSMiroslav Rezanina {
3100b35b2bbaSMiroslav Rezanina     BdrvCoIsAllocatedData *data = opaque;
3101b35b2bbaSMiroslav Rezanina     BlockDriverState *top = data->bs;
3102b35b2bbaSMiroslav Rezanina     BlockDriverState *base = data->base;
3103b35b2bbaSMiroslav Rezanina 
3104b35b2bbaSMiroslav Rezanina     data->ret = bdrv_co_is_allocated_above(top, base, data->sector_num,
3105b35b2bbaSMiroslav Rezanina                                            data->nb_sectors, data->pnum);
3106b35b2bbaSMiroslav Rezanina     data->done = true;
3107b35b2bbaSMiroslav Rezanina }
3108b35b2bbaSMiroslav Rezanina 
3109b35b2bbaSMiroslav Rezanina /*
3110b35b2bbaSMiroslav Rezanina  * Synchronous wrapper around bdrv_co_is_allocated_above().
3111b35b2bbaSMiroslav Rezanina  *
3112b35b2bbaSMiroslav Rezanina  * See bdrv_co_is_allocated_above() for details.
3113b35b2bbaSMiroslav Rezanina  */
3114b35b2bbaSMiroslav Rezanina int bdrv_is_allocated_above(BlockDriverState *top, BlockDriverState *base,
3115b35b2bbaSMiroslav Rezanina                             int64_t sector_num, int nb_sectors, int *pnum)
3116b35b2bbaSMiroslav Rezanina {
3117b35b2bbaSMiroslav Rezanina     Coroutine *co;
3118b35b2bbaSMiroslav Rezanina     BdrvCoIsAllocatedData data = {
3119b35b2bbaSMiroslav Rezanina         .bs = top,
3120b35b2bbaSMiroslav Rezanina         .base = base,
3121b35b2bbaSMiroslav Rezanina         .sector_num = sector_num,
3122b35b2bbaSMiroslav Rezanina         .nb_sectors = nb_sectors,
3123b35b2bbaSMiroslav Rezanina         .pnum = pnum,
3124b35b2bbaSMiroslav Rezanina         .done = false,
3125b35b2bbaSMiroslav Rezanina     };
3126b35b2bbaSMiroslav Rezanina 
3127b35b2bbaSMiroslav Rezanina     co = qemu_coroutine_create(bdrv_is_allocated_above_co_entry);
3128b35b2bbaSMiroslav Rezanina     qemu_coroutine_enter(co, &data);
3129b35b2bbaSMiroslav Rezanina     while (!data.done) {
3130b35b2bbaSMiroslav Rezanina         qemu_aio_wait();
3131b35b2bbaSMiroslav Rezanina     }
3132b35b2bbaSMiroslav Rezanina     return data.ret;
3133b35b2bbaSMiroslav Rezanina }
3134b35b2bbaSMiroslav Rezanina 
3135045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
3136045df330Saliguori {
3137045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
3138045df330Saliguori         return bs->backing_file;
3139045df330Saliguori     else if (bs->encrypted)
3140045df330Saliguori         return bs->filename;
3141045df330Saliguori     else
3142045df330Saliguori         return NULL;
3143045df330Saliguori }
3144045df330Saliguori 
314583f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
314683f64091Sbellard                                char *filename, int filename_size)
314783f64091Sbellard {
314883f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
314983f64091Sbellard }
315083f64091Sbellard 
3151faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
3152faea38e7Sbellard                           const uint8_t *buf, int nb_sectors)
3153faea38e7Sbellard {
3154faea38e7Sbellard     BlockDriver *drv = bs->drv;
3155faea38e7Sbellard     if (!drv)
315619cb3738Sbellard         return -ENOMEDIUM;
3157faea38e7Sbellard     if (!drv->bdrv_write_compressed)
3158faea38e7Sbellard         return -ENOTSUP;
3159fbb7b4e0SKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors))
3160fbb7b4e0SKevin Wolf         return -EIO;
31617cd1e32aSlirans@il.ibm.com 
31621755da16SPaolo Bonzini     assert(!bs->dirty_bitmap);
31637cd1e32aSlirans@il.ibm.com 
3164faea38e7Sbellard     return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
3165faea38e7Sbellard }
3166faea38e7Sbellard 
3167faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3168faea38e7Sbellard {
3169faea38e7Sbellard     BlockDriver *drv = bs->drv;
3170faea38e7Sbellard     if (!drv)
317119cb3738Sbellard         return -ENOMEDIUM;
3172faea38e7Sbellard     if (!drv->bdrv_get_info)
3173faea38e7Sbellard         return -ENOTSUP;
3174faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3175faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3176faea38e7Sbellard }
3177faea38e7Sbellard 
317845566e9cSChristoph Hellwig int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
317945566e9cSChristoph Hellwig                       int64_t pos, int size)
3180178e08a5Saliguori {
3181cf8074b3SKevin Wolf     QEMUIOVector qiov;
3182cf8074b3SKevin Wolf     struct iovec iov = {
3183cf8074b3SKevin Wolf         .iov_base   = (void *) buf,
3184cf8074b3SKevin Wolf         .iov_len    = size,
3185cf8074b3SKevin Wolf     };
3186cf8074b3SKevin Wolf 
3187cf8074b3SKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
3188cf8074b3SKevin Wolf     return bdrv_writev_vmstate(bs, &qiov, pos);
3189cf8074b3SKevin Wolf }
3190cf8074b3SKevin Wolf 
3191cf8074b3SKevin Wolf int bdrv_writev_vmstate(BlockDriverState *bs, QEMUIOVector *qiov, int64_t pos)
3192cf8074b3SKevin Wolf {
3193178e08a5Saliguori     BlockDriver *drv = bs->drv;
3194cf8074b3SKevin Wolf 
3195cf8074b3SKevin Wolf     if (!drv) {
3196178e08a5Saliguori         return -ENOMEDIUM;
3197cf8074b3SKevin Wolf     } else if (drv->bdrv_save_vmstate) {
3198cf8074b3SKevin Wolf         return drv->bdrv_save_vmstate(bs, qiov, pos);
3199cf8074b3SKevin Wolf     } else if (bs->file) {
3200cf8074b3SKevin Wolf         return bdrv_writev_vmstate(bs->file, qiov, pos);
3201cf8074b3SKevin Wolf     }
3202cf8074b3SKevin Wolf 
32037cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3204178e08a5Saliguori }
3205178e08a5Saliguori 
320645566e9cSChristoph Hellwig int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
320745566e9cSChristoph Hellwig                       int64_t pos, int size)
3208178e08a5Saliguori {
3209178e08a5Saliguori     BlockDriver *drv = bs->drv;
3210178e08a5Saliguori     if (!drv)
3211178e08a5Saliguori         return -ENOMEDIUM;
32127cdb1f6dSMORITA Kazutaka     if (drv->bdrv_load_vmstate)
321345566e9cSChristoph Hellwig         return drv->bdrv_load_vmstate(bs, buf, pos, size);
32147cdb1f6dSMORITA Kazutaka     if (bs->file)
32157cdb1f6dSMORITA Kazutaka         return bdrv_load_vmstate(bs->file, buf, pos, size);
32167cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3217178e08a5Saliguori }
3218178e08a5Saliguori 
32198b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
32208b9b0cc2SKevin Wolf {
3221bf736fe3SKevin Wolf     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
32228b9b0cc2SKevin Wolf         return;
32238b9b0cc2SKevin Wolf     }
32248b9b0cc2SKevin Wolf 
3225bf736fe3SKevin Wolf     bs->drv->bdrv_debug_event(bs, event);
322641c695c7SKevin Wolf }
32278b9b0cc2SKevin Wolf 
322841c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
322941c695c7SKevin Wolf                           const char *tag)
323041c695c7SKevin Wolf {
323141c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
323241c695c7SKevin Wolf         bs = bs->file;
323341c695c7SKevin Wolf     }
323441c695c7SKevin Wolf 
323541c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
323641c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
323741c695c7SKevin Wolf     }
323841c695c7SKevin Wolf 
323941c695c7SKevin Wolf     return -ENOTSUP;
324041c695c7SKevin Wolf }
324141c695c7SKevin Wolf 
324241c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
324341c695c7SKevin Wolf {
324441c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_resume) {
324541c695c7SKevin Wolf         bs = bs->file;
324641c695c7SKevin Wolf     }
324741c695c7SKevin Wolf 
324841c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
324941c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
325041c695c7SKevin Wolf     }
325141c695c7SKevin Wolf 
325241c695c7SKevin Wolf     return -ENOTSUP;
325341c695c7SKevin Wolf }
325441c695c7SKevin Wolf 
325541c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
325641c695c7SKevin Wolf {
325741c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
325841c695c7SKevin Wolf         bs = bs->file;
325941c695c7SKevin Wolf     }
326041c695c7SKevin Wolf 
326141c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
326241c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
326341c695c7SKevin Wolf     }
326441c695c7SKevin Wolf 
326541c695c7SKevin Wolf     return false;
32668b9b0cc2SKevin Wolf }
32678b9b0cc2SKevin Wolf 
3268199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3269199630b6SBlue Swirl {
3270199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3271199630b6SBlue Swirl }
3272199630b6SBlue Swirl 
3273b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3274b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3275b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3276b1b1d783SJeff Cody  * the CWD rather than the chain. */
3277e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3278e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3279e8a6bb9cSMarcelo Tosatti {
3280b1b1d783SJeff Cody     char *filename_full = NULL;
3281b1b1d783SJeff Cody     char *backing_file_full = NULL;
3282b1b1d783SJeff Cody     char *filename_tmp = NULL;
3283b1b1d783SJeff Cody     int is_protocol = 0;
3284b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3285b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3286b1b1d783SJeff Cody 
3287b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3288e8a6bb9cSMarcelo Tosatti         return NULL;
3289e8a6bb9cSMarcelo Tosatti     }
3290e8a6bb9cSMarcelo Tosatti 
3291b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
3292b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
3293b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
3294b1b1d783SJeff Cody 
3295b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
3296b1b1d783SJeff Cody 
3297b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3298b1b1d783SJeff Cody 
3299b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
3300b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
3301b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3302b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3303b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3304b1b1d783SJeff Cody                 break;
3305b1b1d783SJeff Cody             }
3306e8a6bb9cSMarcelo Tosatti         } else {
3307b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
3308b1b1d783SJeff Cody              * image's filename path */
3309b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3310b1b1d783SJeff Cody                          backing_file);
3311b1b1d783SJeff Cody 
3312b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
3313b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
3314b1b1d783SJeff Cody                 continue;
3315b1b1d783SJeff Cody             }
3316b1b1d783SJeff Cody 
3317b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
3318b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
3319b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3320b1b1d783SJeff Cody                          curr_bs->backing_file);
3321b1b1d783SJeff Cody 
3322b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
3323b1b1d783SJeff Cody                 continue;
3324b1b1d783SJeff Cody             }
3325b1b1d783SJeff Cody 
3326b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
3327b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3328b1b1d783SJeff Cody                 break;
3329b1b1d783SJeff Cody             }
3330e8a6bb9cSMarcelo Tosatti         }
3331e8a6bb9cSMarcelo Tosatti     }
3332e8a6bb9cSMarcelo Tosatti 
3333b1b1d783SJeff Cody     g_free(filename_full);
3334b1b1d783SJeff Cody     g_free(backing_file_full);
3335b1b1d783SJeff Cody     g_free(filename_tmp);
3336b1b1d783SJeff Cody     return retval;
3337e8a6bb9cSMarcelo Tosatti }
3338e8a6bb9cSMarcelo Tosatti 
3339f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
3340f198fd1cSBenoît Canet {
3341f198fd1cSBenoît Canet     if (!bs->drv) {
3342f198fd1cSBenoît Canet         return 0;
3343f198fd1cSBenoît Canet     }
3344f198fd1cSBenoît Canet 
3345f198fd1cSBenoît Canet     if (!bs->backing_hd) {
3346f198fd1cSBenoît Canet         return 0;
3347f198fd1cSBenoît Canet     }
3348f198fd1cSBenoît Canet 
3349f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3350f198fd1cSBenoît Canet }
3351f198fd1cSBenoît Canet 
335279fac568SJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
335379fac568SJeff Cody {
335479fac568SJeff Cody     BlockDriverState *curr_bs = NULL;
335579fac568SJeff Cody 
335679fac568SJeff Cody     if (!bs) {
335779fac568SJeff Cody         return NULL;
335879fac568SJeff Cody     }
335979fac568SJeff Cody 
336079fac568SJeff Cody     curr_bs = bs;
336179fac568SJeff Cody 
336279fac568SJeff Cody     while (curr_bs->backing_hd) {
336379fac568SJeff Cody         curr_bs = curr_bs->backing_hd;
336479fac568SJeff Cody     }
336579fac568SJeff Cody     return curr_bs;
336679fac568SJeff Cody }
336779fac568SJeff Cody 
3368ea2384d3Sbellard /**************************************************************/
336983f64091Sbellard /* async I/Os */
3370ea2384d3Sbellard 
33713b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
3372f141eafeSaliguori                                  QEMUIOVector *qiov, int nb_sectors,
337383f64091Sbellard                                  BlockDriverCompletionFunc *cb, void *opaque)
3374ea2384d3Sbellard {
3375bbf0a440SStefan Hajnoczi     trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
3376bbf0a440SStefan Hajnoczi 
3377b2a61371SStefan Hajnoczi     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
33788c5873d6SStefan Hajnoczi                                  cb, opaque, false);
337983f64091Sbellard }
338083f64091Sbellard 
3381f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
3382f141eafeSaliguori                                   QEMUIOVector *qiov, int nb_sectors,
338383f64091Sbellard                                   BlockDriverCompletionFunc *cb, void *opaque)
33847674e7bfSbellard {
3385bbf0a440SStefan Hajnoczi     trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
3386bbf0a440SStefan Hajnoczi 
33871a6e115bSStefan Hajnoczi     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
33888c5873d6SStefan Hajnoczi                                  cb, opaque, true);
338983f64091Sbellard }
339083f64091Sbellard 
339140b4f539SKevin Wolf 
339240b4f539SKevin Wolf typedef struct MultiwriteCB {
339340b4f539SKevin Wolf     int error;
339440b4f539SKevin Wolf     int num_requests;
339540b4f539SKevin Wolf     int num_callbacks;
339640b4f539SKevin Wolf     struct {
339740b4f539SKevin Wolf         BlockDriverCompletionFunc *cb;
339840b4f539SKevin Wolf         void *opaque;
339940b4f539SKevin Wolf         QEMUIOVector *free_qiov;
340040b4f539SKevin Wolf     } callbacks[];
340140b4f539SKevin Wolf } MultiwriteCB;
340240b4f539SKevin Wolf 
340340b4f539SKevin Wolf static void multiwrite_user_cb(MultiwriteCB *mcb)
340440b4f539SKevin Wolf {
340540b4f539SKevin Wolf     int i;
340640b4f539SKevin Wolf 
340740b4f539SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
340840b4f539SKevin Wolf         mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
34091e1ea48dSStefan Hajnoczi         if (mcb->callbacks[i].free_qiov) {
34101e1ea48dSStefan Hajnoczi             qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
34111e1ea48dSStefan Hajnoczi         }
34127267c094SAnthony Liguori         g_free(mcb->callbacks[i].free_qiov);
341340b4f539SKevin Wolf     }
341440b4f539SKevin Wolf }
341540b4f539SKevin Wolf 
341640b4f539SKevin Wolf static void multiwrite_cb(void *opaque, int ret)
341740b4f539SKevin Wolf {
341840b4f539SKevin Wolf     MultiwriteCB *mcb = opaque;
341940b4f539SKevin Wolf 
34206d519a5fSStefan Hajnoczi     trace_multiwrite_cb(mcb, ret);
34216d519a5fSStefan Hajnoczi 
3422cb6d3ca0SKevin Wolf     if (ret < 0 && !mcb->error) {
342340b4f539SKevin Wolf         mcb->error = ret;
342440b4f539SKevin Wolf     }
342540b4f539SKevin Wolf 
342640b4f539SKevin Wolf     mcb->num_requests--;
342740b4f539SKevin Wolf     if (mcb->num_requests == 0) {
342840b4f539SKevin Wolf         multiwrite_user_cb(mcb);
34297267c094SAnthony Liguori         g_free(mcb);
343040b4f539SKevin Wolf     }
343140b4f539SKevin Wolf }
343240b4f539SKevin Wolf 
343340b4f539SKevin Wolf static int multiwrite_req_compare(const void *a, const void *b)
343440b4f539SKevin Wolf {
343577be4366SChristoph Hellwig     const BlockRequest *req1 = a, *req2 = b;
343677be4366SChristoph Hellwig 
343777be4366SChristoph Hellwig     /*
343877be4366SChristoph Hellwig      * Note that we can't simply subtract req2->sector from req1->sector
343977be4366SChristoph Hellwig      * here as that could overflow the return value.
344077be4366SChristoph Hellwig      */
344177be4366SChristoph Hellwig     if (req1->sector > req2->sector) {
344277be4366SChristoph Hellwig         return 1;
344377be4366SChristoph Hellwig     } else if (req1->sector < req2->sector) {
344477be4366SChristoph Hellwig         return -1;
344577be4366SChristoph Hellwig     } else {
344677be4366SChristoph Hellwig         return 0;
344777be4366SChristoph Hellwig     }
344840b4f539SKevin Wolf }
344940b4f539SKevin Wolf 
345040b4f539SKevin Wolf /*
345140b4f539SKevin Wolf  * Takes a bunch of requests and tries to merge them. Returns the number of
345240b4f539SKevin Wolf  * requests that remain after merging.
345340b4f539SKevin Wolf  */
345440b4f539SKevin Wolf static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
345540b4f539SKevin Wolf     int num_reqs, MultiwriteCB *mcb)
345640b4f539SKevin Wolf {
345740b4f539SKevin Wolf     int i, outidx;
345840b4f539SKevin Wolf 
345940b4f539SKevin Wolf     // Sort requests by start sector
346040b4f539SKevin Wolf     qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
346140b4f539SKevin Wolf 
346240b4f539SKevin Wolf     // Check if adjacent requests touch the same clusters. If so, combine them,
346340b4f539SKevin Wolf     // filling up gaps with zero sectors.
346440b4f539SKevin Wolf     outidx = 0;
346540b4f539SKevin Wolf     for (i = 1; i < num_reqs; i++) {
346640b4f539SKevin Wolf         int merge = 0;
346740b4f539SKevin Wolf         int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
346840b4f539SKevin Wolf 
3469b6a127a1SPaolo Bonzini         // Handle exactly sequential writes and overlapping writes.
347040b4f539SKevin Wolf         if (reqs[i].sector <= oldreq_last) {
347140b4f539SKevin Wolf             merge = 1;
347240b4f539SKevin Wolf         }
347340b4f539SKevin Wolf 
3474e2a305fbSChristoph Hellwig         if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
3475e2a305fbSChristoph Hellwig             merge = 0;
3476e2a305fbSChristoph Hellwig         }
3477e2a305fbSChristoph Hellwig 
347840b4f539SKevin Wolf         if (merge) {
347940b4f539SKevin Wolf             size_t size;
34807267c094SAnthony Liguori             QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
348140b4f539SKevin Wolf             qemu_iovec_init(qiov,
348240b4f539SKevin Wolf                 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
348340b4f539SKevin Wolf 
348440b4f539SKevin Wolf             // Add the first request to the merged one. If the requests are
348540b4f539SKevin Wolf             // overlapping, drop the last sectors of the first request.
348640b4f539SKevin Wolf             size = (reqs[i].sector - reqs[outidx].sector) << 9;
34871b093c48SMichael Tokarev             qemu_iovec_concat(qiov, reqs[outidx].qiov, 0, size);
348840b4f539SKevin Wolf 
3489b6a127a1SPaolo Bonzini             // We should need to add any zeros between the two requests
3490b6a127a1SPaolo Bonzini             assert (reqs[i].sector <= oldreq_last);
349140b4f539SKevin Wolf 
349240b4f539SKevin Wolf             // Add the second request
34931b093c48SMichael Tokarev             qemu_iovec_concat(qiov, reqs[i].qiov, 0, reqs[i].qiov->size);
349440b4f539SKevin Wolf 
3495cbf1dff2SKevin Wolf             reqs[outidx].nb_sectors = qiov->size >> 9;
349640b4f539SKevin Wolf             reqs[outidx].qiov = qiov;
349740b4f539SKevin Wolf 
349840b4f539SKevin Wolf             mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
349940b4f539SKevin Wolf         } else {
350040b4f539SKevin Wolf             outidx++;
350140b4f539SKevin Wolf             reqs[outidx].sector     = reqs[i].sector;
350240b4f539SKevin Wolf             reqs[outidx].nb_sectors = reqs[i].nb_sectors;
350340b4f539SKevin Wolf             reqs[outidx].qiov       = reqs[i].qiov;
350440b4f539SKevin Wolf         }
350540b4f539SKevin Wolf     }
350640b4f539SKevin Wolf 
350740b4f539SKevin Wolf     return outidx + 1;
350840b4f539SKevin Wolf }
350940b4f539SKevin Wolf 
351040b4f539SKevin Wolf /*
351140b4f539SKevin Wolf  * Submit multiple AIO write requests at once.
351240b4f539SKevin Wolf  *
351340b4f539SKevin Wolf  * On success, the function returns 0 and all requests in the reqs array have
351440b4f539SKevin Wolf  * been submitted. In error case this function returns -1, and any of the
351540b4f539SKevin Wolf  * requests may or may not be submitted yet. In particular, this means that the
351640b4f539SKevin Wolf  * callback will be called for some of the requests, for others it won't. The
351740b4f539SKevin Wolf  * caller must check the error field of the BlockRequest to wait for the right
351840b4f539SKevin Wolf  * callbacks (if error != 0, no callback will be called).
351940b4f539SKevin Wolf  *
352040b4f539SKevin Wolf  * The implementation may modify the contents of the reqs array, e.g. to merge
352140b4f539SKevin Wolf  * requests. However, the fields opaque and error are left unmodified as they
352240b4f539SKevin Wolf  * are used to signal failure for a single request to the caller.
352340b4f539SKevin Wolf  */
352440b4f539SKevin Wolf int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
352540b4f539SKevin Wolf {
352640b4f539SKevin Wolf     MultiwriteCB *mcb;
352740b4f539SKevin Wolf     int i;
352840b4f539SKevin Wolf 
3529301db7c2SRyan Harper     /* don't submit writes if we don't have a medium */
3530301db7c2SRyan Harper     if (bs->drv == NULL) {
3531301db7c2SRyan Harper         for (i = 0; i < num_reqs; i++) {
3532301db7c2SRyan Harper             reqs[i].error = -ENOMEDIUM;
3533301db7c2SRyan Harper         }
3534301db7c2SRyan Harper         return -1;
3535301db7c2SRyan Harper     }
3536301db7c2SRyan Harper 
353740b4f539SKevin Wolf     if (num_reqs == 0) {
353840b4f539SKevin Wolf         return 0;
353940b4f539SKevin Wolf     }
354040b4f539SKevin Wolf 
354140b4f539SKevin Wolf     // Create MultiwriteCB structure
35427267c094SAnthony Liguori     mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
354340b4f539SKevin Wolf     mcb->num_requests = 0;
354440b4f539SKevin Wolf     mcb->num_callbacks = num_reqs;
354540b4f539SKevin Wolf 
354640b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
354740b4f539SKevin Wolf         mcb->callbacks[i].cb = reqs[i].cb;
354840b4f539SKevin Wolf         mcb->callbacks[i].opaque = reqs[i].opaque;
354940b4f539SKevin Wolf     }
355040b4f539SKevin Wolf 
355140b4f539SKevin Wolf     // Check for mergable requests
355240b4f539SKevin Wolf     num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
355340b4f539SKevin Wolf 
35546d519a5fSStefan Hajnoczi     trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
35556d519a5fSStefan Hajnoczi 
3556df9309fbSPaolo Bonzini     /* Run the aio requests. */
3557df9309fbSPaolo Bonzini     mcb->num_requests = num_reqs;
355840b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
3559ad54ae80SPaolo Bonzini         bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
356040b4f539SKevin Wolf             reqs[i].nb_sectors, multiwrite_cb, mcb);
356140b4f539SKevin Wolf     }
356240b4f539SKevin Wolf 
356340b4f539SKevin Wolf     return 0;
356440b4f539SKevin Wolf }
356540b4f539SKevin Wolf 
356683f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
356783f64091Sbellard {
3568d7331bedSStefan Hajnoczi     acb->aiocb_info->cancel(acb);
356983f64091Sbellard }
357083f64091Sbellard 
357198f90dbaSZhi Yong Wu /* block I/O throttling */
357298f90dbaSZhi Yong Wu static bool bdrv_exceed_bps_limits(BlockDriverState *bs, int nb_sectors,
357398f90dbaSZhi Yong Wu                  bool is_write, double elapsed_time, uint64_t *wait)
357498f90dbaSZhi Yong Wu {
357598f90dbaSZhi Yong Wu     uint64_t bps_limit = 0;
3576ae29d6c6SStefan Hajnoczi     uint64_t extension;
357798f90dbaSZhi Yong Wu     double   bytes_limit, bytes_base, bytes_res;
357898f90dbaSZhi Yong Wu     double   slice_time, wait_time;
357998f90dbaSZhi Yong Wu 
358098f90dbaSZhi Yong Wu     if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
358198f90dbaSZhi Yong Wu         bps_limit = bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL];
358298f90dbaSZhi Yong Wu     } else if (bs->io_limits.bps[is_write]) {
358398f90dbaSZhi Yong Wu         bps_limit = bs->io_limits.bps[is_write];
358498f90dbaSZhi Yong Wu     } else {
358598f90dbaSZhi Yong Wu         if (wait) {
358698f90dbaSZhi Yong Wu             *wait = 0;
358798f90dbaSZhi Yong Wu         }
358898f90dbaSZhi Yong Wu 
358998f90dbaSZhi Yong Wu         return false;
359098f90dbaSZhi Yong Wu     }
359198f90dbaSZhi Yong Wu 
359298f90dbaSZhi Yong Wu     slice_time = bs->slice_end - bs->slice_start;
359398f90dbaSZhi Yong Wu     slice_time /= (NANOSECONDS_PER_SECOND);
359498f90dbaSZhi Yong Wu     bytes_limit = bps_limit * slice_time;
35955905fbc9SStefan Hajnoczi     bytes_base  = bs->slice_submitted.bytes[is_write];
359698f90dbaSZhi Yong Wu     if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
35975905fbc9SStefan Hajnoczi         bytes_base += bs->slice_submitted.bytes[!is_write];
359898f90dbaSZhi Yong Wu     }
359998f90dbaSZhi Yong Wu 
360098f90dbaSZhi Yong Wu     /* bytes_base: the bytes of data which have been read/written; and
360198f90dbaSZhi Yong Wu      *             it is obtained from the history statistic info.
360298f90dbaSZhi Yong Wu      * bytes_res: the remaining bytes of data which need to be read/written.
360398f90dbaSZhi Yong Wu      * (bytes_base + bytes_res) / bps_limit: used to calcuate
360498f90dbaSZhi Yong Wu      *             the total time for completing reading/writting all data.
360598f90dbaSZhi Yong Wu      */
360698f90dbaSZhi Yong Wu     bytes_res   = (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
360798f90dbaSZhi Yong Wu 
360898f90dbaSZhi Yong Wu     if (bytes_base + bytes_res <= bytes_limit) {
360998f90dbaSZhi Yong Wu         if (wait) {
361098f90dbaSZhi Yong Wu             *wait = 0;
361198f90dbaSZhi Yong Wu         }
361298f90dbaSZhi Yong Wu 
361398f90dbaSZhi Yong Wu         return false;
361498f90dbaSZhi Yong Wu     }
361598f90dbaSZhi Yong Wu 
361698f90dbaSZhi Yong Wu     /* Calc approx time to dispatch */
361798f90dbaSZhi Yong Wu     wait_time = (bytes_base + bytes_res) / bps_limit - elapsed_time;
361898f90dbaSZhi Yong Wu 
361998f90dbaSZhi Yong Wu     /* When the I/O rate at runtime exceeds the limits,
362098f90dbaSZhi Yong Wu      * bs->slice_end need to be extended in order that the current statistic
362198f90dbaSZhi Yong Wu      * info can be kept until the timer fire, so it is increased and tuned
362298f90dbaSZhi Yong Wu      * based on the result of experiment.
362398f90dbaSZhi Yong Wu      */
3624ae29d6c6SStefan Hajnoczi     extension = wait_time * NANOSECONDS_PER_SECOND;
3625ae29d6c6SStefan Hajnoczi     extension = DIV_ROUND_UP(extension, BLOCK_IO_SLICE_TIME) *
3626ae29d6c6SStefan Hajnoczi                 BLOCK_IO_SLICE_TIME;
3627ae29d6c6SStefan Hajnoczi     bs->slice_end += extension;
362898f90dbaSZhi Yong Wu     if (wait) {
36290775437fSStefan Hajnoczi         *wait = wait_time * NANOSECONDS_PER_SECOND;
363098f90dbaSZhi Yong Wu     }
363198f90dbaSZhi Yong Wu 
363298f90dbaSZhi Yong Wu     return true;
363398f90dbaSZhi Yong Wu }
363498f90dbaSZhi Yong Wu 
363598f90dbaSZhi Yong Wu static bool bdrv_exceed_iops_limits(BlockDriverState *bs, bool is_write,
363698f90dbaSZhi Yong Wu                              double elapsed_time, uint64_t *wait)
363798f90dbaSZhi Yong Wu {
363898f90dbaSZhi Yong Wu     uint64_t iops_limit = 0;
363998f90dbaSZhi Yong Wu     double   ios_limit, ios_base;
364098f90dbaSZhi Yong Wu     double   slice_time, wait_time;
364198f90dbaSZhi Yong Wu 
364298f90dbaSZhi Yong Wu     if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
364398f90dbaSZhi Yong Wu         iops_limit = bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL];
364498f90dbaSZhi Yong Wu     } else if (bs->io_limits.iops[is_write]) {
364598f90dbaSZhi Yong Wu         iops_limit = bs->io_limits.iops[is_write];
364698f90dbaSZhi Yong Wu     } else {
364798f90dbaSZhi Yong Wu         if (wait) {
364898f90dbaSZhi Yong Wu             *wait = 0;
364998f90dbaSZhi Yong Wu         }
365098f90dbaSZhi Yong Wu 
365198f90dbaSZhi Yong Wu         return false;
365298f90dbaSZhi Yong Wu     }
365398f90dbaSZhi Yong Wu 
365498f90dbaSZhi Yong Wu     slice_time = bs->slice_end - bs->slice_start;
365598f90dbaSZhi Yong Wu     slice_time /= (NANOSECONDS_PER_SECOND);
365698f90dbaSZhi Yong Wu     ios_limit  = iops_limit * slice_time;
36575905fbc9SStefan Hajnoczi     ios_base   = bs->slice_submitted.ios[is_write];
365898f90dbaSZhi Yong Wu     if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
36595905fbc9SStefan Hajnoczi         ios_base += bs->slice_submitted.ios[!is_write];
366098f90dbaSZhi Yong Wu     }
366198f90dbaSZhi Yong Wu 
366298f90dbaSZhi Yong Wu     if (ios_base + 1 <= ios_limit) {
366398f90dbaSZhi Yong Wu         if (wait) {
366498f90dbaSZhi Yong Wu             *wait = 0;
366598f90dbaSZhi Yong Wu         }
366698f90dbaSZhi Yong Wu 
366798f90dbaSZhi Yong Wu         return false;
366898f90dbaSZhi Yong Wu     }
366998f90dbaSZhi Yong Wu 
36700775437fSStefan Hajnoczi     /* Calc approx time to dispatch, in seconds */
367198f90dbaSZhi Yong Wu     wait_time = (ios_base + 1) / iops_limit;
367298f90dbaSZhi Yong Wu     if (wait_time > elapsed_time) {
367398f90dbaSZhi Yong Wu         wait_time = wait_time - elapsed_time;
367498f90dbaSZhi Yong Wu     } else {
367598f90dbaSZhi Yong Wu         wait_time = 0;
367698f90dbaSZhi Yong Wu     }
367798f90dbaSZhi Yong Wu 
3678ae29d6c6SStefan Hajnoczi     /* Exceeded current slice, extend it by another slice time */
3679ae29d6c6SStefan Hajnoczi     bs->slice_end += BLOCK_IO_SLICE_TIME;
368098f90dbaSZhi Yong Wu     if (wait) {
36810775437fSStefan Hajnoczi         *wait = wait_time * NANOSECONDS_PER_SECOND;
368298f90dbaSZhi Yong Wu     }
368398f90dbaSZhi Yong Wu 
368498f90dbaSZhi Yong Wu     return true;
368598f90dbaSZhi Yong Wu }
368698f90dbaSZhi Yong Wu 
368798f90dbaSZhi Yong Wu static bool bdrv_exceed_io_limits(BlockDriverState *bs, int nb_sectors,
368898f90dbaSZhi Yong Wu                            bool is_write, int64_t *wait)
368998f90dbaSZhi Yong Wu {
369098f90dbaSZhi Yong Wu     int64_t  now, max_wait;
369198f90dbaSZhi Yong Wu     uint64_t bps_wait = 0, iops_wait = 0;
369298f90dbaSZhi Yong Wu     double   elapsed_time;
369398f90dbaSZhi Yong Wu     int      bps_ret, iops_ret;
369498f90dbaSZhi Yong Wu 
369598f90dbaSZhi Yong Wu     now = qemu_get_clock_ns(vm_clock);
3696e660fb8bSStefan Hajnoczi     if (now > bs->slice_end) {
369798f90dbaSZhi Yong Wu         bs->slice_start = now;
3698ae29d6c6SStefan Hajnoczi         bs->slice_end   = now + BLOCK_IO_SLICE_TIME;
36995905fbc9SStefan Hajnoczi         memset(&bs->slice_submitted, 0, sizeof(bs->slice_submitted));
370098f90dbaSZhi Yong Wu     }
370198f90dbaSZhi Yong Wu 
370298f90dbaSZhi Yong Wu     elapsed_time  = now - bs->slice_start;
370398f90dbaSZhi Yong Wu     elapsed_time  /= (NANOSECONDS_PER_SECOND);
370498f90dbaSZhi Yong Wu 
370598f90dbaSZhi Yong Wu     bps_ret  = bdrv_exceed_bps_limits(bs, nb_sectors,
370698f90dbaSZhi Yong Wu                                       is_write, elapsed_time, &bps_wait);
370798f90dbaSZhi Yong Wu     iops_ret = bdrv_exceed_iops_limits(bs, is_write,
370898f90dbaSZhi Yong Wu                                       elapsed_time, &iops_wait);
370998f90dbaSZhi Yong Wu     if (bps_ret || iops_ret) {
371098f90dbaSZhi Yong Wu         max_wait = bps_wait > iops_wait ? bps_wait : iops_wait;
371198f90dbaSZhi Yong Wu         if (wait) {
371298f90dbaSZhi Yong Wu             *wait = max_wait;
371398f90dbaSZhi Yong Wu         }
371498f90dbaSZhi Yong Wu 
371598f90dbaSZhi Yong Wu         now = qemu_get_clock_ns(vm_clock);
371698f90dbaSZhi Yong Wu         if (bs->slice_end < now + max_wait) {
371798f90dbaSZhi Yong Wu             bs->slice_end = now + max_wait;
371898f90dbaSZhi Yong Wu         }
371998f90dbaSZhi Yong Wu 
372098f90dbaSZhi Yong Wu         return true;
372198f90dbaSZhi Yong Wu     }
372298f90dbaSZhi Yong Wu 
372398f90dbaSZhi Yong Wu     if (wait) {
372498f90dbaSZhi Yong Wu         *wait = 0;
372598f90dbaSZhi Yong Wu     }
372698f90dbaSZhi Yong Wu 
37275905fbc9SStefan Hajnoczi     bs->slice_submitted.bytes[is_write] += (int64_t)nb_sectors *
37285905fbc9SStefan Hajnoczi                                            BDRV_SECTOR_SIZE;
37295905fbc9SStefan Hajnoczi     bs->slice_submitted.ios[is_write]++;
37305905fbc9SStefan Hajnoczi 
373198f90dbaSZhi Yong Wu     return false;
373298f90dbaSZhi Yong Wu }
373383f64091Sbellard 
373483f64091Sbellard /**************************************************************/
373583f64091Sbellard /* async block device emulation */
373683f64091Sbellard 
3737c16b5a2cSChristoph Hellwig typedef struct BlockDriverAIOCBSync {
3738c16b5a2cSChristoph Hellwig     BlockDriverAIOCB common;
3739c16b5a2cSChristoph Hellwig     QEMUBH *bh;
3740c16b5a2cSChristoph Hellwig     int ret;
3741c16b5a2cSChristoph Hellwig     /* vector translation state */
3742c16b5a2cSChristoph Hellwig     QEMUIOVector *qiov;
3743c16b5a2cSChristoph Hellwig     uint8_t *bounce;
3744c16b5a2cSChristoph Hellwig     int is_write;
3745c16b5a2cSChristoph Hellwig } BlockDriverAIOCBSync;
3746c16b5a2cSChristoph Hellwig 
3747c16b5a2cSChristoph Hellwig static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
3748c16b5a2cSChristoph Hellwig {
3749b666d239SKevin Wolf     BlockDriverAIOCBSync *acb =
3750b666d239SKevin Wolf         container_of(blockacb, BlockDriverAIOCBSync, common);
37516a7ad299SDor Laor     qemu_bh_delete(acb->bh);
375236afc451SAvi Kivity     acb->bh = NULL;
3753c16b5a2cSChristoph Hellwig     qemu_aio_release(acb);
3754c16b5a2cSChristoph Hellwig }
3755c16b5a2cSChristoph Hellwig 
3756d7331bedSStefan Hajnoczi static const AIOCBInfo bdrv_em_aiocb_info = {
3757c16b5a2cSChristoph Hellwig     .aiocb_size         = sizeof(BlockDriverAIOCBSync),
3758c16b5a2cSChristoph Hellwig     .cancel             = bdrv_aio_cancel_em,
3759c16b5a2cSChristoph Hellwig };
3760c16b5a2cSChristoph Hellwig 
376183f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
3762beac80cdSbellard {
3763ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = opaque;
3764f141eafeSaliguori 
3765f141eafeSaliguori     if (!acb->is_write)
376603396148SMichael Tokarev         qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size);
3767ceb42de8Saliguori     qemu_vfree(acb->bounce);
3768ce1a14dcSpbrook     acb->common.cb(acb->common.opaque, acb->ret);
37696a7ad299SDor Laor     qemu_bh_delete(acb->bh);
377036afc451SAvi Kivity     acb->bh = NULL;
3771ce1a14dcSpbrook     qemu_aio_release(acb);
3772beac80cdSbellard }
3773beac80cdSbellard 
3774f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
3775f141eafeSaliguori                                             int64_t sector_num,
3776f141eafeSaliguori                                             QEMUIOVector *qiov,
3777f141eafeSaliguori                                             int nb_sectors,
3778f141eafeSaliguori                                             BlockDriverCompletionFunc *cb,
3779f141eafeSaliguori                                             void *opaque,
3780f141eafeSaliguori                                             int is_write)
3781f141eafeSaliguori 
3782ea2384d3Sbellard {
3783ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
378483f64091Sbellard 
3785d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_aiocb_info, bs, cb, opaque);
3786f141eafeSaliguori     acb->is_write = is_write;
3787f141eafeSaliguori     acb->qiov = qiov;
3788e268ca52Saliguori     acb->bounce = qemu_blockalign(bs, qiov->size);
3789ce1a14dcSpbrook     acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
3790f141eafeSaliguori 
3791f141eafeSaliguori     if (is_write) {
3792d5e6b161SMichael Tokarev         qemu_iovec_to_buf(acb->qiov, 0, acb->bounce, qiov->size);
37931ed20acfSStefan Hajnoczi         acb->ret = bs->drv->bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
3794f141eafeSaliguori     } else {
37951ed20acfSStefan Hajnoczi         acb->ret = bs->drv->bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
3796f141eafeSaliguori     }
3797f141eafeSaliguori 
3798ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
3799f141eafeSaliguori 
3800ce1a14dcSpbrook     return &acb->common;
38017a6cba61Spbrook }
38027a6cba61Spbrook 
3803f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
3804f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
3805ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
380683f64091Sbellard {
3807f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
380883f64091Sbellard }
380983f64091Sbellard 
3810f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
3811f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
3812f141eafeSaliguori         BlockDriverCompletionFunc *cb, void *opaque)
3813f141eafeSaliguori {
3814f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
3815f141eafeSaliguori }
3816f141eafeSaliguori 
381768485420SKevin Wolf 
381868485420SKevin Wolf typedef struct BlockDriverAIOCBCoroutine {
381968485420SKevin Wolf     BlockDriverAIOCB common;
382068485420SKevin Wolf     BlockRequest req;
382168485420SKevin Wolf     bool is_write;
3822d318aea9SKevin Wolf     bool *done;
382368485420SKevin Wolf     QEMUBH* bh;
382468485420SKevin Wolf } BlockDriverAIOCBCoroutine;
382568485420SKevin Wolf 
382668485420SKevin Wolf static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
382768485420SKevin Wolf {
3828d318aea9SKevin Wolf     BlockDriverAIOCBCoroutine *acb =
3829d318aea9SKevin Wolf         container_of(blockacb, BlockDriverAIOCBCoroutine, common);
3830d318aea9SKevin Wolf     bool done = false;
3831d318aea9SKevin Wolf 
3832d318aea9SKevin Wolf     acb->done = &done;
3833d318aea9SKevin Wolf     while (!done) {
3834d318aea9SKevin Wolf         qemu_aio_wait();
3835d318aea9SKevin Wolf     }
383668485420SKevin Wolf }
383768485420SKevin Wolf 
3838d7331bedSStefan Hajnoczi static const AIOCBInfo bdrv_em_co_aiocb_info = {
383968485420SKevin Wolf     .aiocb_size         = sizeof(BlockDriverAIOCBCoroutine),
384068485420SKevin Wolf     .cancel             = bdrv_aio_co_cancel_em,
384168485420SKevin Wolf };
384268485420SKevin Wolf 
384335246a68SPaolo Bonzini static void bdrv_co_em_bh(void *opaque)
384468485420SKevin Wolf {
384568485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb = opaque;
384668485420SKevin Wolf 
384768485420SKevin Wolf     acb->common.cb(acb->common.opaque, acb->req.error);
3848d318aea9SKevin Wolf 
3849d318aea9SKevin Wolf     if (acb->done) {
3850d318aea9SKevin Wolf         *acb->done = true;
3851d318aea9SKevin Wolf     }
3852d318aea9SKevin Wolf 
385368485420SKevin Wolf     qemu_bh_delete(acb->bh);
385468485420SKevin Wolf     qemu_aio_release(acb);
385568485420SKevin Wolf }
385668485420SKevin Wolf 
3857b2a61371SStefan Hajnoczi /* Invoke bdrv_co_do_readv/bdrv_co_do_writev */
3858b2a61371SStefan Hajnoczi static void coroutine_fn bdrv_co_do_rw(void *opaque)
3859b2a61371SStefan Hajnoczi {
3860b2a61371SStefan Hajnoczi     BlockDriverAIOCBCoroutine *acb = opaque;
3861b2a61371SStefan Hajnoczi     BlockDriverState *bs = acb->common.bs;
3862b2a61371SStefan Hajnoczi 
3863b2a61371SStefan Hajnoczi     if (!acb->is_write) {
3864b2a61371SStefan Hajnoczi         acb->req.error = bdrv_co_do_readv(bs, acb->req.sector,
3865470c0504SStefan Hajnoczi             acb->req.nb_sectors, acb->req.qiov, 0);
3866b2a61371SStefan Hajnoczi     } else {
3867b2a61371SStefan Hajnoczi         acb->req.error = bdrv_co_do_writev(bs, acb->req.sector,
3868f08f2ddaSStefan Hajnoczi             acb->req.nb_sectors, acb->req.qiov, 0);
3869b2a61371SStefan Hajnoczi     }
3870b2a61371SStefan Hajnoczi 
387135246a68SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
3872b2a61371SStefan Hajnoczi     qemu_bh_schedule(acb->bh);
3873b2a61371SStefan Hajnoczi }
3874b2a61371SStefan Hajnoczi 
387568485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
387668485420SKevin Wolf                                                int64_t sector_num,
387768485420SKevin Wolf                                                QEMUIOVector *qiov,
387868485420SKevin Wolf                                                int nb_sectors,
387968485420SKevin Wolf                                                BlockDriverCompletionFunc *cb,
388068485420SKevin Wolf                                                void *opaque,
38818c5873d6SStefan Hajnoczi                                                bool is_write)
388268485420SKevin Wolf {
388368485420SKevin Wolf     Coroutine *co;
388468485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb;
388568485420SKevin Wolf 
3886d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
388768485420SKevin Wolf     acb->req.sector = sector_num;
388868485420SKevin Wolf     acb->req.nb_sectors = nb_sectors;
388968485420SKevin Wolf     acb->req.qiov = qiov;
389068485420SKevin Wolf     acb->is_write = is_write;
3891d318aea9SKevin Wolf     acb->done = NULL;
389268485420SKevin Wolf 
38938c5873d6SStefan Hajnoczi     co = qemu_coroutine_create(bdrv_co_do_rw);
389468485420SKevin Wolf     qemu_coroutine_enter(co, acb);
389568485420SKevin Wolf 
389668485420SKevin Wolf     return &acb->common;
389768485420SKevin Wolf }
389868485420SKevin Wolf 
389907f07615SPaolo Bonzini static void coroutine_fn bdrv_aio_flush_co_entry(void *opaque)
3900b2e12bc6SChristoph Hellwig {
390107f07615SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb = opaque;
390207f07615SPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
3903b2e12bc6SChristoph Hellwig 
390407f07615SPaolo Bonzini     acb->req.error = bdrv_co_flush(bs);
390507f07615SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
3906b2e12bc6SChristoph Hellwig     qemu_bh_schedule(acb->bh);
3907b2e12bc6SChristoph Hellwig }
3908b2e12bc6SChristoph Hellwig 
390907f07615SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
3910016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque)
3911016f5cf6SAlexander Graf {
391207f07615SPaolo Bonzini     trace_bdrv_aio_flush(bs, opaque);
3913016f5cf6SAlexander Graf 
391407f07615SPaolo Bonzini     Coroutine *co;
391507f07615SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb;
3916016f5cf6SAlexander Graf 
3917d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
3918d318aea9SKevin Wolf     acb->done = NULL;
3919d318aea9SKevin Wolf 
392007f07615SPaolo Bonzini     co = qemu_coroutine_create(bdrv_aio_flush_co_entry);
392107f07615SPaolo Bonzini     qemu_coroutine_enter(co, acb);
3922016f5cf6SAlexander Graf 
3923016f5cf6SAlexander Graf     return &acb->common;
3924016f5cf6SAlexander Graf }
3925016f5cf6SAlexander Graf 
39264265d620SPaolo Bonzini static void coroutine_fn bdrv_aio_discard_co_entry(void *opaque)
39274265d620SPaolo Bonzini {
39284265d620SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb = opaque;
39294265d620SPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
39304265d620SPaolo Bonzini 
39314265d620SPaolo Bonzini     acb->req.error = bdrv_co_discard(bs, acb->req.sector, acb->req.nb_sectors);
39324265d620SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
39334265d620SPaolo Bonzini     qemu_bh_schedule(acb->bh);
39344265d620SPaolo Bonzini }
39354265d620SPaolo Bonzini 
39364265d620SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_discard(BlockDriverState *bs,
39374265d620SPaolo Bonzini         int64_t sector_num, int nb_sectors,
39384265d620SPaolo Bonzini         BlockDriverCompletionFunc *cb, void *opaque)
39394265d620SPaolo Bonzini {
39404265d620SPaolo Bonzini     Coroutine *co;
39414265d620SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb;
39424265d620SPaolo Bonzini 
39434265d620SPaolo Bonzini     trace_bdrv_aio_discard(bs, sector_num, nb_sectors, opaque);
39444265d620SPaolo Bonzini 
3945d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
39464265d620SPaolo Bonzini     acb->req.sector = sector_num;
39474265d620SPaolo Bonzini     acb->req.nb_sectors = nb_sectors;
3948d318aea9SKevin Wolf     acb->done = NULL;
39494265d620SPaolo Bonzini     co = qemu_coroutine_create(bdrv_aio_discard_co_entry);
39504265d620SPaolo Bonzini     qemu_coroutine_enter(co, acb);
39514265d620SPaolo Bonzini 
39524265d620SPaolo Bonzini     return &acb->common;
39534265d620SPaolo Bonzini }
39544265d620SPaolo Bonzini 
3955ea2384d3Sbellard void bdrv_init(void)
3956ea2384d3Sbellard {
39575efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
3958ea2384d3Sbellard }
3959ce1a14dcSpbrook 
3960eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
3961eb852011SMarkus Armbruster {
3962eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
3963eb852011SMarkus Armbruster     bdrv_init();
3964eb852011SMarkus Armbruster }
3965eb852011SMarkus Armbruster 
3966d7331bedSStefan Hajnoczi void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs,
39676bbff9a0Saliguori                    BlockDriverCompletionFunc *cb, void *opaque)
39686bbff9a0Saliguori {
3969ce1a14dcSpbrook     BlockDriverAIOCB *acb;
3970ce1a14dcSpbrook 
3971d7331bedSStefan Hajnoczi     acb = g_slice_alloc(aiocb_info->aiocb_size);
3972d7331bedSStefan Hajnoczi     acb->aiocb_info = aiocb_info;
3973ce1a14dcSpbrook     acb->bs = bs;
3974ce1a14dcSpbrook     acb->cb = cb;
3975ce1a14dcSpbrook     acb->opaque = opaque;
3976ce1a14dcSpbrook     return acb;
3977ce1a14dcSpbrook }
3978ce1a14dcSpbrook 
3979ce1a14dcSpbrook void qemu_aio_release(void *p)
3980ce1a14dcSpbrook {
3981d37c975fSStefan Hajnoczi     BlockDriverAIOCB *acb = p;
3982d7331bedSStefan Hajnoczi     g_slice_free1(acb->aiocb_info->aiocb_size, acb);
3983ce1a14dcSpbrook }
398419cb3738Sbellard 
398519cb3738Sbellard /**************************************************************/
3986f9f05dc5SKevin Wolf /* Coroutine block device emulation */
3987f9f05dc5SKevin Wolf 
3988f9f05dc5SKevin Wolf typedef struct CoroutineIOCompletion {
3989f9f05dc5SKevin Wolf     Coroutine *coroutine;
3990f9f05dc5SKevin Wolf     int ret;
3991f9f05dc5SKevin Wolf } CoroutineIOCompletion;
3992f9f05dc5SKevin Wolf 
3993f9f05dc5SKevin Wolf static void bdrv_co_io_em_complete(void *opaque, int ret)
3994f9f05dc5SKevin Wolf {
3995f9f05dc5SKevin Wolf     CoroutineIOCompletion *co = opaque;
3996f9f05dc5SKevin Wolf 
3997f9f05dc5SKevin Wolf     co->ret = ret;
3998f9f05dc5SKevin Wolf     qemu_coroutine_enter(co->coroutine, NULL);
3999f9f05dc5SKevin Wolf }
4000f9f05dc5SKevin Wolf 
4001f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
4002f9f05dc5SKevin Wolf                                       int nb_sectors, QEMUIOVector *iov,
4003f9f05dc5SKevin Wolf                                       bool is_write)
4004f9f05dc5SKevin Wolf {
4005f9f05dc5SKevin Wolf     CoroutineIOCompletion co = {
4006f9f05dc5SKevin Wolf         .coroutine = qemu_coroutine_self(),
4007f9f05dc5SKevin Wolf     };
4008f9f05dc5SKevin Wolf     BlockDriverAIOCB *acb;
4009f9f05dc5SKevin Wolf 
4010f9f05dc5SKevin Wolf     if (is_write) {
4011a652d160SStefan Hajnoczi         acb = bs->drv->bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
4012f9f05dc5SKevin Wolf                                        bdrv_co_io_em_complete, &co);
4013f9f05dc5SKevin Wolf     } else {
4014a652d160SStefan Hajnoczi         acb = bs->drv->bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
4015f9f05dc5SKevin Wolf                                       bdrv_co_io_em_complete, &co);
4016f9f05dc5SKevin Wolf     }
4017f9f05dc5SKevin Wolf 
401859370aaaSStefan Hajnoczi     trace_bdrv_co_io_em(bs, sector_num, nb_sectors, is_write, acb);
4019f9f05dc5SKevin Wolf     if (!acb) {
4020f9f05dc5SKevin Wolf         return -EIO;
4021f9f05dc5SKevin Wolf     }
4022f9f05dc5SKevin Wolf     qemu_coroutine_yield();
4023f9f05dc5SKevin Wolf 
4024f9f05dc5SKevin Wolf     return co.ret;
4025f9f05dc5SKevin Wolf }
4026f9f05dc5SKevin Wolf 
4027f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
4028f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
4029f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
4030f9f05dc5SKevin Wolf {
4031f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
4032f9f05dc5SKevin Wolf }
4033f9f05dc5SKevin Wolf 
4034f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
4035f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
4036f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
4037f9f05dc5SKevin Wolf {
4038f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
4039f9f05dc5SKevin Wolf }
4040f9f05dc5SKevin Wolf 
404107f07615SPaolo Bonzini static void coroutine_fn bdrv_flush_co_entry(void *opaque)
4042e7a8a783SKevin Wolf {
404307f07615SPaolo Bonzini     RwCo *rwco = opaque;
404407f07615SPaolo Bonzini 
404507f07615SPaolo Bonzini     rwco->ret = bdrv_co_flush(rwco->bs);
404607f07615SPaolo Bonzini }
404707f07615SPaolo Bonzini 
404807f07615SPaolo Bonzini int coroutine_fn bdrv_co_flush(BlockDriverState *bs)
404907f07615SPaolo Bonzini {
4050eb489bb1SKevin Wolf     int ret;
4051eb489bb1SKevin Wolf 
405229cdb251SPaolo Bonzini     if (!bs || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
405307f07615SPaolo Bonzini         return 0;
4054eb489bb1SKevin Wolf     }
4055eb489bb1SKevin Wolf 
4056ca716364SKevin Wolf     /* Write back cached data to the OS even with cache=unsafe */
4057bf736fe3SKevin Wolf     BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_OS);
4058eb489bb1SKevin Wolf     if (bs->drv->bdrv_co_flush_to_os) {
4059eb489bb1SKevin Wolf         ret = bs->drv->bdrv_co_flush_to_os(bs);
4060eb489bb1SKevin Wolf         if (ret < 0) {
4061eb489bb1SKevin Wolf             return ret;
4062eb489bb1SKevin Wolf         }
4063eb489bb1SKevin Wolf     }
4064eb489bb1SKevin Wolf 
4065ca716364SKevin Wolf     /* But don't actually force it to the disk with cache=unsafe */
4066ca716364SKevin Wolf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
4067d4c82329SKevin Wolf         goto flush_parent;
4068ca716364SKevin Wolf     }
4069ca716364SKevin Wolf 
4070bf736fe3SKevin Wolf     BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_DISK);
4071eb489bb1SKevin Wolf     if (bs->drv->bdrv_co_flush_to_disk) {
407229cdb251SPaolo Bonzini         ret = bs->drv->bdrv_co_flush_to_disk(bs);
407307f07615SPaolo Bonzini     } else if (bs->drv->bdrv_aio_flush) {
407407f07615SPaolo Bonzini         BlockDriverAIOCB *acb;
4075e7a8a783SKevin Wolf         CoroutineIOCompletion co = {
4076e7a8a783SKevin Wolf             .coroutine = qemu_coroutine_self(),
4077e7a8a783SKevin Wolf         };
4078e7a8a783SKevin Wolf 
407907f07615SPaolo Bonzini         acb = bs->drv->bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
408007f07615SPaolo Bonzini         if (acb == NULL) {
408129cdb251SPaolo Bonzini             ret = -EIO;
408207f07615SPaolo Bonzini         } else {
4083e7a8a783SKevin Wolf             qemu_coroutine_yield();
408429cdb251SPaolo Bonzini             ret = co.ret;
4085e7a8a783SKevin Wolf         }
408607f07615SPaolo Bonzini     } else {
408707f07615SPaolo Bonzini         /*
408807f07615SPaolo Bonzini          * Some block drivers always operate in either writethrough or unsafe
408907f07615SPaolo Bonzini          * mode and don't support bdrv_flush therefore. Usually qemu doesn't
409007f07615SPaolo Bonzini          * know how the server works (because the behaviour is hardcoded or
409107f07615SPaolo Bonzini          * depends on server-side configuration), so we can't ensure that
409207f07615SPaolo Bonzini          * everything is safe on disk. Returning an error doesn't work because
409307f07615SPaolo Bonzini          * that would break guests even if the server operates in writethrough
409407f07615SPaolo Bonzini          * mode.
409507f07615SPaolo Bonzini          *
409607f07615SPaolo Bonzini          * Let's hope the user knows what he's doing.
409707f07615SPaolo Bonzini          */
409829cdb251SPaolo Bonzini         ret = 0;
409907f07615SPaolo Bonzini     }
410029cdb251SPaolo Bonzini     if (ret < 0) {
410129cdb251SPaolo Bonzini         return ret;
410229cdb251SPaolo Bonzini     }
410329cdb251SPaolo Bonzini 
410429cdb251SPaolo Bonzini     /* Now flush the underlying protocol.  It will also have BDRV_O_NO_FLUSH
410529cdb251SPaolo Bonzini      * in the case of cache=unsafe, so there are no useless flushes.
410629cdb251SPaolo Bonzini      */
4107d4c82329SKevin Wolf flush_parent:
410829cdb251SPaolo Bonzini     return bdrv_co_flush(bs->file);
410907f07615SPaolo Bonzini }
411007f07615SPaolo Bonzini 
41110f15423cSAnthony Liguori void bdrv_invalidate_cache(BlockDriverState *bs)
41120f15423cSAnthony Liguori {
41130f15423cSAnthony Liguori     if (bs->drv && bs->drv->bdrv_invalidate_cache) {
41140f15423cSAnthony Liguori         bs->drv->bdrv_invalidate_cache(bs);
41150f15423cSAnthony Liguori     }
41160f15423cSAnthony Liguori }
41170f15423cSAnthony Liguori 
41180f15423cSAnthony Liguori void bdrv_invalidate_cache_all(void)
41190f15423cSAnthony Liguori {
41200f15423cSAnthony Liguori     BlockDriverState *bs;
41210f15423cSAnthony Liguori 
41220f15423cSAnthony Liguori     QTAILQ_FOREACH(bs, &bdrv_states, list) {
41230f15423cSAnthony Liguori         bdrv_invalidate_cache(bs);
41240f15423cSAnthony Liguori     }
41250f15423cSAnthony Liguori }
41260f15423cSAnthony Liguori 
412707789269SBenoît Canet void bdrv_clear_incoming_migration_all(void)
412807789269SBenoît Canet {
412907789269SBenoît Canet     BlockDriverState *bs;
413007789269SBenoît Canet 
413107789269SBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, list) {
413207789269SBenoît Canet         bs->open_flags = bs->open_flags & ~(BDRV_O_INCOMING);
413307789269SBenoît Canet     }
413407789269SBenoît Canet }
413507789269SBenoît Canet 
413607f07615SPaolo Bonzini int bdrv_flush(BlockDriverState *bs)
413707f07615SPaolo Bonzini {
413807f07615SPaolo Bonzini     Coroutine *co;
413907f07615SPaolo Bonzini     RwCo rwco = {
414007f07615SPaolo Bonzini         .bs = bs,
414107f07615SPaolo Bonzini         .ret = NOT_DONE,
414207f07615SPaolo Bonzini     };
414307f07615SPaolo Bonzini 
414407f07615SPaolo Bonzini     if (qemu_in_coroutine()) {
414507f07615SPaolo Bonzini         /* Fast-path if already in coroutine context */
414607f07615SPaolo Bonzini         bdrv_flush_co_entry(&rwco);
414707f07615SPaolo Bonzini     } else {
414807f07615SPaolo Bonzini         co = qemu_coroutine_create(bdrv_flush_co_entry);
414907f07615SPaolo Bonzini         qemu_coroutine_enter(co, &rwco);
415007f07615SPaolo Bonzini         while (rwco.ret == NOT_DONE) {
415107f07615SPaolo Bonzini             qemu_aio_wait();
415207f07615SPaolo Bonzini         }
415307f07615SPaolo Bonzini     }
415407f07615SPaolo Bonzini 
415507f07615SPaolo Bonzini     return rwco.ret;
415607f07615SPaolo Bonzini }
4157e7a8a783SKevin Wolf 
41584265d620SPaolo Bonzini static void coroutine_fn bdrv_discard_co_entry(void *opaque)
41594265d620SPaolo Bonzini {
41604265d620SPaolo Bonzini     RwCo *rwco = opaque;
41614265d620SPaolo Bonzini 
41624265d620SPaolo Bonzini     rwco->ret = bdrv_co_discard(rwco->bs, rwco->sector_num, rwco->nb_sectors);
41634265d620SPaolo Bonzini }
41644265d620SPaolo Bonzini 
41654265d620SPaolo Bonzini int coroutine_fn bdrv_co_discard(BlockDriverState *bs, int64_t sector_num,
41664265d620SPaolo Bonzini                                  int nb_sectors)
41674265d620SPaolo Bonzini {
41684265d620SPaolo Bonzini     if (!bs->drv) {
41694265d620SPaolo Bonzini         return -ENOMEDIUM;
41704265d620SPaolo Bonzini     } else if (bdrv_check_request(bs, sector_num, nb_sectors)) {
41714265d620SPaolo Bonzini         return -EIO;
41724265d620SPaolo Bonzini     } else if (bs->read_only) {
41734265d620SPaolo Bonzini         return -EROFS;
4174df702c9bSPaolo Bonzini     }
4175df702c9bSPaolo Bonzini 
4176df702c9bSPaolo Bonzini     if (bs->dirty_bitmap) {
41778f0720ecSPaolo Bonzini         bdrv_reset_dirty(bs, sector_num, nb_sectors);
4178df702c9bSPaolo Bonzini     }
4179df702c9bSPaolo Bonzini 
41809e8f1835SPaolo Bonzini     /* Do nothing if disabled.  */
41819e8f1835SPaolo Bonzini     if (!(bs->open_flags & BDRV_O_UNMAP)) {
41829e8f1835SPaolo Bonzini         return 0;
41839e8f1835SPaolo Bonzini     }
41849e8f1835SPaolo Bonzini 
4185df702c9bSPaolo Bonzini     if (bs->drv->bdrv_co_discard) {
41864265d620SPaolo Bonzini         return bs->drv->bdrv_co_discard(bs, sector_num, nb_sectors);
41874265d620SPaolo Bonzini     } else if (bs->drv->bdrv_aio_discard) {
41884265d620SPaolo Bonzini         BlockDriverAIOCB *acb;
41894265d620SPaolo Bonzini         CoroutineIOCompletion co = {
41904265d620SPaolo Bonzini             .coroutine = qemu_coroutine_self(),
41914265d620SPaolo Bonzini         };
41924265d620SPaolo Bonzini 
41934265d620SPaolo Bonzini         acb = bs->drv->bdrv_aio_discard(bs, sector_num, nb_sectors,
41944265d620SPaolo Bonzini                                         bdrv_co_io_em_complete, &co);
41954265d620SPaolo Bonzini         if (acb == NULL) {
41964265d620SPaolo Bonzini             return -EIO;
41974265d620SPaolo Bonzini         } else {
41984265d620SPaolo Bonzini             qemu_coroutine_yield();
41994265d620SPaolo Bonzini             return co.ret;
42004265d620SPaolo Bonzini         }
42014265d620SPaolo Bonzini     } else {
42024265d620SPaolo Bonzini         return 0;
42034265d620SPaolo Bonzini     }
42044265d620SPaolo Bonzini }
42054265d620SPaolo Bonzini 
42064265d620SPaolo Bonzini int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
42074265d620SPaolo Bonzini {
42084265d620SPaolo Bonzini     Coroutine *co;
42094265d620SPaolo Bonzini     RwCo rwco = {
42104265d620SPaolo Bonzini         .bs = bs,
42114265d620SPaolo Bonzini         .sector_num = sector_num,
42124265d620SPaolo Bonzini         .nb_sectors = nb_sectors,
42134265d620SPaolo Bonzini         .ret = NOT_DONE,
42144265d620SPaolo Bonzini     };
42154265d620SPaolo Bonzini 
42164265d620SPaolo Bonzini     if (qemu_in_coroutine()) {
42174265d620SPaolo Bonzini         /* Fast-path if already in coroutine context */
42184265d620SPaolo Bonzini         bdrv_discard_co_entry(&rwco);
42194265d620SPaolo Bonzini     } else {
42204265d620SPaolo Bonzini         co = qemu_coroutine_create(bdrv_discard_co_entry);
42214265d620SPaolo Bonzini         qemu_coroutine_enter(co, &rwco);
42224265d620SPaolo Bonzini         while (rwco.ret == NOT_DONE) {
42234265d620SPaolo Bonzini             qemu_aio_wait();
42244265d620SPaolo Bonzini         }
42254265d620SPaolo Bonzini     }
42264265d620SPaolo Bonzini 
42274265d620SPaolo Bonzini     return rwco.ret;
42284265d620SPaolo Bonzini }
42294265d620SPaolo Bonzini 
4230f9f05dc5SKevin Wolf /**************************************************************/
423119cb3738Sbellard /* removable device support */
423219cb3738Sbellard 
423319cb3738Sbellard /**
423419cb3738Sbellard  * Return TRUE if the media is present
423519cb3738Sbellard  */
423619cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
423719cb3738Sbellard {
423819cb3738Sbellard     BlockDriver *drv = bs->drv;
4239a1aff5bfSMarkus Armbruster 
424019cb3738Sbellard     if (!drv)
424119cb3738Sbellard         return 0;
424219cb3738Sbellard     if (!drv->bdrv_is_inserted)
4243a1aff5bfSMarkus Armbruster         return 1;
4244a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
424519cb3738Sbellard }
424619cb3738Sbellard 
424719cb3738Sbellard /**
42488e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
42498e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
425019cb3738Sbellard  */
425119cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
425219cb3738Sbellard {
425319cb3738Sbellard     BlockDriver *drv = bs->drv;
425419cb3738Sbellard 
42558e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
42568e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
42578e49ca46SMarkus Armbruster     }
42588e49ca46SMarkus Armbruster     return -ENOTSUP;
425919cb3738Sbellard }
426019cb3738Sbellard 
426119cb3738Sbellard /**
426219cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
426319cb3738Sbellard  */
4264f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
426519cb3738Sbellard {
426619cb3738Sbellard     BlockDriver *drv = bs->drv;
426719cb3738Sbellard 
4268822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
4269822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
427019cb3738Sbellard     }
42716f382ed2SLuiz Capitulino 
42726f382ed2SLuiz Capitulino     if (bs->device_name[0] != '\0') {
42736f382ed2SLuiz Capitulino         bdrv_emit_qmp_eject_event(bs, eject_flag);
42746f382ed2SLuiz Capitulino     }
427519cb3738Sbellard }
427619cb3738Sbellard 
427719cb3738Sbellard /**
427819cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
427919cb3738Sbellard  * to eject it manually).
428019cb3738Sbellard  */
4281025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
428219cb3738Sbellard {
428319cb3738Sbellard     BlockDriver *drv = bs->drv;
428419cb3738Sbellard 
4285025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
4286b8c6d095SStefan Hajnoczi 
4287025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
4288025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
428919cb3738Sbellard     }
429019cb3738Sbellard }
4291985a03b0Sths 
4292985a03b0Sths /* needed for generic scsi interface */
4293985a03b0Sths 
4294985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
4295985a03b0Sths {
4296985a03b0Sths     BlockDriver *drv = bs->drv;
4297985a03b0Sths 
4298985a03b0Sths     if (drv && drv->bdrv_ioctl)
4299985a03b0Sths         return drv->bdrv_ioctl(bs, req, buf);
4300985a03b0Sths     return -ENOTSUP;
4301985a03b0Sths }
43027d780669Saliguori 
4303221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
4304221f715dSaliguori         unsigned long int req, void *buf,
43057d780669Saliguori         BlockDriverCompletionFunc *cb, void *opaque)
43067d780669Saliguori {
4307221f715dSaliguori     BlockDriver *drv = bs->drv;
43087d780669Saliguori 
4309221f715dSaliguori     if (drv && drv->bdrv_aio_ioctl)
4310221f715dSaliguori         return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
4311221f715dSaliguori     return NULL;
43127d780669Saliguori }
4313e268ca52Saliguori 
43147b6f9300SMarkus Armbruster void bdrv_set_buffer_alignment(BlockDriverState *bs, int align)
43157b6f9300SMarkus Armbruster {
43167b6f9300SMarkus Armbruster     bs->buffer_alignment = align;
43177b6f9300SMarkus Armbruster }
43187cd1e32aSlirans@il.ibm.com 
4319e268ca52Saliguori void *qemu_blockalign(BlockDriverState *bs, size_t size)
4320e268ca52Saliguori {
4321e268ca52Saliguori     return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
4322e268ca52Saliguori }
43237cd1e32aSlirans@il.ibm.com 
4324c53b1c51SStefan Hajnoczi /*
4325c53b1c51SStefan Hajnoczi  * Check if all memory in this vector is sector aligned.
4326c53b1c51SStefan Hajnoczi  */
4327c53b1c51SStefan Hajnoczi bool bdrv_qiov_is_aligned(BlockDriverState *bs, QEMUIOVector *qiov)
4328c53b1c51SStefan Hajnoczi {
4329c53b1c51SStefan Hajnoczi     int i;
4330c53b1c51SStefan Hajnoczi 
4331c53b1c51SStefan Hajnoczi     for (i = 0; i < qiov->niov; i++) {
4332c53b1c51SStefan Hajnoczi         if ((uintptr_t) qiov->iov[i].iov_base % bs->buffer_alignment) {
4333c53b1c51SStefan Hajnoczi             return false;
4334c53b1c51SStefan Hajnoczi         }
4335c53b1c51SStefan Hajnoczi     }
4336c53b1c51SStefan Hajnoczi 
4337c53b1c51SStefan Hajnoczi     return true;
4338c53b1c51SStefan Hajnoczi }
4339c53b1c51SStefan Hajnoczi 
434050717e94SPaolo Bonzini void bdrv_set_dirty_tracking(BlockDriverState *bs, int granularity)
43417cd1e32aSlirans@il.ibm.com {
43427cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
4343a55eb92cSJan Kiszka 
434450717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
434550717e94SPaolo Bonzini 
434650717e94SPaolo Bonzini     if (granularity) {
434750717e94SPaolo Bonzini         granularity >>= BDRV_SECTOR_BITS;
434850717e94SPaolo Bonzini         assert(!bs->dirty_bitmap);
43498f0720ecSPaolo Bonzini         bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS);
435050717e94SPaolo Bonzini         bs->dirty_bitmap = hbitmap_alloc(bitmap_size, ffs(granularity) - 1);
43517cd1e32aSlirans@il.ibm.com     } else {
4352c6d22830SJan Kiszka         if (bs->dirty_bitmap) {
43538f0720ecSPaolo Bonzini             hbitmap_free(bs->dirty_bitmap);
4354c6d22830SJan Kiszka             bs->dirty_bitmap = NULL;
43557cd1e32aSlirans@il.ibm.com         }
43567cd1e32aSlirans@il.ibm.com     }
43577cd1e32aSlirans@il.ibm.com }
43587cd1e32aSlirans@il.ibm.com 
43597cd1e32aSlirans@il.ibm.com int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
43607cd1e32aSlirans@il.ibm.com {
43618f0720ecSPaolo Bonzini     if (bs->dirty_bitmap) {
43628f0720ecSPaolo Bonzini         return hbitmap_get(bs->dirty_bitmap, sector);
43637cd1e32aSlirans@il.ibm.com     } else {
43647cd1e32aSlirans@il.ibm.com         return 0;
43657cd1e32aSlirans@il.ibm.com     }
43667cd1e32aSlirans@il.ibm.com }
43677cd1e32aSlirans@il.ibm.com 
43688f0720ecSPaolo Bonzini void bdrv_dirty_iter_init(BlockDriverState *bs, HBitmapIter *hbi)
43691755da16SPaolo Bonzini {
43708f0720ecSPaolo Bonzini     hbitmap_iter_init(hbi, bs->dirty_bitmap, 0);
43711755da16SPaolo Bonzini }
43721755da16SPaolo Bonzini 
43731755da16SPaolo Bonzini void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
43741755da16SPaolo Bonzini                     int nr_sectors)
43751755da16SPaolo Bonzini {
43768f0720ecSPaolo Bonzini     hbitmap_set(bs->dirty_bitmap, cur_sector, nr_sectors);
43771755da16SPaolo Bonzini }
43781755da16SPaolo Bonzini 
43797cd1e32aSlirans@il.ibm.com void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
43807cd1e32aSlirans@il.ibm.com                       int nr_sectors)
43817cd1e32aSlirans@il.ibm.com {
43828f0720ecSPaolo Bonzini     hbitmap_reset(bs->dirty_bitmap, cur_sector, nr_sectors);
43837cd1e32aSlirans@il.ibm.com }
4384aaa0eb75SLiran Schour 
4385aaa0eb75SLiran Schour int64_t bdrv_get_dirty_count(BlockDriverState *bs)
4386aaa0eb75SLiran Schour {
43878f0720ecSPaolo Bonzini     if (bs->dirty_bitmap) {
4388acc906c6SPaolo Bonzini         return hbitmap_count(bs->dirty_bitmap);
43898f0720ecSPaolo Bonzini     } else {
43908f0720ecSPaolo Bonzini         return 0;
43918f0720ecSPaolo Bonzini     }
4392aaa0eb75SLiran Schour }
4393f88e1a42SJes Sorensen 
4394db593f25SMarcelo Tosatti void bdrv_set_in_use(BlockDriverState *bs, int in_use)
4395db593f25SMarcelo Tosatti {
4396db593f25SMarcelo Tosatti     assert(bs->in_use != in_use);
4397db593f25SMarcelo Tosatti     bs->in_use = in_use;
4398db593f25SMarcelo Tosatti }
4399db593f25SMarcelo Tosatti 
4400db593f25SMarcelo Tosatti int bdrv_in_use(BlockDriverState *bs)
4401db593f25SMarcelo Tosatti {
4402db593f25SMarcelo Tosatti     return bs->in_use;
4403db593f25SMarcelo Tosatti }
4404db593f25SMarcelo Tosatti 
440528a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
440628a7282aSLuiz Capitulino {
4407d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
440858e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
440928a7282aSLuiz Capitulino }
441028a7282aSLuiz Capitulino 
441128a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
441228a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
441328a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
441428a7282aSLuiz Capitulino {
4415d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
441692aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
441792aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
441892aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
441928a7282aSLuiz Capitulino }
442028a7282aSLuiz Capitulino 
442128a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
442228a7282aSLuiz Capitulino {
4423d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
442428a7282aSLuiz Capitulino }
442528a7282aSLuiz Capitulino 
442628a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
442728a7282aSLuiz Capitulino {
442828a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
442958e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
44303bd293c3SPaolo Bonzini         if (bs->job) {
44313bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
44323bd293c3SPaolo Bonzini         }
443328a7282aSLuiz Capitulino     }
443428a7282aSLuiz Capitulino }
443528a7282aSLuiz Capitulino 
443628a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
443728a7282aSLuiz Capitulino {
44383e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
44393e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
444058e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
444158e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
444228a7282aSLuiz Capitulino     }
444328a7282aSLuiz Capitulino }
444428a7282aSLuiz Capitulino 
4445a597e79cSChristoph Hellwig void
4446a597e79cSChristoph Hellwig bdrv_acct_start(BlockDriverState *bs, BlockAcctCookie *cookie, int64_t bytes,
4447a597e79cSChristoph Hellwig         enum BlockAcctType type)
4448a597e79cSChristoph Hellwig {
4449a597e79cSChristoph Hellwig     assert(type < BDRV_MAX_IOTYPE);
4450a597e79cSChristoph Hellwig 
4451a597e79cSChristoph Hellwig     cookie->bytes = bytes;
4452c488c7f6SChristoph Hellwig     cookie->start_time_ns = get_clock();
4453a597e79cSChristoph Hellwig     cookie->type = type;
4454a597e79cSChristoph Hellwig }
4455a597e79cSChristoph Hellwig 
4456a597e79cSChristoph Hellwig void
4457a597e79cSChristoph Hellwig bdrv_acct_done(BlockDriverState *bs, BlockAcctCookie *cookie)
4458a597e79cSChristoph Hellwig {
4459a597e79cSChristoph Hellwig     assert(cookie->type < BDRV_MAX_IOTYPE);
4460a597e79cSChristoph Hellwig 
4461a597e79cSChristoph Hellwig     bs->nr_bytes[cookie->type] += cookie->bytes;
4462a597e79cSChristoph Hellwig     bs->nr_ops[cookie->type]++;
4463c488c7f6SChristoph Hellwig     bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
4464a597e79cSChristoph Hellwig }
4465a597e79cSChristoph Hellwig 
4466d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
4467f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
4468f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
4469f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
4470f88e1a42SJes Sorensen {
4471f88e1a42SJes Sorensen     QEMUOptionParameter *param = NULL, *create_options = NULL;
4472d220894eSKevin Wolf     QEMUOptionParameter *backing_fmt, *backing_file, *size;
4473f88e1a42SJes Sorensen     BlockDriverState *bs = NULL;
4474f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
447596df67d1SStefan Hajnoczi     BlockDriver *backing_drv = NULL;
4476f88e1a42SJes Sorensen     int ret = 0;
4477f88e1a42SJes Sorensen 
4478f88e1a42SJes Sorensen     /* Find driver and parse its options */
4479f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
4480f88e1a42SJes Sorensen     if (!drv) {
448171c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
4482d92ada22SLuiz Capitulino         return;
4483f88e1a42SJes Sorensen     }
4484f88e1a42SJes Sorensen 
448598289620SKevin Wolf     proto_drv = bdrv_find_protocol(filename, true);
4486f88e1a42SJes Sorensen     if (!proto_drv) {
448771c79813SLuiz Capitulino         error_setg(errp, "Unknown protocol '%s'", filename);
4488d92ada22SLuiz Capitulino         return;
4489f88e1a42SJes Sorensen     }
4490f88e1a42SJes Sorensen 
4491f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
4492f88e1a42SJes Sorensen                                               drv->create_options);
4493f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
4494f88e1a42SJes Sorensen                                               proto_drv->create_options);
4495f88e1a42SJes Sorensen 
4496f88e1a42SJes Sorensen     /* Create parameter list with default values */
4497f88e1a42SJes Sorensen     param = parse_option_parameters("", create_options, param);
4498f88e1a42SJes Sorensen 
4499f88e1a42SJes Sorensen     set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
4500f88e1a42SJes Sorensen 
4501f88e1a42SJes Sorensen     /* Parse -o options */
4502f88e1a42SJes Sorensen     if (options) {
4503f88e1a42SJes Sorensen         param = parse_option_parameters(options, create_options, param);
4504f88e1a42SJes Sorensen         if (param == NULL) {
450571c79813SLuiz Capitulino             error_setg(errp, "Invalid options for file format '%s'.", fmt);
4506f88e1a42SJes Sorensen             goto out;
4507f88e1a42SJes Sorensen         }
4508f88e1a42SJes Sorensen     }
4509f88e1a42SJes Sorensen 
4510f88e1a42SJes Sorensen     if (base_filename) {
4511f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
4512f88e1a42SJes Sorensen                                  base_filename)) {
451371c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
451471c79813SLuiz Capitulino                        fmt);
4515f88e1a42SJes Sorensen             goto out;
4516f88e1a42SJes Sorensen         }
4517f88e1a42SJes Sorensen     }
4518f88e1a42SJes Sorensen 
4519f88e1a42SJes Sorensen     if (base_fmt) {
4520f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
452171c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
452271c79813SLuiz Capitulino                              "format '%s'", fmt);
4523f88e1a42SJes Sorensen             goto out;
4524f88e1a42SJes Sorensen         }
4525f88e1a42SJes Sorensen     }
4526f88e1a42SJes Sorensen 
4527792da93aSJes Sorensen     backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
4528792da93aSJes Sorensen     if (backing_file && backing_file->value.s) {
4529792da93aSJes Sorensen         if (!strcmp(filename, backing_file->value.s)) {
453071c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
453171c79813SLuiz Capitulino                              "same filename as the backing file");
4532792da93aSJes Sorensen             goto out;
4533792da93aSJes Sorensen         }
4534792da93aSJes Sorensen     }
4535792da93aSJes Sorensen 
4536f88e1a42SJes Sorensen     backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
4537f88e1a42SJes Sorensen     if (backing_fmt && backing_fmt->value.s) {
453896df67d1SStefan Hajnoczi         backing_drv = bdrv_find_format(backing_fmt->value.s);
453996df67d1SStefan Hajnoczi         if (!backing_drv) {
454071c79813SLuiz Capitulino             error_setg(errp, "Unknown backing file format '%s'",
454171c79813SLuiz Capitulino                        backing_fmt->value.s);
4542f88e1a42SJes Sorensen             goto out;
4543f88e1a42SJes Sorensen         }
4544f88e1a42SJes Sorensen     }
4545f88e1a42SJes Sorensen 
4546f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
4547f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
4548d220894eSKevin Wolf     size = get_option_parameter(param, BLOCK_OPT_SIZE);
4549d220894eSKevin Wolf     if (size && size->value.n == -1) {
4550f88e1a42SJes Sorensen         if (backing_file && backing_file->value.s) {
4551f88e1a42SJes Sorensen             uint64_t size;
4552f88e1a42SJes Sorensen             char buf[32];
455363090dacSPaolo Bonzini             int back_flags;
455463090dacSPaolo Bonzini 
455563090dacSPaolo Bonzini             /* backing files always opened read-only */
455663090dacSPaolo Bonzini             back_flags =
455763090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
4558f88e1a42SJes Sorensen 
4559f88e1a42SJes Sorensen             bs = bdrv_new("");
4560f88e1a42SJes Sorensen 
4561de9c0cecSKevin Wolf             ret = bdrv_open(bs, backing_file->value.s, NULL, back_flags,
4562de9c0cecSKevin Wolf                             backing_drv);
4563f88e1a42SJes Sorensen             if (ret < 0) {
456471c79813SLuiz Capitulino                 error_setg_errno(errp, -ret, "Could not open '%s'",
456571c79813SLuiz Capitulino                                  backing_file->value.s);
4566f88e1a42SJes Sorensen                 goto out;
4567f88e1a42SJes Sorensen             }
4568f88e1a42SJes Sorensen             bdrv_get_geometry(bs, &size);
4569f88e1a42SJes Sorensen             size *= 512;
4570f88e1a42SJes Sorensen 
4571f88e1a42SJes Sorensen             snprintf(buf, sizeof(buf), "%" PRId64, size);
4572f88e1a42SJes Sorensen             set_option_parameter(param, BLOCK_OPT_SIZE, buf);
4573f88e1a42SJes Sorensen         } else {
457471c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
4575f88e1a42SJes Sorensen             goto out;
4576f88e1a42SJes Sorensen         }
4577f88e1a42SJes Sorensen     }
4578f88e1a42SJes Sorensen 
4579f382d43aSMiroslav Rezanina     if (!quiet) {
4580f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
4581f88e1a42SJes Sorensen         print_option_parameters(param);
4582f88e1a42SJes Sorensen         puts("");
4583f382d43aSMiroslav Rezanina     }
4584f88e1a42SJes Sorensen     ret = bdrv_create(drv, filename, param);
4585f88e1a42SJes Sorensen     if (ret < 0) {
4586f88e1a42SJes Sorensen         if (ret == -ENOTSUP) {
458771c79813SLuiz Capitulino             error_setg(errp,"Formatting or formatting option not supported for "
458871c79813SLuiz Capitulino                             "file format '%s'", fmt);
4589f88e1a42SJes Sorensen         } else if (ret == -EFBIG) {
4590f3f4d2c0SKevin Wolf             const char *cluster_size_hint = "";
4591f3f4d2c0SKevin Wolf             if (get_option_parameter(create_options, BLOCK_OPT_CLUSTER_SIZE)) {
4592f3f4d2c0SKevin Wolf                 cluster_size_hint = " (try using a larger cluster size)";
4593f3f4d2c0SKevin Wolf             }
4594f3f4d2c0SKevin Wolf             error_setg(errp, "The image size is too large for file format '%s'%s",
4595f3f4d2c0SKevin Wolf                        fmt, cluster_size_hint);
4596f88e1a42SJes Sorensen         } else {
459771c79813SLuiz Capitulino             error_setg(errp, "%s: error while creating %s: %s", filename, fmt,
459871c79813SLuiz Capitulino                        strerror(-ret));
4599f88e1a42SJes Sorensen         }
4600f88e1a42SJes Sorensen     }
4601f88e1a42SJes Sorensen 
4602f88e1a42SJes Sorensen out:
4603f88e1a42SJes Sorensen     free_option_parameters(create_options);
4604f88e1a42SJes Sorensen     free_option_parameters(param);
4605f88e1a42SJes Sorensen 
4606f88e1a42SJes Sorensen     if (bs) {
4607f88e1a42SJes Sorensen         bdrv_delete(bs);
4608f88e1a42SJes Sorensen     }
4609f88e1a42SJes Sorensen }
461085d126f3SStefan Hajnoczi 
461185d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
461285d126f3SStefan Hajnoczi {
461385d126f3SStefan Hajnoczi     /* Currently BlockDriverState always uses the main loop AioContext */
461485d126f3SStefan Hajnoczi     return qemu_get_aio_context();
461585d126f3SStefan Hajnoczi }
4616d616b224SStefan Hajnoczi 
4617d616b224SStefan Hajnoczi void bdrv_add_before_write_notifier(BlockDriverState *bs,
4618d616b224SStefan Hajnoczi                                     NotifierWithReturn *notifier)
4619d616b224SStefan Hajnoczi {
4620d616b224SStefan Hajnoczi     notifier_with_return_list_add(&bs->before_write_notifiers, notifier);
4621d616b224SStefan Hajnoczi }
4622