xref: /openbmc/qemu/block.c (revision c488c7f649106d09df76f697adccbe6e72520b26)
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"
27376253ecSaliguori #include "monitor.h"
28ea2384d3Sbellard #include "block_int.h"
295efa9d5aSAnthony Liguori #include "module.h"
30d15e5465SLuiz Capitulino #include "qemu-objects.h"
3168485420SKevin Wolf #include "qemu-coroutine.h"
32fc01f7e7Sbellard 
3371e72a19SJuan Quintela #ifdef CONFIG_BSD
347674e7bfSbellard #include <sys/types.h>
357674e7bfSbellard #include <sys/stat.h>
367674e7bfSbellard #include <sys/ioctl.h>
3772cf2d4fSBlue Swirl #include <sys/queue.h>
38c5e97233Sblueswir1 #ifndef __DragonFly__
397674e7bfSbellard #include <sys/disk.h>
407674e7bfSbellard #endif
41c5e97233Sblueswir1 #endif
427674e7bfSbellard 
4349dc768dSaliguori #ifdef _WIN32
4449dc768dSaliguori #include <windows.h>
4549dc768dSaliguori #endif
4649dc768dSaliguori 
47f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
48f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
49c87c0672Saliguori         BlockDriverCompletionFunc *cb, void *opaque);
50f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
51f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
52ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque);
53b2e12bc6SChristoph Hellwig static BlockDriverAIOCB *bdrv_aio_flush_em(BlockDriverState *bs,
54b2e12bc6SChristoph Hellwig         BlockDriverCompletionFunc *cb, void *opaque);
55016f5cf6SAlexander Graf static BlockDriverAIOCB *bdrv_aio_noop_em(BlockDriverState *bs,
56016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque);
5783f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
5883f64091Sbellard                         uint8_t *buf, int nb_sectors);
5983f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
6083f64091Sbellard                          const uint8_t *buf, int nb_sectors);
6168485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_readv_em(BlockDriverState *bs,
6268485420SKevin Wolf         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
6368485420SKevin Wolf         BlockDriverCompletionFunc *cb, void *opaque);
6468485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_writev_em(BlockDriverState *bs,
6568485420SKevin Wolf         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
6668485420SKevin Wolf         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);
73e7a8a783SKevin Wolf static int coroutine_fn bdrv_co_flush_em(BlockDriverState *bs);
74ec530c81Sbellard 
751b7bdbc1SStefan Hajnoczi static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
761b7bdbc1SStefan Hajnoczi     QTAILQ_HEAD_INITIALIZER(bdrv_states);
777ee930d0Sblueswir1 
788a22f02aSStefan Hajnoczi static QLIST_HEAD(, BlockDriver) bdrv_drivers =
798a22f02aSStefan Hajnoczi     QLIST_HEAD_INITIALIZER(bdrv_drivers);
80ea2384d3Sbellard 
81f9092b10SMarkus Armbruster /* The device to use for VM snapshots */
82f9092b10SMarkus Armbruster static BlockDriverState *bs_snapshots;
83f9092b10SMarkus Armbruster 
84eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
85eb852011SMarkus Armbruster static int use_bdrv_whitelist;
86eb852011SMarkus Armbruster 
879e0b22f4SStefan Hajnoczi #ifdef _WIN32
889e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
899e0b22f4SStefan Hajnoczi {
909e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
919e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
929e0b22f4SStefan Hajnoczi             filename[1] == ':');
939e0b22f4SStefan Hajnoczi }
949e0b22f4SStefan Hajnoczi 
959e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
969e0b22f4SStefan Hajnoczi {
979e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
989e0b22f4SStefan Hajnoczi         filename[2] == '\0')
999e0b22f4SStefan Hajnoczi         return 1;
1009e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
1019e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
1029e0b22f4SStefan Hajnoczi         return 1;
1039e0b22f4SStefan Hajnoczi     return 0;
1049e0b22f4SStefan Hajnoczi }
1059e0b22f4SStefan Hajnoczi #endif
1069e0b22f4SStefan Hajnoczi 
1079e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
1089e0b22f4SStefan Hajnoczi static int path_has_protocol(const char *path)
1099e0b22f4SStefan Hajnoczi {
1109e0b22f4SStefan Hajnoczi #ifdef _WIN32
1119e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
1129e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
1139e0b22f4SStefan Hajnoczi         return 0;
1149e0b22f4SStefan Hajnoczi     }
1159e0b22f4SStefan Hajnoczi #endif
1169e0b22f4SStefan Hajnoczi 
1179e0b22f4SStefan Hajnoczi     return strchr(path, ':') != NULL;
1189e0b22f4SStefan Hajnoczi }
1199e0b22f4SStefan Hajnoczi 
12083f64091Sbellard int path_is_absolute(const char *path)
12183f64091Sbellard {
12283f64091Sbellard     const char *p;
12321664424Sbellard #ifdef _WIN32
12421664424Sbellard     /* specific case for names like: "\\.\d:" */
12521664424Sbellard     if (*path == '/' || *path == '\\')
12621664424Sbellard         return 1;
12721664424Sbellard #endif
12883f64091Sbellard     p = strchr(path, ':');
12983f64091Sbellard     if (p)
13083f64091Sbellard         p++;
13183f64091Sbellard     else
13283f64091Sbellard         p = path;
1333b9f94e1Sbellard #ifdef _WIN32
1343b9f94e1Sbellard     return (*p == '/' || *p == '\\');
1353b9f94e1Sbellard #else
1363b9f94e1Sbellard     return (*p == '/');
1373b9f94e1Sbellard #endif
13883f64091Sbellard }
13983f64091Sbellard 
14083f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
14183f64091Sbellard    path to it by considering it is relative to base_path. URL are
14283f64091Sbellard    supported. */
14383f64091Sbellard void path_combine(char *dest, int dest_size,
14483f64091Sbellard                   const char *base_path,
14583f64091Sbellard                   const char *filename)
14683f64091Sbellard {
14783f64091Sbellard     const char *p, *p1;
14883f64091Sbellard     int len;
14983f64091Sbellard 
15083f64091Sbellard     if (dest_size <= 0)
15183f64091Sbellard         return;
15283f64091Sbellard     if (path_is_absolute(filename)) {
15383f64091Sbellard         pstrcpy(dest, dest_size, filename);
15483f64091Sbellard     } else {
15583f64091Sbellard         p = strchr(base_path, ':');
15683f64091Sbellard         if (p)
15783f64091Sbellard             p++;
15883f64091Sbellard         else
15983f64091Sbellard             p = base_path;
1603b9f94e1Sbellard         p1 = strrchr(base_path, '/');
1613b9f94e1Sbellard #ifdef _WIN32
1623b9f94e1Sbellard         {
1633b9f94e1Sbellard             const char *p2;
1643b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
1653b9f94e1Sbellard             if (!p1 || p2 > p1)
1663b9f94e1Sbellard                 p1 = p2;
1673b9f94e1Sbellard         }
1683b9f94e1Sbellard #endif
16983f64091Sbellard         if (p1)
17083f64091Sbellard             p1++;
17183f64091Sbellard         else
17283f64091Sbellard             p1 = base_path;
17383f64091Sbellard         if (p1 > p)
17483f64091Sbellard             p = p1;
17583f64091Sbellard         len = p - base_path;
17683f64091Sbellard         if (len > dest_size - 1)
17783f64091Sbellard             len = dest_size - 1;
17883f64091Sbellard         memcpy(dest, base_path, len);
17983f64091Sbellard         dest[len] = '\0';
18083f64091Sbellard         pstrcat(dest, dest_size, filename);
18183f64091Sbellard     }
18283f64091Sbellard }
18383f64091Sbellard 
1845efa9d5aSAnthony Liguori void bdrv_register(BlockDriver *bdrv)
185ea2384d3Sbellard {
18668485420SKevin Wolf     if (bdrv->bdrv_co_readv) {
18768485420SKevin Wolf         /* Emulate AIO by coroutines, and sync by AIO */
18868485420SKevin Wolf         bdrv->bdrv_aio_readv = bdrv_co_aio_readv_em;
18968485420SKevin Wolf         bdrv->bdrv_aio_writev = bdrv_co_aio_writev_em;
19068485420SKevin Wolf         bdrv->bdrv_read = bdrv_read_em;
19168485420SKevin Wolf         bdrv->bdrv_write = bdrv_write_em;
192f9f05dc5SKevin Wolf      } else {
193f9f05dc5SKevin Wolf         bdrv->bdrv_co_readv = bdrv_co_readv_em;
194f9f05dc5SKevin Wolf         bdrv->bdrv_co_writev = bdrv_co_writev_em;
195f9f05dc5SKevin Wolf 
196f9f05dc5SKevin Wolf         if (!bdrv->bdrv_aio_readv) {
19783f64091Sbellard             /* add AIO emulation layer */
198f141eafeSaliguori             bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
199f141eafeSaliguori             bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
200eda578e5Saliguori         } else if (!bdrv->bdrv_read) {
20183f64091Sbellard             /* add synchronous IO emulation layer */
20283f64091Sbellard             bdrv->bdrv_read = bdrv_read_em;
20383f64091Sbellard             bdrv->bdrv_write = bdrv_write_em;
20483f64091Sbellard         }
205f9f05dc5SKevin Wolf     }
206b2e12bc6SChristoph Hellwig 
207b2e12bc6SChristoph Hellwig     if (!bdrv->bdrv_aio_flush)
208b2e12bc6SChristoph Hellwig         bdrv->bdrv_aio_flush = bdrv_aio_flush_em;
209b2e12bc6SChristoph Hellwig 
2108a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
211ea2384d3Sbellard }
212b338082bSbellard 
213b338082bSbellard /* create a new block device (by default it is empty) */
214b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
215fc01f7e7Sbellard {
2161b7bdbc1SStefan Hajnoczi     BlockDriverState *bs;
217b338082bSbellard 
2187267c094SAnthony Liguori     bs = g_malloc0(sizeof(BlockDriverState));
219b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
220ea2384d3Sbellard     if (device_name[0] != '\0') {
2211b7bdbc1SStefan Hajnoczi         QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
222ea2384d3Sbellard     }
223b338082bSbellard     return bs;
224b338082bSbellard }
225b338082bSbellard 
226ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
227ea2384d3Sbellard {
228ea2384d3Sbellard     BlockDriver *drv1;
2298a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
2308a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
231ea2384d3Sbellard             return drv1;
232ea2384d3Sbellard         }
2338a22f02aSStefan Hajnoczi     }
234ea2384d3Sbellard     return NULL;
235ea2384d3Sbellard }
236ea2384d3Sbellard 
237eb852011SMarkus Armbruster static int bdrv_is_whitelisted(BlockDriver *drv)
238eb852011SMarkus Armbruster {
239eb852011SMarkus Armbruster     static const char *whitelist[] = {
240eb852011SMarkus Armbruster         CONFIG_BDRV_WHITELIST
241eb852011SMarkus Armbruster     };
242eb852011SMarkus Armbruster     const char **p;
243eb852011SMarkus Armbruster 
244eb852011SMarkus Armbruster     if (!whitelist[0])
245eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
246eb852011SMarkus Armbruster 
247eb852011SMarkus Armbruster     for (p = whitelist; *p; p++) {
248eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
249eb852011SMarkus Armbruster             return 1;
250eb852011SMarkus Armbruster         }
251eb852011SMarkus Armbruster     }
252eb852011SMarkus Armbruster     return 0;
253eb852011SMarkus Armbruster }
254eb852011SMarkus Armbruster 
255eb852011SMarkus Armbruster BlockDriver *bdrv_find_whitelisted_format(const char *format_name)
256eb852011SMarkus Armbruster {
257eb852011SMarkus Armbruster     BlockDriver *drv = bdrv_find_format(format_name);
258eb852011SMarkus Armbruster     return drv && bdrv_is_whitelisted(drv) ? drv : NULL;
259eb852011SMarkus Armbruster }
260eb852011SMarkus Armbruster 
2610e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
2620e7e1989SKevin Wolf     QEMUOptionParameter *options)
263ea2384d3Sbellard {
264ea2384d3Sbellard     if (!drv->bdrv_create)
265ea2384d3Sbellard         return -ENOTSUP;
2660e7e1989SKevin Wolf 
2670e7e1989SKevin Wolf     return drv->bdrv_create(filename, options);
268ea2384d3Sbellard }
269ea2384d3Sbellard 
27084a12e66SChristoph Hellwig int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
27184a12e66SChristoph Hellwig {
27284a12e66SChristoph Hellwig     BlockDriver *drv;
27384a12e66SChristoph Hellwig 
274b50cbabcSMORITA Kazutaka     drv = bdrv_find_protocol(filename);
27584a12e66SChristoph Hellwig     if (drv == NULL) {
27616905d71SStefan Hajnoczi         return -ENOENT;
27784a12e66SChristoph Hellwig     }
27884a12e66SChristoph Hellwig 
27984a12e66SChristoph Hellwig     return bdrv_create(drv, filename, options);
28084a12e66SChristoph Hellwig }
28184a12e66SChristoph Hellwig 
282d5249393Sbellard #ifdef _WIN32
28395389c86Sbellard void get_tmp_filename(char *filename, int size)
284d5249393Sbellard {
2853b9f94e1Sbellard     char temp_dir[MAX_PATH];
2863b9f94e1Sbellard 
2873b9f94e1Sbellard     GetTempPath(MAX_PATH, temp_dir);
2883b9f94e1Sbellard     GetTempFileName(temp_dir, "qem", 0, filename);
289d5249393Sbellard }
290d5249393Sbellard #else
29195389c86Sbellard void get_tmp_filename(char *filename, int size)
292ea2384d3Sbellard {
293ea2384d3Sbellard     int fd;
2947ccfb2ebSblueswir1     const char *tmpdir;
295d5249393Sbellard     /* XXX: race condition possible */
2960badc1eeSaurel32     tmpdir = getenv("TMPDIR");
2970badc1eeSaurel32     if (!tmpdir)
2980badc1eeSaurel32         tmpdir = "/tmp";
2990badc1eeSaurel32     snprintf(filename, size, "%s/vl.XXXXXX", tmpdir);
300ea2384d3Sbellard     fd = mkstemp(filename);
301ea2384d3Sbellard     close(fd);
302ea2384d3Sbellard }
303d5249393Sbellard #endif
304ea2384d3Sbellard 
305f3a5d3f8SChristoph Hellwig /*
306f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
307f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
308f3a5d3f8SChristoph Hellwig  */
309f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
310f3a5d3f8SChristoph Hellwig {
311508c7cb3SChristoph Hellwig     int score_max = 0, score;
312508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
313f3a5d3f8SChristoph Hellwig 
3148a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
315508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
316508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
317508c7cb3SChristoph Hellwig             if (score > score_max) {
318508c7cb3SChristoph Hellwig                 score_max = score;
319508c7cb3SChristoph Hellwig                 drv = d;
320f3a5d3f8SChristoph Hellwig             }
321508c7cb3SChristoph Hellwig         }
322f3a5d3f8SChristoph Hellwig     }
323f3a5d3f8SChristoph Hellwig 
324508c7cb3SChristoph Hellwig     return drv;
325f3a5d3f8SChristoph Hellwig }
326f3a5d3f8SChristoph Hellwig 
327b50cbabcSMORITA Kazutaka BlockDriver *bdrv_find_protocol(const char *filename)
32884a12e66SChristoph Hellwig {
32984a12e66SChristoph Hellwig     BlockDriver *drv1;
33084a12e66SChristoph Hellwig     char protocol[128];
33184a12e66SChristoph Hellwig     int len;
33284a12e66SChristoph Hellwig     const char *p;
33384a12e66SChristoph Hellwig 
33466f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
33566f82ceeSKevin Wolf 
33639508e7aSChristoph Hellwig     /*
33739508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
33839508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
33939508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
34039508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
34139508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
34239508e7aSChristoph Hellwig      */
34384a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
34439508e7aSChristoph Hellwig     if (drv1) {
34584a12e66SChristoph Hellwig         return drv1;
34684a12e66SChristoph Hellwig     }
34739508e7aSChristoph Hellwig 
3489e0b22f4SStefan Hajnoczi     if (!path_has_protocol(filename)) {
34939508e7aSChristoph Hellwig         return bdrv_find_format("file");
35039508e7aSChristoph Hellwig     }
3519e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
3529e0b22f4SStefan Hajnoczi     assert(p != NULL);
35384a12e66SChristoph Hellwig     len = p - filename;
35484a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
35584a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
35684a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
35784a12e66SChristoph Hellwig     protocol[len] = '\0';
35884a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
35984a12e66SChristoph Hellwig         if (drv1->protocol_name &&
36084a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
36184a12e66SChristoph Hellwig             return drv1;
36284a12e66SChristoph Hellwig         }
36384a12e66SChristoph Hellwig     }
36484a12e66SChristoph Hellwig     return NULL;
36584a12e66SChristoph Hellwig }
36684a12e66SChristoph Hellwig 
367c98ac35dSStefan Weil static int find_image_format(const char *filename, BlockDriver **pdrv)
368ea2384d3Sbellard {
36983f64091Sbellard     int ret, score, score_max;
370ea2384d3Sbellard     BlockDriver *drv1, *drv;
37183f64091Sbellard     uint8_t buf[2048];
37283f64091Sbellard     BlockDriverState *bs;
373ea2384d3Sbellard 
374f5edb014SNaphtali Sprei     ret = bdrv_file_open(&bs, filename, 0);
375c98ac35dSStefan Weil     if (ret < 0) {
376c98ac35dSStefan Weil         *pdrv = NULL;
377c98ac35dSStefan Weil         return ret;
378c98ac35dSStefan Weil     }
379f8ea0b00SNicholas Bellinger 
38008a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
38108a00559SKevin Wolf     if (bs->sg || !bdrv_is_inserted(bs)) {
3821a396859SNicholas A. Bellinger         bdrv_delete(bs);
383c98ac35dSStefan Weil         drv = bdrv_find_format("raw");
384c98ac35dSStefan Weil         if (!drv) {
385c98ac35dSStefan Weil             ret = -ENOENT;
386c98ac35dSStefan Weil         }
387c98ac35dSStefan Weil         *pdrv = drv;
388c98ac35dSStefan Weil         return ret;
3891a396859SNicholas A. Bellinger     }
390f8ea0b00SNicholas Bellinger 
39183f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
39283f64091Sbellard     bdrv_delete(bs);
393ea2384d3Sbellard     if (ret < 0) {
394c98ac35dSStefan Weil         *pdrv = NULL;
395c98ac35dSStefan Weil         return ret;
396ea2384d3Sbellard     }
397ea2384d3Sbellard 
398ea2384d3Sbellard     score_max = 0;
39984a12e66SChristoph Hellwig     drv = NULL;
4008a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
40183f64091Sbellard         if (drv1->bdrv_probe) {
402ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
403ea2384d3Sbellard             if (score > score_max) {
404ea2384d3Sbellard                 score_max = score;
405ea2384d3Sbellard                 drv = drv1;
406ea2384d3Sbellard             }
407ea2384d3Sbellard         }
40883f64091Sbellard     }
409c98ac35dSStefan Weil     if (!drv) {
410c98ac35dSStefan Weil         ret = -ENOENT;
411c98ac35dSStefan Weil     }
412c98ac35dSStefan Weil     *pdrv = drv;
413c98ac35dSStefan Weil     return ret;
414ea2384d3Sbellard }
415ea2384d3Sbellard 
41651762288SStefan Hajnoczi /**
41751762288SStefan Hajnoczi  * Set the current 'total_sectors' value
41851762288SStefan Hajnoczi  */
41951762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
42051762288SStefan Hajnoczi {
42151762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
42251762288SStefan Hajnoczi 
423396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
424396759adSNicholas Bellinger     if (bs->sg)
425396759adSNicholas Bellinger         return 0;
426396759adSNicholas Bellinger 
42751762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
42851762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
42951762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
43051762288SStefan Hajnoczi         if (length < 0) {
43151762288SStefan Hajnoczi             return length;
43251762288SStefan Hajnoczi         }
43351762288SStefan Hajnoczi         hint = length >> BDRV_SECTOR_BITS;
43451762288SStefan Hajnoczi     }
43551762288SStefan Hajnoczi 
43651762288SStefan Hajnoczi     bs->total_sectors = hint;
43751762288SStefan Hajnoczi     return 0;
43851762288SStefan Hajnoczi }
43951762288SStefan Hajnoczi 
440c3993cdcSStefan Hajnoczi /**
441c3993cdcSStefan Hajnoczi  * Set open flags for a given cache mode
442c3993cdcSStefan Hajnoczi  *
443c3993cdcSStefan Hajnoczi  * Return 0 on success, -1 if the cache mode was invalid.
444c3993cdcSStefan Hajnoczi  */
445c3993cdcSStefan Hajnoczi int bdrv_parse_cache_flags(const char *mode, int *flags)
446c3993cdcSStefan Hajnoczi {
447c3993cdcSStefan Hajnoczi     *flags &= ~BDRV_O_CACHE_MASK;
448c3993cdcSStefan Hajnoczi 
449c3993cdcSStefan Hajnoczi     if (!strcmp(mode, "off") || !strcmp(mode, "none")) {
450c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
45192196b2fSStefan Hajnoczi     } else if (!strcmp(mode, "directsync")) {
45292196b2fSStefan Hajnoczi         *flags |= BDRV_O_NOCACHE;
453c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writeback")) {
454c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
455c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "unsafe")) {
456c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_CACHE_WB;
457c3993cdcSStefan Hajnoczi         *flags |= BDRV_O_NO_FLUSH;
458c3993cdcSStefan Hajnoczi     } else if (!strcmp(mode, "writethrough")) {
459c3993cdcSStefan Hajnoczi         /* this is the default */
460c3993cdcSStefan Hajnoczi     } else {
461c3993cdcSStefan Hajnoczi         return -1;
462c3993cdcSStefan Hajnoczi     }
463c3993cdcSStefan Hajnoczi 
464c3993cdcSStefan Hajnoczi     return 0;
465c3993cdcSStefan Hajnoczi }
466c3993cdcSStefan Hajnoczi 
467b6ce07aaSKevin Wolf /*
46857915332SKevin Wolf  * Common part for opening disk images and files
46957915332SKevin Wolf  */
47057915332SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, const char *filename,
47157915332SKevin Wolf     int flags, BlockDriver *drv)
47257915332SKevin Wolf {
47357915332SKevin Wolf     int ret, open_flags;
47457915332SKevin Wolf 
47557915332SKevin Wolf     assert(drv != NULL);
47657915332SKevin Wolf 
47766f82ceeSKevin Wolf     bs->file = NULL;
47851762288SStefan Hajnoczi     bs->total_sectors = 0;
47957915332SKevin Wolf     bs->encrypted = 0;
48057915332SKevin Wolf     bs->valid_key = 0;
48157915332SKevin Wolf     bs->open_flags = flags;
48257915332SKevin Wolf     /* buffer_alignment defaulted to 512, drivers can change this value */
48357915332SKevin Wolf     bs->buffer_alignment = 512;
48457915332SKevin Wolf 
48557915332SKevin Wolf     pstrcpy(bs->filename, sizeof(bs->filename), filename);
48657915332SKevin Wolf 
48757915332SKevin Wolf     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv)) {
48857915332SKevin Wolf         return -ENOTSUP;
48957915332SKevin Wolf     }
49057915332SKevin Wolf 
49157915332SKevin Wolf     bs->drv = drv;
4927267c094SAnthony Liguori     bs->opaque = g_malloc0(drv->instance_size);
49357915332SKevin Wolf 
494a6599793SChristoph Hellwig     if (flags & BDRV_O_CACHE_WB)
49557915332SKevin Wolf         bs->enable_write_cache = 1;
49657915332SKevin Wolf 
49757915332SKevin Wolf     /*
49857915332SKevin Wolf      * Clear flags that are internal to the block layer before opening the
49957915332SKevin Wolf      * image.
50057915332SKevin Wolf      */
50157915332SKevin Wolf     open_flags = flags & ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
50257915332SKevin Wolf 
50357915332SKevin Wolf     /*
504ebabb67aSStefan Weil      * Snapshots should be writable.
50557915332SKevin Wolf      */
50657915332SKevin Wolf     if (bs->is_temporary) {
50757915332SKevin Wolf         open_flags |= BDRV_O_RDWR;
50857915332SKevin Wolf     }
50957915332SKevin Wolf 
51066f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
51166f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
51266f82ceeSKevin Wolf         ret = drv->bdrv_file_open(bs, filename, open_flags);
51366f82ceeSKevin Wolf     } else {
51466f82ceeSKevin Wolf         ret = bdrv_file_open(&bs->file, filename, open_flags);
51566f82ceeSKevin Wolf         if (ret >= 0) {
51666f82ceeSKevin Wolf             ret = drv->bdrv_open(bs, open_flags);
51766f82ceeSKevin Wolf         }
51866f82ceeSKevin Wolf     }
51966f82ceeSKevin Wolf 
52057915332SKevin Wolf     if (ret < 0) {
52157915332SKevin Wolf         goto free_and_fail;
52257915332SKevin Wolf     }
52357915332SKevin Wolf 
52457915332SKevin Wolf     bs->keep_read_only = bs->read_only = !(open_flags & BDRV_O_RDWR);
52551762288SStefan Hajnoczi 
52651762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
52751762288SStefan Hajnoczi     if (ret < 0) {
52851762288SStefan Hajnoczi         goto free_and_fail;
52957915332SKevin Wolf     }
53051762288SStefan Hajnoczi 
53157915332SKevin Wolf #ifndef _WIN32
53257915332SKevin Wolf     if (bs->is_temporary) {
53357915332SKevin Wolf         unlink(filename);
53457915332SKevin Wolf     }
53557915332SKevin Wolf #endif
53657915332SKevin Wolf     return 0;
53757915332SKevin Wolf 
53857915332SKevin Wolf free_and_fail:
53966f82ceeSKevin Wolf     if (bs->file) {
54066f82ceeSKevin Wolf         bdrv_delete(bs->file);
54166f82ceeSKevin Wolf         bs->file = NULL;
54266f82ceeSKevin Wolf     }
5437267c094SAnthony Liguori     g_free(bs->opaque);
54457915332SKevin Wolf     bs->opaque = NULL;
54557915332SKevin Wolf     bs->drv = NULL;
54657915332SKevin Wolf     return ret;
54757915332SKevin Wolf }
54857915332SKevin Wolf 
54957915332SKevin Wolf /*
550b6ce07aaSKevin Wolf  * Opens a file using a protocol (file, host_device, nbd, ...)
551b6ce07aaSKevin Wolf  */
55283f64091Sbellard int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
553b338082bSbellard {
55483f64091Sbellard     BlockDriverState *bs;
5556db95603SChristoph Hellwig     BlockDriver *drv;
55683f64091Sbellard     int ret;
5573b0d4f61Sbellard 
558b50cbabcSMORITA Kazutaka     drv = bdrv_find_protocol(filename);
5596db95603SChristoph Hellwig     if (!drv) {
5606db95603SChristoph Hellwig         return -ENOENT;
5616db95603SChristoph Hellwig     }
5626db95603SChristoph Hellwig 
56383f64091Sbellard     bs = bdrv_new("");
564b6ce07aaSKevin Wolf     ret = bdrv_open_common(bs, filename, flags, drv);
56583f64091Sbellard     if (ret < 0) {
56683f64091Sbellard         bdrv_delete(bs);
56783f64091Sbellard         return ret;
5683b0d4f61Sbellard     }
56971d0770cSaliguori     bs->growable = 1;
57083f64091Sbellard     *pbs = bs;
57183f64091Sbellard     return 0;
5723b0d4f61Sbellard }
5733b0d4f61Sbellard 
574b6ce07aaSKevin Wolf /*
575b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
576b6ce07aaSKevin Wolf  */
577d6e9098eSKevin Wolf int bdrv_open(BlockDriverState *bs, const char *filename, int flags,
578ea2384d3Sbellard               BlockDriver *drv)
579ea2384d3Sbellard {
580b6ce07aaSKevin Wolf     int ret;
58133e3963eSbellard 
58283f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
583ea2384d3Sbellard         BlockDriverState *bs1;
584ea2384d3Sbellard         int64_t total_size;
5857c96d46eSaliguori         int is_protocol = 0;
58691a073a9SKevin Wolf         BlockDriver *bdrv_qcow2;
58791a073a9SKevin Wolf         QEMUOptionParameter *options;
588b6ce07aaSKevin Wolf         char tmp_filename[PATH_MAX];
589b6ce07aaSKevin Wolf         char backing_filename[PATH_MAX];
59033e3963eSbellard 
591ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
592ea2384d3Sbellard            instead of opening 'filename' directly */
593ea2384d3Sbellard 
594ea2384d3Sbellard         /* if there is a backing file, use it */
595ea2384d3Sbellard         bs1 = bdrv_new("");
596d6e9098eSKevin Wolf         ret = bdrv_open(bs1, filename, 0, drv);
59751d7c00cSaliguori         if (ret < 0) {
598ea2384d3Sbellard             bdrv_delete(bs1);
59951d7c00cSaliguori             return ret;
600ea2384d3Sbellard         }
6013e82990bSJes Sorensen         total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
6027c96d46eSaliguori 
6037c96d46eSaliguori         if (bs1->drv && bs1->drv->protocol_name)
6047c96d46eSaliguori             is_protocol = 1;
6057c96d46eSaliguori 
606ea2384d3Sbellard         bdrv_delete(bs1);
607ea2384d3Sbellard 
608ea2384d3Sbellard         get_tmp_filename(tmp_filename, sizeof(tmp_filename));
6097c96d46eSaliguori 
6107c96d46eSaliguori         /* Real path is meaningless for protocols */
6117c96d46eSaliguori         if (is_protocol)
6127c96d46eSaliguori             snprintf(backing_filename, sizeof(backing_filename),
6137c96d46eSaliguori                      "%s", filename);
614114cdfa9SKirill A. Shutemov         else if (!realpath(filename, backing_filename))
615114cdfa9SKirill A. Shutemov             return -errno;
6167c96d46eSaliguori 
61791a073a9SKevin Wolf         bdrv_qcow2 = bdrv_find_format("qcow2");
61891a073a9SKevin Wolf         options = parse_option_parameters("", bdrv_qcow2->create_options, NULL);
61991a073a9SKevin Wolf 
6203e82990bSJes Sorensen         set_option_parameter_int(options, BLOCK_OPT_SIZE, total_size);
62191a073a9SKevin Wolf         set_option_parameter(options, BLOCK_OPT_BACKING_FILE, backing_filename);
62291a073a9SKevin Wolf         if (drv) {
62391a073a9SKevin Wolf             set_option_parameter(options, BLOCK_OPT_BACKING_FMT,
62491a073a9SKevin Wolf                 drv->format_name);
62591a073a9SKevin Wolf         }
62691a073a9SKevin Wolf 
62791a073a9SKevin Wolf         ret = bdrv_create(bdrv_qcow2, tmp_filename, options);
628d748768cSJan Kiszka         free_option_parameters(options);
62951d7c00cSaliguori         if (ret < 0) {
63051d7c00cSaliguori             return ret;
631ea2384d3Sbellard         }
63291a073a9SKevin Wolf 
633ea2384d3Sbellard         filename = tmp_filename;
63491a073a9SKevin Wolf         drv = bdrv_qcow2;
635ea2384d3Sbellard         bs->is_temporary = 1;
636ea2384d3Sbellard     }
637ea2384d3Sbellard 
638b6ce07aaSKevin Wolf     /* Find the right image format driver */
6396db95603SChristoph Hellwig     if (!drv) {
640c98ac35dSStefan Weil         ret = find_image_format(filename, &drv);
641ea2384d3Sbellard     }
6426987307cSChristoph Hellwig 
64351d7c00cSaliguori     if (!drv) {
64451d7c00cSaliguori         goto unlink_and_fail;
64583f64091Sbellard     }
646b6ce07aaSKevin Wolf 
647b6ce07aaSKevin Wolf     /* Open the image */
648b6ce07aaSKevin Wolf     ret = bdrv_open_common(bs, filename, flags, drv);
649b6ce07aaSKevin Wolf     if (ret < 0) {
6506987307cSChristoph Hellwig         goto unlink_and_fail;
6516987307cSChristoph Hellwig     }
6526987307cSChristoph Hellwig 
653b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
654b6ce07aaSKevin Wolf     if ((flags & BDRV_O_NO_BACKING) == 0 && bs->backing_file[0] != '\0') {
655b6ce07aaSKevin Wolf         char backing_filename[PATH_MAX];
656b6ce07aaSKevin Wolf         int back_flags;
657b6ce07aaSKevin Wolf         BlockDriver *back_drv = NULL;
658b6ce07aaSKevin Wolf 
659b6ce07aaSKevin Wolf         bs->backing_hd = bdrv_new("");
660df2dbb4aSStefan Hajnoczi 
661df2dbb4aSStefan Hajnoczi         if (path_has_protocol(bs->backing_file)) {
662df2dbb4aSStefan Hajnoczi             pstrcpy(backing_filename, sizeof(backing_filename),
663df2dbb4aSStefan Hajnoczi                     bs->backing_file);
664df2dbb4aSStefan Hajnoczi         } else {
665b6ce07aaSKevin Wolf             path_combine(backing_filename, sizeof(backing_filename),
666b6ce07aaSKevin Wolf                          filename, bs->backing_file);
667df2dbb4aSStefan Hajnoczi         }
668df2dbb4aSStefan Hajnoczi 
669df2dbb4aSStefan Hajnoczi         if (bs->backing_format[0] != '\0') {
670b6ce07aaSKevin Wolf             back_drv = bdrv_find_format(bs->backing_format);
671df2dbb4aSStefan Hajnoczi         }
672b6ce07aaSKevin Wolf 
673b6ce07aaSKevin Wolf         /* backing files always opened read-only */
674b6ce07aaSKevin Wolf         back_flags =
675b6ce07aaSKevin Wolf             flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
676b6ce07aaSKevin Wolf 
677b6ce07aaSKevin Wolf         ret = bdrv_open(bs->backing_hd, backing_filename, back_flags, back_drv);
678b6ce07aaSKevin Wolf         if (ret < 0) {
679b6ce07aaSKevin Wolf             bdrv_close(bs);
680b6ce07aaSKevin Wolf             return ret;
681b6ce07aaSKevin Wolf         }
682b6ce07aaSKevin Wolf         if (bs->is_temporary) {
683b6ce07aaSKevin Wolf             bs->backing_hd->keep_read_only = !(flags & BDRV_O_RDWR);
684b6ce07aaSKevin Wolf         } else {
685b6ce07aaSKevin Wolf             /* base image inherits from "parent" */
686b6ce07aaSKevin Wolf             bs->backing_hd->keep_read_only = bs->keep_read_only;
687b6ce07aaSKevin Wolf         }
688b6ce07aaSKevin Wolf     }
689b6ce07aaSKevin Wolf 
690b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
691b6ce07aaSKevin Wolf         /* call the change callback */
692b6ce07aaSKevin Wolf         bs->media_changed = 1;
693b6ce07aaSKevin Wolf         if (bs->change_cb)
694db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
695b6ce07aaSKevin Wolf     }
696b6ce07aaSKevin Wolf 
697b6ce07aaSKevin Wolf     return 0;
698b6ce07aaSKevin Wolf 
699b6ce07aaSKevin Wolf unlink_and_fail:
700b6ce07aaSKevin Wolf     if (bs->is_temporary) {
701b6ce07aaSKevin Wolf         unlink(filename);
702b6ce07aaSKevin Wolf     }
703b6ce07aaSKevin Wolf     return ret;
704b6ce07aaSKevin Wolf }
705b6ce07aaSKevin Wolf 
706fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
707fc01f7e7Sbellard {
70819cb3738Sbellard     if (bs->drv) {
709f9092b10SMarkus Armbruster         if (bs == bs_snapshots) {
710f9092b10SMarkus Armbruster             bs_snapshots = NULL;
711f9092b10SMarkus Armbruster         }
712557df6acSStefan Hajnoczi         if (bs->backing_hd) {
713ea2384d3Sbellard             bdrv_delete(bs->backing_hd);
714557df6acSStefan Hajnoczi             bs->backing_hd = NULL;
715557df6acSStefan Hajnoczi         }
716ea2384d3Sbellard         bs->drv->bdrv_close(bs);
7177267c094SAnthony Liguori         g_free(bs->opaque);
718ea2384d3Sbellard #ifdef _WIN32
719ea2384d3Sbellard         if (bs->is_temporary) {
720ea2384d3Sbellard             unlink(bs->filename);
721ea2384d3Sbellard         }
72267b915a5Sbellard #endif
723ea2384d3Sbellard         bs->opaque = NULL;
724ea2384d3Sbellard         bs->drv = NULL;
725b338082bSbellard 
72666f82ceeSKevin Wolf         if (bs->file != NULL) {
72766f82ceeSKevin Wolf             bdrv_close(bs->file);
72866f82ceeSKevin Wolf         }
72966f82ceeSKevin Wolf 
730b338082bSbellard         /* call the change callback */
73119cb3738Sbellard         bs->media_changed = 1;
732b338082bSbellard         if (bs->change_cb)
733db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
734b338082bSbellard     }
735b338082bSbellard }
736b338082bSbellard 
7372bc93fedSMORITA Kazutaka void bdrv_close_all(void)
7382bc93fedSMORITA Kazutaka {
7392bc93fedSMORITA Kazutaka     BlockDriverState *bs;
7402bc93fedSMORITA Kazutaka 
7412bc93fedSMORITA Kazutaka     QTAILQ_FOREACH(bs, &bdrv_states, list) {
7422bc93fedSMORITA Kazutaka         bdrv_close(bs);
7432bc93fedSMORITA Kazutaka     }
7442bc93fedSMORITA Kazutaka }
7452bc93fedSMORITA Kazutaka 
746d22b2f41SRyan Harper /* make a BlockDriverState anonymous by removing from bdrv_state list.
747d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
748d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
749d22b2f41SRyan Harper {
750d22b2f41SRyan Harper     if (bs->device_name[0] != '\0') {
751d22b2f41SRyan Harper         QTAILQ_REMOVE(&bdrv_states, bs, list);
752d22b2f41SRyan Harper     }
753d22b2f41SRyan Harper     bs->device_name[0] = '\0';
754d22b2f41SRyan Harper }
755d22b2f41SRyan Harper 
756b338082bSbellard void bdrv_delete(BlockDriverState *bs)
757b338082bSbellard {
75818846deeSMarkus Armbruster     assert(!bs->peer);
75918846deeSMarkus Armbruster 
7601b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
761d22b2f41SRyan Harper     bdrv_make_anon(bs);
76234c6f050Saurel32 
763b338082bSbellard     bdrv_close(bs);
76466f82ceeSKevin Wolf     if (bs->file != NULL) {
76566f82ceeSKevin Wolf         bdrv_delete(bs->file);
76666f82ceeSKevin Wolf     }
76766f82ceeSKevin Wolf 
768f9092b10SMarkus Armbruster     assert(bs != bs_snapshots);
7697267c094SAnthony Liguori     g_free(bs);
770fc01f7e7Sbellard }
771fc01f7e7Sbellard 
77218846deeSMarkus Armbruster int bdrv_attach(BlockDriverState *bs, DeviceState *qdev)
77318846deeSMarkus Armbruster {
77418846deeSMarkus Armbruster     if (bs->peer) {
77518846deeSMarkus Armbruster         return -EBUSY;
77618846deeSMarkus Armbruster     }
77718846deeSMarkus Armbruster     bs->peer = qdev;
77818846deeSMarkus Armbruster     return 0;
77918846deeSMarkus Armbruster }
78018846deeSMarkus Armbruster 
78118846deeSMarkus Armbruster void bdrv_detach(BlockDriverState *bs, DeviceState *qdev)
78218846deeSMarkus Armbruster {
78318846deeSMarkus Armbruster     assert(bs->peer == qdev);
78418846deeSMarkus Armbruster     bs->peer = NULL;
785a19712b0SMarkus Armbruster     bs->change_cb = NULL;
786a19712b0SMarkus Armbruster     bs->change_opaque = NULL;
78718846deeSMarkus Armbruster }
78818846deeSMarkus Armbruster 
78918846deeSMarkus Armbruster DeviceState *bdrv_get_attached(BlockDriverState *bs)
79018846deeSMarkus Armbruster {
79118846deeSMarkus Armbruster     return bs->peer;
79218846deeSMarkus Armbruster }
79318846deeSMarkus Armbruster 
794e97fc193Saliguori /*
795e97fc193Saliguori  * Run consistency checks on an image
796e97fc193Saliguori  *
797e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
798a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
799e076f338SKevin Wolf  * check are stored in res.
800e97fc193Saliguori  */
801e076f338SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res)
802e97fc193Saliguori {
803e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
804e97fc193Saliguori         return -ENOTSUP;
805e97fc193Saliguori     }
806e97fc193Saliguori 
807e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
8089ac228e0SKevin Wolf     return bs->drv->bdrv_check(bs, res);
809e97fc193Saliguori }
810e97fc193Saliguori 
8118a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
8128a426614SKevin Wolf 
81333e3963eSbellard /* commit COW file into the raw image */
81433e3963eSbellard int bdrv_commit(BlockDriverState *bs)
81533e3963eSbellard {
81619cb3738Sbellard     BlockDriver *drv = bs->drv;
817ee181196SKevin Wolf     BlockDriver *backing_drv;
8188a426614SKevin Wolf     int64_t sector, total_sectors;
8198a426614SKevin Wolf     int n, ro, open_flags;
8204dca4b63SNaphtali Sprei     int ret = 0, rw_ret = 0;
8218a426614SKevin Wolf     uint8_t *buf;
8224dca4b63SNaphtali Sprei     char filename[1024];
8234dca4b63SNaphtali Sprei     BlockDriverState *bs_rw, *bs_ro;
82433e3963eSbellard 
82519cb3738Sbellard     if (!drv)
82619cb3738Sbellard         return -ENOMEDIUM;
82733e3963eSbellard 
8284dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
8294dca4b63SNaphtali Sprei         return -ENOTSUP;
8304dca4b63SNaphtali Sprei     }
8314dca4b63SNaphtali Sprei 
8324dca4b63SNaphtali Sprei     if (bs->backing_hd->keep_read_only) {
833ea2384d3Sbellard         return -EACCES;
83433e3963eSbellard     }
83533e3963eSbellard 
836ee181196SKevin Wolf     backing_drv = bs->backing_hd->drv;
8374dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
8384dca4b63SNaphtali Sprei     strncpy(filename, bs->backing_hd->filename, sizeof(filename));
8394dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
8404dca4b63SNaphtali Sprei 
8414dca4b63SNaphtali Sprei     if (ro) {
8424dca4b63SNaphtali Sprei         /* re-open as RW */
8434dca4b63SNaphtali Sprei         bdrv_delete(bs->backing_hd);
8444dca4b63SNaphtali Sprei         bs->backing_hd = NULL;
8454dca4b63SNaphtali Sprei         bs_rw = bdrv_new("");
846ee181196SKevin Wolf         rw_ret = bdrv_open(bs_rw, filename, open_flags | BDRV_O_RDWR,
847ee181196SKevin Wolf             backing_drv);
8484dca4b63SNaphtali Sprei         if (rw_ret < 0) {
8494dca4b63SNaphtali Sprei             bdrv_delete(bs_rw);
8504dca4b63SNaphtali Sprei             /* try to re-open read-only */
8514dca4b63SNaphtali Sprei             bs_ro = bdrv_new("");
852ee181196SKevin Wolf             ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
853ee181196SKevin Wolf                 backing_drv);
8544dca4b63SNaphtali Sprei             if (ret < 0) {
8554dca4b63SNaphtali Sprei                 bdrv_delete(bs_ro);
8564dca4b63SNaphtali Sprei                 /* drive not functional anymore */
8574dca4b63SNaphtali Sprei                 bs->drv = NULL;
8584dca4b63SNaphtali Sprei                 return ret;
8594dca4b63SNaphtali Sprei             }
8604dca4b63SNaphtali Sprei             bs->backing_hd = bs_ro;
8614dca4b63SNaphtali Sprei             return rw_ret;
8624dca4b63SNaphtali Sprei         }
8634dca4b63SNaphtali Sprei         bs->backing_hd = bs_rw;
864ea2384d3Sbellard     }
865ea2384d3Sbellard 
8666ea44308SJan Kiszka     total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
8677267c094SAnthony Liguori     buf = g_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
8688a426614SKevin Wolf 
8698a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
8708a426614SKevin Wolf         if (drv->bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
8718a426614SKevin Wolf 
8728a426614SKevin Wolf             if (bdrv_read(bs, sector, buf, n) != 0) {
8734dca4b63SNaphtali Sprei                 ret = -EIO;
8744dca4b63SNaphtali Sprei                 goto ro_cleanup;
87533e3963eSbellard             }
87633e3963eSbellard 
8778a426614SKevin Wolf             if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
8784dca4b63SNaphtali Sprei                 ret = -EIO;
8794dca4b63SNaphtali Sprei                 goto ro_cleanup;
88033e3963eSbellard             }
88133e3963eSbellard         }
88233e3963eSbellard     }
88395389c86Sbellard 
8841d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
8851d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
8861d44952fSChristoph Hellwig         bdrv_flush(bs);
8871d44952fSChristoph Hellwig     }
88895389c86Sbellard 
8893f5075aeSChristoph Hellwig     /*
8903f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
8913f5075aeSChristoph Hellwig      * stable on disk.
8923f5075aeSChristoph Hellwig      */
8933f5075aeSChristoph Hellwig     if (bs->backing_hd)
8943f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
8954dca4b63SNaphtali Sprei 
8964dca4b63SNaphtali Sprei ro_cleanup:
8977267c094SAnthony Liguori     g_free(buf);
8984dca4b63SNaphtali Sprei 
8994dca4b63SNaphtali Sprei     if (ro) {
9004dca4b63SNaphtali Sprei         /* re-open as RO */
9014dca4b63SNaphtali Sprei         bdrv_delete(bs->backing_hd);
9024dca4b63SNaphtali Sprei         bs->backing_hd = NULL;
9034dca4b63SNaphtali Sprei         bs_ro = bdrv_new("");
904ee181196SKevin Wolf         ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
905ee181196SKevin Wolf             backing_drv);
9064dca4b63SNaphtali Sprei         if (ret < 0) {
9074dca4b63SNaphtali Sprei             bdrv_delete(bs_ro);
9084dca4b63SNaphtali Sprei             /* drive not functional anymore */
9094dca4b63SNaphtali Sprei             bs->drv = NULL;
9104dca4b63SNaphtali Sprei             return ret;
9114dca4b63SNaphtali Sprei         }
9124dca4b63SNaphtali Sprei         bs->backing_hd = bs_ro;
9134dca4b63SNaphtali Sprei         bs->backing_hd->keep_read_only = 0;
9144dca4b63SNaphtali Sprei     }
9154dca4b63SNaphtali Sprei 
9161d44952fSChristoph Hellwig     return ret;
91733e3963eSbellard }
91833e3963eSbellard 
9196ab4b5abSMarkus Armbruster void bdrv_commit_all(void)
9206ab4b5abSMarkus Armbruster {
9216ab4b5abSMarkus Armbruster     BlockDriverState *bs;
9226ab4b5abSMarkus Armbruster 
9236ab4b5abSMarkus Armbruster     QTAILQ_FOREACH(bs, &bdrv_states, list) {
9246ab4b5abSMarkus Armbruster         bdrv_commit(bs);
9256ab4b5abSMarkus Armbruster     }
9266ab4b5abSMarkus Armbruster }
9276ab4b5abSMarkus Armbruster 
928756e6736SKevin Wolf /*
929756e6736SKevin Wolf  * Return values:
930756e6736SKevin Wolf  * 0        - success
931756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
932756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
933756e6736SKevin Wolf  *            image file header
934756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
935756e6736SKevin Wolf  */
936756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
937756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
938756e6736SKevin Wolf {
939756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
940756e6736SKevin Wolf 
941756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
942756e6736SKevin Wolf         return drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
943756e6736SKevin Wolf     } else {
944756e6736SKevin Wolf         return -ENOTSUP;
945756e6736SKevin Wolf     }
946756e6736SKevin Wolf }
947756e6736SKevin Wolf 
94871d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
94971d0770cSaliguori                                    size_t size)
95071d0770cSaliguori {
95171d0770cSaliguori     int64_t len;
95271d0770cSaliguori 
95371d0770cSaliguori     if (!bdrv_is_inserted(bs))
95471d0770cSaliguori         return -ENOMEDIUM;
95571d0770cSaliguori 
95671d0770cSaliguori     if (bs->growable)
95771d0770cSaliguori         return 0;
95871d0770cSaliguori 
95971d0770cSaliguori     len = bdrv_getlength(bs);
96071d0770cSaliguori 
961fbb7b4e0SKevin Wolf     if (offset < 0)
962fbb7b4e0SKevin Wolf         return -EIO;
963fbb7b4e0SKevin Wolf 
964fbb7b4e0SKevin Wolf     if ((offset > len) || (len - offset < size))
96571d0770cSaliguori         return -EIO;
96671d0770cSaliguori 
96771d0770cSaliguori     return 0;
96871d0770cSaliguori }
96971d0770cSaliguori 
97071d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
97171d0770cSaliguori                               int nb_sectors)
97271d0770cSaliguori {
973eb5a3165SJes Sorensen     return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
974eb5a3165SJes Sorensen                                    nb_sectors * BDRV_SECTOR_SIZE);
97571d0770cSaliguori }
97671d0770cSaliguori 
977e7a8a783SKevin Wolf static inline bool bdrv_has_async_rw(BlockDriver *drv)
978e7a8a783SKevin Wolf {
979e7a8a783SKevin Wolf     return drv->bdrv_co_readv != bdrv_co_readv_em
980e7a8a783SKevin Wolf         || drv->bdrv_aio_readv != bdrv_aio_readv_em;
981e7a8a783SKevin Wolf }
982e7a8a783SKevin Wolf 
983e7a8a783SKevin Wolf static inline bool bdrv_has_async_flush(BlockDriver *drv)
984e7a8a783SKevin Wolf {
985e7a8a783SKevin Wolf     return drv->bdrv_aio_flush != bdrv_aio_flush_em;
986e7a8a783SKevin Wolf }
987e7a8a783SKevin Wolf 
98819cb3738Sbellard /* return < 0 if error. See bdrv_write() for the return codes */
989fc01f7e7Sbellard int bdrv_read(BlockDriverState *bs, int64_t sector_num,
990fc01f7e7Sbellard               uint8_t *buf, int nb_sectors)
991fc01f7e7Sbellard {
992ea2384d3Sbellard     BlockDriver *drv = bs->drv;
993fc01f7e7Sbellard 
99419cb3738Sbellard     if (!drv)
99519cb3738Sbellard         return -ENOMEDIUM;
996e7a8a783SKevin Wolf 
997e7a8a783SKevin Wolf     if (bdrv_has_async_rw(drv) && qemu_in_coroutine()) {
998e7a8a783SKevin Wolf         QEMUIOVector qiov;
999e7a8a783SKevin Wolf         struct iovec iov = {
1000e7a8a783SKevin Wolf             .iov_base = (void *)buf,
1001e7a8a783SKevin Wolf             .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
1002e7a8a783SKevin Wolf         };
1003e7a8a783SKevin Wolf 
1004e7a8a783SKevin Wolf         qemu_iovec_init_external(&qiov, &iov, 1);
1005e7a8a783SKevin Wolf         return bdrv_co_readv(bs, sector_num, nb_sectors, &qiov);
1006e7a8a783SKevin Wolf     }
1007e7a8a783SKevin Wolf 
100871d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
100971d0770cSaliguori         return -EIO;
1010b338082bSbellard 
101183f64091Sbellard     return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
101283f64091Sbellard }
1013fc01f7e7Sbellard 
10147cd1e32aSlirans@il.ibm.com static void set_dirty_bitmap(BlockDriverState *bs, int64_t sector_num,
10157cd1e32aSlirans@il.ibm.com                              int nb_sectors, int dirty)
10167cd1e32aSlirans@il.ibm.com {
10177cd1e32aSlirans@il.ibm.com     int64_t start, end;
1018c6d22830SJan Kiszka     unsigned long val, idx, bit;
1019a55eb92cSJan Kiszka 
10206ea44308SJan Kiszka     start = sector_num / BDRV_SECTORS_PER_DIRTY_CHUNK;
1021c6d22830SJan Kiszka     end = (sector_num + nb_sectors - 1) / BDRV_SECTORS_PER_DIRTY_CHUNK;
10227cd1e32aSlirans@il.ibm.com 
10237cd1e32aSlirans@il.ibm.com     for (; start <= end; start++) {
1024c6d22830SJan Kiszka         idx = start / (sizeof(unsigned long) * 8);
1025c6d22830SJan Kiszka         bit = start % (sizeof(unsigned long) * 8);
1026c6d22830SJan Kiszka         val = bs->dirty_bitmap[idx];
1027c6d22830SJan Kiszka         if (dirty) {
10286d59fec1SMarcelo Tosatti             if (!(val & (1UL << bit))) {
1029aaa0eb75SLiran Schour                 bs->dirty_count++;
10306d59fec1SMarcelo Tosatti                 val |= 1UL << bit;
1031aaa0eb75SLiran Schour             }
1032c6d22830SJan Kiszka         } else {
10336d59fec1SMarcelo Tosatti             if (val & (1UL << bit)) {
1034aaa0eb75SLiran Schour                 bs->dirty_count--;
10356d59fec1SMarcelo Tosatti                 val &= ~(1UL << bit);
1036c6d22830SJan Kiszka             }
1037aaa0eb75SLiran Schour         }
1038c6d22830SJan Kiszka         bs->dirty_bitmap[idx] = val;
10397cd1e32aSlirans@il.ibm.com     }
10407cd1e32aSlirans@il.ibm.com }
10417cd1e32aSlirans@il.ibm.com 
104219cb3738Sbellard /* Return < 0 if error. Important errors are:
104319cb3738Sbellard   -EIO         generic I/O error (may happen for all errors)
104419cb3738Sbellard   -ENOMEDIUM   No media inserted.
104519cb3738Sbellard   -EINVAL      Invalid sector number or nb_sectors
104619cb3738Sbellard   -EACCES      Trying to write a read-only device
104719cb3738Sbellard */
1048fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
1049fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
1050fc01f7e7Sbellard {
105183f64091Sbellard     BlockDriver *drv = bs->drv;
1052e7a8a783SKevin Wolf 
105319cb3738Sbellard     if (!bs->drv)
105419cb3738Sbellard         return -ENOMEDIUM;
1055e7a8a783SKevin Wolf 
1056e7a8a783SKevin Wolf     if (bdrv_has_async_rw(drv) && qemu_in_coroutine()) {
1057e7a8a783SKevin Wolf         QEMUIOVector qiov;
1058e7a8a783SKevin Wolf         struct iovec iov = {
1059e7a8a783SKevin Wolf             .iov_base = (void *)buf,
1060e7a8a783SKevin Wolf             .iov_len = nb_sectors * BDRV_SECTOR_SIZE,
1061e7a8a783SKevin Wolf         };
1062e7a8a783SKevin Wolf 
1063e7a8a783SKevin Wolf         qemu_iovec_init_external(&qiov, &iov, 1);
1064e7a8a783SKevin Wolf         return bdrv_co_writev(bs, sector_num, nb_sectors, &qiov);
1065e7a8a783SKevin Wolf     }
1066e7a8a783SKevin Wolf 
10670849bf08Sbellard     if (bs->read_only)
106819cb3738Sbellard         return -EACCES;
106971d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
107071d0770cSaliguori         return -EIO;
107171d0770cSaliguori 
1072c6d22830SJan Kiszka     if (bs->dirty_bitmap) {
10737cd1e32aSlirans@il.ibm.com         set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
10747cd1e32aSlirans@il.ibm.com     }
10757cd1e32aSlirans@il.ibm.com 
1076294cc35fSKevin Wolf     if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
1077294cc35fSKevin Wolf         bs->wr_highest_sector = sector_num + nb_sectors - 1;
1078294cc35fSKevin Wolf     }
1079294cc35fSKevin Wolf 
108083f64091Sbellard     return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
108183f64091Sbellard }
108283f64091Sbellard 
1083eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset,
1084eda578e5Saliguori                void *buf, int count1)
108583f64091Sbellard {
10866ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
108783f64091Sbellard     int len, nb_sectors, count;
108883f64091Sbellard     int64_t sector_num;
10899a8c4cceSKevin Wolf     int ret;
109083f64091Sbellard 
109183f64091Sbellard     count = count1;
109283f64091Sbellard     /* first read to align to sector start */
10936ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
109483f64091Sbellard     if (len > count)
109583f64091Sbellard         len = count;
10966ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
109783f64091Sbellard     if (len > 0) {
10989a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
10999a8c4cceSKevin Wolf             return ret;
11006ea44308SJan Kiszka         memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
110183f64091Sbellard         count -= len;
110283f64091Sbellard         if (count == 0)
110383f64091Sbellard             return count1;
110483f64091Sbellard         sector_num++;
110583f64091Sbellard         buf += len;
110683f64091Sbellard     }
110783f64091Sbellard 
110883f64091Sbellard     /* read the sectors "in place" */
11096ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
111083f64091Sbellard     if (nb_sectors > 0) {
11119a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
11129a8c4cceSKevin Wolf             return ret;
111383f64091Sbellard         sector_num += nb_sectors;
11146ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
111583f64091Sbellard         buf += len;
111683f64091Sbellard         count -= len;
111783f64091Sbellard     }
111883f64091Sbellard 
111983f64091Sbellard     /* add data from the last sector */
112083f64091Sbellard     if (count > 0) {
11219a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
11229a8c4cceSKevin Wolf             return ret;
112383f64091Sbellard         memcpy(buf, tmp_buf, count);
112483f64091Sbellard     }
112583f64091Sbellard     return count1;
112683f64091Sbellard }
112783f64091Sbellard 
1128eda578e5Saliguori int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
1129eda578e5Saliguori                 const void *buf, int count1)
113083f64091Sbellard {
11316ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
113283f64091Sbellard     int len, nb_sectors, count;
113383f64091Sbellard     int64_t sector_num;
11349a8c4cceSKevin Wolf     int ret;
113583f64091Sbellard 
113683f64091Sbellard     count = count1;
113783f64091Sbellard     /* first write to align to sector start */
11386ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
113983f64091Sbellard     if (len > count)
114083f64091Sbellard         len = count;
11416ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
114283f64091Sbellard     if (len > 0) {
11439a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
11449a8c4cceSKevin Wolf             return ret;
11456ea44308SJan Kiszka         memcpy(tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), buf, len);
11469a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
11479a8c4cceSKevin Wolf             return ret;
114883f64091Sbellard         count -= len;
114983f64091Sbellard         if (count == 0)
115083f64091Sbellard             return count1;
115183f64091Sbellard         sector_num++;
115283f64091Sbellard         buf += len;
115383f64091Sbellard     }
115483f64091Sbellard 
115583f64091Sbellard     /* write the sectors "in place" */
11566ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
115783f64091Sbellard     if (nb_sectors > 0) {
11589a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, buf, nb_sectors)) < 0)
11599a8c4cceSKevin Wolf             return ret;
116083f64091Sbellard         sector_num += nb_sectors;
11616ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
116283f64091Sbellard         buf += len;
116383f64091Sbellard         count -= len;
116483f64091Sbellard     }
116583f64091Sbellard 
116683f64091Sbellard     /* add data from the last sector */
116783f64091Sbellard     if (count > 0) {
11689a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
11699a8c4cceSKevin Wolf             return ret;
117083f64091Sbellard         memcpy(tmp_buf, buf, count);
11719a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
11729a8c4cceSKevin Wolf             return ret;
117383f64091Sbellard     }
117483f64091Sbellard     return count1;
117583f64091Sbellard }
117683f64091Sbellard 
1177f08145feSKevin Wolf /*
1178f08145feSKevin Wolf  * Writes to the file and ensures that no writes are reordered across this
1179f08145feSKevin Wolf  * request (acts as a barrier)
1180f08145feSKevin Wolf  *
1181f08145feSKevin Wolf  * Returns 0 on success, -errno in error cases.
1182f08145feSKevin Wolf  */
1183f08145feSKevin Wolf int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
1184f08145feSKevin Wolf     const void *buf, int count)
1185f08145feSKevin Wolf {
1186f08145feSKevin Wolf     int ret;
1187f08145feSKevin Wolf 
1188f08145feSKevin Wolf     ret = bdrv_pwrite(bs, offset, buf, count);
1189f08145feSKevin Wolf     if (ret < 0) {
1190f08145feSKevin Wolf         return ret;
1191f08145feSKevin Wolf     }
1192f08145feSKevin Wolf 
119392196b2fSStefan Hajnoczi     /* No flush needed for cache modes that use O_DSYNC */
119492196b2fSStefan Hajnoczi     if ((bs->open_flags & BDRV_O_CACHE_WB) != 0) {
1195f08145feSKevin Wolf         bdrv_flush(bs);
1196f08145feSKevin Wolf     }
1197f08145feSKevin Wolf 
1198f08145feSKevin Wolf     return 0;
1199f08145feSKevin Wolf }
1200f08145feSKevin Wolf 
1201da1fa91dSKevin Wolf int coroutine_fn bdrv_co_readv(BlockDriverState *bs, int64_t sector_num,
1202da1fa91dSKevin Wolf     int nb_sectors, QEMUIOVector *qiov)
1203da1fa91dSKevin Wolf {
1204da1fa91dSKevin Wolf     BlockDriver *drv = bs->drv;
1205da1fa91dSKevin Wolf 
1206da1fa91dSKevin Wolf     trace_bdrv_co_readv(bs, sector_num, nb_sectors);
1207da1fa91dSKevin Wolf 
1208da1fa91dSKevin Wolf     if (!drv) {
1209da1fa91dSKevin Wolf         return -ENOMEDIUM;
1210da1fa91dSKevin Wolf     }
1211da1fa91dSKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
1212da1fa91dSKevin Wolf         return -EIO;
1213da1fa91dSKevin Wolf     }
1214da1fa91dSKevin Wolf 
1215da1fa91dSKevin Wolf     return drv->bdrv_co_readv(bs, sector_num, nb_sectors, qiov);
1216da1fa91dSKevin Wolf }
1217da1fa91dSKevin Wolf 
1218da1fa91dSKevin Wolf int coroutine_fn bdrv_co_writev(BlockDriverState *bs, int64_t sector_num,
1219da1fa91dSKevin Wolf     int nb_sectors, QEMUIOVector *qiov)
1220da1fa91dSKevin Wolf {
1221da1fa91dSKevin Wolf     BlockDriver *drv = bs->drv;
1222da1fa91dSKevin Wolf 
1223da1fa91dSKevin Wolf     trace_bdrv_co_writev(bs, sector_num, nb_sectors);
1224da1fa91dSKevin Wolf 
1225da1fa91dSKevin Wolf     if (!bs->drv) {
1226da1fa91dSKevin Wolf         return -ENOMEDIUM;
1227da1fa91dSKevin Wolf     }
1228da1fa91dSKevin Wolf     if (bs->read_only) {
1229da1fa91dSKevin Wolf         return -EACCES;
1230da1fa91dSKevin Wolf     }
1231da1fa91dSKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors)) {
1232da1fa91dSKevin Wolf         return -EIO;
1233da1fa91dSKevin Wolf     }
1234da1fa91dSKevin Wolf 
1235da1fa91dSKevin Wolf     if (bs->dirty_bitmap) {
1236da1fa91dSKevin Wolf         set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
1237da1fa91dSKevin Wolf     }
1238da1fa91dSKevin Wolf 
1239da1fa91dSKevin Wolf     if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
1240da1fa91dSKevin Wolf         bs->wr_highest_sector = sector_num + nb_sectors - 1;
1241da1fa91dSKevin Wolf     }
1242da1fa91dSKevin Wolf 
1243da1fa91dSKevin Wolf     return drv->bdrv_co_writev(bs, sector_num, nb_sectors, qiov);
1244da1fa91dSKevin Wolf }
1245da1fa91dSKevin Wolf 
124683f64091Sbellard /**
124783f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
124883f64091Sbellard  */
124983f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
125083f64091Sbellard {
125183f64091Sbellard     BlockDriver *drv = bs->drv;
125251762288SStefan Hajnoczi     int ret;
125383f64091Sbellard     if (!drv)
125419cb3738Sbellard         return -ENOMEDIUM;
125583f64091Sbellard     if (!drv->bdrv_truncate)
125683f64091Sbellard         return -ENOTSUP;
125759f2689dSNaphtali Sprei     if (bs->read_only)
125859f2689dSNaphtali Sprei         return -EACCES;
12598591675fSMarcelo Tosatti     if (bdrv_in_use(bs))
12608591675fSMarcelo Tosatti         return -EBUSY;
126151762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
126251762288SStefan Hajnoczi     if (ret == 0) {
126351762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
1264db97ee6aSChristoph Hellwig         if (bs->change_cb) {
1265db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_SIZE);
1266db97ee6aSChristoph Hellwig         }
126751762288SStefan Hajnoczi     }
126851762288SStefan Hajnoczi     return ret;
126983f64091Sbellard }
127083f64091Sbellard 
127183f64091Sbellard /**
12724a1d5e1fSFam Zheng  * Length of a allocated file in bytes. Sparse files are counted by actual
12734a1d5e1fSFam Zheng  * allocated space. Return < 0 if error or unknown.
12744a1d5e1fSFam Zheng  */
12754a1d5e1fSFam Zheng int64_t bdrv_get_allocated_file_size(BlockDriverState *bs)
12764a1d5e1fSFam Zheng {
12774a1d5e1fSFam Zheng     BlockDriver *drv = bs->drv;
12784a1d5e1fSFam Zheng     if (!drv) {
12794a1d5e1fSFam Zheng         return -ENOMEDIUM;
12804a1d5e1fSFam Zheng     }
12814a1d5e1fSFam Zheng     if (drv->bdrv_get_allocated_file_size) {
12824a1d5e1fSFam Zheng         return drv->bdrv_get_allocated_file_size(bs);
12834a1d5e1fSFam Zheng     }
12844a1d5e1fSFam Zheng     if (bs->file) {
12854a1d5e1fSFam Zheng         return bdrv_get_allocated_file_size(bs->file);
12864a1d5e1fSFam Zheng     }
12874a1d5e1fSFam Zheng     return -ENOTSUP;
12884a1d5e1fSFam Zheng }
12894a1d5e1fSFam Zheng 
12904a1d5e1fSFam Zheng /**
129183f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
129283f64091Sbellard  */
129383f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
129483f64091Sbellard {
129583f64091Sbellard     BlockDriver *drv = bs->drv;
129683f64091Sbellard     if (!drv)
129719cb3738Sbellard         return -ENOMEDIUM;
129851762288SStefan Hajnoczi 
129946a4e4e6SStefan Hajnoczi     if (bs->growable || bs->removable) {
130046a4e4e6SStefan Hajnoczi         if (drv->bdrv_getlength) {
130183f64091Sbellard             return drv->bdrv_getlength(bs);
1302fc01f7e7Sbellard         }
130346a4e4e6SStefan Hajnoczi     }
130446a4e4e6SStefan Hajnoczi     return bs->total_sectors * BDRV_SECTOR_SIZE;
130546a4e4e6SStefan Hajnoczi }
1306fc01f7e7Sbellard 
130719cb3738Sbellard /* return 0 as number of sectors if no device present or error */
130896b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
1309fc01f7e7Sbellard {
131019cb3738Sbellard     int64_t length;
131119cb3738Sbellard     length = bdrv_getlength(bs);
131219cb3738Sbellard     if (length < 0)
131319cb3738Sbellard         length = 0;
131419cb3738Sbellard     else
13156ea44308SJan Kiszka         length = length >> BDRV_SECTOR_BITS;
131619cb3738Sbellard     *nb_sectors_ptr = length;
1317fc01f7e7Sbellard }
1318cf98951bSbellard 
1319f3d54fc4Saliguori struct partition {
1320f3d54fc4Saliguori         uint8_t boot_ind;           /* 0x80 - active */
1321f3d54fc4Saliguori         uint8_t head;               /* starting head */
1322f3d54fc4Saliguori         uint8_t sector;             /* starting sector */
1323f3d54fc4Saliguori         uint8_t cyl;                /* starting cylinder */
1324f3d54fc4Saliguori         uint8_t sys_ind;            /* What partition type */
1325f3d54fc4Saliguori         uint8_t end_head;           /* end head */
1326f3d54fc4Saliguori         uint8_t end_sector;         /* end sector */
1327f3d54fc4Saliguori         uint8_t end_cyl;            /* end cylinder */
1328f3d54fc4Saliguori         uint32_t start_sect;        /* starting sector counting from 0 */
1329f3d54fc4Saliguori         uint32_t nr_sects;          /* nr of sectors in partition */
1330f3d54fc4Saliguori } __attribute__((packed));
1331f3d54fc4Saliguori 
1332f3d54fc4Saliguori /* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */
1333f3d54fc4Saliguori static int guess_disk_lchs(BlockDriverState *bs,
1334f3d54fc4Saliguori                            int *pcylinders, int *pheads, int *psectors)
1335f3d54fc4Saliguori {
1336eb5a3165SJes Sorensen     uint8_t buf[BDRV_SECTOR_SIZE];
1337f3d54fc4Saliguori     int ret, i, heads, sectors, cylinders;
1338f3d54fc4Saliguori     struct partition *p;
1339f3d54fc4Saliguori     uint32_t nr_sects;
1340a38131b6Sblueswir1     uint64_t nb_sectors;
1341f3d54fc4Saliguori 
1342f3d54fc4Saliguori     bdrv_get_geometry(bs, &nb_sectors);
1343f3d54fc4Saliguori 
1344f3d54fc4Saliguori     ret = bdrv_read(bs, 0, buf, 1);
1345f3d54fc4Saliguori     if (ret < 0)
1346f3d54fc4Saliguori         return -1;
1347f3d54fc4Saliguori     /* test msdos magic */
1348f3d54fc4Saliguori     if (buf[510] != 0x55 || buf[511] != 0xaa)
1349f3d54fc4Saliguori         return -1;
1350f3d54fc4Saliguori     for(i = 0; i < 4; i++) {
1351f3d54fc4Saliguori         p = ((struct partition *)(buf + 0x1be)) + i;
1352f3d54fc4Saliguori         nr_sects = le32_to_cpu(p->nr_sects);
1353f3d54fc4Saliguori         if (nr_sects && p->end_head) {
1354f3d54fc4Saliguori             /* We make the assumption that the partition terminates on
1355f3d54fc4Saliguori                a cylinder boundary */
1356f3d54fc4Saliguori             heads = p->end_head + 1;
1357f3d54fc4Saliguori             sectors = p->end_sector & 63;
1358f3d54fc4Saliguori             if (sectors == 0)
1359f3d54fc4Saliguori                 continue;
1360f3d54fc4Saliguori             cylinders = nb_sectors / (heads * sectors);
1361f3d54fc4Saliguori             if (cylinders < 1 || cylinders > 16383)
1362f3d54fc4Saliguori                 continue;
1363f3d54fc4Saliguori             *pheads = heads;
1364f3d54fc4Saliguori             *psectors = sectors;
1365f3d54fc4Saliguori             *pcylinders = cylinders;
1366f3d54fc4Saliguori #if 0
1367f3d54fc4Saliguori             printf("guessed geometry: LCHS=%d %d %d\n",
1368f3d54fc4Saliguori                    cylinders, heads, sectors);
1369f3d54fc4Saliguori #endif
1370f3d54fc4Saliguori             return 0;
1371f3d54fc4Saliguori         }
1372f3d54fc4Saliguori     }
1373f3d54fc4Saliguori     return -1;
1374f3d54fc4Saliguori }
1375f3d54fc4Saliguori 
1376f3d54fc4Saliguori void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs)
1377f3d54fc4Saliguori {
1378f3d54fc4Saliguori     int translation, lba_detected = 0;
1379f3d54fc4Saliguori     int cylinders, heads, secs;
1380a38131b6Sblueswir1     uint64_t nb_sectors;
1381f3d54fc4Saliguori 
1382f3d54fc4Saliguori     /* if a geometry hint is available, use it */
1383f3d54fc4Saliguori     bdrv_get_geometry(bs, &nb_sectors);
1384f3d54fc4Saliguori     bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs);
1385f3d54fc4Saliguori     translation = bdrv_get_translation_hint(bs);
1386f3d54fc4Saliguori     if (cylinders != 0) {
1387f3d54fc4Saliguori         *pcyls = cylinders;
1388f3d54fc4Saliguori         *pheads = heads;
1389f3d54fc4Saliguori         *psecs = secs;
1390f3d54fc4Saliguori     } else {
1391f3d54fc4Saliguori         if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) {
1392f3d54fc4Saliguori             if (heads > 16) {
1393f3d54fc4Saliguori                 /* if heads > 16, it means that a BIOS LBA
1394f3d54fc4Saliguori                    translation was active, so the default
1395f3d54fc4Saliguori                    hardware geometry is OK */
1396f3d54fc4Saliguori                 lba_detected = 1;
1397f3d54fc4Saliguori                 goto default_geometry;
1398f3d54fc4Saliguori             } else {
1399f3d54fc4Saliguori                 *pcyls = cylinders;
1400f3d54fc4Saliguori                 *pheads = heads;
1401f3d54fc4Saliguori                 *psecs = secs;
1402f3d54fc4Saliguori                 /* disable any translation to be in sync with
1403f3d54fc4Saliguori                    the logical geometry */
1404f3d54fc4Saliguori                 if (translation == BIOS_ATA_TRANSLATION_AUTO) {
1405f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
1406f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_NONE);
1407f3d54fc4Saliguori                 }
1408f3d54fc4Saliguori             }
1409f3d54fc4Saliguori         } else {
1410f3d54fc4Saliguori         default_geometry:
1411f3d54fc4Saliguori             /* if no geometry, use a standard physical disk geometry */
1412f3d54fc4Saliguori             cylinders = nb_sectors / (16 * 63);
1413f3d54fc4Saliguori 
1414f3d54fc4Saliguori             if (cylinders > 16383)
1415f3d54fc4Saliguori                 cylinders = 16383;
1416f3d54fc4Saliguori             else if (cylinders < 2)
1417f3d54fc4Saliguori                 cylinders = 2;
1418f3d54fc4Saliguori             *pcyls = cylinders;
1419f3d54fc4Saliguori             *pheads = 16;
1420f3d54fc4Saliguori             *psecs = 63;
1421f3d54fc4Saliguori             if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) {
1422f3d54fc4Saliguori                 if ((*pcyls * *pheads) <= 131072) {
1423f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
1424f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_LARGE);
1425f3d54fc4Saliguori                 } else {
1426f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
1427f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_LBA);
1428f3d54fc4Saliguori                 }
1429f3d54fc4Saliguori             }
1430f3d54fc4Saliguori         }
1431f3d54fc4Saliguori         bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs);
1432f3d54fc4Saliguori     }
1433f3d54fc4Saliguori }
1434f3d54fc4Saliguori 
1435b338082bSbellard void bdrv_set_geometry_hint(BlockDriverState *bs,
1436b338082bSbellard                             int cyls, int heads, int secs)
1437b338082bSbellard {
1438b338082bSbellard     bs->cyls = cyls;
1439b338082bSbellard     bs->heads = heads;
1440b338082bSbellard     bs->secs = secs;
1441b338082bSbellard }
1442b338082bSbellard 
144346d4767dSbellard void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
144446d4767dSbellard {
144546d4767dSbellard     bs->translation = translation;
144646d4767dSbellard }
144746d4767dSbellard 
1448b338082bSbellard void bdrv_get_geometry_hint(BlockDriverState *bs,
1449b338082bSbellard                             int *pcyls, int *pheads, int *psecs)
1450b338082bSbellard {
1451b338082bSbellard     *pcyls = bs->cyls;
1452b338082bSbellard     *pheads = bs->heads;
1453b338082bSbellard     *psecs = bs->secs;
1454b338082bSbellard }
1455b338082bSbellard 
14565bbdbb46SBlue Swirl /* Recognize floppy formats */
14575bbdbb46SBlue Swirl typedef struct FDFormat {
14585bbdbb46SBlue Swirl     FDriveType drive;
14595bbdbb46SBlue Swirl     uint8_t last_sect;
14605bbdbb46SBlue Swirl     uint8_t max_track;
14615bbdbb46SBlue Swirl     uint8_t max_head;
14625bbdbb46SBlue Swirl } FDFormat;
14635bbdbb46SBlue Swirl 
14645bbdbb46SBlue Swirl static const FDFormat fd_formats[] = {
14655bbdbb46SBlue Swirl     /* First entry is default format */
14665bbdbb46SBlue Swirl     /* 1.44 MB 3"1/2 floppy disks */
14675bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 18, 80, 1, },
14685bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 20, 80, 1, },
14695bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 21, 80, 1, },
14705bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 21, 82, 1, },
14715bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 21, 83, 1, },
14725bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 22, 80, 1, },
14735bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 23, 80, 1, },
14745bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 24, 80, 1, },
14755bbdbb46SBlue Swirl     /* 2.88 MB 3"1/2 floppy disks */
14765bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 36, 80, 1, },
14775bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 39, 80, 1, },
14785bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 40, 80, 1, },
14795bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 44, 80, 1, },
14805bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 48, 80, 1, },
14815bbdbb46SBlue Swirl     /* 720 kB 3"1/2 floppy disks */
14825bbdbb46SBlue Swirl     { FDRIVE_DRV_144,  9, 80, 1, },
14835bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 10, 80, 1, },
14845bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 10, 82, 1, },
14855bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 10, 83, 1, },
14865bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 13, 80, 1, },
14875bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 14, 80, 1, },
14885bbdbb46SBlue Swirl     /* 1.2 MB 5"1/4 floppy disks */
14895bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 15, 80, 1, },
14905bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 18, 80, 1, },
14915bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 18, 82, 1, },
14925bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 18, 83, 1, },
14935bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 20, 80, 1, },
14945bbdbb46SBlue Swirl     /* 720 kB 5"1/4 floppy disks */
14955bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  9, 80, 1, },
14965bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 11, 80, 1, },
14975bbdbb46SBlue Swirl     /* 360 kB 5"1/4 floppy disks */
14985bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  9, 40, 1, },
14995bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  9, 40, 0, },
15005bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 10, 41, 1, },
15015bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 10, 42, 1, },
15025bbdbb46SBlue Swirl     /* 320 kB 5"1/4 floppy disks */
15035bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  8, 40, 1, },
15045bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  8, 40, 0, },
15055bbdbb46SBlue Swirl     /* 360 kB must match 5"1/4 better than 3"1/2... */
15065bbdbb46SBlue Swirl     { FDRIVE_DRV_144,  9, 80, 0, },
15075bbdbb46SBlue Swirl     /* end */
15085bbdbb46SBlue Swirl     { FDRIVE_DRV_NONE, -1, -1, 0, },
15095bbdbb46SBlue Swirl };
15105bbdbb46SBlue Swirl 
15115bbdbb46SBlue Swirl void bdrv_get_floppy_geometry_hint(BlockDriverState *bs, int *nb_heads,
15125bbdbb46SBlue Swirl                                    int *max_track, int *last_sect,
15135bbdbb46SBlue Swirl                                    FDriveType drive_in, FDriveType *drive)
15145bbdbb46SBlue Swirl {
15155bbdbb46SBlue Swirl     const FDFormat *parse;
15165bbdbb46SBlue Swirl     uint64_t nb_sectors, size;
15175bbdbb46SBlue Swirl     int i, first_match, match;
15185bbdbb46SBlue Swirl 
15195bbdbb46SBlue Swirl     bdrv_get_geometry_hint(bs, nb_heads, max_track, last_sect);
15205bbdbb46SBlue Swirl     if (*nb_heads != 0 && *max_track != 0 && *last_sect != 0) {
15215bbdbb46SBlue Swirl         /* User defined disk */
15225bbdbb46SBlue Swirl     } else {
15235bbdbb46SBlue Swirl         bdrv_get_geometry(bs, &nb_sectors);
15245bbdbb46SBlue Swirl         match = -1;
15255bbdbb46SBlue Swirl         first_match = -1;
15265bbdbb46SBlue Swirl         for (i = 0; ; i++) {
15275bbdbb46SBlue Swirl             parse = &fd_formats[i];
15285bbdbb46SBlue Swirl             if (parse->drive == FDRIVE_DRV_NONE) {
15295bbdbb46SBlue Swirl                 break;
15305bbdbb46SBlue Swirl             }
15315bbdbb46SBlue Swirl             if (drive_in == parse->drive ||
15325bbdbb46SBlue Swirl                 drive_in == FDRIVE_DRV_NONE) {
15335bbdbb46SBlue Swirl                 size = (parse->max_head + 1) * parse->max_track *
15345bbdbb46SBlue Swirl                     parse->last_sect;
15355bbdbb46SBlue Swirl                 if (nb_sectors == size) {
15365bbdbb46SBlue Swirl                     match = i;
15375bbdbb46SBlue Swirl                     break;
15385bbdbb46SBlue Swirl                 }
15395bbdbb46SBlue Swirl                 if (first_match == -1) {
15405bbdbb46SBlue Swirl                     first_match = i;
15415bbdbb46SBlue Swirl                 }
15425bbdbb46SBlue Swirl             }
15435bbdbb46SBlue Swirl         }
15445bbdbb46SBlue Swirl         if (match == -1) {
15455bbdbb46SBlue Swirl             if (first_match == -1) {
15465bbdbb46SBlue Swirl                 match = 1;
15475bbdbb46SBlue Swirl             } else {
15485bbdbb46SBlue Swirl                 match = first_match;
15495bbdbb46SBlue Swirl             }
15505bbdbb46SBlue Swirl             parse = &fd_formats[match];
15515bbdbb46SBlue Swirl         }
15525bbdbb46SBlue Swirl         *nb_heads = parse->max_head + 1;
15535bbdbb46SBlue Swirl         *max_track = parse->max_track;
15545bbdbb46SBlue Swirl         *last_sect = parse->last_sect;
15555bbdbb46SBlue Swirl         *drive = parse->drive;
15565bbdbb46SBlue Swirl     }
15575bbdbb46SBlue Swirl }
15585bbdbb46SBlue Swirl 
155946d4767dSbellard int bdrv_get_translation_hint(BlockDriverState *bs)
156046d4767dSbellard {
156146d4767dSbellard     return bs->translation;
156246d4767dSbellard }
156346d4767dSbellard 
1564abd7f68dSMarkus Armbruster void bdrv_set_on_error(BlockDriverState *bs, BlockErrorAction on_read_error,
1565abd7f68dSMarkus Armbruster                        BlockErrorAction on_write_error)
1566abd7f68dSMarkus Armbruster {
1567abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
1568abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
1569abd7f68dSMarkus Armbruster }
1570abd7f68dSMarkus Armbruster 
1571abd7f68dSMarkus Armbruster BlockErrorAction bdrv_get_on_error(BlockDriverState *bs, int is_read)
1572abd7f68dSMarkus Armbruster {
1573abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
1574abd7f68dSMarkus Armbruster }
1575abd7f68dSMarkus Armbruster 
15767d0d6950SMarkus Armbruster void bdrv_set_removable(BlockDriverState *bs, int removable)
15777d0d6950SMarkus Armbruster {
15787d0d6950SMarkus Armbruster     bs->removable = removable;
15797d0d6950SMarkus Armbruster     if (removable && bs == bs_snapshots) {
15807d0d6950SMarkus Armbruster         bs_snapshots = NULL;
15817d0d6950SMarkus Armbruster     }
15827d0d6950SMarkus Armbruster }
15837d0d6950SMarkus Armbruster 
1584b338082bSbellard int bdrv_is_removable(BlockDriverState *bs)
1585b338082bSbellard {
1586b338082bSbellard     return bs->removable;
1587b338082bSbellard }
1588b338082bSbellard 
1589b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
1590b338082bSbellard {
1591b338082bSbellard     return bs->read_only;
1592b338082bSbellard }
1593b338082bSbellard 
1594985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
1595985a03b0Sths {
1596985a03b0Sths     return bs->sg;
1597985a03b0Sths }
1598985a03b0Sths 
1599e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
1600e900a7b7SChristoph Hellwig {
1601e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
1602e900a7b7SChristoph Hellwig }
1603e900a7b7SChristoph Hellwig 
160419cb3738Sbellard /* XXX: no longer used */
1605b338082bSbellard void bdrv_set_change_cb(BlockDriverState *bs,
1606db97ee6aSChristoph Hellwig                         void (*change_cb)(void *opaque, int reason),
1607db97ee6aSChristoph Hellwig                         void *opaque)
1608b338082bSbellard {
1609b338082bSbellard     bs->change_cb = change_cb;
1610b338082bSbellard     bs->change_opaque = opaque;
1611b338082bSbellard }
1612b338082bSbellard 
1613ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
1614ea2384d3Sbellard {
1615ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
1616ea2384d3Sbellard         return 1;
1617ea2384d3Sbellard     return bs->encrypted;
1618ea2384d3Sbellard }
1619ea2384d3Sbellard 
1620c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
1621c0f4ce77Saliguori {
1622c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
1623c0f4ce77Saliguori 
1624c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
1625c0f4ce77Saliguori         return 1;
1626c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
1627c0f4ce77Saliguori }
1628c0f4ce77Saliguori 
1629ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
1630ea2384d3Sbellard {
1631ea2384d3Sbellard     int ret;
1632ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
1633ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
1634ea2384d3Sbellard         if (ret < 0)
1635ea2384d3Sbellard             return ret;
1636ea2384d3Sbellard         if (!bs->encrypted)
1637ea2384d3Sbellard             return 0;
1638ea2384d3Sbellard     }
1639fd04a2aeSShahar Havivi     if (!bs->encrypted) {
1640fd04a2aeSShahar Havivi         return -EINVAL;
1641fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
1642fd04a2aeSShahar Havivi         return -ENOMEDIUM;
1643fd04a2aeSShahar Havivi     }
1644c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
1645bb5fc20fSaliguori     if (ret < 0) {
1646bb5fc20fSaliguori         bs->valid_key = 0;
1647bb5fc20fSaliguori     } else if (!bs->valid_key) {
1648bb5fc20fSaliguori         bs->valid_key = 1;
1649bb5fc20fSaliguori         /* call the change callback now, we skipped it on open */
1650bb5fc20fSaliguori         bs->media_changed = 1;
1651bb5fc20fSaliguori         if (bs->change_cb)
1652db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
1653bb5fc20fSaliguori     }
1654c0f4ce77Saliguori     return ret;
1655ea2384d3Sbellard }
1656ea2384d3Sbellard 
1657ea2384d3Sbellard void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
1658ea2384d3Sbellard {
165919cb3738Sbellard     if (!bs->drv) {
1660ea2384d3Sbellard         buf[0] = '\0';
1661ea2384d3Sbellard     } else {
1662ea2384d3Sbellard         pstrcpy(buf, buf_size, bs->drv->format_name);
1663ea2384d3Sbellard     }
1664ea2384d3Sbellard }
1665ea2384d3Sbellard 
1666ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
1667ea2384d3Sbellard                          void *opaque)
1668ea2384d3Sbellard {
1669ea2384d3Sbellard     BlockDriver *drv;
1670ea2384d3Sbellard 
16718a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
1672ea2384d3Sbellard         it(opaque, drv->format_name);
1673ea2384d3Sbellard     }
1674ea2384d3Sbellard }
1675ea2384d3Sbellard 
1676b338082bSbellard BlockDriverState *bdrv_find(const char *name)
1677b338082bSbellard {
1678b338082bSbellard     BlockDriverState *bs;
1679b338082bSbellard 
16801b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
16811b7bdbc1SStefan Hajnoczi         if (!strcmp(name, bs->device_name)) {
1682b338082bSbellard             return bs;
1683b338082bSbellard         }
16841b7bdbc1SStefan Hajnoczi     }
1685b338082bSbellard     return NULL;
1686b338082bSbellard }
1687b338082bSbellard 
16882f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
16892f399b0aSMarkus Armbruster {
16902f399b0aSMarkus Armbruster     if (!bs) {
16912f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
16922f399b0aSMarkus Armbruster     }
16932f399b0aSMarkus Armbruster     return QTAILQ_NEXT(bs, list);
16942f399b0aSMarkus Armbruster }
16952f399b0aSMarkus Armbruster 
169651de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
169781d0912dSbellard {
169881d0912dSbellard     BlockDriverState *bs;
169981d0912dSbellard 
17001b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
170151de9760Saliguori         it(opaque, bs);
170281d0912dSbellard     }
170381d0912dSbellard }
170481d0912dSbellard 
1705ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
1706ea2384d3Sbellard {
1707ea2384d3Sbellard     return bs->device_name;
1708ea2384d3Sbellard }
1709ea2384d3Sbellard 
1710205ef796SKevin Wolf int bdrv_flush(BlockDriverState *bs)
17117a6cba61Spbrook {
1712016f5cf6SAlexander Graf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
1713205ef796SKevin Wolf         return 0;
1714016f5cf6SAlexander Graf     }
1715016f5cf6SAlexander Graf 
1716e7a8a783SKevin Wolf     if (bs->drv && bdrv_has_async_flush(bs->drv) && qemu_in_coroutine()) {
1717e7a8a783SKevin Wolf         return bdrv_co_flush_em(bs);
1718e7a8a783SKevin Wolf     }
1719e7a8a783SKevin Wolf 
1720205ef796SKevin Wolf     if (bs->drv && bs->drv->bdrv_flush) {
1721205ef796SKevin Wolf         return bs->drv->bdrv_flush(bs);
1722205ef796SKevin Wolf     }
1723205ef796SKevin Wolf 
1724205ef796SKevin Wolf     /*
1725205ef796SKevin Wolf      * Some block drivers always operate in either writethrough or unsafe mode
1726205ef796SKevin Wolf      * and don't support bdrv_flush therefore. Usually qemu doesn't know how
1727205ef796SKevin Wolf      * the server works (because the behaviour is hardcoded or depends on
1728205ef796SKevin Wolf      * server-side configuration), so we can't ensure that everything is safe
1729205ef796SKevin Wolf      * on disk. Returning an error doesn't work because that would break guests
1730205ef796SKevin Wolf      * even if the server operates in writethrough mode.
1731205ef796SKevin Wolf      *
1732205ef796SKevin Wolf      * Let's hope the user knows what he's doing.
1733205ef796SKevin Wolf      */
1734205ef796SKevin Wolf     return 0;
17357a6cba61Spbrook }
17367a6cba61Spbrook 
1737c6ca28d6Saliguori void bdrv_flush_all(void)
1738c6ca28d6Saliguori {
1739c6ca28d6Saliguori     BlockDriverState *bs;
1740c6ca28d6Saliguori 
17411b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1742c6ca28d6Saliguori         if (bs->drv && !bdrv_is_read_only(bs) &&
17431b7bdbc1SStefan Hajnoczi             (!bdrv_is_removable(bs) || bdrv_is_inserted(bs))) {
1744c6ca28d6Saliguori             bdrv_flush(bs);
1745c6ca28d6Saliguori         }
17461b7bdbc1SStefan Hajnoczi     }
17471b7bdbc1SStefan Hajnoczi }
1748c6ca28d6Saliguori 
1749f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
1750f2feebbdSKevin Wolf {
1751f2feebbdSKevin Wolf     assert(bs->drv);
1752f2feebbdSKevin Wolf 
1753336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
1754336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
1755f2feebbdSKevin Wolf     }
1756f2feebbdSKevin Wolf 
1757f2feebbdSKevin Wolf     return 1;
1758f2feebbdSKevin Wolf }
1759f2feebbdSKevin Wolf 
1760bb8bf76fSChristoph Hellwig int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
1761bb8bf76fSChristoph Hellwig {
1762bb8bf76fSChristoph Hellwig     if (!bs->drv) {
1763bb8bf76fSChristoph Hellwig         return -ENOMEDIUM;
1764bb8bf76fSChristoph Hellwig     }
1765bb8bf76fSChristoph Hellwig     if (!bs->drv->bdrv_discard) {
1766bb8bf76fSChristoph Hellwig         return 0;
1767bb8bf76fSChristoph Hellwig     }
1768bb8bf76fSChristoph Hellwig     return bs->drv->bdrv_discard(bs, sector_num, nb_sectors);
1769bb8bf76fSChristoph Hellwig }
1770bb8bf76fSChristoph Hellwig 
1771f58c7b35Sths /*
1772f58c7b35Sths  * Returns true iff the specified sector is present in the disk image. Drivers
1773f58c7b35Sths  * not implementing the functionality are assumed to not support backing files,
1774f58c7b35Sths  * hence all their sectors are reported as allocated.
1775f58c7b35Sths  *
1776f58c7b35Sths  * 'pnum' is set to the number of sectors (including and immediately following
1777f58c7b35Sths  * the specified sector) that are known to be in the same
1778f58c7b35Sths  * allocated/unallocated state.
1779f58c7b35Sths  *
1780f58c7b35Sths  * 'nb_sectors' is the max value 'pnum' should be set to.
1781f58c7b35Sths  */
1782f58c7b35Sths int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
1783f58c7b35Sths 	int *pnum)
1784f58c7b35Sths {
1785f58c7b35Sths     int64_t n;
1786f58c7b35Sths     if (!bs->drv->bdrv_is_allocated) {
1787f58c7b35Sths         if (sector_num >= bs->total_sectors) {
1788f58c7b35Sths             *pnum = 0;
1789f58c7b35Sths             return 0;
1790f58c7b35Sths         }
1791f58c7b35Sths         n = bs->total_sectors - sector_num;
1792f58c7b35Sths         *pnum = (n < nb_sectors) ? (n) : (nb_sectors);
1793f58c7b35Sths         return 1;
1794f58c7b35Sths     }
1795f58c7b35Sths     return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum);
1796f58c7b35Sths }
1797f58c7b35Sths 
17982582bfedSLuiz Capitulino void bdrv_mon_event(const BlockDriverState *bdrv,
17992582bfedSLuiz Capitulino                     BlockMonEventAction action, int is_read)
18002582bfedSLuiz Capitulino {
18012582bfedSLuiz Capitulino     QObject *data;
18022582bfedSLuiz Capitulino     const char *action_str;
18032582bfedSLuiz Capitulino 
18042582bfedSLuiz Capitulino     switch (action) {
18052582bfedSLuiz Capitulino     case BDRV_ACTION_REPORT:
18062582bfedSLuiz Capitulino         action_str = "report";
18072582bfedSLuiz Capitulino         break;
18082582bfedSLuiz Capitulino     case BDRV_ACTION_IGNORE:
18092582bfedSLuiz Capitulino         action_str = "ignore";
18102582bfedSLuiz Capitulino         break;
18112582bfedSLuiz Capitulino     case BDRV_ACTION_STOP:
18122582bfedSLuiz Capitulino         action_str = "stop";
18132582bfedSLuiz Capitulino         break;
18142582bfedSLuiz Capitulino     default:
18152582bfedSLuiz Capitulino         abort();
18162582bfedSLuiz Capitulino     }
18172582bfedSLuiz Capitulino 
18182582bfedSLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
18192582bfedSLuiz Capitulino                               bdrv->device_name,
18202582bfedSLuiz Capitulino                               action_str,
18212582bfedSLuiz Capitulino                               is_read ? "read" : "write");
18222582bfedSLuiz Capitulino     monitor_protocol_event(QEVENT_BLOCK_IO_ERROR, data);
18232582bfedSLuiz Capitulino 
18242582bfedSLuiz Capitulino     qobject_decref(data);
18252582bfedSLuiz Capitulino }
18262582bfedSLuiz Capitulino 
1827d15e5465SLuiz Capitulino static void bdrv_print_dict(QObject *obj, void *opaque)
1828b338082bSbellard {
1829d15e5465SLuiz Capitulino     QDict *bs_dict;
1830d15e5465SLuiz Capitulino     Monitor *mon = opaque;
1831b338082bSbellard 
1832d15e5465SLuiz Capitulino     bs_dict = qobject_to_qdict(obj);
1833d15e5465SLuiz Capitulino 
1834d8aeeb31SMarkus Armbruster     monitor_printf(mon, "%s: removable=%d",
1835d15e5465SLuiz Capitulino                         qdict_get_str(bs_dict, "device"),
1836d15e5465SLuiz Capitulino                         qdict_get_bool(bs_dict, "removable"));
1837d15e5465SLuiz Capitulino 
1838d15e5465SLuiz Capitulino     if (qdict_get_bool(bs_dict, "removable")) {
1839d15e5465SLuiz Capitulino         monitor_printf(mon, " locked=%d", qdict_get_bool(bs_dict, "locked"));
1840b338082bSbellard     }
1841d15e5465SLuiz Capitulino 
1842d15e5465SLuiz Capitulino     if (qdict_haskey(bs_dict, "inserted")) {
1843d15e5465SLuiz Capitulino         QDict *qdict = qobject_to_qdict(qdict_get(bs_dict, "inserted"));
1844d15e5465SLuiz Capitulino 
1845376253ecSaliguori         monitor_printf(mon, " file=");
1846d15e5465SLuiz Capitulino         monitor_print_filename(mon, qdict_get_str(qdict, "file"));
1847d15e5465SLuiz Capitulino         if (qdict_haskey(qdict, "backing_file")) {
1848376253ecSaliguori             monitor_printf(mon, " backing_file=");
1849d15e5465SLuiz Capitulino             monitor_print_filename(mon, qdict_get_str(qdict, "backing_file"));
1850fef30743Sths         }
1851d15e5465SLuiz Capitulino         monitor_printf(mon, " ro=%d drv=%s encrypted=%d",
1852d15e5465SLuiz Capitulino                             qdict_get_bool(qdict, "ro"),
1853d15e5465SLuiz Capitulino                             qdict_get_str(qdict, "drv"),
1854d15e5465SLuiz Capitulino                             qdict_get_bool(qdict, "encrypted"));
1855b338082bSbellard     } else {
1856376253ecSaliguori         monitor_printf(mon, " [not inserted]");
1857b338082bSbellard     }
1858d15e5465SLuiz Capitulino 
1859376253ecSaliguori     monitor_printf(mon, "\n");
1860b338082bSbellard }
1861d15e5465SLuiz Capitulino 
1862d15e5465SLuiz Capitulino void bdrv_info_print(Monitor *mon, const QObject *data)
1863d15e5465SLuiz Capitulino {
1864d15e5465SLuiz Capitulino     qlist_iter(qobject_to_qlist(data), bdrv_print_dict, mon);
1865d15e5465SLuiz Capitulino }
1866d15e5465SLuiz Capitulino 
1867d15e5465SLuiz Capitulino void bdrv_info(Monitor *mon, QObject **ret_data)
1868d15e5465SLuiz Capitulino {
1869d15e5465SLuiz Capitulino     QList *bs_list;
1870d15e5465SLuiz Capitulino     BlockDriverState *bs;
1871d15e5465SLuiz Capitulino 
1872d15e5465SLuiz Capitulino     bs_list = qlist_new();
1873d15e5465SLuiz Capitulino 
18741b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1875d15e5465SLuiz Capitulino         QObject *bs_obj;
1876d15e5465SLuiz Capitulino 
1877d8aeeb31SMarkus Armbruster         bs_obj = qobject_from_jsonf("{ 'device': %s, 'type': 'unknown', "
1878d15e5465SLuiz Capitulino                                     "'removable': %i, 'locked': %i }",
1879d8aeeb31SMarkus Armbruster                                     bs->device_name, bs->removable,
1880d15e5465SLuiz Capitulino                                     bs->locked);
1881d15e5465SLuiz Capitulino 
1882d15e5465SLuiz Capitulino         if (bs->drv) {
1883d15e5465SLuiz Capitulino             QObject *obj;
1884d15e5465SLuiz Capitulino             QDict *bs_dict = qobject_to_qdict(bs_obj);
1885d15e5465SLuiz Capitulino 
1886d15e5465SLuiz Capitulino             obj = qobject_from_jsonf("{ 'file': %s, 'ro': %i, 'drv': %s, "
1887d15e5465SLuiz Capitulino                                      "'encrypted': %i }",
1888d15e5465SLuiz Capitulino                                      bs->filename, bs->read_only,
1889d15e5465SLuiz Capitulino                                      bs->drv->format_name,
1890d15e5465SLuiz Capitulino                                      bdrv_is_encrypted(bs));
1891d15e5465SLuiz Capitulino             if (bs->backing_file[0] != '\0') {
1892d15e5465SLuiz Capitulino                 QDict *qdict = qobject_to_qdict(obj);
1893d15e5465SLuiz Capitulino                 qdict_put(qdict, "backing_file",
1894d15e5465SLuiz Capitulino                           qstring_from_str(bs->backing_file));
1895d15e5465SLuiz Capitulino             }
1896d15e5465SLuiz Capitulino 
1897d15e5465SLuiz Capitulino             qdict_put_obj(bs_dict, "inserted", obj);
1898d15e5465SLuiz Capitulino         }
1899d15e5465SLuiz Capitulino         qlist_append_obj(bs_list, bs_obj);
1900d15e5465SLuiz Capitulino     }
1901d15e5465SLuiz Capitulino 
1902d15e5465SLuiz Capitulino     *ret_data = QOBJECT(bs_list);
1903b338082bSbellard }
1904a36e69ddSths 
1905218a536aSLuiz Capitulino static void bdrv_stats_iter(QObject *data, void *opaque)
1906a36e69ddSths {
1907218a536aSLuiz Capitulino     QDict *qdict;
1908218a536aSLuiz Capitulino     Monitor *mon = opaque;
1909218a536aSLuiz Capitulino 
1910218a536aSLuiz Capitulino     qdict = qobject_to_qdict(data);
1911218a536aSLuiz Capitulino     monitor_printf(mon, "%s:", qdict_get_str(qdict, "device"));
1912218a536aSLuiz Capitulino 
1913218a536aSLuiz Capitulino     qdict = qobject_to_qdict(qdict_get(qdict, "stats"));
1914218a536aSLuiz Capitulino     monitor_printf(mon, " rd_bytes=%" PRId64
1915218a536aSLuiz Capitulino                         " wr_bytes=%" PRId64
1916218a536aSLuiz Capitulino                         " rd_operations=%" PRId64
1917218a536aSLuiz Capitulino                         " wr_operations=%" PRId64
1918e8045d67SChristoph Hellwig                         " flush_operations=%" PRId64
1919*c488c7f6SChristoph Hellwig                         " wr_total_time_ns=%" PRId64
1920*c488c7f6SChristoph Hellwig                         " rd_total_time_ns=%" PRId64
1921*c488c7f6SChristoph Hellwig                         " flush_total_time_ns=%" PRId64
1922218a536aSLuiz Capitulino                         "\n",
1923218a536aSLuiz Capitulino                         qdict_get_int(qdict, "rd_bytes"),
1924218a536aSLuiz Capitulino                         qdict_get_int(qdict, "wr_bytes"),
1925218a536aSLuiz Capitulino                         qdict_get_int(qdict, "rd_operations"),
1926e8045d67SChristoph Hellwig                         qdict_get_int(qdict, "wr_operations"),
1927*c488c7f6SChristoph Hellwig                         qdict_get_int(qdict, "flush_operations"),
1928*c488c7f6SChristoph Hellwig                         qdict_get_int(qdict, "wr_total_time_ns"),
1929*c488c7f6SChristoph Hellwig                         qdict_get_int(qdict, "rd_total_time_ns"),
1930*c488c7f6SChristoph Hellwig                         qdict_get_int(qdict, "flush_total_time_ns"));
1931218a536aSLuiz Capitulino }
1932218a536aSLuiz Capitulino 
1933218a536aSLuiz Capitulino void bdrv_stats_print(Monitor *mon, const QObject *data)
1934218a536aSLuiz Capitulino {
1935218a536aSLuiz Capitulino     qlist_iter(qobject_to_qlist(data), bdrv_stats_iter, mon);
1936218a536aSLuiz Capitulino }
1937218a536aSLuiz Capitulino 
1938294cc35fSKevin Wolf static QObject* bdrv_info_stats_bs(BlockDriverState *bs)
1939294cc35fSKevin Wolf {
1940294cc35fSKevin Wolf     QObject *res;
1941294cc35fSKevin Wolf     QDict *dict;
1942294cc35fSKevin Wolf 
1943294cc35fSKevin Wolf     res = qobject_from_jsonf("{ 'stats': {"
1944294cc35fSKevin Wolf                              "'rd_bytes': %" PRId64 ","
1945294cc35fSKevin Wolf                              "'wr_bytes': %" PRId64 ","
1946294cc35fSKevin Wolf                              "'rd_operations': %" PRId64 ","
1947294cc35fSKevin Wolf                              "'wr_operations': %" PRId64 ","
1948e8045d67SChristoph Hellwig                              "'wr_highest_offset': %" PRId64 ","
1949*c488c7f6SChristoph Hellwig                              "'flush_operations': %" PRId64 ","
1950*c488c7f6SChristoph Hellwig                              "'wr_total_time_ns': %" PRId64 ","
1951*c488c7f6SChristoph Hellwig                              "'rd_total_time_ns': %" PRId64 ","
1952*c488c7f6SChristoph Hellwig                              "'flush_total_time_ns': %" PRId64
1953294cc35fSKevin Wolf                              "} }",
1954a597e79cSChristoph Hellwig                              bs->nr_bytes[BDRV_ACCT_READ],
1955a597e79cSChristoph Hellwig                              bs->nr_bytes[BDRV_ACCT_WRITE],
1956a597e79cSChristoph Hellwig                              bs->nr_ops[BDRV_ACCT_READ],
1957a597e79cSChristoph Hellwig                              bs->nr_ops[BDRV_ACCT_WRITE],
19585ffbbc67SBlue Swirl                              bs->wr_highest_sector *
1959e8045d67SChristoph Hellwig                              (uint64_t)BDRV_SECTOR_SIZE,
1960*c488c7f6SChristoph Hellwig                              bs->nr_ops[BDRV_ACCT_FLUSH],
1961*c488c7f6SChristoph Hellwig                              bs->total_time_ns[BDRV_ACCT_WRITE],
1962*c488c7f6SChristoph Hellwig                              bs->total_time_ns[BDRV_ACCT_READ],
1963*c488c7f6SChristoph Hellwig                              bs->total_time_ns[BDRV_ACCT_FLUSH]);
1964294cc35fSKevin Wolf     dict  = qobject_to_qdict(res);
1965294cc35fSKevin Wolf 
1966294cc35fSKevin Wolf     if (*bs->device_name) {
1967294cc35fSKevin Wolf         qdict_put(dict, "device", qstring_from_str(bs->device_name));
1968294cc35fSKevin Wolf     }
1969294cc35fSKevin Wolf 
1970294cc35fSKevin Wolf     if (bs->file) {
1971294cc35fSKevin Wolf         QObject *parent = bdrv_info_stats_bs(bs->file);
1972294cc35fSKevin Wolf         qdict_put_obj(dict, "parent", parent);
1973294cc35fSKevin Wolf     }
1974294cc35fSKevin Wolf 
1975294cc35fSKevin Wolf     return res;
1976294cc35fSKevin Wolf }
1977294cc35fSKevin Wolf 
1978218a536aSLuiz Capitulino void bdrv_info_stats(Monitor *mon, QObject **ret_data)
1979218a536aSLuiz Capitulino {
1980218a536aSLuiz Capitulino     QObject *obj;
1981218a536aSLuiz Capitulino     QList *devices;
1982a36e69ddSths     BlockDriverState *bs;
1983a36e69ddSths 
1984218a536aSLuiz Capitulino     devices = qlist_new();
1985218a536aSLuiz Capitulino 
19861b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1987294cc35fSKevin Wolf         obj = bdrv_info_stats_bs(bs);
1988218a536aSLuiz Capitulino         qlist_append_obj(devices, obj);
1989a36e69ddSths     }
1990218a536aSLuiz Capitulino 
1991218a536aSLuiz Capitulino     *ret_data = QOBJECT(devices);
1992a36e69ddSths }
1993ea2384d3Sbellard 
1994045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
1995045df330Saliguori {
1996045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
1997045df330Saliguori         return bs->backing_file;
1998045df330Saliguori     else if (bs->encrypted)
1999045df330Saliguori         return bs->filename;
2000045df330Saliguori     else
2001045df330Saliguori         return NULL;
2002045df330Saliguori }
2003045df330Saliguori 
200483f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
200583f64091Sbellard                                char *filename, int filename_size)
200683f64091Sbellard {
2007b783e409SKevin Wolf     if (!bs->backing_file) {
200883f64091Sbellard         pstrcpy(filename, filename_size, "");
200983f64091Sbellard     } else {
201083f64091Sbellard         pstrcpy(filename, filename_size, bs->backing_file);
201183f64091Sbellard     }
201283f64091Sbellard }
201383f64091Sbellard 
2014faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
2015faea38e7Sbellard                           const uint8_t *buf, int nb_sectors)
2016faea38e7Sbellard {
2017faea38e7Sbellard     BlockDriver *drv = bs->drv;
2018faea38e7Sbellard     if (!drv)
201919cb3738Sbellard         return -ENOMEDIUM;
2020faea38e7Sbellard     if (!drv->bdrv_write_compressed)
2021faea38e7Sbellard         return -ENOTSUP;
2022fbb7b4e0SKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors))
2023fbb7b4e0SKevin Wolf         return -EIO;
20247cd1e32aSlirans@il.ibm.com 
2025c6d22830SJan Kiszka     if (bs->dirty_bitmap) {
20267cd1e32aSlirans@il.ibm.com         set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
20277cd1e32aSlirans@il.ibm.com     }
20287cd1e32aSlirans@il.ibm.com 
2029faea38e7Sbellard     return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
2030faea38e7Sbellard }
2031faea38e7Sbellard 
2032faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
2033faea38e7Sbellard {
2034faea38e7Sbellard     BlockDriver *drv = bs->drv;
2035faea38e7Sbellard     if (!drv)
203619cb3738Sbellard         return -ENOMEDIUM;
2037faea38e7Sbellard     if (!drv->bdrv_get_info)
2038faea38e7Sbellard         return -ENOTSUP;
2039faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
2040faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
2041faea38e7Sbellard }
2042faea38e7Sbellard 
204345566e9cSChristoph Hellwig int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
204445566e9cSChristoph Hellwig                       int64_t pos, int size)
2045178e08a5Saliguori {
2046178e08a5Saliguori     BlockDriver *drv = bs->drv;
2047178e08a5Saliguori     if (!drv)
2048178e08a5Saliguori         return -ENOMEDIUM;
20497cdb1f6dSMORITA Kazutaka     if (drv->bdrv_save_vmstate)
205045566e9cSChristoph Hellwig         return drv->bdrv_save_vmstate(bs, buf, pos, size);
20517cdb1f6dSMORITA Kazutaka     if (bs->file)
20527cdb1f6dSMORITA Kazutaka         return bdrv_save_vmstate(bs->file, buf, pos, size);
20537cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2054178e08a5Saliguori }
2055178e08a5Saliguori 
205645566e9cSChristoph Hellwig int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
205745566e9cSChristoph Hellwig                       int64_t pos, int size)
2058178e08a5Saliguori {
2059178e08a5Saliguori     BlockDriver *drv = bs->drv;
2060178e08a5Saliguori     if (!drv)
2061178e08a5Saliguori         return -ENOMEDIUM;
20627cdb1f6dSMORITA Kazutaka     if (drv->bdrv_load_vmstate)
206345566e9cSChristoph Hellwig         return drv->bdrv_load_vmstate(bs, buf, pos, size);
20647cdb1f6dSMORITA Kazutaka     if (bs->file)
20657cdb1f6dSMORITA Kazutaka         return bdrv_load_vmstate(bs->file, buf, pos, size);
20667cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2067178e08a5Saliguori }
2068178e08a5Saliguori 
20698b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
20708b9b0cc2SKevin Wolf {
20718b9b0cc2SKevin Wolf     BlockDriver *drv = bs->drv;
20728b9b0cc2SKevin Wolf 
20738b9b0cc2SKevin Wolf     if (!drv || !drv->bdrv_debug_event) {
20748b9b0cc2SKevin Wolf         return;
20758b9b0cc2SKevin Wolf     }
20768b9b0cc2SKevin Wolf 
20778b9b0cc2SKevin Wolf     return drv->bdrv_debug_event(bs, event);
20788b9b0cc2SKevin Wolf 
20798b9b0cc2SKevin Wolf }
20808b9b0cc2SKevin Wolf 
2081faea38e7Sbellard /**************************************************************/
2082faea38e7Sbellard /* handling of snapshots */
2083faea38e7Sbellard 
2084feeee5acSMiguel Di Ciurcio Filho int bdrv_can_snapshot(BlockDriverState *bs)
2085feeee5acSMiguel Di Ciurcio Filho {
2086feeee5acSMiguel Di Ciurcio Filho     BlockDriver *drv = bs->drv;
2087feeee5acSMiguel Di Ciurcio Filho     if (!drv || bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
2088feeee5acSMiguel Di Ciurcio Filho         return 0;
2089feeee5acSMiguel Di Ciurcio Filho     }
2090feeee5acSMiguel Di Ciurcio Filho 
2091feeee5acSMiguel Di Ciurcio Filho     if (!drv->bdrv_snapshot_create) {
2092feeee5acSMiguel Di Ciurcio Filho         if (bs->file != NULL) {
2093feeee5acSMiguel Di Ciurcio Filho             return bdrv_can_snapshot(bs->file);
2094feeee5acSMiguel Di Ciurcio Filho         }
2095feeee5acSMiguel Di Ciurcio Filho         return 0;
2096feeee5acSMiguel Di Ciurcio Filho     }
2097feeee5acSMiguel Di Ciurcio Filho 
2098feeee5acSMiguel Di Ciurcio Filho     return 1;
2099feeee5acSMiguel Di Ciurcio Filho }
2100feeee5acSMiguel Di Ciurcio Filho 
2101199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
2102199630b6SBlue Swirl {
2103199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
2104199630b6SBlue Swirl }
2105199630b6SBlue Swirl 
2106f9092b10SMarkus Armbruster BlockDriverState *bdrv_snapshots(void)
2107f9092b10SMarkus Armbruster {
2108f9092b10SMarkus Armbruster     BlockDriverState *bs;
2109f9092b10SMarkus Armbruster 
21103ac906f7SMarkus Armbruster     if (bs_snapshots) {
2111f9092b10SMarkus Armbruster         return bs_snapshots;
21123ac906f7SMarkus Armbruster     }
2113f9092b10SMarkus Armbruster 
2114f9092b10SMarkus Armbruster     bs = NULL;
2115f9092b10SMarkus Armbruster     while ((bs = bdrv_next(bs))) {
2116f9092b10SMarkus Armbruster         if (bdrv_can_snapshot(bs)) {
21173ac906f7SMarkus Armbruster             bs_snapshots = bs;
21183ac906f7SMarkus Armbruster             return bs;
2119f9092b10SMarkus Armbruster         }
2120f9092b10SMarkus Armbruster     }
2121f9092b10SMarkus Armbruster     return NULL;
2122f9092b10SMarkus Armbruster }
2123f9092b10SMarkus Armbruster 
2124faea38e7Sbellard int bdrv_snapshot_create(BlockDriverState *bs,
2125faea38e7Sbellard                          QEMUSnapshotInfo *sn_info)
2126faea38e7Sbellard {
2127faea38e7Sbellard     BlockDriver *drv = bs->drv;
2128faea38e7Sbellard     if (!drv)
212919cb3738Sbellard         return -ENOMEDIUM;
21307cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_create)
2131faea38e7Sbellard         return drv->bdrv_snapshot_create(bs, sn_info);
21327cdb1f6dSMORITA Kazutaka     if (bs->file)
21337cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_create(bs->file, sn_info);
21347cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2135faea38e7Sbellard }
2136faea38e7Sbellard 
2137faea38e7Sbellard int bdrv_snapshot_goto(BlockDriverState *bs,
2138faea38e7Sbellard                        const char *snapshot_id)
2139faea38e7Sbellard {
2140faea38e7Sbellard     BlockDriver *drv = bs->drv;
21417cdb1f6dSMORITA Kazutaka     int ret, open_ret;
21427cdb1f6dSMORITA Kazutaka 
2143faea38e7Sbellard     if (!drv)
214419cb3738Sbellard         return -ENOMEDIUM;
21457cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_goto)
2146faea38e7Sbellard         return drv->bdrv_snapshot_goto(bs, snapshot_id);
21477cdb1f6dSMORITA Kazutaka 
21487cdb1f6dSMORITA Kazutaka     if (bs->file) {
21497cdb1f6dSMORITA Kazutaka         drv->bdrv_close(bs);
21507cdb1f6dSMORITA Kazutaka         ret = bdrv_snapshot_goto(bs->file, snapshot_id);
21517cdb1f6dSMORITA Kazutaka         open_ret = drv->bdrv_open(bs, bs->open_flags);
21527cdb1f6dSMORITA Kazutaka         if (open_ret < 0) {
21537cdb1f6dSMORITA Kazutaka             bdrv_delete(bs->file);
21547cdb1f6dSMORITA Kazutaka             bs->drv = NULL;
21557cdb1f6dSMORITA Kazutaka             return open_ret;
21567cdb1f6dSMORITA Kazutaka         }
21577cdb1f6dSMORITA Kazutaka         return ret;
21587cdb1f6dSMORITA Kazutaka     }
21597cdb1f6dSMORITA Kazutaka 
21607cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2161faea38e7Sbellard }
2162faea38e7Sbellard 
2163faea38e7Sbellard int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
2164faea38e7Sbellard {
2165faea38e7Sbellard     BlockDriver *drv = bs->drv;
2166faea38e7Sbellard     if (!drv)
216719cb3738Sbellard         return -ENOMEDIUM;
21687cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_delete)
2169faea38e7Sbellard         return drv->bdrv_snapshot_delete(bs, snapshot_id);
21707cdb1f6dSMORITA Kazutaka     if (bs->file)
21717cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_delete(bs->file, snapshot_id);
21727cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2173faea38e7Sbellard }
2174faea38e7Sbellard 
2175faea38e7Sbellard int bdrv_snapshot_list(BlockDriverState *bs,
2176faea38e7Sbellard                        QEMUSnapshotInfo **psn_info)
2177faea38e7Sbellard {
2178faea38e7Sbellard     BlockDriver *drv = bs->drv;
2179faea38e7Sbellard     if (!drv)
218019cb3738Sbellard         return -ENOMEDIUM;
21817cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_list)
2182faea38e7Sbellard         return drv->bdrv_snapshot_list(bs, psn_info);
21837cdb1f6dSMORITA Kazutaka     if (bs->file)
21847cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_list(bs->file, psn_info);
21857cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2186faea38e7Sbellard }
2187faea38e7Sbellard 
218851ef6727Sedison int bdrv_snapshot_load_tmp(BlockDriverState *bs,
218951ef6727Sedison         const char *snapshot_name)
219051ef6727Sedison {
219151ef6727Sedison     BlockDriver *drv = bs->drv;
219251ef6727Sedison     if (!drv) {
219351ef6727Sedison         return -ENOMEDIUM;
219451ef6727Sedison     }
219551ef6727Sedison     if (!bs->read_only) {
219651ef6727Sedison         return -EINVAL;
219751ef6727Sedison     }
219851ef6727Sedison     if (drv->bdrv_snapshot_load_tmp) {
219951ef6727Sedison         return drv->bdrv_snapshot_load_tmp(bs, snapshot_name);
220051ef6727Sedison     }
220151ef6727Sedison     return -ENOTSUP;
220251ef6727Sedison }
220351ef6727Sedison 
2204faea38e7Sbellard #define NB_SUFFIXES 4
2205faea38e7Sbellard 
2206faea38e7Sbellard char *get_human_readable_size(char *buf, int buf_size, int64_t size)
2207faea38e7Sbellard {
2208faea38e7Sbellard     static const char suffixes[NB_SUFFIXES] = "KMGT";
2209faea38e7Sbellard     int64_t base;
2210faea38e7Sbellard     int i;
2211faea38e7Sbellard 
2212faea38e7Sbellard     if (size <= 999) {
2213faea38e7Sbellard         snprintf(buf, buf_size, "%" PRId64, size);
2214faea38e7Sbellard     } else {
2215faea38e7Sbellard         base = 1024;
2216faea38e7Sbellard         for(i = 0; i < NB_SUFFIXES; i++) {
2217faea38e7Sbellard             if (size < (10 * base)) {
2218faea38e7Sbellard                 snprintf(buf, buf_size, "%0.1f%c",
2219faea38e7Sbellard                          (double)size / base,
2220faea38e7Sbellard                          suffixes[i]);
2221faea38e7Sbellard                 break;
2222faea38e7Sbellard             } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
2223faea38e7Sbellard                 snprintf(buf, buf_size, "%" PRId64 "%c",
2224faea38e7Sbellard                          ((size + (base >> 1)) / base),
2225faea38e7Sbellard                          suffixes[i]);
2226faea38e7Sbellard                 break;
2227faea38e7Sbellard             }
2228faea38e7Sbellard             base = base * 1024;
2229faea38e7Sbellard         }
2230faea38e7Sbellard     }
2231faea38e7Sbellard     return buf;
2232faea38e7Sbellard }
2233faea38e7Sbellard 
2234faea38e7Sbellard char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
2235faea38e7Sbellard {
2236faea38e7Sbellard     char buf1[128], date_buf[128], clock_buf[128];
22373b9f94e1Sbellard #ifdef _WIN32
22383b9f94e1Sbellard     struct tm *ptm;
22393b9f94e1Sbellard #else
2240faea38e7Sbellard     struct tm tm;
22413b9f94e1Sbellard #endif
2242faea38e7Sbellard     time_t ti;
2243faea38e7Sbellard     int64_t secs;
2244faea38e7Sbellard 
2245faea38e7Sbellard     if (!sn) {
2246faea38e7Sbellard         snprintf(buf, buf_size,
2247faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
2248faea38e7Sbellard                  "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
2249faea38e7Sbellard     } else {
2250faea38e7Sbellard         ti = sn->date_sec;
22513b9f94e1Sbellard #ifdef _WIN32
22523b9f94e1Sbellard         ptm = localtime(&ti);
22533b9f94e1Sbellard         strftime(date_buf, sizeof(date_buf),
22543b9f94e1Sbellard                  "%Y-%m-%d %H:%M:%S", ptm);
22553b9f94e1Sbellard #else
2256faea38e7Sbellard         localtime_r(&ti, &tm);
2257faea38e7Sbellard         strftime(date_buf, sizeof(date_buf),
2258faea38e7Sbellard                  "%Y-%m-%d %H:%M:%S", &tm);
22593b9f94e1Sbellard #endif
2260faea38e7Sbellard         secs = sn->vm_clock_nsec / 1000000000;
2261faea38e7Sbellard         snprintf(clock_buf, sizeof(clock_buf),
2262faea38e7Sbellard                  "%02d:%02d:%02d.%03d",
2263faea38e7Sbellard                  (int)(secs / 3600),
2264faea38e7Sbellard                  (int)((secs / 60) % 60),
2265faea38e7Sbellard                  (int)(secs % 60),
2266faea38e7Sbellard                  (int)((sn->vm_clock_nsec / 1000000) % 1000));
2267faea38e7Sbellard         snprintf(buf, buf_size,
2268faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
2269faea38e7Sbellard                  sn->id_str, sn->name,
2270faea38e7Sbellard                  get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
2271faea38e7Sbellard                  date_buf,
2272faea38e7Sbellard                  clock_buf);
2273faea38e7Sbellard     }
2274faea38e7Sbellard     return buf;
2275faea38e7Sbellard }
2276faea38e7Sbellard 
2277ea2384d3Sbellard /**************************************************************/
227883f64091Sbellard /* async I/Os */
2279ea2384d3Sbellard 
22803b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
2281f141eafeSaliguori                                  QEMUIOVector *qiov, int nb_sectors,
228283f64091Sbellard                                  BlockDriverCompletionFunc *cb, void *opaque)
2283ea2384d3Sbellard {
228483f64091Sbellard     BlockDriver *drv = bs->drv;
2285ea2384d3Sbellard 
2286bbf0a440SStefan Hajnoczi     trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
2287bbf0a440SStefan Hajnoczi 
228819cb3738Sbellard     if (!drv)
2289ce1a14dcSpbrook         return NULL;
229071d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
229171d0770cSaliguori         return NULL;
229283f64091Sbellard 
2293a597e79cSChristoph Hellwig     return drv->bdrv_aio_readv(bs, sector_num, qiov, nb_sectors,
2294f141eafeSaliguori                                cb, opaque);
229583f64091Sbellard }
229683f64091Sbellard 
22974dcafbb1SMarcelo Tosatti typedef struct BlockCompleteData {
22984dcafbb1SMarcelo Tosatti     BlockDriverCompletionFunc *cb;
22994dcafbb1SMarcelo Tosatti     void *opaque;
23004dcafbb1SMarcelo Tosatti     BlockDriverState *bs;
23014dcafbb1SMarcelo Tosatti     int64_t sector_num;
23024dcafbb1SMarcelo Tosatti     int nb_sectors;
23034dcafbb1SMarcelo Tosatti } BlockCompleteData;
23044dcafbb1SMarcelo Tosatti 
23054dcafbb1SMarcelo Tosatti static void block_complete_cb(void *opaque, int ret)
23064dcafbb1SMarcelo Tosatti {
23074dcafbb1SMarcelo Tosatti     BlockCompleteData *b = opaque;
23084dcafbb1SMarcelo Tosatti 
23094dcafbb1SMarcelo Tosatti     if (b->bs->dirty_bitmap) {
23104dcafbb1SMarcelo Tosatti         set_dirty_bitmap(b->bs, b->sector_num, b->nb_sectors, 1);
23114dcafbb1SMarcelo Tosatti     }
23124dcafbb1SMarcelo Tosatti     b->cb(b->opaque, ret);
23137267c094SAnthony Liguori     g_free(b);
23144dcafbb1SMarcelo Tosatti }
23154dcafbb1SMarcelo Tosatti 
23164dcafbb1SMarcelo Tosatti static BlockCompleteData *blk_dirty_cb_alloc(BlockDriverState *bs,
23174dcafbb1SMarcelo Tosatti                                              int64_t sector_num,
23184dcafbb1SMarcelo Tosatti                                              int nb_sectors,
23194dcafbb1SMarcelo Tosatti                                              BlockDriverCompletionFunc *cb,
23204dcafbb1SMarcelo Tosatti                                              void *opaque)
23214dcafbb1SMarcelo Tosatti {
23227267c094SAnthony Liguori     BlockCompleteData *blkdata = g_malloc0(sizeof(BlockCompleteData));
23234dcafbb1SMarcelo Tosatti 
23244dcafbb1SMarcelo Tosatti     blkdata->bs = bs;
23254dcafbb1SMarcelo Tosatti     blkdata->cb = cb;
23264dcafbb1SMarcelo Tosatti     blkdata->opaque = opaque;
23274dcafbb1SMarcelo Tosatti     blkdata->sector_num = sector_num;
23284dcafbb1SMarcelo Tosatti     blkdata->nb_sectors = nb_sectors;
23294dcafbb1SMarcelo Tosatti 
23304dcafbb1SMarcelo Tosatti     return blkdata;
23314dcafbb1SMarcelo Tosatti }
23324dcafbb1SMarcelo Tosatti 
2333f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
2334f141eafeSaliguori                                   QEMUIOVector *qiov, int nb_sectors,
233583f64091Sbellard                                   BlockDriverCompletionFunc *cb, void *opaque)
23367674e7bfSbellard {
233783f64091Sbellard     BlockDriver *drv = bs->drv;
2338a36e69ddSths     BlockDriverAIOCB *ret;
23394dcafbb1SMarcelo Tosatti     BlockCompleteData *blk_cb_data;
234083f64091Sbellard 
2341bbf0a440SStefan Hajnoczi     trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
2342bbf0a440SStefan Hajnoczi 
234319cb3738Sbellard     if (!drv)
2344ce1a14dcSpbrook         return NULL;
234583f64091Sbellard     if (bs->read_only)
2346ce1a14dcSpbrook         return NULL;
234771d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
234871d0770cSaliguori         return NULL;
234983f64091Sbellard 
2350c6d22830SJan Kiszka     if (bs->dirty_bitmap) {
23514dcafbb1SMarcelo Tosatti         blk_cb_data = blk_dirty_cb_alloc(bs, sector_num, nb_sectors, cb,
23524dcafbb1SMarcelo Tosatti                                          opaque);
23534dcafbb1SMarcelo Tosatti         cb = &block_complete_cb;
23544dcafbb1SMarcelo Tosatti         opaque = blk_cb_data;
23557cd1e32aSlirans@il.ibm.com     }
23567cd1e32aSlirans@il.ibm.com 
2357f141eafeSaliguori     ret = drv->bdrv_aio_writev(bs, sector_num, qiov, nb_sectors,
2358f141eafeSaliguori                                cb, opaque);
2359a36e69ddSths 
2360a36e69ddSths     if (ret) {
2361294cc35fSKevin Wolf         if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
2362294cc35fSKevin Wolf             bs->wr_highest_sector = sector_num + nb_sectors - 1;
2363294cc35fSKevin Wolf         }
2364a36e69ddSths     }
2365a36e69ddSths 
2366a36e69ddSths     return ret;
236783f64091Sbellard }
236883f64091Sbellard 
236940b4f539SKevin Wolf 
237040b4f539SKevin Wolf typedef struct MultiwriteCB {
237140b4f539SKevin Wolf     int error;
237240b4f539SKevin Wolf     int num_requests;
237340b4f539SKevin Wolf     int num_callbacks;
237440b4f539SKevin Wolf     struct {
237540b4f539SKevin Wolf         BlockDriverCompletionFunc *cb;
237640b4f539SKevin Wolf         void *opaque;
237740b4f539SKevin Wolf         QEMUIOVector *free_qiov;
237840b4f539SKevin Wolf         void *free_buf;
237940b4f539SKevin Wolf     } callbacks[];
238040b4f539SKevin Wolf } MultiwriteCB;
238140b4f539SKevin Wolf 
238240b4f539SKevin Wolf static void multiwrite_user_cb(MultiwriteCB *mcb)
238340b4f539SKevin Wolf {
238440b4f539SKevin Wolf     int i;
238540b4f539SKevin Wolf 
238640b4f539SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
238740b4f539SKevin Wolf         mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
23881e1ea48dSStefan Hajnoczi         if (mcb->callbacks[i].free_qiov) {
23891e1ea48dSStefan Hajnoczi             qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
23901e1ea48dSStefan Hajnoczi         }
23917267c094SAnthony Liguori         g_free(mcb->callbacks[i].free_qiov);
2392f8a83245SHerve Poussineau         qemu_vfree(mcb->callbacks[i].free_buf);
239340b4f539SKevin Wolf     }
239440b4f539SKevin Wolf }
239540b4f539SKevin Wolf 
239640b4f539SKevin Wolf static void multiwrite_cb(void *opaque, int ret)
239740b4f539SKevin Wolf {
239840b4f539SKevin Wolf     MultiwriteCB *mcb = opaque;
239940b4f539SKevin Wolf 
24006d519a5fSStefan Hajnoczi     trace_multiwrite_cb(mcb, ret);
24016d519a5fSStefan Hajnoczi 
2402cb6d3ca0SKevin Wolf     if (ret < 0 && !mcb->error) {
240340b4f539SKevin Wolf         mcb->error = ret;
240440b4f539SKevin Wolf     }
240540b4f539SKevin Wolf 
240640b4f539SKevin Wolf     mcb->num_requests--;
240740b4f539SKevin Wolf     if (mcb->num_requests == 0) {
240840b4f539SKevin Wolf         multiwrite_user_cb(mcb);
24097267c094SAnthony Liguori         g_free(mcb);
241040b4f539SKevin Wolf     }
241140b4f539SKevin Wolf }
241240b4f539SKevin Wolf 
241340b4f539SKevin Wolf static int multiwrite_req_compare(const void *a, const void *b)
241440b4f539SKevin Wolf {
241577be4366SChristoph Hellwig     const BlockRequest *req1 = a, *req2 = b;
241677be4366SChristoph Hellwig 
241777be4366SChristoph Hellwig     /*
241877be4366SChristoph Hellwig      * Note that we can't simply subtract req2->sector from req1->sector
241977be4366SChristoph Hellwig      * here as that could overflow the return value.
242077be4366SChristoph Hellwig      */
242177be4366SChristoph Hellwig     if (req1->sector > req2->sector) {
242277be4366SChristoph Hellwig         return 1;
242377be4366SChristoph Hellwig     } else if (req1->sector < req2->sector) {
242477be4366SChristoph Hellwig         return -1;
242577be4366SChristoph Hellwig     } else {
242677be4366SChristoph Hellwig         return 0;
242777be4366SChristoph Hellwig     }
242840b4f539SKevin Wolf }
242940b4f539SKevin Wolf 
243040b4f539SKevin Wolf /*
243140b4f539SKevin Wolf  * Takes a bunch of requests and tries to merge them. Returns the number of
243240b4f539SKevin Wolf  * requests that remain after merging.
243340b4f539SKevin Wolf  */
243440b4f539SKevin Wolf static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
243540b4f539SKevin Wolf     int num_reqs, MultiwriteCB *mcb)
243640b4f539SKevin Wolf {
243740b4f539SKevin Wolf     int i, outidx;
243840b4f539SKevin Wolf 
243940b4f539SKevin Wolf     // Sort requests by start sector
244040b4f539SKevin Wolf     qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
244140b4f539SKevin Wolf 
244240b4f539SKevin Wolf     // Check if adjacent requests touch the same clusters. If so, combine them,
244340b4f539SKevin Wolf     // filling up gaps with zero sectors.
244440b4f539SKevin Wolf     outidx = 0;
244540b4f539SKevin Wolf     for (i = 1; i < num_reqs; i++) {
244640b4f539SKevin Wolf         int merge = 0;
244740b4f539SKevin Wolf         int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
244840b4f539SKevin Wolf 
244940b4f539SKevin Wolf         // This handles the cases that are valid for all block drivers, namely
245040b4f539SKevin Wolf         // exactly sequential writes and overlapping writes.
245140b4f539SKevin Wolf         if (reqs[i].sector <= oldreq_last) {
245240b4f539SKevin Wolf             merge = 1;
245340b4f539SKevin Wolf         }
245440b4f539SKevin Wolf 
245540b4f539SKevin Wolf         // The block driver may decide that it makes sense to combine requests
245640b4f539SKevin Wolf         // even if there is a gap of some sectors between them. In this case,
245740b4f539SKevin Wolf         // the gap is filled with zeros (therefore only applicable for yet
245840b4f539SKevin Wolf         // unused space in format like qcow2).
245940b4f539SKevin Wolf         if (!merge && bs->drv->bdrv_merge_requests) {
246040b4f539SKevin Wolf             merge = bs->drv->bdrv_merge_requests(bs, &reqs[outidx], &reqs[i]);
246140b4f539SKevin Wolf         }
246240b4f539SKevin Wolf 
2463e2a305fbSChristoph Hellwig         if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
2464e2a305fbSChristoph Hellwig             merge = 0;
2465e2a305fbSChristoph Hellwig         }
2466e2a305fbSChristoph Hellwig 
246740b4f539SKevin Wolf         if (merge) {
246840b4f539SKevin Wolf             size_t size;
24697267c094SAnthony Liguori             QEMUIOVector *qiov = g_malloc0(sizeof(*qiov));
247040b4f539SKevin Wolf             qemu_iovec_init(qiov,
247140b4f539SKevin Wolf                 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
247240b4f539SKevin Wolf 
247340b4f539SKevin Wolf             // Add the first request to the merged one. If the requests are
247440b4f539SKevin Wolf             // overlapping, drop the last sectors of the first request.
247540b4f539SKevin Wolf             size = (reqs[i].sector - reqs[outidx].sector) << 9;
247640b4f539SKevin Wolf             qemu_iovec_concat(qiov, reqs[outidx].qiov, size);
247740b4f539SKevin Wolf 
247840b4f539SKevin Wolf             // We might need to add some zeros between the two requests
247940b4f539SKevin Wolf             if (reqs[i].sector > oldreq_last) {
248040b4f539SKevin Wolf                 size_t zero_bytes = (reqs[i].sector - oldreq_last) << 9;
248140b4f539SKevin Wolf                 uint8_t *buf = qemu_blockalign(bs, zero_bytes);
248240b4f539SKevin Wolf                 memset(buf, 0, zero_bytes);
248340b4f539SKevin Wolf                 qemu_iovec_add(qiov, buf, zero_bytes);
248440b4f539SKevin Wolf                 mcb->callbacks[i].free_buf = buf;
248540b4f539SKevin Wolf             }
248640b4f539SKevin Wolf 
248740b4f539SKevin Wolf             // Add the second request
248840b4f539SKevin Wolf             qemu_iovec_concat(qiov, reqs[i].qiov, reqs[i].qiov->size);
248940b4f539SKevin Wolf 
2490cbf1dff2SKevin Wolf             reqs[outidx].nb_sectors = qiov->size >> 9;
249140b4f539SKevin Wolf             reqs[outidx].qiov = qiov;
249240b4f539SKevin Wolf 
249340b4f539SKevin Wolf             mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
249440b4f539SKevin Wolf         } else {
249540b4f539SKevin Wolf             outidx++;
249640b4f539SKevin Wolf             reqs[outidx].sector     = reqs[i].sector;
249740b4f539SKevin Wolf             reqs[outidx].nb_sectors = reqs[i].nb_sectors;
249840b4f539SKevin Wolf             reqs[outidx].qiov       = reqs[i].qiov;
249940b4f539SKevin Wolf         }
250040b4f539SKevin Wolf     }
250140b4f539SKevin Wolf 
250240b4f539SKevin Wolf     return outidx + 1;
250340b4f539SKevin Wolf }
250440b4f539SKevin Wolf 
250540b4f539SKevin Wolf /*
250640b4f539SKevin Wolf  * Submit multiple AIO write requests at once.
250740b4f539SKevin Wolf  *
250840b4f539SKevin Wolf  * On success, the function returns 0 and all requests in the reqs array have
250940b4f539SKevin Wolf  * been submitted. In error case this function returns -1, and any of the
251040b4f539SKevin Wolf  * requests may or may not be submitted yet. In particular, this means that the
251140b4f539SKevin Wolf  * callback will be called for some of the requests, for others it won't. The
251240b4f539SKevin Wolf  * caller must check the error field of the BlockRequest to wait for the right
251340b4f539SKevin Wolf  * callbacks (if error != 0, no callback will be called).
251440b4f539SKevin Wolf  *
251540b4f539SKevin Wolf  * The implementation may modify the contents of the reqs array, e.g. to merge
251640b4f539SKevin Wolf  * requests. However, the fields opaque and error are left unmodified as they
251740b4f539SKevin Wolf  * are used to signal failure for a single request to the caller.
251840b4f539SKevin Wolf  */
251940b4f539SKevin Wolf int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
252040b4f539SKevin Wolf {
252140b4f539SKevin Wolf     BlockDriverAIOCB *acb;
252240b4f539SKevin Wolf     MultiwriteCB *mcb;
252340b4f539SKevin Wolf     int i;
252440b4f539SKevin Wolf 
2525301db7c2SRyan Harper     /* don't submit writes if we don't have a medium */
2526301db7c2SRyan Harper     if (bs->drv == NULL) {
2527301db7c2SRyan Harper         for (i = 0; i < num_reqs; i++) {
2528301db7c2SRyan Harper             reqs[i].error = -ENOMEDIUM;
2529301db7c2SRyan Harper         }
2530301db7c2SRyan Harper         return -1;
2531301db7c2SRyan Harper     }
2532301db7c2SRyan Harper 
253340b4f539SKevin Wolf     if (num_reqs == 0) {
253440b4f539SKevin Wolf         return 0;
253540b4f539SKevin Wolf     }
253640b4f539SKevin Wolf 
253740b4f539SKevin Wolf     // Create MultiwriteCB structure
25387267c094SAnthony Liguori     mcb = g_malloc0(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
253940b4f539SKevin Wolf     mcb->num_requests = 0;
254040b4f539SKevin Wolf     mcb->num_callbacks = num_reqs;
254140b4f539SKevin Wolf 
254240b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
254340b4f539SKevin Wolf         mcb->callbacks[i].cb = reqs[i].cb;
254440b4f539SKevin Wolf         mcb->callbacks[i].opaque = reqs[i].opaque;
254540b4f539SKevin Wolf     }
254640b4f539SKevin Wolf 
254740b4f539SKevin Wolf     // Check for mergable requests
254840b4f539SKevin Wolf     num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
254940b4f539SKevin Wolf 
25506d519a5fSStefan Hajnoczi     trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
25516d519a5fSStefan Hajnoczi 
2552453f9a16SKevin Wolf     /*
2553453f9a16SKevin Wolf      * Run the aio requests. As soon as one request can't be submitted
2554453f9a16SKevin Wolf      * successfully, fail all requests that are not yet submitted (we must
2555453f9a16SKevin Wolf      * return failure for all requests anyway)
2556453f9a16SKevin Wolf      *
2557453f9a16SKevin Wolf      * num_requests cannot be set to the right value immediately: If
2558453f9a16SKevin Wolf      * bdrv_aio_writev fails for some request, num_requests would be too high
2559453f9a16SKevin Wolf      * and therefore multiwrite_cb() would never recognize the multiwrite
2560453f9a16SKevin Wolf      * request as completed. We also cannot use the loop variable i to set it
2561453f9a16SKevin Wolf      * when the first request fails because the callback may already have been
2562453f9a16SKevin Wolf      * called for previously submitted requests. Thus, num_requests must be
2563453f9a16SKevin Wolf      * incremented for each request that is submitted.
2564453f9a16SKevin Wolf      *
2565453f9a16SKevin Wolf      * The problem that callbacks may be called early also means that we need
2566453f9a16SKevin Wolf      * to take care that num_requests doesn't become 0 before all requests are
2567453f9a16SKevin Wolf      * submitted - multiwrite_cb() would consider the multiwrite request
2568453f9a16SKevin Wolf      * completed. A dummy request that is "completed" by a manual call to
2569453f9a16SKevin Wolf      * multiwrite_cb() takes care of this.
2570453f9a16SKevin Wolf      */
2571453f9a16SKevin Wolf     mcb->num_requests = 1;
2572453f9a16SKevin Wolf 
25736d519a5fSStefan Hajnoczi     // Run the aio requests
257440b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
2575453f9a16SKevin Wolf         mcb->num_requests++;
257640b4f539SKevin Wolf         acb = bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
257740b4f539SKevin Wolf             reqs[i].nb_sectors, multiwrite_cb, mcb);
257840b4f539SKevin Wolf 
257940b4f539SKevin Wolf         if (acb == NULL) {
258040b4f539SKevin Wolf             // We can only fail the whole thing if no request has been
258140b4f539SKevin Wolf             // submitted yet. Otherwise we'll wait for the submitted AIOs to
258240b4f539SKevin Wolf             // complete and report the error in the callback.
2583453f9a16SKevin Wolf             if (i == 0) {
25846d519a5fSStefan Hajnoczi                 trace_bdrv_aio_multiwrite_earlyfail(mcb);
258540b4f539SKevin Wolf                 goto fail;
258640b4f539SKevin Wolf             } else {
25876d519a5fSStefan Hajnoczi                 trace_bdrv_aio_multiwrite_latefail(mcb, i);
25887eb58a6cSKevin Wolf                 multiwrite_cb(mcb, -EIO);
258940b4f539SKevin Wolf                 break;
259040b4f539SKevin Wolf             }
259140b4f539SKevin Wolf         }
259240b4f539SKevin Wolf     }
259340b4f539SKevin Wolf 
2594453f9a16SKevin Wolf     /* Complete the dummy request */
2595453f9a16SKevin Wolf     multiwrite_cb(mcb, 0);
2596453f9a16SKevin Wolf 
259740b4f539SKevin Wolf     return 0;
259840b4f539SKevin Wolf 
259940b4f539SKevin Wolf fail:
2600453f9a16SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
2601453f9a16SKevin Wolf         reqs[i].error = -EIO;
2602453f9a16SKevin Wolf     }
26037267c094SAnthony Liguori     g_free(mcb);
260440b4f539SKevin Wolf     return -1;
260540b4f539SKevin Wolf }
260640b4f539SKevin Wolf 
2607b2e12bc6SChristoph Hellwig BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
2608b2e12bc6SChristoph Hellwig         BlockDriverCompletionFunc *cb, void *opaque)
2609b2e12bc6SChristoph Hellwig {
2610b2e12bc6SChristoph Hellwig     BlockDriver *drv = bs->drv;
2611b2e12bc6SChristoph Hellwig 
2612a13aac04SStefan Hajnoczi     trace_bdrv_aio_flush(bs, opaque);
2613a13aac04SStefan Hajnoczi 
2614016f5cf6SAlexander Graf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
2615016f5cf6SAlexander Graf         return bdrv_aio_noop_em(bs, cb, opaque);
2616016f5cf6SAlexander Graf     }
2617016f5cf6SAlexander Graf 
2618b2e12bc6SChristoph Hellwig     if (!drv)
2619b2e12bc6SChristoph Hellwig         return NULL;
2620b2e12bc6SChristoph Hellwig     return drv->bdrv_aio_flush(bs, cb, opaque);
2621b2e12bc6SChristoph Hellwig }
2622b2e12bc6SChristoph Hellwig 
262383f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
262483f64091Sbellard {
26256bbff9a0Saliguori     acb->pool->cancel(acb);
262683f64091Sbellard }
262783f64091Sbellard 
262883f64091Sbellard 
262983f64091Sbellard /**************************************************************/
263083f64091Sbellard /* async block device emulation */
263183f64091Sbellard 
2632c16b5a2cSChristoph Hellwig typedef struct BlockDriverAIOCBSync {
2633c16b5a2cSChristoph Hellwig     BlockDriverAIOCB common;
2634c16b5a2cSChristoph Hellwig     QEMUBH *bh;
2635c16b5a2cSChristoph Hellwig     int ret;
2636c16b5a2cSChristoph Hellwig     /* vector translation state */
2637c16b5a2cSChristoph Hellwig     QEMUIOVector *qiov;
2638c16b5a2cSChristoph Hellwig     uint8_t *bounce;
2639c16b5a2cSChristoph Hellwig     int is_write;
2640c16b5a2cSChristoph Hellwig } BlockDriverAIOCBSync;
2641c16b5a2cSChristoph Hellwig 
2642c16b5a2cSChristoph Hellwig static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
2643c16b5a2cSChristoph Hellwig {
2644b666d239SKevin Wolf     BlockDriverAIOCBSync *acb =
2645b666d239SKevin Wolf         container_of(blockacb, BlockDriverAIOCBSync, common);
26466a7ad299SDor Laor     qemu_bh_delete(acb->bh);
264736afc451SAvi Kivity     acb->bh = NULL;
2648c16b5a2cSChristoph Hellwig     qemu_aio_release(acb);
2649c16b5a2cSChristoph Hellwig }
2650c16b5a2cSChristoph Hellwig 
2651c16b5a2cSChristoph Hellwig static AIOPool bdrv_em_aio_pool = {
2652c16b5a2cSChristoph Hellwig     .aiocb_size         = sizeof(BlockDriverAIOCBSync),
2653c16b5a2cSChristoph Hellwig     .cancel             = bdrv_aio_cancel_em,
2654c16b5a2cSChristoph Hellwig };
2655c16b5a2cSChristoph Hellwig 
265683f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
2657beac80cdSbellard {
2658ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = opaque;
2659f141eafeSaliguori 
2660f141eafeSaliguori     if (!acb->is_write)
2661f141eafeSaliguori         qemu_iovec_from_buffer(acb->qiov, acb->bounce, acb->qiov->size);
2662ceb42de8Saliguori     qemu_vfree(acb->bounce);
2663ce1a14dcSpbrook     acb->common.cb(acb->common.opaque, acb->ret);
26646a7ad299SDor Laor     qemu_bh_delete(acb->bh);
266536afc451SAvi Kivity     acb->bh = NULL;
2666ce1a14dcSpbrook     qemu_aio_release(acb);
2667beac80cdSbellard }
2668beac80cdSbellard 
2669f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
2670f141eafeSaliguori                                             int64_t sector_num,
2671f141eafeSaliguori                                             QEMUIOVector *qiov,
2672f141eafeSaliguori                                             int nb_sectors,
2673f141eafeSaliguori                                             BlockDriverCompletionFunc *cb,
2674f141eafeSaliguori                                             void *opaque,
2675f141eafeSaliguori                                             int is_write)
2676f141eafeSaliguori 
2677ea2384d3Sbellard {
2678ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
267983f64091Sbellard 
2680c16b5a2cSChristoph Hellwig     acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2681f141eafeSaliguori     acb->is_write = is_write;
2682f141eafeSaliguori     acb->qiov = qiov;
2683e268ca52Saliguori     acb->bounce = qemu_blockalign(bs, qiov->size);
2684f141eafeSaliguori 
2685ce1a14dcSpbrook     if (!acb->bh)
2686ce1a14dcSpbrook         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2687f141eafeSaliguori 
2688f141eafeSaliguori     if (is_write) {
2689f141eafeSaliguori         qemu_iovec_to_buffer(acb->qiov, acb->bounce);
2690f141eafeSaliguori         acb->ret = bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
2691f141eafeSaliguori     } else {
2692f141eafeSaliguori         acb->ret = bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
2693f141eafeSaliguori     }
2694f141eafeSaliguori 
2695ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
2696f141eafeSaliguori 
2697ce1a14dcSpbrook     return &acb->common;
26987a6cba61Spbrook }
26997a6cba61Spbrook 
2700f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
2701f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
2702ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
270383f64091Sbellard {
2704f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
270583f64091Sbellard }
270683f64091Sbellard 
2707f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
2708f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
2709f141eafeSaliguori         BlockDriverCompletionFunc *cb, void *opaque)
2710f141eafeSaliguori {
2711f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
2712f141eafeSaliguori }
2713f141eafeSaliguori 
271468485420SKevin Wolf 
271568485420SKevin Wolf typedef struct BlockDriverAIOCBCoroutine {
271668485420SKevin Wolf     BlockDriverAIOCB common;
271768485420SKevin Wolf     BlockRequest req;
271868485420SKevin Wolf     bool is_write;
271968485420SKevin Wolf     QEMUBH* bh;
272068485420SKevin Wolf } BlockDriverAIOCBCoroutine;
272168485420SKevin Wolf 
272268485420SKevin Wolf static void bdrv_aio_co_cancel_em(BlockDriverAIOCB *blockacb)
272368485420SKevin Wolf {
272468485420SKevin Wolf     qemu_aio_flush();
272568485420SKevin Wolf }
272668485420SKevin Wolf 
272768485420SKevin Wolf static AIOPool bdrv_em_co_aio_pool = {
272868485420SKevin Wolf     .aiocb_size         = sizeof(BlockDriverAIOCBCoroutine),
272968485420SKevin Wolf     .cancel             = bdrv_aio_co_cancel_em,
273068485420SKevin Wolf };
273168485420SKevin Wolf 
273268485420SKevin Wolf static void bdrv_co_rw_bh(void *opaque)
273368485420SKevin Wolf {
273468485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb = opaque;
273568485420SKevin Wolf 
273668485420SKevin Wolf     acb->common.cb(acb->common.opaque, acb->req.error);
273768485420SKevin Wolf     qemu_bh_delete(acb->bh);
273868485420SKevin Wolf     qemu_aio_release(acb);
273968485420SKevin Wolf }
274068485420SKevin Wolf 
274168485420SKevin Wolf static void coroutine_fn bdrv_co_rw(void *opaque)
274268485420SKevin Wolf {
274368485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb = opaque;
274468485420SKevin Wolf     BlockDriverState *bs = acb->common.bs;
274568485420SKevin Wolf 
274668485420SKevin Wolf     if (!acb->is_write) {
274768485420SKevin Wolf         acb->req.error = bs->drv->bdrv_co_readv(bs, acb->req.sector,
274868485420SKevin Wolf             acb->req.nb_sectors, acb->req.qiov);
274968485420SKevin Wolf     } else {
275068485420SKevin Wolf         acb->req.error = bs->drv->bdrv_co_writev(bs, acb->req.sector,
275168485420SKevin Wolf             acb->req.nb_sectors, acb->req.qiov);
275268485420SKevin Wolf     }
275368485420SKevin Wolf 
275468485420SKevin Wolf     acb->bh = qemu_bh_new(bdrv_co_rw_bh, acb);
275568485420SKevin Wolf     qemu_bh_schedule(acb->bh);
275668485420SKevin Wolf }
275768485420SKevin Wolf 
275868485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_rw_vector(BlockDriverState *bs,
275968485420SKevin Wolf                                                int64_t sector_num,
276068485420SKevin Wolf                                                QEMUIOVector *qiov,
276168485420SKevin Wolf                                                int nb_sectors,
276268485420SKevin Wolf                                                BlockDriverCompletionFunc *cb,
276368485420SKevin Wolf                                                void *opaque,
276468485420SKevin Wolf                                                bool is_write)
276568485420SKevin Wolf {
276668485420SKevin Wolf     Coroutine *co;
276768485420SKevin Wolf     BlockDriverAIOCBCoroutine *acb;
276868485420SKevin Wolf 
276968485420SKevin Wolf     acb = qemu_aio_get(&bdrv_em_co_aio_pool, bs, cb, opaque);
277068485420SKevin Wolf     acb->req.sector = sector_num;
277168485420SKevin Wolf     acb->req.nb_sectors = nb_sectors;
277268485420SKevin Wolf     acb->req.qiov = qiov;
277368485420SKevin Wolf     acb->is_write = is_write;
277468485420SKevin Wolf 
277568485420SKevin Wolf     co = qemu_coroutine_create(bdrv_co_rw);
277668485420SKevin Wolf     qemu_coroutine_enter(co, acb);
277768485420SKevin Wolf 
277868485420SKevin Wolf     return &acb->common;
277968485420SKevin Wolf }
278068485420SKevin Wolf 
278168485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_readv_em(BlockDriverState *bs,
278268485420SKevin Wolf         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
278368485420SKevin Wolf         BlockDriverCompletionFunc *cb, void *opaque)
278468485420SKevin Wolf {
278568485420SKevin Wolf     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque,
278668485420SKevin Wolf                                  false);
278768485420SKevin Wolf }
278868485420SKevin Wolf 
278968485420SKevin Wolf static BlockDriverAIOCB *bdrv_co_aio_writev_em(BlockDriverState *bs,
279068485420SKevin Wolf         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
279168485420SKevin Wolf         BlockDriverCompletionFunc *cb, void *opaque)
279268485420SKevin Wolf {
279368485420SKevin Wolf     return bdrv_co_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque,
279468485420SKevin Wolf                                  true);
279568485420SKevin Wolf }
279668485420SKevin Wolf 
2797b2e12bc6SChristoph Hellwig static BlockDriverAIOCB *bdrv_aio_flush_em(BlockDriverState *bs,
2798b2e12bc6SChristoph Hellwig         BlockDriverCompletionFunc *cb, void *opaque)
2799b2e12bc6SChristoph Hellwig {
2800b2e12bc6SChristoph Hellwig     BlockDriverAIOCBSync *acb;
2801b2e12bc6SChristoph Hellwig 
2802b2e12bc6SChristoph Hellwig     acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2803b2e12bc6SChristoph Hellwig     acb->is_write = 1; /* don't bounce in the completion hadler */
2804b2e12bc6SChristoph Hellwig     acb->qiov = NULL;
2805b2e12bc6SChristoph Hellwig     acb->bounce = NULL;
2806b2e12bc6SChristoph Hellwig     acb->ret = 0;
2807b2e12bc6SChristoph Hellwig 
2808b2e12bc6SChristoph Hellwig     if (!acb->bh)
2809b2e12bc6SChristoph Hellwig         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2810b2e12bc6SChristoph Hellwig 
2811b2e12bc6SChristoph Hellwig     bdrv_flush(bs);
2812b2e12bc6SChristoph Hellwig     qemu_bh_schedule(acb->bh);
2813b2e12bc6SChristoph Hellwig     return &acb->common;
2814b2e12bc6SChristoph Hellwig }
2815b2e12bc6SChristoph Hellwig 
2816016f5cf6SAlexander Graf static BlockDriverAIOCB *bdrv_aio_noop_em(BlockDriverState *bs,
2817016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque)
2818016f5cf6SAlexander Graf {
2819016f5cf6SAlexander Graf     BlockDriverAIOCBSync *acb;
2820016f5cf6SAlexander Graf 
2821016f5cf6SAlexander Graf     acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2822016f5cf6SAlexander Graf     acb->is_write = 1; /* don't bounce in the completion handler */
2823016f5cf6SAlexander Graf     acb->qiov = NULL;
2824016f5cf6SAlexander Graf     acb->bounce = NULL;
2825016f5cf6SAlexander Graf     acb->ret = 0;
2826016f5cf6SAlexander Graf 
2827016f5cf6SAlexander Graf     if (!acb->bh) {
2828016f5cf6SAlexander Graf         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2829016f5cf6SAlexander Graf     }
2830016f5cf6SAlexander Graf 
2831016f5cf6SAlexander Graf     qemu_bh_schedule(acb->bh);
2832016f5cf6SAlexander Graf     return &acb->common;
2833016f5cf6SAlexander Graf }
2834016f5cf6SAlexander Graf 
283583f64091Sbellard /**************************************************************/
283683f64091Sbellard /* sync block device emulation */
283783f64091Sbellard 
283883f64091Sbellard static void bdrv_rw_em_cb(void *opaque, int ret)
283983f64091Sbellard {
284083f64091Sbellard     *(int *)opaque = ret;
284183f64091Sbellard }
284283f64091Sbellard 
284383f64091Sbellard #define NOT_DONE 0x7fffffff
284483f64091Sbellard 
284583f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
284683f64091Sbellard                         uint8_t *buf, int nb_sectors)
284783f64091Sbellard {
2848ce1a14dcSpbrook     int async_ret;
2849ce1a14dcSpbrook     BlockDriverAIOCB *acb;
2850f141eafeSaliguori     struct iovec iov;
2851f141eafeSaliguori     QEMUIOVector qiov;
285283f64091Sbellard 
285383f64091Sbellard     async_ret = NOT_DONE;
28543f4cb3d3Sblueswir1     iov.iov_base = (void *)buf;
2855eb5a3165SJes Sorensen     iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2856f141eafeSaliguori     qemu_iovec_init_external(&qiov, &iov, 1);
2857f141eafeSaliguori     acb = bdrv_aio_readv(bs, sector_num, &qiov, nb_sectors,
285883f64091Sbellard         bdrv_rw_em_cb, &async_ret);
285965d6b3d8SKevin Wolf     if (acb == NULL) {
286065d6b3d8SKevin Wolf         async_ret = -1;
286165d6b3d8SKevin Wolf         goto fail;
286265d6b3d8SKevin Wolf     }
2863baf35cb9Saliguori 
286483f64091Sbellard     while (async_ret == NOT_DONE) {
286583f64091Sbellard         qemu_aio_wait();
286683f64091Sbellard     }
2867baf35cb9Saliguori 
286865d6b3d8SKevin Wolf 
286965d6b3d8SKevin Wolf fail:
287083f64091Sbellard     return async_ret;
287183f64091Sbellard }
287283f64091Sbellard 
287383f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
287483f64091Sbellard                          const uint8_t *buf, int nb_sectors)
287583f64091Sbellard {
2876ce1a14dcSpbrook     int async_ret;
2877ce1a14dcSpbrook     BlockDriverAIOCB *acb;
2878f141eafeSaliguori     struct iovec iov;
2879f141eafeSaliguori     QEMUIOVector qiov;
288083f64091Sbellard 
288183f64091Sbellard     async_ret = NOT_DONE;
2882f141eafeSaliguori     iov.iov_base = (void *)buf;
2883eb5a3165SJes Sorensen     iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2884f141eafeSaliguori     qemu_iovec_init_external(&qiov, &iov, 1);
2885f141eafeSaliguori     acb = bdrv_aio_writev(bs, sector_num, &qiov, nb_sectors,
288683f64091Sbellard         bdrv_rw_em_cb, &async_ret);
288765d6b3d8SKevin Wolf     if (acb == NULL) {
288865d6b3d8SKevin Wolf         async_ret = -1;
288965d6b3d8SKevin Wolf         goto fail;
289065d6b3d8SKevin Wolf     }
289183f64091Sbellard     while (async_ret == NOT_DONE) {
289283f64091Sbellard         qemu_aio_wait();
289383f64091Sbellard     }
289465d6b3d8SKevin Wolf 
289565d6b3d8SKevin Wolf fail:
289683f64091Sbellard     return async_ret;
289783f64091Sbellard }
2898ea2384d3Sbellard 
2899ea2384d3Sbellard void bdrv_init(void)
2900ea2384d3Sbellard {
29015efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
2902ea2384d3Sbellard }
2903ce1a14dcSpbrook 
2904eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
2905eb852011SMarkus Armbruster {
2906eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
2907eb852011SMarkus Armbruster     bdrv_init();
2908eb852011SMarkus Armbruster }
2909eb852011SMarkus Armbruster 
2910c16b5a2cSChristoph Hellwig void *qemu_aio_get(AIOPool *pool, BlockDriverState *bs,
29116bbff9a0Saliguori                    BlockDriverCompletionFunc *cb, void *opaque)
29126bbff9a0Saliguori {
2913ce1a14dcSpbrook     BlockDriverAIOCB *acb;
2914ce1a14dcSpbrook 
29156bbff9a0Saliguori     if (pool->free_aiocb) {
29166bbff9a0Saliguori         acb = pool->free_aiocb;
29176bbff9a0Saliguori         pool->free_aiocb = acb->next;
2918ce1a14dcSpbrook     } else {
29197267c094SAnthony Liguori         acb = g_malloc0(pool->aiocb_size);
29206bbff9a0Saliguori         acb->pool = pool;
2921ce1a14dcSpbrook     }
2922ce1a14dcSpbrook     acb->bs = bs;
2923ce1a14dcSpbrook     acb->cb = cb;
2924ce1a14dcSpbrook     acb->opaque = opaque;
2925ce1a14dcSpbrook     return acb;
2926ce1a14dcSpbrook }
2927ce1a14dcSpbrook 
2928ce1a14dcSpbrook void qemu_aio_release(void *p)
2929ce1a14dcSpbrook {
29306bbff9a0Saliguori     BlockDriverAIOCB *acb = (BlockDriverAIOCB *)p;
29316bbff9a0Saliguori     AIOPool *pool = acb->pool;
29326bbff9a0Saliguori     acb->next = pool->free_aiocb;
29336bbff9a0Saliguori     pool->free_aiocb = acb;
2934ce1a14dcSpbrook }
293519cb3738Sbellard 
293619cb3738Sbellard /**************************************************************/
2937f9f05dc5SKevin Wolf /* Coroutine block device emulation */
2938f9f05dc5SKevin Wolf 
2939f9f05dc5SKevin Wolf typedef struct CoroutineIOCompletion {
2940f9f05dc5SKevin Wolf     Coroutine *coroutine;
2941f9f05dc5SKevin Wolf     int ret;
2942f9f05dc5SKevin Wolf } CoroutineIOCompletion;
2943f9f05dc5SKevin Wolf 
2944f9f05dc5SKevin Wolf static void bdrv_co_io_em_complete(void *opaque, int ret)
2945f9f05dc5SKevin Wolf {
2946f9f05dc5SKevin Wolf     CoroutineIOCompletion *co = opaque;
2947f9f05dc5SKevin Wolf 
2948f9f05dc5SKevin Wolf     co->ret = ret;
2949f9f05dc5SKevin Wolf     qemu_coroutine_enter(co->coroutine, NULL);
2950f9f05dc5SKevin Wolf }
2951f9f05dc5SKevin Wolf 
2952f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_io_em(BlockDriverState *bs, int64_t sector_num,
2953f9f05dc5SKevin Wolf                                       int nb_sectors, QEMUIOVector *iov,
2954f9f05dc5SKevin Wolf                                       bool is_write)
2955f9f05dc5SKevin Wolf {
2956f9f05dc5SKevin Wolf     CoroutineIOCompletion co = {
2957f9f05dc5SKevin Wolf         .coroutine = qemu_coroutine_self(),
2958f9f05dc5SKevin Wolf     };
2959f9f05dc5SKevin Wolf     BlockDriverAIOCB *acb;
2960f9f05dc5SKevin Wolf 
2961f9f05dc5SKevin Wolf     if (is_write) {
2962f9f05dc5SKevin Wolf         acb = bdrv_aio_writev(bs, sector_num, iov, nb_sectors,
2963f9f05dc5SKevin Wolf                               bdrv_co_io_em_complete, &co);
2964f9f05dc5SKevin Wolf     } else {
2965f9f05dc5SKevin Wolf         acb = bdrv_aio_readv(bs, sector_num, iov, nb_sectors,
2966f9f05dc5SKevin Wolf                              bdrv_co_io_em_complete, &co);
2967f9f05dc5SKevin Wolf     }
2968f9f05dc5SKevin Wolf 
2969f9f05dc5SKevin Wolf     trace_bdrv_co_io(is_write, acb);
2970f9f05dc5SKevin Wolf     if (!acb) {
2971f9f05dc5SKevin Wolf         return -EIO;
2972f9f05dc5SKevin Wolf     }
2973f9f05dc5SKevin Wolf     qemu_coroutine_yield();
2974f9f05dc5SKevin Wolf 
2975f9f05dc5SKevin Wolf     return co.ret;
2976f9f05dc5SKevin Wolf }
2977f9f05dc5SKevin Wolf 
2978f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_readv_em(BlockDriverState *bs,
2979f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
2980f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
2981f9f05dc5SKevin Wolf {
2982f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, false);
2983f9f05dc5SKevin Wolf }
2984f9f05dc5SKevin Wolf 
2985f9f05dc5SKevin Wolf static int coroutine_fn bdrv_co_writev_em(BlockDriverState *bs,
2986f9f05dc5SKevin Wolf                                          int64_t sector_num, int nb_sectors,
2987f9f05dc5SKevin Wolf                                          QEMUIOVector *iov)
2988f9f05dc5SKevin Wolf {
2989f9f05dc5SKevin Wolf     return bdrv_co_io_em(bs, sector_num, nb_sectors, iov, true);
2990f9f05dc5SKevin Wolf }
2991f9f05dc5SKevin Wolf 
2992e7a8a783SKevin Wolf static int coroutine_fn bdrv_co_flush_em(BlockDriverState *bs)
2993e7a8a783SKevin Wolf {
2994e7a8a783SKevin Wolf     CoroutineIOCompletion co = {
2995e7a8a783SKevin Wolf         .coroutine = qemu_coroutine_self(),
2996e7a8a783SKevin Wolf     };
2997e7a8a783SKevin Wolf     BlockDriverAIOCB *acb;
2998e7a8a783SKevin Wolf 
2999e7a8a783SKevin Wolf     acb = bdrv_aio_flush(bs, bdrv_co_io_em_complete, &co);
3000e7a8a783SKevin Wolf     if (!acb) {
3001e7a8a783SKevin Wolf         return -EIO;
3002e7a8a783SKevin Wolf     }
3003e7a8a783SKevin Wolf     qemu_coroutine_yield();
3004e7a8a783SKevin Wolf     return co.ret;
3005e7a8a783SKevin Wolf }
3006e7a8a783SKevin Wolf 
3007f9f05dc5SKevin Wolf /**************************************************************/
300819cb3738Sbellard /* removable device support */
300919cb3738Sbellard 
301019cb3738Sbellard /**
301119cb3738Sbellard  * Return TRUE if the media is present
301219cb3738Sbellard  */
301319cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
301419cb3738Sbellard {
301519cb3738Sbellard     BlockDriver *drv = bs->drv;
301619cb3738Sbellard     int ret;
301719cb3738Sbellard     if (!drv)
301819cb3738Sbellard         return 0;
301919cb3738Sbellard     if (!drv->bdrv_is_inserted)
30204be9762aSMarkus Armbruster         return !bs->tray_open;
302119cb3738Sbellard     ret = drv->bdrv_is_inserted(bs);
302219cb3738Sbellard     return ret;
302319cb3738Sbellard }
302419cb3738Sbellard 
302519cb3738Sbellard /**
302619cb3738Sbellard  * Return TRUE if the media changed since the last call to this
302719cb3738Sbellard  * function. It is currently only used for floppy disks
302819cb3738Sbellard  */
302919cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
303019cb3738Sbellard {
303119cb3738Sbellard     BlockDriver *drv = bs->drv;
303219cb3738Sbellard     int ret;
303319cb3738Sbellard 
303419cb3738Sbellard     if (!drv || !drv->bdrv_media_changed)
303519cb3738Sbellard         ret = -ENOTSUP;
303619cb3738Sbellard     else
303719cb3738Sbellard         ret = drv->bdrv_media_changed(bs);
303819cb3738Sbellard     if (ret == -ENOTSUP)
303919cb3738Sbellard         ret = bs->media_changed;
304019cb3738Sbellard     bs->media_changed = 0;
304119cb3738Sbellard     return ret;
304219cb3738Sbellard }
304319cb3738Sbellard 
304419cb3738Sbellard /**
304519cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
304619cb3738Sbellard  */
3047aea2a33cSMark McLoughlin int bdrv_eject(BlockDriverState *bs, int eject_flag)
304819cb3738Sbellard {
304919cb3738Sbellard     BlockDriver *drv = bs->drv;
305019cb3738Sbellard 
305149aa46bbSMarkus Armbruster     if (eject_flag && bs->locked) {
3052aea2a33cSMark McLoughlin         return -EBUSY;
3053aea2a33cSMark McLoughlin     }
3054aea2a33cSMark McLoughlin 
3055822e1cd1SMarkus Armbruster     if (drv && drv->bdrv_eject) {
3056822e1cd1SMarkus Armbruster         drv->bdrv_eject(bs, eject_flag);
305719cb3738Sbellard     }
30584be9762aSMarkus Armbruster     bs->tray_open = eject_flag;
3059822e1cd1SMarkus Armbruster     return 0;
306019cb3738Sbellard }
306119cb3738Sbellard 
306219cb3738Sbellard int bdrv_is_locked(BlockDriverState *bs)
306319cb3738Sbellard {
306419cb3738Sbellard     return bs->locked;
306519cb3738Sbellard }
306619cb3738Sbellard 
306719cb3738Sbellard /**
306819cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
306919cb3738Sbellard  * to eject it manually).
307019cb3738Sbellard  */
307119cb3738Sbellard void bdrv_set_locked(BlockDriverState *bs, int locked)
307219cb3738Sbellard {
307319cb3738Sbellard     BlockDriver *drv = bs->drv;
307419cb3738Sbellard 
3075b8c6d095SStefan Hajnoczi     trace_bdrv_set_locked(bs, locked);
3076b8c6d095SStefan Hajnoczi 
307719cb3738Sbellard     bs->locked = locked;
307819cb3738Sbellard     if (drv && drv->bdrv_set_locked) {
307919cb3738Sbellard         drv->bdrv_set_locked(bs, locked);
308019cb3738Sbellard     }
308119cb3738Sbellard }
3082985a03b0Sths 
3083985a03b0Sths /* needed for generic scsi interface */
3084985a03b0Sths 
3085985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
3086985a03b0Sths {
3087985a03b0Sths     BlockDriver *drv = bs->drv;
3088985a03b0Sths 
3089985a03b0Sths     if (drv && drv->bdrv_ioctl)
3090985a03b0Sths         return drv->bdrv_ioctl(bs, req, buf);
3091985a03b0Sths     return -ENOTSUP;
3092985a03b0Sths }
30937d780669Saliguori 
3094221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
3095221f715dSaliguori         unsigned long int req, void *buf,
30967d780669Saliguori         BlockDriverCompletionFunc *cb, void *opaque)
30977d780669Saliguori {
3098221f715dSaliguori     BlockDriver *drv = bs->drv;
30997d780669Saliguori 
3100221f715dSaliguori     if (drv && drv->bdrv_aio_ioctl)
3101221f715dSaliguori         return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
3102221f715dSaliguori     return NULL;
31037d780669Saliguori }
3104e268ca52Saliguori 
31057cd1e32aSlirans@il.ibm.com 
31067cd1e32aSlirans@il.ibm.com 
3107e268ca52Saliguori void *qemu_blockalign(BlockDriverState *bs, size_t size)
3108e268ca52Saliguori {
3109e268ca52Saliguori     return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
3110e268ca52Saliguori }
31117cd1e32aSlirans@il.ibm.com 
31127cd1e32aSlirans@il.ibm.com void bdrv_set_dirty_tracking(BlockDriverState *bs, int enable)
31137cd1e32aSlirans@il.ibm.com {
31147cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
3115a55eb92cSJan Kiszka 
3116aaa0eb75SLiran Schour     bs->dirty_count = 0;
31177cd1e32aSlirans@il.ibm.com     if (enable) {
3118c6d22830SJan Kiszka         if (!bs->dirty_bitmap) {
3119c6d22830SJan Kiszka             bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS) +
3120c6d22830SJan Kiszka                     BDRV_SECTORS_PER_DIRTY_CHUNK * 8 - 1;
3121c6d22830SJan Kiszka             bitmap_size /= BDRV_SECTORS_PER_DIRTY_CHUNK * 8;
31227cd1e32aSlirans@il.ibm.com 
31237267c094SAnthony Liguori             bs->dirty_bitmap = g_malloc0(bitmap_size);
31247cd1e32aSlirans@il.ibm.com         }
31257cd1e32aSlirans@il.ibm.com     } else {
3126c6d22830SJan Kiszka         if (bs->dirty_bitmap) {
31277267c094SAnthony Liguori             g_free(bs->dirty_bitmap);
3128c6d22830SJan Kiszka             bs->dirty_bitmap = NULL;
31297cd1e32aSlirans@il.ibm.com         }
31307cd1e32aSlirans@il.ibm.com     }
31317cd1e32aSlirans@il.ibm.com }
31327cd1e32aSlirans@il.ibm.com 
31337cd1e32aSlirans@il.ibm.com int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
31347cd1e32aSlirans@il.ibm.com {
31356ea44308SJan Kiszka     int64_t chunk = sector / (int64_t)BDRV_SECTORS_PER_DIRTY_CHUNK;
31367cd1e32aSlirans@il.ibm.com 
3137c6d22830SJan Kiszka     if (bs->dirty_bitmap &&
3138c6d22830SJan Kiszka         (sector << BDRV_SECTOR_BITS) < bdrv_getlength(bs)) {
31396d59fec1SMarcelo Tosatti         return !!(bs->dirty_bitmap[chunk / (sizeof(unsigned long) * 8)] &
31406d59fec1SMarcelo Tosatti             (1UL << (chunk % (sizeof(unsigned long) * 8))));
31417cd1e32aSlirans@il.ibm.com     } else {
31427cd1e32aSlirans@il.ibm.com         return 0;
31437cd1e32aSlirans@il.ibm.com     }
31447cd1e32aSlirans@il.ibm.com }
31457cd1e32aSlirans@il.ibm.com 
31467cd1e32aSlirans@il.ibm.com void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
31477cd1e32aSlirans@il.ibm.com                       int nr_sectors)
31487cd1e32aSlirans@il.ibm.com {
31497cd1e32aSlirans@il.ibm.com     set_dirty_bitmap(bs, cur_sector, nr_sectors, 0);
31507cd1e32aSlirans@il.ibm.com }
3151aaa0eb75SLiran Schour 
3152aaa0eb75SLiran Schour int64_t bdrv_get_dirty_count(BlockDriverState *bs)
3153aaa0eb75SLiran Schour {
3154aaa0eb75SLiran Schour     return bs->dirty_count;
3155aaa0eb75SLiran Schour }
3156f88e1a42SJes Sorensen 
3157db593f25SMarcelo Tosatti void bdrv_set_in_use(BlockDriverState *bs, int in_use)
3158db593f25SMarcelo Tosatti {
3159db593f25SMarcelo Tosatti     assert(bs->in_use != in_use);
3160db593f25SMarcelo Tosatti     bs->in_use = in_use;
3161db593f25SMarcelo Tosatti }
3162db593f25SMarcelo Tosatti 
3163db593f25SMarcelo Tosatti int bdrv_in_use(BlockDriverState *bs)
3164db593f25SMarcelo Tosatti {
3165db593f25SMarcelo Tosatti     return bs->in_use;
3166db593f25SMarcelo Tosatti }
3167db593f25SMarcelo Tosatti 
3168a597e79cSChristoph Hellwig void
3169a597e79cSChristoph Hellwig bdrv_acct_start(BlockDriverState *bs, BlockAcctCookie *cookie, int64_t bytes,
3170a597e79cSChristoph Hellwig         enum BlockAcctType type)
3171a597e79cSChristoph Hellwig {
3172a597e79cSChristoph Hellwig     assert(type < BDRV_MAX_IOTYPE);
3173a597e79cSChristoph Hellwig 
3174a597e79cSChristoph Hellwig     cookie->bytes = bytes;
3175*c488c7f6SChristoph Hellwig     cookie->start_time_ns = get_clock();
3176a597e79cSChristoph Hellwig     cookie->type = type;
3177a597e79cSChristoph Hellwig }
3178a597e79cSChristoph Hellwig 
3179a597e79cSChristoph Hellwig void
3180a597e79cSChristoph Hellwig bdrv_acct_done(BlockDriverState *bs, BlockAcctCookie *cookie)
3181a597e79cSChristoph Hellwig {
3182a597e79cSChristoph Hellwig     assert(cookie->type < BDRV_MAX_IOTYPE);
3183a597e79cSChristoph Hellwig 
3184a597e79cSChristoph Hellwig     bs->nr_bytes[cookie->type] += cookie->bytes;
3185a597e79cSChristoph Hellwig     bs->nr_ops[cookie->type]++;
3186*c488c7f6SChristoph Hellwig     bs->total_time_ns[cookie->type] += get_clock() - cookie->start_time_ns;
3187a597e79cSChristoph Hellwig }
3188a597e79cSChristoph Hellwig 
3189f88e1a42SJes Sorensen int bdrv_img_create(const char *filename, const char *fmt,
3190f88e1a42SJes Sorensen                     const char *base_filename, const char *base_fmt,
3191f88e1a42SJes Sorensen                     char *options, uint64_t img_size, int flags)
3192f88e1a42SJes Sorensen {
3193f88e1a42SJes Sorensen     QEMUOptionParameter *param = NULL, *create_options = NULL;
3194d220894eSKevin Wolf     QEMUOptionParameter *backing_fmt, *backing_file, *size;
3195f88e1a42SJes Sorensen     BlockDriverState *bs = NULL;
3196f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
319796df67d1SStefan Hajnoczi     BlockDriver *backing_drv = NULL;
3198f88e1a42SJes Sorensen     int ret = 0;
3199f88e1a42SJes Sorensen 
3200f88e1a42SJes Sorensen     /* Find driver and parse its options */
3201f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
3202f88e1a42SJes Sorensen     if (!drv) {
3203f88e1a42SJes Sorensen         error_report("Unknown file format '%s'", fmt);
32044f70f249SJes Sorensen         ret = -EINVAL;
3205f88e1a42SJes Sorensen         goto out;
3206f88e1a42SJes Sorensen     }
3207f88e1a42SJes Sorensen 
3208f88e1a42SJes Sorensen     proto_drv = bdrv_find_protocol(filename);
3209f88e1a42SJes Sorensen     if (!proto_drv) {
3210f88e1a42SJes Sorensen         error_report("Unknown protocol '%s'", filename);
32114f70f249SJes Sorensen         ret = -EINVAL;
3212f88e1a42SJes Sorensen         goto out;
3213f88e1a42SJes Sorensen     }
3214f88e1a42SJes Sorensen 
3215f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
3216f88e1a42SJes Sorensen                                               drv->create_options);
3217f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
3218f88e1a42SJes Sorensen                                               proto_drv->create_options);
3219f88e1a42SJes Sorensen 
3220f88e1a42SJes Sorensen     /* Create parameter list with default values */
3221f88e1a42SJes Sorensen     param = parse_option_parameters("", create_options, param);
3222f88e1a42SJes Sorensen 
3223f88e1a42SJes Sorensen     set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
3224f88e1a42SJes Sorensen 
3225f88e1a42SJes Sorensen     /* Parse -o options */
3226f88e1a42SJes Sorensen     if (options) {
3227f88e1a42SJes Sorensen         param = parse_option_parameters(options, create_options, param);
3228f88e1a42SJes Sorensen         if (param == NULL) {
3229f88e1a42SJes Sorensen             error_report("Invalid options for file format '%s'.", fmt);
32304f70f249SJes Sorensen             ret = -EINVAL;
3231f88e1a42SJes Sorensen             goto out;
3232f88e1a42SJes Sorensen         }
3233f88e1a42SJes Sorensen     }
3234f88e1a42SJes Sorensen 
3235f88e1a42SJes Sorensen     if (base_filename) {
3236f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
3237f88e1a42SJes Sorensen                                  base_filename)) {
3238f88e1a42SJes Sorensen             error_report("Backing file not supported for file format '%s'",
3239f88e1a42SJes Sorensen                          fmt);
32404f70f249SJes Sorensen             ret = -EINVAL;
3241f88e1a42SJes Sorensen             goto out;
3242f88e1a42SJes Sorensen         }
3243f88e1a42SJes Sorensen     }
3244f88e1a42SJes Sorensen 
3245f88e1a42SJes Sorensen     if (base_fmt) {
3246f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
3247f88e1a42SJes Sorensen             error_report("Backing file format not supported for file "
3248f88e1a42SJes Sorensen                          "format '%s'", fmt);
32494f70f249SJes Sorensen             ret = -EINVAL;
3250f88e1a42SJes Sorensen             goto out;
3251f88e1a42SJes Sorensen         }
3252f88e1a42SJes Sorensen     }
3253f88e1a42SJes Sorensen 
3254792da93aSJes Sorensen     backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
3255792da93aSJes Sorensen     if (backing_file && backing_file->value.s) {
3256792da93aSJes Sorensen         if (!strcmp(filename, backing_file->value.s)) {
3257792da93aSJes Sorensen             error_report("Error: Trying to create an image with the "
3258792da93aSJes Sorensen                          "same filename as the backing file");
32594f70f249SJes Sorensen             ret = -EINVAL;
3260792da93aSJes Sorensen             goto out;
3261792da93aSJes Sorensen         }
3262792da93aSJes Sorensen     }
3263792da93aSJes Sorensen 
3264f88e1a42SJes Sorensen     backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
3265f88e1a42SJes Sorensen     if (backing_fmt && backing_fmt->value.s) {
326696df67d1SStefan Hajnoczi         backing_drv = bdrv_find_format(backing_fmt->value.s);
326796df67d1SStefan Hajnoczi         if (!backing_drv) {
3268f88e1a42SJes Sorensen             error_report("Unknown backing file format '%s'",
3269f88e1a42SJes Sorensen                          backing_fmt->value.s);
32704f70f249SJes Sorensen             ret = -EINVAL;
3271f88e1a42SJes Sorensen             goto out;
3272f88e1a42SJes Sorensen         }
3273f88e1a42SJes Sorensen     }
3274f88e1a42SJes Sorensen 
3275f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
3276f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
3277d220894eSKevin Wolf     size = get_option_parameter(param, BLOCK_OPT_SIZE);
3278d220894eSKevin Wolf     if (size && size->value.n == -1) {
3279f88e1a42SJes Sorensen         if (backing_file && backing_file->value.s) {
3280f88e1a42SJes Sorensen             uint64_t size;
3281f88e1a42SJes Sorensen             char buf[32];
3282f88e1a42SJes Sorensen 
3283f88e1a42SJes Sorensen             bs = bdrv_new("");
3284f88e1a42SJes Sorensen 
328596df67d1SStefan Hajnoczi             ret = bdrv_open(bs, backing_file->value.s, flags, backing_drv);
3286f88e1a42SJes Sorensen             if (ret < 0) {
328796df67d1SStefan Hajnoczi                 error_report("Could not open '%s'", backing_file->value.s);
3288f88e1a42SJes Sorensen                 goto out;
3289f88e1a42SJes Sorensen             }
3290f88e1a42SJes Sorensen             bdrv_get_geometry(bs, &size);
3291f88e1a42SJes Sorensen             size *= 512;
3292f88e1a42SJes Sorensen 
3293f88e1a42SJes Sorensen             snprintf(buf, sizeof(buf), "%" PRId64, size);
3294f88e1a42SJes Sorensen             set_option_parameter(param, BLOCK_OPT_SIZE, buf);
3295f88e1a42SJes Sorensen         } else {
3296f88e1a42SJes Sorensen             error_report("Image creation needs a size parameter");
32974f70f249SJes Sorensen             ret = -EINVAL;
3298f88e1a42SJes Sorensen             goto out;
3299f88e1a42SJes Sorensen         }
3300f88e1a42SJes Sorensen     }
3301f88e1a42SJes Sorensen 
3302f88e1a42SJes Sorensen     printf("Formatting '%s', fmt=%s ", filename, fmt);
3303f88e1a42SJes Sorensen     print_option_parameters(param);
3304f88e1a42SJes Sorensen     puts("");
3305f88e1a42SJes Sorensen 
3306f88e1a42SJes Sorensen     ret = bdrv_create(drv, filename, param);
3307f88e1a42SJes Sorensen 
3308f88e1a42SJes Sorensen     if (ret < 0) {
3309f88e1a42SJes Sorensen         if (ret == -ENOTSUP) {
3310f88e1a42SJes Sorensen             error_report("Formatting or formatting option not supported for "
3311f88e1a42SJes Sorensen                          "file format '%s'", fmt);
3312f88e1a42SJes Sorensen         } else if (ret == -EFBIG) {
3313f88e1a42SJes Sorensen             error_report("The image size is too large for file format '%s'",
3314f88e1a42SJes Sorensen                          fmt);
3315f88e1a42SJes Sorensen         } else {
3316f88e1a42SJes Sorensen             error_report("%s: error while creating %s: %s", filename, fmt,
3317f88e1a42SJes Sorensen                          strerror(-ret));
3318f88e1a42SJes Sorensen         }
3319f88e1a42SJes Sorensen     }
3320f88e1a42SJes Sorensen 
3321f88e1a42SJes Sorensen out:
3322f88e1a42SJes Sorensen     free_option_parameters(create_options);
3323f88e1a42SJes Sorensen     free_option_parameters(param);
3324f88e1a42SJes Sorensen 
3325f88e1a42SJes Sorensen     if (bs) {
3326f88e1a42SJes Sorensen         bdrv_delete(bs);
3327f88e1a42SJes Sorensen     }
33284f70f249SJes Sorensen 
33294f70f249SJes Sorensen     return ret;
3330f88e1a42SJes Sorensen }
3331