xref: /openbmc/qemu/block.c (revision d220894e025ea85d604a2cdc1eef541ed516728e)
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"
31fc01f7e7Sbellard 
3271e72a19SJuan Quintela #ifdef CONFIG_BSD
337674e7bfSbellard #include <sys/types.h>
347674e7bfSbellard #include <sys/stat.h>
357674e7bfSbellard #include <sys/ioctl.h>
3672cf2d4fSBlue Swirl #include <sys/queue.h>
37c5e97233Sblueswir1 #ifndef __DragonFly__
387674e7bfSbellard #include <sys/disk.h>
397674e7bfSbellard #endif
40c5e97233Sblueswir1 #endif
417674e7bfSbellard 
4249dc768dSaliguori #ifdef _WIN32
4349dc768dSaliguori #include <windows.h>
4449dc768dSaliguori #endif
4549dc768dSaliguori 
46f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
47f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
48c87c0672Saliguori         BlockDriverCompletionFunc *cb, void *opaque);
49f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
50f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
51ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque);
52b2e12bc6SChristoph Hellwig static BlockDriverAIOCB *bdrv_aio_flush_em(BlockDriverState *bs,
53b2e12bc6SChristoph Hellwig         BlockDriverCompletionFunc *cb, void *opaque);
54016f5cf6SAlexander Graf static BlockDriverAIOCB *bdrv_aio_noop_em(BlockDriverState *bs,
55016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque);
5683f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
5783f64091Sbellard                         uint8_t *buf, int nb_sectors);
5883f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
5983f64091Sbellard                          const uint8_t *buf, int nb_sectors);
60ec530c81Sbellard 
611b7bdbc1SStefan Hajnoczi static QTAILQ_HEAD(, BlockDriverState) bdrv_states =
621b7bdbc1SStefan Hajnoczi     QTAILQ_HEAD_INITIALIZER(bdrv_states);
637ee930d0Sblueswir1 
648a22f02aSStefan Hajnoczi static QLIST_HEAD(, BlockDriver) bdrv_drivers =
658a22f02aSStefan Hajnoczi     QLIST_HEAD_INITIALIZER(bdrv_drivers);
66ea2384d3Sbellard 
67f9092b10SMarkus Armbruster /* The device to use for VM snapshots */
68f9092b10SMarkus Armbruster static BlockDriverState *bs_snapshots;
69f9092b10SMarkus Armbruster 
70eb852011SMarkus Armbruster /* If non-zero, use only whitelisted block drivers */
71eb852011SMarkus Armbruster static int use_bdrv_whitelist;
72eb852011SMarkus Armbruster 
739e0b22f4SStefan Hajnoczi #ifdef _WIN32
749e0b22f4SStefan Hajnoczi static int is_windows_drive_prefix(const char *filename)
759e0b22f4SStefan Hajnoczi {
769e0b22f4SStefan Hajnoczi     return (((filename[0] >= 'a' && filename[0] <= 'z') ||
779e0b22f4SStefan Hajnoczi              (filename[0] >= 'A' && filename[0] <= 'Z')) &&
789e0b22f4SStefan Hajnoczi             filename[1] == ':');
799e0b22f4SStefan Hajnoczi }
809e0b22f4SStefan Hajnoczi 
819e0b22f4SStefan Hajnoczi int is_windows_drive(const char *filename)
829e0b22f4SStefan Hajnoczi {
839e0b22f4SStefan Hajnoczi     if (is_windows_drive_prefix(filename) &&
849e0b22f4SStefan Hajnoczi         filename[2] == '\0')
859e0b22f4SStefan Hajnoczi         return 1;
869e0b22f4SStefan Hajnoczi     if (strstart(filename, "\\\\.\\", NULL) ||
879e0b22f4SStefan Hajnoczi         strstart(filename, "//./", NULL))
889e0b22f4SStefan Hajnoczi         return 1;
899e0b22f4SStefan Hajnoczi     return 0;
909e0b22f4SStefan Hajnoczi }
919e0b22f4SStefan Hajnoczi #endif
929e0b22f4SStefan Hajnoczi 
939e0b22f4SStefan Hajnoczi /* check if the path starts with "<protocol>:" */
949e0b22f4SStefan Hajnoczi static int path_has_protocol(const char *path)
959e0b22f4SStefan Hajnoczi {
969e0b22f4SStefan Hajnoczi #ifdef _WIN32
979e0b22f4SStefan Hajnoczi     if (is_windows_drive(path) ||
989e0b22f4SStefan Hajnoczi         is_windows_drive_prefix(path)) {
999e0b22f4SStefan Hajnoczi         return 0;
1009e0b22f4SStefan Hajnoczi     }
1019e0b22f4SStefan Hajnoczi #endif
1029e0b22f4SStefan Hajnoczi 
1039e0b22f4SStefan Hajnoczi     return strchr(path, ':') != NULL;
1049e0b22f4SStefan Hajnoczi }
1059e0b22f4SStefan Hajnoczi 
10683f64091Sbellard int path_is_absolute(const char *path)
10783f64091Sbellard {
10883f64091Sbellard     const char *p;
10921664424Sbellard #ifdef _WIN32
11021664424Sbellard     /* specific case for names like: "\\.\d:" */
11121664424Sbellard     if (*path == '/' || *path == '\\')
11221664424Sbellard         return 1;
11321664424Sbellard #endif
11483f64091Sbellard     p = strchr(path, ':');
11583f64091Sbellard     if (p)
11683f64091Sbellard         p++;
11783f64091Sbellard     else
11883f64091Sbellard         p = path;
1193b9f94e1Sbellard #ifdef _WIN32
1203b9f94e1Sbellard     return (*p == '/' || *p == '\\');
1213b9f94e1Sbellard #else
1223b9f94e1Sbellard     return (*p == '/');
1233b9f94e1Sbellard #endif
12483f64091Sbellard }
12583f64091Sbellard 
12683f64091Sbellard /* if filename is absolute, just copy it to dest. Otherwise, build a
12783f64091Sbellard    path to it by considering it is relative to base_path. URL are
12883f64091Sbellard    supported. */
12983f64091Sbellard void path_combine(char *dest, int dest_size,
13083f64091Sbellard                   const char *base_path,
13183f64091Sbellard                   const char *filename)
13283f64091Sbellard {
13383f64091Sbellard     const char *p, *p1;
13483f64091Sbellard     int len;
13583f64091Sbellard 
13683f64091Sbellard     if (dest_size <= 0)
13783f64091Sbellard         return;
13883f64091Sbellard     if (path_is_absolute(filename)) {
13983f64091Sbellard         pstrcpy(dest, dest_size, filename);
14083f64091Sbellard     } else {
14183f64091Sbellard         p = strchr(base_path, ':');
14283f64091Sbellard         if (p)
14383f64091Sbellard             p++;
14483f64091Sbellard         else
14583f64091Sbellard             p = base_path;
1463b9f94e1Sbellard         p1 = strrchr(base_path, '/');
1473b9f94e1Sbellard #ifdef _WIN32
1483b9f94e1Sbellard         {
1493b9f94e1Sbellard             const char *p2;
1503b9f94e1Sbellard             p2 = strrchr(base_path, '\\');
1513b9f94e1Sbellard             if (!p1 || p2 > p1)
1523b9f94e1Sbellard                 p1 = p2;
1533b9f94e1Sbellard         }
1543b9f94e1Sbellard #endif
15583f64091Sbellard         if (p1)
15683f64091Sbellard             p1++;
15783f64091Sbellard         else
15883f64091Sbellard             p1 = base_path;
15983f64091Sbellard         if (p1 > p)
16083f64091Sbellard             p = p1;
16183f64091Sbellard         len = p - base_path;
16283f64091Sbellard         if (len > dest_size - 1)
16383f64091Sbellard             len = dest_size - 1;
16483f64091Sbellard         memcpy(dest, base_path, len);
16583f64091Sbellard         dest[len] = '\0';
16683f64091Sbellard         pstrcat(dest, dest_size, filename);
16783f64091Sbellard     }
16883f64091Sbellard }
16983f64091Sbellard 
1705efa9d5aSAnthony Liguori void bdrv_register(BlockDriver *bdrv)
171ea2384d3Sbellard {
172f141eafeSaliguori     if (!bdrv->bdrv_aio_readv) {
17383f64091Sbellard         /* add AIO emulation layer */
174f141eafeSaliguori         bdrv->bdrv_aio_readv = bdrv_aio_readv_em;
175f141eafeSaliguori         bdrv->bdrv_aio_writev = bdrv_aio_writev_em;
176eda578e5Saliguori     } else if (!bdrv->bdrv_read) {
17783f64091Sbellard         /* add synchronous IO emulation layer */
17883f64091Sbellard         bdrv->bdrv_read = bdrv_read_em;
17983f64091Sbellard         bdrv->bdrv_write = bdrv_write_em;
18083f64091Sbellard     }
181b2e12bc6SChristoph Hellwig 
182b2e12bc6SChristoph Hellwig     if (!bdrv->bdrv_aio_flush)
183b2e12bc6SChristoph Hellwig         bdrv->bdrv_aio_flush = bdrv_aio_flush_em;
184b2e12bc6SChristoph Hellwig 
1858a22f02aSStefan Hajnoczi     QLIST_INSERT_HEAD(&bdrv_drivers, bdrv, list);
186ea2384d3Sbellard }
187b338082bSbellard 
188b338082bSbellard /* create a new block device (by default it is empty) */
189b338082bSbellard BlockDriverState *bdrv_new(const char *device_name)
190fc01f7e7Sbellard {
1911b7bdbc1SStefan Hajnoczi     BlockDriverState *bs;
192b338082bSbellard 
193b338082bSbellard     bs = qemu_mallocz(sizeof(BlockDriverState));
194b338082bSbellard     pstrcpy(bs->device_name, sizeof(bs->device_name), device_name);
195ea2384d3Sbellard     if (device_name[0] != '\0') {
1961b7bdbc1SStefan Hajnoczi         QTAILQ_INSERT_TAIL(&bdrv_states, bs, list);
197ea2384d3Sbellard     }
198b338082bSbellard     return bs;
199b338082bSbellard }
200b338082bSbellard 
201ea2384d3Sbellard BlockDriver *bdrv_find_format(const char *format_name)
202ea2384d3Sbellard {
203ea2384d3Sbellard     BlockDriver *drv1;
2048a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
2058a22f02aSStefan Hajnoczi         if (!strcmp(drv1->format_name, format_name)) {
206ea2384d3Sbellard             return drv1;
207ea2384d3Sbellard         }
2088a22f02aSStefan Hajnoczi     }
209ea2384d3Sbellard     return NULL;
210ea2384d3Sbellard }
211ea2384d3Sbellard 
212eb852011SMarkus Armbruster static int bdrv_is_whitelisted(BlockDriver *drv)
213eb852011SMarkus Armbruster {
214eb852011SMarkus Armbruster     static const char *whitelist[] = {
215eb852011SMarkus Armbruster         CONFIG_BDRV_WHITELIST
216eb852011SMarkus Armbruster     };
217eb852011SMarkus Armbruster     const char **p;
218eb852011SMarkus Armbruster 
219eb852011SMarkus Armbruster     if (!whitelist[0])
220eb852011SMarkus Armbruster         return 1;               /* no whitelist, anything goes */
221eb852011SMarkus Armbruster 
222eb852011SMarkus Armbruster     for (p = whitelist; *p; p++) {
223eb852011SMarkus Armbruster         if (!strcmp(drv->format_name, *p)) {
224eb852011SMarkus Armbruster             return 1;
225eb852011SMarkus Armbruster         }
226eb852011SMarkus Armbruster     }
227eb852011SMarkus Armbruster     return 0;
228eb852011SMarkus Armbruster }
229eb852011SMarkus Armbruster 
230eb852011SMarkus Armbruster BlockDriver *bdrv_find_whitelisted_format(const char *format_name)
231eb852011SMarkus Armbruster {
232eb852011SMarkus Armbruster     BlockDriver *drv = bdrv_find_format(format_name);
233eb852011SMarkus Armbruster     return drv && bdrv_is_whitelisted(drv) ? drv : NULL;
234eb852011SMarkus Armbruster }
235eb852011SMarkus Armbruster 
2360e7e1989SKevin Wolf int bdrv_create(BlockDriver *drv, const char* filename,
2370e7e1989SKevin Wolf     QEMUOptionParameter *options)
238ea2384d3Sbellard {
239ea2384d3Sbellard     if (!drv->bdrv_create)
240ea2384d3Sbellard         return -ENOTSUP;
2410e7e1989SKevin Wolf 
2420e7e1989SKevin Wolf     return drv->bdrv_create(filename, options);
243ea2384d3Sbellard }
244ea2384d3Sbellard 
24584a12e66SChristoph Hellwig int bdrv_create_file(const char* filename, QEMUOptionParameter *options)
24684a12e66SChristoph Hellwig {
24784a12e66SChristoph Hellwig     BlockDriver *drv;
24884a12e66SChristoph Hellwig 
249b50cbabcSMORITA Kazutaka     drv = bdrv_find_protocol(filename);
25084a12e66SChristoph Hellwig     if (drv == NULL) {
25116905d71SStefan Hajnoczi         return -ENOENT;
25284a12e66SChristoph Hellwig     }
25384a12e66SChristoph Hellwig 
25484a12e66SChristoph Hellwig     return bdrv_create(drv, filename, options);
25584a12e66SChristoph Hellwig }
25684a12e66SChristoph Hellwig 
257d5249393Sbellard #ifdef _WIN32
25895389c86Sbellard void get_tmp_filename(char *filename, int size)
259d5249393Sbellard {
2603b9f94e1Sbellard     char temp_dir[MAX_PATH];
2613b9f94e1Sbellard 
2623b9f94e1Sbellard     GetTempPath(MAX_PATH, temp_dir);
2633b9f94e1Sbellard     GetTempFileName(temp_dir, "qem", 0, filename);
264d5249393Sbellard }
265d5249393Sbellard #else
26695389c86Sbellard void get_tmp_filename(char *filename, int size)
267ea2384d3Sbellard {
268ea2384d3Sbellard     int fd;
2697ccfb2ebSblueswir1     const char *tmpdir;
270d5249393Sbellard     /* XXX: race condition possible */
2710badc1eeSaurel32     tmpdir = getenv("TMPDIR");
2720badc1eeSaurel32     if (!tmpdir)
2730badc1eeSaurel32         tmpdir = "/tmp";
2740badc1eeSaurel32     snprintf(filename, size, "%s/vl.XXXXXX", tmpdir);
275ea2384d3Sbellard     fd = mkstemp(filename);
276ea2384d3Sbellard     close(fd);
277ea2384d3Sbellard }
278d5249393Sbellard #endif
279ea2384d3Sbellard 
280f3a5d3f8SChristoph Hellwig /*
281f3a5d3f8SChristoph Hellwig  * Detect host devices. By convention, /dev/cdrom[N] is always
282f3a5d3f8SChristoph Hellwig  * recognized as a host CDROM.
283f3a5d3f8SChristoph Hellwig  */
284f3a5d3f8SChristoph Hellwig static BlockDriver *find_hdev_driver(const char *filename)
285f3a5d3f8SChristoph Hellwig {
286508c7cb3SChristoph Hellwig     int score_max = 0, score;
287508c7cb3SChristoph Hellwig     BlockDriver *drv = NULL, *d;
288f3a5d3f8SChristoph Hellwig 
2898a22f02aSStefan Hajnoczi     QLIST_FOREACH(d, &bdrv_drivers, list) {
290508c7cb3SChristoph Hellwig         if (d->bdrv_probe_device) {
291508c7cb3SChristoph Hellwig             score = d->bdrv_probe_device(filename);
292508c7cb3SChristoph Hellwig             if (score > score_max) {
293508c7cb3SChristoph Hellwig                 score_max = score;
294508c7cb3SChristoph Hellwig                 drv = d;
295f3a5d3f8SChristoph Hellwig             }
296508c7cb3SChristoph Hellwig         }
297f3a5d3f8SChristoph Hellwig     }
298f3a5d3f8SChristoph Hellwig 
299508c7cb3SChristoph Hellwig     return drv;
300f3a5d3f8SChristoph Hellwig }
301f3a5d3f8SChristoph Hellwig 
302b50cbabcSMORITA Kazutaka BlockDriver *bdrv_find_protocol(const char *filename)
30384a12e66SChristoph Hellwig {
30484a12e66SChristoph Hellwig     BlockDriver *drv1;
30584a12e66SChristoph Hellwig     char protocol[128];
30684a12e66SChristoph Hellwig     int len;
30784a12e66SChristoph Hellwig     const char *p;
30884a12e66SChristoph Hellwig 
30966f82ceeSKevin Wolf     /* TODO Drivers without bdrv_file_open must be specified explicitly */
31066f82ceeSKevin Wolf 
31139508e7aSChristoph Hellwig     /*
31239508e7aSChristoph Hellwig      * XXX(hch): we really should not let host device detection
31339508e7aSChristoph Hellwig      * override an explicit protocol specification, but moving this
31439508e7aSChristoph Hellwig      * later breaks access to device names with colons in them.
31539508e7aSChristoph Hellwig      * Thanks to the brain-dead persistent naming schemes on udev-
31639508e7aSChristoph Hellwig      * based Linux systems those actually are quite common.
31739508e7aSChristoph Hellwig      */
31884a12e66SChristoph Hellwig     drv1 = find_hdev_driver(filename);
31939508e7aSChristoph Hellwig     if (drv1) {
32084a12e66SChristoph Hellwig         return drv1;
32184a12e66SChristoph Hellwig     }
32239508e7aSChristoph Hellwig 
3239e0b22f4SStefan Hajnoczi     if (!path_has_protocol(filename)) {
32439508e7aSChristoph Hellwig         return bdrv_find_format("file");
32539508e7aSChristoph Hellwig     }
3269e0b22f4SStefan Hajnoczi     p = strchr(filename, ':');
3279e0b22f4SStefan Hajnoczi     assert(p != NULL);
32884a12e66SChristoph Hellwig     len = p - filename;
32984a12e66SChristoph Hellwig     if (len > sizeof(protocol) - 1)
33084a12e66SChristoph Hellwig         len = sizeof(protocol) - 1;
33184a12e66SChristoph Hellwig     memcpy(protocol, filename, len);
33284a12e66SChristoph Hellwig     protocol[len] = '\0';
33384a12e66SChristoph Hellwig     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
33484a12e66SChristoph Hellwig         if (drv1->protocol_name &&
33584a12e66SChristoph Hellwig             !strcmp(drv1->protocol_name, protocol)) {
33684a12e66SChristoph Hellwig             return drv1;
33784a12e66SChristoph Hellwig         }
33884a12e66SChristoph Hellwig     }
33984a12e66SChristoph Hellwig     return NULL;
34084a12e66SChristoph Hellwig }
34184a12e66SChristoph Hellwig 
342c98ac35dSStefan Weil static int find_image_format(const char *filename, BlockDriver **pdrv)
343ea2384d3Sbellard {
34483f64091Sbellard     int ret, score, score_max;
345ea2384d3Sbellard     BlockDriver *drv1, *drv;
34683f64091Sbellard     uint8_t buf[2048];
34783f64091Sbellard     BlockDriverState *bs;
348ea2384d3Sbellard 
349f5edb014SNaphtali Sprei     ret = bdrv_file_open(&bs, filename, 0);
350c98ac35dSStefan Weil     if (ret < 0) {
351c98ac35dSStefan Weil         *pdrv = NULL;
352c98ac35dSStefan Weil         return ret;
353c98ac35dSStefan Weil     }
354f8ea0b00SNicholas Bellinger 
35508a00559SKevin Wolf     /* Return the raw BlockDriver * to scsi-generic devices or empty drives */
35608a00559SKevin Wolf     if (bs->sg || !bdrv_is_inserted(bs)) {
3571a396859SNicholas A. Bellinger         bdrv_delete(bs);
358c98ac35dSStefan Weil         drv = bdrv_find_format("raw");
359c98ac35dSStefan Weil         if (!drv) {
360c98ac35dSStefan Weil             ret = -ENOENT;
361c98ac35dSStefan Weil         }
362c98ac35dSStefan Weil         *pdrv = drv;
363c98ac35dSStefan Weil         return ret;
3641a396859SNicholas A. Bellinger     }
365f8ea0b00SNicholas Bellinger 
36683f64091Sbellard     ret = bdrv_pread(bs, 0, buf, sizeof(buf));
36783f64091Sbellard     bdrv_delete(bs);
368ea2384d3Sbellard     if (ret < 0) {
369c98ac35dSStefan Weil         *pdrv = NULL;
370c98ac35dSStefan Weil         return ret;
371ea2384d3Sbellard     }
372ea2384d3Sbellard 
373ea2384d3Sbellard     score_max = 0;
37484a12e66SChristoph Hellwig     drv = NULL;
3758a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv1, &bdrv_drivers, list) {
37683f64091Sbellard         if (drv1->bdrv_probe) {
377ea2384d3Sbellard             score = drv1->bdrv_probe(buf, ret, filename);
378ea2384d3Sbellard             if (score > score_max) {
379ea2384d3Sbellard                 score_max = score;
380ea2384d3Sbellard                 drv = drv1;
381ea2384d3Sbellard             }
382ea2384d3Sbellard         }
38383f64091Sbellard     }
384c98ac35dSStefan Weil     if (!drv) {
385c98ac35dSStefan Weil         ret = -ENOENT;
386c98ac35dSStefan Weil     }
387c98ac35dSStefan Weil     *pdrv = drv;
388c98ac35dSStefan Weil     return ret;
389ea2384d3Sbellard }
390ea2384d3Sbellard 
39151762288SStefan Hajnoczi /**
39251762288SStefan Hajnoczi  * Set the current 'total_sectors' value
39351762288SStefan Hajnoczi  */
39451762288SStefan Hajnoczi static int refresh_total_sectors(BlockDriverState *bs, int64_t hint)
39551762288SStefan Hajnoczi {
39651762288SStefan Hajnoczi     BlockDriver *drv = bs->drv;
39751762288SStefan Hajnoczi 
398396759adSNicholas Bellinger     /* Do not attempt drv->bdrv_getlength() on scsi-generic devices */
399396759adSNicholas Bellinger     if (bs->sg)
400396759adSNicholas Bellinger         return 0;
401396759adSNicholas Bellinger 
40251762288SStefan Hajnoczi     /* query actual device if possible, otherwise just trust the hint */
40351762288SStefan Hajnoczi     if (drv->bdrv_getlength) {
40451762288SStefan Hajnoczi         int64_t length = drv->bdrv_getlength(bs);
40551762288SStefan Hajnoczi         if (length < 0) {
40651762288SStefan Hajnoczi             return length;
40751762288SStefan Hajnoczi         }
40851762288SStefan Hajnoczi         hint = length >> BDRV_SECTOR_BITS;
40951762288SStefan Hajnoczi     }
41051762288SStefan Hajnoczi 
41151762288SStefan Hajnoczi     bs->total_sectors = hint;
41251762288SStefan Hajnoczi     return 0;
41351762288SStefan Hajnoczi }
41451762288SStefan Hajnoczi 
415b6ce07aaSKevin Wolf /*
41657915332SKevin Wolf  * Common part for opening disk images and files
41757915332SKevin Wolf  */
41857915332SKevin Wolf static int bdrv_open_common(BlockDriverState *bs, const char *filename,
41957915332SKevin Wolf     int flags, BlockDriver *drv)
42057915332SKevin Wolf {
42157915332SKevin Wolf     int ret, open_flags;
42257915332SKevin Wolf 
42357915332SKevin Wolf     assert(drv != NULL);
42457915332SKevin Wolf 
42566f82ceeSKevin Wolf     bs->file = NULL;
42651762288SStefan Hajnoczi     bs->total_sectors = 0;
42757915332SKevin Wolf     bs->encrypted = 0;
42857915332SKevin Wolf     bs->valid_key = 0;
42957915332SKevin Wolf     bs->open_flags = flags;
43057915332SKevin Wolf     /* buffer_alignment defaulted to 512, drivers can change this value */
43157915332SKevin Wolf     bs->buffer_alignment = 512;
43257915332SKevin Wolf 
43357915332SKevin Wolf     pstrcpy(bs->filename, sizeof(bs->filename), filename);
43457915332SKevin Wolf 
43557915332SKevin Wolf     if (use_bdrv_whitelist && !bdrv_is_whitelisted(drv)) {
43657915332SKevin Wolf         return -ENOTSUP;
43757915332SKevin Wolf     }
43857915332SKevin Wolf 
43957915332SKevin Wolf     bs->drv = drv;
44057915332SKevin Wolf     bs->opaque = qemu_mallocz(drv->instance_size);
44157915332SKevin Wolf 
442a6599793SChristoph Hellwig     if (flags & BDRV_O_CACHE_WB)
44357915332SKevin Wolf         bs->enable_write_cache = 1;
44457915332SKevin Wolf 
44557915332SKevin Wolf     /*
44657915332SKevin Wolf      * Clear flags that are internal to the block layer before opening the
44757915332SKevin Wolf      * image.
44857915332SKevin Wolf      */
44957915332SKevin Wolf     open_flags = flags & ~(BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
45057915332SKevin Wolf 
45157915332SKevin Wolf     /*
452ebabb67aSStefan Weil      * Snapshots should be writable.
45357915332SKevin Wolf      */
45457915332SKevin Wolf     if (bs->is_temporary) {
45557915332SKevin Wolf         open_flags |= BDRV_O_RDWR;
45657915332SKevin Wolf     }
45757915332SKevin Wolf 
45866f82ceeSKevin Wolf     /* Open the image, either directly or using a protocol */
45966f82ceeSKevin Wolf     if (drv->bdrv_file_open) {
46066f82ceeSKevin Wolf         ret = drv->bdrv_file_open(bs, filename, open_flags);
46166f82ceeSKevin Wolf     } else {
46266f82ceeSKevin Wolf         ret = bdrv_file_open(&bs->file, filename, open_flags);
46366f82ceeSKevin Wolf         if (ret >= 0) {
46466f82ceeSKevin Wolf             ret = drv->bdrv_open(bs, open_flags);
46566f82ceeSKevin Wolf         }
46666f82ceeSKevin Wolf     }
46766f82ceeSKevin Wolf 
46857915332SKevin Wolf     if (ret < 0) {
46957915332SKevin Wolf         goto free_and_fail;
47057915332SKevin Wolf     }
47157915332SKevin Wolf 
47257915332SKevin Wolf     bs->keep_read_only = bs->read_only = !(open_flags & BDRV_O_RDWR);
47351762288SStefan Hajnoczi 
47451762288SStefan Hajnoczi     ret = refresh_total_sectors(bs, bs->total_sectors);
47551762288SStefan Hajnoczi     if (ret < 0) {
47651762288SStefan Hajnoczi         goto free_and_fail;
47757915332SKevin Wolf     }
47851762288SStefan Hajnoczi 
47957915332SKevin Wolf #ifndef _WIN32
48057915332SKevin Wolf     if (bs->is_temporary) {
48157915332SKevin Wolf         unlink(filename);
48257915332SKevin Wolf     }
48357915332SKevin Wolf #endif
48457915332SKevin Wolf     return 0;
48557915332SKevin Wolf 
48657915332SKevin Wolf free_and_fail:
48766f82ceeSKevin Wolf     if (bs->file) {
48866f82ceeSKevin Wolf         bdrv_delete(bs->file);
48966f82ceeSKevin Wolf         bs->file = NULL;
49066f82ceeSKevin Wolf     }
49157915332SKevin Wolf     qemu_free(bs->opaque);
49257915332SKevin Wolf     bs->opaque = NULL;
49357915332SKevin Wolf     bs->drv = NULL;
49457915332SKevin Wolf     return ret;
49557915332SKevin Wolf }
49657915332SKevin Wolf 
49757915332SKevin Wolf /*
498b6ce07aaSKevin Wolf  * Opens a file using a protocol (file, host_device, nbd, ...)
499b6ce07aaSKevin Wolf  */
50083f64091Sbellard int bdrv_file_open(BlockDriverState **pbs, const char *filename, int flags)
501b338082bSbellard {
50283f64091Sbellard     BlockDriverState *bs;
5036db95603SChristoph Hellwig     BlockDriver *drv;
50483f64091Sbellard     int ret;
5053b0d4f61Sbellard 
506b50cbabcSMORITA Kazutaka     drv = bdrv_find_protocol(filename);
5076db95603SChristoph Hellwig     if (!drv) {
5086db95603SChristoph Hellwig         return -ENOENT;
5096db95603SChristoph Hellwig     }
5106db95603SChristoph Hellwig 
51183f64091Sbellard     bs = bdrv_new("");
512b6ce07aaSKevin Wolf     ret = bdrv_open_common(bs, filename, flags, drv);
51383f64091Sbellard     if (ret < 0) {
51483f64091Sbellard         bdrv_delete(bs);
51583f64091Sbellard         return ret;
5163b0d4f61Sbellard     }
51771d0770cSaliguori     bs->growable = 1;
51883f64091Sbellard     *pbs = bs;
51983f64091Sbellard     return 0;
5203b0d4f61Sbellard }
5213b0d4f61Sbellard 
522b6ce07aaSKevin Wolf /*
523b6ce07aaSKevin Wolf  * Opens a disk image (raw, qcow2, vmdk, ...)
524b6ce07aaSKevin Wolf  */
525d6e9098eSKevin Wolf int bdrv_open(BlockDriverState *bs, const char *filename, int flags,
526ea2384d3Sbellard               BlockDriver *drv)
527ea2384d3Sbellard {
528b6ce07aaSKevin Wolf     int ret;
52933e3963eSbellard 
53083f64091Sbellard     if (flags & BDRV_O_SNAPSHOT) {
531ea2384d3Sbellard         BlockDriverState *bs1;
532ea2384d3Sbellard         int64_t total_size;
5337c96d46eSaliguori         int is_protocol = 0;
53491a073a9SKevin Wolf         BlockDriver *bdrv_qcow2;
53591a073a9SKevin Wolf         QEMUOptionParameter *options;
536b6ce07aaSKevin Wolf         char tmp_filename[PATH_MAX];
537b6ce07aaSKevin Wolf         char backing_filename[PATH_MAX];
53833e3963eSbellard 
539ea2384d3Sbellard         /* if snapshot, we create a temporary backing file and open it
540ea2384d3Sbellard            instead of opening 'filename' directly */
541ea2384d3Sbellard 
542ea2384d3Sbellard         /* if there is a backing file, use it */
543ea2384d3Sbellard         bs1 = bdrv_new("");
544d6e9098eSKevin Wolf         ret = bdrv_open(bs1, filename, 0, drv);
54551d7c00cSaliguori         if (ret < 0) {
546ea2384d3Sbellard             bdrv_delete(bs1);
54751d7c00cSaliguori             return ret;
548ea2384d3Sbellard         }
5493e82990bSJes Sorensen         total_size = bdrv_getlength(bs1) & BDRV_SECTOR_MASK;
5507c96d46eSaliguori 
5517c96d46eSaliguori         if (bs1->drv && bs1->drv->protocol_name)
5527c96d46eSaliguori             is_protocol = 1;
5537c96d46eSaliguori 
554ea2384d3Sbellard         bdrv_delete(bs1);
555ea2384d3Sbellard 
556ea2384d3Sbellard         get_tmp_filename(tmp_filename, sizeof(tmp_filename));
5577c96d46eSaliguori 
5587c96d46eSaliguori         /* Real path is meaningless for protocols */
5597c96d46eSaliguori         if (is_protocol)
5607c96d46eSaliguori             snprintf(backing_filename, sizeof(backing_filename),
5617c96d46eSaliguori                      "%s", filename);
562114cdfa9SKirill A. Shutemov         else if (!realpath(filename, backing_filename))
563114cdfa9SKirill A. Shutemov             return -errno;
5647c96d46eSaliguori 
56591a073a9SKevin Wolf         bdrv_qcow2 = bdrv_find_format("qcow2");
56691a073a9SKevin Wolf         options = parse_option_parameters("", bdrv_qcow2->create_options, NULL);
56791a073a9SKevin Wolf 
5683e82990bSJes Sorensen         set_option_parameter_int(options, BLOCK_OPT_SIZE, total_size);
56991a073a9SKevin Wolf         set_option_parameter(options, BLOCK_OPT_BACKING_FILE, backing_filename);
57091a073a9SKevin Wolf         if (drv) {
57191a073a9SKevin Wolf             set_option_parameter(options, BLOCK_OPT_BACKING_FMT,
57291a073a9SKevin Wolf                 drv->format_name);
57391a073a9SKevin Wolf         }
57491a073a9SKevin Wolf 
57591a073a9SKevin Wolf         ret = bdrv_create(bdrv_qcow2, tmp_filename, options);
576d748768cSJan Kiszka         free_option_parameters(options);
57751d7c00cSaliguori         if (ret < 0) {
57851d7c00cSaliguori             return ret;
579ea2384d3Sbellard         }
58091a073a9SKevin Wolf 
581ea2384d3Sbellard         filename = tmp_filename;
58291a073a9SKevin Wolf         drv = bdrv_qcow2;
583ea2384d3Sbellard         bs->is_temporary = 1;
584ea2384d3Sbellard     }
585ea2384d3Sbellard 
586b6ce07aaSKevin Wolf     /* Find the right image format driver */
5876db95603SChristoph Hellwig     if (!drv) {
588c98ac35dSStefan Weil         ret = find_image_format(filename, &drv);
589ea2384d3Sbellard     }
5906987307cSChristoph Hellwig 
59151d7c00cSaliguori     if (!drv) {
59251d7c00cSaliguori         goto unlink_and_fail;
59383f64091Sbellard     }
594b6ce07aaSKevin Wolf 
595b6ce07aaSKevin Wolf     /* Open the image */
596b6ce07aaSKevin Wolf     ret = bdrv_open_common(bs, filename, flags, drv);
597b6ce07aaSKevin Wolf     if (ret < 0) {
5986987307cSChristoph Hellwig         goto unlink_and_fail;
5996987307cSChristoph Hellwig     }
6006987307cSChristoph Hellwig 
601b6ce07aaSKevin Wolf     /* If there is a backing file, use it */
602b6ce07aaSKevin Wolf     if ((flags & BDRV_O_NO_BACKING) == 0 && bs->backing_file[0] != '\0') {
603b6ce07aaSKevin Wolf         char backing_filename[PATH_MAX];
604b6ce07aaSKevin Wolf         int back_flags;
605b6ce07aaSKevin Wolf         BlockDriver *back_drv = NULL;
606b6ce07aaSKevin Wolf 
607b6ce07aaSKevin Wolf         bs->backing_hd = bdrv_new("");
608df2dbb4aSStefan Hajnoczi 
609df2dbb4aSStefan Hajnoczi         if (path_has_protocol(bs->backing_file)) {
610df2dbb4aSStefan Hajnoczi             pstrcpy(backing_filename, sizeof(backing_filename),
611df2dbb4aSStefan Hajnoczi                     bs->backing_file);
612df2dbb4aSStefan Hajnoczi         } else {
613b6ce07aaSKevin Wolf             path_combine(backing_filename, sizeof(backing_filename),
614b6ce07aaSKevin Wolf                          filename, bs->backing_file);
615df2dbb4aSStefan Hajnoczi         }
616df2dbb4aSStefan Hajnoczi 
617df2dbb4aSStefan Hajnoczi         if (bs->backing_format[0] != '\0') {
618b6ce07aaSKevin Wolf             back_drv = bdrv_find_format(bs->backing_format);
619df2dbb4aSStefan Hajnoczi         }
620b6ce07aaSKevin Wolf 
621b6ce07aaSKevin Wolf         /* backing files always opened read-only */
622b6ce07aaSKevin Wolf         back_flags =
623b6ce07aaSKevin Wolf             flags & ~(BDRV_O_RDWR | BDRV_O_SNAPSHOT | BDRV_O_NO_BACKING);
624b6ce07aaSKevin Wolf 
625b6ce07aaSKevin Wolf         ret = bdrv_open(bs->backing_hd, backing_filename, back_flags, back_drv);
626b6ce07aaSKevin Wolf         if (ret < 0) {
627b6ce07aaSKevin Wolf             bdrv_close(bs);
628b6ce07aaSKevin Wolf             return ret;
629b6ce07aaSKevin Wolf         }
630b6ce07aaSKevin Wolf         if (bs->is_temporary) {
631b6ce07aaSKevin Wolf             bs->backing_hd->keep_read_only = !(flags & BDRV_O_RDWR);
632b6ce07aaSKevin Wolf         } else {
633b6ce07aaSKevin Wolf             /* base image inherits from "parent" */
634b6ce07aaSKevin Wolf             bs->backing_hd->keep_read_only = bs->keep_read_only;
635b6ce07aaSKevin Wolf         }
636b6ce07aaSKevin Wolf     }
637b6ce07aaSKevin Wolf 
638b6ce07aaSKevin Wolf     if (!bdrv_key_required(bs)) {
639b6ce07aaSKevin Wolf         /* call the change callback */
640b6ce07aaSKevin Wolf         bs->media_changed = 1;
641b6ce07aaSKevin Wolf         if (bs->change_cb)
642db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
643b6ce07aaSKevin Wolf     }
644b6ce07aaSKevin Wolf 
645b6ce07aaSKevin Wolf     return 0;
646b6ce07aaSKevin Wolf 
647b6ce07aaSKevin Wolf unlink_and_fail:
648b6ce07aaSKevin Wolf     if (bs->is_temporary) {
649b6ce07aaSKevin Wolf         unlink(filename);
650b6ce07aaSKevin Wolf     }
651b6ce07aaSKevin Wolf     return ret;
652b6ce07aaSKevin Wolf }
653b6ce07aaSKevin Wolf 
654fc01f7e7Sbellard void bdrv_close(BlockDriverState *bs)
655fc01f7e7Sbellard {
65619cb3738Sbellard     if (bs->drv) {
657f9092b10SMarkus Armbruster         if (bs == bs_snapshots) {
658f9092b10SMarkus Armbruster             bs_snapshots = NULL;
659f9092b10SMarkus Armbruster         }
660557df6acSStefan Hajnoczi         if (bs->backing_hd) {
661ea2384d3Sbellard             bdrv_delete(bs->backing_hd);
662557df6acSStefan Hajnoczi             bs->backing_hd = NULL;
663557df6acSStefan Hajnoczi         }
664ea2384d3Sbellard         bs->drv->bdrv_close(bs);
665ea2384d3Sbellard         qemu_free(bs->opaque);
666ea2384d3Sbellard #ifdef _WIN32
667ea2384d3Sbellard         if (bs->is_temporary) {
668ea2384d3Sbellard             unlink(bs->filename);
669ea2384d3Sbellard         }
67067b915a5Sbellard #endif
671ea2384d3Sbellard         bs->opaque = NULL;
672ea2384d3Sbellard         bs->drv = NULL;
673b338082bSbellard 
67466f82ceeSKevin Wolf         if (bs->file != NULL) {
67566f82ceeSKevin Wolf             bdrv_close(bs->file);
67666f82ceeSKevin Wolf         }
67766f82ceeSKevin Wolf 
678b338082bSbellard         /* call the change callback */
67919cb3738Sbellard         bs->media_changed = 1;
680b338082bSbellard         if (bs->change_cb)
681db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
682b338082bSbellard     }
683b338082bSbellard }
684b338082bSbellard 
6852bc93fedSMORITA Kazutaka void bdrv_close_all(void)
6862bc93fedSMORITA Kazutaka {
6872bc93fedSMORITA Kazutaka     BlockDriverState *bs;
6882bc93fedSMORITA Kazutaka 
6892bc93fedSMORITA Kazutaka     QTAILQ_FOREACH(bs, &bdrv_states, list) {
6902bc93fedSMORITA Kazutaka         bdrv_close(bs);
6912bc93fedSMORITA Kazutaka     }
6922bc93fedSMORITA Kazutaka }
6932bc93fedSMORITA Kazutaka 
694d22b2f41SRyan Harper /* make a BlockDriverState anonymous by removing from bdrv_state list.
695d22b2f41SRyan Harper    Also, NULL terminate the device_name to prevent double remove */
696d22b2f41SRyan Harper void bdrv_make_anon(BlockDriverState *bs)
697d22b2f41SRyan Harper {
698d22b2f41SRyan Harper     if (bs->device_name[0] != '\0') {
699d22b2f41SRyan Harper         QTAILQ_REMOVE(&bdrv_states, bs, list);
700d22b2f41SRyan Harper     }
701d22b2f41SRyan Harper     bs->device_name[0] = '\0';
702d22b2f41SRyan Harper }
703d22b2f41SRyan Harper 
704b338082bSbellard void bdrv_delete(BlockDriverState *bs)
705b338082bSbellard {
70618846deeSMarkus Armbruster     assert(!bs->peer);
70718846deeSMarkus Armbruster 
7081b7bdbc1SStefan Hajnoczi     /* remove from list, if necessary */
709d22b2f41SRyan Harper     bdrv_make_anon(bs);
71034c6f050Saurel32 
711b338082bSbellard     bdrv_close(bs);
71266f82ceeSKevin Wolf     if (bs->file != NULL) {
71366f82ceeSKevin Wolf         bdrv_delete(bs->file);
71466f82ceeSKevin Wolf     }
71566f82ceeSKevin Wolf 
716f9092b10SMarkus Armbruster     assert(bs != bs_snapshots);
717b338082bSbellard     qemu_free(bs);
718fc01f7e7Sbellard }
719fc01f7e7Sbellard 
72018846deeSMarkus Armbruster int bdrv_attach(BlockDriverState *bs, DeviceState *qdev)
72118846deeSMarkus Armbruster {
72218846deeSMarkus Armbruster     if (bs->peer) {
72318846deeSMarkus Armbruster         return -EBUSY;
72418846deeSMarkus Armbruster     }
72518846deeSMarkus Armbruster     bs->peer = qdev;
72618846deeSMarkus Armbruster     return 0;
72718846deeSMarkus Armbruster }
72818846deeSMarkus Armbruster 
72918846deeSMarkus Armbruster void bdrv_detach(BlockDriverState *bs, DeviceState *qdev)
73018846deeSMarkus Armbruster {
73118846deeSMarkus Armbruster     assert(bs->peer == qdev);
73218846deeSMarkus Armbruster     bs->peer = NULL;
73318846deeSMarkus Armbruster }
73418846deeSMarkus Armbruster 
73518846deeSMarkus Armbruster DeviceState *bdrv_get_attached(BlockDriverState *bs)
73618846deeSMarkus Armbruster {
73718846deeSMarkus Armbruster     return bs->peer;
73818846deeSMarkus Armbruster }
73918846deeSMarkus Armbruster 
740e97fc193Saliguori /*
741e97fc193Saliguori  * Run consistency checks on an image
742e97fc193Saliguori  *
743e076f338SKevin Wolf  * Returns 0 if the check could be completed (it doesn't mean that the image is
744a1c7273bSStefan Weil  * free of errors) or -errno when an internal error occurred. The results of the
745e076f338SKevin Wolf  * check are stored in res.
746e97fc193Saliguori  */
747e076f338SKevin Wolf int bdrv_check(BlockDriverState *bs, BdrvCheckResult *res)
748e97fc193Saliguori {
749e97fc193Saliguori     if (bs->drv->bdrv_check == NULL) {
750e97fc193Saliguori         return -ENOTSUP;
751e97fc193Saliguori     }
752e97fc193Saliguori 
753e076f338SKevin Wolf     memset(res, 0, sizeof(*res));
7549ac228e0SKevin Wolf     return bs->drv->bdrv_check(bs, res);
755e97fc193Saliguori }
756e97fc193Saliguori 
7578a426614SKevin Wolf #define COMMIT_BUF_SECTORS 2048
7588a426614SKevin Wolf 
75933e3963eSbellard /* commit COW file into the raw image */
76033e3963eSbellard int bdrv_commit(BlockDriverState *bs)
76133e3963eSbellard {
76219cb3738Sbellard     BlockDriver *drv = bs->drv;
763ee181196SKevin Wolf     BlockDriver *backing_drv;
7648a426614SKevin Wolf     int64_t sector, total_sectors;
7658a426614SKevin Wolf     int n, ro, open_flags;
7664dca4b63SNaphtali Sprei     int ret = 0, rw_ret = 0;
7678a426614SKevin Wolf     uint8_t *buf;
7684dca4b63SNaphtali Sprei     char filename[1024];
7694dca4b63SNaphtali Sprei     BlockDriverState *bs_rw, *bs_ro;
77033e3963eSbellard 
77119cb3738Sbellard     if (!drv)
77219cb3738Sbellard         return -ENOMEDIUM;
77333e3963eSbellard 
7744dca4b63SNaphtali Sprei     if (!bs->backing_hd) {
7754dca4b63SNaphtali Sprei         return -ENOTSUP;
7764dca4b63SNaphtali Sprei     }
7774dca4b63SNaphtali Sprei 
7784dca4b63SNaphtali Sprei     if (bs->backing_hd->keep_read_only) {
779ea2384d3Sbellard         return -EACCES;
78033e3963eSbellard     }
78133e3963eSbellard 
782ee181196SKevin Wolf     backing_drv = bs->backing_hd->drv;
7834dca4b63SNaphtali Sprei     ro = bs->backing_hd->read_only;
7844dca4b63SNaphtali Sprei     strncpy(filename, bs->backing_hd->filename, sizeof(filename));
7854dca4b63SNaphtali Sprei     open_flags =  bs->backing_hd->open_flags;
7864dca4b63SNaphtali Sprei 
7874dca4b63SNaphtali Sprei     if (ro) {
7884dca4b63SNaphtali Sprei         /* re-open as RW */
7894dca4b63SNaphtali Sprei         bdrv_delete(bs->backing_hd);
7904dca4b63SNaphtali Sprei         bs->backing_hd = NULL;
7914dca4b63SNaphtali Sprei         bs_rw = bdrv_new("");
792ee181196SKevin Wolf         rw_ret = bdrv_open(bs_rw, filename, open_flags | BDRV_O_RDWR,
793ee181196SKevin Wolf             backing_drv);
7944dca4b63SNaphtali Sprei         if (rw_ret < 0) {
7954dca4b63SNaphtali Sprei             bdrv_delete(bs_rw);
7964dca4b63SNaphtali Sprei             /* try to re-open read-only */
7974dca4b63SNaphtali Sprei             bs_ro = bdrv_new("");
798ee181196SKevin Wolf             ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
799ee181196SKevin Wolf                 backing_drv);
8004dca4b63SNaphtali Sprei             if (ret < 0) {
8014dca4b63SNaphtali Sprei                 bdrv_delete(bs_ro);
8024dca4b63SNaphtali Sprei                 /* drive not functional anymore */
8034dca4b63SNaphtali Sprei                 bs->drv = NULL;
8044dca4b63SNaphtali Sprei                 return ret;
8054dca4b63SNaphtali Sprei             }
8064dca4b63SNaphtali Sprei             bs->backing_hd = bs_ro;
8074dca4b63SNaphtali Sprei             return rw_ret;
8084dca4b63SNaphtali Sprei         }
8094dca4b63SNaphtali Sprei         bs->backing_hd = bs_rw;
810ea2384d3Sbellard     }
811ea2384d3Sbellard 
8126ea44308SJan Kiszka     total_sectors = bdrv_getlength(bs) >> BDRV_SECTOR_BITS;
8138a426614SKevin Wolf     buf = qemu_malloc(COMMIT_BUF_SECTORS * BDRV_SECTOR_SIZE);
8148a426614SKevin Wolf 
8158a426614SKevin Wolf     for (sector = 0; sector < total_sectors; sector += n) {
8168a426614SKevin Wolf         if (drv->bdrv_is_allocated(bs, sector, COMMIT_BUF_SECTORS, &n)) {
8178a426614SKevin Wolf 
8188a426614SKevin Wolf             if (bdrv_read(bs, sector, buf, n) != 0) {
8194dca4b63SNaphtali Sprei                 ret = -EIO;
8204dca4b63SNaphtali Sprei                 goto ro_cleanup;
82133e3963eSbellard             }
82233e3963eSbellard 
8238a426614SKevin Wolf             if (bdrv_write(bs->backing_hd, sector, buf, n) != 0) {
8244dca4b63SNaphtali Sprei                 ret = -EIO;
8254dca4b63SNaphtali Sprei                 goto ro_cleanup;
82633e3963eSbellard             }
82733e3963eSbellard         }
82833e3963eSbellard     }
82995389c86Sbellard 
8301d44952fSChristoph Hellwig     if (drv->bdrv_make_empty) {
8311d44952fSChristoph Hellwig         ret = drv->bdrv_make_empty(bs);
8321d44952fSChristoph Hellwig         bdrv_flush(bs);
8331d44952fSChristoph Hellwig     }
83495389c86Sbellard 
8353f5075aeSChristoph Hellwig     /*
8363f5075aeSChristoph Hellwig      * Make sure all data we wrote to the backing device is actually
8373f5075aeSChristoph Hellwig      * stable on disk.
8383f5075aeSChristoph Hellwig      */
8393f5075aeSChristoph Hellwig     if (bs->backing_hd)
8403f5075aeSChristoph Hellwig         bdrv_flush(bs->backing_hd);
8414dca4b63SNaphtali Sprei 
8424dca4b63SNaphtali Sprei ro_cleanup:
8438a426614SKevin Wolf     qemu_free(buf);
8444dca4b63SNaphtali Sprei 
8454dca4b63SNaphtali Sprei     if (ro) {
8464dca4b63SNaphtali Sprei         /* re-open as RO */
8474dca4b63SNaphtali Sprei         bdrv_delete(bs->backing_hd);
8484dca4b63SNaphtali Sprei         bs->backing_hd = NULL;
8494dca4b63SNaphtali Sprei         bs_ro = bdrv_new("");
850ee181196SKevin Wolf         ret = bdrv_open(bs_ro, filename, open_flags & ~BDRV_O_RDWR,
851ee181196SKevin Wolf             backing_drv);
8524dca4b63SNaphtali Sprei         if (ret < 0) {
8534dca4b63SNaphtali Sprei             bdrv_delete(bs_ro);
8544dca4b63SNaphtali Sprei             /* drive not functional anymore */
8554dca4b63SNaphtali Sprei             bs->drv = NULL;
8564dca4b63SNaphtali Sprei             return ret;
8574dca4b63SNaphtali Sprei         }
8584dca4b63SNaphtali Sprei         bs->backing_hd = bs_ro;
8594dca4b63SNaphtali Sprei         bs->backing_hd->keep_read_only = 0;
8604dca4b63SNaphtali Sprei     }
8614dca4b63SNaphtali Sprei 
8621d44952fSChristoph Hellwig     return ret;
86333e3963eSbellard }
86433e3963eSbellard 
8656ab4b5abSMarkus Armbruster void bdrv_commit_all(void)
8666ab4b5abSMarkus Armbruster {
8676ab4b5abSMarkus Armbruster     BlockDriverState *bs;
8686ab4b5abSMarkus Armbruster 
8696ab4b5abSMarkus Armbruster     QTAILQ_FOREACH(bs, &bdrv_states, list) {
8706ab4b5abSMarkus Armbruster         bdrv_commit(bs);
8716ab4b5abSMarkus Armbruster     }
8726ab4b5abSMarkus Armbruster }
8736ab4b5abSMarkus Armbruster 
874756e6736SKevin Wolf /*
875756e6736SKevin Wolf  * Return values:
876756e6736SKevin Wolf  * 0        - success
877756e6736SKevin Wolf  * -EINVAL  - backing format specified, but no file
878756e6736SKevin Wolf  * -ENOSPC  - can't update the backing file because no space is left in the
879756e6736SKevin Wolf  *            image file header
880756e6736SKevin Wolf  * -ENOTSUP - format driver doesn't support changing the backing file
881756e6736SKevin Wolf  */
882756e6736SKevin Wolf int bdrv_change_backing_file(BlockDriverState *bs,
883756e6736SKevin Wolf     const char *backing_file, const char *backing_fmt)
884756e6736SKevin Wolf {
885756e6736SKevin Wolf     BlockDriver *drv = bs->drv;
886756e6736SKevin Wolf 
887756e6736SKevin Wolf     if (drv->bdrv_change_backing_file != NULL) {
888756e6736SKevin Wolf         return drv->bdrv_change_backing_file(bs, backing_file, backing_fmt);
889756e6736SKevin Wolf     } else {
890756e6736SKevin Wolf         return -ENOTSUP;
891756e6736SKevin Wolf     }
892756e6736SKevin Wolf }
893756e6736SKevin Wolf 
89471d0770cSaliguori static int bdrv_check_byte_request(BlockDriverState *bs, int64_t offset,
89571d0770cSaliguori                                    size_t size)
89671d0770cSaliguori {
89771d0770cSaliguori     int64_t len;
89871d0770cSaliguori 
89971d0770cSaliguori     if (!bdrv_is_inserted(bs))
90071d0770cSaliguori         return -ENOMEDIUM;
90171d0770cSaliguori 
90271d0770cSaliguori     if (bs->growable)
90371d0770cSaliguori         return 0;
90471d0770cSaliguori 
90571d0770cSaliguori     len = bdrv_getlength(bs);
90671d0770cSaliguori 
907fbb7b4e0SKevin Wolf     if (offset < 0)
908fbb7b4e0SKevin Wolf         return -EIO;
909fbb7b4e0SKevin Wolf 
910fbb7b4e0SKevin Wolf     if ((offset > len) || (len - offset < size))
91171d0770cSaliguori         return -EIO;
91271d0770cSaliguori 
91371d0770cSaliguori     return 0;
91471d0770cSaliguori }
91571d0770cSaliguori 
91671d0770cSaliguori static int bdrv_check_request(BlockDriverState *bs, int64_t sector_num,
91771d0770cSaliguori                               int nb_sectors)
91871d0770cSaliguori {
919eb5a3165SJes Sorensen     return bdrv_check_byte_request(bs, sector_num * BDRV_SECTOR_SIZE,
920eb5a3165SJes Sorensen                                    nb_sectors * BDRV_SECTOR_SIZE);
92171d0770cSaliguori }
92271d0770cSaliguori 
92319cb3738Sbellard /* return < 0 if error. See bdrv_write() for the return codes */
924fc01f7e7Sbellard int bdrv_read(BlockDriverState *bs, int64_t sector_num,
925fc01f7e7Sbellard               uint8_t *buf, int nb_sectors)
926fc01f7e7Sbellard {
927ea2384d3Sbellard     BlockDriver *drv = bs->drv;
928fc01f7e7Sbellard 
92919cb3738Sbellard     if (!drv)
93019cb3738Sbellard         return -ENOMEDIUM;
93171d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
93271d0770cSaliguori         return -EIO;
933b338082bSbellard 
93483f64091Sbellard     return drv->bdrv_read(bs, sector_num, buf, nb_sectors);
93583f64091Sbellard }
936fc01f7e7Sbellard 
9377cd1e32aSlirans@il.ibm.com static void set_dirty_bitmap(BlockDriverState *bs, int64_t sector_num,
9387cd1e32aSlirans@il.ibm.com                              int nb_sectors, int dirty)
9397cd1e32aSlirans@il.ibm.com {
9407cd1e32aSlirans@il.ibm.com     int64_t start, end;
941c6d22830SJan Kiszka     unsigned long val, idx, bit;
942a55eb92cSJan Kiszka 
9436ea44308SJan Kiszka     start = sector_num / BDRV_SECTORS_PER_DIRTY_CHUNK;
944c6d22830SJan Kiszka     end = (sector_num + nb_sectors - 1) / BDRV_SECTORS_PER_DIRTY_CHUNK;
9457cd1e32aSlirans@il.ibm.com 
9467cd1e32aSlirans@il.ibm.com     for (; start <= end; start++) {
947c6d22830SJan Kiszka         idx = start / (sizeof(unsigned long) * 8);
948c6d22830SJan Kiszka         bit = start % (sizeof(unsigned long) * 8);
949c6d22830SJan Kiszka         val = bs->dirty_bitmap[idx];
950c6d22830SJan Kiszka         if (dirty) {
9516d59fec1SMarcelo Tosatti             if (!(val & (1UL << bit))) {
952aaa0eb75SLiran Schour                 bs->dirty_count++;
9536d59fec1SMarcelo Tosatti                 val |= 1UL << bit;
954aaa0eb75SLiran Schour             }
955c6d22830SJan Kiszka         } else {
9566d59fec1SMarcelo Tosatti             if (val & (1UL << bit)) {
957aaa0eb75SLiran Schour                 bs->dirty_count--;
9586d59fec1SMarcelo Tosatti                 val &= ~(1UL << bit);
959c6d22830SJan Kiszka             }
960aaa0eb75SLiran Schour         }
961c6d22830SJan Kiszka         bs->dirty_bitmap[idx] = val;
9627cd1e32aSlirans@il.ibm.com     }
9637cd1e32aSlirans@il.ibm.com }
9647cd1e32aSlirans@il.ibm.com 
96519cb3738Sbellard /* Return < 0 if error. Important errors are:
96619cb3738Sbellard   -EIO         generic I/O error (may happen for all errors)
96719cb3738Sbellard   -ENOMEDIUM   No media inserted.
96819cb3738Sbellard   -EINVAL      Invalid sector number or nb_sectors
96919cb3738Sbellard   -EACCES      Trying to write a read-only device
97019cb3738Sbellard */
971fc01f7e7Sbellard int bdrv_write(BlockDriverState *bs, int64_t sector_num,
972fc01f7e7Sbellard                const uint8_t *buf, int nb_sectors)
973fc01f7e7Sbellard {
97483f64091Sbellard     BlockDriver *drv = bs->drv;
97519cb3738Sbellard     if (!bs->drv)
97619cb3738Sbellard         return -ENOMEDIUM;
9770849bf08Sbellard     if (bs->read_only)
97819cb3738Sbellard         return -EACCES;
97971d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
98071d0770cSaliguori         return -EIO;
98171d0770cSaliguori 
982c6d22830SJan Kiszka     if (bs->dirty_bitmap) {
9837cd1e32aSlirans@il.ibm.com         set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
9847cd1e32aSlirans@il.ibm.com     }
9857cd1e32aSlirans@il.ibm.com 
986294cc35fSKevin Wolf     if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
987294cc35fSKevin Wolf         bs->wr_highest_sector = sector_num + nb_sectors - 1;
988294cc35fSKevin Wolf     }
989294cc35fSKevin Wolf 
99083f64091Sbellard     return drv->bdrv_write(bs, sector_num, buf, nb_sectors);
99183f64091Sbellard }
99283f64091Sbellard 
993eda578e5Saliguori int bdrv_pread(BlockDriverState *bs, int64_t offset,
994eda578e5Saliguori                void *buf, int count1)
99583f64091Sbellard {
9966ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
99783f64091Sbellard     int len, nb_sectors, count;
99883f64091Sbellard     int64_t sector_num;
9999a8c4cceSKevin Wolf     int ret;
100083f64091Sbellard 
100183f64091Sbellard     count = count1;
100283f64091Sbellard     /* first read to align to sector start */
10036ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
100483f64091Sbellard     if (len > count)
100583f64091Sbellard         len = count;
10066ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
100783f64091Sbellard     if (len > 0) {
10089a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
10099a8c4cceSKevin Wolf             return ret;
10106ea44308SJan Kiszka         memcpy(buf, tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), len);
101183f64091Sbellard         count -= len;
101283f64091Sbellard         if (count == 0)
101383f64091Sbellard             return count1;
101483f64091Sbellard         sector_num++;
101583f64091Sbellard         buf += len;
101683f64091Sbellard     }
101783f64091Sbellard 
101883f64091Sbellard     /* read the sectors "in place" */
10196ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
102083f64091Sbellard     if (nb_sectors > 0) {
10219a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, buf, nb_sectors)) < 0)
10229a8c4cceSKevin Wolf             return ret;
102383f64091Sbellard         sector_num += nb_sectors;
10246ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
102583f64091Sbellard         buf += len;
102683f64091Sbellard         count -= len;
102783f64091Sbellard     }
102883f64091Sbellard 
102983f64091Sbellard     /* add data from the last sector */
103083f64091Sbellard     if (count > 0) {
10319a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
10329a8c4cceSKevin Wolf             return ret;
103383f64091Sbellard         memcpy(buf, tmp_buf, count);
103483f64091Sbellard     }
103583f64091Sbellard     return count1;
103683f64091Sbellard }
103783f64091Sbellard 
1038eda578e5Saliguori int bdrv_pwrite(BlockDriverState *bs, int64_t offset,
1039eda578e5Saliguori                 const void *buf, int count1)
104083f64091Sbellard {
10416ea44308SJan Kiszka     uint8_t tmp_buf[BDRV_SECTOR_SIZE];
104283f64091Sbellard     int len, nb_sectors, count;
104383f64091Sbellard     int64_t sector_num;
10449a8c4cceSKevin Wolf     int ret;
104583f64091Sbellard 
104683f64091Sbellard     count = count1;
104783f64091Sbellard     /* first write to align to sector start */
10486ea44308SJan Kiszka     len = (BDRV_SECTOR_SIZE - offset) & (BDRV_SECTOR_SIZE - 1);
104983f64091Sbellard     if (len > count)
105083f64091Sbellard         len = count;
10516ea44308SJan Kiszka     sector_num = offset >> BDRV_SECTOR_BITS;
105283f64091Sbellard     if (len > 0) {
10539a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
10549a8c4cceSKevin Wolf             return ret;
10556ea44308SJan Kiszka         memcpy(tmp_buf + (offset & (BDRV_SECTOR_SIZE - 1)), buf, len);
10569a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
10579a8c4cceSKevin Wolf             return ret;
105883f64091Sbellard         count -= len;
105983f64091Sbellard         if (count == 0)
106083f64091Sbellard             return count1;
106183f64091Sbellard         sector_num++;
106283f64091Sbellard         buf += len;
106383f64091Sbellard     }
106483f64091Sbellard 
106583f64091Sbellard     /* write the sectors "in place" */
10666ea44308SJan Kiszka     nb_sectors = count >> BDRV_SECTOR_BITS;
106783f64091Sbellard     if (nb_sectors > 0) {
10689a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, buf, nb_sectors)) < 0)
10699a8c4cceSKevin Wolf             return ret;
107083f64091Sbellard         sector_num += nb_sectors;
10716ea44308SJan Kiszka         len = nb_sectors << BDRV_SECTOR_BITS;
107283f64091Sbellard         buf += len;
107383f64091Sbellard         count -= len;
107483f64091Sbellard     }
107583f64091Sbellard 
107683f64091Sbellard     /* add data from the last sector */
107783f64091Sbellard     if (count > 0) {
10789a8c4cceSKevin Wolf         if ((ret = bdrv_read(bs, sector_num, tmp_buf, 1)) < 0)
10799a8c4cceSKevin Wolf             return ret;
108083f64091Sbellard         memcpy(tmp_buf, buf, count);
10819a8c4cceSKevin Wolf         if ((ret = bdrv_write(bs, sector_num, tmp_buf, 1)) < 0)
10829a8c4cceSKevin Wolf             return ret;
108383f64091Sbellard     }
108483f64091Sbellard     return count1;
108583f64091Sbellard }
108683f64091Sbellard 
1087f08145feSKevin Wolf /*
1088f08145feSKevin Wolf  * Writes to the file and ensures that no writes are reordered across this
1089f08145feSKevin Wolf  * request (acts as a barrier)
1090f08145feSKevin Wolf  *
1091f08145feSKevin Wolf  * Returns 0 on success, -errno in error cases.
1092f08145feSKevin Wolf  */
1093f08145feSKevin Wolf int bdrv_pwrite_sync(BlockDriverState *bs, int64_t offset,
1094f08145feSKevin Wolf     const void *buf, int count)
1095f08145feSKevin Wolf {
1096f08145feSKevin Wolf     int ret;
1097f08145feSKevin Wolf 
1098f08145feSKevin Wolf     ret = bdrv_pwrite(bs, offset, buf, count);
1099f08145feSKevin Wolf     if (ret < 0) {
1100f08145feSKevin Wolf         return ret;
1101f08145feSKevin Wolf     }
1102f08145feSKevin Wolf 
1103f08145feSKevin Wolf     /* No flush needed for cache=writethrough, it uses O_DSYNC */
1104f08145feSKevin Wolf     if ((bs->open_flags & BDRV_O_CACHE_MASK) != 0) {
1105f08145feSKevin Wolf         bdrv_flush(bs);
1106f08145feSKevin Wolf     }
1107f08145feSKevin Wolf 
1108f08145feSKevin Wolf     return 0;
1109f08145feSKevin Wolf }
1110f08145feSKevin Wolf 
1111f08145feSKevin Wolf /*
1112f08145feSKevin Wolf  * Writes to the file and ensures that no writes are reordered across this
1113f08145feSKevin Wolf  * request (acts as a barrier)
1114f08145feSKevin Wolf  *
1115f08145feSKevin Wolf  * Returns 0 on success, -errno in error cases.
1116f08145feSKevin Wolf  */
1117f08145feSKevin Wolf int bdrv_write_sync(BlockDriverState *bs, int64_t sector_num,
1118f08145feSKevin Wolf     const uint8_t *buf, int nb_sectors)
1119f08145feSKevin Wolf {
1120f08145feSKevin Wolf     return bdrv_pwrite_sync(bs, BDRV_SECTOR_SIZE * sector_num,
1121f08145feSKevin Wolf         buf, BDRV_SECTOR_SIZE * nb_sectors);
1122f08145feSKevin Wolf }
1123f08145feSKevin Wolf 
112483f64091Sbellard /**
112583f64091Sbellard  * Truncate file to 'offset' bytes (needed only for file protocols)
112683f64091Sbellard  */
112783f64091Sbellard int bdrv_truncate(BlockDriverState *bs, int64_t offset)
112883f64091Sbellard {
112983f64091Sbellard     BlockDriver *drv = bs->drv;
113051762288SStefan Hajnoczi     int ret;
113183f64091Sbellard     if (!drv)
113219cb3738Sbellard         return -ENOMEDIUM;
113383f64091Sbellard     if (!drv->bdrv_truncate)
113483f64091Sbellard         return -ENOTSUP;
113559f2689dSNaphtali Sprei     if (bs->read_only)
113659f2689dSNaphtali Sprei         return -EACCES;
11378591675fSMarcelo Tosatti     if (bdrv_in_use(bs))
11388591675fSMarcelo Tosatti         return -EBUSY;
113951762288SStefan Hajnoczi     ret = drv->bdrv_truncate(bs, offset);
114051762288SStefan Hajnoczi     if (ret == 0) {
114151762288SStefan Hajnoczi         ret = refresh_total_sectors(bs, offset >> BDRV_SECTOR_BITS);
1142db97ee6aSChristoph Hellwig         if (bs->change_cb) {
1143db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_SIZE);
1144db97ee6aSChristoph Hellwig         }
114551762288SStefan Hajnoczi     }
114651762288SStefan Hajnoczi     return ret;
114783f64091Sbellard }
114883f64091Sbellard 
114983f64091Sbellard /**
115083f64091Sbellard  * Length of a file in bytes. Return < 0 if error or unknown.
115183f64091Sbellard  */
115283f64091Sbellard int64_t bdrv_getlength(BlockDriverState *bs)
115383f64091Sbellard {
115483f64091Sbellard     BlockDriver *drv = bs->drv;
115583f64091Sbellard     if (!drv)
115619cb3738Sbellard         return -ENOMEDIUM;
115751762288SStefan Hajnoczi 
115846a4e4e6SStefan Hajnoczi     if (bs->growable || bs->removable) {
115946a4e4e6SStefan Hajnoczi         if (drv->bdrv_getlength) {
116083f64091Sbellard             return drv->bdrv_getlength(bs);
1161fc01f7e7Sbellard         }
116246a4e4e6SStefan Hajnoczi     }
116346a4e4e6SStefan Hajnoczi     return bs->total_sectors * BDRV_SECTOR_SIZE;
116446a4e4e6SStefan Hajnoczi }
1165fc01f7e7Sbellard 
116619cb3738Sbellard /* return 0 as number of sectors if no device present or error */
116796b8f136Sths void bdrv_get_geometry(BlockDriverState *bs, uint64_t *nb_sectors_ptr)
1168fc01f7e7Sbellard {
116919cb3738Sbellard     int64_t length;
117019cb3738Sbellard     length = bdrv_getlength(bs);
117119cb3738Sbellard     if (length < 0)
117219cb3738Sbellard         length = 0;
117319cb3738Sbellard     else
11746ea44308SJan Kiszka         length = length >> BDRV_SECTOR_BITS;
117519cb3738Sbellard     *nb_sectors_ptr = length;
1176fc01f7e7Sbellard }
1177cf98951bSbellard 
1178f3d54fc4Saliguori struct partition {
1179f3d54fc4Saliguori         uint8_t boot_ind;           /* 0x80 - active */
1180f3d54fc4Saliguori         uint8_t head;               /* starting head */
1181f3d54fc4Saliguori         uint8_t sector;             /* starting sector */
1182f3d54fc4Saliguori         uint8_t cyl;                /* starting cylinder */
1183f3d54fc4Saliguori         uint8_t sys_ind;            /* What partition type */
1184f3d54fc4Saliguori         uint8_t end_head;           /* end head */
1185f3d54fc4Saliguori         uint8_t end_sector;         /* end sector */
1186f3d54fc4Saliguori         uint8_t end_cyl;            /* end cylinder */
1187f3d54fc4Saliguori         uint32_t start_sect;        /* starting sector counting from 0 */
1188f3d54fc4Saliguori         uint32_t nr_sects;          /* nr of sectors in partition */
1189f3d54fc4Saliguori } __attribute__((packed));
1190f3d54fc4Saliguori 
1191f3d54fc4Saliguori /* try to guess the disk logical geometry from the MSDOS partition table. Return 0 if OK, -1 if could not guess */
1192f3d54fc4Saliguori static int guess_disk_lchs(BlockDriverState *bs,
1193f3d54fc4Saliguori                            int *pcylinders, int *pheads, int *psectors)
1194f3d54fc4Saliguori {
1195eb5a3165SJes Sorensen     uint8_t buf[BDRV_SECTOR_SIZE];
1196f3d54fc4Saliguori     int ret, i, heads, sectors, cylinders;
1197f3d54fc4Saliguori     struct partition *p;
1198f3d54fc4Saliguori     uint32_t nr_sects;
1199a38131b6Sblueswir1     uint64_t nb_sectors;
1200f3d54fc4Saliguori 
1201f3d54fc4Saliguori     bdrv_get_geometry(bs, &nb_sectors);
1202f3d54fc4Saliguori 
1203f3d54fc4Saliguori     ret = bdrv_read(bs, 0, buf, 1);
1204f3d54fc4Saliguori     if (ret < 0)
1205f3d54fc4Saliguori         return -1;
1206f3d54fc4Saliguori     /* test msdos magic */
1207f3d54fc4Saliguori     if (buf[510] != 0x55 || buf[511] != 0xaa)
1208f3d54fc4Saliguori         return -1;
1209f3d54fc4Saliguori     for(i = 0; i < 4; i++) {
1210f3d54fc4Saliguori         p = ((struct partition *)(buf + 0x1be)) + i;
1211f3d54fc4Saliguori         nr_sects = le32_to_cpu(p->nr_sects);
1212f3d54fc4Saliguori         if (nr_sects && p->end_head) {
1213f3d54fc4Saliguori             /* We make the assumption that the partition terminates on
1214f3d54fc4Saliguori                a cylinder boundary */
1215f3d54fc4Saliguori             heads = p->end_head + 1;
1216f3d54fc4Saliguori             sectors = p->end_sector & 63;
1217f3d54fc4Saliguori             if (sectors == 0)
1218f3d54fc4Saliguori                 continue;
1219f3d54fc4Saliguori             cylinders = nb_sectors / (heads * sectors);
1220f3d54fc4Saliguori             if (cylinders < 1 || cylinders > 16383)
1221f3d54fc4Saliguori                 continue;
1222f3d54fc4Saliguori             *pheads = heads;
1223f3d54fc4Saliguori             *psectors = sectors;
1224f3d54fc4Saliguori             *pcylinders = cylinders;
1225f3d54fc4Saliguori #if 0
1226f3d54fc4Saliguori             printf("guessed geometry: LCHS=%d %d %d\n",
1227f3d54fc4Saliguori                    cylinders, heads, sectors);
1228f3d54fc4Saliguori #endif
1229f3d54fc4Saliguori             return 0;
1230f3d54fc4Saliguori         }
1231f3d54fc4Saliguori     }
1232f3d54fc4Saliguori     return -1;
1233f3d54fc4Saliguori }
1234f3d54fc4Saliguori 
1235f3d54fc4Saliguori void bdrv_guess_geometry(BlockDriverState *bs, int *pcyls, int *pheads, int *psecs)
1236f3d54fc4Saliguori {
1237f3d54fc4Saliguori     int translation, lba_detected = 0;
1238f3d54fc4Saliguori     int cylinders, heads, secs;
1239a38131b6Sblueswir1     uint64_t nb_sectors;
1240f3d54fc4Saliguori 
1241f3d54fc4Saliguori     /* if a geometry hint is available, use it */
1242f3d54fc4Saliguori     bdrv_get_geometry(bs, &nb_sectors);
1243f3d54fc4Saliguori     bdrv_get_geometry_hint(bs, &cylinders, &heads, &secs);
1244f3d54fc4Saliguori     translation = bdrv_get_translation_hint(bs);
1245f3d54fc4Saliguori     if (cylinders != 0) {
1246f3d54fc4Saliguori         *pcyls = cylinders;
1247f3d54fc4Saliguori         *pheads = heads;
1248f3d54fc4Saliguori         *psecs = secs;
1249f3d54fc4Saliguori     } else {
1250f3d54fc4Saliguori         if (guess_disk_lchs(bs, &cylinders, &heads, &secs) == 0) {
1251f3d54fc4Saliguori             if (heads > 16) {
1252f3d54fc4Saliguori                 /* if heads > 16, it means that a BIOS LBA
1253f3d54fc4Saliguori                    translation was active, so the default
1254f3d54fc4Saliguori                    hardware geometry is OK */
1255f3d54fc4Saliguori                 lba_detected = 1;
1256f3d54fc4Saliguori                 goto default_geometry;
1257f3d54fc4Saliguori             } else {
1258f3d54fc4Saliguori                 *pcyls = cylinders;
1259f3d54fc4Saliguori                 *pheads = heads;
1260f3d54fc4Saliguori                 *psecs = secs;
1261f3d54fc4Saliguori                 /* disable any translation to be in sync with
1262f3d54fc4Saliguori                    the logical geometry */
1263f3d54fc4Saliguori                 if (translation == BIOS_ATA_TRANSLATION_AUTO) {
1264f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
1265f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_NONE);
1266f3d54fc4Saliguori                 }
1267f3d54fc4Saliguori             }
1268f3d54fc4Saliguori         } else {
1269f3d54fc4Saliguori         default_geometry:
1270f3d54fc4Saliguori             /* if no geometry, use a standard physical disk geometry */
1271f3d54fc4Saliguori             cylinders = nb_sectors / (16 * 63);
1272f3d54fc4Saliguori 
1273f3d54fc4Saliguori             if (cylinders > 16383)
1274f3d54fc4Saliguori                 cylinders = 16383;
1275f3d54fc4Saliguori             else if (cylinders < 2)
1276f3d54fc4Saliguori                 cylinders = 2;
1277f3d54fc4Saliguori             *pcyls = cylinders;
1278f3d54fc4Saliguori             *pheads = 16;
1279f3d54fc4Saliguori             *psecs = 63;
1280f3d54fc4Saliguori             if ((lba_detected == 1) && (translation == BIOS_ATA_TRANSLATION_AUTO)) {
1281f3d54fc4Saliguori                 if ((*pcyls * *pheads) <= 131072) {
1282f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
1283f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_LARGE);
1284f3d54fc4Saliguori                 } else {
1285f3d54fc4Saliguori                     bdrv_set_translation_hint(bs,
1286f3d54fc4Saliguori                                               BIOS_ATA_TRANSLATION_LBA);
1287f3d54fc4Saliguori                 }
1288f3d54fc4Saliguori             }
1289f3d54fc4Saliguori         }
1290f3d54fc4Saliguori         bdrv_set_geometry_hint(bs, *pcyls, *pheads, *psecs);
1291f3d54fc4Saliguori     }
1292f3d54fc4Saliguori }
1293f3d54fc4Saliguori 
1294b338082bSbellard void bdrv_set_geometry_hint(BlockDriverState *bs,
1295b338082bSbellard                             int cyls, int heads, int secs)
1296b338082bSbellard {
1297b338082bSbellard     bs->cyls = cyls;
1298b338082bSbellard     bs->heads = heads;
1299b338082bSbellard     bs->secs = secs;
1300b338082bSbellard }
1301b338082bSbellard 
130246d4767dSbellard void bdrv_set_translation_hint(BlockDriverState *bs, int translation)
130346d4767dSbellard {
130446d4767dSbellard     bs->translation = translation;
130546d4767dSbellard }
130646d4767dSbellard 
1307b338082bSbellard void bdrv_get_geometry_hint(BlockDriverState *bs,
1308b338082bSbellard                             int *pcyls, int *pheads, int *psecs)
1309b338082bSbellard {
1310b338082bSbellard     *pcyls = bs->cyls;
1311b338082bSbellard     *pheads = bs->heads;
1312b338082bSbellard     *psecs = bs->secs;
1313b338082bSbellard }
1314b338082bSbellard 
13155bbdbb46SBlue Swirl /* Recognize floppy formats */
13165bbdbb46SBlue Swirl typedef struct FDFormat {
13175bbdbb46SBlue Swirl     FDriveType drive;
13185bbdbb46SBlue Swirl     uint8_t last_sect;
13195bbdbb46SBlue Swirl     uint8_t max_track;
13205bbdbb46SBlue Swirl     uint8_t max_head;
13215bbdbb46SBlue Swirl } FDFormat;
13225bbdbb46SBlue Swirl 
13235bbdbb46SBlue Swirl static const FDFormat fd_formats[] = {
13245bbdbb46SBlue Swirl     /* First entry is default format */
13255bbdbb46SBlue Swirl     /* 1.44 MB 3"1/2 floppy disks */
13265bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 18, 80, 1, },
13275bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 20, 80, 1, },
13285bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 21, 80, 1, },
13295bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 21, 82, 1, },
13305bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 21, 83, 1, },
13315bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 22, 80, 1, },
13325bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 23, 80, 1, },
13335bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 24, 80, 1, },
13345bbdbb46SBlue Swirl     /* 2.88 MB 3"1/2 floppy disks */
13355bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 36, 80, 1, },
13365bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 39, 80, 1, },
13375bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 40, 80, 1, },
13385bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 44, 80, 1, },
13395bbdbb46SBlue Swirl     { FDRIVE_DRV_288, 48, 80, 1, },
13405bbdbb46SBlue Swirl     /* 720 kB 3"1/2 floppy disks */
13415bbdbb46SBlue Swirl     { FDRIVE_DRV_144,  9, 80, 1, },
13425bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 10, 80, 1, },
13435bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 10, 82, 1, },
13445bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 10, 83, 1, },
13455bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 13, 80, 1, },
13465bbdbb46SBlue Swirl     { FDRIVE_DRV_144, 14, 80, 1, },
13475bbdbb46SBlue Swirl     /* 1.2 MB 5"1/4 floppy disks */
13485bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 15, 80, 1, },
13495bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 18, 80, 1, },
13505bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 18, 82, 1, },
13515bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 18, 83, 1, },
13525bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 20, 80, 1, },
13535bbdbb46SBlue Swirl     /* 720 kB 5"1/4 floppy disks */
13545bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  9, 80, 1, },
13555bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 11, 80, 1, },
13565bbdbb46SBlue Swirl     /* 360 kB 5"1/4 floppy disks */
13575bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  9, 40, 1, },
13585bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  9, 40, 0, },
13595bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 10, 41, 1, },
13605bbdbb46SBlue Swirl     { FDRIVE_DRV_120, 10, 42, 1, },
13615bbdbb46SBlue Swirl     /* 320 kB 5"1/4 floppy disks */
13625bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  8, 40, 1, },
13635bbdbb46SBlue Swirl     { FDRIVE_DRV_120,  8, 40, 0, },
13645bbdbb46SBlue Swirl     /* 360 kB must match 5"1/4 better than 3"1/2... */
13655bbdbb46SBlue Swirl     { FDRIVE_DRV_144,  9, 80, 0, },
13665bbdbb46SBlue Swirl     /* end */
13675bbdbb46SBlue Swirl     { FDRIVE_DRV_NONE, -1, -1, 0, },
13685bbdbb46SBlue Swirl };
13695bbdbb46SBlue Swirl 
13705bbdbb46SBlue Swirl void bdrv_get_floppy_geometry_hint(BlockDriverState *bs, int *nb_heads,
13715bbdbb46SBlue Swirl                                    int *max_track, int *last_sect,
13725bbdbb46SBlue Swirl                                    FDriveType drive_in, FDriveType *drive)
13735bbdbb46SBlue Swirl {
13745bbdbb46SBlue Swirl     const FDFormat *parse;
13755bbdbb46SBlue Swirl     uint64_t nb_sectors, size;
13765bbdbb46SBlue Swirl     int i, first_match, match;
13775bbdbb46SBlue Swirl 
13785bbdbb46SBlue Swirl     bdrv_get_geometry_hint(bs, nb_heads, max_track, last_sect);
13795bbdbb46SBlue Swirl     if (*nb_heads != 0 && *max_track != 0 && *last_sect != 0) {
13805bbdbb46SBlue Swirl         /* User defined disk */
13815bbdbb46SBlue Swirl     } else {
13825bbdbb46SBlue Swirl         bdrv_get_geometry(bs, &nb_sectors);
13835bbdbb46SBlue Swirl         match = -1;
13845bbdbb46SBlue Swirl         first_match = -1;
13855bbdbb46SBlue Swirl         for (i = 0; ; i++) {
13865bbdbb46SBlue Swirl             parse = &fd_formats[i];
13875bbdbb46SBlue Swirl             if (parse->drive == FDRIVE_DRV_NONE) {
13885bbdbb46SBlue Swirl                 break;
13895bbdbb46SBlue Swirl             }
13905bbdbb46SBlue Swirl             if (drive_in == parse->drive ||
13915bbdbb46SBlue Swirl                 drive_in == FDRIVE_DRV_NONE) {
13925bbdbb46SBlue Swirl                 size = (parse->max_head + 1) * parse->max_track *
13935bbdbb46SBlue Swirl                     parse->last_sect;
13945bbdbb46SBlue Swirl                 if (nb_sectors == size) {
13955bbdbb46SBlue Swirl                     match = i;
13965bbdbb46SBlue Swirl                     break;
13975bbdbb46SBlue Swirl                 }
13985bbdbb46SBlue Swirl                 if (first_match == -1) {
13995bbdbb46SBlue Swirl                     first_match = i;
14005bbdbb46SBlue Swirl                 }
14015bbdbb46SBlue Swirl             }
14025bbdbb46SBlue Swirl         }
14035bbdbb46SBlue Swirl         if (match == -1) {
14045bbdbb46SBlue Swirl             if (first_match == -1) {
14055bbdbb46SBlue Swirl                 match = 1;
14065bbdbb46SBlue Swirl             } else {
14075bbdbb46SBlue Swirl                 match = first_match;
14085bbdbb46SBlue Swirl             }
14095bbdbb46SBlue Swirl             parse = &fd_formats[match];
14105bbdbb46SBlue Swirl         }
14115bbdbb46SBlue Swirl         *nb_heads = parse->max_head + 1;
14125bbdbb46SBlue Swirl         *max_track = parse->max_track;
14135bbdbb46SBlue Swirl         *last_sect = parse->last_sect;
14145bbdbb46SBlue Swirl         *drive = parse->drive;
14155bbdbb46SBlue Swirl     }
14165bbdbb46SBlue Swirl }
14175bbdbb46SBlue Swirl 
141846d4767dSbellard int bdrv_get_translation_hint(BlockDriverState *bs)
141946d4767dSbellard {
142046d4767dSbellard     return bs->translation;
142146d4767dSbellard }
142246d4767dSbellard 
1423abd7f68dSMarkus Armbruster void bdrv_set_on_error(BlockDriverState *bs, BlockErrorAction on_read_error,
1424abd7f68dSMarkus Armbruster                        BlockErrorAction on_write_error)
1425abd7f68dSMarkus Armbruster {
1426abd7f68dSMarkus Armbruster     bs->on_read_error = on_read_error;
1427abd7f68dSMarkus Armbruster     bs->on_write_error = on_write_error;
1428abd7f68dSMarkus Armbruster }
1429abd7f68dSMarkus Armbruster 
1430abd7f68dSMarkus Armbruster BlockErrorAction bdrv_get_on_error(BlockDriverState *bs, int is_read)
1431abd7f68dSMarkus Armbruster {
1432abd7f68dSMarkus Armbruster     return is_read ? bs->on_read_error : bs->on_write_error;
1433abd7f68dSMarkus Armbruster }
1434abd7f68dSMarkus Armbruster 
14357d0d6950SMarkus Armbruster void bdrv_set_removable(BlockDriverState *bs, int removable)
14367d0d6950SMarkus Armbruster {
14377d0d6950SMarkus Armbruster     bs->removable = removable;
14387d0d6950SMarkus Armbruster     if (removable && bs == bs_snapshots) {
14397d0d6950SMarkus Armbruster         bs_snapshots = NULL;
14407d0d6950SMarkus Armbruster     }
14417d0d6950SMarkus Armbruster }
14427d0d6950SMarkus Armbruster 
1443b338082bSbellard int bdrv_is_removable(BlockDriverState *bs)
1444b338082bSbellard {
1445b338082bSbellard     return bs->removable;
1446b338082bSbellard }
1447b338082bSbellard 
1448b338082bSbellard int bdrv_is_read_only(BlockDriverState *bs)
1449b338082bSbellard {
1450b338082bSbellard     return bs->read_only;
1451b338082bSbellard }
1452b338082bSbellard 
1453985a03b0Sths int bdrv_is_sg(BlockDriverState *bs)
1454985a03b0Sths {
1455985a03b0Sths     return bs->sg;
1456985a03b0Sths }
1457985a03b0Sths 
1458e900a7b7SChristoph Hellwig int bdrv_enable_write_cache(BlockDriverState *bs)
1459e900a7b7SChristoph Hellwig {
1460e900a7b7SChristoph Hellwig     return bs->enable_write_cache;
1461e900a7b7SChristoph Hellwig }
1462e900a7b7SChristoph Hellwig 
146319cb3738Sbellard /* XXX: no longer used */
1464b338082bSbellard void bdrv_set_change_cb(BlockDriverState *bs,
1465db97ee6aSChristoph Hellwig                         void (*change_cb)(void *opaque, int reason),
1466db97ee6aSChristoph Hellwig                         void *opaque)
1467b338082bSbellard {
1468b338082bSbellard     bs->change_cb = change_cb;
1469b338082bSbellard     bs->change_opaque = opaque;
1470b338082bSbellard }
1471b338082bSbellard 
1472ea2384d3Sbellard int bdrv_is_encrypted(BlockDriverState *bs)
1473ea2384d3Sbellard {
1474ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted)
1475ea2384d3Sbellard         return 1;
1476ea2384d3Sbellard     return bs->encrypted;
1477ea2384d3Sbellard }
1478ea2384d3Sbellard 
1479c0f4ce77Saliguori int bdrv_key_required(BlockDriverState *bs)
1480c0f4ce77Saliguori {
1481c0f4ce77Saliguori     BlockDriverState *backing_hd = bs->backing_hd;
1482c0f4ce77Saliguori 
1483c0f4ce77Saliguori     if (backing_hd && backing_hd->encrypted && !backing_hd->valid_key)
1484c0f4ce77Saliguori         return 1;
1485c0f4ce77Saliguori     return (bs->encrypted && !bs->valid_key);
1486c0f4ce77Saliguori }
1487c0f4ce77Saliguori 
1488ea2384d3Sbellard int bdrv_set_key(BlockDriverState *bs, const char *key)
1489ea2384d3Sbellard {
1490ea2384d3Sbellard     int ret;
1491ea2384d3Sbellard     if (bs->backing_hd && bs->backing_hd->encrypted) {
1492ea2384d3Sbellard         ret = bdrv_set_key(bs->backing_hd, key);
1493ea2384d3Sbellard         if (ret < 0)
1494ea2384d3Sbellard             return ret;
1495ea2384d3Sbellard         if (!bs->encrypted)
1496ea2384d3Sbellard             return 0;
1497ea2384d3Sbellard     }
1498fd04a2aeSShahar Havivi     if (!bs->encrypted) {
1499fd04a2aeSShahar Havivi         return -EINVAL;
1500fd04a2aeSShahar Havivi     } else if (!bs->drv || !bs->drv->bdrv_set_key) {
1501fd04a2aeSShahar Havivi         return -ENOMEDIUM;
1502fd04a2aeSShahar Havivi     }
1503c0f4ce77Saliguori     ret = bs->drv->bdrv_set_key(bs, key);
1504bb5fc20fSaliguori     if (ret < 0) {
1505bb5fc20fSaliguori         bs->valid_key = 0;
1506bb5fc20fSaliguori     } else if (!bs->valid_key) {
1507bb5fc20fSaliguori         bs->valid_key = 1;
1508bb5fc20fSaliguori         /* call the change callback now, we skipped it on open */
1509bb5fc20fSaliguori         bs->media_changed = 1;
1510bb5fc20fSaliguori         if (bs->change_cb)
1511db97ee6aSChristoph Hellwig             bs->change_cb(bs->change_opaque, CHANGE_MEDIA);
1512bb5fc20fSaliguori     }
1513c0f4ce77Saliguori     return ret;
1514ea2384d3Sbellard }
1515ea2384d3Sbellard 
1516ea2384d3Sbellard void bdrv_get_format(BlockDriverState *bs, char *buf, int buf_size)
1517ea2384d3Sbellard {
151819cb3738Sbellard     if (!bs->drv) {
1519ea2384d3Sbellard         buf[0] = '\0';
1520ea2384d3Sbellard     } else {
1521ea2384d3Sbellard         pstrcpy(buf, buf_size, bs->drv->format_name);
1522ea2384d3Sbellard     }
1523ea2384d3Sbellard }
1524ea2384d3Sbellard 
1525ea2384d3Sbellard void bdrv_iterate_format(void (*it)(void *opaque, const char *name),
1526ea2384d3Sbellard                          void *opaque)
1527ea2384d3Sbellard {
1528ea2384d3Sbellard     BlockDriver *drv;
1529ea2384d3Sbellard 
15308a22f02aSStefan Hajnoczi     QLIST_FOREACH(drv, &bdrv_drivers, list) {
1531ea2384d3Sbellard         it(opaque, drv->format_name);
1532ea2384d3Sbellard     }
1533ea2384d3Sbellard }
1534ea2384d3Sbellard 
1535b338082bSbellard BlockDriverState *bdrv_find(const char *name)
1536b338082bSbellard {
1537b338082bSbellard     BlockDriverState *bs;
1538b338082bSbellard 
15391b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
15401b7bdbc1SStefan Hajnoczi         if (!strcmp(name, bs->device_name)) {
1541b338082bSbellard             return bs;
1542b338082bSbellard         }
15431b7bdbc1SStefan Hajnoczi     }
1544b338082bSbellard     return NULL;
1545b338082bSbellard }
1546b338082bSbellard 
15472f399b0aSMarkus Armbruster BlockDriverState *bdrv_next(BlockDriverState *bs)
15482f399b0aSMarkus Armbruster {
15492f399b0aSMarkus Armbruster     if (!bs) {
15502f399b0aSMarkus Armbruster         return QTAILQ_FIRST(&bdrv_states);
15512f399b0aSMarkus Armbruster     }
15522f399b0aSMarkus Armbruster     return QTAILQ_NEXT(bs, list);
15532f399b0aSMarkus Armbruster }
15542f399b0aSMarkus Armbruster 
155551de9760Saliguori void bdrv_iterate(void (*it)(void *opaque, BlockDriverState *bs), void *opaque)
155681d0912dSbellard {
155781d0912dSbellard     BlockDriverState *bs;
155881d0912dSbellard 
15591b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
156051de9760Saliguori         it(opaque, bs);
156181d0912dSbellard     }
156281d0912dSbellard }
156381d0912dSbellard 
1564ea2384d3Sbellard const char *bdrv_get_device_name(BlockDriverState *bs)
1565ea2384d3Sbellard {
1566ea2384d3Sbellard     return bs->device_name;
1567ea2384d3Sbellard }
1568ea2384d3Sbellard 
1569205ef796SKevin Wolf int bdrv_flush(BlockDriverState *bs)
15707a6cba61Spbrook {
1571016f5cf6SAlexander Graf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
1572205ef796SKevin Wolf         return 0;
1573016f5cf6SAlexander Graf     }
1574016f5cf6SAlexander Graf 
1575205ef796SKevin Wolf     if (bs->drv && bs->drv->bdrv_flush) {
1576205ef796SKevin Wolf         return bs->drv->bdrv_flush(bs);
1577205ef796SKevin Wolf     }
1578205ef796SKevin Wolf 
1579205ef796SKevin Wolf     /*
1580205ef796SKevin Wolf      * Some block drivers always operate in either writethrough or unsafe mode
1581205ef796SKevin Wolf      * and don't support bdrv_flush therefore. Usually qemu doesn't know how
1582205ef796SKevin Wolf      * the server works (because the behaviour is hardcoded or depends on
1583205ef796SKevin Wolf      * server-side configuration), so we can't ensure that everything is safe
1584205ef796SKevin Wolf      * on disk. Returning an error doesn't work because that would break guests
1585205ef796SKevin Wolf      * even if the server operates in writethrough mode.
1586205ef796SKevin Wolf      *
1587205ef796SKevin Wolf      * Let's hope the user knows what he's doing.
1588205ef796SKevin Wolf      */
1589205ef796SKevin Wolf     return 0;
15907a6cba61Spbrook }
15917a6cba61Spbrook 
1592c6ca28d6Saliguori void bdrv_flush_all(void)
1593c6ca28d6Saliguori {
1594c6ca28d6Saliguori     BlockDriverState *bs;
1595c6ca28d6Saliguori 
15961b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1597c6ca28d6Saliguori         if (bs->drv && !bdrv_is_read_only(bs) &&
15981b7bdbc1SStefan Hajnoczi             (!bdrv_is_removable(bs) || bdrv_is_inserted(bs))) {
1599c6ca28d6Saliguori             bdrv_flush(bs);
1600c6ca28d6Saliguori         }
16011b7bdbc1SStefan Hajnoczi     }
16021b7bdbc1SStefan Hajnoczi }
1603c6ca28d6Saliguori 
1604f2feebbdSKevin Wolf int bdrv_has_zero_init(BlockDriverState *bs)
1605f2feebbdSKevin Wolf {
1606f2feebbdSKevin Wolf     assert(bs->drv);
1607f2feebbdSKevin Wolf 
1608336c1c12SKevin Wolf     if (bs->drv->bdrv_has_zero_init) {
1609336c1c12SKevin Wolf         return bs->drv->bdrv_has_zero_init(bs);
1610f2feebbdSKevin Wolf     }
1611f2feebbdSKevin Wolf 
1612f2feebbdSKevin Wolf     return 1;
1613f2feebbdSKevin Wolf }
1614f2feebbdSKevin Wolf 
1615bb8bf76fSChristoph Hellwig int bdrv_discard(BlockDriverState *bs, int64_t sector_num, int nb_sectors)
1616bb8bf76fSChristoph Hellwig {
1617bb8bf76fSChristoph Hellwig     if (!bs->drv) {
1618bb8bf76fSChristoph Hellwig         return -ENOMEDIUM;
1619bb8bf76fSChristoph Hellwig     }
1620bb8bf76fSChristoph Hellwig     if (!bs->drv->bdrv_discard) {
1621bb8bf76fSChristoph Hellwig         return 0;
1622bb8bf76fSChristoph Hellwig     }
1623bb8bf76fSChristoph Hellwig     return bs->drv->bdrv_discard(bs, sector_num, nb_sectors);
1624bb8bf76fSChristoph Hellwig }
1625bb8bf76fSChristoph Hellwig 
1626f58c7b35Sths /*
1627f58c7b35Sths  * Returns true iff the specified sector is present in the disk image. Drivers
1628f58c7b35Sths  * not implementing the functionality are assumed to not support backing files,
1629f58c7b35Sths  * hence all their sectors are reported as allocated.
1630f58c7b35Sths  *
1631f58c7b35Sths  * 'pnum' is set to the number of sectors (including and immediately following
1632f58c7b35Sths  * the specified sector) that are known to be in the same
1633f58c7b35Sths  * allocated/unallocated state.
1634f58c7b35Sths  *
1635f58c7b35Sths  * 'nb_sectors' is the max value 'pnum' should be set to.
1636f58c7b35Sths  */
1637f58c7b35Sths int bdrv_is_allocated(BlockDriverState *bs, int64_t sector_num, int nb_sectors,
1638f58c7b35Sths 	int *pnum)
1639f58c7b35Sths {
1640f58c7b35Sths     int64_t n;
1641f58c7b35Sths     if (!bs->drv->bdrv_is_allocated) {
1642f58c7b35Sths         if (sector_num >= bs->total_sectors) {
1643f58c7b35Sths             *pnum = 0;
1644f58c7b35Sths             return 0;
1645f58c7b35Sths         }
1646f58c7b35Sths         n = bs->total_sectors - sector_num;
1647f58c7b35Sths         *pnum = (n < nb_sectors) ? (n) : (nb_sectors);
1648f58c7b35Sths         return 1;
1649f58c7b35Sths     }
1650f58c7b35Sths     return bs->drv->bdrv_is_allocated(bs, sector_num, nb_sectors, pnum);
1651f58c7b35Sths }
1652f58c7b35Sths 
16532582bfedSLuiz Capitulino void bdrv_mon_event(const BlockDriverState *bdrv,
16542582bfedSLuiz Capitulino                     BlockMonEventAction action, int is_read)
16552582bfedSLuiz Capitulino {
16562582bfedSLuiz Capitulino     QObject *data;
16572582bfedSLuiz Capitulino     const char *action_str;
16582582bfedSLuiz Capitulino 
16592582bfedSLuiz Capitulino     switch (action) {
16602582bfedSLuiz Capitulino     case BDRV_ACTION_REPORT:
16612582bfedSLuiz Capitulino         action_str = "report";
16622582bfedSLuiz Capitulino         break;
16632582bfedSLuiz Capitulino     case BDRV_ACTION_IGNORE:
16642582bfedSLuiz Capitulino         action_str = "ignore";
16652582bfedSLuiz Capitulino         break;
16662582bfedSLuiz Capitulino     case BDRV_ACTION_STOP:
16672582bfedSLuiz Capitulino         action_str = "stop";
16682582bfedSLuiz Capitulino         break;
16692582bfedSLuiz Capitulino     default:
16702582bfedSLuiz Capitulino         abort();
16712582bfedSLuiz Capitulino     }
16722582bfedSLuiz Capitulino 
16732582bfedSLuiz Capitulino     data = qobject_from_jsonf("{ 'device': %s, 'action': %s, 'operation': %s }",
16742582bfedSLuiz Capitulino                               bdrv->device_name,
16752582bfedSLuiz Capitulino                               action_str,
16762582bfedSLuiz Capitulino                               is_read ? "read" : "write");
16772582bfedSLuiz Capitulino     monitor_protocol_event(QEVENT_BLOCK_IO_ERROR, data);
16782582bfedSLuiz Capitulino 
16792582bfedSLuiz Capitulino     qobject_decref(data);
16802582bfedSLuiz Capitulino }
16812582bfedSLuiz Capitulino 
1682d15e5465SLuiz Capitulino static void bdrv_print_dict(QObject *obj, void *opaque)
1683b338082bSbellard {
1684d15e5465SLuiz Capitulino     QDict *bs_dict;
1685d15e5465SLuiz Capitulino     Monitor *mon = opaque;
1686b338082bSbellard 
1687d15e5465SLuiz Capitulino     bs_dict = qobject_to_qdict(obj);
1688d15e5465SLuiz Capitulino 
1689d8aeeb31SMarkus Armbruster     monitor_printf(mon, "%s: removable=%d",
1690d15e5465SLuiz Capitulino                         qdict_get_str(bs_dict, "device"),
1691d15e5465SLuiz Capitulino                         qdict_get_bool(bs_dict, "removable"));
1692d15e5465SLuiz Capitulino 
1693d15e5465SLuiz Capitulino     if (qdict_get_bool(bs_dict, "removable")) {
1694d15e5465SLuiz Capitulino         monitor_printf(mon, " locked=%d", qdict_get_bool(bs_dict, "locked"));
1695b338082bSbellard     }
1696d15e5465SLuiz Capitulino 
1697d15e5465SLuiz Capitulino     if (qdict_haskey(bs_dict, "inserted")) {
1698d15e5465SLuiz Capitulino         QDict *qdict = qobject_to_qdict(qdict_get(bs_dict, "inserted"));
1699d15e5465SLuiz Capitulino 
1700376253ecSaliguori         monitor_printf(mon, " file=");
1701d15e5465SLuiz Capitulino         monitor_print_filename(mon, qdict_get_str(qdict, "file"));
1702d15e5465SLuiz Capitulino         if (qdict_haskey(qdict, "backing_file")) {
1703376253ecSaliguori             monitor_printf(mon, " backing_file=");
1704d15e5465SLuiz Capitulino             monitor_print_filename(mon, qdict_get_str(qdict, "backing_file"));
1705fef30743Sths         }
1706d15e5465SLuiz Capitulino         monitor_printf(mon, " ro=%d drv=%s encrypted=%d",
1707d15e5465SLuiz Capitulino                             qdict_get_bool(qdict, "ro"),
1708d15e5465SLuiz Capitulino                             qdict_get_str(qdict, "drv"),
1709d15e5465SLuiz Capitulino                             qdict_get_bool(qdict, "encrypted"));
1710b338082bSbellard     } else {
1711376253ecSaliguori         monitor_printf(mon, " [not inserted]");
1712b338082bSbellard     }
1713d15e5465SLuiz Capitulino 
1714376253ecSaliguori     monitor_printf(mon, "\n");
1715b338082bSbellard }
1716d15e5465SLuiz Capitulino 
1717d15e5465SLuiz Capitulino void bdrv_info_print(Monitor *mon, const QObject *data)
1718d15e5465SLuiz Capitulino {
1719d15e5465SLuiz Capitulino     qlist_iter(qobject_to_qlist(data), bdrv_print_dict, mon);
1720d15e5465SLuiz Capitulino }
1721d15e5465SLuiz Capitulino 
1722d15e5465SLuiz Capitulino void bdrv_info(Monitor *mon, QObject **ret_data)
1723d15e5465SLuiz Capitulino {
1724d15e5465SLuiz Capitulino     QList *bs_list;
1725d15e5465SLuiz Capitulino     BlockDriverState *bs;
1726d15e5465SLuiz Capitulino 
1727d15e5465SLuiz Capitulino     bs_list = qlist_new();
1728d15e5465SLuiz Capitulino 
17291b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1730d15e5465SLuiz Capitulino         QObject *bs_obj;
1731d15e5465SLuiz Capitulino 
1732d8aeeb31SMarkus Armbruster         bs_obj = qobject_from_jsonf("{ 'device': %s, 'type': 'unknown', "
1733d15e5465SLuiz Capitulino                                     "'removable': %i, 'locked': %i }",
1734d8aeeb31SMarkus Armbruster                                     bs->device_name, bs->removable,
1735d15e5465SLuiz Capitulino                                     bs->locked);
1736d15e5465SLuiz Capitulino 
1737d15e5465SLuiz Capitulino         if (bs->drv) {
1738d15e5465SLuiz Capitulino             QObject *obj;
1739d15e5465SLuiz Capitulino             QDict *bs_dict = qobject_to_qdict(bs_obj);
1740d15e5465SLuiz Capitulino 
1741d15e5465SLuiz Capitulino             obj = qobject_from_jsonf("{ 'file': %s, 'ro': %i, 'drv': %s, "
1742d15e5465SLuiz Capitulino                                      "'encrypted': %i }",
1743d15e5465SLuiz Capitulino                                      bs->filename, bs->read_only,
1744d15e5465SLuiz Capitulino                                      bs->drv->format_name,
1745d15e5465SLuiz Capitulino                                      bdrv_is_encrypted(bs));
1746d15e5465SLuiz Capitulino             if (bs->backing_file[0] != '\0') {
1747d15e5465SLuiz Capitulino                 QDict *qdict = qobject_to_qdict(obj);
1748d15e5465SLuiz Capitulino                 qdict_put(qdict, "backing_file",
1749d15e5465SLuiz Capitulino                           qstring_from_str(bs->backing_file));
1750d15e5465SLuiz Capitulino             }
1751d15e5465SLuiz Capitulino 
1752d15e5465SLuiz Capitulino             qdict_put_obj(bs_dict, "inserted", obj);
1753d15e5465SLuiz Capitulino         }
1754d15e5465SLuiz Capitulino         qlist_append_obj(bs_list, bs_obj);
1755d15e5465SLuiz Capitulino     }
1756d15e5465SLuiz Capitulino 
1757d15e5465SLuiz Capitulino     *ret_data = QOBJECT(bs_list);
1758b338082bSbellard }
1759a36e69ddSths 
1760218a536aSLuiz Capitulino static void bdrv_stats_iter(QObject *data, void *opaque)
1761a36e69ddSths {
1762218a536aSLuiz Capitulino     QDict *qdict;
1763218a536aSLuiz Capitulino     Monitor *mon = opaque;
1764218a536aSLuiz Capitulino 
1765218a536aSLuiz Capitulino     qdict = qobject_to_qdict(data);
1766218a536aSLuiz Capitulino     monitor_printf(mon, "%s:", qdict_get_str(qdict, "device"));
1767218a536aSLuiz Capitulino 
1768218a536aSLuiz Capitulino     qdict = qobject_to_qdict(qdict_get(qdict, "stats"));
1769218a536aSLuiz Capitulino     monitor_printf(mon, " rd_bytes=%" PRId64
1770218a536aSLuiz Capitulino                         " wr_bytes=%" PRId64
1771218a536aSLuiz Capitulino                         " rd_operations=%" PRId64
1772218a536aSLuiz Capitulino                         " wr_operations=%" PRId64
1773218a536aSLuiz Capitulino                         "\n",
1774218a536aSLuiz Capitulino                         qdict_get_int(qdict, "rd_bytes"),
1775218a536aSLuiz Capitulino                         qdict_get_int(qdict, "wr_bytes"),
1776218a536aSLuiz Capitulino                         qdict_get_int(qdict, "rd_operations"),
1777218a536aSLuiz Capitulino                         qdict_get_int(qdict, "wr_operations"));
1778218a536aSLuiz Capitulino }
1779218a536aSLuiz Capitulino 
1780218a536aSLuiz Capitulino void bdrv_stats_print(Monitor *mon, const QObject *data)
1781218a536aSLuiz Capitulino {
1782218a536aSLuiz Capitulino     qlist_iter(qobject_to_qlist(data), bdrv_stats_iter, mon);
1783218a536aSLuiz Capitulino }
1784218a536aSLuiz Capitulino 
1785294cc35fSKevin Wolf static QObject* bdrv_info_stats_bs(BlockDriverState *bs)
1786294cc35fSKevin Wolf {
1787294cc35fSKevin Wolf     QObject *res;
1788294cc35fSKevin Wolf     QDict *dict;
1789294cc35fSKevin Wolf 
1790294cc35fSKevin Wolf     res = qobject_from_jsonf("{ 'stats': {"
1791294cc35fSKevin Wolf                              "'rd_bytes': %" PRId64 ","
1792294cc35fSKevin Wolf                              "'wr_bytes': %" PRId64 ","
1793294cc35fSKevin Wolf                              "'rd_operations': %" PRId64 ","
1794294cc35fSKevin Wolf                              "'wr_operations': %" PRId64 ","
1795294cc35fSKevin Wolf                              "'wr_highest_offset': %" PRId64
1796294cc35fSKevin Wolf                              "} }",
1797294cc35fSKevin Wolf                              bs->rd_bytes, bs->wr_bytes,
1798294cc35fSKevin Wolf                              bs->rd_ops, bs->wr_ops,
17995ffbbc67SBlue Swirl                              bs->wr_highest_sector *
18005ffbbc67SBlue Swirl                              (uint64_t)BDRV_SECTOR_SIZE);
1801294cc35fSKevin Wolf     dict  = qobject_to_qdict(res);
1802294cc35fSKevin Wolf 
1803294cc35fSKevin Wolf     if (*bs->device_name) {
1804294cc35fSKevin Wolf         qdict_put(dict, "device", qstring_from_str(bs->device_name));
1805294cc35fSKevin Wolf     }
1806294cc35fSKevin Wolf 
1807294cc35fSKevin Wolf     if (bs->file) {
1808294cc35fSKevin Wolf         QObject *parent = bdrv_info_stats_bs(bs->file);
1809294cc35fSKevin Wolf         qdict_put_obj(dict, "parent", parent);
1810294cc35fSKevin Wolf     }
1811294cc35fSKevin Wolf 
1812294cc35fSKevin Wolf     return res;
1813294cc35fSKevin Wolf }
1814294cc35fSKevin Wolf 
1815218a536aSLuiz Capitulino void bdrv_info_stats(Monitor *mon, QObject **ret_data)
1816218a536aSLuiz Capitulino {
1817218a536aSLuiz Capitulino     QObject *obj;
1818218a536aSLuiz Capitulino     QList *devices;
1819a36e69ddSths     BlockDriverState *bs;
1820a36e69ddSths 
1821218a536aSLuiz Capitulino     devices = qlist_new();
1822218a536aSLuiz Capitulino 
18231b7bdbc1SStefan Hajnoczi     QTAILQ_FOREACH(bs, &bdrv_states, list) {
1824294cc35fSKevin Wolf         obj = bdrv_info_stats_bs(bs);
1825218a536aSLuiz Capitulino         qlist_append_obj(devices, obj);
1826a36e69ddSths     }
1827218a536aSLuiz Capitulino 
1828218a536aSLuiz Capitulino     *ret_data = QOBJECT(devices);
1829a36e69ddSths }
1830ea2384d3Sbellard 
1831045df330Saliguori const char *bdrv_get_encrypted_filename(BlockDriverState *bs)
1832045df330Saliguori {
1833045df330Saliguori     if (bs->backing_hd && bs->backing_hd->encrypted)
1834045df330Saliguori         return bs->backing_file;
1835045df330Saliguori     else if (bs->encrypted)
1836045df330Saliguori         return bs->filename;
1837045df330Saliguori     else
1838045df330Saliguori         return NULL;
1839045df330Saliguori }
1840045df330Saliguori 
184183f64091Sbellard void bdrv_get_backing_filename(BlockDriverState *bs,
184283f64091Sbellard                                char *filename, int filename_size)
184383f64091Sbellard {
1844b783e409SKevin Wolf     if (!bs->backing_file) {
184583f64091Sbellard         pstrcpy(filename, filename_size, "");
184683f64091Sbellard     } else {
184783f64091Sbellard         pstrcpy(filename, filename_size, bs->backing_file);
184883f64091Sbellard     }
184983f64091Sbellard }
185083f64091Sbellard 
1851faea38e7Sbellard int bdrv_write_compressed(BlockDriverState *bs, int64_t sector_num,
1852faea38e7Sbellard                           const uint8_t *buf, int nb_sectors)
1853faea38e7Sbellard {
1854faea38e7Sbellard     BlockDriver *drv = bs->drv;
1855faea38e7Sbellard     if (!drv)
185619cb3738Sbellard         return -ENOMEDIUM;
1857faea38e7Sbellard     if (!drv->bdrv_write_compressed)
1858faea38e7Sbellard         return -ENOTSUP;
1859fbb7b4e0SKevin Wolf     if (bdrv_check_request(bs, sector_num, nb_sectors))
1860fbb7b4e0SKevin Wolf         return -EIO;
18617cd1e32aSlirans@il.ibm.com 
1862c6d22830SJan Kiszka     if (bs->dirty_bitmap) {
18637cd1e32aSlirans@il.ibm.com         set_dirty_bitmap(bs, sector_num, nb_sectors, 1);
18647cd1e32aSlirans@il.ibm.com     }
18657cd1e32aSlirans@il.ibm.com 
1866faea38e7Sbellard     return drv->bdrv_write_compressed(bs, sector_num, buf, nb_sectors);
1867faea38e7Sbellard }
1868faea38e7Sbellard 
1869faea38e7Sbellard int bdrv_get_info(BlockDriverState *bs, BlockDriverInfo *bdi)
1870faea38e7Sbellard {
1871faea38e7Sbellard     BlockDriver *drv = bs->drv;
1872faea38e7Sbellard     if (!drv)
187319cb3738Sbellard         return -ENOMEDIUM;
1874faea38e7Sbellard     if (!drv->bdrv_get_info)
1875faea38e7Sbellard         return -ENOTSUP;
1876faea38e7Sbellard     memset(bdi, 0, sizeof(*bdi));
1877faea38e7Sbellard     return drv->bdrv_get_info(bs, bdi);
1878faea38e7Sbellard }
1879faea38e7Sbellard 
188045566e9cSChristoph Hellwig int bdrv_save_vmstate(BlockDriverState *bs, const uint8_t *buf,
188145566e9cSChristoph Hellwig                       int64_t pos, int size)
1882178e08a5Saliguori {
1883178e08a5Saliguori     BlockDriver *drv = bs->drv;
1884178e08a5Saliguori     if (!drv)
1885178e08a5Saliguori         return -ENOMEDIUM;
18867cdb1f6dSMORITA Kazutaka     if (drv->bdrv_save_vmstate)
188745566e9cSChristoph Hellwig         return drv->bdrv_save_vmstate(bs, buf, pos, size);
18887cdb1f6dSMORITA Kazutaka     if (bs->file)
18897cdb1f6dSMORITA Kazutaka         return bdrv_save_vmstate(bs->file, buf, pos, size);
18907cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
1891178e08a5Saliguori }
1892178e08a5Saliguori 
189345566e9cSChristoph Hellwig int bdrv_load_vmstate(BlockDriverState *bs, uint8_t *buf,
189445566e9cSChristoph Hellwig                       int64_t pos, int size)
1895178e08a5Saliguori {
1896178e08a5Saliguori     BlockDriver *drv = bs->drv;
1897178e08a5Saliguori     if (!drv)
1898178e08a5Saliguori         return -ENOMEDIUM;
18997cdb1f6dSMORITA Kazutaka     if (drv->bdrv_load_vmstate)
190045566e9cSChristoph Hellwig         return drv->bdrv_load_vmstate(bs, buf, pos, size);
19017cdb1f6dSMORITA Kazutaka     if (bs->file)
19027cdb1f6dSMORITA Kazutaka         return bdrv_load_vmstate(bs->file, buf, pos, size);
19037cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
1904178e08a5Saliguori }
1905178e08a5Saliguori 
19068b9b0cc2SKevin Wolf void bdrv_debug_event(BlockDriverState *bs, BlkDebugEvent event)
19078b9b0cc2SKevin Wolf {
19088b9b0cc2SKevin Wolf     BlockDriver *drv = bs->drv;
19098b9b0cc2SKevin Wolf 
19108b9b0cc2SKevin Wolf     if (!drv || !drv->bdrv_debug_event) {
19118b9b0cc2SKevin Wolf         return;
19128b9b0cc2SKevin Wolf     }
19138b9b0cc2SKevin Wolf 
19148b9b0cc2SKevin Wolf     return drv->bdrv_debug_event(bs, event);
19158b9b0cc2SKevin Wolf 
19168b9b0cc2SKevin Wolf }
19178b9b0cc2SKevin Wolf 
1918faea38e7Sbellard /**************************************************************/
1919faea38e7Sbellard /* handling of snapshots */
1920faea38e7Sbellard 
1921feeee5acSMiguel Di Ciurcio Filho int bdrv_can_snapshot(BlockDriverState *bs)
1922feeee5acSMiguel Di Ciurcio Filho {
1923feeee5acSMiguel Di Ciurcio Filho     BlockDriver *drv = bs->drv;
1924feeee5acSMiguel Di Ciurcio Filho     if (!drv || bdrv_is_removable(bs) || bdrv_is_read_only(bs)) {
1925feeee5acSMiguel Di Ciurcio Filho         return 0;
1926feeee5acSMiguel Di Ciurcio Filho     }
1927feeee5acSMiguel Di Ciurcio Filho 
1928feeee5acSMiguel Di Ciurcio Filho     if (!drv->bdrv_snapshot_create) {
1929feeee5acSMiguel Di Ciurcio Filho         if (bs->file != NULL) {
1930feeee5acSMiguel Di Ciurcio Filho             return bdrv_can_snapshot(bs->file);
1931feeee5acSMiguel Di Ciurcio Filho         }
1932feeee5acSMiguel Di Ciurcio Filho         return 0;
1933feeee5acSMiguel Di Ciurcio Filho     }
1934feeee5acSMiguel Di Ciurcio Filho 
1935feeee5acSMiguel Di Ciurcio Filho     return 1;
1936feeee5acSMiguel Di Ciurcio Filho }
1937feeee5acSMiguel Di Ciurcio Filho 
1938199630b6SBlue Swirl int bdrv_is_snapshot(BlockDriverState *bs)
1939199630b6SBlue Swirl {
1940199630b6SBlue Swirl     return !!(bs->open_flags & BDRV_O_SNAPSHOT);
1941199630b6SBlue Swirl }
1942199630b6SBlue Swirl 
1943f9092b10SMarkus Armbruster BlockDriverState *bdrv_snapshots(void)
1944f9092b10SMarkus Armbruster {
1945f9092b10SMarkus Armbruster     BlockDriverState *bs;
1946f9092b10SMarkus Armbruster 
19473ac906f7SMarkus Armbruster     if (bs_snapshots) {
1948f9092b10SMarkus Armbruster         return bs_snapshots;
19493ac906f7SMarkus Armbruster     }
1950f9092b10SMarkus Armbruster 
1951f9092b10SMarkus Armbruster     bs = NULL;
1952f9092b10SMarkus Armbruster     while ((bs = bdrv_next(bs))) {
1953f9092b10SMarkus Armbruster         if (bdrv_can_snapshot(bs)) {
19543ac906f7SMarkus Armbruster             bs_snapshots = bs;
19553ac906f7SMarkus Armbruster             return bs;
1956f9092b10SMarkus Armbruster         }
1957f9092b10SMarkus Armbruster     }
1958f9092b10SMarkus Armbruster     return NULL;
1959f9092b10SMarkus Armbruster }
1960f9092b10SMarkus Armbruster 
1961faea38e7Sbellard int bdrv_snapshot_create(BlockDriverState *bs,
1962faea38e7Sbellard                          QEMUSnapshotInfo *sn_info)
1963faea38e7Sbellard {
1964faea38e7Sbellard     BlockDriver *drv = bs->drv;
1965faea38e7Sbellard     if (!drv)
196619cb3738Sbellard         return -ENOMEDIUM;
19677cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_create)
1968faea38e7Sbellard         return drv->bdrv_snapshot_create(bs, sn_info);
19697cdb1f6dSMORITA Kazutaka     if (bs->file)
19707cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_create(bs->file, sn_info);
19717cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
1972faea38e7Sbellard }
1973faea38e7Sbellard 
1974faea38e7Sbellard int bdrv_snapshot_goto(BlockDriverState *bs,
1975faea38e7Sbellard                        const char *snapshot_id)
1976faea38e7Sbellard {
1977faea38e7Sbellard     BlockDriver *drv = bs->drv;
19787cdb1f6dSMORITA Kazutaka     int ret, open_ret;
19797cdb1f6dSMORITA Kazutaka 
1980faea38e7Sbellard     if (!drv)
198119cb3738Sbellard         return -ENOMEDIUM;
19827cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_goto)
1983faea38e7Sbellard         return drv->bdrv_snapshot_goto(bs, snapshot_id);
19847cdb1f6dSMORITA Kazutaka 
19857cdb1f6dSMORITA Kazutaka     if (bs->file) {
19867cdb1f6dSMORITA Kazutaka         drv->bdrv_close(bs);
19877cdb1f6dSMORITA Kazutaka         ret = bdrv_snapshot_goto(bs->file, snapshot_id);
19887cdb1f6dSMORITA Kazutaka         open_ret = drv->bdrv_open(bs, bs->open_flags);
19897cdb1f6dSMORITA Kazutaka         if (open_ret < 0) {
19907cdb1f6dSMORITA Kazutaka             bdrv_delete(bs->file);
19917cdb1f6dSMORITA Kazutaka             bs->drv = NULL;
19927cdb1f6dSMORITA Kazutaka             return open_ret;
19937cdb1f6dSMORITA Kazutaka         }
19947cdb1f6dSMORITA Kazutaka         return ret;
19957cdb1f6dSMORITA Kazutaka     }
19967cdb1f6dSMORITA Kazutaka 
19977cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
1998faea38e7Sbellard }
1999faea38e7Sbellard 
2000faea38e7Sbellard int bdrv_snapshot_delete(BlockDriverState *bs, const char *snapshot_id)
2001faea38e7Sbellard {
2002faea38e7Sbellard     BlockDriver *drv = bs->drv;
2003faea38e7Sbellard     if (!drv)
200419cb3738Sbellard         return -ENOMEDIUM;
20057cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_delete)
2006faea38e7Sbellard         return drv->bdrv_snapshot_delete(bs, snapshot_id);
20077cdb1f6dSMORITA Kazutaka     if (bs->file)
20087cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_delete(bs->file, snapshot_id);
20097cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2010faea38e7Sbellard }
2011faea38e7Sbellard 
2012faea38e7Sbellard int bdrv_snapshot_list(BlockDriverState *bs,
2013faea38e7Sbellard                        QEMUSnapshotInfo **psn_info)
2014faea38e7Sbellard {
2015faea38e7Sbellard     BlockDriver *drv = bs->drv;
2016faea38e7Sbellard     if (!drv)
201719cb3738Sbellard         return -ENOMEDIUM;
20187cdb1f6dSMORITA Kazutaka     if (drv->bdrv_snapshot_list)
2019faea38e7Sbellard         return drv->bdrv_snapshot_list(bs, psn_info);
20207cdb1f6dSMORITA Kazutaka     if (bs->file)
20217cdb1f6dSMORITA Kazutaka         return bdrv_snapshot_list(bs->file, psn_info);
20227cdb1f6dSMORITA Kazutaka     return -ENOTSUP;
2023faea38e7Sbellard }
2024faea38e7Sbellard 
202551ef6727Sedison int bdrv_snapshot_load_tmp(BlockDriverState *bs,
202651ef6727Sedison         const char *snapshot_name)
202751ef6727Sedison {
202851ef6727Sedison     BlockDriver *drv = bs->drv;
202951ef6727Sedison     if (!drv) {
203051ef6727Sedison         return -ENOMEDIUM;
203151ef6727Sedison     }
203251ef6727Sedison     if (!bs->read_only) {
203351ef6727Sedison         return -EINVAL;
203451ef6727Sedison     }
203551ef6727Sedison     if (drv->bdrv_snapshot_load_tmp) {
203651ef6727Sedison         return drv->bdrv_snapshot_load_tmp(bs, snapshot_name);
203751ef6727Sedison     }
203851ef6727Sedison     return -ENOTSUP;
203951ef6727Sedison }
204051ef6727Sedison 
2041faea38e7Sbellard #define NB_SUFFIXES 4
2042faea38e7Sbellard 
2043faea38e7Sbellard char *get_human_readable_size(char *buf, int buf_size, int64_t size)
2044faea38e7Sbellard {
2045faea38e7Sbellard     static const char suffixes[NB_SUFFIXES] = "KMGT";
2046faea38e7Sbellard     int64_t base;
2047faea38e7Sbellard     int i;
2048faea38e7Sbellard 
2049faea38e7Sbellard     if (size <= 999) {
2050faea38e7Sbellard         snprintf(buf, buf_size, "%" PRId64, size);
2051faea38e7Sbellard     } else {
2052faea38e7Sbellard         base = 1024;
2053faea38e7Sbellard         for(i = 0; i < NB_SUFFIXES; i++) {
2054faea38e7Sbellard             if (size < (10 * base)) {
2055faea38e7Sbellard                 snprintf(buf, buf_size, "%0.1f%c",
2056faea38e7Sbellard                          (double)size / base,
2057faea38e7Sbellard                          suffixes[i]);
2058faea38e7Sbellard                 break;
2059faea38e7Sbellard             } else if (size < (1000 * base) || i == (NB_SUFFIXES - 1)) {
2060faea38e7Sbellard                 snprintf(buf, buf_size, "%" PRId64 "%c",
2061faea38e7Sbellard                          ((size + (base >> 1)) / base),
2062faea38e7Sbellard                          suffixes[i]);
2063faea38e7Sbellard                 break;
2064faea38e7Sbellard             }
2065faea38e7Sbellard             base = base * 1024;
2066faea38e7Sbellard         }
2067faea38e7Sbellard     }
2068faea38e7Sbellard     return buf;
2069faea38e7Sbellard }
2070faea38e7Sbellard 
2071faea38e7Sbellard char *bdrv_snapshot_dump(char *buf, int buf_size, QEMUSnapshotInfo *sn)
2072faea38e7Sbellard {
2073faea38e7Sbellard     char buf1[128], date_buf[128], clock_buf[128];
20743b9f94e1Sbellard #ifdef _WIN32
20753b9f94e1Sbellard     struct tm *ptm;
20763b9f94e1Sbellard #else
2077faea38e7Sbellard     struct tm tm;
20783b9f94e1Sbellard #endif
2079faea38e7Sbellard     time_t ti;
2080faea38e7Sbellard     int64_t secs;
2081faea38e7Sbellard 
2082faea38e7Sbellard     if (!sn) {
2083faea38e7Sbellard         snprintf(buf, buf_size,
2084faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
2085faea38e7Sbellard                  "ID", "TAG", "VM SIZE", "DATE", "VM CLOCK");
2086faea38e7Sbellard     } else {
2087faea38e7Sbellard         ti = sn->date_sec;
20883b9f94e1Sbellard #ifdef _WIN32
20893b9f94e1Sbellard         ptm = localtime(&ti);
20903b9f94e1Sbellard         strftime(date_buf, sizeof(date_buf),
20913b9f94e1Sbellard                  "%Y-%m-%d %H:%M:%S", ptm);
20923b9f94e1Sbellard #else
2093faea38e7Sbellard         localtime_r(&ti, &tm);
2094faea38e7Sbellard         strftime(date_buf, sizeof(date_buf),
2095faea38e7Sbellard                  "%Y-%m-%d %H:%M:%S", &tm);
20963b9f94e1Sbellard #endif
2097faea38e7Sbellard         secs = sn->vm_clock_nsec / 1000000000;
2098faea38e7Sbellard         snprintf(clock_buf, sizeof(clock_buf),
2099faea38e7Sbellard                  "%02d:%02d:%02d.%03d",
2100faea38e7Sbellard                  (int)(secs / 3600),
2101faea38e7Sbellard                  (int)((secs / 60) % 60),
2102faea38e7Sbellard                  (int)(secs % 60),
2103faea38e7Sbellard                  (int)((sn->vm_clock_nsec / 1000000) % 1000));
2104faea38e7Sbellard         snprintf(buf, buf_size,
2105faea38e7Sbellard                  "%-10s%-20s%7s%20s%15s",
2106faea38e7Sbellard                  sn->id_str, sn->name,
2107faea38e7Sbellard                  get_human_readable_size(buf1, sizeof(buf1), sn->vm_state_size),
2108faea38e7Sbellard                  date_buf,
2109faea38e7Sbellard                  clock_buf);
2110faea38e7Sbellard     }
2111faea38e7Sbellard     return buf;
2112faea38e7Sbellard }
2113faea38e7Sbellard 
211483f64091Sbellard 
2115ea2384d3Sbellard /**************************************************************/
211683f64091Sbellard /* async I/Os */
2117ea2384d3Sbellard 
21183b69e4b9Saliguori BlockDriverAIOCB *bdrv_aio_readv(BlockDriverState *bs, int64_t sector_num,
2119f141eafeSaliguori                                  QEMUIOVector *qiov, int nb_sectors,
212083f64091Sbellard                                  BlockDriverCompletionFunc *cb, void *opaque)
2121ea2384d3Sbellard {
212283f64091Sbellard     BlockDriver *drv = bs->drv;
2123a36e69ddSths     BlockDriverAIOCB *ret;
2124ea2384d3Sbellard 
2125bbf0a440SStefan Hajnoczi     trace_bdrv_aio_readv(bs, sector_num, nb_sectors, opaque);
2126bbf0a440SStefan Hajnoczi 
212719cb3738Sbellard     if (!drv)
2128ce1a14dcSpbrook         return NULL;
212971d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
213071d0770cSaliguori         return NULL;
213183f64091Sbellard 
2132f141eafeSaliguori     ret = drv->bdrv_aio_readv(bs, sector_num, qiov, nb_sectors,
2133f141eafeSaliguori                               cb, opaque);
2134a36e69ddSths 
2135a36e69ddSths     if (ret) {
2136a36e69ddSths 	/* Update stats even though technically transfer has not happened. */
21376ea44308SJan Kiszka 	bs->rd_bytes += (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
2138a36e69ddSths 	bs->rd_ops ++;
2139a36e69ddSths     }
2140a36e69ddSths 
2141a36e69ddSths     return ret;
214283f64091Sbellard }
214383f64091Sbellard 
21444dcafbb1SMarcelo Tosatti typedef struct BlockCompleteData {
21454dcafbb1SMarcelo Tosatti     BlockDriverCompletionFunc *cb;
21464dcafbb1SMarcelo Tosatti     void *opaque;
21474dcafbb1SMarcelo Tosatti     BlockDriverState *bs;
21484dcafbb1SMarcelo Tosatti     int64_t sector_num;
21494dcafbb1SMarcelo Tosatti     int nb_sectors;
21504dcafbb1SMarcelo Tosatti } BlockCompleteData;
21514dcafbb1SMarcelo Tosatti 
21524dcafbb1SMarcelo Tosatti static void block_complete_cb(void *opaque, int ret)
21534dcafbb1SMarcelo Tosatti {
21544dcafbb1SMarcelo Tosatti     BlockCompleteData *b = opaque;
21554dcafbb1SMarcelo Tosatti 
21564dcafbb1SMarcelo Tosatti     if (b->bs->dirty_bitmap) {
21574dcafbb1SMarcelo Tosatti         set_dirty_bitmap(b->bs, b->sector_num, b->nb_sectors, 1);
21584dcafbb1SMarcelo Tosatti     }
21594dcafbb1SMarcelo Tosatti     b->cb(b->opaque, ret);
21604dcafbb1SMarcelo Tosatti     qemu_free(b);
21614dcafbb1SMarcelo Tosatti }
21624dcafbb1SMarcelo Tosatti 
21634dcafbb1SMarcelo Tosatti static BlockCompleteData *blk_dirty_cb_alloc(BlockDriverState *bs,
21644dcafbb1SMarcelo Tosatti                                              int64_t sector_num,
21654dcafbb1SMarcelo Tosatti                                              int nb_sectors,
21664dcafbb1SMarcelo Tosatti                                              BlockDriverCompletionFunc *cb,
21674dcafbb1SMarcelo Tosatti                                              void *opaque)
21684dcafbb1SMarcelo Tosatti {
21694dcafbb1SMarcelo Tosatti     BlockCompleteData *blkdata = qemu_mallocz(sizeof(BlockCompleteData));
21704dcafbb1SMarcelo Tosatti 
21714dcafbb1SMarcelo Tosatti     blkdata->bs = bs;
21724dcafbb1SMarcelo Tosatti     blkdata->cb = cb;
21734dcafbb1SMarcelo Tosatti     blkdata->opaque = opaque;
21744dcafbb1SMarcelo Tosatti     blkdata->sector_num = sector_num;
21754dcafbb1SMarcelo Tosatti     blkdata->nb_sectors = nb_sectors;
21764dcafbb1SMarcelo Tosatti 
21774dcafbb1SMarcelo Tosatti     return blkdata;
21784dcafbb1SMarcelo Tosatti }
21794dcafbb1SMarcelo Tosatti 
2180f141eafeSaliguori BlockDriverAIOCB *bdrv_aio_writev(BlockDriverState *bs, int64_t sector_num,
2181f141eafeSaliguori                                   QEMUIOVector *qiov, int nb_sectors,
218283f64091Sbellard                                   BlockDriverCompletionFunc *cb, void *opaque)
21837674e7bfSbellard {
218483f64091Sbellard     BlockDriver *drv = bs->drv;
2185a36e69ddSths     BlockDriverAIOCB *ret;
21864dcafbb1SMarcelo Tosatti     BlockCompleteData *blk_cb_data;
218783f64091Sbellard 
2188bbf0a440SStefan Hajnoczi     trace_bdrv_aio_writev(bs, sector_num, nb_sectors, opaque);
2189bbf0a440SStefan Hajnoczi 
219019cb3738Sbellard     if (!drv)
2191ce1a14dcSpbrook         return NULL;
219283f64091Sbellard     if (bs->read_only)
2193ce1a14dcSpbrook         return NULL;
219471d0770cSaliguori     if (bdrv_check_request(bs, sector_num, nb_sectors))
219571d0770cSaliguori         return NULL;
219683f64091Sbellard 
2197c6d22830SJan Kiszka     if (bs->dirty_bitmap) {
21984dcafbb1SMarcelo Tosatti         blk_cb_data = blk_dirty_cb_alloc(bs, sector_num, nb_sectors, cb,
21994dcafbb1SMarcelo Tosatti                                          opaque);
22004dcafbb1SMarcelo Tosatti         cb = &block_complete_cb;
22014dcafbb1SMarcelo Tosatti         opaque = blk_cb_data;
22027cd1e32aSlirans@il.ibm.com     }
22037cd1e32aSlirans@il.ibm.com 
2204f141eafeSaliguori     ret = drv->bdrv_aio_writev(bs, sector_num, qiov, nb_sectors,
2205f141eafeSaliguori                                cb, opaque);
2206a36e69ddSths 
2207a36e69ddSths     if (ret) {
2208a36e69ddSths         /* Update stats even though technically transfer has not happened. */
22096ea44308SJan Kiszka         bs->wr_bytes += (unsigned) nb_sectors * BDRV_SECTOR_SIZE;
2210a36e69ddSths         bs->wr_ops ++;
2211294cc35fSKevin Wolf         if (bs->wr_highest_sector < sector_num + nb_sectors - 1) {
2212294cc35fSKevin Wolf             bs->wr_highest_sector = sector_num + nb_sectors - 1;
2213294cc35fSKevin Wolf         }
2214a36e69ddSths     }
2215a36e69ddSths 
2216a36e69ddSths     return ret;
221783f64091Sbellard }
221883f64091Sbellard 
221940b4f539SKevin Wolf 
222040b4f539SKevin Wolf typedef struct MultiwriteCB {
222140b4f539SKevin Wolf     int error;
222240b4f539SKevin Wolf     int num_requests;
222340b4f539SKevin Wolf     int num_callbacks;
222440b4f539SKevin Wolf     struct {
222540b4f539SKevin Wolf         BlockDriverCompletionFunc *cb;
222640b4f539SKevin Wolf         void *opaque;
222740b4f539SKevin Wolf         QEMUIOVector *free_qiov;
222840b4f539SKevin Wolf         void *free_buf;
222940b4f539SKevin Wolf     } callbacks[];
223040b4f539SKevin Wolf } MultiwriteCB;
223140b4f539SKevin Wolf 
223240b4f539SKevin Wolf static void multiwrite_user_cb(MultiwriteCB *mcb)
223340b4f539SKevin Wolf {
223440b4f539SKevin Wolf     int i;
223540b4f539SKevin Wolf 
223640b4f539SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
223740b4f539SKevin Wolf         mcb->callbacks[i].cb(mcb->callbacks[i].opaque, mcb->error);
22381e1ea48dSStefan Hajnoczi         if (mcb->callbacks[i].free_qiov) {
22391e1ea48dSStefan Hajnoczi             qemu_iovec_destroy(mcb->callbacks[i].free_qiov);
22401e1ea48dSStefan Hajnoczi         }
224140b4f539SKevin Wolf         qemu_free(mcb->callbacks[i].free_qiov);
2242f8a83245SHerve Poussineau         qemu_vfree(mcb->callbacks[i].free_buf);
224340b4f539SKevin Wolf     }
224440b4f539SKevin Wolf }
224540b4f539SKevin Wolf 
224640b4f539SKevin Wolf static void multiwrite_cb(void *opaque, int ret)
224740b4f539SKevin Wolf {
224840b4f539SKevin Wolf     MultiwriteCB *mcb = opaque;
224940b4f539SKevin Wolf 
22506d519a5fSStefan Hajnoczi     trace_multiwrite_cb(mcb, ret);
22516d519a5fSStefan Hajnoczi 
2252cb6d3ca0SKevin Wolf     if (ret < 0 && !mcb->error) {
225340b4f539SKevin Wolf         mcb->error = ret;
225440b4f539SKevin Wolf     }
225540b4f539SKevin Wolf 
225640b4f539SKevin Wolf     mcb->num_requests--;
225740b4f539SKevin Wolf     if (mcb->num_requests == 0) {
225840b4f539SKevin Wolf         multiwrite_user_cb(mcb);
225940b4f539SKevin Wolf         qemu_free(mcb);
226040b4f539SKevin Wolf     }
226140b4f539SKevin Wolf }
226240b4f539SKevin Wolf 
226340b4f539SKevin Wolf static int multiwrite_req_compare(const void *a, const void *b)
226440b4f539SKevin Wolf {
226577be4366SChristoph Hellwig     const BlockRequest *req1 = a, *req2 = b;
226677be4366SChristoph Hellwig 
226777be4366SChristoph Hellwig     /*
226877be4366SChristoph Hellwig      * Note that we can't simply subtract req2->sector from req1->sector
226977be4366SChristoph Hellwig      * here as that could overflow the return value.
227077be4366SChristoph Hellwig      */
227177be4366SChristoph Hellwig     if (req1->sector > req2->sector) {
227277be4366SChristoph Hellwig         return 1;
227377be4366SChristoph Hellwig     } else if (req1->sector < req2->sector) {
227477be4366SChristoph Hellwig         return -1;
227577be4366SChristoph Hellwig     } else {
227677be4366SChristoph Hellwig         return 0;
227777be4366SChristoph Hellwig     }
227840b4f539SKevin Wolf }
227940b4f539SKevin Wolf 
228040b4f539SKevin Wolf /*
228140b4f539SKevin Wolf  * Takes a bunch of requests and tries to merge them. Returns the number of
228240b4f539SKevin Wolf  * requests that remain after merging.
228340b4f539SKevin Wolf  */
228440b4f539SKevin Wolf static int multiwrite_merge(BlockDriverState *bs, BlockRequest *reqs,
228540b4f539SKevin Wolf     int num_reqs, MultiwriteCB *mcb)
228640b4f539SKevin Wolf {
228740b4f539SKevin Wolf     int i, outidx;
228840b4f539SKevin Wolf 
228940b4f539SKevin Wolf     // Sort requests by start sector
229040b4f539SKevin Wolf     qsort(reqs, num_reqs, sizeof(*reqs), &multiwrite_req_compare);
229140b4f539SKevin Wolf 
229240b4f539SKevin Wolf     // Check if adjacent requests touch the same clusters. If so, combine them,
229340b4f539SKevin Wolf     // filling up gaps with zero sectors.
229440b4f539SKevin Wolf     outidx = 0;
229540b4f539SKevin Wolf     for (i = 1; i < num_reqs; i++) {
229640b4f539SKevin Wolf         int merge = 0;
229740b4f539SKevin Wolf         int64_t oldreq_last = reqs[outidx].sector + reqs[outidx].nb_sectors;
229840b4f539SKevin Wolf 
229940b4f539SKevin Wolf         // This handles the cases that are valid for all block drivers, namely
230040b4f539SKevin Wolf         // exactly sequential writes and overlapping writes.
230140b4f539SKevin Wolf         if (reqs[i].sector <= oldreq_last) {
230240b4f539SKevin Wolf             merge = 1;
230340b4f539SKevin Wolf         }
230440b4f539SKevin Wolf 
230540b4f539SKevin Wolf         // The block driver may decide that it makes sense to combine requests
230640b4f539SKevin Wolf         // even if there is a gap of some sectors between them. In this case,
230740b4f539SKevin Wolf         // the gap is filled with zeros (therefore only applicable for yet
230840b4f539SKevin Wolf         // unused space in format like qcow2).
230940b4f539SKevin Wolf         if (!merge && bs->drv->bdrv_merge_requests) {
231040b4f539SKevin Wolf             merge = bs->drv->bdrv_merge_requests(bs, &reqs[outidx], &reqs[i]);
231140b4f539SKevin Wolf         }
231240b4f539SKevin Wolf 
2313e2a305fbSChristoph Hellwig         if (reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1 > IOV_MAX) {
2314e2a305fbSChristoph Hellwig             merge = 0;
2315e2a305fbSChristoph Hellwig         }
2316e2a305fbSChristoph Hellwig 
231740b4f539SKevin Wolf         if (merge) {
231840b4f539SKevin Wolf             size_t size;
231940b4f539SKevin Wolf             QEMUIOVector *qiov = qemu_mallocz(sizeof(*qiov));
232040b4f539SKevin Wolf             qemu_iovec_init(qiov,
232140b4f539SKevin Wolf                 reqs[outidx].qiov->niov + reqs[i].qiov->niov + 1);
232240b4f539SKevin Wolf 
232340b4f539SKevin Wolf             // Add the first request to the merged one. If the requests are
232440b4f539SKevin Wolf             // overlapping, drop the last sectors of the first request.
232540b4f539SKevin Wolf             size = (reqs[i].sector - reqs[outidx].sector) << 9;
232640b4f539SKevin Wolf             qemu_iovec_concat(qiov, reqs[outidx].qiov, size);
232740b4f539SKevin Wolf 
232840b4f539SKevin Wolf             // We might need to add some zeros between the two requests
232940b4f539SKevin Wolf             if (reqs[i].sector > oldreq_last) {
233040b4f539SKevin Wolf                 size_t zero_bytes = (reqs[i].sector - oldreq_last) << 9;
233140b4f539SKevin Wolf                 uint8_t *buf = qemu_blockalign(bs, zero_bytes);
233240b4f539SKevin Wolf                 memset(buf, 0, zero_bytes);
233340b4f539SKevin Wolf                 qemu_iovec_add(qiov, buf, zero_bytes);
233440b4f539SKevin Wolf                 mcb->callbacks[i].free_buf = buf;
233540b4f539SKevin Wolf             }
233640b4f539SKevin Wolf 
233740b4f539SKevin Wolf             // Add the second request
233840b4f539SKevin Wolf             qemu_iovec_concat(qiov, reqs[i].qiov, reqs[i].qiov->size);
233940b4f539SKevin Wolf 
2340cbf1dff2SKevin Wolf             reqs[outidx].nb_sectors = qiov->size >> 9;
234140b4f539SKevin Wolf             reqs[outidx].qiov = qiov;
234240b4f539SKevin Wolf 
234340b4f539SKevin Wolf             mcb->callbacks[i].free_qiov = reqs[outidx].qiov;
234440b4f539SKevin Wolf         } else {
234540b4f539SKevin Wolf             outidx++;
234640b4f539SKevin Wolf             reqs[outidx].sector     = reqs[i].sector;
234740b4f539SKevin Wolf             reqs[outidx].nb_sectors = reqs[i].nb_sectors;
234840b4f539SKevin Wolf             reqs[outidx].qiov       = reqs[i].qiov;
234940b4f539SKevin Wolf         }
235040b4f539SKevin Wolf     }
235140b4f539SKevin Wolf 
235240b4f539SKevin Wolf     return outidx + 1;
235340b4f539SKevin Wolf }
235440b4f539SKevin Wolf 
235540b4f539SKevin Wolf /*
235640b4f539SKevin Wolf  * Submit multiple AIO write requests at once.
235740b4f539SKevin Wolf  *
235840b4f539SKevin Wolf  * On success, the function returns 0 and all requests in the reqs array have
235940b4f539SKevin Wolf  * been submitted. In error case this function returns -1, and any of the
236040b4f539SKevin Wolf  * requests may or may not be submitted yet. In particular, this means that the
236140b4f539SKevin Wolf  * callback will be called for some of the requests, for others it won't. The
236240b4f539SKevin Wolf  * caller must check the error field of the BlockRequest to wait for the right
236340b4f539SKevin Wolf  * callbacks (if error != 0, no callback will be called).
236440b4f539SKevin Wolf  *
236540b4f539SKevin Wolf  * The implementation may modify the contents of the reqs array, e.g. to merge
236640b4f539SKevin Wolf  * requests. However, the fields opaque and error are left unmodified as they
236740b4f539SKevin Wolf  * are used to signal failure for a single request to the caller.
236840b4f539SKevin Wolf  */
236940b4f539SKevin Wolf int bdrv_aio_multiwrite(BlockDriverState *bs, BlockRequest *reqs, int num_reqs)
237040b4f539SKevin Wolf {
237140b4f539SKevin Wolf     BlockDriverAIOCB *acb;
237240b4f539SKevin Wolf     MultiwriteCB *mcb;
237340b4f539SKevin Wolf     int i;
237440b4f539SKevin Wolf 
2375301db7c2SRyan Harper     /* don't submit writes if we don't have a medium */
2376301db7c2SRyan Harper     if (bs->drv == NULL) {
2377301db7c2SRyan Harper         for (i = 0; i < num_reqs; i++) {
2378301db7c2SRyan Harper             reqs[i].error = -ENOMEDIUM;
2379301db7c2SRyan Harper         }
2380301db7c2SRyan Harper         return -1;
2381301db7c2SRyan Harper     }
2382301db7c2SRyan Harper 
238340b4f539SKevin Wolf     if (num_reqs == 0) {
238440b4f539SKevin Wolf         return 0;
238540b4f539SKevin Wolf     }
238640b4f539SKevin Wolf 
238740b4f539SKevin Wolf     // Create MultiwriteCB structure
238840b4f539SKevin Wolf     mcb = qemu_mallocz(sizeof(*mcb) + num_reqs * sizeof(*mcb->callbacks));
238940b4f539SKevin Wolf     mcb->num_requests = 0;
239040b4f539SKevin Wolf     mcb->num_callbacks = num_reqs;
239140b4f539SKevin Wolf 
239240b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
239340b4f539SKevin Wolf         mcb->callbacks[i].cb = reqs[i].cb;
239440b4f539SKevin Wolf         mcb->callbacks[i].opaque = reqs[i].opaque;
239540b4f539SKevin Wolf     }
239640b4f539SKevin Wolf 
239740b4f539SKevin Wolf     // Check for mergable requests
239840b4f539SKevin Wolf     num_reqs = multiwrite_merge(bs, reqs, num_reqs, mcb);
239940b4f539SKevin Wolf 
24006d519a5fSStefan Hajnoczi     trace_bdrv_aio_multiwrite(mcb, mcb->num_callbacks, num_reqs);
24016d519a5fSStefan Hajnoczi 
2402453f9a16SKevin Wolf     /*
2403453f9a16SKevin Wolf      * Run the aio requests. As soon as one request can't be submitted
2404453f9a16SKevin Wolf      * successfully, fail all requests that are not yet submitted (we must
2405453f9a16SKevin Wolf      * return failure for all requests anyway)
2406453f9a16SKevin Wolf      *
2407453f9a16SKevin Wolf      * num_requests cannot be set to the right value immediately: If
2408453f9a16SKevin Wolf      * bdrv_aio_writev fails for some request, num_requests would be too high
2409453f9a16SKevin Wolf      * and therefore multiwrite_cb() would never recognize the multiwrite
2410453f9a16SKevin Wolf      * request as completed. We also cannot use the loop variable i to set it
2411453f9a16SKevin Wolf      * when the first request fails because the callback may already have been
2412453f9a16SKevin Wolf      * called for previously submitted requests. Thus, num_requests must be
2413453f9a16SKevin Wolf      * incremented for each request that is submitted.
2414453f9a16SKevin Wolf      *
2415453f9a16SKevin Wolf      * The problem that callbacks may be called early also means that we need
2416453f9a16SKevin Wolf      * to take care that num_requests doesn't become 0 before all requests are
2417453f9a16SKevin Wolf      * submitted - multiwrite_cb() would consider the multiwrite request
2418453f9a16SKevin Wolf      * completed. A dummy request that is "completed" by a manual call to
2419453f9a16SKevin Wolf      * multiwrite_cb() takes care of this.
2420453f9a16SKevin Wolf      */
2421453f9a16SKevin Wolf     mcb->num_requests = 1;
2422453f9a16SKevin Wolf 
24236d519a5fSStefan Hajnoczi     // Run the aio requests
242440b4f539SKevin Wolf     for (i = 0; i < num_reqs; i++) {
2425453f9a16SKevin Wolf         mcb->num_requests++;
242640b4f539SKevin Wolf         acb = bdrv_aio_writev(bs, reqs[i].sector, reqs[i].qiov,
242740b4f539SKevin Wolf             reqs[i].nb_sectors, multiwrite_cb, mcb);
242840b4f539SKevin Wolf 
242940b4f539SKevin Wolf         if (acb == NULL) {
243040b4f539SKevin Wolf             // We can only fail the whole thing if no request has been
243140b4f539SKevin Wolf             // submitted yet. Otherwise we'll wait for the submitted AIOs to
243240b4f539SKevin Wolf             // complete and report the error in the callback.
2433453f9a16SKevin Wolf             if (i == 0) {
24346d519a5fSStefan Hajnoczi                 trace_bdrv_aio_multiwrite_earlyfail(mcb);
243540b4f539SKevin Wolf                 goto fail;
243640b4f539SKevin Wolf             } else {
24376d519a5fSStefan Hajnoczi                 trace_bdrv_aio_multiwrite_latefail(mcb, i);
24387eb58a6cSKevin Wolf                 multiwrite_cb(mcb, -EIO);
243940b4f539SKevin Wolf                 break;
244040b4f539SKevin Wolf             }
244140b4f539SKevin Wolf         }
244240b4f539SKevin Wolf     }
244340b4f539SKevin Wolf 
2444453f9a16SKevin Wolf     /* Complete the dummy request */
2445453f9a16SKevin Wolf     multiwrite_cb(mcb, 0);
2446453f9a16SKevin Wolf 
244740b4f539SKevin Wolf     return 0;
244840b4f539SKevin Wolf 
244940b4f539SKevin Wolf fail:
2450453f9a16SKevin Wolf     for (i = 0; i < mcb->num_callbacks; i++) {
2451453f9a16SKevin Wolf         reqs[i].error = -EIO;
2452453f9a16SKevin Wolf     }
2453af474591SBruce Rogers     qemu_free(mcb);
245440b4f539SKevin Wolf     return -1;
245540b4f539SKevin Wolf }
245640b4f539SKevin Wolf 
2457b2e12bc6SChristoph Hellwig BlockDriverAIOCB *bdrv_aio_flush(BlockDriverState *bs,
2458b2e12bc6SChristoph Hellwig         BlockDriverCompletionFunc *cb, void *opaque)
2459b2e12bc6SChristoph Hellwig {
2460b2e12bc6SChristoph Hellwig     BlockDriver *drv = bs->drv;
2461b2e12bc6SChristoph Hellwig 
2462a13aac04SStefan Hajnoczi     trace_bdrv_aio_flush(bs, opaque);
2463a13aac04SStefan Hajnoczi 
2464016f5cf6SAlexander Graf     if (bs->open_flags & BDRV_O_NO_FLUSH) {
2465016f5cf6SAlexander Graf         return bdrv_aio_noop_em(bs, cb, opaque);
2466016f5cf6SAlexander Graf     }
2467016f5cf6SAlexander Graf 
2468b2e12bc6SChristoph Hellwig     if (!drv)
2469b2e12bc6SChristoph Hellwig         return NULL;
2470b2e12bc6SChristoph Hellwig     return drv->bdrv_aio_flush(bs, cb, opaque);
2471b2e12bc6SChristoph Hellwig }
2472b2e12bc6SChristoph Hellwig 
247383f64091Sbellard void bdrv_aio_cancel(BlockDriverAIOCB *acb)
247483f64091Sbellard {
24756bbff9a0Saliguori     acb->pool->cancel(acb);
247683f64091Sbellard }
247783f64091Sbellard 
247883f64091Sbellard 
247983f64091Sbellard /**************************************************************/
248083f64091Sbellard /* async block device emulation */
248183f64091Sbellard 
2482c16b5a2cSChristoph Hellwig typedef struct BlockDriverAIOCBSync {
2483c16b5a2cSChristoph Hellwig     BlockDriverAIOCB common;
2484c16b5a2cSChristoph Hellwig     QEMUBH *bh;
2485c16b5a2cSChristoph Hellwig     int ret;
2486c16b5a2cSChristoph Hellwig     /* vector translation state */
2487c16b5a2cSChristoph Hellwig     QEMUIOVector *qiov;
2488c16b5a2cSChristoph Hellwig     uint8_t *bounce;
2489c16b5a2cSChristoph Hellwig     int is_write;
2490c16b5a2cSChristoph Hellwig } BlockDriverAIOCBSync;
2491c16b5a2cSChristoph Hellwig 
2492c16b5a2cSChristoph Hellwig static void bdrv_aio_cancel_em(BlockDriverAIOCB *blockacb)
2493c16b5a2cSChristoph Hellwig {
2494b666d239SKevin Wolf     BlockDriverAIOCBSync *acb =
2495b666d239SKevin Wolf         container_of(blockacb, BlockDriverAIOCBSync, common);
24966a7ad299SDor Laor     qemu_bh_delete(acb->bh);
249736afc451SAvi Kivity     acb->bh = NULL;
2498c16b5a2cSChristoph Hellwig     qemu_aio_release(acb);
2499c16b5a2cSChristoph Hellwig }
2500c16b5a2cSChristoph Hellwig 
2501c16b5a2cSChristoph Hellwig static AIOPool bdrv_em_aio_pool = {
2502c16b5a2cSChristoph Hellwig     .aiocb_size         = sizeof(BlockDriverAIOCBSync),
2503c16b5a2cSChristoph Hellwig     .cancel             = bdrv_aio_cancel_em,
2504c16b5a2cSChristoph Hellwig };
2505c16b5a2cSChristoph Hellwig 
250683f64091Sbellard static void bdrv_aio_bh_cb(void *opaque)
2507beac80cdSbellard {
2508ce1a14dcSpbrook     BlockDriverAIOCBSync *acb = opaque;
2509f141eafeSaliguori 
2510f141eafeSaliguori     if (!acb->is_write)
2511f141eafeSaliguori         qemu_iovec_from_buffer(acb->qiov, acb->bounce, acb->qiov->size);
2512ceb42de8Saliguori     qemu_vfree(acb->bounce);
2513ce1a14dcSpbrook     acb->common.cb(acb->common.opaque, acb->ret);
25146a7ad299SDor Laor     qemu_bh_delete(acb->bh);
251536afc451SAvi Kivity     acb->bh = NULL;
2516ce1a14dcSpbrook     qemu_aio_release(acb);
2517beac80cdSbellard }
2518beac80cdSbellard 
2519f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_rw_vector(BlockDriverState *bs,
2520f141eafeSaliguori                                             int64_t sector_num,
2521f141eafeSaliguori                                             QEMUIOVector *qiov,
2522f141eafeSaliguori                                             int nb_sectors,
2523f141eafeSaliguori                                             BlockDriverCompletionFunc *cb,
2524f141eafeSaliguori                                             void *opaque,
2525f141eafeSaliguori                                             int is_write)
2526f141eafeSaliguori 
2527ea2384d3Sbellard {
2528ce1a14dcSpbrook     BlockDriverAIOCBSync *acb;
252983f64091Sbellard 
2530c16b5a2cSChristoph Hellwig     acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2531f141eafeSaliguori     acb->is_write = is_write;
2532f141eafeSaliguori     acb->qiov = qiov;
2533e268ca52Saliguori     acb->bounce = qemu_blockalign(bs, qiov->size);
2534f141eafeSaliguori 
2535ce1a14dcSpbrook     if (!acb->bh)
2536ce1a14dcSpbrook         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2537f141eafeSaliguori 
2538f141eafeSaliguori     if (is_write) {
2539f141eafeSaliguori         qemu_iovec_to_buffer(acb->qiov, acb->bounce);
2540f141eafeSaliguori         acb->ret = bdrv_write(bs, sector_num, acb->bounce, nb_sectors);
2541f141eafeSaliguori     } else {
2542f141eafeSaliguori         acb->ret = bdrv_read(bs, sector_num, acb->bounce, nb_sectors);
2543f141eafeSaliguori     }
2544f141eafeSaliguori 
2545ce1a14dcSpbrook     qemu_bh_schedule(acb->bh);
2546f141eafeSaliguori 
2547ce1a14dcSpbrook     return &acb->common;
25487a6cba61Spbrook }
25497a6cba61Spbrook 
2550f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_readv_em(BlockDriverState *bs,
2551f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
2552ce1a14dcSpbrook         BlockDriverCompletionFunc *cb, void *opaque)
255383f64091Sbellard {
2554f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 0);
255583f64091Sbellard }
255683f64091Sbellard 
2557f141eafeSaliguori static BlockDriverAIOCB *bdrv_aio_writev_em(BlockDriverState *bs,
2558f141eafeSaliguori         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
2559f141eafeSaliguori         BlockDriverCompletionFunc *cb, void *opaque)
2560f141eafeSaliguori {
2561f141eafeSaliguori     return bdrv_aio_rw_vector(bs, sector_num, qiov, nb_sectors, cb, opaque, 1);
2562f141eafeSaliguori }
2563f141eafeSaliguori 
2564b2e12bc6SChristoph Hellwig static BlockDriverAIOCB *bdrv_aio_flush_em(BlockDriverState *bs,
2565b2e12bc6SChristoph Hellwig         BlockDriverCompletionFunc *cb, void *opaque)
2566b2e12bc6SChristoph Hellwig {
2567b2e12bc6SChristoph Hellwig     BlockDriverAIOCBSync *acb;
2568b2e12bc6SChristoph Hellwig 
2569b2e12bc6SChristoph Hellwig     acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2570b2e12bc6SChristoph Hellwig     acb->is_write = 1; /* don't bounce in the completion hadler */
2571b2e12bc6SChristoph Hellwig     acb->qiov = NULL;
2572b2e12bc6SChristoph Hellwig     acb->bounce = NULL;
2573b2e12bc6SChristoph Hellwig     acb->ret = 0;
2574b2e12bc6SChristoph Hellwig 
2575b2e12bc6SChristoph Hellwig     if (!acb->bh)
2576b2e12bc6SChristoph Hellwig         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2577b2e12bc6SChristoph Hellwig 
2578b2e12bc6SChristoph Hellwig     bdrv_flush(bs);
2579b2e12bc6SChristoph Hellwig     qemu_bh_schedule(acb->bh);
2580b2e12bc6SChristoph Hellwig     return &acb->common;
2581b2e12bc6SChristoph Hellwig }
2582b2e12bc6SChristoph Hellwig 
2583016f5cf6SAlexander Graf static BlockDriverAIOCB *bdrv_aio_noop_em(BlockDriverState *bs,
2584016f5cf6SAlexander Graf         BlockDriverCompletionFunc *cb, void *opaque)
2585016f5cf6SAlexander Graf {
2586016f5cf6SAlexander Graf     BlockDriverAIOCBSync *acb;
2587016f5cf6SAlexander Graf 
2588016f5cf6SAlexander Graf     acb = qemu_aio_get(&bdrv_em_aio_pool, bs, cb, opaque);
2589016f5cf6SAlexander Graf     acb->is_write = 1; /* don't bounce in the completion handler */
2590016f5cf6SAlexander Graf     acb->qiov = NULL;
2591016f5cf6SAlexander Graf     acb->bounce = NULL;
2592016f5cf6SAlexander Graf     acb->ret = 0;
2593016f5cf6SAlexander Graf 
2594016f5cf6SAlexander Graf     if (!acb->bh) {
2595016f5cf6SAlexander Graf         acb->bh = qemu_bh_new(bdrv_aio_bh_cb, acb);
2596016f5cf6SAlexander Graf     }
2597016f5cf6SAlexander Graf 
2598016f5cf6SAlexander Graf     qemu_bh_schedule(acb->bh);
2599016f5cf6SAlexander Graf     return &acb->common;
2600016f5cf6SAlexander Graf }
2601016f5cf6SAlexander Graf 
260283f64091Sbellard /**************************************************************/
260383f64091Sbellard /* sync block device emulation */
260483f64091Sbellard 
260583f64091Sbellard static void bdrv_rw_em_cb(void *opaque, int ret)
260683f64091Sbellard {
260783f64091Sbellard     *(int *)opaque = ret;
260883f64091Sbellard }
260983f64091Sbellard 
261083f64091Sbellard #define NOT_DONE 0x7fffffff
261183f64091Sbellard 
261283f64091Sbellard static int bdrv_read_em(BlockDriverState *bs, int64_t sector_num,
261383f64091Sbellard                         uint8_t *buf, int nb_sectors)
261483f64091Sbellard {
2615ce1a14dcSpbrook     int async_ret;
2616ce1a14dcSpbrook     BlockDriverAIOCB *acb;
2617f141eafeSaliguori     struct iovec iov;
2618f141eafeSaliguori     QEMUIOVector qiov;
261983f64091Sbellard 
262065d6b3d8SKevin Wolf     async_context_push();
262165d6b3d8SKevin Wolf 
262283f64091Sbellard     async_ret = NOT_DONE;
26233f4cb3d3Sblueswir1     iov.iov_base = (void *)buf;
2624eb5a3165SJes Sorensen     iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2625f141eafeSaliguori     qemu_iovec_init_external(&qiov, &iov, 1);
2626f141eafeSaliguori     acb = bdrv_aio_readv(bs, sector_num, &qiov, nb_sectors,
262783f64091Sbellard         bdrv_rw_em_cb, &async_ret);
262865d6b3d8SKevin Wolf     if (acb == NULL) {
262965d6b3d8SKevin Wolf         async_ret = -1;
263065d6b3d8SKevin Wolf         goto fail;
263165d6b3d8SKevin Wolf     }
2632baf35cb9Saliguori 
263383f64091Sbellard     while (async_ret == NOT_DONE) {
263483f64091Sbellard         qemu_aio_wait();
263583f64091Sbellard     }
2636baf35cb9Saliguori 
263765d6b3d8SKevin Wolf 
263865d6b3d8SKevin Wolf fail:
263965d6b3d8SKevin Wolf     async_context_pop();
264083f64091Sbellard     return async_ret;
264183f64091Sbellard }
264283f64091Sbellard 
264383f64091Sbellard static int bdrv_write_em(BlockDriverState *bs, int64_t sector_num,
264483f64091Sbellard                          const uint8_t *buf, int nb_sectors)
264583f64091Sbellard {
2646ce1a14dcSpbrook     int async_ret;
2647ce1a14dcSpbrook     BlockDriverAIOCB *acb;
2648f141eafeSaliguori     struct iovec iov;
2649f141eafeSaliguori     QEMUIOVector qiov;
265083f64091Sbellard 
265165d6b3d8SKevin Wolf     async_context_push();
265265d6b3d8SKevin Wolf 
265383f64091Sbellard     async_ret = NOT_DONE;
2654f141eafeSaliguori     iov.iov_base = (void *)buf;
2655eb5a3165SJes Sorensen     iov.iov_len = nb_sectors * BDRV_SECTOR_SIZE;
2656f141eafeSaliguori     qemu_iovec_init_external(&qiov, &iov, 1);
2657f141eafeSaliguori     acb = bdrv_aio_writev(bs, sector_num, &qiov, nb_sectors,
265883f64091Sbellard         bdrv_rw_em_cb, &async_ret);
265965d6b3d8SKevin Wolf     if (acb == NULL) {
266065d6b3d8SKevin Wolf         async_ret = -1;
266165d6b3d8SKevin Wolf         goto fail;
266265d6b3d8SKevin Wolf     }
266383f64091Sbellard     while (async_ret == NOT_DONE) {
266483f64091Sbellard         qemu_aio_wait();
266583f64091Sbellard     }
266665d6b3d8SKevin Wolf 
266765d6b3d8SKevin Wolf fail:
266865d6b3d8SKevin Wolf     async_context_pop();
266983f64091Sbellard     return async_ret;
267083f64091Sbellard }
2671ea2384d3Sbellard 
2672ea2384d3Sbellard void bdrv_init(void)
2673ea2384d3Sbellard {
26745efa9d5aSAnthony Liguori     module_call_init(MODULE_INIT_BLOCK);
2675ea2384d3Sbellard }
2676ce1a14dcSpbrook 
2677eb852011SMarkus Armbruster void bdrv_init_with_whitelist(void)
2678eb852011SMarkus Armbruster {
2679eb852011SMarkus Armbruster     use_bdrv_whitelist = 1;
2680eb852011SMarkus Armbruster     bdrv_init();
2681eb852011SMarkus Armbruster }
2682eb852011SMarkus Armbruster 
2683c16b5a2cSChristoph Hellwig void *qemu_aio_get(AIOPool *pool, BlockDriverState *bs,
26846bbff9a0Saliguori                    BlockDriverCompletionFunc *cb, void *opaque)
26856bbff9a0Saliguori {
2686ce1a14dcSpbrook     BlockDriverAIOCB *acb;
2687ce1a14dcSpbrook 
26886bbff9a0Saliguori     if (pool->free_aiocb) {
26896bbff9a0Saliguori         acb = pool->free_aiocb;
26906bbff9a0Saliguori         pool->free_aiocb = acb->next;
2691ce1a14dcSpbrook     } else {
26926bbff9a0Saliguori         acb = qemu_mallocz(pool->aiocb_size);
26936bbff9a0Saliguori         acb->pool = pool;
2694ce1a14dcSpbrook     }
2695ce1a14dcSpbrook     acb->bs = bs;
2696ce1a14dcSpbrook     acb->cb = cb;
2697ce1a14dcSpbrook     acb->opaque = opaque;
2698ce1a14dcSpbrook     return acb;
2699ce1a14dcSpbrook }
2700ce1a14dcSpbrook 
2701ce1a14dcSpbrook void qemu_aio_release(void *p)
2702ce1a14dcSpbrook {
27036bbff9a0Saliguori     BlockDriverAIOCB *acb = (BlockDriverAIOCB *)p;
27046bbff9a0Saliguori     AIOPool *pool = acb->pool;
27056bbff9a0Saliguori     acb->next = pool->free_aiocb;
27066bbff9a0Saliguori     pool->free_aiocb = acb;
2707ce1a14dcSpbrook }
270819cb3738Sbellard 
270919cb3738Sbellard /**************************************************************/
271019cb3738Sbellard /* removable device support */
271119cb3738Sbellard 
271219cb3738Sbellard /**
271319cb3738Sbellard  * Return TRUE if the media is present
271419cb3738Sbellard  */
271519cb3738Sbellard int bdrv_is_inserted(BlockDriverState *bs)
271619cb3738Sbellard {
271719cb3738Sbellard     BlockDriver *drv = bs->drv;
271819cb3738Sbellard     int ret;
271919cb3738Sbellard     if (!drv)
272019cb3738Sbellard         return 0;
272119cb3738Sbellard     if (!drv->bdrv_is_inserted)
27224be9762aSMarkus Armbruster         return !bs->tray_open;
272319cb3738Sbellard     ret = drv->bdrv_is_inserted(bs);
272419cb3738Sbellard     return ret;
272519cb3738Sbellard }
272619cb3738Sbellard 
272719cb3738Sbellard /**
272819cb3738Sbellard  * Return TRUE if the media changed since the last call to this
272919cb3738Sbellard  * function. It is currently only used for floppy disks
273019cb3738Sbellard  */
273119cb3738Sbellard int bdrv_media_changed(BlockDriverState *bs)
273219cb3738Sbellard {
273319cb3738Sbellard     BlockDriver *drv = bs->drv;
273419cb3738Sbellard     int ret;
273519cb3738Sbellard 
273619cb3738Sbellard     if (!drv || !drv->bdrv_media_changed)
273719cb3738Sbellard         ret = -ENOTSUP;
273819cb3738Sbellard     else
273919cb3738Sbellard         ret = drv->bdrv_media_changed(bs);
274019cb3738Sbellard     if (ret == -ENOTSUP)
274119cb3738Sbellard         ret = bs->media_changed;
274219cb3738Sbellard     bs->media_changed = 0;
274319cb3738Sbellard     return ret;
274419cb3738Sbellard }
274519cb3738Sbellard 
274619cb3738Sbellard /**
274719cb3738Sbellard  * If eject_flag is TRUE, eject the media. Otherwise, close the tray
274819cb3738Sbellard  */
2749aea2a33cSMark McLoughlin int bdrv_eject(BlockDriverState *bs, int eject_flag)
275019cb3738Sbellard {
275119cb3738Sbellard     BlockDriver *drv = bs->drv;
275219cb3738Sbellard     int ret;
275319cb3738Sbellard 
2754aea2a33cSMark McLoughlin     if (bs->locked) {
2755aea2a33cSMark McLoughlin         return -EBUSY;
2756aea2a33cSMark McLoughlin     }
2757aea2a33cSMark McLoughlin 
275819cb3738Sbellard     if (!drv || !drv->bdrv_eject) {
275919cb3738Sbellard         ret = -ENOTSUP;
276019cb3738Sbellard     } else {
276119cb3738Sbellard         ret = drv->bdrv_eject(bs, eject_flag);
276219cb3738Sbellard     }
276319cb3738Sbellard     if (ret == -ENOTSUP) {
2764aea2a33cSMark McLoughlin         ret = 0;
276519cb3738Sbellard     }
27664be9762aSMarkus Armbruster     if (ret >= 0) {
27674be9762aSMarkus Armbruster         bs->tray_open = eject_flag;
27684be9762aSMarkus Armbruster     }
2769aea2a33cSMark McLoughlin 
2770aea2a33cSMark McLoughlin     return ret;
277119cb3738Sbellard }
277219cb3738Sbellard 
277319cb3738Sbellard int bdrv_is_locked(BlockDriverState *bs)
277419cb3738Sbellard {
277519cb3738Sbellard     return bs->locked;
277619cb3738Sbellard }
277719cb3738Sbellard 
277819cb3738Sbellard /**
277919cb3738Sbellard  * Lock or unlock the media (if it is locked, the user won't be able
278019cb3738Sbellard  * to eject it manually).
278119cb3738Sbellard  */
278219cb3738Sbellard void bdrv_set_locked(BlockDriverState *bs, int locked)
278319cb3738Sbellard {
278419cb3738Sbellard     BlockDriver *drv = bs->drv;
278519cb3738Sbellard 
2786b8c6d095SStefan Hajnoczi     trace_bdrv_set_locked(bs, locked);
2787b8c6d095SStefan Hajnoczi 
278819cb3738Sbellard     bs->locked = locked;
278919cb3738Sbellard     if (drv && drv->bdrv_set_locked) {
279019cb3738Sbellard         drv->bdrv_set_locked(bs, locked);
279119cb3738Sbellard     }
279219cb3738Sbellard }
2793985a03b0Sths 
2794985a03b0Sths /* needed for generic scsi interface */
2795985a03b0Sths 
2796985a03b0Sths int bdrv_ioctl(BlockDriverState *bs, unsigned long int req, void *buf)
2797985a03b0Sths {
2798985a03b0Sths     BlockDriver *drv = bs->drv;
2799985a03b0Sths 
2800985a03b0Sths     if (drv && drv->bdrv_ioctl)
2801985a03b0Sths         return drv->bdrv_ioctl(bs, req, buf);
2802985a03b0Sths     return -ENOTSUP;
2803985a03b0Sths }
28047d780669Saliguori 
2805221f715dSaliguori BlockDriverAIOCB *bdrv_aio_ioctl(BlockDriverState *bs,
2806221f715dSaliguori         unsigned long int req, void *buf,
28077d780669Saliguori         BlockDriverCompletionFunc *cb, void *opaque)
28087d780669Saliguori {
2809221f715dSaliguori     BlockDriver *drv = bs->drv;
28107d780669Saliguori 
2811221f715dSaliguori     if (drv && drv->bdrv_aio_ioctl)
2812221f715dSaliguori         return drv->bdrv_aio_ioctl(bs, req, buf, cb, opaque);
2813221f715dSaliguori     return NULL;
28147d780669Saliguori }
2815e268ca52Saliguori 
28167cd1e32aSlirans@il.ibm.com 
28177cd1e32aSlirans@il.ibm.com 
2818e268ca52Saliguori void *qemu_blockalign(BlockDriverState *bs, size_t size)
2819e268ca52Saliguori {
2820e268ca52Saliguori     return qemu_memalign((bs && bs->buffer_alignment) ? bs->buffer_alignment : 512, size);
2821e268ca52Saliguori }
28227cd1e32aSlirans@il.ibm.com 
28237cd1e32aSlirans@il.ibm.com void bdrv_set_dirty_tracking(BlockDriverState *bs, int enable)
28247cd1e32aSlirans@il.ibm.com {
28257cd1e32aSlirans@il.ibm.com     int64_t bitmap_size;
2826a55eb92cSJan Kiszka 
2827aaa0eb75SLiran Schour     bs->dirty_count = 0;
28287cd1e32aSlirans@il.ibm.com     if (enable) {
2829c6d22830SJan Kiszka         if (!bs->dirty_bitmap) {
2830c6d22830SJan Kiszka             bitmap_size = (bdrv_getlength(bs) >> BDRV_SECTOR_BITS) +
2831c6d22830SJan Kiszka                     BDRV_SECTORS_PER_DIRTY_CHUNK * 8 - 1;
2832c6d22830SJan Kiszka             bitmap_size /= BDRV_SECTORS_PER_DIRTY_CHUNK * 8;
28337cd1e32aSlirans@il.ibm.com 
28347cd1e32aSlirans@il.ibm.com             bs->dirty_bitmap = qemu_mallocz(bitmap_size);
28357cd1e32aSlirans@il.ibm.com         }
28367cd1e32aSlirans@il.ibm.com     } else {
2837c6d22830SJan Kiszka         if (bs->dirty_bitmap) {
28387cd1e32aSlirans@il.ibm.com             qemu_free(bs->dirty_bitmap);
2839c6d22830SJan Kiszka             bs->dirty_bitmap = NULL;
28407cd1e32aSlirans@il.ibm.com         }
28417cd1e32aSlirans@il.ibm.com     }
28427cd1e32aSlirans@il.ibm.com }
28437cd1e32aSlirans@il.ibm.com 
28447cd1e32aSlirans@il.ibm.com int bdrv_get_dirty(BlockDriverState *bs, int64_t sector)
28457cd1e32aSlirans@il.ibm.com {
28466ea44308SJan Kiszka     int64_t chunk = sector / (int64_t)BDRV_SECTORS_PER_DIRTY_CHUNK;
28477cd1e32aSlirans@il.ibm.com 
2848c6d22830SJan Kiszka     if (bs->dirty_bitmap &&
2849c6d22830SJan Kiszka         (sector << BDRV_SECTOR_BITS) < bdrv_getlength(bs)) {
28506d59fec1SMarcelo Tosatti         return !!(bs->dirty_bitmap[chunk / (sizeof(unsigned long) * 8)] &
28516d59fec1SMarcelo Tosatti             (1UL << (chunk % (sizeof(unsigned long) * 8))));
28527cd1e32aSlirans@il.ibm.com     } else {
28537cd1e32aSlirans@il.ibm.com         return 0;
28547cd1e32aSlirans@il.ibm.com     }
28557cd1e32aSlirans@il.ibm.com }
28567cd1e32aSlirans@il.ibm.com 
28577cd1e32aSlirans@il.ibm.com void bdrv_reset_dirty(BlockDriverState *bs, int64_t cur_sector,
28587cd1e32aSlirans@il.ibm.com                       int nr_sectors)
28597cd1e32aSlirans@il.ibm.com {
28607cd1e32aSlirans@il.ibm.com     set_dirty_bitmap(bs, cur_sector, nr_sectors, 0);
28617cd1e32aSlirans@il.ibm.com }
2862aaa0eb75SLiran Schour 
2863aaa0eb75SLiran Schour int64_t bdrv_get_dirty_count(BlockDriverState *bs)
2864aaa0eb75SLiran Schour {
2865aaa0eb75SLiran Schour     return bs->dirty_count;
2866aaa0eb75SLiran Schour }
2867f88e1a42SJes Sorensen 
2868db593f25SMarcelo Tosatti void bdrv_set_in_use(BlockDriverState *bs, int in_use)
2869db593f25SMarcelo Tosatti {
2870db593f25SMarcelo Tosatti     assert(bs->in_use != in_use);
2871db593f25SMarcelo Tosatti     bs->in_use = in_use;
2872db593f25SMarcelo Tosatti }
2873db593f25SMarcelo Tosatti 
2874db593f25SMarcelo Tosatti int bdrv_in_use(BlockDriverState *bs)
2875db593f25SMarcelo Tosatti {
2876db593f25SMarcelo Tosatti     return bs->in_use;
2877db593f25SMarcelo Tosatti }
2878db593f25SMarcelo Tosatti 
2879f88e1a42SJes Sorensen int bdrv_img_create(const char *filename, const char *fmt,
2880f88e1a42SJes Sorensen                     const char *base_filename, const char *base_fmt,
2881f88e1a42SJes Sorensen                     char *options, uint64_t img_size, int flags)
2882f88e1a42SJes Sorensen {
2883f88e1a42SJes Sorensen     QEMUOptionParameter *param = NULL, *create_options = NULL;
2884*d220894eSKevin Wolf     QEMUOptionParameter *backing_fmt, *backing_file, *size;
2885f88e1a42SJes Sorensen     BlockDriverState *bs = NULL;
2886f88e1a42SJes Sorensen     BlockDriver *drv, *proto_drv;
288796df67d1SStefan Hajnoczi     BlockDriver *backing_drv = NULL;
2888f88e1a42SJes Sorensen     int ret = 0;
2889f88e1a42SJes Sorensen 
2890f88e1a42SJes Sorensen     /* Find driver and parse its options */
2891f88e1a42SJes Sorensen     drv = bdrv_find_format(fmt);
2892f88e1a42SJes Sorensen     if (!drv) {
2893f88e1a42SJes Sorensen         error_report("Unknown file format '%s'", fmt);
28944f70f249SJes Sorensen         ret = -EINVAL;
2895f88e1a42SJes Sorensen         goto out;
2896f88e1a42SJes Sorensen     }
2897f88e1a42SJes Sorensen 
2898f88e1a42SJes Sorensen     proto_drv = bdrv_find_protocol(filename);
2899f88e1a42SJes Sorensen     if (!proto_drv) {
2900f88e1a42SJes Sorensen         error_report("Unknown protocol '%s'", filename);
29014f70f249SJes Sorensen         ret = -EINVAL;
2902f88e1a42SJes Sorensen         goto out;
2903f88e1a42SJes Sorensen     }
2904f88e1a42SJes Sorensen 
2905f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
2906f88e1a42SJes Sorensen                                               drv->create_options);
2907f88e1a42SJes Sorensen     create_options = append_option_parameters(create_options,
2908f88e1a42SJes Sorensen                                               proto_drv->create_options);
2909f88e1a42SJes Sorensen 
2910f88e1a42SJes Sorensen     /* Create parameter list with default values */
2911f88e1a42SJes Sorensen     param = parse_option_parameters("", create_options, param);
2912f88e1a42SJes Sorensen 
2913f88e1a42SJes Sorensen     set_option_parameter_int(param, BLOCK_OPT_SIZE, img_size);
2914f88e1a42SJes Sorensen 
2915f88e1a42SJes Sorensen     /* Parse -o options */
2916f88e1a42SJes Sorensen     if (options) {
2917f88e1a42SJes Sorensen         param = parse_option_parameters(options, create_options, param);
2918f88e1a42SJes Sorensen         if (param == NULL) {
2919f88e1a42SJes Sorensen             error_report("Invalid options for file format '%s'.", fmt);
29204f70f249SJes Sorensen             ret = -EINVAL;
2921f88e1a42SJes Sorensen             goto out;
2922f88e1a42SJes Sorensen         }
2923f88e1a42SJes Sorensen     }
2924f88e1a42SJes Sorensen 
2925f88e1a42SJes Sorensen     if (base_filename) {
2926f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FILE,
2927f88e1a42SJes Sorensen                                  base_filename)) {
2928f88e1a42SJes Sorensen             error_report("Backing file not supported for file format '%s'",
2929f88e1a42SJes Sorensen                          fmt);
29304f70f249SJes Sorensen             ret = -EINVAL;
2931f88e1a42SJes Sorensen             goto out;
2932f88e1a42SJes Sorensen         }
2933f88e1a42SJes Sorensen     }
2934f88e1a42SJes Sorensen 
2935f88e1a42SJes Sorensen     if (base_fmt) {
2936f88e1a42SJes Sorensen         if (set_option_parameter(param, BLOCK_OPT_BACKING_FMT, base_fmt)) {
2937f88e1a42SJes Sorensen             error_report("Backing file format not supported for file "
2938f88e1a42SJes Sorensen                          "format '%s'", fmt);
29394f70f249SJes Sorensen             ret = -EINVAL;
2940f88e1a42SJes Sorensen             goto out;
2941f88e1a42SJes Sorensen         }
2942f88e1a42SJes Sorensen     }
2943f88e1a42SJes Sorensen 
2944792da93aSJes Sorensen     backing_file = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
2945792da93aSJes Sorensen     if (backing_file && backing_file->value.s) {
2946792da93aSJes Sorensen         if (!strcmp(filename, backing_file->value.s)) {
2947792da93aSJes Sorensen             error_report("Error: Trying to create an image with the "
2948792da93aSJes Sorensen                          "same filename as the backing file");
29494f70f249SJes Sorensen             ret = -EINVAL;
2950792da93aSJes Sorensen             goto out;
2951792da93aSJes Sorensen         }
2952792da93aSJes Sorensen     }
2953792da93aSJes Sorensen 
2954f88e1a42SJes Sorensen     backing_fmt = get_option_parameter(param, BLOCK_OPT_BACKING_FMT);
2955f88e1a42SJes Sorensen     if (backing_fmt && backing_fmt->value.s) {
295696df67d1SStefan Hajnoczi         backing_drv = bdrv_find_format(backing_fmt->value.s);
295796df67d1SStefan Hajnoczi         if (!backing_drv) {
2958f88e1a42SJes Sorensen             error_report("Unknown backing file format '%s'",
2959f88e1a42SJes Sorensen                          backing_fmt->value.s);
29604f70f249SJes Sorensen             ret = -EINVAL;
2961f88e1a42SJes Sorensen             goto out;
2962f88e1a42SJes Sorensen         }
2963f88e1a42SJes Sorensen     }
2964f88e1a42SJes Sorensen 
2965f88e1a42SJes Sorensen     // The size for the image must always be specified, with one exception:
2966f88e1a42SJes Sorensen     // If we are using a backing file, we can obtain the size from there
2967*d220894eSKevin Wolf     size = get_option_parameter(param, BLOCK_OPT_SIZE);
2968*d220894eSKevin Wolf     if (size && size->value.n == -1) {
2969f88e1a42SJes Sorensen         if (backing_file && backing_file->value.s) {
2970f88e1a42SJes Sorensen             uint64_t size;
2971f88e1a42SJes Sorensen             char buf[32];
2972f88e1a42SJes Sorensen 
2973f88e1a42SJes Sorensen             bs = bdrv_new("");
2974f88e1a42SJes Sorensen 
297596df67d1SStefan Hajnoczi             ret = bdrv_open(bs, backing_file->value.s, flags, backing_drv);
2976f88e1a42SJes Sorensen             if (ret < 0) {
297796df67d1SStefan Hajnoczi                 error_report("Could not open '%s'", backing_file->value.s);
2978f88e1a42SJes Sorensen                 goto out;
2979f88e1a42SJes Sorensen             }
2980f88e1a42SJes Sorensen             bdrv_get_geometry(bs, &size);
2981f88e1a42SJes Sorensen             size *= 512;
2982f88e1a42SJes Sorensen 
2983f88e1a42SJes Sorensen             snprintf(buf, sizeof(buf), "%" PRId64, size);
2984f88e1a42SJes Sorensen             set_option_parameter(param, BLOCK_OPT_SIZE, buf);
2985f88e1a42SJes Sorensen         } else {
2986f88e1a42SJes Sorensen             error_report("Image creation needs a size parameter");
29874f70f249SJes Sorensen             ret = -EINVAL;
2988f88e1a42SJes Sorensen             goto out;
2989f88e1a42SJes Sorensen         }
2990f88e1a42SJes Sorensen     }
2991f88e1a42SJes Sorensen 
2992f88e1a42SJes Sorensen     printf("Formatting '%s', fmt=%s ", filename, fmt);
2993f88e1a42SJes Sorensen     print_option_parameters(param);
2994f88e1a42SJes Sorensen     puts("");
2995f88e1a42SJes Sorensen 
2996f88e1a42SJes Sorensen     ret = bdrv_create(drv, filename, param);
2997f88e1a42SJes Sorensen 
2998f88e1a42SJes Sorensen     if (ret < 0) {
2999f88e1a42SJes Sorensen         if (ret == -ENOTSUP) {
3000f88e1a42SJes Sorensen             error_report("Formatting or formatting option not supported for "
3001f88e1a42SJes Sorensen                          "file format '%s'", fmt);
3002f88e1a42SJes Sorensen         } else if (ret == -EFBIG) {
3003f88e1a42SJes Sorensen             error_report("The image size is too large for file format '%s'",
3004f88e1a42SJes Sorensen                          fmt);
3005f88e1a42SJes Sorensen         } else {
3006f88e1a42SJes Sorensen             error_report("%s: error while creating %s: %s", filename, fmt,
3007f88e1a42SJes Sorensen                          strerror(-ret));
3008f88e1a42SJes Sorensen         }
3009f88e1a42SJes Sorensen     }
3010f88e1a42SJes Sorensen 
3011f88e1a42SJes Sorensen out:
3012f88e1a42SJes Sorensen     free_option_parameters(create_options);
3013f88e1a42SJes Sorensen     free_option_parameters(param);
3014f88e1a42SJes Sorensen 
3015f88e1a42SJes Sorensen     if (bs) {
3016f88e1a42SJes Sorensen         bdrv_delete(bs);
3017f88e1a42SJes Sorensen     }
30184f70f249SJes Sorensen 
30194f70f249SJes Sorensen     return ret;
3020f88e1a42SJes Sorensen }
3021