xref: /openbmc/qemu/block.c (revision b404f72036716ab8ace04b83a8f0a93be4739a6a)
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"
35c13163fbSBenoît Canet #include "block/qapi.h"
36b2023818SLuiz Capitulino #include "qmp-commands.h"
371de7afc9SPaolo Bonzini #include "qemu/timer.h"
38fc01f7e7Sbellard 
3971e72a19SJuan Quintela #ifdef CONFIG_BSD
407674e7bfSbellard #include <sys/types.h>
417674e7bfSbellard #include <sys/stat.h>
427674e7bfSbellard #include <sys/ioctl.h>
4372cf2d4fSBlue Swirl #include <sys/queue.h>
44c5e97233Sblueswir1 #ifndef __DragonFly__
457674e7bfSbellard #include <sys/disk.h>
467674e7bfSbellard #endif
47c5e97233Sblueswir1 #endif
487674e7bfSbellard 
4949dc768dSaliguori #ifdef _WIN32
5049dc768dSaliguori #include <windows.h>
5149dc768dSaliguori #endif
5249dc768dSaliguori 
53e4654d2dSFam Zheng struct BdrvDirtyBitmap {
54e4654d2dSFam Zheng     HBitmap *bitmap;
55e4654d2dSFam Zheng     QLIST_ENTRY(BdrvDirtyBitmap) list;
56e4654d2dSFam Zheng };
57e4654d2dSFam Zheng 
581c9805a3SStefan Hajnoczi #define NOT_DONE 0x7fffffff /* used while emulated sync operation in progress */
591c9805a3SStefan Hajnoczi 
607d4b4ba5SMarkus Armbruster static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load);
61f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
62f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
63c87c0672Saliguori         BlockDriverCompletionFunc *cb, void *opaque);
64f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
65f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
66ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque);
67f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
68f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
69f9f05dc5SKevin Wolf                                          QEMUIOVector *iov);
70f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
71f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
72f9f05dc5SKevin Wolf                                          QEMUIOVector *iov);
73c5fbe571SStefan Hajnoczi static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
74470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
75470c0504SStefan Hajnoczi     BdrvRequestFlags flags);
761c9805a3SStefan Hajnoczi static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
77f08f2ddaSStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
78f08f2ddaSStefan Hajnoczi     BdrvRequestFlags flags);
79b2a61371SStefan Hajnoczi static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
80b2a61371SStefan Hajnoczi                                                int64_t sector_num,
81b2a61371SStefan Hajnoczi                                                QEMUIOVector *qiov,
82b2a61371SStefan Hajnoczi                                                int nb_sectors,
83d20d9b7cSPaolo Bonzini                                                BdrvRequestFlags flags,
84b2a61371SStefan Hajnoczi                                                BlockDriverCompletionFunc *cb,
85b2a61371SStefan Hajnoczi                                                void *opaque,
868c5873d6SStefan Hajnoczi                                                bool is_write);
87b2a61371SStefan Hajnoczi static void coroutine_fn bdrv_co_do_rw(void *opaque);
88621f0589SKevin Wolf static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
89aa7bfbffSPeter Lieven     int64_t sector_num, int nb_sectors, BdrvRequestFlags flags);
90ec530c81Sbellard 
911b7bdbc1SStefan Hajnoczi static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
921b7bdbc1SStefan Hajnoczi     QTAILQ_HEAD_INITIALIZER(bdrv_states);
937ee930d0Sblueswir1 
94dc364f4cSBenoît Canet static QTAILQ_HEAD(, BlockDriverState) graph_bdrv_states =
95dc364f4cSBenoît Canet     QTAILQ_HEAD_INITIALIZER(graph_bdrv_states);
96dc364f4cSBenoît Canet 
978a22f02aSStefan Hajnoczi static QLIST_HEAD(, BlockDriver) bdrv_drivers =
988a22f02aSStefan Hajnoczi     QLIST_HEAD_INITIALIZER(bdrv_drivers);
99ea2384d3Sbellard 
100eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
101eb852011SMarkus Armbruster static int use_bdrv_whitelist;
102eb852011SMarkus Armbruster 
1039e0b22f4SStefan Hajnoczi #ifdef _WIN32
1049e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
1059e0b22f4SStefan Hajnoczi {
1069e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
1079e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
1089e0b22f4SStefan Hajnoczi             filename[1] == ':');
1099e0b22f4SStefan Hajnoczi }
1109e0b22f4SStefan Hajnoczi 
1119e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
1129e0b22f4SStefan Hajnoczi {
1139e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
1149e0b22f4SStefan Hajnoczi         filename[2] == '\0')
1159e0b22f4SStefan Hajnoczi         return 1;
1169e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
1179e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
1189e0b22f4SStefan Hajnoczi         return 1;
1199e0b22f4SStefan Hajnoczi     return 0;
1209e0b22f4SStefan Hajnoczi }
1219e0b22f4SStefan Hajnoczi #endif
1229e0b22f4SStefan Hajnoczi 
1230563e191SZhi Yong Wu /* throttling disk I/O limits */
124cc0681c4SBenoît Canet void bdrv_set_io_limits(BlockDriverState *bs,
125cc0681c4SBenoît Canet                         ThrottleConfig *cfg)
126cc0681c4SBenoît Canet {
127cc0681c4SBenoît Canet     int i;
128cc0681c4SBenoît Canet 
129cc0681c4SBenoît Canet     throttle_config(&bs->throttle_state, cfg);
130cc0681c4SBenoît Canet 
131cc0681c4SBenoît Canet     for (i = 0; i < 2; i++) {
132cc0681c4SBenoît Canet         qemu_co_enter_next(&bs->throttled_reqs[i]);
133cc0681c4SBenoît Canet     }
134cc0681c4SBenoît Canet }
135cc0681c4SBenoît Canet 
136cc0681c4SBenoît Canet /* this function drain all the throttled IOs */
137cc0681c4SBenoît Canet static bool bdrv_start_throttled_reqs(BlockDriverState *bs)
138cc0681c4SBenoît Canet {
139cc0681c4SBenoît Canet     bool drained = false;
140cc0681c4SBenoît Canet     bool enabled = bs->io_limits_enabled;
141cc0681c4SBenoît Canet     int i;
142cc0681c4SBenoît Canet 
143cc0681c4SBenoît Canet     bs->io_limits_enabled = false;
144cc0681c4SBenoît Canet 
145cc0681c4SBenoît Canet     for (i = 0; i < 2; i++) {
146cc0681c4SBenoît Canet         while (qemu_co_enter_next(&bs->throttled_reqs[i])) {
147cc0681c4SBenoît Canet             drained = true;
148cc0681c4SBenoît Canet         }
149cc0681c4SBenoît Canet     }
150cc0681c4SBenoît Canet 
151cc0681c4SBenoît Canet     bs->io_limits_enabled = enabled;
152cc0681c4SBenoît Canet 
153cc0681c4SBenoît Canet     return drained;
154cc0681c4SBenoît Canet }
155cc0681c4SBenoît Canet 
15698f90dbaSZhi Yong Wu void bdrv_io_limits_disable(BlockDriverState *bs)
15798f90dbaSZhi Yong Wu {
15898f90dbaSZhi Yong Wu     bs->io_limits_enabled = false;
15998f90dbaSZhi Yong Wu 
160cc0681c4SBenoît Canet     bdrv_start_throttled_reqs(bs);
16198f90dbaSZhi Yong Wu 
162cc0681c4SBenoît Canet     throttle_destroy(&bs->throttle_state);
16398f90dbaSZhi Yong Wu }
16498f90dbaSZhi Yong Wu 
165cc0681c4SBenoît Canet static void bdrv_throttle_read_timer_cb(void *opaque)
1660563e191SZhi Yong Wu {
1670563e191SZhi Yong Wu     BlockDriverState *bs = opaque;
168cc0681c4SBenoît Canet     qemu_co_enter_next(&bs->throttled_reqs[0]);
1690563e191SZhi Yong Wu }
1700563e191SZhi Yong Wu 
171cc0681c4SBenoît Canet static void bdrv_throttle_write_timer_cb(void *opaque)
172cc0681c4SBenoît Canet {
173cc0681c4SBenoît Canet     BlockDriverState *bs = opaque;
174cc0681c4SBenoît Canet     qemu_co_enter_next(&bs->throttled_reqs[1]);
175cc0681c4SBenoît Canet }
176cc0681c4SBenoît Canet 
177cc0681c4SBenoît Canet /* should be called before bdrv_set_io_limits if a limit is set */
1780563e191SZhi Yong Wu void bdrv_io_limits_enable(BlockDriverState *bs)
1790563e191SZhi Yong Wu {
180cc0681c4SBenoît Canet     assert(!bs->io_limits_enabled);
181cc0681c4SBenoît Canet     throttle_init(&bs->throttle_state,
182cc0681c4SBenoît Canet                   QEMU_CLOCK_VIRTUAL,
183cc0681c4SBenoît Canet                   bdrv_throttle_read_timer_cb,
184cc0681c4SBenoît Canet                   bdrv_throttle_write_timer_cb,
185cc0681c4SBenoît Canet                   bs);
1860563e191SZhi Yong Wu     bs->io_limits_enabled = true;
1870563e191SZhi Yong Wu }
1880563e191SZhi Yong Wu 
189cc0681c4SBenoît Canet /* This function makes an IO wait if needed
190cc0681c4SBenoît Canet  *
191cc0681c4SBenoît Canet  * @nb_sectors: the number of sectors of the IO
192cc0681c4SBenoît Canet  * @is_write:   is the IO a write
19398f90dbaSZhi Yong Wu  */
194cc0681c4SBenoît Canet static void bdrv_io_limits_intercept(BlockDriverState *bs,
195cc0681c4SBenoît Canet                                      int nb_sectors,
196cc0681c4SBenoît Canet                                      bool is_write)
197cc0681c4SBenoît Canet {
198cc0681c4SBenoît Canet     /* does this io must wait */
199cc0681c4SBenoît Canet     bool must_wait = throttle_schedule_timer(&bs->throttle_state, is_write);
20098f90dbaSZhi Yong Wu 
201cc0681c4SBenoît Canet     /* if must wait or any request of this type throttled queue the IO */
202cc0681c4SBenoît Canet     if (must_wait ||
203cc0681c4SBenoît Canet         !qemu_co_queue_empty(&bs->throttled_reqs[is_write])) {
204cc0681c4SBenoît Canet         qemu_co_queue_wait(&bs->throttled_reqs[is_write]);
20598f90dbaSZhi Yong Wu     }
20698f90dbaSZhi Yong Wu 
207cc0681c4SBenoît Canet     /* the IO will be executed, do the accounting */
208cc0681c4SBenoît Canet     throttle_account(&bs->throttle_state,
209cc0681c4SBenoît Canet                      is_write,
210cc0681c4SBenoît Canet                      nb_sectors * BDRV_SECTOR_SIZE);
211cc0681c4SBenoît Canet 
212cc0681c4SBenoît Canet     /* if the next request must wait -> do nothing */
213cc0681c4SBenoît Canet     if (throttle_schedule_timer(&bs->throttle_state, is_write)) {
214cc0681c4SBenoît Canet         return;
215cc0681c4SBenoît Canet     }
216cc0681c4SBenoît Canet 
217cc0681c4SBenoît Canet     /* else queue next request for execution */
218cc0681c4SBenoît Canet     qemu_co_queue_next(&bs->throttled_reqs[is_write]);
21998f90dbaSZhi Yong Wu }
22098f90dbaSZhi Yong Wu 
221339064d5SKevin Wolf size_t bdrv_opt_mem_align(BlockDriverState *bs)
222339064d5SKevin Wolf {
223339064d5SKevin Wolf     if (!bs || !bs->drv) {
224339064d5SKevin Wolf         /* 4k should be on the safe side */
225339064d5SKevin Wolf         return 4096;
226339064d5SKevin Wolf     }
227339064d5SKevin Wolf 
228339064d5SKevin Wolf     return bs->bl.opt_mem_alignment;
229339064d5SKevin Wolf }
230339064d5SKevin Wolf 
2319e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
2329e0b22f4SStefan Hajnoczi static int path_has_protocol(const char *path)
2339e0b22f4SStefan Hajnoczi {
234947995c0SPaolo Bonzini     const char *p;
235947995c0SPaolo Bonzini 
2369e0b22f4SStefan Hajnoczi #ifdef _WIN32
2379e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
2389e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
2399e0b22f4SStefan Hajnoczi         return 0;
2409e0b22f4SStefan Hajnoczi     }
241947995c0SPaolo Bonzini     p = path + strcspn(path, ":/\\");
242947995c0SPaolo Bonzini #else
243947995c0SPaolo Bonzini     p = path + strcspn(path, ":/");
2449e0b22f4SStefan Hajnoczi #endif
2459e0b22f4SStefan Hajnoczi 
246947995c0SPaolo Bonzini     return *p == ':';
2479e0b22f4SStefan Hajnoczi }
2489e0b22f4SStefan Hajnoczi 
24983f64091Sbellard int path_is_absolute(const char *path)
25083f64091Sbellard {
25121664424Sbellard #ifdef _WIN32
25221664424Sbellard     /* specific case for names like: "\\.\d:" */
253f53f4da9SPaolo Bonzini     if (is_windows_drive(path) || is_windows_drive_prefix(path)) {
25421664424Sbellard         return 1;
255f53f4da9SPaolo Bonzini     }
256f53f4da9SPaolo Bonzini     return (*path == '/' || *path == '\\');
2573b9f94e1Sbellard #else
258f53f4da9SPaolo Bonzini     return (*path == '/');
2593b9f94e1Sbellard #endif
26083f64091Sbellard }
26183f64091Sbellard 
26283f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
26383f64091Sbellard    path to it by considering it is relative to base_path. URL are
26483f64091Sbellard    supported. */
26583f64091Sbellard void path_combine(char *dest, int dest_size,
26683f64091Sbellard                   const char *base_path,
26783f64091Sbellard                   const char *filename)
26883f64091Sbellard {
26983f64091Sbellard     const char *p, *p1;
27083f64091Sbellard     int len;
27183f64091Sbellard 
27283f64091Sbellard     if (dest_size <= 0)
27383f64091Sbellard         return;
27483f64091Sbellard     if (path_is_absolute(filename)) {
27583f64091Sbellard         pstrcpy(dest, dest_size, filename);
27683f64091Sbellard     } else {
27783f64091Sbellard         p = strchr(base_path, ':');
27883f64091Sbellard         if (p)
27983f64091Sbellard             p++;
28083f64091Sbellard         else
28183f64091Sbellard             p = base_path;
2823b9f94e1Sbellard         p1 = strrchr(base_path, '/');
2833b9f94e1Sbellard #ifdef _WIN32
2843b9f94e1Sbellard         {
2853b9f94e1Sbellard             const char *p2;
2863b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
2873b9f94e1Sbellard             if (!p1 || p2 > p1)
2883b9f94e1Sbellard                 p1 = p2;
2893b9f94e1Sbellard         }
2903b9f94e1Sbellard #endif
29183f64091Sbellard         if (p1)
29283f64091Sbellard             p1++;
29383f64091Sbellard         else
29483f64091Sbellard             p1 = base_path;
29583f64091Sbellard         if (p1 > p)
29683f64091Sbellard             p = p1;
29783f64091Sbellard         len = p - base_path;
29883f64091Sbellard         if (len > dest_size - 1)
29983f64091Sbellard             len = dest_size - 1;
30083f64091Sbellard         memcpy(dest, base_path, len);
30183f64091Sbellard         dest[len] = '\0';
30283f64091Sbellard         pstrcat(dest, dest_size, filename);
30383f64091Sbellard     }
30483f64091Sbellard }
30583f64091Sbellard 
306dc5a1371SPaolo Bonzini void bdrv_get_full_backing_filename(BlockDriverState *bs, char *dest, size_t sz)
307dc5a1371SPaolo Bonzini {
308dc5a1371SPaolo Bonzini     if (bs->backing_file[0] == '\0' || path_has_protocol(bs->backing_file)) {
309dc5a1371SPaolo Bonzini         pstrcpy(dest, sz, bs->backing_file);
310dc5a1371SPaolo Bonzini     } else {
311dc5a1371SPaolo Bonzini         path_combine(dest, sz, bs->filename, bs->backing_file);
312dc5a1371SPaolo Bonzini     }
313dc5a1371SPaolo Bonzini }
314dc5a1371SPaolo Bonzini 
3155efa9d5aSAnthony Liguori void bdrv_register(BlockDriver *bdrv)
316ea2384d3Sbellard {
3178c5873d6SStefan Hajnoczi     /* Block drivers without coroutine functions need emulation */
3188c5873d6SStefan Hajnoczi     if (!bdrv->bdrv_co_readv) {
319f9f05dc5SKevin Wolf         bdrv->bdrv_co_readv = bdrv_co_readv_em;
320f9f05dc5SKevin Wolf         bdrv->bdrv_co_writev = bdrv_co_writev_em;
321f9f05dc5SKevin Wolf 
322f8c35c1dSStefan Hajnoczi         /* bdrv_co_readv_em()/brdv_co_writev_em() work in terms of aio, so if
323f8c35c1dSStefan Hajnoczi          * the block driver lacks aio we need to emulate that too.
324f8c35c1dSStefan Hajnoczi          */
325f9f05dc5SKevin Wolf         if (!bdrv->bdrv_aio_readv) {
32683f64091Sbellard             /* add AIO emulation layer */
327f141eafeSaliguori             bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
328f141eafeSaliguori             bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
32983f64091Sbellard         }
330f9f05dc5SKevin Wolf     }
331b2e12bc6SChristoph Hellwig 
3328a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
333ea2384d3Sbellard }
334b338082bSbellard 
335b338082bSbellard /* create a new block device (by default it is empty) */
336b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
337fc01f7e7Sbellard {
3381b7bdbc1SStefan Hajnoczi     BlockDriverState *bs;
339b338082bSbellard 
3407267c094SAnthony Liguori     bs = g_malloc0(sizeof(BlockDriverState));
341e4654d2dSFam Zheng     QLIST_INIT(&bs->dirty_bitmaps);
342b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
343ea2384d3Sbellard     if (device_name[0] != '\0') {
344dc364f4cSBenoît Canet         QTAILQ_INSERT_TAIL(&bdrv_states, bs, device_list);
345ea2384d3Sbellard     }
34628a7282aSLuiz Capitulino     bdrv_iostatus_disable(bs);
347d7d512f6SPaolo Bonzini     notifier_list_init(&bs->close_notifiers);
348d616b224SStefan Hajnoczi     notifier_with_return_list_init(&bs->before_write_notifiers);
349cc0681c4SBenoît Canet     qemu_co_queue_init(&bs->throttled_reqs[0]);
350cc0681c4SBenoît Canet     qemu_co_queue_init(&bs->throttled_reqs[1]);
3519fcb0251SFam Zheng     bs->refcnt = 1;
352d7d512f6SPaolo Bonzini 
353b338082bSbellard     return bs;
354b338082bSbellard }
355b338082bSbellard 
356d7d512f6SPaolo Bonzini void bdrv_add_close_notifier(BlockDriverState *bs, Notifier *notify)
357d7d512f6SPaolo Bonzini {
358d7d512f6SPaolo Bonzini     notifier_list_add(&bs->close_notifiers, notify);
359d7d512f6SPaolo Bonzini }
360d7d512f6SPaolo Bonzini 
361ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
362ea2384d3Sbellard {
363ea2384d3Sbellard     BlockDriver *drv1;
3648a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
3658a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
366ea2384d3Sbellard             return drv1;
367ea2384d3Sbellard         }
3688a22f02aSStefan Hajnoczi     }
369ea2384d3Sbellard     return NULL;
370ea2384d3Sbellard }
371ea2384d3Sbellard 
372b64ec4e4SFam Zheng static int bdrv_is_whitelisted(BlockDriver *drv, bool read_only)
373eb852011SMarkus Armbruster {
374b64ec4e4SFam Zheng     static const char *whitelist_rw[] = {
375b64ec4e4SFam Zheng         CONFIG_BDRV_RW_WHITELIST
376b64ec4e4SFam Zheng     };
377b64ec4e4SFam Zheng     static const char *whitelist_ro[] = {
378b64ec4e4SFam Zheng         CONFIG_BDRV_RO_WHITELIST
379eb852011SMarkus Armbruster     };
380eb852011SMarkus Armbruster     const char **p;
381eb852011SMarkus Armbruster 
382b64ec4e4SFam Zheng     if (!whitelist_rw[0] && !whitelist_ro[0]) {
383eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
384b64ec4e4SFam Zheng     }
385eb852011SMarkus Armbruster 
386b64ec4e4SFam Zheng     for (p = whitelist_rw; *p; p++) {
387eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
388eb852011SMarkus Armbruster             return 1;
389eb852011SMarkus Armbruster         }
390eb852011SMarkus Armbruster     }
391b64ec4e4SFam Zheng     if (read_only) {
392b64ec4e4SFam Zheng         for (p = whitelist_ro; *p; p++) {
393b64ec4e4SFam Zheng             if (!strcmp(drv->format_name, *p)) {
394b64ec4e4SFam Zheng                 return 1;
395b64ec4e4SFam Zheng             }
396b64ec4e4SFam Zheng         }
397b64ec4e4SFam Zheng     }
398eb852011SMarkus Armbruster     return 0;
399eb852011SMarkus Armbruster }
400eb852011SMarkus Armbruster 
401b64ec4e4SFam Zheng BlockDriver *bdrv_find_whitelisted_format(const char *format_name,
402b64ec4e4SFam Zheng                                           bool read_only)
403eb852011SMarkus Armbruster {
404eb852011SMarkus Armbruster     BlockDriver *drv = bdrv_find_format(format_name);
405b64ec4e4SFam Zheng     return drv && bdrv_is_whitelisted(drv, read_only) ? drv : NULL;
406eb852011SMarkus Armbruster }
407eb852011SMarkus Armbruster 
4085b7e1542SZhi Yong Wu typedef struct CreateCo {
4095b7e1542SZhi Yong Wu     BlockDriver *drv;
4105b7e1542SZhi Yong Wu     char *filename;
4115b7e1542SZhi Yong Wu     QEMUOptionParameter *options;
4125b7e1542SZhi Yong Wu     int ret;
413cc84d90fSMax Reitz     Error *err;
4145b7e1542SZhi Yong Wu } CreateCo;
4155b7e1542SZhi Yong Wu 
4165b7e1542SZhi Yong Wu static void coroutine_fn bdrv_create_co_entry(void *opaque)
4175b7e1542SZhi Yong Wu {
418cc84d90fSMax Reitz     Error *local_err = NULL;
419cc84d90fSMax Reitz     int ret;
420cc84d90fSMax Reitz 
4215b7e1542SZhi Yong Wu     CreateCo *cco = opaque;
4225b7e1542SZhi Yong Wu     assert(cco->drv);
4235b7e1542SZhi Yong Wu 
424cc84d90fSMax Reitz     ret = cco->drv->bdrv_create(cco->filename, cco->options, &local_err);
425cc84d90fSMax Reitz     if (error_is_set(&local_err)) {
426cc84d90fSMax Reitz         error_propagate(&cco->err, local_err);
427cc84d90fSMax Reitz     }
428cc84d90fSMax Reitz     cco->ret = ret;
4295b7e1542SZhi Yong Wu }
4305b7e1542SZhi Yong Wu 
4310e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
432cc84d90fSMax Reitz     QEMUOptionParameter *options, Error **errp)
433ea2384d3Sbellard {
4345b7e1542SZhi Yong Wu     int ret;
4350e7e1989SKevin Wolf 
4365b7e1542SZhi Yong Wu     Coroutine *co;
4375b7e1542SZhi Yong Wu     CreateCo cco = {
4385b7e1542SZhi Yong Wu         .drv = drv,
4395b7e1542SZhi Yong Wu         .filename = g_strdup(filename),
4405b7e1542SZhi Yong Wu         .options = options,
4415b7e1542SZhi Yong Wu         .ret = NOT_DONE,
442cc84d90fSMax Reitz         .err = NULL,
4435b7e1542SZhi Yong Wu     };
4445b7e1542SZhi Yong Wu 
4455b7e1542SZhi Yong Wu     if (!drv->bdrv_create) {
446cc84d90fSMax Reitz         error_setg(errp, "Driver '%s' does not support image creation", drv->format_name);
44780168bffSLuiz Capitulino         ret = -ENOTSUP;
44880168bffSLuiz Capitulino         goto out;
4495b7e1542SZhi Yong Wu     }
4505b7e1542SZhi Yong Wu 
4515b7e1542SZhi Yong Wu     if (qemu_in_coroutine()) {
4525b7e1542SZhi Yong Wu         /* Fast-path if already in coroutine context */
4535b7e1542SZhi Yong Wu         bdrv_create_co_entry(&cco);
4545b7e1542SZhi Yong Wu     } else {
4555b7e1542SZhi Yong Wu         co = qemu_coroutine_create(bdrv_create_co_entry);
4565b7e1542SZhi Yong Wu         qemu_coroutine_enter(co, &cco);
4575b7e1542SZhi Yong Wu         while (cco.ret == NOT_DONE) {
4585b7e1542SZhi Yong Wu             qemu_aio_wait();
4595b7e1542SZhi Yong Wu         }
4605b7e1542SZhi Yong Wu     }
4615b7e1542SZhi Yong Wu 
4625b7e1542SZhi Yong Wu     ret = cco.ret;
463cc84d90fSMax Reitz     if (ret < 0) {
464cc84d90fSMax Reitz         if (error_is_set(&cco.err)) {
465cc84d90fSMax Reitz             error_propagate(errp, cco.err);
466cc84d90fSMax Reitz         } else {
467cc84d90fSMax Reitz             error_setg_errno(errp, -ret, "Could not create image");
468cc84d90fSMax Reitz         }
469cc84d90fSMax Reitz     }
4705b7e1542SZhi Yong Wu 
47180168bffSLuiz Capitulino out:
47280168bffSLuiz Capitulino     g_free(cco.filename);
4735b7e1542SZhi Yong Wu     return ret;
474ea2384d3Sbellard }
475ea2384d3Sbellard 
476cc84d90fSMax Reitz int bdrv_create_file(const char* filename, QEMUOptionParameter *options,
477cc84d90fSMax Reitz                      Error **errp)
47884a12e66SChristoph Hellwig {
47984a12e66SChristoph Hellwig     BlockDriver *drv;
480cc84d90fSMax Reitz     Error *local_err = NULL;
481cc84d90fSMax Reitz     int ret;
48284a12e66SChristoph Hellwig 
48398289620SKevin Wolf     drv = bdrv_find_protocol(filename, true);
48484a12e66SChristoph Hellwig     if (drv == NULL) {
485cc84d90fSMax Reitz         error_setg(errp, "Could not find protocol for file '%s'", filename);
48616905d71SStefan Hajnoczi         return -ENOENT;
48784a12e66SChristoph Hellwig     }
48884a12e66SChristoph Hellwig 
489cc84d90fSMax Reitz     ret = bdrv_create(drv, filename, options, &local_err);
490cc84d90fSMax Reitz     if (error_is_set(&local_err)) {
491cc84d90fSMax Reitz         error_propagate(errp, local_err);
492cc84d90fSMax Reitz     }
493cc84d90fSMax Reitz     return ret;
49484a12e66SChristoph Hellwig }
49584a12e66SChristoph Hellwig 
496355ef4acSKevin Wolf int bdrv_refresh_limits(BlockDriverState *bs)
497d34682cdSKevin Wolf {
498d34682cdSKevin Wolf     BlockDriver *drv = bs->drv;
499d34682cdSKevin Wolf 
500d34682cdSKevin Wolf     memset(&bs->bl, 0, sizeof(bs->bl));
501d34682cdSKevin Wolf 
502466ad822SKevin Wolf     if (!drv) {
503466ad822SKevin Wolf         return 0;
504466ad822SKevin Wolf     }
505466ad822SKevin Wolf 
506466ad822SKevin Wolf     /* Take some limits from the children as a default */
507466ad822SKevin Wolf     if (bs->file) {
508466ad822SKevin Wolf         bdrv_refresh_limits(bs->file);
509466ad822SKevin Wolf         bs->bl.opt_transfer_length = bs->file->bl.opt_transfer_length;
510339064d5SKevin Wolf         bs->bl.opt_mem_alignment = bs->file->bl.opt_mem_alignment;
511339064d5SKevin Wolf     } else {
512339064d5SKevin Wolf         bs->bl.opt_mem_alignment = 512;
513466ad822SKevin Wolf     }
514466ad822SKevin Wolf 
515466ad822SKevin Wolf     if (bs->backing_hd) {
516466ad822SKevin Wolf         bdrv_refresh_limits(bs->backing_hd);
517466ad822SKevin Wolf         bs->bl.opt_transfer_length =
518466ad822SKevin Wolf             MAX(bs->bl.opt_transfer_length,
519466ad822SKevin Wolf                 bs->backing_hd->bl.opt_transfer_length);
520339064d5SKevin Wolf         bs->bl.opt_mem_alignment =
521339064d5SKevin Wolf             MAX(bs->bl.opt_mem_alignment,
522339064d5SKevin Wolf                 bs->backing_hd->bl.opt_mem_alignment);
523466ad822SKevin Wolf     }
524466ad822SKevin Wolf 
525466ad822SKevin Wolf     /* Then let the driver override it */
526466ad822SKevin Wolf     if (drv->bdrv_refresh_limits) {
527d34682cdSKevin Wolf         return drv->bdrv_refresh_limits(bs);
528d34682cdSKevin Wolf     }
529d34682cdSKevin Wolf 
530d34682cdSKevin Wolf     return 0;
531d34682cdSKevin Wolf }
532d34682cdSKevin Wolf 
533eba25057SJim Meyering /*
534eba25057SJim Meyering  * Create a uniquely-named empty temporary file.
535eba25057SJim Meyering  * Return 0 upon success, otherwise a negative errno value.
536eba25057SJim Meyering  */
537eba25057SJim Meyering int get_tmp_filename(char *filename, int size)
538eba25057SJim Meyering {
539d5249393Sbellard #ifdef _WIN32
5403b9f94e1Sbellard     char temp_dir[MAX_PATH];
541eba25057SJim Meyering     /* GetTempFileName requires that its output buffer (4th param)
542eba25057SJim Meyering        have length MAX_PATH or greater.  */
543eba25057SJim Meyering     assert(size >= MAX_PATH);
544eba25057SJim Meyering     return (GetTempPath(MAX_PATH, temp_dir)
545eba25057SJim Meyering             && GetTempFileName(temp_dir, "qem", 0, filename)
546eba25057SJim Meyering             ? 0 : -GetLastError());
547d5249393Sbellard #else
548ea2384d3Sbellard     int fd;
5497ccfb2ebSblueswir1     const char *tmpdir;
5500badc1eeSaurel32     tmpdir = getenv("TMPDIR");
5510badc1eeSaurel32     if (!tmpdir)
5520badc1eeSaurel32         tmpdir = "/tmp";
553eba25057SJim Meyering     if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) {
554eba25057SJim Meyering         return -EOVERFLOW;
555ea2384d3Sbellard     }
556eba25057SJim Meyering     fd = mkstemp(filename);
557fe235a06SDunrong Huang     if (fd < 0) {
558fe235a06SDunrong Huang         return -errno;
559fe235a06SDunrong Huang     }
560fe235a06SDunrong Huang     if (close(fd) != 0) {
561fe235a06SDunrong Huang         unlink(filename);
562eba25057SJim Meyering         return -errno;
563eba25057SJim Meyering     }
564eba25057SJim Meyering     return 0;
565d5249393Sbellard #endif
566eba25057SJim Meyering }
567ea2384d3Sbellard 
568f3a5d3f8SChristoph Hellwig /*
569f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
570f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
571f3a5d3f8SChristoph Hellwig  */
572f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
573f3a5d3f8SChristoph Hellwig {
574508c7cb3SChristoph Hellwig     int score_max = 0, score;
575508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
576f3a5d3f8SChristoph Hellwig 
5778a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
578508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
579508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
580508c7cb3SChristoph Hellwig             if (score > score_max) {
581508c7cb3SChristoph Hellwig                 score_max = score;
582508c7cb3SChristoph Hellwig                 drv = d;
583f3a5d3f8SChristoph Hellwig             }
584508c7cb3SChristoph Hellwig         }
585f3a5d3f8SChristoph Hellwig     }
586f3a5d3f8SChristoph Hellwig 
587508c7cb3SChristoph Hellwig     return drv;
588f3a5d3f8SChristoph Hellwig }
589f3a5d3f8SChristoph Hellwig 
59098289620SKevin Wolf BlockDriver *bdrv_find_protocol(const char *filename,
59198289620SKevin Wolf                                 bool allow_protocol_prefix)
59284a12e66SChristoph Hellwig {
59384a12e66SChristoph Hellwig     BlockDriver *drv1;
59484a12e66SChristoph Hellwig     char protocol[128];
59584a12e66SChristoph Hellwig     int len;
59684a12e66SChristoph Hellwig     const char *p;
59784a12e66SChristoph Hellwig 
59866f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
59966f82ceeSKevin Wolf 
60039508e7aSChristoph Hellwig     /*
60139508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
60239508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
60339508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
60439508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
60539508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
60639508e7aSChristoph Hellwig      */
60784a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
60839508e7aSChristoph Hellwig     if (drv1) {
60984a12e66SChristoph Hellwig         return drv1;
61084a12e66SChristoph Hellwig     }
61139508e7aSChristoph Hellwig 
61298289620SKevin Wolf     if (!path_has_protocol(filename) || !allow_protocol_prefix) {
61339508e7aSChristoph Hellwig         return bdrv_find_format("file");
61439508e7aSChristoph Hellwig     }
61598289620SKevin Wolf 
6169e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
6179e0b22f4SStefan Hajnoczi     assert(p != NULL);
61884a12e66SChristoph Hellwig     len = p - filename;
61984a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
62084a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
62184a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
62284a12e66SChristoph Hellwig     protocol[len] = '\0';
62384a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
62484a12e66SChristoph Hellwig         if (drv1->protocol_name &&
62584a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
62684a12e66SChristoph Hellwig             return drv1;
62784a12e66SChristoph Hellwig         }
62884a12e66SChristoph Hellwig     }
62984a12e66SChristoph Hellwig     return NULL;
63084a12e66SChristoph Hellwig }
63184a12e66SChristoph Hellwig 
632f500a6d3SKevin Wolf static int find_image_format(BlockDriverState *bs, const char *filename,
63334b5d2c6SMax Reitz                              BlockDriver **pdrv, Error **errp)
634ea2384d3Sbellard {
635f500a6d3SKevin Wolf     int score, score_max;
636ea2384d3Sbellard     BlockDriver *drv1, *drv;
63783f64091Sbellard     uint8_t buf[2048];
638f500a6d3SKevin Wolf     int ret = 0;
639f8ea0b00SNicholas Bellinger 
64008a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
6418e895599SPaolo Bonzini     if (bs->sg || !bdrv_is_inserted(bs) || bdrv_getlength(bs) == 0) {
642c98ac35dSStefan Weil         drv = bdrv_find_format("raw");
643c98ac35dSStefan Weil         if (!drv) {
64434b5d2c6SMax Reitz             error_setg(errp, "Could not find raw image format");
645c98ac35dSStefan Weil             ret = -ENOENT;
646c98ac35dSStefan Weil         }
647c98ac35dSStefan Weil         *pdrv = drv;
648c98ac35dSStefan Weil         return ret;
6491a396859SNicholas A. Bellinger     }
650f8ea0b00SNicholas Bellinger 
65183f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
652ea2384d3Sbellard     if (ret < 0) {
65334b5d2c6SMax Reitz         error_setg_errno(errp, -ret, "Could not read image for determining its "
65434b5d2c6SMax Reitz                          "format");
655c98ac35dSStefan Weil         *pdrv = NULL;
656c98ac35dSStefan Weil         return ret;
657ea2384d3Sbellard     }
658ea2384d3Sbellard 
659ea2384d3Sbellard     score_max = 0;
66084a12e66SChristoph Hellwig     drv = NULL;
6618a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
66283f64091Sbellard         if (drv1->bdrv_probe) {
663ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
664ea2384d3Sbellard             if (score > score_max) {
665ea2384d3Sbellard                 score_max = score;
666ea2384d3Sbellard                 drv = drv1;
667ea2384d3Sbellard             }
668ea2384d3Sbellard         }
66983f64091Sbellard     }
670c98ac35dSStefan Weil     if (!drv) {
67134b5d2c6SMax Reitz         error_setg(errp, "Could not determine image format: No compatible "
67234b5d2c6SMax Reitz                    "driver found");
673c98ac35dSStefan Weil         ret = -ENOENT;
674c98ac35dSStefan Weil     }
675c98ac35dSStefan Weil     *pdrv = drv;
676c98ac35dSStefan Weil     return ret;
677ea2384d3Sbellard }
678ea2384d3Sbellard 
67951762288SStefan Hajnoczi /**
68051762288SStefan Hajnoczi  * Set the current 'total_sectors' value
68151762288SStefan Hajnoczi  */
68251762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
68351762288SStefan Hajnoczi {
68451762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
68551762288SStefan Hajnoczi 
686396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
687396759adSNicholas Bellinger     if (bs->sg)
688396759adSNicholas Bellinger         return 0;
689396759adSNicholas Bellinger 
69051762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
69151762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
69251762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
69351762288SStefan Hajnoczi         if (length < 0) {
69451762288SStefan Hajnoczi             return length;
69551762288SStefan Hajnoczi         }
6967e382003SFam Zheng         hint = DIV_ROUND_UP(length, BDRV_SECTOR_SIZE);
69751762288SStefan Hajnoczi     }
69851762288SStefan Hajnoczi 
69951762288SStefan Hajnoczi     bs->total_sectors = hint;
70051762288SStefan Hajnoczi     return 0;
70151762288SStefan Hajnoczi }
70251762288SStefan Hajnoczi 
703c3993cdcSStefan Hajnoczi /**
7049e8f1835SPaolo Bonzini  * Set open flags for a given discard mode
7059e8f1835SPaolo Bonzini  *
7069e8f1835SPaolo Bonzini  * Return 0 on success, -1 if the discard mode was invalid.
7079e8f1835SPaolo Bonzini  */
7089e8f1835SPaolo Bonzini int bdrv_parse_discard_flags(const char *mode, int *flags)
7099e8f1835SPaolo Bonzini {
7109e8f1835SPaolo Bonzini     *flags &= ~BDRV_O_UNMAP;
7119e8f1835SPaolo Bonzini 
7129e8f1835SPaolo Bonzini     if (!strcmp(mode, "off") || !strcmp(mode, "ignore")) {
7139e8f1835SPaolo Bonzini         /* do nothing */
7149e8f1835SPaolo Bonzini     } else if (!strcmp(mode, "on") || !strcmp(mode, "unmap")) {
7159e8f1835SPaolo Bonzini         *flags |= BDRV_O_UNMAP;
7169e8f1835SPaolo Bonzini     } else {
7179e8f1835SPaolo Bonzini         return -1;
7189e8f1835SPaolo Bonzini     }
7199e8f1835SPaolo Bonzini 
7209e8f1835SPaolo Bonzini     return 0;
7219e8f1835SPaolo Bonzini }
7229e8f1835SPaolo Bonzini 
7239e8f1835SPaolo Bonzini /**
724c3993cdcSStefan Hajnoczi  * Set open flags for a given cache mode
725c3993cdcSStefan Hajnoczi  *
726c3993cdcSStefan Hajnoczi  * Return 0 on success, -1 if the cache mode was invalid.
727c3993cdcSStefan Hajnoczi  */
728c3993cdcSStefan Hajnoczi int bdrv_parse_cache_flags(const char *mode, int *flags)
729c3993cdcSStefan Hajnoczi {
730c3993cdcSStefan Hajnoczi     *flags &= ~BDRV_O_CACHE_MASK;
731c3993cdcSStefan Hajnoczi 
732c3993cdcSStefan Hajnoczi     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
733c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
73492196b2fSStefan Hajnoczi     } else if (!strcmp(mode, "directsync")) {
73592196b2fSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE;
736c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writeback")) {
737c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
738c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "unsafe")) {
739c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
740c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NO_FLUSH;
741c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writethrough")) {
742c3993cdcSStefan Hajnoczi         /* this is the default */
743c3993cdcSStefan Hajnoczi     } else {
744c3993cdcSStefan Hajnoczi         return -1;
745c3993cdcSStefan Hajnoczi     }
746c3993cdcSStefan Hajnoczi 
747c3993cdcSStefan Hajnoczi     return 0;
748c3993cdcSStefan Hajnoczi }
749c3993cdcSStefan Hajnoczi 
75053fec9d3SStefan Hajnoczi /**
75153fec9d3SStefan Hajnoczi  * The copy-on-read flag is actually a reference count so multiple users may
75253fec9d3SStefan Hajnoczi  * use the feature without worrying about clobbering its previous state.
75353fec9d3SStefan Hajnoczi  * Copy-on-read stays enabled until all users have called to disable it.
75453fec9d3SStefan Hajnoczi  */
75553fec9d3SStefan Hajnoczi void bdrv_enable_copy_on_read(BlockDriverState *bs)
75653fec9d3SStefan Hajnoczi {
75753fec9d3SStefan Hajnoczi     bs->copy_on_read++;
75853fec9d3SStefan Hajnoczi }
75953fec9d3SStefan Hajnoczi 
76053fec9d3SStefan Hajnoczi void bdrv_disable_copy_on_read(BlockDriverState *bs)
76153fec9d3SStefan Hajnoczi {
76253fec9d3SStefan Hajnoczi     assert(bs->copy_on_read > 0);
76353fec9d3SStefan Hajnoczi     bs->copy_on_read--;
76453fec9d3SStefan Hajnoczi }
76553fec9d3SStefan Hajnoczi 
7667b272452SKevin Wolf static int bdrv_open_flags(BlockDriverState *bs, int flags)
7677b272452SKevin Wolf {
7687b272452SKevin Wolf     int open_flags = flags | BDRV_O_CACHE_WB;
7697b272452SKevin Wolf 
7707b272452SKevin Wolf     /*
7717b272452SKevin Wolf      * Clear flags that are internal to the block layer before opening the
7727b272452SKevin Wolf      * image.
7737b272452SKevin Wolf      */
7747b272452SKevin Wolf     open_flags &= ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
7757b272452SKevin Wolf 
7767b272452SKevin Wolf     /*
7777b272452SKevin Wolf      * Snapshots should be writable.
7787b272452SKevin Wolf      */
7797b272452SKevin Wolf     if (bs->is_temporary) {
7807b272452SKevin Wolf         open_flags |= BDRV_O_RDWR;
7817b272452SKevin Wolf     }
7827b272452SKevin Wolf 
7837b272452SKevin Wolf     return open_flags;
7847b272452SKevin Wolf }
7857b272452SKevin Wolf 
7866913c0c2SBenoît Canet static int bdrv_assign_node_name(BlockDriverState *bs,
7876913c0c2SBenoît Canet                                  const char *node_name,
7886913c0c2SBenoît Canet                                  Error **errp)
7896913c0c2SBenoît Canet {
7906913c0c2SBenoît Canet     if (!node_name) {
7916913c0c2SBenoît Canet         return 0;
7926913c0c2SBenoît Canet     }
7936913c0c2SBenoît Canet 
7946913c0c2SBenoît Canet     /* empty string node name is invalid */
7956913c0c2SBenoît Canet     if (node_name[0] == '\0') {
7966913c0c2SBenoît Canet         error_setg(errp, "Empty node name");
7976913c0c2SBenoît Canet         return -EINVAL;
7986913c0c2SBenoît Canet     }
7996913c0c2SBenoît Canet 
8006913c0c2SBenoît Canet     /* takes care of avoiding duplicates node names */
8016913c0c2SBenoît Canet     if (bdrv_find_node(node_name)) {
8026913c0c2SBenoît Canet         error_setg(errp, "Duplicate node name");
8036913c0c2SBenoît Canet         return -EINVAL;
8046913c0c2SBenoît Canet     }
8056913c0c2SBenoît Canet 
8066913c0c2SBenoît Canet     /* copy node name into the bs and insert it into the graph list */
8076913c0c2SBenoît Canet     pstrcpy(bs->node_name, sizeof(bs->node_name), node_name);
8086913c0c2SBenoît Canet     QTAILQ_INSERT_TAIL(&graph_bdrv_states, bs, node_list);
8096913c0c2SBenoît Canet 
8106913c0c2SBenoît Canet     return 0;
8116913c0c2SBenoît Canet }
8126913c0c2SBenoît Canet 
813b6ce07aaSKevin Wolf /*
81457915332SKevin Wolf  * Common part for opening disk images and files
815b6ad491aSKevin Wolf  *
816b6ad491aSKevin Wolf  * Removes all processed options from *options.
81757915332SKevin Wolf  */
818f500a6d3SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, BlockDriverState *file,
81934b5d2c6SMax Reitz     QDict *options, int flags, BlockDriver *drv, Error **errp)
82057915332SKevin Wolf {
82157915332SKevin Wolf     int ret, open_flags;
822035fccdfSKevin Wolf     const char *filename;
8236913c0c2SBenoît Canet     const char *node_name = NULL;
82434b5d2c6SMax Reitz     Error *local_err = NULL;
82557915332SKevin Wolf 
82657915332SKevin Wolf     assert(drv != NULL);
8276405875cSPaolo Bonzini     assert(bs->file == NULL);
828707ff828SKevin Wolf     assert(options != NULL && bs->options != options);
82957915332SKevin Wolf 
83045673671SKevin Wolf     if (file != NULL) {
83145673671SKevin Wolf         filename = file->filename;
83245673671SKevin Wolf     } else {
83345673671SKevin Wolf         filename = qdict_get_try_str(options, "filename");
83445673671SKevin Wolf     }
83545673671SKevin Wolf 
83645673671SKevin Wolf     trace_bdrv_open_common(bs, filename ?: "", flags, drv->format_name);
83728dcee10SStefan Hajnoczi 
8386913c0c2SBenoît Canet     node_name = qdict_get_try_str(options, "node-name");
8396913c0c2SBenoît Canet     ret = bdrv_assign_node_name(bs, node_name, errp);
8406913c0c2SBenoît Canet     if (ret < 0) {
8416913c0c2SBenoît Canet         return ret;
8426913c0c2SBenoît Canet     }
8436913c0c2SBenoît Canet     qdict_del(options, "node-name");
8446913c0c2SBenoît Canet 
8455d186eb0SKevin Wolf     /* bdrv_open() with directly using a protocol as drv. This layer is already
8465d186eb0SKevin Wolf      * opened, so assign it to bs (while file becomes a closed BlockDriverState)
8475d186eb0SKevin Wolf      * and return immediately. */
8485d186eb0SKevin Wolf     if (file != NULL && drv->bdrv_file_open) {
8495d186eb0SKevin Wolf         bdrv_swap(file, bs);
8505d186eb0SKevin Wolf         return 0;
8515d186eb0SKevin Wolf     }
8525d186eb0SKevin Wolf 
85357915332SKevin Wolf     bs->open_flags = flags;
8541b7fd729SPaolo Bonzini     bs->guest_block_size = 512;
855c25f53b0SPaolo Bonzini     bs->request_alignment = 512;
8560d51b4deSAsias He     bs->zero_beyond_eof = true;
857b64ec4e4SFam Zheng     open_flags = bdrv_open_flags(bs, flags);
858b64ec4e4SFam Zheng     bs->read_only = !(open_flags & BDRV_O_RDWR);
859b64ec4e4SFam Zheng 
860b64ec4e4SFam Zheng     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv, bs->read_only)) {
8618f94a6e4SKevin Wolf         error_setg(errp,
8628f94a6e4SKevin Wolf                    !bs->read_only && bdrv_is_whitelisted(drv, true)
8638f94a6e4SKevin Wolf                         ? "Driver '%s' can only be used for read-only devices"
8648f94a6e4SKevin Wolf                         : "Driver '%s' is not whitelisted",
8658f94a6e4SKevin Wolf                    drv->format_name);
866b64ec4e4SFam Zheng         return -ENOTSUP;
867b64ec4e4SFam Zheng     }
86857915332SKevin Wolf 
86953fec9d3SStefan Hajnoczi     assert(bs->copy_on_read == 0); /* bdrv_new() and bdrv_close() make it so */
8700ebd24e0SKevin Wolf     if (flags & BDRV_O_COPY_ON_READ) {
8710ebd24e0SKevin Wolf         if (!bs->read_only) {
87253fec9d3SStefan Hajnoczi             bdrv_enable_copy_on_read(bs);
8730ebd24e0SKevin Wolf         } else {
8740ebd24e0SKevin Wolf             error_setg(errp, "Can't use copy-on-read on read-only device");
8750ebd24e0SKevin Wolf             return -EINVAL;
8760ebd24e0SKevin Wolf         }
87753fec9d3SStefan Hajnoczi     }
87853fec9d3SStefan Hajnoczi 
879c2ad1b0cSKevin Wolf     if (filename != NULL) {
88057915332SKevin Wolf         pstrcpy(bs->filename, sizeof(bs->filename), filename);
881c2ad1b0cSKevin Wolf     } else {
882c2ad1b0cSKevin Wolf         bs->filename[0] = '\0';
883c2ad1b0cSKevin Wolf     }
88457915332SKevin Wolf 
88557915332SKevin Wolf     bs->drv = drv;
8867267c094SAnthony Liguori     bs->opaque = g_malloc0(drv->instance_size);
88757915332SKevin Wolf 
88803f541bdSStefan Hajnoczi     bs->enable_write_cache = !!(flags & BDRV_O_CACHE_WB);
889e7c63796SStefan Hajnoczi 
89066f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
89166f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
8925d186eb0SKevin Wolf         assert(file == NULL);
893030be321SBenoît Canet         assert(!drv->bdrv_needs_filename || filename != NULL);
89434b5d2c6SMax Reitz         ret = drv->bdrv_file_open(bs, options, open_flags, &local_err);
895f500a6d3SKevin Wolf     } else {
8962af5ef70SKevin Wolf         if (file == NULL) {
89734b5d2c6SMax Reitz             error_setg(errp, "Can't use '%s' as a block driver for the "
89834b5d2c6SMax Reitz                        "protocol level", drv->format_name);
8992af5ef70SKevin Wolf             ret = -EINVAL;
9002af5ef70SKevin Wolf             goto free_and_fail;
9012af5ef70SKevin Wolf         }
902f500a6d3SKevin Wolf         bs->file = file;
90334b5d2c6SMax Reitz         ret = drv->bdrv_open(bs, options, open_flags, &local_err);
90466f82ceeSKevin Wolf     }
90566f82ceeSKevin Wolf 
90657915332SKevin Wolf     if (ret < 0) {
90734b5d2c6SMax Reitz         if (error_is_set(&local_err)) {
90834b5d2c6SMax Reitz             error_propagate(errp, local_err);
9092fa9aa59SDunrong Huang         } else if (bs->filename[0]) {
9102fa9aa59SDunrong Huang             error_setg_errno(errp, -ret, "Could not open '%s'", bs->filename);
91134b5d2c6SMax Reitz         } else {
91234b5d2c6SMax Reitz             error_setg_errno(errp, -ret, "Could not open image");
91334b5d2c6SMax Reitz         }
91457915332SKevin Wolf         goto free_and_fail;
91557915332SKevin Wolf     }
91657915332SKevin Wolf 
91751762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
91851762288SStefan Hajnoczi     if (ret < 0) {
91934b5d2c6SMax Reitz         error_setg_errno(errp, -ret, "Could not refresh total sector count");
92051762288SStefan Hajnoczi         goto free_and_fail;
92157915332SKevin Wolf     }
92251762288SStefan Hajnoczi 
923d34682cdSKevin Wolf     bdrv_refresh_limits(bs);
924c25f53b0SPaolo Bonzini     assert(bdrv_opt_mem_align(bs) != 0);
925c25f53b0SPaolo Bonzini     assert(bs->request_alignment != 0);
926d34682cdSKevin Wolf 
92757915332SKevin Wolf #ifndef _WIN32
92857915332SKevin Wolf     if (bs->is_temporary) {
929d4cea8dfSDunrong Huang         assert(bs->filename[0] != '\0');
930d4cea8dfSDunrong Huang         unlink(bs->filename);
93157915332SKevin Wolf     }
93257915332SKevin Wolf #endif
93357915332SKevin Wolf     return 0;
93457915332SKevin Wolf 
93557915332SKevin Wolf free_and_fail:
93666f82ceeSKevin Wolf     bs->file = NULL;
9377267c094SAnthony Liguori     g_free(bs->opaque);
93857915332SKevin Wolf     bs->opaque = NULL;
93957915332SKevin Wolf     bs->drv = NULL;
94057915332SKevin Wolf     return ret;
94157915332SKevin Wolf }
94257915332SKevin Wolf 
94357915332SKevin Wolf /*
944b6ce07aaSKevin Wolf  * Opens a file using a protocol (file, host_device, nbd, ...)
945787e4a85SKevin Wolf  *
946787e4a85SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
947787e4a85SKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
948787e4a85SKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
949787e4a85SKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_file_open.
950b6ce07aaSKevin Wolf  */
951787e4a85SKevin Wolf int bdrv_file_open(BlockDriverState **pbs, const char *filename,
95272daa72eSMax Reitz                    const char *reference, QDict *options, int flags,
95372daa72eSMax Reitz                    Error **errp)
954b338082bSbellard {
95572daa72eSMax Reitz     BlockDriverState *bs = NULL;
9566db95603SChristoph Hellwig     BlockDriver *drv;
957c2ad1b0cSKevin Wolf     const char *drvname;
95898289620SKevin Wolf     bool allow_protocol_prefix = false;
95934b5d2c6SMax Reitz     Error *local_err = NULL;
96083f64091Sbellard     int ret;
9613b0d4f61Sbellard 
962707ff828SKevin Wolf     /* NULL means an empty set of options */
963707ff828SKevin Wolf     if (options == NULL) {
964707ff828SKevin Wolf         options = qdict_new();
9653b0d4f61Sbellard     }
966707ff828SKevin Wolf 
96772daa72eSMax Reitz     if (reference) {
96872daa72eSMax Reitz         if (filename || qdict_size(options)) {
96972daa72eSMax Reitz             error_setg(errp, "Cannot reference an existing block device with "
97072daa72eSMax Reitz                        "additional options or a new filename");
97172daa72eSMax Reitz             return -EINVAL;
97272daa72eSMax Reitz         }
97372daa72eSMax Reitz         QDECREF(options);
97472daa72eSMax Reitz 
97572daa72eSMax Reitz         bs = bdrv_find(reference);
97672daa72eSMax Reitz         if (!bs) {
97772daa72eSMax Reitz             error_setg(errp, "Cannot find block device '%s'", reference);
97872daa72eSMax Reitz             return -ENODEV;
97972daa72eSMax Reitz         }
98072daa72eSMax Reitz         bdrv_ref(bs);
98172daa72eSMax Reitz         *pbs = bs;
98272daa72eSMax Reitz         return 0;
98372daa72eSMax Reitz     }
98472daa72eSMax Reitz 
985707ff828SKevin Wolf     bs = bdrv_new("");
986707ff828SKevin Wolf     bs->options = options;
987707ff828SKevin Wolf     options = qdict_clone_shallow(options);
988707ff828SKevin Wolf 
989035fccdfSKevin Wolf     /* Fetch the file name from the options QDict if necessary */
990035fccdfSKevin Wolf     if (!filename) {
991035fccdfSKevin Wolf         filename = qdict_get_try_str(options, "filename");
992035fccdfSKevin Wolf     } else if (filename && !qdict_haskey(options, "filename")) {
993035fccdfSKevin Wolf         qdict_put(options, "filename", qstring_from_str(filename));
99498289620SKevin Wolf         allow_protocol_prefix = true;
995035fccdfSKevin Wolf     } else {
99634b5d2c6SMax Reitz         error_setg(errp, "Can't specify 'file' and 'filename' options at the "
99734b5d2c6SMax Reitz                    "same time");
998035fccdfSKevin Wolf         ret = -EINVAL;
999035fccdfSKevin Wolf         goto fail;
1000035fccdfSKevin Wolf     }
1001035fccdfSKevin Wolf 
1002c2ad1b0cSKevin Wolf     /* Find the right block driver */
1003c2ad1b0cSKevin Wolf     drvname = qdict_get_try_str(options, "driver");
1004c2ad1b0cSKevin Wolf     if (drvname) {
10058f94a6e4SKevin Wolf         drv = bdrv_find_format(drvname);
100634b5d2c6SMax Reitz         if (!drv) {
100734b5d2c6SMax Reitz             error_setg(errp, "Unknown driver '%s'", drvname);
100834b5d2c6SMax Reitz         }
1009c2ad1b0cSKevin Wolf         qdict_del(options, "driver");
1010c2ad1b0cSKevin Wolf     } else if (filename) {
101198289620SKevin Wolf         drv = bdrv_find_protocol(filename, allow_protocol_prefix);
101298289620SKevin Wolf         if (!drv) {
101334b5d2c6SMax Reitz             error_setg(errp, "Unknown protocol");
101498289620SKevin Wolf         }
1015c2ad1b0cSKevin Wolf     } else {
101634b5d2c6SMax Reitz         error_setg(errp, "Must specify either driver or file");
1017c2ad1b0cSKevin Wolf         drv = NULL;
1018c2ad1b0cSKevin Wolf     }
1019c2ad1b0cSKevin Wolf 
1020c2ad1b0cSKevin Wolf     if (!drv) {
102134b5d2c6SMax Reitz         /* errp has been set already */
1022c2ad1b0cSKevin Wolf         ret = -ENOENT;
1023c2ad1b0cSKevin Wolf         goto fail;
1024c2ad1b0cSKevin Wolf     }
1025c2ad1b0cSKevin Wolf 
1026c2ad1b0cSKevin Wolf     /* Parse the filename and open it */
1027c2ad1b0cSKevin Wolf     if (drv->bdrv_parse_filename && filename) {
10286963a30dSKevin Wolf         drv->bdrv_parse_filename(filename, options, &local_err);
10296963a30dSKevin Wolf         if (error_is_set(&local_err)) {
103034b5d2c6SMax Reitz             error_propagate(errp, local_err);
10316963a30dSKevin Wolf             ret = -EINVAL;
10326963a30dSKevin Wolf             goto fail;
10336963a30dSKevin Wolf         }
103456d1b4d2SKevin Wolf         qdict_del(options, "filename");
1035030be321SBenoît Canet     } else if (drv->bdrv_needs_filename && !filename) {
103634b5d2c6SMax Reitz         error_setg(errp, "The '%s' block driver requires a file name",
1037c2ad1b0cSKevin Wolf                    drv->format_name);
1038c2ad1b0cSKevin Wolf         ret = -EINVAL;
1039c2ad1b0cSKevin Wolf         goto fail;
10406963a30dSKevin Wolf     }
10416963a30dSKevin Wolf 
1042505d7583SMax Reitz     if (!drv->bdrv_file_open) {
1043505d7583SMax Reitz         ret = bdrv_open(bs, filename, options, flags, drv, &local_err);
1044505d7583SMax Reitz         options = NULL;
1045505d7583SMax Reitz     } else {
104634b5d2c6SMax Reitz         ret = bdrv_open_common(bs, NULL, options, flags, drv, &local_err);
1047505d7583SMax Reitz     }
1048707ff828SKevin Wolf     if (ret < 0) {
104934b5d2c6SMax Reitz         error_propagate(errp, local_err);
1050707ff828SKevin Wolf         goto fail;
1051707ff828SKevin Wolf     }
1052707ff828SKevin Wolf 
1053707ff828SKevin Wolf     /* Check if any unknown options were used */
1054505d7583SMax Reitz     if (options && (qdict_size(options) != 0)) {
1055707ff828SKevin Wolf         const QDictEntry *entry = qdict_first(options);
105634b5d2c6SMax Reitz         error_setg(errp, "Block protocol '%s' doesn't support the option '%s'",
1057707ff828SKevin Wolf                    drv->format_name, entry->key);
1058707ff828SKevin Wolf         ret = -EINVAL;
1059707ff828SKevin Wolf         goto fail;
1060707ff828SKevin Wolf     }
1061707ff828SKevin Wolf     QDECREF(options);
1062707ff828SKevin Wolf 
106371d0770cSaliguori     bs->growable = 1;
106483f64091Sbellard     *pbs = bs;
106583f64091Sbellard     return 0;
1066707ff828SKevin Wolf 
1067707ff828SKevin Wolf fail:
1068707ff828SKevin Wolf     QDECREF(options);
1069707ff828SKevin Wolf     if (!bs->drv) {
1070707ff828SKevin Wolf         QDECREF(bs->options);
1071707ff828SKevin Wolf     }
10724f6fd349SFam Zheng     bdrv_unref(bs);
1073707ff828SKevin Wolf     return ret;
10743b0d4f61Sbellard }
10753b0d4f61Sbellard 
107631ca6d07SKevin Wolf /*
107731ca6d07SKevin Wolf  * Opens the backing file for a BlockDriverState if not yet open
107831ca6d07SKevin Wolf  *
107931ca6d07SKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
108031ca6d07SKevin Wolf  * empty set of options. The reference to the QDict is transferred to this
108131ca6d07SKevin Wolf  * function (even on failure), so if the caller intends to reuse the dictionary,
108231ca6d07SKevin Wolf  * it needs to use QINCREF() before calling bdrv_file_open.
108331ca6d07SKevin Wolf  */
108434b5d2c6SMax Reitz int bdrv_open_backing_file(BlockDriverState *bs, QDict *options, Error **errp)
10859156df12SPaolo Bonzini {
10869156df12SPaolo Bonzini     char backing_filename[PATH_MAX];
10879156df12SPaolo Bonzini     int back_flags, ret;
10889156df12SPaolo Bonzini     BlockDriver *back_drv = NULL;
108934b5d2c6SMax Reitz     Error *local_err = NULL;
10909156df12SPaolo Bonzini 
10919156df12SPaolo Bonzini     if (bs->backing_hd != NULL) {
109231ca6d07SKevin Wolf         QDECREF(options);
10939156df12SPaolo Bonzini         return 0;
10949156df12SPaolo Bonzini     }
10959156df12SPaolo Bonzini 
109631ca6d07SKevin Wolf     /* NULL means an empty set of options */
109731ca6d07SKevin Wolf     if (options == NULL) {
109831ca6d07SKevin Wolf         options = qdict_new();
109931ca6d07SKevin Wolf     }
110031ca6d07SKevin Wolf 
11019156df12SPaolo Bonzini     bs->open_flags &= ~BDRV_O_NO_BACKING;
11021cb6f506SKevin Wolf     if (qdict_haskey(options, "file.filename")) {
11031cb6f506SKevin Wolf         backing_filename[0] = '\0';
11041cb6f506SKevin Wolf     } else if (bs->backing_file[0] == '\0' && qdict_size(options) == 0) {
110531ca6d07SKevin Wolf         QDECREF(options);
11069156df12SPaolo Bonzini         return 0;
1107dbecebddSFam Zheng     } else {
1108dbecebddSFam Zheng         bdrv_get_full_backing_filename(bs, backing_filename,
1109dbecebddSFam Zheng                                        sizeof(backing_filename));
11109156df12SPaolo Bonzini     }
11119156df12SPaolo Bonzini 
11129156df12SPaolo Bonzini     bs->backing_hd = bdrv_new("");
11139156df12SPaolo Bonzini 
11149156df12SPaolo Bonzini     if (bs->backing_format[0] != '\0') {
11159156df12SPaolo Bonzini         back_drv = bdrv_find_format(bs->backing_format);
11169156df12SPaolo Bonzini     }
11179156df12SPaolo Bonzini 
11189156df12SPaolo Bonzini     /* backing files always opened read-only */
111987a5debdSThibaut LAURENT     back_flags = bs->open_flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT |
112087a5debdSThibaut LAURENT                                     BDRV_O_COPY_ON_READ);
11219156df12SPaolo Bonzini 
112231ca6d07SKevin Wolf     ret = bdrv_open(bs->backing_hd,
112331ca6d07SKevin Wolf                     *backing_filename ? backing_filename : NULL, options,
112434b5d2c6SMax Reitz                     back_flags, back_drv, &local_err);
11259156df12SPaolo Bonzini     if (ret < 0) {
11264f6fd349SFam Zheng         bdrv_unref(bs->backing_hd);
11279156df12SPaolo Bonzini         bs->backing_hd = NULL;
11289156df12SPaolo Bonzini         bs->open_flags |= BDRV_O_NO_BACKING;
1129b04b6b6eSFam Zheng         error_setg(errp, "Could not open backing file: %s",
1130b04b6b6eSFam Zheng                    error_get_pretty(local_err));
1131b04b6b6eSFam Zheng         error_free(local_err);
11329156df12SPaolo Bonzini         return ret;
11339156df12SPaolo Bonzini     }
1134d80ac658SPeter Feiner 
1135d80ac658SPeter Feiner     if (bs->backing_hd->file) {
113661ed2684SMax Reitz         pstrcpy(bs->backing_file, sizeof(bs->backing_file),
113761ed2684SMax Reitz                 bs->backing_hd->file->filename);
1138d80ac658SPeter Feiner     }
1139d80ac658SPeter Feiner 
1140d34682cdSKevin Wolf     /* Recalculate the BlockLimits with the backing file */
1141d34682cdSKevin Wolf     bdrv_refresh_limits(bs);
1142d34682cdSKevin Wolf 
11439156df12SPaolo Bonzini     return 0;
11449156df12SPaolo Bonzini }
11459156df12SPaolo Bonzini 
1146b6ce07aaSKevin Wolf /*
1147da557aacSMax Reitz  * Opens a disk image whose options are given as BlockdevRef in another block
1148da557aacSMax Reitz  * device's options.
1149da557aacSMax Reitz  *
1150da557aacSMax Reitz  * If force_raw is true, bdrv_file_open() will be used, thereby preventing any
1151da557aacSMax Reitz  * image format auto-detection. If it is false and a filename is given,
1152da557aacSMax Reitz  * bdrv_open() will be used for auto-detection.
1153da557aacSMax Reitz  *
1154da557aacSMax Reitz  * If allow_none is true, no image will be opened if filename is false and no
1155da557aacSMax Reitz  * BlockdevRef is given. *pbs will remain unchanged and 0 will be returned.
1156da557aacSMax Reitz  *
1157da557aacSMax Reitz  * bdrev_key specifies the key for the image's BlockdevRef in the options QDict.
1158da557aacSMax Reitz  * That QDict has to be flattened; therefore, if the BlockdevRef is a QDict
1159da557aacSMax Reitz  * itself, all options starting with "${bdref_key}." are considered part of the
1160da557aacSMax Reitz  * BlockdevRef.
1161da557aacSMax Reitz  *
1162da557aacSMax Reitz  * The BlockdevRef will be removed from the options QDict.
1163da557aacSMax Reitz  */
1164da557aacSMax Reitz int bdrv_open_image(BlockDriverState **pbs, const char *filename,
1165da557aacSMax Reitz                     QDict *options, const char *bdref_key, int flags,
1166da557aacSMax Reitz                     bool force_raw, bool allow_none, Error **errp)
1167da557aacSMax Reitz {
1168da557aacSMax Reitz     QDict *image_options;
1169da557aacSMax Reitz     int ret;
1170da557aacSMax Reitz     char *bdref_key_dot;
1171da557aacSMax Reitz     const char *reference;
1172da557aacSMax Reitz 
1173da557aacSMax Reitz     bdref_key_dot = g_strdup_printf("%s.", bdref_key);
1174da557aacSMax Reitz     qdict_extract_subqdict(options, &image_options, bdref_key_dot);
1175da557aacSMax Reitz     g_free(bdref_key_dot);
1176da557aacSMax Reitz 
1177da557aacSMax Reitz     reference = qdict_get_try_str(options, bdref_key);
1178da557aacSMax Reitz     if (!filename && !reference && !qdict_size(image_options)) {
1179da557aacSMax Reitz         if (allow_none) {
1180da557aacSMax Reitz             ret = 0;
1181da557aacSMax Reitz         } else {
1182da557aacSMax Reitz             error_setg(errp, "A block device must be specified for \"%s\"",
1183da557aacSMax Reitz                        bdref_key);
1184da557aacSMax Reitz             ret = -EINVAL;
1185da557aacSMax Reitz         }
1186da557aacSMax Reitz         goto done;
1187da557aacSMax Reitz     }
1188da557aacSMax Reitz 
1189da557aacSMax Reitz     if (filename && !force_raw) {
1190da557aacSMax Reitz         /* If a filename is given and the block driver should be detected
1191da557aacSMax Reitz            automatically (instead of using none), use bdrv_open() in order to do
1192da557aacSMax Reitz            that auto-detection. */
1193da557aacSMax Reitz         BlockDriverState *bs;
1194da557aacSMax Reitz 
1195da557aacSMax Reitz         if (reference) {
1196da557aacSMax Reitz             error_setg(errp, "Cannot reference an existing block device while "
1197da557aacSMax Reitz                        "giving a filename");
1198da557aacSMax Reitz             ret = -EINVAL;
1199da557aacSMax Reitz             goto done;
1200da557aacSMax Reitz         }
1201da557aacSMax Reitz 
1202da557aacSMax Reitz         bs = bdrv_new("");
1203da557aacSMax Reitz         ret = bdrv_open(bs, filename, image_options, flags, NULL, errp);
1204da557aacSMax Reitz         if (ret < 0) {
1205da557aacSMax Reitz             bdrv_unref(bs);
1206da557aacSMax Reitz         } else {
1207da557aacSMax Reitz             *pbs = bs;
1208da557aacSMax Reitz         }
1209da557aacSMax Reitz     } else {
1210da557aacSMax Reitz         ret = bdrv_file_open(pbs, filename, reference, image_options, flags,
1211da557aacSMax Reitz                              errp);
1212da557aacSMax Reitz     }
1213da557aacSMax Reitz 
1214da557aacSMax Reitz done:
1215da557aacSMax Reitz     qdict_del(options, bdref_key);
1216da557aacSMax Reitz     return ret;
1217da557aacSMax Reitz }
1218da557aacSMax Reitz 
1219da557aacSMax Reitz /*
1220b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
1221de9c0cecSKevin Wolf  *
1222de9c0cecSKevin Wolf  * options is a QDict of options to pass to the block drivers, or NULL for an
1223de9c0cecSKevin Wolf  * empty set of options. The reference to the QDict belongs to the block layer
1224de9c0cecSKevin Wolf  * after the call (even on failure), so if the caller intends to reuse the
1225de9c0cecSKevin Wolf  * dictionary, it needs to use QINCREF() before calling bdrv_open.
1226b6ce07aaSKevin Wolf  */
1227de9c0cecSKevin Wolf int bdrv_open(BlockDriverState *bs, const char *filename, QDict *options,
122834b5d2c6SMax Reitz               int flags, BlockDriver *drv, Error **errp)
1229ea2384d3Sbellard {
1230b6ce07aaSKevin Wolf     int ret;
123189c9bc3dSStefan Weil     /* TODO: extra byte is a hack to ensure MAX_PATH space on Windows. */
123289c9bc3dSStefan Weil     char tmp_filename[PATH_MAX + 1];
1233f500a6d3SKevin Wolf     BlockDriverState *file = NULL;
123474fe54f2SKevin Wolf     const char *drvname;
123534b5d2c6SMax Reitz     Error *local_err = NULL;
123633e3963eSbellard 
1237de9c0cecSKevin Wolf     /* NULL means an empty set of options */
1238de9c0cecSKevin Wolf     if (options == NULL) {
1239de9c0cecSKevin Wolf         options = qdict_new();
1240de9c0cecSKevin Wolf     }
1241de9c0cecSKevin Wolf 
1242de9c0cecSKevin Wolf     bs->options = options;
1243b6ad491aSKevin Wolf     options = qdict_clone_shallow(options);
1244de9c0cecSKevin Wolf 
1245de9c0cecSKevin Wolf     /* For snapshot=on, create a temporary qcow2 overlay */
124683f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
1247ea2384d3Sbellard         BlockDriverState *bs1;
1248ea2384d3Sbellard         int64_t total_size;
124991a073a9SKevin Wolf         BlockDriver *bdrv_qcow2;
125008b392e1SKevin Wolf         QEMUOptionParameter *create_options;
12519fd3171aSKevin Wolf         QDict *snapshot_options;
1252c2ad1b0cSKevin Wolf 
1253ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
1254ea2384d3Sbellard            instead of opening 'filename' directly */
1255ea2384d3Sbellard 
12569fd3171aSKevin Wolf         /* Get the required size from the image */
1257ea2384d3Sbellard         bs1 = bdrv_new("");
12589fd3171aSKevin Wolf         QINCREF(options);
1259c9fbb99dSKevin Wolf         ret = bdrv_open(bs1, filename, options, BDRV_O_NO_BACKING,
1260c9fbb99dSKevin Wolf                         drv, &local_err);
126151d7c00cSaliguori         if (ret < 0) {
12624f6fd349SFam Zheng             bdrv_unref(bs1);
1263de9c0cecSKevin Wolf             goto fail;
1264ea2384d3Sbellard         }
12653e82990bSJes Sorensen         total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
12667c96d46eSaliguori 
12674f6fd349SFam Zheng         bdrv_unref(bs1);
1268ea2384d3Sbellard 
12699fd3171aSKevin Wolf         /* Create the temporary image */
1270eba25057SJim Meyering         ret = get_tmp_filename(tmp_filename, sizeof(tmp_filename));
1271eba25057SJim Meyering         if (ret < 0) {
127234b5d2c6SMax Reitz             error_setg_errno(errp, -ret, "Could not get temporary filename");
1273de9c0cecSKevin Wolf             goto fail;
1274eba25057SJim Meyering         }
12757c96d46eSaliguori 
127691a073a9SKevin Wolf         bdrv_qcow2 = bdrv_find_format("qcow2");
127708b392e1SKevin Wolf         create_options = parse_option_parameters("", bdrv_qcow2->create_options,
127808b392e1SKevin Wolf                                                  NULL);
127991a073a9SKevin Wolf 
128008b392e1SKevin Wolf         set_option_parameter_int(create_options, BLOCK_OPT_SIZE, total_size);
128191a073a9SKevin Wolf 
1282cc84d90fSMax Reitz         ret = bdrv_create(bdrv_qcow2, tmp_filename, create_options, &local_err);
128308b392e1SKevin Wolf         free_option_parameters(create_options);
128451d7c00cSaliguori         if (ret < 0) {
128534b5d2c6SMax Reitz             error_setg_errno(errp, -ret, "Could not create temporary overlay "
1286cc84d90fSMax Reitz                              "'%s': %s", tmp_filename,
1287cc84d90fSMax Reitz                              error_get_pretty(local_err));
1288cc84d90fSMax Reitz             error_free(local_err);
1289cc84d90fSMax Reitz             local_err = NULL;
1290de9c0cecSKevin Wolf             goto fail;
1291ea2384d3Sbellard         }
129291a073a9SKevin Wolf 
12939fd3171aSKevin Wolf         /* Prepare a new options QDict for the temporary file, where user
12949fd3171aSKevin Wolf          * options refer to the backing file */
12959fd3171aSKevin Wolf         if (filename) {
12969fd3171aSKevin Wolf             qdict_put(options, "file.filename", qstring_from_str(filename));
12979fd3171aSKevin Wolf         }
12989fd3171aSKevin Wolf         if (drv) {
12999fd3171aSKevin Wolf             qdict_put(options, "driver", qstring_from_str(drv->format_name));
13009fd3171aSKevin Wolf         }
13019fd3171aSKevin Wolf 
13029fd3171aSKevin Wolf         snapshot_options = qdict_new();
13039fd3171aSKevin Wolf         qdict_put(snapshot_options, "backing", options);
13049fd3171aSKevin Wolf         qdict_flatten(snapshot_options);
13059fd3171aSKevin Wolf 
13069fd3171aSKevin Wolf         bs->options = snapshot_options;
13079fd3171aSKevin Wolf         options = qdict_clone_shallow(bs->options);
13089fd3171aSKevin Wolf 
1309ea2384d3Sbellard         filename = tmp_filename;
131091a073a9SKevin Wolf         drv = bdrv_qcow2;
1311ea2384d3Sbellard         bs->is_temporary = 1;
1312ea2384d3Sbellard     }
1313ea2384d3Sbellard 
1314f500a6d3SKevin Wolf     /* Open image file without format layer */
1315be028adcSJeff Cody     if (flags & BDRV_O_RDWR) {
1316be028adcSJeff Cody         flags |= BDRV_O_ALLOW_RDWR;
1317be028adcSJeff Cody     }
1318be028adcSJeff Cody 
1319054963f8SMax Reitz     ret = bdrv_open_image(&file, filename, options, "file",
1320054963f8SMax Reitz                           bdrv_open_flags(bs, flags | BDRV_O_UNMAP), true, true,
13212a05cbe4SMax Reitz                           &local_err);
1322f500a6d3SKevin Wolf     if (ret < 0) {
1323de9c0cecSKevin Wolf         goto fail;
1324f500a6d3SKevin Wolf     }
1325f500a6d3SKevin Wolf 
1326f500a6d3SKevin Wolf     /* Find the right image format driver */
132774fe54f2SKevin Wolf     drvname = qdict_get_try_str(options, "driver");
132874fe54f2SKevin Wolf     if (drvname) {
13298f94a6e4SKevin Wolf         drv = bdrv_find_format(drvname);
133074fe54f2SKevin Wolf         qdict_del(options, "driver");
133106d22aa3SKevin Wolf         if (!drv) {
133206d22aa3SKevin Wolf             error_setg(errp, "Invalid driver: '%s'", drvname);
133306d22aa3SKevin Wolf             ret = -EINVAL;
133406d22aa3SKevin Wolf             goto unlink_and_fail;
133506d22aa3SKevin Wolf         }
133674fe54f2SKevin Wolf     }
133774fe54f2SKevin Wolf 
1338f500a6d3SKevin Wolf     if (!drv) {
13392a05cbe4SMax Reitz         if (file) {
134034b5d2c6SMax Reitz             ret = find_image_format(file, filename, &drv, &local_err);
13412a05cbe4SMax Reitz         } else {
13422a05cbe4SMax Reitz             error_setg(errp, "Must specify either driver or file");
13432a05cbe4SMax Reitz             ret = -EINVAL;
13442a05cbe4SMax Reitz             goto unlink_and_fail;
13452a05cbe4SMax Reitz         }
1346f500a6d3SKevin Wolf     }
1347f500a6d3SKevin Wolf 
1348f500a6d3SKevin Wolf     if (!drv) {
1349f500a6d3SKevin Wolf         goto unlink_and_fail;
1350f500a6d3SKevin Wolf     }
1351f500a6d3SKevin Wolf 
1352b6ce07aaSKevin Wolf     /* Open the image */
135334b5d2c6SMax Reitz     ret = bdrv_open_common(bs, file, options, flags, drv, &local_err);
1354b6ce07aaSKevin Wolf     if (ret < 0) {
13556987307cSChristoph Hellwig         goto unlink_and_fail;
13566987307cSChristoph Hellwig     }
13576987307cSChristoph Hellwig 
13582a05cbe4SMax Reitz     if (file && (bs->file != file)) {
13594f6fd349SFam Zheng         bdrv_unref(file);
1360f500a6d3SKevin Wolf         file = NULL;
1361f500a6d3SKevin Wolf     }
1362f500a6d3SKevin Wolf 
1363b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
13649156df12SPaolo Bonzini     if ((flags & BDRV_O_NO_BACKING) == 0) {
136531ca6d07SKevin Wolf         QDict *backing_options;
136631ca6d07SKevin Wolf 
13675726d872SBenoît Canet         qdict_extract_subqdict(options, &backing_options, "backing.");
136834b5d2c6SMax Reitz         ret = bdrv_open_backing_file(bs, backing_options, &local_err);
1369b6ce07aaSKevin Wolf         if (ret < 0) {
1370b6ad491aSKevin Wolf             goto close_and_fail;
1371b6ce07aaSKevin Wolf         }
1372b6ce07aaSKevin Wolf     }
1373b6ce07aaSKevin Wolf 
1374b6ad491aSKevin Wolf     /* Check if any unknown options were used */
1375b6ad491aSKevin Wolf     if (qdict_size(options) != 0) {
1376b6ad491aSKevin Wolf         const QDictEntry *entry = qdict_first(options);
137734b5d2c6SMax Reitz         error_setg(errp, "Block format '%s' used by device '%s' doesn't "
137834b5d2c6SMax Reitz                    "support the option '%s'", drv->format_name, bs->device_name,
137934b5d2c6SMax Reitz                    entry->key);
1380b6ad491aSKevin Wolf 
1381b6ad491aSKevin Wolf         ret = -EINVAL;
1382b6ad491aSKevin Wolf         goto close_and_fail;
1383b6ad491aSKevin Wolf     }
1384b6ad491aSKevin Wolf     QDECREF(options);
1385b6ad491aSKevin Wolf 
1386b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
13877d4b4ba5SMarkus Armbruster         bdrv_dev_change_media_cb(bs, true);
1388b6ce07aaSKevin Wolf     }
1389b6ce07aaSKevin Wolf 
1390b6ce07aaSKevin Wolf     return 0;
1391b6ce07aaSKevin Wolf 
1392b6ce07aaSKevin Wolf unlink_and_fail:
1393f500a6d3SKevin Wolf     if (file != NULL) {
13944f6fd349SFam Zheng         bdrv_unref(file);
1395f500a6d3SKevin Wolf     }
1396b6ce07aaSKevin Wolf     if (bs->is_temporary) {
1397b6ce07aaSKevin Wolf         unlink(filename);
1398b6ce07aaSKevin Wolf     }
1399de9c0cecSKevin Wolf fail:
1400de9c0cecSKevin Wolf     QDECREF(bs->options);
1401b6ad491aSKevin Wolf     QDECREF(options);
1402de9c0cecSKevin Wolf     bs->options = NULL;
140334b5d2c6SMax Reitz     if (error_is_set(&local_err)) {
140434b5d2c6SMax Reitz         error_propagate(errp, local_err);
140534b5d2c6SMax Reitz     }
1406b6ad491aSKevin Wolf     return ret;
1407de9c0cecSKevin Wolf 
1408b6ad491aSKevin Wolf close_and_fail:
1409b6ad491aSKevin Wolf     bdrv_close(bs);
1410b6ad491aSKevin Wolf     QDECREF(options);
141134b5d2c6SMax Reitz     if (error_is_set(&local_err)) {
141234b5d2c6SMax Reitz         error_propagate(errp, local_err);
141334b5d2c6SMax Reitz     }
1414b6ce07aaSKevin Wolf     return ret;
1415b6ce07aaSKevin Wolf }
1416b6ce07aaSKevin Wolf 
1417e971aa12SJeff Cody typedef struct BlockReopenQueueEntry {
1418e971aa12SJeff Cody      bool prepared;
1419e971aa12SJeff Cody      BDRVReopenState state;
1420e971aa12SJeff Cody      QSIMPLEQ_ENTRY(BlockReopenQueueEntry) entry;
1421e971aa12SJeff Cody } BlockReopenQueueEntry;
1422e971aa12SJeff Cody 
1423e971aa12SJeff Cody /*
1424e971aa12SJeff Cody  * Adds a BlockDriverState to a simple queue for an atomic, transactional
1425e971aa12SJeff Cody  * reopen of multiple devices.
1426e971aa12SJeff Cody  *
1427e971aa12SJeff Cody  * bs_queue can either be an existing BlockReopenQueue that has had QSIMPLE_INIT
1428e971aa12SJeff Cody  * already performed, or alternatively may be NULL a new BlockReopenQueue will
1429e971aa12SJeff Cody  * be created and initialized. This newly created BlockReopenQueue should be
1430e971aa12SJeff Cody  * passed back in for subsequent calls that are intended to be of the same
1431e971aa12SJeff Cody  * atomic 'set'.
1432e971aa12SJeff Cody  *
1433e971aa12SJeff Cody  * bs is the BlockDriverState to add to the reopen queue.
1434e971aa12SJeff Cody  *
1435e971aa12SJeff Cody  * flags contains the open flags for the associated bs
1436e971aa12SJeff Cody  *
1437e971aa12SJeff Cody  * returns a pointer to bs_queue, which is either the newly allocated
1438e971aa12SJeff Cody  * bs_queue, or the existing bs_queue being used.
1439e971aa12SJeff Cody  *
1440e971aa12SJeff Cody  */
1441e971aa12SJeff Cody BlockReopenQueue *bdrv_reopen_queue(BlockReopenQueue *bs_queue,
1442e971aa12SJeff Cody                                     BlockDriverState *bs, int flags)
1443e971aa12SJeff Cody {
1444e971aa12SJeff Cody     assert(bs != NULL);
1445e971aa12SJeff Cody 
1446e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry;
1447e971aa12SJeff Cody     if (bs_queue == NULL) {
1448e971aa12SJeff Cody         bs_queue = g_new0(BlockReopenQueue, 1);
1449e971aa12SJeff Cody         QSIMPLEQ_INIT(bs_queue);
1450e971aa12SJeff Cody     }
1451e971aa12SJeff Cody 
1452e971aa12SJeff Cody     if (bs->file) {
1453e971aa12SJeff Cody         bdrv_reopen_queue(bs_queue, bs->file, flags);
1454e971aa12SJeff Cody     }
1455e971aa12SJeff Cody 
1456e971aa12SJeff Cody     bs_entry = g_new0(BlockReopenQueueEntry, 1);
1457e971aa12SJeff Cody     QSIMPLEQ_INSERT_TAIL(bs_queue, bs_entry, entry);
1458e971aa12SJeff Cody 
1459e971aa12SJeff Cody     bs_entry->state.bs = bs;
1460e971aa12SJeff Cody     bs_entry->state.flags = flags;
1461e971aa12SJeff Cody 
1462e971aa12SJeff Cody     return bs_queue;
1463e971aa12SJeff Cody }
1464e971aa12SJeff Cody 
1465e971aa12SJeff Cody /*
1466e971aa12SJeff Cody  * Reopen multiple BlockDriverStates atomically & transactionally.
1467e971aa12SJeff Cody  *
1468e971aa12SJeff Cody  * The queue passed in (bs_queue) must have been built up previous
1469e971aa12SJeff Cody  * via bdrv_reopen_queue().
1470e971aa12SJeff Cody  *
1471e971aa12SJeff Cody  * Reopens all BDS specified in the queue, with the appropriate
1472e971aa12SJeff Cody  * flags.  All devices are prepared for reopen, and failure of any
1473e971aa12SJeff Cody  * device will cause all device changes to be abandonded, and intermediate
1474e971aa12SJeff Cody  * data cleaned up.
1475e971aa12SJeff Cody  *
1476e971aa12SJeff Cody  * If all devices prepare successfully, then the changes are committed
1477e971aa12SJeff Cody  * to all devices.
1478e971aa12SJeff Cody  *
1479e971aa12SJeff Cody  */
1480e971aa12SJeff Cody int bdrv_reopen_multiple(BlockReopenQueue *bs_queue, Error **errp)
1481e971aa12SJeff Cody {
1482e971aa12SJeff Cody     int ret = -1;
1483e971aa12SJeff Cody     BlockReopenQueueEntry *bs_entry, *next;
1484e971aa12SJeff Cody     Error *local_err = NULL;
1485e971aa12SJeff Cody 
1486e971aa12SJeff Cody     assert(bs_queue != NULL);
1487e971aa12SJeff Cody 
1488e971aa12SJeff Cody     bdrv_drain_all();
1489e971aa12SJeff Cody 
1490e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1491e971aa12SJeff Cody         if (bdrv_reopen_prepare(&bs_entry->state, bs_queue, &local_err)) {
1492e971aa12SJeff Cody             error_propagate(errp, local_err);
1493e971aa12SJeff Cody             goto cleanup;
1494e971aa12SJeff Cody         }
1495e971aa12SJeff Cody         bs_entry->prepared = true;
1496e971aa12SJeff Cody     }
1497e971aa12SJeff Cody 
1498e971aa12SJeff Cody     /* If we reach this point, we have success and just need to apply the
1499e971aa12SJeff Cody      * changes
1500e971aa12SJeff Cody      */
1501e971aa12SJeff Cody     QSIMPLEQ_FOREACH(bs_entry, bs_queue, entry) {
1502e971aa12SJeff Cody         bdrv_reopen_commit(&bs_entry->state);
1503e971aa12SJeff Cody     }
1504e971aa12SJeff Cody 
1505e971aa12SJeff Cody     ret = 0;
1506e971aa12SJeff Cody 
1507e971aa12SJeff Cody cleanup:
1508e971aa12SJeff Cody     QSIMPLEQ_FOREACH_SAFE(bs_entry, bs_queue, entry, next) {
1509e971aa12SJeff Cody         if (ret && bs_entry->prepared) {
1510e971aa12SJeff Cody             bdrv_reopen_abort(&bs_entry->state);
1511e971aa12SJeff Cody         }
1512e971aa12SJeff Cody         g_free(bs_entry);
1513e971aa12SJeff Cody     }
1514e971aa12SJeff Cody     g_free(bs_queue);
1515e971aa12SJeff Cody     return ret;
1516e971aa12SJeff Cody }
1517e971aa12SJeff Cody 
1518e971aa12SJeff Cody 
1519e971aa12SJeff Cody /* Reopen a single BlockDriverState with the specified flags. */
1520e971aa12SJeff Cody int bdrv_reopen(BlockDriverState *bs, int bdrv_flags, Error **errp)
1521e971aa12SJeff Cody {
1522e971aa12SJeff Cody     int ret = -1;
1523e971aa12SJeff Cody     Error *local_err = NULL;
1524e971aa12SJeff Cody     BlockReopenQueue *queue = bdrv_reopen_queue(NULL, bs, bdrv_flags);
1525e971aa12SJeff Cody 
1526e971aa12SJeff Cody     ret = bdrv_reopen_multiple(queue, &local_err);
1527e971aa12SJeff Cody     if (local_err != NULL) {
1528e971aa12SJeff Cody         error_propagate(errp, local_err);
1529e971aa12SJeff Cody     }
1530e971aa12SJeff Cody     return ret;
1531e971aa12SJeff Cody }
1532e971aa12SJeff Cody 
1533e971aa12SJeff Cody 
1534e971aa12SJeff Cody /*
1535e971aa12SJeff Cody  * Prepares a BlockDriverState for reopen. All changes are staged in the
1536e971aa12SJeff Cody  * 'opaque' field of the BDRVReopenState, which is used and allocated by
1537e971aa12SJeff Cody  * the block driver layer .bdrv_reopen_prepare()
1538e971aa12SJeff Cody  *
1539e971aa12SJeff Cody  * bs is the BlockDriverState to reopen
1540e971aa12SJeff Cody  * flags are the new open flags
1541e971aa12SJeff Cody  * queue is the reopen queue
1542e971aa12SJeff Cody  *
1543e971aa12SJeff Cody  * Returns 0 on success, non-zero on error.  On error errp will be set
1544e971aa12SJeff Cody  * as well.
1545e971aa12SJeff Cody  *
1546e971aa12SJeff Cody  * On failure, bdrv_reopen_abort() will be called to clean up any data.
1547e971aa12SJeff Cody  * It is the responsibility of the caller to then call the abort() or
1548e971aa12SJeff Cody  * commit() for any other BDS that have been left in a prepare() state
1549e971aa12SJeff Cody  *
1550e971aa12SJeff Cody  */
1551e971aa12SJeff Cody int bdrv_reopen_prepare(BDRVReopenState *reopen_state, BlockReopenQueue *queue,
1552e971aa12SJeff Cody                         Error **errp)
1553e971aa12SJeff Cody {
1554e971aa12SJeff Cody     int ret = -1;
1555e971aa12SJeff Cody     Error *local_err = NULL;
1556e971aa12SJeff Cody     BlockDriver *drv;
1557e971aa12SJeff Cody 
1558e971aa12SJeff Cody     assert(reopen_state != NULL);
1559e971aa12SJeff Cody     assert(reopen_state->bs->drv != NULL);
1560e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1561e971aa12SJeff Cody 
1562e971aa12SJeff Cody     /* if we are to stay read-only, do not allow permission change
1563e971aa12SJeff Cody      * to r/w */
1564e971aa12SJeff Cody     if (!(reopen_state->bs->open_flags & BDRV_O_ALLOW_RDWR) &&
1565e971aa12SJeff Cody         reopen_state->flags & BDRV_O_RDWR) {
1566e971aa12SJeff Cody         error_set(errp, QERR_DEVICE_IS_READ_ONLY,
1567e971aa12SJeff Cody                   reopen_state->bs->device_name);
1568e971aa12SJeff Cody         goto error;
1569e971aa12SJeff Cody     }
1570e971aa12SJeff Cody 
1571e971aa12SJeff Cody 
1572e971aa12SJeff Cody     ret = bdrv_flush(reopen_state->bs);
1573e971aa12SJeff Cody     if (ret) {
1574e971aa12SJeff Cody         error_set(errp, ERROR_CLASS_GENERIC_ERROR, "Error (%s) flushing drive",
1575e971aa12SJeff Cody                   strerror(-ret));
1576e971aa12SJeff Cody         goto error;
1577e971aa12SJeff Cody     }
1578e971aa12SJeff Cody 
1579e971aa12SJeff Cody     if (drv->bdrv_reopen_prepare) {
1580e971aa12SJeff Cody         ret = drv->bdrv_reopen_prepare(reopen_state, queue, &local_err);
1581e971aa12SJeff Cody         if (ret) {
1582e971aa12SJeff Cody             if (local_err != NULL) {
1583e971aa12SJeff Cody                 error_propagate(errp, local_err);
1584e971aa12SJeff Cody             } else {
1585d8b6895fSLuiz Capitulino                 error_setg(errp, "failed while preparing to reopen image '%s'",
1586e971aa12SJeff Cody                            reopen_state->bs->filename);
1587e971aa12SJeff Cody             }
1588e971aa12SJeff Cody             goto error;
1589e971aa12SJeff Cody         }
1590e971aa12SJeff Cody     } else {
1591e971aa12SJeff Cody         /* It is currently mandatory to have a bdrv_reopen_prepare()
1592e971aa12SJeff Cody          * handler for each supported drv. */
1593e971aa12SJeff Cody         error_set(errp, QERR_BLOCK_FORMAT_FEATURE_NOT_SUPPORTED,
1594e971aa12SJeff Cody                   drv->format_name, reopen_state->bs->device_name,
1595e971aa12SJeff Cody                  "reopening of file");
1596e971aa12SJeff Cody         ret = -1;
1597e971aa12SJeff Cody         goto error;
1598e971aa12SJeff Cody     }
1599e971aa12SJeff Cody 
1600e971aa12SJeff Cody     ret = 0;
1601e971aa12SJeff Cody 
1602e971aa12SJeff Cody error:
1603e971aa12SJeff Cody     return ret;
1604e971aa12SJeff Cody }
1605e971aa12SJeff Cody 
1606e971aa12SJeff Cody /*
1607e971aa12SJeff Cody  * Takes the staged changes for the reopen from bdrv_reopen_prepare(), and
1608e971aa12SJeff Cody  * makes them final by swapping the staging BlockDriverState contents into
1609e971aa12SJeff Cody  * the active BlockDriverState contents.
1610e971aa12SJeff Cody  */
1611e971aa12SJeff Cody void bdrv_reopen_commit(BDRVReopenState *reopen_state)
1612e971aa12SJeff Cody {
1613e971aa12SJeff Cody     BlockDriver *drv;
1614e971aa12SJeff Cody 
1615e971aa12SJeff Cody     assert(reopen_state != NULL);
1616e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1617e971aa12SJeff Cody     assert(drv != NULL);
1618e971aa12SJeff Cody 
1619e971aa12SJeff Cody     /* If there are any driver level actions to take */
1620e971aa12SJeff Cody     if (drv->bdrv_reopen_commit) {
1621e971aa12SJeff Cody         drv->bdrv_reopen_commit(reopen_state);
1622e971aa12SJeff Cody     }
1623e971aa12SJeff Cody 
1624e971aa12SJeff Cody     /* set BDS specific flags now */
1625e971aa12SJeff Cody     reopen_state->bs->open_flags         = reopen_state->flags;
1626e971aa12SJeff Cody     reopen_state->bs->enable_write_cache = !!(reopen_state->flags &
1627e971aa12SJeff Cody                                               BDRV_O_CACHE_WB);
1628e971aa12SJeff Cody     reopen_state->bs->read_only = !(reopen_state->flags & BDRV_O_RDWR);
1629355ef4acSKevin Wolf 
1630355ef4acSKevin Wolf     bdrv_refresh_limits(reopen_state->bs);
1631e971aa12SJeff Cody }
1632e971aa12SJeff Cody 
1633e971aa12SJeff Cody /*
1634e971aa12SJeff Cody  * Abort the reopen, and delete and free the staged changes in
1635e971aa12SJeff Cody  * reopen_state
1636e971aa12SJeff Cody  */
1637e971aa12SJeff Cody void bdrv_reopen_abort(BDRVReopenState *reopen_state)
1638e971aa12SJeff Cody {
1639e971aa12SJeff Cody     BlockDriver *drv;
1640e971aa12SJeff Cody 
1641e971aa12SJeff Cody     assert(reopen_state != NULL);
1642e971aa12SJeff Cody     drv = reopen_state->bs->drv;
1643e971aa12SJeff Cody     assert(drv != NULL);
1644e971aa12SJeff Cody 
1645e971aa12SJeff Cody     if (drv->bdrv_reopen_abort) {
1646e971aa12SJeff Cody         drv->bdrv_reopen_abort(reopen_state);
1647e971aa12SJeff Cody     }
1648e971aa12SJeff Cody }
1649e971aa12SJeff Cody 
1650e971aa12SJeff Cody 
1651fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
1652fc01f7e7Sbellard {
16533e914655SPaolo Bonzini     if (bs->job) {
16543e914655SPaolo Bonzini         block_job_cancel_sync(bs->job);
16553e914655SPaolo Bonzini     }
165658fda173SStefan Hajnoczi     bdrv_drain_all(); /* complete I/O */
165758fda173SStefan Hajnoczi     bdrv_flush(bs);
165858fda173SStefan Hajnoczi     bdrv_drain_all(); /* in case flush left pending I/O */
1659d7d512f6SPaolo Bonzini     notifier_list_notify(&bs->close_notifiers, bs);
16607094f12fSKevin Wolf 
16613cbc002cSPaolo Bonzini     if (bs->drv) {
1662557df6acSStefan Hajnoczi         if (bs->backing_hd) {
16634f6fd349SFam Zheng             bdrv_unref(bs->backing_hd);
1664557df6acSStefan Hajnoczi             bs->backing_hd = NULL;
1665557df6acSStefan Hajnoczi         }
1666ea2384d3Sbellard         bs->drv->bdrv_close(bs);
16677267c094SAnthony Liguori         g_free(bs->opaque);
1668ea2384d3Sbellard #ifdef _WIN32
1669ea2384d3Sbellard         if (bs->is_temporary) {
1670ea2384d3Sbellard             unlink(bs->filename);
1671ea2384d3Sbellard         }
167267b915a5Sbellard #endif
1673ea2384d3Sbellard         bs->opaque = NULL;
1674ea2384d3Sbellard         bs->drv = NULL;
167553fec9d3SStefan Hajnoczi         bs->copy_on_read = 0;
1676a275fa42SPaolo Bonzini         bs->backing_file[0] = '\0';
1677a275fa42SPaolo Bonzini         bs->backing_format[0] = '\0';
16786405875cSPaolo Bonzini         bs->total_sectors = 0;
16796405875cSPaolo Bonzini         bs->encrypted = 0;
16806405875cSPaolo Bonzini         bs->valid_key = 0;
16816405875cSPaolo Bonzini         bs->sg = 0;
16826405875cSPaolo Bonzini         bs->growable = 0;
16830d51b4deSAsias He         bs->zero_beyond_eof = false;
1684de9c0cecSKevin Wolf         QDECREF(bs->options);
1685de9c0cecSKevin Wolf         bs->options = NULL;
1686b338082bSbellard 
168766f82ceeSKevin Wolf         if (bs->file != NULL) {
16884f6fd349SFam Zheng             bdrv_unref(bs->file);
16890ac9377dSPaolo Bonzini             bs->file = NULL;
169066f82ceeSKevin Wolf         }
16919ca11154SPavel Hrdina     }
169266f82ceeSKevin Wolf 
16937d4b4ba5SMarkus Armbruster     bdrv_dev_change_media_cb(bs, false);
169498f90dbaSZhi Yong Wu 
169598f90dbaSZhi Yong Wu     /*throttling disk I/O limits*/
169698f90dbaSZhi Yong Wu     if (bs->io_limits_enabled) {
169798f90dbaSZhi Yong Wu         bdrv_io_limits_disable(bs);
169898f90dbaSZhi Yong Wu     }
1699b338082bSbellard }
1700b338082bSbellard 
17012bc93fedSMORITA Kazutaka void bdrv_close_all(void)
17022bc93fedSMORITA Kazutaka {
17032bc93fedSMORITA Kazutaka     BlockDriverState *bs;
17042bc93fedSMORITA Kazutaka 
1705dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
17062bc93fedSMORITA Kazutaka         bdrv_close(bs);
17072bc93fedSMORITA Kazutaka     }
17082bc93fedSMORITA Kazutaka }
17092bc93fedSMORITA Kazutaka 
171088266f5aSStefan Hajnoczi /* Check if any requests are in-flight (including throttled requests) */
171188266f5aSStefan Hajnoczi static bool bdrv_requests_pending(BlockDriverState *bs)
171288266f5aSStefan Hajnoczi {
171388266f5aSStefan Hajnoczi     if (!QLIST_EMPTY(&bs->tracked_requests)) {
171488266f5aSStefan Hajnoczi         return true;
171588266f5aSStefan Hajnoczi     }
1716cc0681c4SBenoît Canet     if (!qemu_co_queue_empty(&bs->throttled_reqs[0])) {
1717cc0681c4SBenoît Canet         return true;
1718cc0681c4SBenoît Canet     }
1719cc0681c4SBenoît Canet     if (!qemu_co_queue_empty(&bs->throttled_reqs[1])) {
172088266f5aSStefan Hajnoczi         return true;
172188266f5aSStefan Hajnoczi     }
172288266f5aSStefan Hajnoczi     if (bs->file && bdrv_requests_pending(bs->file)) {
172388266f5aSStefan Hajnoczi         return true;
172488266f5aSStefan Hajnoczi     }
172588266f5aSStefan Hajnoczi     if (bs->backing_hd && bdrv_requests_pending(bs->backing_hd)) {
172688266f5aSStefan Hajnoczi         return true;
172788266f5aSStefan Hajnoczi     }
172888266f5aSStefan Hajnoczi     return false;
172988266f5aSStefan Hajnoczi }
173088266f5aSStefan Hajnoczi 
173188266f5aSStefan Hajnoczi static bool bdrv_requests_pending_all(void)
173288266f5aSStefan Hajnoczi {
173388266f5aSStefan Hajnoczi     BlockDriverState *bs;
1734dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
173588266f5aSStefan Hajnoczi         if (bdrv_requests_pending(bs)) {
173688266f5aSStefan Hajnoczi             return true;
173788266f5aSStefan Hajnoczi         }
173888266f5aSStefan Hajnoczi     }
173988266f5aSStefan Hajnoczi     return false;
174088266f5aSStefan Hajnoczi }
174188266f5aSStefan Hajnoczi 
1742922453bcSStefan Hajnoczi /*
1743922453bcSStefan Hajnoczi  * Wait for pending requests to complete across all BlockDriverStates
1744922453bcSStefan Hajnoczi  *
1745922453bcSStefan Hajnoczi  * This function does not flush data to disk, use bdrv_flush_all() for that
1746922453bcSStefan Hajnoczi  * after calling this function.
17474c355d53SZhi Yong Wu  *
17484c355d53SZhi Yong Wu  * Note that completion of an asynchronous I/O operation can trigger any
17494c355d53SZhi Yong Wu  * number of other I/O operations on other devices---for example a coroutine
17504c355d53SZhi Yong Wu  * can be arbitrarily complex and a constant flow of I/O can come until the
17514c355d53SZhi Yong Wu  * coroutine is complete.  Because of this, it is not possible to have a
17524c355d53SZhi Yong Wu  * function to drain a single device's I/O queue.
1753922453bcSStefan Hajnoczi  */
1754922453bcSStefan Hajnoczi void bdrv_drain_all(void)
1755922453bcSStefan Hajnoczi {
175688266f5aSStefan Hajnoczi     /* Always run first iteration so any pending completion BHs run */
175788266f5aSStefan Hajnoczi     bool busy = true;
1758922453bcSStefan Hajnoczi     BlockDriverState *bs;
1759922453bcSStefan Hajnoczi 
176088266f5aSStefan Hajnoczi     while (busy) {
1761dc364f4cSBenoît Canet         QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
17620b06ef3bSStefan Hajnoczi             bdrv_start_throttled_reqs(bs);
17634c355d53SZhi Yong Wu         }
1764922453bcSStefan Hajnoczi 
176588266f5aSStefan Hajnoczi         busy = bdrv_requests_pending_all();
176688266f5aSStefan Hajnoczi         busy |= aio_poll(qemu_get_aio_context(), busy);
1767922453bcSStefan Hajnoczi     }
1768922453bcSStefan Hajnoczi }
1769922453bcSStefan Hajnoczi 
1770dc364f4cSBenoît Canet /* make a BlockDriverState anonymous by removing from bdrv_state and
1771dc364f4cSBenoît Canet  * graph_bdrv_state list.
1772d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
1773d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
1774d22b2f41SRyan Harper {
1775d22b2f41SRyan Harper     if (bs->device_name[0] != '\0') {
1776dc364f4cSBenoît Canet         QTAILQ_REMOVE(&bdrv_states, bs, device_list);
1777d22b2f41SRyan Harper     }
1778d22b2f41SRyan Harper     bs->device_name[0] = '\0';
1779dc364f4cSBenoît Canet     if (bs->node_name[0] != '\0') {
1780dc364f4cSBenoît Canet         QTAILQ_REMOVE(&graph_bdrv_states, bs, node_list);
1781dc364f4cSBenoît Canet     }
1782dc364f4cSBenoît Canet     bs->node_name[0] = '\0';
1783d22b2f41SRyan Harper }
1784d22b2f41SRyan Harper 
1785e023b2e2SPaolo Bonzini static void bdrv_rebind(BlockDriverState *bs)
1786e023b2e2SPaolo Bonzini {
1787e023b2e2SPaolo Bonzini     if (bs->drv && bs->drv->bdrv_rebind) {
1788e023b2e2SPaolo Bonzini         bs->drv->bdrv_rebind(bs);
1789e023b2e2SPaolo Bonzini     }
1790e023b2e2SPaolo Bonzini }
1791e023b2e2SPaolo Bonzini 
17924ddc07caSPaolo Bonzini static void bdrv_move_feature_fields(BlockDriverState *bs_dest,
17934ddc07caSPaolo Bonzini                                      BlockDriverState *bs_src)
17944ddc07caSPaolo Bonzini {
17954ddc07caSPaolo Bonzini     /* move some fields that need to stay attached to the device */
17964ddc07caSPaolo Bonzini     bs_dest->open_flags         = bs_src->open_flags;
17974ddc07caSPaolo Bonzini 
17984ddc07caSPaolo Bonzini     /* dev info */
17994ddc07caSPaolo Bonzini     bs_dest->dev_ops            = bs_src->dev_ops;
18004ddc07caSPaolo Bonzini     bs_dest->dev_opaque         = bs_src->dev_opaque;
18014ddc07caSPaolo Bonzini     bs_dest->dev                = bs_src->dev;
18021b7fd729SPaolo Bonzini     bs_dest->guest_block_size   = bs_src->guest_block_size;
18034ddc07caSPaolo Bonzini     bs_dest->copy_on_read       = bs_src->copy_on_read;
18044ddc07caSPaolo Bonzini 
18054ddc07caSPaolo Bonzini     bs_dest->enable_write_cache = bs_src->enable_write_cache;
18064ddc07caSPaolo Bonzini 
1807cc0681c4SBenoît Canet     /* i/o throttled req */
1808cc0681c4SBenoît Canet     memcpy(&bs_dest->throttle_state,
1809cc0681c4SBenoît Canet            &bs_src->throttle_state,
1810cc0681c4SBenoît Canet            sizeof(ThrottleState));
1811cc0681c4SBenoît Canet     bs_dest->throttled_reqs[0]  = bs_src->throttled_reqs[0];
1812cc0681c4SBenoît Canet     bs_dest->throttled_reqs[1]  = bs_src->throttled_reqs[1];
18134ddc07caSPaolo Bonzini     bs_dest->io_limits_enabled  = bs_src->io_limits_enabled;
18144ddc07caSPaolo Bonzini 
18154ddc07caSPaolo Bonzini     /* r/w error */
18164ddc07caSPaolo Bonzini     bs_dest->on_read_error      = bs_src->on_read_error;
18174ddc07caSPaolo Bonzini     bs_dest->on_write_error     = bs_src->on_write_error;
18184ddc07caSPaolo Bonzini 
18194ddc07caSPaolo Bonzini     /* i/o status */
18204ddc07caSPaolo Bonzini     bs_dest->iostatus_enabled   = bs_src->iostatus_enabled;
18214ddc07caSPaolo Bonzini     bs_dest->iostatus           = bs_src->iostatus;
18224ddc07caSPaolo Bonzini 
18234ddc07caSPaolo Bonzini     /* dirty bitmap */
1824e4654d2dSFam Zheng     bs_dest->dirty_bitmaps      = bs_src->dirty_bitmaps;
18254ddc07caSPaolo Bonzini 
18269fcb0251SFam Zheng     /* reference count */
18279fcb0251SFam Zheng     bs_dest->refcnt             = bs_src->refcnt;
18289fcb0251SFam Zheng 
18294ddc07caSPaolo Bonzini     /* job */
18304ddc07caSPaolo Bonzini     bs_dest->in_use             = bs_src->in_use;
18314ddc07caSPaolo Bonzini     bs_dest->job                = bs_src->job;
18324ddc07caSPaolo Bonzini 
18334ddc07caSPaolo Bonzini     /* keep the same entry in bdrv_states */
18344ddc07caSPaolo Bonzini     pstrcpy(bs_dest->device_name, sizeof(bs_dest->device_name),
18354ddc07caSPaolo Bonzini             bs_src->device_name);
1836dc364f4cSBenoît Canet     bs_dest->device_list = bs_src->device_list;
1837dc364f4cSBenoît Canet 
1838dc364f4cSBenoît Canet     /* keep the same entry in graph_bdrv_states
1839dc364f4cSBenoît Canet      * We do want to swap name but don't want to swap linked list entries
1840dc364f4cSBenoît Canet      */
1841dc364f4cSBenoît Canet     bs_dest->node_list   = bs_src->node_list;
18424ddc07caSPaolo Bonzini }
18434ddc07caSPaolo Bonzini 
18444ddc07caSPaolo Bonzini /*
18454ddc07caSPaolo Bonzini  * Swap bs contents for two image chains while they are live,
18464ddc07caSPaolo Bonzini  * while keeping required fields on the BlockDriverState that is
18474ddc07caSPaolo Bonzini  * actually attached to a device.
18484ddc07caSPaolo Bonzini  *
18494ddc07caSPaolo Bonzini  * This will modify the BlockDriverState fields, and swap contents
18504ddc07caSPaolo Bonzini  * between bs_new and bs_old. Both bs_new and bs_old are modified.
18514ddc07caSPaolo Bonzini  *
18524ddc07caSPaolo Bonzini  * bs_new is required to be anonymous.
18534ddc07caSPaolo Bonzini  *
18544ddc07caSPaolo Bonzini  * This function does not create any image files.
18554ddc07caSPaolo Bonzini  */
18564ddc07caSPaolo Bonzini void bdrv_swap(BlockDriverState *bs_new, BlockDriverState *bs_old)
18574ddc07caSPaolo Bonzini {
18584ddc07caSPaolo Bonzini     BlockDriverState tmp;
18594ddc07caSPaolo Bonzini 
18604ddc07caSPaolo Bonzini     /* bs_new must be anonymous and shouldn't have anything fancy enabled */
18614ddc07caSPaolo Bonzini     assert(bs_new->device_name[0] == '\0');
1862e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs_new->dirty_bitmaps));
18634ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
18644ddc07caSPaolo Bonzini     assert(bs_new->dev == NULL);
18654ddc07caSPaolo Bonzini     assert(bs_new->in_use == 0);
18664ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
1867cc0681c4SBenoît Canet     assert(!throttle_have_timer(&bs_new->throttle_state));
18684ddc07caSPaolo Bonzini 
18694ddc07caSPaolo Bonzini     tmp = *bs_new;
18704ddc07caSPaolo Bonzini     *bs_new = *bs_old;
18714ddc07caSPaolo Bonzini     *bs_old = tmp;
18724ddc07caSPaolo Bonzini 
18734ddc07caSPaolo Bonzini     /* there are some fields that should not be swapped, move them back */
18744ddc07caSPaolo Bonzini     bdrv_move_feature_fields(&tmp, bs_old);
18754ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_old, bs_new);
18764ddc07caSPaolo Bonzini     bdrv_move_feature_fields(bs_new, &tmp);
18774ddc07caSPaolo Bonzini 
18784ddc07caSPaolo Bonzini     /* bs_new shouldn't be in bdrv_states even after the swap!  */
18794ddc07caSPaolo Bonzini     assert(bs_new->device_name[0] == '\0');
18804ddc07caSPaolo Bonzini 
18814ddc07caSPaolo Bonzini     /* Check a few fields that should remain attached to the device */
18824ddc07caSPaolo Bonzini     assert(bs_new->dev == NULL);
18834ddc07caSPaolo Bonzini     assert(bs_new->job == NULL);
18844ddc07caSPaolo Bonzini     assert(bs_new->in_use == 0);
18854ddc07caSPaolo Bonzini     assert(bs_new->io_limits_enabled == false);
1886cc0681c4SBenoît Canet     assert(!throttle_have_timer(&bs_new->throttle_state));
18874ddc07caSPaolo Bonzini 
18884ddc07caSPaolo Bonzini     bdrv_rebind(bs_new);
18894ddc07caSPaolo Bonzini     bdrv_rebind(bs_old);
18904ddc07caSPaolo Bonzini }
18914ddc07caSPaolo Bonzini 
18928802d1fdSJeff Cody /*
18938802d1fdSJeff Cody  * Add new bs contents at the top of an image chain while the chain is
18948802d1fdSJeff Cody  * live, while keeping required fields on the top layer.
18958802d1fdSJeff Cody  *
18968802d1fdSJeff Cody  * This will modify the BlockDriverState fields, and swap contents
18978802d1fdSJeff Cody  * between bs_new and bs_top. Both bs_new and bs_top are modified.
18988802d1fdSJeff Cody  *
1899f6801b83SJeff Cody  * bs_new is required to be anonymous.
1900f6801b83SJeff Cody  *
19018802d1fdSJeff Cody  * This function does not create any image files.
19028802d1fdSJeff Cody  */
19038802d1fdSJeff Cody void bdrv_append(BlockDriverState *bs_new, BlockDriverState *bs_top)
19048802d1fdSJeff Cody {
19054ddc07caSPaolo Bonzini     bdrv_swap(bs_new, bs_top);
19068802d1fdSJeff Cody 
19078802d1fdSJeff Cody     /* The contents of 'tmp' will become bs_top, as we are
19088802d1fdSJeff Cody      * swapping bs_new and bs_top contents. */
19094ddc07caSPaolo Bonzini     bs_top->backing_hd = bs_new;
19104ddc07caSPaolo Bonzini     bs_top->open_flags &= ~BDRV_O_NO_BACKING;
19114ddc07caSPaolo Bonzini     pstrcpy(bs_top->backing_file, sizeof(bs_top->backing_file),
19124ddc07caSPaolo Bonzini             bs_new->filename);
19134ddc07caSPaolo Bonzini     pstrcpy(bs_top->backing_format, sizeof(bs_top->backing_format),
19144ddc07caSPaolo Bonzini             bs_new->drv ? bs_new->drv->format_name : "");
19158802d1fdSJeff Cody }
19168802d1fdSJeff Cody 
19174f6fd349SFam Zheng static void bdrv_delete(BlockDriverState *bs)
1918b338082bSbellard {
1919fa879d62SMarkus Armbruster     assert(!bs->dev);
19203e914655SPaolo Bonzini     assert(!bs->job);
19213e914655SPaolo Bonzini     assert(!bs->in_use);
19224f6fd349SFam Zheng     assert(!bs->refcnt);
1923e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
192418846deeSMarkus Armbruster 
1925e1b5c52eSStefan Hajnoczi     bdrv_close(bs);
1926e1b5c52eSStefan Hajnoczi 
19271b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
1928d22b2f41SRyan Harper     bdrv_make_anon(bs);
192934c6f050Saurel32 
19307267c094SAnthony Liguori     g_free(bs);
1931fc01f7e7Sbellard }
1932fc01f7e7Sbellard 
1933fa879d62SMarkus Armbruster int bdrv_attach_dev(BlockDriverState *bs, void *dev)
1934fa879d62SMarkus Armbruster /* TODO change to DeviceState *dev when all users are qdevified */
193518846deeSMarkus Armbruster {
1936fa879d62SMarkus Armbruster     if (bs->dev) {
193718846deeSMarkus Armbruster         return -EBUSY;
193818846deeSMarkus Armbruster     }
1939fa879d62SMarkus Armbruster     bs->dev = dev;
194028a7282aSLuiz Capitulino     bdrv_iostatus_reset(bs);
194118846deeSMarkus Armbruster     return 0;
194218846deeSMarkus Armbruster }
194318846deeSMarkus Armbruster 
1944fa879d62SMarkus Armbruster /* TODO qdevified devices don't use this, remove when devices are qdevified */
1945fa879d62SMarkus Armbruster void bdrv_attach_dev_nofail(BlockDriverState *bs, void *dev)
194618846deeSMarkus Armbruster {
1947fa879d62SMarkus Armbruster     if (bdrv_attach_dev(bs, dev) < 0) {
1948fa879d62SMarkus Armbruster         abort();
1949fa879d62SMarkus Armbruster     }
1950fa879d62SMarkus Armbruster }
1951fa879d62SMarkus Armbruster 
1952fa879d62SMarkus Armbruster void bdrv_detach_dev(BlockDriverState *bs, void *dev)
1953fa879d62SMarkus Armbruster /* TODO change to DeviceState *dev when all users are qdevified */
1954fa879d62SMarkus Armbruster {
1955fa879d62SMarkus Armbruster     assert(bs->dev == dev);
1956fa879d62SMarkus Armbruster     bs->dev = NULL;
19570e49de52SMarkus Armbruster     bs->dev_ops = NULL;
19580e49de52SMarkus Armbruster     bs->dev_opaque = NULL;
19591b7fd729SPaolo Bonzini     bs->guest_block_size = 512;
196018846deeSMarkus Armbruster }
196118846deeSMarkus Armbruster 
1962fa879d62SMarkus Armbruster /* TODO change to return DeviceState * when all users are qdevified */
1963fa879d62SMarkus Armbruster void *bdrv_get_attached_dev(BlockDriverState *bs)
196418846deeSMarkus Armbruster {
1965fa879d62SMarkus Armbruster     return bs->dev;
196618846deeSMarkus Armbruster }
196718846deeSMarkus Armbruster 
19680e49de52SMarkus Armbruster void bdrv_set_dev_ops(BlockDriverState *bs, const BlockDevOps *ops,
19690e49de52SMarkus Armbruster                       void *opaque)
19700e49de52SMarkus Armbruster {
19710e49de52SMarkus Armbruster     bs->dev_ops = ops;
19720e49de52SMarkus Armbruster     bs->dev_opaque = opaque;
19730e49de52SMarkus Armbruster }
19740e49de52SMarkus Armbruster 
197532c81a4aSPaolo Bonzini void bdrv_emit_qmp_error_event(const BlockDriverState *bdrv,
197632c81a4aSPaolo Bonzini                                enum MonitorEvent ev,
19771ceee0d5SPaolo Bonzini                                BlockErrorAction action, bool is_read)
1978329c0a48SLuiz Capitulino {
1979329c0a48SLuiz Capitulino     QObject *data;
1980329c0a48SLuiz Capitulino     const char *action_str;
1981329c0a48SLuiz Capitulino 
1982329c0a48SLuiz Capitulino     switch (action) {
1983329c0a48SLuiz Capitulino     case BDRV_ACTION_REPORT:
1984329c0a48SLuiz Capitulino         action_str = "report";
1985329c0a48SLuiz Capitulino         break;
1986329c0a48SLuiz Capitulino     case BDRV_ACTION_IGNORE:
1987329c0a48SLuiz Capitulino         action_str = "ignore";
1988329c0a48SLuiz Capitulino         break;
1989329c0a48SLuiz Capitulino     case BDRV_ACTION_STOP:
1990329c0a48SLuiz Capitulino         action_str = "stop";
1991329c0a48SLuiz Capitulino         break;
1992329c0a48SLuiz Capitulino     default:
1993329c0a48SLuiz Capitulino         abort();
1994329c0a48SLuiz Capitulino     }
1995329c0a48SLuiz Capitulino 
1996329c0a48SLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
1997329c0a48SLuiz Capitulino                               bdrv->device_name,
1998329c0a48SLuiz Capitulino                               action_str,
1999329c0a48SLuiz Capitulino                               is_read ? "read" : "write");
200032c81a4aSPaolo Bonzini     monitor_protocol_event(ev, data);
2001329c0a48SLuiz Capitulino 
2002329c0a48SLuiz Capitulino     qobject_decref(data);
2003329c0a48SLuiz Capitulino }
2004329c0a48SLuiz Capitulino 
20056f382ed2SLuiz Capitulino static void bdrv_emit_qmp_eject_event(BlockDriverState *bs, bool ejected)
20066f382ed2SLuiz Capitulino {
20076f382ed2SLuiz Capitulino     QObject *data;
20086f382ed2SLuiz Capitulino 
20096f382ed2SLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'tray-open': %i }",
20106f382ed2SLuiz Capitulino                               bdrv_get_device_name(bs), ejected);
20116f382ed2SLuiz Capitulino     monitor_protocol_event(QEVENT_DEVICE_TRAY_MOVED, data);
20126f382ed2SLuiz Capitulino 
20136f382ed2SLuiz Capitulino     qobject_decref(data);
20146f382ed2SLuiz Capitulino }
20156f382ed2SLuiz Capitulino 
20167d4b4ba5SMarkus Armbruster static void bdrv_dev_change_media_cb(BlockDriverState *bs, bool load)
20170e49de52SMarkus Armbruster {
2018145feb17SMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->change_media_cb) {
20196f382ed2SLuiz Capitulino         bool tray_was_closed = !bdrv_dev_is_tray_open(bs);
20207d4b4ba5SMarkus Armbruster         bs->dev_ops->change_media_cb(bs->dev_opaque, load);
20216f382ed2SLuiz Capitulino         if (tray_was_closed) {
20226f382ed2SLuiz Capitulino             /* tray open */
20236f382ed2SLuiz Capitulino             bdrv_emit_qmp_eject_event(bs, true);
20246f382ed2SLuiz Capitulino         }
20256f382ed2SLuiz Capitulino         if (load) {
20266f382ed2SLuiz Capitulino             /* tray close */
20276f382ed2SLuiz Capitulino             bdrv_emit_qmp_eject_event(bs, false);
20286f382ed2SLuiz Capitulino         }
2029145feb17SMarkus Armbruster     }
2030145feb17SMarkus Armbruster }
2031145feb17SMarkus Armbruster 
20322c6942faSMarkus Armbruster bool bdrv_dev_has_removable_media(BlockDriverState *bs)
20332c6942faSMarkus Armbruster {
20342c6942faSMarkus Armbruster     return !bs->dev || (bs->dev_ops && bs->dev_ops->change_media_cb);
20352c6942faSMarkus Armbruster }
20362c6942faSMarkus Armbruster 
2037025ccaa7SPaolo Bonzini void bdrv_dev_eject_request(BlockDriverState *bs, bool force)
2038025ccaa7SPaolo Bonzini {
2039025ccaa7SPaolo Bonzini     if (bs->dev_ops && bs->dev_ops->eject_request_cb) {
2040025ccaa7SPaolo Bonzini         bs->dev_ops->eject_request_cb(bs->dev_opaque, force);
2041025ccaa7SPaolo Bonzini     }
2042025ccaa7SPaolo Bonzini }
2043025ccaa7SPaolo Bonzini 
2044e4def80bSMarkus Armbruster bool bdrv_dev_is_tray_open(BlockDriverState *bs)
2045e4def80bSMarkus Armbruster {
2046e4def80bSMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->is_tray_open) {
2047e4def80bSMarkus Armbruster         return bs->dev_ops->is_tray_open(bs->dev_opaque);
2048e4def80bSMarkus Armbruster     }
2049e4def80bSMarkus Armbruster     return false;
2050e4def80bSMarkus Armbruster }
2051e4def80bSMarkus Armbruster 
2052145feb17SMarkus Armbruster static void bdrv_dev_resize_cb(BlockDriverState *bs)
2053145feb17SMarkus Armbruster {
2054145feb17SMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->resize_cb) {
2055145feb17SMarkus Armbruster         bs->dev_ops->resize_cb(bs->dev_opaque);
20560e49de52SMarkus Armbruster     }
20570e49de52SMarkus Armbruster }
20580e49de52SMarkus Armbruster 
2059f107639aSMarkus Armbruster bool bdrv_dev_is_medium_locked(BlockDriverState *bs)
2060f107639aSMarkus Armbruster {
2061f107639aSMarkus Armbruster     if (bs->dev_ops && bs->dev_ops->is_medium_locked) {
2062f107639aSMarkus Armbruster         return bs->dev_ops->is_medium_locked(bs->dev_opaque);
2063f107639aSMarkus Armbruster     }
2064f107639aSMarkus Armbruster     return false;
2065f107639aSMarkus Armbruster }
2066f107639aSMarkus Armbruster 
2067e97fc193Saliguori /*
2068e97fc193Saliguori  * Run consistency checks on an image
2069e97fc193Saliguori  *
2070e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
2071a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
2072e076f338SKevin Wolf  * check are stored in res.
2073e97fc193Saliguori  */
20744534ff54SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res, BdrvCheckMode fix)
2075e97fc193Saliguori {
2076e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
2077e97fc193Saliguori         return -ENOTSUP;
2078e97fc193Saliguori     }
2079e97fc193Saliguori 
2080e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
20814534ff54SKevin Wolf     return bs->drv->bdrv_check(bs, res, fix);
2082e97fc193Saliguori }
2083e97fc193Saliguori 
20848a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
20858a426614SKevin Wolf 
208633e3963eSbellard /* commit COW file into the raw image */
208733e3963eSbellard int bdrv_commit(BlockDriverState *bs)
208833e3963eSbellard {
208919cb3738Sbellard     BlockDriver *drv = bs->drv;
209072706ea4SJeff Cody     int64_t sector, total_sectors, length, backing_length;
20918a426614SKevin Wolf     int n, ro, open_flags;
20920bce597dSJeff Cody     int ret = 0;
209372706ea4SJeff Cody     uint8_t *buf = NULL;
2094c2cba3d9SJim Meyering     char filename[PATH_MAX];
209533e3963eSbellard 
209619cb3738Sbellard     if (!drv)
209719cb3738Sbellard         return -ENOMEDIUM;
209833e3963eSbellard 
20994dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
21004dca4b63SNaphtali Sprei         return -ENOTSUP;
21014dca4b63SNaphtali Sprei     }
21024dca4b63SNaphtali Sprei 
21032d3735d3SStefan Hajnoczi     if (bdrv_in_use(bs) || bdrv_in_use(bs->backing_hd)) {
21042d3735d3SStefan Hajnoczi         return -EBUSY;
21052d3735d3SStefan Hajnoczi     }
21062d3735d3SStefan Hajnoczi 
21074dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
2108c2cba3d9SJim Meyering     /* Use pstrcpy (not strncpy): filename must be NUL-terminated. */
2109c2cba3d9SJim Meyering     pstrcpy(filename, sizeof(filename), bs->backing_hd->filename);
21104dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
21114dca4b63SNaphtali Sprei 
21124dca4b63SNaphtali Sprei     if (ro) {
21130bce597dSJeff Cody         if (bdrv_reopen(bs->backing_hd, open_flags | BDRV_O_RDWR, NULL)) {
21140bce597dSJeff Cody             return -EACCES;
21154dca4b63SNaphtali Sprei         }
2116ea2384d3Sbellard     }
2117ea2384d3Sbellard 
211872706ea4SJeff Cody     length = bdrv_getlength(bs);
211972706ea4SJeff Cody     if (length < 0) {
212072706ea4SJeff Cody         ret = length;
212172706ea4SJeff Cody         goto ro_cleanup;
212272706ea4SJeff Cody     }
212372706ea4SJeff Cody 
212472706ea4SJeff Cody     backing_length = bdrv_getlength(bs->backing_hd);
212572706ea4SJeff Cody     if (backing_length < 0) {
212672706ea4SJeff Cody         ret = backing_length;
212772706ea4SJeff Cody         goto ro_cleanup;
212872706ea4SJeff Cody     }
212972706ea4SJeff Cody 
213072706ea4SJeff Cody     /* If our top snapshot is larger than the backing file image,
213172706ea4SJeff Cody      * grow the backing file image if possible.  If not possible,
213272706ea4SJeff Cody      * we must return an error */
213372706ea4SJeff Cody     if (length > backing_length) {
213472706ea4SJeff Cody         ret = bdrv_truncate(bs->backing_hd, length);
213572706ea4SJeff Cody         if (ret < 0) {
213672706ea4SJeff Cody             goto ro_cleanup;
213772706ea4SJeff Cody         }
213872706ea4SJeff Cody     }
213972706ea4SJeff Cody 
214072706ea4SJeff Cody     total_sectors = length >> BDRV_SECTOR_BITS;
21417267c094SAnthony Liguori     buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
21428a426614SKevin Wolf 
21438a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
2144d663640cSPaolo Bonzini         ret = bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n);
2145d663640cSPaolo Bonzini         if (ret < 0) {
2146d663640cSPaolo Bonzini             goto ro_cleanup;
2147d663640cSPaolo Bonzini         }
2148d663640cSPaolo Bonzini         if (ret) {
2149dabfa6ccSKevin Wolf             ret = bdrv_read(bs, sector, buf, n);
2150dabfa6ccSKevin Wolf             if (ret < 0) {
21514dca4b63SNaphtali Sprei                 goto ro_cleanup;
215233e3963eSbellard             }
215333e3963eSbellard 
2154dabfa6ccSKevin Wolf             ret = bdrv_write(bs->backing_hd, sector, buf, n);
2155dabfa6ccSKevin Wolf             if (ret < 0) {
21564dca4b63SNaphtali Sprei                 goto ro_cleanup;
215733e3963eSbellard             }
215833e3963eSbellard         }
215933e3963eSbellard     }
216095389c86Sbellard 
21611d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
21621d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
2163dabfa6ccSKevin Wolf         if (ret < 0) {
2164dabfa6ccSKevin Wolf             goto ro_cleanup;
2165dabfa6ccSKevin Wolf         }
21661d44952fSChristoph Hellwig         bdrv_flush(bs);
21671d44952fSChristoph Hellwig     }
216895389c86Sbellard 
21693f5075aeSChristoph Hellwig     /*
21703f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
21713f5075aeSChristoph Hellwig      * stable on disk.
21723f5075aeSChristoph Hellwig      */
2173dabfa6ccSKevin Wolf     if (bs->backing_hd) {
21743f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
2175dabfa6ccSKevin Wolf     }
21764dca4b63SNaphtali Sprei 
2177dabfa6ccSKevin Wolf     ret = 0;
21784dca4b63SNaphtali Sprei ro_cleanup:
21797267c094SAnthony Liguori     g_free(buf);
21804dca4b63SNaphtali Sprei 
21814dca4b63SNaphtali Sprei     if (ro) {
21820bce597dSJeff Cody         /* ignoring error return here */
21830bce597dSJeff Cody         bdrv_reopen(bs->backing_hd, open_flags & ~BDRV_O_RDWR, NULL);
21844dca4b63SNaphtali Sprei     }
21854dca4b63SNaphtali Sprei 
21861d44952fSChristoph Hellwig     return ret;
218733e3963eSbellard }
218833e3963eSbellard 
2189e8877497SStefan Hajnoczi int bdrv_commit_all(void)
21906ab4b5abSMarkus Armbruster {
21916ab4b5abSMarkus Armbruster     BlockDriverState *bs;
21926ab4b5abSMarkus Armbruster 
2193dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
2194272d2d8eSJeff Cody         if (bs->drv && bs->backing_hd) {
2195e8877497SStefan Hajnoczi             int ret = bdrv_commit(bs);
2196e8877497SStefan Hajnoczi             if (ret < 0) {
2197e8877497SStefan Hajnoczi                 return ret;
21986ab4b5abSMarkus Armbruster             }
21996ab4b5abSMarkus Armbruster         }
2200272d2d8eSJeff Cody     }
2201e8877497SStefan Hajnoczi     return 0;
2202e8877497SStefan Hajnoczi }
22036ab4b5abSMarkus Armbruster 
2204dbffbdcfSStefan Hajnoczi /**
2205dbffbdcfSStefan Hajnoczi  * Remove an active request from the tracked requests list
2206dbffbdcfSStefan Hajnoczi  *
2207dbffbdcfSStefan Hajnoczi  * This function should be called when a tracked request is completing.
2208dbffbdcfSStefan Hajnoczi  */
2209dbffbdcfSStefan Hajnoczi static void tracked_request_end(BdrvTrackedRequest *req)
2210dbffbdcfSStefan Hajnoczi {
2211dbffbdcfSStefan Hajnoczi     QLIST_REMOVE(req, list);
2212f4658285SStefan Hajnoczi     qemu_co_queue_restart_all(&req->wait_queue);
2213dbffbdcfSStefan Hajnoczi }
2214dbffbdcfSStefan Hajnoczi 
2215dbffbdcfSStefan Hajnoczi /**
2216dbffbdcfSStefan Hajnoczi  * Add an active request to the tracked requests list
2217dbffbdcfSStefan Hajnoczi  */
2218dbffbdcfSStefan Hajnoczi static void tracked_request_begin(BdrvTrackedRequest *req,
2219dbffbdcfSStefan Hajnoczi                                   BlockDriverState *bs,
2220dbffbdcfSStefan Hajnoczi                                   int64_t sector_num,
2221dbffbdcfSStefan Hajnoczi                                   int nb_sectors, bool is_write)
2222dbffbdcfSStefan Hajnoczi {
2223dbffbdcfSStefan Hajnoczi     *req = (BdrvTrackedRequest){
2224dbffbdcfSStefan Hajnoczi         .bs = bs,
2225dbffbdcfSStefan Hajnoczi         .sector_num = sector_num,
2226dbffbdcfSStefan Hajnoczi         .nb_sectors = nb_sectors,
2227dbffbdcfSStefan Hajnoczi         .is_write = is_write,
22285f8b6491SStefan Hajnoczi         .co = qemu_coroutine_self(),
2229dbffbdcfSStefan Hajnoczi     };
2230dbffbdcfSStefan Hajnoczi 
2231f4658285SStefan Hajnoczi     qemu_co_queue_init(&req->wait_queue);
2232f4658285SStefan Hajnoczi 
2233dbffbdcfSStefan Hajnoczi     QLIST_INSERT_HEAD(&bs->tracked_requests, req, list);
2234dbffbdcfSStefan Hajnoczi }
2235dbffbdcfSStefan Hajnoczi 
2236d83947acSStefan Hajnoczi /**
2237d83947acSStefan Hajnoczi  * Round a region to cluster boundaries
2238d83947acSStefan Hajnoczi  */
2239343bded4SPaolo Bonzini void bdrv_round_to_clusters(BlockDriverState *bs,
2240d83947acSStefan Hajnoczi                             int64_t sector_num, int nb_sectors,
2241d83947acSStefan Hajnoczi                             int64_t *cluster_sector_num,
2242d83947acSStefan Hajnoczi                             int *cluster_nb_sectors)
2243d83947acSStefan Hajnoczi {
2244d83947acSStefan Hajnoczi     BlockDriverInfo bdi;
2245d83947acSStefan Hajnoczi 
2246d83947acSStefan Hajnoczi     if (bdrv_get_info(bs, &bdi) < 0 || bdi.cluster_size == 0) {
2247d83947acSStefan Hajnoczi         *cluster_sector_num = sector_num;
2248d83947acSStefan Hajnoczi         *cluster_nb_sectors = nb_sectors;
2249d83947acSStefan Hajnoczi     } else {
2250d83947acSStefan Hajnoczi         int64_t c = bdi.cluster_size / BDRV_SECTOR_SIZE;
2251d83947acSStefan Hajnoczi         *cluster_sector_num = QEMU_ALIGN_DOWN(sector_num, c);
2252d83947acSStefan Hajnoczi         *cluster_nb_sectors = QEMU_ALIGN_UP(sector_num - *cluster_sector_num +
2253d83947acSStefan Hajnoczi                                             nb_sectors, c);
2254d83947acSStefan Hajnoczi     }
2255d83947acSStefan Hajnoczi }
2256d83947acSStefan Hajnoczi 
2257f4658285SStefan Hajnoczi static bool tracked_request_overlaps(BdrvTrackedRequest *req,
2258f4658285SStefan Hajnoczi                                      int64_t sector_num, int nb_sectors) {
2259d83947acSStefan Hajnoczi     /*        aaaa   bbbb */
2260d83947acSStefan Hajnoczi     if (sector_num >= req->sector_num + req->nb_sectors) {
2261d83947acSStefan Hajnoczi         return false;
2262d83947acSStefan Hajnoczi     }
2263d83947acSStefan Hajnoczi     /* bbbb   aaaa        */
2264d83947acSStefan Hajnoczi     if (req->sector_num >= sector_num + nb_sectors) {
2265d83947acSStefan Hajnoczi         return false;
2266d83947acSStefan Hajnoczi     }
2267d83947acSStefan Hajnoczi     return true;
2268f4658285SStefan Hajnoczi }
2269f4658285SStefan Hajnoczi 
2270f4658285SStefan Hajnoczi static void coroutine_fn wait_for_overlapping_requests(BlockDriverState *bs,
2271f4658285SStefan Hajnoczi         int64_t sector_num, int nb_sectors)
2272f4658285SStefan Hajnoczi {
2273f4658285SStefan Hajnoczi     BdrvTrackedRequest *req;
2274d83947acSStefan Hajnoczi     int64_t cluster_sector_num;
2275d83947acSStefan Hajnoczi     int cluster_nb_sectors;
2276f4658285SStefan Hajnoczi     bool retry;
2277f4658285SStefan Hajnoczi 
2278d83947acSStefan Hajnoczi     /* If we touch the same cluster it counts as an overlap.  This guarantees
2279d83947acSStefan Hajnoczi      * that allocating writes will be serialized and not race with each other
2280d83947acSStefan Hajnoczi      * for the same cluster.  For example, in copy-on-read it ensures that the
2281d83947acSStefan Hajnoczi      * CoR read and write operations are atomic and guest writes cannot
2282d83947acSStefan Hajnoczi      * interleave between them.
2283d83947acSStefan Hajnoczi      */
2284343bded4SPaolo Bonzini     bdrv_round_to_clusters(bs, sector_num, nb_sectors,
2285d83947acSStefan Hajnoczi                            &cluster_sector_num, &cluster_nb_sectors);
2286d83947acSStefan Hajnoczi 
2287f4658285SStefan Hajnoczi     do {
2288f4658285SStefan Hajnoczi         retry = false;
2289f4658285SStefan Hajnoczi         QLIST_FOREACH(req, &bs->tracked_requests, list) {
2290d83947acSStefan Hajnoczi             if (tracked_request_overlaps(req, cluster_sector_num,
2291d83947acSStefan Hajnoczi                                          cluster_nb_sectors)) {
22925f8b6491SStefan Hajnoczi                 /* Hitting this means there was a reentrant request, for
22935f8b6491SStefan Hajnoczi                  * example, a block driver issuing nested requests.  This must
22945f8b6491SStefan Hajnoczi                  * never happen since it means deadlock.
22955f8b6491SStefan Hajnoczi                  */
22965f8b6491SStefan Hajnoczi                 assert(qemu_coroutine_self() != req->co);
22975f8b6491SStefan Hajnoczi 
2298f4658285SStefan Hajnoczi                 qemu_co_queue_wait(&req->wait_queue);
2299f4658285SStefan Hajnoczi                 retry = true;
2300f4658285SStefan Hajnoczi                 break;
2301f4658285SStefan Hajnoczi             }
2302f4658285SStefan Hajnoczi         }
2303f4658285SStefan Hajnoczi     } while (retry);
2304f4658285SStefan Hajnoczi }
2305f4658285SStefan Hajnoczi 
2306756e6736SKevin Wolf /*
2307756e6736SKevin Wolf  * Return values:
2308756e6736SKevin Wolf  * 0        - success
2309756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
2310756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
2311756e6736SKevin Wolf  *            image file header
2312756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
2313756e6736SKevin Wolf  */
2314756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
2315756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
2316756e6736SKevin Wolf {
2317756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
2318469ef350SPaolo Bonzini     int ret;
2319756e6736SKevin Wolf 
23205f377794SPaolo Bonzini     /* Backing file format doesn't make sense without a backing file */
23215f377794SPaolo Bonzini     if (backing_fmt && !backing_file) {
23225f377794SPaolo Bonzini         return -EINVAL;
23235f377794SPaolo Bonzini     }
23245f377794SPaolo Bonzini 
2325756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
2326469ef350SPaolo Bonzini         ret = drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
2327756e6736SKevin Wolf     } else {
2328469ef350SPaolo Bonzini         ret = -ENOTSUP;
2329756e6736SKevin Wolf     }
2330469ef350SPaolo Bonzini 
2331469ef350SPaolo Bonzini     if (ret == 0) {
2332469ef350SPaolo Bonzini         pstrcpy(bs->backing_file, sizeof(bs->backing_file), backing_file ?: "");
2333469ef350SPaolo Bonzini         pstrcpy(bs->backing_format, sizeof(bs->backing_format), backing_fmt ?: "");
2334469ef350SPaolo Bonzini     }
2335469ef350SPaolo Bonzini     return ret;
2336756e6736SKevin Wolf }
2337756e6736SKevin Wolf 
23386ebdcee2SJeff Cody /*
23396ebdcee2SJeff Cody  * Finds the image layer in the chain that has 'bs' as its backing file.
23406ebdcee2SJeff Cody  *
23416ebdcee2SJeff Cody  * active is the current topmost image.
23426ebdcee2SJeff Cody  *
23436ebdcee2SJeff Cody  * Returns NULL if bs is not found in active's image chain,
23446ebdcee2SJeff Cody  * or if active == bs.
23456ebdcee2SJeff Cody  */
23466ebdcee2SJeff Cody BlockDriverState *bdrv_find_overlay(BlockDriverState *active,
23476ebdcee2SJeff Cody                                     BlockDriverState *bs)
23486ebdcee2SJeff Cody {
23496ebdcee2SJeff Cody     BlockDriverState *overlay = NULL;
23506ebdcee2SJeff Cody     BlockDriverState *intermediate;
23516ebdcee2SJeff Cody 
23526ebdcee2SJeff Cody     assert(active != NULL);
23536ebdcee2SJeff Cody     assert(bs != NULL);
23546ebdcee2SJeff Cody 
23556ebdcee2SJeff Cody     /* if bs is the same as active, then by definition it has no overlay
23566ebdcee2SJeff Cody      */
23576ebdcee2SJeff Cody     if (active == bs) {
23586ebdcee2SJeff Cody         return NULL;
23596ebdcee2SJeff Cody     }
23606ebdcee2SJeff Cody 
23616ebdcee2SJeff Cody     intermediate = active;
23626ebdcee2SJeff Cody     while (intermediate->backing_hd) {
23636ebdcee2SJeff Cody         if (intermediate->backing_hd == bs) {
23646ebdcee2SJeff Cody             overlay = intermediate;
23656ebdcee2SJeff Cody             break;
23666ebdcee2SJeff Cody         }
23676ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
23686ebdcee2SJeff Cody     }
23696ebdcee2SJeff Cody 
23706ebdcee2SJeff Cody     return overlay;
23716ebdcee2SJeff Cody }
23726ebdcee2SJeff Cody 
23736ebdcee2SJeff Cody typedef struct BlkIntermediateStates {
23746ebdcee2SJeff Cody     BlockDriverState *bs;
23756ebdcee2SJeff Cody     QSIMPLEQ_ENTRY(BlkIntermediateStates) entry;
23766ebdcee2SJeff Cody } BlkIntermediateStates;
23776ebdcee2SJeff Cody 
23786ebdcee2SJeff Cody 
23796ebdcee2SJeff Cody /*
23806ebdcee2SJeff Cody  * Drops images above 'base' up to and including 'top', and sets the image
23816ebdcee2SJeff Cody  * above 'top' to have base as its backing file.
23826ebdcee2SJeff Cody  *
23836ebdcee2SJeff Cody  * Requires that the overlay to 'top' is opened r/w, so that the backing file
23846ebdcee2SJeff Cody  * information in 'bs' can be properly updated.
23856ebdcee2SJeff Cody  *
23866ebdcee2SJeff Cody  * E.g., this will convert the following chain:
23876ebdcee2SJeff Cody  * bottom <- base <- intermediate <- top <- active
23886ebdcee2SJeff Cody  *
23896ebdcee2SJeff Cody  * to
23906ebdcee2SJeff Cody  *
23916ebdcee2SJeff Cody  * bottom <- base <- active
23926ebdcee2SJeff Cody  *
23936ebdcee2SJeff Cody  * It is allowed for bottom==base, in which case it converts:
23946ebdcee2SJeff Cody  *
23956ebdcee2SJeff Cody  * base <- intermediate <- top <- active
23966ebdcee2SJeff Cody  *
23976ebdcee2SJeff Cody  * to
23986ebdcee2SJeff Cody  *
23996ebdcee2SJeff Cody  * base <- active
24006ebdcee2SJeff Cody  *
24016ebdcee2SJeff Cody  * Error conditions:
24026ebdcee2SJeff Cody  *  if active == top, that is considered an error
24036ebdcee2SJeff Cody  *
24046ebdcee2SJeff Cody  */
24056ebdcee2SJeff Cody int bdrv_drop_intermediate(BlockDriverState *active, BlockDriverState *top,
24066ebdcee2SJeff Cody                            BlockDriverState *base)
24076ebdcee2SJeff Cody {
24086ebdcee2SJeff Cody     BlockDriverState *intermediate;
24096ebdcee2SJeff Cody     BlockDriverState *base_bs = NULL;
24106ebdcee2SJeff Cody     BlockDriverState *new_top_bs = NULL;
24116ebdcee2SJeff Cody     BlkIntermediateStates *intermediate_state, *next;
24126ebdcee2SJeff Cody     int ret = -EIO;
24136ebdcee2SJeff Cody 
24146ebdcee2SJeff Cody     QSIMPLEQ_HEAD(states_to_delete, BlkIntermediateStates) states_to_delete;
24156ebdcee2SJeff Cody     QSIMPLEQ_INIT(&states_to_delete);
24166ebdcee2SJeff Cody 
24176ebdcee2SJeff Cody     if (!top->drv || !base->drv) {
24186ebdcee2SJeff Cody         goto exit;
24196ebdcee2SJeff Cody     }
24206ebdcee2SJeff Cody 
24216ebdcee2SJeff Cody     new_top_bs = bdrv_find_overlay(active, top);
24226ebdcee2SJeff Cody 
24236ebdcee2SJeff Cody     if (new_top_bs == NULL) {
24246ebdcee2SJeff Cody         /* we could not find the image above 'top', this is an error */
24256ebdcee2SJeff Cody         goto exit;
24266ebdcee2SJeff Cody     }
24276ebdcee2SJeff Cody 
24286ebdcee2SJeff Cody     /* special case of new_top_bs->backing_hd already pointing to base - nothing
24296ebdcee2SJeff Cody      * to do, no intermediate images */
24306ebdcee2SJeff Cody     if (new_top_bs->backing_hd == base) {
24316ebdcee2SJeff Cody         ret = 0;
24326ebdcee2SJeff Cody         goto exit;
24336ebdcee2SJeff Cody     }
24346ebdcee2SJeff Cody 
24356ebdcee2SJeff Cody     intermediate = top;
24366ebdcee2SJeff Cody 
24376ebdcee2SJeff Cody     /* now we will go down through the list, and add each BDS we find
24386ebdcee2SJeff Cody      * into our deletion queue, until we hit the 'base'
24396ebdcee2SJeff Cody      */
24406ebdcee2SJeff Cody     while (intermediate) {
24416ebdcee2SJeff Cody         intermediate_state = g_malloc0(sizeof(BlkIntermediateStates));
24426ebdcee2SJeff Cody         intermediate_state->bs = intermediate;
24436ebdcee2SJeff Cody         QSIMPLEQ_INSERT_TAIL(&states_to_delete, intermediate_state, entry);
24446ebdcee2SJeff Cody 
24456ebdcee2SJeff Cody         if (intermediate->backing_hd == base) {
24466ebdcee2SJeff Cody             base_bs = intermediate->backing_hd;
24476ebdcee2SJeff Cody             break;
24486ebdcee2SJeff Cody         }
24496ebdcee2SJeff Cody         intermediate = intermediate->backing_hd;
24506ebdcee2SJeff Cody     }
24516ebdcee2SJeff Cody     if (base_bs == NULL) {
24526ebdcee2SJeff Cody         /* something went wrong, we did not end at the base. safely
24536ebdcee2SJeff Cody          * unravel everything, and exit with error */
24546ebdcee2SJeff Cody         goto exit;
24556ebdcee2SJeff Cody     }
24566ebdcee2SJeff Cody 
24576ebdcee2SJeff Cody     /* success - we can delete the intermediate states, and link top->base */
24586ebdcee2SJeff Cody     ret = bdrv_change_backing_file(new_top_bs, base_bs->filename,
24596ebdcee2SJeff Cody                                    base_bs->drv ? base_bs->drv->format_name : "");
24606ebdcee2SJeff Cody     if (ret) {
24616ebdcee2SJeff Cody         goto exit;
24626ebdcee2SJeff Cody     }
24636ebdcee2SJeff Cody     new_top_bs->backing_hd = base_bs;
24646ebdcee2SJeff Cody 
2465355ef4acSKevin Wolf     bdrv_refresh_limits(new_top_bs);
24666ebdcee2SJeff Cody 
24676ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
24686ebdcee2SJeff Cody         /* so that bdrv_close() does not recursively close the chain */
24696ebdcee2SJeff Cody         intermediate_state->bs->backing_hd = NULL;
24704f6fd349SFam Zheng         bdrv_unref(intermediate_state->bs);
24716ebdcee2SJeff Cody     }
24726ebdcee2SJeff Cody     ret = 0;
24736ebdcee2SJeff Cody 
24746ebdcee2SJeff Cody exit:
24756ebdcee2SJeff Cody     QSIMPLEQ_FOREACH_SAFE(intermediate_state, &states_to_delete, entry, next) {
24766ebdcee2SJeff Cody         g_free(intermediate_state);
24776ebdcee2SJeff Cody     }
24786ebdcee2SJeff Cody     return ret;
24796ebdcee2SJeff Cody }
24806ebdcee2SJeff Cody 
24816ebdcee2SJeff Cody 
248271d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
248371d0770cSaliguori                                    size_t size)
248471d0770cSaliguori {
248571d0770cSaliguori     int64_t len;
248671d0770cSaliguori 
248771d0770cSaliguori     if (!bdrv_is_inserted(bs))
248871d0770cSaliguori         return -ENOMEDIUM;
248971d0770cSaliguori 
249071d0770cSaliguori     if (bs->growable)
249171d0770cSaliguori         return 0;
249271d0770cSaliguori 
249371d0770cSaliguori     len = bdrv_getlength(bs);
249471d0770cSaliguori 
2495fbb7b4e0SKevin Wolf     if (offset < 0)
2496fbb7b4e0SKevin Wolf         return -EIO;
2497fbb7b4e0SKevin Wolf 
2498fbb7b4e0SKevin Wolf     if ((offset > len) || (len - offset < size))
249971d0770cSaliguori         return -EIO;
250071d0770cSaliguori 
250171d0770cSaliguori     return 0;
250271d0770cSaliguori }
250371d0770cSaliguori 
250471d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
250571d0770cSaliguori                               int nb_sectors)
250671d0770cSaliguori {
2507eb5a3165SJes Sorensen     return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
2508eb5a3165SJes Sorensen                                    nb_sectors * BDRV_SECTOR_SIZE);
250971d0770cSaliguori }
251071d0770cSaliguori 
25111c9805a3SStefan Hajnoczi typedef struct RwCo {
25121c9805a3SStefan Hajnoczi     BlockDriverState *bs;
25131c9805a3SStefan Hajnoczi     int64_t sector_num;
25141c9805a3SStefan Hajnoczi     int nb_sectors;
25151c9805a3SStefan Hajnoczi     QEMUIOVector *qiov;
25161c9805a3SStefan Hajnoczi     bool is_write;
25171c9805a3SStefan Hajnoczi     int ret;
25184105eaaaSPeter Lieven     BdrvRequestFlags flags;
25191c9805a3SStefan Hajnoczi } RwCo;
25201c9805a3SStefan Hajnoczi 
25211c9805a3SStefan Hajnoczi static void coroutine_fn bdrv_rw_co_entry(void *opaque)
2522fc01f7e7Sbellard {
25231c9805a3SStefan Hajnoczi     RwCo *rwco = opaque;
2524fc01f7e7Sbellard 
25251c9805a3SStefan Hajnoczi     if (!rwco->is_write) {
25261c9805a3SStefan Hajnoczi         rwco->ret = bdrv_co_do_readv(rwco->bs, rwco->sector_num,
25274105eaaaSPeter Lieven                                      rwco->nb_sectors, rwco->qiov,
25284105eaaaSPeter Lieven                                      rwco->flags);
25291c9805a3SStefan Hajnoczi     } else {
25301c9805a3SStefan Hajnoczi         rwco->ret = bdrv_co_do_writev(rwco->bs, rwco->sector_num,
25314105eaaaSPeter Lieven                                       rwco->nb_sectors, rwco->qiov,
25324105eaaaSPeter Lieven                                       rwco->flags);
25331c9805a3SStefan Hajnoczi     }
25341c9805a3SStefan Hajnoczi }
2535e7a8a783SKevin Wolf 
25361c9805a3SStefan Hajnoczi /*
25378d3b1a2dSKevin Wolf  * Process a vectored synchronous request using coroutines
25381c9805a3SStefan Hajnoczi  */
25398d3b1a2dSKevin Wolf static int bdrv_rwv_co(BlockDriverState *bs, int64_t sector_num,
25404105eaaaSPeter Lieven                        QEMUIOVector *qiov, bool is_write,
25414105eaaaSPeter Lieven                        BdrvRequestFlags flags)
25421c9805a3SStefan Hajnoczi {
25431c9805a3SStefan Hajnoczi     Coroutine *co;
25441c9805a3SStefan Hajnoczi     RwCo rwco = {
25451c9805a3SStefan Hajnoczi         .bs = bs,
25461c9805a3SStefan Hajnoczi         .sector_num = sector_num,
25478d3b1a2dSKevin Wolf         .nb_sectors = qiov->size >> BDRV_SECTOR_BITS,
25488d3b1a2dSKevin Wolf         .qiov = qiov,
25491c9805a3SStefan Hajnoczi         .is_write = is_write,
25501c9805a3SStefan Hajnoczi         .ret = NOT_DONE,
25514105eaaaSPeter Lieven         .flags = flags,
25521c9805a3SStefan Hajnoczi     };
25538d3b1a2dSKevin Wolf     assert((qiov->size & (BDRV_SECTOR_SIZE - 1)) == 0);
25541c9805a3SStefan Hajnoczi 
2555498e386cSZhi Yong Wu     /**
2556498e386cSZhi Yong Wu      * In sync call context, when the vcpu is blocked, this throttling timer
2557498e386cSZhi Yong Wu      * will not fire; so the I/O throttling function has to be disabled here
2558498e386cSZhi Yong Wu      * if it has been enabled.
2559498e386cSZhi Yong Wu      */
2560498e386cSZhi Yong Wu     if (bs->io_limits_enabled) {
2561498e386cSZhi Yong Wu         fprintf(stderr, "Disabling I/O throttling on '%s' due "
2562498e386cSZhi Yong Wu                         "to synchronous I/O.\n", bdrv_get_device_name(bs));
2563498e386cSZhi Yong Wu         bdrv_io_limits_disable(bs);
2564498e386cSZhi Yong Wu     }
2565498e386cSZhi Yong Wu 
25661c9805a3SStefan Hajnoczi     if (qemu_in_coroutine()) {
25671c9805a3SStefan Hajnoczi         /* Fast-path if already in coroutine context */
25681c9805a3SStefan Hajnoczi         bdrv_rw_co_entry(&rwco);
25691c9805a3SStefan Hajnoczi     } else {
25701c9805a3SStefan Hajnoczi         co = qemu_coroutine_create(bdrv_rw_co_entry);
25711c9805a3SStefan Hajnoczi         qemu_coroutine_enter(co, &rwco);
25721c9805a3SStefan Hajnoczi         while (rwco.ret == NOT_DONE) {
25731c9805a3SStefan Hajnoczi             qemu_aio_wait();
25741c9805a3SStefan Hajnoczi         }
25751c9805a3SStefan Hajnoczi     }
25761c9805a3SStefan Hajnoczi     return rwco.ret;
2577e7a8a783SKevin Wolf }
2578e7a8a783SKevin Wolf 
25798d3b1a2dSKevin Wolf /*
25808d3b1a2dSKevin Wolf  * Process a synchronous request using coroutines
25818d3b1a2dSKevin Wolf  */
25828d3b1a2dSKevin Wolf static int bdrv_rw_co(BlockDriverState *bs, int64_t sector_num, uint8_t *buf,
25834105eaaaSPeter Lieven                       int nb_sectors, bool is_write, BdrvRequestFlags flags)
25848d3b1a2dSKevin Wolf {
25858d3b1a2dSKevin Wolf     QEMUIOVector qiov;
25868d3b1a2dSKevin Wolf     struct iovec iov = {
25878d3b1a2dSKevin Wolf         .iov_base = (void *)buf,
25888d3b1a2dSKevin Wolf         .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
25898d3b1a2dSKevin Wolf     };
25908d3b1a2dSKevin Wolf 
25918d3b1a2dSKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
25924105eaaaSPeter Lieven     return bdrv_rwv_co(bs, sector_num, &qiov, is_write, flags);
25938d3b1a2dSKevin Wolf }
25948d3b1a2dSKevin Wolf 
25951c9805a3SStefan Hajnoczi /* return < 0 if error. See bdrv_write() for the return codes */
25961c9805a3SStefan Hajnoczi int bdrv_read(BlockDriverState *bs, int64_t sector_num,
25971c9805a3SStefan Hajnoczi               uint8_t *buf, int nb_sectors)
25981c9805a3SStefan Hajnoczi {
25994105eaaaSPeter Lieven     return bdrv_rw_co(bs, sector_num, buf, nb_sectors, false, 0);
260083f64091Sbellard }
2601fc01f7e7Sbellard 
260207d27a44SMarkus Armbruster /* Just like bdrv_read(), but with I/O throttling temporarily disabled */
260307d27a44SMarkus Armbruster int bdrv_read_unthrottled(BlockDriverState *bs, int64_t sector_num,
260407d27a44SMarkus Armbruster                           uint8_t *buf, int nb_sectors)
260507d27a44SMarkus Armbruster {
260607d27a44SMarkus Armbruster     bool enabled;
260707d27a44SMarkus Armbruster     int ret;
260807d27a44SMarkus Armbruster 
260907d27a44SMarkus Armbruster     enabled = bs->io_limits_enabled;
261007d27a44SMarkus Armbruster     bs->io_limits_enabled = false;
26114e7395e8SPeter Lieven     ret = bdrv_read(bs, sector_num, buf, nb_sectors);
261207d27a44SMarkus Armbruster     bs->io_limits_enabled = enabled;
261307d27a44SMarkus Armbruster     return ret;
261407d27a44SMarkus Armbruster }
261507d27a44SMarkus Armbruster 
261619cb3738Sbellard /* Return < 0 if error. Important errors are:
261719cb3738Sbellard   -EIO         generic I/O error (may happen for all errors)
261819cb3738Sbellard   -ENOMEDIUM   No media inserted.
261919cb3738Sbellard   -EINVAL      Invalid sector number or nb_sectors
262019cb3738Sbellard   -EACCES      Trying to write a read-only device
262119cb3738Sbellard */
2622fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
2623fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
2624fc01f7e7Sbellard {
26254105eaaaSPeter Lieven     return bdrv_rw_co(bs, sector_num, (uint8_t *)buf, nb_sectors, true, 0);
262683f64091Sbellard }
262783f64091Sbellard 
26288d3b1a2dSKevin Wolf int bdrv_writev(BlockDriverState *bs, int64_t sector_num, QEMUIOVector *qiov)
26298d3b1a2dSKevin Wolf {
26304105eaaaSPeter Lieven     return bdrv_rwv_co(bs, sector_num, qiov, true, 0);
26314105eaaaSPeter Lieven }
26324105eaaaSPeter Lieven 
2633aa7bfbffSPeter Lieven int bdrv_write_zeroes(BlockDriverState *bs, int64_t sector_num,
2634aa7bfbffSPeter Lieven                       int nb_sectors, BdrvRequestFlags flags)
26354105eaaaSPeter Lieven {
26364105eaaaSPeter Lieven     return bdrv_rw_co(bs, sector_num, NULL, nb_sectors, true,
2637aa7bfbffSPeter Lieven                       BDRV_REQ_ZERO_WRITE | flags);
26388d3b1a2dSKevin Wolf }
26398d3b1a2dSKevin Wolf 
2640d75cbb5eSPeter Lieven /*
2641d75cbb5eSPeter Lieven  * Completely zero out a block device with the help of bdrv_write_zeroes.
2642d75cbb5eSPeter Lieven  * The operation is sped up by checking the block status and only writing
2643d75cbb5eSPeter Lieven  * zeroes to the device if they currently do not return zeroes. Optional
2644d75cbb5eSPeter Lieven  * flags are passed through to bdrv_write_zeroes (e.g. BDRV_REQ_MAY_UNMAP).
2645d75cbb5eSPeter Lieven  *
2646d75cbb5eSPeter Lieven  * Returns < 0 on error, 0 on success. For error codes see bdrv_write().
2647d75cbb5eSPeter Lieven  */
2648d75cbb5eSPeter Lieven int bdrv_make_zero(BlockDriverState *bs, BdrvRequestFlags flags)
2649d75cbb5eSPeter Lieven {
2650d75cbb5eSPeter Lieven     int64_t target_size = bdrv_getlength(bs) / BDRV_SECTOR_SIZE;
2651d75cbb5eSPeter Lieven     int64_t ret, nb_sectors, sector_num = 0;
2652d75cbb5eSPeter Lieven     int n;
2653d75cbb5eSPeter Lieven 
2654d75cbb5eSPeter Lieven     for (;;) {
2655d75cbb5eSPeter Lieven         nb_sectors = target_size - sector_num;
2656d75cbb5eSPeter Lieven         if (nb_sectors <= 0) {
2657d75cbb5eSPeter Lieven             return 0;
2658d75cbb5eSPeter Lieven         }
2659d75cbb5eSPeter Lieven         if (nb_sectors > INT_MAX) {
2660d75cbb5eSPeter Lieven             nb_sectors = INT_MAX;
2661d75cbb5eSPeter Lieven         }
2662d75cbb5eSPeter Lieven         ret = bdrv_get_block_status(bs, sector_num, nb_sectors, &n);
26633d94ce60SPeter Lieven         if (ret < 0) {
26643d94ce60SPeter Lieven             error_report("error getting block status at sector %" PRId64 ": %s",
26653d94ce60SPeter Lieven                          sector_num, strerror(-ret));
26663d94ce60SPeter Lieven             return ret;
26673d94ce60SPeter Lieven         }
2668d75cbb5eSPeter Lieven         if (ret & BDRV_BLOCK_ZERO) {
2669d75cbb5eSPeter Lieven             sector_num += n;
2670d75cbb5eSPeter Lieven             continue;
2671d75cbb5eSPeter Lieven         }
2672d75cbb5eSPeter Lieven         ret = bdrv_write_zeroes(bs, sector_num, n, flags);
2673d75cbb5eSPeter Lieven         if (ret < 0) {
2674d75cbb5eSPeter Lieven             error_report("error writing zeroes at sector %" PRId64 ": %s",
2675d75cbb5eSPeter Lieven                          sector_num, strerror(-ret));
2676d75cbb5eSPeter Lieven             return ret;
2677d75cbb5eSPeter Lieven         }
2678d75cbb5eSPeter Lieven         sector_num += n;
2679d75cbb5eSPeter Lieven     }
2680d75cbb5eSPeter Lieven }
2681d75cbb5eSPeter Lieven 
2682eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset,
2683eda578e5Saliguori                void *buf, int count1)
268483f64091Sbellard {
26856ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
268683f64091Sbellard     int len, nb_sectors, count;
268783f64091Sbellard     int64_t sector_num;
26889a8c4cceSKevin Wolf     int ret;
268983f64091Sbellard 
269083f64091Sbellard     count = count1;
269183f64091Sbellard     /* first read to align to sector start */
26926ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
269383f64091Sbellard     if (len > count)
269483f64091Sbellard         len = count;
26956ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
269683f64091Sbellard     if (len > 0) {
26979a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
26989a8c4cceSKevin Wolf             return ret;
26996ea44308SJan Kiszka         memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
270083f64091Sbellard         count -= len;
270183f64091Sbellard         if (count == 0)
270283f64091Sbellard             return count1;
270383f64091Sbellard         sector_num++;
270483f64091Sbellard         buf += len;
270583f64091Sbellard     }
270683f64091Sbellard 
270783f64091Sbellard     /* read the sectors "in place" */
27086ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
270983f64091Sbellard     if (nb_sectors > 0) {
27109a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
27119a8c4cceSKevin Wolf             return ret;
271283f64091Sbellard         sector_num += nb_sectors;
27136ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
271483f64091Sbellard         buf += len;
271583f64091Sbellard         count -= len;
271683f64091Sbellard     }
271783f64091Sbellard 
271883f64091Sbellard     /* add data from the last sector */
271983f64091Sbellard     if (count > 0) {
27209a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
27219a8c4cceSKevin Wolf             return ret;
272283f64091Sbellard         memcpy(buf, tmp_buf, count);
272383f64091Sbellard     }
272483f64091Sbellard     return count1;
272583f64091Sbellard }
272683f64091Sbellard 
27278d3b1a2dSKevin Wolf int bdrv_pwritev(BlockDriverState *bs, int64_t offset, QEMUIOVector *qiov)
272883f64091Sbellard {
27296ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
273083f64091Sbellard     int len, nb_sectors, count;
273183f64091Sbellard     int64_t sector_num;
27329a8c4cceSKevin Wolf     int ret;
273383f64091Sbellard 
27348d3b1a2dSKevin Wolf     count = qiov->size;
27358d3b1a2dSKevin Wolf 
273683f64091Sbellard     /* first write to align to sector start */
27376ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
273883f64091Sbellard     if (len > count)
273983f64091Sbellard         len = count;
27406ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
274183f64091Sbellard     if (len > 0) {
27429a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
27439a8c4cceSKevin Wolf             return ret;
27448d3b1a2dSKevin Wolf         qemu_iovec_to_buf(qiov, 0, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)),
27458d3b1a2dSKevin Wolf                           len);
27469a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
27479a8c4cceSKevin Wolf             return ret;
274883f64091Sbellard         count -= len;
274983f64091Sbellard         if (count == 0)
27508d3b1a2dSKevin Wolf             return qiov->size;
275183f64091Sbellard         sector_num++;
275283f64091Sbellard     }
275383f64091Sbellard 
275483f64091Sbellard     /* write the sectors "in place" */
27556ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
275683f64091Sbellard     if (nb_sectors > 0) {
27578d3b1a2dSKevin Wolf         QEMUIOVector qiov_inplace;
27588d3b1a2dSKevin Wolf 
27598d3b1a2dSKevin Wolf         qemu_iovec_init(&qiov_inplace, qiov->niov);
27608d3b1a2dSKevin Wolf         qemu_iovec_concat(&qiov_inplace, qiov, len,
27618d3b1a2dSKevin Wolf                           nb_sectors << BDRV_SECTOR_BITS);
27628d3b1a2dSKevin Wolf         ret = bdrv_writev(bs, sector_num, &qiov_inplace);
27638d3b1a2dSKevin Wolf         qemu_iovec_destroy(&qiov_inplace);
27648d3b1a2dSKevin Wolf         if (ret < 0) {
27659a8c4cceSKevin Wolf             return ret;
27668d3b1a2dSKevin Wolf         }
27678d3b1a2dSKevin Wolf 
276883f64091Sbellard         sector_num += nb_sectors;
27696ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
277083f64091Sbellard         count -= len;
277183f64091Sbellard     }
277283f64091Sbellard 
277383f64091Sbellard     /* add data from the last sector */
277483f64091Sbellard     if (count > 0) {
27759a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
27769a8c4cceSKevin Wolf             return ret;
27778d3b1a2dSKevin Wolf         qemu_iovec_to_buf(qiov, qiov->size - count, tmp_buf, count);
27789a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
27799a8c4cceSKevin Wolf             return ret;
278083f64091Sbellard     }
27818d3b1a2dSKevin Wolf     return qiov->size;
27828d3b1a2dSKevin Wolf }
27838d3b1a2dSKevin Wolf 
27848d3b1a2dSKevin Wolf int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
27858d3b1a2dSKevin Wolf                 const void *buf, int count1)
27868d3b1a2dSKevin Wolf {
27878d3b1a2dSKevin Wolf     QEMUIOVector qiov;
27888d3b1a2dSKevin Wolf     struct iovec iov = {
27898d3b1a2dSKevin Wolf         .iov_base   = (void *) buf,
27908d3b1a2dSKevin Wolf         .iov_len    = count1,
27918d3b1a2dSKevin Wolf     };
27928d3b1a2dSKevin Wolf 
27938d3b1a2dSKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
27948d3b1a2dSKevin Wolf     return bdrv_pwritev(bs, offset, &qiov);
279583f64091Sbellard }
279683f64091Sbellard 
2797f08145feSKevin Wolf /*
2798f08145feSKevin Wolf  * Writes to the file and ensures that no writes are reordered across this
2799f08145feSKevin Wolf  * request (acts as a barrier)
2800f08145feSKevin Wolf  *
2801f08145feSKevin Wolf  * Returns 0 on success, -errno in error cases.
2802f08145feSKevin Wolf  */
2803f08145feSKevin Wolf int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
2804f08145feSKevin Wolf     const void *buf, int count)
2805f08145feSKevin Wolf {
2806f08145feSKevin Wolf     int ret;
2807f08145feSKevin Wolf 
2808f08145feSKevin Wolf     ret = bdrv_pwrite(bs, offset, buf, count);
2809f08145feSKevin Wolf     if (ret < 0) {
2810f08145feSKevin Wolf         return ret;
2811f08145feSKevin Wolf     }
2812f08145feSKevin Wolf 
2813f05fa4adSPaolo Bonzini     /* No flush needed for cache modes that already do it */
2814f05fa4adSPaolo Bonzini     if (bs->enable_write_cache) {
2815f08145feSKevin Wolf         bdrv_flush(bs);
2816f08145feSKevin Wolf     }
2817f08145feSKevin Wolf 
2818f08145feSKevin Wolf     return 0;
2819f08145feSKevin Wolf }
2820f08145feSKevin Wolf 
2821470c0504SStefan Hajnoczi static int coroutine_fn bdrv_co_do_copy_on_readv(BlockDriverState *bs,
2822ab185921SStefan Hajnoczi         int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
2823ab185921SStefan Hajnoczi {
2824ab185921SStefan Hajnoczi     /* Perform I/O through a temporary buffer so that users who scribble over
2825ab185921SStefan Hajnoczi      * their read buffer while the operation is in progress do not end up
2826ab185921SStefan Hajnoczi      * modifying the image file.  This is critical for zero-copy guest I/O
2827ab185921SStefan Hajnoczi      * where anything might happen inside guest memory.
2828ab185921SStefan Hajnoczi      */
2829ab185921SStefan Hajnoczi     void *bounce_buffer;
2830ab185921SStefan Hajnoczi 
283179c053bdSStefan Hajnoczi     BlockDriver *drv = bs->drv;
2832ab185921SStefan Hajnoczi     struct iovec iov;
2833ab185921SStefan Hajnoczi     QEMUIOVector bounce_qiov;
2834ab185921SStefan Hajnoczi     int64_t cluster_sector_num;
2835ab185921SStefan Hajnoczi     int cluster_nb_sectors;
2836ab185921SStefan Hajnoczi     size_t skip_bytes;
2837ab185921SStefan Hajnoczi     int ret;
2838ab185921SStefan Hajnoczi 
2839ab185921SStefan Hajnoczi     /* Cover entire cluster so no additional backing file I/O is required when
2840ab185921SStefan Hajnoczi      * allocating cluster in the image file.
2841ab185921SStefan Hajnoczi      */
2842343bded4SPaolo Bonzini     bdrv_round_to_clusters(bs, sector_num, nb_sectors,
2843ab185921SStefan Hajnoczi                            &cluster_sector_num, &cluster_nb_sectors);
2844ab185921SStefan Hajnoczi 
2845470c0504SStefan Hajnoczi     trace_bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors,
2846ab185921SStefan Hajnoczi                                    cluster_sector_num, cluster_nb_sectors);
2847ab185921SStefan Hajnoczi 
2848ab185921SStefan Hajnoczi     iov.iov_len = cluster_nb_sectors * BDRV_SECTOR_SIZE;
2849ab185921SStefan Hajnoczi     iov.iov_base = bounce_buffer = qemu_blockalign(bs, iov.iov_len);
2850ab185921SStefan Hajnoczi     qemu_iovec_init_external(&bounce_qiov, &iov, 1);
2851ab185921SStefan Hajnoczi 
285279c053bdSStefan Hajnoczi     ret = drv->bdrv_co_readv(bs, cluster_sector_num, cluster_nb_sectors,
2853ab185921SStefan Hajnoczi                              &bounce_qiov);
2854ab185921SStefan Hajnoczi     if (ret < 0) {
2855ab185921SStefan Hajnoczi         goto err;
2856ab185921SStefan Hajnoczi     }
2857ab185921SStefan Hajnoczi 
285879c053bdSStefan Hajnoczi     if (drv->bdrv_co_write_zeroes &&
285979c053bdSStefan Hajnoczi         buffer_is_zero(bounce_buffer, iov.iov_len)) {
2860621f0589SKevin Wolf         ret = bdrv_co_do_write_zeroes(bs, cluster_sector_num,
2861aa7bfbffSPeter Lieven                                       cluster_nb_sectors, 0);
286279c053bdSStefan Hajnoczi     } else {
2863f05fa4adSPaolo Bonzini         /* This does not change the data on the disk, it is not necessary
2864f05fa4adSPaolo Bonzini          * to flush even in cache=writethrough mode.
2865f05fa4adSPaolo Bonzini          */
286679c053bdSStefan Hajnoczi         ret = drv->bdrv_co_writev(bs, cluster_sector_num, cluster_nb_sectors,
2867ab185921SStefan Hajnoczi                                   &bounce_qiov);
286879c053bdSStefan Hajnoczi     }
286979c053bdSStefan Hajnoczi 
2870ab185921SStefan Hajnoczi     if (ret < 0) {
2871ab185921SStefan Hajnoczi         /* It might be okay to ignore write errors for guest requests.  If this
2872ab185921SStefan Hajnoczi          * is a deliberate copy-on-read then we don't want to ignore the error.
2873ab185921SStefan Hajnoczi          * Simply report it in all cases.
2874ab185921SStefan Hajnoczi          */
2875ab185921SStefan Hajnoczi         goto err;
2876ab185921SStefan Hajnoczi     }
2877ab185921SStefan Hajnoczi 
2878ab185921SStefan Hajnoczi     skip_bytes = (sector_num - cluster_sector_num) * BDRV_SECTOR_SIZE;
287903396148SMichael Tokarev     qemu_iovec_from_buf(qiov, 0, bounce_buffer + skip_bytes,
2880ab185921SStefan Hajnoczi                         nb_sectors * BDRV_SECTOR_SIZE);
2881ab185921SStefan Hajnoczi 
2882ab185921SStefan Hajnoczi err:
2883ab185921SStefan Hajnoczi     qemu_vfree(bounce_buffer);
2884ab185921SStefan Hajnoczi     return ret;
2885ab185921SStefan Hajnoczi }
2886ab185921SStefan Hajnoczi 
2887c5fbe571SStefan Hajnoczi /*
2888d0c7f642SKevin Wolf  * Forwards an already correctly aligned request to the BlockDriver. This
2889d0c7f642SKevin Wolf  * handles copy on read and zeroing after EOF; any other features must be
2890d0c7f642SKevin Wolf  * implemented by the caller.
2891c5fbe571SStefan Hajnoczi  */
2892d0c7f642SKevin Wolf static int coroutine_fn bdrv_aligned_preadv(BlockDriverState *bs,
2893d0c7f642SKevin Wolf     int64_t offset, unsigned int bytes, QEMUIOVector *qiov, int flags)
2894da1fa91dSKevin Wolf {
2895da1fa91dSKevin Wolf     BlockDriver *drv = bs->drv;
2896dbffbdcfSStefan Hajnoczi     BdrvTrackedRequest req;
2897dbffbdcfSStefan Hajnoczi     int ret;
2898da1fa91dSKevin Wolf 
2899d0c7f642SKevin Wolf     int64_t sector_num = offset >> BDRV_SECTOR_BITS;
2900d0c7f642SKevin Wolf     unsigned int nb_sectors = bytes >> BDRV_SECTOR_BITS;
2901da1fa91dSKevin Wolf 
2902d0c7f642SKevin Wolf     assert((offset & (BDRV_SECTOR_SIZE - 1)) == 0);
2903d0c7f642SKevin Wolf     assert((bytes & (BDRV_SECTOR_SIZE - 1)) == 0);
2904d0c7f642SKevin Wolf 
2905d0c7f642SKevin Wolf     /* Handle Copy on Read and associated serialisation */
2906470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2907470c0504SStefan Hajnoczi         bs->copy_on_read_in_flight++;
2908470c0504SStefan Hajnoczi     }
2909470c0504SStefan Hajnoczi 
2910470c0504SStefan Hajnoczi     if (bs->copy_on_read_in_flight) {
2911f4658285SStefan Hajnoczi         wait_for_overlapping_requests(bs, sector_num, nb_sectors);
2912f4658285SStefan Hajnoczi     }
2913f4658285SStefan Hajnoczi 
2914dbffbdcfSStefan Hajnoczi     tracked_request_begin(&req, bs, sector_num, nb_sectors, false);
2915ab185921SStefan Hajnoczi 
2916470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2917ab185921SStefan Hajnoczi         int pnum;
2918ab185921SStefan Hajnoczi 
2919bdad13b9SPaolo Bonzini         ret = bdrv_is_allocated(bs, sector_num, nb_sectors, &pnum);
2920ab185921SStefan Hajnoczi         if (ret < 0) {
2921ab185921SStefan Hajnoczi             goto out;
2922ab185921SStefan Hajnoczi         }
2923ab185921SStefan Hajnoczi 
2924ab185921SStefan Hajnoczi         if (!ret || pnum != nb_sectors) {
2925470c0504SStefan Hajnoczi             ret = bdrv_co_do_copy_on_readv(bs, sector_num, nb_sectors, qiov);
2926ab185921SStefan Hajnoczi             goto out;
2927ab185921SStefan Hajnoczi         }
2928ab185921SStefan Hajnoczi     }
2929ab185921SStefan Hajnoczi 
2930d0c7f642SKevin Wolf     /* Forward the request to the BlockDriver */
2931893a8f62SMORITA Kazutaka     if (!(bs->zero_beyond_eof && bs->growable)) {
2932dbffbdcfSStefan Hajnoczi         ret = drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
2933893a8f62SMORITA Kazutaka     } else {
2934893a8f62SMORITA Kazutaka         /* Read zeros after EOF of growable BDSes */
2935893a8f62SMORITA Kazutaka         int64_t len, total_sectors, max_nb_sectors;
2936893a8f62SMORITA Kazutaka 
2937893a8f62SMORITA Kazutaka         len = bdrv_getlength(bs);
2938893a8f62SMORITA Kazutaka         if (len < 0) {
2939893a8f62SMORITA Kazutaka             ret = len;
2940893a8f62SMORITA Kazutaka             goto out;
2941893a8f62SMORITA Kazutaka         }
2942893a8f62SMORITA Kazutaka 
2943d055a1feSFam Zheng         total_sectors = DIV_ROUND_UP(len, BDRV_SECTOR_SIZE);
2944893a8f62SMORITA Kazutaka         max_nb_sectors = MAX(0, total_sectors - sector_num);
2945893a8f62SMORITA Kazutaka         if (max_nb_sectors > 0) {
2946893a8f62SMORITA Kazutaka             ret = drv->bdrv_co_readv(bs, sector_num,
2947893a8f62SMORITA Kazutaka                                      MIN(nb_sectors, max_nb_sectors), qiov);
2948893a8f62SMORITA Kazutaka         } else {
2949893a8f62SMORITA Kazutaka             ret = 0;
2950893a8f62SMORITA Kazutaka         }
2951893a8f62SMORITA Kazutaka 
2952893a8f62SMORITA Kazutaka         /* Reading beyond end of file is supposed to produce zeroes */
2953893a8f62SMORITA Kazutaka         if (ret == 0 && total_sectors < sector_num + nb_sectors) {
2954893a8f62SMORITA Kazutaka             uint64_t offset = MAX(0, total_sectors - sector_num);
2955893a8f62SMORITA Kazutaka             uint64_t bytes = (sector_num + nb_sectors - offset) *
2956893a8f62SMORITA Kazutaka                               BDRV_SECTOR_SIZE;
2957893a8f62SMORITA Kazutaka             qemu_iovec_memset(qiov, offset * BDRV_SECTOR_SIZE, 0, bytes);
2958893a8f62SMORITA Kazutaka         }
2959893a8f62SMORITA Kazutaka     }
2960ab185921SStefan Hajnoczi 
2961ab185921SStefan Hajnoczi out:
2962dbffbdcfSStefan Hajnoczi     tracked_request_end(&req);
2963470c0504SStefan Hajnoczi 
2964470c0504SStefan Hajnoczi     if (flags & BDRV_REQ_COPY_ON_READ) {
2965470c0504SStefan Hajnoczi         bs->copy_on_read_in_flight--;
2966470c0504SStefan Hajnoczi     }
2967470c0504SStefan Hajnoczi 
2968dbffbdcfSStefan Hajnoczi     return ret;
2969da1fa91dSKevin Wolf }
2970da1fa91dSKevin Wolf 
2971d0c7f642SKevin Wolf /*
2972d0c7f642SKevin Wolf  * Handle a read request in coroutine context
2973d0c7f642SKevin Wolf  */
29741b0288aeSKevin Wolf static int coroutine_fn bdrv_co_do_preadv(BlockDriverState *bs,
29751b0288aeSKevin Wolf     int64_t offset, unsigned int bytes, QEMUIOVector *qiov,
2976d0c7f642SKevin Wolf     BdrvRequestFlags flags)
2977d0c7f642SKevin Wolf {
2978d0c7f642SKevin Wolf     BlockDriver *drv = bs->drv;
29791b0288aeSKevin Wolf     /* TODO Lift BDRV_SECTOR_SIZE restriction in BlockDriver interface */
29801b0288aeSKevin Wolf     uint64_t align = MAX(BDRV_SECTOR_SIZE, bs->request_alignment);
29811b0288aeSKevin Wolf     uint8_t *head_buf = NULL;
29821b0288aeSKevin Wolf     uint8_t *tail_buf = NULL;
29831b0288aeSKevin Wolf     QEMUIOVector local_qiov;
29841b0288aeSKevin Wolf     bool use_local_qiov = false;
2985d0c7f642SKevin Wolf     int ret;
2986d0c7f642SKevin Wolf 
2987d0c7f642SKevin Wolf     if (!drv) {
2988d0c7f642SKevin Wolf         return -ENOMEDIUM;
2989d0c7f642SKevin Wolf     }
29901b0288aeSKevin Wolf     if (bdrv_check_byte_request(bs, offset, bytes)) {
2991d0c7f642SKevin Wolf         return -EIO;
2992d0c7f642SKevin Wolf     }
2993d0c7f642SKevin Wolf 
2994d0c7f642SKevin Wolf     if (bs->copy_on_read) {
2995d0c7f642SKevin Wolf         flags |= BDRV_REQ_COPY_ON_READ;
2996d0c7f642SKevin Wolf     }
2997d0c7f642SKevin Wolf 
2998d0c7f642SKevin Wolf     /* throttling disk I/O */
2999d0c7f642SKevin Wolf     if (bs->io_limits_enabled) {
30001b0288aeSKevin Wolf         /* TODO Switch to byte granularity */
30011b0288aeSKevin Wolf         bdrv_io_limits_intercept(bs, bytes >> BDRV_SECTOR_BITS, false);
3002d0c7f642SKevin Wolf     }
3003d0c7f642SKevin Wolf 
30041b0288aeSKevin Wolf     /* Align read if necessary by padding qiov */
30051b0288aeSKevin Wolf     if (offset & (align - 1)) {
30061b0288aeSKevin Wolf         head_buf = qemu_blockalign(bs, align);
30071b0288aeSKevin Wolf         qemu_iovec_init(&local_qiov, qiov->niov + 2);
30081b0288aeSKevin Wolf         qemu_iovec_add(&local_qiov, head_buf, offset & (align - 1));
30091b0288aeSKevin Wolf         qemu_iovec_concat(&local_qiov, qiov, 0, qiov->size);
30101b0288aeSKevin Wolf         use_local_qiov = true;
30111b0288aeSKevin Wolf 
30121b0288aeSKevin Wolf         bytes += offset & (align - 1);
30131b0288aeSKevin Wolf         offset = offset & ~(align - 1);
30141b0288aeSKevin Wolf     }
30151b0288aeSKevin Wolf 
30161b0288aeSKevin Wolf     if ((offset + bytes) & (align - 1)) {
30171b0288aeSKevin Wolf         if (!use_local_qiov) {
30181b0288aeSKevin Wolf             qemu_iovec_init(&local_qiov, qiov->niov + 1);
30191b0288aeSKevin Wolf             qemu_iovec_concat(&local_qiov, qiov, 0, qiov->size);
30201b0288aeSKevin Wolf             use_local_qiov = true;
30211b0288aeSKevin Wolf         }
30221b0288aeSKevin Wolf         tail_buf = qemu_blockalign(bs, align);
30231b0288aeSKevin Wolf         qemu_iovec_add(&local_qiov, tail_buf,
30241b0288aeSKevin Wolf                        align - ((offset + bytes) & (align - 1)));
30251b0288aeSKevin Wolf 
30261b0288aeSKevin Wolf         bytes = ROUND_UP(bytes, align);
30271b0288aeSKevin Wolf     }
30281b0288aeSKevin Wolf 
30291b0288aeSKevin Wolf     ret = bdrv_aligned_preadv(bs, offset, bytes,
30301b0288aeSKevin Wolf                               use_local_qiov ? &local_qiov : qiov,
30311b0288aeSKevin Wolf                               flags);
30321b0288aeSKevin Wolf 
30331b0288aeSKevin Wolf     if (use_local_qiov) {
30341b0288aeSKevin Wolf         qemu_iovec_destroy(&local_qiov);
30351b0288aeSKevin Wolf         qemu_vfree(head_buf);
30361b0288aeSKevin Wolf         qemu_vfree(tail_buf);
30371b0288aeSKevin Wolf     }
30381b0288aeSKevin Wolf 
3039d0c7f642SKevin Wolf     return ret;
3040d0c7f642SKevin Wolf }
3041d0c7f642SKevin Wolf 
30421b0288aeSKevin Wolf static int coroutine_fn bdrv_co_do_readv(BlockDriverState *bs,
30431b0288aeSKevin Wolf     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
30441b0288aeSKevin Wolf     BdrvRequestFlags flags)
30451b0288aeSKevin Wolf {
30461b0288aeSKevin Wolf     if (nb_sectors < 0 || nb_sectors > (UINT_MAX >> BDRV_SECTOR_BITS)) {
30471b0288aeSKevin Wolf         return -EINVAL;
30481b0288aeSKevin Wolf     }
30491b0288aeSKevin Wolf 
30501b0288aeSKevin Wolf     return bdrv_co_do_preadv(bs, sector_num << BDRV_SECTOR_BITS,
30511b0288aeSKevin Wolf                              nb_sectors << BDRV_SECTOR_BITS, qiov, flags);
30521b0288aeSKevin Wolf }
30531b0288aeSKevin Wolf 
3054c5fbe571SStefan Hajnoczi int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
3055da1fa91dSKevin Wolf     int nb_sectors, QEMUIOVector *qiov)
3056da1fa91dSKevin Wolf {
3057c5fbe571SStefan Hajnoczi     trace_bdrv_co_readv(bs, sector_num, nb_sectors);
3058da1fa91dSKevin Wolf 
3059470c0504SStefan Hajnoczi     return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov, 0);
3060470c0504SStefan Hajnoczi }
3061470c0504SStefan Hajnoczi 
3062470c0504SStefan Hajnoczi int coroutine_fn bdrv_co_copy_on_readv(BlockDriverState *bs,
3063470c0504SStefan Hajnoczi     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov)
3064470c0504SStefan Hajnoczi {
3065470c0504SStefan Hajnoczi     trace_bdrv_co_copy_on_readv(bs, sector_num, nb_sectors);
3066470c0504SStefan Hajnoczi 
3067470c0504SStefan Hajnoczi     return bdrv_co_do_readv(bs, sector_num, nb_sectors, qiov,
3068470c0504SStefan Hajnoczi                             BDRV_REQ_COPY_ON_READ);
3069c5fbe571SStefan Hajnoczi }
3070c5fbe571SStefan Hajnoczi 
3071c31cb707SPeter Lieven /* if no limit is specified in the BlockLimits use a default
3072c31cb707SPeter Lieven  * of 32768 512-byte sectors (16 MiB) per request.
3073c31cb707SPeter Lieven  */
3074c31cb707SPeter Lieven #define MAX_WRITE_ZEROES_DEFAULT 32768
3075c31cb707SPeter Lieven 
3076f08f2ddaSStefan Hajnoczi static int coroutine_fn bdrv_co_do_write_zeroes(BlockDriverState *bs,
3077aa7bfbffSPeter Lieven     int64_t sector_num, int nb_sectors, BdrvRequestFlags flags)
3078f08f2ddaSStefan Hajnoczi {
3079f08f2ddaSStefan Hajnoczi     BlockDriver *drv = bs->drv;
3080f08f2ddaSStefan Hajnoczi     QEMUIOVector qiov;
3081c31cb707SPeter Lieven     struct iovec iov = {0};
3082c31cb707SPeter Lieven     int ret = 0;
3083f08f2ddaSStefan Hajnoczi 
3084c31cb707SPeter Lieven     int max_write_zeroes = bs->bl.max_write_zeroes ?
3085c31cb707SPeter Lieven                            bs->bl.max_write_zeroes : MAX_WRITE_ZEROES_DEFAULT;
3086621f0589SKevin Wolf 
3087c31cb707SPeter Lieven     while (nb_sectors > 0 && !ret) {
3088c31cb707SPeter Lieven         int num = nb_sectors;
3089c31cb707SPeter Lieven 
3090b8d71c09SPaolo Bonzini         /* Align request.  Block drivers can expect the "bulk" of the request
3091b8d71c09SPaolo Bonzini          * to be aligned.
3092b8d71c09SPaolo Bonzini          */
3093b8d71c09SPaolo Bonzini         if (bs->bl.write_zeroes_alignment
3094b8d71c09SPaolo Bonzini             && num > bs->bl.write_zeroes_alignment) {
3095b8d71c09SPaolo Bonzini             if (sector_num % bs->bl.write_zeroes_alignment != 0) {
3096b8d71c09SPaolo Bonzini                 /* Make a small request up to the first aligned sector.  */
3097c31cb707SPeter Lieven                 num = bs->bl.write_zeroes_alignment;
3098c31cb707SPeter Lieven                 num -= sector_num % bs->bl.write_zeroes_alignment;
3099b8d71c09SPaolo Bonzini             } else if ((sector_num + num) % bs->bl.write_zeroes_alignment != 0) {
3100b8d71c09SPaolo Bonzini                 /* Shorten the request to the last aligned sector.  num cannot
3101b8d71c09SPaolo Bonzini                  * underflow because num > bs->bl.write_zeroes_alignment.
3102b8d71c09SPaolo Bonzini                  */
3103b8d71c09SPaolo Bonzini                 num -= (sector_num + num) % bs->bl.write_zeroes_alignment;
3104b8d71c09SPaolo Bonzini             }
3105c31cb707SPeter Lieven         }
3106c31cb707SPeter Lieven 
3107c31cb707SPeter Lieven         /* limit request size */
3108c31cb707SPeter Lieven         if (num > max_write_zeroes) {
3109c31cb707SPeter Lieven             num = max_write_zeroes;
3110c31cb707SPeter Lieven         }
3111c31cb707SPeter Lieven 
3112c31cb707SPeter Lieven         ret = -ENOTSUP;
3113f08f2ddaSStefan Hajnoczi         /* First try the efficient write zeroes operation */
3114f08f2ddaSStefan Hajnoczi         if (drv->bdrv_co_write_zeroes) {
3115c31cb707SPeter Lieven             ret = drv->bdrv_co_write_zeroes(bs, sector_num, num, flags);
3116f08f2ddaSStefan Hajnoczi         }
3117f08f2ddaSStefan Hajnoczi 
3118c31cb707SPeter Lieven         if (ret == -ENOTSUP) {
3119f08f2ddaSStefan Hajnoczi             /* Fall back to bounce buffer if write zeroes is unsupported */
3120c31cb707SPeter Lieven             iov.iov_len = num * BDRV_SECTOR_SIZE;
3121c31cb707SPeter Lieven             if (iov.iov_base == NULL) {
3122b8d71c09SPaolo Bonzini                 iov.iov_base = qemu_blockalign(bs, num * BDRV_SECTOR_SIZE);
3123b8d71c09SPaolo Bonzini                 memset(iov.iov_base, 0, num * BDRV_SECTOR_SIZE);
3124c31cb707SPeter Lieven             }
3125f08f2ddaSStefan Hajnoczi             qemu_iovec_init_external(&qiov, &iov, 1);
3126f08f2ddaSStefan Hajnoczi 
3127c31cb707SPeter Lieven             ret = drv->bdrv_co_writev(bs, sector_num, num, &qiov);
3128b8d71c09SPaolo Bonzini 
3129b8d71c09SPaolo Bonzini             /* Keep bounce buffer around if it is big enough for all
3130b8d71c09SPaolo Bonzini              * all future requests.
3131b8d71c09SPaolo Bonzini              */
3132b8d71c09SPaolo Bonzini             if (num < max_write_zeroes) {
3133b8d71c09SPaolo Bonzini                 qemu_vfree(iov.iov_base);
3134b8d71c09SPaolo Bonzini                 iov.iov_base = NULL;
3135b8d71c09SPaolo Bonzini             }
3136c31cb707SPeter Lieven         }
3137c31cb707SPeter Lieven 
3138c31cb707SPeter Lieven         sector_num += num;
3139c31cb707SPeter Lieven         nb_sectors -= num;
3140c31cb707SPeter Lieven     }
3141f08f2ddaSStefan Hajnoczi 
3142f08f2ddaSStefan Hajnoczi     qemu_vfree(iov.iov_base);
3143f08f2ddaSStefan Hajnoczi     return ret;
3144f08f2ddaSStefan Hajnoczi }
3145f08f2ddaSStefan Hajnoczi 
3146c5fbe571SStefan Hajnoczi /*
3147*b404f720SKevin Wolf  * Forwards an already correctly aligned write request to the BlockDriver.
3148c5fbe571SStefan Hajnoczi  */
3149*b404f720SKevin Wolf static int coroutine_fn bdrv_aligned_pwritev(BlockDriverState *bs,
3150*b404f720SKevin Wolf     int64_t offset, unsigned int bytes, QEMUIOVector *qiov, int flags)
3151c5fbe571SStefan Hajnoczi {
3152c5fbe571SStefan Hajnoczi     BlockDriver *drv = bs->drv;
3153dbffbdcfSStefan Hajnoczi     BdrvTrackedRequest req;
31546b7cb247SStefan Hajnoczi     int ret;
3155da1fa91dSKevin Wolf 
3156*b404f720SKevin Wolf     int64_t sector_num = offset >> BDRV_SECTOR_BITS;
3157*b404f720SKevin Wolf     unsigned int nb_sectors = bytes >> BDRV_SECTOR_BITS;
3158da1fa91dSKevin Wolf 
3159*b404f720SKevin Wolf     assert((offset & (BDRV_SECTOR_SIZE - 1)) == 0);
3160*b404f720SKevin Wolf     assert((bytes & (BDRV_SECTOR_SIZE - 1)) == 0);
3161cc0681c4SBenoît Canet 
3162dbffbdcfSStefan Hajnoczi     tracked_request_begin(&req, bs, sector_num, nb_sectors, true);
3163dbffbdcfSStefan Hajnoczi 
3164d616b224SStefan Hajnoczi     ret = notifier_with_return_list_notify(&bs->before_write_notifiers, &req);
3165d616b224SStefan Hajnoczi 
3166d616b224SStefan Hajnoczi     if (ret < 0) {
3167d616b224SStefan Hajnoczi         /* Do nothing, write notifier decided to fail this request */
3168d616b224SStefan Hajnoczi     } else if (flags & BDRV_REQ_ZERO_WRITE) {
3169aa7bfbffSPeter Lieven         ret = bdrv_co_do_write_zeroes(bs, sector_num, nb_sectors, flags);
3170f08f2ddaSStefan Hajnoczi     } else {
31716b7cb247SStefan Hajnoczi         ret = drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
3172f08f2ddaSStefan Hajnoczi     }
31736b7cb247SStefan Hajnoczi 
3174f05fa4adSPaolo Bonzini     if (ret == 0 && !bs->enable_write_cache) {
3175f05fa4adSPaolo Bonzini         ret = bdrv_co_flush(bs);
3176f05fa4adSPaolo Bonzini     }
3177f05fa4adSPaolo Bonzini 
31781755da16SPaolo Bonzini     bdrv_set_dirty(bs, sector_num, nb_sectors);
3179da1fa91dSKevin Wolf 
3180da1fa91dSKevin Wolf     if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
3181da1fa91dSKevin Wolf         bs->wr_highest_sector = sector_num + nb_sectors - 1;
3182da1fa91dSKevin Wolf     }
3183df2a6f29SPaolo Bonzini     if (bs->growable && ret >= 0) {
3184df2a6f29SPaolo Bonzini         bs->total_sectors = MAX(bs->total_sectors, sector_num + nb_sectors);
3185df2a6f29SPaolo Bonzini     }
3186da1fa91dSKevin Wolf 
3187dbffbdcfSStefan Hajnoczi     tracked_request_end(&req);
3188dbffbdcfSStefan Hajnoczi 
31896b7cb247SStefan Hajnoczi     return ret;
3190da1fa91dSKevin Wolf }
3191da1fa91dSKevin Wolf 
3192*b404f720SKevin Wolf /*
3193*b404f720SKevin Wolf  * Handle a write request in coroutine context
3194*b404f720SKevin Wolf  */
3195*b404f720SKevin Wolf static int coroutine_fn bdrv_co_do_writev(BlockDriverState *bs,
3196*b404f720SKevin Wolf     int64_t sector_num, int nb_sectors, QEMUIOVector *qiov,
3197*b404f720SKevin Wolf     BdrvRequestFlags flags)
3198*b404f720SKevin Wolf {
3199*b404f720SKevin Wolf     int ret;
3200*b404f720SKevin Wolf 
3201*b404f720SKevin Wolf     if (!bs->drv) {
3202*b404f720SKevin Wolf         return -ENOMEDIUM;
3203*b404f720SKevin Wolf     }
3204*b404f720SKevin Wolf     if (bs->read_only) {
3205*b404f720SKevin Wolf         return -EACCES;
3206*b404f720SKevin Wolf     }
3207*b404f720SKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
3208*b404f720SKevin Wolf         return -EIO;
3209*b404f720SKevin Wolf     }
3210*b404f720SKevin Wolf 
3211*b404f720SKevin Wolf     if (bs->copy_on_read_in_flight) {
3212*b404f720SKevin Wolf         wait_for_overlapping_requests(bs, sector_num, nb_sectors);
3213*b404f720SKevin Wolf     }
3214*b404f720SKevin Wolf 
3215*b404f720SKevin Wolf     /* throttling disk I/O */
3216*b404f720SKevin Wolf     if (bs->io_limits_enabled) {
3217*b404f720SKevin Wolf         bdrv_io_limits_intercept(bs, nb_sectors, true);
3218*b404f720SKevin Wolf     }
3219*b404f720SKevin Wolf 
3220*b404f720SKevin Wolf     ret = bdrv_aligned_pwritev(bs, sector_num << BDRV_SECTOR_BITS,
3221*b404f720SKevin Wolf                                nb_sectors << BDRV_SECTOR_BITS, qiov, flags);
3222*b404f720SKevin Wolf 
3223*b404f720SKevin Wolf     return ret;
3224*b404f720SKevin Wolf }
3225*b404f720SKevin Wolf 
3226c5fbe571SStefan Hajnoczi int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
3227c5fbe571SStefan Hajnoczi     int nb_sectors, QEMUIOVector *qiov)
3228c5fbe571SStefan Hajnoczi {
3229c5fbe571SStefan Hajnoczi     trace_bdrv_co_writev(bs, sector_num, nb_sectors);
3230c5fbe571SStefan Hajnoczi 
3231f08f2ddaSStefan Hajnoczi     return bdrv_co_do_writev(bs, sector_num, nb_sectors, qiov, 0);
3232f08f2ddaSStefan Hajnoczi }
3233f08f2ddaSStefan Hajnoczi 
3234f08f2ddaSStefan Hajnoczi int coroutine_fn bdrv_co_write_zeroes(BlockDriverState *bs,
3235aa7bfbffSPeter Lieven                                       int64_t sector_num, int nb_sectors,
3236aa7bfbffSPeter Lieven                                       BdrvRequestFlags flags)
3237f08f2ddaSStefan Hajnoczi {
323894d6ff21SPaolo Bonzini     trace_bdrv_co_write_zeroes(bs, sector_num, nb_sectors, flags);
3239f08f2ddaSStefan Hajnoczi 
3240d32f35cbSPeter Lieven     if (!(bs->open_flags & BDRV_O_UNMAP)) {
3241d32f35cbSPeter Lieven         flags &= ~BDRV_REQ_MAY_UNMAP;
3242d32f35cbSPeter Lieven     }
3243d32f35cbSPeter Lieven 
3244f08f2ddaSStefan Hajnoczi     return bdrv_co_do_writev(bs, sector_num, nb_sectors, NULL,
3245aa7bfbffSPeter Lieven                              BDRV_REQ_ZERO_WRITE | flags);
3246c5fbe571SStefan Hajnoczi }
3247c5fbe571SStefan Hajnoczi 
324883f64091Sbellard /**
324983f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
325083f64091Sbellard  */
325183f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
325283f64091Sbellard {
325383f64091Sbellard     BlockDriver *drv = bs->drv;
325451762288SStefan Hajnoczi     int ret;
325583f64091Sbellard     if (!drv)
325619cb3738Sbellard         return -ENOMEDIUM;
325783f64091Sbellard     if (!drv->bdrv_truncate)
325883f64091Sbellard         return -ENOTSUP;
325959f2689dSNaphtali Sprei     if (bs->read_only)
326059f2689dSNaphtali Sprei         return -EACCES;
32618591675fSMarcelo Tosatti     if (bdrv_in_use(bs))
32628591675fSMarcelo Tosatti         return -EBUSY;
326351762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
326451762288SStefan Hajnoczi     if (ret == 0) {
326551762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
3266145feb17SMarkus Armbruster         bdrv_dev_resize_cb(bs);
326751762288SStefan Hajnoczi     }
326851762288SStefan Hajnoczi     return ret;
326983f64091Sbellard }
327083f64091Sbellard 
327183f64091Sbellard /**
32724a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
32734a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
32744a1d5e1fSFam Zheng  */
32754a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
32764a1d5e1fSFam Zheng {
32774a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
32784a1d5e1fSFam Zheng     if (!drv) {
32794a1d5e1fSFam Zheng         return -ENOMEDIUM;
32804a1d5e1fSFam Zheng     }
32814a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
32824a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
32834a1d5e1fSFam Zheng     }
32844a1d5e1fSFam Zheng     if (bs->file) {
32854a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
32864a1d5e1fSFam Zheng     }
32874a1d5e1fSFam Zheng     return -ENOTSUP;
32884a1d5e1fSFam Zheng }
32894a1d5e1fSFam Zheng 
32904a1d5e1fSFam Zheng /**
329183f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
329283f64091Sbellard  */
329383f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
329483f64091Sbellard {
329583f64091Sbellard     BlockDriver *drv = bs->drv;
329683f64091Sbellard     if (!drv)
329719cb3738Sbellard         return -ENOMEDIUM;
329851762288SStefan Hajnoczi 
3299b94a2610SKevin Wolf     if (drv->has_variable_length) {
3300b94a2610SKevin Wolf         int ret = refresh_total_sectors(bs, bs->total_sectors);
3301b94a2610SKevin Wolf         if (ret < 0) {
3302b94a2610SKevin Wolf             return ret;
3303fc01f7e7Sbellard         }
330446a4e4e6SStefan Hajnoczi     }
330546a4e4e6SStefan Hajnoczi     return bs->total_sectors * BDRV_SECTOR_SIZE;
330646a4e4e6SStefan Hajnoczi }
3307fc01f7e7Sbellard 
330819cb3738Sbellard /* return 0 as number of sectors if no device present or error */
330996b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
3310fc01f7e7Sbellard {
331119cb3738Sbellard     int64_t length;
331219cb3738Sbellard     length = bdrv_getlength(bs);
331319cb3738Sbellard     if (length < 0)
331419cb3738Sbellard         length = 0;
331519cb3738Sbellard     else
33166ea44308SJan Kiszka         length = length >> BDRV_SECTOR_BITS;
331719cb3738Sbellard     *nb_sectors_ptr = length;
3318fc01f7e7Sbellard }
3319cf98951bSbellard 
3320ff06f5f3SPaolo Bonzini void bdrv_set_on_error(BlockDriverState *bs, BlockdevOnError on_read_error,
3321ff06f5f3SPaolo Bonzini                        BlockdevOnError on_write_error)
3322abd7f68dSMarkus Armbruster {
3323abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
3324abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
3325abd7f68dSMarkus Armbruster }
3326abd7f68dSMarkus Armbruster 
33271ceee0d5SPaolo Bonzini BlockdevOnError bdrv_get_on_error(BlockDriverState *bs, bool is_read)
3328abd7f68dSMarkus Armbruster {
3329abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
3330abd7f68dSMarkus Armbruster }
3331abd7f68dSMarkus Armbruster 
33323e1caa5fSPaolo Bonzini BlockErrorAction bdrv_get_error_action(BlockDriverState *bs, bool is_read, int error)
33333e1caa5fSPaolo Bonzini {
33343e1caa5fSPaolo Bonzini     BlockdevOnError on_err = is_read ? bs->on_read_error : bs->on_write_error;
33353e1caa5fSPaolo Bonzini 
33363e1caa5fSPaolo Bonzini     switch (on_err) {
33373e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_ENOSPC:
33383e1caa5fSPaolo Bonzini         return (error == ENOSPC) ? BDRV_ACTION_STOP : BDRV_ACTION_REPORT;
33393e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_STOP:
33403e1caa5fSPaolo Bonzini         return BDRV_ACTION_STOP;
33413e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_REPORT:
33423e1caa5fSPaolo Bonzini         return BDRV_ACTION_REPORT;
33433e1caa5fSPaolo Bonzini     case BLOCKDEV_ON_ERROR_IGNORE:
33443e1caa5fSPaolo Bonzini         return BDRV_ACTION_IGNORE;
33453e1caa5fSPaolo Bonzini     default:
33463e1caa5fSPaolo Bonzini         abort();
33473e1caa5fSPaolo Bonzini     }
33483e1caa5fSPaolo Bonzini }
33493e1caa5fSPaolo Bonzini 
33503e1caa5fSPaolo Bonzini /* This is done by device models because, while the block layer knows
33513e1caa5fSPaolo Bonzini  * about the error, it does not know whether an operation comes from
33523e1caa5fSPaolo Bonzini  * the device or the block layer (from a job, for example).
33533e1caa5fSPaolo Bonzini  */
33543e1caa5fSPaolo Bonzini void bdrv_error_action(BlockDriverState *bs, BlockErrorAction action,
33553e1caa5fSPaolo Bonzini                        bool is_read, int error)
33563e1caa5fSPaolo Bonzini {
33573e1caa5fSPaolo Bonzini     assert(error >= 0);
335832c81a4aSPaolo Bonzini     bdrv_emit_qmp_error_event(bs, QEVENT_BLOCK_IO_ERROR, action, is_read);
33593e1caa5fSPaolo Bonzini     if (action == BDRV_ACTION_STOP) {
33603e1caa5fSPaolo Bonzini         vm_stop(RUN_STATE_IO_ERROR);
33613e1caa5fSPaolo Bonzini         bdrv_iostatus_set_err(bs, error);
33623e1caa5fSPaolo Bonzini     }
33633e1caa5fSPaolo Bonzini }
33643e1caa5fSPaolo Bonzini 
3365b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
3366b338082bSbellard {
3367b338082bSbellard     return bs->read_only;
3368b338082bSbellard }
3369b338082bSbellard 
3370985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
3371985a03b0Sths {
3372985a03b0Sths     return bs->sg;
3373985a03b0Sths }
3374985a03b0Sths 
3375e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
3376e900a7b7SChristoph Hellwig {
3377e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
3378e900a7b7SChristoph Hellwig }
3379e900a7b7SChristoph Hellwig 
3380425b0148SPaolo Bonzini void bdrv_set_enable_write_cache(BlockDriverState *bs, bool wce)
3381425b0148SPaolo Bonzini {
3382425b0148SPaolo Bonzini     bs->enable_write_cache = wce;
338355b110f2SJeff Cody 
338455b110f2SJeff Cody     /* so a reopen() will preserve wce */
338555b110f2SJeff Cody     if (wce) {
338655b110f2SJeff Cody         bs->open_flags |= BDRV_O_CACHE_WB;
338755b110f2SJeff Cody     } else {
338855b110f2SJeff Cody         bs->open_flags &= ~BDRV_O_CACHE_WB;
338955b110f2SJeff Cody     }
3390425b0148SPaolo Bonzini }
3391425b0148SPaolo Bonzini 
3392ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
3393ea2384d3Sbellard {
3394ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
3395ea2384d3Sbellard         return 1;
3396ea2384d3Sbellard     return bs->encrypted;
3397ea2384d3Sbellard }
3398ea2384d3Sbellard 
3399c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
3400c0f4ce77Saliguori {
3401c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
3402c0f4ce77Saliguori 
3403c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
3404c0f4ce77Saliguori         return 1;
3405c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
3406c0f4ce77Saliguori }
3407c0f4ce77Saliguori 
3408ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
3409ea2384d3Sbellard {
3410ea2384d3Sbellard     int ret;
3411ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
3412ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
3413ea2384d3Sbellard         if (ret < 0)
3414ea2384d3Sbellard             return ret;
3415ea2384d3Sbellard         if (!bs->encrypted)
3416ea2384d3Sbellard             return 0;
3417ea2384d3Sbellard     }
3418fd04a2aeSShahar Havivi     if (!bs->encrypted) {
3419fd04a2aeSShahar Havivi         return -EINVAL;
3420fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
3421fd04a2aeSShahar Havivi         return -ENOMEDIUM;
3422fd04a2aeSShahar Havivi     }
3423c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
3424bb5fc20fSaliguori     if (ret < 0) {
3425bb5fc20fSaliguori         bs->valid_key = 0;
3426bb5fc20fSaliguori     } else if (!bs->valid_key) {
3427bb5fc20fSaliguori         bs->valid_key = 1;
3428bb5fc20fSaliguori         /* call the change callback now, we skipped it on open */
34297d4b4ba5SMarkus Armbruster         bdrv_dev_change_media_cb(bs, true);
3430bb5fc20fSaliguori     }
3431c0f4ce77Saliguori     return ret;
3432ea2384d3Sbellard }
3433ea2384d3Sbellard 
3434f8d6bba1SMarkus Armbruster const char *bdrv_get_format_name(BlockDriverState *bs)
3435ea2384d3Sbellard {
3436f8d6bba1SMarkus Armbruster     return bs->drv ? bs->drv->format_name : NULL;
3437ea2384d3Sbellard }
3438ea2384d3Sbellard 
3439ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
3440ea2384d3Sbellard                          void *opaque)
3441ea2384d3Sbellard {
3442ea2384d3Sbellard     BlockDriver *drv;
3443ea2384d3Sbellard 
34448a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
3445ea2384d3Sbellard         it(opaque, drv->format_name);
3446ea2384d3Sbellard     }
3447ea2384d3Sbellard }
3448ea2384d3Sbellard 
3449dc364f4cSBenoît Canet /* This function is to find block backend bs */
3450b338082bSbellard BlockDriverState *bdrv_find(const char *name)
3451b338082bSbellard {
3452b338082bSbellard     BlockDriverState *bs;
3453b338082bSbellard 
3454dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
34551b7bdbc1SStefan Hajnoczi         if (!strcmp(name, bs->device_name)) {
3456b338082bSbellard             return bs;
3457b338082bSbellard         }
34581b7bdbc1SStefan Hajnoczi     }
3459b338082bSbellard     return NULL;
3460b338082bSbellard }
3461b338082bSbellard 
3462dc364f4cSBenoît Canet /* This function is to find a node in the bs graph */
3463dc364f4cSBenoît Canet BlockDriverState *bdrv_find_node(const char *node_name)
3464dc364f4cSBenoît Canet {
3465dc364f4cSBenoît Canet     BlockDriverState *bs;
3466dc364f4cSBenoît Canet 
3467dc364f4cSBenoît Canet     assert(node_name);
3468dc364f4cSBenoît Canet 
3469dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
3470dc364f4cSBenoît Canet         if (!strcmp(node_name, bs->node_name)) {
3471dc364f4cSBenoît Canet             return bs;
3472dc364f4cSBenoît Canet         }
3473dc364f4cSBenoît Canet     }
3474dc364f4cSBenoît Canet     return NULL;
3475dc364f4cSBenoît Canet }
3476dc364f4cSBenoît Canet 
3477c13163fbSBenoît Canet /* Put this QMP function here so it can access the static graph_bdrv_states. */
3478c13163fbSBenoît Canet BlockDeviceInfoList *bdrv_named_nodes_list(void)
3479c13163fbSBenoît Canet {
3480c13163fbSBenoît Canet     BlockDeviceInfoList *list, *entry;
3481c13163fbSBenoît Canet     BlockDriverState *bs;
3482c13163fbSBenoît Canet 
3483c13163fbSBenoît Canet     list = NULL;
3484c13163fbSBenoît Canet     QTAILQ_FOREACH(bs, &graph_bdrv_states, node_list) {
3485c13163fbSBenoît Canet         entry = g_malloc0(sizeof(*entry));
3486c13163fbSBenoît Canet         entry->value = bdrv_block_device_info(bs);
3487c13163fbSBenoît Canet         entry->next = list;
3488c13163fbSBenoît Canet         list = entry;
3489c13163fbSBenoît Canet     }
3490c13163fbSBenoît Canet 
3491c13163fbSBenoît Canet     return list;
3492c13163fbSBenoît Canet }
3493c13163fbSBenoît Canet 
349412d3ba82SBenoît Canet BlockDriverState *bdrv_lookup_bs(const char *device,
349512d3ba82SBenoît Canet                                  const char *node_name,
349612d3ba82SBenoît Canet                                  Error **errp)
349712d3ba82SBenoît Canet {
349812d3ba82SBenoît Canet     BlockDriverState *bs = NULL;
349912d3ba82SBenoît Canet 
350012d3ba82SBenoît Canet     if ((!device && !node_name) || (device && node_name)) {
350112d3ba82SBenoît Canet         error_setg(errp, "Use either device or node-name but not both");
350212d3ba82SBenoît Canet         return NULL;
350312d3ba82SBenoît Canet     }
350412d3ba82SBenoît Canet 
350512d3ba82SBenoît Canet     if (device) {
350612d3ba82SBenoît Canet         bs = bdrv_find(device);
350712d3ba82SBenoît Canet 
350812d3ba82SBenoît Canet         if (!bs) {
350912d3ba82SBenoît Canet             error_set(errp, QERR_DEVICE_NOT_FOUND, device);
351012d3ba82SBenoît Canet             return NULL;
351112d3ba82SBenoît Canet         }
351212d3ba82SBenoît Canet 
351312d3ba82SBenoît Canet         return bs;
351412d3ba82SBenoît Canet     }
351512d3ba82SBenoît Canet 
351612d3ba82SBenoît Canet     bs = bdrv_find_node(node_name);
351712d3ba82SBenoît Canet 
351812d3ba82SBenoît Canet     if (!bs) {
351912d3ba82SBenoît Canet         error_set(errp, QERR_DEVICE_NOT_FOUND, node_name);
352012d3ba82SBenoît Canet         return NULL;
352112d3ba82SBenoît Canet     }
352212d3ba82SBenoît Canet 
352312d3ba82SBenoît Canet     return bs;
352412d3ba82SBenoît Canet }
352512d3ba82SBenoît Canet 
35262f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
35272f399b0aSMarkus Armbruster {
35282f399b0aSMarkus Armbruster     if (!bs) {
35292f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
35302f399b0aSMarkus Armbruster     }
3531dc364f4cSBenoît Canet     return QTAILQ_NEXT(bs, device_list);
35322f399b0aSMarkus Armbruster }
35332f399b0aSMarkus Armbruster 
353451de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
353581d0912dSbellard {
353681d0912dSbellard     BlockDriverState *bs;
353781d0912dSbellard 
3538dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
353951de9760Saliguori         it(opaque, bs);
354081d0912dSbellard     }
354181d0912dSbellard }
354281d0912dSbellard 
3543ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
3544ea2384d3Sbellard {
3545ea2384d3Sbellard     return bs->device_name;
3546ea2384d3Sbellard }
3547ea2384d3Sbellard 
3548c8433287SMarkus Armbruster int bdrv_get_flags(BlockDriverState *bs)
3549c8433287SMarkus Armbruster {
3550c8433287SMarkus Armbruster     return bs->open_flags;
3551c8433287SMarkus Armbruster }
3552c8433287SMarkus Armbruster 
3553f0f0fdfeSKevin Wolf int bdrv_flush_all(void)
3554c6ca28d6Saliguori {
3555c6ca28d6Saliguori     BlockDriverState *bs;
3556f0f0fdfeSKevin Wolf     int result = 0;
3557c6ca28d6Saliguori 
3558dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
3559f0f0fdfeSKevin Wolf         int ret = bdrv_flush(bs);
3560f0f0fdfeSKevin Wolf         if (ret < 0 && !result) {
3561f0f0fdfeSKevin Wolf             result = ret;
3562c6ca28d6Saliguori         }
35631b7bdbc1SStefan Hajnoczi     }
3564c6ca28d6Saliguori 
3565f0f0fdfeSKevin Wolf     return result;
3566f0f0fdfeSKevin Wolf }
3567f0f0fdfeSKevin Wolf 
35683ac21627SPeter Lieven int bdrv_has_zero_init_1(BlockDriverState *bs)
35693ac21627SPeter Lieven {
35703ac21627SPeter Lieven     return 1;
35713ac21627SPeter Lieven }
35723ac21627SPeter Lieven 
3573f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
3574f2feebbdSKevin Wolf {
3575f2feebbdSKevin Wolf     assert(bs->drv);
3576f2feebbdSKevin Wolf 
357711212d8fSPaolo Bonzini     /* If BS is a copy on write image, it is initialized to
357811212d8fSPaolo Bonzini        the contents of the base image, which may not be zeroes.  */
357911212d8fSPaolo Bonzini     if (bs->backing_hd) {
358011212d8fSPaolo Bonzini         return 0;
358111212d8fSPaolo Bonzini     }
3582336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
3583336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
3584f2feebbdSKevin Wolf     }
3585f2feebbdSKevin Wolf 
35863ac21627SPeter Lieven     /* safe default */
35873ac21627SPeter Lieven     return 0;
3588f2feebbdSKevin Wolf }
3589f2feebbdSKevin Wolf 
35904ce78691SPeter Lieven bool bdrv_unallocated_blocks_are_zero(BlockDriverState *bs)
35914ce78691SPeter Lieven {
35924ce78691SPeter Lieven     BlockDriverInfo bdi;
35934ce78691SPeter Lieven 
35944ce78691SPeter Lieven     if (bs->backing_hd) {
35954ce78691SPeter Lieven         return false;
35964ce78691SPeter Lieven     }
35974ce78691SPeter Lieven 
35984ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
35994ce78691SPeter Lieven         return bdi.unallocated_blocks_are_zero;
36004ce78691SPeter Lieven     }
36014ce78691SPeter Lieven 
36024ce78691SPeter Lieven     return false;
36034ce78691SPeter Lieven }
36044ce78691SPeter Lieven 
36054ce78691SPeter Lieven bool bdrv_can_write_zeroes_with_unmap(BlockDriverState *bs)
36064ce78691SPeter Lieven {
36074ce78691SPeter Lieven     BlockDriverInfo bdi;
36084ce78691SPeter Lieven 
36094ce78691SPeter Lieven     if (bs->backing_hd || !(bs->open_flags & BDRV_O_UNMAP)) {
36104ce78691SPeter Lieven         return false;
36114ce78691SPeter Lieven     }
36124ce78691SPeter Lieven 
36134ce78691SPeter Lieven     if (bdrv_get_info(bs, &bdi) == 0) {
36144ce78691SPeter Lieven         return bdi.can_write_zeroes_with_unmap;
36154ce78691SPeter Lieven     }
36164ce78691SPeter Lieven 
36174ce78691SPeter Lieven     return false;
36184ce78691SPeter Lieven }
36194ce78691SPeter Lieven 
3620b6b8a333SPaolo Bonzini typedef struct BdrvCoGetBlockStatusData {
3621376ae3f1SStefan Hajnoczi     BlockDriverState *bs;
3622b35b2bbaSMiroslav Rezanina     BlockDriverState *base;
3623376ae3f1SStefan Hajnoczi     int64_t sector_num;
3624376ae3f1SStefan Hajnoczi     int nb_sectors;
3625376ae3f1SStefan Hajnoczi     int *pnum;
3626b6b8a333SPaolo Bonzini     int64_t ret;
3627376ae3f1SStefan Hajnoczi     bool done;
3628b6b8a333SPaolo Bonzini } BdrvCoGetBlockStatusData;
3629376ae3f1SStefan Hajnoczi 
3630f58c7b35Sths /*
3631f58c7b35Sths  * Returns true iff the specified sector is present in the disk image. Drivers
3632f58c7b35Sths  * not implementing the functionality are assumed to not support backing files,
3633f58c7b35Sths  * hence all their sectors are reported as allocated.
3634f58c7b35Sths  *
3635bd9533e3SStefan Hajnoczi  * If 'sector_num' is beyond the end of the disk image the return value is 0
3636bd9533e3SStefan Hajnoczi  * and 'pnum' is set to 0.
3637bd9533e3SStefan Hajnoczi  *
3638f58c7b35Sths  * 'pnum' is set to the number of sectors (including and immediately following
3639f58c7b35Sths  * the specified sector) that are known to be in the same
3640f58c7b35Sths  * allocated/unallocated state.
3641f58c7b35Sths  *
3642bd9533e3SStefan Hajnoczi  * 'nb_sectors' is the max value 'pnum' should be set to.  If nb_sectors goes
3643bd9533e3SStefan Hajnoczi  * beyond the end of the disk image it will be clamped.
3644f58c7b35Sths  */
3645b6b8a333SPaolo Bonzini static int64_t coroutine_fn bdrv_co_get_block_status(BlockDriverState *bs,
3646bdad13b9SPaolo Bonzini                                                      int64_t sector_num,
3647060f51c9SStefan Hajnoczi                                                      int nb_sectors, int *pnum)
3648f58c7b35Sths {
3649617ccb46SPaolo Bonzini     int64_t length;
3650f58c7b35Sths     int64_t n;
36515daa74a6SPaolo Bonzini     int64_t ret, ret2;
3652bd9533e3SStefan Hajnoczi 
3653617ccb46SPaolo Bonzini     length = bdrv_getlength(bs);
3654617ccb46SPaolo Bonzini     if (length < 0) {
3655617ccb46SPaolo Bonzini         return length;
3656617ccb46SPaolo Bonzini     }
3657617ccb46SPaolo Bonzini 
3658617ccb46SPaolo Bonzini     if (sector_num >= (length >> BDRV_SECTOR_BITS)) {
36596aebab14SStefan Hajnoczi         *pnum = 0;
36606aebab14SStefan Hajnoczi         return 0;
36616aebab14SStefan Hajnoczi     }
3662bd9533e3SStefan Hajnoczi 
36636aebab14SStefan Hajnoczi     n = bs->total_sectors - sector_num;
3664bd9533e3SStefan Hajnoczi     if (n < nb_sectors) {
3665bd9533e3SStefan Hajnoczi         nb_sectors = n;
3666bd9533e3SStefan Hajnoczi     }
3667bd9533e3SStefan Hajnoczi 
3668b6b8a333SPaolo Bonzini     if (!bs->drv->bdrv_co_get_block_status) {
3669bd9533e3SStefan Hajnoczi         *pnum = nb_sectors;
3670918e92d7SPaolo Bonzini         ret = BDRV_BLOCK_DATA;
3671918e92d7SPaolo Bonzini         if (bs->drv->protocol_name) {
3672918e92d7SPaolo Bonzini             ret |= BDRV_BLOCK_OFFSET_VALID | (sector_num * BDRV_SECTOR_SIZE);
3673918e92d7SPaolo Bonzini         }
3674918e92d7SPaolo Bonzini         return ret;
36756aebab14SStefan Hajnoczi     }
36766aebab14SStefan Hajnoczi 
3677415b5b01SPaolo Bonzini     ret = bs->drv->bdrv_co_get_block_status(bs, sector_num, nb_sectors, pnum);
3678415b5b01SPaolo Bonzini     if (ret < 0) {
36793e0a233dSPeter Lieven         *pnum = 0;
3680415b5b01SPaolo Bonzini         return ret;
3681415b5b01SPaolo Bonzini     }
3682415b5b01SPaolo Bonzini 
368392bc50a5SPeter Lieven     if (ret & BDRV_BLOCK_RAW) {
368492bc50a5SPeter Lieven         assert(ret & BDRV_BLOCK_OFFSET_VALID);
368592bc50a5SPeter Lieven         return bdrv_get_block_status(bs->file, ret >> BDRV_SECTOR_BITS,
368692bc50a5SPeter Lieven                                      *pnum, pnum);
368792bc50a5SPeter Lieven     }
368892bc50a5SPeter Lieven 
3689c3d86884SPeter Lieven     if (!(ret & BDRV_BLOCK_DATA) && !(ret & BDRV_BLOCK_ZERO)) {
3690c3d86884SPeter Lieven         if (bdrv_unallocated_blocks_are_zero(bs)) {
3691415b5b01SPaolo Bonzini             ret |= BDRV_BLOCK_ZERO;
36921f9db224SPeter Lieven         } else if (bs->backing_hd) {
3693f0ad5712SPaolo Bonzini             BlockDriverState *bs2 = bs->backing_hd;
3694f0ad5712SPaolo Bonzini             int64_t length2 = bdrv_getlength(bs2);
3695f0ad5712SPaolo Bonzini             if (length2 >= 0 && sector_num >= (length2 >> BDRV_SECTOR_BITS)) {
3696f0ad5712SPaolo Bonzini                 ret |= BDRV_BLOCK_ZERO;
3697f0ad5712SPaolo Bonzini             }
3698f0ad5712SPaolo Bonzini         }
3699415b5b01SPaolo Bonzini     }
37005daa74a6SPaolo Bonzini 
37015daa74a6SPaolo Bonzini     if (bs->file &&
37025daa74a6SPaolo Bonzini         (ret & BDRV_BLOCK_DATA) && !(ret & BDRV_BLOCK_ZERO) &&
37035daa74a6SPaolo Bonzini         (ret & BDRV_BLOCK_OFFSET_VALID)) {
37045daa74a6SPaolo Bonzini         ret2 = bdrv_co_get_block_status(bs->file, ret >> BDRV_SECTOR_BITS,
37055daa74a6SPaolo Bonzini                                         *pnum, pnum);
37065daa74a6SPaolo Bonzini         if (ret2 >= 0) {
37075daa74a6SPaolo Bonzini             /* Ignore errors.  This is just providing extra information, it
37085daa74a6SPaolo Bonzini              * is useful but not necessary.
37095daa74a6SPaolo Bonzini              */
37105daa74a6SPaolo Bonzini             ret |= (ret2 & BDRV_BLOCK_ZERO);
37115daa74a6SPaolo Bonzini         }
37125daa74a6SPaolo Bonzini     }
37135daa74a6SPaolo Bonzini 
3714415b5b01SPaolo Bonzini     return ret;
3715060f51c9SStefan Hajnoczi }
3716060f51c9SStefan Hajnoczi 
3717b6b8a333SPaolo Bonzini /* Coroutine wrapper for bdrv_get_block_status() */
3718b6b8a333SPaolo Bonzini static void coroutine_fn bdrv_get_block_status_co_entry(void *opaque)
3719060f51c9SStefan Hajnoczi {
3720b6b8a333SPaolo Bonzini     BdrvCoGetBlockStatusData *data = opaque;
3721060f51c9SStefan Hajnoczi     BlockDriverState *bs = data->bs;
3722060f51c9SStefan Hajnoczi 
3723b6b8a333SPaolo Bonzini     data->ret = bdrv_co_get_block_status(bs, data->sector_num, data->nb_sectors,
3724060f51c9SStefan Hajnoczi                                          data->pnum);
3725060f51c9SStefan Hajnoczi     data->done = true;
3726060f51c9SStefan Hajnoczi }
3727060f51c9SStefan Hajnoczi 
3728060f51c9SStefan Hajnoczi /*
3729b6b8a333SPaolo Bonzini  * Synchronous wrapper around bdrv_co_get_block_status().
3730060f51c9SStefan Hajnoczi  *
3731b6b8a333SPaolo Bonzini  * See bdrv_co_get_block_status() for details.
3732060f51c9SStefan Hajnoczi  */
3733b6b8a333SPaolo Bonzini int64_t bdrv_get_block_status(BlockDriverState *bs, int64_t sector_num,
3734b6b8a333SPaolo Bonzini                               int nb_sectors, int *pnum)
3735060f51c9SStefan Hajnoczi {
3736376ae3f1SStefan Hajnoczi     Coroutine *co;
3737b6b8a333SPaolo Bonzini     BdrvCoGetBlockStatusData data = {
3738376ae3f1SStefan Hajnoczi         .bs = bs,
3739376ae3f1SStefan Hajnoczi         .sector_num = sector_num,
3740376ae3f1SStefan Hajnoczi         .nb_sectors = nb_sectors,
3741376ae3f1SStefan Hajnoczi         .pnum = pnum,
3742376ae3f1SStefan Hajnoczi         .done = false,
3743376ae3f1SStefan Hajnoczi     };
3744376ae3f1SStefan Hajnoczi 
3745bdad13b9SPaolo Bonzini     if (qemu_in_coroutine()) {
3746bdad13b9SPaolo Bonzini         /* Fast-path if already in coroutine context */
3747b6b8a333SPaolo Bonzini         bdrv_get_block_status_co_entry(&data);
3748bdad13b9SPaolo Bonzini     } else {
3749b6b8a333SPaolo Bonzini         co = qemu_coroutine_create(bdrv_get_block_status_co_entry);
3750376ae3f1SStefan Hajnoczi         qemu_coroutine_enter(co, &data);
3751376ae3f1SStefan Hajnoczi         while (!data.done) {
3752376ae3f1SStefan Hajnoczi             qemu_aio_wait();
3753376ae3f1SStefan Hajnoczi         }
3754bdad13b9SPaolo Bonzini     }
3755376ae3f1SStefan Hajnoczi     return data.ret;
3756376ae3f1SStefan Hajnoczi }
3757f58c7b35Sths 
3758b6b8a333SPaolo Bonzini int coroutine_fn bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num,
3759b6b8a333SPaolo Bonzini                                    int nb_sectors, int *pnum)
3760b6b8a333SPaolo Bonzini {
37614333bb71SPaolo Bonzini     int64_t ret = bdrv_get_block_status(bs, sector_num, nb_sectors, pnum);
37624333bb71SPaolo Bonzini     if (ret < 0) {
37634333bb71SPaolo Bonzini         return ret;
37644333bb71SPaolo Bonzini     }
37654333bb71SPaolo Bonzini     return
37664333bb71SPaolo Bonzini         (ret & BDRV_BLOCK_DATA) ||
37674333bb71SPaolo Bonzini         ((ret & BDRV_BLOCK_ZERO) && !bdrv_has_zero_init(bs));
3768b6b8a333SPaolo Bonzini }
3769b6b8a333SPaolo Bonzini 
3770188a7bbfSPaolo Bonzini /*
3771188a7bbfSPaolo Bonzini  * Given an image chain: ... -> [BASE] -> [INTER1] -> [INTER2] -> [TOP]
3772188a7bbfSPaolo Bonzini  *
3773188a7bbfSPaolo Bonzini  * Return true if the given sector is allocated in any image between
3774188a7bbfSPaolo Bonzini  * BASE and TOP (inclusive).  BASE can be NULL to check if the given
3775188a7bbfSPaolo Bonzini  * sector is allocated in any image of the chain.  Return false otherwise.
3776188a7bbfSPaolo Bonzini  *
3777188a7bbfSPaolo Bonzini  * 'pnum' is set to the number of sectors (including and immediately following
3778188a7bbfSPaolo Bonzini  *  the specified sector) that are known to be in the same
3779188a7bbfSPaolo Bonzini  *  allocated/unallocated state.
3780188a7bbfSPaolo Bonzini  *
3781188a7bbfSPaolo Bonzini  */
37824f578637SPaolo Bonzini int bdrv_is_allocated_above(BlockDriverState *top,
3783188a7bbfSPaolo Bonzini                             BlockDriverState *base,
3784188a7bbfSPaolo Bonzini                             int64_t sector_num,
3785188a7bbfSPaolo Bonzini                             int nb_sectors, int *pnum)
3786188a7bbfSPaolo Bonzini {
3787188a7bbfSPaolo Bonzini     BlockDriverState *intermediate;
3788188a7bbfSPaolo Bonzini     int ret, n = nb_sectors;
3789188a7bbfSPaolo Bonzini 
3790188a7bbfSPaolo Bonzini     intermediate = top;
3791188a7bbfSPaolo Bonzini     while (intermediate && intermediate != base) {
3792188a7bbfSPaolo Bonzini         int pnum_inter;
3793bdad13b9SPaolo Bonzini         ret = bdrv_is_allocated(intermediate, sector_num, nb_sectors,
3794188a7bbfSPaolo Bonzini                                 &pnum_inter);
3795188a7bbfSPaolo Bonzini         if (ret < 0) {
3796188a7bbfSPaolo Bonzini             return ret;
3797188a7bbfSPaolo Bonzini         } else if (ret) {
3798188a7bbfSPaolo Bonzini             *pnum = pnum_inter;
3799188a7bbfSPaolo Bonzini             return 1;
3800188a7bbfSPaolo Bonzini         }
3801188a7bbfSPaolo Bonzini 
3802188a7bbfSPaolo Bonzini         /*
3803188a7bbfSPaolo Bonzini          * [sector_num, nb_sectors] is unallocated on top but intermediate
3804188a7bbfSPaolo Bonzini          * might have
3805188a7bbfSPaolo Bonzini          *
3806188a7bbfSPaolo Bonzini          * [sector_num+x, nr_sectors] allocated.
3807188a7bbfSPaolo Bonzini          */
380863ba17d3SVishvananda Ishaya         if (n > pnum_inter &&
380963ba17d3SVishvananda Ishaya             (intermediate == top ||
381063ba17d3SVishvananda Ishaya              sector_num + pnum_inter < intermediate->total_sectors)) {
3811188a7bbfSPaolo Bonzini             n = pnum_inter;
3812188a7bbfSPaolo Bonzini         }
3813188a7bbfSPaolo Bonzini 
3814188a7bbfSPaolo Bonzini         intermediate = intermediate->backing_hd;
3815188a7bbfSPaolo Bonzini     }
3816188a7bbfSPaolo Bonzini 
3817188a7bbfSPaolo Bonzini     *pnum = n;
3818188a7bbfSPaolo Bonzini     return 0;
3819188a7bbfSPaolo Bonzini }
3820188a7bbfSPaolo Bonzini 
3821045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
3822045df330Saliguori {
3823045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
3824045df330Saliguori         return bs->backing_file;
3825045df330Saliguori     else if (bs->encrypted)
3826045df330Saliguori         return bs->filename;
3827045df330Saliguori     else
3828045df330Saliguori         return NULL;
3829045df330Saliguori }
3830045df330Saliguori 
383183f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
383283f64091Sbellard                                char *filename, int filename_size)
383383f64091Sbellard {
383483f64091Sbellard     pstrcpy(filename, filename_size, bs->backing_file);
383583f64091Sbellard }
383683f64091Sbellard 
3837faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
3838faea38e7Sbellard                           const uint8_t *buf, int nb_sectors)
3839faea38e7Sbellard {
3840faea38e7Sbellard     BlockDriver *drv = bs->drv;
3841faea38e7Sbellard     if (!drv)
384219cb3738Sbellard         return -ENOMEDIUM;
3843faea38e7Sbellard     if (!drv->bdrv_write_compressed)
3844faea38e7Sbellard         return -ENOTSUP;
3845fbb7b4e0SKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors))
3846fbb7b4e0SKevin Wolf         return -EIO;
38477cd1e32aSlirans@il.ibm.com 
3848e4654d2dSFam Zheng     assert(QLIST_EMPTY(&bs->dirty_bitmaps));
38497cd1e32aSlirans@il.ibm.com 
3850faea38e7Sbellard     return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
3851faea38e7Sbellard }
3852faea38e7Sbellard 
3853faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
3854faea38e7Sbellard {
3855faea38e7Sbellard     BlockDriver *drv = bs->drv;
3856faea38e7Sbellard     if (!drv)
385719cb3738Sbellard         return -ENOMEDIUM;
3858faea38e7Sbellard     if (!drv->bdrv_get_info)
3859faea38e7Sbellard         return -ENOTSUP;
3860faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
3861faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
3862faea38e7Sbellard }
3863faea38e7Sbellard 
3864eae041feSMax Reitz ImageInfoSpecific *bdrv_get_specific_info(BlockDriverState *bs)
3865eae041feSMax Reitz {
3866eae041feSMax Reitz     BlockDriver *drv = bs->drv;
3867eae041feSMax Reitz     if (drv && drv->bdrv_get_specific_info) {
3868eae041feSMax Reitz         return drv->bdrv_get_specific_info(bs);
3869eae041feSMax Reitz     }
3870eae041feSMax Reitz     return NULL;
3871eae041feSMax Reitz }
3872eae041feSMax Reitz 
387345566e9cSChristoph Hellwig int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
387445566e9cSChristoph Hellwig                       int64_t pos, int size)
3875178e08a5Saliguori {
3876cf8074b3SKevin Wolf     QEMUIOVector qiov;
3877cf8074b3SKevin Wolf     struct iovec iov = {
3878cf8074b3SKevin Wolf         .iov_base   = (void *) buf,
3879cf8074b3SKevin Wolf         .iov_len    = size,
3880cf8074b3SKevin Wolf     };
3881cf8074b3SKevin Wolf 
3882cf8074b3SKevin Wolf     qemu_iovec_init_external(&qiov, &iov, 1);
3883cf8074b3SKevin Wolf     return bdrv_writev_vmstate(bs, &qiov, pos);
3884cf8074b3SKevin Wolf }
3885cf8074b3SKevin Wolf 
3886cf8074b3SKevin Wolf int bdrv_writev_vmstate(BlockDriverState *bs, QEMUIOVector *qiov, int64_t pos)
3887cf8074b3SKevin Wolf {
3888178e08a5Saliguori     BlockDriver *drv = bs->drv;
3889cf8074b3SKevin Wolf 
3890cf8074b3SKevin Wolf     if (!drv) {
3891178e08a5Saliguori         return -ENOMEDIUM;
3892cf8074b3SKevin Wolf     } else if (drv->bdrv_save_vmstate) {
3893cf8074b3SKevin Wolf         return drv->bdrv_save_vmstate(bs, qiov, pos);
3894cf8074b3SKevin Wolf     } else if (bs->file) {
3895cf8074b3SKevin Wolf         return bdrv_writev_vmstate(bs->file, qiov, pos);
3896cf8074b3SKevin Wolf     }
3897cf8074b3SKevin Wolf 
38987cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3899178e08a5Saliguori }
3900178e08a5Saliguori 
390145566e9cSChristoph Hellwig int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
390245566e9cSChristoph Hellwig                       int64_t pos, int size)
3903178e08a5Saliguori {
3904178e08a5Saliguori     BlockDriver *drv = bs->drv;
3905178e08a5Saliguori     if (!drv)
3906178e08a5Saliguori         return -ENOMEDIUM;
39077cdb1f6dSMORITA Kazutaka     if (drv->bdrv_load_vmstate)
390845566e9cSChristoph Hellwig         return drv->bdrv_load_vmstate(bs, buf, pos, size);
39097cdb1f6dSMORITA Kazutaka     if (bs->file)
39107cdb1f6dSMORITA Kazutaka         return bdrv_load_vmstate(bs->file, buf, pos, size);
39117cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
3912178e08a5Saliguori }
3913178e08a5Saliguori 
39148b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
39158b9b0cc2SKevin Wolf {
3916bf736fe3SKevin Wolf     if (!bs || !bs->drv || !bs->drv->bdrv_debug_event) {
39178b9b0cc2SKevin Wolf         return;
39188b9b0cc2SKevin Wolf     }
39198b9b0cc2SKevin Wolf 
3920bf736fe3SKevin Wolf     bs->drv->bdrv_debug_event(bs, event);
392141c695c7SKevin Wolf }
39228b9b0cc2SKevin Wolf 
392341c695c7SKevin Wolf int bdrv_debug_breakpoint(BlockDriverState *bs, const char *event,
392441c695c7SKevin Wolf                           const char *tag)
392541c695c7SKevin Wolf {
392641c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_breakpoint) {
392741c695c7SKevin Wolf         bs = bs->file;
392841c695c7SKevin Wolf     }
392941c695c7SKevin Wolf 
393041c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_breakpoint) {
393141c695c7SKevin Wolf         return bs->drv->bdrv_debug_breakpoint(bs, event, tag);
393241c695c7SKevin Wolf     }
393341c695c7SKevin Wolf 
393441c695c7SKevin Wolf     return -ENOTSUP;
393541c695c7SKevin Wolf }
393641c695c7SKevin Wolf 
39374cc70e93SFam Zheng int bdrv_debug_remove_breakpoint(BlockDriverState *bs, const char *tag)
39384cc70e93SFam Zheng {
39394cc70e93SFam Zheng     while (bs && bs->drv && !bs->drv->bdrv_debug_remove_breakpoint) {
39404cc70e93SFam Zheng         bs = bs->file;
39414cc70e93SFam Zheng     }
39424cc70e93SFam Zheng 
39434cc70e93SFam Zheng     if (bs && bs->drv && bs->drv->bdrv_debug_remove_breakpoint) {
39444cc70e93SFam Zheng         return bs->drv->bdrv_debug_remove_breakpoint(bs, tag);
39454cc70e93SFam Zheng     }
39464cc70e93SFam Zheng 
39474cc70e93SFam Zheng     return -ENOTSUP;
39484cc70e93SFam Zheng }
39494cc70e93SFam Zheng 
395041c695c7SKevin Wolf int bdrv_debug_resume(BlockDriverState *bs, const char *tag)
395141c695c7SKevin Wolf {
395241c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_resume) {
395341c695c7SKevin Wolf         bs = bs->file;
395441c695c7SKevin Wolf     }
395541c695c7SKevin Wolf 
395641c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_resume) {
395741c695c7SKevin Wolf         return bs->drv->bdrv_debug_resume(bs, tag);
395841c695c7SKevin Wolf     }
395941c695c7SKevin Wolf 
396041c695c7SKevin Wolf     return -ENOTSUP;
396141c695c7SKevin Wolf }
396241c695c7SKevin Wolf 
396341c695c7SKevin Wolf bool bdrv_debug_is_suspended(BlockDriverState *bs, const char *tag)
396441c695c7SKevin Wolf {
396541c695c7SKevin Wolf     while (bs && bs->drv && !bs->drv->bdrv_debug_is_suspended) {
396641c695c7SKevin Wolf         bs = bs->file;
396741c695c7SKevin Wolf     }
396841c695c7SKevin Wolf 
396941c695c7SKevin Wolf     if (bs && bs->drv && bs->drv->bdrv_debug_is_suspended) {
397041c695c7SKevin Wolf         return bs->drv->bdrv_debug_is_suspended(bs, tag);
397141c695c7SKevin Wolf     }
397241c695c7SKevin Wolf 
397341c695c7SKevin Wolf     return false;
39748b9b0cc2SKevin Wolf }
39758b9b0cc2SKevin Wolf 
3976199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
3977199630b6SBlue Swirl {
3978199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
3979199630b6SBlue Swirl }
3980199630b6SBlue Swirl 
3981b1b1d783SJeff Cody /* backing_file can either be relative, or absolute, or a protocol.  If it is
3982b1b1d783SJeff Cody  * relative, it must be relative to the chain.  So, passing in bs->filename
3983b1b1d783SJeff Cody  * from a BDS as backing_file should not be done, as that may be relative to
3984b1b1d783SJeff Cody  * the CWD rather than the chain. */
3985e8a6bb9cSMarcelo Tosatti BlockDriverState *bdrv_find_backing_image(BlockDriverState *bs,
3986e8a6bb9cSMarcelo Tosatti         const char *backing_file)
3987e8a6bb9cSMarcelo Tosatti {
3988b1b1d783SJeff Cody     char *filename_full = NULL;
3989b1b1d783SJeff Cody     char *backing_file_full = NULL;
3990b1b1d783SJeff Cody     char *filename_tmp = NULL;
3991b1b1d783SJeff Cody     int is_protocol = 0;
3992b1b1d783SJeff Cody     BlockDriverState *curr_bs = NULL;
3993b1b1d783SJeff Cody     BlockDriverState *retval = NULL;
3994b1b1d783SJeff Cody 
3995b1b1d783SJeff Cody     if (!bs || !bs->drv || !backing_file) {
3996e8a6bb9cSMarcelo Tosatti         return NULL;
3997e8a6bb9cSMarcelo Tosatti     }
3998e8a6bb9cSMarcelo Tosatti 
3999b1b1d783SJeff Cody     filename_full     = g_malloc(PATH_MAX);
4000b1b1d783SJeff Cody     backing_file_full = g_malloc(PATH_MAX);
4001b1b1d783SJeff Cody     filename_tmp      = g_malloc(PATH_MAX);
4002b1b1d783SJeff Cody 
4003b1b1d783SJeff Cody     is_protocol = path_has_protocol(backing_file);
4004b1b1d783SJeff Cody 
4005b1b1d783SJeff Cody     for (curr_bs = bs; curr_bs->backing_hd; curr_bs = curr_bs->backing_hd) {
4006b1b1d783SJeff Cody 
4007b1b1d783SJeff Cody         /* If either of the filename paths is actually a protocol, then
4008b1b1d783SJeff Cody          * compare unmodified paths; otherwise make paths relative */
4009b1b1d783SJeff Cody         if (is_protocol || path_has_protocol(curr_bs->backing_file)) {
4010b1b1d783SJeff Cody             if (strcmp(backing_file, curr_bs->backing_file) == 0) {
4011b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
4012b1b1d783SJeff Cody                 break;
4013b1b1d783SJeff Cody             }
4014e8a6bb9cSMarcelo Tosatti         } else {
4015b1b1d783SJeff Cody             /* If not an absolute filename path, make it relative to the current
4016b1b1d783SJeff Cody              * image's filename path */
4017b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
4018b1b1d783SJeff Cody                          backing_file);
4019b1b1d783SJeff Cody 
4020b1b1d783SJeff Cody             /* We are going to compare absolute pathnames */
4021b1b1d783SJeff Cody             if (!realpath(filename_tmp, filename_full)) {
4022b1b1d783SJeff Cody                 continue;
4023b1b1d783SJeff Cody             }
4024b1b1d783SJeff Cody 
4025b1b1d783SJeff Cody             /* We need to make sure the backing filename we are comparing against
4026b1b1d783SJeff Cody              * is relative to the current image filename (or absolute) */
4027b1b1d783SJeff Cody             path_combine(filename_tmp, PATH_MAX, curr_bs->filename,
4028b1b1d783SJeff Cody                          curr_bs->backing_file);
4029b1b1d783SJeff Cody 
4030b1b1d783SJeff Cody             if (!realpath(filename_tmp, backing_file_full)) {
4031b1b1d783SJeff Cody                 continue;
4032b1b1d783SJeff Cody             }
4033b1b1d783SJeff Cody 
4034b1b1d783SJeff Cody             if (strcmp(backing_file_full, filename_full) == 0) {
4035b1b1d783SJeff Cody                 retval = curr_bs->backing_hd;
4036b1b1d783SJeff Cody                 break;
4037b1b1d783SJeff Cody             }
4038e8a6bb9cSMarcelo Tosatti         }
4039e8a6bb9cSMarcelo Tosatti     }
4040e8a6bb9cSMarcelo Tosatti 
4041b1b1d783SJeff Cody     g_free(filename_full);
4042b1b1d783SJeff Cody     g_free(backing_file_full);
4043b1b1d783SJeff Cody     g_free(filename_tmp);
4044b1b1d783SJeff Cody     return retval;
4045e8a6bb9cSMarcelo Tosatti }
4046e8a6bb9cSMarcelo Tosatti 
4047f198fd1cSBenoît Canet int bdrv_get_backing_file_depth(BlockDriverState *bs)
4048f198fd1cSBenoît Canet {
4049f198fd1cSBenoît Canet     if (!bs->drv) {
4050f198fd1cSBenoît Canet         return 0;
4051f198fd1cSBenoît Canet     }
4052f198fd1cSBenoît Canet 
4053f198fd1cSBenoît Canet     if (!bs->backing_hd) {
4054f198fd1cSBenoît Canet         return 0;
4055f198fd1cSBenoît Canet     }
4056f198fd1cSBenoît Canet 
4057f198fd1cSBenoît Canet     return 1 + bdrv_get_backing_file_depth(bs->backing_hd);
4058f198fd1cSBenoît Canet }
4059f198fd1cSBenoît Canet 
406079fac568SJeff Cody BlockDriverState *bdrv_find_base(BlockDriverState *bs)
406179fac568SJeff Cody {
406279fac568SJeff Cody     BlockDriverState *curr_bs = NULL;
406379fac568SJeff Cody 
406479fac568SJeff Cody     if (!bs) {
406579fac568SJeff Cody         return NULL;
406679fac568SJeff Cody     }
406779fac568SJeff Cody 
406879fac568SJeff Cody     curr_bs = bs;
406979fac568SJeff Cody 
407079fac568SJeff Cody     while (curr_bs->backing_hd) {
407179fac568SJeff Cody         curr_bs = curr_bs->backing_hd;
407279fac568SJeff Cody     }
407379fac568SJeff Cody     return curr_bs;
407479fac568SJeff Cody }
407579fac568SJeff Cody 
4076ea2384d3Sbellard /**************************************************************/
407783f64091Sbellard /* async I/Os */
4078ea2384d3Sbellard 
40793b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
4080f141eafeSaliguori                                  QEMUIOVector *qiov, int nb_sectors,
408183f64091Sbellard                                  BlockDriverCompletionFunc *cb, void *opaque)
4082ea2384d3Sbellard {
4083bbf0a440SStefan Hajnoczi     trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
4084bbf0a440SStefan Hajnoczi 
4085d20d9b7cSPaolo Bonzini     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors, 0,
40868c5873d6SStefan Hajnoczi                                  cb, opaque, false);
408783f64091Sbellard }
408883f64091Sbellard 
4089f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
4090f141eafeSaliguori                                   QEMUIOVector *qiov, int nb_sectors,
409183f64091Sbellard                                   BlockDriverCompletionFunc *cb, void *opaque)
40927674e7bfSbellard {
4093bbf0a440SStefan Hajnoczi     trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
4094bbf0a440SStefan Hajnoczi 
4095d20d9b7cSPaolo Bonzini     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors, 0,
40968c5873d6SStefan Hajnoczi                                  cb, opaque, true);
409783f64091Sbellard }
409883f64091Sbellard 
4099d5ef94d4SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_write_zeroes(BlockDriverState *bs,
4100d5ef94d4SPaolo Bonzini         int64_t sector_num, int nb_sectors, BdrvRequestFlags flags,
4101d5ef94d4SPaolo Bonzini         BlockDriverCompletionFunc *cb, void *opaque)
4102d5ef94d4SPaolo Bonzini {
4103d5ef94d4SPaolo Bonzini     trace_bdrv_aio_write_zeroes(bs, sector_num, nb_sectors, flags, opaque);
4104d5ef94d4SPaolo Bonzini 
4105d5ef94d4SPaolo Bonzini     return bdrv_co_aio_rw_vector(bs, sector_num, NULL, nb_sectors,
4106d5ef94d4SPaolo Bonzini                                  BDRV_REQ_ZERO_WRITE | flags,
4107d5ef94d4SPaolo Bonzini                                  cb, opaque, true);
4108d5ef94d4SPaolo Bonzini }
4109d5ef94d4SPaolo Bonzini 
411040b4f539SKevin Wolf 
411140b4f539SKevin Wolf typedef struct MultiwriteCB {
411240b4f539SKevin Wolf     int error;
411340b4f539SKevin Wolf     int num_requests;
411440b4f539SKevin Wolf     int num_callbacks;
411540b4f539SKevin Wolf     struct {
411640b4f539SKevin Wolf         BlockDriverCompletionFunc *cb;
411740b4f539SKevin Wolf         void *opaque;
411840b4f539SKevin Wolf         QEMUIOVector *free_qiov;
411940b4f539SKevin Wolf     } callbacks[];
412040b4f539SKevin Wolf } MultiwriteCB;
412140b4f539SKevin Wolf 
412240b4f539SKevin Wolf static void multiwrite_user_cb(MultiwriteCB *mcb)
412340b4f539SKevin Wolf {
412440b4f539SKevin Wolf     int i;
412540b4f539SKevin Wolf 
412640b4f539SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
412740b4f539SKevin Wolf         mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
41281e1ea48dSStefan Hajnoczi         if (mcb->callbacks[i].free_qiov) {
41291e1ea48dSStefan Hajnoczi             qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
41301e1ea48dSStefan Hajnoczi         }
41317267c094SAnthony Liguori         g_free(mcb->callbacks[i].free_qiov);
413240b4f539SKevin Wolf     }
413340b4f539SKevin Wolf }
413440b4f539SKevin Wolf 
413540b4f539SKevin Wolf static void multiwrite_cb(void *opaque, int ret)
413640b4f539SKevin Wolf {
413740b4f539SKevin Wolf     MultiwriteCB *mcb = opaque;
413840b4f539SKevin Wolf 
41396d519a5fSStefan Hajnoczi     trace_multiwrite_cb(mcb, ret);
41406d519a5fSStefan Hajnoczi 
4141cb6d3ca0SKevin Wolf     if (ret < 0 && !mcb->error) {
414240b4f539SKevin Wolf         mcb->error = ret;
414340b4f539SKevin Wolf     }
414440b4f539SKevin Wolf 
414540b4f539SKevin Wolf     mcb->num_requests--;
414640b4f539SKevin Wolf     if (mcb->num_requests == 0) {
414740b4f539SKevin Wolf         multiwrite_user_cb(mcb);
41487267c094SAnthony Liguori         g_free(mcb);
414940b4f539SKevin Wolf     }
415040b4f539SKevin Wolf }
415140b4f539SKevin Wolf 
415240b4f539SKevin Wolf static int multiwrite_req_compare(const void *a, const void *b)
415340b4f539SKevin Wolf {
415477be4366SChristoph Hellwig     const BlockRequest *req1 = a, *req2 = b;
415577be4366SChristoph Hellwig 
415677be4366SChristoph Hellwig     /*
415777be4366SChristoph Hellwig      * Note that we can't simply subtract req2->sector from req1->sector
415877be4366SChristoph Hellwig      * here as that could overflow the return value.
415977be4366SChristoph Hellwig      */
416077be4366SChristoph Hellwig     if (req1->sector > req2->sector) {
416177be4366SChristoph Hellwig         return 1;
416277be4366SChristoph Hellwig     } else if (req1->sector < req2->sector) {
416377be4366SChristoph Hellwig         return -1;
416477be4366SChristoph Hellwig     } else {
416577be4366SChristoph Hellwig         return 0;
416677be4366SChristoph Hellwig     }
416740b4f539SKevin Wolf }
416840b4f539SKevin Wolf 
416940b4f539SKevin Wolf /*
417040b4f539SKevin Wolf  * Takes a bunch of requests and tries to merge them. Returns the number of
417140b4f539SKevin Wolf  * requests that remain after merging.
417240b4f539SKevin Wolf  */
417340b4f539SKevin Wolf static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
417440b4f539SKevin Wolf     int num_reqs, MultiwriteCB *mcb)
417540b4f539SKevin Wolf {
417640b4f539SKevin Wolf     int i, outidx;
417740b4f539SKevin Wolf 
417840b4f539SKevin Wolf     // Sort requests by start sector
417940b4f539SKevin Wolf     qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
418040b4f539SKevin Wolf 
418140b4f539SKevin Wolf     // Check if adjacent requests touch the same clusters. If so, combine them,
418240b4f539SKevin Wolf     // filling up gaps with zero sectors.
418340b4f539SKevin Wolf     outidx = 0;
418440b4f539SKevin Wolf     for (i = 1; i < num_reqs; i++) {
418540b4f539SKevin Wolf         int merge = 0;
418640b4f539SKevin Wolf         int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
418740b4f539SKevin Wolf 
4188b6a127a1SPaolo Bonzini         // Handle exactly sequential writes and overlapping writes.
418940b4f539SKevin Wolf         if (reqs[i].sector <= oldreq_last) {
419040b4f539SKevin Wolf             merge = 1;
419140b4f539SKevin Wolf         }
419240b4f539SKevin Wolf 
4193e2a305fbSChristoph Hellwig         if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
4194e2a305fbSChristoph Hellwig             merge = 0;
4195e2a305fbSChristoph Hellwig         }
4196e2a305fbSChristoph Hellwig 
419740b4f539SKevin Wolf         if (merge) {
419840b4f539SKevin Wolf             size_t size;
41997267c094SAnthony Liguori             QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
420040b4f539SKevin Wolf             qemu_iovec_init(qiov,
420140b4f539SKevin Wolf                 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
420240b4f539SKevin Wolf 
420340b4f539SKevin Wolf             // Add the first request to the merged one. If the requests are
420440b4f539SKevin Wolf             // overlapping, drop the last sectors of the first request.
420540b4f539SKevin Wolf             size = (reqs[i].sector - reqs[outidx].sector) << 9;
42061b093c48SMichael Tokarev             qemu_iovec_concat(qiov, reqs[outidx].qiov, 0, size);
420740b4f539SKevin Wolf 
4208b6a127a1SPaolo Bonzini             // We should need to add any zeros between the two requests
4209b6a127a1SPaolo Bonzini             assert (reqs[i].sector <= oldreq_last);
421040b4f539SKevin Wolf 
421140b4f539SKevin Wolf             // Add the second request
42121b093c48SMichael Tokarev             qemu_iovec_concat(qiov, reqs[i].qiov, 0, reqs[i].qiov->size);
421340b4f539SKevin Wolf 
4214cbf1dff2SKevin Wolf             reqs[outidx].nb_sectors = qiov->size >> 9;
421540b4f539SKevin Wolf             reqs[outidx].qiov = qiov;
421640b4f539SKevin Wolf 
421740b4f539SKevin Wolf             mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
421840b4f539SKevin Wolf         } else {
421940b4f539SKevin Wolf             outidx++;
422040b4f539SKevin Wolf             reqs[outidx].sector     = reqs[i].sector;
422140b4f539SKevin Wolf             reqs[outidx].nb_sectors = reqs[i].nb_sectors;
422240b4f539SKevin Wolf             reqs[outidx].qiov       = reqs[i].qiov;
422340b4f539SKevin Wolf         }
422440b4f539SKevin Wolf     }
422540b4f539SKevin Wolf 
422640b4f539SKevin Wolf     return outidx + 1;
422740b4f539SKevin Wolf }
422840b4f539SKevin Wolf 
422940b4f539SKevin Wolf /*
423040b4f539SKevin Wolf  * Submit multiple AIO write requests at once.
423140b4f539SKevin Wolf  *
423240b4f539SKevin Wolf  * On success, the function returns 0 and all requests in the reqs array have
423340b4f539SKevin Wolf  * been submitted. In error case this function returns -1, and any of the
423440b4f539SKevin Wolf  * requests may or may not be submitted yet. In particular, this means that the
423540b4f539SKevin Wolf  * callback will be called for some of the requests, for others it won't. The
423640b4f539SKevin Wolf  * caller must check the error field of the BlockRequest to wait for the right
423740b4f539SKevin Wolf  * callbacks (if error != 0, no callback will be called).
423840b4f539SKevin Wolf  *
423940b4f539SKevin Wolf  * The implementation may modify the contents of the reqs array, e.g. to merge
424040b4f539SKevin Wolf  * requests. However, the fields opaque and error are left unmodified as they
424140b4f539SKevin Wolf  * are used to signal failure for a single request to the caller.
424240b4f539SKevin Wolf  */
424340b4f539SKevin Wolf int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
424440b4f539SKevin Wolf {
424540b4f539SKevin Wolf     MultiwriteCB *mcb;
424640b4f539SKevin Wolf     int i;
424740b4f539SKevin Wolf 
4248301db7c2SRyan Harper     /* don't submit writes if we don't have a medium */
4249301db7c2SRyan Harper     if (bs->drv == NULL) {
4250301db7c2SRyan Harper         for (i = 0; i < num_reqs; i++) {
4251301db7c2SRyan Harper             reqs[i].error = -ENOMEDIUM;
4252301db7c2SRyan Harper         }
4253301db7c2SRyan Harper         return -1;
4254301db7c2SRyan Harper     }
4255301db7c2SRyan Harper 
425640b4f539SKevin Wolf     if (num_reqs == 0) {
425740b4f539SKevin Wolf         return 0;
425840b4f539SKevin Wolf     }
425940b4f539SKevin Wolf 
426040b4f539SKevin Wolf     // Create MultiwriteCB structure
42617267c094SAnthony Liguori     mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
426240b4f539SKevin Wolf     mcb->num_requests = 0;
426340b4f539SKevin Wolf     mcb->num_callbacks = num_reqs;
426440b4f539SKevin Wolf 
426540b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
426640b4f539SKevin Wolf         mcb->callbacks[i].cb = reqs[i].cb;
426740b4f539SKevin Wolf         mcb->callbacks[i].opaque = reqs[i].opaque;
426840b4f539SKevin Wolf     }
426940b4f539SKevin Wolf 
427040b4f539SKevin Wolf     // Check for mergable requests
427140b4f539SKevin Wolf     num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
427240b4f539SKevin Wolf 
42736d519a5fSStefan Hajnoczi     trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
42746d519a5fSStefan Hajnoczi 
4275df9309fbSPaolo Bonzini     /* Run the aio requests. */
4276df9309fbSPaolo Bonzini     mcb->num_requests = num_reqs;
427740b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
4278d20d9b7cSPaolo Bonzini         bdrv_co_aio_rw_vector(bs, reqs[i].sector, reqs[i].qiov,
4279d20d9b7cSPaolo Bonzini                               reqs[i].nb_sectors, reqs[i].flags,
4280d20d9b7cSPaolo Bonzini                               multiwrite_cb, mcb,
4281d20d9b7cSPaolo Bonzini                               true);
428240b4f539SKevin Wolf     }
428340b4f539SKevin Wolf 
428440b4f539SKevin Wolf     return 0;
428540b4f539SKevin Wolf }
428640b4f539SKevin Wolf 
428783f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
428883f64091Sbellard {
4289d7331bedSStefan Hajnoczi     acb->aiocb_info->cancel(acb);
429083f64091Sbellard }
429183f64091Sbellard 
429283f64091Sbellard /**************************************************************/
429383f64091Sbellard /* async block device emulation */
429483f64091Sbellard 
4295c16b5a2cSChristoph Hellwig typedef struct BlockDriverAIOCBSync {
4296c16b5a2cSChristoph Hellwig     BlockDriverAIOCB common;
4297c16b5a2cSChristoph Hellwig     QEMUBH *bh;
4298c16b5a2cSChristoph Hellwig     int ret;
4299c16b5a2cSChristoph Hellwig     /* vector translation state */
4300c16b5a2cSChristoph Hellwig     QEMUIOVector *qiov;
4301c16b5a2cSChristoph Hellwig     uint8_t *bounce;
4302c16b5a2cSChristoph Hellwig     int is_write;
4303c16b5a2cSChristoph Hellwig } BlockDriverAIOCBSync;
4304c16b5a2cSChristoph Hellwig 
4305c16b5a2cSChristoph Hellwig static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
4306c16b5a2cSChristoph Hellwig {
4307b666d239SKevin Wolf     BlockDriverAIOCBSync *acb =
4308b666d239SKevin Wolf         container_of(blockacb, BlockDriverAIOCBSync, common);
43096a7ad299SDor Laor     qemu_bh_delete(acb->bh);
431036afc451SAvi Kivity     acb->bh = NULL;
4311c16b5a2cSChristoph Hellwig     qemu_aio_release(acb);
4312c16b5a2cSChristoph Hellwig }
4313c16b5a2cSChristoph Hellwig 
4314d7331bedSStefan Hajnoczi static const AIOCBInfo bdrv_em_aiocb_info = {
4315c16b5a2cSChristoph Hellwig     .aiocb_size         = sizeof(BlockDriverAIOCBSync),
4316c16b5a2cSChristoph Hellwig     .cancel             = bdrv_aio_cancel_em,
4317c16b5a2cSChristoph Hellwig };
4318c16b5a2cSChristoph Hellwig 
431983f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
4320beac80cdSbellard {
4321ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = opaque;
4322f141eafeSaliguori 
4323f141eafeSaliguori     if (!acb->is_write)
432403396148SMichael Tokarev         qemu_iovec_from_buf(acb->qiov, 0, acb->bounce, acb->qiov->size);
4325ceb42de8Saliguori     qemu_vfree(acb->bounce);
4326ce1a14dcSpbrook     acb->common.cb(acb->common.opaque, acb->ret);
43276a7ad299SDor Laor     qemu_bh_delete(acb->bh);
432836afc451SAvi Kivity     acb->bh = NULL;
4329ce1a14dcSpbrook     qemu_aio_release(acb);
4330beac80cdSbellard }
4331beac80cdSbellard 
4332f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
4333f141eafeSaliguori                                             int64_t sector_num,
4334f141eafeSaliguori                                             QEMUIOVector *qiov,
4335f141eafeSaliguori                                             int nb_sectors,
4336f141eafeSaliguori                                             BlockDriverCompletionFunc *cb,
4337f141eafeSaliguori                                             void *opaque,
4338f141eafeSaliguori                                             int is_write)
4339f141eafeSaliguori 
4340ea2384d3Sbellard {
4341ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
434283f64091Sbellard 
4343d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_aiocb_info, bs, cb, opaque);
4344f141eafeSaliguori     acb->is_write = is_write;
4345f141eafeSaliguori     acb->qiov = qiov;
4346e268ca52Saliguori     acb->bounce = qemu_blockalign(bs, qiov->size);
4347ce1a14dcSpbrook     acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
4348f141eafeSaliguori 
4349f141eafeSaliguori     if (is_write) {
4350d5e6b161SMichael Tokarev         qemu_iovec_to_buf(acb->qiov, 0, acb->bounce, qiov->size);
43511ed20acfSStefan Hajnoczi         acb->ret = bs->drv->bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
4352f141eafeSaliguori     } else {
43531ed20acfSStefan Hajnoczi         acb->ret = bs->drv->bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
4354f141eafeSaliguori     }
4355f141eafeSaliguori 
4356ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
4357f141eafeSaliguori 
4358ce1a14dcSpbrook     return &acb->common;
43597a6cba61Spbrook }
43607a6cba61Spbrook 
4361f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
4362f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
4363ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
436483f64091Sbellard {
4365f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
436683f64091Sbellard }
436783f64091Sbellard 
4368f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
4369f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
4370f141eafeSaliguori         BlockDriverCompletionFunc *cb, void *opaque)
4371f141eafeSaliguori {
4372f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
4373f141eafeSaliguori }
4374f141eafeSaliguori 
437568485420SKevin Wolf 
437668485420SKevin Wolf typedef struct BlockDriverAIOCBCoroutine {
437768485420SKevin Wolf     BlockDriverAIOCB common;
437868485420SKevin Wolf     BlockRequest req;
437968485420SKevin Wolf     bool is_write;
4380d318aea9SKevin Wolf     bool *done;
438168485420SKevin Wolf     QEMUBH* bh;
438268485420SKevin Wolf } BlockDriverAIOCBCoroutine;
438368485420SKevin Wolf 
438468485420SKevin Wolf static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
438568485420SKevin Wolf {
4386d318aea9SKevin Wolf     BlockDriverAIOCBCoroutine *acb =
4387d318aea9SKevin Wolf         container_of(blockacb, BlockDriverAIOCBCoroutine, common);
4388d318aea9SKevin Wolf     bool done = false;
4389d318aea9SKevin Wolf 
4390d318aea9SKevin Wolf     acb->done = &done;
4391d318aea9SKevin Wolf     while (!done) {
4392d318aea9SKevin Wolf         qemu_aio_wait();
4393d318aea9SKevin Wolf     }
439468485420SKevin Wolf }
439568485420SKevin Wolf 
4396d7331bedSStefan Hajnoczi static const AIOCBInfo bdrv_em_co_aiocb_info = {
439768485420SKevin Wolf     .aiocb_size         = sizeof(BlockDriverAIOCBCoroutine),
439868485420SKevin Wolf     .cancel             = bdrv_aio_co_cancel_em,
439968485420SKevin Wolf };
440068485420SKevin Wolf 
440135246a68SPaolo Bonzini static void bdrv_co_em_bh(void *opaque)
440268485420SKevin Wolf {
440368485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb = opaque;
440468485420SKevin Wolf 
440568485420SKevin Wolf     acb->common.cb(acb->common.opaque, acb->req.error);
4406d318aea9SKevin Wolf 
4407d318aea9SKevin Wolf     if (acb->done) {
4408d318aea9SKevin Wolf         *acb->done = true;
4409d318aea9SKevin Wolf     }
4410d318aea9SKevin Wolf 
441168485420SKevin Wolf     qemu_bh_delete(acb->bh);
441268485420SKevin Wolf     qemu_aio_release(acb);
441368485420SKevin Wolf }
441468485420SKevin Wolf 
4415b2a61371SStefan Hajnoczi /* Invoke bdrv_co_do_readv/bdrv_co_do_writev */
4416b2a61371SStefan Hajnoczi static void coroutine_fn bdrv_co_do_rw(void *opaque)
4417b2a61371SStefan Hajnoczi {
4418b2a61371SStefan Hajnoczi     BlockDriverAIOCBCoroutine *acb = opaque;
4419b2a61371SStefan Hajnoczi     BlockDriverState *bs = acb->common.bs;
4420b2a61371SStefan Hajnoczi 
4421b2a61371SStefan Hajnoczi     if (!acb->is_write) {
4422b2a61371SStefan Hajnoczi         acb->req.error = bdrv_co_do_readv(bs, acb->req.sector,
4423d20d9b7cSPaolo Bonzini             acb->req.nb_sectors, acb->req.qiov, acb->req.flags);
4424b2a61371SStefan Hajnoczi     } else {
4425b2a61371SStefan Hajnoczi         acb->req.error = bdrv_co_do_writev(bs, acb->req.sector,
4426d20d9b7cSPaolo Bonzini             acb->req.nb_sectors, acb->req.qiov, acb->req.flags);
4427b2a61371SStefan Hajnoczi     }
4428b2a61371SStefan Hajnoczi 
442935246a68SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
4430b2a61371SStefan Hajnoczi     qemu_bh_schedule(acb->bh);
4431b2a61371SStefan Hajnoczi }
4432b2a61371SStefan Hajnoczi 
443368485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
443468485420SKevin Wolf                                                int64_t sector_num,
443568485420SKevin Wolf                                                QEMUIOVector *qiov,
443668485420SKevin Wolf                                                int nb_sectors,
4437d20d9b7cSPaolo Bonzini                                                BdrvRequestFlags flags,
443868485420SKevin Wolf                                                BlockDriverCompletionFunc *cb,
443968485420SKevin Wolf                                                void *opaque,
44408c5873d6SStefan Hajnoczi                                                bool is_write)
444168485420SKevin Wolf {
444268485420SKevin Wolf     Coroutine *co;
444368485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb;
444468485420SKevin Wolf 
4445d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
444668485420SKevin Wolf     acb->req.sector = sector_num;
444768485420SKevin Wolf     acb->req.nb_sectors = nb_sectors;
444868485420SKevin Wolf     acb->req.qiov = qiov;
4449d20d9b7cSPaolo Bonzini     acb->req.flags = flags;
445068485420SKevin Wolf     acb->is_write = is_write;
4451d318aea9SKevin Wolf     acb->done = NULL;
445268485420SKevin Wolf 
44538c5873d6SStefan Hajnoczi     co = qemu_coroutine_create(bdrv_co_do_rw);
445468485420SKevin Wolf     qemu_coroutine_enter(co, acb);
445568485420SKevin Wolf 
445668485420SKevin Wolf     return &acb->common;
445768485420SKevin Wolf }
445868485420SKevin Wolf 
445907f07615SPaolo Bonzini static void coroutine_fn bdrv_aio_flush_co_entry(void *opaque)
4460b2e12bc6SChristoph Hellwig {
446107f07615SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb = opaque;
446207f07615SPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
4463b2e12bc6SChristoph Hellwig 
446407f07615SPaolo Bonzini     acb->req.error = bdrv_co_flush(bs);
446507f07615SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
4466b2e12bc6SChristoph Hellwig     qemu_bh_schedule(acb->bh);
4467b2e12bc6SChristoph Hellwig }
4468b2e12bc6SChristoph Hellwig 
446907f07615SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
4470016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque)
4471016f5cf6SAlexander Graf {
447207f07615SPaolo Bonzini     trace_bdrv_aio_flush(bs, opaque);
4473016f5cf6SAlexander Graf 
447407f07615SPaolo Bonzini     Coroutine *co;
447507f07615SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb;
4476016f5cf6SAlexander Graf 
4477d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
4478d318aea9SKevin Wolf     acb->done = NULL;
4479d318aea9SKevin Wolf 
448007f07615SPaolo Bonzini     co = qemu_coroutine_create(bdrv_aio_flush_co_entry);
448107f07615SPaolo Bonzini     qemu_coroutine_enter(co, acb);
4482016f5cf6SAlexander Graf 
4483016f5cf6SAlexander Graf     return &acb->common;
4484016f5cf6SAlexander Graf }
4485016f5cf6SAlexander Graf 
44864265d620SPaolo Bonzini static void coroutine_fn bdrv_aio_discard_co_entry(void *opaque)
44874265d620SPaolo Bonzini {
44884265d620SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb = opaque;
44894265d620SPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
44904265d620SPaolo Bonzini 
44914265d620SPaolo Bonzini     acb->req.error = bdrv_co_discard(bs, acb->req.sector, acb->req.nb_sectors);
44924265d620SPaolo Bonzini     acb->bh = qemu_bh_new(bdrv_co_em_bh, acb);
44934265d620SPaolo Bonzini     qemu_bh_schedule(acb->bh);
44944265d620SPaolo Bonzini }
44954265d620SPaolo Bonzini 
44964265d620SPaolo Bonzini BlockDriverAIOCB *bdrv_aio_discard(BlockDriverState *bs,
44974265d620SPaolo Bonzini         int64_t sector_num, int nb_sectors,
44984265d620SPaolo Bonzini         BlockDriverCompletionFunc *cb, void *opaque)
44994265d620SPaolo Bonzini {
45004265d620SPaolo Bonzini     Coroutine *co;
45014265d620SPaolo Bonzini     BlockDriverAIOCBCoroutine *acb;
45024265d620SPaolo Bonzini 
45034265d620SPaolo Bonzini     trace_bdrv_aio_discard(bs, sector_num, nb_sectors, opaque);
45044265d620SPaolo Bonzini 
4505d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&bdrv_em_co_aiocb_info, bs, cb, opaque);
45064265d620SPaolo Bonzini     acb->req.sector = sector_num;
45074265d620SPaolo Bonzini     acb->req.nb_sectors = nb_sectors;
4508d318aea9SKevin Wolf     acb->done = NULL;
45094265d620SPaolo Bonzini     co = qemu_coroutine_create(bdrv_aio_discard_co_entry);
45104265d620SPaolo Bonzini     qemu_coroutine_enter(co, acb);
45114265d620SPaolo Bonzini 
45124265d620SPaolo Bonzini     return &acb->common;
45134265d620SPaolo Bonzini }
45144265d620SPaolo Bonzini 
4515ea2384d3Sbellard void bdrv_init(void)
4516ea2384d3Sbellard {
45175efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
4518ea2384d3Sbellard }
4519ce1a14dcSpbrook 
4520eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
4521eb852011SMarkus Armbruster {
4522eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
4523eb852011SMarkus Armbruster     bdrv_init();
4524eb852011SMarkus Armbruster }
4525eb852011SMarkus Armbruster 
4526d7331bedSStefan Hajnoczi void *qemu_aio_get(const AIOCBInfo *aiocb_info, BlockDriverState *bs,
45276bbff9a0Saliguori                    BlockDriverCompletionFunc *cb, void *opaque)
45286bbff9a0Saliguori {
4529ce1a14dcSpbrook     BlockDriverAIOCB *acb;
4530ce1a14dcSpbrook 
4531d7331bedSStefan Hajnoczi     acb = g_slice_alloc(aiocb_info->aiocb_size);
4532d7331bedSStefan Hajnoczi     acb->aiocb_info = aiocb_info;
4533ce1a14dcSpbrook     acb->bs = bs;
4534ce1a14dcSpbrook     acb->cb = cb;
4535ce1a14dcSpbrook     acb->opaque = opaque;
4536ce1a14dcSpbrook     return acb;
4537ce1a14dcSpbrook }
4538ce1a14dcSpbrook 
4539ce1a14dcSpbrook void qemu_aio_release(void *p)
4540ce1a14dcSpbrook {
4541d37c975fSStefan Hajnoczi     BlockDriverAIOCB *acb = p;
4542d7331bedSStefan Hajnoczi     g_slice_free1(acb->aiocb_info->aiocb_size, acb);
4543ce1a14dcSpbrook }
454419cb3738Sbellard 
454519cb3738Sbellard /**************************************************************/
4546f9f05dc5SKevin Wolf /* Coroutine block device emulation */
4547f9f05dc5SKevin Wolf 
4548f9f05dc5SKevin Wolf typedef struct CoroutineIOCompletion {
4549f9f05dc5SKevin Wolf     Coroutine *coroutine;
4550f9f05dc5SKevin Wolf     int ret;
4551f9f05dc5SKevin Wolf } CoroutineIOCompletion;
4552f9f05dc5SKevin Wolf 
4553f9f05dc5SKevin Wolf static void bdrv_co_io_em_complete(void *opaque, int ret)
4554f9f05dc5SKevin Wolf {
4555f9f05dc5SKevin Wolf     CoroutineIOCompletion *co = opaque;
4556f9f05dc5SKevin Wolf 
4557f9f05dc5SKevin Wolf     co->ret = ret;
4558f9f05dc5SKevin Wolf     qemu_coroutine_enter(co->coroutine, NULL);
4559f9f05dc5SKevin Wolf }
4560f9f05dc5SKevin Wolf 
4561f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
4562f9f05dc5SKevin Wolf                                       int nb_sectors, QEMUIOVector *iov,
4563f9f05dc5SKevin Wolf                                       bool is_write)
4564f9f05dc5SKevin Wolf {
4565f9f05dc5SKevin Wolf     CoroutineIOCompletion co = {
4566f9f05dc5SKevin Wolf         .coroutine = qemu_coroutine_self(),
4567f9f05dc5SKevin Wolf     };
4568f9f05dc5SKevin Wolf     BlockDriverAIOCB *acb;
4569f9f05dc5SKevin Wolf 
4570f9f05dc5SKevin Wolf     if (is_write) {
4571a652d160SStefan Hajnoczi         acb = bs->drv->bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
4572f9f05dc5SKevin Wolf                                        bdrv_co_io_em_complete, &co);
4573f9f05dc5SKevin Wolf     } else {
4574a652d160SStefan Hajnoczi         acb = bs->drv->bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
4575f9f05dc5SKevin Wolf                                       bdrv_co_io_em_complete, &co);
4576f9f05dc5SKevin Wolf     }
4577f9f05dc5SKevin Wolf 
457859370aaaSStefan Hajnoczi     trace_bdrv_co_io_em(bs, sector_num, nb_sectors, is_write, acb);
4579f9f05dc5SKevin Wolf     if (!acb) {
4580f9f05dc5SKevin Wolf         return -EIO;
4581f9f05dc5SKevin Wolf     }
4582f9f05dc5SKevin Wolf     qemu_coroutine_yield();
4583f9f05dc5SKevin Wolf 
4584f9f05dc5SKevin Wolf     return co.ret;
4585f9f05dc5SKevin Wolf }
4586f9f05dc5SKevin Wolf 
4587f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
4588f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
4589f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
4590f9f05dc5SKevin Wolf {
4591f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
4592f9f05dc5SKevin Wolf }
4593f9f05dc5SKevin Wolf 
4594f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
4595f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
4596f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
4597f9f05dc5SKevin Wolf {
4598f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
4599f9f05dc5SKevin Wolf }
4600f9f05dc5SKevin Wolf 
460107f07615SPaolo Bonzini static void coroutine_fn bdrv_flush_co_entry(void *opaque)
4602e7a8a783SKevin Wolf {
460307f07615SPaolo Bonzini     RwCo *rwco = opaque;
460407f07615SPaolo Bonzini 
460507f07615SPaolo Bonzini     rwco->ret = bdrv_co_flush(rwco->bs);
460607f07615SPaolo Bonzini }
460707f07615SPaolo Bonzini 
460807f07615SPaolo Bonzini int coroutine_fn bdrv_co_flush(BlockDriverState *bs)
460907f07615SPaolo Bonzini {
4610eb489bb1SKevin Wolf     int ret;
4611eb489bb1SKevin Wolf 
461229cdb251SPaolo Bonzini     if (!bs || !bdrv_is_inserted(bs) || bdrv_is_read_only(bs)) {
461307f07615SPaolo Bonzini         return 0;
4614eb489bb1SKevin Wolf     }
4615eb489bb1SKevin Wolf 
4616ca716364SKevin Wolf     /* Write back cached data to the OS even with cache=unsafe */
4617bf736fe3SKevin Wolf     BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_OS);
4618eb489bb1SKevin Wolf     if (bs->drv->bdrv_co_flush_to_os) {
4619eb489bb1SKevin Wolf         ret = bs->drv->bdrv_co_flush_to_os(bs);
4620eb489bb1SKevin Wolf         if (ret < 0) {
4621eb489bb1SKevin Wolf             return ret;
4622eb489bb1SKevin Wolf         }
4623eb489bb1SKevin Wolf     }
4624eb489bb1SKevin Wolf 
4625ca716364SKevin Wolf     /* But don't actually force it to the disk with cache=unsafe */
4626ca716364SKevin Wolf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
4627d4c82329SKevin Wolf         goto flush_parent;
4628ca716364SKevin Wolf     }
4629ca716364SKevin Wolf 
4630bf736fe3SKevin Wolf     BLKDBG_EVENT(bs->file, BLKDBG_FLUSH_TO_DISK);
4631eb489bb1SKevin Wolf     if (bs->drv->bdrv_co_flush_to_disk) {
463229cdb251SPaolo Bonzini         ret = bs->drv->bdrv_co_flush_to_disk(bs);
463307f07615SPaolo Bonzini     } else if (bs->drv->bdrv_aio_flush) {
463407f07615SPaolo Bonzini         BlockDriverAIOCB *acb;
4635e7a8a783SKevin Wolf         CoroutineIOCompletion co = {
4636e7a8a783SKevin Wolf             .coroutine = qemu_coroutine_self(),
4637e7a8a783SKevin Wolf         };
4638e7a8a783SKevin Wolf 
463907f07615SPaolo Bonzini         acb = bs->drv->bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
464007f07615SPaolo Bonzini         if (acb == NULL) {
464129cdb251SPaolo Bonzini             ret = -EIO;
464207f07615SPaolo Bonzini         } else {
4643e7a8a783SKevin Wolf             qemu_coroutine_yield();
464429cdb251SPaolo Bonzini             ret = co.ret;
4645e7a8a783SKevin Wolf         }
464607f07615SPaolo Bonzini     } else {
464707f07615SPaolo Bonzini         /*
464807f07615SPaolo Bonzini          * Some block drivers always operate in either writethrough or unsafe
464907f07615SPaolo Bonzini          * mode and don't support bdrv_flush therefore. Usually qemu doesn't
465007f07615SPaolo Bonzini          * know how the server works (because the behaviour is hardcoded or
465107f07615SPaolo Bonzini          * depends on server-side configuration), so we can't ensure that
465207f07615SPaolo Bonzini          * everything is safe on disk. Returning an error doesn't work because
465307f07615SPaolo Bonzini          * that would break guests even if the server operates in writethrough
465407f07615SPaolo Bonzini          * mode.
465507f07615SPaolo Bonzini          *
465607f07615SPaolo Bonzini          * Let's hope the user knows what he's doing.
465707f07615SPaolo Bonzini          */
465829cdb251SPaolo Bonzini         ret = 0;
465907f07615SPaolo Bonzini     }
466029cdb251SPaolo Bonzini     if (ret < 0) {
466129cdb251SPaolo Bonzini         return ret;
466229cdb251SPaolo Bonzini     }
466329cdb251SPaolo Bonzini 
466429cdb251SPaolo Bonzini     /* Now flush the underlying protocol.  It will also have BDRV_O_NO_FLUSH
466529cdb251SPaolo Bonzini      * in the case of cache=unsafe, so there are no useless flushes.
466629cdb251SPaolo Bonzini      */
4667d4c82329SKevin Wolf flush_parent:
466829cdb251SPaolo Bonzini     return bdrv_co_flush(bs->file);
466907f07615SPaolo Bonzini }
467007f07615SPaolo Bonzini 
46710f15423cSAnthony Liguori void bdrv_invalidate_cache(BlockDriverState *bs)
46720f15423cSAnthony Liguori {
46730f15423cSAnthony Liguori     if (bs->drv && bs->drv->bdrv_invalidate_cache) {
46740f15423cSAnthony Liguori         bs->drv->bdrv_invalidate_cache(bs);
46750f15423cSAnthony Liguori     }
46760f15423cSAnthony Liguori }
46770f15423cSAnthony Liguori 
46780f15423cSAnthony Liguori void bdrv_invalidate_cache_all(void)
46790f15423cSAnthony Liguori {
46800f15423cSAnthony Liguori     BlockDriverState *bs;
46810f15423cSAnthony Liguori 
4682dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
46830f15423cSAnthony Liguori         bdrv_invalidate_cache(bs);
46840f15423cSAnthony Liguori     }
46850f15423cSAnthony Liguori }
46860f15423cSAnthony Liguori 
468707789269SBenoît Canet void bdrv_clear_incoming_migration_all(void)
468807789269SBenoît Canet {
468907789269SBenoît Canet     BlockDriverState *bs;
469007789269SBenoît Canet 
4691dc364f4cSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
469207789269SBenoît Canet         bs->open_flags = bs->open_flags & ~(BDRV_O_INCOMING);
469307789269SBenoît Canet     }
469407789269SBenoît Canet }
469507789269SBenoît Canet 
469607f07615SPaolo Bonzini int bdrv_flush(BlockDriverState *bs)
469707f07615SPaolo Bonzini {
469807f07615SPaolo Bonzini     Coroutine *co;
469907f07615SPaolo Bonzini     RwCo rwco = {
470007f07615SPaolo Bonzini         .bs = bs,
470107f07615SPaolo Bonzini         .ret = NOT_DONE,
470207f07615SPaolo Bonzini     };
470307f07615SPaolo Bonzini 
470407f07615SPaolo Bonzini     if (qemu_in_coroutine()) {
470507f07615SPaolo Bonzini         /* Fast-path if already in coroutine context */
470607f07615SPaolo Bonzini         bdrv_flush_co_entry(&rwco);
470707f07615SPaolo Bonzini     } else {
470807f07615SPaolo Bonzini         co = qemu_coroutine_create(bdrv_flush_co_entry);
470907f07615SPaolo Bonzini         qemu_coroutine_enter(co, &rwco);
471007f07615SPaolo Bonzini         while (rwco.ret == NOT_DONE) {
471107f07615SPaolo Bonzini             qemu_aio_wait();
471207f07615SPaolo Bonzini         }
471307f07615SPaolo Bonzini     }
471407f07615SPaolo Bonzini 
471507f07615SPaolo Bonzini     return rwco.ret;
471607f07615SPaolo Bonzini }
4717e7a8a783SKevin Wolf 
47184265d620SPaolo Bonzini static void coroutine_fn bdrv_discard_co_entry(void *opaque)
47194265d620SPaolo Bonzini {
47204265d620SPaolo Bonzini     RwCo *rwco = opaque;
47214265d620SPaolo Bonzini 
47224265d620SPaolo Bonzini     rwco->ret = bdrv_co_discard(rwco->bs, rwco->sector_num, rwco->nb_sectors);
47234265d620SPaolo Bonzini }
47244265d620SPaolo Bonzini 
47256f14da52SPeter Lieven /* if no limit is specified in the BlockLimits use a default
47266f14da52SPeter Lieven  * of 32768 512-byte sectors (16 MiB) per request.
47276f14da52SPeter Lieven  */
47286f14da52SPeter Lieven #define MAX_DISCARD_DEFAULT 32768
47296f14da52SPeter Lieven 
47304265d620SPaolo Bonzini int coroutine_fn bdrv_co_discard(BlockDriverState *bs, int64_t sector_num,
47314265d620SPaolo Bonzini                                  int nb_sectors)
47324265d620SPaolo Bonzini {
4733d51e9fe5SPaolo Bonzini     int max_discard;
4734d51e9fe5SPaolo Bonzini 
47354265d620SPaolo Bonzini     if (!bs->drv) {
47364265d620SPaolo Bonzini         return -ENOMEDIUM;
47374265d620SPaolo Bonzini     } else if (bdrv_check_request(bs, sector_num, nb_sectors)) {
47384265d620SPaolo Bonzini         return -EIO;
47394265d620SPaolo Bonzini     } else if (bs->read_only) {
47404265d620SPaolo Bonzini         return -EROFS;
4741df702c9bSPaolo Bonzini     }
4742df702c9bSPaolo Bonzini 
47438f0720ecSPaolo Bonzini     bdrv_reset_dirty(bs, sector_num, nb_sectors);
4744df702c9bSPaolo Bonzini 
47459e8f1835SPaolo Bonzini     /* Do nothing if disabled.  */
47469e8f1835SPaolo Bonzini     if (!(bs->open_flags & BDRV_O_UNMAP)) {
47479e8f1835SPaolo Bonzini         return 0;
47489e8f1835SPaolo Bonzini     }
47499e8f1835SPaolo Bonzini 
4750d51e9fe5SPaolo Bonzini     if (!bs->drv->bdrv_co_discard && !bs->drv->bdrv_aio_discard) {
4751d51e9fe5SPaolo Bonzini         return 0;
4752d51e9fe5SPaolo Bonzini     }
47536f14da52SPeter Lieven 
4754d51e9fe5SPaolo Bonzini     max_discard = bs->bl.max_discard ?  bs->bl.max_discard : MAX_DISCARD_DEFAULT;
47556f14da52SPeter Lieven     while (nb_sectors > 0) {
47566f14da52SPeter Lieven         int ret;
47576f14da52SPeter Lieven         int num = nb_sectors;
47586f14da52SPeter Lieven 
47596f14da52SPeter Lieven         /* align request */
47606f14da52SPeter Lieven         if (bs->bl.discard_alignment &&
47616f14da52SPeter Lieven             num >= bs->bl.discard_alignment &&
47626f14da52SPeter Lieven             sector_num % bs->bl.discard_alignment) {
47636f14da52SPeter Lieven             if (num > bs->bl.discard_alignment) {
47646f14da52SPeter Lieven                 num = bs->bl.discard_alignment;
47656f14da52SPeter Lieven             }
47666f14da52SPeter Lieven             num -= sector_num % bs->bl.discard_alignment;
47676f14da52SPeter Lieven         }
47686f14da52SPeter Lieven 
47696f14da52SPeter Lieven         /* limit request size */
47706f14da52SPeter Lieven         if (num > max_discard) {
47716f14da52SPeter Lieven             num = max_discard;
47726f14da52SPeter Lieven         }
47736f14da52SPeter Lieven 
4774d51e9fe5SPaolo Bonzini         if (bs->drv->bdrv_co_discard) {
47756f14da52SPeter Lieven             ret = bs->drv->bdrv_co_discard(bs, sector_num, num);
4776d51e9fe5SPaolo Bonzini         } else {
47774265d620SPaolo Bonzini             BlockDriverAIOCB *acb;
47784265d620SPaolo Bonzini             CoroutineIOCompletion co = {
47794265d620SPaolo Bonzini                 .coroutine = qemu_coroutine_self(),
47804265d620SPaolo Bonzini             };
47814265d620SPaolo Bonzini 
47824265d620SPaolo Bonzini             acb = bs->drv->bdrv_aio_discard(bs, sector_num, nb_sectors,
47834265d620SPaolo Bonzini                                             bdrv_co_io_em_complete, &co);
47844265d620SPaolo Bonzini             if (acb == NULL) {
47854265d620SPaolo Bonzini                 return -EIO;
47864265d620SPaolo Bonzini             } else {
47874265d620SPaolo Bonzini                 qemu_coroutine_yield();
4788d51e9fe5SPaolo Bonzini                 ret = co.ret;
47894265d620SPaolo Bonzini             }
4790d51e9fe5SPaolo Bonzini         }
47917ce21016SPaolo Bonzini         if (ret && ret != -ENOTSUP) {
4792d51e9fe5SPaolo Bonzini             return ret;
4793d51e9fe5SPaolo Bonzini         }
4794d51e9fe5SPaolo Bonzini 
4795d51e9fe5SPaolo Bonzini         sector_num += num;
4796d51e9fe5SPaolo Bonzini         nb_sectors -= num;
4797d51e9fe5SPaolo Bonzini     }
47984265d620SPaolo Bonzini     return 0;
47994265d620SPaolo Bonzini }
48004265d620SPaolo Bonzini 
48014265d620SPaolo Bonzini int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
48024265d620SPaolo Bonzini {
48034265d620SPaolo Bonzini     Coroutine *co;
48044265d620SPaolo Bonzini     RwCo rwco = {
48054265d620SPaolo Bonzini         .bs = bs,
48064265d620SPaolo Bonzini         .sector_num = sector_num,
48074265d620SPaolo Bonzini         .nb_sectors = nb_sectors,
48084265d620SPaolo Bonzini         .ret = NOT_DONE,
48094265d620SPaolo Bonzini     };
48104265d620SPaolo Bonzini 
48114265d620SPaolo Bonzini     if (qemu_in_coroutine()) {
48124265d620SPaolo Bonzini         /* Fast-path if already in coroutine context */
48134265d620SPaolo Bonzini         bdrv_discard_co_entry(&rwco);
48144265d620SPaolo Bonzini     } else {
48154265d620SPaolo Bonzini         co = qemu_coroutine_create(bdrv_discard_co_entry);
48164265d620SPaolo Bonzini         qemu_coroutine_enter(co, &rwco);
48174265d620SPaolo Bonzini         while (rwco.ret == NOT_DONE) {
48184265d620SPaolo Bonzini             qemu_aio_wait();
48194265d620SPaolo Bonzini         }
48204265d620SPaolo Bonzini     }
48214265d620SPaolo Bonzini 
48224265d620SPaolo Bonzini     return rwco.ret;
48234265d620SPaolo Bonzini }
48244265d620SPaolo Bonzini 
4825f9f05dc5SKevin Wolf /**************************************************************/
482619cb3738Sbellard /* removable device support */
482719cb3738Sbellard 
482819cb3738Sbellard /**
482919cb3738Sbellard  * Return TRUE if the media is present
483019cb3738Sbellard  */
483119cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
483219cb3738Sbellard {
483319cb3738Sbellard     BlockDriver *drv = bs->drv;
4834a1aff5bfSMarkus Armbruster 
483519cb3738Sbellard     if (!drv)
483619cb3738Sbellard         return 0;
483719cb3738Sbellard     if (!drv->bdrv_is_inserted)
4838a1aff5bfSMarkus Armbruster         return 1;
4839a1aff5bfSMarkus Armbruster     return drv->bdrv_is_inserted(bs);
484019cb3738Sbellard }
484119cb3738Sbellard 
484219cb3738Sbellard /**
48438e49ca46SMarkus Armbruster  * Return whether the media changed since the last call to this
48448e49ca46SMarkus Armbruster  * function, or -ENOTSUP if we don't know.  Most drivers don't know.
484519cb3738Sbellard  */
484619cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
484719cb3738Sbellard {
484819cb3738Sbellard     BlockDriver *drv = bs->drv;
484919cb3738Sbellard 
48508e49ca46SMarkus Armbruster     if (drv && drv->bdrv_media_changed) {
48518e49ca46SMarkus Armbruster         return drv->bdrv_media_changed(bs);
48528e49ca46SMarkus Armbruster     }
48538e49ca46SMarkus Armbruster     return -ENOTSUP;
485419cb3738Sbellard }
485519cb3738Sbellard 
485619cb3738Sbellard /**
485719cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
485819cb3738Sbellard  */
4859f36f3949SLuiz Capitulino void bdrv_eject(BlockDriverState *bs, bool eject_flag)
486019cb3738Sbellard {
486119cb3738Sbellard     BlockDriver *drv = bs->drv;
486219cb3738Sbellard 
4863822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
4864822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
486519cb3738Sbellard     }
48666f382ed2SLuiz Capitulino 
48676f382ed2SLuiz Capitulino     if (bs->device_name[0] != '\0') {
48686f382ed2SLuiz Capitulino         bdrv_emit_qmp_eject_event(bs, eject_flag);
48696f382ed2SLuiz Capitulino     }
487019cb3738Sbellard }
487119cb3738Sbellard 
487219cb3738Sbellard /**
487319cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
487419cb3738Sbellard  * to eject it manually).
487519cb3738Sbellard  */
4876025e849aSMarkus Armbruster void bdrv_lock_medium(BlockDriverState *bs, bool locked)
487719cb3738Sbellard {
487819cb3738Sbellard     BlockDriver *drv = bs->drv;
487919cb3738Sbellard 
4880025e849aSMarkus Armbruster     trace_bdrv_lock_medium(bs, locked);
4881b8c6d095SStefan Hajnoczi 
4882025e849aSMarkus Armbruster     if (drv && drv->bdrv_lock_medium) {
4883025e849aSMarkus Armbruster         drv->bdrv_lock_medium(bs, locked);
488419cb3738Sbellard     }
488519cb3738Sbellard }
4886985a03b0Sths 
4887985a03b0Sths /* needed for generic scsi interface */
4888985a03b0Sths 
4889985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
4890985a03b0Sths {
4891985a03b0Sths     BlockDriver *drv = bs->drv;
4892985a03b0Sths 
4893985a03b0Sths     if (drv && drv->bdrv_ioctl)
4894985a03b0Sths         return drv->bdrv_ioctl(bs, req, buf);
4895985a03b0Sths     return -ENOTSUP;
4896985a03b0Sths }
48977d780669Saliguori 
4898221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
4899221f715dSaliguori         unsigned long int req, void *buf,
49007d780669Saliguori         BlockDriverCompletionFunc *cb, void *opaque)
49017d780669Saliguori {
4902221f715dSaliguori     BlockDriver *drv = bs->drv;
49037d780669Saliguori 
4904221f715dSaliguori     if (drv && drv->bdrv_aio_ioctl)
4905221f715dSaliguori         return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
4906221f715dSaliguori     return NULL;
49077d780669Saliguori }
4908e268ca52Saliguori 
49091b7fd729SPaolo Bonzini void bdrv_set_guest_block_size(BlockDriverState *bs, int align)
49107b6f9300SMarkus Armbruster {
49111b7fd729SPaolo Bonzini     bs->guest_block_size = align;
49127b6f9300SMarkus Armbruster }
49137cd1e32aSlirans@il.ibm.com 
4914e268ca52Saliguori void *qemu_blockalign(BlockDriverState *bs, size_t size)
4915e268ca52Saliguori {
4916339064d5SKevin Wolf     return qemu_memalign(bdrv_opt_mem_align(bs), size);
4917e268ca52Saliguori }
49187cd1e32aSlirans@il.ibm.com 
4919c53b1c51SStefan Hajnoczi /*
4920c53b1c51SStefan Hajnoczi  * Check if all memory in this vector is sector aligned.
4921c53b1c51SStefan Hajnoczi  */
4922c53b1c51SStefan Hajnoczi bool bdrv_qiov_is_aligned(BlockDriverState *bs, QEMUIOVector *qiov)
4923c53b1c51SStefan Hajnoczi {
4924c53b1c51SStefan Hajnoczi     int i;
4925339064d5SKevin Wolf     size_t alignment = bdrv_opt_mem_align(bs);
4926c53b1c51SStefan Hajnoczi 
4927c53b1c51SStefan Hajnoczi     for (i = 0; i < qiov->niov; i++) {
4928339064d5SKevin Wolf         if ((uintptr_t) qiov->iov[i].iov_base % alignment) {
4929c53b1c51SStefan Hajnoczi             return false;
4930c53b1c51SStefan Hajnoczi         }
4931339064d5SKevin Wolf         if (qiov->iov[i].iov_len % alignment) {
49321ff735bdSKevin Wolf             return false;
49331ff735bdSKevin Wolf         }
4934c53b1c51SStefan Hajnoczi     }
4935c53b1c51SStefan Hajnoczi 
4936c53b1c51SStefan Hajnoczi     return true;
4937c53b1c51SStefan Hajnoczi }
4938c53b1c51SStefan Hajnoczi 
4939e4654d2dSFam Zheng BdrvDirtyBitmap *bdrv_create_dirty_bitmap(BlockDriverState *bs, int granularity)
49407cd1e32aSlirans@il.ibm.com {
49417cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
4942e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
4943a55eb92cSJan Kiszka 
494450717e94SPaolo Bonzini     assert((granularity & (granularity - 1)) == 0);
494550717e94SPaolo Bonzini 
494650717e94SPaolo Bonzini     granularity >>= BDRV_SECTOR_BITS;
4947e4654d2dSFam Zheng     assert(granularity);
49488f0720ecSPaolo Bonzini     bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS);
4949e4654d2dSFam Zheng     bitmap = g_malloc0(sizeof(BdrvDirtyBitmap));
4950e4654d2dSFam Zheng     bitmap->bitmap = hbitmap_alloc(bitmap_size, ffs(granularity) - 1);
4951e4654d2dSFam Zheng     QLIST_INSERT_HEAD(&bs->dirty_bitmaps, bitmap, list);
4952e4654d2dSFam Zheng     return bitmap;
4953e4654d2dSFam Zheng }
4954e4654d2dSFam Zheng 
4955e4654d2dSFam Zheng void bdrv_release_dirty_bitmap(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
4956e4654d2dSFam Zheng {
4957e4654d2dSFam Zheng     BdrvDirtyBitmap *bm, *next;
4958e4654d2dSFam Zheng     QLIST_FOREACH_SAFE(bm, &bs->dirty_bitmaps, list, next) {
4959e4654d2dSFam Zheng         if (bm == bitmap) {
4960e4654d2dSFam Zheng             QLIST_REMOVE(bitmap, list);
4961e4654d2dSFam Zheng             hbitmap_free(bitmap->bitmap);
4962e4654d2dSFam Zheng             g_free(bitmap);
4963e4654d2dSFam Zheng             return;
49647cd1e32aSlirans@il.ibm.com         }
49657cd1e32aSlirans@il.ibm.com     }
49667cd1e32aSlirans@il.ibm.com }
49677cd1e32aSlirans@il.ibm.com 
496821b56835SFam Zheng BlockDirtyInfoList *bdrv_query_dirty_bitmaps(BlockDriverState *bs)
496921b56835SFam Zheng {
497021b56835SFam Zheng     BdrvDirtyBitmap *bm;
497121b56835SFam Zheng     BlockDirtyInfoList *list = NULL;
497221b56835SFam Zheng     BlockDirtyInfoList **plist = &list;
497321b56835SFam Zheng 
497421b56835SFam Zheng     QLIST_FOREACH(bm, &bs->dirty_bitmaps, list) {
497521b56835SFam Zheng         BlockDirtyInfo *info = g_malloc0(sizeof(BlockDirtyInfo));
497621b56835SFam Zheng         BlockDirtyInfoList *entry = g_malloc0(sizeof(BlockDirtyInfoList));
497721b56835SFam Zheng         info->count = bdrv_get_dirty_count(bs, bm);
497821b56835SFam Zheng         info->granularity =
497921b56835SFam Zheng             ((int64_t) BDRV_SECTOR_SIZE << hbitmap_granularity(bm->bitmap));
498021b56835SFam Zheng         entry->value = info;
498121b56835SFam Zheng         *plist = entry;
498221b56835SFam Zheng         plist = &entry->next;
498321b56835SFam Zheng     }
498421b56835SFam Zheng 
498521b56835SFam Zheng     return list;
498621b56835SFam Zheng }
498721b56835SFam Zheng 
4988e4654d2dSFam Zheng int bdrv_get_dirty(BlockDriverState *bs, BdrvDirtyBitmap *bitmap, int64_t sector)
49897cd1e32aSlirans@il.ibm.com {
4990e4654d2dSFam Zheng     if (bitmap) {
4991e4654d2dSFam Zheng         return hbitmap_get(bitmap->bitmap, sector);
49927cd1e32aSlirans@il.ibm.com     } else {
49937cd1e32aSlirans@il.ibm.com         return 0;
49947cd1e32aSlirans@il.ibm.com     }
49957cd1e32aSlirans@il.ibm.com }
49967cd1e32aSlirans@il.ibm.com 
4997e4654d2dSFam Zheng void bdrv_dirty_iter_init(BlockDriverState *bs,
4998e4654d2dSFam Zheng                           BdrvDirtyBitmap *bitmap, HBitmapIter *hbi)
49991755da16SPaolo Bonzini {
5000e4654d2dSFam Zheng     hbitmap_iter_init(hbi, bitmap->bitmap, 0);
50011755da16SPaolo Bonzini }
50021755da16SPaolo Bonzini 
50031755da16SPaolo Bonzini void bdrv_set_dirty(BlockDriverState *bs, int64_t cur_sector,
50041755da16SPaolo Bonzini                     int nr_sectors)
50051755da16SPaolo Bonzini {
5006e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
5007e4654d2dSFam Zheng     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
5008e4654d2dSFam Zheng         hbitmap_set(bitmap->bitmap, cur_sector, nr_sectors);
5009e4654d2dSFam Zheng     }
50101755da16SPaolo Bonzini }
50111755da16SPaolo Bonzini 
5012e4654d2dSFam Zheng void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector, int nr_sectors)
50137cd1e32aSlirans@il.ibm.com {
5014e4654d2dSFam Zheng     BdrvDirtyBitmap *bitmap;
5015e4654d2dSFam Zheng     QLIST_FOREACH(bitmap, &bs->dirty_bitmaps, list) {
5016e4654d2dSFam Zheng         hbitmap_reset(bitmap->bitmap, cur_sector, nr_sectors);
5017e4654d2dSFam Zheng     }
50187cd1e32aSlirans@il.ibm.com }
5019aaa0eb75SLiran Schour 
5020e4654d2dSFam Zheng int64_t bdrv_get_dirty_count(BlockDriverState *bs, BdrvDirtyBitmap *bitmap)
5021aaa0eb75SLiran Schour {
5022e4654d2dSFam Zheng     return hbitmap_count(bitmap->bitmap);
5023aaa0eb75SLiran Schour }
5024f88e1a42SJes Sorensen 
50259fcb0251SFam Zheng /* Get a reference to bs */
50269fcb0251SFam Zheng void bdrv_ref(BlockDriverState *bs)
50279fcb0251SFam Zheng {
50289fcb0251SFam Zheng     bs->refcnt++;
50299fcb0251SFam Zheng }
50309fcb0251SFam Zheng 
50319fcb0251SFam Zheng /* Release a previously grabbed reference to bs.
50329fcb0251SFam Zheng  * If after releasing, reference count is zero, the BlockDriverState is
50339fcb0251SFam Zheng  * deleted. */
50349fcb0251SFam Zheng void bdrv_unref(BlockDriverState *bs)
50359fcb0251SFam Zheng {
50369fcb0251SFam Zheng     assert(bs->refcnt > 0);
50379fcb0251SFam Zheng     if (--bs->refcnt == 0) {
50389fcb0251SFam Zheng         bdrv_delete(bs);
50399fcb0251SFam Zheng     }
50409fcb0251SFam Zheng }
50419fcb0251SFam Zheng 
5042db593f25SMarcelo Tosatti void bdrv_set_in_use(BlockDriverState *bs, int in_use)
5043db593f25SMarcelo Tosatti {
5044db593f25SMarcelo Tosatti     assert(bs->in_use != in_use);
5045db593f25SMarcelo Tosatti     bs->in_use = in_use;
5046db593f25SMarcelo Tosatti }
5047db593f25SMarcelo Tosatti 
5048db593f25SMarcelo Tosatti int bdrv_in_use(BlockDriverState *bs)
5049db593f25SMarcelo Tosatti {
5050db593f25SMarcelo Tosatti     return bs->in_use;
5051db593f25SMarcelo Tosatti }
5052db593f25SMarcelo Tosatti 
505328a7282aSLuiz Capitulino void bdrv_iostatus_enable(BlockDriverState *bs)
505428a7282aSLuiz Capitulino {
5055d6bf279eSLuiz Capitulino     bs->iostatus_enabled = true;
505658e21ef5SLuiz Capitulino     bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
505728a7282aSLuiz Capitulino }
505828a7282aSLuiz Capitulino 
505928a7282aSLuiz Capitulino /* The I/O status is only enabled if the drive explicitly
506028a7282aSLuiz Capitulino  * enables it _and_ the VM is configured to stop on errors */
506128a7282aSLuiz Capitulino bool bdrv_iostatus_is_enabled(const BlockDriverState *bs)
506228a7282aSLuiz Capitulino {
5063d6bf279eSLuiz Capitulino     return (bs->iostatus_enabled &&
506492aa5c6dSPaolo Bonzini            (bs->on_write_error == BLOCKDEV_ON_ERROR_ENOSPC ||
506592aa5c6dSPaolo Bonzini             bs->on_write_error == BLOCKDEV_ON_ERROR_STOP   ||
506692aa5c6dSPaolo Bonzini             bs->on_read_error == BLOCKDEV_ON_ERROR_STOP));
506728a7282aSLuiz Capitulino }
506828a7282aSLuiz Capitulino 
506928a7282aSLuiz Capitulino void bdrv_iostatus_disable(BlockDriverState *bs)
507028a7282aSLuiz Capitulino {
5071d6bf279eSLuiz Capitulino     bs->iostatus_enabled = false;
507228a7282aSLuiz Capitulino }
507328a7282aSLuiz Capitulino 
507428a7282aSLuiz Capitulino void bdrv_iostatus_reset(BlockDriverState *bs)
507528a7282aSLuiz Capitulino {
507628a7282aSLuiz Capitulino     if (bdrv_iostatus_is_enabled(bs)) {
507758e21ef5SLuiz Capitulino         bs->iostatus = BLOCK_DEVICE_IO_STATUS_OK;
50783bd293c3SPaolo Bonzini         if (bs->job) {
50793bd293c3SPaolo Bonzini             block_job_iostatus_reset(bs->job);
50803bd293c3SPaolo Bonzini         }
508128a7282aSLuiz Capitulino     }
508228a7282aSLuiz Capitulino }
508328a7282aSLuiz Capitulino 
508428a7282aSLuiz Capitulino void bdrv_iostatus_set_err(BlockDriverState *bs, int error)
508528a7282aSLuiz Capitulino {
50863e1caa5fSPaolo Bonzini     assert(bdrv_iostatus_is_enabled(bs));
50873e1caa5fSPaolo Bonzini     if (bs->iostatus == BLOCK_DEVICE_IO_STATUS_OK) {
508858e21ef5SLuiz Capitulino         bs->iostatus = error == ENOSPC ? BLOCK_DEVICE_IO_STATUS_NOSPACE :
508958e21ef5SLuiz Capitulino                                          BLOCK_DEVICE_IO_STATUS_FAILED;
509028a7282aSLuiz Capitulino     }
509128a7282aSLuiz Capitulino }
509228a7282aSLuiz Capitulino 
5093a597e79cSChristoph Hellwig void
5094a597e79cSChristoph Hellwig bdrv_acct_start(BlockDriverState *bs, BlockAcctCookie *cookie, int64_t bytes,
5095a597e79cSChristoph Hellwig         enum BlockAcctType type)
5096a597e79cSChristoph Hellwig {
5097a597e79cSChristoph Hellwig     assert(type < BDRV_MAX_IOTYPE);
5098a597e79cSChristoph Hellwig 
5099a597e79cSChristoph Hellwig     cookie->bytes = bytes;
5100c488c7f6SChristoph Hellwig     cookie->start_time_ns = get_clock();
5101a597e79cSChristoph Hellwig     cookie->type = type;
5102a597e79cSChristoph Hellwig }
5103a597e79cSChristoph Hellwig 
5104a597e79cSChristoph Hellwig void
5105a597e79cSChristoph Hellwig bdrv_acct_done(BlockDriverState *bs, BlockAcctCookie *cookie)
5106a597e79cSChristoph Hellwig {
5107a597e79cSChristoph Hellwig     assert(cookie->type < BDRV_MAX_IOTYPE);
5108a597e79cSChristoph Hellwig 
5109a597e79cSChristoph Hellwig     bs->nr_bytes[cookie->type] += cookie->bytes;
5110a597e79cSChristoph Hellwig     bs->nr_ops[cookie->type]++;
5111c488c7f6SChristoph Hellwig     bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
5112a597e79cSChristoph Hellwig }
5113a597e79cSChristoph Hellwig 
5114d92ada22SLuiz Capitulino void bdrv_img_create(const char *filename, const char *fmt,
5115f88e1a42SJes Sorensen                      const char *base_filename, const char *base_fmt,
5116f382d43aSMiroslav Rezanina                      char *options, uint64_t img_size, int flags,
5117f382d43aSMiroslav Rezanina                      Error **errp, bool quiet)
5118f88e1a42SJes Sorensen {
5119f88e1a42SJes Sorensen     QEMUOptionParameter *param = NULL, *create_options = NULL;
5120d220894eSKevin Wolf     QEMUOptionParameter *backing_fmt, *backing_file, *size;
5121f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
512296df67d1SStefan Hajnoczi     BlockDriver *backing_drv = NULL;
5123cc84d90fSMax Reitz     Error *local_err = NULL;
5124f88e1a42SJes Sorensen     int ret = 0;
5125f88e1a42SJes Sorensen 
5126f88e1a42SJes Sorensen     /* Find driver and parse its options */
5127f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
5128f88e1a42SJes Sorensen     if (!drv) {
512971c79813SLuiz Capitulino         error_setg(errp, "Unknown file format '%s'", fmt);
5130d92ada22SLuiz Capitulino         return;
5131f88e1a42SJes Sorensen     }
5132f88e1a42SJes Sorensen 
513398289620SKevin Wolf     proto_drv = bdrv_find_protocol(filename, true);
5134f88e1a42SJes Sorensen     if (!proto_drv) {
513571c79813SLuiz Capitulino         error_setg(errp, "Unknown protocol '%s'", filename);
5136d92ada22SLuiz Capitulino         return;
5137f88e1a42SJes Sorensen     }
5138f88e1a42SJes Sorensen 
5139f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
5140f88e1a42SJes Sorensen                                               drv->create_options);
5141f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
5142f88e1a42SJes Sorensen                                               proto_drv->create_options);
5143f88e1a42SJes Sorensen 
5144f88e1a42SJes Sorensen     /* Create parameter list with default values */
5145f88e1a42SJes Sorensen     param = parse_option_parameters("", create_options, param);
5146f88e1a42SJes Sorensen 
5147f88e1a42SJes Sorensen     set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
5148f88e1a42SJes Sorensen 
5149f88e1a42SJes Sorensen     /* Parse -o options */
5150f88e1a42SJes Sorensen     if (options) {
5151f88e1a42SJes Sorensen         param = parse_option_parameters(options, create_options, param);
5152f88e1a42SJes Sorensen         if (param == NULL) {
515371c79813SLuiz Capitulino             error_setg(errp, "Invalid options for file format '%s'.", fmt);
5154f88e1a42SJes Sorensen             goto out;
5155f88e1a42SJes Sorensen         }
5156f88e1a42SJes Sorensen     }
5157f88e1a42SJes Sorensen 
5158f88e1a42SJes Sorensen     if (base_filename) {
5159f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
5160f88e1a42SJes Sorensen                                  base_filename)) {
516171c79813SLuiz Capitulino             error_setg(errp, "Backing file not supported for file format '%s'",
516271c79813SLuiz Capitulino                        fmt);
5163f88e1a42SJes Sorensen             goto out;
5164f88e1a42SJes Sorensen         }
5165f88e1a42SJes Sorensen     }
5166f88e1a42SJes Sorensen 
5167f88e1a42SJes Sorensen     if (base_fmt) {
5168f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
516971c79813SLuiz Capitulino             error_setg(errp, "Backing file format not supported for file "
517071c79813SLuiz Capitulino                              "format '%s'", fmt);
5171f88e1a42SJes Sorensen             goto out;
5172f88e1a42SJes Sorensen         }
5173f88e1a42SJes Sorensen     }
5174f88e1a42SJes Sorensen 
5175792da93aSJes Sorensen     backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
5176792da93aSJes Sorensen     if (backing_file && backing_file->value.s) {
5177792da93aSJes Sorensen         if (!strcmp(filename, backing_file->value.s)) {
517871c79813SLuiz Capitulino             error_setg(errp, "Error: Trying to create an image with the "
517971c79813SLuiz Capitulino                              "same filename as the backing file");
5180792da93aSJes Sorensen             goto out;
5181792da93aSJes Sorensen         }
5182792da93aSJes Sorensen     }
5183792da93aSJes Sorensen 
5184f88e1a42SJes Sorensen     backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
5185f88e1a42SJes Sorensen     if (backing_fmt && backing_fmt->value.s) {
518696df67d1SStefan Hajnoczi         backing_drv = bdrv_find_format(backing_fmt->value.s);
518796df67d1SStefan Hajnoczi         if (!backing_drv) {
518871c79813SLuiz Capitulino             error_setg(errp, "Unknown backing file format '%s'",
518971c79813SLuiz Capitulino                        backing_fmt->value.s);
5190f88e1a42SJes Sorensen             goto out;
5191f88e1a42SJes Sorensen         }
5192f88e1a42SJes Sorensen     }
5193f88e1a42SJes Sorensen 
5194f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
5195f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
5196d220894eSKevin Wolf     size = get_option_parameter(param, BLOCK_OPT_SIZE);
5197d220894eSKevin Wolf     if (size && size->value.n == -1) {
5198f88e1a42SJes Sorensen         if (backing_file && backing_file->value.s) {
519966f6b814SMax Reitz             BlockDriverState *bs;
5200f88e1a42SJes Sorensen             uint64_t size;
5201f88e1a42SJes Sorensen             char buf[32];
520263090dacSPaolo Bonzini             int back_flags;
520363090dacSPaolo Bonzini 
520463090dacSPaolo Bonzini             /* backing files always opened read-only */
520563090dacSPaolo Bonzini             back_flags =
520663090dacSPaolo Bonzini                 flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
5207f88e1a42SJes Sorensen 
5208f88e1a42SJes Sorensen             bs = bdrv_new("");
5209f88e1a42SJes Sorensen 
5210de9c0cecSKevin Wolf             ret = bdrv_open(bs, backing_file->value.s, NULL, back_flags,
5211cc84d90fSMax Reitz                             backing_drv, &local_err);
5212f88e1a42SJes Sorensen             if (ret < 0) {
5213cc84d90fSMax Reitz                 error_setg_errno(errp, -ret, "Could not open '%s': %s",
5214cc84d90fSMax Reitz                                  backing_file->value.s,
5215cc84d90fSMax Reitz                                  error_get_pretty(local_err));
5216cc84d90fSMax Reitz                 error_free(local_err);
5217cc84d90fSMax Reitz                 local_err = NULL;
521866f6b814SMax Reitz                 bdrv_unref(bs);
5219f88e1a42SJes Sorensen                 goto out;
5220f88e1a42SJes Sorensen             }
5221f88e1a42SJes Sorensen             bdrv_get_geometry(bs, &size);
5222f88e1a42SJes Sorensen             size *= 512;
5223f88e1a42SJes Sorensen 
5224f88e1a42SJes Sorensen             snprintf(buf, sizeof(buf), "%" PRId64, size);
5225f88e1a42SJes Sorensen             set_option_parameter(param, BLOCK_OPT_SIZE, buf);
522666f6b814SMax Reitz 
522766f6b814SMax Reitz             bdrv_unref(bs);
5228f88e1a42SJes Sorensen         } else {
522971c79813SLuiz Capitulino             error_setg(errp, "Image creation needs a size parameter");
5230f88e1a42SJes Sorensen             goto out;
5231f88e1a42SJes Sorensen         }
5232f88e1a42SJes Sorensen     }
5233f88e1a42SJes Sorensen 
5234f382d43aSMiroslav Rezanina     if (!quiet) {
5235f88e1a42SJes Sorensen         printf("Formatting '%s', fmt=%s ", filename, fmt);
5236f88e1a42SJes Sorensen         print_option_parameters(param);
5237f88e1a42SJes Sorensen         puts("");
5238f382d43aSMiroslav Rezanina     }
5239cc84d90fSMax Reitz     ret = bdrv_create(drv, filename, param, &local_err);
5240cc84d90fSMax Reitz     if (ret == -EFBIG) {
5241cc84d90fSMax Reitz         /* This is generally a better message than whatever the driver would
5242cc84d90fSMax Reitz          * deliver (especially because of the cluster_size_hint), since that
5243cc84d90fSMax Reitz          * is most probably not much different from "image too large". */
5244f3f4d2c0SKevin Wolf         const char *cluster_size_hint = "";
5245f3f4d2c0SKevin Wolf         if (get_option_parameter(create_options, BLOCK_OPT_CLUSTER_SIZE)) {
5246f3f4d2c0SKevin Wolf             cluster_size_hint = " (try using a larger cluster size)";
5247f3f4d2c0SKevin Wolf         }
5248cc84d90fSMax Reitz         error_setg(errp, "The image size is too large for file format '%s'"
5249cc84d90fSMax Reitz                    "%s", fmt, cluster_size_hint);
5250cc84d90fSMax Reitz         error_free(local_err);
5251cc84d90fSMax Reitz         local_err = NULL;
5252f88e1a42SJes Sorensen     }
5253f88e1a42SJes Sorensen 
5254f88e1a42SJes Sorensen out:
5255f88e1a42SJes Sorensen     free_option_parameters(create_options);
5256f88e1a42SJes Sorensen     free_option_parameters(param);
5257f88e1a42SJes Sorensen 
5258cc84d90fSMax Reitz     if (error_is_set(&local_err)) {
5259cc84d90fSMax Reitz         error_propagate(errp, local_err);
5260cc84d90fSMax Reitz     }
5261f88e1a42SJes Sorensen }
526285d126f3SStefan Hajnoczi 
526385d126f3SStefan Hajnoczi AioContext *bdrv_get_aio_context(BlockDriverState *bs)
526485d126f3SStefan Hajnoczi {
526585d126f3SStefan Hajnoczi     /* Currently BlockDriverState always uses the main loop AioContext */
526685d126f3SStefan Hajnoczi     return qemu_get_aio_context();
526785d126f3SStefan Hajnoczi }
5268d616b224SStefan Hajnoczi 
5269d616b224SStefan Hajnoczi void bdrv_add_before_write_notifier(BlockDriverState *bs,
5270d616b224SStefan Hajnoczi                                     NotifierWithReturn *notifier)
5271d616b224SStefan Hajnoczi {
5272d616b224SStefan Hajnoczi     notifier_with_return_list_add(&bs->before_write_notifiers, notifier);
5273d616b224SStefan Hajnoczi }
52746f176b48SMax Reitz 
52756f176b48SMax Reitz int bdrv_amend_options(BlockDriverState *bs, QEMUOptionParameter *options)
52766f176b48SMax Reitz {
52776f176b48SMax Reitz     if (bs->drv->bdrv_amend_options == NULL) {
52786f176b48SMax Reitz         return -ENOTSUP;
52796f176b48SMax Reitz     }
52806f176b48SMax Reitz     return bs->drv->bdrv_amend_options(bs, options);
52816f176b48SMax Reitz }
5282f6186f49SBenoît Canet 
5283212a5a8fSBenoît Canet /* Used to recurse on single child block filters.
5284212a5a8fSBenoît Canet  * Single child block filter will store their child in bs->file.
5285212a5a8fSBenoît Canet  */
5286212a5a8fSBenoît Canet bool bdrv_generic_is_first_non_filter(BlockDriverState *bs,
5287212a5a8fSBenoît Canet                                       BlockDriverState *candidate)
5288f6186f49SBenoît Canet {
5289212a5a8fSBenoît Canet     if (!bs->drv) {
5290212a5a8fSBenoît Canet         return false;
5291f6186f49SBenoît Canet     }
5292f6186f49SBenoît Canet 
5293212a5a8fSBenoît Canet     if (!bs->drv->authorizations[BS_IS_A_FILTER]) {
5294212a5a8fSBenoît Canet         if (bs == candidate) {
5295212a5a8fSBenoît Canet             return true;
5296212a5a8fSBenoît Canet         } else {
5297212a5a8fSBenoît Canet             return false;
5298212a5a8fSBenoît Canet         }
5299f6186f49SBenoît Canet     }
5300f6186f49SBenoît Canet 
5301212a5a8fSBenoît Canet     if (!bs->drv->authorizations[BS_FILTER_PASS_DOWN]) {
5302212a5a8fSBenoît Canet         return false;
5303f6186f49SBenoît Canet     }
5304f6186f49SBenoît Canet 
5305212a5a8fSBenoît Canet     if (!bs->file) {
5306212a5a8fSBenoît Canet         return false;
5307212a5a8fSBenoît Canet     }
5308212a5a8fSBenoît Canet 
5309212a5a8fSBenoît Canet     return bdrv_recurse_is_first_non_filter(bs->file, candidate);
5310212a5a8fSBenoît Canet }
5311212a5a8fSBenoît Canet 
5312212a5a8fSBenoît Canet bool bdrv_recurse_is_first_non_filter(BlockDriverState *bs,
5313212a5a8fSBenoît Canet                                       BlockDriverState *candidate)
5314f6186f49SBenoît Canet {
5315212a5a8fSBenoît Canet     if (bs->drv && bs->drv->bdrv_recurse_is_first_non_filter) {
5316212a5a8fSBenoît Canet         return bs->drv->bdrv_recurse_is_first_non_filter(bs, candidate);
5317212a5a8fSBenoît Canet     }
5318212a5a8fSBenoît Canet 
5319212a5a8fSBenoît Canet     return bdrv_generic_is_first_non_filter(bs, candidate);
5320212a5a8fSBenoît Canet }
5321212a5a8fSBenoît Canet 
5322212a5a8fSBenoît Canet /* This function checks if the candidate is the first non filter bs down it's
5323212a5a8fSBenoît Canet  * bs chain. Since we don't have pointers to parents it explore all bs chains
5324212a5a8fSBenoît Canet  * from the top. Some filters can choose not to pass down the recursion.
5325212a5a8fSBenoît Canet  */
5326212a5a8fSBenoît Canet bool bdrv_is_first_non_filter(BlockDriverState *candidate)
5327212a5a8fSBenoît Canet {
5328212a5a8fSBenoît Canet     BlockDriverState *bs;
5329212a5a8fSBenoît Canet 
5330212a5a8fSBenoît Canet     /* walk down the bs forest recursively */
5331212a5a8fSBenoît Canet     QTAILQ_FOREACH(bs, &bdrv_states, device_list) {
5332212a5a8fSBenoît Canet         bool perm;
5333212a5a8fSBenoît Canet 
5334212a5a8fSBenoît Canet         if (!bs->file) {
5335212a5a8fSBenoît Canet             continue;
5336212a5a8fSBenoît Canet         }
5337212a5a8fSBenoît Canet 
5338212a5a8fSBenoît Canet         perm = bdrv_recurse_is_first_non_filter(bs->file, candidate);
5339212a5a8fSBenoît Canet 
5340212a5a8fSBenoît Canet         /* candidate is the first non filter */
5341212a5a8fSBenoît Canet         if (perm) {
5342212a5a8fSBenoît Canet             return true;
5343212a5a8fSBenoît Canet         }
5344212a5a8fSBenoît Canet     }
5345212a5a8fSBenoît Canet 
5346212a5a8fSBenoît Canet     return false;
5347f6186f49SBenoît Canet }
5348