xref: /openbmc/qemu/block.c (revision f3f4d2c09b9cf46903ba38425ec46c44185162bd)
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 
102f9092b10SMarkus Armbruster /* The device to use for VM snapshots */
103f9092b10SMarkus Armbruster static BlockDriverState *bs_snapshots;
104f9092b10SMarkus Armbruster 
105eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
106eb852011SMarkus Armbruster static int use_bdrv_whitelist;
107eb852011SMarkus Armbruster 
1089e0b22f4SStefan Hajnoczi #ifdef _WIN32
1099e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
1109e0b22f4SStefan Hajnoczi {
1119e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
1129e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
1139e0b22f4SStefan Hajnoczi             filename[1] == ':');
1149e0b22f4SStefan Hajnoczi }
1159e0b22f4SStefan Hajnoczi 
1169e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
1179e0b22f4SStefan Hajnoczi {
1189e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
1199e0b22f4SStefan Hajnoczi         filename[2] == '\0')
1209e0b22f4SStefan Hajnoczi         return 1;
1219e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
1229e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
1239e0b22f4SStefan Hajnoczi         return 1;
1249e0b22f4SStefan Hajnoczi     return 0;
1259e0b22f4SStefan Hajnoczi }
1269e0b22f4SStefan Hajnoczi #endif
1279e0b22f4SStefan Hajnoczi 
1280563e191SZhi Yong Wu /* throttling disk I/O limits */
12998f90dbaSZhi Yong Wu void bdrv_io_limits_disable(BlockDriverState *bs)
13098f90dbaSZhi Yong Wu {
13198f90dbaSZhi Yong Wu     bs->io_limits_enabled = false;
13298f90dbaSZhi Yong Wu 
13398f90dbaSZhi Yong Wu     while (qemu_co_queue_next(&bs->throttled_reqs));
13498f90dbaSZhi Yong Wu 
13598f90dbaSZhi Yong Wu     if (bs->block_timer) {
13698f90dbaSZhi Yong Wu         qemu_del_timer(bs->block_timer);
13798f90dbaSZhi Yong Wu         qemu_free_timer(bs->block_timer);
13898f90dbaSZhi Yong Wu         bs->block_timer = NULL;
13998f90dbaSZhi Yong Wu     }
14098f90dbaSZhi Yong Wu 
14198f90dbaSZhi Yong Wu     bs->slice_start = 0;
14298f90dbaSZhi Yong Wu     bs->slice_end   = 0;
14398f90dbaSZhi Yong Wu }
14498f90dbaSZhi Yong Wu 
1450563e191SZhi Yong Wu static void bdrv_block_timer(void *opaque)
1460563e191SZhi Yong Wu {
1470563e191SZhi Yong Wu     BlockDriverState *bs = opaque;
1480563e191SZhi Yong Wu 
1490563e191SZhi Yong Wu     qemu_co_queue_next(&bs->throttled_reqs);
1500563e191SZhi Yong Wu }
1510563e191SZhi Yong Wu 
1520563e191SZhi Yong Wu void bdrv_io_limits_enable(BlockDriverState *bs)
1530563e191SZhi Yong Wu {
1540563e191SZhi Yong Wu     qemu_co_queue_init(&bs->throttled_reqs);
1550563e191SZhi Yong Wu     bs->block_timer = qemu_new_timer_ns(vm_clock, bdrv_block_timer, bs);
1560563e191SZhi Yong Wu     bs->io_limits_enabled = true;
1570563e191SZhi Yong Wu }
1580563e191SZhi Yong Wu 
1590563e191SZhi Yong Wu bool bdrv_io_limits_enabled(BlockDriverState *bs)
1600563e191SZhi Yong Wu {
1610563e191SZhi Yong Wu     BlockIOLimit *io_limits = &bs->io_limits;
1620563e191SZhi Yong Wu     return io_limits->bps[BLOCK_IO_LIMIT_READ]
1630563e191SZhi Yong Wu          || io_limits->bps[BLOCK_IO_LIMIT_WRITE]
1640563e191SZhi Yong Wu          || io_limits->bps[BLOCK_IO_LIMIT_TOTAL]
1650563e191SZhi Yong Wu          || io_limits->iops[BLOCK_IO_LIMIT_READ]
1660563e191SZhi Yong Wu          || io_limits->iops[BLOCK_IO_LIMIT_WRITE]
1670563e191SZhi Yong Wu          || io_limits->iops[BLOCK_IO_LIMIT_TOTAL];
1680563e191SZhi Yong Wu }
1690563e191SZhi Yong Wu 
17098f90dbaSZhi Yong Wu static void bdrv_io_limits_intercept(BlockDriverState *bs,
17198f90dbaSZhi Yong Wu                                      bool is_write, int nb_sectors)
17298f90dbaSZhi Yong Wu {
17398f90dbaSZhi Yong Wu     int64_t wait_time = -1;
17498f90dbaSZhi Yong Wu 
17598f90dbaSZhi Yong Wu     if (!qemu_co_queue_empty(&bs->throttled_reqs)) {
17698f90dbaSZhi Yong Wu         qemu_co_queue_wait(&bs->throttled_reqs);
17798f90dbaSZhi Yong Wu     }
17898f90dbaSZhi Yong Wu 
17998f90dbaSZhi Yong Wu     /* In fact, we hope to keep each request's timing, in FIFO mode. The next
18098f90dbaSZhi Yong Wu      * throttled requests will not be dequeued until the current request is
18198f90dbaSZhi Yong Wu      * allowed to be serviced. So if the current request still exceeds the
18298f90dbaSZhi Yong Wu      * limits, it will be inserted to the head. All requests followed it will
18398f90dbaSZhi Yong Wu      * be still in throttled_reqs queue.
18498f90dbaSZhi Yong Wu      */
18598f90dbaSZhi Yong Wu 
18698f90dbaSZhi Yong Wu     while (bdrv_exceed_io_limits(bs, nb_sectors, is_write, &wait_time)) {
18798f90dbaSZhi Yong Wu         qemu_mod_timer(bs->block_timer,
18898f90dbaSZhi Yong Wu                        wait_time + qemu_get_clock_ns(vm_clock));
18998f90dbaSZhi Yong Wu         qemu_co_queue_wait_insert_head(&bs->throttled_reqs);
19098f90dbaSZhi Yong Wu     }
19198f90dbaSZhi Yong Wu 
19298f90dbaSZhi Yong Wu     qemu_co_queue_next(&bs->throttled_reqs);
19398f90dbaSZhi Yong Wu }
19498f90dbaSZhi Yong Wu 
1959e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
1969e0b22f4SStefan Hajnoczi static int path_has_protocol(const char *path)
1979e0b22f4SStefan Hajnoczi {
198947995c0SPaolo Bonzini     const char *p;
199947995c0SPaolo Bonzini 
2009e0b22f4SStefan Hajnoczi #ifdef _WIN32
2019e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
2029e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
2039e0b22f4SStefan Hajnoczi         return 0;
2049e0b22f4SStefan Hajnoczi     }
205947995c0SPaolo Bonzini     p = path + strcspn(path, ":/\\");
206947995c0SPaolo Bonzini #else
207947995c0SPaolo Bonzini     p = path + strcspn(path, ":/");
2089e0b22f4SStefan Hajnoczi #endif
2099e0b22f4SStefan Hajnoczi 
210947995c0SPaolo Bonzini     return *p == ':';
2119e0b22f4SStefan Hajnoczi }
2129e0b22f4SStefan Hajnoczi 
21383f64091Sbellard int path_is_absolute(const char *path)
21483f64091Sbellard {
21521664424Sbellard #ifdef _WIN32
21621664424Sbellard     /* specific case for names like: "\\.\d:" */
217f53f4da9SPaolo Bonzini     if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
21821664424Sbellard         return 1;
219f53f4da9SPaolo Bonzini     }
220f53f4da9SPaolo Bonzini     return (*path == '/' || *path == '\\');
2213b9f94e1Sbellard #else
222f53f4da9SPaolo Bonzini     return (*path == '/');
2233b9f94e1Sbellard #endif
22483f64091Sbellard }
22583f64091Sbellard 
22683f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
22783f64091Sbellard    path to it by considering it is relative to base_path. URL are
22883f64091Sbellard    supported. */
22983f64091Sbellard void path_combine(char *dest, int dest_size,
23083f64091Sbellard                   const char *base_path,
23183f64091Sbellard                   const char *filename)
23283f64091Sbellard {
23383f64091Sbellard     const char *p, *p1;
23483f64091Sbellard     int len;
23583f64091Sbellard 
23683f64091Sbellard     if (dest_size <= 0)
23783f64091Sbellard         return;
23883f64091Sbellard     if (path_is_absolute(filename)) {
23983f64091Sbellard         pstrcpy(dest, dest_size, filename);
24083f64091Sbellard     } else {
24183f64091Sbellard         p = strchr(base_path, ':');
24283f64091Sbellard         if (p)
24383f64091Sbellard             p++;
24483f64091Sbellard         else
24583f64091Sbellard             p = base_path;
2463b9f94e1Sbellard         p1 = strrchr(base_path, '/');
2473b9f94e1Sbellard #ifdef _WIN32
2483b9f94e1Sbellard         {
2493b9f94e1Sbellard             const char *p2;
2503b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
2513b9f94e1Sbellard             if (!p1 || p2 > p1)
2523b9f94e1Sbellard                 p1 = p2;
2533b9f94e1Sbellard         }
2543b9f94e1Sbellard #endif
25583f64091Sbellard         if (p1)
25683f64091Sbellard             p1++;
25783f64091Sbellard         else
25883f64091Sbellard             p1 = base_path;
25983f64091Sbellard         if (p1 > p)
26083f64091Sbellard             p = p1;
26183f64091Sbellard         len = p - base_path;
26283f64091Sbellard         if (len > dest_size - 1)
26383f64091Sbellard             len = dest_size - 1;
26483f64091Sbellard         memcpy(dest, base_path, len);
26583f64091Sbellard         dest[len] = '\0';
26683f64091Sbellard         pstrcat(dest, dest_size, filename);
26783f64091Sbellard     }
26883f64091Sbellard }
26983f64091Sbellard 
270dc5a1371SPaolo Bonzini void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz)
271dc5a1371SPaolo Bonzini {
272dc5a1371SPaolo Bonzini     if (bs->backing_file[0] == '\0' || path_has_protocol(bs->backing_file)) {
273dc5a1371SPaolo Bonzini         pstrcpy(dest, sz, bs->backing_file);
274dc5a1371SPaolo Bonzini     } else {
275dc5a1371SPaolo Bonzini         path_combine(dest, sz, bs->filename, bs->backing_file);
276dc5a1371SPaolo Bonzini     }
277dc5a1371SPaolo Bonzini }
278dc5a1371SPaolo Bonzini 
2795efa9d5aSAnthony Liguori void bdrv_register(BlockDriver *bdrv)
280ea2384d3Sbellard {
2818c5873d6SStefan Hajnoczi     /* Block drivers without coroutine functions need emulation */
2828c5873d6SStefan Hajnoczi     if (!bdrv->bdrv_co_readv) {
283f9f05dc5SKevin Wolf         bdrv->bdrv_co_readv = bdrv_co_readv_em;
284f9f05dc5SKevin Wolf         bdrv->bdrv_co_writev = bdrv_co_writev_em;
285f9f05dc5SKevin Wolf 
286f8c35c1dSStefan Hajnoczi         /* bdrv_co_readv_em()/brdv_co_writev_em() work in terms of aio, so if
287f8c35c1dSStefan Hajnoczi          * the block driver lacks aio we need to emulate that too.
288f8c35c1dSStefan Hajnoczi          */
289f9f05dc5SKevin Wolf         if (!bdrv->bdrv_aio_readv) {
29083f64091Sbellard             /* add AIO emulation layer */
291f141eafeSaliguori             bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
292f141eafeSaliguori             bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
29383f64091Sbellard         }
294f9f05dc5SKevin Wolf     }
295b2e12bc6SChristoph Hellwig 
2968a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
297ea2384d3Sbellard }
298b338082bSbellard 
299b338082bSbellard /* create a new block device (by default it is empty) */
300b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
301fc01f7e7Sbellard {
3021b7bdbc1SStefan Hajnoczi     BlockDriverState *bs;
303b338082bSbellard 
3047267c094SAnthony Liguori     bs = g_malloc0(sizeof(BlockDriverState));
305b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
306ea2384d3Sbellard     if (device_name[0] != '\0') {
3071b7bdbc1SStefan Hajnoczi         QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
308ea2384d3Sbellard     }
30928a7282aSLuiz Capitulino     bdrv_iostatus_disable(bs);
310d7d512f6SPaolo Bonzini     notifier_list_init(&bs->close_notifiers);
311d7d512f6SPaolo Bonzini 
312b338082bSbellard     return bs;
313b338082bSbellard }
314b338082bSbellard 
315d7d512f6SPaolo Bonzini void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
316d7d512f6SPaolo Bonzini {
317d7d512f6SPaolo Bonzini     notifier_list_add(&bs->close_notifiers, notify);
318d7d512f6SPaolo Bonzini }
319d7d512f6SPaolo Bonzini 
320ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
321ea2384d3Sbellard {
322ea2384d3Sbellard     BlockDriver *drv1;
3238a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
3248a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
325ea2384d3Sbellard             return drv1;
326ea2384d3Sbellard         }
3278a22f02aSStefan Hajnoczi     }
328ea2384d3Sbellard     return NULL;
329ea2384d3Sbellard }
330ea2384d3Sbellard 
331eb852011SMarkus Armbruster static int bdrv_is_whitelisted(BlockDriver *drv)
332eb852011SMarkus Armbruster {
333eb852011SMarkus Armbruster     static const char *whitelist[] = {
334eb852011SMarkus Armbruster         CONFIG_BDRV_WHITELIST
335eb852011SMarkus Armbruster     };
336eb852011SMarkus Armbruster     const char **p;
337eb852011SMarkus Armbruster 
338eb852011SMarkus Armbruster     if (!whitelist[0])
339eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
340eb852011SMarkus Armbruster 
341eb852011SMarkus Armbruster     for (p = whitelist; *p; p++) {
342eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
343eb852011SMarkus Armbruster             return 1;
344eb852011SMarkus Armbruster         }
345eb852011SMarkus Armbruster     }
346eb852011SMarkus Armbruster     return 0;
347eb852011SMarkus Armbruster }
348eb852011SMarkus Armbruster 
349eb852011SMarkus Armbruster BlockDriver *bdrv_find_whitelisted_format(const char *format_name)
350eb852011SMarkus Armbruster {
351eb852011SMarkus Armbruster     BlockDriver *drv = bdrv_find_format(format_name);
352eb852011SMarkus Armbruster     return drv && bdrv_is_whitelisted(drv) ? drv : NULL;
353eb852011SMarkus Armbruster }
354eb852011SMarkus Armbruster 
3555b7e1542SZhi Yong Wu typedef struct CreateCo {
3565b7e1542SZhi Yong Wu     BlockDriver *drv;
3575b7e1542SZhi Yong Wu     char *filename;
3585b7e1542SZhi Yong Wu     QEMUOptionParameter *options;
3595b7e1542SZhi Yong Wu     int ret;
3605b7e1542SZhi Yong Wu } CreateCo;
3615b7e1542SZhi Yong Wu 
3625b7e1542SZhi Yong Wu static void coroutine_fn bdrv_create_co_entry(void *opaque)
3635b7e1542SZhi Yong Wu {
3645b7e1542SZhi Yong Wu     CreateCo *cco = opaque;
3655b7e1542SZhi Yong Wu     assert(cco->drv);
3665b7e1542SZhi Yong Wu 
3675b7e1542SZhi Yong Wu     cco->ret = cco->drv->bdrv_create(cco->filename, cco->options);
3685b7e1542SZhi Yong Wu }
3695b7e1542SZhi Yong Wu 
3700e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
3710e7e1989SKevin Wolf     QEMUOptionParameter *options)
372ea2384d3Sbellard {
3735b7e1542SZhi Yong Wu     int ret;
3740e7e1989SKevin Wolf 
3755b7e1542SZhi Yong Wu     Coroutine *co;
3765b7e1542SZhi Yong Wu     CreateCo cco = {
3775b7e1542SZhi Yong Wu         .drv = drv,
3785b7e1542SZhi Yong Wu         .filename = g_strdup(filename),
3795b7e1542SZhi Yong Wu         .options = options,
3805b7e1542SZhi Yong Wu         .ret = NOT_DONE,
3815b7e1542SZhi Yong Wu     };
3825b7e1542SZhi Yong Wu 
3835b7e1542SZhi Yong Wu     if (!drv->bdrv_create) {
38480168bffSLuiz Capitulino         ret = -ENOTSUP;
38580168bffSLuiz Capitulino         goto out;
3865b7e1542SZhi Yong Wu     }
3875b7e1542SZhi Yong Wu 
3885b7e1542SZhi Yong Wu     if (qemu_in_coroutine()) {
3895b7e1542SZhi Yong Wu         /* Fast-path if already in coroutine context */
3905b7e1542SZhi Yong Wu         bdrv_create_co_entry(&cco);
3915b7e1542SZhi Yong Wu     } else {
3925b7e1542SZhi Yong Wu         co = qemu_coroutine_create(bdrv_create_co_entry);
3935b7e1542SZhi Yong Wu         qemu_coroutine_enter(co, &cco);
3945b7e1542SZhi Yong Wu         while (cco.ret == NOT_DONE) {
3955b7e1542SZhi Yong Wu             qemu_aio_wait();
3965b7e1542SZhi Yong Wu         }
3975b7e1542SZhi Yong Wu     }
3985b7e1542SZhi Yong Wu 
3995b7e1542SZhi Yong Wu     ret = cco.ret;
4005b7e1542SZhi Yong Wu 
40180168bffSLuiz Capitulino out:
40280168bffSLuiz Capitulino     g_free(cco.filename);
4035b7e1542SZhi Yong Wu     return ret;
404ea2384d3Sbellard }
405ea2384d3Sbellard 
40684a12e66SChristoph Hellwig int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
40784a12e66SChristoph Hellwig {
40884a12e66SChristoph Hellwig     BlockDriver *drv;
40984a12e66SChristoph Hellwig 
410b50cbabcSMORITA Kazutaka     drv = bdrv_find_protocol(filename);
41184a12e66SChristoph Hellwig     if (drv == NULL) {
41216905d71SStefan Hajnoczi         return -ENOENT;
41384a12e66SChristoph Hellwig     }
41484a12e66SChristoph Hellwig 
41584a12e66SChristoph Hellwig     return bdrv_create(drv, filename, options);
41684a12e66SChristoph Hellwig }
41784a12e66SChristoph Hellwig 
418eba25057SJim Meyering /*
419eba25057SJim Meyering  * Create a uniquely-named empty temporary file.
420eba25057SJim Meyering  * Return 0 upon success, otherwise a negative errno value.
421eba25057SJim Meyering  */
422eba25057SJim Meyering int get_tmp_filename(char *filename, int size)
423eba25057SJim Meyering {
424d5249393Sbellard #ifdef _WIN32
4253b9f94e1Sbellard     char temp_dir[MAX_PATH];
426eba25057SJim Meyering     /* GetTempFileName requires that its output buffer (4th param)
427eba25057SJim Meyering        have length MAX_PATH or greater.  */
428eba25057SJim Meyering     assert(size >= MAX_PATH);
429eba25057SJim Meyering     return (GetTempPath(MAX_PATH, temp_dir)
430eba25057SJim Meyering             && GetTempFileName(temp_dir, "qem", 0, filename)
431eba25057SJim Meyering             ? 0 : -GetLastError());
432d5249393Sbellard #else
433ea2384d3Sbellard     int fd;
4347ccfb2ebSblueswir1     const char *tmpdir;
4350badc1eeSaurel32     tmpdir = getenv("TMPDIR");
4360badc1eeSaurel32     if (!tmpdir)
4370badc1eeSaurel32         tmpdir = "/tmp";
438eba25057SJim Meyering     if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
439eba25057SJim Meyering         return -EOVERFLOW;
440ea2384d3Sbellard     }
441eba25057SJim Meyering     fd = mkstemp(filename);
442fe235a06SDunrong Huang     if (fd < 0) {
443fe235a06SDunrong Huang         return -errno;
444fe235a06SDunrong Huang     }
445fe235a06SDunrong Huang     if (close(fd) != 0) {
446fe235a06SDunrong Huang         unlink(filename);
447eba25057SJim Meyering         return -errno;
448eba25057SJim Meyering     }
449eba25057SJim Meyering     return 0;
450d5249393Sbellard #endif
451eba25057SJim Meyering }
452ea2384d3Sbellard 
453f3a5d3f8SChristoph Hellwig /*
454f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
455f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
456f3a5d3f8SChristoph Hellwig  */
457f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
458f3a5d3f8SChristoph Hellwig {
459508c7cb3SChristoph Hellwig     int score_max = 0, score;
460508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
461f3a5d3f8SChristoph Hellwig 
4628a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
463508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
464508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
465508c7cb3SChristoph Hellwig             if (score > score_max) {
466508c7cb3SChristoph Hellwig                 score_max = score;
467508c7cb3SChristoph Hellwig                 drv = d;
468f3a5d3f8SChristoph Hellwig             }
469508c7cb3SChristoph Hellwig         }
470f3a5d3f8SChristoph Hellwig     }
471f3a5d3f8SChristoph Hellwig 
472508c7cb3SChristoph Hellwig     return drv;
473f3a5d3f8SChristoph Hellwig }
474f3a5d3f8SChristoph Hellwig 
475b50cbabcSMORITA Kazutaka BlockDriver *bdrv_find_protocol(const char *filename)
47684a12e66SChristoph Hellwig {
47784a12e66SChristoph Hellwig     BlockDriver *drv1;
47884a12e66SChristoph Hellwig     char protocol[128];
47984a12e66SChristoph Hellwig     int len;
48084a12e66SChristoph Hellwig     const char *p;
48184a12e66SChristoph Hellwig 
48266f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
48366f82ceeSKevin Wolf 
48439508e7aSChristoph Hellwig     /*
48539508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
48639508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
48739508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
48839508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
48939508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
49039508e7aSChristoph Hellwig      */
49184a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
49239508e7aSChristoph Hellwig     if (drv1) {
49384a12e66SChristoph Hellwig         return drv1;
49484a12e66SChristoph Hellwig     }
49539508e7aSChristoph Hellwig 
4969e0b22f4SStefan Hajnoczi     if (!path_has_protocol(filename)) {
49739508e7aSChristoph Hellwig         return bdrv_find_format("file");
49839508e7aSChristoph Hellwig     }
4999e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
5009e0b22f4SStefan Hajnoczi     assert(p != NULL);
50184a12e66SChristoph Hellwig     len = p - filename;
50284a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
50384a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
50484a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
50584a12e66SChristoph Hellwig     protocol[len] = '\0';
50684a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
50784a12e66SChristoph Hellwig         if (drv1->protocol_name &&
50884a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
50984a12e66SChristoph Hellwig             return drv1;
51084a12e66SChristoph Hellwig         }
51184a12e66SChristoph Hellwig     }
51284a12e66SChristoph Hellwig     return NULL;
51384a12e66SChristoph Hellwig }
51484a12e66SChristoph Hellwig 
515f500a6d3SKevin Wolf static int find_image_format(BlockDriverState *bs, const char *filename,
516f500a6d3SKevin Wolf                              BlockDriver **pdrv)
517ea2384d3Sbellard {
518f500a6d3SKevin Wolf     int score, score_max;
519ea2384d3Sbellard     BlockDriver *drv1, *drv;
52083f64091Sbellard     uint8_t buf[2048];
521f500a6d3SKevin Wolf     int ret = 0;
522f8ea0b00SNicholas Bellinger 
52308a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
5248e895599SPaolo Bonzini     if (bs->sg || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
525c98ac35dSStefan Weil         drv = bdrv_find_format("raw");
526c98ac35dSStefan Weil         if (!drv) {
527c98ac35dSStefan Weil             ret = -ENOENT;
528c98ac35dSStefan Weil         }
529c98ac35dSStefan Weil         *pdrv = drv;
530c98ac35dSStefan Weil         return ret;
5311a396859SNicholas A. Bellinger     }
532f8ea0b00SNicholas Bellinger 
53383f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
534ea2384d3Sbellard     if (ret < 0) {
535c98ac35dSStefan Weil         *pdrv = NULL;
536c98ac35dSStefan Weil         return ret;
537ea2384d3Sbellard     }
538ea2384d3Sbellard 
539ea2384d3Sbellard     score_max = 0;
54084a12e66SChristoph Hellwig     drv = NULL;
5418a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
54283f64091Sbellard         if (drv1->bdrv_probe) {
543ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
544ea2384d3Sbellard             if (score > score_max) {
545ea2384d3Sbellard                 score_max = score;
546ea2384d3Sbellard                 drv = drv1;
547ea2384d3Sbellard             }
548ea2384d3Sbellard         }
54983f64091Sbellard     }
550c98ac35dSStefan Weil     if (!drv) {
551c98ac35dSStefan Weil         ret = -ENOENT;
552c98ac35dSStefan Weil     }
553c98ac35dSStefan Weil     *pdrv = drv;
554c98ac35dSStefan Weil     return ret;
555ea2384d3Sbellard }
556ea2384d3Sbellard 
55751762288SStefan Hajnoczi /**
55851762288SStefan Hajnoczi  * Set the current 'total_sectors' value
55951762288SStefan Hajnoczi  */
56051762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
56151762288SStefan Hajnoczi {
56251762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
56351762288SStefan Hajnoczi 
564396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
565396759adSNicholas Bellinger     if (bs->sg)
566396759adSNicholas Bellinger         return 0;
567396759adSNicholas Bellinger 
56851762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
56951762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
57051762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
57151762288SStefan Hajnoczi         if (length < 0) {
57251762288SStefan Hajnoczi             return length;
57351762288SStefan Hajnoczi         }
57451762288SStefan Hajnoczi         hint = length >> BDRV_SECTOR_BITS;
57551762288SStefan Hajnoczi     }
57651762288SStefan Hajnoczi 
57751762288SStefan Hajnoczi     bs->total_sectors = hint;
57851762288SStefan Hajnoczi     return 0;
57951762288SStefan Hajnoczi }
58051762288SStefan Hajnoczi 
581c3993cdcSStefan Hajnoczi /**
5829e8f1835SPaolo Bonzini  * Set open flags for a given discard mode
5839e8f1835SPaolo Bonzini  *
5849e8f1835SPaolo Bonzini  * Return 0 on success, -1 if the discard mode was invalid.
5859e8f1835SPaolo Bonzini  */
5869e8f1835SPaolo Bonzini int bdrv_parse_discard_flags(const char *mode, int *flags)
5879e8f1835SPaolo Bonzini {
5889e8f1835SPaolo Bonzini     *flags &= ~BDRV_O_UNMAP;
5899e8f1835SPaolo Bonzini 
5909e8f1835SPaolo Bonzini     if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
5919e8f1835SPaolo Bonzini         /* do nothing */
5929e8f1835SPaolo Bonzini     } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
5939e8f1835SPaolo Bonzini         *flags |= BDRV_O_UNMAP;
5949e8f1835SPaolo Bonzini     } else {
5959e8f1835SPaolo Bonzini         return -1;
5969e8f1835SPaolo Bonzini     }
5979e8f1835SPaolo Bonzini 
5989e8f1835SPaolo Bonzini     return 0;
5999e8f1835SPaolo Bonzini }
6009e8f1835SPaolo Bonzini 
6019e8f1835SPaolo Bonzini /**
602c3993cdcSStefan Hajnoczi  * Set open flags for a given cache mode
603c3993cdcSStefan Hajnoczi  *
604c3993cdcSStefan Hajnoczi  * Return 0 on success, -1 if the cache mode was invalid.
605c3993cdcSStefan Hajnoczi  */
606c3993cdcSStefan Hajnoczi int bdrv_parse_cache_flags(const char *mode, int *flags)
607c3993cdcSStefan Hajnoczi {
608c3993cdcSStefan Hajnoczi     *flags &= ~BDRV_O_CACHE_MASK;
609c3993cdcSStefan Hajnoczi 
610c3993cdcSStefan Hajnoczi     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
611c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
61292196b2fSStefan Hajnoczi     } else if (!strcmp(mode, "directsync")) {
61392196b2fSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE;
614c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writeback")) {
615c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
616c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "unsafe")) {
617c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
618c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NO_FLUSH;
619c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writethrough")) {
620c3993cdcSStefan Hajnoczi         /* this is the default */
621c3993cdcSStefan Hajnoczi     } else {
622c3993cdcSStefan Hajnoczi         return -1;
623c3993cdcSStefan Hajnoczi     }
624c3993cdcSStefan Hajnoczi 
625c3993cdcSStefan Hajnoczi     return 0;
626c3993cdcSStefan Hajnoczi }
627c3993cdcSStefan Hajnoczi 
62853fec9d3SStefan Hajnoczi /**
62953fec9d3SStefan Hajnoczi  * The copy-on-read flag is actually a reference count so multiple users may
63053fec9d3SStefan Hajnoczi  * use the feature without worrying about clobbering its previous state.
63153fec9d3SStefan Hajnoczi  * Copy-on-read stays enabled until all users have called to disable it.
63253fec9d3SStefan Hajnoczi  */
63353fec9d3SStefan Hajnoczi void bdrv_enable_copy_on_read(BlockDriverState *bs)
63453fec9d3SStefan Hajnoczi {
63553fec9d3SStefan Hajnoczi     bs->copy_on_read++;
63653fec9d3SStefan Hajnoczi }
63753fec9d3SStefan Hajnoczi 
63853fec9d3SStefan Hajnoczi void bdrv_disable_copy_on_read(BlockDriverState *bs)
63953fec9d3SStefan Hajnoczi {
64053fec9d3SStefan Hajnoczi     assert(bs->copy_on_read > 0);
64153fec9d3SStefan Hajnoczi     bs->copy_on_read--;
64253fec9d3SStefan Hajnoczi }
64353fec9d3SStefan Hajnoczi 
6447b272452SKevin Wolf static int bdrv_open_flags(BlockDriverState *bs, int flags)
6457b272452SKevin Wolf {
6467b272452SKevin Wolf     int open_flags = flags | BDRV_O_CACHE_WB;
6477b272452SKevin Wolf 
6487b272452SKevin Wolf     /*
6497b272452SKevin Wolf      * Clear flags that are internal to the block layer before opening the
6507b272452SKevin Wolf      * image.
6517b272452SKevin Wolf      */
6527b272452SKevin Wolf     open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
6537b272452SKevin Wolf 
6547b272452SKevin Wolf     /*
6557b272452SKevin Wolf      * Snapshots should be writable.
6567b272452SKevin Wolf      */
6577b272452SKevin Wolf     if (bs->is_temporary) {
6587b272452SKevin Wolf         open_flags |= BDRV_O_RDWR;
6597b272452SKevin Wolf     }
6607b272452SKevin Wolf 
6617b272452SKevin Wolf     return open_flags;
6627b272452SKevin Wolf }
6637b272452SKevin Wolf 
664b6ce07aaSKevin Wolf /*
66557915332SKevin Wolf  * Common part for opening disk images and files
666b6ad491aSKevin Wolf  *
667b6ad491aSKevin Wolf  * Removes all processed options from *options.
66857915332SKevin Wolf  */
669f500a6d3SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
670035fccdfSKevin Wolf     QDict *options, int flags, BlockDriver *drv)
67157915332SKevin Wolf {
67257915332SKevin Wolf     int ret, open_flags;
673035fccdfSKevin Wolf     const char *filename;
67457915332SKevin Wolf 
67557915332SKevin Wolf     assert(drv != NULL);
6766405875cSPaolo Bonzini     assert(bs->file == NULL);
677707ff828SKevin Wolf     assert(options != NULL && bs->options != options);
67857915332SKevin Wolf 
67945673671SKevin Wolf     if (file != NULL) {
68045673671SKevin Wolf         filename = file->filename;
68145673671SKevin Wolf     } else {
68245673671SKevin Wolf         filename = qdict_get_try_str(options, "filename");
68345673671SKevin Wolf     }
68445673671SKevin Wolf 
68545673671SKevin Wolf     trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
68628dcee10SStefan Hajnoczi 
6875d186eb0SKevin Wolf     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv)) {
6885d186eb0SKevin Wolf         return -ENOTSUP;
6895d186eb0SKevin Wolf     }
6905d186eb0SKevin Wolf 
6915d186eb0SKevin Wolf     /* bdrv_open() with directly using a protocol as drv. This layer is already
6925d186eb0SKevin Wolf      * opened, so assign it to bs (while file becomes a closed BlockDriverState)
6935d186eb0SKevin Wolf      * and return immediately. */
6945d186eb0SKevin Wolf     if (file != NULL && drv->bdrv_file_open) {
6955d186eb0SKevin Wolf         bdrv_swap(file, bs);
6965d186eb0SKevin Wolf         return 0;
6975d186eb0SKevin Wolf     }
6985d186eb0SKevin Wolf 
69957915332SKevin Wolf     bs->open_flags = flags;
70057915332SKevin Wolf     bs->buffer_alignment = 512;
70157915332SKevin Wolf 
70253fec9d3SStefan Hajnoczi     assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
70353fec9d3SStefan Hajnoczi     if ((flags & BDRV_O_RDWR) && (flags & BDRV_O_COPY_ON_READ)) {
70453fec9d3SStefan Hajnoczi         bdrv_enable_copy_on_read(bs);
70553fec9d3SStefan Hajnoczi     }
70653fec9d3SStefan Hajnoczi 
707c2ad1b0cSKevin Wolf     if (filename != NULL) {
70857915332SKevin Wolf         pstrcpy(bs->filename, sizeof(bs->filename), filename);
709c2ad1b0cSKevin Wolf     } else {
710c2ad1b0cSKevin Wolf         bs->filename[0] = '\0';
711c2ad1b0cSKevin Wolf     }
71257915332SKevin Wolf 
71357915332SKevin Wolf     bs->drv = drv;
7147267c094SAnthony Liguori     bs->opaque = g_malloc0(drv->instance_size);
71557915332SKevin Wolf 
71603f541bdSStefan Hajnoczi     bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
7177b272452SKevin Wolf     open_flags = bdrv_open_flags(bs, flags);
71857915332SKevin Wolf 
719be028adcSJeff Cody     bs->read_only = !(open_flags & BDRV_O_RDWR);
720e7c63796SStefan Hajnoczi 
72166f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
72266f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
7235d186eb0SKevin Wolf         assert(file == NULL);
724c2ad1b0cSKevin Wolf         assert(drv->bdrv_parse_filename || filename != NULL);
72556d1b4d2SKevin Wolf         ret = drv->bdrv_file_open(bs, options, open_flags);
726f500a6d3SKevin Wolf     } else {
7272af5ef70SKevin Wolf         if (file == NULL) {
7282af5ef70SKevin Wolf             qerror_report(ERROR_CLASS_GENERIC_ERROR, "Can't use '%s' as a "
7292af5ef70SKevin Wolf                           "block driver for the protocol level",
7302af5ef70SKevin Wolf                           drv->format_name);
7312af5ef70SKevin Wolf             ret = -EINVAL;
7322af5ef70SKevin Wolf             goto free_and_fail;
7332af5ef70SKevin Wolf         }
734f500a6d3SKevin Wolf         assert(file != NULL);
735f500a6d3SKevin Wolf         bs->file = file;
736b6ad491aSKevin Wolf         ret = drv->bdrv_open(bs, options, open_flags);
73766f82ceeSKevin Wolf     }
73866f82ceeSKevin Wolf 
73957915332SKevin Wolf     if (ret < 0) {
74057915332SKevin Wolf         goto free_and_fail;
74157915332SKevin Wolf     }
74257915332SKevin Wolf 
74351762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
74451762288SStefan Hajnoczi     if (ret < 0) {
74551762288SStefan Hajnoczi         goto free_and_fail;
74657915332SKevin Wolf     }
74751762288SStefan Hajnoczi 
74857915332SKevin Wolf #ifndef _WIN32
74957915332SKevin Wolf     if (bs->is_temporary) {
750c2ad1b0cSKevin Wolf         assert(filename != NULL);
75157915332SKevin Wolf         unlink(filename);
75257915332SKevin Wolf     }
75357915332SKevin Wolf #endif
75457915332SKevin Wolf     return 0;
75557915332SKevin Wolf 
75657915332SKevin Wolf free_and_fail:
75766f82ceeSKevin Wolf     bs->file = NULL;
7587267c094SAnthony Liguori     g_free(bs->opaque);
75957915332SKevin Wolf     bs->opaque = NULL;
76057915332SKevin Wolf     bs->drv = NULL;
76157915332SKevin Wolf     return ret;
76257915332SKevin Wolf }
76357915332SKevin Wolf 
76457915332SKevin Wolf /*
765b6ce07aaSKevin Wolf  * Opens a file using a protocol (file, host_device, nbd, ...)
766787e4a85SKevin Wolf  *
767787e4a85SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
768787e4a85SKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
769787e4a85SKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
770787e4a85SKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_file_open.
771b6ce07aaSKevin Wolf  */
772787e4a85SKevin Wolf int bdrv_file_open(BlockDriverState **pbs, const char *filename,
773787e4a85SKevin Wolf                    QDict *options, int flags)
774b338082bSbellard {
77583f64091Sbellard     BlockDriverState *bs;
7766db95603SChristoph Hellwig     BlockDriver *drv;
777c2ad1b0cSKevin Wolf     const char *drvname;
77883f64091Sbellard     int ret;
7793b0d4f61Sbellard 
780707ff828SKevin Wolf     /* NULL means an empty set of options */
781707ff828SKevin Wolf     if (options == NULL) {
782707ff828SKevin Wolf         options = qdict_new();
7833b0d4f61Sbellard     }
784707ff828SKevin Wolf 
785707ff828SKevin Wolf     bs = bdrv_new("");
786707ff828SKevin Wolf     bs->options = options;
787707ff828SKevin Wolf     options = qdict_clone_shallow(options);
788707ff828SKevin Wolf 
789035fccdfSKevin Wolf     /* Fetch the file name from the options QDict if necessary */
790035fccdfSKevin Wolf     if (!filename) {
791035fccdfSKevin Wolf         filename = qdict_get_try_str(options, "filename");
792035fccdfSKevin Wolf     } else if (filename && !qdict_haskey(options, "filename")) {
793035fccdfSKevin Wolf         qdict_put(options, "filename", qstring_from_str(filename));
794035fccdfSKevin Wolf     } else {
795035fccdfSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR, "Can't specify 'file' and "
796035fccdfSKevin Wolf                       "'filename' options at the same time");
797035fccdfSKevin Wolf         ret = -EINVAL;
798035fccdfSKevin Wolf         goto fail;
799035fccdfSKevin Wolf     }
800035fccdfSKevin Wolf 
801c2ad1b0cSKevin Wolf     /* Find the right block driver */
802c2ad1b0cSKevin Wolf     drvname = qdict_get_try_str(options, "driver");
803c2ad1b0cSKevin Wolf     if (drvname) {
804c2ad1b0cSKevin Wolf         drv = bdrv_find_whitelisted_format(drvname);
805c2ad1b0cSKevin Wolf         qdict_del(options, "driver");
806c2ad1b0cSKevin Wolf     } else if (filename) {
807c2ad1b0cSKevin Wolf         drv = bdrv_find_protocol(filename);
808c2ad1b0cSKevin Wolf     } else {
809c2ad1b0cSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR,
810c2ad1b0cSKevin Wolf                       "Must specify either driver or file");
811c2ad1b0cSKevin Wolf         drv = NULL;
812c2ad1b0cSKevin Wolf     }
813c2ad1b0cSKevin Wolf 
814c2ad1b0cSKevin Wolf     if (!drv) {
815c2ad1b0cSKevin Wolf         ret = -ENOENT;
816c2ad1b0cSKevin Wolf         goto fail;
817c2ad1b0cSKevin Wolf     }
818c2ad1b0cSKevin Wolf 
819c2ad1b0cSKevin Wolf     /* Parse the filename and open it */
820c2ad1b0cSKevin Wolf     if (drv->bdrv_parse_filename && filename) {
8216963a30dSKevin Wolf         Error *local_err = NULL;
8226963a30dSKevin Wolf         drv->bdrv_parse_filename(filename, options, &local_err);
8236963a30dSKevin Wolf         if (error_is_set(&local_err)) {
8246963a30dSKevin Wolf             qerror_report_err(local_err);
8256963a30dSKevin Wolf             error_free(local_err);
8266963a30dSKevin Wolf             ret = -EINVAL;
8276963a30dSKevin Wolf             goto fail;
8286963a30dSKevin Wolf         }
82956d1b4d2SKevin Wolf         qdict_del(options, "filename");
830c2ad1b0cSKevin Wolf     } else if (!drv->bdrv_parse_filename && !filename) {
831c2ad1b0cSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR,
832c2ad1b0cSKevin Wolf                       "The '%s' block driver requires a file name",
833c2ad1b0cSKevin Wolf                       drv->format_name);
834c2ad1b0cSKevin Wolf         ret = -EINVAL;
835c2ad1b0cSKevin Wolf         goto fail;
8366963a30dSKevin Wolf     }
8376963a30dSKevin Wolf 
838035fccdfSKevin Wolf     ret = bdrv_open_common(bs, NULL, options, flags, drv);
839707ff828SKevin Wolf     if (ret < 0) {
840707ff828SKevin Wolf         goto fail;
841707ff828SKevin Wolf     }
842707ff828SKevin Wolf 
843707ff828SKevin Wolf     /* Check if any unknown options were used */
844707ff828SKevin Wolf     if (qdict_size(options) != 0) {
845707ff828SKevin Wolf         const QDictEntry *entry = qdict_first(options);
846707ff828SKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR, "Block protocol '%s' doesn't "
847707ff828SKevin Wolf                       "support the option '%s'",
848707ff828SKevin Wolf                       drv->format_name, entry->key);
849707ff828SKevin Wolf         ret = -EINVAL;
850707ff828SKevin Wolf         goto fail;
851707ff828SKevin Wolf     }
852707ff828SKevin Wolf     QDECREF(options);
853707ff828SKevin Wolf 
85471d0770cSaliguori     bs->growable = 1;
85583f64091Sbellard     *pbs = bs;
85683f64091Sbellard     return 0;
857707ff828SKevin Wolf 
858707ff828SKevin Wolf fail:
859707ff828SKevin Wolf     QDECREF(options);
860707ff828SKevin Wolf     if (!bs->drv) {
861707ff828SKevin Wolf         QDECREF(bs->options);
862707ff828SKevin Wolf     }
863707ff828SKevin Wolf     bdrv_delete(bs);
864707ff828SKevin Wolf     return ret;
8653b0d4f61Sbellard }
8663b0d4f61Sbellard 
86731ca6d07SKevin Wolf /*
86831ca6d07SKevin Wolf  * Opens the backing file for a BlockDriverState if not yet open
86931ca6d07SKevin Wolf  *
87031ca6d07SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
87131ca6d07SKevin Wolf  * empty set of options. The reference to the QDict is transferred to this
87231ca6d07SKevin Wolf  * function (even on failure), so if the caller intends to reuse the dictionary,
87331ca6d07SKevin Wolf  * it needs to use QINCREF() before calling bdrv_file_open.
87431ca6d07SKevin Wolf  */
87531ca6d07SKevin Wolf int bdrv_open_backing_file(BlockDriverState *bs, QDict *options)
8769156df12SPaolo Bonzini {
8779156df12SPaolo Bonzini     char backing_filename[PATH_MAX];
8789156df12SPaolo Bonzini     int back_flags, ret;
8799156df12SPaolo Bonzini     BlockDriver *back_drv = NULL;
8809156df12SPaolo Bonzini 
8819156df12SPaolo Bonzini     if (bs->backing_hd != NULL) {
88231ca6d07SKevin Wolf         QDECREF(options);
8839156df12SPaolo Bonzini         return 0;
8849156df12SPaolo Bonzini     }
8859156df12SPaolo Bonzini 
88631ca6d07SKevin Wolf     /* NULL means an empty set of options */
88731ca6d07SKevin Wolf     if (options == NULL) {
88831ca6d07SKevin Wolf         options = qdict_new();
88931ca6d07SKevin Wolf     }
89031ca6d07SKevin Wolf 
8919156df12SPaolo Bonzini     bs->open_flags &= ~BDRV_O_NO_BACKING;
8921cb6f506SKevin Wolf     if (qdict_haskey(options, "file.filename")) {
8931cb6f506SKevin Wolf         backing_filename[0] = '\0';
8941cb6f506SKevin Wolf     } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
89531ca6d07SKevin Wolf         QDECREF(options);
8969156df12SPaolo Bonzini         return 0;
8979156df12SPaolo Bonzini     }
8989156df12SPaolo Bonzini 
8999156df12SPaolo Bonzini     bs->backing_hd = bdrv_new("");
9009156df12SPaolo Bonzini     bdrv_get_full_backing_filename(bs, backing_filename,
9019156df12SPaolo Bonzini                                    sizeof(backing_filename));
9029156df12SPaolo Bonzini 
9039156df12SPaolo Bonzini     if (bs->backing_format[0] != '\0') {
9049156df12SPaolo Bonzini         back_drv = bdrv_find_format(bs->backing_format);
9059156df12SPaolo Bonzini     }
9069156df12SPaolo Bonzini 
9079156df12SPaolo Bonzini     /* backing files always opened read-only */
9089156df12SPaolo Bonzini     back_flags = bs->open_flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT);
9099156df12SPaolo Bonzini 
91031ca6d07SKevin Wolf     ret = bdrv_open(bs->backing_hd,
91131ca6d07SKevin Wolf                     *backing_filename ? backing_filename : NULL, options,
912de9c0cecSKevin Wolf                     back_flags, back_drv);
9139156df12SPaolo Bonzini     if (ret < 0) {
9149156df12SPaolo Bonzini         bdrv_delete(bs->backing_hd);
9159156df12SPaolo Bonzini         bs->backing_hd = NULL;
9169156df12SPaolo Bonzini         bs->open_flags |= BDRV_O_NO_BACKING;
9179156df12SPaolo Bonzini         return ret;
9189156df12SPaolo Bonzini     }
9199156df12SPaolo Bonzini     return 0;
9209156df12SPaolo Bonzini }
9219156df12SPaolo Bonzini 
922707ff828SKevin Wolf static void extract_subqdict(QDict *src, QDict **dst, const char *start)
923707ff828SKevin Wolf {
924707ff828SKevin Wolf     const QDictEntry *entry, *next;
925707ff828SKevin Wolf     const char *p;
926707ff828SKevin Wolf 
927707ff828SKevin Wolf     *dst = qdict_new();
928707ff828SKevin Wolf     entry = qdict_first(src);
929707ff828SKevin Wolf 
930707ff828SKevin Wolf     while (entry != NULL) {
931707ff828SKevin Wolf         next = qdict_next(src, entry);
932707ff828SKevin Wolf         if (strstart(entry->key, start, &p)) {
933707ff828SKevin Wolf             qobject_incref(entry->value);
934707ff828SKevin Wolf             qdict_put_obj(*dst, p, entry->value);
935707ff828SKevin Wolf             qdict_del(src, entry->key);
936707ff828SKevin Wolf         }
937707ff828SKevin Wolf         entry = next;
938707ff828SKevin Wolf     }
939707ff828SKevin Wolf }
940707ff828SKevin Wolf 
941b6ce07aaSKevin Wolf /*
942b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
943de9c0cecSKevin Wolf  *
944de9c0cecSKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
945de9c0cecSKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
946de9c0cecSKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
947de9c0cecSKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_open.
948b6ce07aaSKevin Wolf  */
949de9c0cecSKevin Wolf int bdrv_open(BlockDriverState *bs, const char *filename, QDict *options,
950de9c0cecSKevin Wolf               int flags, BlockDriver *drv)
951ea2384d3Sbellard {
952b6ce07aaSKevin Wolf     int ret;
95389c9bc3dSStefan Weil     /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
95489c9bc3dSStefan Weil     char tmp_filename[PATH_MAX + 1];
955f500a6d3SKevin Wolf     BlockDriverState *file = NULL;
956707ff828SKevin Wolf     QDict *file_options = NULL;
95733e3963eSbellard 
958de9c0cecSKevin Wolf     /* NULL means an empty set of options */
959de9c0cecSKevin Wolf     if (options == NULL) {
960de9c0cecSKevin Wolf         options = qdict_new();
961de9c0cecSKevin Wolf     }
962de9c0cecSKevin Wolf 
963de9c0cecSKevin Wolf     bs->options = options;
964b6ad491aSKevin Wolf     options = qdict_clone_shallow(options);
965de9c0cecSKevin Wolf 
966de9c0cecSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay */
96783f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
968ea2384d3Sbellard         BlockDriverState *bs1;
969ea2384d3Sbellard         int64_t total_size;
97091a073a9SKevin Wolf         BlockDriver *bdrv_qcow2;
97108b392e1SKevin Wolf         QEMUOptionParameter *create_options;
972b6ce07aaSKevin Wolf         char backing_filename[PATH_MAX];
97333e3963eSbellard 
974c2ad1b0cSKevin Wolf         if (qdict_size(options) != 0) {
975c2ad1b0cSKevin Wolf             error_report("Can't use snapshot=on with driver-specific options");
976c2ad1b0cSKevin Wolf             ret = -EINVAL;
977c2ad1b0cSKevin Wolf             goto fail;
978c2ad1b0cSKevin Wolf         }
979c2ad1b0cSKevin Wolf         assert(filename != NULL);
980c2ad1b0cSKevin Wolf 
981ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
982ea2384d3Sbellard            instead of opening 'filename' directly */
983ea2384d3Sbellard 
984ea2384d3Sbellard         /* if there is a backing file, use it */
985ea2384d3Sbellard         bs1 = bdrv_new("");
986de9c0cecSKevin Wolf         ret = bdrv_open(bs1, filename, NULL, 0, drv);
98751d7c00cSaliguori         if (ret < 0) {
988ea2384d3Sbellard             bdrv_delete(bs1);
989de9c0cecSKevin Wolf             goto fail;
990ea2384d3Sbellard         }
9913e82990bSJes Sorensen         total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
9927c96d46eSaliguori 
993ea2384d3Sbellard         bdrv_delete(bs1);
994ea2384d3Sbellard 
995eba25057SJim Meyering         ret = get_tmp_filename(tmp_filename, sizeof(tmp_filename));
996eba25057SJim Meyering         if (ret < 0) {
997de9c0cecSKevin Wolf             goto fail;
998eba25057SJim Meyering         }
9997c96d46eSaliguori 
10007c96d46eSaliguori         /* Real path is meaningless for protocols */
10014d70655bSStefan Hajnoczi         if (path_has_protocol(filename)) {
10027c96d46eSaliguori             snprintf(backing_filename, sizeof(backing_filename),
10037c96d46eSaliguori                      "%s", filename);
1004de9c0cecSKevin Wolf         } else if (!realpath(filename, backing_filename)) {
1005de9c0cecSKevin Wolf             ret = -errno;
1006de9c0cecSKevin Wolf             goto fail;
1007de9c0cecSKevin Wolf         }
10087c96d46eSaliguori 
100991a073a9SKevin Wolf         bdrv_qcow2 = bdrv_find_format("qcow2");
101008b392e1SKevin Wolf         create_options = parse_option_parameters("", bdrv_qcow2->create_options,
101108b392e1SKevin Wolf                                                  NULL);
101291a073a9SKevin Wolf 
101308b392e1SKevin Wolf         set_option_parameter_int(create_options, BLOCK_OPT_SIZE, total_size);
101408b392e1SKevin Wolf         set_option_parameter(create_options, BLOCK_OPT_BACKING_FILE,
101508b392e1SKevin Wolf                              backing_filename);
101691a073a9SKevin Wolf         if (drv) {
101708b392e1SKevin Wolf             set_option_parameter(create_options, BLOCK_OPT_BACKING_FMT,
101891a073a9SKevin Wolf                 drv->format_name);
101991a073a9SKevin Wolf         }
102091a073a9SKevin Wolf 
102108b392e1SKevin Wolf         ret = bdrv_create(bdrv_qcow2, tmp_filename, create_options);
102208b392e1SKevin Wolf         free_option_parameters(create_options);
102351d7c00cSaliguori         if (ret < 0) {
1024de9c0cecSKevin Wolf             goto fail;
1025ea2384d3Sbellard         }
102691a073a9SKevin Wolf 
1027ea2384d3Sbellard         filename = tmp_filename;
102891a073a9SKevin Wolf         drv = bdrv_qcow2;
1029ea2384d3Sbellard         bs->is_temporary = 1;
1030ea2384d3Sbellard     }
1031ea2384d3Sbellard 
1032f500a6d3SKevin Wolf     /* Open image file without format layer */
1033be028adcSJeff Cody     if (flags & BDRV_O_RDWR) {
1034be028adcSJeff Cody         flags |= BDRV_O_ALLOW_RDWR;
1035be028adcSJeff Cody     }
1036be028adcSJeff Cody 
1037707ff828SKevin Wolf     extract_subqdict(options, &file_options, "file.");
1038707ff828SKevin Wolf 
1039707ff828SKevin Wolf     ret = bdrv_file_open(&file, filename, file_options,
1040707ff828SKevin Wolf                          bdrv_open_flags(bs, flags));
1041f500a6d3SKevin Wolf     if (ret < 0) {
1042de9c0cecSKevin Wolf         goto fail;
1043f500a6d3SKevin Wolf     }
1044f500a6d3SKevin Wolf 
1045f500a6d3SKevin Wolf     /* Find the right image format driver */
1046f500a6d3SKevin Wolf     if (!drv) {
1047f500a6d3SKevin Wolf         ret = find_image_format(file, filename, &drv);
1048f500a6d3SKevin Wolf     }
1049f500a6d3SKevin Wolf 
1050f500a6d3SKevin Wolf     if (!drv) {
1051f500a6d3SKevin Wolf         goto unlink_and_fail;
1052f500a6d3SKevin Wolf     }
1053f500a6d3SKevin Wolf 
1054b6ce07aaSKevin Wolf     /* Open the image */
1055035fccdfSKevin Wolf     ret = bdrv_open_common(bs, file, options, flags, drv);
1056b6ce07aaSKevin Wolf     if (ret < 0) {
10576987307cSChristoph Hellwig         goto unlink_and_fail;
10586987307cSChristoph Hellwig     }
10596987307cSChristoph Hellwig 
1060f500a6d3SKevin Wolf     if (bs->file != file) {
1061f500a6d3SKevin Wolf         bdrv_delete(file);
1062f500a6d3SKevin Wolf         file = NULL;
1063f500a6d3SKevin Wolf     }
1064f500a6d3SKevin Wolf 
1065b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
10669156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
106731ca6d07SKevin Wolf         QDict *backing_options;
106831ca6d07SKevin Wolf 
106931ca6d07SKevin Wolf         extract_subqdict(options, &backing_options, "backing.");
107031ca6d07SKevin Wolf         ret = bdrv_open_backing_file(bs, backing_options);
1071b6ce07aaSKevin Wolf         if (ret < 0) {
1072b6ad491aSKevin Wolf             goto close_and_fail;
1073b6ce07aaSKevin Wolf         }
1074b6ce07aaSKevin Wolf     }
1075b6ce07aaSKevin Wolf 
1076b6ad491aSKevin Wolf     /* Check if any unknown options were used */
1077b6ad491aSKevin Wolf     if (qdict_size(options) != 0) {
1078b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
1079b6ad491aSKevin Wolf         qerror_report(ERROR_CLASS_GENERIC_ERROR, "Block format '%s' used by "
1080b6ad491aSKevin Wolf             "device '%s' doesn't support the option '%s'",
1081b6ad491aSKevin Wolf             drv->format_name, bs->device_name, entry->key);
1082b6ad491aSKevin Wolf 
1083b6ad491aSKevin Wolf         ret = -EINVAL;
1084b6ad491aSKevin Wolf         goto close_and_fail;
1085b6ad491aSKevin Wolf     }
1086b6ad491aSKevin Wolf     QDECREF(options);
1087b6ad491aSKevin Wolf 
1088b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
10897d4b4ba5SMarkus Armbruster         bdrv_dev_change_media_cb(bs, true);
1090b6ce07aaSKevin Wolf     }
1091b6ce07aaSKevin Wolf 
109298f90dbaSZhi Yong Wu     /* throttling disk I/O limits */
109398f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
109498f90dbaSZhi Yong Wu         bdrv_io_limits_enable(bs);
109598f90dbaSZhi Yong Wu     }
109698f90dbaSZhi Yong Wu 
1097b6ce07aaSKevin Wolf     return 0;
1098b6ce07aaSKevin Wolf 
1099b6ce07aaSKevin Wolf unlink_and_fail:
1100f500a6d3SKevin Wolf     if (file != NULL) {
1101f500a6d3SKevin Wolf         bdrv_delete(file);
1102f500a6d3SKevin Wolf     }
1103b6ce07aaSKevin Wolf     if (bs->is_temporary) {
1104b6ce07aaSKevin Wolf         unlink(filename);
1105b6ce07aaSKevin Wolf     }
1106de9c0cecSKevin Wolf fail:
1107de9c0cecSKevin Wolf     QDECREF(bs->options);
1108b6ad491aSKevin Wolf     QDECREF(options);
1109de9c0cecSKevin Wolf     bs->options = NULL;
1110b6ad491aSKevin Wolf     return ret;
1111de9c0cecSKevin Wolf 
1112b6ad491aSKevin Wolf close_and_fail:
1113b6ad491aSKevin Wolf     bdrv_close(bs);
1114b6ad491aSKevin Wolf     QDECREF(options);
1115b6ce07aaSKevin Wolf     return ret;
1116b6ce07aaSKevin Wolf }
1117b6ce07aaSKevin Wolf 
1118e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1119e971aa12SJeff Cody      bool prepared;
1120e971aa12SJeff Cody      BDRVReopenState state;
1121e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1122e971aa12SJeff Cody } BlockReopenQueueEntry;
1123e971aa12SJeff Cody 
1124e971aa12SJeff Cody /*
1125e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1126e971aa12SJeff Cody  * reopen of multiple devices.
1127e971aa12SJeff Cody  *
1128e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1129e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1130e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1131e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1132e971aa12SJeff Cody  * atomic 'set'.
1133e971aa12SJeff Cody  *
1134e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1135e971aa12SJeff Cody  *
1136e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1137e971aa12SJeff Cody  *
1138e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1139e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1140e971aa12SJeff Cody  *
1141e971aa12SJeff Cody  */
1142e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1143e971aa12SJeff Cody                                     BlockDriverState *bs, int flags)
1144e971aa12SJeff Cody {
1145e971aa12SJeff Cody     assert(bs != NULL);
1146e971aa12SJeff Cody 
1147e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
1148e971aa12SJeff Cody     if (bs_queue == NULL) {
1149e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1150e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1151e971aa12SJeff Cody     }
1152e971aa12SJeff Cody 
1153e971aa12SJeff Cody     if (bs->file) {
1154e971aa12SJeff Cody         bdrv_reopen_queue(bs_queue, bs->file, flags);
1155e971aa12SJeff Cody     }
1156e971aa12SJeff Cody 
1157e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1158e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1159e971aa12SJeff Cody 
1160e971aa12SJeff Cody     bs_entry->state.bs = bs;
1161e971aa12SJeff Cody     bs_entry->state.flags = flags;
1162e971aa12SJeff Cody 
1163e971aa12SJeff Cody     return bs_queue;
1164e971aa12SJeff Cody }
1165e971aa12SJeff Cody 
1166e971aa12SJeff Cody /*
1167e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1168e971aa12SJeff Cody  *
1169e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1170e971aa12SJeff Cody  * via bdrv_reopen_queue().
1171e971aa12SJeff Cody  *
1172e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1173e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1174e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1175e971aa12SJeff Cody  * data cleaned up.
1176e971aa12SJeff Cody  *
1177e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1178e971aa12SJeff Cody  * to all devices.
1179e971aa12SJeff Cody  *
1180e971aa12SJeff Cody  */
1181e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1182e971aa12SJeff Cody {
1183e971aa12SJeff Cody     int ret = -1;
1184e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1185e971aa12SJeff Cody     Error *local_err = NULL;
1186e971aa12SJeff Cody 
1187e971aa12SJeff Cody     assert(bs_queue != NULL);
1188e971aa12SJeff Cody 
1189e971aa12SJeff Cody     bdrv_drain_all();
1190e971aa12SJeff Cody 
1191e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1192e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1193e971aa12SJeff Cody             error_propagate(errp, local_err);
1194e971aa12SJeff Cody             goto cleanup;
1195e971aa12SJeff Cody         }
1196e971aa12SJeff Cody         bs_entry->prepared = true;
1197e971aa12SJeff Cody     }
1198e971aa12SJeff Cody 
1199e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1200e971aa12SJeff Cody      * changes
1201e971aa12SJeff Cody      */
1202e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1203e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1204e971aa12SJeff Cody     }
1205e971aa12SJeff Cody 
1206e971aa12SJeff Cody     ret = 0;
1207e971aa12SJeff Cody 
1208e971aa12SJeff Cody cleanup:
1209e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1210e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1211e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1212e971aa12SJeff Cody         }
1213e971aa12SJeff Cody         g_free(bs_entry);
1214e971aa12SJeff Cody     }
1215e971aa12SJeff Cody     g_free(bs_queue);
1216e971aa12SJeff Cody     return ret;
1217e971aa12SJeff Cody }
1218e971aa12SJeff Cody 
1219e971aa12SJeff Cody 
1220e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1221e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1222e971aa12SJeff Cody {
1223e971aa12SJeff Cody     int ret = -1;
1224e971aa12SJeff Cody     Error *local_err = NULL;
1225e971aa12SJeff Cody     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, bdrv_flags);
1226e971aa12SJeff Cody 
1227e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1228e971aa12SJeff Cody     if (local_err != NULL) {
1229e971aa12SJeff Cody         error_propagate(errp, local_err);
1230e971aa12SJeff Cody     }
1231e971aa12SJeff Cody     return ret;
1232e971aa12SJeff Cody }
1233e971aa12SJeff Cody 
1234e971aa12SJeff Cody 
1235e971aa12SJeff Cody /*
1236e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1237e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1238e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1239e971aa12SJeff Cody  *
1240e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1241e971aa12SJeff Cody  * flags are the new open flags
1242e971aa12SJeff Cody  * queue is the reopen queue
1243e971aa12SJeff Cody  *
1244e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1245e971aa12SJeff Cody  * as well.
1246e971aa12SJeff Cody  *
1247e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1248e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1249e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1250e971aa12SJeff Cody  *
1251e971aa12SJeff Cody  */
1252e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1253e971aa12SJeff Cody                         Error **errp)
1254e971aa12SJeff Cody {
1255e971aa12SJeff Cody     int ret = -1;
1256e971aa12SJeff Cody     Error *local_err = NULL;
1257e971aa12SJeff Cody     BlockDriver *drv;
1258e971aa12SJeff Cody 
1259e971aa12SJeff Cody     assert(reopen_state != NULL);
1260e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1261e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1262e971aa12SJeff Cody 
1263e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1264e971aa12SJeff Cody      * to r/w */
1265e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1266e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
1267e971aa12SJeff Cody         error_set(errp, QERR_DEVICE_IS_READ_ONLY,
1268e971aa12SJeff Cody                   reopen_state->bs->device_name);
1269e971aa12SJeff Cody         goto error;
1270e971aa12SJeff Cody     }
1271e971aa12SJeff Cody 
1272e971aa12SJeff Cody 
1273e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1274e971aa12SJeff Cody     if (ret) {
1275e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1276e971aa12SJeff Cody                   strerror(-ret));
1277e971aa12SJeff Cody         goto error;
1278e971aa12SJeff Cody     }
1279e971aa12SJeff Cody 
1280e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1281e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1282e971aa12SJeff Cody         if (ret) {
1283e971aa12SJeff Cody             if (local_err != NULL) {
1284e971aa12SJeff Cody                 error_propagate(errp, local_err);
1285e971aa12SJeff Cody             } else {
1286e971aa12SJeff Cody                 error_set(errp, QERR_OPEN_FILE_FAILED,
1287e971aa12SJeff Cody                           reopen_state->bs->filename);
1288e971aa12SJeff Cody             }
1289e971aa12SJeff Cody             goto error;
1290e971aa12SJeff Cody         }
1291e971aa12SJeff Cody     } else {
1292e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1293e971aa12SJeff Cody          * handler for each supported drv. */
1294e971aa12SJeff Cody         error_set(errp, QERR_BLOCK_FORMAT_FEATURE_NOT_SUPPORTED,
1295e971aa12SJeff Cody                   drv->format_name, reopen_state->bs->device_name,
1296e971aa12SJeff Cody                  "reopening of file");
1297e971aa12SJeff Cody         ret = -1;
1298e971aa12SJeff Cody         goto error;
1299e971aa12SJeff Cody     }
1300e971aa12SJeff Cody 
1301e971aa12SJeff Cody     ret = 0;
1302e971aa12SJeff Cody 
1303e971aa12SJeff Cody error:
1304e971aa12SJeff Cody     return ret;
1305e971aa12SJeff Cody }
1306e971aa12SJeff Cody 
1307e971aa12SJeff Cody /*
1308e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1309e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1310e971aa12SJeff Cody  * the active BlockDriverState contents.
1311e971aa12SJeff Cody  */
1312e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1313e971aa12SJeff Cody {
1314e971aa12SJeff Cody     BlockDriver *drv;
1315e971aa12SJeff Cody 
1316e971aa12SJeff Cody     assert(reopen_state != NULL);
1317e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1318e971aa12SJeff Cody     assert(drv != NULL);
1319e971aa12SJeff Cody 
1320e971aa12SJeff Cody     /* If there are any driver level actions to take */
1321e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1322e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1323e971aa12SJeff Cody     }
1324e971aa12SJeff Cody 
1325e971aa12SJeff Cody     /* set BDS specific flags now */
1326e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1327e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1328e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1329e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1330e971aa12SJeff Cody }
1331e971aa12SJeff Cody 
1332e971aa12SJeff Cody /*
1333e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1334e971aa12SJeff Cody  * reopen_state
1335e971aa12SJeff Cody  */
1336e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1337e971aa12SJeff Cody {
1338e971aa12SJeff Cody     BlockDriver *drv;
1339e971aa12SJeff Cody 
1340e971aa12SJeff Cody     assert(reopen_state != NULL);
1341e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1342e971aa12SJeff Cody     assert(drv != NULL);
1343e971aa12SJeff Cody 
1344e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1345e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1346e971aa12SJeff Cody     }
1347e971aa12SJeff Cody }
1348e971aa12SJeff Cody 
1349e971aa12SJeff Cody 
1350fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1351fc01f7e7Sbellard {
135280ccf93bSLiu Yuan     bdrv_flush(bs);
13533e914655SPaolo Bonzini     if (bs->job) {
13543e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
13553e914655SPaolo Bonzini     }
13567094f12fSKevin Wolf     bdrv_drain_all();
1357d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
13587094f12fSKevin Wolf 
13593cbc002cSPaolo Bonzini     if (bs->drv) {
1360f9092b10SMarkus Armbruster         if (bs == bs_snapshots) {
1361f9092b10SMarkus Armbruster             bs_snapshots = NULL;
1362f9092b10SMarkus Armbruster         }
1363557df6acSStefan Hajnoczi         if (bs->backing_hd) {
1364ea2384d3Sbellard             bdrv_delete(bs->backing_hd);
1365557df6acSStefan Hajnoczi             bs->backing_hd = NULL;
1366557df6acSStefan Hajnoczi         }
1367ea2384d3Sbellard         bs->drv->bdrv_close(bs);
13687267c094SAnthony Liguori         g_free(bs->opaque);
1369ea2384d3Sbellard #ifdef _WIN32
1370ea2384d3Sbellard         if (bs->is_temporary) {
1371ea2384d3Sbellard             unlink(bs->filename);
1372ea2384d3Sbellard         }
137367b915a5Sbellard #endif
1374ea2384d3Sbellard         bs->opaque = NULL;
1375ea2384d3Sbellard         bs->drv = NULL;
137653fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1377a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1378a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
13796405875cSPaolo Bonzini         bs->total_sectors = 0;
13806405875cSPaolo Bonzini         bs->encrypted = 0;
13816405875cSPaolo Bonzini         bs->valid_key = 0;
13826405875cSPaolo Bonzini         bs->sg = 0;
13836405875cSPaolo Bonzini         bs->growable = 0;
1384de9c0cecSKevin Wolf         QDECREF(bs->options);
1385de9c0cecSKevin Wolf         bs->options = NULL;
1386b338082bSbellard 
138766f82ceeSKevin Wolf         if (bs->file != NULL) {
13880ac9377dSPaolo Bonzini             bdrv_delete(bs->file);
13890ac9377dSPaolo Bonzini             bs->file = NULL;
139066f82ceeSKevin Wolf         }
13919ca11154SPavel Hrdina     }
139266f82ceeSKevin Wolf 
13937d4b4ba5SMarkus Armbruster     bdrv_dev_change_media_cb(bs, false);
139498f90dbaSZhi Yong Wu 
139598f90dbaSZhi Yong Wu     /*throttling disk I/O limits*/
139698f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
139798f90dbaSZhi Yong Wu         bdrv_io_limits_disable(bs);
139898f90dbaSZhi Yong Wu     }
1399b338082bSbellard }
1400b338082bSbellard 
14012bc93fedSMORITA Kazutaka void bdrv_close_all(void)
14022bc93fedSMORITA Kazutaka {
14032bc93fedSMORITA Kazutaka     BlockDriverState *bs;
14042bc93fedSMORITA Kazutaka 
14052bc93fedSMORITA Kazutaka     QTAILQ_FOREACH(bs, &bdrv_states, list) {
14062bc93fedSMORITA Kazutaka         bdrv_close(bs);
14072bc93fedSMORITA Kazutaka     }
14082bc93fedSMORITA Kazutaka }
14092bc93fedSMORITA Kazutaka 
1410922453bcSStefan Hajnoczi /*
1411922453bcSStefan Hajnoczi  * Wait for pending requests to complete across all BlockDriverStates
1412922453bcSStefan Hajnoczi  *
1413922453bcSStefan Hajnoczi  * This function does not flush data to disk, use bdrv_flush_all() for that
1414922453bcSStefan Hajnoczi  * after calling this function.
14154c355d53SZhi Yong Wu  *
14164c355d53SZhi Yong Wu  * Note that completion of an asynchronous I/O operation can trigger any
14174c355d53SZhi Yong Wu  * number of other I/O operations on other devices---for example a coroutine
14184c355d53SZhi Yong Wu  * can be arbitrarily complex and a constant flow of I/O can come until the
14194c355d53SZhi Yong Wu  * coroutine is complete.  Because of this, it is not possible to have a
14204c355d53SZhi Yong Wu  * function to drain a single device's I/O queue.
1421922453bcSStefan Hajnoczi  */
1422922453bcSStefan Hajnoczi void bdrv_drain_all(void)
1423922453bcSStefan Hajnoczi {
1424922453bcSStefan Hajnoczi     BlockDriverState *bs;
14254c355d53SZhi Yong Wu     bool busy;
1426922453bcSStefan Hajnoczi 
14274c355d53SZhi Yong Wu     do {
14284c355d53SZhi Yong Wu         busy = qemu_aio_wait();
14294c355d53SZhi Yong Wu 
14304c355d53SZhi Yong Wu         /* FIXME: We do not have timer support here, so this is effectively
14314c355d53SZhi Yong Wu          * a busy wait.
14324c355d53SZhi Yong Wu          */
14334c355d53SZhi Yong Wu         QTAILQ_FOREACH(bs, &bdrv_states, list) {
14344c355d53SZhi Yong Wu             if (!qemu_co_queue_empty(&bs->throttled_reqs)) {
14354c355d53SZhi Yong Wu                 qemu_co_queue_restart_all(&bs->throttled_reqs);
14364c355d53SZhi Yong Wu                 busy = true;
14374c355d53SZhi Yong Wu             }
14384c355d53SZhi Yong Wu         }
14394c355d53SZhi Yong Wu     } while (busy);
1440922453bcSStefan Hajnoczi 
1441922453bcSStefan Hajnoczi     /* If requests are still pending there is a bug somewhere */
1442922453bcSStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1443922453bcSStefan Hajnoczi         assert(QLIST_EMPTY(&bs->tracked_requests));
1444922453bcSStefan Hajnoczi         assert(qemu_co_queue_empty(&bs->throttled_reqs));
1445922453bcSStefan Hajnoczi     }
1446922453bcSStefan Hajnoczi }
1447922453bcSStefan Hajnoczi 
1448d22b2f41SRyan Harper /* make a BlockDriverState anonymous by removing from bdrv_state list.
1449d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1450d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1451d22b2f41SRyan Harper {
1452d22b2f41SRyan Harper     if (bs->device_name[0] != '\0') {
1453d22b2f41SRyan Harper         QTAILQ_REMOVE(&bdrv_states, bs, list);
1454d22b2f41SRyan Harper     }
1455d22b2f41SRyan Harper     bs->device_name[0] = '\0';
1456d22b2f41SRyan Harper }
1457d22b2f41SRyan Harper 
1458e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
1459e023b2e2SPaolo Bonzini {
1460e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
1461e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
1462e023b2e2SPaolo Bonzini     }
1463e023b2e2SPaolo Bonzini }
1464e023b2e2SPaolo Bonzini 
14654ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
14664ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
14674ddc07caSPaolo Bonzini {
14684ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
14694ddc07caSPaolo Bonzini     bs_dest->open_flags         = bs_src->open_flags;
14704ddc07caSPaolo Bonzini 
14714ddc07caSPaolo Bonzini     /* dev info */
14724ddc07caSPaolo Bonzini     bs_dest->dev_ops            = bs_src->dev_ops;
14734ddc07caSPaolo Bonzini     bs_dest->dev_opaque         = bs_src->dev_opaque;
14744ddc07caSPaolo Bonzini     bs_dest->dev                = bs_src->dev;
14754ddc07caSPaolo Bonzini     bs_dest->buffer_alignment   = bs_src->buffer_alignment;
14764ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
14774ddc07caSPaolo Bonzini 
14784ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
14794ddc07caSPaolo Bonzini 
14804ddc07caSPaolo Bonzini     /* i/o timing parameters */
14814ddc07caSPaolo Bonzini     bs_dest->slice_start        = bs_src->slice_start;
14824ddc07caSPaolo Bonzini     bs_dest->slice_end          = bs_src->slice_end;
14835905fbc9SStefan Hajnoczi     bs_dest->slice_submitted    = bs_src->slice_submitted;
14844ddc07caSPaolo Bonzini     bs_dest->io_limits          = bs_src->io_limits;
14854ddc07caSPaolo Bonzini     bs_dest->throttled_reqs     = bs_src->throttled_reqs;
14864ddc07caSPaolo Bonzini     bs_dest->block_timer        = bs_src->block_timer;
14874ddc07caSPaolo Bonzini     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
14884ddc07caSPaolo Bonzini 
14894ddc07caSPaolo Bonzini     /* r/w error */
14904ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
14914ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
14924ddc07caSPaolo Bonzini 
14934ddc07caSPaolo Bonzini     /* i/o status */
14944ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
14954ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
14964ddc07caSPaolo Bonzini 
14974ddc07caSPaolo Bonzini     /* dirty bitmap */
14984ddc07caSPaolo Bonzini     bs_dest->dirty_bitmap       = bs_src->dirty_bitmap;
14994ddc07caSPaolo Bonzini 
15004ddc07caSPaolo Bonzini     /* job */
15014ddc07caSPaolo Bonzini     bs_dest->in_use             = bs_src->in_use;
15024ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
15034ddc07caSPaolo Bonzini 
15044ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
15054ddc07caSPaolo Bonzini     pstrcpy(bs_dest->device_name, sizeof(bs_dest->device_name),
15064ddc07caSPaolo Bonzini             bs_src->device_name);
15074ddc07caSPaolo Bonzini     bs_dest->list = bs_src->list;
15084ddc07caSPaolo Bonzini }
15094ddc07caSPaolo Bonzini 
15104ddc07caSPaolo Bonzini /*
15114ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
15124ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
15134ddc07caSPaolo Bonzini  * actually attached to a device.
15144ddc07caSPaolo Bonzini  *
15154ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
15164ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
15174ddc07caSPaolo Bonzini  *
15184ddc07caSPaolo Bonzini  * bs_new is required to be anonymous.
15194ddc07caSPaolo Bonzini  *
15204ddc07caSPaolo Bonzini  * This function does not create any image files.
15214ddc07caSPaolo Bonzini  */
15224ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
15234ddc07caSPaolo Bonzini {
15244ddc07caSPaolo Bonzini     BlockDriverState tmp;
15254ddc07caSPaolo Bonzini 
15264ddc07caSPaolo Bonzini     /* bs_new must be anonymous and shouldn't have anything fancy enabled */
15274ddc07caSPaolo Bonzini     assert(bs_new->device_name[0] == '\0');
15284ddc07caSPaolo Bonzini     assert(bs_new->dirty_bitmap == NULL);
15294ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
15304ddc07caSPaolo Bonzini     assert(bs_new->dev == NULL);
15314ddc07caSPaolo Bonzini     assert(bs_new->in_use == 0);
15324ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
15334ddc07caSPaolo Bonzini     assert(bs_new->block_timer == NULL);
15344ddc07caSPaolo Bonzini 
15354ddc07caSPaolo Bonzini     tmp = *bs_new;
15364ddc07caSPaolo Bonzini     *bs_new = *bs_old;
15374ddc07caSPaolo Bonzini     *bs_old = tmp;
15384ddc07caSPaolo Bonzini 
15394ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
15404ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
15414ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
15424ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
15434ddc07caSPaolo Bonzini 
15444ddc07caSPaolo Bonzini     /* bs_new shouldn't be in bdrv_states even after the swap!  */
15454ddc07caSPaolo Bonzini     assert(bs_new->device_name[0] == '\0');
15464ddc07caSPaolo Bonzini 
15474ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
15484ddc07caSPaolo Bonzini     assert(bs_new->dev == NULL);
15494ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
15504ddc07caSPaolo Bonzini     assert(bs_new->in_use == 0);
15514ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
15524ddc07caSPaolo Bonzini     assert(bs_new->block_timer == NULL);
15534ddc07caSPaolo Bonzini 
15544ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
15554ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
15564ddc07caSPaolo Bonzini }
15574ddc07caSPaolo Bonzini 
15588802d1fdSJeff Cody /*
15598802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
15608802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
15618802d1fdSJeff Cody  *
15628802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
15638802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
15648802d1fdSJeff Cody  *
1565f6801b83SJeff Cody  * bs_new is required to be anonymous.
1566f6801b83SJeff Cody  *
15678802d1fdSJeff Cody  * This function does not create any image files.
15688802d1fdSJeff Cody  */
15698802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
15708802d1fdSJeff Cody {
15714ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
15728802d1fdSJeff Cody 
15738802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
15748802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
15754ddc07caSPaolo Bonzini     bs_top->backing_hd = bs_new;
15764ddc07caSPaolo Bonzini     bs_top->open_flags &= ~BDRV_O_NO_BACKING;
15774ddc07caSPaolo Bonzini     pstrcpy(bs_top->backing_file, sizeof(bs_top->backing_file),
15784ddc07caSPaolo Bonzini             bs_new->filename);
15794ddc07caSPaolo Bonzini     pstrcpy(bs_top->backing_format, sizeof(bs_top->backing_format),
15804ddc07caSPaolo Bonzini             bs_new->drv ? bs_new->drv->format_name : "");
15818802d1fdSJeff Cody }
15828802d1fdSJeff Cody 
1583b338082bSbellard void bdrv_delete(BlockDriverState *bs)
1584b338082bSbellard {
1585fa879d62SMarkus Armbruster     assert(!bs->dev);
15863e914655SPaolo Bonzini     assert(!bs->job);
15873e914655SPaolo Bonzini     assert(!bs->in_use);
158818846deeSMarkus Armbruster 
15891b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
1590d22b2f41SRyan Harper     bdrv_make_anon(bs);
159134c6f050Saurel32 
1592b338082bSbellard     bdrv_close(bs);
159366f82ceeSKevin Wolf 
1594f9092b10SMarkus Armbruster     assert(bs != bs_snapshots);
15957267c094SAnthony Liguori     g_free(bs);
1596fc01f7e7Sbellard }
1597fc01f7e7Sbellard 
1598fa879d62SMarkus Armbruster int bdrv_attach_dev(BlockDriverState *bs, void *dev)
1599fa879d62SMarkus Armbruster /* TODO change to DeviceState *dev when all users are qdevified */
160018846deeSMarkus Armbruster {
1601fa879d62SMarkus Armbruster     if (bs->dev) {
160218846deeSMarkus Armbruster         return -EBUSY;
160318846deeSMarkus Armbruster     }
1604fa879d62SMarkus Armbruster     bs->dev = dev;
160528a7282aSLuiz Capitulino     bdrv_iostatus_reset(bs);
160618846deeSMarkus Armbruster     return 0;
160718846deeSMarkus Armbruster }
160818846deeSMarkus Armbruster 
1609fa879d62SMarkus Armbruster /* TODO qdevified devices don't use this, remove when devices are qdevified */
1610fa879d62SMarkus Armbruster void bdrv_attach_dev_nofail(BlockDriverState *bs, void *dev)
161118846deeSMarkus Armbruster {
1612fa879d62SMarkus Armbruster     if (bdrv_attach_dev(bs, dev) < 0) {
1613fa879d62SMarkus Armbruster         abort();
1614fa879d62SMarkus Armbruster     }
1615fa879d62SMarkus Armbruster }
1616fa879d62SMarkus Armbruster 
1617fa879d62SMarkus Armbruster void bdrv_detach_dev(BlockDriverState *bs, void *dev)
1618fa879d62SMarkus Armbruster /* TODO change to DeviceState *dev when all users are qdevified */
1619fa879d62SMarkus Armbruster {
1620fa879d62SMarkus Armbruster     assert(bs->dev == dev);
1621fa879d62SMarkus Armbruster     bs->dev = NULL;
16220e49de52SMarkus Armbruster     bs->dev_ops = NULL;
16230e49de52SMarkus Armbruster     bs->dev_opaque = NULL;
162429e05f20SMarkus Armbruster     bs->buffer_alignment = 512;
162518846deeSMarkus Armbruster }
162618846deeSMarkus Armbruster 
1627fa879d62SMarkus Armbruster /* TODO change to return DeviceState * when all users are qdevified */
1628fa879d62SMarkus Armbruster void *bdrv_get_attached_dev(BlockDriverState *bs)
162918846deeSMarkus Armbruster {
1630fa879d62SMarkus Armbruster     return bs->dev;
163118846deeSMarkus Armbruster }
163218846deeSMarkus Armbruster 
16330e49de52SMarkus Armbruster void bdrv_set_dev_ops(BlockDriverState *bs, const BlockDevOps *ops,
16340e49de52SMarkus Armbruster                       void *opaque)
16350e49de52SMarkus Armbruster {
16360e49de52SMarkus Armbruster     bs->dev_ops = ops;
16370e49de52SMarkus Armbruster     bs->dev_opaque = opaque;
16382c6942faSMarkus Armbruster     if (bdrv_dev_has_removable_media(bs) && bs == bs_snapshots) {
16392c6942faSMarkus Armbruster         bs_snapshots = NULL;
16402c6942faSMarkus Armbruster     }
16410e49de52SMarkus Armbruster }
16420e49de52SMarkus Armbruster 
164332c81a4aSPaolo Bonzini void bdrv_emit_qmp_error_event(const BlockDriverState *bdrv,
164432c81a4aSPaolo Bonzini                                enum MonitorEvent ev,
16451ceee0d5SPaolo Bonzini                                BlockErrorAction action, bool is_read)
1646329c0a48SLuiz Capitulino {
1647329c0a48SLuiz Capitulino     QObject *data;
1648329c0a48SLuiz Capitulino     const char *action_str;
1649329c0a48SLuiz Capitulino 
1650329c0a48SLuiz Capitulino     switch (action) {
1651329c0a48SLuiz Capitulino     case BDRV_ACTION_REPORT:
1652329c0a48SLuiz Capitulino         action_str = "report";
1653329c0a48SLuiz Capitulino         break;
1654329c0a48SLuiz Capitulino     case BDRV_ACTION_IGNORE:
1655329c0a48SLuiz Capitulino         action_str = "ignore";
1656329c0a48SLuiz Capitulino         break;
1657329c0a48SLuiz Capitulino     case BDRV_ACTION_STOP:
1658329c0a48SLuiz Capitulino         action_str = "stop";
1659329c0a48SLuiz Capitulino         break;
1660329c0a48SLuiz Capitulino     default:
1661329c0a48SLuiz Capitulino         abort();
1662329c0a48SLuiz Capitulino     }
1663329c0a48SLuiz Capitulino 
1664329c0a48SLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
1665329c0a48SLuiz Capitulino                               bdrv->device_name,
1666329c0a48SLuiz Capitulino                               action_str,
1667329c0a48SLuiz Capitulino                               is_read ? "read" : "write");
166832c81a4aSPaolo Bonzini     monitor_protocol_event(ev, data);
1669329c0a48SLuiz Capitulino 
1670329c0a48SLuiz Capitulino     qobject_decref(data);
1671329c0a48SLuiz Capitulino }
1672329c0a48SLuiz Capitulino 
16736f382ed2SLuiz Capitulino static void bdrv_emit_qmp_eject_event(BlockDriverState *bs, bool ejected)
16746f382ed2SLuiz Capitulino {
16756f382ed2SLuiz Capitulino     QObject *data;
16766f382ed2SLuiz Capitulino 
16776f382ed2SLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'tray-open': %i }",
16786f382ed2SLuiz Capitulino                               bdrv_get_device_name(bs), ejected);
16796f382ed2SLuiz Capitulino     monitor_protocol_event(QEVENT_DEVICE_TRAY_MOVED, data);
16806f382ed2SLuiz Capitulino 
16816f382ed2SLuiz Capitulino     qobject_decref(data);
16826f382ed2SLuiz Capitulino }
16836f382ed2SLuiz Capitulino 
16847d4b4ba5SMarkus Armbruster static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load)
16850e49de52SMarkus Armbruster {
1686145feb17SMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->change_media_cb) {
16876f382ed2SLuiz Capitulino         bool tray_was_closed = !bdrv_dev_is_tray_open(bs);
16887d4b4ba5SMarkus Armbruster         bs->dev_ops->change_media_cb(bs->dev_opaque, load);
16896f382ed2SLuiz Capitulino         if (tray_was_closed) {
16906f382ed2SLuiz Capitulino             /* tray open */
16916f382ed2SLuiz Capitulino             bdrv_emit_qmp_eject_event(bs, true);
16926f382ed2SLuiz Capitulino         }
16936f382ed2SLuiz Capitulino         if (load) {
16946f382ed2SLuiz Capitulino             /* tray close */
16956f382ed2SLuiz Capitulino             bdrv_emit_qmp_eject_event(bs, false);
16966f382ed2SLuiz Capitulino         }
1697145feb17SMarkus Armbruster     }
1698145feb17SMarkus Armbruster }
1699145feb17SMarkus Armbruster 
17002c6942faSMarkus Armbruster bool bdrv_dev_has_removable_media(BlockDriverState *bs)
17012c6942faSMarkus Armbruster {
17022c6942faSMarkus Armbruster     return !bs->dev || (bs->dev_ops && bs->dev_ops->change_media_cb);
17032c6942faSMarkus Armbruster }
17042c6942faSMarkus Armbruster 
1705025ccaa7SPaolo Bonzini void bdrv_dev_eject_request(BlockDriverState *bs, bool force)
1706025ccaa7SPaolo Bonzini {
1707025ccaa7SPaolo Bonzini     if (bs->dev_ops && bs->dev_ops->eject_request_cb) {
1708025ccaa7SPaolo Bonzini         bs->dev_ops->eject_request_cb(bs->dev_opaque, force);
1709025ccaa7SPaolo Bonzini     }
1710025ccaa7SPaolo Bonzini }
1711025ccaa7SPaolo Bonzini 
1712e4def80bSMarkus Armbruster bool bdrv_dev_is_tray_open(BlockDriverState *bs)
1713e4def80bSMarkus Armbruster {
1714e4def80bSMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->is_tray_open) {
1715e4def80bSMarkus Armbruster         return bs->dev_ops->is_tray_open(bs->dev_opaque);
1716e4def80bSMarkus Armbruster     }
1717e4def80bSMarkus Armbruster     return false;
1718e4def80bSMarkus Armbruster }
1719e4def80bSMarkus Armbruster 
1720145feb17SMarkus Armbruster static void bdrv_dev_resize_cb(BlockDriverState *bs)
1721145feb17SMarkus Armbruster {
1722145feb17SMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->resize_cb) {
1723145feb17SMarkus Armbruster         bs->dev_ops->resize_cb(bs->dev_opaque);
17240e49de52SMarkus Armbruster     }
17250e49de52SMarkus Armbruster }
17260e49de52SMarkus Armbruster 
1727f107639aSMarkus Armbruster bool bdrv_dev_is_medium_locked(BlockDriverState *bs)
1728f107639aSMarkus Armbruster {
1729f107639aSMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->is_medium_locked) {
1730f107639aSMarkus Armbruster         return bs->dev_ops->is_medium_locked(bs->dev_opaque);
1731f107639aSMarkus Armbruster     }
1732f107639aSMarkus Armbruster     return false;
1733f107639aSMarkus Armbruster }
1734f107639aSMarkus Armbruster 
1735e97fc193Saliguori /*
1736e97fc193Saliguori  * Run consistency checks on an image
1737e97fc193Saliguori  *
1738e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
1739a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
1740e076f338SKevin Wolf  * check are stored in res.
1741e97fc193Saliguori  */
17424534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
1743e97fc193Saliguori {
1744e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
1745e97fc193Saliguori         return -ENOTSUP;
1746e97fc193Saliguori     }
1747e97fc193Saliguori 
1748e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
17494534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
1750e97fc193Saliguori }
1751e97fc193Saliguori 
17528a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
17538a426614SKevin Wolf 
175433e3963eSbellard /* commit COW file into the raw image */
175533e3963eSbellard int bdrv_commit(BlockDriverState *bs)
175633e3963eSbellard {
175719cb3738Sbellard     BlockDriver *drv = bs->drv;
17588a426614SKevin Wolf     int64_t sector, total_sectors;
17598a426614SKevin Wolf     int n, ro, open_flags;
17600bce597dSJeff Cody     int ret = 0;
17618a426614SKevin Wolf     uint8_t *buf;
1762c2cba3d9SJim Meyering     char filename[PATH_MAX];
176333e3963eSbellard 
176419cb3738Sbellard     if (!drv)
176519cb3738Sbellard         return -ENOMEDIUM;
176633e3963eSbellard 
17674dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
17684dca4b63SNaphtali Sprei         return -ENOTSUP;
17694dca4b63SNaphtali Sprei     }
17704dca4b63SNaphtali Sprei 
17712d3735d3SStefan Hajnoczi     if (bdrv_in_use(bs) || bdrv_in_use(bs->backing_hd)) {
17722d3735d3SStefan Hajnoczi         return -EBUSY;
17732d3735d3SStefan Hajnoczi     }
17742d3735d3SStefan Hajnoczi 
17754dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
1776c2cba3d9SJim Meyering     /* Use pstrcpy (not strncpy): filename must be NUL-terminated. */
1777c2cba3d9SJim Meyering     pstrcpy(filename, sizeof(filename), bs->backing_hd->filename);
17784dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
17794dca4b63SNaphtali Sprei 
17804dca4b63SNaphtali Sprei     if (ro) {
17810bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
17820bce597dSJeff Cody             return -EACCES;
17834dca4b63SNaphtali Sprei         }
1784ea2384d3Sbellard     }
1785ea2384d3Sbellard 
17866ea44308SJan Kiszka     total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
17877267c094SAnthony Liguori     buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
17888a426614SKevin Wolf 
17898a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
179005c4af54SStefan Hajnoczi         if (bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
17918a426614SKevin Wolf 
17928a426614SKevin Wolf             if (bdrv_read(bs, sector, buf, n) != 0) {
17934dca4b63SNaphtali Sprei                 ret = -EIO;
17944dca4b63SNaphtali Sprei                 goto ro_cleanup;
179533e3963eSbellard             }
179633e3963eSbellard 
17978a426614SKevin Wolf             if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
17984dca4b63SNaphtali Sprei                 ret = -EIO;
17994dca4b63SNaphtali Sprei                 goto ro_cleanup;
180033e3963eSbellard             }
180133e3963eSbellard         }
180233e3963eSbellard     }
180395389c86Sbellard 
18041d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
18051d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
18061d44952fSChristoph Hellwig         bdrv_flush(bs);
18071d44952fSChristoph Hellwig     }
180895389c86Sbellard 
18093f5075aeSChristoph Hellwig     /*
18103f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
18113f5075aeSChristoph Hellwig      * stable on disk.
18123f5075aeSChristoph Hellwig      */
18133f5075aeSChristoph Hellwig     if (bs->backing_hd)
18143f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
18154dca4b63SNaphtali Sprei 
18164dca4b63SNaphtali Sprei ro_cleanup:
18177267c094SAnthony Liguori     g_free(buf);
18184dca4b63SNaphtali Sprei 
18194dca4b63SNaphtali Sprei     if (ro) {
18200bce597dSJeff Cody         /* ignoring error return here */
18210bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
18224dca4b63SNaphtali Sprei     }
18234dca4b63SNaphtali Sprei 
18241d44952fSChristoph Hellwig     return ret;
182533e3963eSbellard }
182633e3963eSbellard 
1827e8877497SStefan Hajnoczi int bdrv_commit_all(void)
18286ab4b5abSMarkus Armbruster {
18296ab4b5abSMarkus Armbruster     BlockDriverState *bs;
18306ab4b5abSMarkus Armbruster 
18316ab4b5abSMarkus Armbruster     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1832272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
1833e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
1834e8877497SStefan Hajnoczi             if (ret < 0) {
1835e8877497SStefan Hajnoczi                 return ret;
18366ab4b5abSMarkus Armbruster             }
18376ab4b5abSMarkus Armbruster         }
1838272d2d8eSJeff Cody     }
1839e8877497SStefan Hajnoczi     return 0;
1840e8877497SStefan Hajnoczi }
18416ab4b5abSMarkus Armbruster 
1842dbffbdcfSStefan Hajnoczi struct BdrvTrackedRequest {
1843dbffbdcfSStefan Hajnoczi     BlockDriverState *bs;
1844dbffbdcfSStefan Hajnoczi     int64_t sector_num;
1845dbffbdcfSStefan Hajnoczi     int nb_sectors;
1846dbffbdcfSStefan Hajnoczi     bool is_write;
1847dbffbdcfSStefan Hajnoczi     QLIST_ENTRY(BdrvTrackedRequest) list;
18485f8b6491SStefan Hajnoczi     Coroutine *co; /* owner, used for deadlock detection */
1849f4658285SStefan Hajnoczi     CoQueue wait_queue; /* coroutines blocked on this request */
1850dbffbdcfSStefan Hajnoczi };
1851dbffbdcfSStefan Hajnoczi 
1852dbffbdcfSStefan Hajnoczi /**
1853dbffbdcfSStefan Hajnoczi  * Remove an active request from the tracked requests list
1854dbffbdcfSStefan Hajnoczi  *
1855dbffbdcfSStefan Hajnoczi  * This function should be called when a tracked request is completing.
1856dbffbdcfSStefan Hajnoczi  */
1857dbffbdcfSStefan Hajnoczi static void tracked_request_end(BdrvTrackedRequest *req)
1858dbffbdcfSStefan Hajnoczi {
1859dbffbdcfSStefan Hajnoczi     QLIST_REMOVE(req, list);
1860f4658285SStefan Hajnoczi     qemu_co_queue_restart_all(&req->wait_queue);
1861dbffbdcfSStefan Hajnoczi }
1862dbffbdcfSStefan Hajnoczi 
1863dbffbdcfSStefan Hajnoczi /**
1864dbffbdcfSStefan Hajnoczi  * Add an active request to the tracked requests list
1865dbffbdcfSStefan Hajnoczi  */
1866dbffbdcfSStefan Hajnoczi static void tracked_request_begin(BdrvTrackedRequest *req,
1867dbffbdcfSStefan Hajnoczi                                   BlockDriverState *bs,
1868dbffbdcfSStefan Hajnoczi                                   int64_t sector_num,
1869dbffbdcfSStefan Hajnoczi                                   int nb_sectors, bool is_write)
1870dbffbdcfSStefan Hajnoczi {
1871dbffbdcfSStefan Hajnoczi     *req = (BdrvTrackedRequest){
1872dbffbdcfSStefan Hajnoczi         .bs = bs,
1873dbffbdcfSStefan Hajnoczi         .sector_num = sector_num,
1874dbffbdcfSStefan Hajnoczi         .nb_sectors = nb_sectors,
1875dbffbdcfSStefan Hajnoczi         .is_write = is_write,
18765f8b6491SStefan Hajnoczi         .co = qemu_coroutine_self(),
1877dbffbdcfSStefan Hajnoczi     };
1878dbffbdcfSStefan Hajnoczi 
1879f4658285SStefan Hajnoczi     qemu_co_queue_init(&req->wait_queue);
1880f4658285SStefan Hajnoczi 
1881dbffbdcfSStefan Hajnoczi     QLIST_INSERT_HEAD(&bs->tracked_requests, req, list);
1882dbffbdcfSStefan Hajnoczi }
1883dbffbdcfSStefan Hajnoczi 
1884d83947acSStefan Hajnoczi /**
1885d83947acSStefan Hajnoczi  * Round a region to cluster boundaries
1886d83947acSStefan Hajnoczi  */
1887343bded4SPaolo Bonzini void bdrv_round_to_clusters(BlockDriverState *bs,
1888d83947acSStefan Hajnoczi                             int64_t sector_num, int nb_sectors,
1889d83947acSStefan Hajnoczi                             int64_t *cluster_sector_num,
1890d83947acSStefan Hajnoczi                             int *cluster_nb_sectors)
1891d83947acSStefan Hajnoczi {
1892d83947acSStefan Hajnoczi     BlockDriverInfo bdi;
1893d83947acSStefan Hajnoczi 
1894d83947acSStefan Hajnoczi     if (bdrv_get_info(bs, &bdi) < 0 || bdi.cluster_size == 0) {
1895d83947acSStefan Hajnoczi         *cluster_sector_num = sector_num;
1896d83947acSStefan Hajnoczi         *cluster_nb_sectors = nb_sectors;
1897d83947acSStefan Hajnoczi     } else {
1898d83947acSStefan Hajnoczi         int64_t c = bdi.cluster_size / BDRV_SECTOR_SIZE;
1899d83947acSStefan Hajnoczi         *cluster_sector_num = QEMU_ALIGN_DOWN(sector_num, c);
1900d83947acSStefan Hajnoczi         *cluster_nb_sectors = QEMU_ALIGN_UP(sector_num - *cluster_sector_num +
1901d83947acSStefan Hajnoczi                                             nb_sectors, c);
1902d83947acSStefan Hajnoczi     }
1903d83947acSStefan Hajnoczi }
1904d83947acSStefan Hajnoczi 
1905f4658285SStefan Hajnoczi static bool tracked_request_overlaps(BdrvTrackedRequest *req,
1906f4658285SStefan Hajnoczi                                      int64_t sector_num, int nb_sectors) {
1907d83947acSStefan Hajnoczi     /*        aaaa   bbbb */
1908d83947acSStefan Hajnoczi     if (sector_num >= req->sector_num + req->nb_sectors) {
1909d83947acSStefan Hajnoczi         return false;
1910d83947acSStefan Hajnoczi     }
1911d83947acSStefan Hajnoczi     /* bbbb   aaaa        */
1912d83947acSStefan Hajnoczi     if (req->sector_num >= sector_num + nb_sectors) {
1913d83947acSStefan Hajnoczi         return false;
1914d83947acSStefan Hajnoczi     }
1915d83947acSStefan Hajnoczi     return true;
1916f4658285SStefan Hajnoczi }
1917f4658285SStefan Hajnoczi 
1918f4658285SStefan Hajnoczi static void coroutine_fn wait_for_overlapping_requests(BlockDriverState *bs,
1919f4658285SStefan Hajnoczi         int64_t sector_num, int nb_sectors)
1920f4658285SStefan Hajnoczi {
1921f4658285SStefan Hajnoczi     BdrvTrackedRequest *req;
1922d83947acSStefan Hajnoczi     int64_t cluster_sector_num;
1923d83947acSStefan Hajnoczi     int cluster_nb_sectors;
1924f4658285SStefan Hajnoczi     bool retry;
1925f4658285SStefan Hajnoczi 
1926d83947acSStefan Hajnoczi     /* If we touch the same cluster it counts as an overlap.  This guarantees
1927d83947acSStefan Hajnoczi      * that allocating writes will be serialized and not race with each other
1928d83947acSStefan Hajnoczi      * for the same cluster.  For example, in copy-on-read it ensures that the
1929d83947acSStefan Hajnoczi      * CoR read and write operations are atomic and guest writes cannot
1930d83947acSStefan Hajnoczi      * interleave between them.
1931d83947acSStefan Hajnoczi      */
1932343bded4SPaolo Bonzini     bdrv_round_to_clusters(bs, sector_num, nb_sectors,
1933d83947acSStefan Hajnoczi                            &cluster_sector_num, &cluster_nb_sectors);
1934d83947acSStefan Hajnoczi 
1935f4658285SStefan Hajnoczi     do {
1936f4658285SStefan Hajnoczi         retry = false;
1937f4658285SStefan Hajnoczi         QLIST_FOREACH(req, &bs->tracked_requests, list) {
1938d83947acSStefan Hajnoczi             if (tracked_request_overlaps(req, cluster_sector_num,
1939d83947acSStefan Hajnoczi                                          cluster_nb_sectors)) {
19405f8b6491SStefan Hajnoczi                 /* Hitting this means there was a reentrant request, for
19415f8b6491SStefan Hajnoczi                  * example, a block driver issuing nested requests.  This must
19425f8b6491SStefan Hajnoczi                  * never happen since it means deadlock.
19435f8b6491SStefan Hajnoczi                  */
19445f8b6491SStefan Hajnoczi                 assert(qemu_coroutine_self() != req->co);
19455f8b6491SStefan Hajnoczi 
1946f4658285SStefan Hajnoczi                 qemu_co_queue_wait(&req->wait_queue);
1947f4658285SStefan Hajnoczi                 retry = true;
1948f4658285SStefan Hajnoczi                 break;
1949f4658285SStefan Hajnoczi             }
1950f4658285SStefan Hajnoczi         }
1951f4658285SStefan Hajnoczi     } while (retry);
1952f4658285SStefan Hajnoczi }
1953f4658285SStefan Hajnoczi 
1954756e6736SKevin Wolf /*
1955756e6736SKevin Wolf  * Return values:
1956756e6736SKevin Wolf  * 0        - success
1957756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
1958756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
1959756e6736SKevin Wolf  *            image file header
1960756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
1961756e6736SKevin Wolf  */
1962756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
1963756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
1964756e6736SKevin Wolf {
1965756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
1966469ef350SPaolo Bonzini     int ret;
1967756e6736SKevin Wolf 
19685f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
19695f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
19705f377794SPaolo Bonzini         return -EINVAL;
19715f377794SPaolo Bonzini     }
19725f377794SPaolo Bonzini 
1973756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
1974469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
1975756e6736SKevin Wolf     } else {
1976469ef350SPaolo Bonzini         ret = -ENOTSUP;
1977756e6736SKevin Wolf     }
1978469ef350SPaolo Bonzini 
1979469ef350SPaolo Bonzini     if (ret == 0) {
1980469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
1981469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
1982469ef350SPaolo Bonzini     }
1983469ef350SPaolo Bonzini     return ret;
1984756e6736SKevin Wolf }
1985756e6736SKevin Wolf 
19866ebdcee2SJeff Cody /*
19876ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
19886ebdcee2SJeff Cody  *
19896ebdcee2SJeff Cody  * active is the current topmost image.
19906ebdcee2SJeff Cody  *
19916ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
19926ebdcee2SJeff Cody  * or if active == bs.
19936ebdcee2SJeff Cody  */
19946ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
19956ebdcee2SJeff Cody                                     BlockDriverState *bs)
19966ebdcee2SJeff Cody {
19976ebdcee2SJeff Cody     BlockDriverState *overlay = NULL;
19986ebdcee2SJeff Cody     BlockDriverState *intermediate;
19996ebdcee2SJeff Cody 
20006ebdcee2SJeff Cody     assert(active != NULL);
20016ebdcee2SJeff Cody     assert(bs != NULL);
20026ebdcee2SJeff Cody 
20036ebdcee2SJeff Cody     /* if bs is the same as active, then by definition it has no overlay
20046ebdcee2SJeff Cody      */
20056ebdcee2SJeff Cody     if (active == bs) {
20066ebdcee2SJeff Cody         return NULL;
20076ebdcee2SJeff Cody     }
20086ebdcee2SJeff Cody 
20096ebdcee2SJeff Cody     intermediate = active;
20106ebdcee2SJeff Cody     while (intermediate->backing_hd) {
20116ebdcee2SJeff Cody         if (intermediate->backing_hd == bs) {
20126ebdcee2SJeff Cody             overlay = intermediate;
20136ebdcee2SJeff Cody             break;
20146ebdcee2SJeff Cody         }
20156ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
20166ebdcee2SJeff Cody     }
20176ebdcee2SJeff Cody 
20186ebdcee2SJeff Cody     return overlay;
20196ebdcee2SJeff Cody }
20206ebdcee2SJeff Cody 
20216ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
20226ebdcee2SJeff Cody     BlockDriverState *bs;
20236ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
20246ebdcee2SJeff Cody } BlkIntermediateStates;
20256ebdcee2SJeff Cody 
20266ebdcee2SJeff Cody 
20276ebdcee2SJeff Cody /*
20286ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
20296ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
20306ebdcee2SJeff Cody  *
20316ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
20326ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
20336ebdcee2SJeff Cody  *
20346ebdcee2SJeff Cody  * E.g., this will convert the following chain:
20356ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
20366ebdcee2SJeff Cody  *
20376ebdcee2SJeff Cody  * to
20386ebdcee2SJeff Cody  *
20396ebdcee2SJeff Cody  * bottom <- base <- active
20406ebdcee2SJeff Cody  *
20416ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
20426ebdcee2SJeff Cody  *
20436ebdcee2SJeff Cody  * base <- intermediate <- top <- active
20446ebdcee2SJeff Cody  *
20456ebdcee2SJeff Cody  * to
20466ebdcee2SJeff Cody  *
20476ebdcee2SJeff Cody  * base <- active
20486ebdcee2SJeff Cody  *
20496ebdcee2SJeff Cody  * Error conditions:
20506ebdcee2SJeff Cody  *  if active == top, that is considered an error
20516ebdcee2SJeff Cody  *
20526ebdcee2SJeff Cody  */
20536ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
20546ebdcee2SJeff Cody                            BlockDriverState *base)
20556ebdcee2SJeff Cody {
20566ebdcee2SJeff Cody     BlockDriverState *intermediate;
20576ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
20586ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
20596ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
20606ebdcee2SJeff Cody     int ret = -EIO;
20616ebdcee2SJeff Cody 
20626ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
20636ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
20646ebdcee2SJeff Cody 
20656ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
20666ebdcee2SJeff Cody         goto exit;
20676ebdcee2SJeff Cody     }
20686ebdcee2SJeff Cody 
20696ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
20706ebdcee2SJeff Cody 
20716ebdcee2SJeff Cody     if (new_top_bs == NULL) {
20726ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
20736ebdcee2SJeff Cody         goto exit;
20746ebdcee2SJeff Cody     }
20756ebdcee2SJeff Cody 
20766ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
20776ebdcee2SJeff Cody      * to do, no intermediate images */
20786ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
20796ebdcee2SJeff Cody         ret = 0;
20806ebdcee2SJeff Cody         goto exit;
20816ebdcee2SJeff Cody     }
20826ebdcee2SJeff Cody 
20836ebdcee2SJeff Cody     intermediate = top;
20846ebdcee2SJeff Cody 
20856ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
20866ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
20876ebdcee2SJeff Cody      */
20886ebdcee2SJeff Cody     while (intermediate) {
20896ebdcee2SJeff Cody         intermediate_state = g_malloc0(sizeof(BlkIntermediateStates));
20906ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
20916ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
20926ebdcee2SJeff Cody 
20936ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
20946ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
20956ebdcee2SJeff Cody             break;
20966ebdcee2SJeff Cody         }
20976ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
20986ebdcee2SJeff Cody     }
20996ebdcee2SJeff Cody     if (base_bs == NULL) {
21006ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
21016ebdcee2SJeff Cody          * unravel everything, and exit with error */
21026ebdcee2SJeff Cody         goto exit;
21036ebdcee2SJeff Cody     }
21046ebdcee2SJeff Cody 
21056ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
21066ebdcee2SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, base_bs->filename,
21076ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
21086ebdcee2SJeff Cody     if (ret) {
21096ebdcee2SJeff Cody         goto exit;
21106ebdcee2SJeff Cody     }
21116ebdcee2SJeff Cody     new_top_bs->backing_hd = base_bs;
21126ebdcee2SJeff Cody 
21136ebdcee2SJeff Cody 
21146ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
21156ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
21166ebdcee2SJeff Cody         intermediate_state->bs->backing_hd = NULL;
21176ebdcee2SJeff Cody         bdrv_delete(intermediate_state->bs);
21186ebdcee2SJeff Cody     }
21196ebdcee2SJeff Cody     ret = 0;
21206ebdcee2SJeff Cody 
21216ebdcee2SJeff Cody exit:
21226ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
21236ebdcee2SJeff Cody         g_free(intermediate_state);
21246ebdcee2SJeff Cody     }
21256ebdcee2SJeff Cody     return ret;
21266ebdcee2SJeff Cody }
21276ebdcee2SJeff Cody 
21286ebdcee2SJeff Cody 
212971d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
213071d0770cSaliguori                                    size_t size)
213171d0770cSaliguori {
213271d0770cSaliguori     int64_t len;
213371d0770cSaliguori 
213471d0770cSaliguori     if (!bdrv_is_inserted(bs))
213571d0770cSaliguori         return -ENOMEDIUM;
213671d0770cSaliguori 
213771d0770cSaliguori     if (bs->growable)
213871d0770cSaliguori         return 0;
213971d0770cSaliguori 
214071d0770cSaliguori     len = bdrv_getlength(bs);
214171d0770cSaliguori 
2142fbb7b4e0SKevin Wolf     if (offset < 0)
2143fbb7b4e0SKevin Wolf         return -EIO;
2144fbb7b4e0SKevin Wolf 
2145fbb7b4e0SKevin Wolf     if ((offset > len) || (len - offset < size))
214671d0770cSaliguori         return -EIO;
214771d0770cSaliguori 
214871d0770cSaliguori     return 0;
214971d0770cSaliguori }
215071d0770cSaliguori 
215171d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
215271d0770cSaliguori                               int nb_sectors)
215371d0770cSaliguori {
2154eb5a3165SJes Sorensen     return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
2155eb5a3165SJes Sorensen                                    nb_sectors * BDRV_SECTOR_SIZE);
215671d0770cSaliguori }
215771d0770cSaliguori 
21581c9805a3SStefan Hajnoczi typedef struct RwCo {
21591c9805a3SStefan Hajnoczi     BlockDriverState *bs;
21601c9805a3SStefan Hajnoczi     int64_t sector_num;
21611c9805a3SStefan Hajnoczi     int nb_sectors;
21621c9805a3SStefan Hajnoczi     QEMUIOVector *qiov;
21631c9805a3SStefan Hajnoczi     bool is_write;
21641c9805a3SStefan Hajnoczi     int ret;
21651c9805a3SStefan Hajnoczi } RwCo;
21661c9805a3SStefan Hajnoczi 
21671c9805a3SStefan Hajnoczi static void coroutine_fn bdrv_rw_co_entry(void *opaque)
2168fc01f7e7Sbellard {
21691c9805a3SStefan Hajnoczi     RwCo *rwco = opaque;
2170fc01f7e7Sbellard 
21711c9805a3SStefan Hajnoczi     if (!rwco->is_write) {
21721c9805a3SStefan Hajnoczi         rwco->ret = bdrv_co_do_readv(rwco->bs, rwco->sector_num,
2173470c0504SStefan Hajnoczi                                      rwco->nb_sectors, rwco->qiov, 0);
21741c9805a3SStefan Hajnoczi     } else {
21751c9805a3SStefan Hajnoczi         rwco->ret = bdrv_co_do_writev(rwco->bs, rwco->sector_num,
2176f08f2ddaSStefan Hajnoczi                                       rwco->nb_sectors, rwco->qiov, 0);
21771c9805a3SStefan Hajnoczi     }
21781c9805a3SStefan Hajnoczi }
2179e7a8a783SKevin Wolf 
21801c9805a3SStefan Hajnoczi /*
21818d3b1a2dSKevin Wolf  * Process a vectored synchronous request using coroutines
21821c9805a3SStefan Hajnoczi  */
21838d3b1a2dSKevin Wolf static int bdrv_rwv_co(BlockDriverState *bs, int64_t sector_num,
21848d3b1a2dSKevin Wolf                        QEMUIOVector *qiov, bool is_write)
21851c9805a3SStefan Hajnoczi {
21861c9805a3SStefan Hajnoczi     Coroutine *co;
21871c9805a3SStefan Hajnoczi     RwCo rwco = {
21881c9805a3SStefan Hajnoczi         .bs = bs,
21891c9805a3SStefan Hajnoczi         .sector_num = sector_num,
21908d3b1a2dSKevin Wolf         .nb_sectors = qiov->size >> BDRV_SECTOR_BITS,
21918d3b1a2dSKevin Wolf         .qiov = qiov,
21921c9805a3SStefan Hajnoczi         .is_write = is_write,
21931c9805a3SStefan Hajnoczi         .ret = NOT_DONE,
21941c9805a3SStefan Hajnoczi     };
21958d3b1a2dSKevin Wolf     assert((qiov->size & (BDRV_SECTOR_SIZE - 1)) == 0);
21961c9805a3SStefan Hajnoczi 
2197498e386cSZhi Yong Wu     /**
2198498e386cSZhi Yong Wu      * In sync call context, when the vcpu is blocked, this throttling timer
2199498e386cSZhi Yong Wu      * will not fire; so the I/O throttling function has to be disabled here
2200498e386cSZhi Yong Wu      * if it has been enabled.
2201498e386cSZhi Yong Wu      */
2202498e386cSZhi Yong Wu     if (bs->io_limits_enabled) {
2203498e386cSZhi Yong Wu         fprintf(stderr, "Disabling I/O throttling on '%s' due "
2204498e386cSZhi Yong Wu                         "to synchronous I/O.\n", bdrv_get_device_name(bs));
2205498e386cSZhi Yong Wu         bdrv_io_limits_disable(bs);
2206498e386cSZhi Yong Wu     }
2207498e386cSZhi Yong Wu 
22081c9805a3SStefan Hajnoczi     if (qemu_in_coroutine()) {
22091c9805a3SStefan Hajnoczi         /* Fast-path if already in coroutine context */
22101c9805a3SStefan Hajnoczi         bdrv_rw_co_entry(&rwco);
22111c9805a3SStefan Hajnoczi     } else {
22121c9805a3SStefan Hajnoczi         co = qemu_coroutine_create(bdrv_rw_co_entry);
22131c9805a3SStefan Hajnoczi         qemu_coroutine_enter(co, &rwco);
22141c9805a3SStefan Hajnoczi         while (rwco.ret == NOT_DONE) {
22151c9805a3SStefan Hajnoczi             qemu_aio_wait();
22161c9805a3SStefan Hajnoczi         }
22171c9805a3SStefan Hajnoczi     }
22181c9805a3SStefan Hajnoczi     return rwco.ret;
2219e7a8a783SKevin Wolf }
2220e7a8a783SKevin Wolf 
22218d3b1a2dSKevin Wolf /*
22228d3b1a2dSKevin Wolf  * Process a synchronous request using coroutines
22238d3b1a2dSKevin Wolf  */
22248d3b1a2dSKevin Wolf static int bdrv_rw_co(BlockDriverState *bs, int64_t sector_num, uint8_t *buf,
22258d3b1a2dSKevin Wolf                       int nb_sectors, bool is_write)
22268d3b1a2dSKevin Wolf {
22278d3b1a2dSKevin Wolf     QEMUIOVector qiov;
22288d3b1a2dSKevin Wolf     struct iovec iov = {
22298d3b1a2dSKevin Wolf         .iov_base = (void *)buf,
22308d3b1a2dSKevin Wolf         .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
22318d3b1a2dSKevin Wolf     };
22328d3b1a2dSKevin Wolf 
22338d3b1a2dSKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
22348d3b1a2dSKevin Wolf     return bdrv_rwv_co(bs, sector_num, &qiov, is_write);
22358d3b1a2dSKevin Wolf }
22368d3b1a2dSKevin Wolf 
22371c9805a3SStefan Hajnoczi /* return < 0 if error. See bdrv_write() for the return codes */
22381c9805a3SStefan Hajnoczi int bdrv_read(BlockDriverState *bs, int64_t sector_num,
22391c9805a3SStefan Hajnoczi               uint8_t *buf, int nb_sectors)
22401c9805a3SStefan Hajnoczi {
22411c9805a3SStefan Hajnoczi     return bdrv_rw_co(bs, sector_num, buf, nb_sectors, false);
224283f64091Sbellard }
2243fc01f7e7Sbellard 
224407d27a44SMarkus Armbruster /* Just like bdrv_read(), but with I/O throttling temporarily disabled */
224507d27a44SMarkus Armbruster int bdrv_read_unthrottled(BlockDriverState *bs, int64_t sector_num,
224607d27a44SMarkus Armbruster                           uint8_t *buf, int nb_sectors)
224707d27a44SMarkus Armbruster {
224807d27a44SMarkus Armbruster     bool enabled;
224907d27a44SMarkus Armbruster     int ret;
225007d27a44SMarkus Armbruster 
225107d27a44SMarkus Armbruster     enabled = bs->io_limits_enabled;
225207d27a44SMarkus Armbruster     bs->io_limits_enabled = false;
225307d27a44SMarkus Armbruster     ret = bdrv_read(bs, 0, buf, 1);
225407d27a44SMarkus Armbruster     bs->io_limits_enabled = enabled;
225507d27a44SMarkus Armbruster     return ret;
225607d27a44SMarkus Armbruster }
225707d27a44SMarkus Armbruster 
225819cb3738Sbellard /* Return < 0 if error. Important errors are:
225919cb3738Sbellard   -EIO         generic I/O error (may happen for all errors)
226019cb3738Sbellard   -ENOMEDIUM   No media inserted.
226119cb3738Sbellard   -EINVAL      Invalid sector number or nb_sectors
226219cb3738Sbellard   -EACCES      Trying to write a read-only device
226319cb3738Sbellard */
2264fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
2265fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
2266fc01f7e7Sbellard {
22671c9805a3SStefan Hajnoczi     return bdrv_rw_co(bs, sector_num, (uint8_t *)buf, nb_sectors, true);
226883f64091Sbellard }
226983f64091Sbellard 
22708d3b1a2dSKevin Wolf int bdrv_writev(BlockDriverState *bs, int64_t sector_num, QEMUIOVector *qiov)
22718d3b1a2dSKevin Wolf {
22728d3b1a2dSKevin Wolf     return bdrv_rwv_co(bs, sector_num, qiov, true);
22738d3b1a2dSKevin Wolf }
22748d3b1a2dSKevin Wolf 
2275eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset,
2276eda578e5Saliguori                void *buf, int count1)
227783f64091Sbellard {
22786ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
227983f64091Sbellard     int len, nb_sectors, count;
228083f64091Sbellard     int64_t sector_num;
22819a8c4cceSKevin Wolf     int ret;
228283f64091Sbellard 
228383f64091Sbellard     count = count1;
228483f64091Sbellard     /* first read to align to sector start */
22856ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
228683f64091Sbellard     if (len > count)
228783f64091Sbellard         len = count;
22886ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
228983f64091Sbellard     if (len > 0) {
22909a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
22919a8c4cceSKevin Wolf             return ret;
22926ea44308SJan Kiszka         memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
229383f64091Sbellard         count -= len;
229483f64091Sbellard         if (count == 0)
229583f64091Sbellard             return count1;
229683f64091Sbellard         sector_num++;
229783f64091Sbellard         buf += len;
229883f64091Sbellard     }
229983f64091Sbellard 
230083f64091Sbellard     /* read the sectors "in place" */
23016ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
230283f64091Sbellard     if (nb_sectors > 0) {
23039a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
23049a8c4cceSKevin Wolf             return ret;
230583f64091Sbellard         sector_num += nb_sectors;
23066ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
230783f64091Sbellard         buf += len;
230883f64091Sbellard         count -= len;
230983f64091Sbellard     }
231083f64091Sbellard 
231183f64091Sbellard     /* add data from the last sector */
231283f64091Sbellard     if (count > 0) {
23139a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23149a8c4cceSKevin Wolf             return ret;
231583f64091Sbellard         memcpy(buf, tmp_buf, count);
231683f64091Sbellard     }
231783f64091Sbellard     return count1;
231883f64091Sbellard }
231983f64091Sbellard 
23208d3b1a2dSKevin Wolf int bdrv_pwritev(BlockDriverState *bs, int64_t offset, QEMUIOVector *qiov)
232183f64091Sbellard {
23226ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
232383f64091Sbellard     int len, nb_sectors, count;
232483f64091Sbellard     int64_t sector_num;
23259a8c4cceSKevin Wolf     int ret;
232683f64091Sbellard 
23278d3b1a2dSKevin Wolf     count = qiov->size;
23288d3b1a2dSKevin Wolf 
232983f64091Sbellard     /* first write to align to sector start */
23306ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
233183f64091Sbellard     if (len > count)
233283f64091Sbellard         len = count;
23336ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
233483f64091Sbellard     if (len > 0) {
23359a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23369a8c4cceSKevin Wolf             return ret;
23378d3b1a2dSKevin Wolf         qemu_iovec_to_buf(qiov, 0, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)),
23388d3b1a2dSKevin Wolf                           len);
23399a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
23409a8c4cceSKevin Wolf             return ret;
234183f64091Sbellard         count -= len;
234283f64091Sbellard         if (count == 0)
23438d3b1a2dSKevin Wolf             return qiov->size;
234483f64091Sbellard         sector_num++;
234583f64091Sbellard     }
234683f64091Sbellard 
234783f64091Sbellard     /* write the sectors "in place" */
23486ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
234983f64091Sbellard     if (nb_sectors > 0) {
23508d3b1a2dSKevin Wolf         QEMUIOVector qiov_inplace;
23518d3b1a2dSKevin Wolf 
23528d3b1a2dSKevin Wolf         qemu_iovec_init(&qiov_inplace, qiov->niov);
23538d3b1a2dSKevin Wolf         qemu_iovec_concat(&qiov_inplace, qiov, len,
23548d3b1a2dSKevin Wolf                           nb_sectors << BDRV_SECTOR_BITS);
23558d3b1a2dSKevin Wolf         ret = bdrv_writev(bs, sector_num, &qiov_inplace);
23568d3b1a2dSKevin Wolf         qemu_iovec_destroy(&qiov_inplace);
23578d3b1a2dSKevin Wolf         if (ret < 0) {
23589a8c4cceSKevin Wolf             return ret;
23598d3b1a2dSKevin Wolf         }
23608d3b1a2dSKevin Wolf 
236183f64091Sbellard         sector_num += nb_sectors;
23626ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
236383f64091Sbellard         count -= len;
236483f64091Sbellard     }
236583f64091Sbellard 
236683f64091Sbellard     /* add data from the last sector */
236783f64091Sbellard     if (count > 0) {
23689a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
23699a8c4cceSKevin Wolf             return ret;
23708d3b1a2dSKevin Wolf         qemu_iovec_to_buf(qiov, qiov->size - count, tmp_buf, count);
23719a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
23729a8c4cceSKevin Wolf             return ret;
237383f64091Sbellard     }
23748d3b1a2dSKevin Wolf     return qiov->size;
23758d3b1a2dSKevin Wolf }
23768d3b1a2dSKevin Wolf 
23778d3b1a2dSKevin Wolf int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
23788d3b1a2dSKevin Wolf                 const void *buf, int count1)
23798d3b1a2dSKevin Wolf {
23808d3b1a2dSKevin Wolf     QEMUIOVector qiov;
23818d3b1a2dSKevin Wolf     struct iovec iov = {
23828d3b1a2dSKevin Wolf         .iov_base   = (void *) buf,
23838d3b1a2dSKevin Wolf         .iov_len    = count1,
23848d3b1a2dSKevin Wolf     };
23858d3b1a2dSKevin Wolf 
23868d3b1a2dSKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
23878d3b1a2dSKevin Wolf     return bdrv_pwritev(bs, offset, &qiov);
238883f64091Sbellard }
238983f64091Sbellard 
2390f08145feSKevin Wolf /*
2391f08145feSKevin Wolf  * Writes to the file and ensures that no writes are reordered across this
2392f08145feSKevin Wolf  * request (acts as a barrier)
2393f08145feSKevin Wolf  *
2394f08145feSKevin Wolf  * Returns 0 on success, -errno in error cases.
2395f08145feSKevin Wolf  */
2396f08145feSKevin Wolf int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
2397f08145feSKevin Wolf     const void *buf, int count)
2398f08145feSKevin Wolf {
2399f08145feSKevin Wolf     int ret;
2400f08145feSKevin Wolf 
2401f08145feSKevin Wolf     ret = bdrv_pwrite(bs, offset, buf, count);
2402f08145feSKevin Wolf     if (ret < 0) {
2403f08145feSKevin Wolf         return ret;
2404f08145feSKevin Wolf     }
2405f08145feSKevin Wolf 
2406f05fa4adSPaolo Bonzini     /* No flush needed for cache modes that already do it */
2407f05fa4adSPaolo Bonzini     if (bs->enable_write_cache) {
2408f08145feSKevin Wolf         bdrv_flush(bs);
2409f08145feSKevin Wolf     }
2410f08145feSKevin Wolf 
2411f08145feSKevin Wolf     return 0;
2412f08145feSKevin Wolf }
2413f08145feSKevin Wolf 
2414470c0504SStefan Hajnoczi static int coroutine_fn bdrv_co_do_copy_on_readv(BlockDriverState *bs,
2415ab185921SStefan Hajnoczi         int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2416ab185921SStefan Hajnoczi {
2417ab185921SStefan Hajnoczi     /* Perform I/O through a temporary buffer so that users who scribble over
2418ab185921SStefan Hajnoczi      * their read buffer while the operation is in progress do not end up
2419ab185921SStefan Hajnoczi      * modifying the image file.  This is critical for zero-copy guest I/O
2420ab185921SStefan Hajnoczi      * where anything might happen inside guest memory.
2421ab185921SStefan Hajnoczi      */
2422ab185921SStefan Hajnoczi     void *bounce_buffer;
2423ab185921SStefan Hajnoczi 
242479c053bdSStefan Hajnoczi     BlockDriver *drv = bs->drv;
2425ab185921SStefan Hajnoczi     struct iovec iov;
2426ab185921SStefan Hajnoczi     QEMUIOVector bounce_qiov;
2427ab185921SStefan Hajnoczi     int64_t cluster_sector_num;
2428ab185921SStefan Hajnoczi     int cluster_nb_sectors;
2429ab185921SStefan Hajnoczi     size_t skip_bytes;
2430ab185921SStefan Hajnoczi     int ret;
2431ab185921SStefan Hajnoczi 
2432ab185921SStefan Hajnoczi     /* Cover entire cluster so no additional backing file I/O is required when
2433ab185921SStefan Hajnoczi      * allocating cluster in the image file.
2434ab185921SStefan Hajnoczi      */
2435343bded4SPaolo Bonzini     bdrv_round_to_clusters(bs, sector_num, nb_sectors,
2436ab185921SStefan Hajnoczi                            &cluster_sector_num, &cluster_nb_sectors);
2437ab185921SStefan Hajnoczi 
2438470c0504SStefan Hajnoczi     trace_bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors,
2439ab185921SStefan Hajnoczi                                    cluster_sector_num, cluster_nb_sectors);
2440ab185921SStefan Hajnoczi 
2441ab185921SStefan Hajnoczi     iov.iov_len = cluster_nb_sectors * BDRV_SECTOR_SIZE;
2442ab185921SStefan Hajnoczi     iov.iov_base = bounce_buffer = qemu_blockalign(bs, iov.iov_len);
2443ab185921SStefan Hajnoczi     qemu_iovec_init_external(&bounce_qiov, &iov, 1);
2444ab185921SStefan Hajnoczi 
244579c053bdSStefan Hajnoczi     ret = drv->bdrv_co_readv(bs, cluster_sector_num, cluster_nb_sectors,
2446ab185921SStefan Hajnoczi                              &bounce_qiov);
2447ab185921SStefan Hajnoczi     if (ret < 0) {
2448ab185921SStefan Hajnoczi         goto err;
2449ab185921SStefan Hajnoczi     }
2450ab185921SStefan Hajnoczi 
245179c053bdSStefan Hajnoczi     if (drv->bdrv_co_write_zeroes &&
245279c053bdSStefan Hajnoczi         buffer_is_zero(bounce_buffer, iov.iov_len)) {
2453621f0589SKevin Wolf         ret = bdrv_co_do_write_zeroes(bs, cluster_sector_num,
245479c053bdSStefan Hajnoczi                                       cluster_nb_sectors);
245579c053bdSStefan Hajnoczi     } else {
2456f05fa4adSPaolo Bonzini         /* This does not change the data on the disk, it is not necessary
2457f05fa4adSPaolo Bonzini          * to flush even in cache=writethrough mode.
2458f05fa4adSPaolo Bonzini          */
245979c053bdSStefan Hajnoczi         ret = drv->bdrv_co_writev(bs, cluster_sector_num, cluster_nb_sectors,
2460ab185921SStefan Hajnoczi                                   &bounce_qiov);
246179c053bdSStefan Hajnoczi     }
246279c053bdSStefan Hajnoczi 
2463ab185921SStefan Hajnoczi     if (ret < 0) {
2464ab185921SStefan Hajnoczi         /* It might be okay to ignore write errors for guest requests.  If this
2465ab185921SStefan Hajnoczi          * is a deliberate copy-on-read then we don't want to ignore the error.
2466ab185921SStefan Hajnoczi          * Simply report it in all cases.
2467ab185921SStefan Hajnoczi          */
2468ab185921SStefan Hajnoczi         goto err;
2469ab185921SStefan Hajnoczi     }
2470ab185921SStefan Hajnoczi 
2471ab185921SStefan Hajnoczi     skip_bytes = (sector_num - cluster_sector_num) * BDRV_SECTOR_SIZE;
247203396148SMichael Tokarev     qemu_iovec_from_buf(qiov, 0, bounce_buffer + skip_bytes,
2473ab185921SStefan Hajnoczi                         nb_sectors * BDRV_SECTOR_SIZE);
2474ab185921SStefan Hajnoczi 
2475ab185921SStefan Hajnoczi err:
2476ab185921SStefan Hajnoczi     qemu_vfree(bounce_buffer);
2477ab185921SStefan Hajnoczi     return ret;
2478ab185921SStefan Hajnoczi }
2479ab185921SStefan Hajnoczi 
2480c5fbe571SStefan Hajnoczi /*
2481c5fbe571SStefan Hajnoczi  * Handle a read request in coroutine context
2482c5fbe571SStefan Hajnoczi  */
2483c5fbe571SStefan Hajnoczi static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
2484470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2485470c0504SStefan Hajnoczi     BdrvRequestFlags flags)
2486da1fa91dSKevin Wolf {
2487da1fa91dSKevin Wolf     BlockDriver *drv = bs->drv;
2488dbffbdcfSStefan Hajnoczi     BdrvTrackedRequest req;
2489dbffbdcfSStefan Hajnoczi     int ret;
2490da1fa91dSKevin Wolf 
2491da1fa91dSKevin Wolf     if (!drv) {
2492da1fa91dSKevin Wolf         return -ENOMEDIUM;
2493da1fa91dSKevin Wolf     }
2494da1fa91dSKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2495da1fa91dSKevin Wolf         return -EIO;
2496da1fa91dSKevin Wolf     }
2497da1fa91dSKevin Wolf 
249898f90dbaSZhi Yong Wu     /* throttling disk read I/O */
249998f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
250098f90dbaSZhi Yong Wu         bdrv_io_limits_intercept(bs, false, nb_sectors);
250198f90dbaSZhi Yong Wu     }
250298f90dbaSZhi Yong Wu 
2503f4658285SStefan Hajnoczi     if (bs->copy_on_read) {
2504470c0504SStefan Hajnoczi         flags |= BDRV_REQ_COPY_ON_READ;
2505470c0504SStefan Hajnoczi     }
2506470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2507470c0504SStefan Hajnoczi         bs->copy_on_read_in_flight++;
2508470c0504SStefan Hajnoczi     }
2509470c0504SStefan Hajnoczi 
2510470c0504SStefan Hajnoczi     if (bs->copy_on_read_in_flight) {
2511f4658285SStefan Hajnoczi         wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2512f4658285SStefan Hajnoczi     }
2513f4658285SStefan Hajnoczi 
2514dbffbdcfSStefan Hajnoczi     tracked_request_begin(&req, bs, sector_num, nb_sectors, false);
2515ab185921SStefan Hajnoczi 
2516470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2517ab185921SStefan Hajnoczi         int pnum;
2518ab185921SStefan Hajnoczi 
2519ab185921SStefan Hajnoczi         ret = bdrv_co_is_allocated(bs, sector_num, nb_sectors, &pnum);
2520ab185921SStefan Hajnoczi         if (ret < 0) {
2521ab185921SStefan Hajnoczi             goto out;
2522ab185921SStefan Hajnoczi         }
2523ab185921SStefan Hajnoczi 
2524ab185921SStefan Hajnoczi         if (!ret || pnum != nb_sectors) {
2525470c0504SStefan Hajnoczi             ret = bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors, qiov);
2526ab185921SStefan Hajnoczi             goto out;
2527ab185921SStefan Hajnoczi         }
2528ab185921SStefan Hajnoczi     }
2529ab185921SStefan Hajnoczi 
2530dbffbdcfSStefan Hajnoczi     ret = drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
2531ab185921SStefan Hajnoczi 
2532ab185921SStefan Hajnoczi out:
2533dbffbdcfSStefan Hajnoczi     tracked_request_end(&req);
2534470c0504SStefan Hajnoczi 
2535470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2536470c0504SStefan Hajnoczi         bs->copy_on_read_in_flight--;
2537470c0504SStefan Hajnoczi     }
2538470c0504SStefan Hajnoczi 
2539dbffbdcfSStefan Hajnoczi     return ret;
2540da1fa91dSKevin Wolf }
2541da1fa91dSKevin Wolf 
2542c5fbe571SStefan Hajnoczi int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
2543da1fa91dSKevin Wolf     int nb_sectors, QEMUIOVector *qiov)
2544da1fa91dSKevin Wolf {
2545c5fbe571SStefan Hajnoczi     trace_bdrv_co_readv(bs, sector_num, nb_sectors);
2546da1fa91dSKevin Wolf 
2547470c0504SStefan Hajnoczi     return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov, 0);
2548470c0504SStefan Hajnoczi }
2549470c0504SStefan Hajnoczi 
2550470c0504SStefan Hajnoczi int coroutine_fn bdrv_co_copy_on_readv(BlockDriverState *bs,
2551470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2552470c0504SStefan Hajnoczi {
2553470c0504SStefan Hajnoczi     trace_bdrv_co_copy_on_readv(bs, sector_num, nb_sectors);
2554470c0504SStefan Hajnoczi 
2555470c0504SStefan Hajnoczi     return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov,
2556470c0504SStefan Hajnoczi                             BDRV_REQ_COPY_ON_READ);
2557c5fbe571SStefan Hajnoczi }
2558c5fbe571SStefan Hajnoczi 
2559f08f2ddaSStefan Hajnoczi static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
2560f08f2ddaSStefan Hajnoczi     int64_t sector_num, int nb_sectors)
2561f08f2ddaSStefan Hajnoczi {
2562f08f2ddaSStefan Hajnoczi     BlockDriver *drv = bs->drv;
2563f08f2ddaSStefan Hajnoczi     QEMUIOVector qiov;
2564f08f2ddaSStefan Hajnoczi     struct iovec iov;
2565f08f2ddaSStefan Hajnoczi     int ret;
2566f08f2ddaSStefan Hajnoczi 
2567621f0589SKevin Wolf     /* TODO Emulate only part of misaligned requests instead of letting block
2568621f0589SKevin Wolf      * drivers return -ENOTSUP and emulate everything */
2569621f0589SKevin Wolf 
2570f08f2ddaSStefan Hajnoczi     /* First try the efficient write zeroes operation */
2571f08f2ddaSStefan Hajnoczi     if (drv->bdrv_co_write_zeroes) {
2572621f0589SKevin Wolf         ret = drv->bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2573621f0589SKevin Wolf         if (ret != -ENOTSUP) {
2574621f0589SKevin Wolf             return ret;
2575621f0589SKevin Wolf         }
2576f08f2ddaSStefan Hajnoczi     }
2577f08f2ddaSStefan Hajnoczi 
2578f08f2ddaSStefan Hajnoczi     /* Fall back to bounce buffer if write zeroes is unsupported */
2579f08f2ddaSStefan Hajnoczi     iov.iov_len  = nb_sectors * BDRV_SECTOR_SIZE;
2580f08f2ddaSStefan Hajnoczi     iov.iov_base = qemu_blockalign(bs, iov.iov_len);
2581f08f2ddaSStefan Hajnoczi     memset(iov.iov_base, 0, iov.iov_len);
2582f08f2ddaSStefan Hajnoczi     qemu_iovec_init_external(&qiov, &iov, 1);
2583f08f2ddaSStefan Hajnoczi 
2584f08f2ddaSStefan Hajnoczi     ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, &qiov);
2585f08f2ddaSStefan Hajnoczi 
2586f08f2ddaSStefan Hajnoczi     qemu_vfree(iov.iov_base);
2587f08f2ddaSStefan Hajnoczi     return ret;
2588f08f2ddaSStefan Hajnoczi }
2589f08f2ddaSStefan Hajnoczi 
2590c5fbe571SStefan Hajnoczi /*
2591c5fbe571SStefan Hajnoczi  * Handle a write request in coroutine context
2592c5fbe571SStefan Hajnoczi  */
2593c5fbe571SStefan Hajnoczi static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
2594f08f2ddaSStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
2595f08f2ddaSStefan Hajnoczi     BdrvRequestFlags flags)
2596c5fbe571SStefan Hajnoczi {
2597c5fbe571SStefan Hajnoczi     BlockDriver *drv = bs->drv;
2598dbffbdcfSStefan Hajnoczi     BdrvTrackedRequest req;
25996b7cb247SStefan Hajnoczi     int ret;
2600da1fa91dSKevin Wolf 
2601da1fa91dSKevin Wolf     if (!bs->drv) {
2602da1fa91dSKevin Wolf         return -ENOMEDIUM;
2603da1fa91dSKevin Wolf     }
2604da1fa91dSKevin Wolf     if (bs->read_only) {
2605da1fa91dSKevin Wolf         return -EACCES;
2606da1fa91dSKevin Wolf     }
2607da1fa91dSKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
2608da1fa91dSKevin Wolf         return -EIO;
2609da1fa91dSKevin Wolf     }
2610da1fa91dSKevin Wolf 
261198f90dbaSZhi Yong Wu     /* throttling disk write I/O */
261298f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
261398f90dbaSZhi Yong Wu         bdrv_io_limits_intercept(bs, true, nb_sectors);
261498f90dbaSZhi Yong Wu     }
261598f90dbaSZhi Yong Wu 
2616470c0504SStefan Hajnoczi     if (bs->copy_on_read_in_flight) {
2617f4658285SStefan Hajnoczi         wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2618f4658285SStefan Hajnoczi     }
2619f4658285SStefan Hajnoczi 
2620dbffbdcfSStefan Hajnoczi     tracked_request_begin(&req, bs, sector_num, nb_sectors, true);
2621dbffbdcfSStefan Hajnoczi 
2622f08f2ddaSStefan Hajnoczi     if (flags & BDRV_REQ_ZERO_WRITE) {
2623f08f2ddaSStefan Hajnoczi         ret = bdrv_co_do_write_zeroes(bs, sector_num, nb_sectors);
2624f08f2ddaSStefan Hajnoczi     } else {
26256b7cb247SStefan Hajnoczi         ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
2626f08f2ddaSStefan Hajnoczi     }
26276b7cb247SStefan Hajnoczi 
2628f05fa4adSPaolo Bonzini     if (ret == 0 && !bs->enable_write_cache) {
2629f05fa4adSPaolo Bonzini         ret = bdrv_co_flush(bs);
2630f05fa4adSPaolo Bonzini     }
2631f05fa4adSPaolo Bonzini 
2632da1fa91dSKevin Wolf     if (bs->dirty_bitmap) {
26331755da16SPaolo Bonzini         bdrv_set_dirty(bs, sector_num, nb_sectors);
2634da1fa91dSKevin Wolf     }
2635da1fa91dSKevin Wolf 
2636da1fa91dSKevin Wolf     if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
2637da1fa91dSKevin Wolf         bs->wr_highest_sector = sector_num + nb_sectors - 1;
2638da1fa91dSKevin Wolf     }
2639da1fa91dSKevin Wolf 
2640dbffbdcfSStefan Hajnoczi     tracked_request_end(&req);
2641dbffbdcfSStefan Hajnoczi 
26426b7cb247SStefan Hajnoczi     return ret;
2643da1fa91dSKevin Wolf }
2644da1fa91dSKevin Wolf 
2645c5fbe571SStefan Hajnoczi int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
2646c5fbe571SStefan Hajnoczi     int nb_sectors, QEMUIOVector *qiov)
2647c5fbe571SStefan Hajnoczi {
2648c5fbe571SStefan Hajnoczi     trace_bdrv_co_writev(bs, sector_num, nb_sectors);
2649c5fbe571SStefan Hajnoczi 
2650f08f2ddaSStefan Hajnoczi     return bdrv_co_do_writev(bs, sector_num, nb_sectors, qiov, 0);
2651f08f2ddaSStefan Hajnoczi }
2652f08f2ddaSStefan Hajnoczi 
2653f08f2ddaSStefan Hajnoczi int coroutine_fn bdrv_co_write_zeroes(BlockDriverState *bs,
2654f08f2ddaSStefan Hajnoczi                                       int64_t sector_num, int nb_sectors)
2655f08f2ddaSStefan Hajnoczi {
2656f08f2ddaSStefan Hajnoczi     trace_bdrv_co_write_zeroes(bs, sector_num, nb_sectors);
2657f08f2ddaSStefan Hajnoczi 
2658f08f2ddaSStefan Hajnoczi     return bdrv_co_do_writev(bs, sector_num, nb_sectors, NULL,
2659f08f2ddaSStefan Hajnoczi                              BDRV_REQ_ZERO_WRITE);
2660c5fbe571SStefan Hajnoczi }
2661c5fbe571SStefan Hajnoczi 
266283f64091Sbellard /**
266383f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
266483f64091Sbellard  */
266583f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
266683f64091Sbellard {
266783f64091Sbellard     BlockDriver *drv = bs->drv;
266851762288SStefan Hajnoczi     int ret;
266983f64091Sbellard     if (!drv)
267019cb3738Sbellard         return -ENOMEDIUM;
267183f64091Sbellard     if (!drv->bdrv_truncate)
267283f64091Sbellard         return -ENOTSUP;
267359f2689dSNaphtali Sprei     if (bs->read_only)
267459f2689dSNaphtali Sprei         return -EACCES;
26758591675fSMarcelo Tosatti     if (bdrv_in_use(bs))
26768591675fSMarcelo Tosatti         return -EBUSY;
267751762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
267851762288SStefan Hajnoczi     if (ret == 0) {
267951762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
2680145feb17SMarkus Armbruster         bdrv_dev_resize_cb(bs);
268151762288SStefan Hajnoczi     }
268251762288SStefan Hajnoczi     return ret;
268383f64091Sbellard }
268483f64091Sbellard 
268583f64091Sbellard /**
26864a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
26874a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
26884a1d5e1fSFam Zheng  */
26894a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
26904a1d5e1fSFam Zheng {
26914a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
26924a1d5e1fSFam Zheng     if (!drv) {
26934a1d5e1fSFam Zheng         return -ENOMEDIUM;
26944a1d5e1fSFam Zheng     }
26954a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
26964a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
26974a1d5e1fSFam Zheng     }
26984a1d5e1fSFam Zheng     if (bs->file) {
26994a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
27004a1d5e1fSFam Zheng     }
27014a1d5e1fSFam Zheng     return -ENOTSUP;
27024a1d5e1fSFam Zheng }
27034a1d5e1fSFam Zheng 
27044a1d5e1fSFam Zheng /**
270583f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
270683f64091Sbellard  */
270783f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
270883f64091Sbellard {
270983f64091Sbellard     BlockDriver *drv = bs->drv;
271083f64091Sbellard     if (!drv)
271119cb3738Sbellard         return -ENOMEDIUM;
271251762288SStefan Hajnoczi 
27132c6942faSMarkus Armbruster     if (bs->growable || bdrv_dev_has_removable_media(bs)) {
271446a4e4e6SStefan Hajnoczi         if (drv->bdrv_getlength) {
271583f64091Sbellard             return drv->bdrv_getlength(bs);
2716fc01f7e7Sbellard         }
271746a4e4e6SStefan Hajnoczi     }
271846a4e4e6SStefan Hajnoczi     return bs->total_sectors * BDRV_SECTOR_SIZE;
271946a4e4e6SStefan Hajnoczi }
2720fc01f7e7Sbellard 
272119cb3738Sbellard /* return 0 as number of sectors if no device present or error */
272296b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
2723fc01f7e7Sbellard {
272419cb3738Sbellard     int64_t length;
272519cb3738Sbellard     length = bdrv_getlength(bs);
272619cb3738Sbellard     if (length < 0)
272719cb3738Sbellard         length = 0;
272819cb3738Sbellard     else
27296ea44308SJan Kiszka         length = length >> BDRV_SECTOR_BITS;
273019cb3738Sbellard     *nb_sectors_ptr = length;
2731fc01f7e7Sbellard }
2732cf98951bSbellard 
27330563e191SZhi Yong Wu /* throttling disk io limits */
27340563e191SZhi Yong Wu void bdrv_set_io_limits(BlockDriverState *bs,
27350563e191SZhi Yong Wu                         BlockIOLimit *io_limits)
27360563e191SZhi Yong Wu {
27370563e191SZhi Yong Wu     bs->io_limits = *io_limits;
27380563e191SZhi Yong Wu     bs->io_limits_enabled = bdrv_io_limits_enabled(bs);
27390563e191SZhi Yong Wu }
27400563e191SZhi Yong Wu 
2741ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
2742ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
2743abd7f68dSMarkus Armbruster {
2744abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
2745abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
2746abd7f68dSMarkus Armbruster }
2747abd7f68dSMarkus Armbruster 
27481ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
2749abd7f68dSMarkus Armbruster {
2750abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
2751abd7f68dSMarkus Armbruster }
2752abd7f68dSMarkus Armbruster 
27533e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
27543e1caa5fSPaolo Bonzini {
27553e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
27563e1caa5fSPaolo Bonzini 
27573e1caa5fSPaolo Bonzini     switch (on_err) {
27583e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
27593e1caa5fSPaolo Bonzini         return (error == ENOSPC) ? BDRV_ACTION_STOP : BDRV_ACTION_REPORT;
27603e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
27613e1caa5fSPaolo Bonzini         return BDRV_ACTION_STOP;
27623e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
27633e1caa5fSPaolo Bonzini         return BDRV_ACTION_REPORT;
27643e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
27653e1caa5fSPaolo Bonzini         return BDRV_ACTION_IGNORE;
27663e1caa5fSPaolo Bonzini     default:
27673e1caa5fSPaolo Bonzini         abort();
27683e1caa5fSPaolo Bonzini     }
27693e1caa5fSPaolo Bonzini }
27703e1caa5fSPaolo Bonzini 
27713e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
27723e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
27733e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
27743e1caa5fSPaolo Bonzini  */
27753e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
27763e1caa5fSPaolo Bonzini                        bool is_read, int error)
27773e1caa5fSPaolo Bonzini {
27783e1caa5fSPaolo Bonzini     assert(error >= 0);
277932c81a4aSPaolo Bonzini     bdrv_emit_qmp_error_event(bs, QEVENT_BLOCK_IO_ERROR, action, is_read);
27803e1caa5fSPaolo Bonzini     if (action == BDRV_ACTION_STOP) {
27813e1caa5fSPaolo Bonzini         vm_stop(RUN_STATE_IO_ERROR);
27823e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
27833e1caa5fSPaolo Bonzini     }
27843e1caa5fSPaolo Bonzini }
27853e1caa5fSPaolo Bonzini 
2786b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
2787b338082bSbellard {
2788b338082bSbellard     return bs->read_only;
2789b338082bSbellard }
2790b338082bSbellard 
2791985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
2792985a03b0Sths {
2793985a03b0Sths     return bs->sg;
2794985a03b0Sths }
2795985a03b0Sths 
2796e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
2797e900a7b7SChristoph Hellwig {
2798e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
2799e900a7b7SChristoph Hellwig }
2800e900a7b7SChristoph Hellwig 
2801425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
2802425b0148SPaolo Bonzini {
2803425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
280455b110f2SJeff Cody 
280555b110f2SJeff Cody     /* so a reopen() will preserve wce */
280655b110f2SJeff Cody     if (wce) {
280755b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
280855b110f2SJeff Cody     } else {
280955b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
281055b110f2SJeff Cody     }
2811425b0148SPaolo Bonzini }
2812425b0148SPaolo Bonzini 
2813ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
2814ea2384d3Sbellard {
2815ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
2816ea2384d3Sbellard         return 1;
2817ea2384d3Sbellard     return bs->encrypted;
2818ea2384d3Sbellard }
2819ea2384d3Sbellard 
2820c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
2821c0f4ce77Saliguori {
2822c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
2823c0f4ce77Saliguori 
2824c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
2825c0f4ce77Saliguori         return 1;
2826c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
2827c0f4ce77Saliguori }
2828c0f4ce77Saliguori 
2829ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
2830ea2384d3Sbellard {
2831ea2384d3Sbellard     int ret;
2832ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
2833ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
2834ea2384d3Sbellard         if (ret < 0)
2835ea2384d3Sbellard             return ret;
2836ea2384d3Sbellard         if (!bs->encrypted)
2837ea2384d3Sbellard             return 0;
2838ea2384d3Sbellard     }
2839fd04a2aeSShahar Havivi     if (!bs->encrypted) {
2840fd04a2aeSShahar Havivi         return -EINVAL;
2841fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
2842fd04a2aeSShahar Havivi         return -ENOMEDIUM;
2843fd04a2aeSShahar Havivi     }
2844c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
2845bb5fc20fSaliguori     if (ret < 0) {
2846bb5fc20fSaliguori         bs->valid_key = 0;
2847bb5fc20fSaliguori     } else if (!bs->valid_key) {
2848bb5fc20fSaliguori         bs->valid_key = 1;
2849bb5fc20fSaliguori         /* call the change callback now, we skipped it on open */
28507d4b4ba5SMarkus Armbruster         bdrv_dev_change_media_cb(bs, true);
2851bb5fc20fSaliguori     }
2852c0f4ce77Saliguori     return ret;
2853ea2384d3Sbellard }
2854ea2384d3Sbellard 
2855f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
2856ea2384d3Sbellard {
2857f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
2858ea2384d3Sbellard }
2859ea2384d3Sbellard 
2860ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
2861ea2384d3Sbellard                          void *opaque)
2862ea2384d3Sbellard {
2863ea2384d3Sbellard     BlockDriver *drv;
2864ea2384d3Sbellard 
28658a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
2866ea2384d3Sbellard         it(opaque, drv->format_name);
2867ea2384d3Sbellard     }
2868ea2384d3Sbellard }
2869ea2384d3Sbellard 
2870b338082bSbellard BlockDriverState *bdrv_find(const char *name)
2871b338082bSbellard {
2872b338082bSbellard     BlockDriverState *bs;
2873b338082bSbellard 
28741b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
28751b7bdbc1SStefan Hajnoczi         if (!strcmp(name, bs->device_name)) {
2876b338082bSbellard             return bs;
2877b338082bSbellard         }
28781b7bdbc1SStefan Hajnoczi     }
2879b338082bSbellard     return NULL;
2880b338082bSbellard }
2881b338082bSbellard 
28822f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
28832f399b0aSMarkus Armbruster {
28842f399b0aSMarkus Armbruster     if (!bs) {
28852f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
28862f399b0aSMarkus Armbruster     }
28872f399b0aSMarkus Armbruster     return QTAILQ_NEXT(bs, list);
28882f399b0aSMarkus Armbruster }
28892f399b0aSMarkus Armbruster 
289051de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
289181d0912dSbellard {
289281d0912dSbellard     BlockDriverState *bs;
289381d0912dSbellard 
28941b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
289551de9760Saliguori         it(opaque, bs);
289681d0912dSbellard     }
289781d0912dSbellard }
289881d0912dSbellard 
2899ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
2900ea2384d3Sbellard {
2901ea2384d3Sbellard     return bs->device_name;
2902ea2384d3Sbellard }
2903ea2384d3Sbellard 
2904c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
2905c8433287SMarkus Armbruster {
2906c8433287SMarkus Armbruster     return bs->open_flags;
2907c8433287SMarkus Armbruster }
2908c8433287SMarkus Armbruster 
2909c6ca28d6Saliguori void bdrv_flush_all(void)
2910c6ca28d6Saliguori {
2911c6ca28d6Saliguori     BlockDriverState *bs;
2912c6ca28d6Saliguori 
29131b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
2914c6ca28d6Saliguori         bdrv_flush(bs);
2915c6ca28d6Saliguori     }
29161b7bdbc1SStefan Hajnoczi }
2917c6ca28d6Saliguori 
2918f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
2919f2feebbdSKevin Wolf {
2920f2feebbdSKevin Wolf     assert(bs->drv);
2921f2feebbdSKevin Wolf 
2922336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
2923336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
2924f2feebbdSKevin Wolf     }
2925f2feebbdSKevin Wolf 
2926f2feebbdSKevin Wolf     return 1;
2927f2feebbdSKevin Wolf }
2928f2feebbdSKevin Wolf 
2929376ae3f1SStefan Hajnoczi typedef struct BdrvCoIsAllocatedData {
2930376ae3f1SStefan Hajnoczi     BlockDriverState *bs;
2931b35b2bbaSMiroslav Rezanina     BlockDriverState *base;
2932376ae3f1SStefan Hajnoczi     int64_t sector_num;
2933376ae3f1SStefan Hajnoczi     int nb_sectors;
2934376ae3f1SStefan Hajnoczi     int *pnum;
2935376ae3f1SStefan Hajnoczi     int ret;
2936376ae3f1SStefan Hajnoczi     bool done;
2937376ae3f1SStefan Hajnoczi } BdrvCoIsAllocatedData;
2938376ae3f1SStefan Hajnoczi 
2939f58c7b35Sths /*
2940f58c7b35Sths  * Returns true iff the specified sector is present in the disk image. Drivers
2941f58c7b35Sths  * not implementing the functionality are assumed to not support backing files,
2942f58c7b35Sths  * hence all their sectors are reported as allocated.
2943f58c7b35Sths  *
2944bd9533e3SStefan Hajnoczi  * If 'sector_num' is beyond the end of the disk image the return value is 0
2945bd9533e3SStefan Hajnoczi  * and 'pnum' is set to 0.
2946bd9533e3SStefan Hajnoczi  *
2947f58c7b35Sths  * 'pnum' is set to the number of sectors (including and immediately following
2948f58c7b35Sths  * the specified sector) that are known to be in the same
2949f58c7b35Sths  * allocated/unallocated state.
2950f58c7b35Sths  *
2951bd9533e3SStefan Hajnoczi  * 'nb_sectors' is the max value 'pnum' should be set to.  If nb_sectors goes
2952bd9533e3SStefan Hajnoczi  * beyond the end of the disk image it will be clamped.
2953f58c7b35Sths  */
2954060f51c9SStefan Hajnoczi int coroutine_fn bdrv_co_is_allocated(BlockDriverState *bs, int64_t sector_num,
2955060f51c9SStefan Hajnoczi                                       int nb_sectors, int *pnum)
2956f58c7b35Sths {
2957f58c7b35Sths     int64_t n;
2958bd9533e3SStefan Hajnoczi 
29596aebab14SStefan Hajnoczi     if (sector_num >= bs->total_sectors) {
29606aebab14SStefan Hajnoczi         *pnum = 0;
29616aebab14SStefan Hajnoczi         return 0;
29626aebab14SStefan Hajnoczi     }
2963bd9533e3SStefan Hajnoczi 
29646aebab14SStefan Hajnoczi     n = bs->total_sectors - sector_num;
2965bd9533e3SStefan Hajnoczi     if (n < nb_sectors) {
2966bd9533e3SStefan Hajnoczi         nb_sectors = n;
2967bd9533e3SStefan Hajnoczi     }
2968bd9533e3SStefan Hajnoczi 
2969bd9533e3SStefan Hajnoczi     if (!bs->drv->bdrv_co_is_allocated) {
2970bd9533e3SStefan Hajnoczi         *pnum = nb_sectors;
29716aebab14SStefan Hajnoczi         return 1;
29726aebab14SStefan Hajnoczi     }
29736aebab14SStefan Hajnoczi 
2974060f51c9SStefan Hajnoczi     return bs->drv->bdrv_co_is_allocated(bs, sector_num, nb_sectors, pnum);
2975060f51c9SStefan Hajnoczi }
2976060f51c9SStefan Hajnoczi 
2977060f51c9SStefan Hajnoczi /* Coroutine wrapper for bdrv_is_allocated() */
2978060f51c9SStefan Hajnoczi static void coroutine_fn bdrv_is_allocated_co_entry(void *opaque)
2979060f51c9SStefan Hajnoczi {
2980060f51c9SStefan Hajnoczi     BdrvCoIsAllocatedData *data = opaque;
2981060f51c9SStefan Hajnoczi     BlockDriverState *bs = data->bs;
2982060f51c9SStefan Hajnoczi 
2983060f51c9SStefan Hajnoczi     data->ret = bdrv_co_is_allocated(bs, data->sector_num, data->nb_sectors,
2984060f51c9SStefan Hajnoczi                                      data->pnum);
2985060f51c9SStefan Hajnoczi     data->done = true;
2986060f51c9SStefan Hajnoczi }
2987060f51c9SStefan Hajnoczi 
2988060f51c9SStefan Hajnoczi /*
2989060f51c9SStefan Hajnoczi  * Synchronous wrapper around bdrv_co_is_allocated().
2990060f51c9SStefan Hajnoczi  *
2991060f51c9SStefan Hajnoczi  * See bdrv_co_is_allocated() for details.
2992060f51c9SStefan Hajnoczi  */
2993060f51c9SStefan Hajnoczi int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
2994060f51c9SStefan Hajnoczi                       int *pnum)
2995060f51c9SStefan Hajnoczi {
2996376ae3f1SStefan Hajnoczi     Coroutine *co;
2997376ae3f1SStefan Hajnoczi     BdrvCoIsAllocatedData data = {
2998376ae3f1SStefan Hajnoczi         .bs = bs,
2999376ae3f1SStefan Hajnoczi         .sector_num = sector_num,
3000376ae3f1SStefan Hajnoczi         .nb_sectors = nb_sectors,
3001376ae3f1SStefan Hajnoczi         .pnum = pnum,
3002376ae3f1SStefan Hajnoczi         .done = false,
3003376ae3f1SStefan Hajnoczi     };
3004376ae3f1SStefan Hajnoczi 
3005376ae3f1SStefan Hajnoczi     co = qemu_coroutine_create(bdrv_is_allocated_co_entry);
3006376ae3f1SStefan Hajnoczi     qemu_coroutine_enter(co, &data);
3007376ae3f1SStefan Hajnoczi     while (!data.done) {
3008376ae3f1SStefan Hajnoczi         qemu_aio_wait();
3009376ae3f1SStefan Hajnoczi     }
3010376ae3f1SStefan Hajnoczi     return data.ret;
3011376ae3f1SStefan Hajnoczi }
3012f58c7b35Sths 
3013188a7bbfSPaolo Bonzini /*
3014188a7bbfSPaolo Bonzini  * Given an image chain: ... -> [BASE] -> [INTER1] -> [INTER2] -> [TOP]
3015188a7bbfSPaolo Bonzini  *
3016188a7bbfSPaolo Bonzini  * Return true if the given sector is allocated in any image between
3017188a7bbfSPaolo Bonzini  * BASE and TOP (inclusive).  BASE can be NULL to check if the given
3018188a7bbfSPaolo Bonzini  * sector is allocated in any image of the chain.  Return false otherwise.
3019188a7bbfSPaolo Bonzini  *
3020188a7bbfSPaolo Bonzini  * 'pnum' is set to the number of sectors (including and immediately following
3021188a7bbfSPaolo Bonzini  *  the specified sector) that are known to be in the same
3022188a7bbfSPaolo Bonzini  *  allocated/unallocated state.
3023188a7bbfSPaolo Bonzini  *
3024188a7bbfSPaolo Bonzini  */
3025188a7bbfSPaolo Bonzini int coroutine_fn bdrv_co_is_allocated_above(BlockDriverState *top,
3026188a7bbfSPaolo Bonzini                                             BlockDriverState *base,
3027188a7bbfSPaolo Bonzini                                             int64_t sector_num,
3028188a7bbfSPaolo Bonzini                                             int nb_sectors, int *pnum)
3029188a7bbfSPaolo Bonzini {
3030188a7bbfSPaolo Bonzini     BlockDriverState *intermediate;
3031188a7bbfSPaolo Bonzini     int ret, n = nb_sectors;
3032188a7bbfSPaolo Bonzini 
3033188a7bbfSPaolo Bonzini     intermediate = top;
3034188a7bbfSPaolo Bonzini     while (intermediate && intermediate != base) {
3035188a7bbfSPaolo Bonzini         int pnum_inter;
3036188a7bbfSPaolo Bonzini         ret = bdrv_co_is_allocated(intermediate, sector_num, nb_sectors,
3037188a7bbfSPaolo Bonzini                                    &pnum_inter);
3038188a7bbfSPaolo Bonzini         if (ret < 0) {
3039188a7bbfSPaolo Bonzini             return ret;
3040188a7bbfSPaolo Bonzini         } else if (ret) {
3041188a7bbfSPaolo Bonzini             *pnum = pnum_inter;
3042188a7bbfSPaolo Bonzini             return 1;
3043188a7bbfSPaolo Bonzini         }
3044188a7bbfSPaolo Bonzini 
3045188a7bbfSPaolo Bonzini         /*
3046188a7bbfSPaolo Bonzini          * [sector_num, nb_sectors] is unallocated on top but intermediate
3047188a7bbfSPaolo Bonzini          * might have
3048188a7bbfSPaolo Bonzini          *
3049188a7bbfSPaolo Bonzini          * [sector_num+x, nr_sectors] allocated.
3050188a7bbfSPaolo Bonzini          */
305163ba17d3SVishvananda Ishaya         if (n > pnum_inter &&
305263ba17d3SVishvananda Ishaya             (intermediate == top ||
305363ba17d3SVishvananda Ishaya              sector_num + pnum_inter < intermediate->total_sectors)) {
3054188a7bbfSPaolo Bonzini             n = pnum_inter;
3055188a7bbfSPaolo Bonzini         }
3056188a7bbfSPaolo Bonzini 
3057188a7bbfSPaolo Bonzini         intermediate = intermediate->backing_hd;
3058188a7bbfSPaolo Bonzini     }
3059188a7bbfSPaolo Bonzini 
3060188a7bbfSPaolo Bonzini     *pnum = n;
3061188a7bbfSPaolo Bonzini     return 0;
3062188a7bbfSPaolo Bonzini }
3063188a7bbfSPaolo Bonzini 
3064b35b2bbaSMiroslav Rezanina /* Coroutine wrapper for bdrv_is_allocated_above() */
3065b35b2bbaSMiroslav Rezanina static void coroutine_fn bdrv_is_allocated_above_co_entry(void *opaque)
3066b35b2bbaSMiroslav Rezanina {
3067b35b2bbaSMiroslav Rezanina     BdrvCoIsAllocatedData *data = opaque;
3068b35b2bbaSMiroslav Rezanina     BlockDriverState *top = data->bs;
3069b35b2bbaSMiroslav Rezanina     BlockDriverState *base = data->base;
3070b35b2bbaSMiroslav Rezanina 
3071b35b2bbaSMiroslav Rezanina     data->ret = bdrv_co_is_allocated_above(top, base, data->sector_num,
3072b35b2bbaSMiroslav Rezanina                                            data->nb_sectors, data->pnum);
3073b35b2bbaSMiroslav Rezanina     data->done = true;
3074b35b2bbaSMiroslav Rezanina }
3075b35b2bbaSMiroslav Rezanina 
3076b35b2bbaSMiroslav Rezanina /*
3077b35b2bbaSMiroslav Rezanina  * Synchronous wrapper around bdrv_co_is_allocated_above().
3078b35b2bbaSMiroslav Rezanina  *
3079b35b2bbaSMiroslav Rezanina  * See bdrv_co_is_allocated_above() for details.
3080b35b2bbaSMiroslav Rezanina  */
3081b35b2bbaSMiroslav Rezanina int bdrv_is_allocated_above(BlockDriverState *top, BlockDriverState *base,
3082b35b2bbaSMiroslav Rezanina                             int64_t sector_num, int nb_sectors, int *pnum)
3083b35b2bbaSMiroslav Rezanina {
3084b35b2bbaSMiroslav Rezanina     Coroutine *co;
3085b35b2bbaSMiroslav Rezanina     BdrvCoIsAllocatedData data = {
3086b35b2bbaSMiroslav Rezanina         .bs = top,
3087b35b2bbaSMiroslav Rezanina         .base = base,
3088b35b2bbaSMiroslav Rezanina         .sector_num = sector_num,
3089b35b2bbaSMiroslav Rezanina         .nb_sectors = nb_sectors,
3090b35b2bbaSMiroslav Rezanina         .pnum = pnum,
3091b35b2bbaSMiroslav Rezanina         .done = false,
3092b35b2bbaSMiroslav Rezanina     };
3093b35b2bbaSMiroslav Rezanina 
3094b35b2bbaSMiroslav Rezanina     co = qemu_coroutine_create(bdrv_is_allocated_above_co_entry);
3095b35b2bbaSMiroslav Rezanina     qemu_coroutine_enter(co, &data);
3096b35b2bbaSMiroslav Rezanina     while (!data.done) {
3097b35b2bbaSMiroslav Rezanina         qemu_aio_wait();
3098b35b2bbaSMiroslav Rezanina     }
3099b35b2bbaSMiroslav Rezanina     return data.ret;
3100b35b2bbaSMiroslav Rezanina }
3101b35b2bbaSMiroslav Rezanina 
3102ac84adacSPaolo Bonzini BlockInfo *bdrv_query_info(BlockDriverState *bs)
3103ac84adacSPaolo Bonzini {
3104ac84adacSPaolo Bonzini     BlockInfo *info = g_malloc0(sizeof(*info));
3105ac84adacSPaolo Bonzini     info->device = g_strdup(bs->device_name);
3106ac84adacSPaolo Bonzini     info->type = g_strdup("unknown");
3107ac84adacSPaolo Bonzini     info->locked = bdrv_dev_is_medium_locked(bs);
3108ac84adacSPaolo Bonzini     info->removable = bdrv_dev_has_removable_media(bs);
3109ac84adacSPaolo Bonzini 
3110ac84adacSPaolo Bonzini     if (bdrv_dev_has_removable_media(bs)) {
3111ac84adacSPaolo Bonzini         info->has_tray_open = true;
3112ac84adacSPaolo Bonzini         info->tray_open = bdrv_dev_is_tray_open(bs);
3113ac84adacSPaolo Bonzini     }
3114ac84adacSPaolo Bonzini 
3115ac84adacSPaolo Bonzini     if (bdrv_iostatus_is_enabled(bs)) {
3116ac84adacSPaolo Bonzini         info->has_io_status = true;
3117ac84adacSPaolo Bonzini         info->io_status = bs->iostatus;
3118ac84adacSPaolo Bonzini     }
3119ac84adacSPaolo Bonzini 
3120b9a9b3a4SPaolo Bonzini     if (bs->dirty_bitmap) {
3121b9a9b3a4SPaolo Bonzini         info->has_dirty = true;
3122b9a9b3a4SPaolo Bonzini         info->dirty = g_malloc0(sizeof(*info->dirty));
3123acc906c6SPaolo Bonzini         info->dirty->count = bdrv_get_dirty_count(bs) * BDRV_SECTOR_SIZE;
312450717e94SPaolo Bonzini         info->dirty->granularity =
312550717e94SPaolo Bonzini             ((int64_t) BDRV_SECTOR_SIZE << hbitmap_granularity(bs->dirty_bitmap));
3126b9a9b3a4SPaolo Bonzini     }
3127b9a9b3a4SPaolo Bonzini 
3128ac84adacSPaolo Bonzini     if (bs->drv) {
3129ac84adacSPaolo Bonzini         info->has_inserted = true;
3130ac84adacSPaolo Bonzini         info->inserted = g_malloc0(sizeof(*info->inserted));
3131ac84adacSPaolo Bonzini         info->inserted->file = g_strdup(bs->filename);
3132ac84adacSPaolo Bonzini         info->inserted->ro = bs->read_only;
3133ac84adacSPaolo Bonzini         info->inserted->drv = g_strdup(bs->drv->format_name);
3134ac84adacSPaolo Bonzini         info->inserted->encrypted = bs->encrypted;
3135ac84adacSPaolo Bonzini         info->inserted->encryption_key_missing = bdrv_key_required(bs);
3136ac84adacSPaolo Bonzini 
3137ac84adacSPaolo Bonzini         if (bs->backing_file[0]) {
3138ac84adacSPaolo Bonzini             info->inserted->has_backing_file = true;
3139ac84adacSPaolo Bonzini             info->inserted->backing_file = g_strdup(bs->backing_file);
3140ac84adacSPaolo Bonzini         }
3141ac84adacSPaolo Bonzini 
3142ac84adacSPaolo Bonzini         info->inserted->backing_file_depth = bdrv_get_backing_file_depth(bs);
3143ac84adacSPaolo Bonzini 
3144ac84adacSPaolo Bonzini         if (bs->io_limits_enabled) {
3145ac84adacSPaolo Bonzini             info->inserted->bps =
3146ac84adacSPaolo Bonzini                            bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL];
3147ac84adacSPaolo Bonzini             info->inserted->bps_rd =
3148ac84adacSPaolo Bonzini                            bs->io_limits.bps[BLOCK_IO_LIMIT_READ];
3149ac84adacSPaolo Bonzini             info->inserted->bps_wr =
3150ac84adacSPaolo Bonzini                            bs->io_limits.bps[BLOCK_IO_LIMIT_WRITE];
3151ac84adacSPaolo Bonzini             info->inserted->iops =
3152ac84adacSPaolo Bonzini                            bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL];
3153ac84adacSPaolo Bonzini             info->inserted->iops_rd =
3154ac84adacSPaolo Bonzini                            bs->io_limits.iops[BLOCK_IO_LIMIT_READ];
3155ac84adacSPaolo Bonzini             info->inserted->iops_wr =
3156ac84adacSPaolo Bonzini                            bs->io_limits.iops[BLOCK_IO_LIMIT_WRITE];
3157ac84adacSPaolo Bonzini         }
3158ac84adacSPaolo Bonzini     }
3159ac84adacSPaolo Bonzini     return info;
3160ac84adacSPaolo Bonzini }
3161ac84adacSPaolo Bonzini 
3162b2023818SLuiz Capitulino BlockInfoList *qmp_query_block(Error **errp)
3163b338082bSbellard {
3164ac84adacSPaolo Bonzini     BlockInfoList *head = NULL, **p_next = &head;
3165d15e5465SLuiz Capitulino     BlockDriverState *bs;
3166d15e5465SLuiz Capitulino 
31671b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
3168b2023818SLuiz Capitulino         BlockInfoList *info = g_malloc0(sizeof(*info));
3169ac84adacSPaolo Bonzini         info->value = bdrv_query_info(bs);
3170d15e5465SLuiz Capitulino 
3171ac84adacSPaolo Bonzini         *p_next = info;
3172ac84adacSPaolo Bonzini         p_next = &info->next;
3173d15e5465SLuiz Capitulino     }
3174d15e5465SLuiz Capitulino 
3175b2023818SLuiz Capitulino     return head;
3176b338082bSbellard }
3177a36e69ddSths 
31789887b616SPaolo Bonzini BlockStats *bdrv_query_stats(const BlockDriverState *bs)
3179a36e69ddSths {
3180f11f57e4SLuiz Capitulino     BlockStats *s;
3181218a536aSLuiz Capitulino 
3182f11f57e4SLuiz Capitulino     s = g_malloc0(sizeof(*s));
3183218a536aSLuiz Capitulino 
3184f11f57e4SLuiz Capitulino     if (bs->device_name[0]) {
3185f11f57e4SLuiz Capitulino         s->has_device = true;
3186f11f57e4SLuiz Capitulino         s->device = g_strdup(bs->device_name);
3187218a536aSLuiz Capitulino     }
3188218a536aSLuiz Capitulino 
3189f11f57e4SLuiz Capitulino     s->stats = g_malloc0(sizeof(*s->stats));
3190f11f57e4SLuiz Capitulino     s->stats->rd_bytes = bs->nr_bytes[BDRV_ACCT_READ];
3191f11f57e4SLuiz Capitulino     s->stats->wr_bytes = bs->nr_bytes[BDRV_ACCT_WRITE];
3192f11f57e4SLuiz Capitulino     s->stats->rd_operations = bs->nr_ops[BDRV_ACCT_READ];
3193f11f57e4SLuiz Capitulino     s->stats->wr_operations = bs->nr_ops[BDRV_ACCT_WRITE];
3194f11f57e4SLuiz Capitulino     s->stats->wr_highest_offset = bs->wr_highest_sector * BDRV_SECTOR_SIZE;
3195f11f57e4SLuiz Capitulino     s->stats->flush_operations = bs->nr_ops[BDRV_ACCT_FLUSH];
3196f11f57e4SLuiz Capitulino     s->stats->wr_total_time_ns = bs->total_time_ns[BDRV_ACCT_WRITE];
3197f11f57e4SLuiz Capitulino     s->stats->rd_total_time_ns = bs->total_time_ns[BDRV_ACCT_READ];
3198f11f57e4SLuiz Capitulino     s->stats->flush_total_time_ns = bs->total_time_ns[BDRV_ACCT_FLUSH];
3199294cc35fSKevin Wolf 
3200294cc35fSKevin Wolf     if (bs->file) {
3201f11f57e4SLuiz Capitulino         s->has_parent = true;
32029887b616SPaolo Bonzini         s->parent = bdrv_query_stats(bs->file);
3203294cc35fSKevin Wolf     }
3204294cc35fSKevin Wolf 
3205f11f57e4SLuiz Capitulino     return s;
3206294cc35fSKevin Wolf }
3207294cc35fSKevin Wolf 
3208f11f57e4SLuiz Capitulino BlockStatsList *qmp_query_blockstats(Error **errp)
3209218a536aSLuiz Capitulino {
32109887b616SPaolo Bonzini     BlockStatsList *head = NULL, **p_next = &head;
3211a36e69ddSths     BlockDriverState *bs;
3212a36e69ddSths 
32131b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
3214f11f57e4SLuiz Capitulino         BlockStatsList *info = g_malloc0(sizeof(*info));
32159887b616SPaolo Bonzini         info->value = bdrv_query_stats(bs);
3216f11f57e4SLuiz Capitulino 
32179887b616SPaolo Bonzini         *p_next = info;
32189887b616SPaolo Bonzini         p_next = &info->next;
3219a36e69ddSths     }
3220218a536aSLuiz Capitulino 
3221f11f57e4SLuiz Capitulino     return head;
3222a36e69ddSths }
3223ea2384d3Sbellard 
3224045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
3225045df330Saliguori {
3226045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
3227045df330Saliguori         return bs->backing_file;
3228045df330Saliguori     else if (bs->encrypted)
3229045df330Saliguori         return bs->filename;
3230045df330Saliguori     else
3231045df330Saliguori         return NULL;
3232045df330Saliguori }
3233045df330Saliguori 
323483f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
323583f64091Sbellard                                char *filename, int filename_size)
323683f64091Sbellard {
323783f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
323883f64091Sbellard }
323983f64091Sbellard 
3240faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
3241faea38e7Sbellard                           const uint8_t *buf, int nb_sectors)
3242faea38e7Sbellard {
3243faea38e7Sbellard     BlockDriver *drv = bs->drv;
3244faea38e7Sbellard     if (!drv)
324519cb3738Sbellard         return -ENOMEDIUM;
3246faea38e7Sbellard     if (!drv->bdrv_write_compressed)
3247faea38e7Sbellard         return -ENOTSUP;
3248fbb7b4e0SKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors))
3249fbb7b4e0SKevin Wolf         return -EIO;
32507cd1e32aSlirans@il.ibm.com 
32511755da16SPaolo Bonzini     assert(!bs->dirty_bitmap);
32527cd1e32aSlirans@il.ibm.com 
3253faea38e7Sbellard     return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
3254faea38e7Sbellard }
3255faea38e7Sbellard 
3256faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3257faea38e7Sbellard {
3258faea38e7Sbellard     BlockDriver *drv = bs->drv;
3259faea38e7Sbellard     if (!drv)
326019cb3738Sbellard         return -ENOMEDIUM;
3261faea38e7Sbellard     if (!drv->bdrv_get_info)
3262faea38e7Sbellard         return -ENOTSUP;
3263faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3264faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3265faea38e7Sbellard }
3266faea38e7Sbellard 
326745566e9cSChristoph Hellwig int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
326845566e9cSChristoph Hellwig                       int64_t pos, int size)
3269178e08a5Saliguori {
3270cf8074b3SKevin Wolf     QEMUIOVector qiov;
3271cf8074b3SKevin Wolf     struct iovec iov = {
3272cf8074b3SKevin Wolf         .iov_base   = (void *) buf,
3273cf8074b3SKevin Wolf         .iov_len    = size,
3274cf8074b3SKevin Wolf     };
3275cf8074b3SKevin Wolf 
3276cf8074b3SKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
3277cf8074b3SKevin Wolf     return bdrv_writev_vmstate(bs, &qiov, pos);
3278cf8074b3SKevin Wolf }
3279cf8074b3SKevin Wolf 
3280cf8074b3SKevin Wolf int bdrv_writev_vmstate(BlockDriverState *bs, QEMUIOVector *qiov, int64_t pos)
3281cf8074b3SKevin Wolf {
3282178e08a5Saliguori     BlockDriver *drv = bs->drv;
3283cf8074b3SKevin Wolf 
3284cf8074b3SKevin Wolf     if (!drv) {
3285178e08a5Saliguori         return -ENOMEDIUM;
3286cf8074b3SKevin Wolf     } else if (drv->bdrv_save_vmstate) {
3287cf8074b3SKevin Wolf         return drv->bdrv_save_vmstate(bs, qiov, pos);
3288cf8074b3SKevin Wolf     } else if (bs->file) {
3289cf8074b3SKevin Wolf         return bdrv_writev_vmstate(bs->file, qiov, pos);
3290cf8074b3SKevin Wolf     }
3291cf8074b3SKevin Wolf 
32927cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3293178e08a5Saliguori }
3294178e08a5Saliguori 
329545566e9cSChristoph Hellwig int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
329645566e9cSChristoph Hellwig                       int64_t pos, int size)
3297178e08a5Saliguori {
3298178e08a5Saliguori     BlockDriver *drv = bs->drv;
3299178e08a5Saliguori     if (!drv)
3300178e08a5Saliguori         return -ENOMEDIUM;
33017cdb1f6dSMORITA Kazutaka     if (drv->bdrv_load_vmstate)
330245566e9cSChristoph Hellwig         return drv->bdrv_load_vmstate(bs, buf, pos, size);
33037cdb1f6dSMORITA Kazutaka     if (bs->file)
33047cdb1f6dSMORITA Kazutaka         return bdrv_load_vmstate(bs->file, buf, pos, size);
33057cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3306178e08a5Saliguori }
3307178e08a5Saliguori 
33088b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
33098b9b0cc2SKevin Wolf {
33108b9b0cc2SKevin Wolf     BlockDriver *drv = bs->drv;
33118b9b0cc2SKevin Wolf 
33128b9b0cc2SKevin Wolf     if (!drv || !drv->bdrv_debug_event) {
33138b9b0cc2SKevin Wolf         return;
33148b9b0cc2SKevin Wolf     }
33158b9b0cc2SKevin Wolf 
33160ed8b6f6SBlue Swirl     drv->bdrv_debug_event(bs, event);
331741c695c7SKevin Wolf }
33188b9b0cc2SKevin Wolf 
331941c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
332041c695c7SKevin Wolf                           const char *tag)
332141c695c7SKevin Wolf {
332241c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
332341c695c7SKevin Wolf         bs = bs->file;
332441c695c7SKevin Wolf     }
332541c695c7SKevin Wolf 
332641c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
332741c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
332841c695c7SKevin Wolf     }
332941c695c7SKevin Wolf 
333041c695c7SKevin Wolf     return -ENOTSUP;
333141c695c7SKevin Wolf }
333241c695c7SKevin Wolf 
333341c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
333441c695c7SKevin Wolf {
333541c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_resume) {
333641c695c7SKevin Wolf         bs = bs->file;
333741c695c7SKevin Wolf     }
333841c695c7SKevin Wolf 
333941c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
334041c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
334141c695c7SKevin Wolf     }
334241c695c7SKevin Wolf 
334341c695c7SKevin Wolf     return -ENOTSUP;
334441c695c7SKevin Wolf }
334541c695c7SKevin Wolf 
334641c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
334741c695c7SKevin Wolf {
334841c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
334941c695c7SKevin Wolf         bs = bs->file;
335041c695c7SKevin Wolf     }
335141c695c7SKevin Wolf 
335241c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
335341c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
335441c695c7SKevin Wolf     }
335541c695c7SKevin Wolf 
335641c695c7SKevin Wolf     return false;
33578b9b0cc2SKevin Wolf }
33588b9b0cc2SKevin Wolf 
3359faea38e7Sbellard /**************************************************************/
3360faea38e7Sbellard /* handling of snapshots */
3361faea38e7Sbellard 
3362feeee5acSMiguel Di Ciurcio Filho int bdrv_can_snapshot(BlockDriverState *bs)
3363feeee5acSMiguel Di Ciurcio Filho {
3364feeee5acSMiguel Di Ciurcio Filho     BlockDriver *drv = bs->drv;
336507b70bfbSMarkus Armbruster     if (!drv || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
3366feeee5acSMiguel Di Ciurcio Filho         return 0;
3367feeee5acSMiguel Di Ciurcio Filho     }
3368feeee5acSMiguel Di Ciurcio Filho 
3369feeee5acSMiguel Di Ciurcio Filho     if (!drv->bdrv_snapshot_create) {
3370feeee5acSMiguel Di Ciurcio Filho         if (bs->file != NULL) {
3371feeee5acSMiguel Di Ciurcio Filho             return bdrv_can_snapshot(bs->file);
3372feeee5acSMiguel Di Ciurcio Filho         }
3373feeee5acSMiguel Di Ciurcio Filho         return 0;
3374feeee5acSMiguel Di Ciurcio Filho     }
3375feeee5acSMiguel Di Ciurcio Filho 
3376feeee5acSMiguel Di Ciurcio Filho     return 1;
3377feeee5acSMiguel Di Ciurcio Filho }
3378feeee5acSMiguel Di Ciurcio Filho 
3379199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3380199630b6SBlue Swirl {
3381199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3382199630b6SBlue Swirl }
3383199630b6SBlue Swirl 
3384f9092b10SMarkus Armbruster BlockDriverState *bdrv_snapshots(void)
3385f9092b10SMarkus Armbruster {
3386f9092b10SMarkus Armbruster     BlockDriverState *bs;
3387f9092b10SMarkus Armbruster 
33883ac906f7SMarkus Armbruster     if (bs_snapshots) {
3389f9092b10SMarkus Armbruster         return bs_snapshots;
33903ac906f7SMarkus Armbruster     }
3391f9092b10SMarkus Armbruster 
3392f9092b10SMarkus Armbruster     bs = NULL;
3393f9092b10SMarkus Armbruster     while ((bs = bdrv_next(bs))) {
3394f9092b10SMarkus Armbruster         if (bdrv_can_snapshot(bs)) {
33953ac906f7SMarkus Armbruster             bs_snapshots = bs;
33963ac906f7SMarkus Armbruster             return bs;
3397f9092b10SMarkus Armbruster         }
3398f9092b10SMarkus Armbruster     }
3399f9092b10SMarkus Armbruster     return NULL;
3400f9092b10SMarkus Armbruster }
3401f9092b10SMarkus Armbruster 
3402faea38e7Sbellard int bdrv_snapshot_create(BlockDriverState *bs,
3403faea38e7Sbellard                          QEMUSnapshotInfo *sn_info)
3404faea38e7Sbellard {
3405faea38e7Sbellard     BlockDriver *drv = bs->drv;
3406faea38e7Sbellard     if (!drv)
340719cb3738Sbellard         return -ENOMEDIUM;
34087cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_create)
3409faea38e7Sbellard         return drv->bdrv_snapshot_create(bs, sn_info);
34107cdb1f6dSMORITA Kazutaka     if (bs->file)
34117cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_create(bs->file, sn_info);
34127cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3413faea38e7Sbellard }
3414faea38e7Sbellard 
3415faea38e7Sbellard int bdrv_snapshot_goto(BlockDriverState *bs,
3416faea38e7Sbellard                        const char *snapshot_id)
3417faea38e7Sbellard {
3418faea38e7Sbellard     BlockDriver *drv = bs->drv;
34197cdb1f6dSMORITA Kazutaka     int ret, open_ret;
34207cdb1f6dSMORITA Kazutaka 
3421faea38e7Sbellard     if (!drv)
342219cb3738Sbellard         return -ENOMEDIUM;
34237cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_goto)
3424faea38e7Sbellard         return drv->bdrv_snapshot_goto(bs, snapshot_id);
34257cdb1f6dSMORITA Kazutaka 
34267cdb1f6dSMORITA Kazutaka     if (bs->file) {
34277cdb1f6dSMORITA Kazutaka         drv->bdrv_close(bs);
34287cdb1f6dSMORITA Kazutaka         ret = bdrv_snapshot_goto(bs->file, snapshot_id);
34291a86938fSKevin Wolf         open_ret = drv->bdrv_open(bs, NULL, bs->open_flags);
34307cdb1f6dSMORITA Kazutaka         if (open_ret < 0) {
34317cdb1f6dSMORITA Kazutaka             bdrv_delete(bs->file);
34327cdb1f6dSMORITA Kazutaka             bs->drv = NULL;
34337cdb1f6dSMORITA Kazutaka             return open_ret;
34347cdb1f6dSMORITA Kazutaka         }
34357cdb1f6dSMORITA Kazutaka         return ret;
34367cdb1f6dSMORITA Kazutaka     }
34377cdb1f6dSMORITA Kazutaka 
34387cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3439faea38e7Sbellard }
3440faea38e7Sbellard 
3441faea38e7Sbellard int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
3442faea38e7Sbellard {
3443faea38e7Sbellard     BlockDriver *drv = bs->drv;
3444faea38e7Sbellard     if (!drv)
344519cb3738Sbellard         return -ENOMEDIUM;
34467cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_delete)
3447faea38e7Sbellard         return drv->bdrv_snapshot_delete(bs, snapshot_id);
34487cdb1f6dSMORITA Kazutaka     if (bs->file)
34497cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_delete(bs->file, snapshot_id);
34507cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3451faea38e7Sbellard }
3452faea38e7Sbellard 
3453faea38e7Sbellard int bdrv_snapshot_list(BlockDriverState *bs,
3454faea38e7Sbellard                        QEMUSnapshotInfo **psn_info)
3455faea38e7Sbellard {
3456faea38e7Sbellard     BlockDriver *drv = bs->drv;
3457faea38e7Sbellard     if (!drv)
345819cb3738Sbellard         return -ENOMEDIUM;
34597cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_list)
3460faea38e7Sbellard         return drv->bdrv_snapshot_list(bs, psn_info);
34617cdb1f6dSMORITA Kazutaka     if (bs->file)
34627cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_list(bs->file, psn_info);
34637cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3464faea38e7Sbellard }
3465faea38e7Sbellard 
346651ef6727Sedison int bdrv_snapshot_load_tmp(BlockDriverState *bs,
346751ef6727Sedison         const char *snapshot_name)
346851ef6727Sedison {
346951ef6727Sedison     BlockDriver *drv = bs->drv;
347051ef6727Sedison     if (!drv) {
347151ef6727Sedison         return -ENOMEDIUM;
347251ef6727Sedison     }
347351ef6727Sedison     if (!bs->read_only) {
347451ef6727Sedison         return -EINVAL;
347551ef6727Sedison     }
347651ef6727Sedison     if (drv->bdrv_snapshot_load_tmp) {
347751ef6727Sedison         return drv->bdrv_snapshot_load_tmp(bs, snapshot_name);
347851ef6727Sedison     }
347951ef6727Sedison     return -ENOTSUP;
348051ef6727Sedison }
348151ef6727Sedison 
3482b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3483b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3484b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3485b1b1d783SJeff Cody  * the CWD rather than the chain. */
3486e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3487e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3488e8a6bb9cSMarcelo Tosatti {
3489b1b1d783SJeff Cody     char *filename_full = NULL;
3490b1b1d783SJeff Cody     char *backing_file_full = NULL;
3491b1b1d783SJeff Cody     char *filename_tmp = NULL;
3492b1b1d783SJeff Cody     int is_protocol = 0;
3493b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3494b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3495b1b1d783SJeff Cody 
3496b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3497e8a6bb9cSMarcelo Tosatti         return NULL;
3498e8a6bb9cSMarcelo Tosatti     }
3499e8a6bb9cSMarcelo Tosatti 
3500b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
3501b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
3502b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
3503b1b1d783SJeff Cody 
3504b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
3505b1b1d783SJeff Cody 
3506b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
3507b1b1d783SJeff Cody 
3508b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
3509b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
3510b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
3511b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
3512b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3513b1b1d783SJeff Cody                 break;
3514b1b1d783SJeff Cody             }
3515e8a6bb9cSMarcelo Tosatti         } else {
3516b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
3517b1b1d783SJeff Cody              * image's filename path */
3518b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3519b1b1d783SJeff Cody                          backing_file);
3520b1b1d783SJeff Cody 
3521b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
3522b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
3523b1b1d783SJeff Cody                 continue;
3524b1b1d783SJeff Cody             }
3525b1b1d783SJeff Cody 
3526b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
3527b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
3528b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
3529b1b1d783SJeff Cody                          curr_bs->backing_file);
3530b1b1d783SJeff Cody 
3531b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
3532b1b1d783SJeff Cody                 continue;
3533b1b1d783SJeff Cody             }
3534b1b1d783SJeff Cody 
3535b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
3536b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
3537b1b1d783SJeff Cody                 break;
3538b1b1d783SJeff Cody             }
3539e8a6bb9cSMarcelo Tosatti         }
3540e8a6bb9cSMarcelo Tosatti     }
3541e8a6bb9cSMarcelo Tosatti 
3542b1b1d783SJeff Cody     g_free(filename_full);
3543b1b1d783SJeff Cody     g_free(backing_file_full);
3544b1b1d783SJeff Cody     g_free(filename_tmp);
3545b1b1d783SJeff Cody     return retval;
3546e8a6bb9cSMarcelo Tosatti }
3547e8a6bb9cSMarcelo Tosatti 
3548f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
3549f198fd1cSBenoît Canet {
3550f198fd1cSBenoît Canet     if (!bs->drv) {
3551f198fd1cSBenoît Canet         return 0;
3552f198fd1cSBenoît Canet     }
3553f198fd1cSBenoît Canet 
3554f198fd1cSBenoît Canet     if (!bs->backing_hd) {
3555f198fd1cSBenoît Canet         return 0;
3556f198fd1cSBenoît Canet     }
3557f198fd1cSBenoît Canet 
3558f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
3559f198fd1cSBenoît Canet }
3560f198fd1cSBenoît Canet 
356179fac568SJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
356279fac568SJeff Cody {
356379fac568SJeff Cody     BlockDriverState *curr_bs = NULL;
356479fac568SJeff Cody 
356579fac568SJeff Cody     if (!bs) {
356679fac568SJeff Cody         return NULL;
356779fac568SJeff Cody     }
356879fac568SJeff Cody 
356979fac568SJeff Cody     curr_bs = bs;
357079fac568SJeff Cody 
357179fac568SJeff Cody     while (curr_bs->backing_hd) {
357279fac568SJeff Cody         curr_bs = curr_bs->backing_hd;
357379fac568SJeff Cody     }
357479fac568SJeff Cody     return curr_bs;
357579fac568SJeff Cody }
357679fac568SJeff Cody 
3577faea38e7Sbellard #define NB_SUFFIXES 4
3578faea38e7Sbellard 
3579faea38e7Sbellard char *get_human_readable_size(char *buf, int buf_size, int64_t size)
3580faea38e7Sbellard {
3581faea38e7Sbellard     static const char suffixes[NB_SUFFIXES] = "KMGT";
3582faea38e7Sbellard     int64_t base;
3583faea38e7Sbellard     int i;
3584faea38e7Sbellard 
3585faea38e7Sbellard     if (size <= 999) {
3586faea38e7Sbellard         snprintf(buf, buf_size, "%" PRId64, size);
3587faea38e7Sbellard     } else {
3588faea38e7Sbellard         base = 1024;
3589faea38e7Sbellard         for(i = 0; i < NB_SUFFIXES; i++) {
3590faea38e7Sbellard             if (size < (10 * base)) {
3591faea38e7Sbellard                 snprintf(buf, buf_size, "%0.1f%c",
3592faea38e7Sbellard                          (double)size / base,
3593faea38e7Sbellard                          suffixes[i]);
3594faea38e7Sbellard                 break;
3595faea38e7Sbellard             } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
3596faea38e7Sbellard                 snprintf(buf, buf_size, "%" PRId64 "%c",
3597faea38e7Sbellard                          ((size + (base >> 1)) / base),
3598faea38e7Sbellard                          suffixes[i]);
3599faea38e7Sbellard                 break;
3600faea38e7Sbellard             }
3601faea38e7Sbellard             base = base * 1024;
3602faea38e7Sbellard         }
3603faea38e7Sbellard     }
3604faea38e7Sbellard     return buf;
3605faea38e7Sbellard }
3606faea38e7Sbellard 
3607faea38e7Sbellard char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
3608faea38e7Sbellard {
3609faea38e7Sbellard     char buf1[128], date_buf[128], clock_buf[128];
3610faea38e7Sbellard     struct tm tm;
3611faea38e7Sbellard     time_t ti;
3612faea38e7Sbellard     int64_t secs;
3613faea38e7Sbellard 
3614faea38e7Sbellard     if (!sn) {
3615faea38e7Sbellard         snprintf(buf, buf_size,
3616faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
3617faea38e7Sbellard                  "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
3618faea38e7Sbellard     } else {
3619faea38e7Sbellard         ti = sn->date_sec;
3620faea38e7Sbellard         localtime_r(&ti, &tm);
3621faea38e7Sbellard         strftime(date_buf, sizeof(date_buf),
3622faea38e7Sbellard                  "%Y-%m-%d %H:%M:%S", &tm);
3623faea38e7Sbellard         secs = sn->vm_clock_nsec / 1000000000;
3624faea38e7Sbellard         snprintf(clock_buf, sizeof(clock_buf),
3625faea38e7Sbellard                  "%02d:%02d:%02d.%03d",
3626faea38e7Sbellard                  (int)(secs / 3600),
3627faea38e7Sbellard                  (int)((secs / 60) % 60),
3628faea38e7Sbellard                  (int)(secs % 60),
3629faea38e7Sbellard                  (int)((sn->vm_clock_nsec / 1000000) % 1000));
3630faea38e7Sbellard         snprintf(buf, buf_size,
3631faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
3632faea38e7Sbellard                  sn->id_str, sn->name,
3633faea38e7Sbellard                  get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
3634faea38e7Sbellard                  date_buf,
3635faea38e7Sbellard                  clock_buf);
3636faea38e7Sbellard     }
3637faea38e7Sbellard     return buf;
3638faea38e7Sbellard }
3639faea38e7Sbellard 
3640ea2384d3Sbellard /**************************************************************/
364183f64091Sbellard /* async I/Os */
3642ea2384d3Sbellard 
36433b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
3644f141eafeSaliguori                                  QEMUIOVector *qiov, int nb_sectors,
364583f64091Sbellard                                  BlockDriverCompletionFunc *cb, void *opaque)
3646ea2384d3Sbellard {
3647bbf0a440SStefan Hajnoczi     trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
3648bbf0a440SStefan Hajnoczi 
3649b2a61371SStefan Hajnoczi     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
36508c5873d6SStefan Hajnoczi                                  cb, opaque, false);
365183f64091Sbellard }
365283f64091Sbellard 
3653f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
3654f141eafeSaliguori                                   QEMUIOVector *qiov, int nb_sectors,
365583f64091Sbellard                                   BlockDriverCompletionFunc *cb, void *opaque)
36567674e7bfSbellard {
3657bbf0a440SStefan Hajnoczi     trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
3658bbf0a440SStefan Hajnoczi 
36591a6e115bSStefan Hajnoczi     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors,
36608c5873d6SStefan Hajnoczi                                  cb, opaque, true);
366183f64091Sbellard }
366283f64091Sbellard 
366340b4f539SKevin Wolf 
366440b4f539SKevin Wolf typedef struct MultiwriteCB {
366540b4f539SKevin Wolf     int error;
366640b4f539SKevin Wolf     int num_requests;
366740b4f539SKevin Wolf     int num_callbacks;
366840b4f539SKevin Wolf     struct {
366940b4f539SKevin Wolf         BlockDriverCompletionFunc *cb;
367040b4f539SKevin Wolf         void *opaque;
367140b4f539SKevin Wolf         QEMUIOVector *free_qiov;
367240b4f539SKevin Wolf     } callbacks[];
367340b4f539SKevin Wolf } MultiwriteCB;
367440b4f539SKevin Wolf 
367540b4f539SKevin Wolf static void multiwrite_user_cb(MultiwriteCB *mcb)
367640b4f539SKevin Wolf {
367740b4f539SKevin Wolf     int i;
367840b4f539SKevin Wolf 
367940b4f539SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
368040b4f539SKevin Wolf         mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
36811e1ea48dSStefan Hajnoczi         if (mcb->callbacks[i].free_qiov) {
36821e1ea48dSStefan Hajnoczi             qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
36831e1ea48dSStefan Hajnoczi         }
36847267c094SAnthony Liguori         g_free(mcb->callbacks[i].free_qiov);
368540b4f539SKevin Wolf     }
368640b4f539SKevin Wolf }
368740b4f539SKevin Wolf 
368840b4f539SKevin Wolf static void multiwrite_cb(void *opaque, int ret)
368940b4f539SKevin Wolf {
369040b4f539SKevin Wolf     MultiwriteCB *mcb = opaque;
369140b4f539SKevin Wolf 
36926d519a5fSStefan Hajnoczi     trace_multiwrite_cb(mcb, ret);
36936d519a5fSStefan Hajnoczi 
3694cb6d3ca0SKevin Wolf     if (ret < 0 && !mcb->error) {
369540b4f539SKevin Wolf         mcb->error = ret;
369640b4f539SKevin Wolf     }
369740b4f539SKevin Wolf 
369840b4f539SKevin Wolf     mcb->num_requests--;
369940b4f539SKevin Wolf     if (mcb->num_requests == 0) {
370040b4f539SKevin Wolf         multiwrite_user_cb(mcb);
37017267c094SAnthony Liguori         g_free(mcb);
370240b4f539SKevin Wolf     }
370340b4f539SKevin Wolf }
370440b4f539SKevin Wolf 
370540b4f539SKevin Wolf static int multiwrite_req_compare(const void *a, const void *b)
370640b4f539SKevin Wolf {
370777be4366SChristoph Hellwig     const BlockRequest *req1 = a, *req2 = b;
370877be4366SChristoph Hellwig 
370977be4366SChristoph Hellwig     /*
371077be4366SChristoph Hellwig      * Note that we can't simply subtract req2->sector from req1->sector
371177be4366SChristoph Hellwig      * here as that could overflow the return value.
371277be4366SChristoph Hellwig      */
371377be4366SChristoph Hellwig     if (req1->sector > req2->sector) {
371477be4366SChristoph Hellwig         return 1;
371577be4366SChristoph Hellwig     } else if (req1->sector < req2->sector) {
371677be4366SChristoph Hellwig         return -1;
371777be4366SChristoph Hellwig     } else {
371877be4366SChristoph Hellwig         return 0;
371977be4366SChristoph Hellwig     }
372040b4f539SKevin Wolf }
372140b4f539SKevin Wolf 
372240b4f539SKevin Wolf /*
372340b4f539SKevin Wolf  * Takes a bunch of requests and tries to merge them. Returns the number of
372440b4f539SKevin Wolf  * requests that remain after merging.
372540b4f539SKevin Wolf  */
372640b4f539SKevin Wolf static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
372740b4f539SKevin Wolf     int num_reqs, MultiwriteCB *mcb)
372840b4f539SKevin Wolf {
372940b4f539SKevin Wolf     int i, outidx;
373040b4f539SKevin Wolf 
373140b4f539SKevin Wolf     // Sort requests by start sector
373240b4f539SKevin Wolf     qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
373340b4f539SKevin Wolf 
373440b4f539SKevin Wolf     // Check if adjacent requests touch the same clusters. If so, combine them,
373540b4f539SKevin Wolf     // filling up gaps with zero sectors.
373640b4f539SKevin Wolf     outidx = 0;
373740b4f539SKevin Wolf     for (i = 1; i < num_reqs; i++) {
373840b4f539SKevin Wolf         int merge = 0;
373940b4f539SKevin Wolf         int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
374040b4f539SKevin Wolf 
3741b6a127a1SPaolo Bonzini         // Handle exactly sequential writes and overlapping writes.
374240b4f539SKevin Wolf         if (reqs[i].sector <= oldreq_last) {
374340b4f539SKevin Wolf             merge = 1;
374440b4f539SKevin Wolf         }
374540b4f539SKevin Wolf 
3746e2a305fbSChristoph Hellwig         if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
3747e2a305fbSChristoph Hellwig             merge = 0;
3748e2a305fbSChristoph Hellwig         }
3749e2a305fbSChristoph Hellwig 
375040b4f539SKevin Wolf         if (merge) {
375140b4f539SKevin Wolf             size_t size;
37527267c094SAnthony Liguori             QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
375340b4f539SKevin Wolf             qemu_iovec_init(qiov,
375440b4f539SKevin Wolf                 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
375540b4f539SKevin Wolf 
375640b4f539SKevin Wolf             // Add the first request to the merged one. If the requests are
375740b4f539SKevin Wolf             // overlapping, drop the last sectors of the first request.
375840b4f539SKevin Wolf             size = (reqs[i].sector - reqs[outidx].sector) << 9;
37591b093c48SMichael Tokarev             qemu_iovec_concat(qiov, reqs[outidx].qiov, 0, size);
376040b4f539SKevin Wolf 
3761b6a127a1SPaolo Bonzini             // We should need to add any zeros between the two requests
3762b6a127a1SPaolo Bonzini             assert (reqs[i].sector <= oldreq_last);
376340b4f539SKevin Wolf 
376440b4f539SKevin Wolf             // Add the second request
37651b093c48SMichael Tokarev             qemu_iovec_concat(qiov, reqs[i].qiov, 0, reqs[i].qiov->size);
376640b4f539SKevin Wolf 
3767cbf1dff2SKevin Wolf             reqs[outidx].nb_sectors = qiov->size >> 9;
376840b4f539SKevin Wolf             reqs[outidx].qiov = qiov;
376940b4f539SKevin Wolf 
377040b4f539SKevin Wolf             mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
377140b4f539SKevin Wolf         } else {
377240b4f539SKevin Wolf             outidx++;
377340b4f539SKevin Wolf             reqs[outidx].sector     = reqs[i].sector;
377440b4f539SKevin Wolf             reqs[outidx].nb_sectors = reqs[i].nb_sectors;
377540b4f539SKevin Wolf             reqs[outidx].qiov       = reqs[i].qiov;
377640b4f539SKevin Wolf         }
377740b4f539SKevin Wolf     }
377840b4f539SKevin Wolf 
377940b4f539SKevin Wolf     return outidx + 1;
378040b4f539SKevin Wolf }
378140b4f539SKevin Wolf 
378240b4f539SKevin Wolf /*
378340b4f539SKevin Wolf  * Submit multiple AIO write requests at once.
378440b4f539SKevin Wolf  *
378540b4f539SKevin Wolf  * On success, the function returns 0 and all requests in the reqs array have
378640b4f539SKevin Wolf  * been submitted. In error case this function returns -1, and any of the
378740b4f539SKevin Wolf  * requests may or may not be submitted yet. In particular, this means that the
378840b4f539SKevin Wolf  * callback will be called for some of the requests, for others it won't. The
378940b4f539SKevin Wolf  * caller must check the error field of the BlockRequest to wait for the right
379040b4f539SKevin Wolf  * callbacks (if error != 0, no callback will be called).
379140b4f539SKevin Wolf  *
379240b4f539SKevin Wolf  * The implementation may modify the contents of the reqs array, e.g. to merge
379340b4f539SKevin Wolf  * requests. However, the fields opaque and error are left unmodified as they
379440b4f539SKevin Wolf  * are used to signal failure for a single request to the caller.
379540b4f539SKevin Wolf  */
379640b4f539SKevin Wolf int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
379740b4f539SKevin Wolf {
379840b4f539SKevin Wolf     MultiwriteCB *mcb;
379940b4f539SKevin Wolf     int i;
380040b4f539SKevin Wolf 
3801301db7c2SRyan Harper     /* don't submit writes if we don't have a medium */
3802301db7c2SRyan Harper     if (bs->drv == NULL) {
3803301db7c2SRyan Harper         for (i = 0; i < num_reqs; i++) {
3804301db7c2SRyan Harper             reqs[i].error = -ENOMEDIUM;
3805301db7c2SRyan Harper         }
3806301db7c2SRyan Harper         return -1;
3807301db7c2SRyan Harper     }
3808301db7c2SRyan Harper 
380940b4f539SKevin Wolf     if (num_reqs == 0) {
381040b4f539SKevin Wolf         return 0;
381140b4f539SKevin Wolf     }
381240b4f539SKevin Wolf 
381340b4f539SKevin Wolf     // Create MultiwriteCB structure
38147267c094SAnthony Liguori     mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
381540b4f539SKevin Wolf     mcb->num_requests = 0;
381640b4f539SKevin Wolf     mcb->num_callbacks = num_reqs;
381740b4f539SKevin Wolf 
381840b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
381940b4f539SKevin Wolf         mcb->callbacks[i].cb = reqs[i].cb;
382040b4f539SKevin Wolf         mcb->callbacks[i].opaque = reqs[i].opaque;
382140b4f539SKevin Wolf     }
382240b4f539SKevin Wolf 
382340b4f539SKevin Wolf     // Check for mergable requests
382440b4f539SKevin Wolf     num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
382540b4f539SKevin Wolf 
38266d519a5fSStefan Hajnoczi     trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
38276d519a5fSStefan Hajnoczi 
3828df9309fbSPaolo Bonzini     /* Run the aio requests. */
3829df9309fbSPaolo Bonzini     mcb->num_requests = num_reqs;
383040b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
3831ad54ae80SPaolo Bonzini         bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
383240b4f539SKevin Wolf             reqs[i].nb_sectors, multiwrite_cb, mcb);
383340b4f539SKevin Wolf     }
383440b4f539SKevin Wolf 
383540b4f539SKevin Wolf     return 0;
383640b4f539SKevin Wolf }
383740b4f539SKevin Wolf 
383883f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
383983f64091Sbellard {
3840d7331bedSStefan Hajnoczi     acb->aiocb_info->cancel(acb);
384183f64091Sbellard }
384283f64091Sbellard 
384398f90dbaSZhi Yong Wu /* block I/O throttling */
384498f90dbaSZhi Yong Wu static bool bdrv_exceed_bps_limits(BlockDriverState *bs, int nb_sectors,
384598f90dbaSZhi Yong Wu                  bool is_write, double elapsed_time, uint64_t *wait)
384698f90dbaSZhi Yong Wu {
384798f90dbaSZhi Yong Wu     uint64_t bps_limit = 0;
3848ae29d6c6SStefan Hajnoczi     uint64_t extension;
384998f90dbaSZhi Yong Wu     double   bytes_limit, bytes_base, bytes_res;
385098f90dbaSZhi Yong Wu     double   slice_time, wait_time;
385198f90dbaSZhi Yong Wu 
385298f90dbaSZhi Yong Wu     if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
385398f90dbaSZhi Yong Wu         bps_limit = bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL];
385498f90dbaSZhi Yong Wu     } else if (bs->io_limits.bps[is_write]) {
385598f90dbaSZhi Yong Wu         bps_limit = bs->io_limits.bps[is_write];
385698f90dbaSZhi Yong Wu     } else {
385798f90dbaSZhi Yong Wu         if (wait) {
385898f90dbaSZhi Yong Wu             *wait = 0;
385998f90dbaSZhi Yong Wu         }
386098f90dbaSZhi Yong Wu 
386198f90dbaSZhi Yong Wu         return false;
386298f90dbaSZhi Yong Wu     }
386398f90dbaSZhi Yong Wu 
386498f90dbaSZhi Yong Wu     slice_time = bs->slice_end - bs->slice_start;
386598f90dbaSZhi Yong Wu     slice_time /= (NANOSECONDS_PER_SECOND);
386698f90dbaSZhi Yong Wu     bytes_limit = bps_limit * slice_time;
38675905fbc9SStefan Hajnoczi     bytes_base  = bs->slice_submitted.bytes[is_write];
386898f90dbaSZhi Yong Wu     if (bs->io_limits.bps[BLOCK_IO_LIMIT_TOTAL]) {
38695905fbc9SStefan Hajnoczi         bytes_base += bs->slice_submitted.bytes[!is_write];
387098f90dbaSZhi Yong Wu     }
387198f90dbaSZhi Yong Wu 
387298f90dbaSZhi Yong Wu     /* bytes_base: the bytes of data which have been read/written; and
387398f90dbaSZhi Yong Wu      *             it is obtained from the history statistic info.
387498f90dbaSZhi Yong Wu      * bytes_res: the remaining bytes of data which need to be read/written.
387598f90dbaSZhi Yong Wu      * (bytes_base + bytes_res) / bps_limit: used to calcuate
387698f90dbaSZhi Yong Wu      *             the total time for completing reading/writting all data.
387798f90dbaSZhi Yong Wu      */
387898f90dbaSZhi Yong Wu     bytes_res   = (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
387998f90dbaSZhi Yong Wu 
388098f90dbaSZhi Yong Wu     if (bytes_base + bytes_res <= bytes_limit) {
388198f90dbaSZhi Yong Wu         if (wait) {
388298f90dbaSZhi Yong Wu             *wait = 0;
388398f90dbaSZhi Yong Wu         }
388498f90dbaSZhi Yong Wu 
388598f90dbaSZhi Yong Wu         return false;
388698f90dbaSZhi Yong Wu     }
388798f90dbaSZhi Yong Wu 
388898f90dbaSZhi Yong Wu     /* Calc approx time to dispatch */
388998f90dbaSZhi Yong Wu     wait_time = (bytes_base + bytes_res) / bps_limit - elapsed_time;
389098f90dbaSZhi Yong Wu 
389198f90dbaSZhi Yong Wu     /* When the I/O rate at runtime exceeds the limits,
389298f90dbaSZhi Yong Wu      * bs->slice_end need to be extended in order that the current statistic
389398f90dbaSZhi Yong Wu      * info can be kept until the timer fire, so it is increased and tuned
389498f90dbaSZhi Yong Wu      * based on the result of experiment.
389598f90dbaSZhi Yong Wu      */
3896ae29d6c6SStefan Hajnoczi     extension = wait_time * NANOSECONDS_PER_SECOND;
3897ae29d6c6SStefan Hajnoczi     extension = DIV_ROUND_UP(extension, BLOCK_IO_SLICE_TIME) *
3898ae29d6c6SStefan Hajnoczi                 BLOCK_IO_SLICE_TIME;
3899ae29d6c6SStefan Hajnoczi     bs->slice_end += extension;
390098f90dbaSZhi Yong Wu     if (wait) {
39010775437fSStefan Hajnoczi         *wait = wait_time * NANOSECONDS_PER_SECOND;
390298f90dbaSZhi Yong Wu     }
390398f90dbaSZhi Yong Wu 
390498f90dbaSZhi Yong Wu     return true;
390598f90dbaSZhi Yong Wu }
390698f90dbaSZhi Yong Wu 
390798f90dbaSZhi Yong Wu static bool bdrv_exceed_iops_limits(BlockDriverState *bs, bool is_write,
390898f90dbaSZhi Yong Wu                              double elapsed_time, uint64_t *wait)
390998f90dbaSZhi Yong Wu {
391098f90dbaSZhi Yong Wu     uint64_t iops_limit = 0;
391198f90dbaSZhi Yong Wu     double   ios_limit, ios_base;
391298f90dbaSZhi Yong Wu     double   slice_time, wait_time;
391398f90dbaSZhi Yong Wu 
391498f90dbaSZhi Yong Wu     if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
391598f90dbaSZhi Yong Wu         iops_limit = bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL];
391698f90dbaSZhi Yong Wu     } else if (bs->io_limits.iops[is_write]) {
391798f90dbaSZhi Yong Wu         iops_limit = bs->io_limits.iops[is_write];
391898f90dbaSZhi Yong Wu     } else {
391998f90dbaSZhi Yong Wu         if (wait) {
392098f90dbaSZhi Yong Wu             *wait = 0;
392198f90dbaSZhi Yong Wu         }
392298f90dbaSZhi Yong Wu 
392398f90dbaSZhi Yong Wu         return false;
392498f90dbaSZhi Yong Wu     }
392598f90dbaSZhi Yong Wu 
392698f90dbaSZhi Yong Wu     slice_time = bs->slice_end - bs->slice_start;
392798f90dbaSZhi Yong Wu     slice_time /= (NANOSECONDS_PER_SECOND);
392898f90dbaSZhi Yong Wu     ios_limit  = iops_limit * slice_time;
39295905fbc9SStefan Hajnoczi     ios_base   = bs->slice_submitted.ios[is_write];
393098f90dbaSZhi Yong Wu     if (bs->io_limits.iops[BLOCK_IO_LIMIT_TOTAL]) {
39315905fbc9SStefan Hajnoczi         ios_base += bs->slice_submitted.ios[!is_write];
393298f90dbaSZhi Yong Wu     }
393398f90dbaSZhi Yong Wu 
393498f90dbaSZhi Yong Wu     if (ios_base + 1 <= ios_limit) {
393598f90dbaSZhi Yong Wu         if (wait) {
393698f90dbaSZhi Yong Wu             *wait = 0;
393798f90dbaSZhi Yong Wu         }
393898f90dbaSZhi Yong Wu 
393998f90dbaSZhi Yong Wu         return false;
394098f90dbaSZhi Yong Wu     }
394198f90dbaSZhi Yong Wu 
39420775437fSStefan Hajnoczi     /* Calc approx time to dispatch, in seconds */
394398f90dbaSZhi Yong Wu     wait_time = (ios_base + 1) / iops_limit;
394498f90dbaSZhi Yong Wu     if (wait_time > elapsed_time) {
394598f90dbaSZhi Yong Wu         wait_time = wait_time - elapsed_time;
394698f90dbaSZhi Yong Wu     } else {
394798f90dbaSZhi Yong Wu         wait_time = 0;
394898f90dbaSZhi Yong Wu     }
394998f90dbaSZhi Yong Wu 
3950ae29d6c6SStefan Hajnoczi     /* Exceeded current slice, extend it by another slice time */
3951ae29d6c6SStefan Hajnoczi     bs->slice_end += BLOCK_IO_SLICE_TIME;
395298f90dbaSZhi Yong Wu     if (wait) {
39530775437fSStefan Hajnoczi         *wait = wait_time * NANOSECONDS_PER_SECOND;
395498f90dbaSZhi Yong Wu     }
395598f90dbaSZhi Yong Wu 
395698f90dbaSZhi Yong Wu     return true;
395798f90dbaSZhi Yong Wu }
395898f90dbaSZhi Yong Wu 
395998f90dbaSZhi Yong Wu static bool bdrv_exceed_io_limits(BlockDriverState *bs, int nb_sectors,
396098f90dbaSZhi Yong Wu                            bool is_write, int64_t *wait)
396198f90dbaSZhi Yong Wu {
396298f90dbaSZhi Yong Wu     int64_t  now, max_wait;
396398f90dbaSZhi Yong Wu     uint64_t bps_wait = 0, iops_wait = 0;
396498f90dbaSZhi Yong Wu     double   elapsed_time;
396598f90dbaSZhi Yong Wu     int      bps_ret, iops_ret;
396698f90dbaSZhi Yong Wu 
396798f90dbaSZhi Yong Wu     now = qemu_get_clock_ns(vm_clock);
3968e660fb8bSStefan Hajnoczi     if (now > bs->slice_end) {
396998f90dbaSZhi Yong Wu         bs->slice_start = now;
3970ae29d6c6SStefan Hajnoczi         bs->slice_end   = now + BLOCK_IO_SLICE_TIME;
39715905fbc9SStefan Hajnoczi         memset(&bs->slice_submitted, 0, sizeof(bs->slice_submitted));
397298f90dbaSZhi Yong Wu     }
397398f90dbaSZhi Yong Wu 
397498f90dbaSZhi Yong Wu     elapsed_time  = now - bs->slice_start;
397598f90dbaSZhi Yong Wu     elapsed_time  /= (NANOSECONDS_PER_SECOND);
397698f90dbaSZhi Yong Wu 
397798f90dbaSZhi Yong Wu     bps_ret  = bdrv_exceed_bps_limits(bs, nb_sectors,
397898f90dbaSZhi Yong Wu                                       is_write, elapsed_time, &bps_wait);
397998f90dbaSZhi Yong Wu     iops_ret = bdrv_exceed_iops_limits(bs, is_write,
398098f90dbaSZhi Yong Wu                                       elapsed_time, &iops_wait);
398198f90dbaSZhi Yong Wu     if (bps_ret || iops_ret) {
398298f90dbaSZhi Yong Wu         max_wait = bps_wait > iops_wait ? bps_wait : iops_wait;
398398f90dbaSZhi Yong Wu         if (wait) {
398498f90dbaSZhi Yong Wu             *wait = max_wait;
398598f90dbaSZhi Yong Wu         }
398698f90dbaSZhi Yong Wu 
398798f90dbaSZhi Yong Wu         now = qemu_get_clock_ns(vm_clock);
398898f90dbaSZhi Yong Wu         if (bs->slice_end < now + max_wait) {
398998f90dbaSZhi Yong Wu             bs->slice_end = now + max_wait;
399098f90dbaSZhi Yong Wu         }
399198f90dbaSZhi Yong Wu 
399298f90dbaSZhi Yong Wu         return true;
399398f90dbaSZhi Yong Wu     }
399498f90dbaSZhi Yong Wu 
399598f90dbaSZhi Yong Wu     if (wait) {
399698f90dbaSZhi Yong Wu         *wait = 0;
399798f90dbaSZhi Yong Wu     }
399898f90dbaSZhi Yong Wu 
39995905fbc9SStefan Hajnoczi     bs->slice_submitted.bytes[is_write] += (int64_t)nb_sectors *
40005905fbc9SStefan Hajnoczi                                            BDRV_SECTOR_SIZE;
40015905fbc9SStefan Hajnoczi     bs->slice_submitted.ios[is_write]++;
40025905fbc9SStefan Hajnoczi 
400398f90dbaSZhi Yong Wu     return false;
400498f90dbaSZhi Yong Wu }
400583f64091Sbellard 
400683f64091Sbellard /**************************************************************/
400783f64091Sbellard /* async block device emulation */
400883f64091Sbellard 
4009c16b5a2cSChristoph Hellwig typedef struct BlockDriverAIOCBSync {
4010c16b5a2cSChristoph Hellwig     BlockDriverAIOCB common;
4011c16b5a2cSChristoph Hellwig     QEMUBH *bh;
4012c16b5a2cSChristoph Hellwig     int ret;
4013c16b5a2cSChristoph Hellwig     /* vector translation state */
4014c16b5a2cSChristoph Hellwig     QEMUIOVector *qiov;
4015c16b5a2cSChristoph Hellwig     uint8_t *bounce;
4016c16b5a2cSChristoph Hellwig     int is_write;
4017c16b5a2cSChristoph Hellwig } BlockDriverAIOCBSync;
4018c16b5a2cSChristoph Hellwig 
4019c16b5a2cSChristoph Hellwig static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
4020c16b5a2cSChristoph Hellwig {
4021b666d239SKevin Wolf     BlockDriverAIOCBSync *acb =
4022b666d239SKevin Wolf         container_of(blockacb, BlockDriverAIOCBSync, common);
40236a7ad299SDor Laor     qemu_bh_delete(acb->bh);
402436afc451SAvi Kivity     acb->bh = NULL;
4025c16b5a2cSChristoph Hellwig     qemu_aio_release(acb);
4026c16b5a2cSChristoph Hellwig }
4027c16b5a2cSChristoph Hellwig 
4028d7331bedSStefan Hajnoczi static const AIOCBInfo bdrv_em_aiocb_info = {
4029c16b5a2cSChristoph Hellwig     .aiocb_size         = sizeof(BlockDriverAIOCBSync),
4030c16b5a2cSChristoph Hellwig     .cancel             = bdrv_aio_cancel_em,
4031c16b5a2cSChristoph Hellwig };
4032c16b5a2cSChristoph Hellwig 
403383f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
4034beac80cdSbellard {
4035ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = opaque;
4036f141eafeSaliguori 
4037f141eafeSaliguori     if (!acb->is_write)
403803396148SMichael Tokarev         qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size);
4039ceb42de8Saliguori     qemu_vfree(acb->bounce);
4040ce1a14dcSpbrook     acb->common.cb(acb->common.opaque, acb->ret);
40416a7ad299SDor Laor     qemu_bh_delete(acb->bh);
404236afc451SAvi Kivity     acb->bh = NULL;
4043ce1a14dcSpbrook     qemu_aio_release(acb);
4044beac80cdSbellard }
4045beac80cdSbellard 
4046f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
4047f141eafeSaliguori                                             int64_t sector_num,
4048f141eafeSaliguori                                             QEMUIOVector *qiov,
4049f141eafeSaliguori                                             int nb_sectors,
4050f141eafeSaliguori                                             BlockDriverCompletionFunc *cb,
4051f141eafeSaliguori                                             void *opaque,
4052f141eafeSaliguori                                             int is_write)
4053f141eafeSaliguori 
4054ea2384d3Sbellard {
4055ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
405683f64091Sbellard 
4057d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_aiocb_info, bs, cb, opaque);
4058f141eafeSaliguori     acb->is_write = is_write;
4059f141eafeSaliguori     acb->qiov = qiov;
4060e268ca52Saliguori     acb->bounce = qemu_blockalign(bs, qiov->size);
4061ce1a14dcSpbrook     acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
4062f141eafeSaliguori 
4063f141eafeSaliguori     if (is_write) {
4064d5e6b161SMichael Tokarev         qemu_iovec_to_buf(acb->qiov, 0, acb->bounce, qiov->size);
40651ed20acfSStefan Hajnoczi         acb->ret = bs->drv->bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
4066f141eafeSaliguori     } else {
40671ed20acfSStefan Hajnoczi         acb->ret = bs->drv->bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
4068f141eafeSaliguori     }
4069f141eafeSaliguori 
4070ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
4071f141eafeSaliguori 
4072ce1a14dcSpbrook     return &acb->common;
40737a6cba61Spbrook }
40747a6cba61Spbrook 
4075f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
4076f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
4077ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
407883f64091Sbellard {
4079f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
408083f64091Sbellard }
408183f64091Sbellard 
4082f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
4083f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
4084f141eafeSaliguori         BlockDriverCompletionFunc *cb, void *opaque)
4085f141eafeSaliguori {
4086f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
4087f141eafeSaliguori }
4088f141eafeSaliguori 
408968485420SKevin Wolf 
409068485420SKevin Wolf typedef struct BlockDriverAIOCBCoroutine {
409168485420SKevin Wolf     BlockDriverAIOCB common;
409268485420SKevin Wolf     BlockRequest req;
409368485420SKevin Wolf     bool is_write;
4094d318aea9SKevin Wolf     bool *done;
409568485420SKevin Wolf     QEMUBH* bh;
409668485420SKevin Wolf } BlockDriverAIOCBCoroutine;
409768485420SKevin Wolf 
409868485420SKevin Wolf static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
409968485420SKevin Wolf {
4100d318aea9SKevin Wolf     BlockDriverAIOCBCoroutine *acb =
4101d318aea9SKevin Wolf         container_of(blockacb, BlockDriverAIOCBCoroutine, common);
4102d318aea9SKevin Wolf     bool done = false;
4103d318aea9SKevin Wolf 
4104d318aea9SKevin Wolf     acb->done = &done;
4105d318aea9SKevin Wolf     while (!done) {
4106d318aea9SKevin Wolf         qemu_aio_wait();
4107d318aea9SKevin Wolf     }
410868485420SKevin Wolf }
410968485420SKevin Wolf 
4110d7331bedSStefan Hajnoczi static const AIOCBInfo bdrv_em_co_aiocb_info = {
411168485420SKevin Wolf     .aiocb_size         = sizeof(BlockDriverAIOCBCoroutine),
411268485420SKevin Wolf     .cancel             = bdrv_aio_co_cancel_em,
411368485420SKevin Wolf };
411468485420SKevin Wolf 
411535246a68SPaolo Bonzini static void bdrv_co_em_bh(void *opaque)
411668485420SKevin Wolf {
411768485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb = opaque;
411868485420SKevin Wolf 
411968485420SKevin Wolf     acb->common.cb(acb->common.opaque, acb->req.error);
4120d318aea9SKevin Wolf 
4121d318aea9SKevin Wolf     if (acb->done) {
4122d318aea9SKevin Wolf         *acb->done = true;
4123d318aea9SKevin Wolf     }
4124d318aea9SKevin Wolf 
412568485420SKevin Wolf     qemu_bh_delete(acb->bh);
412668485420SKevin Wolf     qemu_aio_release(acb);
412768485420SKevin Wolf }
412868485420SKevin Wolf 
4129b2a61371SStefan Hajnoczi /* Invoke bdrv_co_do_readv/bdrv_co_do_writev */
4130b2a61371SStefan Hajnoczi static void coroutine_fn bdrv_co_do_rw(void *opaque)
4131b2a61371SStefan Hajnoczi {
4132b2a61371SStefan Hajnoczi     BlockDriverAIOCBCoroutine *acb = opaque;
4133b2a61371SStefan Hajnoczi     BlockDriverState *bs = acb->common.bs;
4134b2a61371SStefan Hajnoczi 
4135b2a61371SStefan Hajnoczi     if (!acb->is_write) {
4136b2a61371SStefan Hajnoczi         acb->req.error = bdrv_co_do_readv(bs, acb->req.sector,
4137470c0504SStefan Hajnoczi             acb->req.nb_sectors, acb->req.qiov, 0);
4138b2a61371SStefan Hajnoczi     } else {
4139b2a61371SStefan Hajnoczi         acb->req.error = bdrv_co_do_writev(bs, acb->req.sector,
4140f08f2ddaSStefan Hajnoczi             acb->req.nb_sectors, acb->req.qiov, 0);
4141b2a61371SStefan Hajnoczi     }
4142b2a61371SStefan Hajnoczi 
414335246a68SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
4144b2a61371SStefan Hajnoczi     qemu_bh_schedule(acb->bh);
4145b2a61371SStefan Hajnoczi }
4146b2a61371SStefan Hajnoczi 
414768485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
414868485420SKevin Wolf                                                int64_t sector_num,
414968485420SKevin Wolf                                                QEMUIOVector *qiov,
415068485420SKevin Wolf                                                int nb_sectors,
415168485420SKevin Wolf                                                BlockDriverCompletionFunc *cb,
415268485420SKevin Wolf                                                void *opaque,
41538c5873d6SStefan Hajnoczi                                                bool is_write)
415468485420SKevin Wolf {
415568485420SKevin Wolf     Coroutine *co;
415668485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb;
415768485420SKevin Wolf 
4158d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
415968485420SKevin Wolf     acb->req.sector = sector_num;
416068485420SKevin Wolf     acb->req.nb_sectors = nb_sectors;
416168485420SKevin Wolf     acb->req.qiov = qiov;
416268485420SKevin Wolf     acb->is_write = is_write;
4163d318aea9SKevin Wolf     acb->done = NULL;
416468485420SKevin Wolf 
41658c5873d6SStefan Hajnoczi     co = qemu_coroutine_create(bdrv_co_do_rw);
416668485420SKevin Wolf     qemu_coroutine_enter(co, acb);
416768485420SKevin Wolf 
416868485420SKevin Wolf     return &acb->common;
416968485420SKevin Wolf }
417068485420SKevin Wolf 
417107f07615SPaolo Bonzini static void coroutine_fn bdrv_aio_flush_co_entry(void *opaque)
4172b2e12bc6SChristoph Hellwig {
417307f07615SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb = opaque;
417407f07615SPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
4175b2e12bc6SChristoph Hellwig 
417607f07615SPaolo Bonzini     acb->req.error = bdrv_co_flush(bs);
417707f07615SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
4178b2e12bc6SChristoph Hellwig     qemu_bh_schedule(acb->bh);
4179b2e12bc6SChristoph Hellwig }
4180b2e12bc6SChristoph Hellwig 
418107f07615SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
4182016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque)
4183016f5cf6SAlexander Graf {
418407f07615SPaolo Bonzini     trace_bdrv_aio_flush(bs, opaque);
4185016f5cf6SAlexander Graf 
418607f07615SPaolo Bonzini     Coroutine *co;
418707f07615SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb;
4188016f5cf6SAlexander Graf 
4189d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
4190d318aea9SKevin Wolf     acb->done = NULL;
4191d318aea9SKevin Wolf 
419207f07615SPaolo Bonzini     co = qemu_coroutine_create(bdrv_aio_flush_co_entry);
419307f07615SPaolo Bonzini     qemu_coroutine_enter(co, acb);
4194016f5cf6SAlexander Graf 
4195016f5cf6SAlexander Graf     return &acb->common;
4196016f5cf6SAlexander Graf }
4197016f5cf6SAlexander Graf 
41984265d620SPaolo Bonzini static void coroutine_fn bdrv_aio_discard_co_entry(void *opaque)
41994265d620SPaolo Bonzini {
42004265d620SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb = opaque;
42014265d620SPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
42024265d620SPaolo Bonzini 
42034265d620SPaolo Bonzini     acb->req.error = bdrv_co_discard(bs, acb->req.sector, acb->req.nb_sectors);
42044265d620SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
42054265d620SPaolo Bonzini     qemu_bh_schedule(acb->bh);
42064265d620SPaolo Bonzini }
42074265d620SPaolo Bonzini 
42084265d620SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_discard(BlockDriverState *bs,
42094265d620SPaolo Bonzini         int64_t sector_num, int nb_sectors,
42104265d620SPaolo Bonzini         BlockDriverCompletionFunc *cb, void *opaque)
42114265d620SPaolo Bonzini {
42124265d620SPaolo Bonzini     Coroutine *co;
42134265d620SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb;
42144265d620SPaolo Bonzini 
42154265d620SPaolo Bonzini     trace_bdrv_aio_discard(bs, sector_num, nb_sectors, opaque);
42164265d620SPaolo Bonzini 
4217d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
42184265d620SPaolo Bonzini     acb->req.sector = sector_num;
42194265d620SPaolo Bonzini     acb->req.nb_sectors = nb_sectors;
4220d318aea9SKevin Wolf     acb->done = NULL;
42214265d620SPaolo Bonzini     co = qemu_coroutine_create(bdrv_aio_discard_co_entry);
42224265d620SPaolo Bonzini     qemu_coroutine_enter(co, acb);
42234265d620SPaolo Bonzini 
42244265d620SPaolo Bonzini     return &acb->common;
42254265d620SPaolo Bonzini }
42264265d620SPaolo Bonzini 
4227ea2384d3Sbellard void bdrv_init(void)
4228ea2384d3Sbellard {
42295efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
4230ea2384d3Sbellard }
4231ce1a14dcSpbrook 
4232eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
4233eb852011SMarkus Armbruster {
4234eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
4235eb852011SMarkus Armbruster     bdrv_init();
4236eb852011SMarkus Armbruster }
4237eb852011SMarkus Armbruster 
4238d7331bedSStefan Hajnoczi void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs,
42396bbff9a0Saliguori                    BlockDriverCompletionFunc *cb, void *opaque)
42406bbff9a0Saliguori {
4241ce1a14dcSpbrook     BlockDriverAIOCB *acb;
4242ce1a14dcSpbrook 
4243d7331bedSStefan Hajnoczi     acb = g_slice_alloc(aiocb_info->aiocb_size);
4244d7331bedSStefan Hajnoczi     acb->aiocb_info = aiocb_info;
4245ce1a14dcSpbrook     acb->bs = bs;
4246ce1a14dcSpbrook     acb->cb = cb;
4247ce1a14dcSpbrook     acb->opaque = opaque;
4248ce1a14dcSpbrook     return acb;
4249ce1a14dcSpbrook }
4250ce1a14dcSpbrook 
4251ce1a14dcSpbrook void qemu_aio_release(void *p)
4252ce1a14dcSpbrook {
4253d37c975fSStefan Hajnoczi     BlockDriverAIOCB *acb = p;
4254d7331bedSStefan Hajnoczi     g_slice_free1(acb->aiocb_info->aiocb_size, acb);
4255ce1a14dcSpbrook }
425619cb3738Sbellard 
425719cb3738Sbellard /**************************************************************/
4258f9f05dc5SKevin Wolf /* Coroutine block device emulation */
4259f9f05dc5SKevin Wolf 
4260f9f05dc5SKevin Wolf typedef struct CoroutineIOCompletion {
4261f9f05dc5SKevin Wolf     Coroutine *coroutine;
4262f9f05dc5SKevin Wolf     int ret;
4263f9f05dc5SKevin Wolf } CoroutineIOCompletion;
4264f9f05dc5SKevin Wolf 
4265f9f05dc5SKevin Wolf static void bdrv_co_io_em_complete(void *opaque, int ret)
4266f9f05dc5SKevin Wolf {
4267f9f05dc5SKevin Wolf     CoroutineIOCompletion *co = opaque;
4268f9f05dc5SKevin Wolf 
4269f9f05dc5SKevin Wolf     co->ret = ret;
4270f9f05dc5SKevin Wolf     qemu_coroutine_enter(co->coroutine, NULL);
4271f9f05dc5SKevin Wolf }
4272f9f05dc5SKevin Wolf 
4273f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
4274f9f05dc5SKevin Wolf                                       int nb_sectors, QEMUIOVector *iov,
4275f9f05dc5SKevin Wolf                                       bool is_write)
4276f9f05dc5SKevin Wolf {
4277f9f05dc5SKevin Wolf     CoroutineIOCompletion co = {
4278f9f05dc5SKevin Wolf         .coroutine = qemu_coroutine_self(),
4279f9f05dc5SKevin Wolf     };
4280f9f05dc5SKevin Wolf     BlockDriverAIOCB *acb;
4281f9f05dc5SKevin Wolf 
4282f9f05dc5SKevin Wolf     if (is_write) {
4283a652d160SStefan Hajnoczi         acb = bs->drv->bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
4284f9f05dc5SKevin Wolf                                        bdrv_co_io_em_complete, &co);
4285f9f05dc5SKevin Wolf     } else {
4286a652d160SStefan Hajnoczi         acb = bs->drv->bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
4287f9f05dc5SKevin Wolf                                       bdrv_co_io_em_complete, &co);
4288f9f05dc5SKevin Wolf     }
4289f9f05dc5SKevin Wolf 
429059370aaaSStefan Hajnoczi     trace_bdrv_co_io_em(bs, sector_num, nb_sectors, is_write, acb);
4291f9f05dc5SKevin Wolf     if (!acb) {
4292f9f05dc5SKevin Wolf         return -EIO;
4293f9f05dc5SKevin Wolf     }
4294f9f05dc5SKevin Wolf     qemu_coroutine_yield();
4295f9f05dc5SKevin Wolf 
4296f9f05dc5SKevin Wolf     return co.ret;
4297f9f05dc5SKevin Wolf }
4298f9f05dc5SKevin Wolf 
4299f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
4300f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
4301f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
4302f9f05dc5SKevin Wolf {
4303f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
4304f9f05dc5SKevin Wolf }
4305f9f05dc5SKevin Wolf 
4306f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
4307f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
4308f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
4309f9f05dc5SKevin Wolf {
4310f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
4311f9f05dc5SKevin Wolf }
4312f9f05dc5SKevin Wolf 
431307f07615SPaolo Bonzini static void coroutine_fn bdrv_flush_co_entry(void *opaque)
4314e7a8a783SKevin Wolf {
431507f07615SPaolo Bonzini     RwCo *rwco = opaque;
431607f07615SPaolo Bonzini 
431707f07615SPaolo Bonzini     rwco->ret = bdrv_co_flush(rwco->bs);
431807f07615SPaolo Bonzini }
431907f07615SPaolo Bonzini 
432007f07615SPaolo Bonzini int coroutine_fn bdrv_co_flush(BlockDriverState *bs)
432107f07615SPaolo Bonzini {
4322eb489bb1SKevin Wolf     int ret;
4323eb489bb1SKevin Wolf 
432429cdb251SPaolo Bonzini     if (!bs || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
432507f07615SPaolo Bonzini         return 0;
4326eb489bb1SKevin Wolf     }
4327eb489bb1SKevin Wolf 
4328ca716364SKevin Wolf     /* Write back cached data to the OS even with cache=unsafe */
4329eb489bb1SKevin Wolf     if (bs->drv->bdrv_co_flush_to_os) {
4330eb489bb1SKevin Wolf         ret = bs->drv->bdrv_co_flush_to_os(bs);
4331eb489bb1SKevin Wolf         if (ret < 0) {
4332eb489bb1SKevin Wolf             return ret;
4333eb489bb1SKevin Wolf         }
4334eb489bb1SKevin Wolf     }
4335eb489bb1SKevin Wolf 
4336ca716364SKevin Wolf     /* But don't actually force it to the disk with cache=unsafe */
4337ca716364SKevin Wolf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
4338d4c82329SKevin Wolf         goto flush_parent;
4339ca716364SKevin Wolf     }
4340ca716364SKevin Wolf 
4341eb489bb1SKevin Wolf     if (bs->drv->bdrv_co_flush_to_disk) {
434229cdb251SPaolo Bonzini         ret = bs->drv->bdrv_co_flush_to_disk(bs);
434307f07615SPaolo Bonzini     } else if (bs->drv->bdrv_aio_flush) {
434407f07615SPaolo Bonzini         BlockDriverAIOCB *acb;
4345e7a8a783SKevin Wolf         CoroutineIOCompletion co = {
4346e7a8a783SKevin Wolf             .coroutine = qemu_coroutine_self(),
4347e7a8a783SKevin Wolf         };
4348e7a8a783SKevin Wolf 
434907f07615SPaolo Bonzini         acb = bs->drv->bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
435007f07615SPaolo Bonzini         if (acb == NULL) {
435129cdb251SPaolo Bonzini             ret = -EIO;
435207f07615SPaolo Bonzini         } else {
4353e7a8a783SKevin Wolf             qemu_coroutine_yield();
435429cdb251SPaolo Bonzini             ret = co.ret;
4355e7a8a783SKevin Wolf         }
435607f07615SPaolo Bonzini     } else {
435707f07615SPaolo Bonzini         /*
435807f07615SPaolo Bonzini          * Some block drivers always operate in either writethrough or unsafe
435907f07615SPaolo Bonzini          * mode and don't support bdrv_flush therefore. Usually qemu doesn't
436007f07615SPaolo Bonzini          * know how the server works (because the behaviour is hardcoded or
436107f07615SPaolo Bonzini          * depends on server-side configuration), so we can't ensure that
436207f07615SPaolo Bonzini          * everything is safe on disk. Returning an error doesn't work because
436307f07615SPaolo Bonzini          * that would break guests even if the server operates in writethrough
436407f07615SPaolo Bonzini          * mode.
436507f07615SPaolo Bonzini          *
436607f07615SPaolo Bonzini          * Let's hope the user knows what he's doing.
436707f07615SPaolo Bonzini          */
436829cdb251SPaolo Bonzini         ret = 0;
436907f07615SPaolo Bonzini     }
437029cdb251SPaolo Bonzini     if (ret < 0) {
437129cdb251SPaolo Bonzini         return ret;
437229cdb251SPaolo Bonzini     }
437329cdb251SPaolo Bonzini 
437429cdb251SPaolo Bonzini     /* Now flush the underlying protocol.  It will also have BDRV_O_NO_FLUSH
437529cdb251SPaolo Bonzini      * in the case of cache=unsafe, so there are no useless flushes.
437629cdb251SPaolo Bonzini      */
4377d4c82329SKevin Wolf flush_parent:
437829cdb251SPaolo Bonzini     return bdrv_co_flush(bs->file);
437907f07615SPaolo Bonzini }
438007f07615SPaolo Bonzini 
43810f15423cSAnthony Liguori void bdrv_invalidate_cache(BlockDriverState *bs)
43820f15423cSAnthony Liguori {
43830f15423cSAnthony Liguori     if (bs->drv && bs->drv->bdrv_invalidate_cache) {
43840f15423cSAnthony Liguori         bs->drv->bdrv_invalidate_cache(bs);
43850f15423cSAnthony Liguori     }
43860f15423cSAnthony Liguori }
43870f15423cSAnthony Liguori 
43880f15423cSAnthony Liguori void bdrv_invalidate_cache_all(void)
43890f15423cSAnthony Liguori {
43900f15423cSAnthony Liguori     BlockDriverState *bs;
43910f15423cSAnthony Liguori 
43920f15423cSAnthony Liguori     QTAILQ_FOREACH(bs, &bdrv_states, list) {
43930f15423cSAnthony Liguori         bdrv_invalidate_cache(bs);
43940f15423cSAnthony Liguori     }
43950f15423cSAnthony Liguori }
43960f15423cSAnthony Liguori 
439707789269SBenoît Canet void bdrv_clear_incoming_migration_all(void)
439807789269SBenoît Canet {
439907789269SBenoît Canet     BlockDriverState *bs;
440007789269SBenoît Canet 
440107789269SBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, list) {
440207789269SBenoît Canet         bs->open_flags = bs->open_flags & ~(BDRV_O_INCOMING);
440307789269SBenoît Canet     }
440407789269SBenoît Canet }
440507789269SBenoît Canet 
440607f07615SPaolo Bonzini int bdrv_flush(BlockDriverState *bs)
440707f07615SPaolo Bonzini {
440807f07615SPaolo Bonzini     Coroutine *co;
440907f07615SPaolo Bonzini     RwCo rwco = {
441007f07615SPaolo Bonzini         .bs = bs,
441107f07615SPaolo Bonzini         .ret = NOT_DONE,
441207f07615SPaolo Bonzini     };
441307f07615SPaolo Bonzini 
441407f07615SPaolo Bonzini     if (qemu_in_coroutine()) {
441507f07615SPaolo Bonzini         /* Fast-path if already in coroutine context */
441607f07615SPaolo Bonzini         bdrv_flush_co_entry(&rwco);
441707f07615SPaolo Bonzini     } else {
441807f07615SPaolo Bonzini         co = qemu_coroutine_create(bdrv_flush_co_entry);
441907f07615SPaolo Bonzini         qemu_coroutine_enter(co, &rwco);
442007f07615SPaolo Bonzini         while (rwco.ret == NOT_DONE) {
442107f07615SPaolo Bonzini             qemu_aio_wait();
442207f07615SPaolo Bonzini         }
442307f07615SPaolo Bonzini     }
442407f07615SPaolo Bonzini 
442507f07615SPaolo Bonzini     return rwco.ret;
442607f07615SPaolo Bonzini }
4427e7a8a783SKevin Wolf 
44284265d620SPaolo Bonzini static void coroutine_fn bdrv_discard_co_entry(void *opaque)
44294265d620SPaolo Bonzini {
44304265d620SPaolo Bonzini     RwCo *rwco = opaque;
44314265d620SPaolo Bonzini 
44324265d620SPaolo Bonzini     rwco->ret = bdrv_co_discard(rwco->bs, rwco->sector_num, rwco->nb_sectors);
44334265d620SPaolo Bonzini }
44344265d620SPaolo Bonzini 
44354265d620SPaolo Bonzini int coroutine_fn bdrv_co_discard(BlockDriverState *bs, int64_t sector_num,
44364265d620SPaolo Bonzini                                  int nb_sectors)
44374265d620SPaolo Bonzini {
44384265d620SPaolo Bonzini     if (!bs->drv) {
44394265d620SPaolo Bonzini         return -ENOMEDIUM;
44404265d620SPaolo Bonzini     } else if (bdrv_check_request(bs, sector_num, nb_sectors)) {
44414265d620SPaolo Bonzini         return -EIO;
44424265d620SPaolo Bonzini     } else if (bs->read_only) {
44434265d620SPaolo Bonzini         return -EROFS;
4444df702c9bSPaolo Bonzini     }
4445df702c9bSPaolo Bonzini 
4446df702c9bSPaolo Bonzini     if (bs->dirty_bitmap) {
44478f0720ecSPaolo Bonzini         bdrv_reset_dirty(bs, sector_num, nb_sectors);
4448df702c9bSPaolo Bonzini     }
4449df702c9bSPaolo Bonzini 
44509e8f1835SPaolo Bonzini     /* Do nothing if disabled.  */
44519e8f1835SPaolo Bonzini     if (!(bs->open_flags & BDRV_O_UNMAP)) {
44529e8f1835SPaolo Bonzini         return 0;
44539e8f1835SPaolo Bonzini     }
44549e8f1835SPaolo Bonzini 
4455df702c9bSPaolo Bonzini     if (bs->drv->bdrv_co_discard) {
44564265d620SPaolo Bonzini         return bs->drv->bdrv_co_discard(bs, sector_num, nb_sectors);
44574265d620SPaolo Bonzini     } else if (bs->drv->bdrv_aio_discard) {
44584265d620SPaolo Bonzini         BlockDriverAIOCB *acb;
44594265d620SPaolo Bonzini         CoroutineIOCompletion co = {
44604265d620SPaolo Bonzini             .coroutine = qemu_coroutine_self(),
44614265d620SPaolo Bonzini         };
44624265d620SPaolo Bonzini 
44634265d620SPaolo Bonzini         acb = bs->drv->bdrv_aio_discard(bs, sector_num, nb_sectors,
44644265d620SPaolo Bonzini                                         bdrv_co_io_em_complete, &co);
44654265d620SPaolo Bonzini         if (acb == NULL) {
44664265d620SPaolo Bonzini             return -EIO;
44674265d620SPaolo Bonzini         } else {
44684265d620SPaolo Bonzini             qemu_coroutine_yield();
44694265d620SPaolo Bonzini             return co.ret;
44704265d620SPaolo Bonzini         }
44714265d620SPaolo Bonzini     } else {
44724265d620SPaolo Bonzini         return 0;
44734265d620SPaolo Bonzini     }
44744265d620SPaolo Bonzini }
44754265d620SPaolo Bonzini 
44764265d620SPaolo Bonzini int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
44774265d620SPaolo Bonzini {
44784265d620SPaolo Bonzini     Coroutine *co;
44794265d620SPaolo Bonzini     RwCo rwco = {
44804265d620SPaolo Bonzini         .bs = bs,
44814265d620SPaolo Bonzini         .sector_num = sector_num,
44824265d620SPaolo Bonzini         .nb_sectors = nb_sectors,
44834265d620SPaolo Bonzini         .ret = NOT_DONE,
44844265d620SPaolo Bonzini     };
44854265d620SPaolo Bonzini 
44864265d620SPaolo Bonzini     if (qemu_in_coroutine()) {
44874265d620SPaolo Bonzini         /* Fast-path if already in coroutine context */
44884265d620SPaolo Bonzini         bdrv_discard_co_entry(&rwco);
44894265d620SPaolo Bonzini     } else {
44904265d620SPaolo Bonzini         co = qemu_coroutine_create(bdrv_discard_co_entry);
44914265d620SPaolo Bonzini         qemu_coroutine_enter(co, &rwco);
44924265d620SPaolo Bonzini         while (rwco.ret == NOT_DONE) {
44934265d620SPaolo Bonzini             qemu_aio_wait();
44944265d620SPaolo Bonzini         }
44954265d620SPaolo Bonzini     }
44964265d620SPaolo Bonzini 
44974265d620SPaolo Bonzini     return rwco.ret;
44984265d620SPaolo Bonzini }
44994265d620SPaolo Bonzini 
4500f9f05dc5SKevin Wolf /**************************************************************/
450119cb3738Sbellard /* removable device support */
450219cb3738Sbellard 
450319cb3738Sbellard /**
450419cb3738Sbellard  * Return TRUE if the media is present
450519cb3738Sbellard  */
450619cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
450719cb3738Sbellard {
450819cb3738Sbellard     BlockDriver *drv = bs->drv;
4509a1aff5bfSMarkus Armbruster 
451019cb3738Sbellard     if (!drv)
451119cb3738Sbellard         return 0;
451219cb3738Sbellard     if (!drv->bdrv_is_inserted)
4513a1aff5bfSMarkus Armbruster         return 1;
4514a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
451519cb3738Sbellard }
451619cb3738Sbellard 
451719cb3738Sbellard /**
45188e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
45198e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
452019cb3738Sbellard  */
452119cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
452219cb3738Sbellard {
452319cb3738Sbellard     BlockDriver *drv = bs->drv;
452419cb3738Sbellard 
45258e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
45268e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
45278e49ca46SMarkus Armbruster     }
45288e49ca46SMarkus Armbruster     return -ENOTSUP;
452919cb3738Sbellard }
453019cb3738Sbellard 
453119cb3738Sbellard /**
453219cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
453319cb3738Sbellard  */
4534f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
453519cb3738Sbellard {
453619cb3738Sbellard     BlockDriver *drv = bs->drv;
453719cb3738Sbellard 
4538822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
4539822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
454019cb3738Sbellard     }
45416f382ed2SLuiz Capitulino 
45426f382ed2SLuiz Capitulino     if (bs->device_name[0] != '\0') {
45436f382ed2SLuiz Capitulino         bdrv_emit_qmp_eject_event(bs, eject_flag);
45446f382ed2SLuiz Capitulino     }
454519cb3738Sbellard }
454619cb3738Sbellard 
454719cb3738Sbellard /**
454819cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
454919cb3738Sbellard  * to eject it manually).
455019cb3738Sbellard  */
4551025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
455219cb3738Sbellard {
455319cb3738Sbellard     BlockDriver *drv = bs->drv;
455419cb3738Sbellard 
4555025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
4556b8c6d095SStefan Hajnoczi 
4557025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
4558025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
455919cb3738Sbellard     }
456019cb3738Sbellard }
4561985a03b0Sths 
4562985a03b0Sths /* needed for generic scsi interface */
4563985a03b0Sths 
4564985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
4565985a03b0Sths {
4566985a03b0Sths     BlockDriver *drv = bs->drv;
4567985a03b0Sths 
4568985a03b0Sths     if (drv && drv->bdrv_ioctl)
4569985a03b0Sths         return drv->bdrv_ioctl(bs, req, buf);
4570985a03b0Sths     return -ENOTSUP;
4571985a03b0Sths }
45727d780669Saliguori 
4573221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
4574221f715dSaliguori         unsigned long int req, void *buf,
45757d780669Saliguori         BlockDriverCompletionFunc *cb, void *opaque)
45767d780669Saliguori {
4577221f715dSaliguori     BlockDriver *drv = bs->drv;
45787d780669Saliguori 
4579221f715dSaliguori     if (drv && drv->bdrv_aio_ioctl)
4580221f715dSaliguori         return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
4581221f715dSaliguori     return NULL;
45827d780669Saliguori }
4583e268ca52Saliguori 
45847b6f9300SMarkus Armbruster void bdrv_set_buffer_alignment(BlockDriverState *bs, int align)
45857b6f9300SMarkus Armbruster {
45867b6f9300SMarkus Armbruster     bs->buffer_alignment = align;
45877b6f9300SMarkus Armbruster }
45887cd1e32aSlirans@il.ibm.com 
4589e268ca52Saliguori void *qemu_blockalign(BlockDriverState *bs, size_t size)
4590e268ca52Saliguori {
4591e268ca52Saliguori     return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
4592e268ca52Saliguori }
45937cd1e32aSlirans@il.ibm.com 
4594c53b1c51SStefan Hajnoczi /*
4595c53b1c51SStefan Hajnoczi  * Check if all memory in this vector is sector aligned.
4596c53b1c51SStefan Hajnoczi  */
4597c53b1c51SStefan Hajnoczi bool bdrv_qiov_is_aligned(BlockDriverState *bs, QEMUIOVector *qiov)
4598c53b1c51SStefan Hajnoczi {
4599c53b1c51SStefan Hajnoczi     int i;
4600c53b1c51SStefan Hajnoczi 
4601c53b1c51SStefan Hajnoczi     for (i = 0; i < qiov->niov; i++) {
4602c53b1c51SStefan Hajnoczi         if ((uintptr_t) qiov->iov[i].iov_base % bs->buffer_alignment) {
4603c53b1c51SStefan Hajnoczi             return false;
4604c53b1c51SStefan Hajnoczi         }
4605c53b1c51SStefan Hajnoczi     }
4606c53b1c51SStefan Hajnoczi 
4607c53b1c51SStefan Hajnoczi     return true;
4608c53b1c51SStefan Hajnoczi }
4609c53b1c51SStefan Hajnoczi 
461050717e94SPaolo Bonzini void bdrv_set_dirty_tracking(BlockDriverState *bs, int granularity)
46117cd1e32aSlirans@il.ibm.com {
46127cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
4613a55eb92cSJan Kiszka 
461450717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
461550717e94SPaolo Bonzini 
461650717e94SPaolo Bonzini     if (granularity) {
461750717e94SPaolo Bonzini         granularity >>= BDRV_SECTOR_BITS;
461850717e94SPaolo Bonzini         assert(!bs->dirty_bitmap);
46198f0720ecSPaolo Bonzini         bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS);
462050717e94SPaolo Bonzini         bs->dirty_bitmap = hbitmap_alloc(bitmap_size, ffs(granularity) - 1);
46217cd1e32aSlirans@il.ibm.com     } else {
4622c6d22830SJan Kiszka         if (bs->dirty_bitmap) {
46238f0720ecSPaolo Bonzini             hbitmap_free(bs->dirty_bitmap);
4624c6d22830SJan Kiszka             bs->dirty_bitmap = NULL;
46257cd1e32aSlirans@il.ibm.com         }
46267cd1e32aSlirans@il.ibm.com     }
46277cd1e32aSlirans@il.ibm.com }
46287cd1e32aSlirans@il.ibm.com 
46297cd1e32aSlirans@il.ibm.com int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
46307cd1e32aSlirans@il.ibm.com {
46318f0720ecSPaolo Bonzini     if (bs->dirty_bitmap) {
46328f0720ecSPaolo Bonzini         return hbitmap_get(bs->dirty_bitmap, sector);
46337cd1e32aSlirans@il.ibm.com     } else {
46347cd1e32aSlirans@il.ibm.com         return 0;
46357cd1e32aSlirans@il.ibm.com     }
46367cd1e32aSlirans@il.ibm.com }
46377cd1e32aSlirans@il.ibm.com 
46388f0720ecSPaolo Bonzini void bdrv_dirty_iter_init(BlockDriverState *bs, HBitmapIter *hbi)
46391755da16SPaolo Bonzini {
46408f0720ecSPaolo Bonzini     hbitmap_iter_init(hbi, bs->dirty_bitmap, 0);
46411755da16SPaolo Bonzini }
46421755da16SPaolo Bonzini 
46431755da16SPaolo Bonzini void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
46441755da16SPaolo Bonzini                     int nr_sectors)
46451755da16SPaolo Bonzini {
46468f0720ecSPaolo Bonzini     hbitmap_set(bs->dirty_bitmap, cur_sector, nr_sectors);
46471755da16SPaolo Bonzini }
46481755da16SPaolo Bonzini 
46497cd1e32aSlirans@il.ibm.com void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
46507cd1e32aSlirans@il.ibm.com                       int nr_sectors)
46517cd1e32aSlirans@il.ibm.com {
46528f0720ecSPaolo Bonzini     hbitmap_reset(bs->dirty_bitmap, cur_sector, nr_sectors);
46537cd1e32aSlirans@il.ibm.com }
4654aaa0eb75SLiran Schour 
4655aaa0eb75SLiran Schour int64_t bdrv_get_dirty_count(BlockDriverState *bs)
4656aaa0eb75SLiran Schour {
46578f0720ecSPaolo Bonzini     if (bs->dirty_bitmap) {
4658acc906c6SPaolo Bonzini         return hbitmap_count(bs->dirty_bitmap);
46598f0720ecSPaolo Bonzini     } else {
46608f0720ecSPaolo Bonzini         return 0;
46618f0720ecSPaolo Bonzini     }
4662aaa0eb75SLiran Schour }
4663f88e1a42SJes Sorensen 
4664db593f25SMarcelo Tosatti void bdrv_set_in_use(BlockDriverState *bs, int in_use)
4665db593f25SMarcelo Tosatti {
4666db593f25SMarcelo Tosatti     assert(bs->in_use != in_use);
4667db593f25SMarcelo Tosatti     bs->in_use = in_use;
4668db593f25SMarcelo Tosatti }
4669db593f25SMarcelo Tosatti 
4670db593f25SMarcelo Tosatti int bdrv_in_use(BlockDriverState *bs)
4671db593f25SMarcelo Tosatti {
4672db593f25SMarcelo Tosatti     return bs->in_use;
4673db593f25SMarcelo Tosatti }
4674db593f25SMarcelo Tosatti 
467528a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
467628a7282aSLuiz Capitulino {
4677d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
467858e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
467928a7282aSLuiz Capitulino }
468028a7282aSLuiz Capitulino 
468128a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
468228a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
468328a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
468428a7282aSLuiz Capitulino {
4685d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
468692aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
468792aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
468892aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
468928a7282aSLuiz Capitulino }
469028a7282aSLuiz Capitulino 
469128a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
469228a7282aSLuiz Capitulino {
4693d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
469428a7282aSLuiz Capitulino }
469528a7282aSLuiz Capitulino 
469628a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
469728a7282aSLuiz Capitulino {
469828a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
469958e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
47003bd293c3SPaolo Bonzini         if (bs->job) {
47013bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
47023bd293c3SPaolo Bonzini         }
470328a7282aSLuiz Capitulino     }
470428a7282aSLuiz Capitulino }
470528a7282aSLuiz Capitulino 
470628a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
470728a7282aSLuiz Capitulino {
47083e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
47093e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
471058e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
471158e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
471228a7282aSLuiz Capitulino     }
471328a7282aSLuiz Capitulino }
471428a7282aSLuiz Capitulino 
4715a597e79cSChristoph Hellwig void
4716a597e79cSChristoph Hellwig bdrv_acct_start(BlockDriverState *bs, BlockAcctCookie *cookie, int64_t bytes,
4717a597e79cSChristoph Hellwig         enum BlockAcctType type)
4718a597e79cSChristoph Hellwig {
4719a597e79cSChristoph Hellwig     assert(type < BDRV_MAX_IOTYPE);
4720a597e79cSChristoph Hellwig 
4721a597e79cSChristoph Hellwig     cookie->bytes = bytes;
4722c488c7f6SChristoph Hellwig     cookie->start_time_ns = get_clock();
4723a597e79cSChristoph Hellwig     cookie->type = type;
4724a597e79cSChristoph Hellwig }
4725a597e79cSChristoph Hellwig 
4726a597e79cSChristoph Hellwig void
4727a597e79cSChristoph Hellwig bdrv_acct_done(BlockDriverState *bs, BlockAcctCookie *cookie)
4728a597e79cSChristoph Hellwig {
4729a597e79cSChristoph Hellwig     assert(cookie->type < BDRV_MAX_IOTYPE);
4730a597e79cSChristoph Hellwig 
4731a597e79cSChristoph Hellwig     bs->nr_bytes[cookie->type] += cookie->bytes;
4732a597e79cSChristoph Hellwig     bs->nr_ops[cookie->type]++;
4733c488c7f6SChristoph Hellwig     bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
4734a597e79cSChristoph Hellwig }
4735a597e79cSChristoph Hellwig 
4736d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
4737f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
4738f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
4739f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
4740f88e1a42SJes Sorensen {
4741f88e1a42SJes Sorensen     QEMUOptionParameter *param = NULL, *create_options = NULL;
4742d220894eSKevin Wolf     QEMUOptionParameter *backing_fmt, *backing_file, *size;
4743f88e1a42SJes Sorensen     BlockDriverState *bs = NULL;
4744f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
474596df67d1SStefan Hajnoczi     BlockDriver *backing_drv = NULL;
4746f88e1a42SJes Sorensen     int ret = 0;
4747f88e1a42SJes Sorensen 
4748f88e1a42SJes Sorensen     /* Find driver and parse its options */
4749f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
4750f88e1a42SJes Sorensen     if (!drv) {
475171c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
4752d92ada22SLuiz Capitulino         return;
4753f88e1a42SJes Sorensen     }
4754f88e1a42SJes Sorensen 
4755f88e1a42SJes Sorensen     proto_drv = bdrv_find_protocol(filename);
4756f88e1a42SJes Sorensen     if (!proto_drv) {
475771c79813SLuiz Capitulino         error_setg(errp, "Unknown protocol '%s'", filename);
4758d92ada22SLuiz Capitulino         return;
4759f88e1a42SJes Sorensen     }
4760f88e1a42SJes Sorensen 
4761f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
4762f88e1a42SJes Sorensen                                               drv->create_options);
4763f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
4764f88e1a42SJes Sorensen                                               proto_drv->create_options);
4765f88e1a42SJes Sorensen 
4766f88e1a42SJes Sorensen     /* Create parameter list with default values */
4767f88e1a42SJes Sorensen     param = parse_option_parameters("", create_options, param);
4768f88e1a42SJes Sorensen 
4769f88e1a42SJes Sorensen     set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
4770f88e1a42SJes Sorensen 
4771f88e1a42SJes Sorensen     /* Parse -o options */
4772f88e1a42SJes Sorensen     if (options) {
4773f88e1a42SJes Sorensen         param = parse_option_parameters(options, create_options, param);
4774f88e1a42SJes Sorensen         if (param == NULL) {
477571c79813SLuiz Capitulino             error_setg(errp, "Invalid options for file format '%s'.", fmt);
4776f88e1a42SJes Sorensen             goto out;
4777f88e1a42SJes Sorensen         }
4778f88e1a42SJes Sorensen     }
4779f88e1a42SJes Sorensen 
4780f88e1a42SJes Sorensen     if (base_filename) {
4781f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
4782f88e1a42SJes Sorensen                                  base_filename)) {
478371c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
478471c79813SLuiz Capitulino                        fmt);
4785f88e1a42SJes Sorensen             goto out;
4786f88e1a42SJes Sorensen         }
4787f88e1a42SJes Sorensen     }
4788f88e1a42SJes Sorensen 
4789f88e1a42SJes Sorensen     if (base_fmt) {
4790f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
479171c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
479271c79813SLuiz Capitulino                              "format '%s'", fmt);
4793f88e1a42SJes Sorensen             goto out;
4794f88e1a42SJes Sorensen         }
4795f88e1a42SJes Sorensen     }
4796f88e1a42SJes Sorensen 
4797792da93aSJes Sorensen     backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
4798792da93aSJes Sorensen     if (backing_file && backing_file->value.s) {
4799792da93aSJes Sorensen         if (!strcmp(filename, backing_file->value.s)) {
480071c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
480171c79813SLuiz Capitulino                              "same filename as the backing file");
4802792da93aSJes Sorensen             goto out;
4803792da93aSJes Sorensen         }
4804792da93aSJes Sorensen     }
4805792da93aSJes Sorensen 
4806f88e1a42SJes Sorensen     backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
4807f88e1a42SJes Sorensen     if (backing_fmt && backing_fmt->value.s) {
480896df67d1SStefan Hajnoczi         backing_drv = bdrv_find_format(backing_fmt->value.s);
480996df67d1SStefan Hajnoczi         if (!backing_drv) {
481071c79813SLuiz Capitulino             error_setg(errp, "Unknown backing file format '%s'",
481171c79813SLuiz Capitulino                        backing_fmt->value.s);
4812f88e1a42SJes Sorensen             goto out;
4813f88e1a42SJes Sorensen         }
4814f88e1a42SJes Sorensen     }
4815f88e1a42SJes Sorensen 
4816f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
4817f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
4818d220894eSKevin Wolf     size = get_option_parameter(param, BLOCK_OPT_SIZE);
4819d220894eSKevin Wolf     if (size && size->value.n == -1) {
4820f88e1a42SJes Sorensen         if (backing_file && backing_file->value.s) {
4821f88e1a42SJes Sorensen             uint64_t size;
4822f88e1a42SJes Sorensen             char buf[32];
482363090dacSPaolo Bonzini             int back_flags;
482463090dacSPaolo Bonzini 
482563090dacSPaolo Bonzini             /* backing files always opened read-only */
482663090dacSPaolo Bonzini             back_flags =
482763090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
4828f88e1a42SJes Sorensen 
4829f88e1a42SJes Sorensen             bs = bdrv_new("");
4830f88e1a42SJes Sorensen 
4831de9c0cecSKevin Wolf             ret = bdrv_open(bs, backing_file->value.s, NULL, back_flags,
4832de9c0cecSKevin Wolf                             backing_drv);
4833f88e1a42SJes Sorensen             if (ret < 0) {
483471c79813SLuiz Capitulino                 error_setg_errno(errp, -ret, "Could not open '%s'",
483571c79813SLuiz Capitulino                                  backing_file->value.s);
4836f88e1a42SJes Sorensen                 goto out;
4837f88e1a42SJes Sorensen             }
4838f88e1a42SJes Sorensen             bdrv_get_geometry(bs, &size);
4839f88e1a42SJes Sorensen             size *= 512;
4840f88e1a42SJes Sorensen 
4841f88e1a42SJes Sorensen             snprintf(buf, sizeof(buf), "%" PRId64, size);
4842f88e1a42SJes Sorensen             set_option_parameter(param, BLOCK_OPT_SIZE, buf);
4843f88e1a42SJes Sorensen         } else {
484471c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
4845f88e1a42SJes Sorensen             goto out;
4846f88e1a42SJes Sorensen         }
4847f88e1a42SJes Sorensen     }
4848f88e1a42SJes Sorensen 
4849f382d43aSMiroslav Rezanina     if (!quiet) {
4850f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
4851f88e1a42SJes Sorensen         print_option_parameters(param);
4852f88e1a42SJes Sorensen         puts("");
4853f382d43aSMiroslav Rezanina     }
4854f88e1a42SJes Sorensen     ret = bdrv_create(drv, filename, param);
4855f88e1a42SJes Sorensen     if (ret < 0) {
4856f88e1a42SJes Sorensen         if (ret == -ENOTSUP) {
485771c79813SLuiz Capitulino             error_setg(errp,"Formatting or formatting option not supported for "
485871c79813SLuiz Capitulino                             "file format '%s'", fmt);
4859f88e1a42SJes Sorensen         } else if (ret == -EFBIG) {
4860*f3f4d2c0SKevin Wolf             const char *cluster_size_hint = "";
4861*f3f4d2c0SKevin Wolf             if (get_option_parameter(create_options, BLOCK_OPT_CLUSTER_SIZE)) {
4862*f3f4d2c0SKevin Wolf                 cluster_size_hint = " (try using a larger cluster size)";
4863*f3f4d2c0SKevin Wolf             }
4864*f3f4d2c0SKevin Wolf             error_setg(errp, "The image size is too large for file format '%s'%s",
4865*f3f4d2c0SKevin Wolf                        fmt, cluster_size_hint);
4866f88e1a42SJes Sorensen         } else {
486771c79813SLuiz Capitulino             error_setg(errp, "%s: error while creating %s: %s", filename, fmt,
486871c79813SLuiz Capitulino                        strerror(-ret));
4869f88e1a42SJes Sorensen         }
4870f88e1a42SJes Sorensen     }
4871f88e1a42SJes Sorensen 
4872f88e1a42SJes Sorensen out:
4873f88e1a42SJes Sorensen     free_option_parameters(create_options);
4874f88e1a42SJes Sorensen     free_option_parameters(param);
4875f88e1a42SJes Sorensen 
4876f88e1a42SJes Sorensen     if (bs) {
4877f88e1a42SJes Sorensen         bdrv_delete(bs);
4878f88e1a42SJes Sorensen     }
4879f88e1a42SJes Sorensen }
488085d126f3SStefan Hajnoczi 
488185d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
488285d126f3SStefan Hajnoczi {
488385d126f3SStefan Hajnoczi     /* Currently BlockDriverState always uses the main loop AioContext */
488485d126f3SStefan Hajnoczi     return qemu_get_aio_context();
488585d126f3SStefan Hajnoczi }
4886